• Home
• airflow.providers.google
• airflow.providers.google.cloud
• airflow.providers.google.cloud.operators
• airflow.providers.google.cloud.operators.vertex_ai
• airflow.providers.google.cloud.operators.vertex_ai.model_service

airflow.providers.google.cloud.operators.vertex_ai.model_service

This module contains Google Vertex AI operators.

Classes

DeleteModelOperator	Deletes a Model.
GetModelOperator	Retrieves a Model.
ExportModelOperator	Exports a trained, exportable Model to a location specified by the user.
ListModelsOperator	Lists Models in a Location.
UploadModelOperator	Uploads a Model artifact into Vertex AI.
ListModelVersionsOperator	Lists Model versions in a Location.
SetDefaultVersionOnModelOperator	Sets the desired Model version as Default.
AddVersionAliasesOnModelOperator	Adds version aliases for the Model.
DeleteVersionAliasesOnModelOperator	Deletes version aliases for the Model.
DeleteModelVersionOperator	Delete Model version in a Location.

Module Contents

class airflow.providers.google.cloud.operators.vertex_ai.model_service.DeleteModelOperator(*, region, project_id, model_id, retry=DEFAULT, timeout=None, metadata=(), gcp_conn_id='google_cloud_default', impersonation_chain=None, **kwargs)

Bases: airflow.providers.google.cloud.operators.cloud_base.GoogleCloudBaseOperator

Deletes a Model.

Parameters:

• project_id (str) – Required. The ID of the Google Cloud project that the service belongs to.

• region (str) – Required. The ID of the Google Cloud region that the service belongs to.

• model_id (str) – Required. The ID of the Model resource to be deleted. Could be in format projects/{project}/locations/{location}/models/{model_id}@{version_id} or projects/{project}/locations/{location}/models/{model_id}@{version_alias} if model has several versions.

• retry (google.api_core.retry.Retry | google.api_core.gapic_v1.method._MethodDefault) – Designation of what errors, if any, should be retried.

• timeout (float | None) – The timeout for this request.

• metadata (collections.abc.Sequence[tuple[str, str]]) – Strings which should be sent along with the request as metadata.

• gcp_conn_id (str) – The connection ID to use connecting to Google Cloud.

• impersonation_chain (str | collections.abc.Sequence[str] | None) – Optional service account to impersonate using short-term credentials, or chained list of accounts required to get the access_token of the last account in the list, which will be impersonated in the request. If set as a string, the account must grant the originating account the Service Account Token Creator IAM role. If set as a sequence, the identities from the list must grant Service Account Token Creator IAM role to the directly preceding identity, with first account from the list granting this role to the originating account (templated).

template_fields = ('region', 'model_id', 'project_id', 'impersonation_chain')

region

project_id

model_id

retry

timeout = None

metadata = ()

gcp_conn_id = 'google_cloud_default'

impersonation_chain = None

execute(context)

Derive when creating an operator.

Context is the same dictionary used as when rendering jinja templates.

Refer to get_template_context for more context.

class airflow.providers.google.cloud.operators.vertex_ai.model_service.GetModelOperator(*, region, project_id, model_id, retry=DEFAULT, timeout=None, metadata=(), gcp_conn_id='google_cloud_default', impersonation_chain=None, **kwargs)

Bases: airflow.providers.google.cloud.operators.cloud_base.GoogleCloudBaseOperator

Retrieves a Model.

Parameters:

• project_id (str) – Required. The ID of the Google Cloud project that the service belongs to.

• region (str) – Required. The ID of the Google Cloud region that the service belongs to.

• model_id (str) – Required. The ID of the Model resource to be retrieved. Could be in format projects/{project}/locations/{location}/models/{model_id}@{version_id} or projects/{project}/locations/{location}/models/{model_id}@{version_alias} if model has several versions.

• retry (google.api_core.retry.Retry | google.api_core.gapic_v1.method._MethodDefault) – Designation of what errors, if any, should be retried.

• timeout (float | None) – The timeout for this request.

• metadata (collections.abc.Sequence[tuple[str, str]]) – Strings which should be sent along with the request as metadata.

• gcp_conn_id (str) – The connection ID to use connecting to Google Cloud.

• impersonation_chain (str | collections.abc.Sequence[str] | None) – Optional service account to impersonate using short-term credentials, or chained list of accounts required to get the access_token of the last account in the list, which will be impersonated in the request. If set as a string, the account must grant the originating account the Service Account Token Creator IAM role. If set as a sequence, the identities from the list must grant Service Account Token Creator IAM role to the directly preceding identity, with first account from the list granting this role to the originating account (templated).

