Swarm 多智能体框架

Swarm 是由 OpenAI 于 2024 年开发的一个实验性质的多智能体框架,它旨在简化多智能体系统的构建、编排和部署工作。 Swarm 关注的重点是让智能体协作和执行变得轻量、高度可控且易于测试。

Swarm 的核心在于两种原语抽象:智能体 (agent) 和 交接 (handoff) 。智能体包含指令和工具,并且在任何时间都可以选择将对话交接给另一个智能体。这种设计使得智能体之间可以灵活地传递任务,适应不同的场景和需求。 Swarm 适合处理存在大量独立功能和指令的情况,这些功能和指令很难编码成单个提示词。

Swarm 框架安装

Swarm 框架的安装非常简单,可以通过 pip 命令直接安装:

bash
pip install git+ssh://git@github.com/openai/swarm.git

使用 Swarm 时,可以定义智能体并指定它们的行为。例如,以下代码定义了两个智能体,并且用户的指令是与智能体 B 交谈:

from swarm import Swarm, Agent
client = Swarm()
def transfer_to_agent_b():
    return agent_b
agent_a = Agent(
    name="Agent A",
    instructions="You are a helpful agent.",
    functions=[transfer_to_agent_b],
)
agent_b = Agent(
    name="Agent B",
    instructions="Only speak in Haikus.",
)
response = client.run(
    agent=agent_a,
    messages=[{"role": "user", "content": "I want to talk to agent B."}],
)
print(response.messages[-1]["content"])

最终输出的消息将是 Agent B 按照 Haikus 诗歌形式给出的回答。