diff --git a/cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/extraction/extract_summary.py b/cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/extraction/extract_summary.py
index a18ae8b29..c0eb45353 100644
--- a/cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/extraction/extract_summary.py
+++ b/cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/extraction/extract_summary.py
@@ -12,11 +12,23 @@ from cognee.infrastructure.llm.structured_output_framework.llitellm_instructor.l
     read_query_prompt,
 )
 from cognee.shared.data_models import SummarizedCode
-from cognee.tasks.summarization.mock_summary import get_mock_summarized_code
 
 logger = get_logger("extract_summary")
 
 
+def get_mock_summarized_code():
+    """Local mock function to avoid circular imports."""
+    return SummarizedCode(
+        high_level_summary="Mock code summary",
+        key_features=["Mock feature 1", "Mock feature 2"],
+        imports=["mock_import"],
+        constants=["MOCK_CONSTANT"],
+        classes=[],
+        functions=[],
+        workflow_description="Mock workflow description",
+    )
+
+
 async def extract_summary(content: str, response_model: Type[BaseModel]):
     llm_client = get_llm_client()