Copybara import of the project:

-- 255d3de by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: feat: add Gen AI logging public preview API PiperOrigin-RevId: 743220774 Source-Link: googleapis/googleapis@3b446a9 Source-Link: googleapis/googleapis-gen@52443a8 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiNTI0NDNhOGRmZDc0MGZmZDU4OTY1YzI3ZjExMzI4NGQyNGZmYTU2ZiJ9 -- 32983cd by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md COPYBARA_INTEGRATE_REVIEW=#5139 from googleapis:owl-bot-copy ee83524 PiperOrigin-RevId: 743328858

Author: gcf-owl-bot[bot]

google/cloud/aiplatform_v1beta1/__init__.py

@@ -223,19 +223,23 @@
 from .types.endpoint import FasterDeploymentConfig
 from .types.endpoint import PredictRequestResponseLoggingConfig
 from .types.endpoint import PrivateEndpoints
+from .types.endpoint import PublisherModelConfig
 from .types.endpoint import RolloutOptions
 from .types.endpoint_service import CreateEndpointOperationMetadata
 from .types.endpoint_service import CreateEndpointRequest
 from .types.endpoint_service import DeleteEndpointRequest
 from .types.endpoint_service import DeployModelOperationMetadata
 from .types.endpoint_service import DeployModelRequest
 from .types.endpoint_service import DeployModelResponse
+from .types.endpoint_service import FetchPublisherModelConfigRequest
 from .types.endpoint_service import GetEndpointRequest
 from .types.endpoint_service import ListEndpointsRequest
 from .types.endpoint_service import ListEndpointsResponse
 from .types.endpoint_service import MutateDeployedModelOperationMetadata
 from .types.endpoint_service import MutateDeployedModelRequest
 from .types.endpoint_service import MutateDeployedModelResponse
+from .types.endpoint_service import SetPublisherModelConfigOperationMetadata
+from .types.endpoint_service import SetPublisherModelConfigRequest
 from .types.endpoint_service import UndeployModelOperationMetadata
 from .types.endpoint_service import UndeployModelRequest
 from .types.endpoint_service import UndeployModelResponse
@@ -1661,6 +1665,7 @@
  "FetchExamplesResponse",
  "FetchFeatureValuesRequest",
  "FetchFeatureValuesResponse",
+ "FetchPublisherModelConfigRequest",
  "FileData",
  "FileStatus",
  "FilterSplit",
@@ -2051,6 +2056,7 @@
  "PscAutomatedEndpoints",
  "PscInterfaceConfig",
  "PublisherModel",
+ "PublisherModelConfig",
  "PublisherModelView",
  "PurgeArtifactsMetadata",
  "PurgeArtifactsRequest",
@@ -2197,6 +2203,8 @@
  "Session",
  "SessionEvent",
  "SessionServiceClient",
+ "SetPublisherModelConfigOperationMetadata",
+ "SetPublisherModelConfigRequest",
  "SharePointSources",
  "ShieldedVmConfig",
  "SlackSource",

google/cloud/aiplatform_v1beta1/gapic_metadata.json

@@ -468,6 +468,11 @@
  "deploy_model"
  ]
  },
+ "FetchPublisherModelConfig": {
+ "methods": [
+ "fetch_publisher_model_config"
+ ]
+ },
  "GetEndpoint": {
  "methods": [
  "get_endpoint"
@@ -483,6 +488,11 @@
  "mutate_deployed_model"
  ]
  },
+ "SetPublisherModelConfig": {
+ "methods": [
+ "set_publisher_model_config"
+ ]
+ },
  "UndeployModel": {
  "methods": [
  "undeploy_model"
@@ -518,6 +528,11 @@
  "deploy_model"
  ]
  },
+ "FetchPublisherModelConfig": {
+ "methods": [
+ "fetch_publisher_model_config"
+ ]
+ },
  "GetEndpoint": {
  "methods": [
  "get_endpoint"
@@ -533,6 +548,11 @@
  "mutate_deployed_model"
  ]
  },
+ "SetPublisherModelConfig": {
+ "methods": [
+ "set_publisher_model_config"
+ ]
+ },
  "UndeployModel": {
  "methods": [
  "undeploy_model"
@@ -568,6 +588,11 @@
  "deploy_model"
  ]
  },
