google-cloud-ai_platform-v1 0.37.0 → 0.38.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: cd0581c567f247302d00a412853f32674a882726467887682a0d83f143b7e8a8
-  data.tar.gz: a6ae84ef91cefd447b0dcaabaa014f7eaeb06cbd425dd237a242f9f8e0d74dd6
+  metadata.gz: 061ef1a71794c5d397cbff5f8b8c4568063952311d8991223212234360e2e590
+  data.tar.gz: aeae4aa17d5e4768e092d8706062a35d57f1f685d6dfe27e20aa9656f3885e73
 SHA512:
-  metadata.gz: 538f1d0daad928b335c5b05c03c7a022ed572035ae004fd8f63643879b231fb8b7152697131b86fa221318c470c284f6b5a66914e59b3eb53dea2130b3efbe65
-  data.tar.gz: bea7867b7fb33bc8c4ffe75ffc5f3777190d9d1796a4961046473f11e77ac063127cd7a46090cb573203c2b32bb541b8e8c755b09c7e82a1858b2fcc6e2ca17e
+  metadata.gz: a5d8117c2cb825ec88450d7ccde1f1033d6dd9892f6d283ecf2fa64a176236292e856b2be07ba7462cf94057c400ab071669417714745b7d121151c113bb40c6
+  data.tar.gz: 421974eb91087333219ff8e7ea771cb3e821eadf77cafffb92e416f1d33c3a4ff4e0429421111007468e7d08f7e63e0de92e98b0672f8f20b0e1a2f3d2f50d00

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

@@ -427,20 +427,23 @@ module Google
             #     Optional. An expression for filtering the results of the request. For field
             #     names both snake_case and camelCase are supported.
             #
-            #       * `endpoint` supports = and !=. `endpoint` represents the Endpoint ID,
+            #       * `endpoint` supports `=` and `!=`. `endpoint` represents the Endpoint
+            #       ID,
             #         i.e. the last segment of the Endpoint's [resource
             #         name][google.cloud.aiplatform.v1.Endpoint.name].
-            #       * `display_name` supports = and, !=
+            #       * `display_name` supports `=` and `!=`.
             #       * `labels` supports general map functions that is:
             #         * `labels.key=value` - key:value equality
-            #         * `labels.key:* or labels:key - key existence
+            #         * `labels.key:*` or `labels:key` - key existence
             #         * A key including a space must be quoted. `labels."a key"`.
+            #       * `base_model_name` only supports `=`.
             #
             #     Some examples:
             #
             #       * `endpoint=1`
             #       * `displayName="myDisplayName"`
             #       * `labels.myKey="myValue"`
+            #       * `baseModelName="text-bison"`
             #   @param page_size [::Integer]
             #     Optional. The standard list page size.
             #   @param page_token [::String]

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

@@ -446,12 +446,14 @@ module Google
             #         * `labels.key=value` - key:value equality
             #         * `labels.key:* or labels:key - key existence
             #         * A key including a space must be quoted. `labels."a key"`.
+            #       * `base_model_name` only supports =
             #
             #     Some examples:
             #
             #       * `model=1234`
             #       * `displayName="myDisplayName"`
             #       * `labels.myKey="myValue"`
+            #       * `baseModelName="text-bison"`
             #   @param page_size [::Integer]
             #     The standard list page size.
             #   @param page_token [::String]

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

@@ -1128,6 +1128,107 @@ module Google
               raise ::Google::Cloud::Error.from_error(e)
             end
 
+            ##
+            # Batch deletes PipelineJobs
+            # The Operation is atomic. If it fails, none of the PipelineJobs are deleted.
+            # If it succeeds, all of the PipelineJobs are deleted.
+            #
+            # @overload batch_delete_pipeline_jobs(request, options = nil)
+            #   Pass arguments to `batch_delete_pipeline_jobs` via a request object, either of type
+            #   {::Google::Cloud::AIPlatform::V1::BatchDeletePipelineJobsRequest} or an equivalent Hash.
+            #
+            #   @param request [::Google::Cloud::AIPlatform::V1::BatchDeletePipelineJobsRequest, ::Hash]
+            #     A request object representing the call parameters. Required. To specify no
+            #     parameters, or to keep all the default parameter values, pass an empty Hash.
+            #   @param options [::Gapic::CallOptions, ::Hash]
+            #     Overrides the default settings for this call, e.g, timeout, retries, etc. Optional.
+            #
+            # @overload batch_delete_pipeline_jobs(parent: nil, names: nil)
+            #   Pass arguments to `batch_delete_pipeline_jobs` via keyword arguments. Note that at
+            #   least one keyword argument is required. To specify no parameters, or to keep all
+            #   the default parameter values, pass an empty Hash as a request object (see above).
+            #
+            #   @param parent [::String]
+            #     Required. The name of the PipelineJobs' parent resource.
+            #     Format: `projects/{project}/locations/{location}`
+            #   @param names [::Array<::String>]
+            #     Required. The names of the PipelineJobs to delete.
+            #     A maximum of 32 PipelineJobs can be deleted in a batch.
+            #     Format:
+            #     `projects/{project}/locations/{location}/pipelineJobs/{pipelineJob}`
+            #
+            # @yield [response, operation] Access the result along with the RPC operation
+            # @yieldparam response [::Gapic::Operation]
+            # @yieldparam operation [::GRPC::ActiveCall::Operation]
+            #
+            # @return [::Gapic::Operation]
+            #
+            # @raise [::Google::Cloud::Error] if the RPC is aborted.
+            #
+            # @example Basic example
+            #   require "google/cloud/ai_platform/v1"
+            #
+            #   # Create a client object. The client can be reused for multiple calls.
+            #   client = Google::Cloud::AIPlatform::V1::PipelineService::Client.new
+            #
+            #   # Create a request. To set request fields, pass in keyword arguments.
+            #   request = Google::Cloud::AIPlatform::V1::BatchDeletePipelineJobsRequest.new
+            #
+            #   # Call the batch_delete_pipeline_jobs method.
+            #   result = client.batch_delete_pipeline_jobs request
+            #
+            #   # The returned object is of type Gapic::Operation. You can use it to
+            #   # check the status of an operation, cancel it, or wait for results.
+            #   # Here is how to wait for a response.
+            #   result.wait_until_done! timeout: 60
+            #   if result.response?
+            #     p result.response
+            #   else
+            #     puts "No response received."
+            #   end
+            #
+            def batch_delete_pipeline_jobs request, options = nil
+              raise ::ArgumentError, "request must be provided" if request.nil?
+
+              request = ::Gapic::Protobuf.coerce request, to: ::Google::Cloud::AIPlatform::V1::BatchDeletePipelineJobsRequest
+
+              # Converts hash and nil to an options object
+              options = ::Gapic::CallOptions.new(**options.to_h) if options.respond_to? :to_h
+
+              # Customize the options with defaults
+              metadata = @config.rpcs.batch_delete_pipeline_jobs.metadata.to_h
+
+              # Set x-goog-api-client and x-goog-user-project headers
+              metadata[:"x-goog-api-client"] ||= ::Gapic::Headers.x_goog_api_client \
+                lib_name: @config.lib_name, lib_version: @config.lib_version,
+                gapic_version: ::Google::Cloud::AIPlatform::V1::VERSION
+              metadata[:"x-goog-user-project"] = @quota_project_id if @quota_project_id
+
+              header_params = {}
+              if request.parent
+                header_params["parent"] = request.parent
+              end
+
+              request_params_header = header_params.map { |k, v| "#{k}=#{v}" }.join("&")
+              metadata[:"x-goog-request-params"] ||= request_params_header
+
+              options.apply_defaults timeout:      @config.rpcs.batch_delete_pipeline_jobs.timeout,
+                                     metadata:     metadata,
+                                     retry_policy: @config.rpcs.batch_delete_pipeline_jobs.retry_policy
+
+              options.apply_defaults timeout:      @config.timeout,
+                                     metadata:     @config.metadata,
+                                     retry_policy: @config.retry_policy
+
+              @pipeline_service_stub.call_rpc :batch_delete_pipeline_jobs, request, options: options do |response, operation|
+                response = ::Gapic::Operation.new response, @operations_client, options: options
+                yield response, operation if block_given?
+                return response
+              end
+            rescue ::GRPC::BadStatus => e
+              raise ::Google::Cloud::Error.from_error(e)
+            end
+
             ##
             # Cancels a PipelineJob.
             # Starts asynchronous cancellation on the PipelineJob. The server
@@ -1227,6 +1328,111 @@ module Google
               raise ::Google::Cloud::Error.from_error(e)
             end
 
+            ##
+            # Batch cancel PipelineJobs.
+            # Firstly the server will check if all the jobs are in non-terminal states,
+            # and skip the jobs that are already terminated.
+            # If the operation failed, none of the pipeline jobs are cancelled.
+            # The server will poll the states of all the pipeline jobs periodically
+            # to check the cancellation status.
+            # This operation will return an LRO.
+            #
+            # @overload batch_cancel_pipeline_jobs(request, options = nil)
+            #   Pass arguments to `batch_cancel_pipeline_jobs` via a request object, either of type
+            #   {::Google::Cloud::AIPlatform::V1::BatchCancelPipelineJobsRequest} or an equivalent Hash.
+            #
+            #   @param request [::Google::Cloud::AIPlatform::V1::BatchCancelPipelineJobsRequest, ::Hash]
+            #     A request object representing the call parameters. Required. To specify no
+            #     parameters, or to keep all the default parameter values, pass an empty Hash.
+            #   @param options [::Gapic::CallOptions, ::Hash]
+            #     Overrides the default settings for this call, e.g, timeout, retries, etc. Optional.
+            #
+            # @overload batch_cancel_pipeline_jobs(parent: nil, names: nil)
+            #   Pass arguments to `batch_cancel_pipeline_jobs` via keyword arguments. Note that at
+            #   least one keyword argument is required. To specify no parameters, or to keep all
+            #   the default parameter values, pass an empty Hash as a request object (see above).
+            #
+            #   @param parent [::String]
+            #     Required. The name of the PipelineJobs' parent resource.
+            #     Format: `projects/{project}/locations/{location}`
+            #   @param names [::Array<::String>]
+            #     Required. The names of the PipelineJobs to cancel.
+            #     A maximum of 32 PipelineJobs can be cancelled in a batch.
+            #     Format:
+            #     `projects/{project}/locations/{location}/pipelineJobs/{pipelineJob}`
+            #
+            # @yield [response, operation] Access the result along with the RPC operation
+            # @yieldparam response [::Gapic::Operation]
+            # @yieldparam operation [::GRPC::ActiveCall::Operation]
+            #
+            # @return [::Gapic::Operation]
+            #
+            # @raise [::Google::Cloud::Error] if the RPC is aborted.
+            #
+            # @example Basic example
+            #   require "google/cloud/ai_platform/v1"
+            #
+            #   # Create a client object. The client can be reused for multiple calls.
+            #   client = Google::Cloud::AIPlatform::V1::PipelineService::Client.new
+            #
+            #   # Create a request. To set request fields, pass in keyword arguments.
+            #   request = Google::Cloud::AIPlatform::V1::BatchCancelPipelineJobsRequest.new
+            #
+            #   # Call the batch_cancel_pipeline_jobs method.
+            #   result = client.batch_cancel_pipeline_jobs request
+            #
+            #   # The returned object is of type Gapic::Operation. You can use it to
+            #   # check the status of an operation, cancel it, or wait for results.
+            #   # Here is how to wait for a response.
+            #   result.wait_until_done! timeout: 60
+            #   if result.response?
+            #     p result.response
+            #   else
+            #     puts "No response received."
+            #   end
+            #
+            def batch_cancel_pipeline_jobs request, options = nil
+              raise ::ArgumentError, "request must be provided" if request.nil?
+
+              request = ::Gapic::Protobuf.coerce request, to: ::Google::Cloud::AIPlatform::V1::BatchCancelPipelineJobsRequest
+
+              # Converts hash and nil to an options object
+              options = ::Gapic::CallOptions.new(**options.to_h) if options.respond_to? :to_h
+
+              # Customize the options with defaults
+              metadata = @config.rpcs.batch_cancel_pipeline_jobs.metadata.to_h
+
+              # Set x-goog-api-client and x-goog-user-project headers
+              metadata[:"x-goog-api-client"] ||= ::Gapic::Headers.x_goog_api_client \
+                lib_name: @config.lib_name, lib_version: @config.lib_version,
+                gapic_version: ::Google::Cloud::AIPlatform::V1::VERSION
+              metadata[:"x-goog-user-project"] = @quota_project_id if @quota_project_id
+
+              header_params = {}
+              if request.parent
+                header_params["parent"] = request.parent
+              end
+
+              request_params_header = header_params.map { |k, v| "#{k}=#{v}" }.join("&")
+              metadata[:"x-goog-request-params"] ||= request_params_header
+
+              options.apply_defaults timeout:      @config.rpcs.batch_cancel_pipeline_jobs.timeout,
+                                     metadata:     metadata,
+                                     retry_policy: @config.rpcs.batch_cancel_pipeline_jobs.retry_policy
+
+              options.apply_defaults timeout:      @config.timeout,
+                                     metadata:     @config.metadata,
+                                     retry_policy: @config.retry_policy
+
+              @pipeline_service_stub.call_rpc :batch_cancel_pipeline_jobs, request, options: options do |response, operation|
+                response = ::Gapic::Operation.new response, @operations_client, options: options
+                yield response, operation if block_given?
+                return response
+              end
+            rescue ::GRPC::BadStatus => e
+              raise ::Google::Cloud::Error.from_error(e)
+            end
+
             ##
             # Configuration class for the PipelineService API.
             #
@@ -1426,10 +1632,20 @@ module Google
                 #
                 attr_reader :delete_pipeline_job
                 ##
+                # RPC-specific configuration for `batch_delete_pipeline_jobs`
+                # @return [::Gapic::Config::Method]
+                #
+                attr_reader :batch_delete_pipeline_jobs
+                ##
                 # RPC-specific configuration for `cancel_pipeline_job`
                 # @return [::Gapic::Config::Method]
                 #
                 attr_reader :cancel_pipeline_job
+                ##
+                # RPC-specific configuration for `batch_cancel_pipeline_jobs`
+                # @return [::Gapic::Config::Method]
+                #
+                attr_reader :batch_cancel_pipeline_jobs
 
                 # @private
                 def initialize parent_rpcs = nil
@@ -1451,8 +1667,12 @@ module Google
                   @list_pipeline_jobs = ::Gapic::Config::Method.new list_pipeline_jobs_config
                   delete_pipeline_job_config = parent_rpcs.delete_pipeline_job if parent_rpcs.respond_to? :delete_pipeline_job
                   @delete_pipeline_job = ::Gapic::Config::Method.new delete_pipeline_job_config
+                  batch_delete_pipeline_jobs_config = parent_rpcs.batch_delete_pipeline_jobs if parent_rpcs.respond_to? :batch_delete_pipeline_jobs
+                  @batch_delete_pipeline_jobs = ::Gapic::Config::Method.new batch_delete_pipeline_jobs_config
                   cancel_pipeline_job_config = parent_rpcs.cancel_pipeline_job if parent_rpcs.respond_to? :cancel_pipeline_job
                   @cancel_pipeline_job = ::Gapic::Config::Method.new cancel_pipeline_job_config
+                  batch_cancel_pipeline_jobs_config = parent_rpcs.batch_cancel_pipeline_jobs if parent_rpcs.respond_to? :batch_cancel_pipeline_jobs
+                  @batch_cancel_pipeline_jobs = ::Gapic::Config::Method.new batch_cancel_pipeline_jobs_config
 
                   yield self if block_given?
                 end

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

@@ -21,7 +21,7 @@ module Google
   module Cloud
     module AIPlatform
       module V1
-        VERSION = "0.37.0"
+        VERSION = "0.38.0"
       end
     end
   end

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

@@ -5,7 +5,7 @@
 require 'google/protobuf'
 
 
