最强Agent OpenAI Swarm

swarm

安装方法

需要 Python 3.10+

方法一：

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

or

pip install git+https://github.com/openai/swarm.git

方法二：

在openai 官网下载swarm （https://github.com/openai/swarm）放在本地文件夹中，在工程中直接引用。

# import os
# 更换为你的本地路径
# sys.path.append('/root/swarm)

初始化

### 头文件引入
import os
from openai import OpenAI
from swarm import Swarm, Agent

环境变量设置

#临时设置环境变量
 os.environ["OPENAI_API_KEY"] = 'your-openai-key'
 os.environ["OPENAI_BASE_URL"] = 'https://ai.newone.tech/proxy/v1'

client = OpenAI (api_key=api_key,base_url="https://ai.newone.tech/proxy/v1")

使用方法介绍

单个Agent使用

from swarm import Swarm, Agent

swarm_client = Swarm(client)

agent_A = Agent(
    name="英文问答机器人",
    model = 'gpt-4o',
    instructions="无论用户发送的消息是什么语言，请用英文回答.",
    functions=[transfer_to_agent_b],
)

agent_B = Agent(
    name="中文问答机器人",
    model = 'gpt-4o',
    instructions="无论用户发送的消息是什么语言，请用中文回答.",
)

test_message1 = {"role" : "user","content":"你好，好久不见，请介绍下你自己。" }

response_A = swarm_client.run(
    agent = agent_A,
    messages = [test_message1],
)

print(response.messages[-1]["content"])

多个Agent使用

from swarm import Swarm, Agent

swarm_client = Swarm(client)

def transfer_to_agent_b():
    return agent_b

agent_a = Agent(
    name = "Agent A",
    instructions = "你是一个乐于助人的智能体.",
    functions=[transfer_to_agent_b],
)

agent_b = Agent(
    name = "Agent B",
    instructions="只用美好的语句回答.",
)

test_message1 = {"role" : "user","content":"我想与智能体B对话。" }

response  = swarm_client.run(
    agent = agent_a,
    messages = [test_message1],
)

print(response.messages[-1]["content"])