cognee 0.2.2.dev1__py3-none-any.whl → 0.2.3.dev1__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (89) hide show
  1. cognee/api/v1/add/add.py +0 -5
  2. cognee/api/v1/cognify/cognify.py +0 -8
  3. cognee/api/v1/config/config.py +5 -13
  4. cognee/api/v1/datasets/routers/get_datasets_router.py +1 -1
  5. cognee/api/v1/delete/delete.py +1 -1
  6. cognee/api/v1/exceptions/__init__.py +13 -0
  7. cognee/api/v1/{delete → exceptions}/exceptions.py +15 -12
  8. cognee/api/v1/search/search.py +0 -7
  9. cognee/exceptions/__init__.py +5 -5
  10. cognee/exceptions/exceptions.py +37 -17
  11. cognee/infrastructure/data/exceptions/__init__.py +7 -0
  12. cognee/infrastructure/data/exceptions/exceptions.py +22 -0
  13. cognee/infrastructure/data/utils/extract_keywords.py +3 -3
  14. cognee/infrastructure/databases/exceptions/__init__.py +3 -0
  15. cognee/infrastructure/databases/exceptions/exceptions.py +57 -9
  16. cognee/infrastructure/databases/graph/neptune_driver/exceptions.py +15 -10
  17. cognee/infrastructure/databases/hybrid/falkordb/FalkorDBAdapter.py +2 -2
  18. cognee/infrastructure/databases/hybrid/neptune_analytics/NeptuneAnalyticsAdapter.py +4 -5
  19. cognee/infrastructure/databases/vector/chromadb/ChromaDBAdapter.py +2 -2
  20. cognee/infrastructure/databases/vector/embeddings/LiteLLMEmbeddingEngine.py +1 -1
  21. cognee/infrastructure/databases/vector/exceptions/exceptions.py +3 -3
  22. cognee/infrastructure/databases/vector/lancedb/LanceDBAdapter.py +2 -2
  23. cognee/infrastructure/databases/vector/pgvector/PGVectorAdapter.py +4 -3
  24. cognee/infrastructure/llm/exceptions.py +30 -2
  25. cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/anthropic/adapter.py +2 -2
  26. cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/gemini/adapter.py +3 -3
  27. cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/get_llm_client.py +10 -7
  28. cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/openai/adapter.py +6 -4
  29. cognee/infrastructure/loaders/LoaderInterface.py +1 -1
  30. cognee/modules/data/exceptions/exceptions.py +18 -5
  31. cognee/modules/data/methods/delete_data.py +2 -4
  32. cognee/modules/data/processing/document_types/exceptions/exceptions.py +2 -2
  33. cognee/modules/graph/cognee_graph/CogneeGraph.py +6 -4
  34. cognee/modules/graph/cognee_graph/CogneeGraphElements.py +5 -10
  35. cognee/modules/graph/exceptions/__init__.py +2 -0
  36. cognee/modules/graph/exceptions/exceptions.py +25 -3
  37. cognee/modules/ingestion/exceptions/exceptions.py +2 -2
  38. cognee/modules/ontology/exceptions/exceptions.py +4 -4
  39. cognee/modules/pipelines/exceptions/exceptions.py +2 -2
  40. cognee/modules/retrieval/exceptions/exceptions.py +12 -6
  41. cognee/modules/search/exceptions/__init__.py +7 -0
  42. cognee/modules/search/exceptions/exceptions.py +15 -0
  43. cognee/modules/search/methods/search.py +2 -3
  44. cognee/modules/users/exceptions/exceptions.py +6 -6
  45. cognee/shared/exceptions/exceptions.py +2 -2
  46. cognee/tasks/completion/exceptions/exceptions.py +3 -3
  47. cognee/tasks/documents/classify_documents.py +4 -0
  48. cognee/tasks/documents/exceptions/__init__.py +11 -0
  49. cognee/tasks/documents/exceptions/exceptions.py +36 -0
  50. cognee/tasks/documents/extract_chunks_from_documents.py +8 -2
  51. cognee/tasks/graph/exceptions/__init__.py +12 -0
  52. cognee/tasks/graph/exceptions/exceptions.py +41 -0
  53. cognee/tasks/graph/extract_graph_from_data.py +28 -0
  54. cognee/tasks/ingestion/exceptions/__init__.py +8 -0
  55. cognee/tasks/ingestion/exceptions/exceptions.py +12 -0
  56. cognee/tasks/ingestion/resolve_data_directories.py +5 -0
  57. cognee/tasks/storage/add_data_points.py +8 -0
  58. cognee/tasks/storage/exceptions/__init__.py +9 -0
  59. cognee/tasks/storage/exceptions/exceptions.py +13 -0
  60. cognee/tasks/storage/index_data_points.py +1 -1
  61. cognee/tasks/summarization/exceptions/__init__.py +9 -0
  62. cognee/tasks/summarization/exceptions/exceptions.py +14 -0
  63. cognee/tasks/summarization/summarize_text.py +8 -1
  64. cognee/tests/test_delete_by_id.py +1 -1
  65. cognee/tests/unit/modules/graph/cognee_graph_elements_test.py +5 -5
  66. cognee/tests/unit/modules/search/search_methods_test.py +2 -2
  67. {cognee-0.2.2.dev1.dist-info → cognee-0.2.3.dev1.dist-info}/METADATA +4 -4
  68. {cognee-0.2.2.dev1.dist-info → cognee-0.2.3.dev1.dist-info}/RECORD +88 -57
  69. distributed/Dockerfile +34 -0
  70. distributed/__init__.py +0 -0
  71. distributed/app.py +4 -0
  72. distributed/entrypoint.py +71 -0
  73. distributed/entrypoint.sh +5 -0
  74. distributed/modal_image.py +11 -0
  75. distributed/poetry.lock +12238 -0
  76. distributed/pyproject.toml +186 -0
  77. distributed/queues.py +5 -0
  78. distributed/tasks/__init__.py +0 -0
  79. distributed/tasks/queued_add_data_points.py +13 -0
  80. distributed/tasks/queued_add_edges.py +13 -0
  81. distributed/tasks/queued_add_nodes.py +13 -0
  82. distributed/test.py +28 -0
  83. distributed/utils.py +19 -0
  84. distributed/workers/data_point_saving_worker.py +93 -0
  85. distributed/workers/graph_saving_worker.py +104 -0
  86. cognee/infrastructure/databases/exceptions/EmbeddingException.py +0 -20
  87. {cognee-0.2.2.dev1.dist-info → cognee-0.2.3.dev1.dist-info}/WHEEL +0 -0
  88. {cognee-0.2.2.dev1.dist-info → cognee-0.2.3.dev1.dist-info}/licenses/LICENSE +0 -0
  89. {cognee-0.2.2.dev1.dist-info → cognee-0.2.3.dev1.dist-info}/licenses/NOTICE.md +0 -0
@@ -13,26 +13,27 @@ cognee/api/client.py,sha256=Cqaa4CgQuhtueALArap4JSP5O8OJ3TC95bwrtQe1OQA,8682
13
13
  cognee/api/health.py,sha256=tL-WzDW_GS2DeSFhkKRROtx-4x2vi1Kb-dp44n73uFw,12556
14
14
  cognee/api/v1/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
15
15
  cognee/api/v1/add/__init__.py,sha256=JOyEOWtj-L8D_QtNDNCQMdf-Y8TXby4Plvco-5esbmo,21
16
- cognee/api/v1/add/add.py,sha256=0Q88xeR--DBg9Puvnt24RPYifqq_M3tXl7tCCoM-zaw,7106
16
+ cognee/api/v1/add/add.py,sha256=OVP49XIdzutEe6YVG9xHBpyBc6tvN8sO9719CmQkXto,6832
17
17
  cognee/api/v1/add/routers/__init__.py,sha256=4c7wJoaUCQ7nqV_-BGYigevAL2mYORo6LFLciKtmVlU,43
18
18
  cognee/api/v1/add/routers/get_add_router.py,sha256=NVU-69_Hwx_H5UyLvbhlwLkkenaVds4l2Fawd0AAUdY,4500
19
19
  cognee/api/v1/cognify/__init__.py,sha256=EKVvqVlvd4MMx3dcVH3NPvxuQflcuJijo0WldrE8XqQ,29
20
20
  cognee/api/v1/cognify/code_graph_pipeline.py,sha256=vnxSEK6PNDEIPTDs3Fin3Wg7LaWctQdyc_Yikw4TBmU,3834
21
- cognee/api/v1/cognify/cognify.py,sha256=J2vpMnolI9nl-ueqkJ-onGTaj7VEyrj3LAlkY996e5k,12971
21
+ cognee/api/v1/cognify/cognify.py,sha256=fof7s75zGXlrP0BxZDI33GfXcyOc8MpK_9RvCxlSC40,12575
22
22
  cognee/api/v1/cognify/routers/__init__.py,sha256=wiPpOoQbSKje-SfbRQ7HPao0bn8FckRvIv0ipKW5Y90,114
23
23
  cognee/api/v1/cognify/routers/get_code_pipeline_router.py,sha256=uO5KzjXYYkZaxzCYxe8-zKYZwXZLUPsJ5sJY9h7dYtw,2974
24
24
  cognee/api/v1/cognify/routers/get_cognify_router.py,sha256=X6iYzpI8Jttdooy4XzHR3O0K_KXY9jslM1UFK0LMFts,7568
25
25
  cognee/api/v1/config/__init__.py,sha256=D_bzXVg_vtSAW6U7S3qUDSqHU1Lf-cFpVt2rXrpBdnc,27
26
- cognee/api/v1/config/config.py,sha256=i3bMS2KImo2t6gFTnfzYzY0IANMVfhBeed1prSxofVE,6831
26
+ cognee/api/v1/config/config.py,sha256=ckyX0euGNYNXK0tFq5b_OouE6x4VDKFG3uXgMDNUbfk,6522
27
27
  cognee/api/v1/datasets/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