-descriptor_data = "\n1google/cloud/aiplatform/v1/accelerator_type.proto\x12\x1agoogle.cloud.aiplatform.v1*\x85\x02\n\x0f\x41\x63\x63\x65leratorType\x12 \n\x1c\x41\x43\x43\x45LERATOR_TYPE_UNSPECIFIED\x10\x00\x12\x14\n\x10NVIDIA_TESLA_K80\x10\x01\x12\x15\n\x11NVIDIA_TESLA_P100\x10\x02\x12\x15\n\x11NVIDIA_TESLA_V100\x10\x03\x12\x13\n\x0fNVIDIA_TESLA_P4\x10\x04\x12\x13\n\x0fNVIDIA_TESLA_T4\x10\x05\x12\x15\n\x11NVIDIA_TESLA_A100\x10\x08\x12\x14\n\x10NVIDIA_A100_80GB\x10\t\x12\r\n\tNVIDIA_L4\x10\x0b\x12\n\n\x06TPU_V2\x10\x06\x12\n\n\x06TPU_V3\x10\x07\x12\x0e\n\nTPU_V4_POD\x10\nB\xd2\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\x14\x41\x63\x63\x65leratorTypeProtoP\x01Z>cloud.google.com/go/aiplatform/apiv1/aiplatformpb;aiplatformpb\xaa\x02\x1aGoogle.Cloud.AIPlatform.V1\xca\x02\x1aGoogle\\Cloud\\AIPlatform\\V1\xea\x02\x1dGoogle::Cloud::AIPlatform::V1b\x06proto3"
+descriptor_data = "\n1google/cloud/aiplatform/v1/accelerator_type.proto\x12\x1agoogle.cloud.aiplatform.v1*\x9b\x02\n\x0f\x41\x63\x63\x65leratorType\x12 \n\x1c\x41\x43\x43\x45LERATOR_TYPE_UNSPECIFIED\x10\x00\x12\x14\n\x10NVIDIA_TESLA_K80\x10\x01\x12\x15\n\x11NVIDIA_TESLA_P100\x10\x02\x12\x15\n\x11NVIDIA_TESLA_V100\x10\x03\x12\x13\n\x0fNVIDIA_TESLA_P4\x10\x04\x12\x13\n\x0fNVIDIA_TESLA_T4\x10\x05\x12\x15\n\x11NVIDIA_TESLA_A100\x10\x08\x12\x14\n\x10NVIDIA_A100_80GB\x10\t\x12\r\n\tNVIDIA_L4\x10\x0b\x12\x14\n\x10NVIDIA_H100_80GB\x10\r\x12\n\n\x06TPU_V2\x10\x06\x12\n\n\x06TPU_V3\x10\x07\x12\x0e\n\nTPU_V4_POD\x10\nB\xd2\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\x14\x41\x63\x63\x65leratorTypeProtoP\x01Z>cloud.google.com/go/aiplatform/apiv1/aiplatformpb;aiplatformpb\xaa\x02\x1aGoogle.Cloud.AIPlatform.V1\xca\x02\x1aGoogle\\Cloud\\AIPlatform\\V1\xea\x02\x1dGoogle::Cloud::AIPlatform::V1b\x06proto3"
 
 pool = Google::Protobuf::DescriptorPool.generated_pool
 

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

@@ -10,7 +10,7 @@ require 'google/protobuf/duration_pb'
 require 'google/type/date_pb'
 
 
-descriptor_data = "\n(google/cloud/aiplatform/v1/content.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a%google/cloud/aiplatform/v1/tool.proto\x1a\x1egoogle/protobuf/duration.proto\x1a\x16google/type/date.proto\"R\n\x07\x43ontent\x12\x11\n\x04role\x18\x01 \x01(\tB\x03\xe0\x41\x01\x12\x34\n\x05parts\x18\x02 \x03(\x0b\x32 .google.cloud.aiplatform.v1.PartB\x03\xe0\x41\x02\"\x8f\x03\n\x04Part\x12\x13\n\x04text\x18\x01 \x01(\tB\x03\xe0\x41\x01H\x00\x12<\n\x0binline_data\x18\x02 \x01(\x0b\x32 .google.cloud.aiplatform.v1.BlobB\x03\xe0\x41\x01H\x00\x12>\n\tfile_data\x18\x03 \x01(\x0b\x32$.google.cloud.aiplatform.v1.FileDataB\x03\xe0\x41\x01H\x00\x12\x46\n\rfunction_call\x18\x05 \x01(\x0b\x32(.google.cloud.aiplatform.v1.FunctionCallB\x03\xe0\x41\x01H\x00\x12N\n\x11\x66unction_response\x18\x06 \x01(\x0b\x32,.google.cloud.aiplatform.v1.FunctionResponseB\x03\xe0\x41\x01H\x00\x12H\n\x0evideo_metadata\x18\x04 \x01(\x0b\x32).google.cloud.aiplatform.v1.VideoMetadataB\x03\xe0\x41\x01H\x01\x42\x06\n\x04\x64\x61taB\n\n\x08metadata\"1\n\x04\x42lob\x12\x16\n\tmime_type\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x11\n\x04\x64\x61ta\x18\x02 \x01(\x0c\x42\x03\xe0\x41\x02\"9\n\x08\x46ileData\x12\x16\n\tmime_type\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x15\n\x08\x66ile_uri\x18\x02 \x01(\tB\x03\xe0\x41\x02\"y\n\rVideoMetadata\x12\x34\n\x0cstart_offset\x18\x01 \x01(\x0b\x32\x19.google.protobuf.DurationB\x03\xe0\x41\x01\x12\x32\n\nend_offset\x18\x02 \x01(\x0b\x32\x19.google.protobuf.DurationB\x03\xe0\x41\x01\"\x96\x02\n\x10GenerationConfig\x12\x1d\n\x0btemperature\x18\x01 \x01(\x02\x42\x03\xe0\x41\x01H\x00\x88\x01\x01\x12\x17\n\x05top_p\x18\x02 \x01(\x02\x42\x03\xe0\x41\x01H\x01\x88\x01\x01\x12\x17\n\x05top_k\x18\x03 \x01(\x02\x42\x03\xe0\x41\x01H\x02\x88\x01\x01\x12!\n\x0f\x63\x61ndidate_count\x18\x04 \x01(\x05\x42\x03\xe0\x41\x01H\x03\x88\x01\x01\x12#\n\x11max_output_tokens\x18\x05 \x01(\x05\x42\x03\xe0\x41\x01H\x04\x88\x01\x01\x12\x1b\n\x0estop_sequences\x18\x06 \x03(\tB\x03\xe0\x41\x01\x42\x0e\n\x0c_temperatureB\x08\n\x06_top_pB\x08\n\x06_top_kB\x12\n\x10_candidate_countB\x14\n\x12_max_output_tokens\"\xbd\x02\n\rSafetySetting\x12?\n\x08\x63\x61tegory\x18\x01 \x01(\x0e\x32(.google.cloud.aiplatform.v1.HarmCategoryB\x03\xe0\x41\x02\x12T\n\tthreshold\x18\x02 \x01(\x0e\x32<.google.cloud.aiplatform.v1.SafetySetting.HarmBlockThresholdB\x03\xe0\x41\x02\"\x94\x01\n\x12HarmBlockThreshold\x12$\n HARM_BLOCK_THRESHOLD_UNSPECIFIED\x10\x00\x12\x17\n\x13\x42LOCK_LOW_AND_ABOVE\x10\x01\x12\x1a\n\x16\x42LOCK_MEDIUM_AND_ABOVE\x10\x02\x12\x13\n\x0f\x42LOCK_ONLY_HIGH\x10\x03\x12\x0e\n\nBLOCK_NONE\x10\x04\"\x9d\x02\n\x0cSafetyRating\x12?\n\x08\x63\x61tegory\x18\x01 \x01(\x0e\x32(.google.cloud.aiplatform.v1.HarmCategoryB\x03\xe0\x41\x03\x12R\n\x0bprobability\x18\x02 \x01(\x0e\x32\x38.google.cloud.aiplatform.v1.SafetyRating.HarmProbabilityB\x03\xe0\x41\x03\x12\x14\n\x07\x62locked\x18\x03 \x01(\x08\x42\x03\xe0\x41\x03\"b\n\x0fHarmProbability\x12 \n\x1cHARM_PROBABILITY_UNSPECIFIED\x10\x00\x12\x0e\n\nNEGLIGIBLE\x10\x01\x12\x07\n\x03LOW\x10\x02\x12\n\n\x06MEDIUM\x10\x03\x12\x08\n\x04HIGH\x10\x04\"P\n\x10\x43itationMetadata\x12<\n\tcitations\x18\x01 \x03(\x0b\x32$.google.cloud.aiplatform.v1.CitationB\x03\xe0\x41\x03\"\xaa\x01\n\x08\x43itation\x12\x18\n\x0bstart_index\x18\x01 \x01(\x05\x42\x03\xe0\x41\x03\x12\x16\n\tend_index\x18\x02 \x01(\x05\x42\x03\xe0\x41\x03\x12\x10\n\x03uri\x18\x03 \x01(\tB\x03\xe0\x41\x03\x12\x12\n\x05title\x18\x04 \x01(\tB\x03\xe0\x41\x03\x12\x14\n\x07license\x18\x05 \x01(\tB\x03\xe0\x41\x03\x12\x30\n\x10publication_date\x18\x06 \x01(\x0b\x32\x11.google.type.DateB\x03\xe0\x41\x03\"\xe4\x03\n\tCandidate\x12\x12\n\x05index\x18\x01 \x01(\x05\x42\x03\xe0\x41\x03\x12\x39\n\x07\x63ontent\x18\x02 \x01(\x0b\x32#.google.cloud.aiplatform.v1.ContentB\x03\xe0\x41\x03\x12N\n\rfinish_reason\x18\x03 \x01(\x0e\x32\x32.google.cloud.aiplatform.v1.Candidate.FinishReasonB\x03\xe0\x41\x03\x12\x45\n\x0esafety_ratings\x18\x04 \x03(\x0b\x32(.google.cloud.aiplatform.v1.SafetyRatingB\x03\xe0\x41\x03\x12 \n\x0e\x66inish_message\x18\x05 \x01(\tB\x03\xe0\x41\x03H\x00\x88\x01\x01\x12L\n\x11\x63itation_metadata\x18\x06 \x01(\x0b\x32,.google.cloud.aiplatform.v1.CitationMetadataB\x03\xe0\x41\x03\"n\n\x0c\x46inishReason\x12\x1d\n\x19\x46INISH_REASON_UNSPECIFIED\x10\x00\x12\x08\n\x04STOP\x10\x01\x12\x0e\n\nMAX_TOKENS\x10\x02\x12\n\n\x06SAFETY\x10\x03\x12\x0e\n\nRECITATION\x10\x04\x12\t\n\x05OTHER\x10\x05\x42\x11\n\x0f_finish_message*\xb4\x01\n\x0cHarmCategory\x12\x1d\n\x19HARM_CATEGORY_UNSPECIFIED\x10\x00\x12\x1d\n\x19HARM_CATEGORY_HATE_SPEECH\x10\x01\x12#\n\x1fHARM_CATEGORY_DANGEROUS_CONTENT\x10\x02\x12\x1c\n\x18HARM_CATEGORY_HARASSMENT\x10\x03\x12#\n\x1fHARM_CATEGORY_SEXUALLY_EXPLICIT\x10\x04\x42\xca\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\x0c\x43ontentProtoP\x01Z>cloud.google.com/go/aiplatform/apiv1/aiplatformpb;aiplatformpb\xaa\x02\x1aGoogle.Cloud.AIPlatform.V1\xca\x02\x1aGoogle\\Cloud\\AIPlatform\\V1\xea\x02\x1dGoogle::Cloud::AIPlatform::V1b\x06proto3"
+descriptor_data = "\n(google/cloud/aiplatform/v1/content.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a%google/cloud/aiplatform/v1/tool.proto\x1a\x1egoogle/protobuf/duration.proto\x1a\x16google/type/date.proto\"R\n\x07\x43ontent\x12\x11\n\x04role\x18\x01 \x01(\tB\x03\xe0\x41\x01\x12\x34\n\x05parts\x18\x02 \x03(\x0b\x32 .google.cloud.aiplatform.v1.PartB\x03\xe0\x41\x02\"\x8f\x03\n\x04Part\x12\x13\n\x04text\x18\x01 \x01(\tB\x03\xe0\x41\x01H\x00\x12<\n\x0binline_data\x18\x02 \x01(\x0b\x32 .google.cloud.aiplatform.v1.BlobB\x03\xe0\x41\x01H\x00\x12>\n\tfile_data\x18\x03 \x01(\x0b\x32$.google.cloud.aiplatform.v1.FileDataB\x03\xe0\x41\x01H\x00\x12\x46\n\rfunction_call\x18\x05 \x01(\x0b\x32(.google.cloud.aiplatform.v1.FunctionCallB\x03\xe0\x41\x01H\x00\x12N\n\x11\x66unction_response\x18\x06 \x01(\x0b\x32,.google.cloud.aiplatform.v1.FunctionResponseB\x03\xe0\x41\x01H\x00\x12H\n\x0evideo_metadata\x18\x04 \x01(\x0b\x32).google.cloud.aiplatform.v1.VideoMetadataB\x03\xe0\x41\x01H\x01\x42\x06\n\x04\x64\x61taB\n\n\x08metadata\"1\n\x04\x42lob\x12\x16\n\tmime_type\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x11\n\x04\x64\x61ta\x18\x02 \x01(\x0c\x42\x03\xe0\x41\x02\"9\n\x08\x46ileData\x12\x16\n\tmime_type\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x15\n\x08\x66ile_uri\x18\x02 \x01(\tB\x03\xe0\x41\x02\"y\n\rVideoMetadata\x12\x34\n\x0cstart_offset\x18\x01 \x01(\x0b\x32\x19.google.protobuf.DurationB\x03\xe0\x41\x01\x12\x32\n\nend_offset\x18\x02 \x01(\x0b\x32\x19.google.protobuf.DurationB\x03\xe0\x41\x01\"\x96\x02\n\x10GenerationConfig\x12\x1d\n\x0btemperature\x18\x01 \x01(\x02\x42\x03\xe0\x41\x01H\x00\x88\x01\x01\x12\x17\n\x05top_p\x18\x02 \x01(\x02\x42\x03\xe0\x41\x01H\x01\x88\x01\x01\x12\x17\n\x05top_k\x18\x03 \x01(\x02\x42\x03\xe0\x41\x01H\x02\x88\x01\x01\x12!\n\x0f\x63\x61ndidate_count\x18\x04 \x01(\x05\x42\x03\xe0\x41\x01H\x03\x88\x01\x01\x12#\n\x11max_output_tokens\x18\x05 \x01(\x05\x42\x03\xe0\x41\x01H\x04\x88\x01\x01\x12\x1b\n\x0estop_sequences\x18\x06 \x03(\tB\x03\xe0\x41\x01\x42\x0e\n\x0c_temperatureB\x08\n\x06_top_pB\x08\n\x06_top_kB\x12\n\x10_candidate_countB\x14\n\x12_max_output_tokens\"\xbd\x02\n\rSafetySetting\x12?\n\x08\x63\x61tegory\x18\x01 \x01(\x0e\x32(.google.cloud.aiplatform.v1.HarmCategoryB\x03\xe0\x41\x02\x12T\n\tthreshold\x18\x02 \x01(\x0e\x32<.google.cloud.aiplatform.v1.SafetySetting.HarmBlockThresholdB\x03\xe0\x41\x02\"\x94\x01\n\x12HarmBlockThreshold\x12$\n HARM_BLOCK_THRESHOLD_UNSPECIFIED\x10\x00\x12\x17\n\x13\x42LOCK_LOW_AND_ABOVE\x10\x01\x12\x1a\n\x16\x42LOCK_MEDIUM_AND_ABOVE\x10\x02\x12\x13\n\x0f\x42LOCK_ONLY_HIGH\x10\x03\x12\x0e\n\nBLOCK_NONE\x10\x04\"\xbf\x04\n\x0cSafetyRating\x12?\n\x08\x63\x61tegory\x18\x01 \x01(\x0e\x32(.google.cloud.aiplatform.v1.HarmCategoryB\x03\xe0\x41\x03\x12R\n\x0bprobability\x18\x02 \x01(\x0e\x32\x38.google.cloud.aiplatform.v1.SafetyRating.HarmProbabilityB\x03\xe0\x41\x03\x12\x1e\n\x11probability_score\x18\x05 \x01(\x02\x42\x03\xe0\x41\x03\x12L\n\x08severity\x18\x06 \x01(\x0e\x32\x35.google.cloud.aiplatform.v1.SafetyRating.HarmSeverityB\x03\xe0\x41\x03\x12\x1b\n\x0eseverity_score\x18\x07 \x01(\x02\x42\x03\xe0\x41\x03\x12\x14\n\x07\x62locked\x18\x03 \x01(\x08\x42\x03\xe0\x41\x03\"b\n\x0fHarmProbability\x12 \n\x1cHARM_PROBABILITY_UNSPECIFIED\x10\x00\x12\x0e\n\nNEGLIGIBLE\x10\x01\x12\x07\n\x03LOW\x10\x02\x12\n\n\x06MEDIUM\x10\x03\x12\x08\n\x04HIGH\x10\x04\"\x94\x01\n\x0cHarmSeverity\x12\x1d\n\x19HARM_SEVERITY_UNSPECIFIED\x10\x00\x12\x1c\n\x18HARM_SEVERITY_NEGLIGIBLE\x10\x01\x12\x15\n\x11HARM_SEVERITY_LOW\x10\x02\x12\x18\n\x14HARM_SEVERITY_MEDIUM\x10\x03\x12\x16\n\x12HARM_SEVERITY_HIGH\x10\x04\"P\n\x10\x43itationMetadata\x12<\n\tcitations\x18\x01 \x03(\x0b\x32$.google.cloud.aiplatform.v1.CitationB\x03\xe0\x41\x03\"\xaa\x01\n\x08\x43itation\x12\x18\n\x0bstart_index\x18\x01 \x01(\x05\x42\x03\xe0\x41\x03\x12\x16\n\tend_index\x18\x02 \x01(\x05\x42\x03\xe0\x41\x03\x12\x10\n\x03uri\x18\x03 \x01(\tB\x03\xe0\x41\x03\x12\x12\n\x05title\x18\x04 \x01(\tB\x03\xe0\x41\x03\x12\x14\n\x07license\x18\x05 \x01(\tB\x03\xe0\x41\x03\x12\x30\n\x10publication_date\x18\x06 \x01(\x0b\x32\x11.google.type.DateB\x03\xe0\x41\x03\"\xe6\x04\n\tCandidate\x12\x12\n\x05index\x18\x01 \x01(\x05\x42\x03\xe0\x41\x03\x12\x39\n\x07\x63ontent\x18\x02 \x01(\x0b\x32#.google.cloud.aiplatform.v1.ContentB\x03\xe0\x41\x03\x12N\n\rfinish_reason\x18\x03 \x01(\x0e\x32\x32.google.cloud.aiplatform.v1.Candidate.FinishReasonB\x03\xe0\x41\x03\x12\x45\n\x0esafety_ratings\x18\x04 \x03(\x0b\x32(.google.cloud.aiplatform.v1.SafetyRatingB\x03\xe0\x41\x03\x12 \n\x0e\x66inish_message\x18\x05 \x01(\tB\x03\xe0\x41\x03H\x00\x88\x01\x01\x12L\n\x11\x63itation_metadata\x18\x06 \x01(\x0b\x32,.google.cloud.aiplatform.v1.CitationMetadataB\x03\xe0\x41\x03\x12N\n\x12grounding_metadata\x18\x07 \x01(\x0b\x32-.google.cloud.aiplatform.v1.GroundingMetadataB\x03\xe0\x41\x03\"\x9f\x01\n\x0c\x46inishReason\x12\x1d\n\x19\x46INISH_REASON_UNSPECIFIED\x10\x00\x12\x08\n\x04STOP\x10\x01\x12\x0e\n\nMAX_TOKENS\x10\x02\x12\n\n\x06SAFETY\x10\x03\x12\x0e\n\nRECITATION\x10\x04\x12\t\n\x05OTHER\x10\x05\x12\r\n\tBLOCKLIST\x10\x06\x12\x16\n\x12PROHIBITED_CONTENT\x10\x07\x12\x08\n\x04SPII\x10\x08\x42\x11\n\x0f_finish_message\"T\n\x07Segment\x12\x17\n\npart_index\x18\x01 \x01(\x05\x42\x03\xe0\x41\x03\x12\x18\n\x0bstart_index\x18\x02 \x01(\x05\x42\x03\xe0\x41\x03\x12\x16\n\tend_index\x18\x03 \x01(\x05\x42\x03\xe0\x41\x03\"\x91\x02\n\x14GroundingAttribution\x12H\n\x03web\x18\x03 \x01(\x0b\x32\x34.google.cloud.aiplatform.v1.GroundingAttribution.WebB\x03\xe0\x41\x01H\x00\x12\x39\n\x07segment\x18\x01 \x01(\x0b\x32#.google.cloud.aiplatform.v1.SegmentB\x03\xe0\x41\x03\x12%\n\x10\x63onfidence_score\x18\x02 \x01(\x02\x42\x06\xe0\x41\x01\xe0\x41\x03H\x01\x88\x01\x01\x1a+\n\x03Web\x12\x10\n\x03uri\x18\x01 \x01(\tB\x03\xe0\x41\x03\x12\x12\n\x05title\x18\x02 \x01(\tB\x03\xe0\x41\x03\x42\x0b\n\treferenceB\x13\n\x11_confidence_score\"\x8b\x01\n\x11GroundingMetadata\x12\x1f\n\x12web_search_queries\x18\x01 \x03(\tB\x03\xe0\x41\x01\x12U\n\x16grounding_attributions\x18\x02 \x03(\x0b\x32\x30.google.cloud.aiplatform.v1.GroundingAttributionB\x03\xe0\x41\x01*\xb4\x01\n\x0cHarmCategory\x12\x1d\n\x19HARM_CATEGORY_UNSPECIFIED\x10\x00\x12\x1d\n\x19HARM_CATEGORY_HATE_SPEECH\x10\x01\x12#\n\x1fHARM_CATEGORY_DANGEROUS_CONTENT\x10\x02\x12\x1c\n\x18HARM_CATEGORY_HARASSMENT\x10\x03\x12#\n\x1fHARM_CATEGORY_SEXUALLY_EXPLICIT\x10\x04\x42\xca\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\x0c\x43ontentProtoP\x01Z>cloud.google.com/go/aiplatform/apiv1/aiplatformpb;aiplatformpb\xaa\x02\x1aGoogle.Cloud.AIPlatform.V1\xca\x02\x1aGoogle\\Cloud\\AIPlatform\\V1\xea\x02\x1dGoogle::Cloud::AIPlatform::V1b\x06proto3"
 
 pool = Google::Protobuf::DescriptorPool.generated_pool
 
