OpenAI开源Swarm环境搭建&推理测试
毫无疑问,多智能体肯定是 AI 未来重要的研究方向之一,前阵子OpenAI可算是再Open一回,开源了多智能体框架Swarm。OpenAI开源的东西自然引来了极大的关注。OK, 我们开始吧
一、模型介绍
Swarm 关注的重点是让智能体协作和执行变得轻量、高度可控且易于测试。为此,它使用了两种原语抽象:智能体(agent)和交接(handoff)。其中,智能体包含指令和工具,并且在任何时间都可以选择将对话交接给另一个智能体。该团队表示,这些原语很强大,「足以表达工具和智能体网络之间的丰富动态,让你可以针对真实世界问题构建可扩展的解决方案,同时避免陡峭的学习曲线。另外,该团队指出,请注意 Swarm 智能体与 Assistants API 中的 Assistants 无关。之所以名字相似,只是为了方便。Swarm 完全由 Chat Completions API 提供支持,因此在调用之间是无状态的。
在设计上,Swarm 是轻量级、可扩展且高度可定制的。它最适合处理存在大量独立功能和指令的情况——这些功能和指令很难编码成单个提示词。如果开发者想要寻求完全托管的线程以及内置的内存管理和检索,那么 Assistants API 就已经是很好的选择了。但如果开发者想要完全的透明度,并且能够细粒度地控制上下文、步骤和工具调用,那么 Swarm 才是最佳选择。Swarm (几乎)完全运行在客户端,与 Chat Completions API 非常相似,不会在调用之间存储状态。该团队还展示了一个应用示例,包括天气查询智能体、用于在航空公司环境中处理不同客户服务请求的多智能体设置、客服机器人、可以帮助销售和退款的个人智能体等。具体示例请访问 Swarm 代码库。
二、安装环境
docker run -it --rm --gpus=all -v /datas/work/zzq:/workspace python:3.10.11 bash
cd /workspace
git clone
cd /Swarm/swarm-main
pip install . -i
三、推理测试
cd /workspace/Swarm/swarm-main/examples/weather_agent
python run.py
好吧,需要OpenAI的key