1.背景
本qiang~关注到OpenAI两周前发布的轻量级多代理框架Swarm,因此想要深入了解了一下,运行了官方提供的例子,整理并总结一些心得体会~
源码非常简单,各位看官们可以小读一下,本文采用gpt-4o-mini进行验证,如果想免费使用gpt-4o-mini,可私信沟通。
Ps: 发布之后,便在X引起了Swarm涉嫌抄袭,但吃瓜的我还是关注技术内部本身。
2.简介
Swarm项目是一个轻量级multi-agent编排框架,当前主要是为了实验及教学,而非用于生产。
Swarm专注于使agent协调和运行变得轻量级、高度可控且易于测试。
有两个主要抽象模块:Agent和handoffs(切换),Agent封装了instructions和tools,且可以随时选择将对话移交给另一个代理。
3. 适用场景
适用于大量的独立功能和指令难以整合到一个prompt的场景,而是将独立功能和指令通过agent, functions, tools等方式进行链接。
4. Swarm核心模块
4.1 Swarm.run()
类似于openai的chat completions api中的create()函数,接收并返回messages,并在调用之间不保存任何状态。注意,该方法也处理Agent函数执行、切换、上下文变量引用,且可以在返回最终输出前进行多轮对话。
其核心实现了如下循环:
(1) 从当前Agent调用模型返回结果
(2) 执行工具调用并且保存结果
(3) 必要时切换代理
(4) 必要时更新上下文变量
(5) 如果没有新的函数调用,则返回
该方法执行完毕后,会返回一个Response,包含所有相关的更新状态,包括新的messages, 最后被调用的agent以及最新的上下文变量context_variables。
4.2 Agents
Agent简单封装了一组instructions和functions,且具有将执行移交给另一个agent的能力。这些instructions和functions可用于表示非常具体的工作流程或步骤。
Instructions将被直接转化为一个对话的system提示,任何时候,只会存在活跃agent的instructions。
instructions可以是普通的字符串,也可以是返回字符串的函数,该函数可以选择一个context_variables参数,进而传入到swarm.run()中。
Swarm的Agent可以直接调用python函数,通常function应返回一个str,但也可以是一个Agent,如果返回Agent的话,则将执行转移到该Agent。如果Agent的函数调用出现错误,错误响应会添加到聊天中,以便Agent可以正常恢复。如果Agent调用多个函数,他们将按顺序进行。
4.3 切换Agent及更新上下文变量
一个agent可以通过在function中返回它来将其移交给另一个代理。还可以通过返回更完整的Result对象来更新context_variables
5. 实战
本次实战主要运行项目自带的例子airline。这个例子能够很好地说明Swarm的运行机理。
5.1 Agent之间的整体交互流程
交互步骤如下:
(1) 首先经过意图识别Agent,判断是处理航班变更还是行李丢失,并分别通过transfer_to_flight_modification, transfer_to_lost_baggage切换到对应的Agent
(2) 如果是航班变更,则进一步判断是航班取消,还是航班更换,如果是航班取消,通过transfer_to_flight_cancel切换到”航班取消Agent”进行处理,该Agent支持退款、兑换飞行积分、向上级反馈、问题已解决、切换到意图识别Agent等5个方法;如果是航班更换,通过transfer_to_flight_change切换到”航班更换Agent”,该Agent支持更换有效性验证、更换成功、向上级反馈、问题已解决、切换到意图识别Agent等5个方法。
(3) 如果是行李的相关问题,则切换至”行李丢失Agent”,该Agent包含行李搜寻、向上级反馈、问题已解决、切换到意图识别Agent等4个方法
5.2 实战效果
其中User是我的输出,[2024...]是打印的日志,Triage Agent、Flight Modification Agent、Flight cancel traversal是对应的Agent或方法调用大模型后返回的结果。
Starting Swarm CLI��
User:我想要退票
[2024-10-22 09:38:14] Ending turn.
Triage Agent:请问您能提供一下您的航班号以及您希望退票的原因吗?
User:航班号是1919,我的时间有些冲突了
[2024-10-22 09:38:48] Processing tool call: transfer_to_flight_modification with arguments {}
[2024-10-22 09:38:49] Processing tool call: transfer_to_flight_cancel with arguments {}
[2024-10-22 09:38:50] Ending turn.
Triage Agent: transfer_to_flight_modification()
Flight Modification Agent: transfer_to_flight_cancel()
Flight cancel traversal:您提到的航班号是1919。请确认您是否确实是要取消这个航班。如果是的话,我将继续进行处理。
User:是的
[2024-10-22 09:38:57] Ending turn.
Flight cancel traversal:请问您是希望申请退款还是希望获得航班积分?
User:退款吧
[2024-10-22 09:39:05] Processing tool call: initiate_refund with arguments {}
[2024-10-22 09:39:06] Ending turn.
Flight cancel traversal: initiate_refund()
Flight cancel traversal:您的退款请求已被处理,退款将在3-5个工作日内完成。请您耐心等待。请问您还有其他问题吗?
User:没有了,谢谢
[2024-10-22 09:39:16] Processing tool call: case_resolved with arguments {}
[2024-10-22 09:39:20] Ending turn.
Flight cancel traversal: case_resolved()
Flight cancel traversal:感谢您的联系!如果将来有任何问题,请随时与我们联系。祝您有美好的一天!
User:
6.总结
一句话足矣~
本文主要是OpenAI开源multi-agent框架swarm进行介绍,包括其核心组件Agent, instructions,functions,多个Agent之间如何进行切换,以及针对其中的examples进行验证,框架及源码非常简单,看官们可以亲自尝试一下。
Multi-agent的适用场景非常丰富,比如客服、工作流处理等
如果想免费获取使用openai的看官,可私信沟通。
7.参考
(1) Swarm仓库: https://github.com/openai/swarm