Agent Swarm 蜂群模式 / ai #50

参考

一个用于利用 LangGraph 构建Swarm(蜂群式多智能体)系统的 Python 库。Swarm是一种多智能体架构，其中智能体会根据各自的专长动态地将控制权移交给其他智能体。该系统会记住最后一个处于活动状态的智能体，从而确保在后续交互中，对话能与该智能体继续进行。

Swarm 是一种受自然界蚁群、蜂群等生物群体行为启发的 多智能体协作设计范式。在自然界中，单只蚂蚁或蜜蜂的智能非常有限，但成千上万只聚集在一起时，却能完成极其复杂的任务——筑巢、觅食、防御。Swarm 蜂群式正是模仿了这种 "个体简单、群体智能" 的思想。

Agent Swarm (由AIJoe生成)

每个Agent仅依据局部环境信息和简单规则进行决策，通过信息素、消息广播或状态共享等方式间接通信，最终在整体层面涌现出复杂、高效的智能行为。

• 每只"蜜蜂"（Agent）负责采集特定区域的花蜜（子任务）
• 蜜蜂之间通过"舞蹈"（信息传递）告诉同伴哪里花多
• 没有一只蜜蜂知道全局计划，但整个蜂群高效完成了采蜜任务
• 即使几只蜜蜂迷路了，蜂群整体依然正常运转

pip install langgraph-swarm -i https://pypi.tuna.tsinghua.edu.cn/simple  //0.1.0 


from dotenv import dotenv_values
from langchain_openai import ChatOpenAI
from langgraph.checkpoint.memory import InMemorySaver
from langchain.agents import create_agent
from langgraph_swarm import create_handoff_tool, create_swarm

env_vars = dotenv_values('.env')
OPENAI_KEY = env_vars['OPENAI_API_KEY'] 
OPENAI_BASE_URL = env_vars['OPENAI_API_BASE'] 


llm = ChatOpenAI(model="gpt-4o-mini", api_key=OPENAI_KEY,base_url=OPENAI_BASE_URL)


def add(a: int, b: int) -> int:
    """Add two numbers"""
    return a + b

alice = create_agent(
    llm,
    tools=[
        add,
        create_handoff_tool(
            agent_name="Bob",
            description="Transfer to Bob",
        ),
    ],
    system_prompt="You are Alice, an addition expert.",
    name="Alice",
)

bob = create_agent(
    llm,
    tools=[
        create_handoff_tool(
            agent_name="Alice",
            description="Transfer to Alice, she can help with math",
        ),
    ],
    system_prompt="You are Bob, you speak like a pirate.",
    name="Bob",
)

checkpointer = InMemorySaver()
workflow = create_swarm(
    [alice, bob],
    default_active_agent="Alice"
)
app = workflow.compile(checkpointer=checkpointer)

config = {"configurable": {"thread_id": "1"}}
turn_1 = app.invoke(
    {"messages": [{"role": "user", "content": "i'd like to speak to Bob"}]},
    config,
)
print(11, turn_1)
turn_2 = app.invoke(
    {"messages": [{"role": "user", "content": "what's 5 + 7?"}]},
    config,
)
print(22, turn_2)