google-cloud-ai_platform-v1 0.35.0 → 0.37.0

data/lib/google/cloud/aiplatform/v1/tool_pb.rb

@@ -0,0 +1,51 @@
+# frozen_string_literal: true
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: google/cloud/aiplatform/v1/tool.proto
+
+require 'google/protobuf'
+
+require 'google/api/field_behavior_pb'
+require 'google/cloud/aiplatform/v1/openapi_pb'
+require 'google/protobuf/struct_pb'
+
+
+descriptor_data = "\n%google/cloud/aiplatform/v1/tool.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a(google/cloud/aiplatform/v1/openapi.proto\x1a\x1cgoogle/protobuf/struct.proto\"[\n\x04Tool\x12S\n\x15\x66unction_declarations\x18\x01 \x03(\x0b\x32/.google.cloud.aiplatform.v1.FunctionDeclarationB\x03\xe0\x41\x01\"\x7f\n\x13\x46unctionDeclaration\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x18\n\x0b\x64\x65scription\x18\x02 \x01(\tB\x03\xe0\x41\x01\x12;\n\nparameters\x18\x03 \x01(\x0b\x32\".google.cloud.aiplatform.v1.SchemaB\x03\xe0\x41\x01\"M\n\x0c\x46unctionCall\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12*\n\x04\x61rgs\x18\x02 \x01(\x0b\x32\x17.google.protobuf.StructB\x03\xe0\x41\x01\"U\n\x10\x46unctionResponse\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12.\n\x08response\x18\x02 \x01(\x0b\x32\x17.google.protobuf.StructB\x03\xe0\x41\x02\x42\xc7\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\tToolProtoP\x01Z>cloud.google.com/go/aiplatform/apiv1/aiplatformpb;aiplatformpb\xaa\x02\x1aGoogle.Cloud.AIPlatform.V1\xca\x02\x1aGoogle\\Cloud\\AIPlatform\\V1\xea\x02\x1dGoogle::Cloud::AIPlatform::V1b\x06proto3"
+
+pool = Google::Protobuf::DescriptorPool.generated_pool
+
+begin
+  pool.add_serialized_file(descriptor_data)
+rescue TypeError
+  # Compatibility code: will be removed in the next major version.
+  require 'google/protobuf/descriptor_pb'
+  parsed = Google::Protobuf::FileDescriptorProto.decode(descriptor_data)
+  parsed.clear_dependency
+  serialized = parsed.class.encode(parsed)
+  file = pool.add_serialized_file(serialized)
+  warn "Warning: Protobuf detected an import path issue while loading generated file #{__FILE__}"
+  imports = [
+    ["google.cloud.aiplatform.v1.Schema", "google/cloud/aiplatform/v1/openapi.proto"],
+    ["google.protobuf.Struct", "google/protobuf/struct.proto"],
+  ]
+  imports.each do |type_name, expected_filename|
+    import_file = pool.lookup(type_name).file_descriptor
+    if import_file.name != expected_filename
+      warn "- #{file.name} imports #{expected_filename}, but that import was loaded as #{import_file.name}"
+    end
+  end
+  warn "Each proto file must use a consistent fully-qualified name."
+  warn "This will become an error in the next major version."
+end
+
+module Google
+  module Cloud
+    module AIPlatform
+      module V1
+        Tool = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.Tool").msgclass
+        FunctionDeclaration = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.FunctionDeclaration").msgclass
+        FunctionCall = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.FunctionCall").msgclass
+        FunctionResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.FunctionResponse").msgclass
+      end
+    end
+  end
+end

data/lib/google/cloud/aiplatform/v1/training_pipeline_pb.rb

@@ -21,7 +21,7 @@ pool = Google::Protobuf::DescriptorPool.generated_pool
 
 begin
   pool.add_serialized_file(descriptor_data)
-rescue TypeError => e
+rescue TypeError
   # Compatibility code: will be removed in the next major version.
   require 'google/protobuf/descriptor_pb'
   parsed = Google::Protobuf::FileDescriptorProto.decode(descriptor_data)

data/lib/google/cloud/aiplatform/v1/types_pb.rb

@@ -11,7 +11,7 @@ pool = Google::Protobuf::DescriptorPool.generated_pool
 
 begin
   pool.add_serialized_file(descriptor_data)
-rescue TypeError => e
+rescue TypeError
   # Compatibility code: will be removed in the next major version.
   require 'google/protobuf/descriptor_pb'
   parsed = Google::Protobuf::FileDescriptorProto.decode(descriptor_data)

data/lib/google/cloud/aiplatform/v1/unmanaged_container_model_pb.rb

@@ -14,7 +14,7 @@ pool = Google::Protobuf::DescriptorPool.generated_pool
 
 begin
   pool.add_serialized_file(descriptor_data)
-rescue TypeError => e
+rescue TypeError
   # Compatibility code: will be removed in the next major version.
   require 'google/protobuf/descriptor_pb'
   parsed = Google::Protobuf::FileDescriptorProto.decode(descriptor_data)

data/lib/google/cloud/aiplatform/v1/user_action_reference_pb.rb

@@ -11,7 +11,7 @@ pool = Google::Protobuf::DescriptorPool.generated_pool
 
 begin
   pool.add_serialized_file(descriptor_data)
-rescue TypeError => e
+rescue TypeError
   # Compatibility code: will be removed in the next major version.
   require 'google/protobuf/descriptor_pb'
   parsed = Google::Protobuf::FileDescriptorProto.decode(descriptor_data)

data/lib/google/cloud/aiplatform/v1/value_pb.rb

@@ -11,7 +11,7 @@ pool = Google::Protobuf::DescriptorPool.generated_pool
 
 begin
   pool.add_serialized_file(descriptor_data)
-rescue TypeError => e
+rescue TypeError
   # Compatibility code: will be removed in the next major version.
   require 'google/protobuf/descriptor_pb'
   parsed = Google::Protobuf::FileDescriptorProto.decode(descriptor_data)

data/lib/google/cloud/aiplatform/v1/vizier_service_pb.rb

@@ -15,13 +15,13 @@ require 'google/protobuf/empty_pb'
 require 'google/protobuf/timestamp_pb'
 
 
