google-apis-aiplatform_v1 0.52.0 → 0.54.0

data/lib/google/apis/aiplatform_v1/classes.rb

@@ -913,6 +913,11 @@ module Google
         # @return [String]
         attr_accessor :api_key_secret_version
       
+        # The API key string. Either this or `api_key_secret_version` must be set.
+        # Corresponds to the JSON property `apiKeyString`
+        # @return [String]
+        attr_accessor :api_key_string
+      
         def initialize(**args)
            update!(**args)
         end
@@ -920,6 +925,7 @@ module Google
         # Update properties of this object
         def update!(**args)
           @api_key_secret_version = args[:api_key_secret_version] if args.key?(:api_key_secret_version)
+          @api_key_string = args[:api_key_string] if args.key?(:api_key_string)
         end
       end
       
@@ -1260,6 +1266,211 @@ module Google
         end
       end
       
+      # Auth configuration to run the extension.
+      class GoogleCloudAiplatformV1AuthConfig
+        include Google::Apis::Core::Hashable
+      
+        # Config for authentication with API key.
+        # Corresponds to the JSON property `apiKeyConfig`
+        # @return [Google::Apis::AiplatformV1::GoogleCloudAiplatformV1AuthConfigApiKeyConfig]
+        attr_accessor :api_key_config
+      
+        # Type of auth scheme.
+        # Corresponds to the JSON property `authType`
+        # @return [String]
+        attr_accessor :auth_type
+      
+        # Config for Google Service Account Authentication.
+        # Corresponds to the JSON property `googleServiceAccountConfig`
+        # @return [Google::Apis::AiplatformV1::GoogleCloudAiplatformV1AuthConfigGoogleServiceAccountConfig]
+        attr_accessor :google_service_account_config
+      
+        # Config for HTTP Basic Authentication.
+        # Corresponds to the JSON property `httpBasicAuthConfig`
+        # @return [Google::Apis::AiplatformV1::GoogleCloudAiplatformV1AuthConfigHttpBasicAuthConfig]
+        attr_accessor :http_basic_auth_config
+      
+        # Config for user oauth.
+        # Corresponds to the JSON property `oauthConfig`
+        # @return [Google::Apis::AiplatformV1::GoogleCloudAiplatformV1AuthConfigOauthConfig]
+        attr_accessor :oauth_config
+      
+        # Config for user OIDC auth.
+        # Corresponds to the JSON property `oidcConfig`
+        # @return [Google::Apis::AiplatformV1::GoogleCloudAiplatformV1AuthConfigOidcConfig]
+        attr_accessor :oidc_config
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @api_key_config = args[:api_key_config] if args.key?(:api_key_config)
+          @auth_type = args[:auth_type] if args.key?(:auth_type)
+          @google_service_account_config = args[:google_service_account_config] if args.key?(:google_service_account_config)
+          @http_basic_auth_config = args[:http_basic_auth_config] if args.key?(:http_basic_auth_config)
+          @oauth_config = args[:oauth_config] if args.key?(:oauth_config)
+          @oidc_config = args[:oidc_config] if args.key?(:oidc_config)
+        end
+      end
+      
+      # Config for authentication with API key.
+      class GoogleCloudAiplatformV1AuthConfigApiKeyConfig
+        include Google::Apis::Core::Hashable
+      
+        # Optional. The name of the SecretManager secret version resource storing the
+        # API key. Format: `projects/`project`/secrets/`secrete`/versions/`version`` -
+        # If both `api_key_secret` and `api_key_string` are specified, this field takes
+        # precedence over `api_key_string`. - If specified, the `secretmanager.versions.
+        # access` permission should be granted to Vertex AI Extension Service Agent (
+        # https://cloud.google.com/vertex-ai/docs/general/access-control#service-agents)
+        # on the specified resource.
+        # Corresponds to the JSON property `apiKeySecret`
+        # @return [String]
+        attr_accessor :api_key_secret
+      
+        # Optional. The API key to be used in the request directly.
+        # Corresponds to the JSON property `apiKeyString`
+        # @return [String]
+        attr_accessor :api_key_string
+      
+        # Optional. The location of the API key.
+        # Corresponds to the JSON property `httpElementLocation`
+        # @return [String]
+        attr_accessor :http_element_location
+      
+        # Optional. The parameter name of the API key. E.g. If the API request is "https:
+        # //example.com/act?api_key=", "api_key" would be the parameter name.
+        # Corresponds to the JSON property `name`
+        # @return [String]
+        attr_accessor :name
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @api_key_secret = args[:api_key_secret] if args.key?(:api_key_secret)
+          @api_key_string = args[:api_key_string] if args.key?(:api_key_string)
+          @http_element_location = args[:http_element_location] if args.key?(:http_element_location)
+          @name = args[:name] if args.key?(:name)
+        end
+      end
+      
+      # Config for Google Service Account Authentication.
+      class GoogleCloudAiplatformV1AuthConfigGoogleServiceAccountConfig
+        include Google::Apis::Core::Hashable
+      
+        # Optional. The service account that the extension execution service runs as. -
+        # If the service account is specified, the `iam.serviceAccounts.getAccessToken`
+        # permission should be granted to Vertex AI Extension Service Agent (https://
+        # cloud.google.com/vertex-ai/docs/general/access-control#service-agents) on the
+        # specified service account. - If not specified, the Vertex AI Extension Service
+        # Agent will be used to execute the Extension.
+        # Corresponds to the JSON property `serviceAccount`
+        # @return [String]
+        attr_accessor :service_account
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @service_account = args[:service_account] if args.key?(:service_account)
+        end
+      end
+      
+      # Config for HTTP Basic Authentication.
+      class GoogleCloudAiplatformV1AuthConfigHttpBasicAuthConfig
+        include Google::Apis::Core::Hashable
+      
+        # Required. The name of the SecretManager secret version resource storing the
+        # base64 encoded credentials. Format: `projects/`project`/secrets/`secrete`/
+        # versions/`version`` - If specified, the `secretmanager.versions.access`
+        # permission should be granted to Vertex AI Extension Service Agent (https://
+        # cloud.google.com/vertex-ai/docs/general/access-control#service-agents) on the
+        # specified resource.
+        # Corresponds to the JSON property `credentialSecret`
+        # @return [String]
+        attr_accessor :credential_secret
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @credential_secret = args[:credential_secret] if args.key?(:credential_secret)
+        end
+      end
+      
+      # Config for user oauth.
+      class GoogleCloudAiplatformV1AuthConfigOauthConfig
+        include Google::Apis::Core::Hashable
+      
+        # Access token for extension endpoint. Only used to propagate token from [[
+        # ExecuteExtensionRequest.runtime_auth_config]] at request time.
+        # Corresponds to the JSON property `accessToken`
+        # @return [String]
+        attr_accessor :access_token
+      
+        # The service account used to generate access tokens for executing the Extension.
+        # - If the service account is specified, the `iam.serviceAccounts.
+        # getAccessToken` permission should be granted to Vertex AI Extension Service
+        # Agent (https://cloud.google.com/vertex-ai/docs/general/access-control#service-
+        # agents) on the provided service account.
+        # Corresponds to the JSON property `serviceAccount`
+        # @return [String]
+        attr_accessor :service_account
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @access_token = args[:access_token] if args.key?(:access_token)
+          @service_account = args[:service_account] if args.key?(:service_account)
+        end
+      end
+      
+      # Config for user OIDC auth.
+      class GoogleCloudAiplatformV1AuthConfigOidcConfig
+        include Google::Apis::Core::Hashable
+      
+        # OpenID Connect formatted ID token for extension endpoint. Only used to
+        # propagate token from [[ExecuteExtensionRequest.runtime_auth_config]] at
+        # request time.
+        # Corresponds to the JSON property `idToken`
+        # @return [String]
+        attr_accessor :id_token
+      
+        # The service account used to generate an OpenID Connect (OIDC)-compatible JWT
+        # token signed by the Google OIDC Provider (accounts.google.com) for extension
+        # endpoint (https://cloud.google.com/iam/docs/create-short-lived-credentials-
+        # direct#sa-credentials-oidc). - The audience for the token will be set to the
+        # URL in the server url defined in the OpenApi spec. - If the service account is
+        # provided, the service account should grant `iam.serviceAccounts.getOpenIdToken`
+        # permission to Vertex AI Extension Service Agent (https://cloud.google.com/
+        # vertex-ai/docs/general/access-control#service-agents).
+        # Corresponds to the JSON property `serviceAccount`
+        # @return [String]
+        attr_accessor :service_account
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @id_token = args[:id_token] if args.key?(:id_token)
+          @service_account = args[:service_account] if args.key?(:service_account)
+        end
+      end
+      
       # A description of resources that to large degree are decided by Vertex AI, and
       # require only a modest additional configuration. Each Model supporting these
       # resources documents its specific guidelines.
