google-apis-aiplatform_v1beta1 0.17.0 → 0.19.0

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

@@ -135,7 +135,7 @@ module Google
         # @return [Google::Apis::AiplatformV1beta1::CloudAiLargeModelsVisionImageRaiScores]
         attr_accessor :image_rai_scores
       
-        # RAI info for image
+        # RAI info for image.
         # Corresponds to the JSON property `raiInfo`
         # @return [Google::Apis::AiplatformV1beta1::CloudAiLargeModelsVisionRaiInfo]
         attr_accessor :rai_info
@@ -145,6 +145,11 @@ module Google
         # @return [Google::Apis::AiplatformV1beta1::CloudAiLargeModelsVisionSemanticFilterResponse]
         attr_accessor :semantic_filter_response
       
+        # Text/Expanded text input for imagen.
+        # Corresponds to the JSON property `text`
+        # @return [String]
+        attr_accessor :text
+      
         # Path to another storage (typically Google Cloud Storage).
         # Corresponds to the JSON property `uri`
         # @return [String]
@@ -161,6 +166,7 @@ module Google
           @image_rai_scores = args[:image_rai_scores] if args.key?(:image_rai_scores)
           @rai_info = args[:rai_info] if args.key?(:rai_info)
           @semantic_filter_response = args[:semantic_filter_response] if args.key?(:semantic_filter_response)
+          @text = args[:text] if args.key?(:text)
           @uri = args[:uri] if args.key?(:uri)
         end
       end
@@ -1060,11 +1066,31 @@ module Google
         # @return [String]
         attr_accessor :category
       
+        # The influential terms that could potentially block the response.
+        # Corresponds to the JSON property `influentialTerms`
+        # @return [Array<Google::Apis::AiplatformV1beta1::CloudAiNlLlmProtoServiceSafetyRatingInfluentialTerm>]
+        attr_accessor :influential_terms
+      
         # Harm probability levels in the content.
         # Corresponds to the JSON property `probability`
         # @return [String]
         attr_accessor :probability
       
+        # Harm probability score.
+        # Corresponds to the JSON property `probabilityScore`
+        # @return [Float]
+        attr_accessor :probability_score
+      
+        # Harm severity levels in the content.
+        # Corresponds to the JSON property `severity`
+        # @return [String]
+        attr_accessor :severity
+      
+        # Harm severity score.
+        # Corresponds to the JSON property `severityScore`
+        # @return [Float]
+        attr_accessor :severity_score
+      
         def initialize(**args)
            update!(**args)
         end
@@ -1073,7 +1099,48 @@ module Google
         def update!(**args)
           @blocked = args[:blocked] if args.key?(:blocked)
           @category = args[:category] if args.key?(:category)
+          @influential_terms = args[:influential_terms] if args.key?(:influential_terms)
           @probability = args[:probability] if args.key?(:probability)
+          @probability_score = args[:probability_score] if args.key?(:probability_score)
+          @severity = args[:severity] if args.key?(:severity)
+          @severity_score = args[:severity_score] if args.key?(:severity_score)
+        end
+      end
+      
+      # The influential term that could potentially block the response.
+      class CloudAiNlLlmProtoServiceSafetyRatingInfluentialTerm
+        include Google::Apis::Core::Hashable
+      
+        # The beginning offset of the influential term.
+        # Corresponds to the JSON property `beginOffset`
+        # @return [Fixnum]
+        attr_accessor :begin_offset
+      
+        # The confidence score of the influential term.
+        # Corresponds to the JSON property `confidence`
+        # @return [Float]
+        attr_accessor :confidence
+      
+        # The source of the influential term, prompt or response.
+        # Corresponds to the JSON property `source`
+        # @return [String]
+        attr_accessor :source
+      
+        # The influential term.
+        # Corresponds to the JSON property `term`
+        # @return [String]
+        attr_accessor :term
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @begin_offset = args[:begin_offset] if args.key?(:begin_offset)
+          @confidence = args[:confidence] if args.key?(:confidence)
+          @source = args[:source] if args.key?(:source)
+          @term = args[:term] if args.key?(:term)
         end
       end
       
@@ -3049,6 +3116,11 @@ module Google
         # @return [String]
         attr_accessor :finish_reason
       
+        # Metadata returned to client when grounding is enabled.
+        # Corresponds to the JSON property `groundingMetadata`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1GroundingMetadata]
+        attr_accessor :grounding_metadata
+      
         # Output only. Index of the candidate.
         # Corresponds to the JSON property `index`
         # @return [Fixnum]
@@ -3070,6 +3142,7 @@ module Google
           @content = args[:content] if args.key?(:content)
           @finish_message = args[:finish_message] if args.key?(:finish_message)
           @finish_reason = args[:finish_reason] if args.key?(:finish_reason)
+          @grounding_metadata = args[:grounding_metadata] if args.key?(:grounding_metadata)
           @index = args[:index] if args.key?(:index)
           @safety_ratings = args[:safety_ratings] if args.key?(:safety_ratings)
         end
@@ -4077,13 +4150,6 @@ module Google
         # @return [String]
         attr_accessor :pipeline_job_id
       
-        # Optional. Whether to do component level validations before job creation.
-        # Currently we only support Google First Party Component/Pipelines.
-        # Corresponds to the JSON property `preflightValidations`
-        # @return [Boolean]
-        attr_accessor :preflight_validations
-        alias_method :preflight_validations?, :preflight_validations
-      
         def initialize(**args)
            update!(**args)
         end
@@ -4093,7 +4159,6 @@ module Google
           @parent = args[:parent] if args.key?(:parent)
           @pipeline_job = args[:pipeline_job] if args.key?(:pipeline_job)
           @pipeline_job_id = args[:pipeline_job_id] if args.key?(:pipeline_job_id)
