Tutoriel : automatisation déclenchée par un alias d’artifact du registre

Ce tutoriel vous guide dans la création d’une automatisation de registre déclenchée par les métadonnées d’un artifact : lorsqu’un artifact de votre registre reçoit un alias spécifique (par exemple, production), W&B envoie une requête POST à votre webhook. Utilisez ce schéma pour notifier des systèmes en aval, tels que des pipelines de déploiement, des services d’astreinte ou des canaux de notification, chaque fois que vous promouvez un modèle vers une étape définie. Ce tutoriel s’adresse aux ingénieurs ML et aux praticiens MLOps qui gèrent le cycle de vie des modèles dans le registre.

Pour savoir comment créer une automatisation de projet, consultez Tutoriel : automatisation d’alerte en cas d’échec de run dans un projet.

Prérequis

Un webhook configuré dans Paramètres de l’équipe.
Un registre W&B contenant au moins une collection, ou un registre existant à réutiliser.

Créer une automatisation de registre

Configurez une automatisation limitée au registre afin que W&B envoie une requête POST à votre webhook lorsqu’un artifact de n’importe quelle collection du registre reçoit un alias spécifique (par exemple, production).

Ouvrez le registre, cliquez sur l’onglet Automations, puis sur Create automation.
Choisissez l’événement An artifact alias is added. Saisissez une Alias regex correspondant à l’alias qui vous intéresse (par exemple, production ou staging).
Cliquez sur Next step. Définissez Action type sur Webhooks, puis sélectionnez votre webhook. Si le webhook attend une charge utile, collez un corps JSON et utilisez des variables de charge utile telles que ${artifact_collection_name} et ${artifact_version_string}.
Cliquez sur Next step. Attribuez un nom à l’automatisation et, si vous le souhaitez, une description, puis cliquez sur Create automation.

Pour plus d’informations, voir Créer une automatisation webhook (onglet Registry).

Tester l’automatisation

Pour vérifier que l’automatisation fonctionne de bout en bout, déclenchez l’événement configuré en ajoutant l’alias à une version d’artifact. Ajoutez l’alias (par exemple, production) à une version d’artifact dans le registre, depuis W&B App ou via l’API publique. Par exemple :

import wandb

with wandb.init(project="my-project") as run:
    artifact = wandb.Artifact("my-model", type="model")
    # ... journaliser des fichiers ou des métadonnées dans l’artifact si nécessaire ...
    run.log_artifact(artifact)
    run.wait()  # Vérifier que l’artifact est bien journalisé avant de continuer

# Ajouter un alias à la dernière version de la collection
api = wandb.Api()
collection = api.artifact_collection(name="my-model", type_name="model")
version = next(collection.versions())  # Obtenir la dernière version

version.aliases.append("production")
version.save()
print("Added alias 'production' to", version.name)

Peu de temps après, votre point de terminaison de webhook reçoit une requête POST contenant la charge utile que vous avez configurée. Vous disposez désormais d’une automatisation de registre opérationnelle, qui se déclenche chaque fois qu’un alias correspondant est appliqué à un artifact de ce registre.

Aller plus loin

Pour approfondir les automatisations W&B au-delà de ce tutoriel, consultez les ressources suivantes :

Événements et portée des automatisations pour tous les types d’événements de projet et de registre.
Créer une automatisation Slack et Créer une automatisation webhook pour tous les détails sur l’interface utilisateur et la charge utile.

​Prérequis

​Créer une automatisation de registre

​Tester l’automatisation

​Aller plus loin

Prérequis

Créer une automatisation de registre

Tester l’automatisation

Aller plus loin