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

fix: fix session tests

Browse source
This commit is contained in:
Andrej Milicevic 2026-01-15 20:34:39 +01:00
parent d180e2aeae
commit 3820988cb3
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
cognee/modules/retrieval/graph_completion_cot_retriever.py
View file

@ -171,11 +171,11 @@ class GraphCompletionCotRetriever(GraphCompletionRetriever):
async def get_completion( async def get_completion(
self, self,
query: Optional[str] = None, query: Optional[str] = None,
query_batch: Optional[List[str]] = None,
context: Optional[List[Edge]] = None, context: Optional[List[Edge]] = None,
session_id: Optional[str] = None, session_id: Optional[str] = None,
max_iter=4, max_iter=4,
response_model: Type = str, response_model: Type = str,
query_batch: Optional[List[str]] = None,
) -> List[Any]: ) -> List[Any]:
""" """
Generate completion responses based on a user query and contextual information. Generate completion responses based on a user query and contextual information.