Copybara import of the project:

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

feat: add ValueType.STRUCT to Feature
feat: add encryption_spec to FeatureOnlineStore
feat: add struct_value to FeatureValue
feat: add encryption_spec, service_account, disable_container_logging to DeploymentResourcePool
feat: add model_reference to Dataset
feat: add tool_config to GenerateContentRequest
feat: add dataplex_config to MetadataStore
feat: add model_reference to DatasetVersion
feat: add INVALID_SPARSE_DIMENSIONS, INVALID_SPARSE_EMBEDDING, INVALID_EMBEDDING to NearestNeighborSearchOperationMetadata.RecordError
feat: add valid_sparse_record_count, invalid_sparse_record_count to NearestNeighborSearchOperationMetadata.ContentValidationStats
feat: add sparse_embedding to IndexDatapoint
feat: add sparse_vectors_count to IndexStats
feat: add sparse_distance to FindNeighborsResponse.Neighbor
feat: add more fields in FindNeighborsRequest.Query
feat: add UpdateNotebookRuntimeTemplate to NotebookService
feat: add direct_notebook_source to NotebookExecutionJob
feat: add encryption_spec to NotebookRuntimeTemplate
feat: add idle_shutdown_config, encryption_spec, satisfies_pzs, satisfies_pzi to NotebookRuntime
feat: add RaySpec to PersistentResource
fix!: An existing message `Segment` is removed
fix!: An existing message `GroundingAttribution` is removed
fix!: An existing field `grounding_attributions` is removed from message `.google.cloud.aiplatform.v1beta1.GroundingMetadata`
fix!: An existing field `disable_attribution` is removed from message `.google.cloud.aiplatform.v1beta1.GoogleSearchRetrieval`
docs: A comment for field `feature_vector` in message `.google.cloud.aiplatform.v1beta1.IndexDatapoint` is changed
docs: A comment for field `vectors_count` in message `.google.cloud.aiplatform.v1beta1.IndexStats` is changed
docs: A comment for enum value `EMBEDDING_SIZE_MISMATCH` in enum `RecordErrorType` is changed
docs: A comment for field `exec` in message `.google.cloud.aiplatform.v1beta1.Probe` is changed

PiperOrigin-RevId: 638338486

Source-Link: googleapis/googleapis@c2a9b99

Source-Link: googleapis/googleapis-gen@c4b0bd6
Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiYzRiMGJkNjg2ZGQ2M2Q1YTM4ZGZiMjI0NmQxODBjMjExNTk3NzZhOCJ9

--
36a620c 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

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

feat: add ValueType.STRUCT to Feature
feat: add encryption_spec to FeatureOnlineStore
feat: add struct_value to FeatureValue
feat: add encryption_spec, service_account, disable_container_logging to DeploymentResourcePool
feat: add model_reference to Dataset
feat: add ChatCompletions to PredictionService
feat: add new GenAiCacheService and CachedContent
feat: add cached_content to GenerationContentRequest
feat: add dataplex_config to MetadataStore
feat: add model_reference to DatasetVersion
feat: add INVALID_SPARSE_DIMENSIONS, INVALID_SPARSE_EMBEDDING, INVALID_EMBEDDING to NearestNeighborSearchOperationMetadata.RecordError
feat: add valid_sparse_record_count, invalid_sparse_record_count to NearestNeighborSearchOperationMetadata.ContentValidationStats
feat: add sparse_embedding to IndexDatapoint
feat: add sparse_vectors_count to IndexStats
feat: add sparse_distance to FindNeighborsResponse.Neighbor
feat: add more fields in FindNeighborsRequest.Query
feat: add progress_percentage to ImportRagFilesOperationMetadata
feat: add UpdateNotebookRuntimeTemplate to NotebookService
feat: add direct_notebook_source to NotebookExecutionJob
feat: add encryption_spec to NotebookRuntimeTemplate
feat: add idle_shutdown_config, encryption_spec, satisfies_pzs, satisfies_pzi to NotebookRuntime
feat: add UpdateReasoningEngine to ReasoningEngineService
fix!: An existing field `app_id` is renamed to `engine_id` in message `.google.cloud.aiplatform.v1beta1.RuntimeConfig`
docs: A comment for field `description` in message `.google.cloud.aiplatform.v1beta1.ExtensionManifest` is changed
docs: A comment for field `serving_config_name` in message `.google.cloud.aiplatform.v1beta1.RuntimeConfig` is changed
docs: A comment for field `update_mask` in message `.google.cloud.aiplatform.v1beta1.UpdateExtensionRequest` is changed
docs: A comment for field `feature_vector` in message `.google.cloud.aiplatform.v1beta1.IndexDatapoint` is changed
docs: A comment for field `vectors_count` in message `.google.cloud.aiplatform.v1beta1.IndexStats` is changed
docs: A comment for enum value `EMBEDDING_SIZE_MISMATCH` in enum `RecordErrorType` is changed
docs: A comment for field `exec` in message `.google.cloud.aiplatform.v1beta1.Probe` is changed
docs: A comment for field `create_notebook_execution_job_request` in message `.google.cloud.aiplatform.v1beta1.Schedule` is changed
docs: A comment for field `INVALID_EMBEDDING` in message `NearestNeighborSearchOperationMetadata.RecordError` is changed

