W&B run を関数の Call のトレースにリンクする

pip install wandb weave

wandb login

import wandb
import weave

def example_wandb(projname):
    # projname を entity と project に分割する
    entity, project = projname.split("/", 1)

    # トレース用に Weave コンテキストを初期化する
    weave.init(projname)

    # トレース可能な Op を定義する
    @weave.op()
    def say(message: str) -> str:
        return f"I said: {message}"

    # 1 つ目の 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})

    # 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 ユーザー名/project に置き換えてください
    example_wandb("your-team-name/your-project-name")

python weave_trace_with_wandb.py