@@ -53,10 +53,15 @@ module Google
         SafetySetting::HarmBlockThreshold = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.SafetySetting.HarmBlockThreshold").enummodule
         SafetyRating = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.SafetyRating").msgclass
         SafetyRating::HarmProbability = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.SafetyRating.HarmProbability").enummodule
+        SafetyRating::HarmSeverity = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.SafetyRating.HarmSeverity").enummodule
         CitationMetadata = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.CitationMetadata").msgclass
         Citation = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.Citation").msgclass
         Candidate = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.Candidate").msgclass
         Candidate::FinishReason = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.Candidate.FinishReason").enummodule
+        Segment = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.Segment").msgclass
+        GroundingAttribution = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.GroundingAttribution").msgclass
+        GroundingAttribution::Web = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.GroundingAttribution.Web").msgclass
+        GroundingMetadata = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.GroundingMetadata").msgclass
         HarmCategory = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.HarmCategory").enummodule
       end
     end

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

@@ -6,10 +6,11 @@ require 'google/protobuf'
 
 require 'google/api/field_behavior_pb'
 require 'google/api/resource_pb'
+require 'google/protobuf/struct_pb'
 require 'google/protobuf/timestamp_pb'
 
 
-descriptor_data = "\n0google/cloud/aiplatform/v1/dataset_version.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a\x1fgoogle/protobuf/timestamp.proto\"\xd1\x02\n\x0e\x44\x61tasetVersion\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x03\x12\x34\n\x0b\x63reate_time\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x34\n\x0bupdate_time\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x0c\n\x04\x65tag\x18\x03 \x01(\t\x12#\n\x16\x62ig_query_dataset_name\x18\x04 \x01(\tB\x03\xe0\x41\x03:\x8c\x01\xea\x41\x88\x01\n(aiplatform.googleapis.com/DatasetVersion\x12\\projects/{project}/locations/{location}/datasets/{dataset}/datasetVersions/{dataset_version}B\xd1\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\x13\x44\x61tasetVersionProtoP\x01Z>cloud.google.com/go/aiplatform/apiv1/aiplatformpb;aiplatformpb\xaa\x02\x1aGoogle.Cloud.AIPlatform.V1\xca\x02\x1aGoogle\\Cloud\\AIPlatform\\V1\xea\x02\x1dGoogle::Cloud::AIPlatform::V1b\x06proto3"
+descriptor_data = "\n0google/cloud/aiplatform/v1/dataset_version.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1fgoogle/protobuf/timestamp.proto\"\x99\x03\n\x0e\x44\x61tasetVersion\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x03\x12\x34\n\x0b\x63reate_time\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x34\n\x0bupdate_time\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x0c\n\x04\x65tag\x18\x03 \x01(\t\x12#\n\x16\x62ig_query_dataset_name\x18\x04 \x01(\tB\x03\xe0\x41\x03\x12\x14\n\x0c\x64isplay_name\x18\x07 \x01(\t\x12\x30\n\x08metadata\x18\x08 \x01(\x0b\x32\x16.google.protobuf.ValueB\x06\xe0\x41\x03\xe0\x41\x02:\x8c\x01\xea\x41\x88\x01\n(aiplatform.googleapis.com/DatasetVersion\x12\\projects/{project}/locations/{location}/datasets/{dataset}/datasetVersions/{dataset_version}B\xd1\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\x13\x44\x61tasetVersionProtoP\x01Z>cloud.google.com/go/aiplatform/apiv1/aiplatformpb;aiplatformpb\xaa\x02\x1aGoogle.Cloud.AIPlatform.V1\xca\x02\x1aGoogle\\Cloud\\AIPlatform\\V1\xea\x02\x1dGoogle::Cloud::AIPlatform::V1b\x06proto3"
 
 pool = Google::Protobuf::DescriptorPool.generated_pool
 
@@ -25,6 +26,7 @@ rescue TypeError
   warn "Warning: Protobuf detected an import path issue while loading generated file #{__FILE__}"
   imports = [
     ["google.protobuf.Timestamp", "google/protobuf/timestamp.proto"],
+    ["google.protobuf.Value", "google/protobuf/struct.proto"],
   ]
   imports.each do |type_name, expected_filename|
     import_file = pool.lookup(type_name).file_descriptor

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

@@ -9,7 +9,7 @@ require 'google/api/resource_pb'
 require 'google/protobuf/timestamp_pb'
 
 
-descriptor_data = "\n-google/cloud/aiplatform/v1/feature_view.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a\x1fgoogle/protobuf/timestamp.proto\"\xf5\x07\n\x0b\x46\x65\x61tureView\x12W\n\x10\x62ig_query_source\x18\x06 \x01(\x0b\x32\x36.google.cloud.aiplatform.v1.FeatureView.BigQuerySourceB\x03\xe0\x41\x01H\x00\x12\x65\n\x17\x66\x65\x61ture_registry_source\x18\t \x01(\x0b\x32=.google.cloud.aiplatform.v1.FeatureView.FeatureRegistrySourceB\x03\xe0\x41\x01H\x00\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x08\x12\x34\n\x0b\x63reate_time\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x34\n\x0bupdate_time\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x11\n\x04\x65tag\x18\x04 \x01(\tB\x03\xe0\x41\x01\x12H\n\x06labels\x18\x05 \x03(\x0b\x32\x33.google.cloud.aiplatform.v1.FeatureView.LabelsEntryB\x03\xe0\x41\x01\x12G\n\x0bsync_config\x18\x07 \x01(\x0b\x32\x32.google.cloud.aiplatform.v1.FeatureView.SyncConfig\x1a\x42\n\x0e\x42igQuerySource\x12\x10\n\x03uri\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x1e\n\x11\x65ntity_id_columns\x18\x02 \x03(\tB\x03\xe0\x41\x02\x1a\x1a\n\nSyncConfig\x12\x0c\n\x04\x63ron\x18\x01 \x01(\t\x1a\xc9\x01\n\x15\x46\x65\x61tureRegistrySource\x12g\n\x0e\x66\x65\x61ture_groups\x18\x01 \x03(\x0b\x32J.google.cloud.aiplatform.v1.FeatureView.FeatureRegistrySource.FeatureGroupB\x03\xe0\x41\x02\x1aG\n\x0c\x46\x65\x61tureGroup\x12\x1d\n\x10\x66\x65\x61ture_group_id\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x18\n\x0b\x66\x65\x61ture_ids\x18\x02 \x03(\tB\x03\xe0\x41\x02\x1a-\n\x0bLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01:\x9b\x01\xea\x41\x97\x01\n%aiplatform.googleapis.com/FeatureView\x12nprojects/{project}/locations/{location}/featureOnlineStores/{feature_online_store}/featureViews/{feature_view}B\x08\n\x06sourceB\xce\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\x10\x46\x65\x61tureViewProtoP\x01Z>cloud.google.com/go/aiplatform/apiv1/aiplatformpb;aiplatformpb\xaa\x02\x1aGoogle.Cloud.AIPlatform.V1\xca\x02\x1aGoogle\\Cloud\\AIPlatform\\V1\xea\x02\x1dGoogle::Cloud::AIPlatform::V1b\x06proto3"
+descriptor_data = "\n-google/cloud/aiplatform/v1/feature_view.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a\x1fgoogle/protobuf/timestamp.proto\"\xaa\x08\n\x0b\x46\x65\x61tureView\x12W\n\x10\x62ig_query_source\x18\x06 \x01(\x0b\x32\x36.google.cloud.aiplatform.v1.FeatureView.BigQuerySourceB\x03\xe0\x41\x01H\x00\x12\x65\n\x17\x66\x65\x61ture_registry_source\x18\t \x01(\x0b\x32=.google.cloud.aiplatform.v1.FeatureView.FeatureRegistrySourceB\x03\xe0\x41\x01H\x00\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x08\x12\x34\n\x0b\x63reate_time\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x34\n\x0bupdate_time\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x11\n\x04\x65tag\x18\x04 \x01(\tB\x03\xe0\x41\x01\x12H\n\x06labels\x18\x05 \x03(\x0b\x32\x33.google.cloud.aiplatform.v1.FeatureView.LabelsEntryB\x03\xe0\x41\x01\x12G\n\x0bsync_config\x18\x07 \x01(\x0b\x32\x32.google.cloud.aiplatform.v1.FeatureView.SyncConfig\x1a\x42\n\x0e\x42igQuerySource\x12\x10\n\x03uri\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x1e\n\x11\x65ntity_id_columns\x18\x02 \x03(\tB\x03\xe0\x41\x02\x1a\x1a\n\nSyncConfig\x12\x0c\n\x04\x63ron\x18\x01 \x01(\t\x1a\xfe\x01\n\x15\x46\x65\x61tureRegistrySource\x12g\n\x0e\x66\x65\x61ture_groups\x18\x01 \x03(\x0b\x32J.google.cloud.aiplatform.v1.FeatureView.FeatureRegistrySource.FeatureGroupB\x03\xe0\x41\x02\x12 \n\x0eproject_number\x18\x02 \x01(\x03\x42\x03\xe0\x41\x01H\x00\x88\x01\x01\x1aG\n\x0c\x46\x65\x61tureGroup\x12\x1d\n\x10\x66\x65\x61ture_group_id\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x18\n\x0b\x66\x65\x61ture_ids\x18\x02 \x03(\tB\x03\xe0\x41\x02\x42\x11\n\x0f_project_number\x1a-\n\x0bLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01:\x9b\x01\xea\x41\x97\x01\n%aiplatform.googleapis.com/FeatureView\x12nprojects/{project}/locations/{location}/featureOnlineStores/{feature_online_store}/featureViews/{feature_view}B\x08\n\x06sourceB\xce\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\x10\x46\x65\x61tureViewProtoP\x01Z>cloud.google.com/go/aiplatform/apiv1/aiplatformpb;aiplatformpb\xaa\x02\x1aGoogle.Cloud.AIPlatform.V1\xca\x02\x1aGoogle\\Cloud\\AIPlatform\\V1\xea\x02\x1dGoogle::Cloud::AIPlatform::V1b\x06proto3"
 
 pool = Google::Protobuf::DescriptorPool.generated_pool
 

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

@@ -11,7 +11,7 @@ require 'google/rpc/status_pb'
 require 'google/type/interval_pb'
 
 
-descriptor_data = "\n2google/cloud/aiplatform/v1/feature_view_sync.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x17google/rpc/status.proto\x1a\x1agoogle/type/interval.proto\"\xfd\x02\n\x0f\x46\x65\x61tureViewSync\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x08\x12\x34\n\x0b\x63reate_time\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12,\n\x08run_time\x18\x05 \x01(\x0b\x32\x15.google.type.IntervalB\x03\xe0\x41\x03\x12-\n\x0c\x66inal_status\x18\x04 \x01(\x0b\x32\x12.google.rpc.StatusB\x03\xe0\x41\x03:\xc3\x01\xea\x41\xbf\x01\n)aiplatform.googleapis.com/FeatureViewSync\x12\x91\x01projects/{project}/locations/{location}/featureOnlineStores/{feature_online_store}/featureViews/{feature_view}/featureViewSyncs/feature_view_syncB\xd2\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\x14\x46\x65\x61tureViewSyncProtoP\x01Z>cloud.google.com/go/aiplatform/apiv1/aiplatformpb;aiplatformpb\xaa\x02\x1aGoogle.Cloud.AIPlatform.V1\xca\x02\x1aGoogle\\Cloud\\AIPlatform\\V1\xea\x02\x1dGoogle::Cloud::Aiplatform::V1b\x06proto3"
+descriptor_data = "\n2google/cloud/aiplatform/v1/feature_view_sync.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x17google/rpc/status.proto\x1a\x1agoogle/type/interval.proto\"\x92\x04\n\x0f\x46\x65\x61tureViewSync\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x08\x12\x34\n\x0b\x63reate_time\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12,\n\x08run_time\x18\x05 \x01(\x0b\x32\x15.google.type.IntervalB\x03\xe0\x41\x03\x12-\n\x0c\x66inal_status\x18\x04 \x01(\x0b\x32\x12.google.rpc.StatusB\x03\xe0\x41\x03\x12R\n\x0csync_summary\x18\x06 \x01(\x0b\x32\x37.google.cloud.aiplatform.v1.FeatureViewSync.SyncSummaryB\x03\xe0\x41\x03\x1a?\n\x0bSyncSummary\x12\x17\n\nrow_synced\x18\x01 \x01(\x03\x42\x03\xe0\x41\x03\x12\x17\n\ntotal_slot\x18\x02 \x01(\x03\x42\x03\xe0\x41\x03:\xc3\x01\xea\x41\xbf\x01\n)aiplatform.googleapis.com/FeatureViewSync\x12\x91\x01projects/{project}/locations/{location}/featureOnlineStores/{feature_online_store}/featureViews/{feature_view}/featureViewSyncs/feature_view_syncB\xd2\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\x14\x46\x65\x61tureViewSyncProtoP\x01Z>cloud.google.com/go/aiplatform/apiv1/aiplatformpb;aiplatformpb\xaa\x02\x1aGoogle.Cloud.AIPlatform.V1\xca\x02\x1aGoogle\\Cloud\\AIPlatform\\V1\xea\x02\x1dGoogle::Cloud::Aiplatform::V1b\x06proto3"
 
 pool = Google::Protobuf::DescriptorPool.generated_pool
 