PiperOrigin-RevId: 638343512

Source-Link: googleapis/googleapis@41b263f

Source-Link: googleapis/googleapis-gen@d6ce0c1
Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiZDZjZTBjMTVkYmFiNzc2ODJkZWRhY2MzMDZhNmI4YjNmMGY4NTAwZiJ9

--
06184d9 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=#3843 from googleapis:owl-bot-copy abd1d2f
PiperOrigin-RevId: 638427889

Author: gcf-owl-bot[bot]

docs/aiplatform_v1beta1/gen_ai_cache_service.rst

@@ -0,0 +1,10 @@
+GenAiCacheService
+-----------------------------------
+
+.. automodule:: google.cloud.aiplatform_v1beta1.services.gen_ai_cache_service
+    :members:
+    :inherited-members:
+
+.. automodule:: google.cloud.aiplatform_v1beta1.services.gen_ai_cache_service.pagers
+    :members:
+    :inherited-members:

docs/aiplatform_v1beta1/gen_ai_tuning_service.rst

@@ -0,0 +1,10 @@
+GenAiTuningService
+------------------------------------
+
+.. automodule:: google.cloud.aiplatform_v1beta1.services.gen_ai_tuning_service
+    :members:
+    :inherited-members:
+
+.. automodule:: google.cloud.aiplatform_v1beta1.services.gen_ai_tuning_service.pagers
+    :members:
+    :inherited-members:

docs/aiplatform_v1beta1/services_.rst

@@ -14,6 +14,8 @@ Services for Google Cloud Aiplatform v1beta1 API
     feature_registry_service
     featurestore_online_serving_service
     featurestore_service
+    gen_ai_cache_service
+    gen_ai_tuning_service
     index_endpoint_service
     index_service
     job_service

google/cloud/aiplatform_v1/__init__.py

@@ -96,13 +96,11 @@
 from .types.content import Content
 from .types.content import FileData
 from .types.content import GenerationConfig
-from .types.content import GroundingAttribution
 from .types.content import GroundingMetadata
 from .types.content import Part
 from .types.content import SafetyRating
 from .types.content import SafetySetting
 from .types.content import SearchEntryPoint
-from .types.content import Segment
 from .types.content import VideoMetadata
 from .types.content import HarmCategory
 from .types.context import Context
@@ -154,6 +152,7 @@
 from .types.dataset_service import SearchDataItemsRequest
 from .types.dataset_service import SearchDataItemsResponse
 from .types.dataset_service import UpdateDatasetRequest
+from .types.dataset_service import UpdateDatasetVersionRequest
 from .types.dataset_version import DatasetVersion
 from .types.deployed_index_ref import DeployedIndexRef
 from .types.deployed_model_ref import DeployedModelRef
@@ -276,6 +275,8 @@
 from .types.featurestore_online_service import ReadFeatureValuesRequest
 from .types.featurestore_online_service import ReadFeatureValuesResponse
 from .types.featurestore_online_service import StreamingReadFeatureValuesRequest
+from .types.featurestore_online_service import StructFieldValue
+from .types.featurestore_online_service import StructValue
 from .types.featurestore_online_service import WriteFeatureValuesPayload
 from .types.featurestore_online_service import WriteFeatureValuesRequest
 from .types.featurestore_online_service import WriteFeatureValuesResponse
@@ -585,6 +586,7 @@
 from .types.notebook_service import StartNotebookRuntimeOperationMetadata
 from .types.notebook_service import StartNotebookRuntimeRequest
 from .types.notebook_service import StartNotebookRuntimeResponse
