google-apis-aiplatform_v1beta1 0.63.0 → 0.65.0

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

@@ -3661,6 +3661,12 @@ module Google
         # @return [String]
         attr_accessor :candidate
       
+        # Optional. Intermediate events (such as tool calls and responses) that led to
+        # the final response.
+        # Corresponds to the JSON property `events`
+        # @return [Array<Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1Content>]
+        attr_accessor :events
+      
         # Text response.
         # Corresponds to the JSON property `text`
         # @return [String]
@@ -3678,6 +3684,7 @@ module Google
         # Update properties of this object
         def update!(**args)
           @candidate = args[:candidate] if args.key?(:candidate)
+          @events = args[:events] if args.key?(:events)
           @text = args[:text] if args.key?(:text)
           @value = args[:value] if args.key?(:value)
         end
@@ -6756,7 +6763,7 @@ module Google
       class GoogleCloudAiplatformV1beta1DedicatedResourcesScaleToZeroSpec
         include Google::Apis::Core::Hashable
       
-        # Optional. Duration of no traffic before scaling to zero. [MinValue=3600] (5
+        # Optional. Duration of no traffic before scaling to zero. [MinValue=300] (5
         # minutes) [MaxValue=28800] (8 hours)
         # Corresponds to the JSON property `idleScaledownPeriod`
         # @return [String]
@@ -8409,6 +8416,107 @@ module Google
         end
       end
       
+      # Request message for PredictionService.EmbedContent.
+      class GoogleCloudAiplatformV1beta1EmbedContentRequest
+        include Google::Apis::Core::Hashable
+      
+        # Optional. Whether to silently truncate the input content if it's longer than
+        # the maximum sequence length.
+        # Corresponds to the JSON property `autoTruncate`
+        # @return [Boolean]
+        attr_accessor :auto_truncate
+        alias_method :auto_truncate?, :auto_truncate
+      
+        # 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
+      
+        # Optional. Optional reduced dimension for the output embedding. If set,
+        # excessive values in the output embedding are truncated from the end.
+        # Corresponds to the JSON property `outputDimensionality`
+        # @return [Fixnum]
+        attr_accessor :output_dimensionality
+      
+        # Optional. The task type of the embedding.
+        # Corresponds to the JSON property `taskType`
+        # @return [String]
+        attr_accessor :task_type
+      
+        # Optional. An optional title for the text.
+        # Corresponds to the JSON property `title`
+        # @return [String]
+        attr_accessor :title
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @auto_truncate = args[:auto_truncate] if args.key?(:auto_truncate)
+          @content = args[:content] if args.key?(:content)
+          @output_dimensionality = args[:output_dimensionality] if args.key?(:output_dimensionality)
+          @task_type = args[:task_type] if args.key?(:task_type)
+          @title = args[:title] if args.key?(:title)
+        end
+      end
+      
+      # Response message for PredictionService.EmbedContent.
+      class GoogleCloudAiplatformV1beta1EmbedContentResponse
+        include Google::Apis::Core::Hashable
+      
+        # A list of floats representing an embedding.
+        # Corresponds to the JSON property `embedding`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1EmbedContentResponseEmbedding]
+        attr_accessor :embedding
+      
+        # Whether the input content was truncated before generating the embedding.
+        # Corresponds to the JSON property `truncated`
+        # @return [Boolean]
+        attr_accessor :truncated
+        alias_method :truncated?, :truncated
+      
+        # Usage metadata about the content generation request and response. This message
+        # provides a detailed breakdown of token usage and other relevant metrics.
+        # Corresponds to the JSON property `usageMetadata`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1UsageMetadata]
+        attr_accessor :usage_metadata
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @embedding = args[:embedding] if args.key?(:embedding)
+          @truncated = args[:truncated] if args.key?(:truncated)
+          @usage_metadata = args[:usage_metadata] if args.key?(:usage_metadata)
+        end
+      end
+      
+      # A list of floats representing an embedding.
+      class GoogleCloudAiplatformV1beta1EmbedContentResponseEmbedding
+        include Google::Apis::Core::Hashable
+      
+        # Embedding vector values.
+        # Corresponds to the JSON property `values`
+        # @return [Array<Float>]
+        attr_accessor :values
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @values = args[:values] if args.key?(:values)
+        end
+      end
+      
       # Request message for ModelGardenService.EnableModel.
       class GoogleCloudAiplatformV1beta1EnableModelRequest
         include Google::Apis::Core::Hashable
@@ -9620,6 +9728,11 @@ module Google
       class GoogleCloudAiplatformV1beta1EvaluationInstance
         include Google::Apis::Core::Hashable
       
+        # Contains data specific to agent evaluations.
+        # Corresponds to the JSON property `agentData`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1EvaluationInstanceAgentData]
+        attr_accessor :agent_data
+      
         # Instance data specified as a map.
         # Corresponds to the JSON property `otherData`
         # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1EvaluationInstanceMapInstance]
@@ -9653,6 +9766,7 @@ module Google
       
         # Update properties of this object
         def update!(**args)
+          @agent_data = args[:agent_data] if args.key?(:agent_data)
           @other_data = args[:other_data] if args.key?(:other_data)
           @prompt = args[:prompt] if args.key?(:prompt)
           @reference = args[:reference] if args.key?(:reference)
@@ -9661,6 +9775,150 @@ module Google
         end
       end
       