-          @preflight_validations = args[:preflight_validations] if args.key?(:preflight_validations)
         end
       end
       
@@ -4910,12 +4975,23 @@ module Google
         # @return [String]
         attr_accessor :create_time
       
+        # The user-defined name of the DatasetVersion. The name can be up to 128
+        # characters long and can consist of any UTF-8 characters.
+        # Corresponds to the JSON property `displayName`
+        # @return [String]
+        attr_accessor :display_name
+      
         # Used to perform consistent read-modify-write updates. If not set, a blind "
         # overwrite" update happens.
         # Corresponds to the JSON property `etag`
         # @return [String]
         attr_accessor :etag
       
+        # Required. Additional information about the DatasetVersion.
+        # Corresponds to the JSON property `metadata`
+        # @return [Object]
+        attr_accessor :metadata
+      
         # Output only. The resource name of the DatasetVersion.
         # Corresponds to the JSON property `name`
         # @return [String]
@@ -4934,7 +5010,9 @@ module Google
         def update!(**args)
           @big_query_dataset_name = args[:big_query_dataset_name] if args.key?(:big_query_dataset_name)
           @create_time = args[:create_time] if args.key?(:create_time)
+          @display_name = args[:display_name] if args.key?(:display_name)
           @etag = args[:etag] if args.key?(:etag)
+          @metadata = args[:metadata] if args.key?(:metadata)
           @name = args[:name] if args.key?(:name)
           @update_time = args[:update_time] if args.key?(:update_time)
         end
@@ -6066,6 +6144,11 @@ module Google
         # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1PredictRequestResponseLoggingConfig]
         attr_accessor :predict_request_response_logging_config
       
+        # Represents configuration for private service connect.
+        # Corresponds to the JSON property `privateServiceConnectConfig`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1PrivateServiceConnectConfig]
+        attr_accessor :private_service_connect_config
+      
         # A map from a DeployedModel's ID to the percentage of this Endpoint's traffic
         # that should be forwarded to that DeployedModel. If a DeployedModel's ID is not
         # listed in this map, then it receives no traffic. The traffic percentage values
@@ -6098,6 +6181,7 @@ module Google
           @name = args[:name] if args.key?(:name)
           @network = args[:network] if args.key?(:network)
           @predict_request_response_logging_config = args[:predict_request_response_logging_config] if args.key?(:predict_request_response_logging_config)
+          @private_service_connect_config = args[:private_service_connect_config] if args.key?(:private_service_connect_config)
           @traffic_split = args[:traffic_split] if args.key?(:traffic_split)
           @update_time = args[:update_time] if args.key?(:update_time)
         end
@@ -7454,63 +7538,6 @@ module Google
         end
       end
       
-      # Details of EndpointService.ExportEndpoint operation.
-      class GoogleCloudAiplatformV1beta1ExportEndpointOperationMetadata
-        include Google::Apis::Core::Hashable
-      
-        # Generic Metadata shared by all operations.
-        # Corresponds to the JSON property `genericMetadata`
-        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1GenericOperationMetadata]
-        attr_accessor :generic_metadata
-      
-        def initialize(**args)
-           update!(**args)
-        end
-      
-        # Update properties of this object
-        def update!(**args)
-          @generic_metadata = args[:generic_metadata] if args.key?(:generic_metadata)
-        end
-      end
-      
-      # Response message of EndpointService.ExportEndpoint operation.
-      class GoogleCloudAiplatformV1beta1ExportEndpointResponse
-        include Google::Apis::Core::Hashable
-      
-        # Describes the output of the ExportEndpoint.
-        # Corresponds to the JSON property `outputInfo`
-        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1ExportEndpointResponseOutputInfo]
-        attr_accessor :output_info
-      
-        def initialize(**args)
-           update!(**args)
-        end
-      
-        # Update properties of this object
-        def update!(**args)
-          @output_info = args[:output_info] if args.key?(:output_info)
-        end
-      end
-      
-      # Describes the output of the ExportEndpoint.
-      class GoogleCloudAiplatformV1beta1ExportEndpointResponseOutputInfo
-        include Google::Apis::Core::Hashable
-      
-        # The BigQuery location for the output content.
-        # Corresponds to the JSON property `bigQueryDestination`
-        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1BigQueryDestination]
-        attr_accessor :big_query_destination
-      
-        def initialize(**args)
-           update!(**args)
-        end
-      
-        # Update properties of this object
-        def update!(**args)
-          @big_query_destination = args[:big_query_destination] if args.key?(:big_query_destination)
-        end
-      end
-      
       # Details of operations that exports Features values.
       class GoogleCloudAiplatformV1beta1ExportFeatureValuesOperationMetadata
         include Google::Apis::Core::Hashable
@@ -7943,6 +7970,12 @@ module Google
         # @return [String]
         attr_accessor :name
       
+        # Entity responsible for maintaining this feature. Can be comma separated list
+        # of email addresses or URIs.
+        # Corresponds to the JSON property `pointOfContact`
+        # @return [String]
+        attr_accessor :point_of_contact
+      
         # Output only. Only applicable for Vertex AI Feature Store (Legacy). Timestamp
         # when this EntityType was most recently updated.
         # Corresponds to the JSON property `updateTime`
@@ -7977,6 +8010,7 @@ module Google
           @monitoring_stats = args[:monitoring_stats] if args.key?(:monitoring_stats)
           @monitoring_stats_anomalies = args[:monitoring_stats_anomalies] if args.key?(:monitoring_stats_anomalies)
           @name = args[:name] if args.key?(:name)
