cognee/notebooks
lxobr 9cc357ac1c
Feat/cog 1365 unify retrievers (#572)
<!-- .github/pull_request_template.md -->

## Description
<!-- Provide a clear description of the changes in this PR -->
- Created the `BaseRetriever` class to unify all the retrievers and
searches.
- Implemented seven specialized retrievers (summaries, chunks,
completions, graph, graph-summary, insights, code) with consistent
get_context/get_completion interfaces.
- Added json context dumping feature in the current completion
implementations to enable context comparisons.
- Built a comparison framework to validate old vs new implementations.
## 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


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

- **New Features**
- Introduced multiple retrieval classes for enhanced search
capabilities, including `BaseRetriever`, `ChunksRetriever`,
`CodeRetriever`, `CompletionRetriever`, `GraphCompletionRetriever`,
`GraphSummaryCompletionRetriever`, `InsightsRetriever`, and
`SummariesRetriever`.
- Enhanced query completions with optional context saving for improved
data persistence.
- Implemented advanced tools to compare retrieval outcomes across
different implementations.

- **Refactor**
- Streamlined internal module organization and updated references for
increased maintainability and consistency.
- Added comments indicating future maintenance tasks related to code
merging.
<!-- end of auto-generated comment: release notes by coderabbit.ai -->

---------

Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>
2025-02-27 12:13:21 +01:00
..
data Cognee simple document example (#521) 2025-02-11 13:58:35 +01:00
cognee_code_graph_demo.ipynb fix: notebooks errors (#565) 2025-02-19 14:07:11 -08:00
cognee_demo.ipynb feat: improve tests add macos runners (#540) 2025-02-15 04:19:19 +01:00
cognee_graphiti_demo.ipynb Feat/cog 1365 unify retrievers (#572) 2025-02-27 12:13:21 +01:00
cognee_hotpot_eval.ipynb chore: Fix and update visualization (#518) 2025-02-11 19:25:01 +01:00
cognee_llama_index.ipynb fix: simplify code pipeline (#529) 2025-02-12 23:58:48 +01:00
cognee_multimedia_demo.ipynb refactor: Refactor search so graph completion is used by default (#505) 2025-02-07 17:16:34 +01:00
cognee_simple_demo.ipynb Cognee simple document example (#521) 2025-02-11 13:58:35 +01:00
graphrag_vs_rag.ipynb GraphRAG vs RAG notebook (#503) 2025-02-18 01:21:14 +01:00
hr_demo.ipynb chore: Fix and update visualization (#518) 2025-02-11 19:25:01 +01:00
llama_index_cognee_integration.ipynb Merge remote-tracking branch 'origin/dev' 2025-02-13 00:02:02 +01:00