|
|
@@ -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]
|