+          @point_of_contact = args[:point_of_contact] if args.key?(:point_of_contact)
           @update_time = args[:update_time] if args.key?(:update_time)
           @value_type = args[:value_type] if args.key?(:value_type)
           @version_column_name = args[:version_column_name] if args.key?(:version_column_name)
@@ -8055,8 +8089,8 @@ module Google
         # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1BigQuerySource]
         attr_accessor :big_query_source
       
-        # Optional. Columns to construct entity_id / row keys. Currently only supports 1
-        # entity_id_column. If not provided defaults to `entity_id`.
+        # Optional. Columns to construct entity_id / row keys. If not provided defaults
+        # to `entity_id`.
         # Corresponds to the JSON property `entityIdColumns`
         # @return [Array<String>]
         attr_accessor :entity_id_columns
@@ -8667,6 +8701,23 @@ module Google
         # @return [String]
         attr_accessor :name
       
+        # Output only. A Service Account unique to this FeatureView. The role bigquery.
+        # dataViewer should be granted to this service account to allow Vertex AI
+        # Feature Store to sync data to the online store.
+        # Corresponds to the JSON property `serviceAccountEmail`
+        # @return [String]
+        attr_accessor :service_account_email
+      
+        # Optional. Service agent type used during data sync. By default, the Vertex AI
+        # Service Agent is used. When using an IAM Policy to isolate this FeatureView
+        # within a project (https://cloud.google.com/vertex-ai/docs/featurestore/latest/
+        # resource-policy) a separate service account should be provisioned by setting
+        # this field to `SERVICE_AGENT_TYPE_FEATURE_VIEW`. This will generate a separate
+        # service account to access the BigQuery source table.
+        # Corresponds to the JSON property `serviceAgentType`
+        # @return [String]
+        attr_accessor :service_agent_type
+      
         # Configuration for Sync. Only one option is set.
         # Corresponds to the JSON property `syncConfig`
         # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1FeatureViewSyncConfig]
@@ -8694,6 +8745,8 @@ module Google
           @feature_registry_source = args[:feature_registry_source] if args.key?(:feature_registry_source)
           @labels = args[:labels] if args.key?(:labels)
           @name = args[:name] if args.key?(:name)
+          @service_account_email = args[:service_account_email] if args.key?(:service_account_email)
+          @service_agent_type = args[:service_agent_type] if args.key?(:service_agent_type)
           @sync_config = args[:sync_config] if args.key?(:sync_config)
           @update_time = args[:update_time] if args.key?(:update_time)
           @vector_search_config = args[:vector_search_config] if args.key?(:vector_search_config)
@@ -8704,8 +8757,7 @@ module Google
       class GoogleCloudAiplatformV1beta1FeatureViewBigQuerySource
         include Google::Apis::Core::Hashable
       
-        # Required. Columns to construct entity_id / row keys. Start by supporting 1
-        # only.
+        # Required. Columns to construct entity_id / row keys.
         # Corresponds to the JSON property `entityIdColumns`
         # @return [Array<String>]
         attr_accessor :entity_id_columns
@@ -8731,6 +8783,11 @@ module Google
       class GoogleCloudAiplatformV1beta1FeatureViewDataKey
         include Google::Apis::Core::Hashable
       
+        # ID that is comprised from several parts (columns).
+        # Corresponds to the JSON property `compositeKey`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1FeatureViewDataKeyCompositeKey]
+        attr_accessor :composite_key
+      
         # String key to use for lookup.
         # Corresponds to the JSON property `key`
         # @return [String]
@@ -8742,10 +8799,31 @@ module Google
       
         # Update properties of this object
         def update!(**args)
+          @composite_key = args[:composite_key] if args.key?(:composite_key)
           @key = args[:key] if args.key?(:key)
         end
       end
       
+      # ID that is comprised from several parts (columns).
+      class GoogleCloudAiplatformV1beta1FeatureViewDataKeyCompositeKey
+        include Google::Apis::Core::Hashable
+      
+        # Parts to construct Entity ID. Should match with the same ID columns as defined
+        # in FeatureView in the same order.
+        # Corresponds to the JSON property `parts`
+        # @return [Array<String>]
+        attr_accessor :parts
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @parts = args[:parts] if args.key?(:parts)
+        end
+      end
+      
       # A Feature Registry source for features that need to be synced to Online Store.
       class GoogleCloudAiplatformV1beta1FeatureViewFeatureRegistrySource
         include Google::Apis::Core::Hashable
@@ -8755,6 +8833,11 @@ module Google
         # @return [Array<Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1FeatureViewFeatureRegistrySourceFeatureGroup>]
         attr_accessor :feature_groups
       
+        # Optional. The project number of the parent project of the Feature Groups.
+        # Corresponds to the JSON property `projectNumber`
+        # @return [Fixnum]
+        attr_accessor :project_number
+      
         def initialize(**args)
            update!(**args)
         end
@@ -8762,6 +8845,7 @@ module Google
         # Update properties of this object
         def update!(**args)
           @feature_groups = args[:feature_groups] if args.key?(:feature_groups)
+          @project_number = args[:project_number] if args.key?(:project_number)
         end
       end
       
@@ -8828,6 +8912,12 @@ module Google
         # @return [Google::Apis::AiplatformV1beta1::GoogleTypeInterval]
         attr_accessor :run_time
       
+        # Summary from the Sync job. For continuous syncs, the summary is updated
+        # periodically. For batch syncs, it gets updated on completion of the sync.
+        # Corresponds to the JSON property `syncSummary`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1FeatureViewSyncSyncSummary]
+        attr_accessor :sync_summary
+      
         def initialize(**args)
            update!(**args)
         end
