google-cloud-ai_platform-v1 0.48.0 → 0.55.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 1a2575dd0d4500a8eecf92ce9e1c7176a51effffc1957f000d124ffdea20cef9
-  data.tar.gz: 826b04b88754cfaf266530b6573764aad6d8e42ef3a89e596842fc806c4e626f
+  metadata.gz: 292afb58f3cdf4ff9b22dbbc1ca24184910ed35aff12f3b2788d131d594c4f72
+  data.tar.gz: 7a0fd8ad2d6c4cc1708b03a0ad6cb8463ce464a69c8ff3f26c0c9c215a66d0e4
 SHA512:
-  metadata.gz: a57dcb05530edefa51f6fdb19c77c56e20620221eb065a3720d639285df129719cf7e37929999cdb0037922218206790d19eee0ad2ebe678966162edec54d1e2
-  data.tar.gz: 9f21df7f67c69370fb2db35fc8941b47f03e3f73887c7c59190bad8fcf82c7f5d8e7aa6bbdb4d03cf545b66538636aca35dc43fc95f04471ee86afed165b2961
+  metadata.gz: e29510b89343e267a882cdf9d2f83ffd901e5317c223472d598065a328291db04c9df721aba7461c7b4a6d687a7fb2481c2e812ad5484bb51602ebe71c40f832
+  data.tar.gz: 8fd987400b2a37917694c9dbafebca57f2cde883519e0c02d9de26fd60c826495c3ff1bfd398cbc8bd830bc81e60d9de7d5cfe4727584fb38588f97ab045a1d7

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

@@ -114,6 +114,22 @@ module Google
                 ],
                 body: nil
               ),
+              Gapic::Rest::GrpcTranscoder::HttpBinding.create_with_validation(
+                uri_method: :post,
+                uri_template: "/v1/{resource}:getIamPolicy",
+                matches: [
+                  ["resource", %r{^projects/[^/]+/locations/[^/]+/featureOnlineStores/[^/]+/?$}, false]
+                ],
+                body: nil
+              ),
+              Gapic::Rest::GrpcTranscoder::HttpBinding.create_with_validation(
+                uri_method: :post,
+                uri_template: "/v1/{resource}:getIamPolicy",
+                matches: [
+                  ["resource", %r{^projects/[^/]+/locations/[^/]+/featureOnlineStores/[^/]+/featureViews/[^/]+/?$}, false]
+                ],
+                body: nil
+              ),
               Gapic::Rest::GrpcTranscoder::HttpBinding.create_with_validation(
                 uri_method: :post,
                 uri_template: "/ui/{resource}:getIamPolicy",
@@ -213,6 +229,22 @@ module Google
                 ],
                 body: "*"
               ),
+              Gapic::Rest::GrpcTranscoder::HttpBinding.create_with_validation(
+                uri_method: :post,
+                uri_template: "/v1/{resource}:setIamPolicy",
+                matches: [
+                  ["resource", %r{^projects/[^/]+/locations/[^/]+/featureOnlineStores/[^/]+/?$}, false]
+                ],
+                body: "*"
+              ),
+              Gapic::Rest::GrpcTranscoder::HttpBinding.create_with_validation(
+                uri_method: :post,
+                uri_template: "/v1/{resource}:setIamPolicy",
+                matches: [
+                  ["resource", %r{^projects/[^/]+/locations/[^/]+/featureOnlineStores/[^/]+/featureViews/[^/]+/?$}, false]
+                ],
+                body: "*"
+              ),
               Gapic::Rest::GrpcTranscoder::HttpBinding.create_with_validation(
                 uri_method: :post,
                 uri_template: "/ui/{resource}:setIamPolicy",
@@ -304,6 +336,22 @@ module Google
                 ],
                 body: nil
               ),
