feat: add enable_private_service_connect field to Endpoint feat: add id field to DeployedModel feat: add service_attachment field to PrivateEndpoints feat: add endpoint_id to CreateEndpointRequest and method signature to CreateEndpoint feat: add method... (#879)

* feat: add enable_private_service_connect field to Endpoint feat: add id field to DeployedModel feat: add service_attachment field to PrivateEndpoints feat: add endpoint_id to CreateEndpointRequest and method signature to CreateEndpoint feat: add method signature to CreateFeatureStore, CreateEntityType, CreateFeature feat: add network and enable_private_service_connect to IndexEndpoint feat: add service_attachment to IndexPrivateEndpoints feat: add stratified_split field to training_pipeline InputDataConfig

PiperOrigin-RevId: 413691586

Source-Link: googleapis/googleapis@5915024

Source-Link: googleapis/googleapis-gen@afe7901
Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiYWZlNzkwMWU5MmMwYmFiMDg4OGNkMjk3MTlhNmI3NThkN2UwZGQzNSJ9

* 🦉 Updates from OwlBot

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

Co-authored-by: Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>

Author: gcf-owl-bot[bot]

google/cloud/aiplatform_v1/__init__.py

@@ -112,6 +112,7 @@
 from .types.event import Event
 from .types.execution import Execution
 from .types.explanation import Attribution
+from .types.explanation import BlurBaselineConfig
 from .types.explanation import Explanation
 from .types.explanation import ExplanationMetadataOverride
 from .types.explanation import ExplanationParameters
@@ -187,6 +188,9 @@
 from .types.index_endpoint_service import GetIndexEndpointRequest
 from .types.index_endpoint_service import ListIndexEndpointsRequest
 from .types.index_endpoint_service import ListIndexEndpointsResponse
+from .types.index_endpoint_service import MutateDeployedIndexOperationMetadata
+from .types.index_endpoint_service import MutateDeployedIndexRequest
+from .types.index_endpoint_service import MutateDeployedIndexResponse
 from .types.index_endpoint_service import UndeployIndexOperationMetadata
 from .types.index_endpoint_service import UndeployIndexRequest
 from .types.index_endpoint_service import UndeployIndexResponse
@@ -445,12 +449,14 @@
 from .types.training_pipeline import FractionSplit
 from .types.training_pipeline import InputDataConfig
 from .types.training_pipeline import PredefinedSplit
+from .types.training_pipeline import StratifiedSplit
 from .types.training_pipeline import TimestampSplit
 from .types.training_pipeline import TrainingPipeline
 from .types.types import BoolArray
 from .types.types import DoubleArray
 from .types.types import Int64Array
 from .types.types import StringArray
+from .types.unmanaged_container_model import UnmanagedContainerModel
 from .types.user_action_reference import UserActionReference
 from .types.value import Value
 from .types.vizier_service import AddTrialMeasurementRequest
@@ -527,6 +533,7 @@
     "BatchReadTensorboardTimeSeriesDataResponse",
     "BigQueryDestination",
     "BigQuerySource",
+    "BlurBaselineConfig",
     "BoolArray",
     "CancelBatchPredictionJobRequest",
     "CancelCustomJobRequest",
@@ -810,6 +817,9 @@
     "ModelMonitoringObjectiveConfig",
     "ModelMonitoringStatsAnomalies",
     "ModelServiceClient",
+    "MutateDeployedIndexOperationMetadata",
+    "MutateDeployedIndexRequest",
+    "MutateDeployedIndexResponse",
     "NearestNeighborSearchOperationMetadata",
     "PauseModelDeploymentMonitoringJobRequest",
     "PipelineJob",
@@ -862,6 +872,7 @@
     "SpecialistPool",
     "SpecialistPoolServiceClient",
     "StopTrialRequest",
+    "StratifiedSplit",
     "StreamingReadFeatureValuesRequest",
     "StringArray",
     "Study",
@@ -891,6 +902,7 @@
     "UndeployModelOperationMetadata",
     "UndeployModelRequest",
     "UndeployModelResponse",
+    "UnmanagedContainerModel",
     "UpdateArtifactRequest",
     "UpdateContextRequest",
     "UpdateDatasetRequest",

google/cloud/aiplatform_v1/gapic_metadata.json

@@ -481,6 +481,11 @@
                 "list_index_endpoints"
               ]
             },
