google-apis-aiplatform_v1beta1 0.50.0 → 0.52.0

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

@@ -3059,7 +3059,8 @@ module Google
       
         # Optional. Display name of the blob. Used to provide a label or filename to
         # distinguish blobs. This field is only returned in PromptMessage for prompt
-        # management. It is not currently used in the Gemini GenerateContent calls.
+        # management. It is currently used in the Gemini GenerateContent calls only when
+        # server side tools (code_execution, google_search, and url_context) are enabled.
         # Corresponds to the JSON property `displayName`
         # @return [String]
         attr_accessor :display_name
@@ -3457,6 +3458,11 @@ module Google
         # @return [Array<Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1SafetyRating>]
         attr_accessor :safety_ratings
       
+        # Metadata related to url context retrieval tool.
+        # Corresponds to the JSON property `urlContextMetadata`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1UrlContextMetadata]
+        attr_accessor :url_context_metadata
+      
         def initialize(**args)
            update!(**args)
         end
@@ -3472,6 +3478,7 @@ module Google
           @index = args[:index] if args.key?(:index)
           @logprobs_result = args[:logprobs_result] if args.key?(:logprobs_result)
           @safety_ratings = args[:safety_ratings] if args.key?(:safety_ratings)
+          @url_context_metadata = args[:url_context_metadata] if args.key?(:url_context_metadata)
         end
       end
       
@@ -7319,12 +7326,12 @@ module Google
         # @return [String]
         attr_accessor :id
       
-        # Required. The resource name of the Model that this is the deployment of. Note
-        # that the Model may be in a different location than the DeployedModel's
-        # Endpoint. The resource name may contain version id or version alias to specify
-        # the version. Example: `projects/`project`/locations/`location`/models/`model`@
-        # 2` or `projects/`project`/locations/`location`/models/`model`@golden` if no
-        # version is specified, the default version will be deployed.
+        # The resource name of the Model that this is the deployment of. Note that the
+        # Model may be in a different location than the DeployedModel's Endpoint. The
+        # resource name may contain version id or version alias to specify the version.
+        # Example: `projects/`project`/locations/`location`/models/`model`@2` or `
+        # projects/`project`/locations/`location`/models/`model`@golden` if no version
+        # is specified, the default version will be deployed.
         # Corresponds to the JSON property `model`
         # @return [String]
         attr_accessor :model
@@ -7828,6 +7835,43 @@ module Google
         end
       end
       
+      # DNS peering configuration. These configurations are used to create DNS peering
+      # zones in the Vertex tenant project VPC, enabling resolution of records within
+      # the specified domain hosted in the target network's Cloud DNS.
+      class GoogleCloudAiplatformV1beta1DnsPeeringConfig
+        include Google::Apis::Core::Hashable
+      
+        # Required. The DNS name suffix of the zone being peered to, e.g., "my-internal-
+        # domain.corp.". Must end with a dot.
+        # Corresponds to the JSON property `domain`
+        # @return [String]
+        attr_accessor :domain
+      
+        # Required. The VPC network name in the target_project where the DNS zone
+        # specified by 'domain' is visible.
+        # Corresponds to the JSON property `targetNetwork`
+        # @return [String]
+        attr_accessor :target_network
+      
+        # Required. The project ID hosting the Cloud DNS managed zone that contains the '
+        # domain'. The Vertex AI Service Agent requires the dns.peer role on this
+        # project.
+        # Corresponds to the JSON property `targetProject`
+        # @return [String]
+        attr_accessor :target_project
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @domain = args[:domain] if args.key?(:domain)
+          @target_network = args[:target_network] if args.key?(:target_network)
+          @target_project = args[:target_project] if args.key?(:target_project)
+        end
+      end
+      
       # A list of double values.
       class GoogleCloudAiplatformV1beta1DoubleArray
         include Google::Apis::Core::Hashable
@@ -13672,8 +13716,9 @@ module Google
       
         # Optional. Display name of the file data. Used to provide a label or filename
         # to distinguish file datas. This field is only returned in PromptMessage for
-        # prompt management. It is not currently used in the Gemini GenerateContent
-        # calls.
+        # prompt management. It is currently used in the Gemini GenerateContent calls
+        # only when server side tools (code_execution, google_search, and url_context)
+        # are enabled.
         # Corresponds to the JSON property `displayName`
         # @return [String]
         attr_accessor :display_name
@@ -14893,6 +14938,132 @@ module Google
         end
       end
       