+              Gapic::Rest::GrpcTranscoder::HttpBinding.create_with_validation(
+                uri_method: :post,
+                uri_template: "/v1/{resource}:testIamPermissions",
+                matches: [
+                  ["resource", %r{^projects/[^/]+/locations/[^/]+/featureOnlineStores/[^/]+/?$}, false]
+                ],
+                body: nil
+              ),
+              Gapic::Rest::GrpcTranscoder::HttpBinding.create_with_validation(
+                uri_method: :post,
+                uri_template: "/v1/{resource}:testIamPermissions",
+                matches: [
+                  ["resource", %r{^projects/[^/]+/locations/[^/]+/featureOnlineStores/[^/]+/featureViews/[^/]+/?$}, false]
+                ],
+                body: nil
+              ),
               Gapic::Rest::GrpcTranscoder::HttpBinding.create_with_validation(
                 uri_method: :post,
                 uri_template: "/ui/{resource}:testIamPermissions",

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

@@ -151,6 +151,13 @@ module Google
               @quota_project_id = @config.quota_project
               @quota_project_id ||= credentials.quota_project_id if credentials.respond_to? :quota_project_id
 
+              @operations_client = Operations.new do |config|
+                config.credentials = credentials
+                config.quota_project = @quota_project_id
+                config.endpoint = @config.endpoint
+                config.universe_domain = @config.universe_domain
+              end
+
               @gen_ai_tuning_service_stub = ::Gapic::ServiceStub.new(
                 ::Google::Cloud::AIPlatform::V1::GenAiTuningService::Stub,
                 credentials: credentials,
@@ -177,6 +184,13 @@ module Google
               end
             end
 
+            ##
+            # Get the associated client for long-running operations.
+            #
+            # @return [::Google::Cloud::AIPlatform::V1::GenAiTuningService::Operations]
+            #
+            attr_reader :operations_client
+
             ##
             # Get the associated client for mix-in of the Locations.
             #
@@ -568,6 +582,114 @@ module Google
               raise ::Google::Cloud::Error.from_error(e)
             end
 
+            ##
+            # Rebase a TunedModel.
+            # Creates a LongRunningOperation that takes a legacy Tuned GenAI model
+            # Reference and creates a TuningJob based on newly available model.
+            #
+            # @overload rebase_tuned_model(request, options = nil)
+            #   Pass arguments to `rebase_tuned_model` via a request object, either of type
+            #   {::Google::Cloud::AIPlatform::V1::RebaseTunedModelRequest} or an equivalent Hash.
+            #
+            #   @param request [::Google::Cloud::AIPlatform::V1::RebaseTunedModelRequest, ::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 rebase_tuned_model(parent: nil, tuned_model_ref: nil, tuning_job: nil, artifact_destination: nil, deploy_to_same_endpoint: nil)
+            #   Pass arguments to `rebase_tuned_model` 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 resource name of the Location into which to rebase the Model.
+            #     Format: `projects/{project}/locations/{location}`
+            #   @param tuned_model_ref [::Google::Cloud::AIPlatform::V1::TunedModelRef, ::Hash]
+            #     Required. TunedModel reference to retrieve the legacy model information.
+            #   @param tuning_job [::Google::Cloud::AIPlatform::V1::TuningJob, ::Hash]
+            #     Optional. The TuningJob to be updated. Users can use this TuningJob field
+            #     to overwrite tuning configs.
+            #   @param artifact_destination [::Google::Cloud::AIPlatform::V1::GcsDestination, ::Hash]
+            #     Optional. The Google Cloud Storage location to write the artifacts.
+            #   @param deploy_to_same_endpoint [::Boolean]
+            #     Optional. By default, bison to gemini migration will always create new
+            #     model/endpoint, but for gemini-1.0 to gemini-1.5 migration, we default
+            #     deploy to the same endpoint. See details in this Section.
+            #
+            # @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::GenAiTuningService::Client.new
+            #
+            #   # Create a request. To set request fields, pass in keyword arguments.
+            #   request = Google::Cloud::AIPlatform::V1::RebaseTunedModelRequest.new
+            #
+            #   # Call the rebase_tuned_model method.
+            #   result = client.rebase_tuned_model 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 rebase_tuned_model request, options = nil
+              raise ::ArgumentError, "request must be provided" if request.nil?
+
+              request = ::Gapic::Protobuf.coerce request, to: ::Google::Cloud::AIPlatform::V1::RebaseTunedModelRequest
+
+              # 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.rebase_tuned_model.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
+
+              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.rebase_tuned_model.timeout,
+                                     metadata:     metadata,
+                                     retry_policy: @config.rpcs.rebase_tuned_model.retry_policy
+
+              options.apply_defaults timeout:      @config.timeout,
+                                     metadata:     @config.metadata,
+                                     retry_policy: @config.retry_policy
+
+              @gen_ai_tuning_service_stub.call_rpc :rebase_tuned_model, 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 GenAiTuningService API.
             #
@@ -741,6 +863,11 @@ module Google
                 # @return [::Gapic::Config::Method]
                 #
                 attr_reader :cancel_tuning_job
+                ##
+                # RPC-specific configuration for `rebase_tuned_model`
+                # @return [::Gapic::Config::Method]
+                #
+                attr_reader :rebase_tuned_model
 
                 # @private
                 def initialize parent_rpcs = nil
@@ -752,6 +879,8 @@ module Google
                   @list_tuning_jobs = ::Gapic::Config::Method.new list_tuning_jobs_config
                   cancel_tuning_job_config = parent_rpcs.cancel_tuning_job if parent_rpcs.respond_to? :cancel_tuning_job
                   @cancel_tuning_job = ::Gapic::Config::Method.new cancel_tuning_job_config
+                  rebase_tuned_model_config = parent_rpcs.rebase_tuned_model if parent_rpcs.respond_to? :rebase_tuned_model
+                  @rebase_tuned_model = ::Gapic::Config::Method.new rebase_tuned_model_config
 
                   yield self if block_given?
                 end