메인 콘텐츠로 건너뛰기
이 튜토리얼에서는 아티팩트 메타데이터를 기반으로 트리거되는 레지스트리 자동화를 구축하는 방법을 안내합니다. 레지스트리의 아티팩트에 특정 별칭(예: 프로덕션)이 추가되면 W&B가 사용자의 웹훅으로 POST 요청을 보냅니다. 이 패턴을 사용하면 모델을 잘 알려진 단계로 승격할 때마다 배포 파이프라인, 페이징 서비스, 알림 채널과 같은 다운스트림 시스템에 알림을 보낼 수 있습니다. 이 튜토리얼은 W&B Registry에서 모델 라이프사이클을 관리하는 ML 엔지니어와 MLOps 실무자를 대상으로 합니다.


프로젝트 자동화를 만드는 방법은 튜토리얼: 프로젝트 run 실패 알림 자동화를 참조하세요.

​
사전 요구 사항

  • 팀 settings에서 구성한 웹훅.
  • 하나 이상의 컬렉션이 있는 W&B 레지스트리 또는 기존 레지스트리.

​
레지스트리 자동화 만들기

레지스트리 범위 자동화를 설정하세요. 그러면 레지스트리의 어떤 컬렉션에 있는 아티팩트에 특정 별칭(예: 프로덕션)이 지정될 때 W&B가 사용자의 웹훅으로 POST 요청을 보냅니다.
  1. 레지스트리를 열고 Automations 탭을 클릭한 다음 Create automation을 클릭하세요.
  2. 이벤트 An artifact alias is added를 선택하세요. 원하는 별칭과 일치하는 Alias regex를 입력하세요(예: production 또는 staging).
  3. Next step을 클릭하세요. Action type을 Webhooks로 설정하고 웹훅을 선택하세요. 웹훅이 페이로드를 받는 경우 JSON 본문을 붙여넣고 ${artifact_collection_name} 및 ${artifact_version_string} 같은 payload variables를 사용하세요.
  4. Next step을 클릭하세요. 자동화 이름과 선택 사항인 설명을 입력한 다음 Create automation을 클릭하세요.
자세한 내용은 Create a webhook automation (레지스트리 탭)을 참조하세요.

​
자동화 테스트하기

자동화가 엔드투엔드로 실행되는지 확인하려면, 아티팩트 버전에 별칭을 추가해 설정된 이벤트를 트리거하세요. W&B App 또는 공개 API를 사용해 레지스트리의 아티팩트 버전에 별칭(예: 프로덕션)을 추가하세요. 예를 들어 다음과 같습니다: 
import wandb

with wandb.init(project="my-project") as run:
    artifact = wandb.Artifact("my-model", type="model")
    # ... 필요에 따라 아티팩트에 파일 또는 메타데이터를 로깅합니다 ...
    run.log_artifact(artifact)
    run.wait()  # 아티팩트 로깅이 완료될 때까지 대기합니다

# 컬렉션의 최신 버전에 별칭을 추가합니다
api = wandb.Api()
collection = api.artifact_collection(name="my-model", type_name="model")
version = next(collection.versions())  # 최신 버전을 조회합니다

version.aliases.append("production")
version.save()
print("Added alias 'production' to", version.name)
잠시 후 웹훅 엔드포인트로 설정한 페이로드가 포함된 POST 요청이 도착합니다. 이제 이 레지스트리에서 아티팩트에 일치하는 별칭이 적용될 때마다 실행되는 레지스트리 자동화가 정상적으로 작동합니다.

​
더 알아보기

이 튜토리얼에서 다룬 내용 외에 W&B 자동화에 대해 더 알아보려면 다음 자료를 참조하세요: