feat: add latent_space_source to ExplanationMetadata in aiplatform v1 explanation_metadata.proto

Google APIs · copybara-github · commit 058bff383bb3 · 2022-05-24T08:28:15.000-07:00
feat: add scaling to OnlineServingConfig in aiplatform v1 featurestore.proto
feat: add template_metadata to PipelineJob in aiplatform v1 pipeline_job.proto

PiperOrigin-RevId: 450687287
diff --git a/google/cloud/aiplatform/v1/endpoint.proto b/google/cloud/aiplatform/v1/endpoint.proto
@@ -154,8 +154,8 @@ message DeployedModel {
   // This value should be 1-10 characters, and valid characters are /[0-9]/.
   string id = 1 [(google.api.field_behavior) = IMMUTABLE];
 
-  // Required. The name of the Model that this is the deployment of. Note that the Model
-  // may be in a different location than the DeployedModel's Endpoint.
+  // Required. The resource name of the Model that this is the deployment of. Note that
+  // the Model may be in a different location than the DeployedModel's Endpoint.
   string model = 2 [
     (google.api.field_behavior) = REQUIRED,
     (google.api.resource_reference) = {
diff --git a/google/cloud/aiplatform/v1/explanation_metadata.proto b/google/cloud/aiplatform/v1/explanation_metadata.proto
@@ -392,4 +392,7 @@ message ExplanationMetadata {
   // than the one given on input. The output URI will point to a location where
   // the user only has a read access.
   string feature_attributions_schema_uri = 3;
+
+  // Name of the source to generate embeddings for example based explanations.
+  string latent_space_source = 5;
 }
diff --git a/google/cloud/aiplatform/v1/featurestore.proto b/google/cloud/aiplatform/v1/featurestore.proto
@@ -41,11 +41,29 @@ message Featurestore {
   // OnlineServingConfig specifies the details for provisioning online serving
   // resources.
   message OnlineServingConfig {
+    // Online serving scaling configuration. If min_node_count and
+    // max_node_count are set to the same value, the cluster will be configured
+    // with the fixed number of node (no auto-scaling).
+    message Scaling {
+      // Required. The minimum number of nodes to scale down to. Must be greater than or
+      // equal to 1.
+      int32 min_node_count = 1 [(google.api.field_behavior) = REQUIRED];
+
+      // The maximum number of nodes to scale up to. Must be greater than
+      // min_node_count, and less than or equal to 10 times of 'min_node_count'.
+      int32 max_node_count = 2;
+    }
+
     // The number of nodes for the online store. The number of nodes doesn't
     // scale automatically, but you can manually update the number of
     // nodes. If set to 0, the featurestore will not have an
     // online store and cannot be used for online serving.
     int32 fixed_node_count = 2;
+
+    // Online serving scaling configuration.
+    // Only one of `fixed_node_count` and `scaling` can be set. Setting one will
+    // reset the other.
+    Scaling scaling = 4;
   }
 
   // Possible states a featurestore can have.
diff --git a/google/cloud/aiplatform/v1/manual_batch_tuning_parameters.proto b/google/cloud/aiplatform/v1/manual_batch_tuning_parameters.proto
@@ -34,6 +34,6 @@ message ManualBatchTuningParameters {
   // speeds up the batch operation's execution, but too high value will result
   // in a whole batch not fitting in a machine's memory, and the whole
   // operation will fail.
-  // The default value is 4.
+  // The default value is 64.
   int32 batch_size = 1 [(google.api.field_behavior) = IMMUTABLE];
 }
diff --git a/google/cloud/aiplatform/v1/model_monitoring.proto b/google/cloud/aiplatform/v1/model_monitoring.proto
@@ -55,6 +55,8 @@ message ModelMonitoringObjectiveConfig {
     //
     // "csv"
     // The source file is a CSV file.
+    // "jsonl"
+    // The source file is a JSONL file.
     string data_format = 2;
 
     // The target field name the model is to predict.
diff --git a/google/cloud/aiplatform/v1/pipeline_job.proto b/google/cloud/aiplatform/v1/pipeline_job.proto
@@ -150,6 +150,27 @@ message PipelineJob {
   string network = 18 [(google.api.resource_reference) = {
                          type: "compute.googleapis.com/Network"
                        }];
+
+  // A template uri from where the [PipelineJob.pipeline_spec][google.cloud.aiplatform.v1.PipelineJob.pipeline_spec], if empty, will
+  // be downloaded.
+  string template_uri = 19;
+
+  // Output only. Pipeline template metadata. Will fill up fields if
+  // [PipelineJob.template_uri][google.cloud.aiplatform.v1.PipelineJob.template_uri] is from supported template registry.
+  PipelineTemplateMetadata template_metadata = 20 [(google.api.field_behavior) = OUTPUT_ONLY];
+}
+
+// Pipeline template metadata if [PipelineJob.template_uri][google.cloud.aiplatform.v1.PipelineJob.template_uri] is from supported
+// template registry. Currently, the only supported registry is Artifact
+// Registry.
+message PipelineTemplateMetadata {
+  // The version_name in artifact registry.
+  //
+  // Will always be presented in output if the [PipelineJob.template_uri][google.cloud.aiplatform.v1.PipelineJob.template_uri] is
+  // from supported template registry.
+  //
+  // Format is "sha256:abcdef123456...".
+  string version = 3;
 }
 
 // The runtime detail of PipelineJob.
diff --git a/google/cloud/aiplatform/v1/pipeline_service.proto b/google/cloud/aiplatform/v1/pipeline_service.proto
@@ -323,7 +323,7 @@ message ListPipelineJobsRequest {
   //   Values must be in RFC 3339 format.
   // * `labels`: Supports key-value equality and key presence.
   // * `template_uri`: Supports `=`, `!=` comparisons, and `:` wildcard.
-  // * `template_metadata.version_name`: Supports `=`, `!=` comparisons, and `:`
+  // * `template_metadata.version`: Supports `=`, `!=` comparisons, and `:`
   // wildcard.
   //
   // Filter expressions can be combined together using logical operators

Original file line number	Diff line number	Diff line change
`@@ -392,4 +392,7 @@ message ExplanationMetadata {`
`392`	`392`	`// than the one given on input. The output URI will point to a location where`
`393`	`393`	`// the user only has a read access.`
`394`	`394`	`string feature_attributions_schema_uri = 3;`
	`395`	`+`
	`396`	`+ // Name of the source to generate embeddings for example based explanations.`
	`397`	`+ string latent_space_source = 5;`
`395`	`398`	`}`
Original file line number	Diff line number	Diff line change
`@@ -34,6 +34,6 @@ message ManualBatchTuningParameters {`
`34`	`34`	`// speeds up the batch operation's execution, but too high value will result`
`35`	`35`	`// in a whole batch not fitting in a machine's memory, and the whole`
`36`	`36`	`// operation will fail.`
`37`		`- // The default value is 4.`
	`37`	`+ // The default value is 64.`
`38`	`38`	`int32 batch_size = 1 [(google.api.field_behavior) = IMMUTABLE];`
`39`	`39`	`}`