+      # Configuration for an Agent.
+      class GoogleCloudAiplatformV1beta1EvaluationInstanceAgentConfig
+        include Google::Apis::Core::Hashable
+      
+        # Instance data used to populate placeholders in a metric prompt template.
+        # Corresponds to the JSON property `developerInstruction`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1EvaluationInstanceInstanceData]
+        attr_accessor :developer_instruction
+      
+        # Represents a list of tools for an agent.
+        # Corresponds to the JSON property `tools`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1EvaluationInstanceAgentConfigTools]
+        attr_accessor :tools
+      
+        # A JSON string containing a list of tools available to an agent with info such
+        # as name, description, parameters and required parameters.
+        # Corresponds to the JSON property `toolsText`
+        # @return [String]
+        attr_accessor :tools_text
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @developer_instruction = args[:developer_instruction] if args.key?(:developer_instruction)
+          @tools = args[:tools] if args.key?(:tools)
+          @tools_text = args[:tools_text] if args.key?(:tools_text)
+        end
+      end
+      
+      # Represents a list of tools for an agent.
+      class GoogleCloudAiplatformV1beta1EvaluationInstanceAgentConfigTools
+        include Google::Apis::Core::Hashable
+      
+        # Optional. List of tools: each tool can have multiple function declarations.
+        # Corresponds to the JSON property `tool`
+        # @return [Array<Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1Tool>]
+        attr_accessor :tool
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @tool = args[:tool] if args.key?(:tool)
+        end
+      end
+      
+      # Contains data specific to agent evaluations.
+      class GoogleCloudAiplatformV1beta1EvaluationInstanceAgentData
+        include Google::Apis::Core::Hashable
+      
+        # Configuration for an Agent.
+        # Corresponds to the JSON property `agentConfig`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1EvaluationInstanceAgentConfig]
+        attr_accessor :agent_config
+      
+        # Instance data used to populate placeholders in a metric prompt template.
+        # Corresponds to the JSON property `developerInstruction`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1EvaluationInstanceInstanceData]
+        attr_accessor :developer_instruction
+      
+        # Represents a list of events for an agent.
+        # Corresponds to the JSON property `events`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1EvaluationInstanceAgentDataEvents]
+        attr_accessor :events
+      
+        # A JSON string containing a sequence of events.
+        # Corresponds to the JSON property `eventsText`
+        # @return [String]
+        attr_accessor :events_text
+      
+        # Represents a list of tools for an agent.
+        # Corresponds to the JSON property `tools`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1EvaluationInstanceAgentDataTools]
+        attr_accessor :tools
+      
+        # A JSON string containing a list of tools available to an agent with info such
+        # as name, description, parameters and required parameters. Example: [ ` "name":
+        # "search_actors", "description": "Search for actors in a movie. Returns a list
+        # of actors, their roles, their birthdate, and their place of birth.", "
+        # parameters": [ ` "name": "movie_name", "description": "The name of the movie."
+        # `, ` "name": "character_name", "description": "The name of the character." ` ],
+        # "required": ["movie_name", "character_name"] ` ]
+        # Corresponds to the JSON property `toolsText`
+        # @return [String]
+        attr_accessor :tools_text
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @agent_config = args[:agent_config] if args.key?(:agent_config)
+          @developer_instruction = args[:developer_instruction] if args.key?(:developer_instruction)
+          @events = args[:events] if args.key?(:events)
+          @events_text = args[:events_text] if args.key?(:events_text)
+          @tools = args[:tools] if args.key?(:tools)
+          @tools_text = args[:tools_text] if args.key?(:tools_text)
+        end
+      end
+      
+      # Represents a list of events for an agent.
+      class GoogleCloudAiplatformV1beta1EvaluationInstanceAgentDataEvents
+        include Google::Apis::Core::Hashable
+      
+        # Optional. A list of events.
+        # Corresponds to the JSON property `event`
+        # @return [Array<Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1Content>]
+        attr_accessor :event
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @event = args[:event] if args.key?(:event)
+        end
+      end
+      
+      # Represents a list of tools for an agent.
+      class GoogleCloudAiplatformV1beta1EvaluationInstanceAgentDataTools
+        include Google::Apis::Core::Hashable
+      
+        # Optional. List of tools: each tool can have multiple function declarations.
+        # Corresponds to the JSON property `tool`
+        # @return [Array<Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1Tool>]
+        attr_accessor :tool
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @tool = args[:tool] if args.key?(:tool)
+        end
+      end
+      
       # Instance data used to populate placeholders in a metric prompt template.
       class GoogleCloudAiplatformV1beta1EvaluationInstanceInstanceData
         include Google::Apis::Core::Hashable
@@ -10278,6 +10536,11 @@ module Google
       class GoogleCloudAiplatformV1beta1EvaluationRunInferenceConfig
         include Google::Apis::Core::Hashable
       
+        # Configuration that describes an agent.
+        # Corresponds to the JSON property `agentConfig`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1EvaluationRunInferenceConfigAgentConfig]
+        attr_accessor :agent_config
+      
         # Generation config.
         # Corresponds to the JSON property `generationConfig`
         # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1GenerationConfig]
@@ -10297,11 +10560,40 @@ module Google
       
         # Update properties of this object
         def update!(**args)
+          @agent_config = args[:agent_config] if args.key?(:agent_config)
           @generation_config = args[:generation_config] if args.key?(:generation_config)
           @model = args[:model] if args.key?(:model)
         end
       end
       
+      # Configuration that describes an agent.
+      class GoogleCloudAiplatformV1beta1EvaluationRunInferenceConfigAgentConfig
+        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 `developerInstruction`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1Content]
+        attr_accessor :developer_instruction
+      
+        # Optional. The tools available to the agent.
+        # Corresponds to the JSON property `tools`
+        # @return [Array<Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1Tool>]
+        attr_accessor :tools
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @developer_instruction = args[:developer_instruction] if args.key?(:developer_instruction)
+          @tools = args[:tools] if args.key?(:tools)
+        end
+      end
+      
       # The metric used for evaluation runs.
       class GoogleCloudAiplatformV1beta1EvaluationRunMetric
         include Google::Apis::Core::Hashable
