فهرست منبع

v0.0.3-修改launch_review接口入参格式问题,暂时关闭文档处理阶段模型校对

WangXuMing 3 ماه پیش
والد
کامیت
d2270ef09c
3فایلهای تغییر یافته به همراه14 افزوده شده و 13 حذف شده
  1. 1 1
      config/config.ini
  2. 12 12
      core/construction_review/component/document_processor.py
  3. 1 0
      views/construction_review/launch_review.py

+ 1 - 1
config/config.ini

@@ -1,7 +1,7 @@
 
 
 [model]
-MODEL_TYPE=qwen_local_14b
+MODEL_TYPE=gemini
 
 
 

+ 12 - 12
core/construction_review/component/document_processor.py

@@ -100,18 +100,18 @@ class DocumentProcessor:
             classified_items = None
             target_level = self.config.target_level
             
-            if self.llm_classifier:
-                try:
-                    logger.info(f"步骤2: 使用LLM对{target_level}级目录进行分类")
-                    classification_result = self.llm_classifier.classify(
-                        toc_info['toc_items'],
-                        target_level=target_level
-                    )
-                    if classification_result:
-                        classified_items = classification_result['items']
-                        logger.info(f"分类完成,共分类 {len(classified_items)} 个目录项")
-                except Exception as e:
-                    logger.warning(f"LLM分类失败,使用目录项直接处理: {str(e)}")
+            # if self.llm_classifier:
+            #     try:
+            #         logger.info(f"步骤2: 使用LLM对{target_level}级目录进行分类")
+            #         classification_result = self.llm_classifier.classify(
+            #             toc_info['toc_items'],
+            #             target_level=target_level
+            #         )
+            #         if classification_result:
+            #             classified_items = classification_result['items']
+            #             logger.info(f"分类完成,共分类 {len(classified_items)} 个目录项")
+            #     except Exception as e:
+            #         logger.warning(f"LLM分类失败,使用目录项直接处理: {str(e)}")
             
             # 如果没有分类结果,使用原始目录项(筛选目标层级)
             if not classified_items:

+ 1 - 0
views/construction_review/launch_review.py

@@ -89,6 +89,7 @@ def format_sse_event(event_type: str, data: str) -> str:
 class LaunchReviewRequest(BaseModel):
     """启动审查请求模型"""
     callback_task_id: str = Field(..., description="回调任务ID,从文件上传接口获取")
+    user_id: str = Field(..., description="用户标识")
     review_config: List[str] = Field(
         ...,
         description="审查配置列表,包含的项为启用状态"