+      # Request message for MemoryBankService.GenerateMemories.
+      class GoogleCloudAiplatformV1beta1GenerateMemoriesRequest
+        include Google::Apis::Core::Hashable
+      
+        # Defines a direct source of content from which to generate the memories.
+        # Corresponds to the JSON property `directContentsSource`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1GenerateMemoriesRequestDirectContentsSource]
+        attr_accessor :direct_contents_source
+      
+        # Optional. If true, generated memories will not be consolidated with existing
+        # memories; all generated memories will be added as new memories regardless of
+        # whether they are duplicates of or contradictory to existing memories. By
+        # default, memory consolidation is enabled.
+        # Corresponds to the JSON property `disableConsolidation`
+        # @return [Boolean]
+        attr_accessor :disable_consolidation
+        alias_method :disable_consolidation?, :disable_consolidation
+      
+        # Optional. The scope of the memories that should be generated. Memories will be
+        # consolidated across memories with the same scope. Must be provided unless the
+        # scope is defined in the source content. If `scope` is provided, it will
+        # override the scope defined in the source content. Scope values cannot contain
+        # the wildcard character '*'.
+        # Corresponds to the JSON property `scope`
+        # @return [Hash<String,String>]
+        attr_accessor :scope
+      
+        # Defines an Agent Engine Session from which to generate the memories. If `scope`
+        # is not provided, the scope will be extracted from the Session (i.e. `"user_id"
+        # : sesison.user_id`).
+        # Corresponds to the JSON property `vertexSessionSource`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1GenerateMemoriesRequestVertexSessionSource]
+        attr_accessor :vertex_session_source
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @direct_contents_source = args[:direct_contents_source] if args.key?(:direct_contents_source)
+          @disable_consolidation = args[:disable_consolidation] if args.key?(:disable_consolidation)
+          @scope = args[:scope] if args.key?(:scope)
+          @vertex_session_source = args[:vertex_session_source] if args.key?(:vertex_session_source)
+        end
+      end
+      
+      # Defines a direct source of content from which to generate the memories.
+      class GoogleCloudAiplatformV1beta1GenerateMemoriesRequestDirectContentsSource
+        include Google::Apis::Core::Hashable
+      
+        # Required. The source content (i.e. chat history) to generate memories from.
+        # Corresponds to the JSON property `events`
+        # @return [Array<Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1GenerateMemoriesRequestDirectContentsSourceEvent>]
+        attr_accessor :events
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @events = args[:events] if args.key?(:events)
+        end
+      end
+      
+      # A single piece of conversation from which to generate memories.
+      class GoogleCloudAiplatformV1beta1GenerateMemoriesRequestDirectContentsSourceEvent
+        include Google::Apis::Core::Hashable
+      
+        # The base structured datatype containing multi-part content of a message. A `
+        # Content` includes a `role` field designating the producer of the `Content` and
+        # a `parts` field containing multi-part data that contains the content of the
+        # message turn.
+        # Corresponds to the JSON property `content`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1Content]
+        attr_accessor :content
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @content = args[:content] if args.key?(:content)
+        end
+      end
+      
+      # Defines an Agent Engine Session from which to generate the memories. If `scope`
+      # is not provided, the scope will be extracted from the Session (i.e. `"user_id"
+      # : sesison.user_id`).
+      class GoogleCloudAiplatformV1beta1GenerateMemoriesRequestVertexSessionSource
+        include Google::Apis::Core::Hashable
+      
+        # Optional. End time (exclusive) of the time range. If not set, the end time is
+        # unbounded.
+        # Corresponds to the JSON property `endTime`
+        # @return [String]
+        attr_accessor :end_time
+      
+        # Required. The resource name of the Session to generate memories for. Format: `
+        # projects/`project`/locations/`location`/reasoningEngines/`reasoning_engine`/
+        # sessions/`session``
+        # Corresponds to the JSON property `session`
+        # @return [String]
+        attr_accessor :session
+      
+        # Optional. Time range to define which session events should be used to generate
+        # memories. Start time (inclusive) of the time range. If not set, the start time
+        # is unbounded.
+        # Corresponds to the JSON property `startTime`
+        # @return [String]
+        attr_accessor :start_time
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @end_time = args[:end_time] if args.key?(:end_time)
+          @session = args[:session] if args.key?(:session)
+          @start_time = args[:start_time] if args.key?(:start_time)
+        end
+      end
+      
       # Generate video response.
       class GoogleCloudAiplatformV1beta1GenerateVideoResponse
         include Google::Apis::Core::Hashable
@@ -15609,7 +15780,9 @@ module Google
         include Google::Apis::Core::Hashable
       
         # Confidence score of the support references. Ranges from 0 to 1. 1 is the most
-        # confident. This list must have the same size as the grounding_chunk_indices.
+        # confident. For Gemini 2.0 and before, this list must have the same size as the
+        # grounding_chunk_indices. For Gemini 2.5 and after, this list will be empty and
+        # should be ignored.
         # Corresponds to the JSON property `confidenceScores`
         # @return [Array<Float>]
         attr_accessor :confidence_scores