@@ -3180,8 +3391,9 @@ module Google
         end
       end
       
-      # Result of executing the [ExecutableCode]. Always follows a `part` containing
-      # the [ExecutableCode].
+      # Result of executing the [ExecutableCode]. Only generated when using the [
+      # CodeExecution] tool, and always follows a `part` containing the [
+      # ExecutableCode].
       class GoogleCloudAiplatformV1CodeExecutionResult
         include Google::Apis::Core::Hashable
       
@@ -3301,6 +3513,34 @@ module Google
         end
       end
       
+      # Colab image of the runtime.
+      class GoogleCloudAiplatformV1ColabImage
+        include Google::Apis::Core::Hashable
+      
+        # Output only. A human-readable description of the specified colab image release,
+        # populated by the system. Example: "Python 3.10", "Latest - current Python 3.
+        # 11"
+        # Corresponds to the JSON property `description`
+        # @return [String]
+        attr_accessor :description
+      
+        # Optional. The release name of the NotebookRuntime Colab image, e.g. "py310".
+        # If not specified, detault to the latest release.
+        # Corresponds to the JSON property `releaseName`
+        # @return [String]
+        attr_accessor :release_name
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @description = args[:description] if args.key?(:description)
+          @release_name = args[:release_name] if args.key?(:release_name)
+        end
+      end
+      
       # Input for Comet metric.
       class GoogleCloudAiplatformV1CometInput
         include Google::Apis::Core::Hashable
