JOJO
fe6fba3958
- 将多智能体/子智能体prompt从代码提取到prompts/multi_agent/和prompts/sub_agent/ - 多智能体模式添加可用的子智能体动态prompt并冻结 - 重启后自动从conversation.json恢复多智能体idle任务 - 修复新对话侧边栏显示其他对话子智能体的问题 - 简化子智能体弹窗输出样式 - sleep工具在多智能体模式下移除wait_sub_agent_ids参数
106 lines
3.4 KiB
Python
106 lines
3.4 KiB
Python
"""多智能体模式的系统提示词。
|
||
|
||
所有 prompt 正文均从 prompts/multi_agent/ 下的文本文件加载，避免在代码中硬编码。
|
||
"""
|
||
from __future__ import annotations
|
||
|
||
from pathlib import Path
|
||
from typing import List, Optional
|
||
|
||
try:
|
||
from config import PROMPTS_DIR
|
||
except ImportError:
|
||
import sys
|
||
|
||
project_root = Path(__file__).resolve().parents[2]
|
||
if str(project_root) not in sys.path:
|
||
sys.path.insert(0, str(project_root))
|
||
from config import PROMPTS_DIR
|
||
|
||
|
||
_MULTI_AGENT_PROMPTS_DIR = Path(PROMPTS_DIR) / "multi_agent"
|
||
_TEMPLATE_CACHE: dict[str, str] = {}
|
||
|
||
|
||
def _load_template(name: str) -> str:
|
||
"""从 prompts/multi_agent/<name>.txt 加载模板，带缓存。"""
|
||
cached = _TEMPLATE_CACHE.get(name)
|
||
if cached is not None:
|
||
return cached
|
||
|
||
template_path = _MULTI_AGENT_PROMPTS_DIR / f"{name}.txt"
|
||
if not template_path.exists():
|
||
raise FileNotFoundError(f"多智能体 prompt 模板缺失: {template_path}")
|
||
|
||
content = template_path.read_text(encoding="utf-8")
|
||
_TEMPLATE_CACHE[name] = content
|
||
return content
|
||
|
||
|
||
def _format_template(name: str, **kwargs) -> str:
|
||
"""加载模板并用 str.format 填充占位符。"""
|
||
template = _load_template(name)
|
||
return template.format(**kwargs)
|
||
|
||
|
||
# 为兼容外部仍可通过常量访问正文，但值来自文件；首次访问时加载。
|
||
def __getattr__(name: str) -> str:
|
||
if name == "MULTI_AGENT_MASTER_PROMPT_BODY":
|
||
return _load_template("master")
|
||
if name == "MULTI_AGENT_SUB_AGENT_PROMPT_BODY":
|
||
return _load_template("sub_agent")
|
||
raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
|
||
|
||
|
||
def build_multi_agent_master_prompt(workspace_path: str, base: str = "") -> str:
|
||
"""构造主智能体（Team Leader）的系统提示词。
|
||
|
||
`base` 一般为现有 MainTerminal 的 base 提示词（环境/工具概览等），
|
||
我们在末尾追加多智能体模式专属正文。
|
||
"""
|
||
body = _load_template("master")
|
||
if base and base.strip():
|
||
return f"{base.rstrip()}\n\n{body}\n"
|
||
return f"{body}\n"
|
||
|
||
|
||
def build_multi_agent_sub_agent_prompt(role_body: str, display_name: str, workspace_path: str) -> str:
|
||
"""构造子智能体的系统提示词。
|
||
|
||
`role_body` 为该角色 Markdown 文件 frontmatter 之后的自定义 prompt。
|
||
`display_name` 为该实例的显示名（如 `UI Operator_1`）。
|
||
"""
|
||
return _format_template("sub_agent", display_name=display_name, role_body=role_body.strip())
|
||
|
||
|
||
def build_available_agents_prompt() -> str:
|
||
"""构造「可用的子智能体角色」动态 prompt。
|
||
|
||
列出当前可创建的所有角色名称与说明，供 Team Leader 参考。
|
||
若没有任何可用角色，返回空字符串。
|
||
"""
|
||
try:
|
||
from modules.multi_agent.role_store import list_roles
|
||
|
||
roles = list_roles()
|
||
except Exception:
|
||
return ""
|
||
|
||
if not roles:
|
||
return ""
|
||
|
||
lines: List[str] = []
|
||
for role in roles:
|
||
name = getattr(role, "name", "") or ""
|
||
description = getattr(role, "description", "") or ""
|
||
role_id = getattr(role, "role_id", "") or ""
|
||
if description:
|
||
lines.append(f"- {name}（role_id: {role_id}）：{description}")
|
||
else:
|
||
lines.append(f"- {name}（role_id: {role_id}）")
|
||
|
||
if not lines:
|
||
return ""
|
||
|
||
return _format_template("available_agents", agents_list="\n".join(lines))
|