@@ -16399,6 +16572,11 @@ module Google
         # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1RagFileChunkingConfig]
         attr_accessor :rag_file_chunking_config
       
+        # Metadata config for RagFile.
+        # Corresponds to the JSON property `ragFileMetadataConfig`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1RagFileMetadataConfig]
+        attr_accessor :rag_file_metadata_config
+      
         # Specifies the parsing config for RagFiles.
         # Corresponds to the JSON property `ragFileParsingConfig`
         # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1RagFileParsingConfig]
@@ -16445,6 +16623,7 @@ module Google
           @partial_failure_bigquery_sink = args[:partial_failure_bigquery_sink] if args.key?(:partial_failure_bigquery_sink)
           @partial_failure_gcs_sink = args[:partial_failure_gcs_sink] if args.key?(:partial_failure_gcs_sink)
           @rag_file_chunking_config = args[:rag_file_chunking_config] if args.key?(:rag_file_chunking_config)
+          @rag_file_metadata_config = args[:rag_file_metadata_config] if args.key?(:rag_file_metadata_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)
@@ -18058,6 +18237,32 @@ module Google
         end
       end
       
+      # Response message for MemoryBankService.ListMemories.
+      class GoogleCloudAiplatformV1beta1ListMemoriesResponse
+        include Google::Apis::Core::Hashable
+      
+        # List of Memories in the requested page.
+        # Corresponds to the JSON property `memories`
+        # @return [Array<Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1Memory>]
+        attr_accessor :memories
+      
+        # A token to retrieve the next page of results. Pass to ListMemoriesRequest.
+        # page_token to obtain that page.
+        # Corresponds to the JSON property `nextPageToken`
+        # @return [String]
+        attr_accessor :next_page_token
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @memories = args[:memories] if args.key?(:memories)
+          @next_page_token = args[:next_page_token] if args.key?(:next_page_token)
+        end
+      end
+      
       # Response message for MetadataService.ListMetadataSchemas.
       class GoogleCloudAiplatformV1beta1ListMetadataSchemasResponse
         include Google::Apis::Core::Hashable
@@ -19193,6 +19398,64 @@ module Google
         end
       end
       
+      # A memory.
+      class GoogleCloudAiplatformV1beta1Memory
+        include Google::Apis::Core::Hashable
+      
+        # Output only. Timestamp when this Memory was created.
+        # Corresponds to the JSON property `createTime`
+        # @return [String]
+        attr_accessor :create_time
+      
+        # Optional. Description of the Memory.
+        # Corresponds to the JSON property `description`
+        # @return [String]
+        attr_accessor :description
+      
+        # Optional. Display name of the Memory.
+        # Corresponds to the JSON property `displayName`
+        # @return [String]
+        attr_accessor :display_name
+      
+        # Required. Semantic knowledge extracted from the source content.
+        # Corresponds to the JSON property `fact`
+        # @return [String]
+        attr_accessor :fact
+      
+        # Identifier. The resource name of the Memory. Format: `projects/`project`/
+        # locations/`location`/reasoningEngines/`reasoning_engine`/memories/`memory``
+        # Corresponds to the JSON property `name`
+        # @return [String]
+        attr_accessor :name
+      
+        # Required. Immutable. The scope of the Memory. Memories are isolated within
+        # their scope. The scope is defined when creating or generating memories. Scope
+        # values cannot contain the wildcard character '*'.
+        # Corresponds to the JSON property `scope`
+        # @return [Hash<String,String>]
+        attr_accessor :scope
+      
+        # Output only. Timestamp when this Memory was most recently updated.
+        # Corresponds to the JSON property `updateTime`
+        # @return [String]
+        attr_accessor :update_time
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @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)
+          @fact = args[:fact] if args.key?(:fact)
+          @name = args[:name] if args.key?(:name)
+          @scope = args[:scope] if args.key?(:scope)
+          @update_time = args[:update_time] if args.key?(:update_time)
+        end
+      end
+      
       # Request message for ModelService.MergeVersionAliases.
       class GoogleCloudAiplatformV1beta1MergeVersionAliasesRequest
         include Google::Apis::Core::Hashable
@@ -20423,6 +20686,16 @@ module Google
         # @return [String]
         attr_accessor :image_uri
       
+        # Immutable. Invoke route prefix for the custom container. "/*" is the only
+        # supported value right now. By setting this field, any non-root route on this
+        # model will be accessible with [PredictionService.Invoke] eg: "/invoke/foo/bar".
+        # Only one of `predict_route` or `invoke_route_prefix` can be set, and we
+        # default to using `predict_route` if this field is not set. If this field is
+        # set, the Model can only be deployed to dedicated endpoint.
+        # Corresponds to the JSON property `invokeRoutePrefix`
+        # @return [String]
+        attr_accessor :invoke_route_prefix
+      
         # Probe describes a health check to be performed against a container to
         # determine whether it is alive or ready to receive traffic.
         # Corresponds to the JSON property `livenessProbe`