-descriptor_data = "\n/google/cloud/aiplatform/v1/vizier_service.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1cgoogle/api/annotations.proto\x1a\x17google/api/client.proto\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a*google/cloud/aiplatform/v1/operation.proto\x1a&google/cloud/aiplatform/v1/study.proto\x1a#google/longrunning/operations.proto\x1a\x1bgoogle/protobuf/empty.proto\x1a\x1fgoogle/protobuf/timestamp.proto\"H\n\x0fGetStudyRequest\x12\x35\n\x04name\x18\x01 \x01(\tB\'\xe0\x41\x02\xfa\x41!\n\x1f\x61iplatform.googleapis.com/Study\"\x86\x01\n\x12\x43reateStudyRequest\x12\x39\n\x06parent\x18\x01 \x01(\tB)\xe0\x41\x02\xfa\x41#\n!locations.googleapis.com/Location\x12\x35\n\x05study\x18\x02 \x01(\x0b\x32!.google.cloud.aiplatform.v1.StudyB\x03\xe0\x41\x02\"\x80\x01\n\x12ListStudiesRequest\x12\x39\n\x06parent\x18\x01 \x01(\tB)\xe0\x41\x02\xfa\x41#\n!locations.googleapis.com/Location\x12\x17\n\npage_token\x18\x02 \x01(\tB\x03\xe0\x41\x01\x12\x16\n\tpage_size\x18\x03 \x01(\x05\x42\x03\xe0\x41\x01\"b\n\x13ListStudiesResponse\x12\x32\n\x07studies\x18\x01 \x03(\x0b\x32!.google.cloud.aiplatform.v1.Study\x12\x17\n\x0fnext_page_token\x18\x02 \x01(\t\"K\n\x12\x44\x65leteStudyRequest\x12\x35\n\x04name\x18\x01 \x01(\tB\'\xe0\x41\x02\xfa\x41!\n\x1f\x61iplatform.googleapis.com/Study\"j\n\x12LookupStudyRequest\x12\x39\n\x06parent\x18\x01 \x01(\tB)\xe0\x41\x02\xfa\x41#\n!locations.googleapis.com/Location\x12\x19\n\x0c\x64isplay_name\x18\x02 \x01(\tB\x03\xe0\x41\x02\"\xc7\x01\n\x14SuggestTrialsRequest\x12\x37\n\x06parent\x18\x01 \x01(\tB\'\xe0\x41\x02\xfa\x41!\n\x1f\x61iplatform.googleapis.com/Study\x12\x1d\n\x10suggestion_count\x18\x02 \x01(\x05\x42\x03\xe0\x41\x02\x12\x16\n\tclient_id\x18\x03 \x01(\tB\x03\xe0\x41\x02\x12?\n\x08\x63ontexts\x18\x04 \x03(\x0b\x32(.google.cloud.aiplatform.v1.TrialContextB\x03\xe0\x41\x01\"\xe6\x01\n\x15SuggestTrialsResponse\x12\x31\n\x06trials\x18\x01 \x03(\x0b\x32!.google.cloud.aiplatform.v1.Trial\x12<\n\x0bstudy_state\x18\x02 \x01(\x0e\x32\'.google.cloud.aiplatform.v1.Study.State\x12.\n\nstart_time\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12,\n\x08\x65nd_time\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\"z\n\x15SuggestTrialsMetadata\x12N\n\x10generic_metadata\x18\x01 \x01(\x0b\x32\x34.google.cloud.aiplatform.v1.GenericOperationMetadata\x12\x11\n\tclient_id\x18\x02 \x01(\t\"\x84\x01\n\x12\x43reateTrialRequest\x12\x37\n\x06parent\x18\x01 \x01(\tB\'\xe0\x41\x02\xfa\x41!\n\x1f\x61iplatform.googleapis.com/Study\x12\x35\n\x05trial\x18\x02 \x01(\x0b\x32!.google.cloud.aiplatform.v1.TrialB\x03\xe0\x41\x02\"H\n\x0fGetTrialRequest\x12\x35\n\x04name\x18\x01 \x01(\tB\'\xe0\x41\x02\xfa\x41!\n\x1f\x61iplatform.googleapis.com/Trial\"}\n\x11ListTrialsRequest\x12\x37\n\x06parent\x18\x01 \x01(\tB\'\xe0\x41\x02\xfa\x41!\n\x1f\x61iplatform.googleapis.com/Study\x12\x17\n\npage_token\x18\x02 \x01(\tB\x03\xe0\x41\x01\x12\x16\n\tpage_size\x18\x03 \x01(\x05\x42\x03\xe0\x41\x01\"`\n\x12ListTrialsResponse\x12\x31\n\x06trials\x18\x01 \x03(\x0b\x32!.google.cloud.aiplatform.v1.Trial\x12\x17\n\x0fnext_page_token\x18\x02 \x01(\t\"\x9c\x01\n\x1a\x41\x64\x64TrialMeasurementRequest\x12;\n\ntrial_name\x18\x01 \x01(\tB\'\xe0\x41\x02\xfa\x41!\n\x1f\x61iplatform.googleapis.com/Trial\x12\x41\n\x0bmeasurement\x18\x03 \x01(\x0b\x32\'.google.cloud.aiplatform.v1.MeasurementB\x03\xe0\x41\x02\"\xd5\x01\n\x14\x43ompleteTrialRequest\x12\x35\n\x04name\x18\x01 \x01(\tB\'\xe0\x41\x02\xfa\x41!\n\x1f\x61iplatform.googleapis.com/Trial\x12G\n\x11\x66inal_measurement\x18\x02 \x01(\x0b\x32\'.google.cloud.aiplatform.v1.MeasurementB\x03\xe0\x41\x01\x12\x1d\n\x10trial_infeasible\x18\x03 \x01(\x08\x42\x03\xe0\x41\x01\x12\x1e\n\x11infeasible_reason\x18\x04 \x01(\tB\x03\xe0\x41\x01\"K\n\x12\x44\x65leteTrialRequest\x12\x35\n\x04name\x18\x01 \x01(\tB\'\xe0\x41\x02\xfa\x41!\n\x1f\x61iplatform.googleapis.com/Trial\"b\n#CheckTrialEarlyStoppingStateRequest\x12;\n\ntrial_name\x18\x01 \x01(\tB\'\xe0\x41\x02\xfa\x41!\n\x1f\x61iplatform.googleapis.com/Trial\";\n$CheckTrialEarlyStoppingStateResponse\x12\x13\n\x0bshould_stop\x18\x01 \x01(\x08\"\x95\x01\n%CheckTrialEarlyStoppingStateMetatdata\x12N\n\x10generic_metadata\x18\x01 \x01(\x0b\x32\x34.google.cloud.aiplatform.v1.GenericOperationMetadata\x12\r\n\x05study\x18\x02 \x01(\t\x12\r\n\x05trial\x18\x03 \x01(\t\"I\n\x10StopTrialRequest\x12\x35\n\x04name\x18\x01 \x01(\tB\'\xe0\x41\x02\xfa\x41!\n\x1f\x61iplatform.googleapis.com/Trial\"S\n\x18ListOptimalTrialsRequest\x12\x37\n\x06parent\x18\x01 \x01(\tB\'\xe0\x41\x02\xfa\x41!\n\x1f\x61iplatform.googleapis.com/Study\"V\n\x19ListOptimalTrialsResponse\x12\x39\n\x0eoptimal_trials\x18\x01 \x03(\x0b\x32!.google.cloud.aiplatform.v1.Trial2\xd4\x16\n\rVizierService\x12\xab\x01\n\x0b\x43reateStudy\x12..google.cloud.aiplatform.v1.CreateStudyRequest\x1a!.google.cloud.aiplatform.v1.Study\"I\x82\xd3\xe4\x93\x02\x34\"+/v1/{parent=projects/*/locations/*}/studies:\x05study\xda\x41\x0cparent,study\x12\x96\x01\n\x08GetStudy\x12+.google.cloud.aiplatform.v1.GetStudyRequest\x1a!.google.cloud.aiplatform.v1.Study\":\x82\xd3\xe4\x93\x02-\x12+/v1/{name=projects/*/locations/*/studies/*}\xda\x41\x04name\x12\xac\x01\n\x0bListStudies\x12..google.cloud.aiplatform.v1.ListStudiesRequest\x1a/.google.cloud.aiplatform.v1.ListStudiesResponse\"<\x82\xd3\xe4\x93\x02-\x12+/v1/{parent=projects/*/locations/*}/studies\xda\x41\x06parent\x12\x91\x01\n\x0b\x44\x65leteStudy\x12..google.cloud.aiplatform.v1.DeleteStudyRequest\x1a\x16.google.protobuf.Empty\":\x82\xd3\xe4\x93\x02-*+/v1/{name=projects/*/locations/*/studies/*}\xda\x41\x04name\x12\xa8\x01\n\x0bLookupStudy\x12..google.cloud.aiplatform.v1.LookupStudyRequest\x1a!.google.cloud.aiplatform.v1.Study\"F\x82\xd3\xe4\x93\x02\x37\"2/v1/{parent=projects/*/locations/*}/studies:lookup:\x01*\xda\x41\x06parent\x12\xda\x01\n\rSuggestTrials\x12\x30.google.cloud.aiplatform.v1.SuggestTrialsRequest\x1a\x1d.google.longrunning.Operation\"x\x82\xd3\xe4\x93\x02\x41\"</v1/{parent=projects/*/locations/*/studies/*}/trials:suggest:\x01*\xca\x41.\n\x15SuggestTrialsResponse\x12\x15SuggestTrialsMetadata\x12\xb4\x01\n\x0b\x43reateTrial\x12..google.cloud.aiplatform.v1.CreateTrialRequest\x1a!.google.cloud.aiplatform.v1.Trial\"R\x82\xd3\xe4\x93\x02=\"4/v1/{parent=projects/*/locations/*/studies/*}/trials:\x05trial\xda\x41\x0cparent,trial\x12\x9f\x01\n\x08GetTrial\x12+.google.cloud.aiplatform.v1.GetTrialRequest\x1a!.google.cloud.aiplatform.v1.Trial\"C\x82\xd3\xe4\x93\x02\x36\x12\x34/v1/{name=projects/*/locations/*/studies/*/trials/*}\xda\x41\x04name\x12\xb2\x01\n\nListTrials\x12-.google.cloud.aiplatform.v1.ListTrialsRequest\x1a..google.cloud.aiplatform.v1.ListTrialsResponse\"E\x82\xd3\xe4\x93\x02\x36\x12\x34/v1/{parent=projects/*/locations/*/studies/*}/trials\xda\x41\x06parent\x12\xcb\x01\n\x13\x41\x64\x64TrialMeasurement\x12\x36.google.cloud.aiplatform.v1.AddTrialMeasurementRequest\x1a!.google.cloud.aiplatform.v1.Trial\"Y\x82\xd3\xe4\x93\x02S\"N/v1/{trial_name=projects/*/locations/*/studies/*/trials/*}:addTrialMeasurement:\x01*\x12\xae\x01\n\rCompleteTrial\x12\x30.google.cloud.aiplatform.v1.CompleteTrialRequest\x1a!.google.cloud.aiplatform.v1.Trial\"H\x82\xd3\xe4\x93\x02\x42\"=/v1/{name=projects/*/locations/*/studies/*/trials/*}:complete:\x01*\x12\x9a\x01\n\x0b\x44\x65leteTrial\x12..google.cloud.aiplatform.v1.DeleteTrialRequest\x1a\x16.google.protobuf.Empty\"C\x82\xd3\xe4\x93\x02\x36*4/v1/{name=projects/*/locations/*/studies/*/trials/*}\xda\x41\x04name\x12\xb3\x02\n\x1c\x43heckTrialEarlyStoppingState\x12?.google.cloud.aiplatform.v1.CheckTrialEarlyStoppingStateRequest\x1a\x1d.google.longrunning.Operation\"\xb2\x01\x82\xd3\xe4\x93\x02\\\"W/v1/{trial_name=projects/*/locations/*/studies/*/trials/*}:checkTrialEarlyStoppingState:\x01*\xca\x41M\n$CheckTrialEarlyStoppingStateResponse\x12%CheckTrialEarlyStoppingStateMetatdata\x12\xa2\x01\n\tStopTrial\x12,.google.cloud.aiplatform.v1.StopTrialRequest\x1a!.google.cloud.aiplatform.v1.Trial\"D\x82\xd3\xe4\x93\x02>\"9/v1/{name=projects/*/locations/*/studies/*/trials/*}:stop:\x01*\x12\xdc\x01\n\x11ListOptimalTrials\x12\x34.google.cloud.aiplatform.v1.ListOptimalTrialsRequest\x1a\x35.google.cloud.aiplatform.v1.ListOptimalTrialsResponse\"Z\x82\xd3\xe4\x93\x02K\"F/v1/{parent=projects/*/locations/*/studies/*}/trials:listOptimalTrials:\x01*\xda\x41\x06parent\x1aM\xca\x41\x19\x61iplatform.googleapis.com\xd2\x41.https://www.googleapis.com/auth/cloud-platformB\xd0\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\x12VizierServiceProtoP\x01Z>cloud.google.com/go/aiplatform/apiv1/aiplatformpb;aiplatformpb\xaa\x02\x1aGoogle.Cloud.AIPlatform.V1\xca\x02\x1aGoogle\\Cloud\\AIPlatform\\V1\xea\x02\x1dGoogle::Cloud::AIPlatform::V1b\x06proto3"
+descriptor_data = "\n/google/cloud/aiplatform/v1/vizier_service.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1cgoogle/api/annotations.proto\x1a\x17google/api/client.proto\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a*google/cloud/aiplatform/v1/operation.proto\x1a&google/cloud/aiplatform/v1/study.proto\x1a#google/longrunning/operations.proto\x1a\x1bgoogle/protobuf/empty.proto\x1a\x1fgoogle/protobuf/timestamp.proto\"H\n\x0fGetStudyRequest\x12\x35\n\x04name\x18\x01 \x01(\tB\'\xe0\x41\x02\xfa\x41!\n\x1f\x61iplatform.googleapis.com/Study\"\x86\x01\n\x12\x43reateStudyRequest\x12\x39\n\x06parent\x18\x01 \x01(\tB)\xe0\x41\x02\xfa\x41#\n!locations.googleapis.com/Location\x12\x35\n\x05study\x18\x02 \x01(\x0b\x32!.google.cloud.aiplatform.v1.StudyB\x03\xe0\x41\x02\"\x80\x01\n\x12ListStudiesRequest\x12\x39\n\x06parent\x18\x01 \x01(\tB)\xe0\x41\x02\xfa\x41#\n!locations.googleapis.com/Location\x12\x17\n\npage_token\x18\x02 \x01(\tB\x03\xe0\x41\x01\x12\x16\n\tpage_size\x18\x03 \x01(\x05\x42\x03\xe0\x41\x01\"b\n\x13ListStudiesResponse\x12\x32\n\x07studies\x18\x01 \x03(\x0b\x32!.google.cloud.aiplatform.v1.Study\x12\x17\n\x0fnext_page_token\x18\x02 \x01(\t\"K\n\x12\x44\x65leteStudyRequest\x12\x35\n\x04name\x18\x01 \x01(\tB\'\xe0\x41\x02\xfa\x41!\n\x1f\x61iplatform.googleapis.com/Study\"j\n\x12LookupStudyRequest\x12\x39\n\x06parent\x18\x01 \x01(\tB)\xe0\x41\x02\xfa\x41#\n!locations.googleapis.com/Location\x12\x19\n\x0c\x64isplay_name\x18\x02 \x01(\tB\x03\xe0\x41\x02\"\xc7\x01\n\x14SuggestTrialsRequest\x12\x37\n\x06parent\x18\x01 \x01(\tB\'\xe0\x41\x02\xfa\x41!\n\x1f\x61iplatform.googleapis.com/Study\x12\x1d\n\x10suggestion_count\x18\x02 \x01(\x05\x42\x03\xe0\x41\x02\x12\x16\n\tclient_id\x18\x03 \x01(\tB\x03\xe0\x41\x02\x12?\n\x08\x63ontexts\x18\x04 \x03(\x0b\x32(.google.cloud.aiplatform.v1.TrialContextB\x03\xe0\x41\x01\"\xe6\x01\n\x15SuggestTrialsResponse\x12\x31\n\x06trials\x18\x01 \x03(\x0b\x32!.google.cloud.aiplatform.v1.Trial\x12<\n\x0bstudy_state\x18\x02 \x01(\x0e\x32\'.google.cloud.aiplatform.v1.Study.State\x12.\n\nstart_time\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12,\n\x08\x65nd_time\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\"z\n\x15SuggestTrialsMetadata\x12N\n\x10generic_metadata\x18\x01 \x01(\x0b\x32\x34.google.cloud.aiplatform.v1.GenericOperationMetadata\x12\x11\n\tclient_id\x18\x02 \x01(\t\"\x84\x01\n\x12\x43reateTrialRequest\x12\x37\n\x06parent\x18\x01 \x01(\tB\'\xe0\x41\x02\xfa\x41!\n\x1f\x61iplatform.googleapis.com/Study\x12\x35\n\x05trial\x18\x02 \x01(\x0b\x32!.google.cloud.aiplatform.v1.TrialB\x03\xe0\x41\x02\"H\n\x0fGetTrialRequest\x12\x35\n\x04name\x18\x01 \x01(\tB\'\xe0\x41\x02\xfa\x41!\n\x1f\x61iplatform.googleapis.com/Trial\"}\n\x11ListTrialsRequest\x12\x37\n\x06parent\x18\x01 \x01(\tB\'\xe0\x41\x02\xfa\x41!\n\x1f\x61iplatform.googleapis.com/Study\x12\x17\n\npage_token\x18\x02 \x01(\tB\x03\xe0\x41\x01\x12\x16\n\tpage_size\x18\x03 \x01(\x05\x42\x03\xe0\x41\x01\"`\n\x12ListTrialsResponse\x12\x31\n\x06trials\x18\x01 \x03(\x0b\x32!.google.cloud.aiplatform.v1.Trial\x12\x17\n\x0fnext_page_token\x18\x02 \x01(\t\"\x9c\x01\n\x1a\x41\x64\x64TrialMeasurementRequest\x12;\n\ntrial_name\x18\x01 \x01(\tB\'\xe0\x41\x02\xfa\x41!\n\x1f\x61iplatform.googleapis.com/Trial\x12\x41\n\x0bmeasurement\x18\x03 \x01(\x0b\x32\'.google.cloud.aiplatform.v1.MeasurementB\x03\xe0\x41\x02\"\xd5\x01\n\x14\x43ompleteTrialRequest\x12\x35\n\x04name\x18\x01 \x01(\tB\'\xe0\x41\x02\xfa\x41!\n\x1f\x61iplatform.googleapis.com/Trial\x12G\n\x11\x66inal_measurement\x18\x02 \x01(\x0b\x32\'.google.cloud.aiplatform.v1.MeasurementB\x03\xe0\x41\x01\x12\x1d\n\x10trial_infeasible\x18\x03 \x01(\x08\x42\x03\xe0\x41\x01\x12\x1e\n\x11infeasible_reason\x18\x04 \x01(\tB\x03\xe0\x41\x01\"K\n\x12\x44\x65leteTrialRequest\x12\x35\n\x04name\x18\x01 \x01(\tB\'\xe0\x41\x02\xfa\x41!\n\x1f\x61iplatform.googleapis.com/Trial\"b\n#CheckTrialEarlyStoppingStateRequest\x12;\n\ntrial_name\x18\x01 \x01(\tB\'\xe0\x41\x02\xfa\x41!\n\x1f\x61iplatform.googleapis.com/Trial\";\n$CheckTrialEarlyStoppingStateResponse\x12\x13\n\x0bshould_stop\x18\x01 \x01(\x08\"\x95\x01\n%CheckTrialEarlyStoppingStateMetatdata\x12N\n\x10generic_metadata\x18\x01 \x01(\x0b\x32\x34.google.cloud.aiplatform.v1.GenericOperationMetadata\x12\r\n\x05study\x18\x02 \x01(\t\x12\r\n\x05trial\x18\x03 \x01(\t\"I\n\x10StopTrialRequest\x12\x35\n\x04name\x18\x01 \x01(\tB\'\xe0\x41\x02\xfa\x41!\n\x1f\x61iplatform.googleapis.com/Trial\"S\n\x18ListOptimalTrialsRequest\x12\x37\n\x06parent\x18\x01 \x01(\tB\'\xe0\x41\x02\xfa\x41!\n\x1f\x61iplatform.googleapis.com/Study\"V\n\x19ListOptimalTrialsResponse\x12\x39\n\x0eoptimal_trials\x18\x01 \x03(\x0b\x32!.google.cloud.aiplatform.v1.Trial2\xd4\x16\n\rVizierService\x12\xab\x01\n\x0b\x43reateStudy\x12..google.cloud.aiplatform.v1.CreateStudyRequest\x1a!.google.cloud.aiplatform.v1.Study\"I\xda\x41\x0cparent,study\x82\xd3\xe4\x93\x02\x34\"+/v1/{parent=projects/*/locations/*}/studies:\x05study\x12\x96\x01\n\x08GetStudy\x12+.google.cloud.aiplatform.v1.GetStudyRequest\x1a!.google.cloud.aiplatform.v1.Study\":\xda\x41\x04name\x82\xd3\xe4\x93\x02-\x12+/v1/{name=projects/*/locations/*/studies/*}\x12\xac\x01\n\x0bListStudies\x12..google.cloud.aiplatform.v1.ListStudiesRequest\x1a/.google.cloud.aiplatform.v1.ListStudiesResponse\"<\xda\x41\x06parent\x82\xd3\xe4\x93\x02-\x12+/v1/{parent=projects/*/locations/*}/studies\x12\x91\x01\n\x0b\x44\x65leteStudy\x12..google.cloud.aiplatform.v1.DeleteStudyRequest\x1a\x16.google.protobuf.Empty\":\xda\x41\x04name\x82\xd3\xe4\x93\x02-*+/v1/{name=projects/*/locations/*/studies/*}\x12\xa8\x01\n\x0bLookupStudy\x12..google.cloud.aiplatform.v1.LookupStudyRequest\x1a!.google.cloud.aiplatform.v1.Study\"F\xda\x41\x06parent\x82\xd3\xe4\x93\x02\x37\"2/v1/{parent=projects/*/locations/*}/studies:lookup:\x01*\x12\xda\x01\n\rSuggestTrials\x12\x30.google.cloud.aiplatform.v1.SuggestTrialsRequest\x1a\x1d.google.longrunning.Operation\"x\xca\x41.\n\x15SuggestTrialsResponse\x12\x15SuggestTrialsMetadata\x82\xd3\xe4\x93\x02\x41\"</v1/{parent=projects/*/locations/*/studies/*}/trials:suggest:\x01*\x12\xb4\x01\n\x0b\x43reateTrial\x12..google.cloud.aiplatform.v1.CreateTrialRequest\x1a!.google.cloud.aiplatform.v1.Trial\"R\xda\x41\x0cparent,trial\x82\xd3\xe4\x93\x02=\"4/v1/{parent=projects/*/locations/*/studies/*}/trials:\x05trial\x12\x9f\x01\n\x08GetTrial\x12+.google.cloud.aiplatform.v1.GetTrialRequest\x1a!.google.cloud.aiplatform.v1.Trial\"C\xda\x41\x04name\x82\xd3\xe4\x93\x02\x36\x12\x34/v1/{name=projects/*/locations/*/studies/*/trials/*}\x12\xb2\x01\n\nListTrials\x12-.google.cloud.aiplatform.v1.ListTrialsRequest\x1a..google.cloud.aiplatform.v1.ListTrialsResponse\"E\xda\x41\x06parent\x82\xd3\xe4\x93\x02\x36\x12\x34/v1/{parent=projects/*/locations/*/studies/*}/trials\x12\xcb\x01\n\x13\x41\x64\x64TrialMeasurement\x12\x36.google.cloud.aiplatform.v1.AddTrialMeasurementRequest\x1a!.google.cloud.aiplatform.v1.Trial\"Y\x82\xd3\xe4\x93\x02S\"N/v1/{trial_name=projects/*/locations/*/studies/*/trials/*}:addTrialMeasurement:\x01*\x12\xae\x01\n\rCompleteTrial\x12\x30.google.cloud.aiplatform.v1.CompleteTrialRequest\x1a!.google.cloud.aiplatform.v1.Trial\"H\x82\xd3\xe4\x93\x02\x42\"=/v1/{name=projects/*/locations/*/studies/*/trials/*}:complete:\x01*\x12\x9a\x01\n\x0b\x44\x65leteTrial\x12..google.cloud.aiplatform.v1.DeleteTrialRequest\x1a\x16.google.protobuf.Empty\"C\xda\x41\x04name\x82\xd3\xe4\x93\x02\x36*4/v1/{name=projects/*/locations/*/studies/*/trials/*}\x12\xb3\x02\n\x1c\x43heckTrialEarlyStoppingState\x12?.google.cloud.aiplatform.v1.CheckTrialEarlyStoppingStateRequest\x1a\x1d.google.longrunning.Operation\"\xb2\x01\xca\x41M\n$CheckTrialEarlyStoppingStateResponse\x12%CheckTrialEarlyStoppingStateMetatdata\x82\xd3\xe4\x93\x02\\\"W/v1/{trial_name=projects/*/locations/*/studies/*/trials/*}:checkTrialEarlyStoppingState:\x01*\x12\xa2\x01\n\tStopTrial\x12,.google.cloud.aiplatform.v1.StopTrialRequest\x1a!.google.cloud.aiplatform.v1.Trial\"D\x82\xd3\xe4\x93\x02>\"9/v1/{name=projects/*/locations/*/studies/*/trials/*}:stop:\x01*\x12\xdc\x01\n\x11ListOptimalTrials\x12\x34.google.cloud.aiplatform.v1.ListOptimalTrialsRequest\x1a\x35.google.cloud.aiplatform.v1.ListOptimalTrialsResponse\"Z\xda\x41\x06parent\x82\xd3\xe4\x93\x02K\"F/v1/{parent=projects/*/locations/*/studies/*}/trials:listOptimalTrials:\x01*\x1aM\xca\x41\x19\x61iplatform.googleapis.com\xd2\x41.https://www.googleapis.com/auth/cloud-platformB\xd0\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\x12VizierServiceProtoP\x01Z>cloud.google.com/go/aiplatform/apiv1/aiplatformpb;aiplatformpb\xaa\x02\x1aGoogle.Cloud.AIPlatform.V1\xca\x02\x1aGoogle\\Cloud\\AIPlatform\\V1\xea\x02\x1dGoogle::Cloud::AIPlatform::V1b\x06proto3"
 
 pool = Google::Protobuf::DescriptorPool.generated_pool
 
 begin
   pool.add_serialized_file(descriptor_data)
