cognee 0.3.4.dev4__py3-none-any.whl → 0.3.6__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 (184) hide show
  1. cognee/api/client.py +16 -7
  2. cognee/api/health.py +5 -9
  3. cognee/api/v1/add/add.py +3 -1
  4. cognee/api/v1/cognify/cognify.py +44 -7
  5. cognee/api/v1/permissions/routers/get_permissions_router.py +8 -4
  6. cognee/api/v1/search/search.py +3 -0
  7. cognee/api/v1/ui/__init__.py +1 -1
  8. cognee/api/v1/ui/ui.py +215 -150
  9. cognee/api/v1/update/__init__.py +1 -0
  10. cognee/api/v1/update/routers/__init__.py +1 -0
  11. cognee/api/v1/update/routers/get_update_router.py +90 -0
  12. cognee/api/v1/update/update.py +100 -0
  13. cognee/base_config.py +5 -2
  14. cognee/cli/_cognee.py +28 -10
  15. cognee/cli/commands/delete_command.py +34 -2
  16. cognee/eval_framework/corpus_builder/task_getters/get_default_tasks_by_indices.py +2 -2
  17. cognee/eval_framework/evaluation/direct_llm_eval_adapter.py +3 -2
  18. cognee/eval_framework/modal_eval_dashboard.py +9 -1
  19. cognee/infrastructure/databases/graph/config.py +9 -9
  20. cognee/infrastructure/databases/graph/get_graph_engine.py +4 -21
  21. cognee/infrastructure/databases/graph/kuzu/adapter.py +60 -9
  22. cognee/infrastructure/databases/hybrid/neptune_analytics/NeptuneAnalyticsAdapter.py +3 -3
  23. cognee/infrastructure/databases/relational/config.py +4 -4
  24. cognee/infrastructure/databases/relational/create_relational_engine.py +11 -3
  25. cognee/infrastructure/databases/vector/chromadb/ChromaDBAdapter.py +7 -3
  26. cognee/infrastructure/databases/vector/config.py +7 -7
  27. cognee/infrastructure/databases/vector/create_vector_engine.py +7 -15
  28. cognee/infrastructure/databases/vector/embeddings/EmbeddingEngine.py +9 -0
  29. cognee/infrastructure/databases/vector/embeddings/FastembedEmbeddingEngine.py +11 -0
  30. cognee/infrastructure/databases/vector/embeddings/LiteLLMEmbeddingEngine.py +19 -2
  31. cognee/infrastructure/databases/vector/embeddings/OllamaEmbeddingEngine.py +11 -0
  32. cognee/infrastructure/databases/vector/embeddings/config.py +8 -0
  33. cognee/infrastructure/databases/vector/embeddings/get_embedding_engine.py +5 -0
  34. cognee/infrastructure/databases/vector/lancedb/LanceDBAdapter.py +11 -10
  35. cognee/infrastructure/databases/vector/pgvector/PGVectorAdapter.py +48 -38
  36. cognee/infrastructure/databases/vector/vector_db_interface.py +8 -4
  37. cognee/infrastructure/files/storage/S3FileStorage.py +15 -5
  38. cognee/infrastructure/files/storage/s3_config.py +1 -0
  39. cognee/infrastructure/files/utils/open_data_file.py +7 -14
  40. cognee/infrastructure/llm/LLMGateway.py +19 -117
  41. cognee/infrastructure/llm/config.py +28 -13
  42. cognee/infrastructure/llm/{structured_output_framework/litellm_instructor/extraction → extraction}/extract_categories.py +2 -1
  43. cognee/infrastructure/llm/{structured_output_framework/litellm_instructor/extraction → extraction}/extract_event_entities.py +3 -2
  44. cognee/infrastructure/llm/{structured_output_framework/litellm_instructor/extraction → extraction}/extract_summary.py +3 -2
  45. cognee/infrastructure/llm/{structured_output_framework/litellm_instructor/extraction → extraction}/knowledge_graph/extract_content_graph.py +2 -1
  46. cognee/infrastructure/llm/{structured_output_framework/litellm_instructor/extraction → extraction}/knowledge_graph/extract_event_graph.py +3 -2
  47. cognee/infrastructure/llm/prompts/read_query_prompt.py +3 -2
  48. cognee/infrastructure/llm/prompts/show_prompt.py +35 -0
  49. cognee/infrastructure/llm/prompts/test.txt +1 -0
  50. cognee/infrastructure/llm/structured_output_framework/baml/baml_client/__init__.py +2 -2
  51. cognee/infrastructure/llm/structured_output_framework/baml/baml_client/async_client.py +50 -397
  52. cognee/infrastructure/llm/structured_output_framework/baml/baml_client/inlinedbaml.py +2 -3
  53. cognee/infrastructure/llm/structured_output_framework/baml/baml_client/parser.py +8 -88
  54. cognee/infrastructure/llm/structured_output_framework/baml/baml_client/runtime.py +78 -0
  55. cognee/infrastructure/llm/structured_output_framework/baml/baml_client/stream_types.py +2 -99
  56. cognee/infrastructure/llm/structured_output_framework/baml/baml_client/sync_client.py +49 -401
  57. cognee/infrastructure/llm/structured_output_framework/baml/baml_client/type_builder.py +19 -882
  58. cognee/infrastructure/llm/structured_output_framework/baml/baml_client/type_map.py +2 -34
  59. cognee/infrastructure/llm/structured_output_framework/baml/baml_client/types.py +2 -107
  60. cognee/infrastructure/llm/structured_output_framework/baml/baml_src/acreate_structured_output.baml +26 -0
  61. cognee/infrastructure/llm/structured_output_framework/baml/baml_src/extraction/__init__.py +1 -2
  62. cognee/infrastructure/llm/structured_output_framework/baml/baml_src/extraction/acreate_structured_output.py +76 -0
  63. cognee/infrastructure/llm/structured_output_framework/baml/baml_src/extraction/create_dynamic_baml_type.py +122 -0
  64. cognee/infrastructure/llm/structured_output_framework/baml/baml_src/generators.baml +3 -3
  65. cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/anthropic/adapter.py +0 -32
  66. cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/gemini/adapter.py +107 -98
  67. cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/generic_llm_api/adapter.py +5 -6
  68. cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/get_llm_client.py +5 -6
  69. cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/llm_interface.py +0 -26
  70. cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/openai/adapter.py +17 -67
  71. cognee/infrastructure/llm/tokenizer/Gemini/adapter.py +8 -7
  72. cognee/infrastructure/llm/utils.py +4 -4
  73. cognee/infrastructure/loaders/LoaderEngine.py +5 -2
  74. cognee/infrastructure/loaders/external/__init__.py +7 -0
  75. cognee/infrastructure/loaders/external/advanced_pdf_loader.py +244 -0
  76. cognee/infrastructure/loaders/supported_loaders.py +7 -0
  77. cognee/modules/data/methods/create_authorized_dataset.py +9 -0
  78. cognee/modules/data/methods/get_authorized_dataset.py +1 -1
  79. cognee/modules/data/methods/get_authorized_dataset_by_name.py +11 -0
  80. cognee/modules/data/methods/get_deletion_counts.py +92 -0
  81. cognee/modules/graph/cognee_graph/CogneeGraph.py +1 -1
  82. cognee/modules/graph/utils/expand_with_nodes_and_edges.py +22 -8
  83. cognee/modules/graph/utils/retrieve_existing_edges.py +0 -2
  84. cognee/modules/ingestion/data_types/TextData.py +0 -1
  85. cognee/modules/observability/get_observe.py +14 -0
  86. cognee/modules/observability/observers.py +1 -0
  87. cognee/modules/ontology/base_ontology_resolver.py +42 -0
  88. cognee/modules/ontology/get_default_ontology_resolver.py +41 -0
  89. cognee/modules/ontology/matching_strategies.py +53 -0
  90. cognee/modules/ontology/models.py +20 -0
  91. cognee/modules/ontology/ontology_config.py +24 -0
  92. cognee/modules/ontology/ontology_env_config.py +45 -0
  93. cognee/modules/ontology/rdf_xml/{OntologyResolver.py → RDFLibOntologyResolver.py} +20 -28
  94. cognee/modules/pipelines/layers/resolve_authorized_user_dataset.py +21 -24
  95. cognee/modules/pipelines/layers/resolve_authorized_user_datasets.py +3 -3
  96. cognee/modules/retrieval/code_retriever.py +2 -1
  97. cognee/modules/retrieval/context_providers/TripletSearchContextProvider.py +1 -4
  98. cognee/modules/retrieval/graph_completion_cot_retriever.py +6 -5
  99. cognee/modules/retrieval/graph_completion_retriever.py +0 -3
  100. cognee/modules/retrieval/insights_retriever.py +1 -1
  101. cognee/modules/retrieval/jaccard_retrival.py +60 -0
  102. cognee/modules/retrieval/lexical_retriever.py +123 -0
  103. cognee/modules/retrieval/natural_language_retriever.py +2 -1
  104. cognee/modules/retrieval/temporal_retriever.py +3 -2
  105. cognee/modules/retrieval/utils/brute_force_triplet_search.py +2 -12
  106. cognee/modules/retrieval/utils/completion.py +4 -7
  107. cognee/modules/search/methods/get_search_type_tools.py +7 -0
  108. cognee/modules/search/methods/no_access_control_search.py +1 -1
  109. cognee/modules/search/methods/search.py +32 -13
  110. cognee/modules/search/types/SearchType.py +1 -0
  111. cognee/modules/users/permissions/methods/authorized_give_permission_on_datasets.py +12 -0
  112. cognee/modules/users/permissions/methods/check_permission_on_dataset.py +11 -0
  113. cognee/modules/users/permissions/methods/get_all_user_permission_datasets.py +10 -0
  114. cognee/modules/users/permissions/methods/get_document_ids_for_user.py +10 -0
  115. cognee/modules/users/permissions/methods/get_principal.py +9 -0
  116. cognee/modules/users/permissions/methods/get_principal_datasets.py +11 -0
  117. cognee/modules/users/permissions/methods/get_role.py +10 -0
  118. cognee/modules/users/permissions/methods/get_specific_user_permission_datasets.py +3 -3
  119. cognee/modules/users/permissions/methods/get_tenant.py +9 -0
  120. cognee/modules/users/permissions/methods/give_default_permission_to_role.py +9 -0
  121. cognee/modules/users/permissions/methods/give_default_permission_to_tenant.py +9 -0
  122. cognee/modules/users/permissions/methods/give_default_permission_to_user.py +9 -0
  123. cognee/modules/users/permissions/methods/give_permission_on_dataset.py +10 -0
  124. cognee/modules/users/roles/methods/add_user_to_role.py +11 -0
  125. cognee/modules/users/roles/methods/create_role.py +12 -1
  126. cognee/modules/users/tenants/methods/add_user_to_tenant.py +12 -0
  127. cognee/modules/users/tenants/methods/create_tenant.py +12 -1
  128. cognee/modules/visualization/cognee_network_visualization.py +13 -9
  129. cognee/shared/data_models.py +0 -1
  130. cognee/shared/utils.py +0 -32
  131. cognee/tasks/chunk_naive_llm_classifier/chunk_naive_llm_classifier.py +2 -2
  132. cognee/tasks/codingagents/coding_rule_associations.py +3 -2
  133. cognee/tasks/entity_completion/entity_extractors/llm_entity_extractor.py +3 -2
  134. cognee/tasks/graph/cascade_extract/utils/extract_content_nodes_and_relationship_names.py +3 -2
  135. cognee/tasks/graph/cascade_extract/utils/extract_edge_triplets.py +3 -2
  136. cognee/tasks/graph/cascade_extract/utils/extract_nodes.py +3 -2
  137. cognee/tasks/graph/extract_graph_from_code.py +2 -2
  138. cognee/tasks/graph/extract_graph_from_data.py +55 -12
  139. cognee/tasks/graph/extract_graph_from_data_v2.py +16 -4
  140. cognee/tasks/ingestion/migrate_relational_database.py +132 -41
  141. cognee/tasks/ingestion/resolve_data_directories.py +4 -1
  142. cognee/tasks/schema/ingest_database_schema.py +134 -0
  143. cognee/tasks/schema/models.py +40 -0
  144. cognee/tasks/storage/index_data_points.py +1 -1
  145. cognee/tasks/storage/index_graph_edges.py +3 -1
  146. cognee/tasks/summarization/summarize_code.py +2 -2
  147. cognee/tasks/summarization/summarize_text.py +2 -2
  148. cognee/tasks/temporal_graph/enrich_events.py +2 -2
  149. cognee/tasks/temporal_graph/extract_events_and_entities.py +2 -2
  150. cognee/tests/cli_tests/cli_unit_tests/test_cli_commands.py +13 -4
  151. cognee/tests/cli_tests/cli_unit_tests/test_cli_edge_cases.py +13 -3
  152. cognee/tests/test_advanced_pdf_loader.py +141 -0
  153. cognee/tests/test_chromadb.py +40 -0
  154. cognee/tests/test_cognee_server_start.py +6 -1
  155. cognee/tests/test_data/Quantum_computers.txt +9 -0
  156. cognee/tests/test_lancedb.py +211 -0
  157. cognee/tests/test_pgvector.py +40 -0
  158. cognee/tests/test_relational_db_migration.py +76 -0
  159. cognee/tests/unit/infrastructure/databases/test_index_graph_edges.py +2 -1
  160. cognee/tests/unit/modules/ontology/test_ontology_adapter.py +330 -13
  161. cognee/tests/unit/modules/retrieval/graph_completion_retriever_context_extension_test.py +0 -4
  162. cognee/tests/unit/modules/retrieval/graph_completion_retriever_cot_test.py +0 -4
  163. cognee/tests/unit/modules/retrieval/graph_completion_retriever_test.py +0 -4
  164. {cognee-0.3.4.dev4.dist-info → cognee-0.3.6.dist-info}/METADATA +92 -96
  165. {cognee-0.3.4.dev4.dist-info → cognee-0.3.6.dist-info}/RECORD +172 -160
  166. cognee/infrastructure/data/utils/extract_keywords.py +0 -48
  167. cognee/infrastructure/databases/hybrid/falkordb/FalkorDBAdapter.py +0 -1227
  168. cognee/infrastructure/llm/structured_output_framework/baml/baml_src/extract_categories.baml +0 -109
  169. cognee/infrastructure/llm/structured_output_framework/baml/baml_src/extract_content_graph.baml +0 -343
  170. cognee/infrastructure/llm/structured_output_framework/baml/baml_src/extraction/extract_categories.py +0 -0
  171. cognee/infrastructure/llm/structured_output_framework/baml/baml_src/extraction/extract_summary.py +0 -89
  172. cognee/infrastructure/llm/structured_output_framework/baml/baml_src/extraction/knowledge_graph/__init__.py +0 -0
  173. cognee/infrastructure/llm/structured_output_framework/baml/baml_src/extraction/knowledge_graph/extract_content_graph.py +0 -44
  174. cognee/tasks/graph/infer_data_ontology.py +0 -309
  175. cognee/tests/test_falkordb.py +0 -174
  176. distributed/poetry.lock +0 -12238
  177. distributed/pyproject.toml +0 -186
  178. /cognee/infrastructure/llm/{structured_output_framework/litellm_instructor/extraction → extraction}/__init__.py +0 -0
  179. /cognee/infrastructure/llm/{structured_output_framework/litellm_instructor/extraction → extraction}/knowledge_graph/__init__.py +0 -0
  180. /cognee/infrastructure/llm/{structured_output_framework/litellm_instructor/extraction → extraction}/texts.json +0 -0
  181. {cognee-0.3.4.dev4.dist-info → cognee-0.3.6.dist-info}/WHEEL +0 -0
  182. {cognee-0.3.4.dev4.dist-info → cognee-0.3.6.dist-info}/entry_points.txt +0 -0
  183. {cognee-0.3.4.dev4.dist-info → cognee-0.3.6.dist-info}/licenses/LICENSE +0 -0
  184. {cognee-0.3.4.dev4.dist-info → cognee-0.3.6.dist-info}/licenses/NOTICE.md +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: cognee
