google-cloud-ai_platform-v1 1.18.0 → 1.20.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 57ee1ae593b90393d3654f02590f976690ce701238ad7e442feb6313f36a3766
-  data.tar.gz: 760f73329a3c77df547af8c8b0f28da4e852d152e80243b68cc89ebded345056
+  metadata.gz: f160dfb751785100f39c1eadc987168d1b70f2027e6ac152d84310bf2e2310fd
+  data.tar.gz: 201173b1f49652e454f6f02608b069b7656838e49144cec7ac231bb3d1c59c4a
 SHA512:
-  metadata.gz: 864b66adebd6ea23f06200fbbe05d34c5aaa92093790e3a29335f7056a686cdecbdc5df07a81c7e0dfe2b98c3cc0c4c90ffe991c0a64ffa1440b86e8f89cfee2
-  data.tar.gz: 9639021a58313d1b1b8dcc06c885e949aa31c307816d0cde717162f8790efb433ca13e24f9c7418dd716ad05b3ed2d46bc1e8c16f6dd6cfa0e2a14ba7e1bc6a3
+  metadata.gz: da6fc27ac30ce5cb4552c3c54b5e5fe6ed50ba4b7a1f581b4b022934603f548b5bdd7e30222adfe43434dfb865232fd583ce99d40e6b8662552c8ed3eca55015
+  data.tar.gz: ab69f022166cdfd4242708b18d3c9b034ff6aa1db8e87c334b468a0a10389bcbb9109949c1e8f92812e3a76b6f68f0a8b57f99ec0d9005d990beadc81a8bd86a

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

@@ -1066,9 +1066,9 @@ module Google
 
             ##
             # Updates an existing deployed model. Updatable fields include
-            # `min_replica_count`, `max_replica_count`, `autoscaling_metric_specs`,
-            # `disable_container_logging` (v1 only), and `enable_container_logging`
-            # (v1beta1 only).
+            # `min_replica_count`, `max_replica_count`, `required_replica_count`,
+            # `autoscaling_metric_specs`, `disable_container_logging` (v1 only), and
+            # `enable_container_logging` (v1beta1 only).
             #
             # @overload mutate_deployed_model(request, options = nil)
             #   Pass arguments to `mutate_deployed_model` via a request object, either of type
@@ -1099,6 +1099,8 @@ module Google
             #     * `max_replica_count` in either
             #     {::Google::Cloud::AIPlatform::V1::DedicatedResources DedicatedResources} or
             #     {::Google::Cloud::AIPlatform::V1::AutomaticResources AutomaticResources}
+            #     * `required_replica_count` in
+            #     {::Google::Cloud::AIPlatform::V1::DedicatedResources DedicatedResources}
             #     * {::Google::Cloud::AIPlatform::V1::DedicatedResources#autoscaling_metric_specs autoscaling_metric_specs}
             #     * `disable_container_logging` (v1 only)
             #     * `enable_container_logging` (v1beta1 only)

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

@@ -1005,9 +1005,9 @@ module Google
 
               ##
               # Updates an existing deployed model. Updatable fields include
-              # `min_replica_count`, `max_replica_count`, `autoscaling_metric_specs`,
-              # `disable_container_logging` (v1 only), and `enable_container_logging`
-              # (v1beta1 only).
+              # `min_replica_count`, `max_replica_count`, `required_replica_count`,
+              # `autoscaling_metric_specs`, `disable_container_logging` (v1 only), and
+              # `enable_container_logging` (v1beta1 only).
               #
               # @overload mutate_deployed_model(request, options = nil)
               #   Pass arguments to `mutate_deployed_model` via a request object, either of type
@@ -1038,6 +1038,8 @@ module Google
               #     * `max_replica_count` in either
               #     {::Google::Cloud::AIPlatform::V1::DedicatedResources DedicatedResources} or
               #     {::Google::Cloud::AIPlatform::V1::AutomaticResources AutomaticResources}
+              #     * `required_replica_count` in
+              #     {::Google::Cloud::AIPlatform::V1::DedicatedResources DedicatedResources}
               #     * {::Google::Cloud::AIPlatform::V1::DedicatedResources#autoscaling_metric_specs autoscaling_metric_specs}
               #     * `disable_container_logging` (v1 only)
               #     * `enable_container_logging` (v1beta1 only)

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

@@ -402,6 +402,86 @@ module Google
               raise ::Google::Cloud::Error.from_error(e)
             end
 
+            ##
+            # Bidirectional streaming RPC to directly write to feature values in a
+            # feature view. Requests may not have a one-to-one mapping to responses and
+            # responses may be returned out-of-order to reduce latency.
+            #
+            # @param request [::Gapic::StreamInput, ::Enumerable<::Google::Cloud::AIPlatform::V1::FeatureViewDirectWriteRequest, ::Hash>]
+            #   An enumerable of {::Google::Cloud::AIPlatform::V1::FeatureViewDirectWriteRequest} instances.
+            # @param options [::Gapic::CallOptions, ::Hash]
+            #   Overrides the default settings for this call, e.g, timeout, retries, etc. Optional.
+            #
+            # @yield [response, operation] Access the result along with the RPC operation
+            # @yieldparam response [::Enumerable<::Google::Cloud::AIPlatform::V1::FeatureViewDirectWriteResponse>]
+            # @yieldparam operation [::GRPC::ActiveCall::Operation]
+            #
+            # @return [::Enumerable<::Google::Cloud::AIPlatform::V1::FeatureViewDirectWriteResponse>]
+            #
+            # @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::FeatureOnlineStoreService::Client.new
+            #
+            #   # Create an input stream.
+            #   input = Gapic::StreamInput.new
+            #
+            #   # Call the feature_view_direct_write method to start streaming.
+            #   output = client.feature_view_direct_write input
+            #
+            #   # Send requests on the stream. For each request object, set fields by
+            #   # passing keyword arguments. Be sure to close the stream when done.
+            #   input << Google::Cloud::AIPlatform::V1::FeatureViewDirectWriteRequest.new
+            #   input << Google::Cloud::AIPlatform::V1::FeatureViewDirectWriteRequest.new
+            #   input.close
+            #
+            #   # The returned object is a streamed enumerable yielding elements of type
+            #   # ::Google::Cloud::AIPlatform::V1::FeatureViewDirectWriteResponse
+            #   output.each do |current_response|
+            #     p current_response
+            #   end
+            #
+            def feature_view_direct_write request, options = nil
+              unless request.is_a? ::Enumerable
+                raise ::ArgumentError, "request must be an Enumerable" unless request.respond_to? :to_enum
+                request = request.to_enum
+              end
+
+              request = request.lazy.map do |req|
+                ::Gapic::Protobuf.coerce req, to: ::Google::Cloud::AIPlatform::V1::FeatureViewDirectWriteRequest
+              end
+
+              # 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.feature_view_direct_write.metadata.to_h
+
+              # Set x-goog-api-client, x-goog-user-project and x-goog-api-version 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-api-version"] = API_VERSION unless API_VERSION.empty?
+              metadata[:"x-goog-user-project"] = @quota_project_id if @quota_project_id
+
+              options.apply_defaults timeout:      @config.rpcs.feature_view_direct_write.timeout,
+                                     metadata:     metadata,
+                                     retry_policy: @config.rpcs.feature_view_direct_write.retry_policy
+
+              options.apply_defaults timeout:      @config.timeout,
+                                     metadata:     @config.metadata,
+                                     retry_policy: @config.retry_policy
+
+              @feature_online_store_service_stub.call_rpc :feature_view_direct_write, request, options: options do |response, operation|
+                yield response, operation if block_given?
+              end
+            rescue ::GRPC::BadStatus => e
+              raise ::Google::Cloud::Error.from_error(e)
+            end
+
             ##
             # Configuration class for the FeatureOnlineStoreService API.
             #
@@ -578,6 +658,11 @@ module Google
                 # @return [::Gapic::Config::Method]
                 #
                 attr_reader :search_nearest_entities
+                ##
+                # RPC-specific configuration for `feature_view_direct_write`
+                # @return [::Gapic::Config::Method]
+                #
+                attr_reader :feature_view_direct_write
 
                 # @private
                 def initialize parent_rpcs = nil
@@ -585,6 +670,8 @@ module Google
                   @fetch_feature_values = ::Gapic::Config::Method.new fetch_feature_values_config
                   search_nearest_entities_config = parent_rpcs.search_nearest_entities if parent_rpcs.respond_to? :search_nearest_entities
                   @search_nearest_entities = ::Gapic::Config::Method.new search_nearest_entities_config
+                  feature_view_direct_write_config = parent_rpcs.feature_view_direct_write if parent_rpcs.respond_to? :feature_view_direct_write
+                  @feature_view_direct_write = ::Gapic::Config::Method.new feature_view_direct_write_config
 
                   yield self if block_given?
                 end

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

@@ -2545,7 +2545,6 @@ module Google
             #     containing the substring `foo` and description containing the substring
             #     `bar`.
             #
-            #
             #     Besides field queries, the following exact-match filters are
             #     supported. The exact-match filters do not support wildcards. Unlike
             #     field-restricted queries, exact-match filters are case-sensitive.

data/lib/google/cloud/ai_platform/v1/featurestore_service/rest/client.rb

@@ -2400,7 +2400,6 @@ module Google
               #     containing the substring `foo` and description containing the substring
               #     `bar`.
               #
-              #
               #     Besides field queries, the following exact-match filters are
               #     supported. The exact-match filters do not support wildcards. Unlike
               #     field-restricted queries, exact-match filters are case-sensitive.

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

@@ -1262,7 +1262,7 @@ module Google
             #   @param options [::Gapic::CallOptions, ::Hash]
             #     Overrides the default settings for this call, e.g, timeout, retries, etc. Optional.
             #
-            # @overload generate_content(model: nil, contents: nil, system_instruction: nil, cached_content: nil, tools: nil, tool_config: nil, labels: nil, safety_settings: nil, generation_config: nil)
+            # @overload generate_content(model: nil, contents: nil, system_instruction: nil, cached_content: nil, tools: nil, tool_config: nil, labels: nil, safety_settings: nil, model_armor_config: nil, generation_config: nil)
             #   Pass arguments to `generate_content` 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).
@@ -1313,6 +1313,9 @@ module Google
             #   @param safety_settings [::Array<::Google::Cloud::AIPlatform::V1::SafetySetting, ::Hash>]
             #     Optional. Per request settings for blocking unsafe content.
             #     Enforced on GenerateContentResponse.candidates.
+            #   @param model_armor_config [::Google::Cloud::AIPlatform::V1::ModelArmorConfig, ::Hash]
+            #     Optional. Settings for prompt and response sanitization using the Model
+            #     Armor service. If supplied, safety_settings must not be supplied.
             #   @param generation_config [::Google::Cloud::AIPlatform::V1::GenerationConfig, ::Hash]
             #     Optional. Generation config.
             #
@@ -1393,7 +1396,7 @@ module Google
             #   @param options [::Gapic::CallOptions, ::Hash]
             #     Overrides the default settings for this call, e.g, timeout, retries, etc. Optional.
             #
-            # @overload stream_generate_content(model: nil, contents: nil, system_instruction: nil, cached_content: nil, tools: nil, tool_config: nil, labels: nil, safety_settings: nil, generation_config: nil)
+            # @overload stream_generate_content(model: nil, contents: nil, system_instruction: nil, cached_content: nil, tools: nil, tool_config: nil, labels: nil, safety_settings: nil, model_armor_config: nil, generation_config: nil)
             #   Pass arguments to `stream_generate_content` 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).
