W&B run에 연결하여 함수 호출 트레이스하기

pip install wandb weave

wandb login

import wandb
import weave

def example_wandb(projname):
    # projname을 entity와 프로젝트로 분리
    entity, project = projname.split("/", 1)

    # 트레이싱을 위한 Weave 컨텍스트 초기화
    weave.init(projname)

    # 트레이스 가능한 Op 정의
    @weave.op()
    def say(message: str) -> str:
        return f"I said: {message}"

    # 첫 번째 W&B run
    with wandb.init(
        entity=entity,
        project=project,
        notes="Experiment 1",
        tags=["baseline", "paper1"],
    ) as run:
        say("Hello, world!")
        say("How are you!")
        run.log({"messages": 2})

    # 두 번째 W&B run
    with wandb.init(
        entity=entity,
        project=project,
        notes="Experiment 2",
        tags=["baseline", "paper1"],
    ) as run:
        say("Hello, world from experiment 2!")
        say("How are you!")
        run.log({"messages": 2})

if __name__ == "__main__":
    # 실제 W&B 사용자 이름/프로젝트로 바꾸세요
    example_wandb("your-team-name/your-project-name")

python weave_trace_with_wandb.py