From 96f2a2f22bff2265d3ad534eda8adb225de6ff79 Mon Sep 17 00:00:00 2001 From: hajdul88 <52442977+hajdul88@users.noreply.github.com> Date: Thu, 16 Oct 2025 15:54:48 +0200 Subject: [PATCH] ruff format --- cognee/modules/retrieval/graph_completion_cot_retriever.py | 2 +- cognee/modules/retrieval/graph_completion_retriever.py | 1 + cognee/modules/retrieval/utils/session_cache.py | 1 - 3 files changed, 2 insertions(+), 2 deletions(-) diff --git a/cognee/modules/retrieval/graph_completion_cot_retriever.py b/cognee/modules/retrieval/graph_completion_cot_retriever.py index aa221a9b0..2a44e3b15 100644 --- a/cognee/modules/retrieval/graph_completion_cot_retriever.py +++ b/cognee/modules/retrieval/graph_completion_cot_retriever.py @@ -150,7 +150,7 @@ class GraphCompletionCotRetriever(GraphCompletionRetriever): cache_config = CacheConfig() user = session_user.get() user_id = getattr(user, "id", None) - + if user_id and cache_config.caching: context_summary = await summarize_text(context_text) await save_to_session_cache( diff --git a/cognee/modules/retrieval/graph_completion_retriever.py b/cognee/modules/retrieval/graph_completion_retriever.py index 61b7bb444..d1d9d5092 100644 --- a/cognee/modules/retrieval/graph_completion_retriever.py +++ b/cognee/modules/retrieval/graph_completion_retriever.py @@ -19,6 +19,7 @@ from cognee.modules.engine.models.node_set import NodeSet from cognee.infrastructure.databases.graph import get_graph_engine from cognee.context_global_variables import session_user from cognee.infrastructure.databases.cache.config import CacheConfig + logger = get_logger("GraphCompletionRetriever") diff --git a/cognee/modules/retrieval/utils/session_cache.py b/cognee/modules/retrieval/utils/session_cache.py index bc0c25536..5ffacb1ab 100644 --- a/cognee/modules/retrieval/utils/session_cache.py +++ b/cognee/modules/retrieval/utils/session_cache.py @@ -45,4 +45,3 @@ async def save_to_session_cache( context=context_summary, answer=answer, ) -