googleapis
diff --git a/‎google/cloud/aiplatform_v1/__init__.py‎
Lines changed: 8 additions & 0 deletions b/‎google/cloud/aiplatform_v1/__init__.py‎
Lines changed: 8 additions & 0 deletions
diff --git a/‎google/cloud/aiplatform_v1/gapic_metadata.json‎
Lines changed: 15 additions & 0 deletions b/‎google/cloud/aiplatform_v1/gapic_metadata.json‎
Lines changed: 15 additions & 0 deletions
diff --git a/‎google/cloud/aiplatform_v1/services/endpoint_service/async_client.py‎
Lines changed: 5 additions & 2 deletions b/‎google/cloud/aiplatform_v1/services/endpoint_service/async_client.py‎
Lines changed: 5 additions & 2 deletions
diff --git a/‎google/cloud/aiplatform_v1/services/endpoint_service/client.py‎
Lines changed: 5 additions & 2 deletions b/‎google/cloud/aiplatform_v1/services/endpoint_service/client.py‎
Lines changed: 5 additions & 2 deletions
diff --git a/‎google/cloud/aiplatform_v1/services/endpoint_service/transports/grpc.py‎
Lines changed: 3 additions & 2 deletions b/‎google/cloud/aiplatform_v1/services/endpoint_service/transports/grpc.py‎
Lines changed: 3 additions & 2 deletions
diff --git a/‎google/cloud/aiplatform_v1/services/endpoint_service/transports/grpc_asyncio.py‎
Lines changed: 3 additions & 2 deletions b/‎google/cloud/aiplatform_v1/services/endpoint_service/transports/grpc_asyncio.py‎
Lines changed: 3 additions & 2 deletions
diff --git a/‎google/cloud/aiplatform_v1/services/feature_online_store_service/async_client.py‎
Lines changed: 100 additions & 0 deletions b/‎google/cloud/aiplatform_v1/services/feature_online_store_service/async_client.py‎
Lines changed: 100 additions & 0 deletions
diff --git a/‎google/cloud/aiplatform_v1/services/feature_online_store_service/client.py‎
Lines changed: 97 additions & 0 deletions b/‎google/cloud/aiplatform_v1/services/feature_online_store_service/client.py‎
Lines changed: 97 additions & 0 deletions
@@ -119,6 +119,7 @@
 from .types.content import GroundingSupport
 from .types.content import LogprobsResult
 from .types.content import ModalityTokenCount
+from .types.content import ModelArmorConfig
 from .types.content import Part
 from .types.content import RetrievalMetadata
 from .types.content import SafetyRating
@@ -198,6 +199,7 @@
     UpdateDeploymentResourcePoolOperationMetadata,
 )
 from .types.deployment_resource_pool_service import UpdateDeploymentResourcePoolRequest
+from .types.deployment_stage import DeploymentStage
 from .types.encryption_spec import EncryptionSpec
 from .types.endpoint import ClientConnectionConfig
 from .types.endpoint import DeployedModel
@@ -393,6 +395,8 @@
 from .types.feature_online_store_admin_service import UpdateFeatureViewOperationMetadata
 from .types.feature_online_store_admin_service import UpdateFeatureViewRequest
 from .types.feature_online_store_service import FeatureViewDataKey
+from .types.feature_online_store_service import FeatureViewDirectWriteRequest
+from .types.feature_online_store_service import FeatureViewDirectWriteResponse
 from .types.feature_online_store_service import FetchFeatureValuesRequest
 from .types.feature_online_store_service import FetchFeatureValuesResponse
 from .types.feature_online_store_service import NearestNeighborQuery
@@ -1319,6 +1323,7 @@
     "DeployedModelRef",
     "DeploymentResourcePool",
     "DeploymentResourcePoolServiceClient",
+    "DeploymentStage",
     "DestinationFeatureSetting",
     "DirectPredictRequest",
     "DirectPredictResponse",
@@ -1392,6 +1397,8 @@
     "FeatureView",
     "FeatureViewDataFormat",
     "FeatureViewDataKey",
+    "FeatureViewDirectWriteRequest",
+    "FeatureViewDirectWriteResponse",
     "FeatureViewSync",
     "Featurestore",
     "FeaturestoreMonitoringConfig",
@@ -1644,6 +1651,7 @@
     "Modality",
     "ModalityTokenCount",
     "Model",
+    "ModelArmorConfig",
     "ModelContainerSpec",
     "ModelDeploymentMonitoringBigQueryTable",
     "ModelDeploymentMonitoringJob",
 
