跳转到内容

AI SDK 集成

Agents SDK 开箱即用,可通过 Responses API 或 Chat Completions API 使用 OpenAI 模型。不过,如果您想使用其他模型,Vercel AI SDK 提供了一系列受支持的模型,可以通过此适配器将它们接入 Agents SDK。

  1. 安装扩展包,以安装 AI SDK 适配器:

    Terminal window
    npm install @openai/agents-extensions
  2. Vercel AI SDK 中选择所需的模型包并进行安装:

    Terminal window
    npm install @ai-sdk/openai
  3. 导入适配器和模型,以连接到您的智能体:

    导入适配器
    import { openai } from '@ai-sdk/openai';
    import { aisdk } from '@openai/agents-extensions/ai-sdk';
  4. 初始化供智能体使用的模型实例:

    创建模型
    import { openai } from '@ai-sdk/openai';
    import { aisdk } from '@openai/agents-extensions/ai-sdk';
    const model = aisdk(openai('gpt-5.4'));
AI SDK 设置
import { Agent, run } from '@openai/agents';
// Import the model package you installed
import { openai } from '@ai-sdk/openai';
// Import the adapter
import { aisdk } from '@openai/agents-extensions/ai-sdk';
// Create a model instance to be used by the agent
const model = aisdk(openai('gpt-5.4'));
// Create an agent with the model
const agent = new Agent({
name: 'My Agent',
instructions: 'You are a helpful assistant.',
model,
});
// Run the agent with the new model
run(agent, 'What is the capital of Germany?');

如果需要随消息发送提供商特定的选项,请通过 providerMetadata 传递。其值会直接转发给底层 AI SDK 模型。例如,Agents SDK 中的以下 providerData

Agents SDK providerData
const providerData = {
anthropic: {
cacheControl: {
type: 'ephemeral',
},
},
};

在使用 AI SDK 集成时会转换为

AI SDK providerMetadata
const providerMetadata = {
anthropic: {
cacheControl: {
type: 'ephemeral',
},
},
};

对于提供商为 openai.responses 的 AI SDK 模型,当模型使用 AI SDK 规范版本 v3 或 v4 时,适配器会在流式和非流式请求中转发 modelSettings.promptCacheRetention。它会将 Agents SDK 值 'in-memory' 映射为 AI SDK 提供商值 'in_memory''24h'null 则保持不变直接转发。

AI SDK v2 模型不支持此选项,因此当设置了该选项时,适配器会在发起模型请求之前抛出 UserError。显式设置的 modelSettings.providerData.providerOptions.openai.promptCacheRetention 值优先于 modelSettings.promptCacheRetention,并且适配器不会将该设置转发给其他 AI SDK 提供商。

此设置控制旧版最大保留策略,且仅受部分模型支持。有关当前模型兼容性和较新的缓存生命周期控制,请参阅提示缓存保留

AI SDK 适配器会将 Agents SDK 的 input_file 内容转换为 AI SDK v2、v3 和 v4 模型所需的文件部分格式。您可以通过以下方式提供 PDF:

  • Base64 数据 URL,例如 data:application/pdf;base64,...
  • 非空的原始 Base64,并提供 .pdf 文件名或 providerData.mediaType
  • 公共 HTTP(S) URL,并使用 .pdf 路径或提供 providerData.mediaType

当文件名或 URL 路径以 .pdf 结尾时,适配器会推断其类型为 application/pdf。否则,请显式设置 providerData: { mediaType: 'application/pdf' }。适配器会将公共 URL 直接转发给 AI SDK 模型,而不会下载文件,因此所选提供商和模型必须支持基于 URL 的 PDF 文件部分;如果不支持,请使用 Base64 表示形式。AI SDK 适配器不支持 OpenAI 文件 ID;请传递文件数据或公共 URL,或者直接使用 OpenAI Responses 模型。

某些提供商会将结构化输出作为带有额外包裹内容的纯文本返回,例如 JSON 代码围栏。如果需要在 Agents 运行时验证最终输出之前执行提供商特定的清理,请在创建适配器时传入 transformOutputText

规范化最终输出文本
import { openai } from '@ai-sdk/openai';
import { aisdk } from '@openai/agents-extensions/ai-sdk';
const model = aisdk(openai('gpt-5.4'), {
transformOutputText(text) {
return text.match(/```(?:json)?\s*([\s\S]*?)\s*```/)?.[1]?.trim() ?? text;
},
});

对于非流式响应,transformOutputText 会对最终确定的助手文本运行;对于流式响应,它会对最终的 response_done 事件运行。它不会修改增量 output_text_delta 事件。

modelSettings.retry 同样适用于由 AI SDK 支持的模型,因为重试由 Agents 运行时实现,而不只由默认 OpenAI 提供商实现。

这意味着您可以采用与其他场景相同的重试配置:

  • AgentRunner 或两者上设置 modelSettings.retry
  • 组合使用 retryPolicies,例如 networkError()httpStatus([...])providerSuggested()
  • 请注意,只有当封装的 AI SDK 模型能够通过适配器提供重试建议时,providerSuggested() 才能发挥作用。