@@ -7946,8 +8186,9 @@ module Google
       end
       
       # Code generated by the model that is meant to be executed, and the result
-      # returned to the model. Generated when using the [FunctionDeclaration] tool and
-      # [FunctionCallingConfig] mode is set to [Mode.CODE].
+      # returned to the model. Generated when using the [CodeExecution] tool, in which
+      # the code will be automatically executed, and a corresponding [
+      # CodeExecutionResult] will also be generated.
       class GoogleCloudAiplatformV1ExecutableCode
         include Google::Apis::Core::Hashable
       
@@ -9245,6 +9486,104 @@ module Google
         end
       end
       
+      # Retrieve from data source powered by external API for grounding. The external
+      # API is not owned by Google, but need to follow the pre-defined API spec.
+      class GoogleCloudAiplatformV1ExternalApi
+        include Google::Apis::Core::Hashable
+      
+        # The generic reusable api auth config. Deprecated. Please use AuthConfig (
+        # google/cloud/aiplatform/master/auth.proto) instead.
+        # Corresponds to the JSON property `apiAuth`
+        # @return [Google::Apis::AiplatformV1::GoogleCloudAiplatformV1ApiAuth]
+        attr_accessor :api_auth
+      
+        # The API spec that the external API implements.
+        # Corresponds to the JSON property `apiSpec`
+        # @return [String]
+        attr_accessor :api_spec
+      
+        # Auth configuration to run the extension.
+        # Corresponds to the JSON property `authConfig`
+        # @return [Google::Apis::AiplatformV1::GoogleCloudAiplatformV1AuthConfig]
+        attr_accessor :auth_config
+      
+        # The search parameters to use for the ELASTIC_SEARCH spec.
+        # Corresponds to the JSON property `elasticSearchParams`
+        # @return [Google::Apis::AiplatformV1::GoogleCloudAiplatformV1ExternalApiElasticSearchParams]
+        attr_accessor :elastic_search_params
+      
+        # The endpoint of the external API. The system will call the API at this
+        # endpoint to retrieve the data for grounding. Example: https://acme.com:443/
+        # search
+        # Corresponds to the JSON property `endpoint`
+        # @return [String]
+        attr_accessor :endpoint
+      
+        # The search parameters to use for SIMPLE_SEARCH spec.
+        # Corresponds to the JSON property `simpleSearchParams`
+        # @return [Google::Apis::AiplatformV1::GoogleCloudAiplatformV1ExternalApiSimpleSearchParams]
+        attr_accessor :simple_search_params
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @api_auth = args[:api_auth] if args.key?(:api_auth)
+          @api_spec = args[:api_spec] if args.key?(:api_spec)
+          @auth_config = args[:auth_config] if args.key?(:auth_config)
+          @elastic_search_params = args[:elastic_search_params] if args.key?(:elastic_search_params)
+          @endpoint = args[:endpoint] if args.key?(:endpoint)
+          @simple_search_params = args[:simple_search_params] if args.key?(:simple_search_params)
+        end
+      end
+      
+      # The search parameters to use for the ELASTIC_SEARCH spec.
+      class GoogleCloudAiplatformV1ExternalApiElasticSearchParams
+        include Google::Apis::Core::Hashable
+      
+        # The ElasticSearch index to use.
+        # Corresponds to the JSON property `index`
+        # @return [String]
+        attr_accessor :index
+      
+        # Optional. Number of hits (chunks) to request. When specified, it is passed to
+        # Elasticsearch as the `num_hits` param.
+        # Corresponds to the JSON property `numHits`
+        # @return [Fixnum]
+        attr_accessor :num_hits
+      
+        # The ElasticSearch search template to use.
+        # Corresponds to the JSON property `searchTemplate`
+        # @return [String]
+        attr_accessor :search_template
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @index = args[:index] if args.key?(:index)
+          @num_hits = args[:num_hits] if args.key?(:num_hits)
+          @search_template = args[:search_template] if args.key?(:search_template)
+        end
+      end
+      
+      # The search parameters to use for SIMPLE_SEARCH spec.
+      class GoogleCloudAiplatformV1ExternalApiSimpleSearchParams
+        include Google::Apis::Core::Hashable
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+        end
+      end
+      
       # The fact used in grounding.
       class GoogleCloudAiplatformV1Fact
         include Google::Apis::Core::Hashable
