diff --git a/cognee/modules/retrieval/EntityCompletionRetriever.py b/cognee/modules/retrieval/EntityCompletionRetriever.py
index d9804647a..0901f9005 100644
--- a/cognee/modules/retrieval/EntityCompletionRetriever.py
+++ b/cognee/modules/retrieval/EntityCompletionRetriever.py
@@ -6,7 +6,10 @@ from cognee.infrastructure.entities.BaseEntityExtractor import BaseEntityExtract
 from cognee.infrastructure.context.BaseContextProvider import BaseContextProvider
 from cognee.modules.retrieval.base_retriever import BaseRetriever
 from cognee.modules.retrieval.utils.completion import generate_completion, summarize_text
-from cognee.modules.retrieval.utils.session_cache import save_to_session_cache, get_conversation_history
+from cognee.modules.retrieval.utils.session_cache import (
+    save_to_session_cache,
+    get_conversation_history,
+)
 from cognee.context_global_variables import session_user
 from cognee.infrastructure.databases.cache.config import CacheConfig
 
@@ -121,7 +124,7 @@ class EntityCompletionRetriever(BaseRetriever):
 
             if session_save:
                 conversation_history = await get_conversation_history(session_id=session_id)
-                
+
                 context_summary, completion = await asyncio.gather(
                     summarize_text(str(context)),
                     generate_completion(
diff --git a/cognee/modules/retrieval/completion_retriever.py b/cognee/modules/retrieval/completion_retriever.py
index b6fc8a48d..f72ef6d33 100644
--- a/cognee/modules/retrieval/completion_retriever.py
+++ b/cognee/modules/retrieval/completion_retriever.py
@@ -4,7 +4,10 @@ from typing import Any, Optional
 from cognee.shared.logging_utils import get_logger
 from cognee.infrastructure.databases.vector import get_vector_engine
 from cognee.modules.retrieval.utils.completion import generate_completion, summarize_text
-from cognee.modules.retrieval.utils.session_cache import save_to_session_cache, get_conversation_history
+from cognee.modules.retrieval.utils.session_cache import (
+    save_to_session_cache,
+    get_conversation_history,
+)
 from cognee.modules.retrieval.base_retriever import BaseRetriever
 from cognee.modules.retrieval.exceptions.exceptions import NoDataError
 from cognee.infrastructure.databases.vector.exceptions import CollectionNotFoundError
@@ -105,7 +108,7 @@ class CompletionRetriever(BaseRetriever):
 
         if session_save:
             conversation_history = await get_conversation_history(session_id=session_id)
-            
+
             context_summary, completion = await asyncio.gather(
                 summarize_text(context),
                 generate_completion(
diff --git a/cognee/modules/retrieval/graph_completion_context_extension_retriever.py b/cognee/modules/retrieval/graph_completion_context_extension_retriever.py
index a1f7821bf..4755910ae 100644
--- a/cognee/modules/retrieval/graph_completion_context_extension_retriever.py
+++ b/cognee/modules/retrieval/graph_completion_context_extension_retriever.py
@@ -4,7 +4,10 @@ from cognee.modules.graph.cognee_graph.CogneeGraphElements import Edge
 from cognee.shared.logging_utils import get_logger
 from cognee.modules.retrieval.graph_completion_retriever import GraphCompletionRetriever
 from cognee.modules.retrieval.utils.completion import generate_completion, summarize_text
-from cognee.modules.retrieval.utils.session_cache import save_to_session_cache, get_conversation_history
+from cognee.modules.retrieval.utils.session_cache import (
+    save_to_session_cache,
+    get_conversation_history,
+)
 from cognee.context_global_variables import session_user
 from cognee.infrastructure.databases.cache.config import CacheConfig
 
@@ -130,7 +133,7 @@ class GraphCompletionContextExtensionRetriever(GraphCompletionRetriever):
 
         if session_save:
             conversation_history = await get_conversation_history(session_id=session_id)
-            
+
             context_summary, completion = await asyncio.gather(
                 summarize_text(context_text),
                 generate_completion(
diff --git a/cognee/modules/retrieval/graph_completion_cot_retriever.py b/cognee/modules/retrieval/graph_completion_cot_retriever.py
index e95cd717d..c6755cb3d 100644
--- a/cognee/modules/retrieval/graph_completion_cot_retriever.py
+++ b/cognee/modules/retrieval/graph_completion_cot_retriever.py
@@ -5,7 +5,10 @@ from cognee.shared.logging_utils import get_logger
 
 from cognee.modules.retrieval.graph_completion_retriever import GraphCompletionRetriever
 from cognee.modules.retrieval.utils.completion import generate_completion, summarize_text
-from cognee.modules.retrieval.utils.session_cache import save_to_session_cache, get_conversation_history
+from cognee.modules.retrieval.utils.session_cache import (
+    save_to_session_cache,
+    get_conversation_history,
+)
 from cognee.infrastructure.llm.LLMGateway import LLMGateway
 from cognee.infrastructure.llm.prompts import render_prompt, read_query_prompt
 from cognee.context_global_variables import session_user
@@ -92,13 +95,13 @@ class GraphCompletionCotRetriever(GraphCompletionRetriever):
         followup_question = ""
         triplets = []
         completion = ""
-        
+
         # Retrieve conversation history if session saving is enabled
         cache_config = CacheConfig()
         user = session_user.get()
         user_id = getattr(user, "id", None)
         session_save = user_id and cache_config.caching
-        
+
         conversation_history = ""
         if session_save:
             conversation_history = await get_conversation_history(session_id=session_id)
diff --git a/cognee/modules/retrieval/temporal_retriever.py b/cognee/modules/retrieval/temporal_retriever.py
index efd4b4749..e48734b90 100644
--- a/cognee/modules/retrieval/temporal_retriever.py
+++ b/cognee/modules/retrieval/temporal_retriever.py
@@ -6,7 +6,10 @@ from typing import Any, Optional, List, Type
 from operator import itemgetter
 from cognee.infrastructure.databases.vector import get_vector_engine
 from cognee.modules.retrieval.utils.completion import generate_completion, summarize_text
-from cognee.modules.retrieval.utils.session_cache import save_to_session_cache, get_conversation_history
+from cognee.modules.retrieval.utils.session_cache import (
+    save_to_session_cache,
+    get_conversation_history,
+)
 from cognee.infrastructure.databases.graph import get_graph_engine
 from cognee.infrastructure.llm.prompts import render_prompt
 from cognee.infrastructure.llm import LLMGateway
@@ -172,7 +175,7 @@ class TemporalRetriever(GraphCompletionRetriever):
 
             if session_save:
                 conversation_history = await get_conversation_history(session_id=session_id)
-                
+
                 context_summary, completion = await asyncio.gather(
                     summarize_text(context),
                     generate_completion(