追踪
Agents SDK内置追踪功能,可在智能体运行期间收集全面的事件记录,包括 LLM 生成、工具调用、任务转移、安全防护措施,乃至发生的自定义事件。借助追踪控制面板,你可以在开发和生产环境中调试、可视化和监控工作流。
!!!note
追踪默认启用。你可以通过以下三种常用方式禁用它:
1. 设置环境变量 `OPENAI_AGENTS_DISABLE_TRACING=1`,在全局范围内禁用追踪
2. 使用 [`set_tracing_disabled(True)`][agents.set_tracing_disabled],在代码中全局禁用追踪
3. 将 [`agents.run.RunConfig.tracing_disabled`][] 设置为 `True`,为单次运行禁用追踪
对于根据零数据保留(ZDR)政策使用OpenAI API 的组织,追踪功能不可用。
追踪与跨度 {#traces-and-spans}
- 追踪表示一次“工作流”的端到端操作。它们由跨度组成。追踪具有以下属性:
workflow_name:逻辑工作流或应用的名称,例如“代码生成”或“客户服务”。trace_id:追踪的唯一 ID。如果未传入,则会自动生成。格式必须为trace_<32_alphanumeric>。group_id:可选的组 ID,用于关联来自同一对话的多个追踪。例如,你可以使用聊天线程 ID。disabled:如果为 True,则不会记录该追踪。metadata:追踪的可选元数据。
- 跨度表示具有开始和结束时间的操作。跨度具有以下属性:
started_at和ended_at时间戳。trace_id,表示它们所属的追踪parent_id,指向此跨度的父跨度(如果有)span_data,即有关该跨度的信息。例如,AgentSpanData包含有关智能体的信息,GenerationSpanData包含有关 LLM 生成的信息,依此类推。
默认追踪 {#default-tracing}
默认情况下,SDK 会追踪以下内容:
- 整个
Runner.\{run, run_sync, run_streamed\}()都封装在一个trace()中。 - 每次运行器调用都封装在一个
task_span()中。 - 每个模型轮次都封装在一个
turn_span()中。 - 每次智能体运行时,都会封装在
agent_span()中 - LLM 生成封装在
generation_span()中 - 每次函数工具调用都分别封装在
function_span()中 - 安全防护措施封装在
guardrail_span()中 - 任务转移封装在
handoff_span()中 - 音频输入(语音转文本)封装在一个
transcription_span()中 - 音频输出(文本转语音)封装在一个
speech_span()中 - SDK 可能会将相关的音频跨度归入一个
speech_group_span()之下
默认情况下,追踪名称是字面字符串 Agent workflow。使用 trace 时可以设置此名称,也可以通过 [RunConfig][agents.run.RunConfig] 配置名称及其他属性。
如果你希望采用更紧凑的层级结构,可以为某次运行禁用自动生成的任务跨度和轮次跨度。智能体、生成、函数、安全防护措施、任务转移和自定义跨度仍会被记录。
from agents import RunConfig, Runner
result = await Runner.run(
agent,
"Hello",
run_config=RunConfig(tracing={"include_task_and_turn_spans": False}),
)此外,你还可以设置自定义追踪处理器,将追踪推送到其他目标位置(作为替代目标或次要目标)。
长时间运行的工作进程与即时导出 {#long-running-workers-and-immediate-exports}
默认的 [BatchTraceProcessor][agents.tracing.processors.BatchTraceProcessor] 每隔几秒在后台导出一次追踪;如果内存队列达到大小触发阈值,则会更早导出;进程退出时还会执行最后一次刷新。对于 Celery、RQ、Dramatiq 或 FastAPI 后台任务等长时间运行的工作进程,这意味着追踪通常无需任何额外代码即可自动导出,但它们可能不会在每项作业完成后立即显示在追踪控制面板中。
如果需要保证在一个工作单元结束时立即交付,请在追踪上下文退出后调用 [flush_traces()][agents.tracing.flush_traces]。
from agents import Runner, flush_traces, trace
@celery_app.task
def run_agent_task(prompt: str):
try:
with trace("celery_task"):
result = Runner.run_sync(agent, prompt)
return result.final_output
finally:
flush_traces()from fastapi import BackgroundTasks, FastAPI
from agents import Runner, flush_traces, trace
app = FastAPI()
def process_in_background(prompt: str) -> None:
try:
with trace("background_job"):
Runner.run_sync(agent, prompt)
finally:
flush_traces()
@app.post("/run")
async def run(prompt: str, background_tasks: BackgroundTasks):
background_tasks.add_task(process_in_background, prompt)
return {"status": "queued"}[flush_traces()][agents.tracing.flush_traces] 会阻塞,直到当前已缓冲的追踪和跨度完成导出,因此应在 trace() 关闭后调用它,以避免刷新尚未构建完整的追踪。如果默认导出延迟可以接受,则可以跳过此调用。
禁用追踪会阻止默认提供程序创建新的追踪和跨度,但不会丢弃其处理器已缓冲的数据。通过 set_tracing_disabled(True) 或 OPENAI_AGENTS_DISABLE_TRACING=1 禁用追踪后,[flush_traces()][agents.tracing.flush_traces] 仍会继续刷新这些已缓冲的数据。
高层级追踪 {#higher-level-traces}
有时,你可能希望对 run() 的多次调用都属于同一个追踪。为此,可以将整个代码封装在一个 trace() 中。
from agents import Agent, Runner, trace
async def main():
agent = Agent(name="Joke generator", instructions="Tell funny jokes.")
with trace("Joke workflow"): # (1)!
first_result = await Runner.run(agent, "Tell me a joke")
second_result = await Runner.run(agent, f"Rate this joke: {first_result.final_output}")
print(f"Joke: {first_result.final_output}")
print(f"Rating: {second_result.final_output}")- 由于对
Runner.run的两次调用都封装在一个with trace()中,因此两次运行会成为同一个整体追踪的一部分,而不是各自创建单独的追踪。
追踪的创建 {#creating-traces}
你可以使用 [trace()][agents.tracing.trace] 函数创建追踪。追踪需要启动和结束。你有以下两种方式:
- 推荐:将追踪用作上下文管理器,即
with trace(...) as my_trace。这样会在正确的时机自动启动和结束追踪。 - 你也可以手动调用 [
trace.start()][agents.tracing.Trace.start] 和 [trace.finish()][agents.tracing.Trace.finish]。
当前追踪通过 Python 的 contextvar 进行跟踪。这意味着它可自动处理并发。如果手动启动和结束追踪,请将 mark_as_current 传给 start(),并将 reset_current 传给 finish(),以更新当前追踪。
跨度的创建 {#creating-spans}
你可以使用各种 [*_span()][agents.tracing.create] 方法创建跨度。通常无需手动创建跨度。你可以使用 [custom_span()][agents.tracing.custom_span] 函数跟踪自定义跨度信息。
跨度会自动成为当前追踪的一部分,并嵌套在最近的当前跨度下;当前跨度通过 Python 的 contextvar 进行跟踪。
敏感数据 {#sensitive-data}
某些跨度可能会捕获潜在的敏感数据。
generation_span() 会存储 LLM 生成的输入和输出,function_span() 会存储函数调用的输入和输出。这些内容可能包含敏感数据,因此可以通过 [RunConfig.trace_include_sensitive_data][agents.run.RunConfig.trace_include_sensitive_data] 禁止捕获这些数据。
对于需要审批的函数工具,暂停以等待审批的跨度不会将 SDK 的内部结果包装器存储为工具输出。如果应用使用自定义拒绝消息来拒绝调用,则仅当 trace_include_sensitive_data 为 True 时,函数跨度才会将该消息存储为输出和错误文本。当该设置为 False 时,跨度会省略输出,并使用通用错误文本 Tool execution rejected。
同样,音频跨度默认包含输入和输出音频的 base64 编码 PCM 数据。你可以通过配置 [VoicePipelineConfig.trace_include_sensitive_audio_data][agents.voice.pipeline_config.VoicePipelineConfig.trace_include_sensitive_audio_data] 禁止捕获这些音频数据。
默认情况下,trace_include_sensitive_data 为 True。你可以在运行应用前,将 OPENAI_AGENTS_TRACE_INCLUDE_SENSITIVE_DATA 环境变量导出为 true/1 或 false/0,从而无需编写代码即可设置默认值。
当 trace_include_sensitive_data 为 False 时,Responses 模型跨度会省略请求输入和响应输出。对于对OpenAI官方端点的调用,跨度仍会包含 Responses API 的 response_id 作为关联元数据。对于自定义端点,SDK 会从经过脱敏的跨度中省略该标识符。
自定义追踪处理器 {#custom-tracing-processors}
追踪的高层级架构如下:
- 初始化时,我们会创建一个全局 [
TraceProvider][agents.tracing.provider.TraceProvider],负责创建追踪。 - 我们为
TraceProvider配置一个 [BatchTraceProcessor][agents.tracing.processors.BatchTraceProcessor],它会将追踪和跨度分批发送给 [BackendSpanExporter][agents.tracing.processors.BackendSpanExporter],后者再将跨度和追踪分批导出到OpenAI后端。
如果要自定义此默认设置,将追踪发送到替代或额外的后端,或者修改导出器行为,你有以下两种选择:
- [
add_trace_processor()][agents.tracing.add_trace_processor] 允许添加一个额外的追踪处理器,在追踪和跨度准备就绪时接收它们。这样,你可以在将追踪发送到OpenAI后端的同时执行自己的处理。 - [
set_trace_processors()][agents.tracing.set_trace_processors] 允许使用你自己的追踪处理器替换默认处理器。这意味着,除非你包含一个能够执行发送操作的TracingProcessor,否则追踪不会发送到OpenAI后端。
非OpenAI模型的追踪 {#tracing-with-non-openai-models}
使用非OpenAI模型时,可以向追踪导出器提供OpenAI API 密钥,从而无需禁用追踪,即可在OpenAI追踪控制面板中免费使用追踪功能。有关适配器选择和设置注意事项,请参阅模型指南中的第三方适配器部分。
import os
from agents import set_tracing_export_api_key, Agent
from agents.extensions.models.any_llm_model import AnyLLMModel
tracing_api_key = os.environ["OPENAI_API_KEY"]
set_tracing_export_api_key(tracing_api_key)
model = AnyLLMModel(
model="your-provider/your-model-name",
api_key="your-api-key",
)
agent = Agent(
name="Assistant",
model=model,
)如果只需为单次运行使用不同的追踪密钥,请通过 RunConfig 传入,而不要更改全局导出器。
from agents import Runner, RunConfig
await Runner.run(
agent,
input="Hello",
run_config=RunConfig(tracing={"api_key": "sk-tracing-123"}),
)补充说明 {#additional-notes}
- 在OpenAI追踪控制面板中查看免费追踪。
生态系统集成 {#ecosystem-integrations}
以下社区和供应商集成支持OpenAI Agents SDK的追踪 API 接口。