跳至内容

OpenAI Agents SDK

OpenAI Agents SDK 让您能够以轻量级、易用的方式构建基于代理的AI应用,抽象层极少。它是我们之前代理实验项目Swarm的生产环境升级版。Agents SDK提供了一组非常精简的基础组件:

  • 代理,即配备指令和工具的LLM模型
  • 交接,允许代理将特定任务委托给其他代理
  • 防护栏,用于验证代理的输入

结合Python使用,这些基础组件功能强大,足以表达工具与智能体之间的复杂关系,让您无需陡峭的学习曲线就能构建现实世界的应用程序。此外,该SDK内置追踪功能,可帮助您可视化、调试智能体流程,还能进行评估甚至为您的应用微调模型。

为何要使用Agents SDK

该SDK遵循两个核心设计原则:

  1. 功能足够丰富值得使用,同时核心概念足够精简便于快速上手。
  2. 开箱即用效果出色,但您可以根据需要精确自定义功能。

以下是该SDK的主要功能:

  • 代理循环:内置的代理循环,负责调用工具、将结果发送给LLM,并循环执行直到LLM完成任务。
  • Python优先:利用内置语言特性来编排和链接代理,无需学习新的抽象概念。
  • 任务交接:一个强大的功能,用于在多个代理之间协调和委派工作。
  • 防护栏:与您的代理并行运行输入验证和检查,如果检查失败则提前中断。
  • 函数工具:将任何Python函数转化为工具,具备自动模式生成和Pydantic驱动的验证功能。
  • 追踪功能:内置的追踪系统,可让您可视化、调试和监控工作流程,同时还能使用OpenAI提供的评估、微调和蒸馏工具套件。

安装

pip install openai-agents

Hello world 示例

from agents import Agent, Runner

agent = Agent(name="Assistant", instructions="You are a helpful assistant")

result = Runner.run_sync(agent, "Write a haiku about recursion in programming.")
print(result.final_output)

# Code within the code,
# Functions calling themselves,
# Infinite loop's dance.

(如果运行此程序,请确保设置OPENAI_API_KEY环境变量)

export OPENAI_API_KEY=sk-...