有关使用 aisdk(openai(...)) 的完整示例,请参阅 examples/ai-sdk/retry.ts。有关重试 API 本身,包括流式传输和有状态后续请求的安全边界,请参阅模型

@openai/agents-extensions 中有两种相关集成:

  • @openai/agents-extensions/ai-sdk 会适配 AI SDK 模型,使 Agent 可以在该模型上运行。
  • @openai/agents-extensions/ai-sdk-ui 会适配流式 Agents SDK 运行,使 AI SDK UI 路由能够返回标准的流式 Response
  • @openai/agents-extensions/ai-sdk 适配器仍处于测试阶段,因此建议使用所选提供商进行仔细测试,尤其是规模较小的提供商。
  • 如果您使用 OpenAI 模型,建议优先使用默认的 OpenAI 模型提供商,而不是此适配器。
  • 受支持的 AI SDK 提供商必须公开 specificationVersion v2v3v4。如果需要较旧的 v1 提供商样式,请将 examples/ai-sdk-v1 中的模块复制到您的项目中。
  • 通过此适配器使用计算机工具时,需要提供显示元数据。请确保工具同时包含 environmentdimensions 元数据。
  • 此处不支持延迟式 Responses 工具加载流程,包括 toolNamespace()、设置了 deferLoading: true 的函数工具,以及 toolSearchTool()。如果需要工具搜索,请直接使用 OpenAI Responses 模型。请参阅工具模型
  • AI SDK 模型适配器不支持程序化工具调用。它会拒绝 programmaticToolCallingTool()allowedCallers 包含 'programmatic' 的工具、程序化工具调用历史项以及 Responses outputSchema。如需使用这些功能,请直接使用 OpenAI Responses 模型。

适配器会保留函数工具返回的 ToolOutputImage 值,包括远程 URL、Base64 数据和 OpenAI 文件 ID。AI SDK v2 模型接收 media 部分。AI SDK v3 模型接收 image-urlimage-dataimage-file-id 部分。AI SDK v4 模型接收 file 部分,其数据表示为 URL、内联数据或文件 ID 的提供商引用。这样,每个受支持的模型版本都可以使用原始图像表示形式。

有关完整示例,请参阅 examples/ai-sdk/image-tool-output.ts

@openai/agents-extensions/ai-sdk-ui 提供了响应辅助函数,用于将 Agents SDK 流接入 AI SDK UI 路由:

  • createAiSdkTextStreamResponse(source, options?) 用于纯文本流式响应。
  • createAiSdkUiMessageStream(source) 用于更底层的 ReadableStream<UIMessageChunk>
  • createAiSdkUiMessageStreamResponse(source, options?) 用于 UIMessageChunk 流式响应。

这些辅助函数接受 StreamedRunResult、类似流的源或兼容的包装对象。响应辅助函数会返回带有适合流式传输标头的 Response

当路由需要直接返回 AI SDK 响应时,请使用 createAiSdkUiMessageStreamResponse(...)。如果您希望自行控制响应或渲染层,同时仍使用由 Agents SDK 维护的 AI SDK UIMessageChunk 转换,请使用 createAiSdkUiMessageStream(...)。如果只需要纯文本,请使用 createAiSdkTextStreamResponse(...)

尽管 AI SDK 模型适配器无法发起程序化工具调用流程,UI 流辅助函数仍可封装使用该流程的流式 OpenAI Responses 运行。程序项会作为 programmatic_tool_calling 工具输入发出,匹配的程序结果则作为 program_output 工具输出发出。

响应辅助函数还通过 options 接受可选的响应设置:

  • headers:要合并到流式响应中的其他响应标头。
  • status:返回的 Response 的 HTTP 状态码。
  • statusText:返回的 Response 的 HTTP 状态文本。

底层 UI 消息流示例:

UI 消息流
import { Agent, run } from '@openai/agents';
import { createAiSdkUiMessageStream } from '@openai/agents-extensions/ai-sdk-ui';
const agent = new Agent({
name: 'Assistant',
instructions: 'Reply with a short answer.',
});
export async function createStream() {
const stream = await run(agent, 'Hello there.', { stream: true });
return createAiSdkUiMessageStream(stream);
}

用于 UI 消息流式传输的 Next.js 路由示例:

UI 消息流响应
import { Agent, run } from '@openai/agents';
import { createAiSdkUiMessageStreamResponse } from '@openai/agents-extensions/ai-sdk-ui';
const agent = new Agent({
name: 'Assistant',
instructions: 'Reply with a short answer.',
});
export async function POST() {
const stream = await run(agent, 'Hello there.', { stream: true });
return createAiSdkUiMessageStreamResponse(stream);
}

用于纯文本流式传输的 Next.js 路由示例:

文本流响应
import { Agent, run } from '@openai/agents';
import { createAiSdkTextStreamResponse } from '@openai/agents-extensions/ai-sdk-ui';
const agent = new Agent({
name: 'Assistant',
instructions: 'Reply with a short answer.',
});
export async function POST() {
const stream = await run(agent, 'Hello there.', { stream: true });
return createAiSdkTextStreamResponse(stream);
}

有关端到端用法,请参阅此仓库中的 examples/ai-sdk-ui 应用。