google-cloud-ai_platform-v1 0.5.0 → 0.8.0

data/lib/google/cloud/ai_platform/v1/specialist_pool_service/client.rb

@@ -18,6 +18,8 @@
 
 require "google/cloud/errors"
 require "google/cloud/aiplatform/v1/specialist_pool_service_pb"
+require "google/cloud/location"
+require "google/iam/v1/iam_policy"
 
 module Google
   module Cloud
@@ -144,6 +146,18 @@ module Google
                 config.endpoint = @config.endpoint
               end
 
+              @location_client = Google::Cloud::Location::Locations::Client.new do |config|
+                config.credentials = credentials
+                config.quota_project = @quota_project_id
+                config.endpoint = @config.endpoint
+              end
+
+              @iam_policy_client = Google::Iam::V1::IAMPolicy::Client.new do |config|
+                config.credentials = credentials
+                config.quota_project = @quota_project_id
+                config.endpoint = @config.endpoint
+              end
+
               @specialist_pool_service_stub = ::Gapic::ServiceStub.new(
                 ::Google::Cloud::AIPlatform::V1::SpecialistPoolService::Stub,
                 credentials:  credentials,
@@ -160,6 +174,20 @@ module Google
             #
             attr_reader :operations_client
 
+            ##
+            # Get the associated client for mix-in of the Locations.
+            #
+            # @return [Google::Cloud::Location::Locations::Client]
+            #
+            attr_reader :location_client
+
+            ##
+            # Get the associated client for mix-in of the IAMPolicy.
+            #
+            # @return [Google::Iam::V1::IAMPolicy::Client]
+            #
+            attr_reader :iam_policy_client
+
             # Service calls
 
             ##

data/lib/google/cloud/ai_platform/v1/tensorboard_service/client.rb

@@ -18,6 +18,8 @@
 
 require "google/cloud/errors"
 require "google/cloud/aiplatform/v1/tensorboard_service_pb"
+require "google/cloud/location"
+require "google/iam/v1/iam_policy"
 
 module Google
   module Cloud
@@ -139,6 +141,18 @@ module Google
                 config.endpoint = @config.endpoint
               end
 
+              @location_client = Google::Cloud::Location::Locations::Client.new do |config|
+                config.credentials = credentials
+                config.quota_project = @quota_project_id
+                config.endpoint = @config.endpoint
+              end
+
+              @iam_policy_client = Google::Iam::V1::IAMPolicy::Client.new do |config|
+                config.credentials = credentials
+                config.quota_project = @quota_project_id
+                config.endpoint = @config.endpoint
+              end
+
               @tensorboard_service_stub = ::Gapic::ServiceStub.new(
                 ::Google::Cloud::AIPlatform::V1::TensorboardService::Stub,
                 credentials:  credentials,
@@ -155,6 +169,20 @@ module Google
             #
             attr_reader :operations_client
 
+            ##
+            # Get the associated client for mix-in of the Locations.
+            #
+            # @return [Google::Cloud::Location::Locations::Client]
+            #
+            attr_reader :location_client
+
+            ##
+            # Get the associated client for mix-in of the IAMPolicy.
+            #
+            # @return [Google::Iam::V1::IAMPolicy::Client]
+            #
+            attr_reader :iam_policy_client
+
             # Service calls
 
             ##

data/lib/google/cloud/ai_platform/v1/version.rb

@@ -21,7 +21,7 @@ module Google
   module Cloud
     module AIPlatform
       module V1
-        VERSION = "0.5.0"
+        VERSION = "0.8.0"
       end
     end
   end

data/lib/google/cloud/ai_platform/v1/vizier_service/client.rb

@@ -18,6 +18,8 @@
 
 require "google/cloud/errors"
 require "google/cloud/aiplatform/v1/vizier_service_pb"
+require "google/cloud/location"
+require "google/iam/v1/iam_policy"
 
 module Google
   module Cloud
@@ -143,6 +145,18 @@ module Google
                 config.endpoint = @config.endpoint
               end
 
+              @location_client = Google::Cloud::Location::Locations::Client.new do |config|
+                config.credentials = credentials
+                config.quota_project = @quota_project_id
+                config.endpoint = @config.endpoint
+              end
+
+              @iam_policy_client = Google::Iam::V1::IAMPolicy::Client.new do |config|
+                config.credentials = credentials
+                config.quota_project = @quota_project_id
+                config.endpoint = @config.endpoint
+              end
+
               @vizier_service_stub = ::Gapic::ServiceStub.new(
                 ::Google::Cloud::AIPlatform::V1::VizierService::Stub,
                 credentials:  credentials,
@@ -159,6 +173,20 @@ module Google
             #
             attr_reader :operations_client
 
+            ##
+            # Get the associated client for mix-in of the Locations.
+            #
+            # @return [Google::Cloud::Location::Locations::Client]
+            #
+            attr_reader :location_client
+
+            ##
+            # Get the associated client for mix-in of the IAMPolicy.
+            #
+            # @return [Google::Iam::V1::IAMPolicy::Client]
+            #
+            attr_reader :iam_policy_client
+
             # Service calls
 
             ##

data/lib/google/cloud/aiplatform/v1/batch_prediction_job_pb.rb

@@ -23,6 +23,7 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       optional :name, :string, 1
       optional :display_name, :string, 2
       optional :model, :string, 3
+      optional :model_version_id, :string, 30
       optional :unmanaged_container_model, :message, 28, "google.cloud.aiplatform.v1.UnmanagedContainerModel"
       optional :input_config, :message, 4, "google.cloud.aiplatform.v1.BatchPredictionJob.InputConfig"
       optional :model_parameters, :message, 5, "google.protobuf.Value"

data/lib/google/cloud/aiplatform/v1/completion_stats_pb.rb

@@ -11,6 +11,7 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       optional :successful_count, :int64, 1
       optional :failed_count, :int64, 2
       optional :incomplete_count, :int64, 3
+      optional :successful_forecast_point_count, :int64, 5
     end
   end
 end

data/lib/google/cloud/aiplatform/v1/dataset_service_pb.rb

@@ -12,6 +12,7 @@ require 'google/cloud/aiplatform/v1/annotation_spec_pb'
 require 'google/cloud/aiplatform/v1/data_item_pb'
 require 'google/cloud/aiplatform/v1/dataset_pb'
 require 'google/cloud/aiplatform/v1/operation_pb'
+require 'google/cloud/aiplatform/v1/saved_query_pb'
 require 'google/longrunning/operations_pb'
 require 'google/protobuf/field_mask_pb'
 
@@ -79,6 +80,18 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       repeated :data_items, :message, 1, "google.cloud.aiplatform.v1.DataItem"
       optional :next_page_token, :string, 2
     end
+    add_message "google.cloud.aiplatform.v1.ListSavedQueriesRequest" do
+      optional :parent, :string, 1
+      optional :filter, :string, 2
+      optional :page_size, :int32, 3
+      optional :page_token, :string, 4
+      optional :read_mask, :message, 5, "google.protobuf.FieldMask"
+      optional :order_by, :string, 6
+    end
+    add_message "google.cloud.aiplatform.v1.ListSavedQueriesResponse" do
+      repeated :saved_queries, :message, 1, "google.cloud.aiplatform.v1.SavedQuery"
+      optional :next_page_token, :string, 2
+    end
     add_message "google.cloud.aiplatform.v1.GetAnnotationSpecRequest" do
       optional :name, :string, 1
       optional :read_mask, :message, 2, "google.protobuf.FieldMask"
@@ -117,6 +130,8 @@ module Google
         ExportDataOperationMetadata = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ExportDataOperationMetadata").msgclass
         ListDataItemsRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ListDataItemsRequest").msgclass
         ListDataItemsResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ListDataItemsResponse").msgclass
+        ListSavedQueriesRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ListSavedQueriesRequest").msgclass
+        ListSavedQueriesResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ListSavedQueriesResponse").msgclass
         GetAnnotationSpecRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.GetAnnotationSpecRequest").msgclass
         ListAnnotationsRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ListAnnotationsRequest").msgclass
         ListAnnotationsResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ListAnnotationsResponse").msgclass

data/lib/google/cloud/aiplatform/v1/dataset_service_services_pb.rb

@@ -50,6 +50,8 @@ module Google
             rpc :ExportData, ::Google::Cloud::AIPlatform::V1::ExportDataRequest, ::Google::Longrunning::Operation
             # Lists DataItems in a Dataset.
             rpc :ListDataItems, ::Google::Cloud::AIPlatform::V1::ListDataItemsRequest, ::Google::Cloud::AIPlatform::V1::ListDataItemsResponse
+            # Lists SavedQueries in a Dataset.
+            rpc :ListSavedQueries, ::Google::Cloud::AIPlatform::V1::ListSavedQueriesRequest, ::Google::Cloud::AIPlatform::V1::ListSavedQueriesResponse
             # Gets an AnnotationSpec.
             rpc :GetAnnotationSpec, ::Google::Cloud::AIPlatform::V1::GetAnnotationSpecRequest, ::Google::Cloud::AIPlatform::V1::AnnotationSpec
             # Lists Annotations belongs to a dataitem

data/lib/google/cloud/aiplatform/v1/endpoint_pb.rb

@@ -32,6 +32,7 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
     add_message "google.cloud.aiplatform.v1.DeployedModel" do
       optional :id, :string, 1
       optional :model, :string, 2
+      optional :model_version_id, :string, 18
       optional :display_name, :string, 3
       optional :create_time, :message, 6, "google.protobuf.Timestamp"
       optional :explanation_spec, :message, 9, "google.cloud.aiplatform.v1.ExplanationSpec"

data/lib/google/cloud/aiplatform/v1/explanation_pb.rb

@@ -11,6 +11,7 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
   add_file("google/cloud/aiplatform/v1/explanation.proto", :syntax => :proto3) do
     add_message "google.cloud.aiplatform.v1.Explanation" do
       repeated :attributions, :message, 1, "google.cloud.aiplatform.v1.Attribution"
+      repeated :neighbors, :message, 2, "google.cloud.aiplatform.v1.Neighbor"
     end
     add_message "google.cloud.aiplatform.v1.ModelExplanation" do
       repeated :mean_attributions, :message, 1, "google.cloud.aiplatform.v1.Attribution"
@@ -24,6 +25,10 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       optional :approximation_error, :double, 6
       optional :output_name, :string, 7
     end
+    add_message "google.cloud.aiplatform.v1.Neighbor" do
+      optional :neighbor_id, :string, 1
+      optional :neighbor_distance, :double, 2
+    end
     add_message "google.cloud.aiplatform.v1.ExplanationSpec" do
       optional :parameters, :message, 1, "google.cloud.aiplatform.v1.ExplanationParameters"
       optional :metadata, :message, 2, "google.cloud.aiplatform.v1.ExplanationMetadata"
@@ -70,6 +75,7 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
     add_message "google.cloud.aiplatform.v1.ExplanationSpecOverride" do
       optional :parameters, :message, 1, "google.cloud.aiplatform.v1.ExplanationParameters"
       optional :metadata, :message, 2, "google.cloud.aiplatform.v1.ExplanationMetadataOverride"
+      optional :examples_override, :message, 3, "google.cloud.aiplatform.v1.ExamplesOverride"
     end
     add_message "google.cloud.aiplatform.v1.ExplanationMetadataOverride" do
       map :inputs, :string, :message, 1, "google.cloud.aiplatform.v1.ExplanationMetadataOverride.InputMetadataOverride"
@@ -77,6 +83,23 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
     add_message "google.cloud.aiplatform.v1.ExplanationMetadataOverride.InputMetadataOverride" do
       repeated :input_baselines, :message, 1, "google.protobuf.Value"
     end
+    add_message "google.cloud.aiplatform.v1.ExamplesOverride" do
+      optional :neighbor_count, :int32, 1
+      optional :crowding_count, :int32, 2
+      repeated :restrictions, :message, 3, "google.cloud.aiplatform.v1.ExamplesRestrictionsNamespace"
+      optional :return_embeddings, :bool, 4
+      optional :data_format, :enum, 5, "google.cloud.aiplatform.v1.ExamplesOverride.DataFormat"
+    end
+    add_enum "google.cloud.aiplatform.v1.ExamplesOverride.DataFormat" do
+      value :DATA_FORMAT_UNSPECIFIED, 0
+      value :INSTANCES, 1
+      value :EMBEDDINGS, 2
+    end
+    add_message "google.cloud.aiplatform.v1.ExamplesRestrictionsNamespace" do
+      optional :namespace_name, :string, 1
+      repeated :allow, :string, 2
+      repeated :deny, :string, 3
+    end
   end
 end
 
@@ -87,6 +110,7 @@ module Google
         Explanation = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.Explanation").msgclass
         ModelExplanation = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ModelExplanation").msgclass
         Attribution = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.Attribution").msgclass
+        Neighbor = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.Neighbor").msgclass
         ExplanationSpec = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ExplanationSpec").msgclass
         ExplanationParameters = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ExplanationParameters").msgclass
         SampledShapleyAttribution = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.SampledShapleyAttribution").msgclass
@@ -99,6 +123,9 @@ module Google
         ExplanationSpecOverride = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ExplanationSpecOverride").msgclass
         ExplanationMetadataOverride = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ExplanationMetadataOverride").msgclass
         ExplanationMetadataOverride::InputMetadataOverride = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ExplanationMetadataOverride.InputMetadataOverride").msgclass
+        ExamplesOverride = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ExamplesOverride").msgclass
+        ExamplesOverride::DataFormat = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ExamplesOverride.DataFormat").enummodule
+        ExamplesRestrictionsNamespace = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ExamplesRestrictionsNamespace").msgclass
       end
     end
   end

data/lib/google/cloud/aiplatform/v1/model_deployment_monitoring_job_pb.rb

@@ -73,6 +73,7 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
     end
     add_message "google.cloud.aiplatform.v1.ModelDeploymentMonitoringScheduleConfig" do
       optional :monitor_interval, :message, 1, "google.protobuf.Duration"
+      optional :monitor_window, :message, 2, "google.protobuf.Duration"
     end
     add_message "google.cloud.aiplatform.v1.ModelMonitoringStatsAnomalies" do
       optional :objective, :enum, 1, "google.cloud.aiplatform.v1.ModelDeploymentMonitoringObjectiveType"

data/lib/google/cloud/aiplatform/v1/model_monitoring_pb.rb

@@ -27,10 +27,12 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
     add_message "google.cloud.aiplatform.v1.ModelMonitoringObjectiveConfig.TrainingPredictionSkewDetectionConfig" do
       map :skew_thresholds, :string, :message, 1, "google.cloud.aiplatform.v1.ThresholdConfig"
       map :attribution_score_skew_thresholds, :string, :message, 2, "google.cloud.aiplatform.v1.ThresholdConfig"
+      optional :default_skew_threshold, :message, 6, "google.cloud.aiplatform.v1.ThresholdConfig"
     end
     add_message "google.cloud.aiplatform.v1.ModelMonitoringObjectiveConfig.PredictionDriftDetectionConfig" do
       map :drift_thresholds, :string, :message, 1, "google.cloud.aiplatform.v1.ThresholdConfig"
       map :attribution_score_drift_thresholds, :string, :message, 2, "google.cloud.aiplatform.v1.ThresholdConfig"
+      optional :default_drift_threshold, :message, 5, "google.cloud.aiplatform.v1.ThresholdConfig"
     end
     add_message "google.cloud.aiplatform.v1.ModelMonitoringObjectiveConfig.ExplanationConfig" do
       optional :enable_feature_attributes, :bool, 1

data/lib/google/cloud/aiplatform/v1/model_pb.rb

@@ -16,8 +16,13 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
   add_file("google/cloud/aiplatform/v1/model.proto", :syntax => :proto3) do
     add_message "google.cloud.aiplatform.v1.Model" do
       optional :name, :string, 1
+      optional :version_id, :string, 28
+      repeated :version_aliases, :string, 29
+      optional :version_create_time, :message, 31, "google.protobuf.Timestamp"
+      optional :version_update_time, :message, 32, "google.protobuf.Timestamp"
       optional :display_name, :string, 2
       optional :description, :string, 3
+      optional :version_description, :string, 30
       optional :predict_schemata, :message, 4, "google.cloud.aiplatform.v1.PredictSchemata"
       optional :metadata_schema_uri, :string, 5
       optional :metadata, :message, 6, "google.protobuf.Value"

data/lib/google/cloud/aiplatform/v1/model_service_pb.rb

@@ -19,6 +19,8 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
   add_file("google/cloud/aiplatform/v1/model_service.proto", :syntax => :proto3) do
     add_message "google.cloud.aiplatform.v1.UploadModelRequest" do
       optional :parent, :string, 1
+      optional :parent_model, :string, 4
+      optional :model_id, :string, 5
       optional :model, :message, 2, "google.cloud.aiplatform.v1.Model"
     end
     add_message "google.cloud.aiplatform.v1.UploadModelOperationMetadata" do
@@ -26,6 +28,7 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
     end
     add_message "google.cloud.aiplatform.v1.UploadModelResponse" do
       optional :model, :string, 1
+      optional :model_version_id, :string, 2
     end
     add_message "google.cloud.aiplatform.v1.GetModelRequest" do
       optional :name, :string, 1
@@ -42,6 +45,17 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       repeated :models, :message, 1, "google.cloud.aiplatform.v1.Model"
       optional :next_page_token, :string, 2
     end
+    add_message "google.cloud.aiplatform.v1.ListModelVersionsRequest" do
+      optional :name, :string, 1
+      optional :page_size, :int32, 2
+      optional :page_token, :string, 3
+      optional :filter, :string, 4
+      optional :read_mask, :message, 5, "google.protobuf.FieldMask"
+    end
+    add_message "google.cloud.aiplatform.v1.ListModelVersionsResponse" do
+      repeated :models, :message, 1, "google.cloud.aiplatform.v1.Model"
+      optional :next_page_token, :string, 2
+    end
     add_message "google.cloud.aiplatform.v1.UpdateModelRequest" do
       optional :model, :message, 1, "google.cloud.aiplatform.v1.Model"
       optional :update_mask, :message, 2, "google.protobuf.FieldMask"
@@ -49,6 +63,13 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
     add_message "google.cloud.aiplatform.v1.DeleteModelRequest" do
       optional :name, :string, 1
     end
+    add_message "google.cloud.aiplatform.v1.DeleteModelVersionRequest" do
+      optional :name, :string, 1
+    end
+    add_message "google.cloud.aiplatform.v1.MergeVersionAliasesRequest" do
+      optional :name, :string, 1
+      repeated :version_aliases, :string, 2
+    end
     add_message "google.cloud.aiplatform.v1.ExportModelRequest" do
       optional :name, :string, 1
       optional :output_config, :message, 2, "google.cloud.aiplatform.v1.ExportModelRequest.OutputConfig"
@@ -72,6 +93,13 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       optional :parent, :string, 1
       optional :model_evaluation, :message, 2, "google.cloud.aiplatform.v1.ModelEvaluation"
     end
+    add_message "google.cloud.aiplatform.v1.BatchImportModelEvaluationSlicesRequest" do
+      optional :parent, :string, 1
+      repeated :model_evaluation_slices, :message, 2, "google.cloud.aiplatform.v1.ModelEvaluationSlice"
+    end
+    add_message "google.cloud.aiplatform.v1.BatchImportModelEvaluationSlicesResponse" do
+      repeated :imported_model_evaluation_slices, :string, 1
+    end
     add_message "google.cloud.aiplatform.v1.GetModelEvaluationRequest" do
       optional :name, :string, 1
     end
@@ -113,14 +141,20 @@ module Google
         GetModelRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.GetModelRequest").msgclass
         ListModelsRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ListModelsRequest").msgclass
         ListModelsResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ListModelsResponse").msgclass
+        ListModelVersionsRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ListModelVersionsRequest").msgclass
+        ListModelVersionsResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ListModelVersionsResponse").msgclass
         UpdateModelRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.UpdateModelRequest").msgclass
         DeleteModelRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.DeleteModelRequest").msgclass
+        DeleteModelVersionRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.DeleteModelVersionRequest").msgclass
+        MergeVersionAliasesRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.MergeVersionAliasesRequest").msgclass
         ExportModelRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ExportModelRequest").msgclass
         ExportModelRequest::OutputConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ExportModelRequest.OutputConfig").msgclass
         ExportModelOperationMetadata = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ExportModelOperationMetadata").msgclass
         ExportModelOperationMetadata::OutputInfo = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ExportModelOperationMetadata.OutputInfo").msgclass
         ExportModelResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ExportModelResponse").msgclass
         ImportModelEvaluationRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ImportModelEvaluationRequest").msgclass
+        BatchImportModelEvaluationSlicesRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.BatchImportModelEvaluationSlicesRequest").msgclass
+        BatchImportModelEvaluationSlicesResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.BatchImportModelEvaluationSlicesResponse").msgclass
         GetModelEvaluationRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.GetModelEvaluationRequest").msgclass
         ListModelEvaluationsRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ListModelEvaluationsRequest").msgclass
         ListModelEvaluationsResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ListModelEvaluationsResponse").msgclass

data/lib/google/cloud/aiplatform/v1/model_service_services_pb.rb

@@ -39,6 +39,8 @@ module Google
             rpc :GetModel, ::Google::Cloud::AIPlatform::V1::GetModelRequest, ::Google::Cloud::AIPlatform::V1::Model
             # Lists Models in a Location.
             rpc :ListModels, ::Google::Cloud::AIPlatform::V1::ListModelsRequest, ::Google::Cloud::AIPlatform::V1::ListModelsResponse
+            # Lists versions of the specified model.
+            rpc :ListModelVersions, ::Google::Cloud::AIPlatform::V1::ListModelVersionsRequest, ::Google::Cloud::AIPlatform::V1::ListModelVersionsResponse
             # Updates a Model.
             rpc :UpdateModel, ::Google::Cloud::AIPlatform::V1::UpdateModelRequest, ::Google::Cloud::AIPlatform::V1::Model
             # Deletes a Model.
@@ -47,12 +49,22 @@ module Google
             # [DeployedModel][google.cloud.aiplatform.v1.DeployedModel] based on the model in its
             # [deployed_models][google.cloud.aiplatform.v1.Endpoint.deployed_models] field.
             rpc :DeleteModel, ::Google::Cloud::AIPlatform::V1::DeleteModelRequest, ::Google::Longrunning::Operation
+            # Deletes a Model version.
+            #
+            # Model version can only be deleted if there are no [DeployedModels][]
+            # created from it. Deleting the only version in the Model is not allowed. Use
+            # [DeleteModel][google.cloud.aiplatform.v1.ModelService.DeleteModel] for deleting the Model instead.
+            rpc :DeleteModelVersion, ::Google::Cloud::AIPlatform::V1::DeleteModelVersionRequest, ::Google::Longrunning::Operation
+            # Merges a set of aliases for a Model version.
+            rpc :MergeVersionAliases, ::Google::Cloud::AIPlatform::V1::MergeVersionAliasesRequest, ::Google::Cloud::AIPlatform::V1::Model
             # Exports a trained, exportable Model to a location specified by the
             # user. A Model is considered to be exportable if it has at least one
             # [supported export format][google.cloud.aiplatform.v1.Model.supported_export_formats].
             rpc :ExportModel, ::Google::Cloud::AIPlatform::V1::ExportModelRequest, ::Google::Longrunning::Operation
             # Imports an externally generated ModelEvaluation.
             rpc :ImportModelEvaluation, ::Google::Cloud::AIPlatform::V1::ImportModelEvaluationRequest, ::Google::Cloud::AIPlatform::V1::ModelEvaluation
+            # Imports a list of externally generated ModelEvaluationSlice.
+            rpc :BatchImportModelEvaluationSlices, ::Google::Cloud::AIPlatform::V1::BatchImportModelEvaluationSlicesRequest, ::Google::Cloud::AIPlatform::V1::BatchImportModelEvaluationSlicesResponse
             # Gets a ModelEvaluation.
             rpc :GetModelEvaluation, ::Google::Cloud::AIPlatform::V1::GetModelEvaluationRequest, ::Google::Cloud::AIPlatform::V1::ModelEvaluation
             # Lists ModelEvaluations in a Model.

data/lib/google/cloud/aiplatform/v1/prediction_service_pb.rb

@@ -22,6 +22,7 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       repeated :predictions, :message, 1, "google.protobuf.Value"
       optional :deployed_model_id, :string, 2
       optional :model, :string, 3
+      optional :model_version_id, :string, 5
       optional :model_display_name, :string, 4
     end
     add_message "google.cloud.aiplatform.v1.RawPredictRequest" do

data/lib/google/cloud/aiplatform/v1/saved_query_pb.rb

@@ -0,0 +1,36 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: google/cloud/aiplatform/v1/saved_query.proto
+
+require 'google/protobuf'
+
+require 'google/api/field_behavior_pb'
+require 'google/api/resource_pb'
+require 'google/protobuf/struct_pb'
+require 'google/protobuf/timestamp_pb'
+
+Google::Protobuf::DescriptorPool.generated_pool.build do
+  add_file("google/cloud/aiplatform/v1/saved_query.proto", :syntax => :proto3) do
+    add_message "google.cloud.aiplatform.v1.SavedQuery" do
+      optional :name, :string, 1
+      optional :display_name, :string, 2
+      optional :metadata, :message, 12, "google.protobuf.Value"
+      optional :create_time, :message, 3, "google.protobuf.Timestamp"
+      optional :update_time, :message, 4, "google.protobuf.Timestamp"
+      optional :annotation_filter, :string, 5
+      optional :problem_type, :string, 6
+      optional :annotation_spec_count, :int32, 10
+      optional :etag, :string, 8
+      optional :support_automl_training, :bool, 9
+    end
+  end
+end
+
+module Google
+  module Cloud
+    module AIPlatform
+      module V1
+        SavedQuery = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.SavedQuery").msgclass
+      end
+    end
+  end
+end

data/lib/google/cloud/aiplatform/v1/training_pipeline_pb.rb

@@ -23,6 +23,8 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       optional :training_task_inputs, :message, 5, "google.protobuf.Value"
       optional :training_task_metadata, :message, 6, "google.protobuf.Value"
       optional :model_to_upload, :message, 7, "google.cloud.aiplatform.v1.Model"
+      optional :model_id, :string, 22
+      optional :parent_model, :string, 21
       optional :state, :enum, 9, "google.cloud.aiplatform.v1.PipelineState"
       optional :error, :message, 10, "google.rpc.Status"
       optional :create_time, :message, 11, "google.protobuf.Timestamp"
@@ -36,6 +38,7 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       optional :dataset_id, :string, 1
       optional :annotations_filter, :string, 6
       optional :annotation_schema_uri, :string, 9
+      optional :saved_query_id, :string, 7
       oneof :split do
         optional :fraction_split, :message, 2, "google.cloud.aiplatform.v1.FractionSplit"
         optional :filter_split, :message, 3, "google.cloud.aiplatform.v1.FilterSplit"

data/proto_docs/google/cloud/aiplatform/v1/batch_prediction_job.rb

@@ -38,6 +38,12 @@ module Google
         #     Starting this job has no impact on any existing deployments of the Model
         #     and their resources.
         #     Exactly one of model and unmanaged_container_model must be set.
+        #
+        #     The model resource name may contain version id or version alias to specify
+        #     the version, if no version is specified, the default version will be used.
+        # @!attribute [r] model_version_id
+        #   @return [::String]
+        #     Output only. The version ID of the Model that produces the predictions via this job.
         # @!attribute [rw] unmanaged_container_model
         #   @return [::Google::Cloud::AIPlatform::V1::UnmanagedContainerModel]
         #     Contains model information necessary to perform batch prediction without

data/proto_docs/google/cloud/aiplatform/v1/completion_stats.rb

@@ -36,6 +36,10 @@ module Google
         #     processing had not been finished (either in successful or failed state).
         #     Set to -1 if the number is unknown (for example, the operation failed
         #     before the total entity number could be collected).
+        # @!attribute [r] successful_forecast_point_count
+        #   @return [::Integer]
+        #     Output only. The number of the successful forecast points that are generated by the
+        #     forecasting model. This is ONLY used by the forecasting batch prediction.
         class CompletionStats
           include ::Google::Protobuf::MessageExts
           extend ::Google::Protobuf::MessageExts::ClassMethods

data/proto_docs/google/cloud/aiplatform/v1/dataset_service.rb

@@ -243,6 +243,45 @@ module Google
           extend ::Google::Protobuf::MessageExts::ClassMethods
         end
 
+        # Request message for {::Google::Cloud::AIPlatform::V1::DatasetService::Client#list_saved_queries DatasetService.ListSavedQueries}.
+        # @!attribute [rw] parent
+        #   @return [::String]
+        #     Required. The resource name of the Dataset to list SavedQueries from.
+        #     Format:
+        #     `projects/{project}/locations/{location}/datasets/{dataset}`
+        # @!attribute [rw] filter
+        #   @return [::String]
+        #     The standard list filter.
+        # @!attribute [rw] page_size
+        #   @return [::Integer]
+        #     The standard list page size.
+        # @!attribute [rw] page_token
+        #   @return [::String]
+        #     The standard list page token.
+        # @!attribute [rw] read_mask
+        #   @return [::Google::Protobuf::FieldMask]
+        #     Mask specifying which fields to read.
+        # @!attribute [rw] order_by
+        #   @return [::String]
+        #     A comma-separated list of fields to order by, sorted in ascending order.
+        #     Use "desc" after a field name for descending.
+        class ListSavedQueriesRequest
+          include ::Google::Protobuf::MessageExts
+          extend ::Google::Protobuf::MessageExts::ClassMethods
+        end
+
+        # Response message for {::Google::Cloud::AIPlatform::V1::DatasetService::Client#list_saved_queries DatasetService.ListSavedQueries}.
+        # @!attribute [rw] saved_queries
+        #   @return [::Array<::Google::Cloud::AIPlatform::V1::SavedQuery>]
+        #     A list of SavedQueries that match the specified filter in the request.
+        # @!attribute [rw] next_page_token
+        #   @return [::String]
+        #     The standard List next-page token.
+        class ListSavedQueriesResponse
+          include ::Google::Protobuf::MessageExts
+          extend ::Google::Protobuf::MessageExts::ClassMethods
+        end
+
         # Request message for {::Google::Cloud::AIPlatform::V1::DatasetService::Client#get_annotation_spec DatasetService.GetAnnotationSpec}.
         # @!attribute [rw] name
         #   @return [::String]

data/proto_docs/google/cloud/aiplatform/v1/endpoint.rb

@@ -148,6 +148,12 @@ module Google
         #   @return [::String]
         #     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.
+        #
+        #     The resource name may contain version id or version alias to specify the
+        #     version, if no version is specified, the default version will be deployed.
+        # @!attribute [r] model_version_id
+        #   @return [::String]
+        #     Output only. The version ID of the model that is deployed.
         # @!attribute [rw] display_name
         #   @return [::String]
         #     The display name of the DeployedModel. If not provided upon creation,