template_fields = ('region', 'model_id', 'project_id', 'impersonation_chain')

operator_extra_links

region

project_id

model_id

retry

timeout = None

metadata = ()

gcp_conn_id = 'google_cloud_default'

impersonation_chain = None

execute(context)

Derive when creating an operator.

Context is the same dictionary used as when rendering jinja templates.

Refer to get_template_context for more context.

class airflow.providers.google.cloud.operators.vertex_ai.model_service.ExportModelOperator(*, project_id, region, model_id, output_config, retry=DEFAULT, timeout=None, metadata=(), gcp_conn_id='google_cloud_default', impersonation_chain=None, **kwargs)

Bases: airflow.providers.google.cloud.operators.cloud_base.GoogleCloudBaseOperator

Exports a trained, exportable Model to a location specified by the user.

Parameters:

• project_id (str) – Required. The ID of the Google Cloud project that the service belongs to.

• region (str) – Required. The ID of the Google Cloud region that the service belongs to.

• model_id (str) – Required. The ID of the Model to export. Could be in format projects/{project}/locations/{location}/models/{model_id}@{version_id} or projects/{project}/locations/{location}/models/{model_id}@{version_alias} if model has several versions.

• output_config (google.cloud.aiplatform_v1.types.model_service.ExportModelRequest.OutputConfig | dict) – Required. The desired output location and configuration.

• retry (google.api_core.retry.Retry | google.api_core.gapic_v1.method._MethodDefault) – Designation of what errors, if any, should be retried.

• timeout (float | None) – The timeout for this request.

• metadata (collections.abc.Sequence[tuple[str, str]]) – Strings which should be sent along with the request as metadata.

• gcp_conn_id (str) – The connection ID to use connecting to Google Cloud.

• impersonation_chain (str | collections.abc.Sequence[str] | None) – Optional service account to impersonate using short-term credentials, or chained list of accounts required to get the access_token of the last account in the list, which will be impersonated in the request. If set as a string, the account must grant the originating account the Service Account Token Creator IAM role. If set as a sequence, the identities from the list must grant Service Account Token Creator IAM role to the directly preceding identity, with first account from the list granting this role to the originating account (templated).

template_fields = ('region', 'model_id', 'project_id', 'impersonation_chain')

operator_extra_links

region

project_id

model_id

output_config

retry

timeout = None

metadata = ()

gcp_conn_id = 'google_cloud_default'

impersonation_chain = None

execute(context)

Derive when creating an operator.

Context is the same dictionary used as when rendering jinja templates.

Refer to get_template_context for more context.

class airflow.providers.google.cloud.operators.vertex_ai.model_service.ListModelsOperator(*, region, project_id, filter=None, page_size=None, page_token=None, read_mask=None, order_by=None, retry=DEFAULT, timeout=None, metadata=(), gcp_conn_id='google_cloud_default', impersonation_chain=None, **kwargs)

Bases: airflow.providers.google.cloud.operators.cloud_base.GoogleCloudBaseOperator

Lists Models in a Location.

Parameters:

• project_id (str) – Required. The ID of the Google Cloud project that the service belongs to.

• region (str) – Required. The ID of the Google Cloud region that the service belongs to.

• retry (google.api_core.retry.Retry | google.api_core.gapic_v1.method._MethodDefault) – Designation of what errors, if any, should be retried.