3
- Version: 0.3.4.dev4
3
+ Version: 0.3.6
4
4
  Summary: Cognee - is a library for enriching LLM context with a semantic layer for better understanding and reasoning.
5
5
  Project-URL: Homepage, https://www.cognee.ai
6
6
  Project-URL: Repository, https://github.com/topoteretes/cognee
@@ -20,51 +20,44 @@ Requires-Dist: aiofiles<24.0.0,>=23.2.1
20
20
  Requires-Dist: aiohttp<4.0.0,>=3.11.14
21
21
  Requires-Dist: aiosqlite<1.0.0,>=0.20.0
22
22
  Requires-Dist: alembic<2,>=1.13.3
23
- Requires-Dist: baml-py<0.202.0,>=0.201.0
24
- Requires-Dist: dlt[sqlalchemy]<2,>=1.9.0
25
23
  Requires-Dist: fastapi-users[sqlalchemy]<15.0.0,>=14.0.1
26
24
  Requires-Dist: fastapi<1.0.0,>=0.115.7
25
+ Requires-Dist: fastembed<=0.6.0
27
26
  Requires-Dist: filetype<2.0.0,>=1.2.0
27
+ Requires-Dist: gunicorn<24,>=20.1.0
28
28
  Requires-Dist: instructor<2.0.0,>=1.9.1
