cognee 0.3.4.dev4__py3-none-any.whl → 0.3.5__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.
- cognee/api/client.py +16 -7
- cognee/api/health.py +5 -9
- cognee/api/v1/add/add.py +3 -1
- cognee/api/v1/cognify/cognify.py +44 -7
- cognee/api/v1/permissions/routers/get_permissions_router.py +8 -4
- cognee/api/v1/search/search.py +3 -0
- cognee/api/v1/ui/__init__.py +1 -1
- cognee/api/v1/ui/ui.py +215 -150
- cognee/api/v1/update/__init__.py +1 -0
- cognee/api/v1/update/routers/__init__.py +1 -0
- cognee/api/v1/update/routers/get_update_router.py +90 -0
- cognee/api/v1/update/update.py +100 -0
- cognee/base_config.py +5 -2
- cognee/cli/_cognee.py +28 -10
- cognee/cli/commands/delete_command.py +34 -2
- cognee/eval_framework/corpus_builder/task_getters/get_default_tasks_by_indices.py +2 -2
- cognee/eval_framework/evaluation/direct_llm_eval_adapter.py +3 -2
- cognee/eval_framework/modal_eval_dashboard.py +9 -1
- cognee/infrastructure/databases/graph/config.py +9 -9
- cognee/infrastructure/databases/graph/get_graph_engine.py +4 -21
- cognee/infrastructure/databases/graph/kuzu/adapter.py +60 -9
- cognee/infrastructure/databases/hybrid/neptune_analytics/NeptuneAnalyticsAdapter.py +3 -3
- cognee/infrastructure/databases/relational/config.py +4 -4
- cognee/infrastructure/databases/relational/create_relational_engine.py +11 -3
- cognee/infrastructure/databases/vector/chromadb/ChromaDBAdapter.py +7 -3
- cognee/infrastructure/databases/vector/config.py +7 -7
- cognee/infrastructure/databases/vector/create_vector_engine.py +7 -15
- cognee/infrastructure/databases/vector/embeddings/EmbeddingEngine.py +9 -0
- cognee/infrastructure/databases/vector/embeddings/FastembedEmbeddingEngine.py +11 -0
- cognee/infrastructure/databases/vector/embeddings/LiteLLMEmbeddingEngine.py +19 -2
- cognee/infrastructure/databases/vector/embeddings/OllamaEmbeddingEngine.py +11 -0
- cognee/infrastructure/databases/vector/embeddings/config.py +8 -0
- cognee/infrastructure/databases/vector/embeddings/get_embedding_engine.py +5 -0
- cognee/infrastructure/databases/vector/lancedb/LanceDBAdapter.py +11 -10
- cognee/infrastructure/databases/vector/pgvector/PGVectorAdapter.py +48 -38
- cognee/infrastructure/databases/vector/vector_db_interface.py +8 -4
- cognee/infrastructure/files/storage/S3FileStorage.py +15 -5
- cognee/infrastructure/files/storage/s3_config.py +1 -0
- cognee/infrastructure/files/utils/open_data_file.py +7 -14
- cognee/infrastructure/llm/LLMGateway.py +19 -117
- cognee/infrastructure/llm/config.py +28 -13
- cognee/infrastructure/llm/{structured_output_framework/litellm_instructor/extraction → extraction}/extract_categories.py +2 -1
- cognee/infrastructure/llm/{structured_output_framework/litellm_instructor/extraction → extraction}/extract_event_entities.py +3 -2
- cognee/infrastructure/llm/{structured_output_framework/litellm_instructor/extraction → extraction}/extract_summary.py +3 -2
- cognee/infrastructure/llm/{structured_output_framework/litellm_instructor/extraction → extraction}/knowledge_graph/extract_content_graph.py +2 -1
- cognee/infrastructure/llm/{structured_output_framework/litellm_instructor/extraction → extraction}/knowledge_graph/extract_event_graph.py +3 -2
- cognee/infrastructure/llm/prompts/read_query_prompt.py +3 -2
- cognee/infrastructure/llm/prompts/show_prompt.py +35 -0
- cognee/infrastructure/llm/prompts/test.txt +1 -0
- cognee/infrastructure/llm/structured_output_framework/baml/baml_client/__init__.py +2 -2
- cognee/infrastructure/llm/structured_output_framework/baml/baml_client/async_client.py +50 -397
- cognee/infrastructure/llm/structured_output_framework/baml/baml_client/inlinedbaml.py +2 -3
- cognee/infrastructure/llm/structured_output_framework/baml/baml_client/parser.py +8 -88
- cognee/infrastructure/llm/structured_output_framework/baml/baml_client/runtime.py +78 -0
- cognee/infrastructure/llm/structured_output_framework/baml/baml_client/stream_types.py +2 -99
- cognee/infrastructure/llm/structured_output_framework/baml/baml_client/sync_client.py +49 -401
- cognee/infrastructure/llm/structured_output_framework/baml/baml_client/type_builder.py +19 -882
- cognee/infrastructure/llm/structured_output_framework/baml/baml_client/type_map.py +2 -34
- cognee/infrastructure/llm/structured_output_framework/baml/baml_client/types.py +2 -107
- cognee/infrastructure/llm/structured_output_framework/baml/baml_src/acreate_structured_output.baml +26 -0
- cognee/infrastructure/llm/structured_output_framework/baml/baml_src/extraction/__init__.py +1 -2
- cognee/infrastructure/llm/structured_output_framework/baml/baml_src/extraction/acreate_structured_output.py +76 -0
- cognee/infrastructure/llm/structured_output_framework/baml/baml_src/extraction/create_dynamic_baml_type.py +122 -0
- cognee/infrastructure/llm/structured_output_framework/baml/baml_src/generators.baml +3 -3
- cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/anthropic/adapter.py +0 -32
- cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/gemini/adapter.py +107 -98
- cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/generic_llm_api/adapter.py +5 -6
- cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/get_llm_client.py +5 -6
- cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/llm_interface.py +0 -26
- cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/openai/adapter.py +17 -67
- cognee/infrastructure/llm/tokenizer/Gemini/adapter.py +8 -7
- cognee/infrastructure/llm/utils.py +4 -4
- cognee/infrastructure/loaders/LoaderEngine.py +5 -2
- cognee/infrastructure/loaders/external/__init__.py +7 -0
- cognee/infrastructure/loaders/external/advanced_pdf_loader.py +244 -0
- cognee/infrastructure/loaders/supported_loaders.py +7 -0
- cognee/modules/data/methods/create_authorized_dataset.py +9 -0
- cognee/modules/data/methods/get_authorized_dataset.py +1 -1
- cognee/modules/data/methods/get_authorized_dataset_by_name.py +11 -0
- cognee/modules/data/methods/get_deletion_counts.py +92 -0
- cognee/modules/graph/cognee_graph/CogneeGraph.py +1 -1
- cognee/modules/graph/utils/expand_with_nodes_and_edges.py +22 -8
- cognee/modules/graph/utils/retrieve_existing_edges.py +0 -2
- cognee/modules/ingestion/data_types/TextData.py +0 -1
- cognee/modules/observability/get_observe.py +14 -0
- cognee/modules/observability/observers.py +1 -0
- cognee/modules/ontology/base_ontology_resolver.py +42 -0
- cognee/modules/ontology/get_default_ontology_resolver.py +41 -0
- cognee/modules/ontology/matching_strategies.py +53 -0
- cognee/modules/ontology/models.py +20 -0
- cognee/modules/ontology/ontology_config.py +24 -0
- cognee/modules/ontology/ontology_env_config.py +45 -0
- cognee/modules/ontology/rdf_xml/{OntologyResolver.py → RDFLibOntologyResolver.py} +20 -28
- cognee/modules/pipelines/layers/resolve_authorized_user_dataset.py +21 -24
- cognee/modules/pipelines/layers/resolve_authorized_user_datasets.py +3 -3
- cognee/modules/retrieval/code_retriever.py +2 -1
- cognee/modules/retrieval/context_providers/TripletSearchContextProvider.py +1 -4
- cognee/modules/retrieval/graph_completion_cot_retriever.py +6 -5
- cognee/modules/retrieval/graph_completion_retriever.py +0 -3
- cognee/modules/retrieval/insights_retriever.py +1 -1
- cognee/modules/retrieval/jaccard_retrival.py +60 -0
- cognee/modules/retrieval/lexical_retriever.py +123 -0
- cognee/modules/retrieval/natural_language_retriever.py +2 -1
- cognee/modules/retrieval/temporal_retriever.py +3 -2
- cognee/modules/retrieval/utils/brute_force_triplet_search.py +2 -12
- cognee/modules/retrieval/utils/completion.py +4 -7
- cognee/modules/search/methods/get_search_type_tools.py +7 -0
- cognee/modules/search/methods/no_access_control_search.py +1 -1
- cognee/modules/search/methods/search.py +32 -13
- cognee/modules/search/types/SearchType.py +1 -0
- cognee/modules/users/permissions/methods/authorized_give_permission_on_datasets.py +12 -0
- cognee/modules/users/permissions/methods/check_permission_on_dataset.py +11 -0
- cognee/modules/users/permissions/methods/get_all_user_permission_datasets.py +10 -0
- cognee/modules/users/permissions/methods/get_document_ids_for_user.py +10 -0
- cognee/modules/users/permissions/methods/get_principal.py +9 -0
- cognee/modules/users/permissions/methods/get_principal_datasets.py +11 -0
- cognee/modules/users/permissions/methods/get_role.py +10 -0
- cognee/modules/users/permissions/methods/get_specific_user_permission_datasets.py +3 -3
- cognee/modules/users/permissions/methods/get_tenant.py +9 -0
- cognee/modules/users/permissions/methods/give_default_permission_to_role.py +9 -0
- cognee/modules/users/permissions/methods/give_default_permission_to_tenant.py +9 -0
- cognee/modules/users/permissions/methods/give_default_permission_to_user.py +9 -0
- cognee/modules/users/permissions/methods/give_permission_on_dataset.py +10 -0
- cognee/modules/users/roles/methods/add_user_to_role.py +11 -0
- cognee/modules/users/roles/methods/create_role.py +12 -1
- cognee/modules/users/tenants/methods/add_user_to_tenant.py +12 -0
- cognee/modules/users/tenants/methods/create_tenant.py +12 -1
- cognee/modules/visualization/cognee_network_visualization.py +13 -9
- cognee/shared/data_models.py +0 -1
- cognee/shared/utils.py +0 -32
- cognee/tasks/chunk_naive_llm_classifier/chunk_naive_llm_classifier.py +2 -2
- cognee/tasks/codingagents/coding_rule_associations.py +3 -2
- cognee/tasks/entity_completion/entity_extractors/llm_entity_extractor.py +3 -2
- cognee/tasks/graph/cascade_extract/utils/extract_content_nodes_and_relationship_names.py +3 -2
- cognee/tasks/graph/cascade_extract/utils/extract_edge_triplets.py +3 -2
- cognee/tasks/graph/cascade_extract/utils/extract_nodes.py +3 -2
- cognee/tasks/graph/extract_graph_from_code.py +2 -2
- cognee/tasks/graph/extract_graph_from_data.py +55 -12
- cognee/tasks/graph/extract_graph_from_data_v2.py +16 -4
- cognee/tasks/ingestion/migrate_relational_database.py +132 -41
- cognee/tasks/ingestion/resolve_data_directories.py +4 -1
- cognee/tasks/schema/ingest_database_schema.py +134 -0
- cognee/tasks/schema/models.py +40 -0
- cognee/tasks/storage/index_data_points.py +1 -1
- cognee/tasks/storage/index_graph_edges.py +3 -1
- cognee/tasks/summarization/summarize_code.py +2 -2
- cognee/tasks/summarization/summarize_text.py +2 -2
- cognee/tasks/temporal_graph/enrich_events.py +2 -2
- cognee/tasks/temporal_graph/extract_events_and_entities.py +2 -2
- cognee/tests/cli_tests/cli_unit_tests/test_cli_commands.py +13 -4
- cognee/tests/cli_tests/cli_unit_tests/test_cli_edge_cases.py +13 -3
- cognee/tests/test_advanced_pdf_loader.py +141 -0
- cognee/tests/test_chromadb.py +40 -0
- cognee/tests/test_cognee_server_start.py +6 -1
- cognee/tests/test_data/Quantum_computers.txt +9 -0
- cognee/tests/test_lancedb.py +211 -0
- cognee/tests/test_pgvector.py +40 -0
- cognee/tests/test_relational_db_migration.py +76 -0
- cognee/tests/unit/infrastructure/databases/test_index_graph_edges.py +2 -1
- cognee/tests/unit/modules/ontology/test_ontology_adapter.py +330 -13
- cognee/tests/unit/modules/retrieval/graph_completion_retriever_context_extension_test.py +0 -4
- cognee/tests/unit/modules/retrieval/graph_completion_retriever_cot_test.py +0 -4
- cognee/tests/unit/modules/retrieval/graph_completion_retriever_test.py +0 -4
- {cognee-0.3.4.dev4.dist-info → cognee-0.3.5.dist-info}/METADATA +92 -96
- {cognee-0.3.4.dev4.dist-info → cognee-0.3.5.dist-info}/RECORD +173 -159
- distributed/pyproject.toml +0 -1
- cognee/infrastructure/data/utils/extract_keywords.py +0 -48
- cognee/infrastructure/databases/hybrid/falkordb/FalkorDBAdapter.py +0 -1227
- cognee/infrastructure/llm/structured_output_framework/baml/baml_src/extract_categories.baml +0 -109
- cognee/infrastructure/llm/structured_output_framework/baml/baml_src/extract_content_graph.baml +0 -343
- cognee/infrastructure/llm/structured_output_framework/baml/baml_src/extraction/extract_categories.py +0 -0
- cognee/infrastructure/llm/structured_output_framework/baml/baml_src/extraction/extract_summary.py +0 -89
- cognee/infrastructure/llm/structured_output_framework/baml/baml_src/extraction/knowledge_graph/__init__.py +0 -0
- cognee/infrastructure/llm/structured_output_framework/baml/baml_src/extraction/knowledge_graph/extract_content_graph.py +0 -44
- cognee/tasks/graph/infer_data_ontology.py +0 -309
- cognee/tests/test_falkordb.py +0 -174
- /cognee/infrastructure/llm/{structured_output_framework/litellm_instructor/extraction → extraction}/__init__.py +0 -0
- /cognee/infrastructure/llm/{structured_output_framework/litellm_instructor/extraction → extraction}/knowledge_graph/__init__.py +0 -0
- /cognee/infrastructure/llm/{structured_output_framework/litellm_instructor/extraction → extraction}/texts.json +0 -0
- {cognee-0.3.4.dev4.dist-info → cognee-0.3.5.dist-info}/WHEEL +0 -0
- {cognee-0.3.4.dev4.dist-info → cognee-0.3.5.dist-info}/entry_points.txt +0 -0
- {cognee-0.3.4.dev4.dist-info → cognee-0.3.5.dist-info}/licenses/LICENSE +0 -0
- {cognee-0.3.4.dev4.dist-info → cognee-0.3.5.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.
|
|
3
|
+
Version: 0.3.5
|
|
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.
|
|
30
|
+
Requires-Dist: kuzu==0.11.2
|
|
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
|
|
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
|
|
41
|
-
Requires-Dist: openai
|
|
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
|
|
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
|
|
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
|
-
|
|
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
|
|
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
|
-
|
|
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
|
-
|
|
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
|
-
|
|
231
|
+
## Self-Hosted (Open Source)
|
|
240
232
|
|
|
241
|
-
### With pip
|
|
242
233
|
|
|
243
|
-
|
|
244
|
-
pip install cognee
|
|
245
|
-
```
|
|
234
|
+
### 📦 Installation
|
|
246
235
|
|
|
247
|
-
|
|
236
|
+
You can install Cognee using either **pip**, **poetry**, **uv** or any other python package manager.
|
|
248
237
|
|
|
249
|
-
|
|
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
|
-
|
|
240
|
+
#### With uv
|
|
253
241
|
|
|
254
242
|
```bash
|
|
255
|
-
uv
|
|
243
|
+
uv pip install cognee
|
|
256
244
|
```
|
|
257
245
|
|
|
258
|
-
|
|
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
|
-
|
|
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
|
-
|
|
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("
|
|
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("
|
|
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
|
-
|
|
299
|
+
Cognee turns documents into AI memory.
|
|
303
300
|
|
|
304
301
|
```
|
|
302
|
+
##### Via CLI
|
|
305
303
|
|
|
306
|
-
|
|
304
|
+
Let's get the basics covered
|
|
307
305
|
|
|
308
|
-
|
|
309
|
-
|
|
310
|
-
</div>
|
|
311
|
-
|
|
312
|
-
</div>
|
|
313
|
-
|
|
314
|
-
## Cognee UI
|
|
306
|
+
```
|
|
307
|
+
cognee-cli add "Cognee turns documents into AI memory."
|
|
315
308
|
|
|
316
|
-
|
|
309
|
+
cognee-cli cognify
|
|
317
310
|
|
|
318
|
-
|
|
311
|
+
cognee-cli search "What does cognee do?"
|
|
312
|
+
cognee-cli delete --all
|
|
319
313
|
|
|
320
|
-
|
|
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
|
-
|
|
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
|
-
###
|
|
324
|
+
### Hosted Platform
|
|
331
325
|
|
|
332
|
-
|
|
326
|
+
Get up and running in minutes with automatic updates, analytics, and enterprise security.
|
|
333
327
|
|
|
334
|
-
|
|
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
|
-
##
|
|
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
|
-
##
|
|
353
|
+
## Code of Conduct
|
|
368
354
|
|
|
369
|
-
|
|
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
|
-
|
|
357
|
+
## Citation
|
|
372
358
|
|
|
373
|
-
|
|
359
|
+
We now have a paper you can cite:
|
|
374
360
|
|
|
375
|
-
|
|
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
|
+
```
|