@@ -45,6 +45,7 @@ module Google
     module Aiplatform
       module V1
         FeatureViewSync = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.FeatureViewSync").msgclass
+        FeatureViewSync::SyncSummary = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.FeatureViewSync.SyncSummary").msgclass
       end
     end
   end

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

@@ -12,7 +12,7 @@ require 'google/protobuf/struct_pb'
 require 'google/protobuf/timestamp_pb'
 
 
-descriptor_data = "\n&google/cloud/aiplatform/v1/index.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a\x33google/cloud/aiplatform/v1/deployed_index_ref.proto\x1a\x30google/cloud/aiplatform/v1/encryption_spec.proto\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1fgoogle/protobuf/timestamp.proto\"\xec\x06\n\x05Index\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x03\x12\x19\n\x0c\x64isplay_name\x18\x02 \x01(\tB\x03\xe0\x41\x02\x12\x13\n\x0b\x64\x65scription\x18\x03 \x01(\t\x12 \n\x13metadata_schema_uri\x18\x04 \x01(\tB\x03\xe0\x41\x05\x12(\n\x08metadata\x18\x06 \x01(\x0b\x32\x16.google.protobuf.Value\x12K\n\x10\x64\x65ployed_indexes\x18\x07 \x03(\x0b\x32,.google.cloud.aiplatform.v1.DeployedIndexRefB\x03\xe0\x41\x03\x12\x0c\n\x04\x65tag\x18\x08 \x01(\t\x12=\n\x06labels\x18\t \x03(\x0b\x32-.google.cloud.aiplatform.v1.Index.LabelsEntry\x12\x34\n\x0b\x63reate_time\x18\n \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x34\n\x0bupdate_time\x18\x0b \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12@\n\x0bindex_stats\x18\x0e \x01(\x0b\x32&.google.cloud.aiplatform.v1.IndexStatsB\x03\xe0\x41\x03\x12U\n\x13index_update_method\x18\x10 \x01(\x0e\x32\x33.google.cloud.aiplatform.v1.Index.IndexUpdateMethodB\x03\xe0\x41\x05\x12H\n\x0f\x65ncryption_spec\x18\x11 \x01(\x0b\x32*.google.cloud.aiplatform.v1.EncryptionSpecB\x03\xe0\x41\x05\x1a-\n\x0bLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"]\n\x11IndexUpdateMethod\x12#\n\x1fINDEX_UPDATE_METHOD_UNSPECIFIED\x10\x00\x12\x10\n\x0c\x42\x41TCH_UPDATE\x10\x01\x12\x11\n\rSTREAM_UPDATE\x10\x02:]\xea\x41Z\n\x1f\x61iplatform.googleapis.com/Index\x12\x37projects/{project}/locations/{location}/indexes/{index}\"\xf4\x05\n\x0eIndexDatapoint\x12\x19\n\x0c\x64\x61tapoint_id\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x1b\n\x0e\x66\x65\x61ture_vector\x18\x02 \x03(\x02\x42\x03\xe0\x41\x02\x12N\n\trestricts\x18\x04 \x03(\x0b\x32\x36.google.cloud.aiplatform.v1.IndexDatapoint.RestrictionB\x03\xe0\x41\x01\x12]\n\x11numeric_restricts\x18\x06 \x03(\x0b\x32=.google.cloud.aiplatform.v1.IndexDatapoint.NumericRestrictionB\x03\xe0\x41\x01\x12Q\n\x0c\x63rowding_tag\x18\x05 \x01(\x0b\x32\x36.google.cloud.aiplatform.v1.IndexDatapoint.CrowdingTagB\x03\xe0\x41\x01\x1aG\n\x0bRestriction\x12\x11\n\tnamespace\x18\x01 \x01(\t\x12\x12\n\nallow_list\x18\x02 \x03(\t\x12\x11\n\tdeny_list\x18\x03 \x03(\t\x1a\xb3\x02\n\x12NumericRestriction\x12\x13\n\tvalue_int\x18\x02 \x01(\x03H\x00\x12\x15\n\x0bvalue_float\x18\x03 \x01(\x02H\x00\x12\x16\n\x0cvalue_double\x18\x04 \x01(\x01H\x00\x12\x11\n\tnamespace\x18\x01 \x01(\t\x12R\n\x02op\x18\x05 \x01(\x0e\x32\x46.google.cloud.aiplatform.v1.IndexDatapoint.NumericRestriction.Operator\"i\n\x08Operator\x12\x18\n\x14OPERATOR_UNSPECIFIED\x10\x00\x12\x08\n\x04LESS\x10\x01\x12\x0e\n\nLESS_EQUAL\x10\x02\x12\t\n\x05\x45QUAL\x10\x03\x12\x11\n\rGREATER_EQUAL\x10\x04\x12\x0b\n\x07GREATER\x10\x05\x42\x07\n\x05Value\x1a)\n\x0b\x43rowdingTag\x12\x1a\n\x12\x63rowding_attribute\x18\x01 \x01(\t\"C\n\nIndexStats\x12\x1a\n\rvectors_count\x18\x01 \x01(\x03\x42\x03\xe0\x41\x03\x12\x19\n\x0cshards_count\x18\x02 \x01(\x05\x42\x03\xe0\x41\x03\x42\xc8\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\nIndexProtoP\x01Z>cloud.google.com/go/aiplatform/apiv1/aiplatformpb;aiplatformpb\xaa\x02\x1aGoogle.Cloud.AIPlatform.V1\xca\x02\x1aGoogle\\Cloud\\AIPlatform\\V1\xea\x02\x1dGoogle::Cloud::AIPlatform::V1b\x06proto3"
+descriptor_data = "\n&google/cloud/aiplatform/v1/index.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a\x33google/cloud/aiplatform/v1/deployed_index_ref.proto\x1a\x30google/cloud/aiplatform/v1/encryption_spec.proto\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1fgoogle/protobuf/timestamp.proto\"\xec\x06\n\x05Index\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x03\x12\x19\n\x0c\x64isplay_name\x18\x02 \x01(\tB\x03\xe0\x41\x02\x12\x13\n\x0b\x64\x65scription\x18\x03 \x01(\t\x12 \n\x13metadata_schema_uri\x18\x04 \x01(\tB\x03\xe0\x41\x05\x12(\n\x08metadata\x18\x06 \x01(\x0b\x32\x16.google.protobuf.Value\x12K\n\x10\x64\x65ployed_indexes\x18\x07 \x03(\x0b\x32,.google.cloud.aiplatform.v1.DeployedIndexRefB\x03\xe0\x41\x03\x12\x0c\n\x04\x65tag\x18\x08 \x01(\t\x12=\n\x06labels\x18\t \x03(\x0b\x32-.google.cloud.aiplatform.v1.Index.LabelsEntry\x12\x34\n\x0b\x63reate_time\x18\n \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x34\n\x0bupdate_time\x18\x0b \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12@\n\x0bindex_stats\x18\x0e \x01(\x0b\x32&.google.cloud.aiplatform.v1.IndexStatsB\x03\xe0\x41\x03\x12U\n\x13index_update_method\x18\x10 \x01(\x0e\x32\x33.google.cloud.aiplatform.v1.Index.IndexUpdateMethodB\x03\xe0\x41\x05\x12H\n\x0f\x65ncryption_spec\x18\x11 \x01(\x0b\x32*.google.cloud.aiplatform.v1.EncryptionSpecB\x03\xe0\x41\x05\x1a-\n\x0bLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"]\n\x11IndexUpdateMethod\x12#\n\x1fINDEX_UPDATE_METHOD_UNSPECIFIED\x10\x00\x12\x10\n\x0c\x42\x41TCH_UPDATE\x10\x01\x12\x11\n\rSTREAM_UPDATE\x10\x02:]\xea\x41Z\n\x1f\x61iplatform.googleapis.com/Index\x12\x37projects/{project}/locations/{location}/indexes/{index}\"\x83\x06\n\x0eIndexDatapoint\x12\x19\n\x0c\x64\x61tapoint_id\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x1b\n\x0e\x66\x65\x61ture_vector\x18\x02 \x03(\x02\x42\x03\xe0\x41\x02\x12N\n\trestricts\x18\x04 \x03(\x0b\x32\x36.google.cloud.aiplatform.v1.IndexDatapoint.RestrictionB\x03\xe0\x41\x01\x12]\n\x11numeric_restricts\x18\x06 \x03(\x0b\x32=.google.cloud.aiplatform.v1.IndexDatapoint.NumericRestrictionB\x03\xe0\x41\x01\x12Q\n\x0c\x63rowding_tag\x18\x05 \x01(\x0b\x32\x36.google.cloud.aiplatform.v1.IndexDatapoint.CrowdingTagB\x03\xe0\x41\x01\x1aG\n\x0bRestriction\x12\x11\n\tnamespace\x18\x01 \x01(\t\x12\x12\n\nallow_list\x18\x02 \x03(\t\x12\x11\n\tdeny_list\x18\x03 \x03(\t\x1a\xc2\x02\n\x12NumericRestriction\x12\x13\n\tvalue_int\x18\x02 \x01(\x03H\x00\x12\x15\n\x0bvalue_float\x18\x03 \x01(\x02H\x00\x12\x16\n\x0cvalue_double\x18\x04 \x01(\x01H\x00\x12\x11\n\tnamespace\x18\x01 \x01(\t\x12R\n\x02op\x18\x05 \x01(\x0e\x32\x46.google.cloud.aiplatform.v1.IndexDatapoint.NumericRestriction.Operator\"x\n\x08Operator\x12\x18\n\x14OPERATOR_UNSPECIFIED\x10\x00\x12\x08\n\x04LESS\x10\x01\x12\x0e\n\nLESS_EQUAL\x10\x02\x12\t\n\x05\x45QUAL\x10\x03\x12\x11\n\rGREATER_EQUAL\x10\x04\x12\x0b\n\x07GREATER\x10\x05\x12\r\n\tNOT_EQUAL\x10\x06\x42\x07\n\x05Value\x1a)\n\x0b\x43rowdingTag\x12\x1a\n\x12\x63rowding_attribute\x18\x01 \x01(\t\"C\n\nIndexStats\x12\x1a\n\rvectors_count\x18\x01 \x01(\x03\x42\x03\xe0\x41\x03\x12\x19\n\x0cshards_count\x18\x02 \x01(\x05\x42\x03\xe0\x41\x03\x42\xc8\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\nIndexProtoP\x01Z>cloud.google.com/go/aiplatform/apiv1/aiplatformpb;aiplatformpb\xaa\x02\x1aGoogle.Cloud.AIPlatform.V1\xca\x02\x1aGoogle\\Cloud\\AIPlatform\\V1\xea\x02\x1dGoogle::Cloud::AIPlatform::V1b\x06proto3"
 
 pool = Google::Protobuf::DescriptorPool.generated_pool
 

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

@@ -15,7 +15,7 @@ require 'google/protobuf/struct_pb'
 require 'google/protobuf/timestamp_pb'
 
 
