董宏凯andCursor
eaf1e075f2
将 backend 配置中的 SQLite/checkpointer 改为相对路径以便本地开发，并在离线部署下禁止并行席位使用 web_search，统一走技能检索。 Co-authored-by: Cursor <cursoragent@cursor.com>
169 lines
9.1 KiB
Python
169 lines
9.1 KiB
Python
"""圆桌各角色「单轮 run 参数」的**单一数据源**。
|
||
|
||
历史问题：前端手动路径（``multi_agent.py`` 的 ``_leader_run`` / ``_special_run``）
|
||
和后台进程内网关（``roundtable_inprocess_gateway.py``）各自硬编码了一份「每角色
|
||
工具裁剪 + thinking + reasoning」，结果**漂移**出差异（后台席位 / 报告少禁了
|
||
``web_search``，能联网搜索而前端不能）。
|
||
|
||
本模块把这套策略收敛成唯一定义，两条路径都从这里取，保证「后台挂起」与「前端
|
||
手动调用」传参**逐字一致**，且以后改一处即可，不再漂移。
|
||
|
||
角色（``role``）：
|
||
- ``leader`` —— 总控派活：短决策，关 thinking + low reasoning（首字节快）；
|
||
**只保留 ``agent_orchestration`` / ``ask_clarification`` / ``present_files``**，
|
||
其余「干活类」工具（``write_file`` / ``bash`` / ``str_replace`` / ``read_file`` /
|
||
``ls`` / ``view_image`` / ``task`` / ``write_todos`` / ``memory`` / ``hindsight_*``）
|
||
一律禁用——否则弱模型（如 deepseek-chat）会概率性地**自己写文件、跑 bash 干活**
|
||
而不是派活（已在生产日志里观测到总控 ``wc -l outputs/*.md`` 自检自己写的报告）。
|
||
总控的唯一职责是「派活 / 澄清 / 收口呈现」，没有任何理由动文件系统或执行命令。
|
||
并配合 ``SkillStopMiddleware`` 在 ``agent_orchestration`` 调用后停图，便于网关从
|
||
AIMessage.tool_calls 解析派活列表。
|
||
- ``seat`` —— 子智能体交付：开 thinking + medium reasoning；**可写文件**
|
||
（``write_file`` / ``str_replace`` 产出 md 等成品文档），但禁派活 / 澄清 /
|
||
展示文件 / 看图 / 联网搜索 / **``bash``**。禁 ``bash`` 的原因：圆桌席位是
|
||
「文档产出者」，弱模型爱写 ``.py`` 脚本再跑出产物（乱且抓不到预览），与
|
||
``report`` 角色禁 ``bash`` 同理；让它**直接 write_file 写成品文档**。
|
||
- ``seat_parallel`` —— **DAG 快速(fast)模式的并行席位（方案 A，2026-06）**：**禁写文件**
|
||
（``write_file`` / ``str_replace``，产物走正文）+ 禁 ``bash`` + 禁「打断 / 越权」类，
|
||
**放开 ``web_search`` 与技能**（搜索不碰 sandbox；技能靠 ``read_file`` SKILL.md）——
|
||
否则并行席位几乎调不动工具、产生不了步骤（步骤条不显示）。
|
||
「写文件产出 md」归 **沙箱(sandbox)模式**（走 ``seat`` policy，单列 + 沙箱跟随预览）；
|
||
fast 模式快、双列、产物走正文 —— 用户明确要求 fast 不生成文件。详见
|
||
``docs/roundtable-dag-orchestration.md`` §4 / §12.1 / §12.8 / §12.10。
|
||
与 ``seat`` 的差异：``seat`` 禁 ``web_search``（旧「圆桌一律禁搜索」），
|
||
``seat_parallel`` **放开搜索**（本期诉求）。
|
||
- ``report`` —— Step3 报告绘制：在 ``seat``（已禁 ``bash``）基础上**额外禁
|
||
``str_replace``**：
|
||
* ``bash`` 已在 ``seat`` 禁 → 杜绝写 Python 脚本再跑出 HTML（脚本产物落在
|
||
outputs，但前端虚拟预览只抓 ``write_file`` 的 ``args.content``，抓不到脚本
|
||
产物 → 预览停在半截 ``write_file``，正是「没写完就去生成 py 文件」的现象）；
|
||
* 禁 ``str_replace`` → 强制返工用整文件 ``write_file`` 重写
|
||
（``str_replace`` 没有 ``content`` 字段，会让前端虚拟预览读到 undefined
|
||
而清空沙箱）。
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
from dataclasses import dataclass
|
||
|
||
# ── 每角色基础工具裁剪（唯一定义处）─────────────────────────────────────────────
|
||
#
|
||
# 注意：``seat`` / ``report`` 都包含 ``web_search`` —— 圆桌研讨里联网搜索一律禁用
|
||
# （与前端 ``_special_run`` 的 excluded_tools 对齐）。
|
||
#
|
||
# ``leader`` 用「白名单」思路反推黑名单：总控**只允许** agent_orchestration（派活）、
|
||
# ask_clarification（澄清）、present_files（收口呈现子智能体产出的文件）三件事，
|
||
# 因此把其余所有工具全部禁掉。关键是 write_file / bash / str_replace —— 留着它们，
|
||
# 弱模型会绕过派活直接自己写报告（生产日志已实锤）。memory / hindsight_* 一并禁，
|
||
# 配合 ``memory_injection_enabled=False`` 让单例总控不被上轮会话的记忆带偏席位。
|
||
_LEADER_EXCLUDED: list[str] = [
|
||
"web_search",
|
||
"view_image",
|
||
"bash",
|
||
"ls",
|
||
"read_file",
|
||
"write_file",
|
||
"str_replace",
|
||
"task",
|
||
"write_todos",
|
||
"memory",
|
||
"hindsight_recall",
|
||
"hindsight_reflect",
|
||
"hindsight_retain",
|
||
"read_peer_delivery", # 总控靠 _collect_delivery_status 直读各席位,不需要这个席位侧工具
|
||
]
|
||
# seat：席位**可以写文件**(write_file / str_replace)产出 md 等成品文档,但**禁 bash** ——
|
||
# 圆桌席位是「文档产出者」,不该跑 shell/python 脚本(弱模型爱写 .py 脚本再跑生成产物,既乱又
|
||
# 抓不到产物预览,与 report 角色禁 bash 同理)。禁 web_search(圆桌串行一律禁搜索,旧决策)。
|
||
_SEAT_EXCLUDED: list[str] = [
|
||
"web_search",
|
||
"ask_clarification",
|
||
"present_files",
|
||
"view_image",
|
||
"agent_orchestration",
|
||
"bash",
|
||
]
|
||
# 并行席位 = **快速(fast)模式**(双列):**禁写文件**(产物走正文)+ 禁跑命令 +「打断/越权」类。
|
||
# **禁 web_search** —— 离线内网部署无外网,席位检索一律走「技能(skill)」(靠 read_file 读
|
||
# SKILL.md,不联网),与 leader / seat / report 三角色保持一致(2026-06 离线部署决策)。
|
||
# 「写文件」归 **沙箱(sandbox)模式**(seat policy,单列 + 沙箱跟随预览);fast 模式快、双列、
|
||
# 产物走正文 —— 用户明确要求 fast 不生成文件(2026-06 §12.10)。
|
||
_SEAT_PARALLEL_EXCLUDED: list[str] = [
|
||
"web_search", # 离线部署无外网,检索走技能(skill),不联网
|
||
"ask_clarification", # 并行批次不打断问澄清
|
||
"present_files", # 产物走正文,不呈现文件
|
||
"view_image",
|
||
"agent_orchestration", # 席位不派活
|
||
"bash", # 禁跑脚本(防 .py 脚本生成产物)
|
||
"write_file", # fast 模式产物走正文,不写文件(写文件归 sandbox 模式)
|
||
"str_replace",
|
||
]
|
||
# report：在 seat 基础上额外禁 str_replace(bash 已在 seat 禁)——强制整文件 write_file 重写,
|
||
# 让前端虚拟预览能拿到完整 args.content。report 是 Step3 单独绘制,不读前序席位 → 禁 read_peer_delivery。
|
||
_REPORT_EXCLUDED: list[str] = [*_SEAT_EXCLUDED, "str_replace", "read_peer_delivery"]
|
||
|
||
# leader 内禀的 skill_stop：``agent_orchestration`` 调用后停图。调用方可在此基础上
|
||
# 再叠加自己的 skill_stop（见 ``merge_skill_stop_names``）。
|
||
_LEADER_SKILL_STOP: list[str] = ["agent_orchestration"]
|
||
|
||
|
||
@dataclass(frozen=True)
|
||
class RoundtableRunPolicy:
|
||
"""一个角色单轮 run 的归一化参数。
|
||
|
||
与 ``multi_agent._run_payload`` 的入参一一对应：``excluded_tools`` /
|
||
``skill_stop_names`` / ``thinking_enabled`` / ``reasoning_effort``。
|
||
"""
|
||
|
||
excluded_tools: list[str]
|
||
skill_stop_names: list[str]
|
||
thinking_enabled: bool
|
||
reasoning_effort: str
|
||
|
||
|
||
def roundtable_run_policy(role: str) -> RoundtableRunPolicy:
|
||
"""返回某角色（``leader`` / ``seat`` / ``seat_parallel`` / ``report``）的单轮 run 参数。
|
||
|
||
返回值里的列表是**新副本**，调用方可安全地再 append/merge 而不污染本模块常量。
|
||
未知角色抛 ``ValueError``（早失败，避免悄悄退化到错误的工具面）。
|
||
"""
|
||
if role == "leader":
|
||
return RoundtableRunPolicy(
|
||
excluded_tools=list(_LEADER_EXCLUDED),
|
||
skill_stop_names=list(_LEADER_SKILL_STOP),
|
||
thinking_enabled=False,
|
||
reasoning_effort="low",
|
||
)
|
||
if role == "seat":
|
||
return RoundtableRunPolicy(
|
||
excluded_tools=list(_SEAT_EXCLUDED),
|
||
skill_stop_names=[],
|
||
thinking_enabled=True,
|
||
reasoning_effort="medium",
|
||
)
|
||
if role == "seat_parallel":
|
||
# 同 seat（thinking + medium），但额外禁写文件 / 跑命令（防并行沙箱打架）。
|
||
return RoundtableRunPolicy(
|
||
excluded_tools=list(_SEAT_PARALLEL_EXCLUDED),
|
||
skill_stop_names=[],
|
||
thinking_enabled=True,
|
||
reasoning_effort="medium",
|
||
)
|
||
if role == "report":
|
||
return RoundtableRunPolicy(
|
||
excluded_tools=list(_REPORT_EXCLUDED),
|
||
skill_stop_names=[],
|
||
thinking_enabled=True,
|
||
reasoning_effort="medium",
|
||
)
|
||
raise ValueError(f"unknown roundtable role: {role!r} (expected leader/seat/seat_parallel/report)")
|
||
|
||
|
||
def merge_skill_stop_names(policy_names: list[str], caller_names: list[str] | None) -> list[str]:
|
||
"""把角色内禀 skill_stop 与调用方追加的 skill_stop 合并去重（保序）。
|
||
|
||
取代历史上 ``list({"agent_orchestration", *skill_stop_names})`` 的 set 写法
|
||
（set 顺序不确定）；这里用 dict.fromkeys 保序去重，结果确定、便于断言。
|
||
"""
|
||
return list(dict.fromkeys([*policy_names, *(caller_names or [])]))
|