28
28
  cognee/api/v1/datasets/datasets.py,sha256=GVPjzE_CMb2arIg0qn9aXpbN2DUQFPzIBq1a-uIbTdc,1325
29
29
  cognee/api/v1/datasets/routers/__init__.py,sha256=F-hptvX-CC9cUHoKJVIFSI5hZ_wPjaN-rpvdA4rXWTk,53
30
- cognee/api/v1/datasets/routers/get_datasets_router.py,sha256=nyTDoQA0b2QvTBs2hzw-sXp-zmPfPHYcSHUvToaeAPA,16530
30
+ cognee/api/v1/datasets/routers/get_datasets_router.py,sha256=V1_1H6aSAlD45isIaiM-qLpN7v6QmL4YfJHEu38hJOo,16523
31
31
  cognee/api/v1/delete/__init__.py,sha256=ZtgOK5grmkjGh7I5SlEYUxF7beiGUjEf0ZEYcdzpCJ8,27
32
- cognee/api/v1/delete/delete.py,sha256=R7C14XBLty6NXVzNN7Yubu1sUnvimV4GbJuS2fIPoog,9955
33
- cognee/api/v1/delete/exceptions.py,sha256=wF9oNxAFJ8NCMxVV4FFO2biA9u0UXjFI5QsvUNuUlLk,1499
32
+ cognee/api/v1/delete/delete.py,sha256=pzggh8cSQdxNo27VAnhJ4gnM8uU4Q-USCUfcjzO-wzE,9948
34
33
  cognee/api/v1/delete/routers/__init__.py,sha256=5UDk3LT37tyW9EnwYMzCIT2AW9zzHMvfLiDiVCrnKFk,49
35
34
  cognee/api/v1/delete/routers/get_delete_router.py,sha256=7o0wsxt0dfk_f800ilY5n5IexbnY_YPXt1QJKoVQiY8,1822
35
+ cognee/api/v1/exceptions/__init__.py,sha256=DHX5lYPEz3wNiZXhLSZqRSO2z1bX-EFQobmJ2OFSgmU,293
36
+ cognee/api/v1/exceptions/exceptions.py,sha256=tR4HyexthHLNfAWxaELallTezFl-26bkEytvrDFsanY,1677
36
37
  cognee/api/v1/permissions/routers/__init__.py,sha256=ljE3YnrzlMcVfThmkR5GSIxkm7sQVyibaLNtYQL4HO0,59
37
38
  cognee/api/v1/permissions/routers/get_permissions_router.py,sha256=tqd-J__UBlstTWnQocesdjVM9JnYO5rtJhhFj-Zv1_o,8316
38
39
  cognee/api/v1/prune/__init__.py,sha256=FEr5tTlX7wf3X4aFff6NPlVhNrPyqx7RBoJ71bJN1cY,25
@@ -45,7 +46,7 @@ cognee/api/v1/responses/routers/__init__.py,sha256=X2qishwGRVFXawnvkZ5bv420PuPRL
45
46
  cognee/api/v1/responses/routers/default_tools.py,sha256=9qqzEZhrt3_YMKzUA06ke8P-2WeLXhYpKgVW6mLHlzw,3004
46
47
  cognee/api/v1/responses/routers/get_responses_router.py,sha256=ggbLhY9IXaInCgIs5TUuOCkFW64xmTKZQsc2ENq2Ocs,5979
47
48
  cognee/api/v1/search/__init__.py,sha256=Sqw60DcOj4Bnvt-EWFknT31sPcvROIRKCWLr5pbkFr4,39
48
- cognee/api/v1/search/search.py,sha256=PziKJkIgpvReEmeAHemm35RudOYAWuBIuK1_o7lszZ4,8659
49
+ cognee/api/v1/search/search.py,sha256=ujdhyq7u6j2HbkPWhzGE3Iz4tUL0-WedRlHEB45QpwY,8172
49
50
  cognee/api/v1/search/routers/__init__.py,sha256=6RebeLX_2NTRxIMPH_mGuLztPxnGnMJK1y_O93CtRm8,49
50
51
  cognee/api/v1/search/routers/get_search_router.py,sha256=dLKA29Cpz-dfdtLxEPow1MSWIOjVEYR5niq2RvBpvYE,4892
51
52
  cognee/api/v1/settings/routers/__init__.py,sha256=wj_UYAXNMPCkn6Mo1YB01dCBiV9DQwTIf6OWjnGRpf8,53
@@ -99,8 +100,8 @@ cognee/eval_framework/evaluation/metrics/__init__.py,sha256=47DEQpj8HBSa-_TImW-5
99
100
  cognee/eval_framework/evaluation/metrics/context_coverage.py,sha256=uggEXMUbgZOD6_7V_d8kpjWRBBz1Cejfx8RFuV6Qyys,2028
100
101
  cognee/eval_framework/evaluation/metrics/exact_match.py,sha256=HYNb0i5iKyQoQxKJaO5FeYE5KEKqaAcreNFyWpj_0Ms,628
101
102
  cognee/eval_framework/evaluation/metrics/f1.py,sha256=uP4lZrfkASK-VhfCosiYiSAkjdW0IAjJdzNamfC3GDg,1652
102
- cognee/exceptions/__init__.py,sha256=2LmTY8bwENaYnu2XkCy9kSzsYCttM93NBbq0y9vG_GU,334
103
- cognee/exceptions/exceptions.py,sha256=-AUP0P3wrrsxW6nISRv7phMRFZS17bets8f-_Qz2IRE,2201
103
+ cognee/exceptions/__init__.py,sha256=KvLZYOil6IhnAvzXHz1VQArcESDMeMtPwfLnBRXBzLw,344
104
+ cognee/exceptions/exceptions.py,sha256=7_pH-37LHtC7OWsjX_qQXdvHjq05Px14PUckFLZeRls,2767
104
105
  cognee/infrastructure/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
105
106
  cognee/infrastructure/context/BaseContextProvider.py,sha256=1M_rFFKWXOQSJszXedGhVyIW5rzL7YqxBTAXPjZqOF4,1194
106
107
  cognee/infrastructure/context/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -112,12 +113,13 @@ cognee/infrastructure/data/chunking/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQ
112
113
  cognee/infrastructure/data/chunking/config.py,sha256=d1y0B1DpQJy61wIRhYumVtfkcibWI9NpBrHaRhlmiFQ,1664
113
114
  cognee/infrastructure/data/chunking/create_chunking_engine.py,sha256=MrLAGTCaSDQ3sP2FrUBUmp3M07T5ABvngHPmqmSd2NM,2475
114
115
  cognee/infrastructure/data/chunking/get_chunking_engine.py,sha256=CRQmvelqDZ5q3qDSIPUZnc25c_O72YIEnDEoyHyHn_Q,374
116
+ cognee/infrastructure/data/exceptions/__init__.py,sha256=okwA2jmaYhAHct3qsktxsQvz5L3CJSxFu7gipldxIjo,168
117
+ cognee/infrastructure/data/exceptions/exceptions.py,sha256=yelSoDCkRAlEY9SyCHODYR4aXQ2JLZgFQ9SAGmHj3JI,635
115
118
  cognee/infrastructure/data/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
116
- cognee/infrastructure/data/utils/extract_keywords.py,sha256=m7ZqhxePAcfSecOAtv68w7jZtUSD6V3U_8pivfTIkJw,1460
119
+ cognee/infrastructure/data/utils/extract_keywords.py,sha256=skL2noylmW8yiuNwoe7CILea2artb-_Vy2i6kbQtPBo,1439
117
120
  cognee/infrastructure/databases/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
118
- cognee/infrastructure/databases/exceptions/EmbeddingException.py,sha256=i_fiMVFETKZZHZBn5r0WOmOds7TxCiwmBaef3m2I6KU,681
119
- cognee/infrastructure/databases/exceptions/__init__.py,sha256=hLmSPEyQF-Jn_kJ_c_Li8dPzwS9oy9NJy-GAgU-AnbQ,237
120
- cognee/infrastructure/databases/exceptions/exceptions.py,sha256=P4M-VoC2as3V90kGznxeYEY9jluZeDXH9fXLBXcdLsY,2944
121
+ cognee/infrastructure/databases/exceptions/__init__.py,sha256=KB7jkhb_ikPJ9TelNE5G0ToSgVFWljCofREnk9uM5KE,336
122
+ cognee/infrastructure/databases/exceptions/exceptions.py,sha256=FPmEWeYE10Q5J9sT93SLRL8mIUkd2jHnoy-bZQ40keM,4663
121
123
  cognee/infrastructure/databases/graph/__init__.py,sha256=iw96qByJlPswPoV1Op-fLDG8_chM0r1fy_3-74m7BQ4,133
122
124
  cognee/infrastructure/databases/graph/config.py,sha256=gwCszzeTR2KFOSsCLkZZSBwEnWf8RQH0_mXnpUlA8ZY,5158
123
125
  cognee/infrastructure/databases/graph/get_graph_engine.py,sha256=c51tHFsqc2EUVgOp21DCQFWRH7ytzYZyjo4phOJZf_4,6887
@@ -136,12 +138,12 @@ cognee/infrastructure/databases/graph/neo4j_driver/deadlock_retry.py,sha256=A4e_
136
138
  cognee/infrastructure/databases/graph/neo4j_driver/neo4j_metrics_utils.py,sha256=1Q2domzVvCy6c1dgnS6HmeyIMlwiFcatLmvcA8xvvr8,5940
137
139
  cognee/infrastructure/databases/graph/neptune_driver/__init__.py,sha256=SBEqsn1oynfB5IkTb9VqyQVQpNinbxLUUgAUGmIa5_k,334
138
140
  cognee/infrastructure/databases/graph/neptune_driver/adapter.py,sha256=dsyBFnM7_Le0LmuOaexX_n1pZHEriDI_2mDSbcZYdvk,52723