-rescue TypeError => e
+rescue TypeError
   # Compatibility code: will be removed in the next major version.
   require 'google/protobuf/descriptor_pb'
   parsed = Google::Protobuf::FileDescriptorProto.decode(descriptor_data)

data/proto_docs/google/api/client.rb

@@ -21,6 +21,7 @@ module Google
   module Api
     # Required information for every language.
     # @!attribute [rw] reference_docs_uri
+    #   @deprecated This field is deprecated and may be removed in the next major version update.
     #   @return [::String]
     #     Link to automatically generated reference documentation.  Example:
     #     https://cloud.google.com/nodejs/docs/reference/asset/latest

data/proto_docs/google/cloud/aiplatform/v1/batch_prediction_job.rb

@@ -323,7 +323,7 @@ module Google
           #     {::Google::Cloud::AIPlatform::V1::BatchPredictionJob::InstanceConfig#excluded_fields excluded_fields}
           #     must be empty.
           #
-          #     The input must be JSONL with objects at each line, CSV, BigQuery
+          #     The input must be JSONL with objects at each line, BigQuery
           #     or TfRecord.
           # @!attribute [rw] excluded_fields
           #   @return [::Array<::String>]
@@ -338,7 +338,7 @@ module Google
           #     {::Google::Cloud::AIPlatform::V1::BatchPredictionJob::InstanceConfig#included_fields included_fields}
           #     must be empty.
           #