29
29
  Requires-Dist: jinja2<4,>=3.1.3
30
- Requires-Dist: kuzu==0.11.0
30
+ Requires-Dist: kuzu==0.11.3
31
31
  Requires-Dist: lancedb<1.0.0,>=0.24.0
32
- Requires-Dist: langfuse<3,>=2.32.0
33
32
  Requires-Dist: limits<5,>=4.4.1
34
- Requires-Dist: litellm<2.0.0,>=1.71.0
35
- Requires-Dist: matplotlib<4,>=3.8.3
33
+ Requires-Dist: litellm>=1.76.0
36
34
  Requires-Dist: nbformat<6.0.0,>=5.7.0
37
35
  Requires-Dist: networkx<4,>=3.4.2
38
- Requires-Dist: nltk<4.0.0,>=3.9.1
39
36
  Requires-Dist: numpy<=4.0.0,>=1.26.4
40
- Requires-Dist: onnxruntime<2.0.0,>=1.0.0
41
- Requires-Dist: openai<2.0.0,>=1.80.1
42
- Requires-Dist: pandas<3.0.0,>=2.2.2
43
- Requires-Dist: pre-commit<5,>=4.0.1
37
+ Requires-Dist: onnxruntime<=1.22.1
38
+ Requires-Dist: openai>=1.80.1
44
39
  Requires-Dist: pydantic-settings<3,>=2.2.1