@@ -10316,6 +10608,11 @@ module Google
         # @return [String]
         attr_accessor :metric
       
+        # The metric used for running evaluations.
+        # Corresponds to the JSON property `metricConfig`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1Metric]
+        attr_accessor :metric_config
+      
         # Specification for a pre-defined metric.
         # Corresponds to the JSON property `predefinedMetricSpec`
         # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1EvaluationRunMetricPredefinedMetricSpec]
@@ -10334,6 +10631,7 @@ module Google
         def update!(**args)
           @llm_based_metric_spec = args[:llm_based_metric_spec] if args.key?(:llm_based_metric_spec)
           @metric = args[:metric] if args.key?(:metric)
+          @metric_config = args[:metric_config] if args.key?(:metric_config)
           @predefined_metric_spec = args[:predefined_metric_spec] if args.key?(:predefined_metric_spec)
           @rubric_based_metric_spec = args[:rubric_based_metric_spec] if args.key?(:rubric_based_metric_spec)
         end
@@ -10667,12 +10965,6 @@ module Google
         # @return [String]
         attr_accessor :transfer_agent
       
-        # Deprecated. If set, the event transfers to the specified agent.
-        # Corresponds to the JSON property `transferToAgent`
-        # @return [Boolean]
-        attr_accessor :transfer_to_agent
-        alias_method :transfer_to_agent?, :transfer_to_agent
-      
         def initialize(**args)
            update!(**args)
         end
@@ -10685,7 +10977,6 @@ module Google
           @skip_summarization = args[:skip_summarization] if args.key?(:skip_summarization)
           @state_delta = args[:state_delta] if args.key?(:state_delta)
           @transfer_agent = args[:transfer_agent] if args.key?(:transfer_agent)
-          @transfer_to_agent = args[:transfer_to_agent] if args.key?(:transfer_to_agent)
         end
       end
       
@@ -16218,6 +16509,12 @@ module Google
         # @return [String]
         attr_accessor :name
       
+        # Optional. Ordered `Parts` that constitute a function response. Parts may have
+        # different IANA MIME types.
+        # Corresponds to the JSON property `parts`
+        # @return [Array<Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1FunctionResponsePart>]
+        attr_accessor :parts
+      
         # Required. The function response in JSON object format. Use "output" key to
         # specify function output and "error" key to specify error details (if any). If "
         # output" and "error" keys are not specified, then whole "response" is treated
@@ -16234,10 +16531,113 @@ module Google
         def update!(**args)
           @id = args[:id] if args.key?(:id)
           @name = args[:name] if args.key?(:name)
+          @parts = args[:parts] if args.key?(:parts)
           @response = args[:response] if args.key?(:response)
         end
       end
       
+      # Raw media bytes for function response. Text should not be sent as raw bytes,
+      # use the 'text' field.
+      class GoogleCloudAiplatformV1beta1FunctionResponseBlob
+        include Google::Apis::Core::Hashable
+      
+        # Required. Raw bytes.
+        # Corresponds to the JSON property `data`
+        # NOTE: Values are automatically base64 encoded/decoded in the client library.
+        # @return [String]
+        attr_accessor :data
+      
+        # 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 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
+      
+        # Required. The IANA standard MIME type of the source data.
+        # Corresponds to the JSON property `mimeType`
+        # @return [String]
+        attr_accessor :mime_type
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @data = args[:data] if args.key?(:data)
+          @display_name = args[:display_name] if args.key?(:display_name)
+          @mime_type = args[:mime_type] if args.key?(:mime_type)
+        end
+      end
+      
+      # URI based data for function response.
+      class GoogleCloudAiplatformV1beta1FunctionResponseFileData
+        include Google::Apis::Core::Hashable
+      
+        # 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 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
+      
+        # Required. URI.
+        # Corresponds to the JSON property `fileUri`
+        # @return [String]
+        attr_accessor :file_uri
+      
+        # Required. The IANA standard MIME type of the source data.
+        # Corresponds to the JSON property `mimeType`
+        # @return [String]
+        attr_accessor :mime_type
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @display_name = args[:display_name] if args.key?(:display_name)
+          @file_uri = args[:file_uri] if args.key?(:file_uri)
+          @mime_type = args[:mime_type] if args.key?(:mime_type)
+        end
+      end
+      
+      # A datatype containing media that is part of a `FunctionResponse` message. A `
+      # FunctionResponsePart` consists of data which has an associated datatype. A `
+      # FunctionResponsePart` can only contain one of the accepted types in `
+      # FunctionResponsePart.data`. A `FunctionResponsePart` must have a fixed IANA
+      # MIME type identifying the type and subtype of the media if the `inline_data`
+      # field is filled with raw bytes.
+      class GoogleCloudAiplatformV1beta1FunctionResponsePart
+        include Google::Apis::Core::Hashable
+      
+        # URI based data for function response.
+        # Corresponds to the JSON property `fileData`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1FunctionResponseFileData]
+        attr_accessor :file_data
+      
+        # Raw media bytes for function response. Text should not be sent as raw bytes,
+        # use the 'text' field.
+        # Corresponds to the JSON property `inlineData`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1FunctionResponseBlob]
+        attr_accessor :inline_data
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @file_data = args[:file_data] if args.key?(:file_data)
+          @inline_data = args[:inline_data] if args.key?(:inline_data)
+        end
+      end
+      
       # The Google Cloud Storage location where the output is to be written to.
       class GoogleCloudAiplatformV1beta1GcsDestination
         include Google::Apis::Core::Hashable