@@ -12171,6 +12510,13 @@ module Google
         # @return [Fixnum]
         attr_accessor :candidate_count
       
+        # Optional. If enabled, the model will detect emotions and adapt its responses
+        # accordingly.
+        # Corresponds to the JSON property `enableAffectiveDialog`
+        # @return [Boolean]
+        attr_accessor :enable_affective_dialog
+        alias_method :enable_affective_dialog?, :enable_affective_dialog
+      
         # Optional. Frequency penalties.
         # Corresponds to the JSON property `frequencyPenalty`
         # @return [Float]
@@ -12271,6 +12617,7 @@ module Google
         def update!(**args)
           @audio_timestamp = args[:audio_timestamp] if args.key?(:audio_timestamp)
           @candidate_count = args[:candidate_count] if args.key?(:candidate_count)
+          @enable_affective_dialog = args[:enable_affective_dialog] if args.key?(:enable_affective_dialog)
           @frequency_penalty = args[:frequency_penalty] if args.key?(:frequency_penalty)
           @logprobs = args[:logprobs] if args.key?(:logprobs)
           @max_output_tokens = args[:max_output_tokens] if args.key?(:max_output_tokens)
@@ -13335,6 +13682,16 @@ module Google
         # @return [Google::Apis::AiplatformV1::GoogleCloudAiplatformV1RagFileTransformationConfig]
         attr_accessor :rag_file_transformation_config
       
+        # Rebuilds the ANN index to optimize for recall on the imported data. Only
+        # applicable for RagCorpora running on RagManagedDb with `retrieval_strategy`
+        # set to `ANN`. The rebuild will be performed using the existing ANN config set
+        # on the RagCorpus. To change the ANN config, please use the UpdateRagCorpus API.
+        # Default is false, i.e., index is not rebuilt.
+        # Corresponds to the JSON property `rebuildAnnIndex`
+        # @return [Boolean]
+        attr_accessor :rebuild_ann_index
+        alias_method :rebuild_ann_index?, :rebuild_ann_index
+      
         # The SharePointSources to pass to ImportRagFiles.
         # Corresponds to the JSON property `sharePointSources`
         # @return [Google::Apis::AiplatformV1::GoogleCloudAiplatformV1SharePointSources]
@@ -13361,6 +13718,7 @@ module Google
           @partial_failure_gcs_sink = args[:partial_failure_gcs_sink] if args.key?(:partial_failure_gcs_sink)
           @rag_file_parsing_config = args[:rag_file_parsing_config] if args.key?(:rag_file_parsing_config)
           @rag_file_transformation_config = args[:rag_file_transformation_config] if args.key?(:rag_file_transformation_config)
