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, ) -