Weights & Biases

提示

這是由社群維護的文件，如果您遇到 bug，請提出 issue https://github.com/BerriAI/litellm

Weights & Biases 協助 AI 開發者更快打造更好的模型 https://wandb.ai

資訊

我們想了解如何讓 callbacks 變得更好！歡迎認識 LiteLLM 的 founders 或 加入我們的 discord

先決條件

請確認您已為此整合執行 uv add wandb

uv add wandb litellm

快速開始

只需 2 行程式碼，即可使用 Weights & Biases 立即記錄您 跨所有提供者 的回應

litellm.success_callback = ["wandb"]

# uv add wandb 
import litellm
import os

os.environ["WANDB_API_KEY"] = ""
# LLM API Keys
os.environ['OPENAI_API_KEY']=""

# set wandb as a callback, litellm will send the data to Weights & Biases
litellm.success_callback = ["wandb"] 
 
# openai call
response = litellm.completion(
  model="gpt-3.5-turbo",
  messages=[
    {"role": "user", "content": "Hi 👋 - i'm openai"}
  ]
)

支援與聯絡 founders

• 預約示範 👋
• 社群 Discord 💭
• 我們的電子郵件 ✉️ ishaan@berri.ai / krrish@berri.ai