@@ -20491,6 +20764,7 @@ module Google
           @health_probe = args[:health_probe] if args.key?(:health_probe)
           @health_route = args[:health_route] if args.key?(:health_route)
           @image_uri = args[:image_uri] if args.key?(:image_uri)
+          @invoke_route_prefix = args[:invoke_route_prefix] if args.key?(:invoke_route_prefix)
           @liveness_probe = args[:liveness_probe] if args.key?(:liveness_probe)
           @ports = args[:ports] if args.key?(:ports)
           @predict_route = args[:predict_route] if args.key?(:predict_route)
@@ -24768,14 +25042,17 @@ 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.
+        # Output only. Error message if the PSC service automation failed.
+        # Corresponds to the JSON property `errorMessage`
+        # @return [String]
+        attr_accessor :error_message
+      
+        # Output only. Forwarding rule created by the PSC service automation.
         # 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.
+        # Output only. IP address rule created by the PSC service automation.
         # Corresponds to the JSON property `ipAddress`
         # @return [String]
         attr_accessor :ip_address
@@ -24794,16 +25071,23 @@ module Google
         # @return [String]
         attr_accessor :project_id
       
+        # Output only. The state of the PSC service automation.
+        # Corresponds to the JSON property `state`
+        # @return [String]
+        attr_accessor :state
+      
         def initialize(**args)
            update!(**args)
         end
       
         # Update properties of this object
         def update!(**args)
+          @error_message = args[:error_message] if args.key?(:error_message)
           @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)
+          @state = args[:state] if args.key?(:state)
         end
       end
       
@@ -27101,6 +27385,14 @@ module Google
       class GoogleCloudAiplatformV1beta1PscInterfaceConfig
         include Google::Apis::Core::Hashable
       
+        # Optional. DNS peering configurations. When specified, Vertex AI will attempt
+        # to configure DNS peering zones in the tenant project VPC to resolve the
+        # specified domains using the target network's Cloud DNS. The user must grant
+        # the dns.peer role to the Vertex AI Service Agent on the target project.
+        # Corresponds to the JSON property `dnsPeeringConfigs`
+        # @return [Array<Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1DnsPeeringConfig>]
+        attr_accessor :dns_peering_configs
+      
         # Optional. The name of the Compute Engine [network attachment](https://cloud.
         # google.com/vpc/docs/about-network-attachments) to attach to the resource
         # within the region and user project. To specify this field, you must have
@@ -27117,6 +27409,7 @@ module Google
       
         # Update properties of this object
         def update!(**args)
+          @dns_peering_configs = args[:dns_peering_configs] if args.key?(:dns_peering_configs)
           @network_attachment = args[:network_attachment] if args.key?(:network_attachment)
         end
       end
@@ -27520,6 +27813,12 @@ module Google
         # @return [String]
         attr_accessor :resource_use_case
       
+        # Optional. For notebook resource, whether the notebook supports Workbench.
+        # Corresponds to the JSON property `supportsWorkbench`
+        # @return [Boolean]
+        attr_accessor :supports_workbench
+        alias_method :supports_workbench?, :supports_workbench
+      
         # Required.
         # Corresponds to the JSON property `title`
         # @return [String]
@@ -27535,6 +27834,7 @@ module Google
           @resource_description = args[:resource_description] if args.key?(:resource_description)
           @resource_title = args[:resource_title] if args.key?(:resource_title)
           @resource_use_case = args[:resource_use_case] if args.key?(:resource_use_case)
+          @supports_workbench = args[:supports_workbench] if args.key?(:supports_workbench)
           @title = args[:title] if args.key?(:title)
         end
       end
@@ -29127,6 +29427,12 @@ module Google
         # @return [String]
         attr_accessor :update_time
       
+        # Output only. The metadata for metadata search. The user_metadata Needs to be
+        # in JSON format.
+        # Corresponds to the JSON property `userMetadata`
+        # @return [String]
+        attr_accessor :user_metadata
+      
         def initialize(**args)
            update!(**args)
         end
@@ -29147,6 +29453,7 @@ module Google
           @size_bytes = args[:size_bytes] if args.key?(:size_bytes)
           @slack_source = args[:slack_source] if args.key?(:slack_source)
           @update_time = args[:update_time] if args.key?(:update_time)
+          @user_metadata = args[:user_metadata] if args.key?(:user_metadata)
         end
       end
       
@@ -29206,6 +29513,55 @@ module Google
         end
       end
       