+          @rebuild_ann_index = args[:rebuild_ann_index] if args.key?(:rebuild_ann_index)
           @share_point_sources = args[:share_point_sources] if args.key?(:share_point_sources)
           @slack_source = args[:slack_source] if args.key?(:slack_source)
         end
@@ -20174,6 +20532,11 @@ module Google
       class GoogleCloudAiplatformV1NotebookSoftwareConfig
         include Google::Apis::Core::Hashable
       
+        # Colab image of the runtime.
+        # Corresponds to the JSON property `colabImage`
+        # @return [Google::Apis::AiplatformV1::GoogleCloudAiplatformV1ColabImage]
+        attr_accessor :colab_image
+      
         # Optional. Environment variables to be passed to the container. Maximum limit
         # is 100.
         # Corresponds to the JSON property `env`
@@ -20191,6 +20554,7 @@ module Google
       
         # Update properties of this object
         def update!(**args)
+          @colab_image = args[:colab_image] if args.key?(:colab_image)
           @env = args[:env] if args.key?(:env)
           @post_startup_script_config = args[:post_startup_script_config] if args.key?(:post_startup_script_config)
         end
@@ -20220,6 +20584,18 @@ module Google
       class GoogleCloudAiplatformV1PscAutomationConfig
         include Google::Apis::Core::Hashable
       
+        # Output only. Forwarding rule created by the PSC service automation. Note for
+        # Vector search, use IndexPrivateEndpoint instead.
+        # Corresponds to the JSON property `forwardingRule`
+        # @return [String]
+        attr_accessor :forwarding_rule
+      
+        # Output only. IP address rule created by the PSC service automation. Note for
+        # Vector search, use IndexPrivateEndpoint instead.
+        # Corresponds to the JSON property `ipAddress`
+        # @return [String]
+        attr_accessor :ip_address
+      
         # Required. The full name of the Google Compute Engine [network](https://cloud.
         # google.com/compute/docs/networks-and-firewalls#networks). [Format](https://
         # cloud.google.com/compute/docs/reference/rest/v1/networks/insert): `projects/`
@@ -20240,6 +20616,8 @@ module Google
       
         # Update properties of this object
         def update!(**args)
+          @forwarding_rule = args[:forwarding_rule] if args.key?(:forwarding_rule)
+          @ip_address = args[:ip_address] if args.key?(:ip_address)
           @network = args[:network] if args.key?(:network)
           @project_id = args[:project_id] if args.key?(:project_id)
         end
@@ -20632,15 +21010,17 @@ module Google
       class GoogleCloudAiplatformV1Part
         include Google::Apis::Core::Hashable
       
-        # Result of executing the [ExecutableCode]. Always follows a `part` containing
-        # the [ExecutableCode].
+        # Result of executing the [ExecutableCode]. Only generated when using the [
+        # CodeExecution] tool, and always follows a `part` containing the [
+        # ExecutableCode].
         # Corresponds to the JSON property `codeExecutionResult`
         # @return [Google::Apis::AiplatformV1::GoogleCloudAiplatformV1CodeExecutionResult]
         attr_accessor :code_execution_result
       
         # Code generated by the model that is meant to be executed, and the result
-        # returned to the model. Generated when using the [FunctionDeclaration] tool and
-        # [FunctionCallingConfig] mode is set to [Mode.CODE].
+        # returned to the model. Generated when using the [CodeExecution] tool, in which
+        # the code will be automatically executed, and a corresponding [
+        # CodeExecutionResult] will also be generated.
         # Corresponds to the JSON property `executableCode`
         # @return [Google::Apis::AiplatformV1::GoogleCloudAiplatformV1ExecutableCode]
         attr_accessor :executable_code
@@ -20675,6 +21055,12 @@ module Google
         # @return [String]
         attr_accessor :text
       
+        # Output only. Indicates if the part is thought from the model.
+        # Corresponds to the JSON property `thought`
+        # @return [Boolean]
+        attr_accessor :thought
+        alias_method :thought?, :thought
+      
         # Metadata describes the input video content.
         # Corresponds to the JSON property `videoMetadata`
         # @return [Google::Apis::AiplatformV1::GoogleCloudAiplatformV1VideoMetadata]