• filter (str | None) – An expression for filtering the results of the request. For field names both snake_case and camelCase are supported. - model supports = and !=. model represents the Model ID, Could be in format the last segment of the Model’s [resource name][google.cloud.aiplatform.v1.Model.name]. - display_name supports = and != - labels supports general map functions that is: – labels.key=value - key:value equality – `labels.key:* or labels:key - key existence – A key including a space must be quoted. labels."a key".

• page_size (int | None) – The standard list page size.

• page_token (str | None) – The standard list page token. Typically obtained via [ListModelsResponse.next_page_token][google.cloud.aiplatform.v1.ListModelsResponse.next_page_token] of the previous [ModelService.ListModels][google.cloud.aiplatform.v1.ModelService.ListModels] call.

• read_mask (str | None) – Mask specifying which fields to read.

• order_by (str | None) – A comma-separated list of fields to order by, sorted in ascending order. Use “desc” after a field name for descending.

• timeout (float | None) – The timeout for this request.

• metadata (collections.abc.Sequence[tuple[str, str]]) – Strings which should be sent along with the request as metadata.

• gcp_conn_id (str) – The connection ID to use connecting to Google Cloud.

• impersonation_chain (str | collections.abc.Sequence[str] | None) – Optional service account to impersonate using short-term credentials, or chained list of accounts required to get the access_token of the last account in the list, which will be impersonated in the request. If set as a string, the account must grant the originating account the Service Account Token Creator IAM role. If set as a sequence, the identities from the list must grant Service Account Token Creator IAM role to the directly preceding identity, with first account from the list granting this role to the originating account (templated).

template_fields = ('region', 'project_id', 'impersonation_chain')

operator_extra_links

region

project_id

filter = None

page_size = None

page_token = None

read_mask = None

order_by = None

retry

timeout = None

metadata = ()

gcp_conn_id = 'google_cloud_default'

impersonation_chain = None

execute(context)

Derive when creating an operator.

Context is the same dictionary used as when rendering jinja templates.

Refer to get_template_context for more context.

class airflow.providers.google.cloud.operators.vertex_ai.model_service.UploadModelOperator(*, project_id, region, model, parent_model=None, retry=DEFAULT, timeout=None, metadata=(), gcp_conn_id='google_cloud_default', impersonation_chain=None, **kwargs)

Bases: airflow.providers.google.cloud.operators.cloud_base.GoogleCloudBaseOperator

Uploads a Model artifact into Vertex AI.

Parameters:

• project_id (str) – Required. The ID of the Google Cloud project that the service belongs to.

• region (str) – Required. The ID of the Google Cloud region that the service belongs to.

• model (google.cloud.aiplatform_v1.types.Model | dict) – Required. The Model to create. Creating model with the name that already exists leads to creating new version of existing model.

• parent_model (str | None) – The ID of the parent model to create a new version under.

• retry (google.api_core.retry.Retry | google.api_core.gapic_v1.method._MethodDefault) – Designation of what errors, if any, should be retried.

• timeout (float | None) – The timeout for this request.

• metadata (collections.abc.Sequence[tuple[str, str]]) – Strings which should be sent along with the request as metadata.

• gcp_conn_id (str) – The connection ID to use connecting to Google Cloud.

• impersonation_chain (str | collections.abc.Sequence[str] | None) – Optional service account to impersonate using short-term credentials, or chained list of accounts required to get the access_token of the last account in the list, which will be impersonated in the request. If set as a string, the account must grant the originating account the Service Account Token Creator IAM role. If set as a sequence, the identities from the list must grant Service Account Token Creator IAM role to the directly preceding identity, with first account from the list granting this role to the originating account (templated).

template_fields = ('region', 'project_id', 'model', 'parent_model', 'impersonation_chain')

operator_extra_links

project_id

region

model

parent_model = None

retry

timeout = None

metadata = ()

gcp_conn_id = 'google_cloud_default'

impersonation_chain = None

execute(context)

Derive when creating an operator.

Context is the same dictionary used as when rendering jinja templates.

Refer to get_template_context for more context.

class airflow.providers.google.cloud.operators.vertex_ai.model_service.ListModelVersionsOperator(*, region, project_id, model_id, retry=DEFAULT, timeout=None, metadata=(), gcp_conn_id='google_cloud_default', impersonation_chain=None, **kwargs)

Bases: airflow.providers.google.cloud.operators.cloud_base.GoogleCloudBaseOperator

Lists Model versions in a Location.

Parameters:

• project_id (str) – Required. The ID of the Google Cloud project that the service belongs to.

• region (str) – Required. The ID of the Google Cloud region that the service belongs to.

• model_id (str) – Required. The ID of the model to list versions for. Could be in format projects/{project}/locations/{location}/models/{model_id}@{version_id} or projects/{project}/locations/{location}/models/{model_id}@{version_alias} if model has several versions.

• retry (google.api_core.retry.Retry | google.api_core.gapic_v1.method._MethodDefault) – Designation of what errors, if any, should be retried.

• timeout (float | None) – The timeout for this request.

• metadata (collections.abc.Sequence[tuple[str, str]]) – Strings which should be sent along with the request as metadata.

• gcp_conn_id (str) – The connection ID to use connecting to Google Cloud.

• impersonation_chain (str | collections.abc.Sequence[str] | None) – Optional service account to impersonate using short-term credentials, or chained list of accounts required to get the access_token of the last account in the list, which will be impersonated in the request. If set as a string, the account must grant the originating account the Service Account Token Creator IAM role. If set as a sequence, the identities from the list must grant Service Account Token Creator IAM role to the directly preceding identity, with first account from the list granting this role to the originating account (templated).

template_fields = ('model_id', 'region', 'project_id', 'impersonation_chain')

region

project_id

model_id

retry

timeout = None

metadata = ()

gcp_conn_id = 'google_cloud_default'

impersonation_chain = None

execute(context)

Derive when creating an operator.

Context is the same dictionary used as when rendering jinja templates.

Refer to get_template_context for more context.

class airflow.providers.google.cloud.operators.vertex_ai.model_service.SetDefaultVersionOnModelOperator(*, region, project_id, model_id, retry=DEFAULT, timeout=None, metadata=(), gcp_conn_id='google_cloud_default', impersonation_chain=None, **kwargs)

Bases: airflow.providers.google.cloud.operators.cloud_base.GoogleCloudBaseOperator

Sets the desired Model version as Default.

Parameters:

• project_id (str) – Required. The ID of the Google Cloud project that the service belongs to.

• region (str) – Required. The ID of the Google Cloud region that the service belongs to.

• model_id (str) – Required. The ID of the model to set as default. Should be in format projects/{project}/locations/{location}/models/{model_id}@{version_id} or projects/{project}/locations/{location}/models/{model_id}@{version_alias} if model has several versions.

• retry (google.api_core.retry.Retry | google.api_core.gapic_v1.method._MethodDefault) – Designation of what errors, if any, should be retried.

• timeout (float | None) – The timeout for this request.

• metadata (collections.abc.Sequence[tuple[str, str]]) – Strings which should be sent along with the request as metadata.

• gcp_conn_id (str) – The connection ID to use connecting to Google Cloud.

• impersonation_chain (str | collections.abc.Sequence[str] | None) – Optional service account to impersonate using short-term credentials, or chained list of accounts required to get the access_token of the last account in the list, which will be impersonated in the request. If set as a string, the account must grant the originating account the Service Account Token Creator IAM role. If set as a sequence, the identities from the list must grant Service Account Token Creator IAM role to the directly preceding identity, with first account from the list granting this role to the originating account (templated).

template_fields = ('model_id', 'project_id', 'impersonation_chain')

operator_extra_links

region

project_id

model_id

retry

timeout = None

metadata = ()

gcp_conn_id = 'google_cloud_default'

impersonation_chain = None

execute(context)

Derive when creating an operator.

Context is the same dictionary used as when rendering jinja templates.

Refer to get_template_context for more context.

class airflow.providers.google.cloud.operators.vertex_ai.model_service.AddVersionAliasesOnModelOperator(*, region, project_id, model_id, version_aliases, retry=DEFAULT, timeout=None, metadata=(), gcp_conn_id='google_cloud_default', impersonation_chain=None, **kwargs)

Bases: airflow.providers.google.cloud.operators.cloud_base.GoogleCloudBaseOperator

Adds version aliases for the Model.

Parameters:

• project_id (str) – Required. The ID of the Google Cloud project that the service belongs to.

• region (str) – Required. The ID of the Google Cloud region that the service belongs to.

• model_id (str) – Required. The ID of the model to add version aliases for. Should be in format projects/{project}/locations/{location}/models/{model_id}@{version_id} or projects/{project}/locations/{location}/models/{model_id}@{version_alias}.

• version_aliases (collections.abc.Sequence[str]) – List of version aliases to be added to model version.

• retry (google.api_core.retry.Retry | google.api_core.gapic_v1.method._MethodDefault) – Designation of what errors, if any, should be retried.

• timeout (float | None) – The timeout for this request.

• metadata (collections.abc.Sequence[tuple[str, str]]) – Strings which should be sent along with the request as metadata.

• gcp_conn_id (str) – The connection ID to use connecting to Google Cloud.

• impersonation_chain (str | collections.abc.Sequence[str] | None) – Optional service account to impersonate using short-term credentials, or chained list of accounts required to get the access_token of the last account in the list, which will be impersonated in the request. If set as a string, the account must grant the originating account the Service Account Token Creator IAM role. If set as a sequence, the identities from the list must grant Service Account Token Creator IAM role to the directly preceding identity, with first account from the list granting this role to the originating account (templated).

template_fields = ('model_id', 'project_id', 'impersonation_chain')

operator_extra_links

region

project_id

model_id

version_aliases

retry

timeout = None

metadata = ()

gcp_conn_id = 'google_cloud_default'

impersonation_chain = None

execute(context)

Derive when creating an operator.

Context is the same dictionary used as when rendering jinja templates.

Refer to get_template_context for more context.

class airflow.providers.google.cloud.operators.vertex_ai.model_service.DeleteVersionAliasesOnModelOperator(*, region, project_id, model_id, version_aliases, retry=DEFAULT, timeout=None, metadata=(), gcp_conn_id='google_cloud_default', impersonation_chain=None, **kwargs)

Bases: airflow.providers.google.cloud.operators.cloud_base.GoogleCloudBaseOperator

Deletes version aliases for the Model.

Parameters:

• project_id (str) – Required. The ID of the Google Cloud project that the service belongs to.

• region (str) – Required. The ID of the Google Cloud region that the service belongs to.

• model_id (str) – Required. The ID of the model to delete version aliases from. Should be in format projects/{project}/locations/{location}/models/{model_id}@{version_id} or projects/{project}/locations/{location}/models/{model_id}@{version_alias}.

• version_aliases (collections.abc.Sequence[str]) – List of version aliases to be deleted from model version.

• retry (google.api_core.retry.Retry | google.api_core.gapic_v1.method._MethodDefault) – Designation of what errors, if any, should be retried.

• timeout (float | None) – The timeout for this request.

• metadata (collections.abc.Sequence[tuple[str, str]]) – Strings which should be sent along with the request as metadata.

• gcp_conn_id (str) – The connection ID to use connecting to Google Cloud.

• impersonation_chain (str | collections.abc.Sequence[str] | None) – Optional service account to impersonate using short-term credentials, or chained list of accounts required to get the access_token of the last account in the list, which will be impersonated in the request. If set as a string, the account must grant the originating account the Service Account Token Creator IAM role. If set as a sequence, the identities from the list must grant Service Account Token Creator IAM role to the directly preceding identity, with first account from the list granting this role to the originating account (templated).

template_fields = ('model_id', 'project_id', 'impersonation_chain')

operator_extra_links

region

project_id

model_id

version_aliases

retry

timeout = None

metadata = ()

gcp_conn_id = 'google_cloud_default'

impersonation_chain = None

execute(context)

Derive when creating an operator.

Context is the same dictionary used as when rendering jinja templates.

Refer to get_template_context for more context.

class airflow.providers.google.cloud.operators.vertex_ai.model_service.DeleteModelVersionOperator(*, region, project_id, model_id, retry=DEFAULT, timeout=None, metadata=(), gcp_conn_id='google_cloud_default', impersonation_chain=None, **kwargs)

Bases: airflow.providers.google.cloud.operators.cloud_base.GoogleCloudBaseOperator

Delete Model version in a Location.

Parameters:

• project_id (str) – Required. The ID of the Google Cloud project that the service belongs to.

• region (str) – Required. The ID of the Google Cloud region that the service belongs to.

• model_id (str) – Required. The ID of the Model in which to delete version. Should be in format projects/{project}/locations/{location}/models/{model_id}@{version_id} or projects/{project}/locations/{location}/models/{model_id}@{version_alias} several versions.

• retry (google.api_core.retry.Retry | google.api_core.gapic_v1.method._MethodDefault) – Designation of what errors, if any, should be retried.

• timeout (float | None) – The timeout for this request.

• metadata (collections.abc.Sequence[tuple[str, str]]) – Strings which should be sent along with the request as metadata.

• gcp_conn_id (str) – The connection ID to use connecting to Google Cloud.

• impersonation_chain (str | collections.abc.Sequence[str] | None) – Optional service account to impersonate using short-term credentials, or chained list of accounts required to get the access_token of the last account in the list, which will be impersonated in the request. If set as a string, the account must grant the originating account the Service Account Token Creator IAM role. If set as a sequence, the identities from the list must grant Service Account Token Creator IAM role to the directly preceding identity, with first account from the list granting this role to the originating account (templated).

template_fields = ('model_id', 'project_id', 'impersonation_chain')

region

project_id

model_id

retry

timeout = None

metadata = ()

gcp_conn_id = 'google_cloud_default'

impersonation_chain = None

execute(context)

Derive when creating an operator.

Context is the same dictionary used as when rendering jinja templates.

Refer to get_template_context for more context.

Previous Next

Was this entry helpful?