cognee/.github/workflows
Dmitrii Galkin e147fa5bde
feat: Add support for ChromaDB (#622)
<!-- .github/pull_request_template.md -->

## Description
<!-- Provide a clear description of the changes in this PR -->

# Add Support for ChromaDB

## Summary
This PR adds support for ChromaDB as a vector database option in the
Cognee application. ChromaDB is a modern, open-source embedding database
designed for AI applications.

## Changes
- Created a new ChromaDBAdapter implementation for vector database
operations
- Added comprehensive test suite for ChromaDB functionality
- Updated docker-compose.yml to include ChromaDB service
- Modified environment configuration to support ChromaDB settings
- Updated vector engine creation logic to support ChromaDB as an option

## Technical Details
- Implemented `ChromaDBAdapter.py` (347 lines) with full CRUD operations
for vector data
- Created test suite (`test_chromadb.py`) with 171 lines of test
coverage
- Updated vector engine creation process to dynamically select ChromaDB
when configured
- Modified settings router to accommodate new database option
- Updated environment template with ChromaDB configuration options

## Docker Changes
- Added ChromaDB service to docker-compose.yml with appropriate
configuration

This PR enhances Cognee's flexibility by providing an alternative vector
database option, allowing users to choose the most appropriate database
for their specific use case.



## DCO Affirmation
I affirm that all code in every commit of this pull request conforms to
the terms of the Topoteretes Developer Certificate of Origin

Tested with UI + tests.


<!-- This is an auto-generated comment: release notes by coderabbit.ai
-->
## Summary by CodeRabbit

- **New Features**
- Expanded vector database integration by adding support for Chromadb,
enabling enhanced data management and search functionalities.
- **Tests**
- Added automated tests to validate the Chromadb integration and related
operations.
- **Chores**
- Updated configuration guidance and dependency management to include
Chromadb.
  - Provided an optional container deployment template for Chromadb.
- Added a new entry to ignore the `.chromadb_data/` directory in version
control.
- Introduced a new GitHub Actions workflow for testing Chromadb
integration.
<!-- end of auto-generated comment: release notes by coderabbit.ai -->

---------

Co-authored-by: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Co-authored-by: Vasilije <8619304+Vasilije1990@users.noreply.github.com>
2025-03-13 15:13:04 +01:00
..
approve_dco.yaml Update approve_dco.yaml 2025-02-09 15:47:21 +01:00
cd.yaml feat: implement unit tests and extensive checks around the get_graph_from_model [COG-754] (#491) 2025-01-31 18:17:23 +01:00
cd_prd.yaml Version 0.1.21 (#431) 2025-01-10 19:37:50 +01:00
ci.yaml Version 0.1.21 (#431) 2025-01-10 19:37:50 +01:00
clean_stale_pr.yaml Update clean_stale_pr.yaml 2025-02-09 15:40:03 +01:00
community_greetings.yml Version 0.1.21 (#431) 2025-01-10 19:37:50 +01:00
docker_compose.yml Version 0.1.21 (#431) 2025-01-10 19:37:50 +01:00
dockerhub-mcp.yml fix: change context of docker build 2025-02-26 20:22:07 +01:00
dockerhub.yml feat: codegraph improvements and new CODE search [COG-1351] (#581) 2025-02-26 20:15:02 +01:00
pr_lint Version 0.1.21 (#431) 2025-01-10 19:37:50 +01:00
py_lint.yml Version 0.1.21 (#431) 2025-01-10 19:37:50 +01:00
release_discord_action.yml Version 0.1.21 (#431) 2025-01-10 19:37:50 +01:00
reusable_notebook.yml feat: Add gemini support [COG-1023] (#485) 2025-01-31 18:03:23 +01:00
reusable_python_example.yml chore: Change Code graph gh action to use OpenAI API (#499) 2025-02-05 22:20:07 +01:00
ruff_format.yaml Update ruff_format.yaml (#513) 2025-02-09 15:48:19 +01:00
ruff_lint.yaml Update ruff_lint.yaml 2025-02-09 15:47:45 +01:00
test_chromadb.yml feat: Add support for ChromaDB (#622) 2025-03-13 15:13:04 +01:00
test_code_graph_example.yml fix: simplify code pipeline (#529) 2025-02-12 23:58:48 +01:00
test_cognee_llama_index_notebook.yml feat: Add gemini support [COG-1023] (#485) 2025-01-31 18:03:23 +01:00
test_cognee_multimedia_notebook.yml feat: Add gemini support [COG-1023] (#485) 2025-01-31 18:03:23 +01:00
test_cognee_server_start.yml feat: user authorization [COG-1189] (#593) 2025-03-13 13:33:42 +01:00
test_deduplication.yml feat: Add gemini support [COG-1023] (#485) 2025-01-31 18:03:23 +01:00
test_descriptive_graph_metrics.yml feat: Implement optional neo4j metrics and improve tests [cog-1262] (#556) 2025-02-19 16:24:59 +01:00
test_dynamic_steps_example.yml refactor: Refactor search so graph completion is used by default (#505) 2025-02-07 17:16:34 +01:00
test_dynamic_steps_example_windows.yml feat: Add gemini support [COG-1023] (#485) 2025-01-31 18:03:23 +01:00
test_eval_framework.yml Feature/cog 1312 integrating evaluation framework into dreamify (#562) 2025-03-03 19:55:47 +01:00
test_gemini.yml feat: Draft ollama test (#566) 2025-02-28 20:15:12 +01:00
test_graphrag_vs_rag_notebook.yml GraphRAG vs RAG notebook (#503) 2025-02-18 01:21:14 +01:00
test_llama_index_cognee_integration_notebook.yml fix: Resolve llama-index integration issue with new cognee version [ COG-1270] (#515) 2025-02-11 18:28:31 +01:00
test_milvus.yml feat: Add gemini support [COG-1023] (#485) 2025-01-31 18:03:23 +01:00
test_multimedia_example.yaml refactor: Refactor search so graph completion is used by default (#505) 2025-02-07 17:16:34 +01:00
test_multimetric_qa_eval_run.yaml feat: Add gemini support [COG-1023] (#485) 2025-01-31 18:03:23 +01:00
test_neo4j.yml feat: Add gemini support [COG-1023] (#485) 2025-01-31 18:03:23 +01:00
test_notebook.yml feat: Add gemini support [COG-1023] (#485) 2025-01-31 18:03:23 +01:00
test_ollama.yml feat: Draft ollama test (#566) 2025-02-28 20:15:12 +01:00
test_pgvector.yml feat: Add gemini support [COG-1023] (#485) 2025-01-31 18:03:23 +01:00
test_python_3_10.yml Test: Parse context pieces separately in MusiqueQAAdapter and adjust tests [cog-1234] (#561) 2025-02-20 14:23:53 +01:00
test_python_3_11.yml Test: Parse context pieces separately in MusiqueQAAdapter and adjust tests [cog-1234] (#561) 2025-02-20 14:23:53 +01:00
test_python_3_12.yml Test: Parse context pieces separately in MusiqueQAAdapter and adjust tests [cog-1234] (#561) 2025-02-20 14:23:53 +01:00
test_qdrant.yml feat: Add gemini support [COG-1023] (#485) 2025-01-31 18:03:23 +01:00
test_simple_example.yml refactor: Refactor search so graph completion is used by default (#505) 2025-02-07 17:16:34 +01:00
test_telemetry.yml test: Add testing of cognee telemetry (#573) 2025-02-27 13:23:16 +01:00
test_weaviate.yml feat: Add gemini support [COG-1023] (#485) 2025-01-31 18:03:23 +01:00
update-contributors.yml Create update-contributors.yml 2025-02-09 15:45:03 +01:00