@@ -20693,6 +21079,7 @@ module Google
           @function_response = args[:function_response] if args.key?(:function_response)
           @inline_data = args[:inline_data] if args.key?(:inline_data)
           @text = args[:text] if args.key?(:text)
+          @thought = args[:thought] if args.key?(:thought)
           @video_metadata = args[:video_metadata] if args.key?(:video_metadata)
         end
       end
@@ -23953,7 +24340,7 @@ module Google
         # @return [Google::Apis::AiplatformV1::GoogleCloudAiplatformV1RagFileParsingConfigLayoutParser]
         attr_accessor :layout_parser
       
-        # Specifies the advanced parsing for RagFiles.
+        # Specifies the LLM parsing for RagFiles.
         # Corresponds to the JSON property `llmParser`
         # @return [Google::Apis::AiplatformV1::GoogleCloudAiplatformV1RagFileParsingConfigLlmParser]
         attr_accessor :llm_parser
@@ -24002,7 +24389,7 @@ module Google
         end
       end
       
-      # Specifies the advanced parsing for RagFiles.
+      # Specifies the LLM parsing for RagFiles.
       class GoogleCloudAiplatformV1RagFileParsingConfigLlmParser
         include Google::Apis::Core::Hashable
       
@@ -24278,6 +24665,67 @@ module Google
       class GoogleCloudAiplatformV1RagVectorDbConfigRagManagedDb
         include Google::Apis::Core::Hashable
       
+        # Config for ANN search. RagManagedDb uses a tree-based structure to partition
+        # data and facilitate faster searches. As a tradeoff, it requires longer
+        # indexing time and manual triggering of index rebuild via the ImportRagFiles
+        # and UpdateRagCorpus API.
+        # Corresponds to the JSON property `ann`
+        # @return [Google::Apis::AiplatformV1::GoogleCloudAiplatformV1RagVectorDbConfigRagManagedDbAnn]
+        attr_accessor :ann
+      
+        # Config for KNN search.
+        # Corresponds to the JSON property `knn`
+        # @return [Google::Apis::AiplatformV1::GoogleCloudAiplatformV1RagVectorDbConfigRagManagedDbKnn]
+        attr_accessor :knn
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @ann = args[:ann] if args.key?(:ann)
+          @knn = args[:knn] if args.key?(:knn)
+        end
+      end
+      
+      # Config for ANN search. RagManagedDb uses a tree-based structure to partition
+      # data and facilitate faster searches. As a tradeoff, it requires longer
+      # indexing time and manual triggering of index rebuild via the ImportRagFiles
+      # and UpdateRagCorpus API.
+      class GoogleCloudAiplatformV1RagVectorDbConfigRagManagedDbAnn
+        include Google::Apis::Core::Hashable
+      
+        # Number of leaf nodes in the tree-based structure. Each leaf node contains
+        # groups of closely related vectors along with their corresponding centroid.
+        # Recommended value is 10 * sqrt(num of RagFiles in your RagCorpus). Default
+        # value is 500.
+        # Corresponds to the JSON property `leafCount`
+        # @return [Fixnum]
+        attr_accessor :leaf_count
+      
+        # The depth of the tree-based structure. Only depth values of 2 and 3 are
+        # supported. Recommended value is 2 if you have if you have O(10K) files in the
+        # RagCorpus and set this to 3 if more than that. Default value is 2.
+        # Corresponds to the JSON property `treeDepth`
+        # @return [Fixnum]
+        attr_accessor :tree_depth
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @leaf_count = args[:leaf_count] if args.key?(:leaf_count)
+          @tree_depth = args[:tree_depth] if args.key?(:tree_depth)
+        end
+      end
+      
+      # Config for KNN search.
+      class GoogleCloudAiplatformV1RagVectorDbConfigRagManagedDbKnn
+        include Google::Apis::Core::Hashable
+      
         def initialize(**args)
            update!(**args)
         end
@@ -25345,6 +25793,12 @@ module Google
         attr_accessor :disable_attribution
         alias_method :disable_attribution?, :disable_attribution
       