+from .types.notebook_service import UpdateNotebookRuntimeTemplateRequest
 from .types.notebook_service import UpgradeNotebookRuntimeOperationMetadata
 from .types.notebook_service import UpgradeNotebookRuntimeRequest
 from .types.notebook_service import UpgradeNotebookRuntimeResponse
@@ -593,6 +595,7 @@
 from .types.operation import DeleteOperationMetadata
 from .types.operation import GenericOperationMetadata
 from .types.persistent_resource import PersistentResource
+from .types.persistent_resource import RayMetricSpec
 from .types.persistent_resource import RaySpec
 from .types.persistent_resource import ResourcePool
 from .types.persistent_resource import ResourceRuntime
@@ -739,11 +742,13 @@
 from .types.tensorboard_service import WriteTensorboardRunDataResponse
 from .types.tensorboard_time_series import TensorboardTimeSeries
 from .types.tool import FunctionCall
+from .types.tool import FunctionCallingConfig
 from .types.tool import FunctionDeclaration
 from .types.tool import FunctionResponse
 from .types.tool import GoogleSearchRetrieval
 from .types.tool import Retrieval
 from .types.tool import Tool
+from .types.tool import ToolConfig
 from .types.tool import VertexAISearch
 from .types.training_pipeline import FilterSplit
 from .types.training_pipeline import FractionSplit
@@ -1085,6 +1090,7 @@
     "FindNeighborsResponse",
     "FractionSplit",
     "FunctionCall",
+    "FunctionCallingConfig",
     "FunctionDeclaration",
     "FunctionResponse",
     "GcsDestination",
@@ -1140,7 +1146,6 @@
     "GetTrialRequest",
     "GetTuningJobRequest",
     "GoogleSearchRetrieval",
-    "GroundingAttribution",
     "GroundingMetadata",
     "HarmCategory",
     "HyperparameterTuningJob",
@@ -1360,6 +1365,7 @@
     "QueryDeployedModelsResponse",
     "QueryExecutionInputsAndOutputsRequest",
     "RawPredictRequest",
+    "RayMetricSpec",
     "RaySpec",
     "ReadFeatureValuesRequest",
     "ReadFeatureValuesResponse",
@@ -1410,7 +1416,6 @@
     "SearchModelDeploymentMonitoringStatsAnomaliesResponse",
     "SearchNearestEntitiesRequest",
     "SearchNearestEntitiesResponse",
-    "Segment",
     "ServiceAccountSpec",
     "ShieldedVmConfig",
     "SmoothGradConfig",
@@ -1432,6 +1437,8 @@
     "StreamingRawPredictResponse",
     "StreamingReadFeatureValuesRequest",
     "StringArray",
+    "StructFieldValue",
+    "StructValue",
     "Study",
     "StudySpec",
     "StudyTimeConstraint",
@@ -1460,6 +1467,7 @@
     "TimestampSplit",
     "TokensInfo",
     "Tool",
+    "ToolConfig",
     "TrainingConfig",
     "TrainingPipeline",
     "Trial",
@@ -1478,6 +1486,7 @@
     "UpdateArtifactRequest",
     "UpdateContextRequest",
     "UpdateDatasetRequest",
+    "UpdateDatasetVersionRequest",
     "UpdateDeploymentResourcePoolOperationMetadata",
     "UpdateEndpointRequest",
     "UpdateEntityTypeRequest",
@@ -1501,6 +1510,7 @@
     "UpdateModelDeploymentMonitoringJobOperationMetadata",
     "UpdateModelDeploymentMonitoringJobRequest",
     "UpdateModelRequest",
+    "UpdateNotebookRuntimeTemplateRequest",
     "UpdatePersistentResourceOperationMetadata",
     "UpdatePersistentResourceRequest",
     "UpdateScheduleRequest",

google/cloud/aiplatform_v1/gapic_metadata.json

@@ -99,6 +99,11 @@
               "methods": [
                 "update_dataset"
               ]
+            },
+            "UpdateDatasetVersion": {
+              "methods": [
+                "update_dataset_version"
+              ]
             }
           }
         },
@@ -194,6 +199,11 @@
               "methods": [
                 "update_dataset"
               ]
+            },
+            "UpdateDatasetVersion": {
+              "methods": [
+                "update_dataset_version"
+              ]
             }
           }
         },
@@ -289,6 +299,11 @@
               "methods": [
                 "update_dataset"
               ]
+            },
+            "UpdateDatasetVersion": {
+              "methods": [
+                "update_dataset_version"
+              ]
             }
           }
         }