@@ -16896,6 +17296,11 @@ module Google
       class GoogleCloudAiplatformV1beta1GenerateInstanceRubricsRequest
         include Google::Apis::Core::Hashable
       
+        # Configuration for an Agent.
+        # Corresponds to the JSON property `agentConfig`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1EvaluationInstanceAgentConfig]
+        attr_accessor :agent_config
+      
         # Required. The prompt to generate rubrics from. For single-turn queries, this
         # is a single instance. For multi-turn queries, this is a repeated field that
         # contains conversation history + latest request.
@@ -16919,6 +17324,7 @@ module Google
       
         # Update properties of this object
         def update!(**args)
+          @agent_config = args[:agent_config] if args.key?(:agent_config)
           @contents = args[:contents] if args.key?(:contents)
           @predefined_rubric_generation_spec = args[:predefined_rubric_generation_spec] if args.key?(:predefined_rubric_generation_spec)
           @rubric_generation_spec = args[:rubric_generation_spec] if args.key?(:rubric_generation_spec)
@@ -16968,6 +17374,29 @@ module Google
         attr_accessor :disable_consolidation
         alias_method :disable_consolidation?, :disable_consolidation
       
+        # Optional. If true, no revisions will be created for this request.
+        # Corresponds to the JSON property `disableMemoryRevisions`
+        # @return [Boolean]
+        attr_accessor :disable_memory_revisions
+        alias_method :disable_memory_revisions?, :disable_memory_revisions
+      
+        # Optional. Timestamp of when the revision is considered expired. If not set,
+        # the memory revision will be kept until manually deleted.
+        # Corresponds to the JSON property `revisionExpireTime`
+        # @return [String]
+        attr_accessor :revision_expire_time
+      
+        # Optional. Labels to be applied to the generated memory revisions. For example,
+        # you can use this to label a revision with its data source.
+        # Corresponds to the JSON property `revisionLabels`
+        # @return [Hash<String,String>]
+        attr_accessor :revision_labels
+      
+        # Optional. The TTL for the revision. The expiration time is computed: now + TTL.
+        # Corresponds to the JSON property `revisionTtl`
+        # @return [String]
+        attr_accessor :revision_ttl
+      
         # 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
@@ -16993,6 +17422,10 @@ module Google
           @direct_contents_source = args[:direct_contents_source] if args.key?(:direct_contents_source)
           @direct_memories_source = args[:direct_memories_source] if args.key?(:direct_memories_source)
           @disable_consolidation = args[:disable_consolidation] if args.key?(:disable_consolidation)
+          @disable_memory_revisions = args[:disable_memory_revisions] if args.key?(:disable_memory_revisions)
+          @revision_expire_time = args[:revision_expire_time] if args.key?(:revision_expire_time)
+          @revision_labels = args[:revision_labels] if args.key?(:revision_labels)
+          @revision_ttl = args[:revision_ttl] if args.key?(:revision_ttl)
           @scope = args[:scope] if args.key?(:scope)
           @vertex_session_source = args[:vertex_session_source] if args.key?(:vertex_session_source)
         end
@@ -17069,6 +17502,11 @@ module Google
         # @return [String]
         attr_accessor :fact
       
+        # Optional. The topics that the consolidated memories should be associated with.
+        # Corresponds to the JSON property `topics`
+        # @return [Array<Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1MemoryTopicId>]
+        attr_accessor :topics
+      
         def initialize(**args)
            update!(**args)
         end
@@ -17076,6 +17514,7 @@ module Google
         # Update properties of this object
         def update!(**args)
           @fact = args[:fact] if args.key?(:fact)
+          @topics = args[:topics] if args.key?(:topics)
         end
       end
       
@@ -18269,6 +18708,16 @@ module Google
         # @return [String]
         attr_accessor :aspect_ratio
       
+        # The image output format for generated images.
+        # Corresponds to the JSON property `imageOutputOptions`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1ImageConfigImageOutputOptions]
+        attr_accessor :image_output_options
+      
+        # Optional. Controls whether the model can generate people.
+        # Corresponds to the JSON property `personGeneration`
+        # @return [String]
+        attr_accessor :person_generation
+      
         def initialize(**args)
            update!(**args)
         end
@@ -18276,6 +18725,33 @@ module Google
         # Update properties of this object
         def update!(**args)
           @aspect_ratio = args[:aspect_ratio] if args.key?(:aspect_ratio)
+          @image_output_options = args[:image_output_options] if args.key?(:image_output_options)
+          @person_generation = args[:person_generation] if args.key?(:person_generation)
+        end
+      end
+      
+      # The image output format for generated images.
+      class GoogleCloudAiplatformV1beta1ImageConfigImageOutputOptions
+        include Google::Apis::Core::Hashable
+      
+        # Optional. The compression quality of the output image.
+        # Corresponds to the JSON property `compressionQuality`
+        # @return [Fixnum]
+        attr_accessor :compression_quality
+      
+        # Optional. The image format that the output should be saved as.
+        # Corresponds to the JSON property `mimeType`
+        # @return [String]
+        attr_accessor :mime_type
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @compression_quality = args[:compression_quality] if args.key?(:compression_quality)
+          @mime_type = args[:mime_type] if args.key?(:mime_type)
         end
       end
       
@@ -19677,6 +20153,25 @@ module Google
         end
       end
       