-descriptor_data = "\n&google/cloud/aiplatform/v1/model.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a\x33google/cloud/aiplatform/v1/deployed_model_ref.proto\x1a\x30google/cloud/aiplatform/v1/encryption_spec.proto\x1a(google/cloud/aiplatform/v1/env_var.proto\x1a,google/cloud/aiplatform/v1/explanation.proto\x1a\x1egoogle/protobuf/duration.proto\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1fgoogle/protobuf/timestamp.proto\"\xff\x12\n\x05Model\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x1a\n\nversion_id\x18\x1c \x01(\tB\x06\xe0\x41\x05\xe0\x41\x03\x12\x17\n\x0fversion_aliases\x18\x1d \x03(\t\x12<\n\x13version_create_time\x18\x1f \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12<\n\x13version_update_time\x18  \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x19\n\x0c\x64isplay_name\x18\x02 \x01(\tB\x03\xe0\x41\x02\x12\x13\n\x0b\x64\x65scription\x18\x03 \x01(\t\x12\x1b\n\x13version_description\x18\x1e \x01(\t\x12\x45\n\x10predict_schemata\x18\x04 \x01(\x0b\x32+.google.cloud.aiplatform.v1.PredictSchemata\x12 \n\x13metadata_schema_uri\x18\x05 \x01(\tB\x03\xe0\x41\x05\x12-\n\x08metadata\x18\x06 \x01(\x0b\x32\x16.google.protobuf.ValueB\x03\xe0\x41\x05\x12U\n\x18supported_export_formats\x18\x14 \x03(\x0b\x32..google.cloud.aiplatform.v1.Model.ExportFormatB\x03\xe0\x41\x03\x12M\n\x11training_pipeline\x18\x07 \x01(\tB2\xe0\x41\x03\xfa\x41,\n*aiplatform.googleapis.com/TrainingPipeline\x12\x43\n\x0cpipeline_job\x18/ \x01(\tB-\xe0\x41\x01\xfa\x41\'\n%aiplatform.googleapis.com/PipelineJob\x12K\n\x0e\x63ontainer_spec\x18\t \x01(\x0b\x32..google.cloud.aiplatform.v1.ModelContainerSpecB\x03\xe0\x41\x04\x12\x19\n\x0c\x61rtifact_uri\x18\x1a \x01(\tB\x03\xe0\x41\x05\x12l\n$supported_deployment_resources_types\x18\n \x03(\x0e\x32\x39.google.cloud.aiplatform.v1.Model.DeploymentResourcesTypeB\x03\xe0\x41\x03\x12,\n\x1fsupported_input_storage_formats\x18\x0b \x03(\tB\x03\xe0\x41\x03\x12-\n supported_output_storage_formats\x18\x0c \x03(\tB\x03\xe0\x41\x03\x12\x34\n\x0b\x63reate_time\x18\r \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x34\n\x0bupdate_time\x18\x0e \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12J\n\x0f\x64\x65ployed_models\x18\x0f \x03(\x0b\x32,.google.cloud.aiplatform.v1.DeployedModelRefB\x03\xe0\x41\x03\x12\x45\n\x10\x65xplanation_spec\x18\x17 \x01(\x0b\x32+.google.cloud.aiplatform.v1.ExplanationSpec\x12\x0c\n\x04\x65tag\x18\x10 \x01(\t\x12=\n\x06labels\x18\x11 \x03(\x0b\x32-.google.cloud.aiplatform.v1.Model.LabelsEntry\x12?\n\ndata_stats\x18\x15 \x01(\x0b\x32+.google.cloud.aiplatform.v1.Model.DataStats\x12\x43\n\x0f\x65ncryption_spec\x18\x18 \x01(\x0b\x32*.google.cloud.aiplatform.v1.EncryptionSpec\x12K\n\x11model_source_info\x18& \x01(\x0b\x32+.google.cloud.aiplatform.v1.ModelSourceInfoB\x03\xe0\x41\x03\x12U\n\x13original_model_info\x18\" \x01(\x0b\x32\x33.google.cloud.aiplatform.v1.Model.OriginalModelInfoB\x03\xe0\x41\x03\x12\x1e\n\x11metadata_artifact\x18, \x01(\tB\x03\xe0\x41\x03\x1a\xd5\x01\n\x0c\x45xportFormat\x12\x0f\n\x02id\x18\x01 \x01(\tB\x03\xe0\x41\x03\x12\x62\n\x13\x65xportable_contents\x18\x02 \x03(\x0e\x32@.google.cloud.aiplatform.v1.Model.ExportFormat.ExportableContentB\x03\xe0\x41\x03\"P\n\x11\x45xportableContent\x12\"\n\x1e\x45XPORTABLE_CONTENT_UNSPECIFIED\x10\x00\x12\x0c\n\x08\x41RTIFACT\x10\x01\x12\t\n\x05IMAGE\x10\x02\x1a\xdc\x01\n\tDataStats\x12!\n\x19training_data_items_count\x18\x01 \x01(\x03\x12#\n\x1bvalidation_data_items_count\x18\x02 \x01(\x03\x12\x1d\n\x15test_data_items_count\x18\x03 \x01(\x03\x12\"\n\x1atraining_annotations_count\x18\x04 \x01(\x03\x12$\n\x1cvalidation_annotations_count\x18\x05 \x01(\x03\x12\x1e\n\x16test_annotations_count\x18\x06 \x01(\x03\x1aK\n\x11OriginalModelInfo\x12\x36\n\x05model\x18\x01 \x01(\tB\'\xe0\x41\x03\xfa\x41!\n\x1f\x61iplatform.googleapis.com/Model\x1a-\n\x0bLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\x8c\x01\n\x17\x44\x65ploymentResourcesType\x12)\n%DEPLOYMENT_RESOURCES_TYPE_UNSPECIFIED\x10\x00\x12\x17\n\x13\x44\x45\x44ICATED_RESOURCES\x10\x01\x12\x17\n\x13\x41UTOMATIC_RESOURCES\x10\x02\x12\x14\n\x10SHARED_RESOURCES\x10\x03:\\\xea\x41Y\n\x1f\x61iplatform.googleapis.com/Model\x12\x36projects/{project}/locations/{location}/models/{model}\"(\n\x13LargeModelReference\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x02\"{\n\x0fPredictSchemata\x12 \n\x13instance_schema_uri\x18\x01 \x01(\tB\x03\xe0\x41\x05\x12\"\n\x15parameters_schema_uri\x18\x02 \x01(\tB\x03\xe0\x41\x05\x12\"\n\x15prediction_schema_uri\x18\x03 \x01(\tB\x03\xe0\x41\x05\"\x93\x04\n\x12ModelContainerSpec\x12\x19\n\timage_uri\x18\x01 \x01(\tB\x06\xe0\x41\x02\xe0\x41\x05\x12\x14\n\x07\x63ommand\x18\x02 \x03(\tB\x03\xe0\x41\x05\x12\x11\n\x04\x61rgs\x18\x03 \x03(\tB\x03\xe0\x41\x05\x12\x34\n\x03\x65nv\x18\x04 \x03(\x0b\x32\".google.cloud.aiplatform.v1.EnvVarB\x03\xe0\x41\x05\x12\x34\n\x05ports\x18\x05 \x03(\x0b\x32 .google.cloud.aiplatform.v1.PortB\x03\xe0\x41\x05\x12\x1a\n\rpredict_route\x18\x06 \x01(\tB\x03\xe0\x41\x05\x12\x19\n\x0chealth_route\x18\x07 \x01(\tB\x03\xe0\x41\x05\x12\x39\n\ngrpc_ports\x18\t \x03(\x0b\x32 .google.cloud.aiplatform.v1.PortB\x03\xe0\x41\x05\x12:\n\x12\x64\x65ployment_timeout\x18\n \x01(\x0b\x32\x19.google.protobuf.DurationB\x03\xe0\x41\x05\x12\"\n\x15shared_memory_size_mb\x18\x0b \x01(\x03\x42\x03\xe0\x41\x05\x12=\n\rstartup_probe\x18\x0c \x01(\x0b\x32!.google.cloud.aiplatform.v1.ProbeB\x03\xe0\x41\x05\x12<\n\x0chealth_probe\x18\r \x01(\x0b\x32!.google.cloud.aiplatform.v1.ProbeB\x03\xe0\x41\x05\"\x1e\n\x04Port\x12\x16\n\x0e\x63ontainer_port\x18\x03 \x01(\x05\"\x93\x02\n\x0fModelSourceInfo\x12P\n\x0bsource_type\x18\x01 \x01(\x0e\x32;.google.cloud.aiplatform.v1.ModelSourceInfo.ModelSourceType\x12\x0c\n\x04\x63opy\x18\x02 \x01(\x08\"\x9f\x01\n\x0fModelSourceType\x12!\n\x1dMODEL_SOURCE_TYPE_UNSPECIFIED\x10\x00\x12\n\n\x06\x41UTOML\x10\x01\x12\n\n\x06\x43USTOM\x10\x02\x12\x08\n\x04\x42QML\x10\x03\x12\x10\n\x0cMODEL_GARDEN\x10\x04\x12\t\n\x05GENIE\x10\x05\x12\x19\n\x15\x43USTOM_TEXT_EMBEDDING\x10\x06\x12\x0f\n\x0bMARKETPLACE\x10\x07\"\xa3\x01\n\x05Probe\x12<\n\x04\x65xec\x18\x01 \x01(\x0b\x32,.google.cloud.aiplatform.v1.Probe.ExecActionH\x00\x12\x16\n\x0eperiod_seconds\x18\x02 \x01(\x05\x12\x17\n\x0ftimeout_seconds\x18\x03 \x01(\x05\x1a\x1d\n\nExecAction\x12\x0f\n\x07\x63ommand\x18\x01 \x03(\tB\x0c\n\nprobe_typeB\xc8\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\nModelProtoP\x01Z>cloud.google.com/go/aiplatform/apiv1/aiplatformpb;aiplatformpb\xaa\x02\x1aGoogle.Cloud.AIPlatform.V1\xca\x02\x1aGoogle\\Cloud\\AIPlatform\\V1\xea\x02\x1dGoogle::Cloud::AIPlatform::V1b\x06proto3"
+descriptor_data = "\n&google/cloud/aiplatform/v1/model.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a\x33google/cloud/aiplatform/v1/deployed_model_ref.proto\x1a\x30google/cloud/aiplatform/v1/encryption_spec.proto\x1a(google/cloud/aiplatform/v1/env_var.proto\x1a,google/cloud/aiplatform/v1/explanation.proto\x1a\x1egoogle/protobuf/duration.proto\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1fgoogle/protobuf/timestamp.proto\"\xff\x14\n\x05Model\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x1a\n\nversion_id\x18\x1c \x01(\tB\x06\xe0\x41\x05\xe0\x41\x03\x12\x17\n\x0fversion_aliases\x18\x1d \x03(\t\x12<\n\x13version_create_time\x18\x1f \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12<\n\x13version_update_time\x18  \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x19\n\x0c\x64isplay_name\x18\x02 \x01(\tB\x03\xe0\x41\x02\x12\x13\n\x0b\x64\x65scription\x18\x03 \x01(\t\x12\x1b\n\x13version_description\x18\x1e \x01(\t\x12\x45\n\x10predict_schemata\x18\x04 \x01(\x0b\x32+.google.cloud.aiplatform.v1.PredictSchemata\x12 \n\x13metadata_schema_uri\x18\x05 \x01(\tB\x03\xe0\x41\x05\x12-\n\x08metadata\x18\x06 \x01(\x0b\x32\x16.google.protobuf.ValueB\x03\xe0\x41\x05\x12U\n\x18supported_export_formats\x18\x14 \x03(\x0b\x32..google.cloud.aiplatform.v1.Model.ExportFormatB\x03\xe0\x41\x03\x12M\n\x11training_pipeline\x18\x07 \x01(\tB2\xe0\x41\x03\xfa\x41,\n*aiplatform.googleapis.com/TrainingPipeline\x12\x43\n\x0cpipeline_job\x18/ \x01(\tB-\xe0\x41\x01\xfa\x41\'\n%aiplatform.googleapis.com/PipelineJob\x12K\n\x0e\x63ontainer_spec\x18\t \x01(\x0b\x32..google.cloud.aiplatform.v1.ModelContainerSpecB\x03\xe0\x41\x04\x12\x19\n\x0c\x61rtifact_uri\x18\x1a \x01(\tB\x03\xe0\x41\x05\x12l\n$supported_deployment_resources_types\x18\n \x03(\x0e\x32\x39.google.cloud.aiplatform.v1.Model.DeploymentResourcesTypeB\x03\xe0\x41\x03\x12,\n\x1fsupported_input_storage_formats\x18\x0b \x03(\tB\x03\xe0\x41\x03\x12-\n supported_output_storage_formats\x18\x0c \x03(\tB\x03\xe0\x41\x03\x12\x34\n\x0b\x63reate_time\x18\r \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x34\n\x0bupdate_time\x18\x0e \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12J\n\x0f\x64\x65ployed_models\x18\x0f \x03(\x0b\x32,.google.cloud.aiplatform.v1.DeployedModelRefB\x03\xe0\x41\x03\x12\x45\n\x10\x65xplanation_spec\x18\x17 \x01(\x0b\x32+.google.cloud.aiplatform.v1.ExplanationSpec\x12\x0c\n\x04\x65tag\x18\x10 \x01(\t\x12=\n\x06labels\x18\x11 \x03(\x0b\x32-.google.cloud.aiplatform.v1.Model.LabelsEntry\x12?\n\ndata_stats\x18\x15 \x01(\x0b\x32+.google.cloud.aiplatform.v1.Model.DataStats\x12\x43\n\x0f\x65ncryption_spec\x18\x18 \x01(\x0b\x32*.google.cloud.aiplatform.v1.EncryptionSpec\x12K\n\x11model_source_info\x18& \x01(\x0b\x32+.google.cloud.aiplatform.v1.ModelSourceInfoB\x03\xe0\x41\x03\x12U\n\x13original_model_info\x18\" \x01(\x0b\x32\x33.google.cloud.aiplatform.v1.Model.OriginalModelInfoB\x03\xe0\x41\x03\x12\x1e\n\x11metadata_artifact\x18, \x01(\tB\x03\xe0\x41\x03\x12Q\n\x11\x62\x61se_model_source\x18\x32 \x01(\x0b\x32\x31.google.cloud.aiplatform.v1.Model.BaseModelSourceB\x03\xe0\x41\x01\x1a\xd5\x01\n\x0c\x45xportFormat\x12\x0f\n\x02id\x18\x01 \x01(\tB\x03\xe0\x41\x03\x12\x62\n\x13\x65xportable_contents\x18\x02 \x03(\x0e\x32@.google.cloud.aiplatform.v1.Model.ExportFormat.ExportableContentB\x03\xe0\x41\x03\"P\n\x11\x45xportableContent\x12\"\n\x1e\x45XPORTABLE_CONTENT_UNSPECIFIED\x10\x00\x12\x0c\n\x08\x41RTIFACT\x10\x01\x12\t\n\x05IMAGE\x10\x02\x1a\xdc\x01\n\tDataStats\x12!\n\x19training_data_items_count\x18\x01 \x01(\x03\x12#\n\x1bvalidation_data_items_count\x18\x02 \x01(\x03\x12\x1d\n\x15test_data_items_count\x18\x03 \x01(\x03\x12\"\n\x1atraining_annotations_count\x18\x04 \x01(\x03\x12$\n\x1cvalidation_annotations_count\x18\x05 \x01(\x03\x12\x1e\n\x16test_annotations_count\x18\x06 \x01(\x03\x1aK\n\x11OriginalModelInfo\x12\x36\n\x05model\x18\x01 \x01(\tB\'\xe0\x41\x03\xfa\x41!\n\x1f\x61iplatform.googleapis.com/Model\x1a\xaa\x01\n\x0f\x42\x61seModelSource\x12L\n\x13model_garden_source\x18\x01 \x01(\x0b\x32-.google.cloud.aiplatform.v1.ModelGardenSourceH\x00\x12?\n\x0cgenie_source\x18\x02 \x01(\x0b\x32\'.google.cloud.aiplatform.v1.GenieSourceH\x00\x42\x08\n\x06source\x1a-\n\x0bLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\x8c\x01\n\x17\x44\x65ploymentResourcesType\x12)\n%DEPLOYMENT_RESOURCES_TYPE_UNSPECIFIED\x10\x00\x12\x17\n\x13\x44\x45\x44ICATED_RESOURCES\x10\x01\x12\x17\n\x13\x41UTOMATIC_RESOURCES\x10\x02\x12\x14\n\x10SHARED_RESOURCES\x10\x03:\\\xea\x41Y\n\x1f\x61iplatform.googleapis.com/Model\x12\x36projects/{project}/locations/{location}/models/{model}\"(\n\x13LargeModelReference\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x02\"3\n\x11ModelGardenSource\x12\x1e\n\x11public_model_name\x18\x01 \x01(\tB\x03\xe0\x41\x02\"*\n\x0bGenieSource\x12\x1b\n\x0e\x62\x61se_model_uri\x18\x01 \x01(\tB\x03\xe0\x41\x02\"{\n\x0fPredictSchemata\x12 \n\x13instance_schema_uri\x18\x01 \x01(\tB\x03\xe0\x41\x05\x12\"\n\x15parameters_schema_uri\x18\x02 \x01(\tB\x03\xe0\x41\x05\x12\"\n\x15prediction_schema_uri\x18\x03 \x01(\tB\x03\xe0\x41\x05\"\x93\x04\n\x12ModelContainerSpec\x12\x19\n\timage_uri\x18\x01 \x01(\tB\x06\xe0\x41\x02\xe0\x41\x05\x12\x14\n\x07\x63ommand\x18\x02 \x03(\tB\x03\xe0\x41\x05\x12\x11\n\x04\x61rgs\x18\x03 \x03(\tB\x03\xe0\x41\x05\x12\x34\n\x03\x65nv\x18\x04 \x03(\x0b\x32\".google.cloud.aiplatform.v1.EnvVarB\x03\xe0\x41\x05\x12\x34\n\x05ports\x18\x05 \x03(\x0b\x32 .google.cloud.aiplatform.v1.PortB\x03\xe0\x41\x05\x12\x1a\n\rpredict_route\x18\x06 \x01(\tB\x03\xe0\x41\x05\x12\x19\n\x0chealth_route\x18\x07 \x01(\tB\x03\xe0\x41\x05\x12\x39\n\ngrpc_ports\x18\t \x03(\x0b\x32 .google.cloud.aiplatform.v1.PortB\x03\xe0\x41\x05\x12:\n\x12\x64\x65ployment_timeout\x18\n \x01(\x0b\x32\x19.google.protobuf.DurationB\x03\xe0\x41\x05\x12\"\n\x15shared_memory_size_mb\x18\x0b \x01(\x03\x42\x03\xe0\x41\x05\x12=\n\rstartup_probe\x18\x0c \x01(\x0b\x32!.google.cloud.aiplatform.v1.ProbeB\x03\xe0\x41\x05\x12<\n\x0chealth_probe\x18\r \x01(\x0b\x32!.google.cloud.aiplatform.v1.ProbeB\x03\xe0\x41\x05\"\x1e\n\x04Port\x12\x16\n\x0e\x63ontainer_port\x18\x03 \x01(\x05\"\x93\x02\n\x0fModelSourceInfo\x12P\n\x0bsource_type\x18\x01 \x01(\x0e\x32;.google.cloud.aiplatform.v1.ModelSourceInfo.ModelSourceType\x12\x0c\n\x04\x63opy\x18\x02 \x01(\x08\"\x9f\x01\n\x0fModelSourceType\x12!\n\x1dMODEL_SOURCE_TYPE_UNSPECIFIED\x10\x00\x12\n\n\x06\x41UTOML\x10\x01\x12\n\n\x06\x43USTOM\x10\x02\x12\x08\n\x04\x42QML\x10\x03\x12\x10\n\x0cMODEL_GARDEN\x10\x04\x12\t\n\x05GENIE\x10\x05\x12\x19\n\x15\x43USTOM_TEXT_EMBEDDING\x10\x06\x12\x0f\n\x0bMARKETPLACE\x10\x07\"\xa3\x01\n\x05Probe\x12<\n\x04\x65xec\x18\x01 \x01(\x0b\x32,.google.cloud.aiplatform.v1.Probe.ExecActionH\x00\x12\x16\n\x0eperiod_seconds\x18\x02 \x01(\x05\x12\x17\n\x0ftimeout_seconds\x18\x03 \x01(\x05\x1a\x1d\n\nExecAction\x12\x0f\n\x07\x63ommand\x18\x01 \x03(\tB\x0c\n\nprobe_typeB\xc8\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\nModelProtoP\x01Z>cloud.google.com/go/aiplatform/apiv1/aiplatformpb;aiplatformpb\xaa\x02\x1aGoogle.Cloud.AIPlatform.V1\xca\x02\x1aGoogle\\Cloud\\AIPlatform\\V1\xea\x02\x1dGoogle::Cloud::AIPlatform::V1b\x06proto3"
 
 pool = Google::Protobuf::DescriptorPool.generated_pool
 
@@ -57,8 +57,11 @@ module Google
         Model::ExportFormat::ExportableContent = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.Model.ExportFormat.ExportableContent").enummodule
         Model::DataStats = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.Model.DataStats").msgclass
         Model::OriginalModelInfo = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.Model.OriginalModelInfo").msgclass
+        Model::BaseModelSource = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.Model.BaseModelSource").msgclass
         Model::DeploymentResourcesType = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.Model.DeploymentResourcesType").enummodule
         LargeModelReference = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.LargeModelReference").msgclass
+        ModelGardenSource = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ModelGardenSource").msgclass
+        GenieSource = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.GenieSource").msgclass
         PredictSchemata = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.PredictSchemata").msgclass
         ModelContainerSpec = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ModelContainerSpec").msgclass
         Port = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.Port").msgclass

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

@@ -16,7 +16,7 @@ require 'google/protobuf/empty_pb'
 require 'google/protobuf/field_mask_pb'
 
 
