google-apis-aiplatform_v1beta1 0.46.0 → 0.48.0

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

@@ -934,6 +934,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
@@ -941,6 +946,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
       
@@ -3717,8 +3723,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 GoogleCloudAiplatformV1beta1CodeExecutionResult
         include Google::Apis::Core::Hashable
       
@@ -3838,6 +3845,34 @@ module Google
         end
       end
       
+      # Colab image of the runtime.
+      class GoogleCloudAiplatformV1beta1ColabImage
+        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 GoogleCloudAiplatformV1beta1CometInput
         include Google::Apis::Core::Hashable
@@ -9412,8 +9447,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 GoogleCloudAiplatformV1beta1ExecutableCode
         include Google::Apis::Core::Hashable
       
@@ -10984,6 +11020,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 GoogleCloudAiplatformV1beta1ExternalApi
+        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::AiplatformV1beta1::GoogleCloudAiplatformV1beta1ApiAuth]
+        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::AiplatformV1beta1::GoogleCloudAiplatformV1beta1AuthConfig]
+        attr_accessor :auth_config
+      
+        # The search parameters to use for the ELASTIC_SEARCH spec.
+        # Corresponds to the JSON property `elasticSearchParams`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1ExternalApiElasticSearchParams]
+        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::AiplatformV1beta1::GoogleCloudAiplatformV1beta1ExternalApiSimpleSearchParams]
+        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 GoogleCloudAiplatformV1beta1ExternalApiElasticSearchParams
+        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 GoogleCloudAiplatformV1beta1ExternalApiSimpleSearchParams
+        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 GoogleCloudAiplatformV1beta1Fact
         include Google::Apis::Core::Hashable
@@ -11883,7 +12017,7 @@ module Google
       
         # Optional. Drift threshold. If calculated difference with baseline data larger
         # than threshold, it will be considered as the feature has drift. If not present,
-        # the threshold will be default to 0.3.
+        # the threshold will be default to 0.3. Must be in range [0, 1).
         # Corresponds to the JSON property `driftThreshold`
         # @return [Float]
         attr_accessor :drift_threshold
@@ -14824,6 +14958,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]
@@ -14929,6 +15070,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)
@@ -15960,6 +16102,184 @@ module Google
         end
       end
       
