mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-10-29 03:18:14 +08:00
6d4ce83837
严格检查全部宿主源码，清零类型错误预算并移除按模块类型豁免。 按原因、所属方法、错误码及语法指纹锁定必要兼容例外和站点类型桩， CI 阻断新增忽略、检查范围退化、本地桩替代真实接口及新的类型错误。 修复空值、异步和供应方类型边界，保持插件与 API 兼容合同， 并补充存储、订阅、通知、Agent 和工作流边界的回归覆盖。
102 lines
4.4 KiB
Python
102 lines
4.4 KiB
Python
"""
|
||
在当前 Agent 会话的持久 Python 内核中处理多次只读工具结果。
|
||
"""
|
||
|
||
import json
|
||
from pathlib import Path
|
||
from typing import Any, ClassVar
|
||
|
||
from pydantic import BaseModel, Field
|
||
|
||
from app.agent.code.authority import CURRENT_CELL
|
||
from app.agent.code.manager import get_code_session_manager
|
||
from app.agent.tools.base import MoviePilotTool
|
||
from app.agent.tools.tags import ToolTag
|
||
from app.runtime.settings import get_runtime_setting
|
||
|
||
|
||
class ExecuteCodeInput(BaseModel):
|
||
"""
|
||
代码与显式重置状态；身份、工具集合和预算由宿主提供，不由模型选择。
|
||
"""
|
||
|
||
code: str = Field(
|
||
...,
|
||
min_length=1,
|
||
max_length=200000,
|
||
description="在持久化会话内核中执行的 Python 代码",
|
||
)
|
||
reset: bool = Field(
|
||
False,
|
||
description="执行本单元前是否丢弃此前所有 Python 变量和导入",
|
||
)
|
||
|
||
|
||
class ExecuteCodeTool(MoviePilotTool):
|
||
"""
|
||
与管理员命令执行使用相同主机权限边界；工具 RPC 仅开放安全只读能力。
|
||
|
||
:cvar result_max_chars: 禁用基类再次截断，由代码内核自身管理输出配额和文件回退
|
||
"""
|
||
|
||
name: str = Field(
|
||
default="execute_code",
|
||
description="供模型调用的稳定工具名称",
|
||
)
|
||
description: str = Field(
|
||
default=(
|
||
"需要循环、分页、筛选、关联或聚合三个及以上只读工具调用时运行 Python。"
|
||
"单次调用或结果需要完整模型推理时使用普通工具调用。"
|
||
"变量、导入和已加载数据在当前对话的多次调用间保留。"
|
||
"通过 from moviepilot_tools import moviepilot_api, read_file, search_web 导入助手函数；"
|
||
"仅当前模型请求已启用的助手可用，返回值为解析后的 JSON。"
|
||
"先加载对应领域 Skill，使用其精确 moviepilot_api 操作标识与参数。"
|
||
"API 助手仅允许安全读取，禁止写入、秘密读取及敏感性未知的读取。"
|
||
"moviepilot_tools 还提供 json_parse、shell_quote 和 retry。"
|
||
"每单元限制 300 秒、50 次工具调用、标准输出 50KB、标准错误 10KB，完整标准输出最多另存 5MB。"
|
||
"只打印有用答案，过大输出会返回可分段读取的文件。"
|
||
"异常保留变量；"
|
||
"超时、取消、重置或 sys.exit 会丢失内核状态。"
|
||
"这是在宿主机以管理员权限运行 Python，不是操作系统沙箱，无需逐次向用户确认。"
|
||
),
|
||
description="供模型理解工具用途、输入约束和调用边界的说明",
|
||
)
|
||
args_schema: type[BaseModel] = Field(
|
||
default=ExecuteCodeInput,
|
||
description="工具输入参数的验证模型或 JSON Schema",
|
||
)
|
||
require_admin: bool = Field(
|
||
default=True,
|
||
description="是否仅允许系统管理员调用",
|
||
)
|
||
tags: list[str] = Field(
|
||
default=[ToolTag.Admin, ToolTag.Command, ToolTag.Write],
|
||
description="工具的能力和权限分类标签，初始化时会统一归一化",
|
||
)
|
||
result_max_chars: ClassVar[int | None] = None
|
||
|
||
def get_tool_message(self, **kwargs: Any) -> str:
|
||
"""
|
||
不把完整程序或参数注入用户通知。
|
||
"""
|
||
return '执行 Python 数据分析'
|
||
|
||
def _get_run_timeout_seconds(self, **kwargs: Any) -> None:
|
||
"""
|
||
内核拥有300秒截止时间及进程回收，外层通用工具定时器不抢先丢弃清理结果。
|
||
"""
|
||
return None
|
||
|
||
async def run(self, **kwargs: Any) -> str: # type: ignore[override]
|
||
"""
|
||
只有真实宿主工具链绑定的当前 cell 可执行，直接外部调用不能获得默认身份。
|
||
"""
|
||
arguments = ExecuteCodeInput(**kwargs)
|
||
authority = CURRENT_CELL.get()
|
||
if authority is None or not authority.available() or authority.scope.user_id != str(self._user_id):
|
||
return json.dumps({'success': False, 'error': '当前入口没有可用的 Python 会话身份。'}, ensure_ascii=False)
|
||
root = Path(get_runtime_setting('CONFIG_PATH')) / 'agent' / 'runtime' / 'code'
|
||
manager = get_code_session_manager(root)
|
||
result = await manager.execute(authority, arguments.code, reset=arguments.reset, cwd=Path(get_runtime_setting('ROOT_PATH')))
|
||
return json.dumps(result, ensure_ascii=False)
|