+            "MutateDeployedIndex": {
+              "methods": [
+                "mutate_deployed_index"
+              ]
+            },
             "UndeployIndex": {
               "methods": [
                 "undeploy_index"
@@ -521,6 +526,11 @@
                 "list_index_endpoints"
               ]
             },
+            "MutateDeployedIndex": {
+              "methods": [
+                "mutate_deployed_index"
+              ]
+            },
             "UndeployIndex": {
               "methods": [
                 "undeploy_index"

google/cloud/aiplatform_v1/services/endpoint_service/async_client.py

@@ -191,6 +191,7 @@ async def create_endpoint(
         *,
         parent: str = None,
         endpoint: gca_endpoint.Endpoint = None,
+        endpoint_id: str = None,
         retry: OptionalRetry = gapic_v1.method.DEFAULT,
         timeout: float = None,
         metadata: Sequence[Tuple[str, str]] = (),
@@ -214,6 +215,21 @@ async def create_endpoint(
                 This corresponds to the ``endpoint`` field
                 on the ``request`` instance; if ``request`` is provided, this
                 should not be set.
+            endpoint_id (:class:`str`):
+                Immutable. The ID to use for endpoint, which will become
+                the final component of the endpoint resource name. If
+                not provided, Vertex AI will generate a value for this
+                ID.
+
+                This value should be 1-10 characters, and valid
+                characters are /[0-9]/. When using HTTP/JSON, this field
+                is populated based on a query string argument, such as
+                ``?endpoint_id=12345``. This is the fallback for fields
+                that are not included in either the URI or the body.
+
+                This corresponds to the ``endpoint_id`` field
+                on the ``request`` instance; if ``request`` is provided, this
+                should not be set.
             retry (google.api_core.retry.Retry): Designation of what errors, if any,
                 should be retried.
             timeout (float): The timeout for this request.
@@ -231,7 +247,7 @@ async def create_endpoint(
         # Create or coerce a protobuf request object.
         # Sanity check: If we got a request object, we should *not* have
         # gotten any keyword arguments that map to the request.
-        has_flattened_params = any([parent, endpoint])
+        has_flattened_params = any([parent, endpoint, endpoint_id])
         if request is not None and has_flattened_params:
             raise ValueError(
                 "If the `request` argument is set, then none of "
@@ -246,6 +262,8 @@ async def create_endpoint(
             request.parent = parent
         if endpoint is not None:
             request.endpoint = endpoint
+        if endpoint_id is not None:
+            request.endpoint_id = endpoint_id
 
         # Wrap the RPC method; this adds retry and timeout information,
         # and friendly error handling.

google/cloud/aiplatform_v1/services/endpoint_service/client.py

@@ -424,6 +424,7 @@ def create_endpoint(
         *,
         parent: str = None,
         endpoint: gca_endpoint.Endpoint = None,
+        endpoint_id: str = None,
         retry: OptionalRetry = gapic_v1.method.DEFAULT,
         timeout: float = None,
         metadata: Sequence[Tuple[str, str]] = (),
@@ -447,6 +448,21 @@ def create_endpoint(
                 This corresponds to the ``endpoint`` field
                 on the ``request`` instance; if ``request`` is provided, this
                 should not be set.
+            endpoint_id (str):
+                Immutable. The ID to use for endpoint, which will become
+                the final component of the endpoint resource name. If
+                not provided, Vertex AI will generate a value for this
+                ID.
+
+                This value should be 1-10 characters, and valid
+                characters are /[0-9]/. When using HTTP/JSON, this field
+                is populated based on a query string argument, such as
+                ``?endpoint_id=12345``. This is the fallback for fields
+                that are not included in either the URI or the body.
+
+                This corresponds to the ``endpoint_id`` field
+                on the ``request`` instance; if ``request`` is provided, this
+                should not be set.
             retry (google.api_core.retry.Retry): Designation of what errors, if any,
                 should be retried.
             timeout (float): The timeout for this request.
@@ -464,7 +480,7 @@ def create_endpoint(
         # Create or coerce a protobuf request object.
         # Sanity check: If we got a request object, we should *not* have
         # gotten any keyword arguments that map to the request.
-        has_flattened_params = any([parent, endpoint])
+        has_flattened_params = any([parent, endpoint, endpoint_id])
         if request is not None and has_flattened_params:
             raise ValueError(
                 "If the `request` argument is set, then none of "
@@ -483,6 +499,8 @@ def create_endpoint(
                 request.parent = parent
             if endpoint is not None:
                 request.endpoint = endpoint
+            if endpoint_id is not None:
+                request.endpoint_id = endpoint_id
 
         # Wrap the RPC method; this adds retry and timeout information,
         # and friendly error handling.

google/cloud/aiplatform_v1/services/featurestore_service/async_client.py

@@ -196,6 +196,7 @@ async def create_featurestore(
         *,
         parent: str = None,
         featurestore: gca_featurestore.Featurestore = None,
+        featurestore_id: str = None,
         retry: OptionalRetry = gapic_v1.method.DEFAULT,
         timeout: float = None,
         metadata: Sequence[Tuple[str, str]] = (),
@@ -220,6 +221,21 @@ async def create_featurestore(
                 This corresponds to the ``featurestore`` field
                 on the ``request`` instance; if ``request`` is provided, this
                 should not be set.
+            featurestore_id (:class:`str`):
+                Required. The ID to use for this Featurestore, which
+                will become the final component of the Featurestore's
+                resource name.
+
+                This value may be up to 60 characters, and valid
+                characters are ``[a-z0-9_]``. The first character cannot
+                be a number.
+
+                The value must be unique within the project and
+                location.
+
+                This corresponds to the ``featurestore_id`` field
+                on the ``request`` instance; if ``request`` is provided, this
+                should not be set.
             retry (google.api_core.retry.Retry): Designation of what errors, if any,
                 should be retried.
             timeout (float): The timeout for this request.
@@ -230,7 +246,7 @@ async def create_featurestore(
             google.api_core.operation_async.AsyncOperation:
                 An object representing a long-running operation.
 
-                The result type for the operation will be :class:`google.cloud.aiplatform_v1.types.Featurestore` Vertex Feature Store provides a centralized repository for organizing,
+                The result type for the operation will be :class:`google.cloud.aiplatform_v1.types.Featurestore` Vertex AI Feature Store provides a centralized repository for organizing,
                    storing, and serving ML features. The Featurestore is
                    a top-level container for your features and their
                    values.
@@ -239,7 +255,7 @@ async def create_featurestore(
         # Create or coerce a protobuf request object.
         # Sanity check: If we got a request object, we should *not* have
         # gotten any keyword arguments that map to the request.
-        has_flattened_params = any([parent, featurestore])
+        has_flattened_params = any([parent, featurestore, featurestore_id])
         if request is not None and has_flattened_params:
             raise ValueError(
                 "If the `request` argument is set, then none of "
@@ -254,6 +270,8 @@ async def create_featurestore(
             request.parent = parent
         if featurestore is not None:
             request.featurestore = featurestore
+        if featurestore_id is not None:
+            request.featurestore_id = featurestore_id
 
         # Wrap the RPC method; this adds retry and timeout information,
         # and friendly error handling.
@@ -313,7 +331,7 @@ async def get_featurestore(
 
         Returns:
             google.cloud.aiplatform_v1.types.Featurestore:
-                Vertex Feature Store provides a
+                Vertex AI Feature Store provides a
                 centralized repository for organizing,
                 storing, and serving ML features. The
                 Featurestore is a top-level container
@@ -490,7 +508,7 @@ async def update_featurestore(
             google.api_core.operation_async.AsyncOperation:
                 An object representing a long-running operation.
 
-                The result type for the operation will be :class:`google.cloud.aiplatform_v1.types.Featurestore` Vertex Feature Store provides a centralized repository for organizing,
+                The result type for the operation will be :class:`google.cloud.aiplatform_v1.types.Featurestore` Vertex AI Feature Store provides a centralized repository for organizing,
                    storing, and serving ML features. The Featurestore is
                    a top-level container for your features and their
                    values.
@@ -659,6 +677,7 @@ async def create_entity_type(
         *,
         parent: str = None,
         entity_type: gca_entity_type.EntityType = None,
+        entity_type_id: str = None,
         retry: OptionalRetry = gapic_v1.method.DEFAULT,
         timeout: float = None,
         metadata: Sequence[Tuple[str, str]] = (),
@@ -682,6 +701,20 @@ async def create_entity_type(
                 This corresponds to the ``entity_type`` field
                 on the ``request`` instance; if ``request`` is provided, this
                 should not be set.
+            entity_type_id (:class:`str`):
+                Required. The ID to use for the EntityType, which will
+                become the final component of the EntityType's resource
+                name.
+
+                This value may be up to 60 characters, and valid
+                characters are ``[a-z0-9_]``. The first character cannot
+                be a number.
+
+                The value must be unique within a featurestore.
+
+                This corresponds to the ``entity_type_id`` field
+                on the ``request`` instance; if ``request`` is provided, this
+                should not be set.
             retry (google.api_core.retry.Retry): Designation of what errors, if any,
                 should be retried.
             timeout (float): The timeout for this request.
@@ -701,7 +734,7 @@ async def create_entity_type(
         # Create or coerce a protobuf request object.
         # Sanity check: If we got a request object, we should *not* have
         # gotten any keyword arguments that map to the request.
-        has_flattened_params = any([parent, entity_type])
+        has_flattened_params = any([parent, entity_type, entity_type_id])
         if request is not None and has_flattened_params:
             raise ValueError(
                 "If the `request` argument is set, then none of "
@@ -716,6 +749,8 @@ async def create_entity_type(
             request.parent = parent
         if entity_type is not None:
             request.entity_type = entity_type
+        if entity_type_id is not None:
+            request.entity_type_id = entity_type_id
 
         # Wrap the RPC method; this adds retry and timeout information,
         # and friendly error handling.
@@ -1115,6 +1150,7 @@ async def create_feature(
         *,
         parent: str = None,
         feature: gca_feature.Feature = None,
+        feature_id: str = None,
         retry: OptionalRetry = gapic_v1.method.DEFAULT,
         timeout: float = None,
         metadata: Sequence[Tuple[str, str]] = (),
@@ -1138,6 +1174,20 @@ async def create_feature(
                 This corresponds to the ``feature`` field
                 on the ``request`` instance; if ``request`` is provided, this
                 should not be set.
+            feature_id (:class:`str`):
+                Required. The ID to use for the Feature, which will
+                become the final component of the Feature's resource
+                name.
+
+                This value may be up to 60 characters, and valid
+                characters are ``[a-z0-9_]``. The first character cannot
+                be a number.
+
+                The value must be unique within an EntityType.
+
+                This corresponds to the ``feature_id`` field
+                on the ``request`` instance; if ``request`` is provided, this
+                should not be set.
             retry (google.api_core.retry.Retry): Designation of what errors, if any,
                 should be retried.
             timeout (float): The timeout for this request.
@@ -1156,7 +1206,7 @@ async def create_feature(
         # Create or coerce a protobuf request object.
         # Sanity check: If we got a request object, we should *not* have
         # gotten any keyword arguments that map to the request.
-        has_flattened_params = any([parent, feature])
+        has_flattened_params = any([parent, feature, feature_id])
         if request is not None and has_flattened_params:
             raise ValueError(
                 "If the `request` argument is set, then none of "
@@ -1171,6 +1221,8 @@ async def create_feature(
             request.parent = parent
         if feature is not None:
             request.feature = feature
+        if feature_id is not None:
+            request.feature_id = feature_id
 
         # Wrap the RPC method; this adds retry and timeout information,
         # and friendly error handling.