45
40
  Requires-Dist: pydantic<3.0.0,>=2.10.5
46
- Requires-Dist: pylance<1.0.0,>=0.22.0
41
+ Requires-Dist: pylance<=0.36.0,>=0.22.0
47
42
  Requires-Dist: pympler<2.0.0,>=1.1
48
43
  Requires-Dist: pypdf<7.0.0,>=4.1.0
49
44
  Requires-Dist: python-dotenv<2.0.0,>=1.0.1
50
45
  Requires-Dist: python-magic-bin<0.5; platform_system == 'Windows'
51
46
  Requires-Dist: python-multipart<1.0.0,>=0.0.20
52
47
  Requires-Dist: rdflib<7.2.0,>=7.1.4
53
- Requires-Dist: s3fs[boto3]==2025.3.2
54
- Requires-Dist: scikit-learn<2,>=1.6.1
55
- Requires-Dist: sentry-sdk[fastapi]<3,>=2.9.0
56
48
  Requires-Dist: sqlalchemy<3.0.0,>=2.0.39
57
49
  Requires-Dist: structlog<26,>=25.2.0
58
50
  Requires-Dist: tiktoken<1.0.0,>=0.8.0
59
51
  Requires-Dist: typing-extensions<5.0.0,>=4.12.2
52
+ Requires-Dist: uvicorn<1.0.0,>=0.34.0
53
+ Requires-Dist: websockets<16.0.0,>=15.0.1
60
54
  Provides-Extra: anthropic