+      # An extracted memory that is the intermediate result before consolidation.
+      class GoogleCloudAiplatformV1beta1IntermediateExtractedMemory
+        include Google::Apis::Core::Hashable
+      
+        # Output only. The fact of the extracted memory.
+        # Corresponds to the JSON property `fact`
+        # @return [String]
+        attr_accessor :fact
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @fact = args[:fact] if args.key?(:fact)
+        end
+      end
+      
       # Request message for [InternalOsServiceStateInstance].
       class GoogleCloudAiplatformV1beta1InternalOsServiceStateInstance
         include Google::Apis::Core::Hashable
@@ -20746,6 +21241,32 @@ module Google
         end
       end
       
+      # Response message for MemoryBankService.ListMemoryRevisions.
+      class GoogleCloudAiplatformV1beta1ListMemoryRevisionsResponse
+        include Google::Apis::Core::Hashable
+      
+        # The list of Memory Revisions in the request page.
+        # Corresponds to the JSON property `memoryRevisions`
+        # @return [Array<Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1MemoryRevision>]
+        attr_accessor :memory_revisions
+      
+        # A token, which can be sent as `page_token` to retrieve the next page. If this
+        # field is omitted, there are no subsequent pages.
+        # 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)
+          @memory_revisions = args[:memory_revisions] if args.key?(:memory_revisions)
+          @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
@@ -21974,6 +22495,12 @@ module Google
         # @return [String]
         attr_accessor :description
       
+        # Optional. Input only. If true, no revision will be created for this request.
+        # Corresponds to the JSON property `disableMemoryRevisions`
+        # @return [Boolean]
+        attr_accessor :disable_memory_revisions
+        alias_method :disable_memory_revisions?, :disable_memory_revisions
+      
         # Optional. Display name of the Memory.
         # Corresponds to the JSON property `displayName`
         # @return [String]
@@ -21997,6 +22524,18 @@ module Google
         # @return [String]
         attr_accessor :name
       
+        # Optional. Input only. Timestamp of when the revision is considered expired. If
+        # not set, the memory revision will be kept until manually deleted.
+        # Corresponds to the JSON property `revisionExpireTime`
+        # @return [String]
+        attr_accessor :revision_expire_time
+      
+        # Optional. Input only. The TTL for the revision. The expiration time is
+        # computed: now + TTL.
+        # Corresponds to the JSON property `revisionTtl`
+        # @return [String]
+        attr_accessor :revision_ttl
+      
         # 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 '*'.
@@ -22004,6 +22543,11 @@ module Google
         # @return [Hash<String,String>]
         attr_accessor :scope
       
+        # Optional. The Topics of the Memory.
+        # Corresponds to the JSON property `topics`
+        # @return [Array<Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1MemoryTopicId>]
+        attr_accessor :topics
+      
         # Optional. Input only. The TTL for this resource. The expiration time is
         # computed: now + TTL.
         # Corresponds to the JSON property `ttl`
@@ -22023,11 +22567,15 @@ module Google
         def update!(**args)
           @create_time = args[:create_time] if args.key?(:create_time)
           @description = args[:description] if args.key?(:description)
+          @disable_memory_revisions = args[:disable_memory_revisions] if args.key?(:disable_memory_revisions)
           @display_name = args[:display_name] if args.key?(:display_name)
           @expire_time = args[:expire_time] if args.key?(:expire_time)
           @fact = args[:fact] if args.key?(:fact)
           @name = args[:name] if args.key?(:name)
+          @revision_expire_time = args[:revision_expire_time] if args.key?(:revision_expire_time)
+          @revision_ttl = args[:revision_ttl] if args.key?(:revision_ttl)
           @scope = args[:scope] if args.key?(:scope)
+          @topics = args[:topics] if args.key?(:topics)
           @ttl = args[:ttl] if args.key?(:ttl)
           @update_time = args[:update_time] if args.key?(:update_time)
         end
@@ -22238,6 +22786,62 @@ module Google
         end
       end
       
+      # A revision of a Memory.
+      class GoogleCloudAiplatformV1beta1MemoryRevision
+        include Google::Apis::Core::Hashable
+      
+        # Output only. Timestamp when this Memory Revision was created.
+        # Corresponds to the JSON property `createTime`
+        # @return [String]
+        attr_accessor :create_time
+      
+        # Output only. Timestamp of when this resource is considered expired.
+        # Corresponds to the JSON property `expireTime`
+        # @return [String]
+        attr_accessor :expire_time
+      
+        # Output only. The extracted memories from the source content before
+        # consolidation when the memory was updated via GenerateMemories. This
+        # information was used to modify an existing Memory via Consolidation.
+        # Corresponds to the JSON property `extractedMemories`
+        # @return [Array<Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1IntermediateExtractedMemory>]
+        attr_accessor :extracted_memories
+      
+        # Output only. The fact of the Memory Revision. This corresponds to the `fact`
+        # field of the parent Memory at the time of revision creation.
+        # Corresponds to the JSON property `fact`
+        # @return [String]
+        attr_accessor :fact
+      
+        # Output only. The labels of the Memory Revision. These labels are applied to
+        # the MemoryRevision when it is created based on `GenerateMemoriesRequest.
+        # revision_labels`.
+        # Corresponds to the JSON property `labels`
+        # @return [Hash<String,String>]
+        attr_accessor :labels
+      
+        # Identifier. The resource name of the Memory Revision. Format: `projects/`
+        # project`/locations/`location`/reasoningEngines/`reasoning_engine`/memories/`
+        # memory`/revisions/`memory_revision``
+        # 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)
+          @create_time = args[:create_time] if args.key?(:create_time)
+          @expire_time = args[:expire_time] if args.key?(:expire_time)
+          @extracted_memories = args[:extracted_memories] if args.key?(:extracted_memories)
+          @fact = args[:fact] if args.key?(:fact)
+          @labels = args[:labels] if args.key?(:labels)
+          @name = args[:name] if args.key?(:name)
+        end
+      end
+      
       # A memory topic identifier. This will be used to label a Memory and to restrict
       # which topics are eligible for generation or retrieval.
       class GoogleCloudAiplatformV1beta1MemoryTopicId