139
- cognee/infrastructure/databases/graph/neptune_driver/exceptions.py,sha256=-Pg1nlRqXtYWUACX3KIKt6w_3lkJ6Nsl9CawSN_v33w,4065
141
+ cognee/infrastructure/databases/graph/neptune_driver/exceptions.py,sha256=n00tVChhQcV3XJ2nytIjyxez-3RkgDvR5gRPiM6aPAQ,4166
140
142
  cognee/infrastructure/databases/graph/neptune_driver/neptune_utils.py,sha256=n9iggmPO-pxFG36F2vITenvyye-PeMe1l5gKVbL6PHI,6367
141
143
  cognee/infrastructure/databases/graph/networkx/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
142
144
  cognee/infrastructure/databases/graph/networkx/adapter.py,sha256=XRKWjngt-9zwggizSayMfQIes-nTZQ73RsnSv1YVWZw,36549
143
- cognee/infrastructure/databases/hybrid/falkordb/FalkorDBAdapter.py,sha256=5jlwt0ES609r1Smo49N_SOnrL6_lSOJr2bXGAeNkYOs,41904
144
- cognee/infrastructure/databases/hybrid/neptune_analytics/NeptuneAnalyticsAdapter.py,sha256=oK3a6NW91G0-qAiGWkWqugukCMuVzaChUPzvwioxMGk,17638
145
+ cognee/infrastructure/databases/hybrid/falkordb/FalkorDBAdapter.py,sha256=C8S81Xz_uWlyWkpZlSE9sYlMizcE6efPwcc1O-L-QSQ,41886
146
+ cognee/infrastructure/databases/hybrid/neptune_analytics/NeptuneAnalyticsAdapter.py,sha256=Y-P1YoajimqIc0T0YoN3xjTbW3zEdlir4zsTAcnBE7A,17617
145
147
  cognee/infrastructure/databases/hybrid/neptune_analytics/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
146
148
  cognee/infrastructure/databases/relational/ModelBase.py,sha256=YOyt3zsuYBP73PcF2WCN6wgUBrWmgE2qUwsAhXKESes,304
147
149
  cognee/infrastructure/databases/relational/__init__.py,sha256=-0HhaGTyxMK0Fx8Sf3LgPAmfRWQomwRltIhSdFE-6hc,300
@@ -162,26 +164,26 @@ cognee/infrastructure/databases/vector/supported_databases.py,sha256=0UIYcQ15p7-
162
164
  cognee/infrastructure/databases/vector/use_vector_adapter.py,sha256=ab2x6-sxVDu_tf4zWChN_ngqv8LaLYk2VCtBjZEyjaM,174
163
165
  cognee/infrastructure/databases/vector/utils.py,sha256=WHPSMFsN2XK72uURvCl_jlzQa-N3XKPhrDnB6GKmBtM,1224
164
166
  cognee/infrastructure/databases/vector/vector_db_interface.py,sha256=EUpRVyMyS0MOQwFEgxwRa_9MY1vYotCyO6CONM81r94,7118
165
- cognee/infrastructure/databases/vector/chromadb/ChromaDBAdapter.py,sha256=uBjGzJhZ8Jhf6pBiqLys8eTJFbBJW91_DVea4IyBg2c,18967
167
+ cognee/infrastructure/databases/vector/chromadb/ChromaDBAdapter.py,sha256=oWNPHp6qyiIJOmsqpTFkBbZPSFmwn4eK-pvExNvgIY0,18949
166
168
  cognee/infrastructure/databases/vector/chromadb/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
167
169
  cognee/infrastructure/databases/vector/embeddings/EmbeddingEngine.py,sha256=boNJ55dxJQ_ImW1_DDjToQa0Hos9mkeRYwfCI7UPLn0,983
168
170
  cognee/infrastructure/databases/vector/embeddings/FastembedEmbeddingEngine.py,sha256=-iXdNnLL5xgCrjA7XVJaWEhrjFme7qJQQoqM9zTPmZQ,3765
169
- cognee/infrastructure/databases/vector/embeddings/LiteLLMEmbeddingEngine.py,sha256=20Hds-jt6WshA2eY1hptr2R3h5w0u8c318eowFHSalU,7271
171
+ cognee/infrastructure/databases/vector/embeddings/LiteLLMEmbeddingEngine.py,sha256=No5mqIw-e6Qe67r8TIFu6XLPYo5DMQnP9vnzswI04HQ,7252
170
172
  cognee/infrastructure/databases/vector/embeddings/OllamaEmbeddingEngine.py,sha256=QFLvFe_asd8PwzMV3tgyACnzymIDiPFaMeLUmGyhcXE,4113
171
173
  cognee/infrastructure/databases/vector/embeddings/__init__.py,sha256=Akv-ShdXjHw-BE00Gw55GgGxIMr0SZ9FHi3RlpsJmiE,55
172
174
  cognee/infrastructure/databases/vector/embeddings/config.py,sha256=shM9paWuDfnlIPYawd1inPQfU0tYcP0qWa1zZQqItvA,2188
173
175
  cognee/infrastructure/databases/vector/embeddings/embedding_rate_limiter.py,sha256=HuMWHaw1WQyuq7PQX3QiYKam16htOPJIgk4rSHGLMs4,17702
174
176
  cognee/infrastructure/databases/vector/embeddings/get_embedding_engine.py,sha256=H41H6tev_Z5891oq1vKSqLIe6U_2xLKjl70iE-xjZh0,3832
175
177
  cognee/infrastructure/databases/vector/exceptions/__init__.py,sha256=zVTMxoY3oCF_FuZ1IR2tfOu1iWKo7a3Eyvq3cEzNXm8,48
176
- cognee/infrastructure/databases/vector/exceptions/exceptions.py,sha256=PGf4gxATW0z_AaOQjgNYaSZF0GAJcGW5Hlvu_gjN1FM,744
177
- cognee/infrastructure/databases/vector/lancedb/LanceDBAdapter.py,sha256=uz9GC5uaJADdUvtjMC9m9x4SaYV1XD_kqrGOAuGH9O8,12951
178
+ cognee/infrastructure/databases/vector/exceptions/exceptions.py,sha256=-LljxC5Um4pDdsGlZEAfIugpOo4ZdMtoLsJ_5LjQqw8,768
179
+ cognee/infrastructure/databases/vector/lancedb/LanceDBAdapter.py,sha256=0dDeHXpg_Pjay0zx-w8cuf9d1CEBdvEJp-uTxzGhnWM,12933
178
180
  cognee/infrastructure/databases/vector/lancedb/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
179
181
  cognee/infrastructure/databases/vector/models/CollectionConfig.py,sha256=cKhdEqD8wgY0WRz1Nrc-zPi7xmd4ZNB5xaTrvJGe-CY,598
180
182
  cognee/infrastructure/databases/vector/models/PayloadSchema.py,sha256=stwZWlZxotNcdkmg0BfdkiLZHtjrZ2T06FU4L2aZ5kI,69
181
183
  cognee/infrastructure/databases/vector/models/ScoredResult.py,sha256=7BfCp1m9JUlod1dIpi09cRAiQos5MZXE8sBYo0PXNWU,619
182
184
  cognee/infrastructure/databases/vector/models/VectorConfig.py,sha256=oJ1oiRcW2Jxq7T01URJPlcexhZ0PT-PsVwbk_057aUI,323
183
185
  cognee/infrastructure/databases/vector/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
184
- cognee/infrastructure/databases/vector/pgvector/PGVectorAdapter.py,sha256=TMoBnHEBa9Jhq9wtITVnnCnGGWS2nLiPi4-16hVeP1M,15344
186
+ cognee/infrastructure/databases/vector/pgvector/PGVectorAdapter.py,sha256=zBWonA3cmGtNwX1rWosReDPmY2HyIKl2lmGZ_U7eBsQ,15309
185
187
  cognee/infrastructure/databases/vector/pgvector/__init__.py,sha256=BZmSlUWky0Vp9_4vvjeQX13YldCzfmSPLq4tvN6U6Ds,55
186
188
  cognee/infrastructure/databases/vector/pgvector/create_db_and_tables.py,sha256=j-HrXr5mSB3S1NPfoe_9T0cxtji8xTKnMgT2pujkers,517
187
189
  cognee/infrastructure/databases/vector/pgvector/serialize_data.py,sha256=Or69QQkCwBSVSrjfeB7APJWprvwnAtxJ2bmxUNkODNs,1111
@@ -216,7 +218,7 @@ cognee/infrastructure/files/utils/open_data_file.py,sha256=CO654MXgbDXQFAJzrKXxS
216
218
  cognee/infrastructure/llm/LLMGateway.py,sha256=s6DHvltGr3wRSTGLXp3kF61UYfDPZNy_xgp-mpdr_nE,5438
217
219
  cognee/infrastructure/llm/__init__.py,sha256=-nEQSe4WmjCfj-M63QGg7DRpaHA4GjcIH5Ox1zgeZLE,356
218
220
  cognee/infrastructure/llm/config.py,sha256=lOTsN913buH1njLnatMQ1_MGqn5GcJkZj8VBHczPopU,7276
219
- cognee/infrastructure/llm/exceptions.py,sha256=yFMVCtsMoBOMdDosLv3Tku5NKh6W2JNTpG_KjLju_RE,113
221
+ cognee/infrastructure/llm/exceptions.py,sha256=1EDvHVC6Gw3oAKZp9yfeW8HUvV8Rt0JlJxRr2af6LE8,976
220
222
  cognee/infrastructure/llm/utils.py,sha256=UObLsHCUqZPv3-5PxUSZ6cog0FC484NjaajsUiExOK4,3806
221
223
  cognee/infrastructure/llm/prompts/__init__.py,sha256=vkBlEYbi73tIMtDDs8lhi9BQoC_30Uo5OBsFlcMFw5Q,90
222
224
  cognee/infrastructure/llm/prompts/answer_hotpot_question.txt,sha256=Je7BpKqW-hceHqB7aCy0wFHhdAquNksnkMxwL2hpb8g,219
