gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: Rename COMPLETION to RAG_COMPLETION (#701)

Browse source 
<!-- .github/pull_request_template.md -->

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

## 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 commit is contained in:
Vasilije 2025-04-07 11:46:48 +02:00 • committed by GitHub
parent e12242b9d0
commit cd0d321eda
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
cognee/modules/search/methods/search.py
View file

@ -57,7 +57,7 @@ async def specific_search(
SearchType.SUMMARIES: SummariesRetriever().get_completion,
SearchType.INSIGHTS: InsightsRetriever().get_completion,
SearchType.CHUNKS: ChunksRetriever().get_completion,
SearchType.COMPLETION: CompletionRetriever(
SearchType.RAG_COMPLETION: CompletionRetriever(
system_prompt_path=system_prompt_path
).get_completion,
SearchType.GRAPH_COMPLETION: GraphCompletionRetriever(

2
cognee/modules/search/types/SearchType.py
View file

@ -5,7 +5,7 @@ class SearchType(Enum):
SUMMARIES = "SUMMARIES"
INSIGHTS = "INSIGHTS"
CHUNKS = "CHUNKS"
COMPLETION = "COMPLETION"
RAG_COMPLETION = "RAG_COMPLETION"
GRAPH_COMPLETION = "GRAPH_COMPLETION"
GRAPH_SUMMARY_COMPLETION = "GRAPH_SUMMARY_COMPLETION"
CODE = "CODE"

2
cognee/tests/test_custom_model.py
View file

@ -77,7 +77,7 @@ async def main():
print(completion)
# Completion query that uses document chunks to form context.
completion = await cognee.search(SearchType.COMPLETION, "What is Python?")
completion = await cognee.search(SearchType.RAG_COMPLETION, "What is Python?")
assert len(completion) != 0, "Completion search didn't return any result."
print("Completion result is:")
print(completion)

2
notebooks/graphrag_vs_rag.ipynb
View file

@ -160,7 +160,7 @@
"metadata": {},
"cell_type": "code",
"source": [
"search_results = await cognee.search(SearchType.COMPLETION, \"Tell me who are the people mentioned?\")\n",
"search_results = await cognee.search(SearchType.RAG_COMPLETION, \"Tell me who are the people mentioned?\")\n",
"\n",
"print(\"\\n\\nAnswer based on RAG:\\n\")\n",
"for result in search_results:\n",