@@ -27962,6 +28566,11 @@ module Google
         # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1Content]
         attr_accessor :content
       
+        # Optional. The target model to optimize the prompt for.
+        # Corresponds to the JSON property `optimizationTarget`
+        # @return [String]
+        attr_accessor :optimization_target
+      
         def initialize(**args)
            update!(**args)
         end
@@ -27969,6 +28578,7 @@ module Google
         # Update properties of this object
         def update!(**args)
           @content = args[:content] if args.key?(:content)
+          @optimization_target = args[:optimization_target] if args.key?(:optimization_target)
         end
       end
       
@@ -29931,6 +30541,12 @@ module Google
         # @return [Array<Object>]
         attr_accessor :instances
       
+        # Optional. The user labels for Imagen billing usage only. Only Imagen supports
+        # labels. For other use cases, it will be ignored.
+        # Corresponds to the JSON property `labels`
+        # @return [Hash<String,String>]
+        attr_accessor :labels
+      
         # The parameters that govern the prediction. The schema of the parameters may be
         # specified via Endpoint's DeployedModels' Model's PredictSchemata's
         # parameters_schema_uri.
@@ -29945,6 +30561,7 @@ module Google
         # Update properties of this object
         def update!(**args)
           @instances = args[:instances] if args.key?(:instances)
+          @labels = args[:labels] if args.key?(:labels)
           @parameters = args[:parameters] if args.key?(:parameters)
         end
       end
@@ -30226,11 +30843,6 @@ module Google
       class GoogleCloudAiplatformV1beta1PreferenceOptimizationSpec
         include Google::Apis::Core::Hashable
       
-        # Evaluation Config for Tuning Job.
-        # Corresponds to the JSON property `evaluationConfig`
-        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1EvaluationConfig]
-        attr_accessor :evaluation_config
-      
         # Optional. If set to true, disable intermediate checkpoints for Preference
         # Optimization and only the last checkpoint will be exported. Otherwise, enable
         # intermediate checkpoints for Preference Optimization. Default is false.
@@ -30262,7 +30874,6 @@ module Google
       
         # Update properties of this object
         def update!(**args)
-          @evaluation_config = args[:evaluation_config] if args.key?(:evaluation_config)
           @export_last_checkpoint_only = args[:export_last_checkpoint_only] if args.key?(:export_last_checkpoint_only)
           @hyper_parameters = args[:hyper_parameters] if args.key?(:hyper_parameters)
           @training_dataset_uri = args[:training_dataset_uri] if args.key?(:training_dataset_uri)
@@ -34136,6 +34747,13 @@ module Google
         # @return [Array<Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1MemoryBankCustomizationConfig>]
         attr_accessor :customization_configs
       
+        # If true, no memory revisions will be created for any requests to the Memory
+        # Bank.
+        # Corresponds to the JSON property `disableMemoryRevisions`
+        # @return [Boolean]
+        attr_accessor :disable_memory_revisions
+        alias_method :disable_memory_revisions?, :disable_memory_revisions
+      
         # Configuration for how to generate memories.
         # Corresponds to the JSON property `generationConfig`
         # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1ReasoningEngineContextSpecMemoryBankConfigGenerationConfig]
@@ -34159,6 +34777,7 @@ module Google
         # Update properties of this object
         def update!(**args)
           @customization_configs = args[:customization_configs] if args.key?(:customization_configs)
+          @disable_memory_revisions = args[:disable_memory_revisions] if args.key?(:disable_memory_revisions)
           @generation_config = args[:generation_config] if args.key?(:generation_config)
           @similarity_search_config = args[:similarity_search_config] if args.key?(:similarity_search_config)
           @ttl_config = args[:ttl_config] if args.key?(:ttl_config)
@@ -34223,6 +34842,13 @@ module Google
         # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1ReasoningEngineContextSpecMemoryBankConfigTtlConfigGranularTtlConfig]
         attr_accessor :granular_ttl_config
       
+        # Optional. The default TTL duration of the memory revisions in the Memory Bank.
+        # This applies to all operations that create a memory revision. If not set, a
+        # default TTL of 365 days will be used.
+        # Corresponds to the JSON property `memoryRevisionDefaultTtl`
+        # @return [String]
+        attr_accessor :memory_revision_default_ttl
+      
         def initialize(**args)
            update!(**args)
         end
@@ -34231,6 +34857,7 @@ module Google
         def update!(**args)
           @default_ttl = args[:default_ttl] if args.key?(:default_ttl)
           @granular_ttl_config = args[:granular_ttl_config] if args.key?(:granular_ttl_config)
+          @memory_revision_default_ttl = args[:memory_revision_default_ttl] if args.key?(:memory_revision_default_ttl)
         end
       end
       
@@ -34307,6 +34934,11 @@ module Google
         # @return [String]
         attr_accessor :service_account
       
+        # Specification for deploying from source code.
+        # Corresponds to the JSON property `sourceCodeSpec`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1ReasoningEngineSpecSourceCodeSpec]
+        attr_accessor :source_code_spec
+      
         def initialize(**args)
            update!(**args)
         end
@@ -34318,6 +34950,7 @@ module Google
           @deployment_spec = args[:deployment_spec] if args.key?(:deployment_spec)
           @package_spec = args[:package_spec] if args.key?(:package_spec)
           @service_account = args[:service_account] if args.key?(:service_account)
+          @source_code_spec = args[:source_code_spec] if args.key?(:source_code_spec)
         end
       end
       
