Définir le nom d'affichage d'un Appel

Définir ou remplacer le nom d’affichage d’un Appel dans le Tracing W&B Weave

Les Ops produisent des Appels. Un Op est une fonction ou une méthode que vous décorez avec @weave.op. Par défaut, le nom de l’Op correspond au nom de la fonction, et les Appels associés ont le même nom d’affichage.

Vous pouvez remplacer le nom d’affichage de tous les Appels d’un Op donné de plusieurs façons.

Python

1. Modifiez le nom d’affichage au moment d’appeler l’Op. L’exemple suivant utilise le dictionnaire __weave pour définir le nom d’affichage de l’Appel, qui prévaut sur le nom d’affichage de l’Op :

result = my_function("World", __weave={"display_name": "My Custom Display Name"})

2. Modifiez le nom d’affichage Appel par Appel. L’exemple suivant utilise la méthode Op.call pour renvoyer un objet call, que vous pouvez ensuite utiliser pour définir le nom d’affichage avec call.set_display_name :

result, call = my_function.call("World")
call.set_display_name("My Custom Display Name")

3. Modifiez le nom d’affichage de tous les Appels d’un Op donné. L’exemple suivant définit le nouveau nom d’affichage directement dans le décorateur de fonction @weave.op afin qu’il s’applique à tous les Appels de l’Op :

@weave.op(call_display_name="My Custom Display Name")
def my_function(name: str):
    return f"Hello, {name}!"

call_display_name peut aussi être une fonction qui prend un objet call et renvoie une chaîne. Weave transmet automatiquement l’objet call lorsque la fonction s’exécute. Vous pouvez donc l’utiliser pour générer dynamiquement des noms à partir du nom de la fonction, des entrées de l’Appel, des champs, etc.

Un cas d’usage courant consiste à ajouter un horodatage au nom de la fonction.

from datetime import datetime

@weave.op(call_display_name=lambda call: f"{call.func_name}__{datetime.now()}")
def func():
    return ...

Vous pouvez également journaliser des métadonnées personnalisées avec .attributes.

def custom_attribute_name(call):
    model = call.attributes["model"]
    revision = call.attributes["revision"]
    now = call.attributes["date"]

    return f"{model}__{revision}__{now}"

@weave.op(call_display_name=custom_attribute_name)
def func():
    return ...

with weave.attributes(
    {
        "model": "finetuned-llama-3.1-8b",
        "revision": "v0.1.2",
        "date": "2024-08-01",
    }
):
    func()  # le nom d'affichage sera "finetuned-llama-3.1-8b__v0.1.2__2024-08-01"

    with weave.attributes(
        {
            "model": "finetuned-gpt-4o",
            "revision": "v0.1.3",
            "date": "2024-08-02",
        }
    ):
        func()  # le nom d'affichage sera "finetuned-gpt-4o__v0.1.3__2024-08-02"

4. Modifiez le nom d’affichage de l’Op lui-même. Les Appels associés à un Op ont le même nom d’affichage. Si vous remplacez le nom de l’Op lui-même, le nom d’affichage de l’Appel change également. Vous pouvez le faire de deux façons :

• Définissez la propriété name de l’Op avant que des Appels ne soient enregistrés :

my_function.name = "My Custom Op Name"

• Définissez l’option name sur le décorateur de l’Op :

@weave.op(name="My Custom Op Name")

TypeScript

Pour remplacer le nom par défaut d’un appel, utilisez l’option callDisplayName lors de l’appel à weave.op().

const extractDinosOp = weave.op(extractDinos, {
callDisplayName: (input: string) => `Your New Display Name`
});

Vous pouvez aussi mettre à jour le nom d’affichage d’un appel après l’exécution.