+      # Metadata config for RagFile.
+      class GoogleCloudAiplatformV1beta1RagFileMetadataConfig
+        include Google::Apis::Core::Hashable
+      
+        # The Google Cloud Storage location for the input content.
+        # Corresponds to the JSON property `gcsMetadataSchemaSource`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1GcsSource]
+        attr_accessor :gcs_metadata_schema_source
+      
+        # The Google Cloud Storage location for the input content.
+        # Corresponds to the JSON property `gcsMetadataSource`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1GcsSource]
+        attr_accessor :gcs_metadata_source
+      
+        # The Google Drive location for the input content.
+        # Corresponds to the JSON property `googleDriveMetadataSchemaSource`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1GoogleDriveSource]
+        attr_accessor :google_drive_metadata_schema_source
+      
+        # The Google Drive location for the input content.
+        # Corresponds to the JSON property `googleDriveMetadataSource`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1GoogleDriveSource]
+        attr_accessor :google_drive_metadata_source
+      
+        # Inline metadata schema source. Must be a JSON string.
+        # Corresponds to the JSON property `inlineMetadataSchemaSource`
+        # @return [String]
+        attr_accessor :inline_metadata_schema_source
+      
+        # Inline metadata source. Must be a JSON string.
+        # Corresponds to the JSON property `inlineMetadataSource`
+        # @return [String]
+        attr_accessor :inline_metadata_source
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @gcs_metadata_schema_source = args[:gcs_metadata_schema_source] if args.key?(:gcs_metadata_schema_source)
+          @gcs_metadata_source = args[:gcs_metadata_source] if args.key?(:gcs_metadata_source)
+          @google_drive_metadata_schema_source = args[:google_drive_metadata_schema_source] if args.key?(:google_drive_metadata_schema_source)
+          @google_drive_metadata_source = args[:google_drive_metadata_source] if args.key?(:google_drive_metadata_source)
+          @inline_metadata_schema_source = args[:inline_metadata_schema_source] if args.key?(:inline_metadata_schema_source)
+          @inline_metadata_source = args[:inline_metadata_source] if args.key?(:inline_metadata_source)
+        end
+      end
+      
       # Specifies the parsing config for RagFiles.
       class GoogleCloudAiplatformV1beta1RagFileParsingConfig
         include Google::Apis::Core::Hashable
@@ -29377,19 +29733,34 @@ module Google
       
         # Basic tier is a cost-effective and low compute tier suitable for the following
         # cases: * Experimenting with RagManagedDb. * Small data size. * Latency
-        # insensitive workload. * Only using RAG Engine with external vector DBs.
+        # insensitive workload. * Only using RAG Engine with external vector DBs. NOTE:
+        # This is the default tier if not explicitly chosen.
         # Corresponds to the JSON property `basic`
         # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1RagManagedDbConfigBasic]
         attr_accessor :basic
       
         # Enterprise tier offers production grade performance along with autoscaling
         # functionality. It is suitable for customers with large amounts of data or
-        # performance sensitive workloads. NOTE: This is the default tier if not
-        # explicitly chosen.
+        # performance sensitive workloads.
         # Corresponds to the JSON property `enterprise`
         # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1RagManagedDbConfigEnterprise]
         attr_accessor :enterprise
       
+        # Scaled tier offers production grade performance along with autoscaling
+        # functionality. It is suitable for customers with large amounts of data or
+        # performance sensitive workloads.
+        # Corresponds to the JSON property `scaled`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1RagManagedDbConfigScaled]
+        attr_accessor :scaled
+      
+        # Disables the RAG Engine service and deletes all your data held within this
+        # service. This will halt the billing of the service. NOTE: Once deleted the
+        # data cannot be recovered. To start using RAG Engine again, you will need to
+        # update the tier by calling the UpdateRagEngineConfig API.
+        # Corresponds to the JSON property `unprovisioned`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1RagManagedDbConfigUnprovisioned]
+        attr_accessor :unprovisioned
+      
         def initialize(**args)
            update!(**args)
         end
@@ -29398,12 +29769,15 @@ module Google
         def update!(**args)
           @basic = args[:basic] if args.key?(:basic)
           @enterprise = args[:enterprise] if args.key?(:enterprise)
+          @scaled = args[:scaled] if args.key?(:scaled)
+          @unprovisioned = args[:unprovisioned] if args.key?(:unprovisioned)
         end
       end
       
       # Basic tier is a cost-effective and low compute tier suitable for the following
       # cases: * Experimenting with RagManagedDb. * Small data size. * Latency
-      # insensitive workload. * Only using RAG Engine with external vector DBs.
+      # insensitive workload. * Only using RAG Engine with external vector DBs. NOTE:
+      # This is the default tier if not explicitly chosen.
       class GoogleCloudAiplatformV1beta1RagManagedDbConfigBasic
         include Google::Apis::Core::Hashable
       