@@ -290,19 +292,19 @@ cognee/infrastructure/llm/structured_output_framework/litellm_instructor/extract
290
292
  cognee/infrastructure/llm/structured_output_framework/litellm_instructor/extraction/knowledge_graph/__init__.py,sha256=sIVB6UrWo0rdSBpo-O4jzUWs8bBBar4kcSMrTBAeHSw,57
291
293
  cognee/infrastructure/llm/structured_output_framework/litellm_instructor/extraction/knowledge_graph/extract_content_graph.py,sha256=fO6uErXag5tpNWz5D33xtWyzodGIZDjltCKe9t72x1U,924
292
294
  cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
293
- cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/get_llm_client.py,sha256=PeRrErjJwPmevpNJu3I1PTrlDOIOuSF5TZff1U__4W8,4889
295
+ cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/get_llm_client.py,sha256=dbrDKdYEMvXxWGOYZkXx38AAOSFrt3VuTeLmRSoOY5k,4807
294
296
  cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/llm_interface.py,sha256=126jfQhTEAbmsVsc4wyf20dK-C2AFJQ0sVmNPZFEet0,2194
295
297
  cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/rate_limiter.py,sha256=ie_zMYnUzMcW4okP4P41mEC31EML2ztdU7bEQQdg99U,16763
296
298
  cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/anthropic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
297
- cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/anthropic/adapter.py,sha256=J-rQvR40ySrbi0raAM1pu04HpUTGVAbVd80DsyGt0yU,3107
299
+ cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/anthropic/adapter.py,sha256=tWUbSLibKkOQBDKjB6mnDDyX8WhZDHGzzaYxJjG35gc,3107
298
300
  cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/gemini/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
299
- cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/gemini/adapter.py,sha256=NXXYh9965j5cI4zHtvDAEeAvfiCYvtvxr3ixXXu4vNM,5025
301
+ cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/gemini/adapter.py,sha256=cfLZ5g1TPyFQWSQ7hVD4tjI_X8JJCmhK8LO3jD3DLTg,5034
300
302
  cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/generic_llm_api/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
301
303
  cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/generic_llm_api/adapter.py,sha256=iwh1NcOSlyRwrUfn_CX7AWuTWhOlekF3Xtx3cQ45Qls,5457
302
304
  cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/ollama/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
303
305
  cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/ollama/adapter.py,sha256=Hv1mlonsYY-5QXFiuOU45aTc-_GO-H_8qu1F-gpS5-s,5457
304
306
  cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/openai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
305
- cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/openai/adapter.py,sha256=J2ErwIp5yFbjNSWIl5qQGegqjJyWAF2-WQo1MtSruzw,11131
307
+ cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/openai/adapter.py,sha256=jkfzYS8erAtDZJa2a-k4vv3O1_AtQ_KZzgSteBwH8-0,11107
306
308
  cognee/infrastructure/llm/tokenizer/__init__.py,sha256=PfvDIZITALjM6CXtUEQ3NyZYxt0QYgHjqXDQYoRKB7o,212
307
309
  cognee/infrastructure/llm/tokenizer/tokenizer_interface.py,sha256=CdpGUFWJwEnj0A2HrD2i3hfFze-hLixUHf-g7p57HWU,1359
308
310
  cognee/infrastructure/llm/tokenizer/Gemini/__init__.py,sha256=x2WAZ-pdVL1UtdpgEQtz8TOjGTXA4OatzSfY7lF0oEo,37
@@ -314,7 +316,7 @@ cognee/infrastructure/llm/tokenizer/Mistral/adapter.py,sha256=ZaAuCl9ka-ANzykzzH
314
316
  cognee/infrastructure/llm/tokenizer/TikToken/__init__.py,sha256=m1P0VJKiWAKiftLRLLDfEVSuufIxy-T24yMjVm4FsgY,39
315
317
  cognee/infrastructure/llm/tokenizer/TikToken/adapter.py,sha256=IDFWGQe5Z8PmhDu0bRoQLreYqicUtzGCEp-_9F4_amE,3581
316
318
  cognee/infrastructure/loaders/LoaderEngine.py,sha256=brNSioTh6Ya6yMWAnVhqX5G93iixSbyM16cUvbt5z5k,5064
317
- cognee/infrastructure/loaders/LoaderInterface.py,sha256=PKkmQRQ9RHyVyL8mwg1BCe5pcRl8JQYgayQLUJimlIM,1967
319
+ cognee/infrastructure/loaders/LoaderInterface.py,sha256=sJjcHAiEkzraBfKcgj9NoLi3WZnQqnf1H2qXewJwoxw,1932
318
320
  cognee/infrastructure/loaders/__init__.py,sha256=onz7sA9r6p5Vq28IsD12eOqF0JixUvB3WDBO4t8hY0g,643
319
321
  cognee/infrastructure/loaders/create_loader_engine.py,sha256=mgo1RlLJ4E3exdEE03CDsrpdeMoucwrOql-1hZ8Ko00,964
320
322
  cognee/infrastructure/loaders/get_loader_engine.py,sha256=cPJefAHFAWU1HXQoWqCpwh8vRtoz1poPSw6wKyECg8k,517
@@ -346,12 +348,12 @@ cognee/modules/data/deletion/__init__.py,sha256=bEOcuz1MPPd7Cqqwd7B4B22Xa81oYdUO
346
348
  cognee/modules/data/deletion/prune_data.py,sha256=Aus7o3PDJPEaTh2u0yCHkT92rEesS0NRu12vwBE2D_s,278
347
349
  cognee/modules/data/deletion/prune_system.py,sha256=zeLynZssM4LWHNMJyOH99wxOn-cRpqN6Aa1PdnOzH98,601
348
350
  cognee/modules/data/exceptions/__init__.py,sha256=8eKBra_q0CcaShB6jca-EM1gLTMRXxc9iWAJCyoFqXs,266
349
- cognee/modules/data/exceptions/exceptions.py,sha256=q7OHVS3n88sN4uwLZs9bcOx5QGRD2vjI9D_tdlehR7U,1284
351
+ cognee/modules/data/exceptions/exceptions.py,sha256=fYdHFXwOGQAyTcJGbvrtZjTGb4ipnwh2FK5CPHxiZlE,1720
350
352
  cognee/modules/data/methods/__init__.py,sha256=5M_c7ce_6tF0x5B0dVL_EsNsD_4j_wk7T6vW9uof7Gk,729
351
353
  cognee/modules/data/methods/add_model_class_to_graph.py,sha256=8FbPcrBZz2nInb95-5lgc1i4Ptn__Z8Bvzu0J4YxGj4,2601
352
354
  cognee/modules/data/methods/check_dataset_name.py,sha256=aP-207gdZS3SGWKK0MAxkMTNk1XVsXTuUq9bcU7RedA,172
353
355
  cognee/modules/data/methods/create_dataset.py,sha256=S1FXs4cTCQ9P6_loo_--iPfr4IPK7PPlUxDqgIeXpFU,1084
354
- cognee/modules/data/methods/delete_data.py,sha256=TlsaPCMR3YCafGL5IAtotea3WnL4AJ5jaPZZAZC0QDc,677
356
+ cognee/modules/data/methods/delete_data.py,sha256=0fPHBt22oe_PMyBRtfRftx9dhG6Hbz-6p5UqFb60h2k,604
355
357
  cognee/modules/data/methods/delete_dataset.py,sha256=SOv8tNABGsJex10B6LUtMuMhyNMDF0ToZNk4vX5uYQU,293
356
358
  cognee/modules/data/methods/get_authorized_dataset.py,sha256=Cyp5bwhR4qVagoOrw345Dbk0JQ_yhDyjPkRU1ExbSQQ,746
357
359
  cognee/modules/data/methods/get_authorized_existing_datasets.py,sha256=BlvEDCiSfCbgz79WH-N5I8jV-06K-tWn23mGETaQvio,1442
@@ -385,7 +387,7 @@ cognee/modules/data/processing/document_types/TextDocument.py,sha256=lANyFq-BzYD
385
387
  cognee/modules/data/processing/document_types/UnstructuredDocument.py,sha256=Et-_of-Xm2oihEkYtnAD5YaNmTTPW9V94t7ZxvqGp4M,1263
386
388
  cognee/modules/data/processing/document_types/__init__.py,sha256=US1CmchkTIbWgzRFIwniSTVI_p_cmHuyDaxQwjTZmhI,244
387
389
  cognee/modules/data/processing/document_types/exceptions/__init__.py,sha256=NA4yw1l-TVr06nzaK9vpBAECRIHrp_d0BAGAhLtCcjc,164
388
- cognee/modules/data/processing/document_types/exceptions/exceptions.py,sha256=pE24qj7DpxmlouKQHjdj7dqykzj1wlaAsXyltOva-lA,429
390
+ cognee/modules/data/processing/document_types/exceptions/exceptions.py,sha256=ssW__awPcmYfr7nSLDdfP20LhkPEJv3hDCA-I-3yUsw,435
389
391
  cognee/modules/engine/models/ColumnValue.py,sha256=jHd1M2vkvW39iyTSf-ahbB0A-7zxfXWFQcFz-gW0aHc,193
390
392
  cognee/modules/engine/models/Entity.py,sha256=R7Fklijl-1bXHCGmhzuDNS_KMD7SVNHE3-Aqi3IHC5c,291
391
393
  cognee/modules/engine/models/EntityType.py,sha256=eSxP03PTBeWUMEvNFSP93n6kM2Hb5H2A30rQ4QTfiZc,456
@@ -401,11 +403,11 @@ cognee/modules/engine/utils/generate_node_id.py,sha256=sHDyJVL9qFFoSXYsrlsE2VC7Y
401
403
  cognee/modules/engine/utils/generate_node_name.py,sha256=UMYi0h2Vx--cj8Ppji13cq0D_71JtvPYUDVUUPQKCkw,83
402
404
  cognee/modules/graph/relationship_manager.py,sha256=J4AA45FvF5z7EtrhGGrLa_-rW_xZ7-J52awOohzv1jA,1635