+ "FetchPublisherModelConfig": {
+ "methods": [
+ "fetch_publisher_model_config"
+ ]
+ },
  "GetEndpoint": {
  "methods": [
  "get_endpoint"
@@ -583,6 +608,11 @@
  "mutate_deployed_model"
  ]
  },
+ "SetPublisherModelConfig": {
+ "methods": [
+ "set_publisher_model_config"
+ ]
+ },
  "UndeployModel": {
  "methods": [
  "undeploy_model"

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

@@ -1613,6 +1613,259 @@ async def sample_mutate_deployed_model():
  # Done; return the response.
  return response
 
+ async def set_publisher_model_config(
+ self,
+ request: Optional[
+ Union[endpoint_service.SetPublisherModelConfigRequest, dict]
+ ] = None,
+ *,
+ name: Optional[str] = None,
+ publisher_model_config: Optional[endpoint.PublisherModelConfig] = None,
+ retry: OptionalRetry = gapic_v1.method.DEFAULT,
+ timeout: Union[float, object] = gapic_v1.method.DEFAULT,
+ metadata: Sequence[Tuple[str, Union[str, bytes]]] = (),
+ ) -> operation_async.AsyncOperation:
+ r"""Sets (creates or updates) configs of publisher
+ models. For example, sets the request/response logging
+ config.
+
+ .. 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_v1beta1
+
+ async def sample_set_publisher_model_config():
+ # Create a client
+ client = aiplatform_v1beta1.EndpointServiceAsyncClient()
+
+ # Initialize request argument(s)
+ request = aiplatform_v1beta1.SetPublisherModelConfigRequest(
+ name="name_value",
+ )
+
+ # Make the request
+ operation = client.set_publisher_model_config(request=request)
+
+ print("Waiting for operation to complete...")
+
+ response = (await operation).result()
+
+ # Handle the response
+ print(response)
+
+ Args:
+ request (Optional[Union[google.cloud.aiplatform_v1beta1.types.SetPublisherModelConfigRequest, dict]]):
+ The request object. Request message for
+ [EndpointService.SetPublisherModelConfig][google.cloud.aiplatform.v1beta1.EndpointService.SetPublisherModelConfig].
+ name (:class:`str`):
+ Required. The name of the publisher model, in the format
+ of
+ ``projects/{project}/locations/{location}/publishers/{publisher}/models/{model}``.
+
+ This corresponds to the ``name`` field
+ on the ``request`` instance; if ``request`` is provided, this
+ should not be set.
+ publisher_model_config (:class:`google.cloud.aiplatform_v1beta1.types.PublisherModelConfig`):
+ Required. The publisher model config.
+ This corresponds to the ``publisher_model_config`` field
+ on the ``request`` instance; if ``request`` is provided, this
+ should not be set.
+ 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:
+ 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_v1beta1.types.PublisherModelConfig`
+ This message contains configs of a publisher model.
+
+ """
+ # Create or coerce a protobuf request object.
+ # - Quick check: If we got a request object, we should *not* have
+ # gotten any keyword arguments that map to the request.
+ flattened_params = [name, publisher_model_config]
+ has_flattened_params = (
+ len([param for param in flattened_params if param is not None]) > 0
+ )
+ if request is not None and has_flattened_params:
+ raise ValueError(
+ "If the `request` argument is set, then none of "
+ "the individual field arguments should be set."
+ )
+
+ # - Use the request object if provided (there's no risk of modifying the input as
+ # there are no flattened fields), or create one.
+ if not isinstance(request, endpoint_service.SetPublisherModelConfigRequest):
+ request = endpoint_service.SetPublisherModelConfigRequest(request)
+
+ # If we have keyword arguments corresponding to fields on the
+ # request, apply these.
+ if name is not None:
+ request.name = name
+ if publisher_model_config is not None:
+ request.publisher_model_config = publisher_model_config
+
+ # Wrap the RPC method; this adds retry and timeout information,
+ # and friendly error handling.
+ rpc = self._client._transport._wrapped_methods[
+ self._client._transport.set_publisher_model_config
+ ]
+
+ # Certain fields should be provided within the metadata header;
+ # add these here.
+ metadata = tuple(metadata) + (
+ gapic_v1.routing_header.to_grpc_metadata((("name", request.name),)),
+ )
+
+ # Validate the universe domain.
+ self._client._validate_universe_domain()
+
+ # Send the request.
+ response = await rpc(
+ request,
+ retry=retry,
+ timeout=timeout,
+ metadata=metadata,
+ )
+
+ # Wrap the response in an operation future.
+ response = operation_async.from_gapic(
+ response,
+ self._client._transport.operations_client,
+ endpoint.PublisherModelConfig,
+ metadata_type=endpoint_service.SetPublisherModelConfigOperationMetadata,
+ )
+
+ # Done; return the response.
+ return response
+
+ async def fetch_publisher_model_config(
+ self,
+ request: Optional[
+ Union[endpoint_service.FetchPublisherModelConfigRequest, dict]
+ ] = None,
+ *,
+ name: Optional[str] = None,
+ retry: OptionalRetry = gapic_v1.method.DEFAULT,
+ timeout: Union[float, object] = gapic_v1.method.DEFAULT,
+ metadata: Sequence[Tuple[str, Union[str, bytes]]] = (),
+ ) -> endpoint.PublisherModelConfig:
+ r"""Fetches the configs of publisher models.
+
+ .. 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_v1beta1
+
+ async def sample_fetch_publisher_model_config():
+ # Create a client
+ client = aiplatform_v1beta1.EndpointServiceAsyncClient()
+
+ # Initialize request argument(s)
+ request = aiplatform_v1beta1.FetchPublisherModelConfigRequest(
+ name="name_value",
+ )
+
+ # Make the request
+ response = await client.fetch_publisher_model_config(request=request)
+
+ # Handle the response
+ print(response)
+
+ Args:
+ request (Optional[Union[google.cloud.aiplatform_v1beta1.types.FetchPublisherModelConfigRequest, dict]]):
+ The request object. Request message for
+ [EndpointService.FetchPublisherModelConfig][google.cloud.aiplatform.v1beta1.EndpointService.FetchPublisherModelConfig].
+ name (:class:`str`):
+ Required. The name of the publisher model, in the format
+ of
+ ``projects/{project}/locations/{location}/publishers/{publisher}/models/{model}``.
+
+ This corresponds to the ``name`` field
+ on the ``request`` instance; if ``request`` is provided, this
+ should not be set.
+ 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:
+ google.cloud.aiplatform_v1beta1.types.PublisherModelConfig:
+ This message contains configs of a
+ publisher model.
+
+ """
+ # Create or coerce a protobuf request object.
+ # - Quick check: If we got a request object, we should *not* have
+ # gotten any keyword arguments that map to the request.
+ flattened_params = [name]
+ has_flattened_params = (
+ len([param for param in flattened_params if param is not None]) > 0
+ )
+ if request is not None and has_flattened_params:
+ raise ValueError(
+ "If the `request` argument is set, then none of "
+ "the individual field arguments should be set."
+ )
+
+ # - Use the request object if provided (there's no risk of modifying the input as
+ # there are no flattened fields), or create one.
+ if not isinstance(request, endpoint_service.FetchPublisherModelConfigRequest):
+ request = endpoint_service.FetchPublisherModelConfigRequest(request)
+
+ # If we have keyword arguments corresponding to fields on the
+ # request, apply these.
+ if name is not None:
+ request.name = name
+
+ # Wrap the RPC method; this adds retry and timeout information,
+ # and friendly error handling.
+ rpc = self._client._transport._wrapped_methods[
+ self._client._transport.fetch_publisher_model_config
+ ]
+
+ # Certain fields should be provided within the metadata header;
+ # add these here.
+ metadata = tuple(metadata) + (
+ gapic_v1.routing_header.to_grpc_metadata((("name", request.name),)),
+ )
+
+ # Validate the universe domain.
+ self._client._validate_universe_domain()
+
+ # Send the request.
+ response = await rpc(
+ request,
+ retry=retry,
+ timeout=timeout,
+ metadata=metadata,
+ )
+
+ # Done; return the response.
+ return response
+
  async def list_operations(
  self,
  request: Optional[operations_pb2.ListOperationsRequest] = None,