cognee 0.5.0__py3-none-any.whl → 0.5.0.dev0__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 (131) hide show
  1. cognee/api/client.py +5 -1
  2. cognee/api/v1/add/add.py +1 -2
  3. cognee/api/v1/cognify/code_graph_pipeline.py +119 -0
  4. cognee/api/v1/cognify/cognify.py +16 -24
  5. cognee/api/v1/cognify/routers/__init__.py +1 -0
  6. cognee/api/v1/cognify/routers/get_code_pipeline_router.py +90 -0
  7. cognee/api/v1/cognify/routers/get_cognify_router.py +1 -3
  8. cognee/api/v1/datasets/routers/get_datasets_router.py +3 -3
  9. cognee/api/v1/ontologies/ontologies.py +37 -12
  10. cognee/api/v1/ontologies/routers/get_ontology_router.py +25 -27
  11. cognee/api/v1/search/search.py +0 -4
  12. cognee/api/v1/ui/ui.py +68 -38
  13. cognee/context_global_variables.py +16 -61
  14. cognee/eval_framework/answer_generation/answer_generation_executor.py +0 -10
  15. cognee/eval_framework/answer_generation/run_question_answering_module.py +1 -1
  16. cognee/eval_framework/corpus_builder/task_getters/get_cascade_graph_tasks.py +2 -0
  17. cognee/eval_framework/corpus_builder/task_getters/get_default_tasks_by_indices.py +4 -4
  18. cognee/eval_framework/eval_config.py +2 -2
  19. cognee/eval_framework/modal_run_eval.py +28 -16
  20. cognee/infrastructure/databases/graph/config.py +0 -3
  21. cognee/infrastructure/databases/graph/get_graph_engine.py +0 -1
  22. cognee/infrastructure/databases/graph/graph_db_interface.py +0 -15
  23. cognee/infrastructure/databases/graph/kuzu/adapter.py +0 -228
  24. cognee/infrastructure/databases/graph/neo4j_driver/adapter.py +1 -80
  25. cognee/infrastructure/databases/utils/__init__.py +0 -3
  26. cognee/infrastructure/databases/utils/get_or_create_dataset_database.py +48 -62
  27. cognee/infrastructure/databases/vector/config.py +0 -2
  28. cognee/infrastructure/databases/vector/create_vector_engine.py +0 -1
  29. cognee/infrastructure/databases/vector/embeddings/FastembedEmbeddingEngine.py +6 -8
  30. cognee/infrastructure/databases/vector/embeddings/LiteLLMEmbeddingEngine.py +7 -9
  31. cognee/infrastructure/databases/vector/embeddings/OllamaEmbeddingEngine.py +10 -11
  32. cognee/infrastructure/databases/vector/embeddings/embedding_rate_limiter.py +544 -0
  33. cognee/infrastructure/databases/vector/lancedb/LanceDBAdapter.py +0 -2
  34. cognee/infrastructure/databases/vector/vector_db_interface.py +0 -35
  35. cognee/infrastructure/files/storage/s3_config.py +0 -2
  36. cognee/infrastructure/llm/LLMGateway.py +2 -5
  37. cognee/infrastructure/llm/config.py +0 -35
  38. cognee/infrastructure/llm/extraction/knowledge_graph/extract_content_graph.py +2 -2
  39. cognee/infrastructure/llm/structured_output_framework/baml/baml_src/extraction/acreate_structured_output.py +8 -23
  40. cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/anthropic/adapter.py +16 -17
  41. cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/gemini/adapter.py +37 -40
  42. cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/generic_llm_api/adapter.py +36 -39
  43. cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/get_llm_client.py +1 -19
  44. cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/mistral/adapter.py +9 -11
  45. cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/ollama/adapter.py +21 -23
  46. cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/openai/adapter.py +34 -42
  47. cognee/modules/cognify/config.py +0 -2
  48. cognee/modules/data/deletion/prune_system.py +2 -52
  49. cognee/modules/data/methods/delete_dataset.py +0 -26
  50. cognee/modules/engine/models/__init__.py +0 -1
  51. cognee/modules/graph/cognee_graph/CogneeGraph.py +37 -85
  52. cognee/modules/graph/cognee_graph/CogneeGraphElements.py +3 -8
  53. cognee/modules/memify/memify.py +7 -1
  54. cognee/modules/pipelines/operations/pipeline.py +2 -18
  55. cognee/modules/retrieval/__init__.py +1 -1
  56. cognee/modules/retrieval/code_retriever.py +232 -0
  57. cognee/modules/retrieval/graph_completion_context_extension_retriever.py +0 -4
  58. cognee/modules/retrieval/graph_completion_cot_retriever.py +0 -4
  59. cognee/modules/retrieval/graph_completion_retriever.py +0 -10
  60. cognee/modules/retrieval/graph_summary_completion_retriever.py +0 -4
  61. cognee/modules/retrieval/temporal_retriever.py +0 -4
  62. cognee/modules/retrieval/utils/brute_force_triplet_search.py +10 -42
  63. cognee/modules/run_custom_pipeline/run_custom_pipeline.py +1 -8
  64. cognee/modules/search/methods/get_search_type_tools.py +8 -54
  65. cognee/modules/search/methods/no_access_control_search.py +0 -4
  66. cognee/modules/search/methods/search.py +0 -21
  67. cognee/modules/search/types/SearchType.py +1 -1
  68. cognee/modules/settings/get_settings.py +0 -19
  69. cognee/modules/users/methods/get_authenticated_user.py +2 -2
  70. cognee/modules/users/models/DatasetDatabase.py +3 -15
  71. cognee/shared/logging_utils.py +0 -4
  72. cognee/tasks/code/enrich_dependency_graph_checker.py +35 -0
  73. cognee/tasks/code/get_local_dependencies_checker.py +20 -0
  74. cognee/tasks/code/get_repo_dependency_graph_checker.py +35 -0
  75. cognee/tasks/documents/__init__.py +1 -0
  76. cognee/tasks/documents/check_permissions_on_dataset.py +26 -0
  77. cognee/tasks/graph/extract_graph_from_data.py +10 -9
  78. cognee/tasks/repo_processor/__init__.py +2 -0
  79. cognee/tasks/repo_processor/get_local_dependencies.py +335 -0
  80. cognee/tasks/repo_processor/get_non_code_files.py +158 -0
  81. cognee/tasks/repo_processor/get_repo_file_dependencies.py +243 -0
  82. cognee/tasks/storage/add_data_points.py +2 -142
  83. cognee/tests/test_cognee_server_start.py +4 -2
  84. cognee/tests/test_conversation_history.py +1 -23
  85. cognee/tests/test_delete_bmw_example.py +60 -0
  86. cognee/tests/test_search_db.py +1 -37
  87. cognee/tests/unit/api/test_ontology_endpoint.py +89 -77
  88. cognee/tests/unit/infrastructure/mock_embedding_engine.py +7 -3
  89. cognee/tests/unit/infrastructure/test_embedding_rate_limiting_realistic.py +5 -0
  90. cognee/tests/unit/modules/graph/cognee_graph_elements_test.py +2 -2
  91. cognee/tests/unit/modules/graph/cognee_graph_test.py +0 -406
  92. {cognee-0.5.0.dist-info → cognee-0.5.0.dev0.dist-info}/METADATA +89 -76
  93. {cognee-0.5.0.dist-info → cognee-0.5.0.dev0.dist-info}/RECORD +97 -118
  94. {cognee-0.5.0.dist-info → cognee-0.5.0.dev0.dist-info}/WHEEL +1 -1
  95. cognee/api/v1/ui/node_setup.py +0 -360
  96. cognee/api/v1/ui/npm_utils.py +0 -50
  97. cognee/eval_framework/Dockerfile +0 -29
  98. cognee/infrastructure/databases/dataset_database_handler/__init__.py +0 -3
  99. cognee/infrastructure/databases/dataset_database_handler/dataset_database_handler_interface.py +0 -80
  100. cognee/infrastructure/databases/dataset_database_handler/supported_dataset_database_handlers.py +0 -18
  101. cognee/infrastructure/databases/dataset_database_handler/use_dataset_database_handler.py +0 -10
  102. cognee/infrastructure/databases/graph/kuzu/KuzuDatasetDatabaseHandler.py +0 -81
  103. cognee/infrastructure/databases/graph/neo4j_driver/Neo4jAuraDevDatasetDatabaseHandler.py +0 -168
  104. cognee/infrastructure/databases/utils/get_graph_dataset_database_handler.py +0 -10
  105. cognee/infrastructure/databases/utils/get_vector_dataset_database_handler.py +0 -10
  106. cognee/infrastructure/databases/utils/resolve_dataset_database_connection_info.py +0 -30
  107. cognee/infrastructure/databases/vector/lancedb/LanceDBDatasetDatabaseHandler.py +0 -50
  108. cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/bedrock/__init__.py +0 -5
  109. cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/bedrock/adapter.py +0 -153
  110. cognee/memify_pipelines/create_triplet_embeddings.py +0 -53
  111. cognee/modules/engine/models/Triplet.py +0 -9
  112. cognee/modules/retrieval/register_retriever.py +0 -10
  113. cognee/modules/retrieval/registered_community_retrievers.py +0 -1
  114. cognee/modules/retrieval/triplet_retriever.py +0 -182
  115. cognee/shared/rate_limiting.py +0 -30
  116. cognee/tasks/memify/get_triplet_datapoints.py +0 -289
  117. cognee/tests/integration/retrieval/test_triplet_retriever.py +0 -84
  118. cognee/tests/integration/tasks/test_add_data_points.py +0 -139
  119. cognee/tests/integration/tasks/test_get_triplet_datapoints.py +0 -69
  120. cognee/tests/test_dataset_database_handler.py +0 -137
  121. cognee/tests/test_dataset_delete.py +0 -76
  122. cognee/tests/test_edge_centered_payload.py +0 -170
  123. cognee/tests/test_pipeline_cache.py +0 -164
  124. cognee/tests/unit/infrastructure/llm/test_llm_config.py +0 -46
  125. cognee/tests/unit/modules/memify_tasks/test_get_triplet_datapoints.py +0 -214
  126. cognee/tests/unit/modules/retrieval/test_brute_force_triplet_search.py +0 -608
  127. cognee/tests/unit/modules/retrieval/triplet_retriever_test.py +0 -83
  128. cognee/tests/unit/tasks/storage/test_add_data_points.py +0 -288
  129. {cognee-0.5.0.dist-info → cognee-0.5.0.dev0.dist-info}/entry_points.txt +0 -0
  130. {cognee-0.5.0.dist-info → cognee-0.5.0.dev0.dist-info}/licenses/LICENSE +0 -0
  131. {cognee-0.5.0.dist-info → cognee-0.5.0.dev0.dist-info}/licenses/NOTICE.md +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: cognee