403
405
  cognee/modules/graph/cognee_graph/CogneeAbstractGraph.py,sha256=rZM8fv4BPlrBdVKedMsqED53tCtQDzypVUryM1xkf7E,1093
404
- cognee/modules/graph/cognee_graph/CogneeGraph.py,sha256=w1t0Xj-m1dXbSedxIljWk7jPJHw0Oh0YFNIjH6yUc5o,7514
405
- cognee/modules/graph/cognee_graph/CogneeGraphElements.py,sha256=tdwA8Z7RyTmlmdNG3A3Vnwi5YM14QEu18nHYmmbROns,5517
406
+ cognee/modules/graph/cognee_graph/CogneeGraph.py,sha256=0TUWZ3DFTMYnM_DQeT-da8v2zG3DUXGrT7lxjhw3-tU,7465
407
+ cognee/modules/graph/cognee_graph/CogneeGraphElements.py,sha256=XuniVXP9tM1ozcBCZpwquPNcSAdG1TFTNrhm0dPi1vM,5347
406
408
  cognee/modules/graph/cognee_graph/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
407
- cognee/modules/graph/exceptions/__init__.py,sha256=WUh8QcjerEyiBFItLvPfG_YaP5mgG9NFZeROROjSMbM,205
408
- cognee/modules/graph/exceptions/exceptions.py,sha256=TJbE8XJDsFDvAzR7jYJRLbS3nmZTFCAAuBJJ4tPcg2s,770
409
+ cognee/modules/graph/exceptions/__init__.py,sha256=PxHp8gF6DA773RNHzEeh17Nvc0O6EtOxAl5EJiWr9Yo,263
410
+ cognee/modules/graph/exceptions/exceptions.py,sha256=XqJqUsDAN0QQa3IufoVgK5lUn9Eq_n1VcXAclX0pbHk,1492
409
411
  cognee/modules/graph/methods/__init__.py,sha256=jVNRdhln1xkJQbl4a-8DuBALcV0l0dCk0zXUbShBFKc,63
410
412
  cognee/modules/graph/methods/get_formatted_graph_data.py,sha256=0n6pasPhTgOI3ZMYxtT4qIAnvcCvhP_oT-wZANCBhzo,1685
411
413
  cognee/modules/graph/models/EdgeType.py,sha256=Q1gvrrAfVL5RHU6YmRlHMZ7wNx3zB3EYwiDf7CFjIHk,194
@@ -428,19 +430,19 @@ cognee/modules/ingestion/data_types/S3BinaryData.py,sha256=Kdd4R2anhhIPQZ-5xihcW
428
430
  cognee/modules/ingestion/data_types/TextData.py,sha256=mysYnoxa1nTazTqN9hDng9PAHBxrz4F4-0qd11tsP5s,1018
429
431
  cognee/modules/ingestion/data_types/__init__.py,sha256=6EUoDx0yDYM3_nyyifVfoWVYIZaI7ESNL75FmWRQZmg,207
430
432
  cognee/modules/ingestion/exceptions/__init__.py,sha256=3vPYqemi-ztg3fCknAT6QvNG2mNG6ZTXuyDKlFbPUNU,174
431
- cognee/modules/ingestion/exceptions/exceptions.py,sha256=6AmiKoEaZ_gLtgsGxS_Zg6d1tZFbElp1RMJAZPyjfkI,373
433
+ cognee/modules/ingestion/exceptions/exceptions.py,sha256=Jv9OiRLDGMULkeEnAwb7BTGWfDlhvekDhxAp6mWmxdc,387
432
434
  cognee/modules/metrics/operations/__init__.py,sha256=MZ3xbVdfEKqfLct8WnbyFVyZmkBfl-77GoBQgktilUM,63
433
435
  cognee/modules/metrics/operations/get_pipeline_run_metrics.py,sha256=upIWnzKeJT1_XbL_ABdGxW-Ai7mO3AqMK35BNmItIQQ,2434
434
436
  cognee/modules/observability/get_observe.py,sha256=chRw4jmpmrwEvDecF9sgApm23IOzVgCbwkKEAyz1_AI,264
435
437
  cognee/modules/observability/observers.py,sha256=SKQSWWyGDG0QY2_bqsFgfpLUb7OUL4WFf8tDZYe5JMM,157
436
438
  cognee/modules/ontology/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
437
439
  cognee/modules/ontology/exceptions/__init__.py,sha256=GPIE6-a9ZWypIFGe8q00qQRc-UMBYLAnTc_jOatF9Jk,214
438
- cognee/modules/ontology/exceptions/exceptions.py,sha256=6b_M4Sl2Jpf9ltgaUKMM9iETzrVPJUpy0r0C8xjazyg,989
440
+ cognee/modules/ontology/exceptions/exceptions.py,sha256=Rv7NMdsSqTo0zLZiNkn7GAKxyH1g0rc4Vm_lChA7-Ws,1001
439
441
  cognee/modules/ontology/rdf_xml/OntologyResolver.py,sha256=mmIVlvSy8s8guS5OBueAPqsyBgEe4hlOpl8CQNmkFNI,7786
440
442
  cognee/modules/ontology/rdf_xml/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
441
443
  cognee/modules/pipelines/__init__.py,sha256=KlOaKgr8hhDie8kpGPRUy5p1e6bvKmWQsolmvpCMWqU,178
442
444
  cognee/modules/pipelines/exceptions/__init__.py,sha256=s0mB-aroyDgHg7cqSYpuyO_IpToVAAyxkDUvjs3tHhw,47
443
- cognee/modules/pipelines/exceptions/exceptions.py,sha256=aaycBo10GsKMwxeg4z4U0YkIubUEtQBJGsuCIdyosdM,370
445
+ cognee/modules/pipelines/exceptions/exceptions.py,sha256=HISOGOPI_J69ggIy3xALJhqXPHAsK1Yzydyu68jtlmo,376
444
446
  cognee/modules/pipelines/methods/__init__.py,sha256=zO7zvcH6gHU4N-6-wwnbtNOuHhgoUKf9MCwA_wYm5MI,116
445
447
  cognee/modules/pipelines/methods/get_pipeline_run.py,sha256=M4uOiCytjS-TPU9WkFtCJMzRD6E8O8ZsSFJ_farfIwk,473
446
448
  cognee/modules/pipelines/methods/get_pipeline_run_by_dataset.py,sha256=ZP1MJ6wFzMxbn-wpNkHrwKrTavObSuHrNQvtxa4laCw,1063
@@ -490,14 +492,16 @@ cognee/modules/retrieval/context_providers/__init__.py,sha256=47DEQpj8HBSa-_TImW
490
492
  cognee/modules/retrieval/entity_extractors/DummyEntityExtractor.py,sha256=DdnzyMAAeRKT_dwLmWff4FVfOMSPlXjFmPbJ1abfVhU,566
491
493
  cognee/modules/retrieval/entity_extractors/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
492
494
  cognee/modules/retrieval/exceptions/__init__.py,sha256=9yC54Z5HmoDnti9_yFLXK9_l3aHAlCTDfPGMMTN7WfM,187
493
- cognee/modules/retrieval/exceptions/exceptions.py,sha256=ILB3aMg3F8yI6XRxs21tFasySXNRxMeASJAjrk7eIVg,1168
495
+ cognee/modules/retrieval/exceptions/exceptions.py,sha256=T5cMVXoW_JhtUeIJap3veN7l2thgb0W5MN90bunzl24,1390
494
496
  cognee/modules/retrieval/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
495
497
  cognee/modules/retrieval/utils/brute_force_triplet_search.py,sha256=B1Yrnc6vRjZ1OSKYDrXkM2J0UwdyQdkXJ-pwZnI9Yss,7651
496
498
  cognee/modules/retrieval/utils/completion.py,sha256=5q1zs3OT4XCORBLpt2JTJLMZ4n-Es6amkWilU-IuKcQ,999
497
499
  cognee/modules/retrieval/utils/description_to_codepart_search.py,sha256=ZvGwJt1_6GyZM_0JSzxP42lAgtMflyZpj-T53Se3WgU,6331
498
500
  cognee/modules/retrieval/utils/stop_words.py,sha256=HP8l2leoLf6u7tnWHrYhgVMY_TX6yetGIae8DTsTEH4,883
501
+ cognee/modules/search/exceptions/__init__.py,sha256=7lH9BznC2274FD481U_8hfrxWonzJ8Mcv4oAkFFveqc,172
502
+ cognee/modules/search/exceptions/exceptions.py,sha256=Zc5Y0M-r-UnSSlpKzHKBplfjZ-Kcvmx912Cuw7wBg9g,431
499
503
  cognee/modules/search/methods/__init__.py,sha256=jGfRvNwM5yIzj025gaVhcx7nCupRSXbUUnFjYVjL_Js,27
500
- cognee/modules/search/methods/search.py,sha256=m8gRk55Y9Nz_81__uIPeXW1g3i1OKsxm7iJ9r0i1HXI,7731
504
+ cognee/modules/search/methods/search.py,sha256=K9shRD79K_yhoxegTgtu-AoGorL3n8tGdaUZSgbDAYc,7730
501
505
  cognee/modules/search/models/Query.py,sha256=9WcF5Z1oCFtA4O-7An37eNAPX3iyygO4B5NSwhx7iIg,558
502
506
  cognee/modules/search/models/Result.py,sha256=U7QtoNzAtZnUDwGWhjVfcalHQd4daKtYYvJz2BeWQ4w,564
503
507
  cognee/modules/search/operations/__init__.py,sha256=AwJl6v9BTpocoefEZLk-flo1EtydYb46NSUoNFHkhX0,156
@@ -529,7 +533,7 @@ cognee/modules/users/authentication/default/default_jwt_strategy.py,sha256=bRcV2
529
533
  cognee/modules/users/authentication/default/default_transport.py,sha256=RuaTV0Xa1g_AtQXDb54UlJTjSoKDfVqLJmKNRAbJIa0,318
