fix: fix session tests

2026-01-15 20:34:39 +01:00 · 2026-01-15 20:34:39 +01:00 · 3820988cb3
commit 3820988cb3
parent d180e2aeae
1 changed files with 1 additions and 1 deletions
--- a/cognee/modules/retrieval/graph_completion_cot_retriever.py
+++ b/cognee/modules/retrieval/graph_completion_cot_retriever.py
@ -171,11 +171,11 @@ class GraphCompletionCotRetriever(GraphCompletionRetriever):
    async def get_completion(
        self,
        query: Optional[str] = None,
-        query_batch: Optional[List[str]] = None,
        context: Optional[List[Edge]] = None,
        session_id: Optional[str] = None,
        max_iter=4,
        response_model: Type = str,
+        query_batch: Optional[List[str]] = None,
    ) -> List[Any]:
        """
        Generate completion responses based on a user query and contextual information.