多角色转换 / `transfer_to_agent`（★★）

源文件：chapter10/multi-role-transfer/README.md

实验 10-2：多角色转换 / transfer_to_agent（★★）

《深入理解 AI Agent》配套代码。演示共享上下文下的链式移交（handoff）：
一个会话里存在多个专业角色 Agent（各有独立系统提示词与专属工具集），
通过一个 transfer_to_agent(target_role, reason) 工具在角色间自主移交控制权。

这个实验想说明什么

• 与 10-1（软件开发单任务的预定义阶段流水线）不同，10-2 强调跨领域、
  由 Agent 自主判断该切换到哪个专业角色——不是预先规划好的线性流程，
  而是根据任务进展动态切换。
• 因为共享同一段对话历史，移交时完整历史天然保留，
  新角色自动继承此前所有内容（无需显式传参）。
• 机制重点是「自主角色移交」，而非工具本身多强，因此工具用轻量真实实现 / 可控 mock。

架构

                        共享对话历史 history（user/assistant/tool 消息，全程保留）
                                        ▲   ▲
   每轮调用大模型时：                     │   │
   [ 当前角色的 system prompt ] + history ┘   └ 只暴露 [ 当前角色工具集 + transfer_to_agent ]

   模型两种动作：
     ① 调用自己的专属工具（普通 function calling）
     ② 调用 transfer_to_agent(target_role, reason)
        → 编排器换掉「系统提示词 + 工具集」，history 原样不动
        → 新角色继承全部历史（共享上下文）
​

5 个角色（roles.py）：

每个角色都额外持有 transfer_to_agent，可自主把控制权交给同事。

代码结构：

• tools.py —— 各角色专属工具的实现 + OpenAI function-calling schema
• roles.py —— 5 个角色定义（系统提示词 + 工具集）+ transfer_to_agent schema
• orchestrator.py —— 移交编排器（共享历史 + 换系统提示词/工具集的主循环，含防死循环/拒绝自我移交）
• demo.py —— 一条命令的演示入口

运行方式

pip install -r requirements.txt

# 配置 key（二选一）
export OPENAI_API_KEY=sk-...        # 直接 export
# 或： cp env.example .env 后填写

python demo.py
​

可配环境变量（均有默认值）：
OPENAI_API_KEY、OPENAI_BASE_URL（默认 https://api.openai.com/v1）、
OPENAI_MODEL（默认 gpt-5.6-luna）。

通用回退：优先用 OPENAI_API_KEY 直连 OpenAI；若未设置该变量但设了
OPENROUTER_API_KEY，则自动改走 OpenRouter，并把模型名映射到其命名空间
（gpt-5.6-luna → openai/gpt-5.6-luna）。提示：gpt-5.6 系列直连 OpenAI 需组织验证，
只填 OPENROUTER_API_KEY（不填 OPENAI_API_KEY）即可强制走 OpenRouter，更省事。

命令行参数

所有参数均可选，不传则行为与最初版本完全一致（跑默认 cagr 场景）。运行
python demo.py --help 查看完整中文说明。

例：

python demo.py --list-roles            # 离线看角色/场景清单，不调用 API
python demo.py --scenario coding       # 路由到 coding 角色的场景
python demo.py --task "帮我调研并总结…" # 自定义任务
python demo.py --role research         # 从 research 角色起步
python demo.py --interactive           # 交互式多轮，输入 exit 退出
​

三个内置场景（SCENARIOS）：cagr（默认，新能源汽车销量→CAGR→投资总结）、
solar（同类链路换一组光伏装机数据）、coding（路由到 coding 角色用
execute_python 真正跑斐波那契脚本，再由 writing/triage 收尾）。

演示说明

demo.py 抛出一个需要多次跨领域切换的复合任务：

查中国 2021—2023 三年新能源汽车销量 → 算出年均复合增长率(CAGR) → 写成一段面向投资人的中文总结

预期看到 Agent 自主完成移交链：

triage → research → data_analysis → writing
​