@@ -34435,6 +35068,95 @@ module Google
         end
       end
       
+      # Specification for deploying from source code.
+      class GoogleCloudAiplatformV1beta1ReasoningEngineSpecSourceCodeSpec
+        include Google::Apis::Core::Hashable
+      
+        # Specifies source code provided as a byte stream.
+        # Corresponds to the JSON property `inlineSource`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1ReasoningEngineSpecSourceCodeSpecInlineSource]
+        attr_accessor :inline_source
+      
+        # Specification for running a Python application from source.
+        # Corresponds to the JSON property `pythonSpec`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1ReasoningEngineSpecSourceCodeSpecPythonSpec]
+        attr_accessor :python_spec
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @inline_source = args[:inline_source] if args.key?(:inline_source)
+          @python_spec = args[:python_spec] if args.key?(:python_spec)
+        end
+      end
+      
+      # Specifies source code provided as a byte stream.
+      class GoogleCloudAiplatformV1beta1ReasoningEngineSpecSourceCodeSpecInlineSource
+        include Google::Apis::Core::Hashable
+      
+        # Required. Input only. The application source code archive, provided as a
+        # compressed tarball (.tar.gz) file.
+        # Corresponds to the JSON property `sourceArchive`
+        # NOTE: Values are automatically base64 encoded/decoded in the client library.
+        # @return [String]
+        attr_accessor :source_archive
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @source_archive = args[:source_archive] if args.key?(:source_archive)
+        end
+      end
+      
+      # Specification for running a Python application from source.
+      class GoogleCloudAiplatformV1beta1ReasoningEngineSpecSourceCodeSpecPythonSpec
+        include Google::Apis::Core::Hashable
+      
+        # Optional. The Python module to load as the entrypoint, specified as a fully
+        # qualified module name. For example: path.to.agent. If not specified, defaults
+        # to "agent". The project root will be added to Python sys.path, allowing
+        # imports to be specified relative to the root.
+        # Corresponds to the JSON property `entrypointModule`
+        # @return [String]
+        attr_accessor :entrypoint_module
+      
+        # Optional. The name of the callable object within the `entrypoint_module` to
+        # use as the application If not specified, defaults to "root_agent".
+        # Corresponds to the JSON property `entrypointObject`
+        # @return [String]
+        attr_accessor :entrypoint_object
+      
+        # Optional. The path to the requirements file, relative to the source root. If
+        # not specified, defaults to "requirements.txt".
+        # Corresponds to the JSON property `requirementsFile`
+        # @return [String]
+        attr_accessor :requirements_file
+      
+        # Optional. The version of Python to use. Support version includes 3.9, 3.10, 3.
+        # 11, 3.12, 3.13. If not specified, default value is 3.10.
+        # Corresponds to the JSON property `version`
+        # @return [String]
+        attr_accessor :version
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @entrypoint_module = args[:entrypoint_module] if args.key?(:entrypoint_module)
+          @entrypoint_object = args[:entrypoint_object] if args.key?(:entrypoint_object)
+          @requirements_file = args[:requirements_file] if args.key?(:requirements_file)
+          @version = args[:version] if args.key?(:version)
+        end
+      end
+      
       # Request message for GenAiTuningService.RebaseTunedModel.
       class GoogleCloudAiplatformV1beta1RebaseTunedModelRequest
         include Google::Apis::Core::Hashable
@@ -35325,6 +36047,13 @@ module Google
       class GoogleCloudAiplatformV1beta1RetrieveMemoriesRequest
         include Google::Apis::Core::Hashable
       
+        # Optional. The standard list filter that will be applied to the retrieved
+        # memories. More detail in [AIP-160](https://google.aip.dev/160). Supported
+        # fields: * `fact` * `create_time` * `update_time`
+        # Corresponds to the JSON property `filter`
+        # @return [String]
+        attr_accessor :filter
+      
         # 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.
@@ -35348,6 +36077,7 @@ module Google
       
         # Update properties of this object
         def update!(**args)
+          @filter = args[:filter] if args.key?(:filter)
           @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)
@@ -35465,6 +36195,25 @@ module Google
         end
       end
       
+      # Request message for MemoryBankService.RollbackMemory.
+      class GoogleCloudAiplatformV1beta1RollbackMemoryRequest
+        include Google::Apis::Core::Hashable
+      
+        # Required. The ID of the revision to rollback to.
+        # Corresponds to the JSON property `targetRevisionId`
+        # @return [Fixnum]
+        attr_accessor :target_revision_id
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @target_revision_id = args[:target_revision_id] if args.key?(:target_revision_id)
+        end
+      end
+      
       # Configuration for rolling deployments.
       class GoogleCloudAiplatformV1beta1RolloutOptions
         include Google::Apis::Core::Hashable
@@ -48030,6 +48779,11 @@ module Google
         # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1ToolCodeExecution]
         attr_accessor :code_execution
       
+        # Tool to support computer use.
+        # Corresponds to the JSON property `computerUse`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1ToolComputerUse]
+        attr_accessor :computer_use
+      
         # Tool to search public web data, powered by Vertex AI Search and Sec4
         # compliance.
         # Corresponds to the JSON property `enterpriseWebSearch`
@@ -48079,6 +48833,7 @@ module Google
         # Update properties of this object
         def update!(**args)
           @code_execution = args[:code_execution] if args.key?(:code_execution)
+          @computer_use = args[:computer_use] if args.key?(:computer_use)
           @enterprise_web_search = args[:enterprise_web_search] if args.key?(:enterprise_web_search)
           @function_declarations = args[:function_declarations] if args.key?(:function_declarations)
           @google_maps = args[:google_maps] if args.key?(:google_maps)