61
55
  Requires-Dist: anthropic>=0.27; extra == 'anthropic'
62
56
  Provides-Extra: api
63
- Requires-Dist: gunicorn<24,>=20.1.0; extra == 'api'
64
- Requires-Dist: uvicorn<1.0.0,>=0.34.0; extra == 'api'
65
- Requires-Dist: websockets<16.0.0,>=15.0.1; extra == 'api'
66
57
  Provides-Extra: aws
67
58
  Requires-Dist: s3fs[boto3]==2025.3.2; extra == 'aws'
59
+ Provides-Extra: baml
60
+ Requires-Dist: baml-py==0.206.0; extra == 'baml'
68
61
  Provides-Extra: chromadb
69
62
  Requires-Dist: chromadb<0.7,>=0.6; extra == 'chromadb'
70
63
  Requires-Dist: pypika==0.48.9; extra == 'chromadb'
@@ -86,30 +79,29 @@ Requires-Dist: mkdocs-minify-plugin<0.9,>=0.8.0; extra == 'dev'
86
79
  Requires-Dist: mkdocstrings[python]<0.27,>=0.26.2; extra == 'dev'
87
80
  Requires-Dist: mypy<2,>=1.7.1; extra == 'dev'
88
81
  Requires-Dist: notebook<8,>=7.1.0; extra == 'dev'
82
+ Requires-Dist: pre-commit<5,>=4.0.1; extra == 'dev'
89
83
  Requires-Dist: pylint<4,>=3.0.3; extra == 'dev'
90
84
  Requires-Dist: pytest-asyncio<0.22,>=0.21.1; extra == 'dev'
91
85
  Requires-Dist: pytest-cov<7.0.0,>=6.1.1; extra == 'dev'
92
86
  Requires-Dist: pytest<8,>=7.4.0; extra == 'dev'
93
- Requires-Dist: ruff<1.0.0,>=0.9.2; extra == 'dev'
87
+ Requires-Dist: ruff<=0.13.1,>=0.9.2; extra == 'dev'
94
88
  Requires-Dist: tweepy<5.0.0,>=4.14.0; extra == 'dev'
95
89
  Provides-Extra: distributed
96
90
  Requires-Dist: modal<2.0.0,>=1.0.5; extra == 'distributed'
91
+ Provides-Extra: dlt
92
+ Requires-Dist: dlt[sqlalchemy]<2,>=1.9.0; extra == 'dlt'
97
93
  Provides-Extra: docs
