jupyter-agent 2025.6.104__py3-none-any.whl → 2025.6.105__py3-none-any.whl

jupyter_agent/bot_outputs.py

@@ -11,9 +11,11 @@ import datetime
 import jinja2
 
 from enum import Enum
-from typing import Optional, Dict, Any
+from typing import Optional, Dict, List, Tuple, Any, Type
 from pydantic import BaseModel, Field
 from IPython.display import display, Markdown
+from .bot_evaluation import BaseEvaluationRecord
+from .bot_actions import ActionBase
 from .utils import no_indent, no_wrap
 
 STAGE_SWITCHER_SCRIPT = no_wrap(
@@ -173,38 +175,6 @@ LOGGING_LEVELS = {
 }
 
 
-class BaseEvalutionRecord(BaseModel):
-    timestamp: float = 0
-    notebook_name: str = ""
-    eval_type: str = "BASE"
-    cell_index: int = -1
-    execution_duration: float = 0.0
-    is_success: bool = False
-    correct_score: float = 0.0
-
-
-class StageEvalutionRecord(BaseEvalutionRecord):
-    eval_type: str = "STAGE"
-    flow: str = ""
-    stage: str = ""
-    coding_score: float = 0.0
-    important_score: float = 0.0
-    user_supply_score: float = 0.0
-
-
-class FlowEvalutionRecord(BaseEvalutionRecord):
-    eval_type: str = "FLOW"
-    flow: str = ""
-    stage_count: int = 0
-    planning_score: float = 0.0
-
-
-class NotebookEvalutionRecord(BaseEvalutionRecord):
-    eval_type: str = "NOTEBOOK"
-    flow_count: int = 0
-    planning_score: float = 0.0
-
-
 class AgentOutput:
     """
     AgentOutput 是一个用于在 Jupyter Notebook 中显示 Agent 输出的类。
@@ -221,6 +191,7 @@ class AgentOutput:
         )
         self.template = self.jinja_env.from_string(AGENT_OUTPUT_TEMPLEATE)
         self.handler = None
+        self._is_dirty = True
         self._latest_display_tm = 0
         self._contents = {}
         self._active_stage = None
@@ -228,6 +199,7 @@ class AgentOutput:
         self._agent_data = {}
         self._logging_records = []
         self._evaluation_records = []
+        self._action_records = []
 
     @property
     def content(self):
@@ -263,11 +235,16 @@ class AgentOutput:
             )
         if self._evaluation_records:
             metadata["jupyter-agent-evaluation-records"] = [record.model_dump() for record in self._evaluation_records]
+        if self._action_records:
+            metadata["jupyter-agent-action-records"] = [record.model_dump() for record in self._action_records]
         return metadata
 
     def display(self, stage=None, force=False, wait=True):
-        if stage is not None:
+        if stage is not None and stage != self._active_stage:
             self._active_stage = stage
+            self._is_dirty = True
+        if not self._is_dirty and not force:
+            return
         if not force and time.time() - self._latest_display_tm < 1:
             if wait:
                 time.sleep(1 - (time.time() - self._latest_display_tm))
@@ -278,6 +255,7 @@ class AgentOutput:
         else:
             self.handler.update(Markdown(self.content), metadata=self.metadata)
         self._latest_display_tm = time.time()
+        self._is_dirty = False
 
     def clear(self, stage=None, clear_metadata=False):
         if stage is None:
@@ -286,6 +264,7 @@ class AgentOutput:
             self._contents[stage] = []
         if clear_metadata:
             self._agent_data = {}
+        self._is_dirty = True
         self.display(force=False, wait=False)
 
     def output_block(
@@ -305,6 +284,7 @@ class AgentOutput:
                 "code_language": code_language,
             }
         )
+        self._is_dirty = True
         self.display(stage, force=False, wait=False)
 
     def output_text(self, content, stage=None, code_language="python"):
@@ -320,6 +300,7 @@ class AgentOutput:
             self._contents[stage][-1]["content"] += "\n" + content
         else:
             self._contents[stage].append({"type": "text", "content": content, "code_language": code_language})
+        self._is_dirty = True
         self.display(stage, force=False, wait=False)
 
     def output_markdown(self, content, stage=None):
@@ -328,12 +309,14 @@ class AgentOutput:
         if stage not in self._contents:
             self._contents[stage] = []
         self._contents[stage].append({"type": "markdown", "content": content})
+        self._is_dirty = True
         self.display(stage, force=False, wait=False)
 
     def output_agent_data(self, **kwargs):
         self.log(f"output agent data {kwargs}", level="DEBUG")
         self._agent_data.update(kwargs)
-        self._agent_data_timestamp = int(time.time() * 1000)
+        self._agent_data_timestamp = time.time()
+        self._is_dirty = True
         self.display(force=False, wait=False)
 
     def log(self, msg, level="INFO"):
@@ -355,11 +338,12 @@ class AgentOutput:
                     "content": content,
                 }
             )
+        self._is_dirty = True
         self.display(force=False, wait=False)
 
-    def log_evaluation(self, record: BaseEvalutionRecord):
+    def log_evaluation(self, record: BaseEvaluationRecord):
         assert isinstance(
-            record, BaseEvalutionRecord
+            record, BaseEvaluationRecord
         ), "record must be an instance of BaseEvalutionRecord or its subclass"
         if record.timestamp == 0:
             record.timestamp = time.time()
@@ -369,6 +353,16 @@ class AgentOutput:
             f"success: {record.is_success} correct: {record.correct_score:.2f}",
             level="INFO",
         )
+        self._is_dirty = True
+        self.display(force=False, wait=False)
+
+    def log_action(self, record: ActionBase):
+        assert isinstance(record, ActionBase), "record must be an instance of BaseActionRecord or its subclass"
+        if record.timestamp == 0:
+            record.timestamp = time.time()
+        self._action_records.append(record)
+        self.log(f"Action: {record.action} from {record.source}", level="INFO")
+        self._is_dirty = True
         self.display(force=False, wait=False)
 
 
@@ -414,14 +408,14 @@ def output_agent_data(**kwargs):
     get_output().output_agent_data(**kwargs)
 
 
-def output_evaluation(record: BaseEvalutionRecord):
-    """
-    输出评估记录到 AgentOutput 中。
-    :param record: 评估记录对象，必须是 BaseEvalutionRecord 的子类。
-    """
+def output_evaluation(record: BaseEvaluationRecord):
     get_output().log_evaluation(record)
 
 
+def output_action(record: ActionBase):
+    get_output().log_action(record)
+
+
 def clear_output(stage=None, clear_metadata=False):
     get_output().clear(stage, clear_metadata)
 
@@ -534,6 +528,6 @@ _A = output_agent_data
 _L = log
 _D = lambda msg: log(msg, level="DEBUG")
 _I = lambda msg: log(msg, level="INFO")
-_W = lambda msg: log(msg, level="WARNING")
+_W = lambda msg: log(msg, level="WARN")
 _E = lambda msg: log(msg, level="ERROR")
 _F = lambda msg: log(msg, level="FATAL")

jupyter_agent/utils.py

@@ -91,37 +91,6 @@ class TeeOutputCapture(capture_output):
         return CapturedIO(stdout, stderr, outputs)
 
 
-class RequestUserPrompt(BaseModel):
-    prompt: str = Field(
-        description="需要用户补充详细信息的Prompt",
-        examples=["请补充与...相关的详细的信息", "请确认...是否...", "请提供..."],
-    )
-    example: Optional[str] = Field(None, description="示例", examples=["..."])
-
-
-class UserPromptResponse(BaseModel):
-    prompt: str = Field(description="需要用户补充详细信息的Prompt", examples=["..."])
-    response: str = Field(description="用户补充的详细信息", examples=["..."])
-
-
-def request_user_response(prompts: list[RequestUserPrompt]) -> list[UserPromptResponse]:
-    responses = []
-    for prompt in prompts:
-        response = input(f"{prompt.prompt} (例如: {prompt.example})")
-        responses.append(UserPromptResponse(prompt=prompt.prompt, response=response))
-    return responses
-
-
-def format_user_prompts(prompts: list[RequestUserPrompt], title="用户补充详细信息") -> str:
-    result = "```markdown\n"
-    result += f"### {title}\n\n"
-    result += "\n".join(
-        [f"- **Issue**: {prompt.prompt} (例如: {prompt.example})\n- **Reply**: " for prompt in prompts]
-    )
-    result += "\n```\n"
-    return result
-
-
 def no_indent(text: str) -> str:
     return re.sub(r"^\s+", "", text, flags=re.MULTILINE)
 
@@ -136,3 +105,23 @@ def no_newline(text: str) -> str:
 
 def no_space(text: str) -> str:
     return re.sub(r"\s+", "", text, flags=re.MULTILINE)
+
+
+class EnvironmentCapbilities(BaseModel):
+    save_metadata: bool = False
+    user_confirm: bool = False
+    user_supply_info: bool = False
+    set_cell_content: bool = False
+
+
+__env_capbilities = EnvironmentCapbilities()
+
+
+def get_env_capbilities() -> EnvironmentCapbilities:
+    return __env_capbilities
+
+
+def set_env_capbilities(env_capbilities: EnvironmentCapbilities):
+    global __env_capbilities
+
+    __env_capbilities = env_capbilities

{jupyter_agent-2025.6.104.dist-info → jupyter_agent-2025.6.105.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: jupyter-agent
-Version: 2025.6.104
+Version: 2025.6.105
 Summary: 调用LLM实现Jupyter代码的自动生成、执行、调试等功能
 Author: viewstar000
 License: MIT
@@ -10,6 +10,7 @@ Classifier: Operating System :: OS Independent
 Requires-Python: >=3.12
 Description-Content-Type: text/markdown
 License-File: LICENSE
+Requires-Dist: bottle
 Requires-Dist: ipynbname
 Requires-Dist: ipython
 Requires-Dist: jinja2
@@ -95,15 +96,21 @@ pip install /path/to/jupyter-agent/dist/jupyter_agent-xxxx-py3-none-any.whl
 # 设置当前Notebook的路径，当无法自动获取时需要手工指定，以Vscode中的Notebook为例
 %config BotMagics.notebook_path = globals()["__vsc_ipynb_file__"]
 
-# 设置是否保存任务数据到Metadata，只有Vscode中安装了jupyter-agent-extension后才支持
+# 设置运行环境是否保存任务数据到Metadata，默认为False，仅在Vscode中安装jupyter-agent-extension后或在评估模式下支持
 %config BotMagics.support_save_meta = True
+# 设置运行环境是否设置单元格内容，默认为False，权在Vscode中安装jupyter-agent-extension后或在评估模式下支持
+%config BotMagics.support_set_cell_content = True
 
 # 设置日志级别，可选值为DEBUG、INFO、WARN、ERROR、FATAL，默认为INFO
 %config BotMagics.logging_level = 'DEBUG'
 
+# 开启自动评估功能，默认为False，调用LLM对当前结果进行打分，目前仅实现了对子任务的整体打分
+%config BotMagics.enable_evaluating = True
+# 开启模拟用户补充信息功能，默认为False，调用LLM模拟对Agent的提问进行补充，用于自动评估
+%config BotMagics.enable_supply_mocking = True
+
 # 设置是否显示思考过程，默认为True
 %config BotMagics.display_think = True
-
 # 设置是否显示发送给出LLM的消息和LLM的回答，默认为False
 %config BotMagics.display_message = True
 %config BotMagics.display_response = True
@@ -151,6 +158,14 @@ pip install /path/to/jupyter-agent/dist/jupyter_agent-xxxx-py3-none-any.whl
 
 更详细用法可参考[示例Notebook](https://github.com/viewstar000/jupyter-agent/blob/main/examples/data_loader.ipynb)
 
+### 评估模式
+
+工具提供了`bot_eval`命令用于在评估模式下执行notebook。在评估模式下，工具会顺序执行所有有单元格，直到例全局目标完成。
+
+```bash
+bot_eval [-o output_eval.ipynb] [-e output_eval.jsonl] input.ipynb
+```
+
 ## 贡献
 
 欢迎提交 issue 或 pull request 参与贡献。
@@ -237,12 +252,19 @@ Advanced Configuration:
 # Set the current notebook path, when it is not automatically obtained, it needs to be manually specified, for example, in Vscode Notebook
 %config BotMagics.notebook_path = globals()["__vsc_ipynb_file__"]
 
-# Set whether to save task data to Metadata, only Vscode installed with jupyter-agent-extension supports
+# Set whether to save task data to Metadata, only Vscode installed with jupyter-agent-extension or evaluation mode supports this.
 %config BotMagics.support_save_meta = True
+# Set whether to set cell content, only Vscode installed with jupyter-agent-extension or evaluation mode supports this.
+%config BotMagics.support_set_cell_content = True
 
 # Set the log level, available values are DEBUG、INFO、WARN、ERROR、FATAL, default is INFO
 %config BotMagics.logging_level = 'DEBUG'
 
+# Enable automatic evaluation, default is False, call LLM to evaluate the overall result of the subtask
+%config BotMagics.enable_evaluating = True
+# Enable the simulation of user filling in information, default is False, call LLM to simulate the question of the agent to fill in
+%config BotMagics.enable_supply_mocking = True
+
 # Set whether to display thinking process, default is True
 %config BotMagics.display_think = True
 
@@ -290,6 +312,14 @@ After generating code for a subtask, the tool will call the corresponding agent
 
 For more details, please refer to [example notebook](https://github.com/viewstar000/jupyter-agent/blob/main/examples/data_loader.ipynb)
 
+### Evaluation mode
+
+Use `bot_eval` command to evaluate the code generated by the agent in evaluation mode. The evaluation mode will execute all cells in order and stop when the global goal is completed.
+
+```python
+bot_eval [-o output_eval.ipynb] [-e output_eval.jsonl] input.ipynb
+```
+
 ## Contributing
 
 Welcome to submit issues or pull requests to participate in contributions.

jupyter_agent-2025.6.105.dist-info/RECORD

@@ -0,0 +1,40 @@
+jupyter_agent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+jupyter_agent/bot_actions.py,sha256=Zq9_nfh4SJdMxkjqcTyQzS0RY4RwofaRkGq_4aexO2o,8453
+jupyter_agent/bot_chat.py,sha256=4zjkHtyOabT1bvGO-n4ZTMw0XREU_XDlgfLCI5gpxsw,8834
+jupyter_agent/bot_contexts.py,sha256=f9PmrA8dTH8v4WLdNL_ZSMJUzMVLuRldLWYwL4vXQ6c,19489
+jupyter_agent/bot_evaluation.py,sha256=t4SH6Gq4BmSyyRMozyQ2623XNGmgtCi9CTNRvOqzuRM,14266
+jupyter_agent/bot_magics.py,sha256=fOStbhYwg60hUbuZwqYIbMrn_TBg0JfTwFT92pysUFw,9869
+jupyter_agent/bot_outputs.py,sha256=QDzReXLqZsU7RAPR4F9JEotxAtIe9YA3ZklCJ9U_jVg,16239
+jupyter_agent/utils.py,sha256=8XKXXZB1EgCwIJEqYJigA8C84FzVTc2xdcF-y5kO3kY,3634
+jupyter_agent/bot_agents/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+jupyter_agent/bot_agents/base.py,sha256=FwCl1Z9E-nIL99qheGEcKO0H_PEd0I9OzEXd23cmQrg,12308
+jupyter_agent/bot_agents/master_planner.py,sha256=aXXgtyarBmDFNAypdEbpQoxzgGkU-botX7bE9q9w3sM,1392
+jupyter_agent/bot_agents/output_task_result.py,sha256=4DeAmDzjUI_0yfb26f5sLIwa74aJRzEQXbMRSyYMv1g,761
+jupyter_agent/bot_agents/request_user_supply.py,sha256=wL_DyIWPd5-Jdq0boTCAhF-nxf_gPmzXgSPM6jOBFFU,6128
+jupyter_agent/bot_agents/task_code_executor.py,sha256=M3oeEBlmsNz89f-yk3_nzsWKGH2C0o7AH059D_J94D8,2206
+jupyter_agent/bot_agents/task_coder.py,sha256=7fXq9nk1yH3F_mJfCMZBktHmxGfgmpuChMQbpEuL0w4,1783
+jupyter_agent/bot_agents/task_debuger.py,sha256=77pa_Awgvzxm3XkFA1oZsGr8SPJkjApKMtkmoySShmI,1367
+jupyter_agent/bot_agents/task_planner_v3.py,sha256=qrvgfsgWs-tD8L1NWzyv9-fwamNTEm6etJYBy-Ipz28,8487
+jupyter_agent/bot_agents/task_reasoner.py,sha256=OHYzPEb1e4eOGdKfo9655tTEdZRqz-s-Vdgp-2G0Jxc,1425
+jupyter_agent/bot_agents/task_structrue_reasoner.py,sha256=NQ0V9uY4UelRH5Glw3PwS9HiN4I74A9FxNPcMC5P0lg,3792
+jupyter_agent/bot_agents/task_structrue_summarier.py,sha256=bMrNlb_Xrs4FNosuTBPmMmmHIoPyCIjSUVR_0j7aZZ8,3898
+jupyter_agent/bot_agents/task_summarier.py,sha256=85pW9mSqI-9nfuKr2n-h36OpD3QLA0WPJ1T7eWyww-A,1747
+jupyter_agent/bot_agents/task_verifier.py,sha256=z45jo38c9U4FnS87-AnI4PGki14uuCRhyWVPtAbkl70,2478
+jupyter_agent/bot_agents/task_verify_summarier.py,sha256=usW7RDE8-XAZXAgyHkWU1ZmY-kwi-_ePVXBngYFtb_M,4899
+jupyter_agent/bot_evaluators/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+jupyter_agent/bot_evaluators/base.py,sha256=yMmeFCNrQRhO9Y4s1OeRAXgxD_m-20a14hM1XFvmN-I,1374
+jupyter_agent/bot_evaluators/dummy_flow.py,sha256=W0BWJKMgXIilZY0i8eP_SNVgqTUd1CT_uqMBs5aygVA,473
+jupyter_agent/bot_evaluators/dummy_global.py,sha256=yZ8fo2xfVN8gZTpBfs8EJ4dcv2t4ls6aXxn3Mo7bNSk,483
+jupyter_agent/bot_evaluators/dummy_task.py,sha256=owh6g6ItPXXYjesplzNMxVcKAU_kktWtuJhqRzZ05V4,475
+jupyter_agent/bot_evaluators/flow_global_planning.py,sha256=kOLd0dCoqrMi6zbe5chXrwxmdahtt8QqX3UnAZgk3AQ,2419
+jupyter_agent/bot_evaluators/flow_task_executor.py,sha256=gzHlKkP9K5fICYgUY5BKAzjwqn3xScxklohqoUCJaZk,4450
+jupyter_agent/bot_flows/__init__.py,sha256=Xe7EbC6bt04Nc4Yr0e--FVvBJCxkZCZkwYL9oahMBtI,338
+jupyter_agent/bot_flows/base.py,sha256=ADNs_-F4XLvEthRyjjBgGkv4KOOYIJRIdaRgjQXHq_k,14116
+jupyter_agent/bot_flows/master_planner.py,sha256=DAsMLXzrKCZaCNoEuPK3A5yWY_PxIcN3bwzYrlpx0pU,794
+jupyter_agent/bot_flows/task_executor_v3.py,sha256=_K6FOM2ZnMZAa8B_Rt1HXDVB20eCQl4Eew2KyfDPyEU,4685
+jupyter_agent-2025.6.105.dist-info/licenses/LICENSE,sha256=nWMmSIg7OepTIDX_OPP0-T9ImeCBBoog7eJxm5awtcM,1068
+jupyter_agent-2025.6.105.dist-info/METADATA,sha256=QqX2ahmch7Ofn_l8GHRjW5BpA1pvlm9kpOTbc5QpnIM,11699
+jupyter_agent-2025.6.105.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+jupyter_agent-2025.6.105.dist-info/entry_points.txt,sha256=063AB86wSrC_V-iiEEqxTlR4uz-T7VH_YagIpmKFQC0,63
+jupyter_agent-2025.6.105.dist-info/top_level.txt,sha256=c3USTBZ7DZGuvLKlEW-QfGIx0tzn98iCEn3bpdYnDtE,14
+jupyter_agent-2025.6.105.dist-info/RECORD,,

jupyter_agent/bot_agents/task_planner_v1.py

@@ -1,158 +0,0 @@
-"""
-Copyright (c) 2025 viewstar000
-
-This software is released under the MIT License.
-https://opensource.org/licenses/MIT
-"""
-
-from enum import Enum
-from pydantic import BaseModel, Field
-from IPython.display import Markdown
-from .base import BaseChatAgent, AgentOutputFormat, AgentModelType
-from ..bot_outputs import _D, _I, _W, _E, _F, _M, _B, _C, _O
-from ..bot_outputs import ReplyType
-
-
-TASK_PLANNER_PROMPT = """\
-**角色定义**：
-
-你是一名任务调度专家，负责根据全局分析规划，逐步推进并协调各子任务的执行。
-
-**任务要求**：
-
-- 审查全局目标与已完成的子任务结果，判断是否已实现整体目标：
-  - 若目标已达成，终止流程并输出最终结果。
-  - 若目标未达成，请根据目标规划说明和已完成子任务，制定下一个**子任务**的执行计划
-    - 子任务的划分应严格遵守全局目标规划说明的要求
-    - 协调代码生成Agent、结果验证Agent、结果总结Agent共同完成当前子任务
-    - 子任务的执行计划具体包括：
-      1. 子任务工作描述：
-          - 简明阐述子任务目标、输入与输出约束
-      2. 子任务代码生成Prompt：
-          - 需生成的代码类型（如数据处理、建模、可视化等）
-          - 具体输入（数据、变量、参数等）
-          - 预期输出形式（变量名、图表、文本等）
-      3. 子任务结果验证Prompt：
-          - 检测子任务代码执行结果是否符合当前子任务的要求
-          - 检测条件只考虑当前子任务的要求，不要考虑全局目标
-      4. 子任务结果总结Prompt：
-          - 说明本子任务结果总结的要点和输出要素，以便后续子任务使用
-  - 若需要用户提供更多的信息，给出详细的提示信息
-
-{% include "TASK_OUTPUT_FORMAT" %}
-
----
-
-{% include "TASK_CONTEXTS" %}
-
----
-
-{% include "CODE_CONTEXTS" %}
-
----
-
-{% if task.subject and task.issue %}
-**当前子任务信息**:
-
-### 当前子任务目标：
-{{ task.subject }}
-
-### 当前子任务代码：
-```python
-{{ task.source }}
-```
-
-### 当前子任务输出：
-{{ task.output }}
-
-### 当前子任务存在的问题：
-{{ task.issue }}
-
----
-
-请参考上述信息重新规划当前子任务：
-{% else %}
-请按要求规划下一个子任务：
-{% endif %}
-
-"""
-
-
-class TaskPlannerState(Enum):
-    PLANNED = "planned"  # 任务规划完成
-    REQUEST_INFO = "request_info"  # 需要用户补充更详细的信息
-    GLOBAL_FINISHED = "global_finished"  # 全局目标已达成
-
-
-class TaskPlannerOutput(BaseModel):
-    state: TaskPlannerState = Field(
-        description=(
-            "任务规划状态: "
-            f"{TaskPlannerState.PLANNED}, 完成规划, 可以开始执行下一步; "
-            f"{TaskPlannerState.REQUEST_INFO}, 需要用户补充更详细的信息; "
-            f"{TaskPlannerState.GLOBAL_FINISHED}, 全局目标是否已达成, 无需继续规划; "
-        ),
-        examples=[TaskPlannerState.PLANNED.value],
-    )
-    subtask_subject: str = Field(
-        "",
-        description='子任务简要描述, 在 state="planned" 时必填',
-        examples=["对...进行...处理，输出..."],
-    )
-    subtask_coding_prompt: str = Field(
-        "",
-        description='子任务代码生成Prompt, 在 state="planned" 时必填',
-        examples=["请基于...，计算...，并保存结果为..."],
-    )
-    subtask_verify_prompt: str = Field(
-        "",
-        description='子任务结果验证Prompt, 在 state="planned" 时必填',
-        examples=["请验证当前任务结果是否符合以下条件：..."],
-    )
-    subtask_summary_prompt: str = Field(
-        "",
-        description='子任务结果总结Prompt, 在 state="planned" 时必填',
-        examples=["请对当前任务的结果进行总结，输出以下要素：..."],
-    )
-    request_info_prompt: str = Field(
-        "",
-        description='需要用户补充更详细的信息的 Prompt, 在 state="request_info" 时必填',
-        examples=["请补充与...相关的详细的信息"],
-    )
-
-
-class TaskPlannerAgentV1(BaseChatAgent):
-    """任务规划器代理类"""
-
-    PROMPT = TASK_PLANNER_PROMPT
-    OUTPUT_FORMAT = AgentOutputFormat.JSON
-    OUTPUT_JSON_SCHEMA = TaskPlannerOutput
-    MODEL_TYPE = AgentModelType.PLANNER
-
-    def on_reply(self, reply: TaskPlannerOutput):
-        """执行规划逻辑"""
-        if reply.state == TaskPlannerState.GLOBAL_FINISHED:
-            _C(Markdown("全局目标已达成，任务完成！"), reply_type=ReplyType.TASK_RESULT)
-            return False, reply.state
-        elif reply.state == TaskPlannerState.REQUEST_INFO:
-            assert reply.request_info_prompt, "Request info prompt is empty"
-            _O(Markdown(f"### 需要补充更详细的信息\n\n{reply.request_info_prompt}"), reply_type=ReplyType.TASK_ISSUE)
-            return True, reply.state
-        elif reply.state == TaskPlannerState.PLANNED:
-            assert reply.subtask_subject, "Subtask subject is empty"
-            assert reply.subtask_coding_prompt, "Subtask coding prompt is empty"
-            assert reply.subtask_verify_prompt, "Subtask verify prompt is empty"
-            assert reply.subtask_summary_prompt, "Subtask summary prompt is empty"
-            _M(
-                f"### 子任务: {reply.subtask_subject}\n"
-                f"- Coding: {reply.subtask_coding_prompt}\n"
-                f"- Verify: {reply.subtask_verify_prompt}\n"
-                f"- Summary: {reply.subtask_summary_prompt}\n"
-            )
-            self.task.set_data("subject", reply.subtask_subject)
-            self.task.set_data("coding_prompt", reply.subtask_coding_prompt)
-            self.task.set_data("verify_prompt", reply.subtask_verify_prompt)
-            self.task.set_data("summary_prompt", reply.subtask_summary_prompt)
-            return False, reply.state
-        else:
-            raise ValueError(f"Unknown task planner state: {reply.state}")