530
534
  cognee/modules/users/authentication/methods/authenticate_user.py,sha256=TlzNEMK9gVRBFIAEof2ywz9gNq-M7rhlLAXLmdnQrmU,1012
531
535
  cognee/modules/users/exceptions/__init__.py,sha256=wDaNSgJtRXA8xMw3qhPX3FS8dFOz4S2fJ4bx_wPotgc,276
532
- cognee/modules/users/exceptions/exceptions.py,sha256=JApidGpWUq9bCSX7Lb6jGm48X0YrVH6qhb3nKO0_VyM,1597
536
+ cognee/modules/users/exceptions/exceptions.py,sha256=J4xBeR0VtKMpr8sQwHD2BOWSzBfPdUn700Z0bFCBxwM,1639
533
537
  cognee/modules/users/methods/__init__.py,sha256=mAjzPq6LMIsn5vDpM8ZNFQ7KUG-On3z80VXASX-FeQ8,313
534
538
  cognee/modules/users/methods/create_default_user.py,sha256=MRdbYof0NJZARO6t3wUr0hw6Kk0lJ_aE3kOvGy9HbW8,555
535
539
  cognee/modules/users/methods/create_user.py,sha256=SHeeb-_7c36KE-JMoXB6Ug0cllcIrlCFV86pW3TjkIw,2490
@@ -585,7 +589,7 @@ cognee/shared/encode_uuid.py,sha256=-EVtObzdnhSXKb5gz-SH1abicaaeiJ2rqcrLMr8V_cs,
585
589
  cognee/shared/logging_utils.py,sha256=H5-x2rrzVqDWf-tenU6F3UK8zApkXZ8BHMPp_OCKM9s,16321
586
590
  cognee/shared/utils.py,sha256=_SbEc1206iuY6mgr0CYOkBbkWpsYQu_e4BYHuLTc_q4,55207
587
591
  cognee/shared/exceptions/__init__.py,sha256=NNi2QcqIxj8tZNuQP5FC-kJCw1e5jcRaoQAmkQEpyjM,179
588
- cognee/shared/exceptions/exceptions.py,sha256=4mZlslg6iKVMI10w2ZLzsGQnh00WNVpqv2zyJ72bx18,347
592
+ cognee/shared/exceptions/exceptions.py,sha256=-sCKV-NReNDLSa98iZqT22s2rUfk0geMGJVsYRyzy3Q,361
589
593
  cognee/tasks/chunk_naive_llm_classifier/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
590
594
  cognee/tasks/chunk_naive_llm_classifier/chunk_naive_llm_classifier.py,sha256=LAaFbVsiUGhNTpPNdGkN4iq-f0t9d6gNhwzLp500rgk,7819
591
595
  cognee/tasks/chunks/__init__.py,sha256=t0Nr_k9yZ5H1BM9osO4KIG-gLBSypxeije5FLSNzFB0,208
@@ -598,18 +602,20 @@ cognee/tasks/code/get_local_dependencies_checker.py,sha256=RTLc1j031XiTKUwFgMcZw
598
602
  cognee/tasks/code/get_repo_dependency_graph_checker.py,sha256=4OVnDrh87AOrlDY-jNLzPWKcodlaxoxDUTaHddDwUAA,1213
599
603
  cognee/tasks/completion/__init__.py,sha256=6wAb89ES_bsb2JUvirXbCoAXScLRCoRJALHMHwqzuLw,67
600
604
  cognee/tasks/completion/exceptions/__init__.py,sha256=xpeAGtV2DfxByZaT0BJoZrEwgEU_PJKboVWhl-BSUgw,177
601
- cognee/tasks/completion/exceptions/exceptions.py,sha256=L8gs73LYUF10LVOn08zgXA8hCWQzR6gnzxg4a5FB4wM,598
605
+ cognee/tasks/completion/exceptions/exceptions.py,sha256=yQ1dwOtgH9tsiWip062X-HpN_19qJcBDWf9fEZIyyFw,619
602
606
  cognee/tasks/documents/__init__.py,sha256=EJfLDJpbVuQy_IXmj5dIQDJ3lXWcNbjo4LdPXJIgDkg,195
603
607
  cognee/tasks/documents/check_permissions_on_dataset.py,sha256=vdpRBqDeMJJqhYgD7adkLTveRVhifQiUNA8xXdnUVD0,981
604
- cognee/tasks/documents/classify_documents.py,sha256=y-JyMW9y-yZBx0vGZE3c2dg9RtQFAKnKOGHSa5KknCo,4025
605
- cognee/tasks/documents/extract_chunks_from_documents.py,sha256=BiO8olCglyA7XeVbd0QH8PyAtuXRc3UakKu1JPvIdpQ,1985
608
+ cognee/tasks/documents/classify_documents.py,sha256=WALiguFaRXfKJtVbJtTDwS56_UWu2kOkA4H8KBTm8gA,4205
609
+ cognee/tasks/documents/extract_chunks_from_documents.py,sha256=vOrWlhRCrbNUM8KM263rQZbskbbX7B4o79U0YrJ9IRg,2320
610
+ cognee/tasks/documents/exceptions/__init__.py,sha256=izXT-CbmvzqBOyDkJRDtW6RFi_V9SC--EqY2uEEO04E,234
611
+ cognee/tasks/documents/exceptions/exceptions.py,sha256=47BvCT3PpuwCKMPLlFQjjVY6B_rsCcxgCbbA39X1QUU,1090
606
612
  cognee/tasks/entity_completion/entity_extractors/llm_entity_extractor.py,sha256=bGZOLuVHWLReMPi98sClexZMkjG6eGxprlKLU7P_fpc,2570
607
613
  cognee/tasks/entity_completion/entity_extractors/regex_entity_config.json,sha256=JIq8wK0i4c5u5TfjdmUZGCYjmra0vm43KBFhRwI_Oa0,2432
608
614
  cognee/tasks/entity_completion/entity_extractors/regex_entity_config.py,sha256=HfQVEhrDbZekylrs2vjepN5Zrl7q4Z_HIUNA-P37_Y4,3835
609
615
  cognee/tasks/entity_completion/entity_extractors/regex_entity_extractor.py,sha256=0Hv4mp--ke7YLLAKX84323XWspF_x4oqgp1LYO_EGHo,3056
610
616
  cognee/tasks/graph/__init__.py,sha256=SLY4uxR1sWWlyOaWcRhUPy2XJ7spC2mtVftv4ugVdWg,122
611
617
  cognee/tasks/graph/extract_graph_from_code.py,sha256=1Y_v-vp5XfMA2RQQwVWTgRg1DfNOMDsQeegjmwj8joI,1043
612
- cognee/tasks/graph/extract_graph_from_data.py,sha256=HcJFMaXguygzi7cbjhi9H72CNWJCGyUk3hDsBOfueR4,2518
618
+ cognee/tasks/graph/extract_graph_from_data.py,sha256=xWGlT2L2cg-nvLGfOuQGuoYUJj8bqJJEn72IoibHZCU,3807
613
619
  cognee/tasks/graph/extract_graph_from_data_v2.py,sha256=0mb5Vs0eTPbGgttPCZr0j2IjR4ztzh5k7zKHkwzeYMM,1742
614
620
  cognee/tasks/graph/infer_data_ontology.py,sha256=lzhwhSelhaSOjtBDtcB46gki3wFywyzfpyVQOMLrHq8,12214
615
621
  cognee/tasks/graph/models.py,sha256=DukW6jeBeCMHxVDYfbJH2p7xStAblWx_i45P0RAr6n8,2831
@@ -624,27 +630,35 @@ cognee/tasks/graph/cascade_extract/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5
624
630
  cognee/tasks/graph/cascade_extract/utils/extract_content_nodes_and_relationship_names.py,sha256=IOTbodNnWFAD8yB8kkFfK91cZP68zdyu_rIVMVo1DDE,2306
625
631
  cognee/tasks/graph/cascade_extract/utils/extract_edge_triplets.py,sha256=vmfLRYrF8oxUwht4GZQ2VEZzy2sqd7pLUvthk8zbxR4,2261
626
632
  cognee/tasks/graph/cascade_extract/utils/extract_nodes.py,sha256=cX0Z8q4q5MopztY_hAJ874ZKem3XA2H_sdNCcXlq1u0,1455
633
+ cognee/tasks/graph/exceptions/__init__.py,sha256=9Qsr4LQ38vFf6VX9ACHDwBwotFpHJIcyVI6fURZN8Zo,271
634
+ cognee/tasks/graph/exceptions/exceptions.py,sha256=gzEnP2M3o_z0VEVntZAv5ej81qz1lJYoghHmvOjmO6I,1319
627
635
  cognee/tasks/ingestion/__init__.py,sha256=TqqVr_LsoBRWn-F8mnWTd-3eQS4QUWj4nGbL8bXKjo0,234
628
636
  cognee/tasks/ingestion/data_item_to_text_file.py,sha256=F5Zg7S8cniGRLmMzvxH7OglXklr1ST-rrMxn8EayQjY,3136
629
637
  cognee/tasks/ingestion/get_dlt_destination.py,sha256=vzky_-TFlnaREbdbndAkXwwlb-0gRq8vDaQ2OOyOSQ4,2075
630
638
  cognee/tasks/ingestion/ingest_data.py,sha256=AWdQ5YKzYZB4nkra1Zm5SJ0F-bXOcsnLdpp83eAX68w,8280
631
639
  cognee/tasks/ingestion/migrate_relational_database.py,sha256=dbHu7TAMjq9zH69bT9q1ziQwXCoZ-VvJY5uY5aer0nc,10260