@@ -29418,8 +29792,7 @@ module Google
       
       # Enterprise tier offers production grade performance along with autoscaling
       # functionality. It is suitable for customers with large amounts of data or
-      # performance sensitive workloads. NOTE: This is the default tier if not
-      # explicitly chosen.
+      # performance sensitive workloads.
       class GoogleCloudAiplatformV1beta1RagManagedDbConfigEnterprise
         include Google::Apis::Core::Hashable
       
@@ -29432,6 +29805,37 @@ module Google
         end
       end
       
+      # Scaled tier offers production grade performance along with autoscaling
+      # functionality. It is suitable for customers with large amounts of data or
+      # performance sensitive workloads.
+      class GoogleCloudAiplatformV1beta1RagManagedDbConfigScaled
+        include Google::Apis::Core::Hashable
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+        end
+      end
+      
+      # Disables the RAG Engine service and deletes all your data held within this
+      # service. This will halt the billing of the service. NOTE: Once deleted the
+      # data cannot be recovered. To start using RAG Engine again, you will need to
+      # update the tier by calling the UpdateRagEngineConfig API.
+      class GoogleCloudAiplatformV1beta1RagManagedDbConfigUnprovisioned
+        include Google::Apis::Core::Hashable
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+        end
+      end
+      
       # A query to retrieve relevant contexts.
       class GoogleCloudAiplatformV1beta1RagQuery
         include Google::Apis::Core::Hashable
@@ -30352,6 +30756,11 @@ module Google
       class GoogleCloudAiplatformV1beta1ReasoningEngine
         include Google::Apis::Core::Hashable
       
+        # Configuration for how Agent Engine sub-resources should manage context.
+        # Corresponds to the JSON property `contextSpec`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1ReasoningEngineContextSpec]
+        attr_accessor :context_spec
+      
         # Output only. Timestamp when this ReasoningEngine was created.
         # Corresponds to the JSON property `createTime`
         # @return [String]
@@ -30395,6 +30804,7 @@ module Google
       
         # Update properties of this object
         def update!(**args)
+          @context_spec = args[:context_spec] if args.key?(:context_spec)
           @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)
@@ -30405,6 +30815,92 @@ module Google
         end
       end
       
+      # Configuration for how Agent Engine sub-resources should manage context.
+      class GoogleCloudAiplatformV1beta1ReasoningEngineContextSpec
+        include Google::Apis::Core::Hashable
+      
+        # Specification for a Memory Bank.
+        # Corresponds to the JSON property `memoryBankConfig`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1ReasoningEngineContextSpecMemoryBankConfig]
+        attr_accessor :memory_bank_config
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @memory_bank_config = args[:memory_bank_config] if args.key?(:memory_bank_config)
+        end
+      end
+      
+      # Specification for a Memory Bank.
+      class GoogleCloudAiplatformV1beta1ReasoningEngineContextSpecMemoryBankConfig
+        include Google::Apis::Core::Hashable
+      
+        # Configuration for how to generate memories.
+        # Corresponds to the JSON property `generationConfig`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1ReasoningEngineContextSpecMemoryBankConfigGenerationConfig]
+        attr_accessor :generation_config
+      
+        # Configuration for how to perform similarity search on memories.
+        # Corresponds to the JSON property `similaritySearchConfig`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1ReasoningEngineContextSpecMemoryBankConfigSimilaritySearchConfig]
+        attr_accessor :similarity_search_config
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @generation_config = args[:generation_config] if args.key?(:generation_config)
+          @similarity_search_config = args[:similarity_search_config] if args.key?(:similarity_search_config)
+        end
+      end
+      
+      # Configuration for how to generate memories.
+      class GoogleCloudAiplatformV1beta1ReasoningEngineContextSpecMemoryBankConfigGenerationConfig
+        include Google::Apis::Core::Hashable
+      
+        # Required. The model used to generate memories. Format: `projects/`project`/
+        # locations/`location`/publishers/google/models/`model`` or `projects/`project`/
+        # locations/`location`/endpoints/`endpoint``.
+        # Corresponds to the JSON property `model`
+        # @return [String]
+        attr_accessor :model
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @model = args[:model] if args.key?(:model)
+        end
+      end
+      
+      # Configuration for how to perform similarity search on memories.
+      class GoogleCloudAiplatformV1beta1ReasoningEngineContextSpecMemoryBankConfigSimilaritySearchConfig
+        include Google::Apis::Core::Hashable
+      
+        # Required. The model used to generate embeddings to lookup similar memories.
+        # Format: `projects/`project`/locations/`location`/publishers/google/models/`
+        # model`` or `projects/`project`/locations/`location`/endpoints/`endpoint``.
+        # Corresponds to the JSON property `embeddingModel`
+        # @return [String]
+        attr_accessor :embedding_model
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @embedding_model = args[:embedding_model] if args.key?(:embedding_model)
+        end
+      end
+      
       # ReasoningEngine configurations
       class GoogleCloudAiplatformV1beta1ReasoningEngineSpec
         include Google::Apis::Core::Hashable
