airflow.providers.google.cloud.hooks.vertex_ai.model_service
¶
This module contains a Google Cloud Vertex AI hook.
Module Contents¶
Classes¶
Hook for Google Cloud Vertex AI Endpoint Service APIs. |
- class airflow.providers.google.cloud.hooks.vertex_ai.model_service.ModelServiceHook(gcp_conn_id='google_cloud_default', delegate_to=None, impersonation_chain=None)[source]¶
Bases:
airflow.providers.google.common.hooks.base_google.GoogleBaseHook
Hook for Google Cloud Vertex AI Endpoint Service APIs.
- wait_for_operation(self, operation, timeout=None)[source]¶
Waits for long-lasting operation to complete.
- delete_model(self, project_id, region, model, retry=DEFAULT, timeout=None, metadata=())[source]¶
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 (str) -- Required. The name of the Model resource to be deleted.
retry (Union[google.api_core.retry.Retry, google.api_core.gapic_v1.method._MethodDefault]) -- Designation of what errors, if any, should be retried.
timeout (Optional[float]) -- The timeout for this request.
metadata (Sequence[Tuple[str, str]]) -- Strings which should be sent along with the request as metadata.
- export_model(self, project_id, region, model, output_config, retry=DEFAULT, timeout=None, metadata=())[source]¶
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 (str) -- Required. The resource name of the Model to export.
output_config (Union[google.cloud.aiplatform_v1.types.model_service.ExportModelRequest.OutputConfig, Dict]) -- Required. The desired output location and configuration.
retry (Union[google.api_core.retry.Retry, google.api_core.gapic_v1.method._MethodDefault]) -- Designation of what errors, if any, should be retried.
timeout (Optional[float]) -- The timeout for this request.
metadata (Sequence[Tuple[str, str]]) -- Strings which should be sent along with the request as metadata.
- list_models(self, project_id, region, filter=None, page_size=None, page_token=None, read_mask=None, order_by=None, retry=DEFAULT, timeout=None, metadata=())[source]¶
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.
filter (Optional[str]) -- 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, i.e. 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 (Optional[int]) -- The standard list page size.
page_token (Optional[str]) -- 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 (Optional[str]) -- Mask specifying which fields to read.
order_by (Optional[str]) -- A comma-separated list of fields to order by, sorted in ascending order. Use "desc" after a field name for descending.
retry (Union[google.api_core.retry.Retry, google.api_core.gapic_v1.method._MethodDefault]) -- Designation of what errors, if any, should be retried.
timeout (Optional[float]) -- The timeout for this request.
metadata (Sequence[Tuple[str, str]]) -- Strings which should be sent along with the request as metadata.
- upload_model(self, project_id, region, model, retry=DEFAULT, timeout=None, metadata=())[source]¶
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 (Union[google.cloud.aiplatform_v1.types.Model, Dict]) -- Required. The Model to create.
retry (Union[google.api_core.retry.Retry, google.api_core.gapic_v1.method._MethodDefault]) -- Designation of what errors, if any, should be retried.
timeout (Optional[float]) -- The timeout for this request.
metadata (Sequence[Tuple[str, str]]) -- Strings which should be sent along with the request as metadata.