追踪 OpenAI Swarm🐝

警告

OpenAI Swarm 集成已被弃用，因为该库已被新的 OpenAI Agents SDK 取代。请考虑迁移到新的 SDK 以获取最新功能和支持。

MLflow 追踪 提供了对 OpenAI Swarm 的自动追踪能力，这是一个由 OpenAI 开发的多智能体框架。通过调用 mlflow.openai.autolog() 函数为 OpenAI 启用自动追踪，MLflow 将捕获嵌套追踪并在调用 OpenAI SDK 时将其记录到活动的 MLflow 实验中。

import mlflow

mlflow.openai.autolog()

除了对 OpenAI 进行基本的 LLM 调用追踪外，MLflow 还会捕获 Swarm 智能体操作的中间步骤以及智能体进行的所有工具调用。

提示

MLflow OpenAI 集成不仅限于追踪。MLflow 为 OpenAI 提供了完整的追踪体验，包括模型追踪、提示管理和评估。请查看 MLflow OpenAI Flavor 以了解更多信息！

基本示例

import mlflow
from swarm import Swarm, Agent

# Calling the autolog API will enable trace logging by default.
mlflow.openai.autolog()

# Optional: Set a tracking URI and an experiment
mlflow.set_tracking_uri("https://:5000")
mlflow.set_experiment("OpenAI Swarm")

# Define a simple multi-agent workflow using OpenAI Swarm
client = Swarm()


def transfer_to_agent_b():
    return agent_b


agent_a = Agent(
    name="Agent A",
    instructions="You are a helpful agent.",
    functions=[transfer_to_agent_b],
)

agent_b = Agent(
    name="Agent B",
    instructions="Only speak in Haikus.",
)

response = client.run(
    agent=agent_a,
    messages=[{"role": "user", "content": "I want to talk to agent B."}],
)

禁用自动跟踪

可以通过调用 mlflow.openai.autolog(disable=True) 或 mlflow.autolog(disable=True) 全局禁用 OpenAI Swarm 的自动追踪。