@@ -31268,6 +31764,150 @@ module Google
         end
       end
       
+      # Request message for MemoryBankService.RetrieveMemories.
+      class GoogleCloudAiplatformV1beta1RetrieveMemoriesRequest
+        include Google::Apis::Core::Hashable
+      
+        # Required. The scope of the memories to retrieve. A memory must have exactly
+        # the same scope (`Memory.scope`) as the scope provided here to be retrieved (
+        # same keys and values). Order does not matter, but it is case-sensitive.
+        # Corresponds to the JSON property `scope`
+        # @return [Hash<String,String>]
+        attr_accessor :scope
+      
+        # Parameters for semantic similarity search based retrieval.
+        # Corresponds to the JSON property `similaritySearchParams`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1RetrieveMemoriesRequestSimilaritySearchParams]
+        attr_accessor :similarity_search_params
+      
+        # Parameters for simple (non-similarity search) retrieval.
+        # Corresponds to the JSON property `simpleRetrievalParams`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1RetrieveMemoriesRequestSimpleRetrievalParams]
+        attr_accessor :simple_retrieval_params
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @scope = args[:scope] if args.key?(:scope)
+          @similarity_search_params = args[:similarity_search_params] if args.key?(:similarity_search_params)
+          @simple_retrieval_params = args[:simple_retrieval_params] if args.key?(:simple_retrieval_params)
+        end
+      end
+      
+      # Parameters for semantic similarity search based retrieval.
+      class GoogleCloudAiplatformV1beta1RetrieveMemoriesRequestSimilaritySearchParams
+        include Google::Apis::Core::Hashable
+      
+        # Required. Query to use for similarity search retrieval. If provided, then the
+        # parent ReasoningEngine must have ReasoningEngineContextSpec.MemoryBankConfig.
+        # SimilaritySearchConfig set.
+        # Corresponds to the JSON property `searchQuery`
+        # @return [String]
+        attr_accessor :search_query
+      
+        # Optional. The maximum number of memories to return. The service may return
+        # fewer than this value. If unspecified, at most 3 memories will be returned.
+        # The maximum value is 100; values above 100 will be coerced to 100.
+        # Corresponds to the JSON property `topK`
+        # @return [Fixnum]
+        attr_accessor :top_k
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @search_query = args[:search_query] if args.key?(:search_query)
+          @top_k = args[:top_k] if args.key?(:top_k)
+        end
+      end
+      
+      # Parameters for simple (non-similarity search) retrieval.
+      class GoogleCloudAiplatformV1beta1RetrieveMemoriesRequestSimpleRetrievalParams
+        include Google::Apis::Core::Hashable
+      
+        # Optional. The maximum number of memories to return. The service may return
+        # fewer than this value. If unspecified, at most 3 memories will be returned.
+        # The maximum value is 100; values above 100 will be coerced to 100.
+        # Corresponds to the JSON property `pageSize`
+        # @return [Fixnum]
+        attr_accessor :page_size
+      
+        # Optional. A page token, received from a previous `RetrieveMemories` call.
+        # Provide this to retrieve the subsequent page.
+        # Corresponds to the JSON property `pageToken`
+        # @return [String]
+        attr_accessor :page_token
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @page_size = args[:page_size] if args.key?(:page_size)
+          @page_token = args[:page_token] if args.key?(:page_token)
+        end
+      end
+      
+      # Response message for MemoryBankService.RetrieveMemories.
+      class GoogleCloudAiplatformV1beta1RetrieveMemoriesResponse
+        include Google::Apis::Core::Hashable
+      
+        # A token that can be sent as `page_token` to retrieve the next page. If this
+        # field is omitted, there are no subsequent pages. This token is not set if
+        # similarity search was used for retrieval.
+        # Corresponds to the JSON property `nextPageToken`
+        # @return [String]
+        attr_accessor :next_page_token
+      
+        # The retrieved memories.
+        # Corresponds to the JSON property `retrievedMemories`
+        # @return [Array<Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1RetrieveMemoriesResponseRetrievedMemory>]
+        attr_accessor :retrieved_memories
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @next_page_token = args[:next_page_token] if args.key?(:next_page_token)
+          @retrieved_memories = args[:retrieved_memories] if args.key?(:retrieved_memories)
+        end
+      end
+      
+      # A retrieved memory.
+      class GoogleCloudAiplatformV1beta1RetrieveMemoriesResponseRetrievedMemory
+        include Google::Apis::Core::Hashable
+      
+        # The distance between the query and the retrieved Memory. Smaller values
+        # indicate more similar memories. This is only set if similarity search was used
+        # for retrieval.
+        # Corresponds to the JSON property `distance`
+        # @return [Float]
+        attr_accessor :distance
+      
+        # A memory.
+        # Corresponds to the JSON property `memory`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1Memory]
+        attr_accessor :memory
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @distance = args[:distance] if args.key?(:distance)
+          @memory = args[:memory] if args.key?(:memory)
+        end
+      end
+      
       # Configuration for rolling deployments.
       class GoogleCloudAiplatformV1beta1RolloutOptions
         include Google::Apis::Core::Hashable