@@ -8838,6 +8928,7 @@ module Google
           @final_status = args[:final_status] if args.key?(:final_status)
           @name = args[:name] if args.key?(:name)
           @run_time = args[:run_time] if args.key?(:run_time)
+          @sync_summary = args[:sync_summary] if args.key?(:sync_summary)
         end
       end
       
@@ -8864,6 +8955,32 @@ module Google
         end
       end
       
+      # Summary from the Sync job. For continuous syncs, the summary is updated
+      # periodically. For batch syncs, it gets updated on completion of the sync.
+      class GoogleCloudAiplatformV1beta1FeatureViewSyncSyncSummary
+        include Google::Apis::Core::Hashable
+      
+        # Output only. Total number of rows synced.
+        # Corresponds to the JSON property `rowSynced`
+        # @return [Fixnum]
+        attr_accessor :row_synced
+      
+        # Output only. BigQuery slot milliseconds consumed for the sync job.
+        # Corresponds to the JSON property `totalSlot`
+        # @return [Fixnum]
+        attr_accessor :total_slot
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @row_synced = args[:row_synced] if args.key?(:row_synced)
+          @total_slot = args[:total_slot] if args.key?(:total_slot)
+        end
+      end
+      
       # Deprecated. Use IndexConfig instead.
       class GoogleCloudAiplatformV1beta1FeatureViewVectorSearchConfig
         include Google::Apis::Core::Hashable
@@ -9871,7 +9988,7 @@ module Google
         # Optional. A list of `Tools` the model may use to generate the next response. A
         # `Tool` is a piece of code that enables the system to interact with external
         # systems to perform an action, or set of actions, outside of knowledge and
-        # scope of the model. The only supported tool is currently `Function`
+        # scope of the model.
         # Corresponds to the JSON property `tools`
         # @return [Array<Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1Tool>]
         attr_accessor :tools
@@ -10065,6 +10182,130 @@ module Google
         end
       end
       
+      # Contains information about the source of the models generated from Generative
+      # AI Studio.
+      class GoogleCloudAiplatformV1beta1GenieSource
+        include Google::Apis::Core::Hashable
+      
+        # Required. The public base model URI.
+        # Corresponds to the JSON property `baseModelUri`
+        # @return [String]
+        attr_accessor :base_model_uri
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @base_model_uri = args[:base_model_uri] if args.key?(:base_model_uri)
+        end
+      end
+      
+      # Tool to retrieve public web data for grounding, powered by Google.
+      class GoogleCloudAiplatformV1beta1GoogleSearchRetrieval
+        include Google::Apis::Core::Hashable
+      
+        # Optional. Disable using the result from this tool in detecting grounding
+        # attribution. This does not affect how the result is given to the model for
+        # generation.
+        # Corresponds to the JSON property `disableAttribution`
+        # @return [Boolean]
+        attr_accessor :disable_attribution
+        alias_method :disable_attribution?, :disable_attribution
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @disable_attribution = args[:disable_attribution] if args.key?(:disable_attribution)
+        end
+      end
+      
+      # Grounding attribution.
+      class GoogleCloudAiplatformV1beta1GroundingAttribution
+        include Google::Apis::Core::Hashable
+      
+        # Optional. Output only. Confidence score of the attribution. Ranges from 0 to 1.
+        # 1 is the most confident.
+        # Corresponds to the JSON property `confidenceScore`
+        # @return [Float]
+        attr_accessor :confidence_score
+      
+        # Segment of the content.
+        # Corresponds to the JSON property `segment`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1Segment]
+        attr_accessor :segment
+      
+        # Attribution from the web.
+        # Corresponds to the JSON property `web`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1GroundingAttributionWeb]
+        attr_accessor :web
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @confidence_score = args[:confidence_score] if args.key?(:confidence_score)
+          @segment = args[:segment] if args.key?(:segment)
+          @web = args[:web] if args.key?(:web)
+        end
+      end
+      
+      # Attribution from the web.
+      class GoogleCloudAiplatformV1beta1GroundingAttributionWeb
+        include Google::Apis::Core::Hashable
+      
+        # Output only. Title of the attribution.
+        # Corresponds to the JSON property `title`
+        # @return [String]
+        attr_accessor :title
+      
+        # Output only. URI reference of the attribution.
+        # Corresponds to the JSON property `uri`
+        # @return [String]
+        attr_accessor :uri
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @title = args[:title] if args.key?(:title)
+          @uri = args[:uri] if args.key?(:uri)
+        end
+      end
+      
+      # Metadata returned to client when grounding is enabled.
+      class GoogleCloudAiplatformV1beta1GroundingMetadata
+        include Google::Apis::Core::Hashable
+      
+        # Optional. List of grounding attributions.
+        # Corresponds to the JSON property `groundingAttributions`
+        # @return [Array<Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1GroundingAttribution>]
+        attr_accessor :grounding_attributions
+      
+        # Optional. Web search queries for the following-up web search.
+        # Corresponds to the JSON property `webSearchQueries`
+        # @return [Array<String>]
+        attr_accessor :web_search_queries
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @grounding_attributions = args[:grounding_attributions] if args.key?(:grounding_attributions)
+          @web_search_queries = args[:web_search_queries] if args.key?(:web_search_queries)
+        end
+      end
+      
       # Represents a HyperparameterTuningJob. A HyperparameterTuningJob has a Study
       # specification and multiple CustomJobs with identical CustomJob specification.
       class GoogleCloudAiplatformV1beta1HyperparameterTuningJob
@@ -13248,6 +13489,12 @@ module Google
         # @return [String]
         attr_accessor :artifact_uri
       