-          #     The input must be JSONL with objects at each line, CSV, BigQuery
+          #     The input must be JSONL with objects at each line, BigQuery
           #     or TfRecord.
           class InstanceConfig
             include ::Google::Protobuf::MessageExts

data/proto_docs/google/cloud/aiplatform/v1/content.rb

@@ -0,0 +1,312 @@
+# frozen_string_literal: true
+
+# Copyright 2023 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# Auto-generated by gapic-generator-ruby. DO NOT EDIT!
+
+
+module Google
+  module Cloud
+    module AIPlatform
+      module V1
+        # 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.
+        # @!attribute [rw] role
+        #   @return [::String]
+        #     Optional. The producer of the content. Must be either 'user' or 'model'.
+        #
+        #     Useful to set for multi-turn conversations, otherwise can be left blank
+        #     or unset.
+        # @!attribute [rw] parts
+        #   @return [::Array<::Google::Cloud::AIPlatform::V1::Part>]
+        #     Required. Ordered `Parts` that constitute a single message. Parts may have
+        #     different IANA MIME types.
+        class Content
+          include ::Google::Protobuf::MessageExts
+          extend ::Google::Protobuf::MessageExts::ClassMethods
+        end
+
+        # A datatype containing media that is part of a multi-part `Content` message.
+        #
+        # A `Part` consists of data which has an associated datatype. A `Part` can only
+        # contain one of the accepted types in `Part.data`.
+        #
+        # A `Part` must have a fixed IANA MIME type identifying the type and subtype
+        # of the media if `inline_data` or `file_data` field is filled with raw bytes.
+        # @!attribute [rw] text
+        #   @return [::String]
+        #     Optional. Text part (can be code).
+        # @!attribute [rw] inline_data
+        #   @return [::Google::Cloud::AIPlatform::V1::Blob]
+        #     Optional. Inlined bytes data.
+        # @!attribute [rw] file_data
+        #   @return [::Google::Cloud::AIPlatform::V1::FileData]
+        #     Optional. URI based data.
+        # @!attribute [rw] function_call
+        #   @return [::Google::Cloud::AIPlatform::V1::FunctionCall]
+        #     Optional. A predicted [FunctionCall] returned from the model that
+        #     contains a string representing the [FunctionDeclaration.name] with the
+        #     parameters and their values.
+        # @!attribute [rw] function_response
+        #   @return [::Google::Cloud::AIPlatform::V1::FunctionResponse]
+        #     Optional. The result output of a [FunctionCall] that contains a string
+        #     representing the [FunctionDeclaration.name] and a structured JSON object
+        #     containing any output from the function call. It is used as context to
+        #     the model.
+        # @!attribute [rw] video_metadata
+        #   @return [::Google::Cloud::AIPlatform::V1::VideoMetadata]
+        #     Optional. Video metadata. The metadata should only be specified while the
+        #     video data is presented in inline_data or file_data.
+        class Part
+          include ::Google::Protobuf::MessageExts
+          extend ::Google::Protobuf::MessageExts::ClassMethods
+        end
+
+        # Raw media bytes.
+        #
+        # Text should not be sent as raw bytes, use the 'text' field.
+        # @!attribute [rw] mime_type
+        #   @return [::String]
+        #     Required. The IANA standard MIME type of the source data.
+        # @!attribute [rw] data
+        #   @return [::String]
+        #     Required. Raw bytes for media formats.
+        class Blob
+          include ::Google::Protobuf::MessageExts
+          extend ::Google::Protobuf::MessageExts::ClassMethods
+        end
+
+        # URI based data.
+        # @!attribute [rw] mime_type
+        #   @return [::String]
+        #     Required. The IANA standard MIME type of the source data.
+        # @!attribute [rw] file_uri
+        #   @return [::String]
+        #     Required. URI.
+        class FileData
+          include ::Google::Protobuf::MessageExts
+          extend ::Google::Protobuf::MessageExts::ClassMethods
+        end
+
+        # Metadata describes the input video content.
+        # @!attribute [rw] start_offset
+        #   @return [::Google::Protobuf::Duration]
+        #     Optional. The start offset of the video.
+        # @!attribute [rw] end_offset
+        #   @return [::Google::Protobuf::Duration]
+        #     Optional. The end offset of the video.
+        class VideoMetadata
+          include ::Google::Protobuf::MessageExts
+          extend ::Google::Protobuf::MessageExts::ClassMethods
+        end
+
+        # Generation config.
+        # @!attribute [rw] temperature
+        #   @return [::Float]
+        #     Optional. Controls the randomness of predictions.
+        # @!attribute [rw] top_p
+        #   @return [::Float]
+        #     Optional. If specified, nucleus sampling will be used.
+        # @!attribute [rw] top_k
+        #   @return [::Float]
+        #     Optional. If specified, top-k sampling will be used.
+        # @!attribute [rw] candidate_count
+        #   @return [::Integer]
+        #     Optional. Number of candidates to generate.
+        # @!attribute [rw] max_output_tokens
+        #   @return [::Integer]
+        #     Optional. The maximum number of output tokens to generate per message.
+        # @!attribute [rw] stop_sequences
+        #   @return [::Array<::String>]
+        #     Optional. Stop sequences.
+        class GenerationConfig
+          include ::Google::Protobuf::MessageExts
+          extend ::Google::Protobuf::MessageExts::ClassMethods
+        end
+
+        # Safety settings.
+        # @!attribute [rw] category
+        #   @return [::Google::Cloud::AIPlatform::V1::HarmCategory]
+        #     Required. Harm category.
+        # @!attribute [rw] threshold
+        #   @return [::Google::Cloud::AIPlatform::V1::SafetySetting::HarmBlockThreshold]
+        #     Required. The harm block threshold.
+        class SafetySetting
+          include ::Google::Protobuf::MessageExts
+          extend ::Google::Protobuf::MessageExts::ClassMethods
+
+          # Probability based thresholds levels for blocking.
+          module HarmBlockThreshold
+            # Unspecified harm block threshold.
+            HARM_BLOCK_THRESHOLD_UNSPECIFIED = 0
+
+            # Block low threshold and above (i.e. block more).
+            BLOCK_LOW_AND_ABOVE = 1
+
+            # Block medium threshold and above.
+            BLOCK_MEDIUM_AND_ABOVE = 2
+
+            # Block only high threshold (i.e. block less).
+            BLOCK_ONLY_HIGH = 3
+
+            # Block none.
+            BLOCK_NONE = 4
+          end
+        end
+
+        # Safety rating corresponding to the generated content.
+        # @!attribute [r] category
+        #   @return [::Google::Cloud::AIPlatform::V1::HarmCategory]
+        #     Output only. Harm category.
+        # @!attribute [r] probability
+        #   @return [::Google::Cloud::AIPlatform::V1::SafetyRating::HarmProbability]
+        #     Output only. Harm probability levels in the content.
+        # @!attribute [r] blocked
+        #   @return [::Boolean]
+        #     Output only. Indicates whether the content was filtered out because of this
+        #     rating.
+        class SafetyRating
+          include ::Google::Protobuf::MessageExts
+          extend ::Google::Protobuf::MessageExts::ClassMethods
+
+          # Harm probability levels in the content.
+          module HarmProbability
+            # Harm probability unspecified.
+            HARM_PROBABILITY_UNSPECIFIED = 0
+
+            # Negligible level of harm.
+            NEGLIGIBLE = 1
+
+            # Low level of harm.
+            LOW = 2
+
+            # Medium level of harm.
+            MEDIUM = 3
+
+            # High level of harm.
+            HIGH = 4
+          end
+        end
+
+        # A collection of source attributions for a piece of content.
+        # @!attribute [r] citations
+        #   @return [::Array<::Google::Cloud::AIPlatform::V1::Citation>]
+        #     Output only. List of citations.
+        class CitationMetadata
+          include ::Google::Protobuf::MessageExts
+          extend ::Google::Protobuf::MessageExts::ClassMethods
+        end
+
+        # Source attributions for content.
+        # @!attribute [r] start_index
+        #   @return [::Integer]
+        #     Output only. Start index into the content.
+        # @!attribute [r] end_index
+        #   @return [::Integer]
+        #     Output only. End index into the content.
+        # @!attribute [r] uri
+        #   @return [::String]
+        #     Output only. Url reference of the attribution.
+        # @!attribute [r] title
+        #   @return [::String]
+        #     Output only. Title of the attribution.
+        # @!attribute [r] license
+        #   @return [::String]
+        #     Output only. License of the attribution.
+        # @!attribute [r] publication_date
+        #   @return [::Google::Type::Date]
+        #     Output only. Publication date of the attribution.
+        class Citation
+          include ::Google::Protobuf::MessageExts
+          extend ::Google::Protobuf::MessageExts::ClassMethods
+        end
+
+        # A response candidate generated from the model.
+        # @!attribute [r] index
+        #   @return [::Integer]
+        #     Output only. Index of the candidate.
+        # @!attribute [r] content
+        #   @return [::Google::Cloud::AIPlatform::V1::Content]
+        #     Output only. Content parts of the candidate.
+        # @!attribute [r] finish_reason
+        #   @return [::Google::Cloud::AIPlatform::V1::Candidate::FinishReason]
+        #     Output only. The reason why the model stopped generating tokens.
+        #     If empty, the model has not stopped generating the tokens.
+        # @!attribute [r] safety_ratings
+        #   @return [::Array<::Google::Cloud::AIPlatform::V1::SafetyRating>]
+        #     Output only. List of ratings for the safety of a response candidate.
+        #
+        #     There is at most one rating per category.
+        # @!attribute [r] finish_message
+        #   @return [::String]
+        #     Output only. Describes the reason the mode stopped generating tokens in
+        #     more detail. This is only filled when `finish_reason` is set.
+        # @!attribute [r] citation_metadata
+        #   @return [::Google::Cloud::AIPlatform::V1::CitationMetadata]
+        #     Output only. Source attribution of the generated content.
+        class Candidate
+          include ::Google::Protobuf::MessageExts
+          extend ::Google::Protobuf::MessageExts::ClassMethods
+
+          # The reason why the model stopped generating tokens.
+          # If empty, the model has not stopped generating the tokens.
+          module FinishReason
+            # The finish reason is unspecified.
+            FINISH_REASON_UNSPECIFIED = 0
+
+            # Natural stop point of the model or provided stop sequence.
+            STOP = 1
+
+            # The maximum number of tokens as specified in the request was reached.
+            MAX_TOKENS = 2
+
+            # The token generation was stopped as the response was flagged for safety
+            # reasons. NOTE: When streaming the Candidate.content will be empty if
+            # content filters blocked the output.
+            SAFETY = 3
+
+            # The token generation was stopped as the response was flagged for
+            # unauthorized citations.
+            RECITATION = 4
+
+            # All other reasons that stopped the token generation
+            OTHER = 5
+          end
+        end
+
+        # Harm categories that will block the content.
+        module HarmCategory
+          # The harm category is unspecified.
+          HARM_CATEGORY_UNSPECIFIED = 0
+
+          # The harm category is hate speech.
+          HARM_CATEGORY_HATE_SPEECH = 1
+
+          # The harm category is dangerous content.
+          HARM_CATEGORY_DANGEROUS_CONTENT = 2
+
+          # The harm category is harassment.
+          HARM_CATEGORY_HARASSMENT = 3
+
+          # The harm category is sexually explicit content.
+          HARM_CATEGORY_SEXUALLY_EXPLICIT = 4
+        end
+      end
+    end
+  end
+end

