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 <[email protected]>
googleapis · Nov 21, 2024 · 39dc74e · 39dc74e
1 parent ced8054
commit 39dc74e
Show file tree

Hide file tree

Showing 20 changed files with 16,802 additions and 1,789 deletions.
diff --git a/packages/google-cloud-aiplatform/README.md b/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) |

diff --git a/packages/google-cloud-aiplatform/protos/google/cloud/aiplatform/v1/endpoint.proto b/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;
+}
diff --git a/packages/google-cloud-aiplatform/protos/google/cloud/aiplatform/v1/endpoint_service.proto b/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 {

diff --git a/...es/google-cloud-aiplatform/protos/google/cloud/aiplatform/v1/notebook_execution_job.proto b/...es/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.