+      # Request message for IndexService.ImportIndex.
+      class GoogleCloudAiplatformV1beta1ImportIndexRequest
+        include Google::Apis::Core::Hashable
+      
+        # Configuration for importing data from an external source.
+        # Corresponds to the JSON property `config`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1ImportIndexRequestConnectorConfig]
+        attr_accessor :config
+      
+        # Optional. If true, completely replace existing index data. Must be true for
+        # streaming update indexes.
+        # Corresponds to the JSON property `isCompleteOverwrite`
+        # @return [Boolean]
+        attr_accessor :is_complete_overwrite
+        alias_method :is_complete_overwrite?, :is_complete_overwrite
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @config = args[:config] if args.key?(:config)
+          @is_complete_overwrite = args[:is_complete_overwrite] if args.key?(:is_complete_overwrite)
+        end
+      end
+      
+      # Configuration for importing data from an external source.
+      class GoogleCloudAiplatformV1beta1ImportIndexRequestConnectorConfig
+        include Google::Apis::Core::Hashable
+      
+        # Configuration for importing data from a BigQuery table.
+        # Corresponds to the JSON property `bigQuerySourceConfig`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1ImportIndexRequestConnectorConfigBigQuerySourceConfig]
+        attr_accessor :big_query_source_config
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @big_query_source_config = args[:big_query_source_config] if args.key?(:big_query_source_config)
+        end
+      end
+      
+      # Configuration for importing data from a BigQuery table.
+      class GoogleCloudAiplatformV1beta1ImportIndexRequestConnectorConfigBigQuerySourceConfig
+        include Google::Apis::Core::Hashable
+      
+        # Mapping of datapoint fields to column names for columnar data sources.
+        # Corresponds to the JSON property `datapointFieldMapping`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1ImportIndexRequestConnectorConfigDatapointFieldMapping]
+        attr_accessor :datapoint_field_mapping
+      
+        # Required. The path to the BigQuery table containing the index data, in the
+        # format of `bq://..`.
+        # Corresponds to the JSON property `tablePath`
+        # @return [String]
+        attr_accessor :table_path
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @datapoint_field_mapping = args[:datapoint_field_mapping] if args.key?(:datapoint_field_mapping)
+          @table_path = args[:table_path] if args.key?(:table_path)
+        end
+      end
+      
+      # Mapping of datapoint fields to column names for columnar data sources.
+      class GoogleCloudAiplatformV1beta1ImportIndexRequestConnectorConfigDatapointFieldMapping
+        include Google::Apis::Core::Hashable
+      
+        # Required. The column with the vector embeddings for each data point.
+        # Corresponds to the JSON property `embeddingColumn`
+        # @return [String]
+        attr_accessor :embedding_column
+      
+        # Required. The column with unique identifiers for each data point.
+        # Corresponds to the JSON property `idColumn`
+        # @return [String]
+        attr_accessor :id_column
+      
+        # Optional. List of columns containing metadata to be included in the index.
+        # Corresponds to the JSON property `metadataColumns`
+        # @return [Array<String>]
+        attr_accessor :metadata_columns
+      
+        # Optional. List of restricts for numeric values.
+        # Corresponds to the JSON property `numericRestricts`
+        # @return [Array<Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1ImportIndexRequestConnectorConfigDatapointFieldMappingNumericRestrict>]
+        attr_accessor :numeric_restricts
+      
+        # Optional. List of restricts for string values.
+        # Corresponds to the JSON property `restricts`
+        # @return [Array<Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1ImportIndexRequestConnectorConfigDatapointFieldMappingRestrict>]
+        attr_accessor :restricts
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @embedding_column = args[:embedding_column] if args.key?(:embedding_column)
+          @id_column = args[:id_column] if args.key?(:id_column)
+          @metadata_columns = args[:metadata_columns] if args.key?(:metadata_columns)
+          @numeric_restricts = args[:numeric_restricts] if args.key?(:numeric_restricts)
+          @restricts = args[:restricts] if args.key?(:restricts)
+        end
+      end
+      
+      # Restrictions on numeric values.
+      class GoogleCloudAiplatformV1beta1ImportIndexRequestConnectorConfigDatapointFieldMappingNumericRestrict
+        include Google::Apis::Core::Hashable
+      
+        # Required. The namespace of the restrict.
+        # Corresponds to the JSON property `namespace`
+        # @return [String]
+        attr_accessor :namespace
+      
+        # Optional. The column containing the numeric value.
+        # Corresponds to the JSON property `valueColumn`
+        # @return [String]
+        attr_accessor :value_column
+      
+        # Required. Numeric type of the restrict. Must be consistent for all datapoints
+        # within the namespace.
+        # Corresponds to the JSON property `valueType`
+        # @return [String]
+        attr_accessor :value_type
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @namespace = args[:namespace] if args.key?(:namespace)
+          @value_column = args[:value_column] if args.key?(:value_column)
+          @value_type = args[:value_type] if args.key?(:value_type)
+        end
+      end
+      
+      # Restrictions on string values.
+      class GoogleCloudAiplatformV1beta1ImportIndexRequestConnectorConfigDatapointFieldMappingRestrict
+        include Google::Apis::Core::Hashable
+      
+        # Optional. The columns containing the allow values.
+        # Corresponds to the JSON property `allowColumn`
+        # @return [Array<String>]
+        attr_accessor :allow_column
+      
+        # Optional. The columns containing the deny values.
+        # Corresponds to the JSON property `denyColumn`
+        # @return [Array<String>]
+        attr_accessor :deny_column
+      
+        # Required. The namespace of the restrict in the index.
+        # Corresponds to the JSON property `namespace`
+        # @return [String]
+        attr_accessor :namespace
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @allow_column = args[:allow_column] if args.key?(:allow_column)
+          @deny_column = args[:deny_column] if args.key?(:deny_column)
+          @namespace = args[:namespace] if args.key?(:namespace)
+        end
+      end
+      
       # Request message for ModelService.ImportModelEvaluation
       class GoogleCloudAiplatformV1beta1ImportModelEvaluationRequest
         include Google::Apis::Core::Hashable
@@ -16052,6 +16372,16 @@ module Google
         # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1RagFileTransformationConfig]
         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::AiplatformV1beta1::GoogleCloudAiplatformV1beta1SharePointSources]
@@ -16080,6 +16410,7 @@ module Google
           @rag_file_chunking_config = args[:rag_file_chunking_config] if args.key?(:rag_file_chunking_config)
           @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
@@ -20928,6 +21259,12 @@ module Google
         # @return [String]
         attr_accessor :display_name
       
+        # Represents a customer-managed encryption key spec that can be applied to a top-
+        # level resource.
+        # Corresponds to the JSON property `encryptionSpec`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1EncryptionSpec]
+        attr_accessor :encryption_spec
+      
         # Specification of Model explanation.
         # Corresponds to the JSON property `explanationSpec`
         # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1ExplanationSpec]
@@ -20997,6 +21334,7 @@ module Google
         def update!(**args)
           @create_time = args[:create_time] if args.key?(:create_time)
           @display_name = args[:display_name] if args.key?(:display_name)
+          @encryption_spec = args[:encryption_spec] if args.key?(:encryption_spec)
           @explanation_spec = args[:explanation_spec] if args.key?(:explanation_spec)
           @model_monitoring_schema = args[:model_monitoring_schema] if args.key?(:model_monitoring_schema)
           @model_monitoring_target = args[:model_monitoring_target] if args.key?(:model_monitoring_target)
@@ -24341,6 +24679,11 @@ module Google
       class GoogleCloudAiplatformV1beta1NotebookSoftwareConfig
         include Google::Apis::Core::Hashable
       
+        # Colab image of the runtime.
+        # Corresponds to the JSON property `colabImage`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1ColabImage]
+        attr_accessor :colab_image
+      
         # Optional. Environment variables to be passed to the container. Maximum limit
         # is 100.
         # Corresponds to the JSON property `env`