3
- Version: 0.5.0
3
+ Version: 0.5.0.dev0
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
@@ -16,9 +16,8 @@ Classifier: Operating System :: Microsoft :: Windows
16
16
  Classifier: Operating System :: POSIX :: Linux
17
17
  Classifier: Topic :: Software Development :: Libraries
18
18
  Requires-Python: <3.14,>=3.10
19
- Requires-Dist: aiofiles>=23.2.1
19
+ Requires-Dist: aiofiles<24.0.0,>=23.2.1
20
20
  Requires-Dist: aiohttp<4.0.0,>=3.11.14
21
- Requires-Dist: aiolimiter>=1.2.1
22
21
  Requires-Dist: aiosqlite<1.0.0,>=0.20.0
23
22
  Requires-Dist: alembic<2,>=1.13.3
24
23
  Requires-Dist: diskcache>=5.6.3
@@ -41,7 +40,7 @@ Requires-Dist: numpy<=4.0.0,>=1.26.4
41
40
  Requires-Dist: onnxruntime<=1.22.1
42
41
  Requires-Dist: openai>=1.80.1
43
42
  Requires-Dist: pydantic-settings<3,>=2.2.1
44
- Requires-Dist: pydantic<2.12.0,>=2.10.5
43
+ Requires-Dist: pydantic<3.0.0,>=2.10.5
45
44
  Requires-Dist: pylance<=0.36.0,>=0.22.0