632
- cognee/tasks/ingestion/resolve_data_directories.py,sha256=CDmIQeSBXi1Nejamgg3j2SebPPoeNvtpqJI168NWQlM,3054
640
+ cognee/tasks/ingestion/resolve_data_directories.py,sha256=Q8wDNY62S9Rutj5EHYe7nB51p6BonUIsm7E_ZBy7WpY,3251
633
641
  cognee/tasks/ingestion/save_data_item_to_storage.py,sha256=OB5h30L1cTB_1jqyPyx7p-eq-HDZmjOzEo7NDyWkl5c,2366
634
642
  cognee/tasks/ingestion/transform_data.py,sha256=cbI1FX86-Ft3pGRRHedjarXst9TR7O9ce39bYRhbf2Y,1506
643
+ cognee/tasks/ingestion/exceptions/__init__.py,sha256=GhFSvM6ChVsm16b-zaCkbluH9zCX72Wy-QLyMXyEAe4,240
644
+ cognee/tasks/ingestion/exceptions/exceptions.py,sha256=3wgLiK4G77nMc95-STtqs3LPWcRtzH_bDamikG4uyD0,385
635
645
  cognee/tasks/repo_processor/__init__.py,sha256=TOl17meDnorn5m8mJ_7p3CAbwqGLhXM5Tx5va6cpZZE,116
636
646
  cognee/tasks/repo_processor/get_local_dependencies.py,sha256=uxed867uImtgJPCcvJ1rqWT5_Rzml8p__ZCUMvCJgVQ,11464
637
647
  cognee/tasks/repo_processor/get_non_code_files.py,sha256=SBHdEFI4GP34eRVeCrvC_GvnMVEiL0rP1Itvvgge3ec,3400
638
648
  cognee/tasks/repo_processor/get_repo_file_dependencies.py,sha256=X2DSilt9fhnBGRfS10lNbA7VLdqUQ9k7JATHSTBdDho,4735
639
649
  cognee/tasks/storage/__init__.py,sha256=pbFosH4bALh7TtftAz_hX6RDYN65lYfSiq8IXIHTjm4,143
640
- cognee/tasks/storage/add_data_points.py,sha256=qiAlUe5unr986VfB-STJdAPv8W-HfFIS4YxbeiJ5neU,1326
641
- cognee/tasks/storage/index_data_points.py,sha256=tzPk-MeQqDw7ts8s5olfcV6oOsQCAm8qBUval-cHHvw,4438
650
+ cognee/tasks/storage/add_data_points.py,sha256=prDPsi7EKZpXldko2zU4M8HmhNW-wg1-8z2Pqs14wBc,1707
651
+ cognee/tasks/storage/index_data_points.py,sha256=laM3qiREp22XUSJixK8tm5Db1XVzJKnhBijLz5tLDf4,4419
642
652
  cognee/tasks/storage/index_graph_edges.py,sha256=LVeaTiX3f3IDbH1Q3-7KfeJjXi_WTDXLXtGRrFdCIyY,3068
653
+ cognee/tasks/storage/exceptions/__init__.py,sha256=YSP4OY_TFhr017IdJxVK-2XzsaW_PeiHkXDVYTIvYTU,192
654
+ cognee/tasks/storage/exceptions/exceptions.py,sha256=bT485p0CghSmPGL2XOhxFMTdQaTvtnG1IunKgtZz2EI,396
643
655
  cognee/tasks/summarization/__init__.py,sha256=fYjE2Bbm8zUmE6up6Xz0zA9w2uPvvU67mRIzrqZZIjs,86
644
656
  cognee/tasks/summarization/mock_summary.py,sha256=6d-K67Sb5QzgfaPx-h4fxfCvHzx_FSYmhL6CB6SP3U0,1997
645
657
  cognee/tasks/summarization/models.py,sha256=bI0lI2TCUtpQ0SebYud0sMhntuuKp9wtTi0qqvVHDfM,1259
646
658
  cognee/tasks/summarization/summarize_code.py,sha256=GONddNe4AAywGrUn5eWhj8OHOm068IEAxT6cKWpn4-M,1138
647
- cognee/tasks/summarization/summarize_text.py,sha256=IdcRJzCoqxOwBwiIpy5A32-0T0JaE0_zSVewcbg-eb4,1895
659
+ cognee/tasks/summarization/summarize_text.py,sha256=RCxm377UrSLhYVZvMJr_SHpmqj7gx6JE9XkOu_n68i4,2261
660
+ cognee/tasks/summarization/exceptions/__init__.py,sha256=FYdwzQQ4CWCSHZ2oNB4shYV_qZd-EZwmyEn83jkshzo,180
661
+ cognee/tasks/summarization/exceptions/exceptions.py,sha256=s7Muy3usrg8klHkPqHPtYkejdAkKZ5YyAP7E6L8GA5o,412
648
662
  cognee/tasks/temporal_awareness/__init__.py,sha256=2eOrrzKF86I_1smZ4QdrjfBQ-PWk9mb2k3X-EHBmO24,172
649
663
  cognee/tasks/temporal_awareness/build_graph_with_temporal_awareness.py,sha256=oWklT-qrhw2ttEYOW76bdAjoYYkLk8zIkSVXdzAF_y8,752
650
664
  cognee/tasks/temporal_awareness/graphiti_model.py,sha256=e5mEkTM1Od02UfLhhVbCu6RxVVZJOFGO_FuY717ELb0,820
@@ -655,7 +669,7 @@ cognee/tests/test_chromadb.py,sha256=D9JEN0xbFxNLgp8UJTVAjpwob9S-LOQC-hSaMVvYhR8
655
669
  cognee/tests/test_cognee_server_start.py,sha256=kcIbzu72ZZUlPZ51c_DpSCCwx3X9mNvYZrVcxHfZaJs,4226
656
670
  cognee/tests/test_custom_model.py,sha256=vypoJkF5YACJ6UAzV7lQFRmtRjVYEoPcUS8Rylgc1Wg,3465
657
671
  cognee/tests/test_deduplication.py,sha256=1wSVq58rwFQOE5JtUcO3T92BBzzGTkC49yiaausjOac,8365
658
- cognee/tests/test_delete_by_id.py,sha256=KX76nBzUiQTmdsMRtZOl2DZb5i5iGg0vMz9BS1PdbAc,13182
672
+ cognee/tests/test_delete_by_id.py,sha256=ROCGHYpI9bLBC-6d6F1VUq4kGPr0fzutbcSO3CGKbC8,13175
659
673
  cognee/tests/test_deletion.py,sha256=3IKAxhXaRcBo1Ry0BPhdpEM0lz5sDDIxv_v4cgUhrPE,5112
660
674
  cognee/tests/test_edge_ingestion.py,sha256=TrJAa3skdVnFQSZ_qwtJui2t8UqkC9q4hIzWjkre3ew,3183
661
675
  cognee/tests/test_falkordb.py,sha256=WS_ZvB10mipumYCIXNggmhdMHZXbtAjmUJofVnwLRMI,7729
@@ -724,7 +738,7 @@ cognee/tests/unit/interfaces/graph/get_graph_from_model_circular_test.py,sha256=
724
738
  cognee/tests/unit/interfaces/graph/get_graph_from_model_unit_test.py,sha256=Hgl8zw-LJa1ylmy627T2s5lsmYBTIUdSXP4gw5HZYoA,6763
725
739
  cognee/tests/unit/interfaces/graph/test_weighted_edges.py,sha256=Xi-iVyhVJ4YzAgG-eGvx9kZLtwCMhq2Ca43xNkaO_K4,13315
726
740
  cognee/tests/unit/modules/data/test_open_data_file.py,sha256=IjmTE1AAQ9SqJV8kUTjC9aCCYu-0NWqwhV_gynlxoCY,4443
727
- cognee/tests/unit/modules/graph/cognee_graph_elements_test.py,sha256=LnIdzLGXT4fq9DLbtpCKGaOHtA2A3Hn9fbuKyn4qbKo,5147
741
+ cognee/tests/unit/modules/graph/cognee_graph_elements_test.py,sha256=Ns8JKOMQML8tHKEg-9e9WG-spNZTGNvwmNETuySJqK0,5048
728
742
  cognee/tests/unit/modules/graph/cognee_graph_test.py,sha256=FKZSVi6dDqFDO_kyRDQmOx_mL4trFY0n20un3GlDERk,2259
729
743
  cognee/tests/unit/modules/ontology/test_ontology_adapter.py,sha256=eB8X66cs5RHIaSDEotgsqpcFidpjBD6LNmUfD6D6ATo,4606
730
744
  cognee/tests/unit/modules/pipelines/run_task_from_queue_test.py,sha256=X2clLQYoPgzmk0QWDmDpJIKShSVh8e8xS76PMP7qeIg,1705
@@ -738,7 +752,7 @@ cognee/tests/unit/modules/retrieval/insights_retriever_test.py,sha256=xkbxlNiHY6
738
752
  cognee/tests/unit/modules/retrieval/rag_completion_retriever_test.py,sha256=3A5FvbN4-x8N7wccB9vbVVv9neLFRM6KkZjTkLu0MSo,6101
739
753
  cognee/tests/unit/modules/retrieval/summaries_retriever_test.py,sha256=IfhDyVuKUrjCEy22-Mva9w7li2mtPZT9FlNIFvpFMKw,4950
740
754
  cognee/tests/unit/modules/retriever/test_description_to_codepart_search.py,sha256=oayCbXQtvmTnlgOuR67w_r278TGMEv-puaTR_jI6weo,4164
741
- cognee/tests/unit/modules/search/search_methods_test.py,sha256=ESOhiciRHQ6OhzdCKBsz4yQOLOfmkJ6Gw3Wbd66KSH8,7418
755
+ cognee/tests/unit/modules/search/search_methods_test.py,sha256=BRZITRlRm4oADNk1-xBh8f4Y41IhpybdaPIqKwKFUTU,7451
742
756
  cognee/tests/unit/modules/visualization/visualization_test.py,sha256=JuNsyqAbEWgO5QZP1lCE0_MDQDJ75WhXLXPyat0mhVw,929
743
757
  cognee/tests/unit/processing/chunks/chunk_by_paragraph_2_test.py,sha256=KGZZq1cMgn-im9hgfeNW74zobSLSOwuzQKzwNBmW6sM,2487
