cognee 0.4.0__py3-none-any.whl → 0.4.1__py3-none-any.whl

cognee/infrastructure/databases/vector/embeddings/OllamaEmbeddingEngine.py

@@ -124,7 +124,10 @@ class OllamaEmbeddingEngine(EmbeddingEngine):
                 self.endpoint, json=payload, headers=headers, timeout=60.0
             ) as response:
                 data = await response.json()
-                return data["embeddings"][0]
+                if "embeddings" in data:
+                    return data["embeddings"][0]
+                else:
+                    return data["data"][0]["embedding"]
 
     def get_vector_size(self) -> int:
         """

cognee/tests/integration/web_url_crawler/test_default_url_crawler.py

@@ -5,7 +5,7 @@ from cognee.tasks.web_scraper import DefaultUrlCrawler
 @pytest.mark.asyncio
 async def test_fetch():
     crawler = DefaultUrlCrawler()
-    url = "https://en.wikipedia.org/wiki/Large_language_model"
+    url = "http://example.com/"
     results = await crawler.fetch_urls(url)
     assert len(results) == 1
     assert isinstance(results, dict)

cognee/tests/integration/web_url_crawler/test_tavily_crawler.py

@@ -11,7 +11,7 @@ skip_in_ci = pytest.mark.skipif(
 @skip_in_ci
 @pytest.mark.asyncio
 async def test_fetch():
-    url = "https://en.wikipedia.org/wiki/Large_language_model"
+    url = "http://example.com/"
     results = await fetch_with_tavily(url)
     assert isinstance(results, dict)
     assert len(results) == 1

cognee/tests/integration/web_url_crawler/test_url_adding_e2e.py

@@ -14,9 +14,7 @@ async def test_url_saves_as_html_file():
     await cognee.prune.prune_system(metadata=True)
 
     try:
-        original_file_path = await save_data_item_to_storage(
-            "https://en.wikipedia.org/wiki/Large_language_model"
-        )
+        original_file_path = await save_data_item_to_storage("http://example.com/")
         file_path = get_data_file_path(original_file_path)
         assert file_path.endswith(".html")
         file = Path(file_path)
@@ -44,9 +42,7 @@ async def test_saved_html_is_valid():
     await cognee.prune.prune_system(metadata=True)
 
     try:
-        original_file_path = await save_data_item_to_storage(
-            "https://en.wikipedia.org/wiki/Large_language_model"
-        )
+        original_file_path = await save_data_item_to_storage("http://example.com/")
         file_path = get_data_file_path(original_file_path)
         content = Path(file_path).read_text()
 
@@ -72,7 +68,7 @@ async def test_add_url():
     await cognee.prune.prune_data()
     await cognee.prune.prune_system(metadata=True)
 
-    await cognee.add("https://en.wikipedia.org/wiki/Large_language_model")
+    await cognee.add("http://example.com/")
 
 
 skip_in_ci = pytest.mark.skipif(
@@ -88,7 +84,7 @@ async def test_add_url_with_tavily():
     await cognee.prune.prune_data()
     await cognee.prune.prune_system(metadata=True)
 
-    await cognee.add("https://en.wikipedia.org/wiki/Large_language_model")
+    await cognee.add("http://example.com/")
 
 
 @pytest.mark.asyncio
@@ -98,7 +94,7 @@ async def test_add_url_without_incremental_loading():
 
     try:
         await cognee.add(
-            "https://en.wikipedia.org/wiki/Large_language_model",
+            "http://example.com/",
             incremental_loading=False,
         )
     except Exception as e:
@@ -112,7 +108,7 @@ async def test_add_url_with_incremental_loading():
 
     try:
         await cognee.add(
-            "https://en.wikipedia.org/wiki/Large_language_model",
+            "http://example.com/",
             incremental_loading=True,
         )
     except Exception as e:
@@ -125,7 +121,7 @@ async def test_add_url_can_define_preferred_loader_as_list_of_str():
     await cognee.prune.prune_system(metadata=True)
 
     await cognee.add(
-        "https://en.wikipedia.org/wiki/Large_language_model",
+        "http://example.com/",
         preferred_loaders=["beautiful_soup_loader"],
     )
 
@@ -144,7 +140,7 @@ async def test_add_url_with_extraction_rules():
 
     try:
         await cognee.add(
-            "https://en.wikipedia.org/wiki/Large_language_model",
+            "http://example.com/",
             preferred_loaders={"beautiful_soup_loader": {"extraction_rules": extraction_rules}},
         )
     except Exception as e:
@@ -163,9 +159,7 @@ async def test_loader_is_none_by_default():
     }
 
     try:
-        original_file_path = await save_data_item_to_storage(
-            "https://en.wikipedia.org/wiki/Large_language_model"
-        )
+        original_file_path = await save_data_item_to_storage("http://example.com/")
         file_path = get_data_file_path(original_file_path)
         assert file_path.endswith(".html")
         file = Path(file_path)
@@ -196,9 +190,7 @@ async def test_beautiful_soup_loader_is_selected_loader_if_preferred_loader_prov
     }
 
     try:
-        original_file_path = await save_data_item_to_storage(
-            "https://en.wikipedia.org/wiki/Large_language_model"
-        )
+        original_file_path = await save_data_item_to_storage("http://example.com/")
         file_path = get_data_file_path(original_file_path)
         assert file_path.endswith(".html")
         file = Path(file_path)
@@ -225,9 +217,7 @@ async def test_beautiful_soup_loader_works_with_and_without_arguments():
     await cognee.prune.prune_system(metadata=True)
 
     try:
-        original_file_path = await save_data_item_to_storage(
-            "https://en.wikipedia.org/wiki/Large_language_model"
-        )
+        original_file_path = await save_data_item_to_storage("http://example.com/")
         file_path = get_data_file_path(original_file_path)
         assert file_path.endswith(".html")
         file = Path(file_path)
@@ -263,9 +253,7 @@ async def test_beautiful_soup_loader_successfully_loads_file_if_required_args_pr
     await cognee.prune.prune_system(metadata=True)
 
     try:
-        original_file_path = await save_data_item_to_storage(
-            "https://en.wikipedia.org/wiki/Large_language_model"
-        )
+        original_file_path = await save_data_item_to_storage("http://example.com/")
         file_path = get_data_file_path(original_file_path)
         assert file_path.endswith(".html")
         file = Path(file_path)
@@ -302,9 +290,7 @@ async def test_beautiful_soup_loads_file_successfully():
     }
 
     try:
-        original_file_path = await save_data_item_to_storage(
-            "https://en.wikipedia.org/wiki/Large_language_model"
-        )
+        original_file_path = await save_data_item_to_storage("http://example.com/")
         file_path = get_data_file_path(original_file_path)
         assert file_path.endswith(".html")
         original_file = Path(file_path)

{cognee-0.4.0.dist-info → cognee-0.4.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cognee
-Version: 0.4.0
+Version: 0.4.1
 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
@@ -28,7 +28,8 @@ Requires-Dist: gunicorn<24,>=20.1.0
 Requires-Dist: instructor<2.0.0,>=1.9.1
 Requires-Dist: jinja2<4,>=3.1.3
 Requires-Dist: kuzu==0.11.3
-Requires-Dist: lancedb<1.0.0,>=0.24.0
+Requires-Dist: lance-namespace<=0.0.21
+Requires-Dist: lancedb<=0.25.3,>=0.24.0
 Requires-Dist: limits<5,>=4.4.1
 Requires-Dist: litellm>=1.76.0
 Requires-Dist: mistralai>=1.9.10
@@ -249,7 +250,7 @@ Let’s try Cognee in just a few lines of code. For detailed setup and configura
 
 ### Prerequisites
 
-- Python 3.10 to 3.12
+- Python 3.10 to 3.13
 
 ### Step 1: Install Cognee
 

{cognee-0.4.0.dist-info → cognee-0.4.1.dist-info}/RECORD

@@ -204,7 +204,7 @@ cognee/infrastructure/databases/vector/chromadb/__init__.py,sha256=47DEQpj8HBSa-
 cognee/infrastructure/databases/vector/embeddings/EmbeddingEngine.py,sha256=I-FXxTSRtb0y00U5eJr2o8n4j4DcC3_mEjEya70BPQU,1158
 cognee/infrastructure/databases/vector/embeddings/FastembedEmbeddingEngine.py,sha256=r1NTOo2aMwjbb9-yfHCU_IS-VZ9p3ZdRGRKWZmcIpso,4521
 cognee/infrastructure/databases/vector/embeddings/LiteLLMEmbeddingEngine.py,sha256=_rSMGNPjjfrV7Xr2xZWvs1RPRVF1nj-1nlBk0cGgh9A,8321
-cognee/infrastructure/databases/vector/embeddings/OllamaEmbeddingEngine.py,sha256=KDjo5qknAj761l0IQzbC0zUcl4pYIxF4AS3MkCVp1jc,4961
+cognee/infrastructure/databases/vector/embeddings/OllamaEmbeddingEngine.py,sha256=QGU9BpzlTJyWESvjUiFOuAOjkj2LCJTq9FOizPzIi00,5084
 cognee/infrastructure/databases/vector/embeddings/__init__.py,sha256=Akv-ShdXjHw-BE00Gw55GgGxIMr0SZ9FHi3RlpsJmiE,55
 cognee/infrastructure/databases/vector/embeddings/config.py,sha256=w7zaQEBNjnYXQi2N5gTCIooDzwGI3HCyyeWt-Q5WIKw,2539
 cognee/infrastructure/databases/vector/embeddings/embedding_rate_limiter.py,sha256=TyCoo_SipQ6JNy5eqXY2shrZnhb2JVjt9xOsJltOCdw,17598
@@ -849,9 +849,9 @@ cognee/tests/integration/documents/PdfDocument_test.py,sha256=IY0Cck8J2gEyuJHPK0
 cognee/tests/integration/documents/TextDocument_test.py,sha256=aSYfyvSQLceZ1c5NqV5Jf5eGA3BL_adP6iwWnT9eMCg,2159
 cognee/tests/integration/documents/UnstructuredDocument_test.py,sha256=nZktosptjw85V1_2iAwlOaYghA4cmqEX62RvQSgU_NY,4006
 cognee/tests/integration/documents/async_gen_zip.py,sha256=h98Q6cxhwb49iaYm4NZ-GmbNDAux-BKplofNgf4aIpc,317
-cognee/tests/integration/web_url_crawler/test_default_url_crawler.py,sha256=7R0I9BrUa4emt73B15r8UrBblYi7PUEFj5_8NxZUGFY,380
-cognee/tests/integration/web_url_crawler/test_tavily_crawler.py,sha256=6gWk6DGz7msY5FyQfhaXuybj0L8Jej96Dy29BqlOtbM,524
-cognee/tests/integration/web_url_crawler/test_url_adding_e2e.py,sha256=Uw2PSwuZ_-pUFp5HaxxVQZInsq1RxOpzgQLerh-6FX4,11684
+cognee/tests/integration/web_url_crawler/test_default_url_crawler.py,sha256=Qk__D7-SwpE5YfCiXoIDF3LgablRMhtoSSGfCVYY-PM,349
+cognee/tests/integration/web_url_crawler/test_tavily_crawler.py,sha256=tro2Isg-zqEEkD03oCWzYV8n5KlqGaUP69RcVVaeYDc,493
+cognee/tests/integration/web_url_crawler/test_url_adding_e2e.py,sha256=F1eTCI9Q9kVwOh5DuDf-uVxz3DqZvJj2HGlAiREIX_8,11127
 cognee/tests/subprocesses/reader.py,sha256=NW5zbXhWUcFXyN9RRAW2lzxCvEYV8hno6gBmE18O0b8,954
 cognee/tests/subprocesses/simple_cognify_1.py,sha256=WE2hG50rFwceKNL07PeAYu-Mrs74pjmdPEQrqZiTf8s,869
 cognee/tests/subprocesses/simple_cognify_2.py,sha256=nv0gJZCLn0iwY7SumiGlIiGJc1tFCyiHhAUzw0sjLn8,872
@@ -942,9 +942,9 @@ 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=kmaQy2A2J7W3k9Gd5lyoiT0XYOaJmEM8MbkKVOFOQVU,4729
 distributed/workers/graph_saving_worker.py,sha256=b5OPLLUq0OBALGekdp73JKxU0GrMlVbO4AfIhmACKkQ,4724
-cognee-0.4.0.dist-info/METADATA,sha256=0gKutXhioeXNoqHwCwj-RYDSf4ZUYXhu9xq4VrknNL0,15317
-cognee-0.4.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-cognee-0.4.0.dist-info/entry_points.txt,sha256=fAozOD9Vs4kgYwRhBiZoLCIXu-OSZqVxKGv45l19uok,88
-cognee-0.4.0.dist-info/licenses/LICENSE,sha256=pHHjSQj1DD8SDppW88MMs04TPk7eAanL1c5xj8NY7NQ,11344
-cognee-0.4.0.dist-info/licenses/NOTICE.md,sha256=6L3saP3kSpcingOxDh-SGjMS8GY79Rlh2dBNLaO0o5c,339
-cognee-0.4.0.dist-info/RECORD,,
+cognee-0.4.1.dist-info/METADATA,sha256=Xv_KcVeblOMxrXCaUpcHGbOnEXhwRHiVtidZw1cJHss,15358
+cognee-0.4.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+cognee-0.4.1.dist-info/entry_points.txt,sha256=fAozOD9Vs4kgYwRhBiZoLCIXu-OSZqVxKGv45l19uok,88
+cognee-0.4.1.dist-info/licenses/LICENSE,sha256=pHHjSQj1DD8SDppW88MMs04TPk7eAanL1c5xj8NY7NQ,11344
+cognee-0.4.1.dist-info/licenses/NOTICE.md,sha256=6L3saP3kSpcingOxDh-SGjMS8GY79Rlh2dBNLaO0o5c,339
+cognee-0.4.1.dist-info/RECORD,,

{cognee-0.4.0.dist-info → cognee-0.4.1.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: hatchling 1.27.0
+Generator: hatchling 1.28.0
 Root-Is-Purelib: true
 Tag: py3-none-any