46
45
  Requires-Dist: pympler<2.0.0,>=1.1
47
46
  Requires-Dist: pypdf<7.0.0,>=4.1.0
@@ -99,8 +98,7 @@ Provides-Extra: docling
99
98
  Requires-Dist: docling>=2.54; extra == 'docling'
100
99
  Requires-Dist: transformers>=4.55; extra == 'docling'
101
100
  Provides-Extra: docs
102
- Requires-Dist: lxml<5,>=4.9.3; (python_version < '3.13') and extra == 'docs'
103
- Requires-Dist: lxml<6,>=5; (python_version >= '3.13') and extra == 'docs'
101
+ Requires-Dist: lxml<6.0.0; extra == 'docs'
104
102
  Requires-Dist: unstructured[csv,doc,docx,epub,md,odt,org,pdf,ppt,pptx,rst,rtf,tsv,xlsx]<19,>=0.18.1; extra == 'docs'
105
103
  Provides-Extra: evals
106
104
  Requires-Dist: gdown<6,>=5.2.0; extra == 'evals'
@@ -147,8 +145,7 @@ Requires-Dist: redis<6.0.0,>=5.0.3; extra == 'redis'
147
145
  Provides-Extra: scraping
148
146
  Requires-Dist: apscheduler<=3.11.0,>=3.10.0; extra == 'scraping'