+        # User input field to specify the base model source. Currently it only supports
+        # specifing the Model Garden models and Genie models.
+        # Corresponds to the JSON property `baseModelSource`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1ModelBaseModelSource]
+        attr_accessor :base_model_source
+      
         # Specification of a container for serving predictions. Some fields in this
         # message correspond to fields in the [Kubernetes Container v1 core
         # specification](https://kubernetes.io/docs/reference/generated/kubernetes-api/
@@ -13460,6 +13707,7 @@ module Google
         # Update properties of this object
         def update!(**args)
           @artifact_uri = args[:artifact_uri] if args.key?(:artifact_uri)
+          @base_model_source = args[:base_model_source] if args.key?(:base_model_source)
           @container_spec = args[:container_spec] if args.key?(:container_spec)
           @create_time = args[:create_time] if args.key?(:create_time)
           @deployed_models = args[:deployed_models] if args.key?(:deployed_models)
@@ -13490,6 +13738,34 @@ module Google
         end
       end
       
+      # User input field to specify the base model source. Currently it only supports
+      # specifing the Model Garden models and Genie models.
+      class GoogleCloudAiplatformV1beta1ModelBaseModelSource
+        include Google::Apis::Core::Hashable
+      
+        # Contains information about the source of the models generated from Generative
+        # AI Studio.
+        # Corresponds to the JSON property `genieSource`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1GenieSource]
+        attr_accessor :genie_source
+      
+        # Contains information about the source of the models generated from Model
+        # Garden.
+        # Corresponds to the JSON property `modelGardenSource`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1ModelGardenSource]
+        attr_accessor :model_garden_source
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @genie_source = args[:genie_source] if args.key?(:genie_source)
+          @model_garden_source = args[:model_garden_source] if args.key?(:model_garden_source)
+        end
+      end
+      
       # Specification of a container for serving predictions. Some fields in this
       # message correspond to fields in the [Kubernetes Container v1 core
       # specification](https://kubernetes.io/docs/reference/generated/kubernetes-api/
@@ -14435,6 +14711,26 @@ module Google
         end
       end
       
+      # Contains information about the source of the models generated from Model
+      # Garden.
+      class GoogleCloudAiplatformV1beta1ModelGardenSource
+        include Google::Apis::Core::Hashable
+      
+        # Required. The model garden source model resource name.
+        # Corresponds to the JSON property `publicModelName`
+        # @return [String]
+        attr_accessor :public_model_name
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @public_model_name = args[:public_model_name] if args.key?(:public_model_name)
+        end
+      end
+      
       # The alert config for model monitoring.
       class GoogleCloudAiplatformV1beta1ModelMonitoringAlertConfig
         include Google::Apis::Core::Hashable
@@ -16504,6 +16800,12 @@ module Google
         # @return [Hash<String,Object>]
         attr_accessor :pipeline_spec
       
+        # Optional. Whether to do component level validations before job creation.
+        # Corresponds to the JSON property `preflightValidations`
+        # @return [Boolean]
+        attr_accessor :preflight_validations
+        alias_method :preflight_validations?, :preflight_validations
+      
         # A list of names for the reserved ip ranges under the VPC network that can be
         # used for this Pipeline Job's workload. If set, we will deploy the Pipeline Job'
         # s workload within the provided ip ranges. Otherwise, the job will be deployed
@@ -16578,6 +16880,7 @@ module Google
           @name = args[:name] if args.key?(:name)
           @network = args[:network] if args.key?(:network)
           @pipeline_spec = args[:pipeline_spec] if args.key?(:pipeline_spec)
+          @preflight_validations = args[:preflight_validations] if args.key?(:preflight_validations)
           @reserved_ip_ranges = args[:reserved_ip_ranges] if args.key?(:reserved_ip_ranges)
           @runtime_config = args[:runtime_config] if args.key?(:runtime_config)
           @schedule_name = args[:schedule_name] if args.key?(:schedule_name)
@@ -16625,11 +16928,6 @@ module Google
       class GoogleCloudAiplatformV1beta1PipelineJobRuntimeConfig
         include Google::Apis::Core::Hashable
       
-        # The default runtime for the PipelineJob.
-        # Corresponds to the JSON property `defaultRuntime`
-        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1PipelineJobRuntimeConfigDefaultRuntime]
-        attr_accessor :default_runtime
-      
         # Represents the failure policy of a pipeline. Currently, the default of a
         # pipeline is that the pipeline will continue to run until no more tasks can be
         # executed, also known as PIPELINE_FAILURE_POLICY_FAIL_SLOW. However, if a
@@ -16680,7 +16978,6 @@ module Google
       
         # Update properties of this object
         def update!(**args)
-          @default_runtime = args[:default_runtime] if args.key?(:default_runtime)
           @failure_policy = args[:failure_policy] if args.key?(:failure_policy)
           @gcs_output_directory = args[:gcs_output_directory] if args.key?(:gcs_output_directory)
           @input_artifacts = args[:input_artifacts] if args.key?(:input_artifacts)
@@ -16689,26 +16986,6 @@ module Google
         end
       end
       
-      # The default runtime for the PipelineJob.
-      class GoogleCloudAiplatformV1beta1PipelineJobRuntimeConfigDefaultRuntime
-        include Google::Apis::Core::Hashable
-      
-        # Persistent resource based runtime detail. For more information, refer to https:
-        # //cloud.google.com/vertex-ai/docs/training/persistent-resource-overview
-        # Corresponds to the JSON property `persistentResourceRuntimeDetail`
-        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1PipelineJobRuntimeConfigPersistentResourceRuntimeDetail]
-        attr_accessor :persistent_resource_runtime_detail
-      
-        def initialize(**args)
-           update!(**args)
-        end
-      
-        # Update properties of this object
-        def update!(**args)
-          @persistent_resource_runtime_detail = args[:persistent_resource_runtime_detail] if args.key?(:persistent_resource_runtime_detail)
-        end
-      end
-      
       # The type of an input artifact.
       class GoogleCloudAiplatformV1beta1PipelineJobRuntimeConfigInputArtifact
         include Google::Apis::Core::Hashable