@@ -24358,6 +24701,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
@@ -24387,6 +24731,18 @@ module Google
       class GoogleCloudAiplatformV1beta1PscAutomationConfig
         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/`
@@ -24407,6 +24763,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
@@ -24799,15 +25157,17 @@ module Google
       class GoogleCloudAiplatformV1beta1Part
         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::AiplatformV1beta1::GoogleCloudAiplatformV1beta1CodeExecutionResult]
         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::AiplatformV1beta1::GoogleCloudAiplatformV1beta1ExecutableCode]
         attr_accessor :executable_code
@@ -28355,6 +28715,12 @@ module Google
         # @return [String]
         attr_accessor :display_name
       
+        # Represents a customer-managed encryption key spec that can be applied to a top-
+        # level resource.
+        # Corresponds to the JSON property `encryptionSpec`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1EncryptionSpec]
+        attr_accessor :encryption_spec
+      
         # Output only. The resource name of the RagCorpus.
         # Corresponds to the JSON property `name`
         # @return [String]
@@ -28400,6 +28766,7 @@ module Google
           @create_time = args[:create_time] if args.key?(:create_time)
           @description = args[:description] if args.key?(:description)
           @display_name = args[:display_name] if args.key?(:display_name)
+          @encryption_spec = args[:encryption_spec] if args.key?(:encryption_spec)
           @name = args[:name] if args.key?(:name)
           @rag_embedding_model_config = args[:rag_embedding_model_config] if args.key?(:rag_embedding_model_config)
           @rag_files_count = args[:rag_files_count] if args.key?(:rag_files_count)
@@ -28745,7 +29112,7 @@ module Google
         # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1RagFileParsingConfigLayoutParser]
         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::AiplatformV1beta1::GoogleCloudAiplatformV1beta1RagFileParsingConfigLlmParser]
         attr_accessor :llm_parser
@@ -28832,7 +29199,7 @@ module Google
         end
       end
       
-      # Specifies the advanced parsing for RagFiles.
+      # Specifies the LLM parsing for RagFiles.
       class GoogleCloudAiplatformV1beta1RagFileParsingConfigLlmParser
         include Google::Apis::Core::Hashable
       
@@ -29253,6 +29620,67 @@ module Google
       class GoogleCloudAiplatformV1beta1RagVectorDbConfigRagManagedDb
         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::AiplatformV1beta1::GoogleCloudAiplatformV1beta1RagVectorDbConfigRagManagedDbAnn]
+        attr_accessor :ann
+      
+        # Config for KNN search.
+        # Corresponds to the JSON property `knn`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1RagVectorDbConfigRagManagedDbKnn]
+        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 GoogleCloudAiplatformV1beta1RagVectorDbConfigRagManagedDbAnn
+        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 GoogleCloudAiplatformV1beta1RagVectorDbConfigRagManagedDbKnn
+        include Google::Apis::Core::Hashable
+      
         def initialize(**args)
            update!(**args)
         end
@@ -30546,6 +30974,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::AiplatformV1beta1::GoogleCloudAiplatformV1beta1ExternalApi]
+        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
@@ -30565,6 +30999,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
@@ -39356,9 +39791,8 @@ module Google
         # @return [String]
         attr_accessor :display_name
       
-        # Required. Identifier. The resource name of the session. Format: 'projects/`
-        # project`/locations/`location`/reasoningEngines/`reasoning_engine`/sessions/`
-        # session`'.
+        # Identifier. The resource name of the session. Format: 'projects/`project`/
+        # locations/`location`/reasoningEngines/`reasoning_engine`/sessions/`session`'.
         # Corresponds to the JSON property `name`
         # @return [String]
         attr_accessor :name
@@ -39436,9 +39870,9 @@ module Google
         # @return [String]
         attr_accessor :invocation_id
       
-        # Required. Identifier. The resource name of the event. Format:`projects/`
-        # project`/locations/`location`/reasoningEngines/`reasoning_engine`/sessions/`
-        # session`/events/`event``.
+        # Identifier. The resource name of the event. Format:`projects/`project`/
+        # locations/`location`/reasoningEngines/`reasoning_engine`/sessions/`session`/
+        # events/`event``.
         # Corresponds to the JSON property `name`
         # @return [String]
         attr_accessor :name
@@ -41942,14 +42376,16 @@ module Google
         # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1SupervisedHyperParameters]
         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
@@ -45526,6 +45962,13 @@ module Google
       class GoogleCloudAiplatformV1beta1VertexAiSearch
         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::AiplatformV1beta1::GoogleCloudAiplatformV1beta1VertexAiSearchDataStoreSpec>]
+        attr_accessor :data_store_specs
+      
         # Optional. Fully-qualified Vertex AI Search data store resource ID. Format: `
         # projects/`project`/locations/`location`/collections/`collection`/dataStores/`
         # dataStore``
@@ -45557,6 +46000,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)
@@ -45564,6 +46008,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 GoogleCloudAiplatformV1beta1VertexAiSearchDataStoreSpec
+        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 GoogleCloudAiplatformV1beta1VertexAiSearchConfig
         include Google::Apis::Core::Hashable