ZengChao 1 mesiac pred
rodič
commit
f0812d4382

+ 13 - 15
core/construction_review/component/ai_review_engine.py

@@ -46,22 +46,24 @@
 └── set_review_location_label()   # 设置审查位置标签
 """
 
-import time
-import json
 import asyncio
-from enum import Enum
+import concurrent.futures
+import json
+import time
 from dataclasses import dataclass
-from typing import Dict, List, Any
-from core.base.task_models import TaskFileInfo  
-from foundation.infrastructure.config.config import config_handler
-from foundation.observability.logger.loggering import server_logger as logger
-from foundation.ai.rag.retrieval.query_rewrite import query_rewrite_manager
-from foundation.ai.rag.retrieval.entities_enhance import entity_enhance
+from enum import Enum
+from typing import Any, Dict, List, Optional, Sequence
+
+from core.base.task_models import TaskFileInfo
+from core.construction_review.component.infrastructure.milvus import MilvusConfig, MilvusManager
+from core.construction_review.component.infrastructure.relevance import is_relevant_async
 from core.construction_review.component.reviewers.base_reviewer import BaseReviewer
 from core.construction_review.component.reviewers.outline_reviewer import OutlineReviewer
 from core.construction_review.component.reviewers.utils.text_split import split_text
-from core.construction_review.component.infrastructure.milvus import MilvusManager, MilvusConfig
-
+from foundation.ai.rag.retrieval.entities_enhance import entity_enhance
+from foundation.ai.rag.retrieval.query_rewrite import query_rewrite_manager
+from foundation.infrastructure.config.config import config_handler
+from foundation.observability.logger.loggering import server_logger as logger
 
 
 @dataclass
@@ -368,10 +370,6 @@ class AIReviewEngine(BaseReviewer):
         # 使用bfp_result_list 获取 parent_id ,通过parent_id 获取父文档内容 utils_test\Milvus_Test\test_查询接口.py
         # llm 异步相关度分析  判断父文档是否与query_content 审查条文相关
         # 如果相关,则追加到 bfp_result,如果不相关则,则跳过
-        import asyncio
-        import concurrent.futures
-        from typing import Any, Dict, List, Optional, Sequence
-        from core.construction_review.component.infrastructure.relevance import is_relevant_async
         PARENT_COLLECTION = "rag_parent_hybrid"  # TODO: 改成你的父段 collection
         PARENT_TEXT_FIELD = "text"                   # TODO: 改成你的父段字段名
         PARENT_OUTPUT_FIELDS: Sequence[str] = ["parent_id", PARENT_TEXT_FIELD]