@@ -16731,27 +17008,6 @@ module Google
         end
       end
       
-      # Persistent resource based runtime detail. For more information, refer to https:
-      # //cloud.google.com/vertex-ai/docs/training/persistent-resource-overview
-      class GoogleCloudAiplatformV1beta1PipelineJobRuntimeConfigPersistentResourceRuntimeDetail
-        include Google::Apis::Core::Hashable
-      
-        # Persistent resource name. Format: `projects/`project`/locations/`location`/
-        # persistentResources/`persistent_resource``
-        # Corresponds to the JSON property `persistentResourceName`
-        # @return [String]
-        attr_accessor :persistent_resource_name
-      
-        def initialize(**args)
-           update!(**args)
-        end
-      
-        # Update properties of this object
-        def update!(**args)
-          @persistent_resource_name = args[:persistent_resource_name] if args.key?(:persistent_resource_name)
-        end
-      end
-      
       # The runtime detail of a task execution.
       class GoogleCloudAiplatformV1beta1PipelineTaskDetail
         include Google::Apis::Core::Hashable
@@ -18257,6 +18513,26 @@ module Google
         end
       end
       
+      # Configuration for the Ray metrics.
+      class GoogleCloudAiplatformV1beta1RayMetricSpec
+        include Google::Apis::Core::Hashable
+      
+        # Optional. Flag to disable the Ray metrics collection.
+        # Corresponds to the JSON property `disabled`
+        # @return [Boolean]
+        attr_accessor :disabled
+        alias_method :disabled?, :disabled
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @disabled = args[:disabled] if args.key?(:disabled)
+        end
+      end
+      
       # Configuration information for the Ray cluster. For experimental launch, Ray
       # cluster creation and Persistent cluster creation are 1:1 mapping: We will
       # provision all the nodes within the Persistent cluster as Ray nodes.
@@ -18280,6 +18556,11 @@ module Google
         # @return [String]
         attr_accessor :image_uri
       
+        # Configuration for the Ray metrics.
+        # Corresponds to the JSON property `rayMetricSpec`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1RayMetricSpec]
+        attr_accessor :ray_metric_spec
+      
         # Optional. Required if image_uri isn't set. A map of resource_pool_id to
         # prebuild Ray image if user need to use different images for different head/
         # worker pools. This map needs to cover all the resource pool ids. Example: ` "
@@ -18297,6 +18578,7 @@ module Google
         def update!(**args)
           @head_node_resource_pool_id = args[:head_node_resource_pool_id] if args.key?(:head_node_resource_pool_id)
           @image_uri = args[:image_uri] if args.key?(:image_uri)
+          @ray_metric_spec = args[:ray_metric_spec] if args.key?(:ray_metric_spec)
           @resource_pool_images = args[:resource_pool_images] if args.key?(:resource_pool_images)
         end
       end
@@ -19012,6 +19294,35 @@ module Google
         end
       end
       
+      # Defines a retrieval tool that model can call to access external knowledge.
+      class GoogleCloudAiplatformV1beta1Retrieval
+        include Google::Apis::Core::Hashable
+      
+        # Optional. Disable using the result from this tool in detecting grounding
+        # attribution. This does not affect how the result is given to the model for
+        # generation.
+        # Corresponds to the JSON property `disableAttribution`
+        # @return [Boolean]
+        attr_accessor :disable_attribution
+        alias_method :disable_attribution?, :disable_attribution
+      
+        # Retrieve from Vertex AI Search datastore for grounding. See https://cloud.
+        # google.com/vertex-ai-search-and-conversation
+        # Corresponds to the JSON property `vertexAiSearch`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1VertexAiSearch]
+        attr_accessor :vertex_ai_search
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @disable_attribution = args[:disable_attribution] if args.key?(:disable_attribution)
+          @vertex_ai_search = args[:vertex_ai_search] if args.key?(:vertex_ai_search)
+        end
+      end
+      
       # Safety rating corresponding to the generated content.
       class GoogleCloudAiplatformV1beta1SafetyRating
         include Google::Apis::Core::Hashable
@@ -19033,6 +19344,21 @@ module Google
         # @return [String]
         attr_accessor :probability
       
+        # Output only. Harm probability score.
+        # Corresponds to the JSON property `probabilityScore`
+        # @return [Float]
+        attr_accessor :probability_score
+      
+        # Output only. Harm severity levels in the content.
+        # Corresponds to the JSON property `severity`
+        # @return [String]
+        attr_accessor :severity
+      
+        # Output only. Harm severity score.
+        # Corresponds to the JSON property `severityScore`
+        # @return [Float]
+        attr_accessor :severity_score
+      
         def initialize(**args)
            update!(**args)
         end
@@ -19042,6 +19368,9 @@ module Google
           @blocked = args[:blocked] if args.key?(:blocked)
           @category = args[:category] if args.key?(:category)
           @probability = args[:probability] if args.key?(:probability)
+          @probability_score = args[:probability_score] if args.key?(:probability_score)
+          @severity = args[:severity] if args.key?(:severity)
+          @severity_score = args[:severity_score] if args.key?(:severity_score)
         end
       end
       
@@ -25709,6 +26038,39 @@ module Google
         end
       end
       
