Sfoglia il codice sorgente

v0.0.3-sse
- check_result json规则优化2.0

WangXuMing 2 mesi fa
parent
commit
9acd8523f8
3 ha cambiato i file con 4809 aggiunte e 1 eliminazioni
  1. 9 0
      core/base/progress_manager.py
  2. 4799 0
      logs/agent_info.log.1
  3. 1 1
      temp/AI审查结果.json

+ 9 - 0
core/base/progress_manager.py

@@ -50,6 +50,12 @@ class SSECallbackManager:
         self._callbacks.clear()
         logger.info("已清空所有SSE回调")
 
+    def force_close_sse(self, callback_task_id: str):
+        """强制关闭SSE连接"""
+        if callback_task_id in self._callbacks:
+            del self._callbacks[callback_task_id]
+            logger.info(f"强制关闭SSE连接: {callback_task_id}")
+
 sse_callback_manager = SSECallbackManager()
 
 class ProgressManager:
@@ -314,6 +320,9 @@ class ProgressManager:
             # 取消SSE回调注册,避免重复推送
             sse_callback_manager.unregister_callback(callback_task_id)
 
+            # 强制关闭SSE连接
+            sse_callback_manager.force_close_sse(callback_task_id)
+
             logger.info(f"任务关闭: {callback_task_id}")
         except Exception as e:
             logger.error(f"标记任务完成失败: {str(e)}")

File diff suppressed because it is too large
+ 4799 - 0
logs/agent_info.log.1


File diff suppressed because it is too large
+ 1 - 1
temp/AI审查结果.json


Some files were not shown because too many files changed in this diff