data/proto_docs/google/cloud/aiplatform/v1/custom_job.rb

@@ -219,6 +219,21 @@ module Google
         #     Optional. The Experiment Run associated with this job.
         #     Format:
         #     `projects/{project}/locations/{location}/metadataStores/{metadataStores}/contexts/{experiment-name}-{experiment-run-name}`
+        # @!attribute [rw] models
+        #   @return [::Array<::String>]
+        #     Optional. The name of the Model resources for which to generate a mapping
+        #     to artifact URIs. Applicable only to some of the Google-provided custom
+        #     jobs. Format: `projects/{project}/locations/{location}/models/{model}`
+        #
+        #     In order to retrieve a specific version of the model, also provide
+        #     the version ID or version alias.
+        #       Example: `projects/{project}/locations/{location}/models/{model}@2`
+        #                  or
+        #                `projects/{project}/locations/{location}/models/{model}@golden`
+        #     If no version ID or alias is specified, the "default" version will be
+        #     returned. The "default" version alias is created for the first version of
+        #     the model, and can be moved to other versions later on. There will be
+        #     exactly one default version.
         class CustomJobSpec
           include ::Google::Protobuf::MessageExts
           extend ::Google::Protobuf::MessageExts::ClassMethods

data/proto_docs/google/cloud/aiplatform/v1/dataset.rb