+      # Segment of the content.
+      class GoogleCloudAiplatformV1beta1Segment
+        include Google::Apis::Core::Hashable
+      
+        # Output only. End index in the given Part, measured in bytes. Offset from the
+        # start of the Part, exclusive, starting at zero.
+        # Corresponds to the JSON property `endIndex`
+        # @return [Fixnum]
+        attr_accessor :end_index
+      
+        # Output only. The index of a Part object within its parent Content object.
+        # Corresponds to the JSON property `partIndex`
+        # @return [Fixnum]
+        attr_accessor :part_index
+      
+        # Output only. Start index in the given Part, measured in bytes. Offset from the
+        # start of the Part, inclusive, starting at zero.
+        # Corresponds to the JSON property `startIndex`
+        # @return [Fixnum]
+        attr_accessor :start_index
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @end_index = args[:end_index] if args.key?(:end_index)
+          @part_index = args[:part_index] if args.key?(:part_index)
+          @start_index = args[:start_index] if args.key?(:start_index)
+        end
+      end
+      
       # Configuration for the use of custom service account to run the workloads.
       class GoogleCloudAiplatformV1beta1ServiceAccountSpec
         include Google::Apis::Core::Hashable
@@ -27755,20 +28117,32 @@ module Google
       
       # Tool details that the model may use to generate response. A `Tool` is a piece
       # of code that enables the system to interact with external systems to perform
-      # an action, or set of actions, outside of knowledge and scope of the model.
+      # an action, or set of actions, outside of knowledge and scope of the model. A
+      # Tool object should contain exactly one type of Tool (e.g FunctionDeclaration,
+      # Retrieval or GoogleSearchRetrieval).
       class GoogleCloudAiplatformV1beta1Tool
         include Google::Apis::Core::Hashable
       
-        # Optional. One or more function declarations to be passed to the model along
-        # with the current user query. Model may decide to call a subset of these
-        # functions by populating FunctionCall in the response. User should provide a
-        # FunctionResponse for each function call in the next turn. Based on the
-        # function responses, Model will generate the final response back to the user.
-        # Maximum 64 function declarations can be provided.
+        # Optional. Function tool type. One or more function declarations to be passed
+        # to the model along with the current user query. Model may decide to call a
+        # subset of these functions by populating FunctionCall in the response. User
+        # should provide a FunctionResponse for each function call in the next turn.
+        # Based on the function responses, Model will generate the final response back
+        # to the user. Maximum 64 function declarations can be provided.
         # Corresponds to the JSON property `functionDeclarations`
         # @return [Array<Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1FunctionDeclaration>]
         attr_accessor :function_declarations
       
+        # Tool to retrieve public web data for grounding, powered by Google.
+        # Corresponds to the JSON property `googleSearchRetrieval`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1GoogleSearchRetrieval]
+        attr_accessor :google_search_retrieval
+      
+        # Defines a retrieval tool that model can call to access external knowledge.
+        # Corresponds to the JSON property `retrieval`
+        # @return [Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1Retrieval]
+        attr_accessor :retrieval
+      
         def initialize(**args)
            update!(**args)
         end
@@ -27776,6 +28150,8 @@ module Google
         # Update properties of this object
         def update!(**args)
           @function_declarations = args[:function_declarations] if args.key?(:function_declarations)
+          @google_search_retrieval = args[:google_search_retrieval] if args.key?(:google_search_retrieval)
+          @retrieval = args[:retrieval] if args.key?(:retrieval)
         end
       end
       
@@ -28806,6 +29182,27 @@ module Google
         end
       end
       
+      # Retrieve from Vertex AI Search datastore for grounding. See https://cloud.
+      # google.com/vertex-ai-search-and-conversation
+      class GoogleCloudAiplatformV1beta1VertexAiSearch
+        include Google::Apis::Core::Hashable
+      
+        # Required. Fully-qualified Vertex AI Search's datastore resource ID. projects/<>
+        # /locations/<>/collections/<>/dataStores/<>
+        # Corresponds to the JSON property `datastore`
+        # @return [String]
+        attr_accessor :datastore
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @datastore = args[:datastore] if args.key?(:datastore)
+        end
+      end
+      
       # Metadata describes the input video content.
       class GoogleCloudAiplatformV1beta1VideoMetadata
         include Google::Apis::Core::Hashable
@@ -30168,7 +30565,8 @@ module Google
       
         # The recitation action for one given input. When its segments contain different
         # actions, the overall action will be returned in the precedence of BLOCK > CITE
-        # > NO_ACTION.
+        # > NO_ACTION. When the given input is not found in any source, the recitation
+        # action will not be specified.
         # Corresponds to the JSON property `recitationAction`
         # @return [String]
         attr_accessor :recitation_action
@@ -30522,7 +30920,8 @@ module Google
       
         # The recitation action for one given input. When its segments contain different
         # actions, the overall action will be returned in the precedence of BLOCK > CITE
-        # > NO_ACTION.
+        # > NO_ACTION. When the given input is not found in any source, the recitation
+        # action will be NO_ACTION.
         # Corresponds to the JSON property `recitationAction`
         # @return [String]
         attr_accessor :recitation_action
@@ -30734,6 +31133,125 @@ module Google
         end
       end
       