-descriptor_data = "\n1google/cloud/aiplatform/v1/pipeline_service.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1cgoogle/api/annotations.proto\x1a\x17google/api/client.proto\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a*google/cloud/aiplatform/v1/operation.proto\x1a-google/cloud/aiplatform/v1/pipeline_job.proto\x1a\x32google/cloud/aiplatform/v1/training_pipeline.proto\x1a#google/longrunning/operations.proto\x1a\x1bgoogle/protobuf/empty.proto\x1a google/protobuf/field_mask.proto\"\xa8\x01\n\x1d\x43reateTrainingPipelineRequest\x12\x39\n\x06parent\x18\x01 \x01(\tB)\xe0\x41\x02\xfa\x41#\n!locations.googleapis.com/Location\x12L\n\x11training_pipeline\x18\x02 \x01(\x0b\x32,.google.cloud.aiplatform.v1.TrainingPipelineB\x03\xe0\x41\x02\"^\n\x1aGetTrainingPipelineRequest\x12@\n\x04name\x18\x01 \x01(\tB2\xe0\x41\x02\xfa\x41,\n*aiplatform.googleapis.com/TrainingPipeline\"\xbf\x01\n\x1cListTrainingPipelinesRequest\x12\x39\n\x06parent\x18\x01 \x01(\tB)\xe0\x41\x02\xfa\x41#\n!locations.googleapis.com/Location\x12\x0e\n\x06\x66ilter\x18\x02 \x01(\t\x12\x11\n\tpage_size\x18\x03 \x01(\x05\x12\x12\n\npage_token\x18\x04 \x01(\t\x12-\n\tread_mask\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.FieldMask\"\x82\x01\n\x1dListTrainingPipelinesResponse\x12H\n\x12training_pipelines\x18\x01 \x03(\x0b\x32,.google.cloud.aiplatform.v1.TrainingPipeline\x12\x17\n\x0fnext_page_token\x18\x02 \x01(\t\"a\n\x1d\x44\x65leteTrainingPipelineRequest\x12@\n\x04name\x18\x01 \x01(\tB2\xe0\x41\x02\xfa\x41,\n*aiplatform.googleapis.com/TrainingPipeline\"a\n\x1d\x43\x61ncelTrainingPipelineRequest\x12@\n\x04name\x18\x01 \x01(\tB2\xe0\x41\x02\xfa\x41,\n*aiplatform.googleapis.com/TrainingPipeline\"\xb2\x01\n\x18\x43reatePipelineJobRequest\x12\x39\n\x06parent\x18\x01 \x01(\tB)\xe0\x41\x02\xfa\x41#\n!locations.googleapis.com/Location\x12\x42\n\x0cpipeline_job\x18\x02 \x01(\x0b\x32\'.google.cloud.aiplatform.v1.PipelineJobB\x03\xe0\x41\x02\x12\x17\n\x0fpipeline_job_id\x18\x03 \x01(\t\"T\n\x15GetPipelineJobRequest\x12;\n\x04name\x18\x01 \x01(\tB-\xe0\x41\x02\xfa\x41\'\n%aiplatform.googleapis.com/PipelineJob\"\xcc\x01\n\x17ListPipelineJobsRequest\x12\x39\n\x06parent\x18\x01 \x01(\tB)\xe0\x41\x02\xfa\x41#\n!locations.googleapis.com/Location\x12\x0e\n\x06\x66ilter\x18\x02 \x01(\t\x12\x11\n\tpage_size\x18\x03 \x01(\x05\x12\x12\n\npage_token\x18\x04 \x01(\t\x12\x10\n\x08order_by\x18\x06 \x01(\t\x12-\n\tread_mask\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.FieldMask\"s\n\x18ListPipelineJobsResponse\x12>\n\rpipeline_jobs\x18\x01 \x03(\x0b\x32\'.google.cloud.aiplatform.v1.PipelineJob\x12\x17\n\x0fnext_page_token\x18\x02 \x01(\t\"W\n\x18\x44\x65letePipelineJobRequest\x12;\n\x04name\x18\x01 \x01(\tB-\xe0\x41\x02\xfa\x41\'\n%aiplatform.googleapis.com/PipelineJob\"W\n\x18\x43\x61ncelPipelineJobRequest\x12;\n\x04name\x18\x01 \x01(\tB-\xe0\x41\x02\xfa\x41\'\n%aiplatform.googleapis.com/PipelineJob2\xfc\x10\n\x0fPipelineService\x12\xee\x01\n\x16\x43reateTrainingPipeline\x12\x39.google.cloud.aiplatform.v1.CreateTrainingPipelineRequest\x1a,.google.cloud.aiplatform.v1.TrainingPipeline\"k\xda\x41\x18parent,training_pipeline\x82\xd3\xe4\x93\x02J\"5/v1/{parent=projects/*/locations/*}/trainingPipelines:\x11training_pipeline\x12\xc1\x01\n\x13GetTrainingPipeline\x12\x36.google.cloud.aiplatform.v1.GetTrainingPipelineRequest\x1a,.google.cloud.aiplatform.v1.TrainingPipeline\"D\xda\x41\x04name\x82\xd3\xe4\x93\x02\x37\x12\x35/v1/{name=projects/*/locations/*/trainingPipelines/*}\x12\xd4\x01\n\x15ListTrainingPipelines\x12\x38.google.cloud.aiplatform.v1.ListTrainingPipelinesRequest\x1a\x39.google.cloud.aiplatform.v1.ListTrainingPipelinesResponse\"F\xda\x41\x06parent\x82\xd3\xe4\x93\x02\x37\x12\x35/v1/{parent=projects/*/locations/*}/trainingPipelines\x12\xeb\x01\n\x16\x44\x65leteTrainingPipeline\x12\x39.google.cloud.aiplatform.v1.DeleteTrainingPipelineRequest\x1a\x1d.google.longrunning.Operation\"w\xca\x41\x30\n\x15google.protobuf.Empty\x12\x17\x44\x65leteOperationMetadata\xda\x41\x04name\x82\xd3\xe4\x93\x02\x37*5/v1/{name=projects/*/locations/*/trainingPipelines/*}\x12\xbb\x01\n\x16\x43\x61ncelTrainingPipeline\x12\x39.google.cloud.aiplatform.v1.CancelTrainingPipelineRequest\x1a\x16.google.protobuf.Empty\"N\xda\x41\x04name\x82\xd3\xe4\x93\x02\x41\"</v1/{name=projects/*/locations/*/trainingPipelines/*}:cancel:\x01*\x12\xe0\x01\n\x11\x43reatePipelineJob\x12\x34.google.cloud.aiplatform.v1.CreatePipelineJobRequest\x1a\'.google.cloud.aiplatform.v1.PipelineJob\"l\xda\x41#parent,pipeline_job,pipeline_job_id\x82\xd3\xe4\x93\x02@\"0/v1/{parent=projects/*/locations/*}/pipelineJobs:\x0cpipeline_job\x12\xad\x01\n\x0eGetPipelineJob\x12\x31.google.cloud.aiplatform.v1.GetPipelineJobRequest\x1a\'.google.cloud.aiplatform.v1.PipelineJob\"?\xda\x41\x04name\x82\xd3\xe4\x93\x02\x32\x12\x30/v1/{name=projects/*/locations/*/pipelineJobs/*}\x12\xc0\x01\n\x10ListPipelineJobs\x12\x33.google.cloud.aiplatform.v1.ListPipelineJobsRequest\x1a\x34.google.cloud.aiplatform.v1.ListPipelineJobsResponse\"A\xda\x41\x06parent\x82\xd3\xe4\x93\x02\x32\x12\x30/v1/{parent=projects/*/locations/*}/pipelineJobs\x12\xdc\x01\n\x11\x44\x65letePipelineJob\x12\x34.google.cloud.aiplatform.v1.DeletePipelineJobRequest\x1a\x1d.google.longrunning.Operation\"r\xca\x41\x30\n\x15google.protobuf.Empty\x12\x17\x44\x65leteOperationMetadata\xda\x41\x04name\x82\xd3\xe4\x93\x02\x32*0/v1/{name=projects/*/locations/*/pipelineJobs/*}\x12\xac\x01\n\x11\x43\x61ncelPipelineJob\x12\x34.google.cloud.aiplatform.v1.CancelPipelineJobRequest\x1a\x16.google.protobuf.Empty\"I\xda\x41\x04name\x82\xd3\xe4\x93\x02<\"7/v1/{name=projects/*/locations/*/pipelineJobs/*}:cancel:\x01*\x1aM\xca\x41\x19\x61iplatform.googleapis.com\xd2\x41.https://www.googleapis.com/auth/cloud-platformB\xd2\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\x14PipelineServiceProtoP\x01Z>cloud.google.com/go/aiplatform/apiv1/aiplatformpb;aiplatformpb\xaa\x02\x1aGoogle.Cloud.AIPlatform.V1\xca\x02\x1aGoogle\\Cloud\\AIPlatform\\V1\xea\x02\x1dGoogle::Cloud::AIPlatform::V1b\x06proto3"
+descriptor_data = "\n1google/cloud/aiplatform/v1/pipeline_service.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1cgoogle/api/annotations.proto\x1a\x17google/api/client.proto\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a*google/cloud/aiplatform/v1/operation.proto\x1a-google/cloud/aiplatform/v1/pipeline_job.proto\x1a\x32google/cloud/aiplatform/v1/training_pipeline.proto\x1a#google/longrunning/operations.proto\x1a\x1bgoogle/protobuf/empty.proto\x1a google/protobuf/field_mask.proto\"z\n(BatchCancelPipelineJobsOperationMetadata\x12N\n\x10generic_metadata\x18\x01 \x01(\x0b\x32\x34.google.cloud.aiplatform.v1.GenericOperationMetadata\"\xa8\x01\n\x1d\x43reateTrainingPipelineRequest\x12\x39\n\x06parent\x18\x01 \x01(\tB)\xe0\x41\x02\xfa\x41#\n!locations.googleapis.com/Location\x12L\n\x11training_pipeline\x18\x02 \x01(\x0b\x32,.google.cloud.aiplatform.v1.TrainingPipelineB\x03\xe0\x41\x02\"^\n\x1aGetTrainingPipelineRequest\x12@\n\x04name\x18\x01 \x01(\tB2\xe0\x41\x02\xfa\x41,\n*aiplatform.googleapis.com/TrainingPipeline\"\xbf\x01\n\x1cListTrainingPipelinesRequest\x12\x39\n\x06parent\x18\x01 \x01(\tB)\xe0\x41\x02\xfa\x41#\n!locations.googleapis.com/Location\x12\x0e\n\x06\x66ilter\x18\x02 \x01(\t\x12\x11\n\tpage_size\x18\x03 \x01(\x05\x12\x12\n\npage_token\x18\x04 \x01(\t\x12-\n\tread_mask\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.FieldMask\"\x82\x01\n\x1dListTrainingPipelinesResponse\x12H\n\x12training_pipelines\x18\x01 \x03(\x0b\x32,.google.cloud.aiplatform.v1.TrainingPipeline\x12\x17\n\x0fnext_page_token\x18\x02 \x01(\t\"a\n\x1d\x44\x65leteTrainingPipelineRequest\x12@\n\x04name\x18\x01 \x01(\tB2\xe0\x41\x02\xfa\x41,\n*aiplatform.googleapis.com/TrainingPipeline\"a\n\x1d\x43\x61ncelTrainingPipelineRequest\x12@\n\x04name\x18\x01 \x01(\tB2\xe0\x41\x02\xfa\x41,\n*aiplatform.googleapis.com/TrainingPipeline\"\xb2\x01\n\x18\x43reatePipelineJobRequest\x12\x39\n\x06parent\x18\x01 \x01(\tB)\xe0\x41\x02\xfa\x41#\n!locations.googleapis.com/Location\x12\x42\n\x0cpipeline_job\x18\x02 \x01(\x0b\x32\'.google.cloud.aiplatform.v1.PipelineJobB\x03\xe0\x41\x02\x12\x17\n\x0fpipeline_job_id\x18\x03 \x01(\t\"T\n\x15GetPipelineJobRequest\x12;\n\x04name\x18\x01 \x01(\tB-\xe0\x41\x02\xfa\x41\'\n%aiplatform.googleapis.com/PipelineJob\"\xcc\x01\n\x17ListPipelineJobsRequest\x12\x39\n\x06parent\x18\x01 \x01(\tB)\xe0\x41\x02\xfa\x41#\n!locations.googleapis.com/Location\x12\x0e\n\x06\x66ilter\x18\x02 \x01(\t\x12\x11\n\tpage_size\x18\x03 \x01(\x05\x12\x12\n\npage_token\x18\x04 \x01(\t\x12\x10\n\x08order_by\x18\x06 \x01(\t\x12-\n\tread_mask\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.FieldMask\"s\n\x18ListPipelineJobsResponse\x12>\n\rpipeline_jobs\x18\x01 \x03(\x0b\x32\'.google.cloud.aiplatform.v1.PipelineJob\x12\x17\n\x0fnext_page_token\x18\x02 \x01(\t\"W\n\x18\x44\x65letePipelineJobRequest\x12;\n\x04name\x18\x01 \x01(\tB-\xe0\x41\x02\xfa\x41\'\n%aiplatform.googleapis.com/PipelineJob\"\x9d\x01\n\x1e\x42\x61tchDeletePipelineJobsRequest\x12=\n\x06parent\x18\x01 \x01(\tB-\xe0\x41\x02\xfa\x41\'\x12%aiplatform.googleapis.com/PipelineJob\x12<\n\x05names\x18\x02 \x03(\tB-\xe0\x41\x02\xfa\x41\'\n%aiplatform.googleapis.com/PipelineJob\"a\n\x1f\x42\x61tchDeletePipelineJobsResponse\x12>\n\rpipeline_jobs\x18\x01 \x03(\x0b\x32\'.google.cloud.aiplatform.v1.PipelineJob\"W\n\x18\x43\x61ncelPipelineJobRequest\x12;\n\x04name\x18\x01 \x01(\tB-\xe0\x41\x02\xfa\x41\'\n%aiplatform.googleapis.com/PipelineJob\"\x9d\x01\n\x1e\x42\x61tchCancelPipelineJobsRequest\x12=\n\x06parent\x18\x01 \x01(\tB-\xe0\x41\x02\xfa\x41\'\x12%aiplatform.googleapis.com/PipelineJob\x12<\n\x05names\x18\x02 \x03(\tB-\xe0\x41\x02\xfa\x41\'\n%aiplatform.googleapis.com/PipelineJob\"a\n\x1f\x42\x61tchCancelPipelineJobsResponse\x12>\n\rpipeline_jobs\x18\x01 \x03(\x0b\x32\'.google.cloud.aiplatform.v1.PipelineJob2\xa7\x15\n\x0fPipelineService\x12\xee\x01\n\x16\x43reateTrainingPipeline\x12\x39.google.cloud.aiplatform.v1.CreateTrainingPipelineRequest\x1a,.google.cloud.aiplatform.v1.TrainingPipeline\"k\xda\x41\x18parent,training_pipeline\x82\xd3\xe4\x93\x02J\"5/v1/{parent=projects/*/locations/*}/trainingPipelines:\x11training_pipeline\x12\xc1\x01\n\x13GetTrainingPipeline\x12\x36.google.cloud.aiplatform.v1.GetTrainingPipelineRequest\x1a,.google.cloud.aiplatform.v1.TrainingPipeline\"D\xda\x41\x04name\x82\xd3\xe4\x93\x02\x37\x12\x35/v1/{name=projects/*/locations/*/trainingPipelines/*}\x12\xd4\x01\n\x15ListTrainingPipelines\x12\x38.google.cloud.aiplatform.v1.ListTrainingPipelinesRequest\x1a\x39.google.cloud.aiplatform.v1.ListTrainingPipelinesResponse\"F\xda\x41\x06parent\x82\xd3\xe4\x93\x02\x37\x12\x35/v1/{parent=projects/*/locations/*}/trainingPipelines\x12\xeb\x01\n\x16\x44\x65leteTrainingPipeline\x12\x39.google.cloud.aiplatform.v1.DeleteTrainingPipelineRequest\x1a\x1d.google.longrunning.Operation\"w\xca\x41\x30\n\x15google.protobuf.Empty\x12\x17\x44\x65leteOperationMetadata\xda\x41\x04name\x82\xd3\xe4\x93\x02\x37*5/v1/{name=projects/*/locations/*/trainingPipelines/*}\x12\xbb\x01\n\x16\x43\x61ncelTrainingPipeline\x12\x39.google.cloud.aiplatform.v1.CancelTrainingPipelineRequest\x1a\x16.google.protobuf.Empty\"N\xda\x41\x04name\x82\xd3\xe4\x93\x02\x41\"</v1/{name=projects/*/locations/*/trainingPipelines/*}:cancel:\x01*\x12\xe0\x01\n\x11\x43reatePipelineJob\x12\x34.google.cloud.aiplatform.v1.CreatePipelineJobRequest\x1a\'.google.cloud.aiplatform.v1.PipelineJob\"l\xda\x41#parent,pipeline_job,pipeline_job_id\x82\xd3\xe4\x93\x02@\"0/v1/{parent=projects/*/locations/*}/pipelineJobs:\x0cpipeline_job\x12\xad\x01\n\x0eGetPipelineJob\x12\x31.google.cloud.aiplatform.v1.GetPipelineJobRequest\x1a\'.google.cloud.aiplatform.v1.PipelineJob\"?\xda\x41\x04name\x82\xd3\xe4\x93\x02\x32\x12\x30/v1/{name=projects/*/locations/*/pipelineJobs/*}\x12\xc0\x01\n\x10ListPipelineJobs\x12\x33.google.cloud.aiplatform.v1.ListPipelineJobsRequest\x1a\x34.google.cloud.aiplatform.v1.ListPipelineJobsResponse\"A\xda\x41\x06parent\x82\xd3\xe4\x93\x02\x32\x12\x30/v1/{parent=projects/*/locations/*}/pipelineJobs\x12\xdc\x01\n\x11\x44\x65letePipelineJob\x12\x34.google.cloud.aiplatform.v1.DeletePipelineJobRequest\x1a\x1d.google.longrunning.Operation\"r\xca\x41\x30\n\x15google.protobuf.Empty\x12\x17\x44\x65leteOperationMetadata\xda\x41\x04name\x82\xd3\xe4\x93\x02\x32*0/v1/{name=projects/*/locations/*/pipelineJobs/*}\x12\x8a\x02\n\x17\x42\x61tchDeletePipelineJobs\x12:.google.cloud.aiplatform.v1.BatchDeletePipelineJobsRequest\x1a\x1d.google.longrunning.Operation\"\x93\x01\xca\x41:\n\x1f\x42\x61tchDeletePipelineJobsResponse\x12\x17\x44\x65leteOperationMetadata\xda\x41\x0cparent,names\x82\xd3\xe4\x93\x02\x41\"</v1/{parent=projects/*/locations/*}/pipelineJobs:batchDelete:\x01*\x12\xac\x01\n\x11\x43\x61ncelPipelineJob\x12\x34.google.cloud.aiplatform.v1.CancelPipelineJobRequest\x1a\x16.google.protobuf.Empty\"I\xda\x41\x04name\x82\xd3\xe4\x93\x02<\"7/v1/{name=projects/*/locations/*/pipelineJobs/*}:cancel:\x01*\x12\x9b\x02\n\x17\x42\x61tchCancelPipelineJobs\x12:.google.cloud.aiplatform.v1.BatchCancelPipelineJobsRequest\x1a\x1d.google.longrunning.Operation\"\xa4\x01\xca\x41K\n\x1f\x42\x61tchCancelPipelineJobsResponse\x12(BatchCancelPipelineJobsOperationMetadata\xda\x41\x0cparent,names\x82\xd3\xe4\x93\x02\x41\"</v1/{parent=projects/*/locations/*}/pipelineJobs:batchCancel:\x01*\x1aM\xca\x41\x19\x61iplatform.googleapis.com\xd2\x41.https://www.googleapis.com/auth/cloud-platformB\xd2\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\x14PipelineServiceProtoP\x01Z>cloud.google.com/go/aiplatform/apiv1/aiplatformpb;aiplatformpb\xaa\x02\x1aGoogle.Cloud.AIPlatform.V1\xca\x02\x1aGoogle\\Cloud\\AIPlatform\\V1\xea\x02\x1dGoogle::Cloud::AIPlatform::V1b\x06proto3"
 
 pool = Google::Protobuf::DescriptorPool.generated_pool
 