149
147
  Requires-Dist: beautifulsoup4>=4.13.1; extra == 'scraping'
150
- Requires-Dist: lxml<5,>=4.9.3; (python_version < '3.13') and extra == 'scraping'
151
- Requires-Dist: lxml<6,>=5; (python_version >= '3.13') and extra == 'scraping'
148
+ Requires-Dist: lxml>=4.9.3; extra == 'scraping'
152
149
  Requires-Dist: playwright>=1.9.0; extra == 'scraping'
153
150
  Requires-Dist: protego>=0.1; extra == 'scraping'
154
151
  Requires-Dist: tavily-python>=0.7.12; extra == 'scraping'
@@ -161,27 +158,27 @@ Description-Content-Type: text/markdown
161
158
 
162
159
  <br />
163
160
 
164
- Cognee - Accurate and Persistent AI Memory
161
+ cognee - Memory for AI Agents in 6 lines of code
165
162
 
166
163
  <p align="center">
167
164
  <a href="https://www.youtube.com/watch?v=1bezuvLwJmw&t=2s">Demo</a>
168
165
  .
169
- <a href="https://docs.cognee.ai/">Docs</a>
170
- .
171
- <a href="https://cognee.ai">Learn More</a>
166
+ <a href="https://cognee.ai">Learn more</a>
172
167
  ·
173
168
  <a href="https://discord.gg/NQPKmU5CCg">Join Discord</a>
174
169
  ·
175
170
  <a href="https://www.reddit.com/r/AIMemory/">Join r/AIMemory</a>
176
171
  .
177
- <a href="https://github.com/topoteretes/cognee-community">Community Plugins & Add-ons</a>
172
+ <a href="https://docs.cognee.ai/">Docs</a>
173
+ .
174
+ <a href="https://github.com/topoteretes/cognee-community">cognee community repo</a>
178
175
  </p>
179
176
 
180
177
 