@@ -48230,6 +48985,36 @@ module Google
         end
       end
       
+      # Tool to support computer use.
+      class GoogleCloudAiplatformV1beta1ToolComputerUse
+        include Google::Apis::Core::Hashable
+      
+        # Required. The environment being operated.
+        # Corresponds to the JSON property `environment`
+        # @return [String]
+        attr_accessor :environment
+      
+        # Optional. By default, [predefined functions](https://cloud.google.com/vertex-
+        # ai/generative-ai/docs/computer-use#supported-actions) are included in the
+        # final model call. Some of them can be explicitly excluded from being
+        # automatically included. This can serve two purposes: 1. Using a more
+        # restricted / different action space. 2. Improving the definitions /
+        # instructions of predefined functions.
+        # Corresponds to the JSON property `excludedPredefinedFunctions`
+        # @return [Array<String>]
+        attr_accessor :excluded_predefined_functions
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @environment = args[:environment] if args.key?(:environment)
+          @excluded_predefined_functions = args[:excluded_predefined_functions] if args.key?(:excluded_predefined_functions)
+        end
+      end
+      
       # Tool config. This config is shared for all tools provided in the request.
       class GoogleCloudAiplatformV1beta1ToolConfig
         include Google::Apis::Core::Hashable
@@ -50910,6 +51695,97 @@ module Google
         end
       end
       
+      # Usage metadata about the content generation request and response. This message
+      # provides a detailed breakdown of token usage and other relevant metrics.
+      class GoogleCloudAiplatformV1beta1UsageMetadata
+        include Google::Apis::Core::Hashable
+      
+        # Output only. A detailed breakdown of the token count for each modality in the
+        # cached content.
+        # Corresponds to the JSON property `cacheTokensDetails`
+        # @return [Array<Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1ModalityTokenCount>]
+        attr_accessor :cache_tokens_details
+      
+        # Output only. The number of tokens in the cached content that was used for this
+        # request.
+        # Corresponds to the JSON property `cachedContentTokenCount`
+        # @return [Fixnum]
+        attr_accessor :cached_content_token_count
+      
+        # The total number of tokens in the generated candidates.
+        # Corresponds to the JSON property `candidatesTokenCount`
+        # @return [Fixnum]
+        attr_accessor :candidates_token_count
+      
+        # Output only. A detailed breakdown of the token count for each modality in the
+        # generated candidates.
+        # Corresponds to the JSON property `candidatesTokensDetails`
+        # @return [Array<Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1ModalityTokenCount>]
+        attr_accessor :candidates_tokens_details
+      
+        # The total number of tokens in the prompt. This includes any text, images, or
+        # other media provided in the request. When `cached_content` is set, this also
+        # includes the number of tokens in the cached content.
+        # Corresponds to the JSON property `promptTokenCount`
+        # @return [Fixnum]
+        attr_accessor :prompt_token_count
+      
+        # Output only. A detailed breakdown of the token count for each modality in the
+        # prompt.
+        # Corresponds to the JSON property `promptTokensDetails`
+        # @return [Array<Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1ModalityTokenCount>]
+        attr_accessor :prompt_tokens_details
+      
+        # Output only. The number of tokens that were part of the model's generated "
+        # thoughts" output, if applicable.
+        # Corresponds to the JSON property `thoughtsTokenCount`
+        # @return [Fixnum]
+        attr_accessor :thoughts_token_count
+      
+        # Output only. The number of tokens in the results from tool executions, which
+        # are provided back to the model as input, if applicable.
+        # Corresponds to the JSON property `toolUsePromptTokenCount`
+        # @return [Fixnum]
+        attr_accessor :tool_use_prompt_token_count
+      
+        # Output only. A detailed breakdown by modality of the token counts from the
+        # results of tool executions, which are provided back to the model as input.
+        # Corresponds to the JSON property `toolUsePromptTokensDetails`
+        # @return [Array<Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1ModalityTokenCount>]
+        attr_accessor :tool_use_prompt_tokens_details
+      
+        # The total number of tokens for the entire request. This is the sum of `
+        # prompt_token_count`, `candidates_token_count`, `tool_use_prompt_token_count`,
+        # and `thoughts_token_count`.
+        # Corresponds to the JSON property `totalTokenCount`
+        # @return [Fixnum]
+        attr_accessor :total_token_count
+      
+        # Output only. The traffic type for this request.
+        # Corresponds to the JSON property `trafficType`
+        # @return [String]
+        attr_accessor :traffic_type
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @cache_tokens_details = args[:cache_tokens_details] if args.key?(:cache_tokens_details)
+          @cached_content_token_count = args[:cached_content_token_count] if args.key?(:cached_content_token_count)
+          @candidates_token_count = args[:candidates_token_count] if args.key?(:candidates_token_count)
+          @candidates_tokens_details = args[:candidates_tokens_details] if args.key?(:candidates_tokens_details)
+          @prompt_token_count = args[:prompt_token_count] if args.key?(:prompt_token_count)
+          @prompt_tokens_details = args[:prompt_tokens_details] if args.key?(:prompt_tokens_details)
+          @thoughts_token_count = args[:thoughts_token_count] if args.key?(:thoughts_token_count)
+          @tool_use_prompt_token_count = args[:tool_use_prompt_token_count] if args.key?(:tool_use_prompt_token_count)
+          @tool_use_prompt_tokens_details = args[:tool_use_prompt_tokens_details] if args.key?(:tool_use_prompt_tokens_details)
+          @total_token_count = args[:total_token_count] if args.key?(:total_token_count)
+          @traffic_type = args[:traffic_type] if args.key?(:traffic_type)
+        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