+        # Retrieve from data source powered by external API for grounding. The external
+        # API is not owned by Google, but need to follow the pre-defined API spec.
+        # Corresponds to the JSON property `externalApi`
+        # @return [Google::Apis::AiplatformV1::GoogleCloudAiplatformV1ExternalApi]
+        attr_accessor :external_api
+      
         # Retrieve from Vertex AI Search datastore or engine for grounding. datastore
         # and engine are mutually exclusive. See https://cloud.google.com/products/agent-
         # builder
@@ -25364,6 +25818,7 @@ module Google
         # Update properties of this object
         def update!(**args)
           @disable_attribution = args[:disable_attribution] if args.key?(:disable_attribution)
+          @external_api = args[:external_api] if args.key?(:external_api)
           @vertex_ai_search = args[:vertex_ai_search] if args.key?(:vertex_ai_search)
           @vertex_rag_store = args[:vertex_rag_store] if args.key?(:vertex_rag_store)
         end
@@ -35750,14 +36205,16 @@ module Google
         # @return [Google::Apis::AiplatformV1::GoogleCloudAiplatformV1SupervisedHyperParameters]
         attr_accessor :hyper_parameters
       
-        # Required. Cloud Storage path to file containing training dataset for tuning.
-        # The dataset must be formatted as a JSONL file.
+        # Required. Training dataset used for tuning. The dataset can be specified as
+        # either a Cloud Storage path to a JSONL file or as the resource name of a
+        # Vertex Multimodal Dataset.
         # Corresponds to the JSON property `trainingDatasetUri`
         # @return [String]
         attr_accessor :training_dataset_uri
       
-        # Optional. Cloud Storage path to file containing validation dataset for tuning.
-        # The dataset must be formatted as a JSONL file.
+        # Optional. Validation dataset used for tuning. The dataset can be specified as
+        # either a Cloud Storage path to a JSONL file or as the resource name of a
+        # Vertex Multimodal Dataset.
         # Corresponds to the JSON property `validationDatasetUri`
         # @return [String]
         attr_accessor :validation_dataset_uri
@@ -39105,6 +39562,13 @@ module Google
       class GoogleCloudAiplatformV1VertexAiSearch
         include Google::Apis::Core::Hashable
       
+        # Specifications that define the specific DataStores to be searched, along with
+        # configurations for those data stores. This is only considered for Engines with
+        # multiple data stores. It should only be set if engine is used.
+        # Corresponds to the JSON property `dataStoreSpecs`
+        # @return [Array<Google::Apis::AiplatformV1::GoogleCloudAiplatformV1VertexAiSearchDataStoreSpec>]
+        attr_accessor :data_store_specs
+      
         # Optional. Fully-qualified Vertex AI Search data store resource ID. Format: `
         # projects/`project`/locations/`location`/collections/`collection`/dataStores/`
         # dataStore``
@@ -39136,6 +39600,7 @@ module Google
       
         # Update properties of this object
         def update!(**args)
+          @data_store_specs = args[:data_store_specs] if args.key?(:data_store_specs)
           @datastore = args[:datastore] if args.key?(:datastore)
           @engine = args[:engine] if args.key?(:engine)
           @filter = args[:filter] if args.key?(:filter)
@@ -39143,6 +39608,37 @@ module Google
         end
       end
       
+      # Define data stores within engine to filter on in a search call and
+      # configurations for those data stores. For more information, see https://cloud.
+      # google.com/generative-ai-app-builder/docs/reference/rpc/google.cloud.
+      # discoveryengine.v1#datastorespec
+      class GoogleCloudAiplatformV1VertexAiSearchDataStoreSpec
+        include Google::Apis::Core::Hashable
+      
+        # Full resource name of DataStore, such as Format: `projects/`project`/locations/
+        # `location`/collections/`collection`/dataStores/`dataStore``
+        # Corresponds to the JSON property `dataStore`
+        # @return [String]
+        attr_accessor :data_store
+      
+        # Optional. Filter specification to filter documents in the data store specified
+        # by data_store field. For more information on filtering, see [Filtering](https:/
+        # /cloud.google.com/generative-ai-app-builder/docs/filter-search-metadata)
+        # Corresponds to the JSON property `filter`
+        # @return [String]
+        attr_accessor :filter
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @data_store = args[:data_store] if args.key?(:data_store)
+          @filter = args[:filter] if args.key?(:filter)
+        end
+      end
+      
       # Config for the Vertex AI Search.
       class GoogleCloudAiplatformV1VertexAiSearchConfig
         include Google::Apis::Core::Hashable