​
概述

此示例演示如何使用 Weights & Biases (WandB) 的 Weave 来记录来自 Agno 代理的模型调用。

​
代码

import weave
from agno.agent import Agent
from agno.models.openai import OpenAIChat

# 创建并配置代理
agent = Agent(model=OpenAIChat(id="gpt-4o"), markdown=True, debug_mode=True)

# 使用您的项目名称初始化 Weave
weave.init("agno")

# 定义一个运行代理的函数，并用 weave.op() 装饰
@weave.op()
def run(content: str):
    return agent.run(content)

# 使用该函数记录模型调用
run("分享一个两句话的恐怖故事")

​
用法

1

安装 Weave

pip install agno openai weave
2

使用 WandB 进行身份验证

  • 前往 WandB 并从此处复制您的 API 密钥。
  • 在终端提示时输入您的 API 密钥，或将其导出为环境变量：
export WANDB_API_KEY=<your-api-key>
3

运行代理

python cookbook/observability/weave_op.py

​
注意事项

  • 初始化: 调用 weave.init("project-name") 以您的项目名称初始化 Weave。
  • 装饰器: 使用 @weave.op() 来装饰您希望通过 Weave 记录的函数。