cognee 0.2.1.dev7__py3-none-any.whl → 0.2.2.dev0__py3-none-any.whl

cognee/modules/settings/save_vector_db_config.py

@@ -6,7 +6,7 @@ from cognee.infrastructure.databases.vector import get_vectordb_config
 class VectorDBConfig(BaseModel):
     url: str
     api_key: str
-    provider: Union[Literal["lancedb"], Literal["qdrant"], Literal["weaviate"], Literal["pgvector"]]
+    provider: Union[Literal["lancedb"], Literal["qdrant"], Literal["pgvector"]]
 
 
 async def save_vector_db_config(vector_db_config: VectorDBConfig):

cognee/tests/unit/modules/retrieval/graph_completion_retriever_test.py

@@ -28,18 +28,38 @@ class TestGraphCompletionRetriever:
 
         class Company(DataPoint):
             name: str
+            description: str
 
         class Person(DataPoint):
             name: str
+            description: str
             works_for: Company
 
-        company1 = Company(name="Figma")
-        company2 = Company(name="Canva")
-        person1 = Person(name="Steve Rodger", works_for=company1)
-        person2 = Person(name="Ike Loma", works_for=company1)
-        person3 = Person(name="Jason Statham", works_for=company1)
-        person4 = Person(name="Mike Broski", works_for=company2)
-        person5 = Person(name="Christina Mayer", works_for=company2)
+        company1 = Company(name="Figma", description="Figma is a company")
+        company2 = Company(name="Canva", description="Canvas is a company")
+        person1 = Person(
+            name="Steve Rodger",
+            description="This is description about Steve Rodger",
+            works_for=company1,
+        )
+        person2 = Person(
+            name="Ike Loma", description="This is description about Ike Loma", works_for=company1
+        )
+        person3 = Person(
+            name="Jason Statham",
+            description="This is description about Jason Statham",
+            works_for=company1,
+        )
+        person4 = Person(
+            name="Mike Broski",
+            description="This is description about Mike Broski",
+            works_for=company2,
+        )
+        person5 = Person(
+            name="Christina Mayer",
+            description="This is description about Christina Mayer",
+            works_for=company2,
+        )
 
         entities = [company1, company2, person1, person2, person3, person4, person5]
 
@@ -49,8 +69,63 @@ class TestGraphCompletionRetriever:
 
         context = await retriever.get_context("Who works at Canva?")
 
-        assert "Mike Broski --[works_for]--> Canva" in context, "Failed to get Mike Broski"
-        assert "Christina Mayer --[works_for]--> Canva" in context, "Failed to get Christina Mayer"
+        # Ensure the top-level sections are present
+        assert "Nodes:" in context, "Missing 'Nodes:' section in context"
+        assert "Connections:" in context, "Missing 'Connections:' section in context"
+
+        # --- Nodes headers ---
+        assert "Node: Steve Rodger" in context, "Missing node header for Steve Rodger"
+        assert "Node: Figma" in context, "Missing node header for Figma"
+        assert "Node: Ike Loma" in context, "Missing node header for Ike Loma"
+        assert "Node: Jason Statham" in context, "Missing node header for Jason Statham"
+        assert "Node: Mike Broski" in context, "Missing node header for Mike Broski"
+        assert "Node: Canva" in context, "Missing node header for Canva"
+        assert "Node: Christina Mayer" in context, "Missing node header for Christina Mayer"
+
+        # --- Node contents ---
+        assert (
+            "__node_content_start__\nThis is description about Steve Rodger\n__node_content_end__"
+            in context
+        ), "Description block for Steve Rodger altered"
+        assert "__node_content_start__\nFigma is a company\n__node_content_end__" in context, (
+            "Description block for Figma altered"
+        )
+        assert (
+            "__node_content_start__\nThis is description about Ike Loma\n__node_content_end__"
+            in context
+        ), "Description block for Ike Loma altered"
+        assert (
+            "__node_content_start__\nThis is description about Jason Statham\n__node_content_end__"
+            in context
+        ), "Description block for Jason Statham altered"
+        assert (
+            "__node_content_start__\nThis is description about Mike Broski\n__node_content_end__"
+            in context
+        ), "Description block for Mike Broski altered"
+        assert "__node_content_start__\nCanvas is a company\n__node_content_end__" in context, (
+            "Description block for Canva altered"
+        )
+        assert (
+            "__node_content_start__\nThis is description about Christina Mayer\n__node_content_end__"
+            in context
+        ), "Description block for Christina Mayer altered"
+
+        # --- Connections ---
+        assert "Steve Rodger --[works_for]--> Figma" in context, (
+            "Connection Steve Rodger→Figma missing or changed"
+        )
+        assert "Ike Loma --[works_for]--> Figma" in context, (
+            "Connection Ike Loma→Figma missing or changed"
+        )
+        assert "Jason Statham --[works_for]--> Figma" in context, (
+            "Connection Jason Statham→Figma missing or changed"
+        )
+        assert "Mike Broski --[works_for]--> Canva" in context, (
+            "Connection Mike Broski→Canva missing or changed"
+        )
+        assert "Christina Mayer --[works_for]--> Canva" in context, (
+            "Connection Christina Mayer→Canva missing or changed"
+        )
 
     @pytest.mark.asyncio
     async def test_graph_completion_context_complex(self):

