Просмотр исходного кода

v0.0.4-debug
- 优化_send_start_review_progress 推送bug

WangXuMing 2 месяцев назад
Родитель
Сommit
f617ef9e43

+ 1 - 1
core/construction_review/component/reviewers/prep_basis_reviewer.py

@@ -474,7 +474,7 @@ class BasisReviewService:
             try:
                 await progress_manager.update_stage_progress(
                     callback_task_id=callback_task_id,
-                    stage_name="编制依据审查",
+                    stage_name="AI审查",
                     current=0,
                     status="processing",
                     message=f"开始编制依据审查,共{len(items)}项编制依据",

+ 12 - 13
core/construction_review/workflows/ai_review_workflow.py

@@ -282,8 +282,8 @@ class AIReviewWorkflow:
                     "messages": [AIMessage(content=f"没有可审查的单元,任务ID: {state['callback_task_id']}")]
                 }
 
-
-            await self.core_fun._send_start_review_progress(state, total_units)
+            logger.info(f"开始核心审查,任务ID: {state['callback_task_id']}")
+            await self.core_fun._send_start_review_progress(state, total_units,'core_review')
             # 2. 执行基础并发审查
             logger.info(f"开始执行并发审查,任务ID: {state['callback_task_id']}")
             successful_results = await self.core_fun._execute_concurrent_reviews(review_chunks, total_units, state)
@@ -827,22 +827,21 @@ class AIReviewCoreFun:
                         message=f"开始大纲审查",
                         event_type="processing"
                     )
-                elif  review_type is "prpe_basis":
-                    await state["progress_manager"].update_stage_progress(
-                        callback_task_id=state["callback_task_id"],
-                        stage_name="AI审查",
-                        current=0,
-                        total=total_units,
-                        status="processing",
-                        message=f"开始编制依据审查",
-                        event_type="processing"
-                    )
+                # elif  review_type is "prpe_basis":
+                #     await state["progress_manager"].update_stage_progress(
+                #         callback_task_id=state["callback_task_id"],
+                #         stage_name="AI审查",
+                #         current=0,
+                #         total=total_units,
+                #         status="processing",
+                #         message=f"开始编制依据审查",
+                #         event_type="processing"
+                #     )
                 else:
                     await state["progress_manager"].update_stage_progress(
                         callback_task_id=state["callback_task_id"],
                         stage_name="AI审查",
                         current=0,
-                        total=total_units,
                         status="processing",
                         message=f"开始核心审查,共 {total_units} 个审查单元",
                         event_type="processing"