@@ -1444,6 +1447,9 @@ module Google
             #   @param safety_settings [::Array<::Google::Cloud::AIPlatform::V1::SafetySetting, ::Hash>]
             #     Optional. Per request settings for blocking unsafe content.
             #     Enforced on GenerateContentResponse.candidates.
+            #   @param model_armor_config [::Google::Cloud::AIPlatform::V1::ModelArmorConfig, ::Hash]
+            #     Optional. Settings for prompt and response sanitization using the Model
+            #     Armor service. If supplied, safety_settings must not be supplied.
             #   @param generation_config [::Google::Cloud::AIPlatform::V1::GenerationConfig, ::Hash]
             #     Optional. Generation config.
             #

data/lib/google/cloud/ai_platform/v1/prediction_service/paths.rb

@@ -107,6 +107,25 @@ module Google
               "projects/#{project}/locations/#{location}/ragCorpora/#{rag_corpus}"
             end
 
+            ##
+            # Create a fully-qualified Template resource string.
+            #
+            # The resource will be in the following format:
+            #
+            # `projects/{project}/locations/{location}/templates/{template}`
+            #
+            # @param project [String]
+            # @param location [String]
+            # @param template [String]
+            #
+            # @return [::String]
+            def template_path project:, location:, template:
+              raise ::ArgumentError, "project cannot contain /" if project.to_s.include? "/"
+              raise ::ArgumentError, "location cannot contain /" if location.to_s.include? "/"
+
+              "projects/#{project}/locations/#{location}/templates/#{template}"
+            end
+
             extend self
           end
         end

data/lib/google/cloud/ai_platform/v1/prediction_service/rest/client.rb

@@ -897,7 +897,7 @@ module Google
               #   @param options [::Gapic::CallOptions, ::Hash]
               #     Overrides the default settings for this call, e.g, timeout, retries etc. Optional.
               #
-              # @overload generate_content(model: nil, contents: nil, system_instruction: nil, cached_content: nil, tools: nil, tool_config: nil, labels: nil, safety_settings: nil, generation_config: nil)
+              # @overload generate_content(model: nil, contents: nil, system_instruction: nil, cached_content: nil, tools: nil, tool_config: nil, labels: nil, safety_settings: nil, model_armor_config: nil, generation_config: nil)
               #   Pass arguments to `generate_content` 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).
@@ -948,6 +948,9 @@ module Google
               #   @param safety_settings [::Array<::Google::Cloud::AIPlatform::V1::SafetySetting, ::Hash>]
               #     Optional. Per request settings for blocking unsafe content.
               #     Enforced on GenerateContentResponse.candidates.
+              #   @param model_armor_config [::Google::Cloud::AIPlatform::V1::ModelArmorConfig, ::Hash]
+              #     Optional. Settings for prompt and response sanitization using the Model
+              #     Armor service. If supplied, safety_settings must not be supplied.
               #   @param generation_config [::Google::Cloud::AIPlatform::V1::GenerationConfig, ::Hash]
               #     Optional. Generation config.
               # @yield [result, operation] Access the result along with the TransportOperation object
@@ -1021,7 +1024,7 @@ module Google
               #   @param options [::Gapic::CallOptions, ::Hash]
               #     Overrides the default settings for this call, e.g, timeout, retries etc. Optional.
               #
-              # @overload stream_generate_content(model: nil, contents: nil, system_instruction: nil, cached_content: nil, tools: nil, tool_config: nil, labels: nil, safety_settings: nil, generation_config: nil)
+              # @overload stream_generate_content(model: nil, contents: nil, system_instruction: nil, cached_content: nil, tools: nil, tool_config: nil, labels: nil, safety_settings: nil, model_armor_config: nil, generation_config: nil)
               #   Pass arguments to `stream_generate_content` 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).
@@ -1072,6 +1075,9 @@ module Google
               #   @param safety_settings [::Array<::Google::Cloud::AIPlatform::V1::SafetySetting, ::Hash>]
               #     Optional. Per request settings for blocking unsafe content.
               #     Enforced on GenerateContentResponse.candidates.
+              #   @param model_armor_config [::Google::Cloud::AIPlatform::V1::ModelArmorConfig, ::Hash]
+              #     Optional. Settings for prompt and response sanitization using the Model
+              #     Armor service. If supplied, safety_settings must not be supplied.
               #   @param generation_config [::Google::Cloud::AIPlatform::V1::GenerationConfig, ::Hash]
               #     Optional. Generation config.
               # @return [::Enumerable<::Google::Cloud::AIPlatform::V1::GenerateContentResponse>]

data/lib/google/cloud/ai_platform/v1/reasoning_engine_service/paths.rb

@@ -41,6 +41,25 @@ module Google
               "projects/#{project}/locations/#{location}"
             end
 
+            ##
+            # Create a fully-qualified NetworkAttachment resource string.
+            #
+            # The resource will be in the following format:
+            #
+            # `projects/{project}/regions/{region}/networkAttachments/{networkattachment}`
+            #
+            # @param project [String]
+            # @param region [String]
+            # @param networkattachment [String]
+            #
+            # @return [::String]
+            def network_attachment_path project:, region:, networkattachment:
+              raise ::ArgumentError, "project cannot contain /" if project.to_s.include? "/"
+              raise ::ArgumentError, "region cannot contain /" if region.to_s.include? "/"
+
+              "projects/#{project}/regions/#{region}/networkAttachments/#{networkattachment}"
+            end
+
             ##
             # Create a fully-qualified ReasoningEngine resource string.
             #

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

@@ -21,7 +21,7 @@ module Google
   module Cloud
     module AIPlatform
       module V1
-        VERSION = "1.18.0"
+        VERSION = "1.20.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*\xf6\x02\n\x0f\x41\x63\x63\x65leratorType\x12 \n\x1c\x41\x43\x43\x45LERATOR_TYPE_UNSPECIFIED\x10\x00\x12\x18\n\x10NVIDIA_TESLA_K80\x10\x01\x1a\x02\x08\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\x19\n\x15NVIDIA_H100_MEGA_80GB\x10\x0e\x12\x15\n\x11NVIDIA_H200_141GB\x10\x0f\x12\x0f\n\x0bNVIDIA_B200\x10\x10\x12\n\n\x06TPU_V2\x10\x06\x12\n\n\x06TPU_V3\x10\x07\x12\x0e\n\nTPU_V4_POD\x10\n\x12\x12\n\x0eTPU_V5_LITEPOD\x10\x0c\x42\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*\x88\x03\n\x0f\x41\x63\x63\x65leratorType\x12 \n\x1c\x41\x43\x43\x45LERATOR_TYPE_UNSPECIFIED\x10\x00\x12\x18\n\x10NVIDIA_TESLA_K80\x10\x01\x1a\x02\x08\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\x19\n\x15NVIDIA_H100_MEGA_80GB\x10\x0e\x12\x15\n\x11NVIDIA_H200_141GB\x10\x0f\x12\x0f\n\x0bNVIDIA_B200\x10\x10\x12\x10\n\x0cNVIDIA_GB200\x10\x11\x12\n\n\x06TPU_V2\x10\x06\x12\n\n\x06TPU_V3\x10\x07\x12\x0e\n\nTPU_V4_POD\x10\n\x12\x12\n\x0eTPU_V5_LITEPOD\x10\x0c\x42\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

@@ -5,6 +5,7 @@
 require 'google/protobuf'
 
 require 'google/api/field_behavior_pb'
+require 'google/api/resource_pb'
 require 'google/cloud/aiplatform/v1/openapi_pb'
 require 'google/cloud/aiplatform/v1/tool_pb'
 require 'google/cloud/aiplatform/v1/vertex_rag_data_pb'
