Weights & Biases

Weights & Biases

Weights & Biases는 AI 개발자가 더 나은 모델을 더 빠르게 구축하도록 도와줘요 https://wandb.ai

출처: 문서

본문

Weights & Biases는 AI 개발자가 더 나은 모델을 더 빠르게 구축하도록 도와줘요.

사전 준비 (Pre-Requisites)

이 통합을 위해 uv add wandb를 실행했는지 확인하세요.

uv add wandb litellm

빠른 시작 (Quick Start)

단 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-5.6-luna",
  messages=[
    {"role": "user", "content": "Hi 👋 - i'm openai"}
  ]
)

더 알아보기 (Learn more)