+      # Stores all metadata relating to AIDA DoConversation.
+      class LearningGenaiRootCodeyChatMetadata
+        include Google::Apis::Core::Hashable
+      
+        # Indicates the programming language of the code if the message is a code chunk.
+        # Corresponds to the JSON property `codeLanguage`
+        # @return [String]
+        attr_accessor :code_language
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @code_language = args[:code_language] if args.key?(:code_language)
+        end
+      end
+      
+      # Describes a sample at a checkpoint for post-processing.
+      class LearningGenaiRootCodeyCheckpoint
+        include Google::Apis::Core::Hashable
+      
+        # Metadata describing what was truncated at each checkpoint.
+        # Corresponds to the JSON property `codeyTruncatorMetadata`
+        # @return [Google::Apis::AiplatformV1beta1::LearningGenaiRootCodeyTruncatorMetadata]
+        attr_accessor :codey_truncator_metadata
+      
+        # Current state of the sample after truncator.
+        # Corresponds to the JSON property `currentSample`
+        # @return [String]
+        attr_accessor :current_sample
+      
+        # Postprocessor run that yielded this checkpoint.
+        # Corresponds to the JSON property `postInferenceStep`
+        # @return [String]
+        attr_accessor :post_inference_step
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @codey_truncator_metadata = args[:codey_truncator_metadata] if args.key?(:codey_truncator_metadata)
+          @current_sample = args[:current_sample] if args.key?(:current_sample)
+          @post_inference_step = args[:post_inference_step] if args.key?(:post_inference_step)
+        end
+      end
+      
+      # Stores all metadata relating to Completion.
+      class LearningGenaiRootCodeyCompletionMetadata
+        include Google::Apis::Core::Hashable
+      
+        # 
+        # Corresponds to the JSON property `checkpoints`
+        # @return [Array<Google::Apis::AiplatformV1beta1::LearningGenaiRootCodeyCheckpoint>]
+        attr_accessor :checkpoints
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @checkpoints = args[:checkpoints] if args.key?(:checkpoints)
+        end
+      end
+      
+      # Top-level wrapper used to store all things codey-related.
+      class LearningGenaiRootCodeyOutput
+        include Google::Apis::Core::Hashable
+      
+        # Stores all metadata relating to AIDA DoConversation.
+        # Corresponds to the JSON property `codeyChatMetadata`
+        # @return [Google::Apis::AiplatformV1beta1::LearningGenaiRootCodeyChatMetadata]
+        attr_accessor :codey_chat_metadata
+      
+        # Stores all metadata relating to Completion.
+        # Corresponds to the JSON property `codeyCompletionMetadata`
+        # @return [Google::Apis::AiplatformV1beta1::LearningGenaiRootCodeyCompletionMetadata]
+        attr_accessor :codey_completion_metadata
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @codey_chat_metadata = args[:codey_chat_metadata] if args.key?(:codey_chat_metadata)
+          @codey_completion_metadata = args[:codey_completion_metadata] if args.key?(:codey_completion_metadata)
+        end
+      end
+      
+      # Metadata describing what was truncated at each checkpoint.
+      class LearningGenaiRootCodeyTruncatorMetadata
+        include Google::Apis::Core::Hashable
+      
+        # Index of the current sample that trims off truncated text.
+        # Corresponds to the JSON property `cutoffIndex`
+        # @return [Fixnum]
+        attr_accessor :cutoff_index
+      
+        # Text that was truncated at a specific checkpoint.
+        # Corresponds to the JSON property `truncatedText`
+        # @return [String]
+        attr_accessor :truncated_text
+      
+        def initialize(**args)
+           update!(**args)
+        end
+      
+        # Update properties of this object
+        def update!(**args)
+          @cutoff_index = args[:cutoff_index] if args.key?(:cutoff_index)
+          @truncated_text = args[:truncated_text] if args.key?(:truncated_text)
+        end
+      end
+      
       # 
       class LearningGenaiRootDataProviderOutput
         include Google::Apis::Core::Hashable
@@ -31952,6 +32470,11 @@ module Google
         # @return [Google::Apis::AiplatformV1beta1::LearningGenaiRootClassifierOutputSummary]
         attr_accessor :classifier_summary
       
+        # Top-level wrapper used to store all things codey-related.
+        # Corresponds to the JSON property `codeyOutput`
+        # @return [Google::Apis::AiplatformV1beta1::LearningGenaiRootCodeyOutput]
+        attr_accessor :codey_output
+      
         # 
         # Corresponds to the JSON property `currentStreamTextLength`
         # @return [Fixnum]
@@ -31984,6 +32507,12 @@ module Google
         # @return [Google::Apis::AiplatformV1beta1::LearningGenaiRootGroundingMetadata]
         attr_accessor :grounding_metadata
       
+        # Applies to streaming response message only. Whether the message is a code.
+        # Corresponds to the JSON property `isCode`
+        # @return [Boolean]
+        attr_accessor :is_code
+        alias_method :is_code?, :is_code
+      
         # Applies to Response message only. Indicates whether the message is a fallback
         # and the response would have otherwise been empty.
         # Corresponds to the JSON property `isFallback`
@@ -32076,12 +32605,14 @@ module Google
         # Update properties of this object
         def update!(**args)
           @classifier_summary = args[:classifier_summary] if args.key?(:classifier_summary)
+          @codey_output = args[:codey_output] if args.key?(:codey_output)
           @current_stream_text_length = args[:current_stream_text_length] if args.key?(:current_stream_text_length)
           @deleted = args[:deleted] if args.key?(:deleted)
           @filter_meta = args[:filter_meta] if args.key?(:filter_meta)
           @final_message_score = args[:final_message_score] if args.key?(:final_message_score)
           @finish_reason = args[:finish_reason] if args.key?(:finish_reason)
           @grounding_metadata = args[:grounding_metadata] if args.key?(:grounding_metadata)
+          @is_code = args[:is_code] if args.key?(:is_code)
           @is_fallback = args[:is_fallback] if args.key?(:is_fallback)
           @langid_result = args[:langid_result] if args.key?(:langid_result)
           @language = args[:language] if args.key?(:language)