@@ -13,7 +14,7 @@ require 'google/protobuf/struct_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/openapi.proto\x1a%google/cloud/aiplatform/v1/tool.proto\x1a\x30google/cloud/aiplatform/v1/vertex_rag_data.proto\x1a\x1egoogle/protobuf/duration.proto\x1a\x1cgoogle/protobuf/struct.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\"\xec\x04\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\x12J\n\x0f\x65xecutable_code\x18\x08 \x01(\x0b\x32*.google.cloud.aiplatform.v1.ExecutableCodeB\x03\xe0\x41\x01H\x00\x12U\n\x15\x63ode_execution_result\x18\t \x01(\x0b\x32/.google.cloud.aiplatform.v1.CodeExecutionResultB\x03\xe0\x41\x01H\x00\x12\x16\n\x07thought\x18\n \x01(\x08\x42\x03\xe0\x41\x01H\x00\x12 \n\x11thought_signature\x18\x0b \x01(\x0c\x42\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\"\xf8\x0c\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\x12#\n\x11response_logprobs\x18\x12 \x01(\x08\x42\x03\xe0\x41\x01H\x05\x88\x01\x01\x12\x1a\n\x08logprobs\x18\x07 \x01(\x05\x42\x03\xe0\x41\x01H\x06\x88\x01\x01\x12\"\n\x10presence_penalty\x18\x08 \x01(\x02\x42\x03\xe0\x41\x01H\x07\x88\x01\x01\x12#\n\x11\x66requency_penalty\x18\t \x01(\x02\x42\x03\xe0\x41\x01H\x08\x88\x01\x01\x12\x16\n\x04seed\x18\x0c \x01(\x05\x42\x03\xe0\x41\x01H\t\x88\x01\x01\x12\x1f\n\x12response_mime_type\x18\r \x01(\tB\x03\xe0\x41\x01\x12\x45\n\x0fresponse_schema\x18\x10 \x01(\x0b\x32\".google.cloud.aiplatform.v1.SchemaB\x03\xe0\x41\x01H\n\x88\x01\x01\x12>\n\x14response_json_schema\x18\x1c \x01(\x0b\x32\x16.google.protobuf.ValueB\x03\xe0\x41\x01H\x0b\x88\x01\x01\x12\\\n\x0erouting_config\x18\x11 \x01(\x0b\x32:.google.cloud.aiplatform.v1.GenerationConfig.RoutingConfigB\x03\xe0\x41\x01H\x0c\x88\x01\x01\x12Y\n\x0fthinking_config\x18\x19 \x01(\x0b\x32;.google.cloud.aiplatform.v1.GenerationConfig.ThinkingConfigB\x03\xe0\x41\x01\x1a\xc2\x04\n\rRoutingConfig\x12_\n\tauto_mode\x18\x01 \x01(\x0b\x32J.google.cloud.aiplatform.v1.GenerationConfig.RoutingConfig.AutoRoutingModeH\x00\x12\x63\n\x0bmanual_mode\x18\x02 \x01(\x0b\x32L.google.cloud.aiplatform.v1.GenerationConfig.RoutingConfig.ManualRoutingModeH\x00\x1a\x9b\x02\n\x0f\x41utoRoutingMode\x12\x88\x01\n\x18model_routing_preference\x18\x01 \x01(\x0e\x32\x61.google.cloud.aiplatform.v1.GenerationConfig.RoutingConfig.AutoRoutingMode.ModelRoutingPreferenceH\x00\x88\x01\x01\"`\n\x16ModelRoutingPreference\x12\x0b\n\x07UNKNOWN\x10\x00\x12\x16\n\x12PRIORITIZE_QUALITY\x10\x01\x12\x0c\n\x08\x42\x41LANCED\x10\x02\x12\x13\n\x0fPRIORITIZE_COST\x10\x03\x42\x1b\n\x19_model_routing_preference\x1a;\n\x11ManualRoutingMode\x12\x17\n\nmodel_name\x18\x01 \x01(\tH\x00\x88\x01\x01\x42\r\n\x0b_model_nameB\x10\n\x0erouting_config\x1a\x80\x01\n\x0eThinkingConfig\x12\"\n\x10include_thoughts\x18\x01 \x01(\x08\x42\x03\xe0\x41\x01H\x00\x88\x01\x01\x12!\n\x0fthinking_budget\x18\x03 \x01(\x05\x42\x03\xe0\x41\x01H\x01\x88\x01\x01\x42\x13\n\x11_include_thoughtsB\x12\n\x10_thinking_budgetB\x0e\n\x0c_temperatureB\x08\n\x06_top_pB\x08\n\x06_top_kB\x12\n\x10_candidate_countB\x14\n\x12_max_output_tokensB\x14\n\x12_response_logprobsB\x0b\n\t_logprobsB\x13\n\x11_presence_penaltyB\x14\n\x12_frequency_penaltyB\x07\n\x05_seedB\x12\n\x10_response_schemaB\x17\n\x15_response_json_schemaB\x11\n\x0f_routing_config\"\xeb\x03\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\x12N\n\x06method\x18\x04 \x01(\x0e\x32\x39.google.cloud.aiplatform.v1.SafetySetting.HarmBlockMethodB\x03\xe0\x41\x01\"\x9d\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\x12\x07\n\x03OFF\x10\x05\"S\n\x0fHarmBlockMethod\x12!\n\x1dHARM_BLOCK_METHOD_UNSPECIFIED\x10\x00\x12\x0c\n\x08SEVERITY\x10\x01\x12\x0f\n\x0bPROBABILITY\x10\x02\"\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\"\xcf\x06\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\x12\x12\n\x05score\x18\x08 \x01(\x01\x42\x03\xe0\x41\x03\x12\x19\n\x0c\x61vg_logprobs\x18\t \x01(\x01\x42\x03\xe0\x41\x03\x12H\n\x0flogprobs_result\x18\n \x01(\x0b\x32*.google.cloud.aiplatform.v1.LogprobsResultB\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\x12Q\n\x14url_context_metadata\x18\x0b \x01(\x0b\x32..google.cloud.aiplatform.v1.UrlContextMetadataB\x03\xe0\x41\x03\"\xbc\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\x12\x1b\n\x17MALFORMED_FUNCTION_CALL\x10\tB\x11\n\x0f_finish_message\"X\n\x12UrlContextMetadata\x12\x42\n\x0curl_metadata\x18\x01 \x03(\x0b\x32\'.google.cloud.aiplatform.v1.UrlMetadataB\x03\xe0\x41\x03\"\xfc\x01\n\x0bUrlMetadata\x12\x15\n\rretrieved_url\x18\x01 \x01(\t\x12X\n\x14url_retrieval_status\x18\x02 \x01(\x0e\x32:.google.cloud.aiplatform.v1.UrlMetadata.UrlRetrievalStatus\"|\n\x12UrlRetrievalStatus\x12$\n URL_RETRIEVAL_STATUS_UNSPECIFIED\x10\x00\x12 \n\x1cURL_RETRIEVAL_STATUS_SUCCESS\x10\x01\x12\x1e\n\x1aURL_RETRIEVAL_STATUS_ERROR\x10\x02\"\x8f\x03\n\x0eLogprobsResult\x12P\n\x0etop_candidates\x18\x01 \x03(\x0b\x32\x38.google.cloud.aiplatform.v1.LogprobsResult.TopCandidates\x12O\n\x11\x63hosen_candidates\x18\x02 \x03(\x0b\x32\x34.google.cloud.aiplatform.v1.LogprobsResult.Candidate\x1a\x7f\n\tCandidate\x12\x12\n\x05token\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x15\n\x08token_id\x18\x03 \x01(\x05H\x01\x88\x01\x01\x12\x1c\n\x0flog_probability\x18\x02 \x01(\x02H\x02\x88\x01\x01\x42\x08\n\x06_tokenB\x0b\n\t_token_idB\x12\n\x10_log_probability\x1aY\n\rTopCandidates\x12H\n\ncandidates\x18\x01 \x03(\x0b\x32\x34.google.cloud.aiplatform.v1.LogprobsResult.Candidate\"g\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\x12\x11\n\x04text\x18\x04 \x01(\tB\x03\xe0\x41\x03\"\xad\x03\n\x0eGroundingChunk\x12=\n\x03web\x18\x01 \x01(\x0b\x32..google.cloud.aiplatform.v1.GroundingChunk.WebH\x00\x12X\n\x11retrieved_context\x18\x02 \x01(\x0b\x32;.google.cloud.aiplatform.v1.GroundingChunk.RetrievedContextH\x00\x1a=\n\x03Web\x12\x10\n\x03uri\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x12\n\x05title\x18\x02 \x01(\tH\x01\x88\x01\x01\x42\x06\n\x04_uriB\x08\n\x06_title\x1a\xb4\x01\n\x10RetrievedContext\x12\x39\n\trag_chunk\x18\x04 \x01(\x0b\x32$.google.cloud.aiplatform.v1.RagChunkH\x00\x12\x10\n\x03uri\x18\x01 \x01(\tH\x01\x88\x01\x01\x12\x12\n\x05title\x18\x02 \x01(\tH\x02\x88\x01\x01\x12\x11\n\x04text\x18\x03 \x01(\tH\x03\x88\x01\x01\x42\x11\n\x0f\x63ontext_detailsB\x06\n\x04_uriB\x08\n\x06_titleB\x07\n\x05_textB\x0c\n\nchunk_type\"\x95\x01\n\x10GroundingSupport\x12\x39\n\x07segment\x18\x01 \x01(\x0b\x32#.google.cloud.aiplatform.v1.SegmentH\x00\x88\x01\x01\x12\x1f\n\x17grounding_chunk_indices\x18\x02 \x03(\x05\x12\x19\n\x11\x63onfidence_scores\x18\x03 \x03(\x02\x42\n\n\x08_segment\"\xa3\x03\n\x11GroundingMetadata\x12\x1f\n\x12web_search_queries\x18\x01 \x03(\tB\x03\xe0\x41\x01\x12R\n\x12search_entry_point\x18\x04 \x01(\x0b\x32,.google.cloud.aiplatform.v1.SearchEntryPointB\x03\xe0\x41\x01H\x00\x88\x01\x01\x12\x44\n\x10grounding_chunks\x18\x05 \x03(\x0b\x32*.google.cloud.aiplatform.v1.GroundingChunk\x12M\n\x12grounding_supports\x18\x06 \x03(\x0b\x32,.google.cloud.aiplatform.v1.GroundingSupportB\x03\xe0\x41\x01\x12V\n\x12retrieval_metadata\x18\x07 \x01(\x0b\x32-.google.cloud.aiplatform.v1.RetrievalMetadataB\x06\xe0\x41\x01\xe0\x41\x03H\x01\x88\x01\x01\x42\x15\n\x13_search_entry_pointB\x15\n\x13_retrieval_metadata\"H\n\x10SearchEntryPoint\x12\x1d\n\x10rendered_content\x18\x01 \x01(\tB\x03\xe0\x41\x01\x12\x15\n\x08sdk_blob\x18\x02 \x01(\x0c\x42\x03\xe0\x41\x01\"G\n\x11RetrievalMetadata\x12\x32\n%google_search_dynamic_retrieval_score\x18\x02 \x01(\x02\x42\x03\xe0\x41\x01\"a\n\x12ModalityTokenCount\x12\x36\n\x08modality\x18\x01 \x01(\x0e\x32$.google.cloud.aiplatform.v1.Modality\x12\x13\n\x0btoken_count\x18\x02 \x01(\x05*\xdb\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\x12%\n\x1dHARM_CATEGORY_CIVIC_INTEGRITY\x10\x05\x1a\x02\x08\x01*]\n\x08Modality\x12\x18\n\x14MODALITY_UNSPECIFIED\x10\x00\x12\x08\n\x04TEXT\x10\x01\x12\t\n\x05IMAGE\x10\x02\x12\t\n\x05VIDEO\x10\x03\x12\t\n\x05\x41UDIO\x10\x04\x12\x0c\n\x08\x44OCUMENT\x10\x05\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\x19google/api/resource.proto\x1a(google/cloud/aiplatform/v1/openapi.proto\x1a%google/cloud/aiplatform/v1/tool.proto\x1a\x30google/cloud/aiplatform/v1/vertex_rag_data.proto\x1a\x1egoogle/protobuf/duration.proto\x1a\x1cgoogle/protobuf/struct.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\"\xe8\x04\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\x12J\n\x0f\x65xecutable_code\x18\x08 \x01(\x0b\x32*.google.cloud.aiplatform.v1.ExecutableCodeB\x03\xe0\x41\x01H\x00\x12U\n\x15\x63ode_execution_result\x18\t \x01(\x0b\x32/.google.cloud.aiplatform.v1.CodeExecutionResultB\x03\xe0\x41\x01H\x00\x12\x14\n\x07thought\x18\n \x01(\x08\x42\x03\xe0\x41\x01\x12\x1e\n\x11thought_signature\x18\x0b \x01(\x0c\x42\x03\xe0\x41\x01\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\"\xf8\x0c\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\x12#\n\x11response_logprobs\x18\x12 \x01(\x08\x42\x03\xe0\x41\x01H\x05\x88\x01\x01\x12\x1a\n\x08logprobs\x18\x07 \x01(\x05\x42\x03\xe0\x41\x01H\x06\x88\x01\x01\x12\"\n\x10presence_penalty\x18\x08 \x01(\x02\x42\x03\xe0\x41\x01H\x07\x88\x01\x01\x12#\n\x11\x66requency_penalty\x18\t \x01(\x02\x42\x03\xe0\x41\x01H\x08\x88\x01\x01\x12\x16\n\x04seed\x18\x0c \x01(\x05\x42\x03\xe0\x41\x01H\t\x88\x01\x01\x12\x1f\n\x12response_mime_type\x18\r \x01(\tB\x03\xe0\x41\x01\x12\x45\n\x0fresponse_schema\x18\x10 \x01(\x0b\x32\".google.cloud.aiplatform.v1.SchemaB\x03\xe0\x41\x01H\n\x88\x01\x01\x12>\n\x14response_json_schema\x18\x1c \x01(\x0b\x32\x16.google.protobuf.ValueB\x03\xe0\x41\x01H\x0b\x88\x01\x01\x12\\\n\x0erouting_config\x18\x11 \x01(\x0b\x32:.google.cloud.aiplatform.v1.GenerationConfig.RoutingConfigB\x03\xe0\x41\x01H\x0c\x88\x01\x01\x12Y\n\x0fthinking_config\x18\x19 \x01(\x0b\x32;.google.cloud.aiplatform.v1.GenerationConfig.ThinkingConfigB\x03\xe0\x41\x01\x1a\xc2\x04\n\rRoutingConfig\x12_\n\tauto_mode\x18\x01 \x01(\x0b\x32J.google.cloud.aiplatform.v1.GenerationConfig.RoutingConfig.AutoRoutingModeH\x00\x12\x63\n\x0bmanual_mode\x18\x02 \x01(\x0b\x32L.google.cloud.aiplatform.v1.GenerationConfig.RoutingConfig.ManualRoutingModeH\x00\x1a\x9b\x02\n\x0f\x41utoRoutingMode\x12\x88\x01\n\x18model_routing_preference\x18\x01 \x01(\x0e\x32\x61.google.cloud.aiplatform.v1.GenerationConfig.RoutingConfig.AutoRoutingMode.ModelRoutingPreferenceH\x00\x88\x01\x01\"`\n\x16ModelRoutingPreference\x12\x0b\n\x07UNKNOWN\x10\x00\x12\x16\n\x12PRIORITIZE_QUALITY\x10\x01\x12\x0c\n\x08\x42\x41LANCED\x10\x02\x12\x13\n\x0fPRIORITIZE_COST\x10\x03\x42\x1b\n\x19_model_routing_preference\x1a;\n\x11ManualRoutingMode\x12\x17\n\nmodel_name\x18\x01 \x01(\tH\x00\x88\x01\x01\x42\r\n\x0b_model_nameB\x10\n\x0erouting_config\x1a\x80\x01\n\x0eThinkingConfig\x12\"\n\x10include_thoughts\x18\x01 \x01(\x08\x42\x03\xe0\x41\x01H\x00\x88\x01\x01\x12!\n\x0fthinking_budget\x18\x03 \x01(\x05\x42\x03\xe0\x41\x01H\x01\x88\x01\x01\x42\x13\n\x11_include_thoughtsB\x12\n\x10_thinking_budgetB\x0e\n\x0c_temperatureB\x08\n\x06_top_pB\x08\n\x06_top_kB\x12\n\x10_candidate_countB\x14\n\x12_max_output_tokensB\x14\n\x12_response_logprobsB\x0b\n\t_logprobsB\x13\n\x11_presence_penaltyB\x14\n\x12_frequency_penaltyB\x07\n\x05_seedB\x12\n\x10_response_schemaB\x17\n\x15_response_json_schemaB\x11\n\x0f_routing_config\"\xeb\x03\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\x12N\n\x06method\x18\x04 \x01(\x0e\x32\x39.google.cloud.aiplatform.v1.SafetySetting.HarmBlockMethodB\x03\xe0\x41\x01\"\x9d\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\x12\x07\n\x03OFF\x10\x05\"S\n\x0fHarmBlockMethod\x12!\n\x1dHARM_BLOCK_METHOD_UNSPECIFIED\x10\x00\x12\x0c\n\x08SEVERITY\x10\x01\x12\x0f\n\x0bPROBABILITY\x10\x02\"\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\"\xe0\x06\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\x12\x12\n\x05score\x18\x08 \x01(\x01\x42\x03\xe0\x41\x03\x12\x19\n\x0c\x61vg_logprobs\x18\t \x01(\x01\x42\x03\xe0\x41\x03\x12H\n\x0flogprobs_result\x18\n \x01(\x0b\x32*.google.cloud.aiplatform.v1.LogprobsResultB\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\x12Q\n\x14url_context_metadata\x18\x0b \x01(\x0b\x32..google.cloud.aiplatform.v1.UrlContextMetadataB\x03\xe0\x41\x03\"\xcd\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\x12\x1b\n\x17MALFORMED_FUNCTION_CALL\x10\t\x12\x0f\n\x0bMODEL_ARMOR\x10\nB\x11\n\x0f_finish_message\"X\n\x12UrlContextMetadata\x12\x42\n\x0curl_metadata\x18\x01 \x03(\x0b\x32\'.google.cloud.aiplatform.v1.UrlMetadataB\x03\xe0\x41\x03\"\xfc\x01\n\x0bUrlMetadata\x12\x15\n\rretrieved_url\x18\x01 \x01(\t\x12X\n\x14url_retrieval_status\x18\x02 \x01(\x0e\x32:.google.cloud.aiplatform.v1.UrlMetadata.UrlRetrievalStatus\"|\n\x12UrlRetrievalStatus\x12$\n URL_RETRIEVAL_STATUS_UNSPECIFIED\x10\x00\x12 \n\x1cURL_RETRIEVAL_STATUS_SUCCESS\x10\x01\x12\x1e\n\x1aURL_RETRIEVAL_STATUS_ERROR\x10\x02\"\x8f\x03\n\x0eLogprobsResult\x12P\n\x0etop_candidates\x18\x01 \x03(\x0b\x32\x38.google.cloud.aiplatform.v1.LogprobsResult.TopCandidates\x12O\n\x11\x63hosen_candidates\x18\x02 \x03(\x0b\x32\x34.google.cloud.aiplatform.v1.LogprobsResult.Candidate\x1a\x7f\n\tCandidate\x12\x12\n\x05token\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x15\n\x08token_id\x18\x03 \x01(\x05H\x01\x88\x01\x01\x12\x1c\n\x0flog_probability\x18\x02 \x01(\x02H\x02\x88\x01\x01\x42\x08\n\x06_tokenB\x0b\n\t_token_idB\x12\n\x10_log_probability\x1aY\n\rTopCandidates\x12H\n\ncandidates\x18\x01 \x03(\x0b\x32\x34.google.cloud.aiplatform.v1.LogprobsResult.Candidate\"g\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\x12\x11\n\x04text\x18\x04 \x01(\tB\x03\xe0\x41\x03\"\xa1\x05\n\x0eGroundingChunk\x12=\n\x03web\x18\x01 \x01(\x0b\x32..google.cloud.aiplatform.v1.GroundingChunk.WebH\x00\x12X\n\x11retrieved_context\x18\x02 \x01(\x0b\x32;.google.cloud.aiplatform.v1.GroundingChunk.RetrievedContextH\x00\x12?\n\x04maps\x18\x03 \x01(\x0b\x32/.google.cloud.aiplatform.v1.GroundingChunk.MapsH\x00\x1a=\n\x03Web\x12\x10\n\x03uri\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x12\n\x05title\x18\x02 \x01(\tH\x01\x88\x01\x01\x42\x06\n\x04_uriB\x08\n\x06_title\x1a\xe7\x01\n\x10RetrievedContext\x12\x39\n\trag_chunk\x18\x04 \x01(\x0b\x32$.google.cloud.aiplatform.v1.RagChunkH\x00\x12\x10\n\x03uri\x18\x01 \x01(\tH\x01\x88\x01\x01\x12\x12\n\x05title\x18\x02 \x01(\tH\x02\x88\x01\x01\x12\x11\n\x04text\x18\x03 \x01(\tH\x03\x88\x01\x01\x12\x1f\n\rdocument_name\x18\x06 \x01(\tB\x03\xe0\x41\x03H\x04\x88\x01\x01\x42\x11\n\x0f\x63ontext_detailsB\x06\n\x04_uriB\x08\n\x06_titleB\x07\n\x05_textB\x10\n\x0e_document_name\x1a~\n\x04Maps\x12\x10\n\x03uri\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x12\n\x05title\x18\x02 \x01(\tH\x01\x88\x01\x01\x12\x11\n\x04text\x18\x03 \x01(\tH\x02\x88\x01\x01\x12\x15\n\x08place_id\x18\x04 \x01(\tH\x03\x88\x01\x01\x42\x06\n\x04_uriB\x08\n\x06_titleB\x07\n\x05_textB\x0b\n\t_place_idB\x0c\n\nchunk_type\"\x95\x01\n\x10GroundingSupport\x12\x39\n\x07segment\x18\x01 \x01(\x0b\x32#.google.cloud.aiplatform.v1.SegmentH\x00\x88\x01\x01\x12\x1f\n\x17grounding_chunk_indices\x18\x02 \x03(\x05\x12\x19\n\x11\x63onfidence_scores\x18\x03 \x03(\x02\x42\n\n\x08_segment\"\xff\x03\n\x11GroundingMetadata\x12\x1f\n\x12web_search_queries\x18\x01 \x03(\tB\x03\xe0\x41\x01\x12R\n\x12search_entry_point\x18\x04 \x01(\x0b\x32,.google.cloud.aiplatform.v1.SearchEntryPointB\x03\xe0\x41\x01H\x00\x88\x01\x01\x12\x44\n\x10grounding_chunks\x18\x05 \x03(\x0b\x32*.google.cloud.aiplatform.v1.GroundingChunk\x12M\n\x12grounding_supports\x18\x06 \x03(\x0b\x32,.google.cloud.aiplatform.v1.GroundingSupportB\x03\xe0\x41\x01\x12V\n\x12retrieval_metadata\x18\x07 \x01(\x0b\x32-.google.cloud.aiplatform.v1.RetrievalMetadataB\x06\xe0\x41\x01\xe0\x41\x03H\x01\x88\x01\x01\x12\x35\n google_maps_widget_context_token\x18\x08 \x01(\tB\x06\xe0\x41\x01\xe0\x41\x03H\x02\x88\x01\x01\x42\x15\n\x13_search_entry_pointB\x15\n\x13_retrieval_metadataB#\n!_google_maps_widget_context_token\"H\n\x10SearchEntryPoint\x12\x1d\n\x10rendered_content\x18\x01 \x01(\tB\x03\xe0\x41\x01\x12\x15\n\x08sdk_blob\x18\x02 \x01(\x0c\x42\x03\xe0\x41\x01\"G\n\x11RetrievalMetadata\x12\x32\n%google_search_dynamic_retrieval_score\x18\x02 \x01(\x02\x42\x03\xe0\x41\x01\"\xa8\x01\n\x10ModelArmorConfig\x12H\n\x14prompt_template_name\x18\x01 \x01(\tB*\xe0\x41\x01\xfa\x41$\n\"modelarmor.googleapis.com/Template\x12J\n\x16response_template_name\x18\x02 \x01(\tB*\xe0\x41\x01\xfa\x41$\n\"modelarmor.googleapis.com/Template\"a\n\x12ModalityTokenCount\x12\x36\n\x08modality\x18\x01 \x01(\x0e\x32$.google.cloud.aiplatform.v1.Modality\x12\x13\n\x0btoken_count\x18\x02 \x01(\x05*\xdb\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\x12%\n\x1dHARM_CATEGORY_CIVIC_INTEGRITY\x10\x05\x1a\x02\x08\x01*]\n\x08Modality\x12\x18\n\x14MODALITY_UNSPECIFIED\x10\x00\x12\x08\n\x04TEXT\x10\x01\x12\t\n\x05IMAGE\x10\x02\x12\t\n\x05VIDEO\x10\x03\x12\t\n\x05\x41UDIO\x10\x04\x12\x0c\n\x08\x44OCUMENT\x10\x05\x42\xaf\x02\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::V1\xea\x41\x62\n\"modelarmor.googleapis.com/Template\x12<projects/{project}/locations/{location}/templates/{template}b\x06proto3"
 
 pool = Google::Protobuf::DescriptorPool.generated_pool
 
