Lier un run W&B aux appels de fonction tracés

pip install wandb weave

wandb login

import wandb
import weave

def example_wandb(projname):
    # Diviser projname en entité et projet
    entity, project = projname.split("/", 1)

    # Initialiser le contexte Weave pour le traçage
    weave.init(projname)

    # Définir une Op traçable
    @weave.op()
    def say(message: str) -> str:
        return f"I said: {message}"

    # Premier run W&B
    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})

    # Deuxième run W&B
    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__":
    # Remplacez ceci par votre nom d'utilisateur/projet W&B réel
    example_wandb("your-team-name/your-project-name")

python weave_trace_with_wandb.py