@@ -3262,6 +3277,11 @@
                 "start_notebook_runtime"
               ]
             },
+            "UpdateNotebookRuntimeTemplate": {
+              "methods": [
+                "update_notebook_runtime_template"
+              ]
+            },
             "UpgradeNotebookRuntime": {
               "methods": [
                 "upgrade_notebook_runtime"
@@ -3317,6 +3337,11 @@
                 "start_notebook_runtime"
               ]
             },
+            "UpdateNotebookRuntimeTemplate": {
+              "methods": [
+                "update_notebook_runtime_template"
+              ]
+            },
             "UpgradeNotebookRuntime": {
               "methods": [
                 "upgrade_notebook_runtime"
@@ -3372,6 +3397,11 @@
                 "start_notebook_runtime"
               ]
             },
+            "UpdateNotebookRuntimeTemplate": {
+              "methods": [
+                "update_notebook_runtime_template"
+              ]
+            },
             "UpgradeNotebookRuntime": {
               "methods": [
                 "upgrade_notebook_runtime"

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

@@ -1324,6 +1324,131 @@ async def sample_create_dataset_version():
         # Done; return the response.
         return response
 
+    async def update_dataset_version(
+        self,
+        request: Optional[
+            Union[dataset_service.UpdateDatasetVersionRequest, dict]
+        ] = None,
+        *,
+        dataset_version: Optional[gca_dataset_version.DatasetVersion] = None,
+        update_mask: Optional[field_mask_pb2.FieldMask] = None,
+        retry: OptionalRetry = gapic_v1.method.DEFAULT,
+        timeout: Union[float, object] = gapic_v1.method.DEFAULT,
+        metadata: Sequence[Tuple[str, str]] = (),
+    ) -> gca_dataset_version.DatasetVersion:
+        r"""Updates a DatasetVersion.
+
+        .. 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_update_dataset_version():
+                # Create a client
+                client = aiplatform_v1.DatasetServiceAsyncClient()
+
+                # Initialize request argument(s)
+                dataset_version = aiplatform_v1.DatasetVersion()
+                dataset_version.metadata.null_value = "NULL_VALUE"
+
+                request = aiplatform_v1.UpdateDatasetVersionRequest(
+                    dataset_version=dataset_version,
+                )
+
+                # Make the request
+                response = await client.update_dataset_version(request=request)
+
+                # Handle the response
+                print(response)
+
+        Args:
+            request (Optional[Union[google.cloud.aiplatform_v1.types.UpdateDatasetVersionRequest, dict]]):
+                The request object. Request message for
+                [DatasetService.UpdateDatasetVersion][google.cloud.aiplatform.v1.DatasetService.UpdateDatasetVersion].
+            dataset_version (:class:`google.cloud.aiplatform_v1.types.DatasetVersion`):
+                Required. The DatasetVersion which
+                replaces the resource on the server.
+
+                This corresponds to the ``dataset_version`` field
+                on the ``request`` instance; if ``request`` is provided, this
+                should not be set.
+            update_mask (:class:`google.protobuf.field_mask_pb2.FieldMask`):
+                Required. The update mask applies to the resource. For
+                the ``FieldMask`` definition, see
+                [google.protobuf.FieldMask][google.protobuf.FieldMask].
+                Updatable fields:
+
+                -  ``display_name``
+
+                This corresponds to the ``update_mask`` 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, str]]): Strings which should be
+                sent along with the request as metadata.
+
+        Returns:
+            google.cloud.aiplatform_v1.types.DatasetVersion:
+                Describes the dataset version.
+        """
+        # 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.
+        has_flattened_params = any([dataset_version, update_mask])
+        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, dataset_service.UpdateDatasetVersionRequest):
+            request = dataset_service.UpdateDatasetVersionRequest(request)
+
+        # If we have keyword arguments corresponding to fields on the
+        # request, apply these.
+        if dataset_version is not None:
+            request.dataset_version = dataset_version
+        if update_mask is not None:
+            request.update_mask = update_mask
+
+        # Wrap the RPC method; this adds retry and timeout information,
+        # and friendly error handling.
+        rpc = self._client._transport._wrapped_methods[
+            self._client._transport.update_dataset_version
+        ]
+
+        # Certain fields should be provided within the metadata header;
+        # add these here.
+        metadata = tuple(metadata) + (
+            gapic_v1.routing_header.to_grpc_metadata(
+                (("dataset_version.name", request.dataset_version.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 delete_dataset_version(
         self,
         request: Optional[