181
178
  [![GitHub forks](https://img.shields.io/github/forks/topoteretes/cognee.svg?style=social&label=Fork&maxAge=2592000)](https://GitHub.com/topoteretes/cognee/network/)
182
179
  [![GitHub stars](https://img.shields.io/github/stars/topoteretes/cognee.svg?style=social&label=Star&maxAge=2592000)](https://GitHub.com/topoteretes/cognee/stargazers/)
183
180
  [![GitHub commits](https://badgen.net/github/commits/topoteretes/cognee)](https://GitHub.com/topoteretes/cognee/commit/)
184
- [![GitHub tag](https://badgen.net/github/tag/topoteretes/cognee)](https://github.com/topoteretes/cognee/tags/)
181
+ [![Github tag](https://badgen.net/github/tag/topoteretes/cognee)](https://github.com/topoteretes/cognee/tags/)
185
182
  [![Downloads](https://static.pepy.tech/badge/cognee)](https://pepy.tech/project/cognee)
186
183
  [![License](https://img.shields.io/github/license/topoteretes/cognee?colorA=00C586&colorB=000000)](https://github.com/topoteretes/cognee/blob/main/LICENSE)
187
184
  [![Contributors](https://img.shields.io/github/contributors/topoteretes/cognee?colorA=00C586&colorB=000000)](https://github.com/topoteretes/cognee/graphs/contributors)
@@ -197,7 +194,11 @@ Description-Content-Type: text/markdown
197
194
  </a>
198
195
  </p>
199
196
 
200
- Use your data to build personalized and dynamic memory for AI Agents. Cognee lets you replace RAG with scalable and modular ECL (Extract, Cognify, Load) pipelines.
197
+
198
+
199
+
200
+
201
+ Build dynamic memory for Agents and replace RAG using scalable, modular ECL (Extract, Cognify, Load) pipelines.
201
202
 
202
203
  <p align="center">
203
204
  🌐 Available Languages
@@ -205,7 +206,7 @@ Use your data to build personalized and dynamic memory for AI Agents. Cognee let
205
206
  <!-- Keep these links. Translations will automatically update with the README. -->
206
207
  <a href="https://www.readme-i18n.com/topoteretes/cognee?lang=de">Deutsch</a> |
207
208
  <a href="https://www.readme-i18n.com/topoteretes/cognee?lang=es">Español</a> |
208
- <a href="https://www.readme-i18n.com/topoteretes/cognee?lang=fr">Français</a> |
209
+ <a href="https://www.readme-i18n.com/topoteretes/cognee?lang=fr">français</a> |
209
210
  <a href="https://www.readme-i18n.com/topoteretes/cognee?lang=ja">日本語</a> |
210
211
  <a href="https://www.readme-i18n.com/topoteretes/cognee?lang=ko">한국어</a> |
211
212
  <a href="https://www.readme-i18n.com/topoteretes/cognee?lang=pt">Português</a> |
@@ -219,65 +220,69 @@ Use your data to build personalized and dynamic memory for AI Agents. Cognee let
219
220
  </div>
220
221
  </div>
221
222
 
222
- ## About Cognee
223
223
 
224
- Cognee is an open-source tool and platform that transforms your raw data into persistent and dynamic AI memory for Agents. It combines vector search with graph databases to make your documents both searchable by meaning and connected by relationships.
225
224
 
226
- You can use Cognee in two ways:
225
+ ## Get Started
226
+
227
+ Get started quickly with a Google Colab <a href="https://colab.research.google.com/drive/12Vi9zID-M3fpKpKiaqDBvkk98ElkRPWy?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>
228
+
227
229
 
228
- 1. [Self-host Cognee Open Source](https://docs.cognee.ai/getting-started/installation), which stores all data locally by default.
229
- 2. [Connect to Cognee Cloud](https://platform.cognee.ai/), and get the same OSS stack on managed infrastructure for easier development and productionization.
230
+ ## About cognee
230
231
 
231
- ### Cognee Open Source (self-hosted):
232
+ cognee works locally and stores your data on your device.
233
+ Our hosted solution is just our deployment of OSS cognee on Modal, with the goal of making development and productionization easier.
232
234
 
233
- - Interconnects any type of data — including past conversations, files, images, and audio transcriptions
234
- - Replaces traditional RAG systems with a unified memory layer built on graphs and vectors
235
- - Reduces developer effort and infrastructure cost while improving quality and precision
236
- - Provides Pythonic data pipelines for ingestion from 30+ data sources
237
- - Offers high customizability through user-defined tasks, modular pipelines, and built-in search endpoints
235
+ Self-hosted package:
238
236
 
239
- ### Cognee Cloud (managed):
240
- - Hosted web UI dashboard
241
- - Automatic version updates
242
- - Resource usage analytics
243
- - GDPR compliant, enterprise-grade security
237
+ - Interconnects any kind of documents: past conversations, files, images, and audio transcriptions
238
+ - Replaces RAG systems with a memory layer based on graphs and vectors
239
+ - Reduces developer effort and cost, while increasing quality and precision
240
+ - Provides Pythonic data pipelines that manage data ingestion from 30+ data sources
241
+ - Is highly customizable with custom tasks, pipelines, and a set of built-in search endpoints
244
242
 
245
- ## Basic Usage & Feature Guide
243
+ Hosted platform:
244
+ - Includes a managed UI and a [hosted solution](https://www.cognee.ai)
246
245
 
247
- To learn more, [check out this short, end-to-end Colab walkthrough](https://colab.research.google.com/drive/12Vi9zID-M3fpKpKiaqDBvkk98ElkRPWy?usp=sharing) of Cognee's core features.
248
246
 
249
- [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/drive/12Vi9zID-M3fpKpKiaqDBvkk98ElkRPWy?usp=sharing)
250
247
 
251
- ## Quickstart
248
+ ## Self-Hosted (Open Source)
252
249
 
253
- Let’s try Cognee in just a few lines of code. For detailed setup and configuration, see the [Cognee Docs](https://docs.cognee.ai/getting-started/installation#environment-configuration).
254
250
 
255
- ### Prerequisites
251
+ ### 📦 Installation
256
252
 
257
- - Python 3.10 to 3.13
253
+ You can install Cognee using either **pip**, **poetry**, **uv** or any other python package manager..
258
254
 
259
- ### Step 1: Install Cognee
255
+ Cognee supports Python 3.10 to 3.12
260
256
 
261
- You can install Cognee with **pip**, **poetry**, **uv**, or your preferred Python package manager.
257
+ #### With uv
262
258
 
263
259
  ```bash
264
260
  uv pip install cognee
265
261
  ```
266
262
 
267
- ### Step 2: Configure the LLM
268
- ```python
263
+ Detailed instructions can be found in our [docs](https://docs.cognee.ai/getting-started/installation#environment-configuration)
264
+
265
+ ### 💻 Basic Usage
266
+
267
+ #### Setup
268
+
269
+ ```
269
270
  import os
270
271
  os.environ["LLM_API_KEY"] = "YOUR OPENAI_API_KEY"
272
+
271
273
  ```
272
- Alternatively, create a `.env` file using our [template](https://github.com/topoteretes/cognee/blob/main/.env.template).
273
274
 
274
- To integrate other LLM providers, see our [LLM Provider Documentation](https://docs.cognee.ai/setup-configuration/llm-providers).
275
+ 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>
276
+ To use different LLM providers, for more info check out our <a href="https://docs.cognee.ai/setup-configuration/llm-providers">documentation</a>
277
+
278
+
279
+ #### Simple example
275
280
 
276
- ### Step 3: Run the Pipeline
277
281
 
278
- Cognee will take your documents, generate a knowledge graph from them and then query the graph based on combined relationships.
279
282
 
280
- Now, run a minimal pipeline:
283
+ ##### Python
284
+
285
+ This script will run the default pipeline:
281
286
 
282
287
  ```python
283
288
  import cognee
@@ -295,7 +300,7 @@ async def main():
295
300
  await cognee.memify()
296
301
 
297
302
  # Query the knowledge graph
298
- results = await cognee.search("What does Cognee do?")
303
+ results = await cognee.search("What does cognee do?")
299
304
 
300
305
  # Display the results
301
306
  for result in results:
@@ -306,61 +311,69 @@ if __name__ == '__main__':
306
311
  asyncio.run(main())
307
312
 
308
313
  ```
309
-
310
- As you can see, the output is generated from the document we previously stored in Cognee:
311
-
312
- ```bash
313
- Cognee turns documents into AI memory.
314
+ Example output:
314
315
  ```
316
+ Cognee turns documents into AI memory.
315
317
 
316
- ### Use the Cognee CLI
318
+ ```
319
+ ##### Via CLI
317
320
 
318
- As an alternative, you can get started with these essential commands:
321
+ Let's get the basics covered
319
322
 
320
- ```bash
323
+ ```
321
324
  cognee-cli add "Cognee turns documents into AI memory."
322
325
 
323
326
  cognee-cli cognify
324
327
 
325
- cognee-cli search "What does Cognee do?"
328
+ cognee-cli search "What does cognee do?"
326
329
  cognee-cli delete --all
327
330
 
328
331
  ```
329
-
330
- To open the local UI, run:
331
- ```bash
332
+ or run
333
+ ```
332
334
  cognee-cli -ui
333
335
  ```
334
336
 
335
- ## Demos & Examples
336
337
 
337
- See Cognee in action:
338
+ </div>
339
+
340
+
341
+ ### Hosted Platform
342
+
343
+ Get up and running in minutes with automatic updates, analytics, and enterprise security.
344
+
345
+ 1. Sign up on [cogwit](https://www.cognee.ai)
346
+ 2. Add your API key to local UI and sync your data to Cogwit
347
+
348
+
349
+
350
+
351
+ ## Demos
338
352
 
339
- ### Persistent Agent Memory
353
+ 1. Cogwit Beta demo:
340
354
 
341
- [Cognee Memory for LangGraph Agents](https://github.com/user-attachments/assets/e113b628-7212-4a2b-b288-0be39a93a1c3)
355
+ [Cogwit Beta](https://github.com/user-attachments/assets/fa520cd2-2913-4246-a444-902ea5242cb0)
342
356
 
343
- ### Simple GraphRAG
357
+ 2. Simple GraphRAG demo
344
358
 
345
- [Watch Demo](https://github.com/user-attachments/assets/f2186b2e-305a-42b0-9c2d-9f4473f15df8)
359
+ [Simple GraphRAG demo](https://github.com/user-attachments/assets/d80b0776-4eb9-4b8e-aa22-3691e2d44b8f)
346
360
 
347
- ### Cognee with Ollama
361
+ 3. cognee with Ollama
348
362
 
349
- [Watch Demo](https://github.com/user-attachments/assets/39672858-f774-4136-b957-1e2de67b8981)
363
+ [cognee with local models](https://github.com/user-attachments/assets/8621d3e8-ecb8-4860-afb2-5594f2ee17db)
350
364
 
351
365
 
352
- ## Community & Support
366
+ ## Contributing
367
+ 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.
353
368
 
354
- ### Contributing
355
- We welcome contributions from the community! Your input helps make Cognee better for everyone. See [`CONTRIBUTING.md`](CONTRIBUTING.md) to get started.
356
369
 
357
- ### Code of Conduct
370
+ ## Code of Conduct
358
371
 
359
- We're committed to fostering an inclusive and respectful community. Read our [Code of Conduct](https://github.com/topoteretes/cognee/blob/main/CODE_OF_CONDUCT.md) for guidelines.
372
+ 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
373
 
361
- ## Research & Citation
374
+ ## Citation
362
375
 
363
- We recently published a research paper on optimizing knowledge graphs for LLM reasoning:
376
+ We now have a paper you can cite:
364
377
 
365
378
  ```bibtex
366
379
  @misc{markovic2025optimizinginterfaceknowledgegraphs,