From 0132be8604a4423daf970349940a8e238fb45682 Mon Sep 17 00:00:00 2001 From: vasilije Date: Sun, 6 Jul 2025 18:35:22 +0200 Subject: [PATCH] fix circular import --- .../extraction/extract_summary.py | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) 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()