@@ -80,10 +81,12 @@ module Google
         GroundingChunk = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.GroundingChunk").msgclass
         GroundingChunk::Web = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.GroundingChunk.Web").msgclass
         GroundingChunk::RetrievedContext = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.GroundingChunk.RetrievedContext").msgclass
+        GroundingChunk::Maps = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.GroundingChunk.Maps").msgclass
         GroundingSupport = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.GroundingSupport").msgclass
         GroundingMetadata = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.GroundingMetadata").msgclass
         SearchEntryPoint = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.SearchEntryPoint").msgclass
         RetrievalMetadata = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.RetrievalMetadata").msgclass
+        ModelArmorConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ModelArmorConfig").msgclass
         ModalityTokenCount = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ModalityTokenCount").msgclass
         HarmCategory = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.HarmCategory").enummodule
         Modality = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.Modality").enummodule

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

@@ -0,0 +1,42 @@
+# frozen_string_literal: true
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: google/cloud/aiplatform/v1/deployment_stage.proto
+
+require 'google/protobuf'
+
+
+descriptor_data = "\n1google/cloud/aiplatform/v1/deployment_stage.proto\x12\x1agoogle.cloud.aiplatform.v1*\x81\x02\n\x0f\x44\x65ploymentStage\x12 \n\x1c\x44\x45PLOYMENT_STAGE_UNSPECIFIED\x10\x00\x12\x17\n\x13STARTING_DEPLOYMENT\x10\x05\x12\x13\n\x0fPREPARING_MODEL\x10\x06\x12\x1c\n\x18\x43REATING_SERVING_CLUSTER\x10\x07\x12\x1b\n\x17\x41\x44\x44ING_NODES_TO_CLUSTER\x10\x08\x12\x1b\n\x17GETTING_CONTAINER_IMAGE\x10\t\x12\x19\n\x15STARTING_MODEL_SERVER\x10\x03\x12\x10\n\x0c\x46INISHING_UP\x10\x04\x12\x19\n\x15\x44\x45PLOYMENT_TERMINATED\x10\nB\xd2\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\x14\x44\x65ploymentStageProtoP\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
+
+begin
+  pool.add_serialized_file(descriptor_data)
+rescue TypeError
+  # Compatibility code: will be removed in the next major version.
+  require 'google/protobuf/descriptor_pb'
+  parsed = Google::Protobuf::FileDescriptorProto.decode(descriptor_data)
+  parsed.clear_dependency
+  serialized = parsed.class.encode(parsed)
+  file = pool.add_serialized_file(serialized)
+  warn "Warning: Protobuf detected an import path issue while loading generated file #{__FILE__}"
+  imports = [
+  ]
+  imports.each do |type_name, expected_filename|
+    import_file = pool.lookup(type_name).file_descriptor
+    if import_file.name != expected_filename
+      warn "- #{file.name} imports #{expected_filename}, but that import was loaded as #{import_file.name}"
+    end
+  end
+  warn "Each proto file must use a consistent fully-qualified name."
+  warn "This will become an error in the next major version."
+end
+
+module Google
+  module Cloud
+    module AIPlatform
+      module V1
+        DeploymentStage = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.DeploymentStage").enummodule
+      end
+    end
+  end
+end

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