744
758
  cognee/tests/unit/processing/chunks/chunk_by_paragraph_test.py,sha256=GRLoeusPg-jzaSLXhZv0zNpmt8gd5vNrCYCAzR5W1TE,2658
@@ -746,8 +760,25 @@ cognee/tests/unit/processing/chunks/chunk_by_sentence_test.py,sha256=j0zcTrGBe_s
746
760
  cognee/tests/unit/processing/chunks/chunk_by_word_test.py,sha256=tkZJgZMAIwyqqjc9aEeor4r4_1CD7PlyXQCCTiWw8U4,1177
747
761
  cognee/tests/unit/processing/chunks/test_input.py,sha256=T1TFdZWBuV7Mwm6kD-It2sMCRTozGjgP8YcMNgT7j_E,9463
748
762
  cognee/tests/unit/processing/utils/utils_test.py,sha256=VQ03g2r2thojM7dLnY7egtXXCpJuxcEcSxPtTZa4b0U,1503
749
- cognee-0.2.2.dev1.dist-info/METADATA,sha256=V7LPZ7zEEZGxPB45SaRiaYdICRmjFSEMiTMhjrK8vf0,14674
750
- cognee-0.2.2.dev1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
751
- cognee-0.2.2.dev1.dist-info/licenses/LICENSE,sha256=pHHjSQj1DD8SDppW88MMs04TPk7eAanL1c5xj8NY7NQ,11344
752
- cognee-0.2.2.dev1.dist-info/licenses/NOTICE.md,sha256=6L3saP3kSpcingOxDh-SGjMS8GY79Rlh2dBNLaO0o5c,339
753
- cognee-0.2.2.dev1.dist-info/RECORD,,
763
+ distributed/Dockerfile,sha256=co4eAKKe8wOWmvdrbvBP_S7aYQSho6SdDFgnYrLarX4,756
764
+ distributed/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
765
+ distributed/app.py,sha256=ZbnFimZvsgDuXWA6F-5o1KQJcJUGbJkuqS9MZbtdBSA,62
766
+ distributed/entrypoint.py,sha256=tLF8eEfW5oG6obyzBGhJsqy9SXm6xfr5HYU45sMtQEw,2221
767
+ distributed/entrypoint.sh,sha256=6j9Lu_6sxws7E-XGOdtfMWmdm8D3awjtGblB0JsL1I0,94
768
+ distributed/modal_image.py,sha256=TSS0mF8-bjTiMHzyQu2xgq3AX74MRnJKyDgFroHvwiQ,320
769
+ distributed/poetry.lock,sha256=KcXMRPszF6A1SKWSqp1Kwg0CFuq4yJ7nn3TUWJ8LNrA,952845
770
+ distributed/pyproject.toml,sha256=cAU_AZEkP3G2hCtqKFUMVkb2yXMrBn5ZAURlGUoYZjU,4990
771
+ distributed/queues.py,sha256=FF2nyfu1rsDTNKEEC2IAn_A0sULudt4JFxdffSyRPb0,212
772
+ distributed/test.py,sha256=NftfFht_2TP898KwdyJNE6FRYeaDGt4l_74KPVVsGJc,552
773
+ distributed/utils.py,sha256=rW2s7xkMPbh6IANFRVaXPI1pJf9SKRufUw4SfIFIFME,586
774
+ distributed/tasks/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
775
+ distributed/tasks/queued_add_data_points.py,sha256=thOrQldx4NlEILPoftYtd4jKtYuio3AtugDUdzLPBB4,568
776
+ distributed/tasks/queued_add_edges.py,sha256=kz1DHE05y-kNHORQJjYWHUi6Q1QWUp_v3Dl13ymEZLM,452
777
+ distributed/tasks/queued_add_nodes.py,sha256=aqK4Ij--ADwUWknxYpiwbYrpa6CcvFfqHWbUZW4Kh3A,452
778
+ distributed/workers/data_point_saving_worker.py,sha256=jFmA0-P_0Ru2IUDrSug0wML-5goAKrGtlBm5BA5Ryw4,3229
779
+ distributed/workers/graph_saving_worker.py,sha256=oUYl99CdhlrPAIsUOHbHnS3d4XhGoV0_OIbCO8wYzRg,3648
780
+ cognee-0.2.3.dev1.dist-info/METADATA,sha256=AW7NrSApkKDg8kMH_vwAQv91uYxIpQC37x9SItRFxsg,14675
781
+ cognee-0.2.3.dev1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
782
+ cognee-0.2.3.dev1.dist-info/licenses/LICENSE,sha256=pHHjSQj1DD8SDppW88MMs04TPk7eAanL1c5xj8NY7NQ,11344
783
+ cognee-0.2.3.dev1.dist-info/licenses/NOTICE.md,sha256=6L3saP3kSpcingOxDh-SGjMS8GY79Rlh2dBNLaO0o5c,339
784
+ cognee-0.2.3.dev1.dist-info/RECORD,,
distributed/Dockerfile ADDED
@@ -0,0 +1,34 @@
1
+ FROM python:3.11-slim
2
+
3
+ # Set environment variables
4
+ ENV PIP_NO_CACHE_DIR=true
5
+ ENV PATH="${PATH}:/root/.poetry/bin"
6
+ ENV PYTHONPATH=/app
7
+ ENV RUN_MODE=modal
8
+ ENV SKIP_MIGRATIONS=true
9
+ ENV COGNEE_DISTRIBUTED=true
10
+
11
+ # System dependencies
12
+ RUN apt-get update && apt-get install -y \
13
+ gcc \
14
+ libpq-dev \
15
+ git \
16
+ curl \
17
+ build-essential \
18
+ && rm -rf /var/lib/apt/lists/*
19
+
20
+ WORKDIR /app
21
+
22
+ COPY pyproject.toml poetry.lock README.md /app/
23
+
24
+ RUN pip install poetry
25
+
26
+ RUN poetry config virtualenvs.create false
27
+
28
+ RUN poetry install --extras neo4j --extras postgres --extras aws --extras distributed --no-root
29
+
30
+ COPY cognee/ /app/cognee
31
+ COPY distributed/ /app/distributed
32
+ RUN chmod +x /app/distributed/entrypoint.sh
33
+
34
+ ENTRYPOINT ["/app/distributed/entrypoint.sh"]
File without changes
distributed/app.py ADDED
@@ -0,0 +1,4 @@
1
+ from modal import App
2
+
3
+
4
+ app = App("cognee_modal_distributed")
@@ -0,0 +1,71 @@
1
+ import os
2
+ import asyncio
3
+
4
+ import cognee
5
+ from cognee.api.v1.prune import prune
6
+ from cognee.shared.logging_utils import get_logger
7
+ from cognee.modules.engine.operations.setup import setup
8
+
9
+ from distributed.app import app
10
+ from distributed.queues import add_nodes_and_edges_queue, add_data_points_queue
11
+ from distributed.workers.graph_saving_worker import graph_saving_worker
12
+ from distributed.workers.data_point_saving_worker import data_point_saving_worker
13
+
14
+ logger = get_logger()
15
+
16
+
17
+ os.environ["COGNEE_DISTRIBUTED"] = "True"
18
+
19
+
20
+ @app.local_entrypoint()
21
+ async def main():
22
+ # Clear queues
23
+ await add_nodes_and_edges_queue.clear.aio()
24
+ await add_data_points_queue.clear.aio()
25
+
26
+ number_of_graph_saving_workers = 1 # Total number of graph_saving_worker to spawn
27
+ number_of_data_point_saving_workers = 5 # Total number of graph_saving_worker to spawn
28
+
29
+ results = []
30
+ consumer_futures = []
31
+
32
+ # await prune.prune_data() # We don't want to delete files on s3
33
+ # Delete DBs and saved files from metastore
34
+ await prune.prune_system(metadata=True)
35
+
36
+ await setup()
37
+
38
+ # Start graph_saving_worker functions
39
+ for _ in range(number_of_graph_saving_workers):
40
+ worker_future = graph_saving_worker.spawn()
41
+ consumer_futures.append(worker_future)
42
+
43
+ # Start data_point_saving_worker functions
44
+ for _ in range(number_of_data_point_saving_workers):
45
+ worker_future = data_point_saving_worker.spawn()
46
+ consumer_futures.append(worker_future)
47
+
48
+ s3_bucket_path = os.getenv("S3_BUCKET_PATH")
49
+ s3_data_path = "s3://" + s3_bucket_path
50
+
51
+ await cognee.add(s3_data_path, dataset_name="s3-files")
52
+
53
+ await cognee.cognify(datasets=["s3-files"])
54
+
55
+ # Push empty tuple into the queue to signal the end of data.
56
+ await add_nodes_and_edges_queue.put.aio(())
57
+ await add_data_points_queue.put.aio(())
58
+
59
+ for consumer_future in consumer_futures:
60
+ try:
61
+ print("Finished but waiting for saving workers to finish.")
62
+ consumer_final = consumer_future.get()
63
+ print(f"All workers are done: {consumer_final}")
64
+ except Exception as e:
65
+ logger.error(e)
66
+
67
+ print(results)
68
+
69
+
70
+ if __name__ == "__main__":
71
+ asyncio.run(main())
@@ -0,0 +1,5 @@
1
+ #!/usr/bin/env bash
2
+
3
+ echo "$@"
4
+
5
+ exec "$@" # Runs the command passed to the entrypoint script.
@@ -0,0 +1,11 @@
1
+ import pathlib
2
+ from os import path
3
+ from modal import Image
4
+ from logging import getLogger
5
+
6
+ logger = getLogger("modal_image_creation")
7
+
8
+ image = Image.from_dockerfile(
9
+ path=pathlib.Path(path.join(path.dirname(__file__), "Dockerfile")).resolve(),
10
+ force_build=False,
11
+ ).add_local_python_source("cognee", "entrypoint")