@@ -31768,6 +32408,13 @@ module Google
         # @return [String]
         attr_accessor :category
       
+        # Output only. The overwritten threshold for the safety category of Gemini 2.0
+        # image out. If minors are detected in the output image, the threshold of each
+        # safety category will be overwritten if user sets a lower threshold.
+        # Corresponds to the JSON property `overwrittenThreshold`
+        # @return [String]
+        attr_accessor :overwritten_threshold
+      
         # Output only. Harm probability levels in the content.
         # Corresponds to the JSON property `probability`
         # @return [String]
@@ -31796,6 +32443,7 @@ module Google
         def update!(**args)
           @blocked = args[:blocked] if args.key?(:blocked)
           @category = args[:category] if args.key?(:category)
+          @overwritten_threshold = args[:overwritten_threshold] if args.key?(:overwritten_threshold)
           @probability = args[:probability] if args.key?(:probability)
           @probability_score = args[:probability_score] if args.key?(:probability_score)
           @severity = args[:severity] if args.key?(:severity)
@@ -42262,7 +42910,8 @@ module Google
         # @return [Fixnum]
         attr_accessor :epoch_count
       
-        # Optional. Multiplier for adjusting the default learning rate.
+        # Optional. Multiplier for adjusting the default learning rate. Mutually
+        # exclusive with `learning_rate`.
         # Corresponds to the JSON property `learningRateMultiplier`
         # @return [Float]
         attr_accessor :learning_rate_multiplier
@@ -45850,6 +46499,16 @@ module Google
         # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1RagFileChunkingConfig]
         attr_accessor :rag_file_chunking_config
       
+        # Metadata config for RagFile.
+        # Corresponds to the JSON property `ragFileMetadataConfig`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1RagFileMetadataConfig]
+        attr_accessor :rag_file_metadata_config
+      
+        # Specifies the parsing config for RagFiles.
+        # Corresponds to the JSON property `ragFileParsingConfig`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1RagFileParsingConfig]
+        attr_accessor :rag_file_parsing_config
+      
         # Specifies the transformation config for RagFiles.
         # Corresponds to the JSON property `ragFileTransformationConfig`
         # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1RagFileTransformationConfig]
@@ -45862,6 +46521,8 @@ module Google
         # Update properties of this object
         def update!(**args)
           @rag_file_chunking_config = args[:rag_file_chunking_config] if args.key?(:rag_file_chunking_config)
+          @rag_file_metadata_config = args[:rag_file_metadata_config] if args.key?(:rag_file_metadata_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)
         end
       end
@@ -46054,6 +46715,50 @@ module Google
         end
       end
       
+      # Metadata related to url context retrieval tool.
+      class GoogleCloudAiplatformV1beta1UrlContextMetadata
+        include Google::Apis::Core::Hashable
+      
+        # Output only. List of url context.
+        # Corresponds to the JSON property `urlMetadata`
+        # @return [Array<Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1UrlMetadata>]
+        attr_accessor :url_metadata
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @url_metadata = args[:url_metadata] if args.key?(:url_metadata)
+        end
+      end
+      
+      # Context of the a single url retrieval.
+      class GoogleCloudAiplatformV1beta1UrlMetadata
+        include Google::Apis::Core::Hashable
+      
+        # Retrieved url by the tool.
+        # Corresponds to the JSON property `retrievedUrl`
+        # @return [String]
+        attr_accessor :retrieved_url
+      
+        # Status of the url retrieval.
+        # Corresponds to the JSON property `urlRetrievalStatus`
+        # @return [String]
+        attr_accessor :url_retrieval_status
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @retrieved_url = args[:retrieved_url] if args.key?(:retrieved_url)
+          @url_retrieval_status = args[:url_retrieval_status] if args.key?(:url_retrieval_status)
+        end
+      end
+      
       # References an API call. It contains more information about long running
       # operation and Jobs that are triggered by the API call.
       class GoogleCloudAiplatformV1beta1UserActionReference