feat: add a v1 UpdateEndpointLongRunning API (#5812)

* feat: add a `nfs_mounts` to RaySpec in PersistentResource API

PiperOrigin-RevId: 696288226

Source-Link: googleapis/googleapis@50d87b4

Source-Link: googleapis/googleapis-gen@f635540
Copy-Tag: eyJwIjoicGFja2FnZXMvZ29vZ2xlLWNsb3VkLWFpcGxhdGZvcm0vLk93bEJvdC55YW1sIiwiaCI6ImY2MzU1NDA0NjAwYmZlNzFmMGY4MzRmYWVlNDM5ZDZkYTE2MjI1NTQifQ==

* feat: add CustomEnvironmentSpec to NotebookExecutionJob

PiperOrigin-RevId: 697694162

Source-Link: googleapis/googleapis@7b30cb5

Source-Link: googleapis/googleapis-gen@7243ba9
Copy-Tag: eyJwIjoicGFja2FnZXMvZ29vZ2xlLWNsb3VkLWFpcGxhdGZvcm0vLk93bEJvdC55YW1sIiwiaCI6IjcyNDNiYTliYzk0ZDQzYWFjMzRhOWRmMzk2NGY4ZTU3YjVlMzYzYWMifQ==

* feat: add CustomEnvironmentSpec to NotebookExecutionJob

PiperOrigin-RevId: 697698546

Source-Link: googleapis/googleapis@caa9439

Source-Link: googleapis/googleapis-gen@d16c0f0
Copy-Tag: eyJwIjoicGFja2FnZXMvZ29vZ2xlLWNsb3VkLWFpcGxhdGZvcm0vLk93bEJvdC55YW1sIiwiaCI6ImQxNmMwZjAyNzE5OWFhZjhiMTU3MzdkMGM4MzgzODEzNjE5ODNjNmYifQ==

* feat: Trajectory eval metrics added to evaluation service proto
docs: Fixed typo for field `use_strict_string_match` in message `.google.cloud.aiplatform.v1beta1.ToolParameterKVMatchSpec`

PiperOrigin-RevId: 697705080

Source-Link: googleapis/googleapis@231f232

Source-Link: googleapis/googleapis-gen@1c17fe0
Copy-Tag: eyJwIjoicGFja2FnZXMvZ29vZ2xlLWNsb3VkLWFpcGxhdGZvcm0vLk93bEJvdC55YW1sIiwiaCI6IjFjMTdmZTAwZmRmYzBhMDk3MzhhNDNkMjdiM2YyYjczZTQ1Zjc0NzcifQ==

* feat: add a v1 UpdateEndpointLongRunning API

PiperOrigin-RevId: 697715678

Source-Link: googleapis/googleapis@4ab02f3

Source-Link: googleapis/googleapis-gen@829605e
Copy-Tag: eyJwIjoicGFja2FnZXMvZ29vZ2xlLWNsb3VkLWFpcGxhdGZvcm0vLk93bEJvdC55YW1sIiwiaCI6IjgyOTYwNWUxNmRlZTFkOGQ3OTVkNGRlYTQ1YjRjMTJkOGQ5ODNhMWMifQ==

* chore: run post-processor

* 🦉 Updates from OwlBot post-processor

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>
Co-authored-by: Sofia Leon <sofialeon@google.com>

Author: gcf-owl-bot[bot]

packages/google-cloud-aiplatform/README.md

@@ -135,6 +135,7 @@ Samples are in the [`samples/`](https://github.com/googleapis/google-cloud-node/
 | Endpoint_service.mutate_deployed_model | [source code](https://github.com/googleapis/google-cloud-node/blob/main/packages/google-cloud-aiplatform/samples/generated/v1/endpoint_service.mutate_deployed_model.js) | [![Open in Cloud Shell][shell_img]](https://console.cloud.google.com/cloudshell/open?git_repo=https://github.com/googleapis/google-cloud-node&page=editor&open_in_editor=packages/google-cloud-aiplatform/samples/generated/v1/endpoint_service.mutate_deployed_model.js,packages/google-cloud-aiplatform/samples/README.md) |
 | Endpoint_service.undeploy_model | [source code](https://github.com/googleapis/google-cloud-node/blob/main/packages/google-cloud-aiplatform/samples/generated/v1/endpoint_service.undeploy_model.js) | [![Open in Cloud Shell][shell_img]](https://console.cloud.google.com/cloudshell/open?git_repo=https://github.com/googleapis/google-cloud-node&page=editor&open_in_editor=packages/google-cloud-aiplatform/samples/generated/v1/endpoint_service.undeploy_model.js,packages/google-cloud-aiplatform/samples/README.md) |
 | Endpoint_service.update_endpoint | [source code](https://github.com/googleapis/google-cloud-node/blob/main/packages/google-cloud-aiplatform/samples/generated/v1/endpoint_service.update_endpoint.js) | [![Open in Cloud Shell][shell_img]](https://console.cloud.google.com/cloudshell/open?git_repo=https://github.com/googleapis/google-cloud-node&page=editor&open_in_editor=packages/google-cloud-aiplatform/samples/generated/v1/endpoint_service.update_endpoint.js,packages/google-cloud-aiplatform/samples/README.md) |
+| Endpoint_service.update_endpoint_long_running | [source code](https://github.com/googleapis/google-cloud-node/blob/main/packages/google-cloud-aiplatform/samples/generated/v1/endpoint_service.update_endpoint_long_running.js) | [![Open in Cloud Shell][shell_img]](https://console.cloud.google.com/cloudshell/open?git_repo=https://github.com/googleapis/google-cloud-node&page=editor&open_in_editor=packages/google-cloud-aiplatform/samples/generated/v1/endpoint_service.update_endpoint_long_running.js,packages/google-cloud-aiplatform/samples/README.md) |
 | Evaluation_service.evaluate_instances | [source code](https://github.com/googleapis/google-cloud-node/blob/main/packages/google-cloud-aiplatform/samples/generated/v1/evaluation_service.evaluate_instances.js) | [![Open in Cloud Shell][shell_img]](https://console.cloud.google.com/cloudshell/open?git_repo=https://github.com/googleapis/google-cloud-node&page=editor&open_in_editor=packages/google-cloud-aiplatform/samples/generated/v1/evaluation_service.evaluate_instances.js,packages/google-cloud-aiplatform/samples/README.md) |
 | Feature_online_store_admin_service.create_feature_online_store | [source code](https://github.com/googleapis/google-cloud-node/blob/main/packages/google-cloud-aiplatform/samples/generated/v1/feature_online_store_admin_service.create_feature_online_store.js) | [![Open in Cloud Shell][shell_img]](https://console.cloud.google.com/cloudshell/open?git_repo=https://github.com/googleapis/google-cloud-node&page=editor&open_in_editor=packages/google-cloud-aiplatform/samples/generated/v1/feature_online_store_admin_service.create_feature_online_store.js,packages/google-cloud-aiplatform/samples/README.md) |
 | Feature_online_store_admin_service.create_feature_view | [source code](https://github.com/googleapis/google-cloud-node/blob/main/packages/google-cloud-aiplatform/samples/generated/v1/feature_online_store_admin_service.create_feature_view.js) | [![Open in Cloud Shell][shell_img]](https://console.cloud.google.com/cloudshell/open?git_repo=https://github.com/googleapis/google-cloud-node&page=editor&open_in_editor=packages/google-cloud-aiplatform/samples/generated/v1/feature_online_store_admin_service.create_feature_view.js,packages/google-cloud-aiplatform/samples/README.md) |

packages/google-cloud-aiplatform/protos/google/cloud/aiplatform/v1/endpoint.proto

@@ -23,6 +23,7 @@ import "google/cloud/aiplatform/v1/explanation.proto";
 import "google/cloud/aiplatform/v1/io.proto";
 import "google/cloud/aiplatform/v1/machine_resources.proto";
 import "google/cloud/aiplatform/v1/service_networking.proto";
+import "google/protobuf/duration.proto";
 import "google/protobuf/timestamp.proto";
 
 option csharp_namespace = "Google.Cloud.AIPlatform.V1";
@@ -167,6 +168,9 @@ message Endpoint {
   string dedicated_endpoint_dns = 25
       [(google.api.field_behavior) = OUTPUT_ONLY];
 
+  // Configurations that are applied to the endpoint for online prediction.
+  ClientConnectionConfig client_connection_config = 23;
+
   // Output only. Reserved for future use.
   bool satisfies_pzs = 27 [(google.api.field_behavior) = OUTPUT_ONLY];
 
@@ -294,6 +298,9 @@ message DeployedModel {
   PrivateEndpoints private_endpoints = 14
       [(google.api.field_behavior) = OUTPUT_ONLY];
 
+  // Configuration for faster model deployment.
+  FasterDeploymentConfig faster_deployment_config = 23;
+
   // System labels to apply to Model Garden deployments.
   // System labels are managed by Google for internal use only.
   map<string, string> system_labels = 28;
@@ -336,3 +343,15 @@ message PredictRequestResponseLoggingConfig {
   // given, a new table will be created with name `request_response_logging`
   BigQueryDestination bigquery_destination = 3;
 }
+
+// Configuration for faster model deployment.
+message FasterDeploymentConfig {
+  // If true, enable fast tryout feature for this deployed model.
+  bool fast_tryout_enabled = 2;
+}
+
+// Configurations (e.g. inference timeout) that are applied on your endpoints.
+message ClientConnectionConfig {
+  // Customizable online prediction request timeout.
+  google.protobuf.Duration inference_timeout = 1;
+}

packages/google-cloud-aiplatform/protos/google/cloud/aiplatform/v1/endpoint_service.proto

@@ -80,6 +80,20 @@ service EndpointService {
     option (google.api.method_signature) = "endpoint,update_mask";
   }
 
+  // Updates an Endpoint with a long running operation.
+  rpc UpdateEndpointLongRunning(UpdateEndpointLongRunningRequest)
+      returns (google.longrunning.Operation) {
+    option (google.api.http) = {
+      post: "/v1/{endpoint.name=projects/*/locations/*/endpoints/*}:update"
+      body: "*"
+    };
+    option (google.api.method_signature) = "endpoint";
+    option (google.longrunning.operation_info) = {
+      response_type: "Endpoint"
+      metadata_type: "UpdateEndpointOperationMetadata"
+    };
+  }
+
   // Deletes an Endpoint.
   rpc DeleteEndpoint(DeleteEndpointRequest)
       returns (google.longrunning.Operation) {
@@ -280,6 +294,22 @@ message UpdateEndpointRequest {
       [(google.api.field_behavior) = REQUIRED];
 }
 
+// Request message for
+// [EndpointService.UpdateEndpointLongRunning][google.cloud.aiplatform.v1.EndpointService.UpdateEndpointLongRunning].
+message UpdateEndpointLongRunningRequest {
+  // Required. The Endpoint which replaces the resource on the server. Currently
+  // we only support updating the `client_connection_config` field, all the
+  // other fields' update will be blocked.
+  Endpoint endpoint = 1 [(google.api.field_behavior) = REQUIRED];
+}
+
+// Runtime operation information for
+// [EndpointService.UpdateEndpointLongRunning][google.cloud.aiplatform.v1.EndpointService.UpdateEndpointLongRunning].
+message UpdateEndpointOperationMetadata {
+  // The operation generic information.
+  GenericOperationMetadata generic_metadata = 1;
+}
+
 // Request message for
 // [EndpointService.DeleteEndpoint][google.cloud.aiplatform.v1.EndpointService.DeleteEndpoint].
 message DeleteEndpointRequest {

packages/google-cloud-aiplatform/protos/google/cloud/aiplatform/v1/notebook_execution_job.proto

@@ -20,6 +20,8 @@ import "google/api/field_behavior.proto";
 import "google/api/resource.proto";
 import "google/cloud/aiplatform/v1/encryption_spec.proto";
 import "google/cloud/aiplatform/v1/job_state.proto";
+import "google/cloud/aiplatform/v1/machine_resources.proto";
+import "google/cloud/aiplatform/v1/network_spec.proto";
 import "google/protobuf/duration.proto";
 import "google/protobuf/timestamp.proto";
 import "google/rpc/status.proto";
@@ -70,6 +72,18 @@ message NotebookExecutionJob {
     bytes content = 1;
   }
 
+  // Compute configuration to use for an execution job.
+  message CustomEnvironmentSpec {
+    // The specification of a single machine for the execution job.
+    MachineSpec machine_spec = 1;
+
+    // The specification of a persistent disk to attach for the execution job.
+    PersistentDiskSpec persistent_disk_spec = 2;
+
+    // The network configuration to use for the execution job.
+    NetworkSpec network_spec = 3;
+  }
+
   // The input notebook.
   oneof notebook_source {
     // The Dataform Repository pointing to a single file notebook repository.
@@ -90,6 +104,9 @@ message NotebookExecutionJob {
         [(google.api.resource_reference) = {
           type: "aiplatform.googleapis.com/NotebookRuntimeTemplate"
         }];
+
+    // The custom compute configuration for an execution job.
+    CustomEnvironmentSpec custom_environment_spec = 16;
   }
 
   // The location to store the notebook execution result.