@@ -8,6 +8,7 @@ require 'google/api/annotations_pb'
 require 'google/api/client_pb'
 require 'google/api/field_behavior_pb'
 require 'google/api/resource_pb'
+require 'google/cloud/aiplatform/v1/deployment_stage_pb'
 require 'google/cloud/aiplatform/v1/endpoint_pb'
 require 'google/cloud/aiplatform/v1/operation_pb'
 require 'google/longrunning/operations_pb'
@@ -15,7 +16,7 @@ require 'google/protobuf/empty_pb'
 require 'google/protobuf/field_mask_pb'
 
 
-descriptor_data = "\n1google/cloud/aiplatform/v1/endpoint_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/endpoint.proto\x1a*google/cloud/aiplatform/v1/operation.proto\x1a#google/longrunning/operations.proto\x1a\x1bgoogle/protobuf/empty.proto\x1a google/protobuf/field_mask.proto\"\xa9\x01\n\x15\x43reateEndpointRequest\x12\x39\n\x06parent\x18\x01 \x01(\tB)\xe0\x41\x02\xfa\x41#\n!locations.googleapis.com/Location\x12;\n\x08\x65ndpoint\x18\x02 \x01(\x0b\x32$.google.cloud.aiplatform.v1.EndpointB\x03\xe0\x41\x02\x12\x18\n\x0b\x65ndpoint_id\x18\x04 \x01(\tB\x03\xe0\x41\x05\"q\n\x1f\x43reateEndpointOperationMetadata\x12N\n\x10generic_metadata\x18\x01 \x01(\x0b\x32\x34.google.cloud.aiplatform.v1.GenericOperationMetadata\"N\n\x12GetEndpointRequest\x12\x38\n\x04name\x18\x01 \x01(\tB*\xe0\x41\x02\xfa\x41$\n\"aiplatform.googleapis.com/Endpoint\"\xdd\x01\n\x14ListEndpointsRequest\x12\x39\n\x06parent\x18\x01 \x01(\tB)\xe0\x41\x02\xfa\x41#\n!locations.googleapis.com/Location\x12\x13\n\x06\x66ilter\x18\x02 \x01(\tB\x03\xe0\x41\x01\x12\x16\n\tpage_size\x18\x03 \x01(\x05\x42\x03\xe0\x41\x01\x12\x17\n\npage_token\x18\x04 \x01(\tB\x03\xe0\x41\x01\x12\x32\n\tread_mask\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.FieldMaskB\x03\xe0\x41\x01\x12\x10\n\x08order_by\x18\x06 \x01(\t\"i\n\x15ListEndpointsResponse\x12\x37\n\tendpoints\x18\x01 \x03(\x0b\x32$.google.cloud.aiplatform.v1.Endpoint\x12\x17\n\x0fnext_page_token\x18\x02 \x01(\t\"\x8a\x01\n\x15UpdateEndpointRequest\x12;\n\x08\x65ndpoint\x18\x01 \x01(\x0b\x32$.google.cloud.aiplatform.v1.EndpointB\x03\xe0\x41\x02\x12\x34\n\x0bupdate_mask\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.FieldMaskB\x03\xe0\x41\x02\"_\n UpdateEndpointLongRunningRequest\x12;\n\x08\x65ndpoint\x18\x01 \x01(\x0b\x32$.google.cloud.aiplatform.v1.EndpointB\x03\xe0\x41\x02\"q\n\x1fUpdateEndpointOperationMetadata\x12N\n\x10generic_metadata\x18\x01 \x01(\x0b\x32\x34.google.cloud.aiplatform.v1.GenericOperationMetadata\"Q\n\x15\x44\x65leteEndpointRequest\x12\x38\n\x04name\x18\x01 \x01(\tB*\xe0\x41\x02\xfa\x41$\n\"aiplatform.googleapis.com/Endpoint\"\xa8\x02\n\x12\x44\x65ployModelRequest\x12<\n\x08\x65ndpoint\x18\x01 \x01(\tB*\xe0\x41\x02\xfa\x41$\n\"aiplatform.googleapis.com/Endpoint\x12\x46\n\x0e\x64\x65ployed_model\x18\x02 \x01(\x0b\x32).google.cloud.aiplatform.v1.DeployedModelB\x03\xe0\x41\x02\x12W\n\rtraffic_split\x18\x03 \x03(\x0b\x32@.google.cloud.aiplatform.v1.DeployModelRequest.TrafficSplitEntry\x1a\x33\n\x11TrafficSplitEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x05:\x02\x38\x01\"X\n\x13\x44\x65ployModelResponse\x12\x41\n\x0e\x64\x65ployed_model\x18\x01 \x01(\x0b\x32).google.cloud.aiplatform.v1.DeployedModel\"n\n\x1c\x44\x65ployModelOperationMetadata\x12N\n\x10generic_metadata\x18\x01 \x01(\x0b\x32\x34.google.cloud.aiplatform.v1.GenericOperationMetadata\"\x84\x02\n\x14UndeployModelRequest\x12<\n\x08\x65ndpoint\x18\x01 \x01(\tB*\xe0\x41\x02\xfa\x41$\n\"aiplatform.googleapis.com/Endpoint\x12\x1e\n\x11\x64\x65ployed_model_id\x18\x02 \x01(\tB\x03\xe0\x41\x02\x12Y\n\rtraffic_split\x18\x03 \x03(\x0b\x32\x42.google.cloud.aiplatform.v1.UndeployModelRequest.TrafficSplitEntry\x1a\x33\n\x11TrafficSplitEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x05:\x02\x38\x01\"\x17\n\x15UndeployModelResponse\"p\n\x1eUndeployModelOperationMetadata\x12N\n\x10generic_metadata\x18\x01 \x01(\x0b\x32\x34.google.cloud.aiplatform.v1.GenericOperationMetadata\"\xd8\x01\n\x1aMutateDeployedModelRequest\x12<\n\x08\x65ndpoint\x18\x01 \x01(\tB*\xe0\x41\x02\xfa\x41$\n\"aiplatform.googleapis.com/Endpoint\x12\x46\n\x0e\x64\x65ployed_model\x18\x02 \x01(\x0b\x32).google.cloud.aiplatform.v1.DeployedModelB\x03\xe0\x41\x02\x12\x34\n\x0bupdate_mask\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.FieldMaskB\x03\xe0\x41\x02\"`\n\x1bMutateDeployedModelResponse\x12\x41\n\x0e\x64\x65ployed_model\x18\x01 \x01(\x0b\x32).google.cloud.aiplatform.v1.DeployedModel\"v\n$MutateDeployedModelOperationMetadata\x12N\n\x10generic_metadata\x18\x01 \x01(\x0b\x32\x34.google.cloud.aiplatform.v1.GenericOperationMetadata2\xad\x11\n\x0f\x45ndpointService\x12\x82\x02\n\x0e\x43reateEndpoint\x12\x31.google.cloud.aiplatform.v1.CreateEndpointRequest\x1a\x1d.google.longrunning.Operation\"\x9d\x01\xca\x41+\n\x08\x45ndpoint\x12\x1f\x43reateEndpointOperationMetadata\xda\x41\x0fparent,endpoint\xda\x41\x1bparent,endpoint,endpoint_id\x82\xd3\xe4\x93\x02\x39\"-/v1/{parent=projects/*/locations/*}/endpoints:\x08\x65ndpoint\x12\xa1\x01\n\x0bGetEndpoint\x12..google.cloud.aiplatform.v1.GetEndpointRequest\x1a$.google.cloud.aiplatform.v1.Endpoint\"<\xda\x41\x04name\x82\xd3\xe4\x93\x02/\x12-/v1/{name=projects/*/locations/*/endpoints/*}\x12\xb4\x01\n\rListEndpoints\x12\x30.google.cloud.aiplatform.v1.ListEndpointsRequest\x1a\x31.google.cloud.aiplatform.v1.ListEndpointsResponse\">\xda\x41\x06parent\x82\xd3\xe4\x93\x02/\x12-/v1/{parent=projects/*/locations/*}/endpoints\x12\xca\x01\n\x0eUpdateEndpoint\x12\x31.google.cloud.aiplatform.v1.UpdateEndpointRequest\x1a$.google.cloud.aiplatform.v1.Endpoint\"_\xda\x41\x14\x65ndpoint,update_mask\x82\xd3\xe4\x93\x02\x42\x32\x36/v1/{endpoint.name=projects/*/locations/*/endpoints/*}:\x08\x65ndpoint\x12\xfc\x01\n\x19UpdateEndpointLongRunning\x12<.google.cloud.aiplatform.v1.UpdateEndpointLongRunningRequest\x1a\x1d.google.longrunning.Operation\"\x81\x01\xca\x41+\n\x08\x45ndpoint\x12\x1fUpdateEndpointOperationMetadata\xda\x41\x08\x65ndpoint\x82\xd3\xe4\x93\x02\x42\"=/v1/{endpoint.name=projects/*/locations/*/endpoints/*}:update:\x01*\x12\xd3\x01\n\x0e\x44\x65leteEndpoint\x12\x31.google.cloud.aiplatform.v1.DeleteEndpointRequest\x1a\x1d.google.longrunning.Operation\"o\xca\x41\x30\n\x15google.protobuf.Empty\x12\x17\x44\x65leteOperationMetadata\xda\x41\x04name\x82\xd3\xe4\x93\x02/*-/v1/{name=projects/*/locations/*/endpoints/*}\x12\x85\x02\n\x0b\x44\x65ployModel\x12..google.cloud.aiplatform.v1.DeployModelRequest\x1a\x1d.google.longrunning.Operation\"\xa6\x01\xca\x41\x33\n\x13\x44\x65ployModelResponse\x12\x1c\x44\x65ployModelOperationMetadata\xda\x41%endpoint,deployed_model,traffic_split\x82\xd3\xe4\x93\x02\x42\"=/v1/{endpoint=projects/*/locations/*/endpoints/*}:deployModel:\x01*\x12\x92\x02\n\rUndeployModel\x12\x30.google.cloud.aiplatform.v1.UndeployModelRequest\x1a\x1d.google.longrunning.Operation\"\xaf\x01\xca\x41\x37\n\x15UndeployModelResponse\x12\x1eUndeployModelOperationMetadata\xda\x41(endpoint,deployed_model_id,traffic_split\x82\xd3\xe4\x93\x02\x44\"?/v1/{endpoint=projects/*/locations/*/endpoints/*}:undeployModel:\x01*\x12\xab\x02\n\x13MutateDeployedModel\x12\x36.google.cloud.aiplatform.v1.MutateDeployedModelRequest\x1a\x1d.google.longrunning.Operation\"\xbc\x01\xca\x41\x43\n\x1bMutateDeployedModelResponse\x12$MutateDeployedModelOperationMetadata\xda\x41#endpoint,deployed_model,update_mask\x82\xd3\xe4\x93\x02J\"E/v1/{endpoint=projects/*/locations/*/endpoints/*}:mutateDeployedModel:\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\x14\x45ndpointServiceProtoP\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/endpoint_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\x31google/cloud/aiplatform/v1/deployment_stage.proto\x1a)google/cloud/aiplatform/v1/endpoint.proto\x1a*google/cloud/aiplatform/v1/operation.proto\x1a#google/longrunning/operations.proto\x1a\x1bgoogle/protobuf/empty.proto\x1a google/protobuf/field_mask.proto\"\xa9\x01\n\x15\x43reateEndpointRequest\x12\x39\n\x06parent\x18\x01 \x01(\tB)\xe0\x41\x02\xfa\x41#\n!locations.googleapis.com/Location\x12;\n\x08\x65ndpoint\x18\x02 \x01(\x0b\x32$.google.cloud.aiplatform.v1.EndpointB\x03\xe0\x41\x02\x12\x18\n\x0b\x65ndpoint_id\x18\x04 \x01(\tB\x03\xe0\x41\x05\"\xbd\x01\n\x1f\x43reateEndpointOperationMetadata\x12N\n\x10generic_metadata\x18\x01 \x01(\x0b\x32\x34.google.cloud.aiplatform.v1.GenericOperationMetadata\x12J\n\x10\x64\x65ployment_stage\x18\x02 \x01(\x0e\x32+.google.cloud.aiplatform.v1.DeploymentStageB\x03\xe0\x41\x03\"N\n\x12GetEndpointRequest\x12\x38\n\x04name\x18\x01 \x01(\tB*\xe0\x41\x02\xfa\x41$\n\"aiplatform.googleapis.com/Endpoint\"\xdd\x01\n\x14ListEndpointsRequest\x12\x39\n\x06parent\x18\x01 \x01(\tB)\xe0\x41\x02\xfa\x41#\n!locations.googleapis.com/Location\x12\x13\n\x06\x66ilter\x18\x02 \x01(\tB\x03\xe0\x41\x01\x12\x16\n\tpage_size\x18\x03 \x01(\x05\x42\x03\xe0\x41\x01\x12\x17\n\npage_token\x18\x04 \x01(\tB\x03\xe0\x41\x01\x12\x32\n\tread_mask\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.FieldMaskB\x03\xe0\x41\x01\x12\x10\n\x08order_by\x18\x06 \x01(\t\"i\n\x15ListEndpointsResponse\x12\x37\n\tendpoints\x18\x01 \x03(\x0b\x32$.google.cloud.aiplatform.v1.Endpoint\x12\x17\n\x0fnext_page_token\x18\x02 \x01(\t\"\x8a\x01\n\x15UpdateEndpointRequest\x12;\n\x08\x65ndpoint\x18\x01 \x01(\x0b\x32$.google.cloud.aiplatform.v1.EndpointB\x03\xe0\x41\x02\x12\x34\n\x0bupdate_mask\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.FieldMaskB\x03\xe0\x41\x02\"_\n UpdateEndpointLongRunningRequest\x12;\n\x08\x65ndpoint\x18\x01 \x01(\x0b\x32$.google.cloud.aiplatform.v1.EndpointB\x03\xe0\x41\x02\"q\n\x1fUpdateEndpointOperationMetadata\x12N\n\x10generic_metadata\x18\x01 \x01(\x0b\x32\x34.google.cloud.aiplatform.v1.GenericOperationMetadata\"Q\n\x15\x44\x65leteEndpointRequest\x12\x38\n\x04name\x18\x01 \x01(\tB*\xe0\x41\x02\xfa\x41$\n\"aiplatform.googleapis.com/Endpoint\"\xa8\x02\n\x12\x44\x65ployModelRequest\x12<\n\x08\x65ndpoint\x18\x01 \x01(\tB*\xe0\x41\x02\xfa\x41$\n\"aiplatform.googleapis.com/Endpoint\x12\x46\n\x0e\x64\x65ployed_model\x18\x02 \x01(\x0b\x32).google.cloud.aiplatform.v1.DeployedModelB\x03\xe0\x41\x02\x12W\n\rtraffic_split\x18\x03 \x03(\x0b\x32@.google.cloud.aiplatform.v1.DeployModelRequest.TrafficSplitEntry\x1a\x33\n\x11TrafficSplitEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x05:\x02\x38\x01\"X\n\x13\x44\x65ployModelResponse\x12\x41\n\x0e\x64\x65ployed_model\x18\x01 \x01(\x0b\x32).google.cloud.aiplatform.v1.DeployedModel\"\xba\x01\n\x1c\x44\x65ployModelOperationMetadata\x12N\n\x10generic_metadata\x18\x01 \x01(\x0b\x32\x34.google.cloud.aiplatform.v1.GenericOperationMetadata\x12J\n\x10\x64\x65ployment_stage\x18\x02 \x01(\x0e\x32+.google.cloud.aiplatform.v1.DeploymentStageB\x03\xe0\x41\x03\"\x84\x02\n\x14UndeployModelRequest\x12<\n\x08\x65ndpoint\x18\x01 \x01(\tB*\xe0\x41\x02\xfa\x41$\n\"aiplatform.googleapis.com/Endpoint\x12\x1e\n\x11\x64\x65ployed_model_id\x18\x02 \x01(\tB\x03\xe0\x41\x02\x12Y\n\rtraffic_split\x18\x03 \x03(\x0b\x32\x42.google.cloud.aiplatform.v1.UndeployModelRequest.TrafficSplitEntry\x1a\x33\n\x11TrafficSplitEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x05:\x02\x38\x01\"\x17\n\x15UndeployModelResponse\"p\n\x1eUndeployModelOperationMetadata\x12N\n\x10generic_metadata\x18\x01 \x01(\x0b\x32\x34.google.cloud.aiplatform.v1.GenericOperationMetadata\"\xd8\x01\n\x1aMutateDeployedModelRequest\x12<\n\x08\x65ndpoint\x18\x01 \x01(\tB*\xe0\x41\x02\xfa\x41$\n\"aiplatform.googleapis.com/Endpoint\x12\x46\n\x0e\x64\x65ployed_model\x18\x02 \x01(\x0b\x32).google.cloud.aiplatform.v1.DeployedModelB\x03\xe0\x41\x02\x12\x34\n\x0bupdate_mask\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.FieldMaskB\x03\xe0\x41\x02\"`\n\x1bMutateDeployedModelResponse\x12\x41\n\x0e\x64\x65ployed_model\x18\x01 \x01(\x0b\x32).google.cloud.aiplatform.v1.DeployedModel\"v\n$MutateDeployedModelOperationMetadata\x12N\n\x10generic_metadata\x18\x01 \x01(\x0b\x32\x34.google.cloud.aiplatform.v1.GenericOperationMetadata2\xad\x11\n\x0f\x45ndpointService\x12\x82\x02\n\x0e\x43reateEndpoint\x12\x31.google.cloud.aiplatform.v1.CreateEndpointRequest\x1a\x1d.google.longrunning.Operation\"\x9d\x01\xca\x41+\n\x08\x45ndpoint\x12\x1f\x43reateEndpointOperationMetadata\xda\x41\x0fparent,endpoint\xda\x41\x1bparent,endpoint,endpoint_id\x82\xd3\xe4\x93\x02\x39\"-/v1/{parent=projects/*/locations/*}/endpoints:\x08\x65ndpoint\x12\xa1\x01\n\x0bGetEndpoint\x12..google.cloud.aiplatform.v1.GetEndpointRequest\x1a$.google.cloud.aiplatform.v1.Endpoint\"<\xda\x41\x04name\x82\xd3\xe4\x93\x02/\x12-/v1/{name=projects/*/locations/*/endpoints/*}\x12\xb4\x01\n\rListEndpoints\x12\x30.google.cloud.aiplatform.v1.ListEndpointsRequest\x1a\x31.google.cloud.aiplatform.v1.ListEndpointsResponse\">\xda\x41\x06parent\x82\xd3\xe4\x93\x02/\x12-/v1/{parent=projects/*/locations/*}/endpoints\x12\xca\x01\n\x0eUpdateEndpoint\x12\x31.google.cloud.aiplatform.v1.UpdateEndpointRequest\x1a$.google.cloud.aiplatform.v1.Endpoint\"_\xda\x41\x14\x65ndpoint,update_mask\x82\xd3\xe4\x93\x02\x42\x32\x36/v1/{endpoint.name=projects/*/locations/*/endpoints/*}:\x08\x65ndpoint\x12\xfc\x01\n\x19UpdateEndpointLongRunning\x12<.google.cloud.aiplatform.v1.UpdateEndpointLongRunningRequest\x1a\x1d.google.longrunning.Operation\"\x81\x01\xca\x41+\n\x08\x45ndpoint\x12\x1fUpdateEndpointOperationMetadata\xda\x41\x08\x65ndpoint\x82\xd3\xe4\x93\x02\x42\"=/v1/{endpoint.name=projects/*/locations/*/endpoints/*}:update:\x01*\x12\xd3\x01\n\x0e\x44\x65leteEndpoint\x12\x31.google.cloud.aiplatform.v1.DeleteEndpointRequest\x1a\x1d.google.longrunning.Operation\"o\xca\x41\x30\n\x15google.protobuf.Empty\x12\x17\x44\x65leteOperationMetadata\xda\x41\x04name\x82\xd3\xe4\x93\x02/*-/v1/{name=projects/*/locations/*/endpoints/*}\x12\x85\x02\n\x0b\x44\x65ployModel\x12..google.cloud.aiplatform.v1.DeployModelRequest\x1a\x1d.google.longrunning.Operation\"\xa6\x01\xca\x41\x33\n\x13\x44\x65ployModelResponse\x12\x1c\x44\x65ployModelOperationMetadata\xda\x41%endpoint,deployed_model,traffic_split\x82\xd3\xe4\x93\x02\x42\"=/v1/{endpoint=projects/*/locations/*/endpoints/*}:deployModel:\x01*\x12\x92\x02\n\rUndeployModel\x12\x30.google.cloud.aiplatform.v1.UndeployModelRequest\x1a\x1d.google.longrunning.Operation\"\xaf\x01\xca\x41\x37\n\x15UndeployModelResponse\x12\x1eUndeployModelOperationMetadata\xda\x41(endpoint,deployed_model_id,traffic_split\x82\xd3\xe4\x93\x02\x44\"?/v1/{endpoint=projects/*/locations/*/endpoints/*}:undeployModel:\x01*\x12\xab\x02\n\x13MutateDeployedModel\x12\x36.google.cloud.aiplatform.v1.MutateDeployedModelRequest\x1a\x1d.google.longrunning.Operation\"\xbc\x01\xca\x41\x43\n\x1bMutateDeployedModelResponse\x12$MutateDeployedModelOperationMetadata\xda\x41#endpoint,deployed_model,update_mask\x82\xd3\xe4\x93\x02J\"E/v1/{endpoint=projects/*/locations/*/endpoints/*}:mutateDeployedModel:\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\x14\x45ndpointServiceProtoP\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/endpoint_service_services_pb.rb