98
- Requires-Dist: unstructured[csv,doc,docx,epub,md,odt,org,ppt,pptx,rst,rtf,tsv,xlsx]<19,>=0.18.1; extra == 'docs'
94
+ Requires-Dist: unstructured[csv,doc,docx,epub,md,odt,org,pdf,ppt,pptx,rst,rtf,tsv,xlsx]<19,>=0.18.1; extra == 'docs'
99
95
  Provides-Extra: evals
100
96
  Requires-Dist: gdown<6,>=5.2.0; extra == 'evals'
97
+ Requires-Dist: matplotlib<4,>=3.8.3; extra == 'evals'
98
+ Requires-Dist: pandas<3.0.0,>=2.2.2; extra == 'evals'
101
99
  Requires-Dist: plotly<7,>=6.0.0; extra == 'evals'
102
- Provides-Extra: falkordb
103
- Requires-Dist: falkordb<2.0.0,>=1.0.9; extra == 'falkordb'
104
- Provides-Extra: gemini
105
- Requires-Dist: google-generativeai<0.9,>=0.8.4; extra == 'gemini'
100
+ Requires-Dist: scikit-learn<2,>=1.6.1; extra == 'evals'
106
101
  Provides-Extra: graphiti
107
102
  Requires-Dist: graphiti-core<0.8,>=0.7.0; extra == 'graphiti'
108
103
  Provides-Extra: groq
109
104
  Requires-Dist: groq<1.0.0,>=0.8.0; extra == 'groq'
110
- Provides-Extra: gui
111
- Requires-Dist: pyside6<7,>=6.8.3; extra == 'gui'
112
- Requires-Dist: qasync<0.28,>=0.27.1; extra == 'gui'
113
105
  Provides-Extra: huggingface
114
106
  Requires-Dist: transformers<5,>=4.46.3; extra == 'huggingface'
115
107
  Provides-Extra: langchain
@@ -119,6 +111,9 @@ Provides-Extra: llama-index
119
111
  Requires-Dist: llama-index-core<0.13,>=0.12.11; extra == 'llama-index'
120
112
  Provides-Extra: mistral
121
113
  Requires-Dist: mistral-common<2,>=1.5.2; extra == 'mistral'
114
+ Provides-Extra: monitoring
115
+ Requires-Dist: langfuse<3,>=2.32.0; extra == 'monitoring'
116
+ Requires-Dist: sentry-sdk[fastapi]<3,>=2.9.0; extra == 'monitoring'
122
117
  Provides-Extra: neo4j
123
118
  Requires-Dist: neo4j<6,>=5.28.0; extra == 'neo4j'
124
119
  Provides-Extra: neptune
@@ -146,7 +141,7 @@ Description-Content-Type: text/markdown
146
141
 
147
142
  <br />
148
143
 
149
- cognee - Memory for AI Agents in 5 lines of code
144
+ cognee - Memory for AI Agents in 6 lines of code
150
145
 
151
146
  <p align="center">
152
147
  <a href="https://www.youtube.com/watch?v=1bezuvLwJmw&t=2s">Demo</a>
@@ -184,11 +179,9 @@ Description-Content-Type: text/markdown
184
179
 
185
180
 
186
181
 