{cognee-0.2.1.dev7.dist-info → cognee-0.2.2.dev0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cognee
-Version: 0.2.1.dev7
+Version: 0.2.2.dev0
 Summary: Cognee - is a library for enriching LLM context with a semantic layer for better understanding and reasoning.
 Project-URL: Homepage, https://www.cognee.ai
 Project-URL: Repository, https://github.com/topoteretes/cognee
@@ -134,8 +134,6 @@ Provides-Extra: posthog
 Requires-Dist: posthog<4,>=3.5.0; extra == 'posthog'
 Provides-Extra: qdrant
 Requires-Dist: qdrant-client<2,>=1.14.2; extra == 'qdrant'
-Provides-Extra: weaviate
-Requires-Dist: weaviate-client<5.0.0,>=4.9.6; extra == 'weaviate'
 Description-Content-Type: text/markdown
 
 <div align="center">
@@ -156,7 +154,7 @@ Description-Content-Type: text/markdown
   ·
   <a href="https://www.reddit.com/r/AIMemory/">Join r/AIMemory</a>
   .
-  <a href="https://www.docs.cognee.ai">Docs</a>
+  <a href="https://docs.cognee.ai/">Docs</a>
   .
   <a href="https://github.com/topoteretes/cognee-community">cognee community repo</a>
   </p>
@@ -183,7 +181,7 @@ Description-Content-Type: text/markdown
 
 
 
-**🚀 We are launching Cognee SaaS: Sign up [here](https://www.cognee.ai/waitlist) for the hosted beta!**
+**🚀 We launched Cogwit beta (Fully-hosted AI Memory): Sign up [here](https://platform.cognee.ai/)! 🚀** 
 
 Build dynamic memory for Agents and replace RAG using scalable, modular ECL (Extract, Cognify, Load) pipelines.
 
@@ -325,9 +323,9 @@ Try cognee UI out locally [here](https://docs.cognee.ai/how-to-guides/cognee-ui)
 
 ## Demos
 
-1. What is AI memory:
+1. Cogwit Beta demo:
 
-[Learn about cognee](https://github.com/user-attachments/assets/8b2a0050-5ec4-424c-b417-8269971503f0)
+[Cogwit Beta](https://github.com/user-attachments/assets/fa520cd2-2913-4246-a444-902ea5242cb0)
 
 2. Simple GraphRAG demo
 

{cognee-0.2.1.dev7.dist-info → cognee-0.2.2.dev0.dist-info}/RECORD

@@ -9,7 +9,7 @@ cognee/version.py,sha256=isN9gXpAwkYR82cmhx-2ani8KylouByqM6iULr85Hyk,874
 cognee/api/.env.example,sha256=T3e9QDX8feK8cGBQUaRqORg1Y-1e-EFpByrvqehJqC4,265
 cognee/api/DTO.py,sha256=D-IN7PpNI6ypf7fhLif1wrsA-OV12th9B-1iTyu63qM,357
 cognee/api/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-cognee/api/client.py,sha256=gUhSLqfB4tNIXvwdeIM-La32ufetI5NMwWqf4h10vkI,7230
+cognee/api/client.py,sha256=rbzxvaCNn7d4P8kcFqPC_8SMTSC15NWow3XYBVENtG8,7268
 cognee/api/v1/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cognee/api/v1/add/__init__.py,sha256=JOyEOWtj-L8D_QtNDNCQMdf-Y8TXby4Plvco-5esbmo,21
 cognee/api/v1/add/add.py,sha256=r8FpH3qkLhr7Tx82vlut7tG2NvuqPu2RLV1okHIJYhY,6969
@@ -26,7 +26,7 @@ cognee/api/v1/config/config.py,sha256=cHnSh1_8RFTzpHF45nBFR-1wCBZk1wRvfyEgNsW7Wy
 cognee/api/v1/datasets/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cognee/api/v1/datasets/datasets.py,sha256=GVPjzE_CMb2arIg0qn9aXpbN2DUQFPzIBq1a-uIbTdc,1325
 cognee/api/v1/datasets/routers/__init__.py,sha256=F-hptvX-CC9cUHoKJVIFSI5hZ_wPjaN-rpvdA4rXWTk,53
-cognee/api/v1/datasets/routers/get_datasets_router.py,sha256=mylM2wCLGbOLiTAW-jDok2lIlF4Kj-F1DbfXbPF5ODg,16757
+cognee/api/v1/datasets/routers/get_datasets_router.py,sha256=B_5d7tMo8je6dw2tuhTB6DSR4sZGL7il2jnZ_FyJWSA,16767
 cognee/api/v1/delete/__init__.py,sha256=ZtgOK5grmkjGh7I5SlEYUxF7beiGUjEf0ZEYcdzpCJ8,27
 cognee/api/v1/delete/delete.py,sha256=n67mpiTAdwJytsitlg5O7r7bzkYqbRFt8n2KkCuG590,9987
 cognee/api/v1/delete/exceptions.py,sha256=wF9oNxAFJ8NCMxVV4FFO2biA9u0UXjFI5QsvUNuUlLk,1499
@@ -124,13 +124,14 @@ cognee/infrastructure/databases/graph/graph_db_interface.py,sha256=nHQweHqpOpWM6
 cognee/infrastructure/databases/graph/supported_databases.py,sha256=0UIYcQ15p7-rq5y_2A-E9ydcXyP6frdg8T5e5ECDDMI,25
 cognee/infrastructure/databases/graph/use_graph_adapter.py,sha256=a_T2NIhSw-cxn909ejiAYcMCFBh13j79TpaZJhokWxc,173
 cognee/infrastructure/databases/graph/kuzu/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-cognee/infrastructure/databases/graph/kuzu/adapter.py,sha256=hws_7bn9bRXbivaTwUz9BJ1wz9CuLSiMdnb6J0M0DrQ,60183
+cognee/infrastructure/databases/graph/kuzu/adapter.py,sha256=PEKvi4Jr7hbcdKCkfnpd-NuPHBvxNO7GE-qr2h-JMT0,61126
+cognee/infrastructure/databases/graph/kuzu/kuzu_migrate.py,sha256=prJxwlZCWJ6w6upVO2KrsfHu2UGb5BwrvSCyU_rkLEQ,10536
 cognee/infrastructure/databases/graph/kuzu/remote_kuzu_adapter.py,sha256=j0GOZl2yTvB_9JbulPec9gfQLoiTH-t932-uW3Ufr_0,7324
 cognee/infrastructure/databases/graph/kuzu/show_remote_kuzu_stats.py,sha256=l5TQUORnoCusIrPNbTuNDzVvUUAMjQNak-9I8KT7N3I,1044
 cognee/infrastructure/databases/graph/memgraph/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cognee/infrastructure/databases/graph/memgraph/memgraph_adapter.py,sha256=eoOpDEn6lw6rVXxj00Ek3lkLjxLDLtTmQbjezMYf850,34376
 cognee/infrastructure/databases/graph/neo4j_driver/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-cognee/infrastructure/databases/graph/neo4j_driver/adapter.py,sha256=FlHzdHRTPVBCxMF-3Ki9OB86XakcTEXeivha-Q_7XI4,39038
+cognee/infrastructure/databases/graph/neo4j_driver/adapter.py,sha256=ZXR9X9ytlWIJMZbTFN41Wjr4cliK5ncEzMLWFHAEEbs,40448
 cognee/infrastructure/databases/graph/neo4j_driver/deadlock_retry.py,sha256=A4e_8YMZ2TUQ5WxtGvaBHpSDOFKTmO0zGCTX9gnKSrM,2103
 cognee/infrastructure/databases/graph/neo4j_driver/neo4j_metrics_utils.py,sha256=1Q2domzVvCy6c1dgnS6HmeyIMlwiFcatLmvcA8xvvr8,5940
 cognee/infrastructure/databases/graph/networkx/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -143,13 +144,13 @@ cognee/infrastructure/databases/relational/create_db_and_tables.py,sha256=sldlFg
 cognee/infrastructure/databases/relational/create_relational_engine.py,sha256=qEAlSftp641CPmn-ud5N1NHW9EL5FxkUjYMXik_g4JM,1533
 cognee/infrastructure/databases/relational/get_migration_relational_engine.py,sha256=5RtH281iIQo3vqgwmKT0nuiJp9jNd7vw6xRUjc5xIDM,1070
 cognee/infrastructure/databases/relational/get_relational_engine.py,sha256=De51ieg9eFhRLX08k9oNc-oszvt_9J5DHebqI1qI8_U,741
-cognee/infrastructure/databases/relational/sqlalchemy/SqlAlchemyAdapter.py,sha256=rL2w3rEzHls_avgt0WSDQub31QaZXC2zAlAaJSqdP8k,27188
+cognee/infrastructure/databases/relational/sqlalchemy/SqlAlchemyAdapter.py,sha256=Xy43nHzxmINcePdaOrAw7msvPYaJV9-C_AFqOQxjh24,27373
 cognee/infrastructure/databases/relational/sqlalchemy/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cognee/infrastructure/databases/utils/__init__.py,sha256=4C0ncZG-O6bOFJpKgscCHu6D5vodLWRIKpe-WT4Ijbs,75
 cognee/infrastructure/databases/utils/get_or_create_dataset_database.py,sha256=wn7pRgeX-BU0L191_6pgT9P54uhVQlGMPqxQdvIlv4Y,2101
 cognee/infrastructure/databases/vector/__init__.py,sha256=7MdGJ3Mxdh2RyDq39rcjD99liIa-yGXxDUzq--1qQZs,291
 cognee/infrastructure/databases/vector/config.py,sha256=xKHkUAs8pohFtWC1-5GMFIKYr4jOfKZtIZYTG5qHr70,2598
-cognee/infrastructure/databases/vector/create_vector_engine.py,sha256=Zh9mBGva83nw5TbOUucVn_9lIo7SIjCKnAvynbw5PLo,4405
+cognee/infrastructure/databases/vector/create_vector_engine.py,sha256=zWedSFdmFz9ZwWwGnaCLtskmjhxfHjug1pNRl5dYKvw,4078
 cognee/infrastructure/databases/vector/get_vector_engine.py,sha256=y4TMWJ6B6DxwKF9PMfjB6WqujPnVhf0oR2j35Q-KhvA,272
 cognee/infrastructure/databases/vector/supported_databases.py,sha256=0UIYcQ15p7-rq5y_2A-E9ydcXyP6frdg8T5e5ECDDMI,25
 cognee/infrastructure/databases/vector/use_vector_adapter.py,sha256=ab2x6-sxVDu_tf4zWChN_ngqv8LaLYk2VCtBjZEyjaM,174
@@ -318,7 +319,7 @@ cognee/modules/data/methods/get_datasets.py,sha256=EZyDPGzZaZL2yC8yuWDz0HFgUCptf
 cognee/modules/data/methods/get_datasets_by_name.py,sha256=PT8QWKBiqfmwx2AtDxtaq-sWCJJOjJWI_7teZosv6XQ,731
 cognee/modules/data/methods/get_unique_dataset_id.py,sha256=ngWFcPpMIpEZbIUNQCICeOpM5o5Xy2i2z8WISlQrAlE,333
 cognee/modules/data/methods/load_or_create_datasets.py,sha256=4Ka7pAQtzsG8UgtOrcJvwe-zZ0B4NvhaukkBJS8rH8s,2195
-cognee/modules/data/models/Data.py,sha256=wX1UE9vlhFI87KKGLK0oxRkXTphRerjLRSdIMnt7UUA,1720
+cognee/modules/data/models/Data.py,sha256=nN3Yaym5XPPdMl5dkf7by6ZJZSpGS4-PWkLl-YOa8D4,1736
 cognee/modules/data/models/Dataset.py,sha256=PKBeZ6yAW3kidbgGcA5aLGW7-fEj5JVgUwKL5DJcHoo,1302
 cognee/modules/data/models/DatasetData.py,sha256=wIUqZGXLznRUObz6nl3wuW_OD2hiyBZX2H93lhaD4eI,498
 cognee/modules/data/models/GraphMetrics.py,sha256=_GCQBYfHbPmfH154AKJPJNQUSG0SOwI_Db-SPfqjIus,1123
@@ -337,7 +338,7 @@ cognee/modules/data/processing/document_types/Document.py,sha256=v94JqsQmKDqog2J
 cognee/modules/data/processing/document_types/ImageDocument.py,sha256=6wgsRLYwL-RLWKNdo9n65kegsvBZkRFtjlptgtxJqKg,721
 cognee/modules/data/processing/document_types/PdfDocument.py,sha256=MQ6qx2C5chtD6KDBhqV70IKbeCAWBsrb9UrkoM8Vdco,1189
 cognee/modules/data/processing/document_types/TextDocument.py,sha256=lANyFq-BzYDQcd_LNNj5xUCuQREUiIGL3dCv_83VW60,778
-cognee/modules/data/processing/document_types/UnstructuredDocument.py,sha256=VanNEqLo6cDyrubtIT4PNA4oaeOK2LgJ8u0qgrBYIjo,1438
+cognee/modules/data/processing/document_types/UnstructuredDocument.py,sha256=Et-_of-Xm2oihEkYtnAD5YaNmTTPW9V94t7ZxvqGp4M,1263
 cognee/modules/data/processing/document_types/__init__.py,sha256=US1CmchkTIbWgzRFIwniSTVI_p_cmHuyDaxQwjTZmhI,244
 cognee/modules/data/processing/document_types/exceptions/__init__.py,sha256=NA4yw1l-TVr06nzaK9vpBAECRIHrp_d0BAGAhLtCcjc,164
 cognee/modules/data/processing/document_types/exceptions/exceptions.py,sha256=pE24qj7DpxmlouKQHjdj7dqykzj1wlaAsXyltOva-lA,429
@@ -355,13 +356,13 @@ cognee/modules/engine/utils/generate_node_id.py,sha256=sHDyJVL9qFFoSXYsrlsE2VC7Y
 cognee/modules/engine/utils/generate_node_name.py,sha256=UMYi0h2Vx--cj8Ppji13cq0D_71JtvPYUDVUUPQKCkw,83
 cognee/modules/graph/relationship_manager.py,sha256=J4AA45FvF5z7EtrhGGrLa_-rW_xZ7-J52awOohzv1jA,1635
 cognee/modules/graph/cognee_graph/CogneeAbstractGraph.py,sha256=rZM8fv4BPlrBdVKedMsqED53tCtQDzypVUryM1xkf7E,1093
-cognee/modules/graph/cognee_graph/CogneeGraph.py,sha256=46PG-zV42I8RHvVrA7equsCaCkQrYPZnBJ34P-GbZxI,7283
+cognee/modules/graph/cognee_graph/CogneeGraph.py,sha256=uSgRwvL6XXUt_JNTFjAQ9MBWBCxwKytbZvuKia9bTGE,7932
 cognee/modules/graph/cognee_graph/CogneeGraphElements.py,sha256=tdwA8Z7RyTmlmdNG3A3Vnwi5YM14QEu18nHYmmbROns,5517
 cognee/modules/graph/cognee_graph/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cognee/modules/graph/exceptions/__init__.py,sha256=WUh8QcjerEyiBFItLvPfG_YaP5mgG9NFZeROROjSMbM,205
 cognee/modules/graph/exceptions/exceptions.py,sha256=TJbE8XJDsFDvAzR7jYJRLbS3nmZTFCAAuBJJ4tPcg2s,770
 cognee/modules/graph/methods/__init__.py,sha256=jVNRdhln1xkJQbl4a-8DuBALcV0l0dCk0zXUbShBFKc,63
-cognee/modules/graph/methods/get_formatted_graph_data.py,sha256=6U473M-HFFbUATY-HvdgOUDBF2womrmhW_1HSTUDixA,1383
+cognee/modules/graph/methods/get_formatted_graph_data.py,sha256=ww3RvPS99LXeIQDYqgVAo0VQ_mL4WHGE7aaknK_MnWo,1388
 cognee/modules/graph/models/EdgeType.py,sha256=Q1gvrrAfVL5RHU6YmRlHMZ7wNx3zB3EYwiDf7CFjIHk,194
 cognee/modules/graph/utils/__init__.py,sha256=8a8de1VYmEfUuNTfViaOIVV4kBIBdXYS5_USbOqKJyg,394
 cognee/modules/graph/utils/convert_node_to_data_point.py,sha256=5EFBNDL2xZlWIZDskzB8fnmsNGy7b6k5SOdZRvLvKLM,622
@@ -411,9 +412,9 @@ cognee/modules/pipelines/operations/log_pipeline_run_initiated.py,sha256=5Gz7T1Y
 cognee/modules/pipelines/operations/log_pipeline_run_start.py,sha256=0mxQJNwIEjlFd2xYDFo87SxMrOY0C2r-o6Ss2AHm2Zw,1203
 cognee/modules/pipelines/operations/pipeline.py,sha256=bDTW_HSMR8aXHvC5gXAdzfJMU1u8eRprEBDNfekACB4,7258
 cognee/modules/pipelines/operations/run_parallel.py,sha256=FtSBWv3-FKoVf2slsISQsBEW6yBroXzdNlnvmBOqNA0,479
-cognee/modules/pipelines/operations/run_tasks.py,sha256=I_Me-rofYmh95gJ7zCW5V8vLFaHh2kktfExEm17zAms,3927
+cognee/modules/pipelines/operations/run_tasks.py,sha256=GV8r3sh_2V7RnbsdbvtaJAngSEfZmnqVxMjIHlEAWYs,3933
 cognee/modules/pipelines/operations/run_tasks_base.py,sha256=zfOabXKhKyoFlZ-kqcGLuqELBt15OsyVhkgWhqvGn6w,2619
-cognee/modules/pipelines/operations/run_tasks_distributed.py,sha256=tAPZA3lxLpxXAdLPUSzr3bsH2-ttq-G9p5SeGqCaYdA,2862
+cognee/modules/pipelines/operations/run_tasks_distributed.py,sha256=E8z3a4a-DaH84jLA2rCi1FV18cAl-UN6BIESmlKM6X8,2868
 cognee/modules/pipelines/operations/run_tasks_with_telemetry.py,sha256=S2RnZSH0fGhXt4giYMrYKyuYNOGfQ2iJUF63AyMAuAc,1674
 cognee/modules/pipelines/queues/pipeline_run_info_queues.py,sha256=Ud-gjKuvfaVK2QXSOR4rASG6DT-13z_XHW3-9IOdRNI,985
 cognee/modules/pipelines/tasks/task.py,sha256=VIdABgACBM8hIZ3gxyUDKZlZveRKREuBuqIUemOdTb0,3332
@@ -423,17 +424,17 @@ cognee/modules/pipelines/utils/generate_pipeline_run_id.py,sha256=uWe8vzD4pcZWCK
 cognee/modules/retrieval/EntityCompletionRetriever.py,sha256=AUi0hTaYLE6dZbuOwVj-HNSGukCCbvXA8GuBnmUp1_E,3977
 cognee/modules/retrieval/__init__.py,sha256=skqAG7z2GDGZ6mKs9Kaxev69i5v5vgFNfmrFj3eLKm0,66
 cognee/modules/retrieval/base_retriever.py,sha256=nuzdpWQ3OmWQ0Y0rW8eMLFgbT8zMrNj3Yo-bRwyykWE,489
-cognee/modules/retrieval/chunks_retriever.py,sha256=wasL8rboYbWjmh_Wujnrm78QdVXvxG_HPa-tld_x0FU,2622
-cognee/modules/retrieval/code_retriever.py,sha256=OgC_6RoAJN_vA74UG5HqKbDwBNICu_n3TOQbrJRGFOA,5874
-cognee/modules/retrieval/completion_retriever.py,sha256=aOSJDu1SGE04RVsq4p1q2jHT3Ti2F1s8Xvfg3SqjXw8,3396
+cognee/modules/retrieval/chunks_retriever.py,sha256=ff9VGIEaaedO-p237PdmZ6jPJwgHzS0CihvDingsNiI,3548
+cognee/modules/retrieval/code_retriever.py,sha256=khn5flX_05EWcUHmmyAs4nz0xrzG1I0QvGGwWSGn2xM,8889
+cognee/modules/retrieval/completion_retriever.py,sha256=plfB63jEJW7o96jxw0dYRYY0inHehmjgHigwMFw8Jt8,3533
 cognee/modules/retrieval/cypher_search_retriever.py,sha256=qzXIrUp3edeKnYQqd2nJCRndCJX9hCle2t78a-A28w0,2817
 cognee/modules/retrieval/graph_completion_context_extension_retriever.py,sha256=fqZo7AI2AR_OsfGjZl_ewAU17es_AgsJVQKKcbuvLTk,4067
 cognee/modules/retrieval/graph_completion_cot_retriever.py,sha256=Y_G44FcVOICvR3k1FD-5emkSDUDkbOBElukuBT3pKKE,5473
-cognee/modules/retrieval/graph_completion_retriever.py,sha256=IYTEutwZ7ZlQFKFDCABNlCFdrAbhxkGjbhFRE8tHkZ0,7158
+cognee/modules/retrieval/graph_completion_retriever.py,sha256=K6Gm2Q6t0eYhU6cWloAfIiMHYwVJfkAu7K-r1lewUrM,7184
 cognee/modules/retrieval/graph_summary_completion_retriever.py,sha256=oOwZGChdjah-MqKg0ZQRfHYk1y0_8x7Xakh472qlFE0,2263
-cognee/modules/retrieval/insights_retriever.py,sha256=EiVKCiel6ypSoXxZt07qgDgalpINrZaL-RcABTtxr6Y,4243
-cognee/modules/retrieval/natural_language_retriever.py,sha256=-XjICp9P9AfE5JGmpSk97ZySCYGhduCHt4e88wCJWFA,6227
-cognee/modules/retrieval/summaries_retriever.py,sha256=vHZhvIwHOnXXPHgvYNDVBJ_pYIuhXAvYf4yMTqnvSAA,2429
+cognee/modules/retrieval/insights_retriever.py,sha256=2qSJuMC-lsbCR-OYU4FQfDT_-UI-lX4wR7qtGRcfwhI,4397
+cognee/modules/retrieval/natural_language_retriever.py,sha256=MsUrOs7inzsoNT_dNzLp_jp2958Ks330hLJWbnB4AjI,6084
+cognee/modules/retrieval/summaries_retriever.py,sha256=UgO6v6zpHqhFrEWLcsFr12zYZisiUWMyS5jiwp6zEak,3374
 cognee/modules/retrieval/context_providers/DummyContextProvider.py,sha256=9GsvINc7ekRyRWO5IefFGyytRYqsSlhpwAOw6Q691cA,419
 cognee/modules/retrieval/context_providers/SummarizedTripletSearchContextProvider.py,sha256=ypO6yWLxvmRsj_5dyYdvXTbztJmB_ioLrgyG6bF5WGA,894
 cognee/modules/retrieval/context_providers/TripletSearchContextProvider.py,sha256=m_ndKKPrPUtuWRphwc43n0CXWtgwu90NjNOvvQOPcjs,3761
@@ -443,7 +444,7 @@ cognee/modules/retrieval/entity_extractors/__init__.py,sha256=47DEQpj8HBSa-_TImW
 cognee/modules/retrieval/exceptions/__init__.py,sha256=9yC54Z5HmoDnti9_yFLXK9_l3aHAlCTDfPGMMTN7WfM,187
 cognee/modules/retrieval/exceptions/exceptions.py,sha256=ILB3aMg3F8yI6XRxs21tFasySXNRxMeASJAjrk7eIVg,1168
 cognee/modules/retrieval/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-cognee/modules/retrieval/utils/brute_force_triplet_search.py,sha256=pXHpcvNrCNIC3CNNXJJUbxetBJW4GRGjV-r2Q2lq0x4,6924
+cognee/modules/retrieval/utils/brute_force_triplet_search.py,sha256=B1Yrnc6vRjZ1OSKYDrXkM2J0UwdyQdkXJ-pwZnI9Yss,7651
 cognee/modules/retrieval/utils/completion.py,sha256=wAc0M3CGnwuw28u5q_eaoH2SJmF7tAdXvLEwodv5IQA,1121
 cognee/modules/retrieval/utils/description_to_codepart_search.py,sha256=Mjgdn_QR0vTJlc72vK5daCysvekAaxCKEvWE97I_dF4,6381
 cognee/modules/retrieval/utils/stop_words.py,sha256=HP8l2leoLf6u7tnWHrYhgVMY_TX6yetGIae8DTsTEH4,883
@@ -461,9 +462,9 @@ cognee/modules/search/types/SearchType.py,sha256=KQTL9P6LOlbjB1z_LT1tovmQ4TC98dP
 cognee/modules/search/types/__init__.py,sha256=tOM_-qzqR4_4V5YZPXB_g9AUj9pobGMmCdNDRIpCPNs,35
 cognee/modules/settings/__init__.py,sha256=_SZQgCQnnnIHLJuKOMO9uWzXNBQxwYHHMUSBp0qa2uQ,210
 cognee/modules/settings/get_current_settings.py,sha256=R2lOusG5Q2PMa2-2vDndh3Lm7nXyZVkdzTV7vQHT81Y,1642
-cognee/modules/settings/get_settings.py,sha256=-bPdLJEkgQuQuHY6E-nCRG4NWhr3A9P39SsFAN6Mwls,4393
+cognee/modules/settings/get_settings.py,sha256=692NgItiBjLeLdL57AJ0coSBrCKvEleNpZ8olU9eyvc,4306
 cognee/modules/settings/save_llm_config.py,sha256=fvvDJc_RGkqthrfD7pw7TNFuFc3-Y3QlJWpVl9OsVw8,504
-cognee/modules/settings/save_vector_db_config.py,sha256=5kbFnsQed7KsaA_R02QZQOgYGv6aXRZEWMQJA1BALx4,692
+cognee/modules/settings/save_vector_db_config.py,sha256=r3kmCNniYuraBH5sSP6jcsVP2-DB8QWSUzDIq7OgSog,671
 cognee/modules/storage/utils/__init__.py,sha256=PcUVyMCZZw5hf3GPxB-vq-FUo1BBaSWL7sTKmZsZnkM,1848
 cognee/modules/users/__init__.py,sha256=QRpoulUGvGvpoTvnPcVqmLavZEK1vSCOhoNomK6vwM4,80
 cognee/modules/users/get_fastapi_users.py,sha256=jm0nMEnzHuJoiCCO1FzgHHzilLRlIuontUr9Jn-f0sc,594
@@ -624,7 +625,6 @@ cognee/tests/test_s3_file_storage.py,sha256=62tvIFyh_uTP0TFF9Ck4Y-sxWPW-cwJKYEJU
 cognee/tests/test_search_db.py,sha256=e3J19yX8IgRI0DSlOp6o-TZjvSLmKn8Fbch8b2BolCA,8996
 cognee/tests/test_starter_pipelines.py,sha256=X1J8RDD0bFMKnRETyi5nyaF4TYdmUIu0EuD3WQwShNs,2475
 cognee/tests/test_telemetry.py,sha256=FIneuVofSKWFYqxNC88sT_P5GPzgfjVyqDCf2TYBE2E,4130
-cognee/tests/test_weaviate.py,sha256=1mR_4VABrjoJ-KJu0cyG1txJ1Szj861PkSUN-NNNsLY,6142
 cognee/tests/integration/documents/AudioDocument_test.py,sha256=0mJnlWRc7gWqOxAUfdSSIxntcUrzkPXhlsd-MFsiRoM,2790
 cognee/tests/integration/documents/ImageDocument_test.py,sha256=vrb3uti0RF6a336LLI95i8fso3hOFw9AFe1NxPnOf6k,2802
 cognee/tests/integration/documents/PdfDocument_test.py,sha256=27idYdl_eN6r92A4vUqLZdl9qrlf3pFfGcMhk7TYpsk,1803
@@ -679,7 +679,7 @@ cognee/tests/unit/modules/pipelines/run_tasks_with_context_test.py,sha256=Bi5XgQ
 cognee/tests/unit/modules/retrieval/chunks_retriever_test.py,sha256=qJcwBW65PNOfWpvxh77EFd1d73o0MGJ9-mbAmrMhUEI,5891
 cognee/tests/unit/modules/retrieval/graph_completion_retriever_context_extension_test.py,sha256=TBhRhQVZFJ7m-GjFuI7lP5F8oj04XPKvVt_cGSLWmlM,6748
 cognee/tests/unit/modules/retrieval/graph_completion_retriever_cot_test.py,sha256=RqptHU4PT4FNtcPBJi7Y1ww1NJb0jqvL2gHf8wRbylA,6563
-cognee/tests/unit/modules/retrieval/graph_completion_retriever_test.py,sha256=G5rM6jEsNJ-EZEXS1_Lj-LBpttrWHuEfcji2pCdhjRI,5571
+cognee/tests/unit/modules/retrieval/graph_completion_retriever_test.py,sha256=zTbGpe-4yZBuLXom0Ml1mDCOEUWBLQ1TdLUBTQrLfvQ,8867
 cognee/tests/unit/modules/retrieval/insights_retriever_test.py,sha256=xkbxlNiHY6evVbBYMncllXDNs3nNC_jZeYP47oT8vG0,8592
 cognee/tests/unit/modules/retrieval/rag_completion_retriever_test.py,sha256=3A5FvbN4-x8N7wccB9vbVVv9neLFRM6KkZjTkLu0MSo,6101
 cognee/tests/unit/modules/retrieval/summaries_retriever_test.py,sha256=IfhDyVuKUrjCEy22-Mva9w7li2mtPZT9FlNIFvpFMKw,4950
@@ -707,8 +707,8 @@ distributed/tasks/queued_add_edges.py,sha256=kz1DHE05y-kNHORQJjYWHUi6Q1QWUp_v3Dl
 distributed/tasks/queued_add_nodes.py,sha256=aqK4Ij--ADwUWknxYpiwbYrpa6CcvFfqHWbUZW4Kh3A,452
 distributed/workers/data_point_saving_worker.py,sha256=jFmA0-P_0Ru2IUDrSug0wML-5goAKrGtlBm5BA5Ryw4,3229
 distributed/workers/graph_saving_worker.py,sha256=oUYl99CdhlrPAIsUOHbHnS3d4XhGoV0_OIbCO8wYzRg,3648
-cognee-0.2.1.dev7.dist-info/METADATA,sha256=UpjBgmYmVpDoMcQBNk1Zquv6XWdJbDWbBUxFGPlV1qs,14536
-cognee-0.2.1.dev7.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-cognee-0.2.1.dev7.dist-info/licenses/LICENSE,sha256=pHHjSQj1DD8SDppW88MMs04TPk7eAanL1c5xj8NY7NQ,11344
-cognee-0.2.1.dev7.dist-info/licenses/NOTICE.md,sha256=6L3saP3kSpcingOxDh-SGjMS8GY79Rlh2dBNLaO0o5c,339
-cognee-0.2.1.dev7.dist-info/RECORD,,
+cognee-0.2.2.dev0.dist-info/METADATA,sha256=omEOYqvxhpAQfopA9SvfbqG7h_qRcTJOQInxBd6jLiI,14437
+cognee-0.2.2.dev0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+cognee-0.2.2.dev0.dist-info/licenses/LICENSE,sha256=pHHjSQj1DD8SDppW88MMs04TPk7eAanL1c5xj8NY7NQ,11344
+cognee-0.2.2.dev0.dist-info/licenses/NOTICE.md,sha256=6L3saP3kSpcingOxDh-SGjMS8GY79Rlh2dBNLaO0o5c,339
+cognee-0.2.2.dev0.dist-info/RECORD,,

cognee/tests/test_weaviate.py

@@ -1,94 +0,0 @@
-import os
-import pathlib
-import cognee
-from cognee.infrastructure.files.storage import get_storage_config
-from cognee.modules.search.operations import get_history
-from cognee.modules.users.methods import get_default_user
-from cognee.shared.logging_utils import get_logger
-from cognee.modules.search.types import SearchType
-
-logger = get_logger()
-
-
-async def main():
-    cognee.config.set_vector_db_provider("weaviate")
-    data_directory_path = str(
-        pathlib.Path(
-            os.path.join(pathlib.Path(__file__).parent, ".data_storage/test_weaviate")
-        ).resolve()
-    )
-    cognee.config.data_root_directory(data_directory_path)
-    cognee_directory_path = str(
-        pathlib.Path(
-            os.path.join(pathlib.Path(__file__).parent, ".cognee_system/test_weaviate")
-        ).resolve()
-    )
-    cognee.config.system_root_directory(cognee_directory_path)
-
-    await cognee.prune.prune_data()
-    await cognee.prune.prune_system(metadata=True)
-
-    dataset_name = "cs_explanations"
-
-    explanation_file_path = os.path.join(
-        pathlib.Path(__file__).parent, "test_data/Natural_language_processing.txt"
-    )
-    await cognee.add([explanation_file_path], dataset_name)
-
-    text = """A quantum computer is a computer that takes advantage of quantum mechanical phenomena.
-    At small scales, physical matter exhibits properties of both particles and waves, and quantum computing leverages this behavior, specifically quantum superposition and entanglement, using specialized hardware that supports the preparation and manipulation of quantum states.
-    Classical physics cannot explain the operation of these quantum devices, and a scalable quantum computer could perform some calculations exponentially faster (with respect to input size scaling) than any modern "classical" computer. In particular, a large-scale quantum computer could break widely used encryption schemes and aid physicists in performing physical simulations; however, the current state of the technology is largely experimental and impractical, with several obstacles to useful applications. Moreover, scalable quantum computers do not hold promise for many practical tasks, and for many important tasks quantum speedups are proven impossible.
-    The basic unit of information in quantum computing is the qubit, similar to the bit in traditional digital electronics. Unlike a classical bit, a qubit can exist in a superposition of its two "basis" states. When measuring a qubit, the result is a probabilistic output of a classical bit, therefore making quantum computers nondeterministic in general. If a quantum computer manipulates the qubit in a particular way, wave interference effects can amplify the desired measurement results. The design of quantum algorithms involves creating procedures that allow a quantum computer to perform calculations efficiently and quickly.
-    Physically engineering high-quality qubits has proven challenging. If a physical qubit is not sufficiently isolated from its environment, it suffers from quantum decoherence, introducing noise into calculations. Paradoxically, perfectly isolating qubits is also undesirable because quantum computations typically need to initialize qubits, perform controlled qubit interactions, and measure the resulting quantum states. Each of those operations introduces errors and suffers from noise, and such inaccuracies accumulate.
-    In principle, a non-quantum (classical) computer can solve the same computational problems as a quantum computer, given enough time. Quantum advantage comes in the form of time complexity rather than computability, and quantum complexity theory shows that some quantum algorithms for carefully selected tasks require exponentially fewer computational steps than the best known non-quantum algorithms. Such tasks can in theory be solved on a large-scale quantum computer whereas classical computers would not finish computations in any reasonable amount of time. However, quantum speedup is not universal or even typical across computational tasks, since basic tasks such as sorting are proven to not allow any asymptotic quantum speedup. Claims of quantum supremacy have drawn significant attention to the discipline, but are demonstrated on contrived tasks, while near-term practical use cases remain limited.
-    """
-
-    await cognee.add([text], dataset_name)
-
-    await cognee.cognify([dataset_name])
-
-    from cognee.infrastructure.databases.vector import get_vector_engine
-
-    vector_engine = get_vector_engine()
-    random_node = (await vector_engine.search("Entity_name", "Quantum computer"))[0]
-    random_node_name = random_node.payload["text"]
-
-    search_results = await cognee.search(
-        query_text=random_node_name, query_type=SearchType.INSIGHTS
-    )
-    assert len(search_results) != 0, "The search results list is empty."
-    print("\n\nExtracted sentences are:\n")
-    for result in search_results:
-        print(f"{result}\n")
-
-    search_results = await cognee.search(query_type=SearchType.CHUNKS, query_text=random_node_name)
-    assert len(search_results) != 0, "The search results list is empty."
-    print("\n\nExtracted chunks are:\n")
-    for result in search_results:
-        print(f"{result}\n")
-
-    search_results = await cognee.search(
-        query_type=SearchType.SUMMARIES, query_text=random_node_name
-    )
-    assert len(search_results) != 0, "Query related summaries don't exist."
-    print("\nExtracted summaries are:\n")
-    for result in search_results:
-        print(f"{result}\n")
-
-    user = await get_default_user()
-    history = await get_history(user.id)
-    assert len(history) == 6, "Search history is not correct."
-
-    await cognee.prune.prune_data()
-    data_root_directory = get_storage_config()["data_root_directory"]
-    assert not os.path.isdir(data_root_directory), "Local data files are not deleted"
-
-    await cognee.prune.prune_system(metadata=True)
-    collections = await get_vector_engine().client.collections.list_all()
-    assert len(collections) == 0, "Weaviate vector database is not empty"
-
-
-if __name__ == "__main__":
-    import asyncio
-
-    asyncio.run(main())