@@ -51,9 +51,9 @@ module Google
             # freeing all resources it's using.
             rpc :UndeployModel, ::Google::Cloud::AIPlatform::V1::UndeployModelRequest, ::Google::Longrunning::Operation
             # Updates an existing deployed model. Updatable fields include
-            # `min_replica_count`, `max_replica_count`, `autoscaling_metric_specs`,
-            # `disable_container_logging` (v1 only), and `enable_container_logging`
-            # (v1beta1 only).
+            # `min_replica_count`, `max_replica_count`, `required_replica_count`,
+            # `autoscaling_metric_specs`, `disable_container_logging` (v1 only), and
+            # `enable_container_logging` (v1beta1 only).
             rpc :MutateDeployedModel, ::Google::Cloud::AIPlatform::V1::MutateDeployedModelRequest, ::Google::Longrunning::Operation
           end
 

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

@@ -10,9 +10,11 @@ require 'google/api/field_behavior_pb'
 require 'google/api/resource_pb'
 require 'google/cloud/aiplatform/v1/featurestore_online_service_pb'
 require 'google/protobuf/struct_pb'
+require 'google/protobuf/timestamp_pb'
+require 'google/rpc/status_pb'
 
 
-descriptor_data = "\n=google/cloud/aiplatform/v1/feature_online_store_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/featurestore_online_service.proto\x1a\x1cgoogle/protobuf/struct.proto\"\xa5\x01\n\x12\x46\x65\x61tureViewDataKey\x12\r\n\x03key\x18\x01 \x01(\tH\x00\x12T\n\rcomposite_key\x18\x02 \x01(\x0b\x32;.google.cloud.aiplatform.v1.FeatureViewDataKey.CompositeKeyH\x00\x1a\x1d\n\x0c\x43ompositeKey\x12\r\n\x05parts\x18\x01 \x03(\tB\x0b\n\tkey_oneof\"\xf4\x01\n\x19\x46\x65tchFeatureValuesRequest\x12\x43\n\x0c\x66\x65\x61ture_view\x18\x01 \x01(\tB-\xe0\x41\x02\xfa\x41\'\n%aiplatform.googleapis.com/FeatureView\x12\x45\n\x08\x64\x61ta_key\x18\x06 \x01(\x0b\x32..google.cloud.aiplatform.v1.FeatureViewDataKeyB\x03\xe0\x41\x01\x12K\n\x0b\x64\x61ta_format\x18\x07 \x01(\x0e\x32\x31.google.cloud.aiplatform.v1.FeatureViewDataFormatB\x03\xe0\x41\x01\"\xfe\x03\n\x1a\x46\x65tchFeatureValuesResponse\x12\x65\n\nkey_values\x18\x03 \x01(\x0b\x32O.google.cloud.aiplatform.v1.FetchFeatureValuesResponse.FeatureNameValuePairListH\x00\x12/\n\x0cproto_struct\x18\x02 \x01(\x0b\x32\x17.google.protobuf.StructH\x00\x12@\n\x08\x64\x61ta_key\x18\x04 \x01(\x0b\x32..google.cloud.aiplatform.v1.FeatureViewDataKey\x1a\xfb\x01\n\x18\x46\x65\x61tureNameValuePairList\x12v\n\x08\x66\x65\x61tures\x18\x01 \x03(\x0b\x32\x64.google.cloud.aiplatform.v1.FetchFeatureValuesResponse.FeatureNameValuePairList.FeatureNameValuePair\x1ag\n\x14\x46\x65\x61tureNameValuePair\x12\x39\n\x05value\x18\x02 \x01(\x0b\x32(.google.cloud.aiplatform.v1.FeatureValueH\x00\x12\x0c\n\x04name\x18\x01 \x01(\tB\x06\n\x04\x64\x61taB\x08\n\x06\x66ormat\"\xa3\x08\n\x14NearestNeighborQuery\x12\x18\n\tentity_id\x18\x01 \x01(\tB\x03\xe0\x41\x01H\x00\x12T\n\tembedding\x18\x02 \x01(\x0b\x32:.google.cloud.aiplatform.v1.NearestNeighborQuery.EmbeddingB\x03\xe0\x41\x01H\x00\x12\x1b\n\x0eneighbor_count\x18\x03 \x01(\x05\x42\x03\xe0\x41\x01\x12Z\n\x0estring_filters\x18\x04 \x03(\x0b\x32=.google.cloud.aiplatform.v1.NearestNeighborQuery.StringFilterB\x03\xe0\x41\x01\x12\\\n\x0fnumeric_filters\x18\x08 \x03(\x0b\x32>.google.cloud.aiplatform.v1.NearestNeighborQuery.NumericFilterB\x03\xe0\x41\x01\x12\x32\n%per_crowding_attribute_neighbor_count\x18\x05 \x01(\x05\x42\x03\xe0\x41\x01\x12T\n\nparameters\x18\x07 \x01(\x0b\x32;.google.cloud.aiplatform.v1.NearestNeighborQuery.ParametersB\x03\xe0\x41\x01\x1a\x1f\n\tEmbedding\x12\x12\n\x05value\x18\x01 \x03(\x02\x42\x03\xe0\x41\x01\x1aV\n\x0cStringFilter\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x19\n\x0c\x61llow_tokens\x18\x02 \x03(\tB\x03\xe0\x41\x01\x12\x18\n\x0b\x64\x65ny_tokens\x18\x03 \x03(\tB\x03\xe0\x41\x01\x1a\xcf\x02\n\rNumericFilter\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\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12]\n\x02op\x18\x05 \x01(\x0e\x32G.google.cloud.aiplatform.v1.NearestNeighborQuery.NumericFilter.OperatorB\x03\xe0\x41\x01H\x01\x88\x01\x01\"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\x05ValueB\x05\n\x03_op\x1a\x63\n\nParameters\x12,\n\x1f\x61pproximate_neighbor_candidates\x18\x01 \x01(\x05\x42\x03\xe0\x41\x01\x12\'\n\x1aleaf_nodes_search_fraction\x18\x02 \x01(\x01\x42\x03\xe0\x41\x01\x42\n\n\x08instance\"\xca\x01\n\x1cSearchNearestEntitiesRequest\x12\x43\n\x0c\x66\x65\x61ture_view\x18\x01 \x01(\tB-\xe0\x41\x02\xfa\x41\'\n%aiplatform.googleapis.com/FeatureView\x12\x44\n\x05query\x18\x02 \x01(\x0b\x32\x30.google.cloud.aiplatform.v1.NearestNeighborQueryB\x03\xe0\x41\x02\x12\x1f\n\x12return_full_entity\x18\x03 \x01(\x08\x42\x03\xe0\x41\x01\"\xe1\x01\n\x10NearestNeighbors\x12H\n\tneighbors\x18\x01 \x03(\x0b\x32\x35.google.cloud.aiplatform.v1.NearestNeighbors.Neighbor\x1a\x82\x01\n\x08Neighbor\x12\x11\n\tentity_id\x18\x01 \x01(\t\x12\x10\n\x08\x64istance\x18\x02 \x01(\x01\x12Q\n\x11\x65ntity_key_values\x18\x03 \x01(\x0b\x32\x36.google.cloud.aiplatform.v1.FetchFeatureValuesResponse\"h\n\x1dSearchNearestEntitiesResponse\x12G\n\x11nearest_neighbors\x18\x01 \x01(\x0b\x32,.google.cloud.aiplatform.v1.NearestNeighbors*b\n\x15\x46\x65\x61tureViewDataFormat\x12(\n$FEATURE_VIEW_DATA_FORMAT_UNSPECIFIED\x10\x00\x12\r\n\tKEY_VALUE\x10\x01\x12\x10\n\x0cPROTO_STRUCT\x10\x02\x32\xf8\x04\n\x19\x46\x65\x61tureOnlineStoreService\x12\x8b\x02\n\x12\x46\x65tchFeatureValues\x12\x35.google.cloud.aiplatform.v1.FetchFeatureValuesRequest\x1a\x36.google.cloud.aiplatform.v1.FetchFeatureValuesResponse\"\x85\x01\xda\x41\x16\x66\x65\x61ture_view, data_key\x82\xd3\xe4\x93\x02\x66\"a/v1/{feature_view=projects/*/locations/*/featureOnlineStores/*/featureViews/*}:fetchFeatureValues:\x01*\x12\xfd\x01\n\x15SearchNearestEntities\x12\x38.google.cloud.aiplatform.v1.SearchNearestEntitiesRequest\x1a\x39.google.cloud.aiplatform.v1.SearchNearestEntitiesResponse\"o\x82\xd3\xe4\x93\x02i\"d/v1/{feature_view=projects/*/locations/*/featureOnlineStores/*/featureViews/*}:searchNearestEntities:\x01*\x1aM\xca\x41\x19\x61iplatform.googleapis.com\xd2\x41.https://www.googleapis.com/auth/cloud-platformB\xdc\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\x1e\x46\x65\x61tureOnlineStoreServiceProtoP\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_online_store_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/featurestore_online_service.proto\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x17google/rpc/status.proto\"\xa5\x01\n\x12\x46\x65\x61tureViewDataKey\x12\r\n\x03key\x18\x01 \x01(\tH\x00\x12T\n\rcomposite_key\x18\x02 \x01(\x0b\x32;.google.cloud.aiplatform.v1.FeatureViewDataKey.CompositeKeyH\x00\x1a\x1d\n\x0c\x43ompositeKey\x12\r\n\x05parts\x18\x01 \x03(\tB\x0b\n\tkey_oneof\"\xf4\x01\n\x19\x46\x65tchFeatureValuesRequest\x12\x43\n\x0c\x66\x65\x61ture_view\x18\x01 \x01(\tB-\xe0\x41\x02\xfa\x41\'\n%aiplatform.googleapis.com/FeatureView\x12\x45\n\x08\x64\x61ta_key\x18\x06 \x01(\x0b\x32..google.cloud.aiplatform.v1.FeatureViewDataKeyB\x03\xe0\x41\x01\x12K\n\x0b\x64\x61ta_format\x18\x07 \x01(\x0e\x32\x31.google.cloud.aiplatform.v1.FeatureViewDataFormatB\x03\xe0\x41\x01\"\xfe\x03\n\x1a\x46\x65tchFeatureValuesResponse\x12\x65\n\nkey_values\x18\x03 \x01(\x0b\x32O.google.cloud.aiplatform.v1.FetchFeatureValuesResponse.FeatureNameValuePairListH\x00\x12/\n\x0cproto_struct\x18\x02 \x01(\x0b\x32\x17.google.protobuf.StructH\x00\x12@\n\x08\x64\x61ta_key\x18\x04 \x01(\x0b\x32..google.cloud.aiplatform.v1.FeatureViewDataKey\x1a\xfb\x01\n\x18\x46\x65\x61tureNameValuePairList\x12v\n\x08\x66\x65\x61tures\x18\x01 \x03(\x0b\x32\x64.google.cloud.aiplatform.v1.FetchFeatureValuesResponse.FeatureNameValuePairList.FeatureNameValuePair\x1ag\n\x14\x46\x65\x61tureNameValuePair\x12\x39\n\x05value\x18\x02 \x01(\x0b\x32(.google.cloud.aiplatform.v1.FeatureValueH\x00\x12\x0c\n\x04name\x18\x01 \x01(\tB\x06\n\x04\x64\x61taB\x08\n\x06\x66ormat\"\xa3\x08\n\x14NearestNeighborQuery\x12\x18\n\tentity_id\x18\x01 \x01(\tB\x03\xe0\x41\x01H\x00\x12T\n\tembedding\x18\x02 \x01(\x0b\x32:.google.cloud.aiplatform.v1.NearestNeighborQuery.EmbeddingB\x03\xe0\x41\x01H\x00\x12\x1b\n\x0eneighbor_count\x18\x03 \x01(\x05\x42\x03\xe0\x41\x01\x12Z\n\x0estring_filters\x18\x04 \x03(\x0b\x32=.google.cloud.aiplatform.v1.NearestNeighborQuery.StringFilterB\x03\xe0\x41\x01\x12\\\n\x0fnumeric_filters\x18\x08 \x03(\x0b\x32>.google.cloud.aiplatform.v1.NearestNeighborQuery.NumericFilterB\x03\xe0\x41\x01\x12\x32\n%per_crowding_attribute_neighbor_count\x18\x05 \x01(\x05\x42\x03\xe0\x41\x01\x12T\n\nparameters\x18\x07 \x01(\x0b\x32;.google.cloud.aiplatform.v1.NearestNeighborQuery.ParametersB\x03\xe0\x41\x01\x1a\x1f\n\tEmbedding\x12\x12\n\x05value\x18\x01 \x03(\x02\x42\x03\xe0\x41\x01\x1aV\n\x0cStringFilter\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x19\n\x0c\x61llow_tokens\x18\x02 \x03(\tB\x03\xe0\x41\x01\x12\x18\n\x0b\x64\x65ny_tokens\x18\x03 \x03(\tB\x03\xe0\x41\x01\x1a\xcf\x02\n\rNumericFilter\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\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12]\n\x02op\x18\x05 \x01(\x0e\x32G.google.cloud.aiplatform.v1.NearestNeighborQuery.NumericFilter.OperatorB\x03\xe0\x41\x01H\x01\x88\x01\x01\"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\x05ValueB\x05\n\x03_op\x1a\x63\n\nParameters\x12,\n\x1f\x61pproximate_neighbor_candidates\x18\x01 \x01(\x05\x42\x03\xe0\x41\x01\x12\'\n\x1aleaf_nodes_search_fraction\x18\x02 \x01(\x01\x42\x03\xe0\x41\x01\x42\n\n\x08instance\"\xca\x01\n\x1cSearchNearestEntitiesRequest\x12\x43\n\x0c\x66\x65\x61ture_view\x18\x01 \x01(\tB-\xe0\x41\x02\xfa\x41\'\n%aiplatform.googleapis.com/FeatureView\x12\x44\n\x05query\x18\x02 \x01(\x0b\x32\x30.google.cloud.aiplatform.v1.NearestNeighborQueryB\x03\xe0\x41\x02\x12\x1f\n\x12return_full_entity\x18\x03 \x01(\x08\x42\x03\xe0\x41\x01\"\xe1\x01\n\x10NearestNeighbors\x12H\n\tneighbors\x18\x01 \x03(\x0b\x32\x35.google.cloud.aiplatform.v1.NearestNeighbors.Neighbor\x1a\x82\x01\n\x08Neighbor\x12\x11\n\tentity_id\x18\x01 \x01(\t\x12\x10\n\x08\x64istance\x18\x02 \x01(\x01\x12Q\n\x11\x65ntity_key_values\x18\x03 \x01(\x0b\x32\x36.google.cloud.aiplatform.v1.FetchFeatureValuesResponse\"h\n\x1dSearchNearestEntitiesResponse\x12G\n\x11nearest_neighbors\x18\x01 \x01(\x0b\x32,.google.cloud.aiplatform.v1.NearestNeighbors\"\x8b\x04\n\x1d\x46\x65\x61tureViewDirectWriteRequest\x12@\n\x0c\x66\x65\x61ture_view\x18\x01 \x01(\tB*\xfa\x41\'\n%aiplatform.googleapis.com/FeatureView\x12{\n\x1b\x64\x61ta_key_and_feature_values\x18\x02 \x03(\x0b\x32Q.google.cloud.aiplatform.v1.FeatureViewDirectWriteRequest.DataKeyAndFeatureValuesB\x03\xe0\x41\x02\x1a\xaa\x02\n\x17\x44\x61taKeyAndFeatureValues\x12@\n\x08\x64\x61ta_key\x18\x01 \x01(\x0b\x32..google.cloud.aiplatform.v1.FeatureViewDataKey\x12k\n\x08\x66\x65\x61tures\x18\x02 \x03(\x0b\x32Y.google.cloud.aiplatform.v1.FeatureViewDirectWriteRequest.DataKeyAndFeatureValues.Feature\x1a`\n\x07\x46\x65\x61ture\x12\x39\n\x05value\x18\x03 \x01(\x0b\x32(.google.cloud.aiplatform.v1.FeatureValueH\x00\x12\x0c\n\x04name\x18\x01 \x01(\tB\x0c\n\ndata_oneof\"\xb8\x02\n\x1e\x46\x65\x61tureViewDirectWriteResponse\x12\"\n\x06status\x18\x01 \x01(\x0b\x32\x12.google.rpc.Status\x12\x61\n\x0fwrite_responses\x18\x02 \x03(\x0b\x32H.google.cloud.aiplatform.v1.FeatureViewDirectWriteResponse.WriteResponse\x1a\x8e\x01\n\rWriteResponse\x12@\n\x08\x64\x61ta_key\x18\x01 \x01(\x0b\x32..google.cloud.aiplatform.v1.FeatureViewDataKey\x12;\n\x17online_store_write_time\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.Timestamp*b\n\x15\x46\x65\x61tureViewDataFormat\x12(\n$FEATURE_VIEW_DATA_FORMAT_UNSPECIFIED\x10\x00\x12\r\n\tKEY_VALUE\x10\x01\x12\x10\n\x0cPROTO_STRUCT\x10\x02\x32\xf5\x06\n\x19\x46\x65\x61tureOnlineStoreService\x12\x8b\x02\n\x12\x46\x65tchFeatureValues\x12\x35.google.cloud.aiplatform.v1.FetchFeatureValuesRequest\x1a\x36.google.cloud.aiplatform.v1.FetchFeatureValuesResponse\"\x85\x01\xda\x41\x16\x66\x65\x61ture_view, data_key\x82\xd3\xe4\x93\x02\x66\"a/v1/{feature_view=projects/*/locations/*/featureOnlineStores/*/featureViews/*}:fetchFeatureValues:\x01*\x12\xfd\x01\n\x15SearchNearestEntities\x12\x38.google.cloud.aiplatform.v1.SearchNearestEntitiesRequest\x1a\x39.google.cloud.aiplatform.v1.SearchNearestEntitiesResponse\"o\x82\xd3\xe4\x93\x02i\"d/v1/{feature_view=projects/*/locations/*/featureOnlineStores/*/featureViews/*}:searchNearestEntities:\x01*\x12\xfa\x01\n\x16\x46\x65\x61tureViewDirectWrite\x12\x39.google.cloud.aiplatform.v1.FeatureViewDirectWriteRequest\x1a:.google.cloud.aiplatform.v1.FeatureViewDirectWriteResponse\"e\x82\xd3\xe4\x93\x02_\"Z/v1/{feature_view=projects/*/locations/*/featureOnlineStores/*/featureViews/*}:directWrite:\x01*(\x01\x30\x01\x1aM\xca\x41\x19\x61iplatform.googleapis.com\xd2\x41.https://www.googleapis.com/auth/cloud-platformB\xdc\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\x1e\x46\x65\x61tureOnlineStoreServiceProtoP\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
 