187
- **🚀 We launched Cogwit beta (Fully-hosted AI Memory): Sign up [here](https://platform.cognee.ai/)! 🚀**
188
182
 
189
- Build dynamic memory for Agents and replace RAG using scalable, modular ECL (Extract, Cognify, Load) pipelines.
190
183
 
191
- More on [use-cases](https://docs.cognee.ai/use-cases) and [evals](https://github.com/topoteretes/cognee/tree/main/evals)
184
+ Build dynamic memory for Agents and replace RAG using scalable, modular ECL (Extract, Cognify, Load) pipelines.
192
185
 
193
186
  <p align="center">
194
187
  🌐 Available Languages
@@ -211,53 +204,50 @@ More on [use-cases](https://docs.cognee.ai/use-cases) and [evals](https://github
211
204
  </div>
212
205
 
213
206
 
214
- ## Features
215
-
216
- - Interconnect and retrieve your past conversations, documents, images and audio transcriptions
217
- - Replaces RAG systems and reduces developer effort, and cost.
218
- - Load data to graph and vector databases using only Pydantic
219
- - Manipulate your data while ingesting from 30+ data sources
220
207
 
221
208
  ## Get Started
222
209
 
223
210
  Get started quickly with a Google Colab <a href="https://colab.research.google.com/drive/1jHbWVypDgCLwjE71GSXhRL3YxYhCZzG1?usp=sharing">notebook</a> , <a href="https://deepnote.com/workspace/cognee-382213d0-0444-4c89-8265-13770e333c02/project/cognee-demo-78ffacb9-5832-4611-bb1a-560386068b30/notebook/Notebook-1-75b24cda566d4c24ab348f7150792601?utm_source=share-modal&utm_medium=product-shared-content&utm_campaign=notebook&utm_content=78ffacb9-5832-4611-bb1a-560386068b30">Deepnote notebook</a> or <a href="https://github.com/topoteretes/cognee/tree/main/cognee-starter-kit">starter repo</a>
224
211
 
225
212
 
213
+ ## About cognee
226
214
 
215
+ cognee works locally and stores your data on your device.
216
+ Our hosted solution is just our deployment of OSS cognee on Modal, with the goal of making development and productionization easier.
227
217
 
228
- ## Contributing
229
- Your contributions are at the core of making this a true open source project. Any contributions you make are **greatly appreciated**. See [`CONTRIBUTING.md`](CONTRIBUTING.md) for more information.
230
-
231
-
218
+ Self-hosted package:
232
219
 
220
+ - Interconnects any kind of documents: past conversations, files, images, and audio transcriptions
221
+ - Replaces RAG systems with a memory layer based on graphs and vectors
222
+ - Reduces developer effort and cost, while increasing quality and precision
223
+ - Provides Pythonic data pipelines that manage data ingestion from 30+ data sources
224
+ - Is highly customizable with custom tasks, pipelines, and a set of built-in search endpoints
233
225
 
226
+ Hosted platform:
227
+ - Includes a managed UI and a [hosted solution](https://www.cognee.ai)
234
228
 
235
- ## 📦 Installation
236
229
 
237
- You can install Cognee using either **pip**, **poetry**, **uv** or any other python package manager.
238
230
 
239
- Cognee supports Python 3.10 to 3.13
231
+ ## Self-Hosted (Open Source)
240
232
 
241
- ### With pip
242
233
 
243
- ```bash
244
- pip install cognee
245
- ```
234
+ ### 📦 Installation
246
235
 
247
- ## Local Cognee installation
236
+ You can install Cognee using either **pip**, **poetry**, **uv** or any other python package manager.
248
237
 
249
- You can install the local Cognee repo using **uv**, **pip** and **poetry**.
250
- For local pip installation please make sure your pip version is above version 21.3.
238
+ Cognee supports Python 3.10 to 3.12
251
239
 
252
- ### with UV with all optional dependencies
240
+ #### With uv
253
241
 
254
242
  ```bash
255
- uv sync --all-extras
243
+ uv pip install cognee
256
244
  ```
257
245
 
258
- ## 💻 Basic Usage
246
+ Detailed instructions can be found in our [docs](https://docs.cognee.ai/getting-started/installation#environment-configuration)
247
+
248
+ ### 💻 Basic Usage
259
249
 
260
- ### Setup
250
+ #### Setup
261
251
 
262
252
  ```
263
253
  import os
@@ -266,10 +256,14 @@ os.environ["LLM_API_KEY"] = "YOUR OPENAI_API_KEY"
266
256
  ```
267
257
 
268
258
  You can also set the variables by creating .env file, using our <a href="https://github.com/topoteretes/cognee/blob/main/.env.template">template.</a>
269
- To use different LLM providers, for more info check out our <a href="https://docs.cognee.ai">documentation</a>
259
+ To use different LLM providers, for more info check out our <a href="https://docs.cognee.ai/setup-configuration/llm-providers">documentation</a>
260
+
261
+
262
+ #### Simple example
263
+
270
264
 
271
265
 
272
- ### Simple example
266
+ ##### Python
273
267
 
274
268
  This script will run the default pipeline:
275
269
 
@@ -280,13 +274,16 @@ import asyncio
280
274
 
281
275
  async def main():
282
276
  # Add text to cognee
283
- await cognee.add("Natural language processing (NLP) is an interdisciplinary subfield of computer science and information retrieval.")
277
+ await cognee.add("Cognee turns documents into AI memory.")
284
278
 
285
279
  # Generate the knowledge graph
286
280
  await cognee.cognify()
287
281
 
282
+ # Add memory algorithms to the graph
283
+ await cognee.memify()
284
+
288
285
  # Query the knowledge graph
289
- results = await cognee.search("Tell me about NLP")
286
+ results = await cognee.search("What does cognee do?")
290
287
 
291
288
  # Display the results
292
289
  for result in results:
@@ -299,43 +296,38 @@ if __name__ == '__main__':
299
296
  ```
300
297
  Example output:
301
298
  ```
302
- Natural Language Processing (NLP) is a cross-disciplinary and interdisciplinary field that involves computer science and information retrieval. It focuses on the interaction between computers and human language, enabling machines to understand and process natural language.
299
+ Cognee turns documents into AI memory.
303
300
 
304
301
  ```
302
+ ##### Via CLI
305
303
 
306
- ## Our paper is out! <a href="https://arxiv.org/abs/2505.24478" target="_blank" rel="noopener noreferrer">Read here</a>
304
+ Let's get the basics covered
307
305
 
308
- <div style="text-align: center">
309
- <img src="assets/cognee-paper.png" alt="cognee paper" width="100%" />
310
- </div>
311
-
312
- </div>
313
-
314
- ## Cognee UI
306
+ ```
307
+ cognee-cli add "Cognee turns documents into AI memory."
315
308
 
316
- You can also cognify your files and query using cognee UI.
309
+ cognee-cli cognify
317
310
 
318
- <img src="assets/cognee-new-ui.webp" width="100%" alt="Cognee UI 2"></a>
311
+ cognee-cli search "What does cognee do?"
312
+ cognee-cli delete --all
319
313
 
320
- ### Installation for UI
314
+ ```
315
+ or run
316
+ ```
317
+ cognee-cli -ui
318
+ ```
321
319
 
322
- To use the cognee UI with full functionality, you need to install cognee with API dependencies:
323
320
 
324
- ```bash
325
- pip install 'cognee[api]'
326
- ```
321
+ </div>
327
322
 
328
- The UI requires backend server functionality (uvicorn and other API dependencies) which are not included in the default cognee installation to keep it lightweight.
329
323
 
330
- ### Running the UI
324
+ ### Hosted Platform
331
325
 
332
- Try cognee UI by running ``` cognee-cli -ui ``` command on your terminal.
326
+ Get up and running in minutes with automatic updates, analytics, and enterprise security.
333
327
 
334
- ## Understand our architecture
328
+ 1. Sign up on [cogwit](https://www.cognee.ai)
329
+ 2. Add your API key to local UI and sync your data to Cogwit
335
330
 
336
- <div style="text-align: center">
337
- <img src="assets/cognee_diagram.png" alt="cognee concept diagram" width="100%" />
338
- </div>
339
331
 
340
332
 
341
333
 
@@ -354,22 +346,26 @@ Try cognee UI by running ``` cognee-cli -ui ``` command on your terminal.
354
346
  [cognee with local models](https://github.com/user-attachments/assets/8621d3e8-ecb8-4860-afb2-5594f2ee17db)
355
347
 
356
348
 
357
- ## Code of Conduct
358
-
359
- We are committed to making open source an enjoyable and respectful experience for our community. See <a href="https://github.com/topoteretes/cognee/blob/main/CODE_OF_CONDUCT.md"><code>CODE_OF_CONDUCT</code></a> for more information.
360
-
361
- ## 💫 Contributors
349
+ ## Contributing
350
+ Your contributions are at the core of making this a true open source project. Any contributions you make are **greatly appreciated**. See [`CONTRIBUTING.md`](CONTRIBUTING.md) for more information.
362
351
 
363
- <a href="https://github.com/topoteretes/cognee/graphs/contributors">
364
- <img alt="contributors" src="https://contrib.rocks/image?repo=topoteretes/cognee"/>
365
- </a>
366
352
 
367
- ## Sponsors
353
+ ## Code of Conduct
368
354
 
369
- Thanks to the following companies for sponsoring the ongoing development of cognee.
355
+ We are committed to making open source an enjoyable and respectful experience for our community. See <a href="https://github.com/topoteretes/cognee/blob/main/CODE_OF_CONDUCT.md"><code>CODE_OF_CONDUCT</code></a> for more information.
370
356
 
371
- - [GitHub's Secure Open Source Fund](https://resources.github.com/github-secure-open-source-fund/)
357
+ ## Citation
372
358
 
373
- ## Star History
359
+ We now have a paper you can cite:
374
360
 
375
- [![Star History Chart](https://api.star-history.com/svg?repos=topoteretes/cognee&type=Date)](https://star-history.com/#topoteretes/cognee&Date)
361
+ ```bibtex
362
+ @misc{markovic2025optimizinginterfaceknowledgegraphs,
363
+ title={Optimizing the Interface Between Knowledge Graphs and LLMs for Complex Reasoning},
364
+ author={Vasilije Markovic and Lazar Obradovic and Laszlo Hajdu and Jovan Pavlovic},
365
+ year={2025},
366
+ eprint={2505.24478},
367
+ archivePrefix={arXiv},
368
+ primaryClass={cs.AI},
369
+ url={https://arxiv.org/abs/2505.24478},
370
+ }
371
+ ```