@@ -184,15 +184,78 @@ module Google
         # @!attribute [rw] fraction_split
         #   @return [::Google::Cloud::AIPlatform::V1::ExportFractionSplit]
         #     Split based on fractions defining the size of each set.
+        # @!attribute [rw] filter_split
+        #   @return [::Google::Cloud::AIPlatform::V1::ExportFilterSplit]
+        #     Split based on the provided filters for each set.
         # @!attribute [rw] annotations_filter
         #   @return [::String]
         #     An expression for filtering what part of the Dataset is to be exported.
         #     Only Annotations that match this filter will be exported. The filter syntax
         #     is the same as in
         #     {::Google::Cloud::AIPlatform::V1::DatasetService::Client#list_annotations ListAnnotations}.
+        # @!attribute [rw] saved_query_id
+        #   @return [::String]
+        #     The ID of a SavedQuery (annotation set) under the Dataset specified by
+        #     [dataset_id][] used for filtering Annotations for training.
+        #
+        #     Only used for custom training data export use cases.
+        #     Only applicable to Datasets that have SavedQueries.
+        #
+        #     Only Annotations that are associated with this SavedQuery are used in
+        #     respectively training. When used in conjunction with
+        #     {::Google::Cloud::AIPlatform::V1::ExportDataConfig#annotations_filter annotations_filter},
+        #     the Annotations used for training are filtered by both
+        #     {::Google::Cloud::AIPlatform::V1::ExportDataConfig#saved_query_id saved_query_id}
+        #     and
+        #     {::Google::Cloud::AIPlatform::V1::ExportDataConfig#annotations_filter annotations_filter}.
+        #
+        #     Only one of
+        #     {::Google::Cloud::AIPlatform::V1::ExportDataConfig#saved_query_id saved_query_id}
+        #     and
+        #     {::Google::Cloud::AIPlatform::V1::ExportDataConfig#annotation_schema_uri annotation_schema_uri}
+        #     should be specified as both of them represent the same thing: problem type.
+        # @!attribute [rw] annotation_schema_uri
+        #   @return [::String]
+        #     The Cloud Storage URI that points to a YAML file describing the annotation
+        #     schema. The schema is defined as an OpenAPI 3.0.2 [Schema
+        #     Object](https://github.com/OAI/OpenAPI-Specification/blob/main/versions/3.0.2.md#schemaObject).
+        #     The schema files that can be used here are found in
+        #     gs://google-cloud-aiplatform/schema/dataset/annotation/, note that the
+        #     chosen schema must be consistent with
+        #     {::Google::Cloud::AIPlatform::V1::Dataset#metadata_schema_uri metadata} of the
+        #     Dataset specified by [dataset_id][].
+        #
+        #     Only used for custom training data export use cases.
+        #     Only applicable to Datasets that have DataItems and Annotations.
+        #
+        #     Only Annotations that both match this schema and belong to DataItems not
+        #     ignored by the split method are used in respectively training, validation
+        #     or test role, depending on the role of the DataItem they are on.
+        #
+        #     When used in conjunction with
+        #     {::Google::Cloud::AIPlatform::V1::ExportDataConfig#annotations_filter annotations_filter},
+        #     the Annotations used for training are filtered by both
+        #     {::Google::Cloud::AIPlatform::V1::ExportDataConfig#annotations_filter annotations_filter}
+        #     and
+        #     {::Google::Cloud::AIPlatform::V1::ExportDataConfig#annotation_schema_uri annotation_schema_uri}.
+        # @!attribute [rw] export_use
+        #   @return [::Google::Cloud::AIPlatform::V1::ExportDataConfig::ExportUse]
+        #     Indicates the usage of the exported files.
         class ExportDataConfig
           include ::Google::Protobuf::MessageExts
           extend ::Google::Protobuf::MessageExts::ClassMethods