@@ -825,6 +825,11 @@
         "grpc": {
           "libraryClient": "FeatureOnlineStoreServiceClient",
           "rpcs": {
+            "FeatureViewDirectWrite": {
+              "methods": [
+                "feature_view_direct_write"
+              ]
+            },
             "FetchFeatureValues": {
               "methods": [
                 "fetch_feature_values"
@@ -840,6 +845,11 @@
         "grpc-async": {
           "libraryClient": "FeatureOnlineStoreServiceAsyncClient",
           "rpcs": {
+            "FeatureViewDirectWrite": {
+              "methods": [
+                "feature_view_direct_write"
+              ]
+            },
             "FetchFeatureValues": {
               "methods": [
                 "fetch_feature_values"
@@ -855,6 +865,11 @@
         "rest": {
           "libraryClient": "FeatureOnlineStoreServiceClient",
           "rpcs": {
+            "FeatureViewDirectWrite": {
+              "methods": [
+                "feature_view_direct_write"
+              ]
+            },
             "FetchFeatureValues": {
               "methods": [
                 "fetch_feature_values"
 
@@ -1460,8 +1460,9 @@ async def mutate_deployed_model(
     ) -> operation_async.AsyncOperation:
         r"""Updates an existing deployed model. Updatable fields include
         ``min_replica_count``, ``max_replica_count``,
-        ``autoscaling_metric_specs``, ``disable_container_logging`` (v1
-        only), and ``enable_container_logging`` (v1beta1 only).
+        ``required_replica_count``, ``autoscaling_metric_specs``,
+        ``disable_container_logging`` (v1 only), and
+        ``enable_container_logging`` (v1beta1 only).
 
         .. code-block:: python
 
@@ -1521,6 +1522,8 @@ async def sample_mutate_deployed_model():
                    [DedicatedResources][google.cloud.aiplatform.v1.DedicatedResources]
                    or
                    [AutomaticResources][google.cloud.aiplatform.v1.AutomaticResources]
+                -  ``required_replica_count`` in
+                   [DedicatedResources][google.cloud.aiplatform.v1.DedicatedResources]
                 -  [autoscaling_metric_specs][google.cloud.aiplatform.v1.DedicatedResources.autoscaling_metric_specs]
                 -  ``disable_container_logging`` (v1 only)
                 -  ``enable_container_logging`` (v1beta1 only)
 
@@ -1986,8 +1986,9 @@ def mutate_deployed_model(
     ) -> gac_operation.Operation:
         r"""Updates an existing deployed model. Updatable fields include
         ``min_replica_count``, ``max_replica_count``,
-        ``autoscaling_metric_specs``, ``disable_container_logging`` (v1
-        only), and ``enable_container_logging`` (v1beta1 only).
+        ``required_replica_count``, ``autoscaling_metric_specs``,
+        ``disable_container_logging`` (v1 only), and
+        ``enable_container_logging`` (v1beta1 only).
 
         .. code-block:: python
 
@@ -2047,6 +2048,8 @@ def sample_mutate_deployed_model():
                    [DedicatedResources][google.cloud.aiplatform.v1.DedicatedResources]
                    or
                    [AutomaticResources][google.cloud.aiplatform.v1.AutomaticResources]
+                -  ``required_replica_count`` in
+                   [DedicatedResources][google.cloud.aiplatform.v1.DedicatedResources]
                 -  [autoscaling_metric_specs][google.cloud.aiplatform.v1.DedicatedResources.autoscaling_metric_specs]
                 -  ``disable_container_logging`` (v1 only)
                 -  ``enable_container_logging`` (v1beta1 only)
 
@@ -571,8 +571,9 @@ def mutate_deployed_model(
 
         Updates an existing deployed model. Updatable fields include
         ``min_replica_count``, ``max_replica_count``,
-        ``autoscaling_metric_specs``, ``disable_container_logging`` (v1
-        only), and ``enable_container_logging`` (v1beta1 only).
+        ``required_replica_count``, ``autoscaling_metric_specs``,
+        ``disable_container_logging`` (v1 only), and
+        ``enable_container_logging`` (v1beta1 only).
 
         Returns:
             Callable[[~.MutateDeployedModelRequest],
 
@@ -592,8 +592,9 @@ def mutate_deployed_model(
 
         Updates an existing deployed model. Updatable fields include
         ``min_replica_count``, ``max_replica_count``,
-        ``autoscaling_metric_specs``, ``disable_container_logging`` (v1
-        only), and ``enable_container_logging`` (v1beta1 only).
+        ``required_replica_count``, ``autoscaling_metric_specs``,
+        ``disable_container_logging`` (v1 only), and
+        ``enable_container_logging`` (v1beta1 only).
 
         Returns:
             Callable[[~.MutateDeployedModelRequest],
 
@@ -23,6 +23,9 @@
     MutableMapping,
     MutableSequence,
     Optional,
+    AsyncIterable,
+    Awaitable,
+    AsyncIterator,
     Sequence,
     Tuple,
     Type,
@@ -51,6 +54,7 @@
 from google.iam.v1 import policy_pb2  # type: ignore
 from google.longrunning import operations_pb2  # type: ignore
 from google.protobuf import struct_pb2  # type: ignore
+from google.rpc import status_pb2  # type: ignore
 from .transports.base import FeatureOnlineStoreServiceTransport, DEFAULT_CLIENT_INFO
 from .transports.grpc_asyncio import FeatureOnlineStoreServiceGrpcAsyncIOTransport
 from .client import FeatureOnlineStoreServiceClient
@@ -533,6 +537,102 @@ async def sample_search_nearest_entities():
         # Done; return the response.
         return response
 
+    def feature_view_direct_write(
+        self,
+        requests: Optional[
+            AsyncIterator[feature_online_store_service.FeatureViewDirectWriteRequest]
+        ] = None,
+        *,
+        retry: OptionalRetry = gapic_v1.method.DEFAULT,
+        timeout: Union[float, object] = gapic_v1.method.DEFAULT,
+        metadata: Sequence[Tuple[str, Union[str, bytes]]] = (),
+    ) -> Awaitable[
+        AsyncIterable[feature_online_store_service.FeatureViewDirectWriteResponse]
+    ]:
+        r"""Bidirectional streaming RPC to directly write to
+        feature values in a feature view. Requests may not have
+        a one-to-one mapping to responses and responses may be
+        returned out-of-order to reduce latency.
+
+        .. code-block:: python
+
+            # This snippet has been automatically generated and should be regarded as a
+            # code template only.
+            # It will require modifications to work:
+            # - It may require correct/in-range values for request initialization.
+            # - It may require specifying regional endpoints when creating the service
+            #   client as shown in:
+            #   https://googleapis.dev/python/google-api-core/latest/client_options.html
+            from google.cloud import aiplatform_v1
+
+            async def sample_feature_view_direct_write():
+                # Create a client
+                client = aiplatform_v1.FeatureOnlineStoreServiceAsyncClient()
+
+                # Initialize request argument(s)
+                request = aiplatform_v1.FeatureViewDirectWriteRequest(
+                )
+
+                # This method expects an iterator which contains
+                # 'aiplatform_v1.FeatureViewDirectWriteRequest' objects
+                # Here we create a generator that yields a single `request` for
+                # demonstrative purposes.
+                requests = [request]
+
+                def request_generator():
+                    for request in requests:
+                        yield request
+
+                # Make the request
+                stream = await client.feature_view_direct_write(requests=request_generator())
+
+                # Handle the response
+                async for response in stream:
+                    print(response)
+
+        Args:
+            requests (AsyncIterator[`google.cloud.aiplatform_v1.types.FeatureViewDirectWriteRequest`]):
+                The request object AsyncIterator. Request message for
+                [FeatureOnlineStoreService.FeatureViewDirectWrite][google.cloud.aiplatform.v1.FeatureOnlineStoreService.FeatureViewDirectWrite].
+            retry (google.api_core.retry_async.AsyncRetry): Designation of what errors, if any,
+                should be retried.
+            timeout (float): The timeout for this request.
+            metadata (Sequence[Tuple[str, Union[str, bytes]]]): Key/value pairs which should be
+                sent along with the request as metadata. Normally, each value must be of type `str`,
+                but for metadata keys ending with the suffix `-bin`, the corresponding values must
+                be of type `bytes`.
+
+        Returns:
+            AsyncIterable[google.cloud.aiplatform_v1.types.FeatureViewDirectWriteResponse]:
+                Response message for
+                   [FeatureOnlineStoreService.FeatureViewDirectWrite][google.cloud.aiplatform.v1.FeatureOnlineStoreService.FeatureViewDirectWrite].
+
+        """
+
+        # Wrap the RPC method; this adds retry and timeout information,
+        # and friendly error handling.
+        rpc = self._client._transport._wrapped_methods[
+            self._client._transport.feature_view_direct_write
+        ]
+
+        # Certain fields should be provided within the metadata header;
+        # add these here.
+        metadata = tuple(metadata) + (gapic_v1.routing_header.to_grpc_metadata(()),)
+
+        # Validate the universe domain.
+        self._client._validate_universe_domain()
+
+        # Send the request.
+        response = rpc(
+            requests,
+            retry=retry,
+            timeout=timeout,
+            metadata=metadata,
+        )
+
+        # Done; return the response.
+        return response
+
     async def list_operations(
         self,
         request: Optional[operations_pb2.ListOperationsRequest] = None,
 
@@ -26,6 +26,8 @@
     MutableMapping,
     MutableSequence,
     Optional,
+    Iterable,
+    Iterator,
     Sequence,
     Tuple,
     Type,
@@ -67,6 +69,7 @@
 from google.iam.v1 import policy_pb2  # type: ignore
 from google.longrunning import operations_pb2  # type: ignore
 from google.protobuf import struct_pb2  # type: ignore
+from google.rpc import status_pb2  # type: ignore
 from .transports.base import FeatureOnlineStoreServiceTransport, DEFAULT_CLIENT_INFO
 from .transports.grpc import FeatureOnlineStoreServiceGrpcTransport
 from .transports.grpc_asyncio import FeatureOnlineStoreServiceGrpcAsyncIOTransport
@@ -988,6 +991,100 @@ def sample_search_nearest_entities():
         # Done; return the response.
         return response
 
+    def feature_view_direct_write(
+        self,
+        requests: Optional[
+            Iterator[feature_online_store_service.FeatureViewDirectWriteRequest]
+        ] = None,
+        *,
+        retry: OptionalRetry = gapic_v1.method.DEFAULT,
+        timeout: Union[float, object] = gapic_v1.method.DEFAULT,
+        metadata: Sequence[Tuple[str, Union[str, bytes]]] = (),
+    ) -> Iterable[feature_online_store_service.FeatureViewDirectWriteResponse]:
+        r"""Bidirectional streaming RPC to directly write to
+        feature values in a feature view. Requests may not have
+        a one-to-one mapping to responses and responses may be
+        returned out-of-order to reduce latency.
+
+        .. code-block:: python
+
+            # This snippet has been automatically generated and should be regarded as a
+            # code template only.
+            # It will require modifications to work:
+            # - It may require correct/in-range values for request initialization.
+            # - It may require specifying regional endpoints when creating the service
+            #   client as shown in:
+            #   https://googleapis.dev/python/google-api-core/latest/client_options.html
+            from google.cloud import aiplatform_v1
+
+            def sample_feature_view_direct_write():
+                # Create a client
+                client = aiplatform_v1.FeatureOnlineStoreServiceClient()
+
+                # Initialize request argument(s)
+                request = aiplatform_v1.FeatureViewDirectWriteRequest(
+                )
+
+                # This method expects an iterator which contains
+                # 'aiplatform_v1.FeatureViewDirectWriteRequest' objects
+                # Here we create a generator that yields a single `request` for
+                # demonstrative purposes.
+                requests = [request]
+
+                def request_generator():
+                    for request in requests:
+                        yield request
+
+                # Make the request
+                stream = client.feature_view_direct_write(requests=request_generator())
+
+                # Handle the response
+                for response in stream:
+                    print(response)
+
+        Args:
+            requests (Iterator[google.cloud.aiplatform_v1.types.FeatureViewDirectWriteRequest]):
+                The request object iterator. Request message for
+                [FeatureOnlineStoreService.FeatureViewDirectWrite][google.cloud.aiplatform.v1.FeatureOnlineStoreService.FeatureViewDirectWrite].
+            retry (google.api_core.retry.Retry): Designation of what errors, if any,
+                should be retried.
+            timeout (float): The timeout for this request.
+            metadata (Sequence[Tuple[str, Union[str, bytes]]]): Key/value pairs which should be
+                sent along with the request as metadata. Normally, each value must be of type `str`,
+                but for metadata keys ending with the suffix `-bin`, the corresponding values must
+                be of type `bytes`.
+
+        Returns:
+            Iterable[google.cloud.aiplatform_v1.types.FeatureViewDirectWriteResponse]:
+                Response message for
+                   [FeatureOnlineStoreService.FeatureViewDirectWrite][google.cloud.aiplatform.v1.FeatureOnlineStoreService.FeatureViewDirectWrite].
+
+        """
+
+        # Wrap the RPC method; this adds retry and timeout information,
+        # and friendly error handling.
+        rpc = self._transport._wrapped_methods[
+            self._transport.feature_view_direct_write
+        ]
+
+        # Certain fields should be provided within the metadata header;
+        # add these here.
+        metadata = tuple(metadata) + (gapic_v1.routing_header.to_grpc_metadata(()),)
+
+        # Validate the universe domain.
+        self._validate_universe_domain()
+
+        # Send the request.
+        response = rpc(
+            requests,
+            retry=retry,
+            timeout=timeout,
+            metadata=metadata,
+        )
+
+        # Done; return the response.
+        return response
+
     def __enter__(self) -> "FeatureOnlineStoreServiceClient":
         return self