Procházet zdrojové kódy

v0.0.4-优化prep_basis_review_results响应格式

WangXuMing před 2 měsíci
rodič
revize
2057192a63

+ 1 - 1
config/config.ini

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

+ 7 - 11
core/construction_review/component/ai_review_engine.py

@@ -569,10 +569,9 @@ class AIReviewEngine(BaseReviewer):
             if not review_content or not review_content.strip():
                 logger.warning("没有可执行的编制依据审查任务")
                 return {
-                    "success": False,
                     "error_message": "编制依据内容为空,无法进行审查",
                     "review_content": review_content,
-                    "review_results": [],
+                    "prep_basis_review_results": [],
                     "total_basis_items": 0,
                     "execution_time": time.time() - start_time
                 }
@@ -590,21 +589,21 @@ class AIReviewEngine(BaseReviewer):
                     # 调用带有SSE推送功能的review_all方法
                     from core.construction_review.component.reviewers.prep_basis_reviewer import BasisReviewService
                     async with BasisReviewService(max_concurrent=max_concurrent) as service:
-                        review_results = await service.review_all(
+                        prep_basis_review_results = await service.review_all(
                             review_content,
                             collection_name="already_basis",
                             progress_manager=progress_manager,
                             callback_task_id=callback_task_id
                         )
 
-                    logger.info(f"编制依据审查完成,批次数量: {len(review_results)}")
+                    logger.info(f"编制依据审查完成,批次数量: {len(prep_basis_review_results)}")
 
                     # 统计审查结果
                     total_items = 0
                     valid_items = 0
                     standard_items = 0
 
-                    for batch in review_results:
+                    for batch in prep_basis_review_results:
                         if isinstance(batch, list):
                             total_items += len(batch)
                             for item in batch:
@@ -618,18 +617,16 @@ class AIReviewEngine(BaseReviewer):
             except Exception as e:
                 logger.error(f"编制依据异步审查失败: {str(e)}")
                 return {
-                    "success": False,
                     "error_message": f"编制依据审查失败: {str(e)}",
                     "review_content": review_content,
-                    "review_results": [],
+                    "prep_basis_review_results": [],
                     "total_basis_items": 0,
                     "execution_time": time.time() - start_time
                 }
 
             # 返回完整结果
             return {
-                "success": True,
-                "review_results": review_results,
+                "prep_basis_review_results": prep_basis_review_results,
                 "total_basis_items": total_items,
                 "valid_items": valid_items,
                 "standard_items": standard_items,
@@ -643,10 +640,9 @@ class AIReviewEngine(BaseReviewer):
             logger.error(error_msg, exc_info=True)
 
             return {
-                "success": False,
                 "error_message": error_msg,
                 "execution_time": execution_time,
-                "review_results": [],
+                "prep_basis_review_results": [],
                 "total_basis_items": 0,
                 "review_content": review_data.get('content', '')
             }