@@ -31,6 +31,7 @@ rescue TypeError
   file = pool.add_serialized_file(serialized)
   warn "Warning: Protobuf detected an import path issue while loading generated file #{__FILE__}"
   imports = [
+    ["google.cloud.aiplatform.v1.GenericOperationMetadata", "google/cloud/aiplatform/v1/operation.proto"],
     ["google.cloud.aiplatform.v1.TrainingPipeline", "google/cloud/aiplatform/v1/training_pipeline.proto"],
     ["google.protobuf.FieldMask", "google/protobuf/field_mask.proto"],
     ["google.cloud.aiplatform.v1.PipelineJob", "google/cloud/aiplatform/v1/pipeline_job.proto"],
@@ -49,6 +50,7 @@ module Google
   module Cloud
     module AIPlatform
       module V1
+        BatchCancelPipelineJobsOperationMetadata = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.BatchCancelPipelineJobsOperationMetadata").msgclass
         CreateTrainingPipelineRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.CreateTrainingPipelineRequest").msgclass
         GetTrainingPipelineRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.GetTrainingPipelineRequest").msgclass
         ListTrainingPipelinesRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ListTrainingPipelinesRequest").msgclass
@@ -60,7 +62,11 @@ module Google
         ListPipelineJobsRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ListPipelineJobsRequest").msgclass
         ListPipelineJobsResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ListPipelineJobsResponse").msgclass
         DeletePipelineJobRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.DeletePipelineJobRequest").msgclass
+        BatchDeletePipelineJobsRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.BatchDeletePipelineJobsRequest").msgclass
+        BatchDeletePipelineJobsResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.BatchDeletePipelineJobsResponse").msgclass
         CancelPipelineJobRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.CancelPipelineJobRequest").msgclass
+        BatchCancelPipelineJobsRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.BatchCancelPipelineJobsRequest").msgclass
+        BatchCancelPipelineJobsResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.BatchCancelPipelineJobsResponse").msgclass
       end
     end
   end

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

@@ -67,6 +67,10 @@ module Google
             rpc :ListPipelineJobs, ::Google::Cloud::AIPlatform::V1::ListPipelineJobsRequest, ::Google::Cloud::AIPlatform::V1::ListPipelineJobsResponse
             # Deletes a PipelineJob.
             rpc :DeletePipelineJob, ::Google::Cloud::AIPlatform::V1::DeletePipelineJobRequest, ::Google::Longrunning::Operation
+            # Batch deletes PipelineJobs
+            # The Operation is atomic. If it fails, none of the PipelineJobs are deleted.
+            # If it succeeds, all of the PipelineJobs are deleted.
+            rpc :BatchDeletePipelineJobs, ::Google::Cloud::AIPlatform::V1::BatchDeletePipelineJobsRequest, ::Google::Longrunning::Operation
             # Cancels a PipelineJob.
             # Starts asynchronous cancellation on the PipelineJob. The server
             # makes a best effort to cancel the pipeline, but success is not
@@ -81,6 +85,14 @@ module Google
             # [PipelineJob.state][google.cloud.aiplatform.v1.PipelineJob.state] is set to
             # `CANCELLED`.
             rpc :CancelPipelineJob, ::Google::Cloud::AIPlatform::V1::CancelPipelineJobRequest, ::Google::Protobuf::Empty
+            # Batch cancel PipelineJobs.
+            # Firstly the server will check if all the jobs are in non-terminal states,
+            # and skip the jobs that are already terminated.
+            # If the operation failed, none of the pipeline jobs are cancelled.
+            # The server will poll the states of all the pipeline jobs periodically
+            # to check the cancellation status.
+            # This operation will return an LRO.
+            rpc :BatchCancelPipelineJobs, ::Google::Cloud::AIPlatform::V1::BatchCancelPipelineJobsRequest, ::Google::Longrunning::Operation
           end
 
           Stub = Service.rpc_stub_class

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

@@ -5,11 +5,12 @@
 require 'google/protobuf'
 
 require 'google/api/field_behavior_pb'
+require 'google/api/resource_pb'
 require 'google/cloud/aiplatform/v1/openapi_pb'
 require 'google/protobuf/struct_pb'
 
 
-descriptor_data = "\n%google/cloud/aiplatform/v1/tool.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a(google/cloud/aiplatform/v1/openapi.proto\x1a\x1cgoogle/protobuf/struct.proto\"[\n\x04Tool\x12S\n\x15\x66unction_declarations\x18\x01 \x03(\x0b\x32/.google.cloud.aiplatform.v1.FunctionDeclarationB\x03\xe0\x41\x01\"\x7f\n\x13\x46unctionDeclaration\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x18\n\x0b\x64\x65scription\x18\x02 \x01(\tB\x03\xe0\x41\x01\x12;\n\nparameters\x18\x03 \x01(\x0b\x32\".google.cloud.aiplatform.v1.SchemaB\x03\xe0\x41\x01\"M\n\x0c\x46unctionCall\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12*\n\x04\x61rgs\x18\x02 \x01(\x0b\x32\x17.google.protobuf.StructB\x03\xe0\x41\x01\"U\n\x10\x46unctionResponse\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12.\n\x08response\x18\x02 \x01(\x0b\x32\x17.google.protobuf.StructB\x03\xe0\x41\x02\x42\xc7\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\tToolProtoP\x01Z>cloud.google.com/go/aiplatform/apiv1/aiplatformpb;aiplatformpb\xaa\x02\x1aGoogle.Cloud.AIPlatform.V1\xca\x02\x1aGoogle\\Cloud\\AIPlatform\\V1\xea\x02\x1dGoogle::Cloud::AIPlatform::V1b\x06proto3"
+descriptor_data = "\n%google/cloud/aiplatform/v1/tool.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a(google/cloud/aiplatform/v1/openapi.proto\x1a\x1cgoogle/protobuf/struct.proto\"\xf3\x01\n\x04Tool\x12S\n\x15\x66unction_declarations\x18\x01 \x03(\x0b\x32/.google.cloud.aiplatform.v1.FunctionDeclarationB\x03\xe0\x41\x01\x12=\n\tretrieval\x18\x02 \x01(\x0b\x32%.google.cloud.aiplatform.v1.RetrievalB\x03\xe0\x41\x01\x12W\n\x17google_search_retrieval\x18\x03 \x01(\x0b\x32\x31.google.cloud.aiplatform.v1.GoogleSearchRetrievalB\x03\xe0\x41\x01\"\x7f\n\x13\x46unctionDeclaration\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x18\n\x0b\x64\x65scription\x18\x02 \x01(\tB\x03\xe0\x41\x01\x12;\n\nparameters\x18\x03 \x01(\x0b\x32\".google.cloud.aiplatform.v1.SchemaB\x03\xe0\x41\x01\"M\n\x0c\x46unctionCall\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12*\n\x04\x61rgs\x18\x02 \x01(\x0b\x32\x17.google.protobuf.StructB\x03\xe0\x41\x01\"U\n\x10\x46unctionResponse\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12.\n\x08response\x18\x02 \x01(\x0b\x32\x17.google.protobuf.StructB\x03\xe0\x41\x02\"\x7f\n\tRetrieval\x12\x46\n\x10vertex_ai_search\x18\x02 \x01(\x0b\x32*.google.cloud.aiplatform.v1.VertexAISearchH\x00\x12 \n\x13\x64isable_attribution\x18\x03 \x01(\x08\x42\x03\xe0\x41\x01\x42\x08\n\x06source\"(\n\x0eVertexAISearch\x12\x16\n\tdatastore\x18\x01 \x01(\tB\x03\xe0\x41\x02\"9\n\x15GoogleSearchRetrieval\x12 \n\x13\x64isable_attribution\x18\x01 \x01(\x08\x42\x03\xe0\x41\x01\x42\xc7\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\tToolProtoP\x01Z>cloud.google.com/go/aiplatform/apiv1/aiplatformpb;aiplatformpb\xaa\x02\x1aGoogle.Cloud.AIPlatform.V1\xca\x02\x1aGoogle\\Cloud\\AIPlatform\\V1\xea\x02\x1dGoogle::Cloud::AIPlatform::V1b\x06proto3"
 
 pool = Google::Protobuf::DescriptorPool.generated_pool
 
@@ -45,6 +46,9 @@ module Google
         FunctionDeclaration = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.FunctionDeclaration").msgclass
         FunctionCall = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.FunctionCall").msgclass
         FunctionResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.FunctionResponse").msgclass
+        Retrieval = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.Retrieval").msgclass
+        VertexAISearch = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.VertexAISearch").msgclass
+        GoogleSearchRetrieval = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.GoogleSearchRetrieval").msgclass
       end
     end
   end

data/proto_docs/google/api/client.rb

@@ -118,6 +118,10 @@ module Google
     #   @return [::String]
     #     Optional link to proto reference documentation.  Example:
     #     https://cloud.google.com/pubsub/lite/docs/reference/rpc
+    # @!attribute [rw] rest_reference_documentation_uri
+    #   @return [::String]
+    #     Optional link to REST reference documentation.  Example:
+    #     https://cloud.google.com/pubsub/lite/docs/reference/rest
     class Publishing
       include ::Google::Protobuf::MessageExts
       extend ::Google::Protobuf::MessageExts::ClassMethods

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

@@ -50,6 +50,9 @@ module Google
           # Nvidia L4 GPU.
           NVIDIA_L4 = 11
 
+          # Nvidia H100 80Gb GPU.
+          NVIDIA_H100_80GB = 13
+
           # TPU v2.
           TPU_V2 = 6
 

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

@@ -176,6 +176,15 @@ module Google
         # @!attribute [r] probability
         #   @return [::Google::Cloud::AIPlatform::V1::SafetyRating::HarmProbability]
         #     Output only. Harm probability levels in the content.
+        # @!attribute [r] probability_score
+        #   @return [::Float]
+        #     Output only. Harm probability score.
+        # @!attribute [r] severity
+        #   @return [::Google::Cloud::AIPlatform::V1::SafetyRating::HarmSeverity]
+        #     Output only. Harm severity levels in the content.
+        # @!attribute [r] severity_score
+        #   @return [::Float]
+        #     Output only. Harm severity score.
         # @!attribute [r] blocked
         #   @return [::Boolean]
         #     Output only. Indicates whether the content was filtered out because of this
@@ -201,6 +210,24 @@ module Google
             # High level of harm.
             HIGH = 4
           end
+
+          # Harm severity levels.
+          module HarmSeverity
+            # Harm severity unspecified.
+            HARM_SEVERITY_UNSPECIFIED = 0
+
+            # Negligible level of harm severity.
+            HARM_SEVERITY_NEGLIGIBLE = 1
+
+            # Low level of harm severity.
+            HARM_SEVERITY_LOW = 2
+
+            # Medium level of harm severity.
+            HARM_SEVERITY_MEDIUM = 3
+
+            # High level of harm severity.
+            HARM_SEVERITY_HIGH = 4
+          end
         end
 
         # A collection of source attributions for a piece of content.
@@ -259,6 +286,9 @@ module Google
         # @!attribute [r] citation_metadata
         #   @return [::Google::Cloud::AIPlatform::V1::CitationMetadata]
         #     Output only. Source attribution of the generated content.
+        # @!attribute [r] grounding_metadata
+        #   @return [::Google::Cloud::AIPlatform::V1::GroundingMetadata]
+        #     Output only. Metadata specifies sources used to ground generated content.
         class Candidate
           include ::Google::Protobuf::MessageExts
           extend ::Google::Protobuf::MessageExts::ClassMethods
@@ -286,9 +316,78 @@ module Google
 
             # All other reasons that stopped the token generation
             OTHER = 5
+
+            # The token generation was stopped as the response was flagged for the
+            # terms which are included from the terminology blocklist.
+            BLOCKLIST = 6
+
+            # The token generation was stopped as the response was flagged for
+            # the prohibited contents.
+            PROHIBITED_CONTENT = 7
+
+            # The token generation was stopped as the response was flagged for
+            # Sensitive Personally Identifiable Information (SPII) contents.
+            SPII = 8
+          end
+        end
+
+        # Segment of the content.
+        # @!attribute [r] part_index
+        #   @return [::Integer]
+        #     Output only. The index of a Part object within its parent Content object.
+        # @!attribute [r] start_index
+        #   @return [::Integer]
+        #     Output only. Start index in the given Part, measured in bytes. Offset from
+        #     the start of the Part, inclusive, starting at zero.
+        # @!attribute [r] end_index
+        #   @return [::Integer]
+        #     Output only. End index in the given Part, measured in bytes. Offset from
+        #     the start of the Part, exclusive, starting at zero.
+        class Segment
+          include ::Google::Protobuf::MessageExts
+          extend ::Google::Protobuf::MessageExts::ClassMethods
+        end
+
+        # Grounding attribution.
+        # @!attribute [rw] web
+        #   @return [::Google::Cloud::AIPlatform::V1::GroundingAttribution::Web]
+        #     Optional. Attribution from the web.
+        # @!attribute [r] segment
+        #   @return [::Google::Cloud::AIPlatform::V1::Segment]
+        #     Output only. Segment of the content this attribution belongs to.
+        # @!attribute [r] confidence_score
+        #   @return [::Float]
+        #     Optional. Output only. Confidence score of the attribution. Ranges from 0
+        #     to 1. 1 is the most confident.
+        class GroundingAttribution
+          include ::Google::Protobuf::MessageExts
+          extend ::Google::Protobuf::MessageExts::ClassMethods
+
+          # Attribution from the web.
+          # @!attribute [r] uri
+          #   @return [::String]
+          #     Output only. URI reference of the attribution.
+          # @!attribute [r] title
+          #   @return [::String]
+          #     Output only. Title of the attribution.
+          class Web
+            include ::Google::Protobuf::MessageExts
+            extend ::Google::Protobuf::MessageExts::ClassMethods
           end
         end
 
+        # Metadata returned to client when grounding is enabled.
+        # @!attribute [rw] web_search_queries
+        #   @return [::Array<::String>]
+        #     Optional. Web search queries for the following-up web search.
+        # @!attribute [rw] grounding_attributions
+        #   @return [::Array<::Google::Cloud::AIPlatform::V1::GroundingAttribution>]
+        #     Optional. List of grounding attributions.
+        class GroundingMetadata
+          include ::Google::Protobuf::MessageExts
+          extend ::Google::Protobuf::MessageExts::ClassMethods
+        end
+
         # Harm categories that will block the content.
         module HarmCategory
           # The harm category is unspecified.

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

@@ -38,6 +38,14 @@ module Google
         # @!attribute [r] big_query_dataset_name
         #   @return [::String]
         #     Output only. Name of the associated BigQuery dataset.
+        # @!attribute [rw] display_name
+        #   @return [::String]
+        #     The user-defined name of the DatasetVersion.
+        #     The name can be up to 128 characters long and can consist of any UTF-8
+        #     characters.
+        # @!attribute [r] metadata
+        #   @return [::Google::Protobuf::Value]
+        #     Required. Output only. Additional information about the DatasetVersion.
         class DatasetVersion
           include ::Google::Protobuf::MessageExts
           extend ::Google::Protobuf::MessageExts::ClassMethods

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

@@ -84,20 +84,23 @@ module Google
         #     Optional. An expression for filtering the results of the request. For field
         #     names both snake_case and camelCase are supported.
         #
-        #       * `endpoint` supports = and !=. `endpoint` represents the Endpoint ID,
+        #       * `endpoint` supports `=` and `!=`. `endpoint` represents the Endpoint
+        #       ID,
         #         i.e. the last segment of the Endpoint's [resource
         #         name][google.cloud.aiplatform.v1.Endpoint.name].
-        #       * `display_name` supports = and, !=
+        #       * `display_name` supports `=` and `!=`.
         #       * `labels` supports general map functions that is:
         #         * `labels.key=value` - key:value equality
-        #         * `labels.key:* or labels:key - key existence
+        #         * `labels.key:*` or `labels:key` - key existence
         #         * A key including a space must be quoted. `labels."a key"`.
+        #       * `base_model_name` only supports `=`.
         #
         #     Some examples:
         #
         #       * `endpoint=1`
         #       * `displayName="myDisplayName"`
         #       * `labels.myKey="myValue"`
+        #       * `baseModelName="text-bison"`
         # @!attribute [rw] page_size
         #   @return [::Integer]
         #     Optional. The standard list page size.

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

@@ -69,8 +69,8 @@ module Google
           #     BigQuery Table or View.
           # @!attribute [rw] entity_id_columns
           #   @return [::Array<::String>]
-          #     Optional. Columns to construct entity_id / row keys. Currently only
-          #     supports 1 entity_id_column. If not provided defaults to `entity_id`.
+          #     Optional. Columns to construct entity_id / row keys.
+          #     If not provided defaults to `entity_id`.
           class BigQuery
             include ::Google::Protobuf::MessageExts
             extend ::Google::Protobuf::MessageExts::ClassMethods

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

@@ -73,8 +73,7 @@ module Google
           #     trigger based on FeatureView.SyncConfig.
           # @!attribute [rw] entity_id_columns
           #   @return [::Array<::String>]
-          #     Required. Columns to construct entity_id / row keys. Start by supporting
-          #     1 only.
+          #     Required. Columns to construct entity_id / row keys.
           class BigQuerySource
             include ::Google::Protobuf::MessageExts
             extend ::Google::Protobuf::MessageExts::ClassMethods
@@ -99,6 +98,9 @@ module Google
           # @!attribute [rw] feature_groups
           #   @return [::Array<::Google::Cloud::AIPlatform::V1::FeatureView::FeatureRegistrySource::FeatureGroup>]
           #     Required. List of features that need to be synced to Online Store.
+          # @!attribute [rw] project_number
+          #   @return [::Integer]
+          #     Optional. The project number of the parent project of the Feature Groups.
           class FeatureRegistrySource
             include ::Google::Protobuf::MessageExts
             extend ::Google::Protobuf::MessageExts::ClassMethods

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

@@ -38,9 +38,25 @@ module Google
         # @!attribute [r] final_status
         #   @return [::Google::Rpc::Status]
         #     Output only. Final status of the FeatureViewSync.
+        # @!attribute [r] sync_summary
+        #   @return [::Google::Cloud::Aiplatform::V1::FeatureViewSync::SyncSummary]
+        #     Output only. Summary of the sync job.
         class FeatureViewSync
           include ::Google::Protobuf::MessageExts
           extend ::Google::Protobuf::MessageExts::ClassMethods
+
+          # Summary from the Sync job. For continuous syncs, the summary is updated
+          # periodically. For batch syncs, it gets updated on completion of the sync.
+          # @!attribute [r] row_synced
+          #   @return [::Integer]
+          #     Output only. Total number of rows synced.
+          # @!attribute [r] total_slot
+          #   @return [::Integer]
+          #     Output only. BigQuery slot milliseconds consumed for the sync job.
+          class SyncSummary
+            include ::Google::Protobuf::MessageExts
+            extend ::Google::Protobuf::MessageExts::ClassMethods
+          end
         end
       end
     end

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

@@ -208,6 +208,9 @@ module Google
 
               # Datapoints are eligible iff their value is > the query's.
               GREATER = 5
+
+              # Datapoints are eligible iff their value is != the query's.
+              NOT_EQUAL = 6
             end
           end
 

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

@@ -295,6 +295,10 @@ module Google
         #     MetadataStore when creating the Model. The Artifact resource name pattern
         #     is
         #     `projects/{project}/locations/{location}/metadataStores/{metadata_store}/artifacts/{artifact}`.
+        # @!attribute [rw] base_model_source
+        #   @return [::Google::Cloud::AIPlatform::V1::Model::BaseModelSource]
+        #     Optional. User input field to specify the base model source. Currently it
+        #     only supports specifing the Model Garden models and Genie models.
         class Model
           include ::Google::Protobuf::MessageExts
           extend ::Google::Protobuf::MessageExts::ClassMethods
@@ -394,6 +398,19 @@ module Google
             extend ::Google::Protobuf::MessageExts::ClassMethods
           end
 
+          # User input field to specify the base model source. Currently it only
+          # supports specifing the Model Garden models and Genie models.
+          # @!attribute [rw] model_garden_source
+          #   @return [::Google::Cloud::AIPlatform::V1::ModelGardenSource]
+          #     Source information of Model Garden models.
+          # @!attribute [rw] genie_source
+          #   @return [::Google::Cloud::AIPlatform::V1::GenieSource]
+          #     Information about the base model of Genie models.
+          class BaseModelSource
+            include ::Google::Protobuf::MessageExts
+            extend ::Google::Protobuf::MessageExts::ClassMethods
+          end
+
           # @!attribute [rw] key
           #   @return [::String]
           # @!attribute [rw] value
@@ -437,6 +454,26 @@ module Google
           extend ::Google::Protobuf::MessageExts::ClassMethods
         end
 
+        # Contains information about the source of the models generated from Model
+        # Garden.
+        # @!attribute [rw] public_model_name
+        #   @return [::String]
+        #     Required. The model garden source model resource name.
+        class ModelGardenSource
+          include ::Google::Protobuf::MessageExts
+          extend ::Google::Protobuf::MessageExts::ClassMethods
+        end
+
+        # Contains information about the source of the models generated from Generative
+        # AI Studio.
+        # @!attribute [rw] base_model_uri
+        #   @return [::String]
+        #     Required. The public base model URI.
+        class GenieSource
+          include ::Google::Protobuf::MessageExts
+          extend ::Google::Protobuf::MessageExts::ClassMethods
+        end
+
         # Contains the schemata used in Model's predictions and explanations via
         # {::Google::Cloud::AIPlatform::V1::PredictionService::Client#predict PredictionService.Predict},
         # {::Google::Cloud::AIPlatform::V1::PredictionService::Client#explain PredictionService.Explain}

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

@@ -121,12 +121,14 @@ module Google
         #         * `labels.key=value` - key:value equality
         #         * `labels.key:* or labels:key - key existence
         #         * A key including a space must be quoted. `labels."a key"`.
+        #       * `base_model_name` only supports =
         #
         #     Some examples:
         #
         #       * `model=1234`
         #       * `displayName="myDisplayName"`
         #       * `labels.myKey="myValue"`
+        #       * `baseModelName="text-bison"`
         # @!attribute [rw] page_size
         #   @return [::Integer]
         #     The standard list page size.

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

@@ -21,6 +21,16 @@ module Google
   module Cloud
     module AIPlatform
       module V1
+        # Runtime operation information for
+        # {::Google::Cloud::AIPlatform::V1::PipelineService::Client#batch_cancel_pipeline_jobs PipelineService.BatchCancelPipelineJobs}.
+        # @!attribute [rw] generic_metadata
+        #   @return [::Google::Cloud::AIPlatform::V1::GenericOperationMetadata]
+        #     The common part of the operation metadata.
+        class BatchCancelPipelineJobsOperationMetadata
+          include ::Google::Protobuf::MessageExts
+          extend ::Google::Protobuf::MessageExts::ClassMethods
+        end
+
         # Request message for
         # {::Google::Cloud::AIPlatform::V1::PipelineService::Client#create_training_pipeline PipelineService.CreateTrainingPipeline}.
         # @!attribute [rw] parent
@@ -272,6 +282,33 @@ module Google
           extend ::Google::Protobuf::MessageExts::ClassMethods
         end
 
+        # Request message for
+        # {::Google::Cloud::AIPlatform::V1::PipelineService::Client#batch_delete_pipeline_jobs PipelineService.BatchDeletePipelineJobs}.
+        # @!attribute [rw] parent
+        #   @return [::String]
+        #     Required. The name of the PipelineJobs' parent resource.
+        #     Format: `projects/{project}/locations/{location}`
+        # @!attribute [rw] names
+        #   @return [::Array<::String>]
+        #     Required. The names of the PipelineJobs to delete.
+        #     A maximum of 32 PipelineJobs can be deleted in a batch.
+        #     Format:
+        #     `projects/{project}/locations/{location}/pipelineJobs/{pipelineJob}`
+        class BatchDeletePipelineJobsRequest
+          include ::Google::Protobuf::MessageExts
+          extend ::Google::Protobuf::MessageExts::ClassMethods
+        end
+
+        # Response message for
+        # {::Google::Cloud::AIPlatform::V1::PipelineService::Client#batch_delete_pipeline_jobs PipelineService.BatchDeletePipelineJobs}.
+        # @!attribute [rw] pipeline_jobs
+        #   @return [::Array<::Google::Cloud::AIPlatform::V1::PipelineJob>]
+        #     PipelineJobs deleted.
+        class BatchDeletePipelineJobsResponse
+          include ::Google::Protobuf::MessageExts
+          extend ::Google::Protobuf::MessageExts::ClassMethods
+        end
+
         # Request message for
         # {::Google::Cloud::AIPlatform::V1::PipelineService::Client#cancel_pipeline_job PipelineService.CancelPipelineJob}.
         # @!attribute [rw] name
@@ -283,6 +320,33 @@ module Google
           include ::Google::Protobuf::MessageExts
           extend ::Google::Protobuf::MessageExts::ClassMethods
         end
+
+        # Request message for
+        # {::Google::Cloud::AIPlatform::V1::PipelineService::Client#batch_cancel_pipeline_jobs PipelineService.BatchCancelPipelineJobs}.
+        # @!attribute [rw] parent
+        #   @return [::String]
+        #     Required. The name of the PipelineJobs' parent resource.
+        #     Format: `projects/{project}/locations/{location}`
+        # @!attribute [rw] names
+        #   @return [::Array<::String>]
+        #     Required. The names of the PipelineJobs to cancel.
+        #     A maximum of 32 PipelineJobs can be cancelled in a batch.
+        #     Format:
+        #     `projects/{project}/locations/{location}/pipelineJobs/{pipelineJob}`
+        class BatchCancelPipelineJobsRequest
+          include ::Google::Protobuf::MessageExts
+          extend ::Google::Protobuf::MessageExts::ClassMethods
+        end
+
+        # Response message for
+        # {::Google::Cloud::AIPlatform::V1::PipelineService::Client#batch_cancel_pipeline_jobs PipelineService.BatchCancelPipelineJobs}.
+        # @!attribute [rw] pipeline_jobs
+        #   @return [::Array<::Google::Cloud::AIPlatform::V1::PipelineJob>]
+        #     PipelineJobs cancelled.
+        class BatchCancelPipelineJobsResponse
+          include ::Google::Protobuf::MessageExts
+          extend ::Google::Protobuf::MessageExts::ClassMethods
+        end
       end
     end
   end

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

@@ -279,16 +279,23 @@ module Google
             # The model launch stage is unspecified.
             LAUNCH_STAGE_UNSPECIFIED = 0
 
-            # Used to indicate the PublisherModel is at Experimental launch stage.
+            # Used to indicate the PublisherModel is at Experimental launch stage,
+            # available to a small set of customers.
             EXPERIMENTAL = 1
 
-            # Used to indicate the PublisherModel is at Private Preview launch stage.
+            # Used to indicate the PublisherModel is at Private Preview launch stage,
+            # only available to a small set of customers, although a larger set of
+            # customers than an Experimental launch. Previews are the first launch
+            # stage used to get feedback from customers.
             PRIVATE_PREVIEW = 2
 
-            # Used to indicate the PublisherModel is at Public Preview launch stage.
+            # Used to indicate the PublisherModel is at Public Preview launch stage,
+            # available to all customers, although not supported for production
+            # workloads.
             PUBLIC_PREVIEW = 3
 
-            # Used to indicate the PublisherModel is at GA launch stage.
+            # Used to indicate the PublisherModel is at GA launch stage, available to
+            # all customers and ready for production workload.
             GA = 4
           end
 

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

@@ -25,17 +25,29 @@ module Google
         #
         # A `Tool` is a piece of code that enables the system to interact with
         # external systems to perform an action, or set of actions, outside of
-        # knowledge and scope of the model.
+        # knowledge and scope of the model. A Tool object should contain exactly
+        # one type of Tool (e.g FunctionDeclaration, Retrieval or
+        # GoogleSearchRetrieval).
         # @!attribute [rw] function_declarations
         #   @return [::Array<::Google::Cloud::AIPlatform::V1::FunctionDeclaration>]
-        #     Optional. One or more function declarations to be passed to the model along
-        #     with the current user query. Model may decide to call a subset of these
-        #     functions by populating [FunctionCall][content.part.function_call] in the
-        #     response. User should provide a
-        #     [FunctionResponse][content.part.function_response] for each function call
-        #     in the next turn. Based on the function responses, Model will generate the
-        #     final response back to the user. Maximum 64 function declarations can be
-        #     provided.
+        #     Optional. Function tool type.
+        #     One or more function declarations to be passed to the model along with the
+        #     current user query. Model may decide to call a subset of these functions
+        #     by populating [FunctionCall][content.part.function_call] in the response.
+        #     User should provide a [FunctionResponse][content.part.function_response]
+        #     for each function call in the next turn. Based on the function responses,
+        #     Model will generate the final response back to the user.
+        #     Maximum 64 function declarations can be provided.
+        # @!attribute [rw] retrieval
+        #   @return [::Google::Cloud::AIPlatform::V1::Retrieval]
+        #     Optional. Retrieval tool type.
+        #     System will always execute the provided retrieval tool(s) to get external
+        #     knowledge to answer the prompt. Retrieval results are presented to the
+        #     model for generation.
+        # @!attribute [rw] google_search_retrieval
+        #   @return [::Google::Cloud::AIPlatform::V1::GoogleSearchRetrieval]
+        #     Optional. GoogleSearchRetrieval tool type.
+        #     Specialized retrieval tool that is powered by Google search.
         class Tool
           include ::Google::Protobuf::MessageExts
           extend ::Google::Protobuf::MessageExts::ClassMethods
@@ -106,6 +118,43 @@ module Google
           include ::Google::Protobuf::MessageExts
           extend ::Google::Protobuf::MessageExts::ClassMethods
         end
+
+        # Defines a retrieval tool that model can call to access external knowledge.
+        # @!attribute [rw] vertex_ai_search
+        #   @return [::Google::Cloud::AIPlatform::V1::VertexAISearch]
+        #     Set to use data source powered by Vertex AI Search.
+        # @!attribute [rw] disable_attribution
+        #   @return [::Boolean]
+        #     Optional. Disable using the result from this tool in detecting grounding
+        #     attribution. This does not affect how the result is given to the model for
+        #     generation.
+        class Retrieval
+          include ::Google::Protobuf::MessageExts
+          extend ::Google::Protobuf::MessageExts::ClassMethods
+        end
+
+        # Retrieve from Vertex AI Search datastore for grounding.
+        # See https://cloud.google.com/vertex-ai-search-and-conversation
+        # @!attribute [rw] datastore
+        #   @return [::String]
+        #     Required. Fully-qualified Vertex AI Search's datastore resource ID.
+        #     Format:
+        #     projects/\\{project}/locations/\\{location}/collections/\\{collection}/dataStores/\\{dataStore}
+        class VertexAISearch
+          include ::Google::Protobuf::MessageExts
+          extend ::Google::Protobuf::MessageExts::ClassMethods
+        end
+
+        # Tool to retrieve public web data for grounding, powered by Google.
+        # @!attribute [rw] disable_attribution
+        #   @return [::Boolean]
+        #     Optional. Disable using the result from this tool in detecting grounding
+        #     attribution. This does not affect how the result is given to the model for
+        #     generation.
+        class GoogleSearchRetrieval
+          include ::Google::Protobuf::MessageExts
+          extend ::Google::Protobuf::MessageExts::ClassMethods
+        end
       end
     end
   end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: google-cloud-ai_platform-v1
 version: !ruby/object:Gem::Version
-  version: 0.37.0
+  version: 0.38.0
 platform: ruby
 authors:
 - Google LLC
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2024-02-26 00:00:00.000000000 Z
+date: 2024-03-18 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: gapic-common