Weave

概述

此示例演示如何使用 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. 运行代理

   Mac

   python cookbook/observability/weave_op.py
   

注意事项

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