• triage 判断第一步要查数据，移交 research；
• research 用 web_search 查到三年销量，移交 data_analysis；
• data_analysis 用 calculate 算出 CAGR ≈ 64.22%，移交 writing；
• writing 综合此前历史里的销量数据与 CAGR，直接写出最终成稿。

writing 从未自己检索或计算，却能引用准确的销量数字和增长率——
这正是共享上下文的证据。运行结束会打印完整移交链、每次移交的 from→to 与 reason，
以及各角色分工总览（谁调用了哪些专属工具、谁产出了最终回复），一眼看清
「同一段历史上不同专业角色各司其职地接力」。

注：真实 LLM 输出有随机性，某次运行的具体措辞/步数可能略有不同，但移交机制一致。

预期输出示例（真实运行截取）

以下是一次 python demo.py（model=gpt-5.6-luna，经 OpenRouter 路由）真实运行的关键片段，未做任何编造或修饰：

=== 角色花名册（共 5 个专业角色）===
• triage — 前台分诊（默认入口）
    工具集: ['transfer_to_agent']
    系统提示词(首句): 你是通用助理系统的『前台分诊』角色，也是默认入口。
• research — 信息检索专家
    工具集: ['web_search', 'transfer_to_agent']
    ...（其余角色略，完整列表见上方角色表）

┌── 当前角色: 信息检索专家 (research)  工具: ['web_search', 'transfer_to_agent']
└── 🔧 调用工具 web_search args={'query': '中国 2021年 2022年 2023年 新能源汽车销量 乘联会 中汽协'}
    → 【检索结果·中国乘用车市场信息联席会/中汽协】…2021 年：352.1 万辆 / 2022 年：688.7 万辆 / 2023 年：949.5 万辆
┌── 当前角色: 数据分析专家 (data_analysis)  工具: ['calculate', 'descriptive_stats', 'transfer_to_agent']
└── 🔧 调用工具 calculate args={'expression': '(949.5/352.1)**(1/2)-1'}
    → (949.5/352.1)**(1/2)-1 = 0.6421562289791105

================ 运行汇总 ================
自主移交链: triage → research → data_analysis → writing → triage
移交次数: 4
  1. triage → research  |  reason: 需要先检索中国2021、2022、2023年新能源汽车销量及可靠来源，为后续CAGR计算和投资人摘要提供数据依据。
  2. research → data_analysis  |  reason: 已检索到2021、2022、2023年新能源汽车销量，请计算2021至2023年的两年CAGR，并给出结果供后续写作。
  3. data_analysis → writing  |  reason: 销量数据与CAGR已完成：2021年352.1万辆、2022年688.7万辆、2023年949.5万辆；2021—2023年CAGR=(949.5/352.1)^(1/2)-1=64.22%。请据此写不超过120字的投资人中文总结。
  4. writing → triage  |  reason: 已完成投资人摘要并核对篇幅（101字符，不超过120字）…请做最终收尾确认。

各角色分工（谁用了什么工具、谁产出最终回复）:
  triage        : （仅路由/移交，未用专属工具）  ⇒ 产出最终回复
  research      : web_search
  data_analysis : calculate
  writing       : count_characters

最终成果:
据中汽协公开数据，中国新能源汽车销量由2021年的352.1万辆增至2022年的688.7万辆、2023年的949.5万辆。2021—2023年两年CAGR达64.2%，市场保持高速扩张，成长潜力显著。
​

局限

• 默认模型为 gpt-5.6-luna；移交是否按预期链路发生，很大程度依赖所选模型的指令遵循能力，换模型效果可能不同。
• research 角色的 web_search 是内置知识库 mock，并非真实联网检索，仅能命中预置的少量关键词（新能源汽车销量、光伏装机、Python GIL），换查询词可能查不到。
• 真实 LLM 输出存在随机性：具体移交步数、每次 reason 的措辞、是否途经 coding 角色等，不同次运行可能不同，但移交机制本身一致。
• orchestrator.py 设有 max_steps（默认 20）硬上限，以及「同一 (角色, 工具, 参数) 连续调用 ≥3 次」的纠偏提示，用于防止模型死循环；这是兜底保护，不代表每次运行都会用满这些步数。