@@ -29,6 +31,8 @@ rescue TypeError
   imports = [
     ["google.protobuf.Struct", "google/protobuf/struct.proto"],
     ["google.cloud.aiplatform.v1.FeatureValue", "google/cloud/aiplatform/v1/featurestore_online_service.proto"],
+    ["google.rpc.Status", "google/rpc/status.proto"],
+    ["google.protobuf.Timestamp", "google/protobuf/timestamp.proto"],
   ]
   imports.each do |type_name, expected_filename|
     import_file = pool.lookup(type_name).file_descriptor
@@ -60,6 +64,11 @@ module Google
         NearestNeighbors = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.NearestNeighbors").msgclass
         NearestNeighbors::Neighbor = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.NearestNeighbors.Neighbor").msgclass
         SearchNearestEntitiesResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.SearchNearestEntitiesResponse").msgclass
+        FeatureViewDirectWriteRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.FeatureViewDirectWriteRequest").msgclass
+        FeatureViewDirectWriteRequest::DataKeyAndFeatureValues = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.FeatureViewDirectWriteRequest.DataKeyAndFeatureValues").msgclass
+        FeatureViewDirectWriteRequest::DataKeyAndFeatureValues::Feature = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.FeatureViewDirectWriteRequest.DataKeyAndFeatureValues.Feature").msgclass
+        FeatureViewDirectWriteResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.FeatureViewDirectWriteResponse").msgclass
+        FeatureViewDirectWriteResponse::WriteResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.FeatureViewDirectWriteResponse.WriteResponse").msgclass
         FeatureViewDataFormat = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.FeatureViewDataFormat").enummodule
       end
     end

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

@@ -39,6 +39,10 @@ module Google
             # Search only works for indexable feature view; if a feature view isn't
             # indexable, returns Invalid argument response.
             rpc :SearchNearestEntities, ::Google::Cloud::AIPlatform::V1::SearchNearestEntitiesRequest, ::Google::Cloud::AIPlatform::V1::SearchNearestEntitiesResponse
+            # Bidirectional streaming RPC to directly write to feature values in a
+            # feature view. Requests may not have a one-to-one mapping to responses and
+            # responses may be returned out-of-order to reduce latency.
+            rpc :FeatureViewDirectWrite, stream(::Google::Cloud::AIPlatform::V1::FeatureViewDirectWriteRequest), stream(::Google::Cloud::AIPlatform::V1::FeatureViewDirectWriteResponse)
           end
 
           Stub = Service.rpc_stub_class