mlflow.webhooks

MLflow webhooks module.

This module provides webhook functionality for MLflow model registry events.

class mlflow.webhooks.ModelVersionAliasCreatedPayload[source]

Bases: TypedDict

Payload sent when an alias is created for a model version.

Example payload:

{
    "name": "example_model",
    "alias": "example_alias",
    "version": "1",
}

alias: str: The alias being created.

classmethod example() → mlflow.webhooks.types.ModelVersionAliasCreatedPayload[source]

name: str: The name of the registered model.

version: str: The version of the model the alias is being assigned to.

class mlflow.webhooks.ModelVersionAliasDeletedPayload[source]

Bases: TypedDict

Payload sent when an alias is deleted from a model version.

Example payload:

{
    "name": "example_model",
    "alias": "example_alias",
}

alias: str: The alias being deleted.

classmethod example() → mlflow.webhooks.types.ModelVersionAliasDeletedPayload[source]

name: str: The name of the registered model.

class mlflow.webhooks.ModelVersionCreatedPayload[source]

Bases: TypedDict

Payload sent when a new model version is created.

Example payload:

{
    "name": "example_model",
    "version": "1",
    "source": "models:/123",
    "run_id": "abcd1234abcd5678",
    "tags": {"example_key": "example_value"},
    "description": "An example model version",
}

description: str | None: Description of the model version.

classmethod example() → mlflow.webhooks.types.ModelVersionCreatedPayload[source]

name: str: The name of the registered model.

run_id: str | None: The run ID associated with the model version, if applicable.

source: str: The source URI of the model version.

tags: dict[str, str]: Tags associated with the model version.

version: str: The version of the model.

class mlflow.webhooks.ModelVersionTagDeletedPayload[source]

Bases: TypedDict

Payload sent when a tag is deleted from a model version.

Example payload:

{
    "name": "example_model",
    "version": "1",
    "key": "example_key",
}

classmethod example() → mlflow.webhooks.types.ModelVersionTagDeletedPayload[source]

key: str: The tag key being deleted.

name: str: The name of the registered model.

version: str: The version of the model.

class mlflow.webhooks.ModelVersionTagSetPayload[source]

Bases: TypedDict

Payload sent when a tag is set on a model version.

Example payload:

{
    "name": "example_model",
    "version": "1",
    "key": "example_key",
    "value": "example_value",
}

classmethod example() → mlflow.webhooks.types.ModelVersionTagSetPayload[source]

key: str: The tag key being set.

name: str: The name of the registered model.

value: str: The tag value being set.

version: str: The version of the model.

class mlflow.webhooks.RegisteredModelCreatedPayload[source]

Bases: TypedDict

Payload sent when a new registered model is created.

Example payload:

{
    "name": "example_model",
    "tags": {"example_key": "example_value"},
    "description": "An example registered model",
}

description: str | None: Description of the registered model.

classmethod example() → mlflow.webhooks.types.RegisteredModelCreatedPayload[source]

name: str: The name of the registered model.

tags: dict[str, str]: Tags associated with the registered model.