Merge branch 'main' into dependabot/pip/llm/langchain-core-0.1.35

api/chunks/migrations/0001_initial.py

@@ -1,4 +1,4 @@
-# Generated by Django 5.0.2 on 2024-03-25 19:36
+# Generated by Django 5.0.2 on 2024-03-29 09:07
 
 import django.db.models.deletion
 import pgvector.django

api/chunks/tests.py

@@ -7,7 +7,6 @@
 def create_document():
     return Document.objects.create(
         text="example",
-        embedding=list(range(1, 11)),
     )
 
 

api/documents/migrations/0001_initial.py

@@ -1,4 +1,4 @@
-# Generated by Django 5.0.2 on 2024-03-25 19:36
+# Generated by Django 5.0.2 on 2024-03-29 09:07
 
 import pgvector.django
 from django.db import migrations, models

api/documents/migrations/0002_remove_document_embedding.py

@@ -0,0 +1,16 @@
+# Generated by Django 5.0.2 on 2024-03-29 09:08
+
+from django.db import migrations
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("documents", "0001_initial"),
+    ]
+
+    operations = [
+        migrations.RemoveField(
+            model_name="document",
+            name="embedding",
+        ),
+    ]

api/documents/models.py

@@ -1,7 +1,5 @@
 from django.db import models
-from pgvector.django import VectorField
 
 
 class Document(models.Model):
     text = models.TextField[str, str]()
-    embedding = VectorField[list[float], VectorField](dimensions=10)

api/documents/schemas.py

@@ -3,10 +3,8 @@
 
 class DocumentIn(Schema):
     text: str
-    embedding: list[float]
 
 
 class DocumentOut(Schema):
     id: int
     text: str
-    embedding: list[float]

api/documents/tests.py

@@ -4,7 +4,7 @@
 
 
 def create_document():
-    return Document.objects.create(text="example", embedding=list(range(1, 11)))
+    return Document.objects.create(text="example")
 
 
 class DocumentModelTests(TestCase):
@@ -31,9 +31,9 @@ def test_put_method(self):
         ).exists(), "Can't create document object"
         document.text = "notexample"
         document.save()
-        assert all(
-            a == b for a, b in zip(document.embedding, range(1, 10))
-        ), "Can't update document object"
+        assert Document.objects.filter(
+            text="notexample"
+        ).exists(), "Can't update document object"
 
     @pytest.mark.django_db
     def test_delete_method(self):

api/pyproject.toml

@@ -7,7 +7,7 @@ readme = "README.md"
 
 [tool.poetry.dependencies]
 python = "^3.11"
-django = "^4.2.7"
+django = "^4.2.11"
 uvicorn = "^0.24.0.post1"
 channels = "^4.0.0"
 psycopg2-binary = "^2.9.9"