+
+          # ExportUse indicates the usage of the exported files. It restricts file
+          # destination, format, annotations to be exported, whether to allow
+          # unannotated data to be exported and whether to clone files to temp Cloud
+          # Storage bucket.
+          module ExportUse
+            # Regular user export.
+            EXPORT_USE_UNSPECIFIED = 0
+
+            # Export for custom code training.
+            CUSTOM_CODE_TRAINING = 6
+          end
         end
 
         # Assigns the input data to training, validation, and test sets as per the
@@ -214,6 +277,45 @@ module Google
           include ::Google::Protobuf::MessageExts
           extend ::Google::Protobuf::MessageExts::ClassMethods
         end
+
+        # Assigns input data to training, validation, and test sets based on the given
+        # filters, data pieces not matched by any filter are ignored. Currently only
+        # supported for Datasets containing DataItems.
+        # If any of the filters in this message are to match nothing, then they can be
+        # set as '-' (the minus sign).
+        #
+        # Supported only for unstructured Datasets.
+        # @!attribute [rw] training_filter
+        #   @return [::String]
+        #     Required. A filter on DataItems of the Dataset. DataItems that match
+        #     this filter are used to train the Model. A filter with same syntax
+        #     as the one used in
+        #     {::Google::Cloud::AIPlatform::V1::DatasetService::Client#list_data_items DatasetService.ListDataItems}
+        #     may be used. If a single DataItem is matched by more than one of the
+        #     FilterSplit filters, then it is assigned to the first set that applies to
+        #     it in the training, validation, test order.
+        # @!attribute [rw] validation_filter
+        #   @return [::String]
+        #     Required. A filter on DataItems of the Dataset. DataItems that match
+        #     this filter are used to validate the Model. A filter with same syntax
+        #     as the one used in
+        #     {::Google::Cloud::AIPlatform::V1::DatasetService::Client#list_data_items DatasetService.ListDataItems}
+        #     may be used. If a single DataItem is matched by more than one of the
+        #     FilterSplit filters, then it is assigned to the first set that applies to
+        #     it in the training, validation, test order.
+        # @!attribute [rw] test_filter
+        #   @return [::String]
+        #     Required. A filter on DataItems of the Dataset. DataItems that match
+        #     this filter are used to test the Model. A filter with same syntax
+        #     as the one used in
+        #     {::Google::Cloud::AIPlatform::V1::DatasetService::Client#list_data_items DatasetService.ListDataItems}
+        #     may be used. If a single DataItem is matched by more than one of the
+        #     FilterSplit filters, then it is assigned to the first set that applies to
+        #     it in the training, validation, test order.
+        class ExportFilterSplit
+          include ::Google::Protobuf::MessageExts
+          extend ::Google::Protobuf::MessageExts::ClassMethods
+        end
       end
     end
   end