jarvis-ai-assistant 0.7.8__py3-none-any.whl → 1.0.2__py3-none-any.whl

jarvis/jarvis_agent/task_analyzer.py

@@ -4,11 +4,15 @@
 负责处理任务分析和方法论生成功能
 """
 
-from jarvis.jarvis_utils.globals import get_interrupt, set_interrupt
-
+from jarvis.jarvis_agent.events import AFTER_TOOL_CALL
+from jarvis.jarvis_agent.events import BEFORE_SUMMARY
+from jarvis.jarvis_agent.events import BEFORE_TOOL_CALL
+from jarvis.jarvis_agent.events import TASK_COMPLETED
 from jarvis.jarvis_agent.prompts import get_task_analysis_prompt
 from jarvis.jarvis_agent.utils import join_prompts
-from jarvis.jarvis_agent.events import BEFORE_TOOL_CALL, AFTER_TOOL_CALL, BEFORE_SUMMARY, TASK_COMPLETED
+from jarvis.jarvis_utils.globals import get_interrupt
+from jarvis.jarvis_utils.globals import set_interrupt
+from jarvis.jarvis_utils.output import PrettyOutput
 
 
 class TaskAnalyzer:
@@ -45,7 +49,7 @@ class TaskAnalyzer:
             self._process_analysis_loop()
 
         except Exception:
-            print("❌ 分析失败")
+            PrettyOutput.auto_print("❌ 分析失败")
         finally:
             # 标记已完成一次分析，避免事件回调重复执行
             self._analysis_done = True
@@ -58,7 +62,7 @@ class TaskAnalyzer:
         """准备分析提示"""
         # 检查是否有 save_memory 工具（工具可用性）
         has_save_memory = False
-        # 检查是否有 generate_new_tool 工具
+        # 检查是否有 meta_agent 工具（原 generate_new_tool，自举式工具生成器）
         has_generate_new_tool = False
         try:
             tool_registry = self.agent.get_tool_registry()
@@ -66,25 +70,24 @@ class TaskAnalyzer:
                 # 检查 save_memory 工具
                 save_memory_tool = tool_registry.get_tool("save_memory")
                 has_save_memory = save_memory_tool is not None
-                
-                # 检查 generate_new_tool 工具
-                generate_tool = tool_registry.get_tool("generate_new_tool")
+
+                # 检查 meta_agent 工具
+                generate_tool = tool_registry.get_tool("meta_agent")
                 has_generate_new_tool = generate_tool is not None
         except Exception:
             pass
-        
+
         # 根据配置获取相应的提示词
         analysis_prompt = get_task_analysis_prompt(
-            has_save_memory=has_save_memory,
-            has_generate_new_tool=has_generate_new_tool
+            has_save_memory=has_save_memory, has_generate_new_tool=has_generate_new_tool
         )
-        
+
         return join_prompts([analysis_prompt, satisfaction_feedback])
 
     def _process_analysis_loop(self):
         """处理分析循环"""
         while True:
-            response = self.agent.model.chat_until_success(self.agent.session.prompt)  # type: ignore
+            response = self.agent.model.chat_until_success(self.agent.session.prompt)
             self.agent.session.prompt = ""
 
             # 处理用户中断
@@ -150,16 +153,19 @@ class TaskAnalyzer:
 
     def _handle_interrupt_with_tool_calls(self, user_input: str) -> str:
         """处理有工具调用时的中断"""
-        if self.agent.confirm_callback("检测到有工具调用，是否继续处理工具调用？", True):
-            return join_prompts([
-                f"被用户中断，用户补充信息为：{user_input}",
-                "用户同意继续工具调用。"
-            ])
+        if self.agent.confirm_callback(
+            "检测到有工具调用，是否继续处理工具调用？", False
+        ):
+            return join_prompts(
+                [f"被用户中断，用户补充信息为：{user_input}", "用户同意继续工具调用。"]
+            )
         else:
-            return join_prompts([
-                f"被用户中断，用户补充信息为：{user_input}",
-                "检测到有工具调用，但被用户拒绝执行。请根据用户的补充信息重新考虑下一步操作。"
-            ])
+            return join_prompts(
+                [
+                    f"被用户中断，用户补充信息为：{user_input}",
+                    "检测到有工具调用，但被用户拒绝执行。请根据用户的补充信息重新考虑下一步操作。",
+                ]
+            )
 
     def collect_satisfaction_feedback(self, auto_completed: bool) -> str:
         """收集满意度反馈"""
@@ -190,7 +196,7 @@ class TaskAnalyzer:
     # 事件订阅与处理（旁路）
     # -----------------------
     def _subscribe_events(self) -> None:
-        bus = self.agent.get_event_bus()  # type: ignore[attr-defined]
+        bus = self.agent.get_event_bus()
         # 在生成总结前触发（保持与原顺序一致）
         bus.subscribe(BEFORE_SUMMARY, self._on_before_summary)
         # 当无需总结时，作为兜底触发分析