google-cloud-ai_platform-v1 0.40.0 → 0.42.0

Sign up to get free protection for your applications and to get access to all the features.
Files changed (73) hide show
  1. checksums.yaml +4 -4
  2. data/lib/google/cloud/ai_platform/v1/dataset_service/client.rb +99 -0
  3. data/lib/google/cloud/ai_platform/v1/dataset_service/rest/client.rb +92 -0
  4. data/lib/google/cloud/ai_platform/v1/dataset_service/rest/operations.rb +280 -0
  5. data/lib/google/cloud/ai_platform/v1/dataset_service/rest/service_stub.rb +60 -0
  6. data/lib/google/cloud/ai_platform/v1/deployment_resource_pool_service/rest/operations.rb +280 -0
  7. data/lib/google/cloud/ai_platform/v1/endpoint_service/rest/operations.rb +280 -0
  8. data/lib/google/cloud/ai_platform/v1/feature_online_store_admin_service/rest/operations.rb +280 -0
  9. data/lib/google/cloud/ai_platform/v1/feature_registry_service/rest/operations.rb +280 -0
  10. data/lib/google/cloud/ai_platform/v1/featurestore_service/rest/operations.rb +280 -0
  11. data/lib/google/cloud/ai_platform/v1/index_endpoint_service/rest/operations.rb +280 -0
  12. data/lib/google/cloud/ai_platform/v1/index_service/rest/operations.rb +280 -0
  13. data/lib/google/cloud/ai_platform/v1/job_service/rest/operations.rb +280 -0
  14. data/lib/google/cloud/ai_platform/v1/metadata_service/rest/operations.rb +280 -0
  15. data/lib/google/cloud/ai_platform/v1/migration_service/rest/operations.rb +280 -0
  16. data/lib/google/cloud/ai_platform/v1/model_service/rest/operations.rb +280 -0
  17. data/lib/google/cloud/ai_platform/v1/notebook_service/client.rb +100 -0
  18. data/lib/google/cloud/ai_platform/v1/notebook_service/rest/client.rb +93 -0
  19. data/lib/google/cloud/ai_platform/v1/notebook_service/rest/operations.rb +280 -0
  20. data/lib/google/cloud/ai_platform/v1/notebook_service/rest/service_stub.rb +60 -0
  21. data/lib/google/cloud/ai_platform/v1/persistent_resource_service/rest/operations.rb +280 -0
  22. data/lib/google/cloud/ai_platform/v1/pipeline_service/rest/operations.rb +280 -0
  23. data/lib/google/cloud/ai_platform/v1/prediction_service/client.rb +8 -2
  24. data/lib/google/cloud/ai_platform/v1/prediction_service/credentials.rb +2 -1
  25. data/lib/google/cloud/ai_platform/v1/prediction_service/rest/client.rb +8 -2
  26. data/lib/google/cloud/ai_platform/v1/schedule_service/rest/operations.rb +280 -0
  27. data/lib/google/cloud/ai_platform/v1/specialist_pool_service/rest/operations.rb +280 -0
  28. data/lib/google/cloud/ai_platform/v1/tensorboard_service/rest/operations.rb +280 -0
  29. data/lib/google/cloud/ai_platform/v1/version.rb +1 -1
  30. data/lib/google/cloud/ai_platform/v1/vizier_service/rest/operations.rb +280 -0
  31. data/lib/google/cloud/aiplatform/v1/content_pb.rb +3 -4
  32. data/lib/google/cloud/aiplatform/v1/dataset_pb.rb +1 -1
  33. data/lib/google/cloud/aiplatform/v1/dataset_service_pb.rb +3 -2
  34. data/lib/google/cloud/aiplatform/v1/dataset_service_services_pb.rb +2 -0
  35. data/lib/google/cloud/aiplatform/v1/dataset_version_pb.rb +1 -1
  36. data/lib/google/cloud/aiplatform/v1/deployment_resource_pool_pb.rb +3 -1
  37. data/lib/google/cloud/aiplatform/v1/feature_online_store_pb.rb +3 -1
  38. data/lib/google/cloud/aiplatform/v1/feature_pb.rb +1 -1
  39. data/lib/google/cloud/aiplatform/v1/featurestore_online_service_pb.rb +3 -1
  40. data/lib/google/cloud/aiplatform/v1/index_pb.rb +2 -1
  41. data/lib/google/cloud/aiplatform/v1/index_service_pb.rb +1 -1
  42. data/lib/google/cloud/aiplatform/v1/match_service_pb.rb +2 -1
  43. data/lib/google/cloud/aiplatform/v1/metadata_store_pb.rb +2 -1
  44. data/lib/google/cloud/aiplatform/v1/notebook_runtime_pb.rb +3 -1
  45. data/lib/google/cloud/aiplatform/v1/notebook_service_pb.rb +2 -1
  46. data/lib/google/cloud/aiplatform/v1/notebook_service_services_pb.rb +2 -0
  47. data/lib/google/cloud/aiplatform/v1/persistent_resource_pb.rb +2 -1
  48. data/lib/google/cloud/aiplatform/v1/pipeline_job_pb.rb +1 -1
  49. data/lib/google/cloud/aiplatform/v1/prediction_service_pb.rb +1 -1
  50. data/lib/google/cloud/aiplatform/v1/tool_pb.rb +4 -1
  51. data/lib/google/cloud/aiplatform/v1/tuning_job_pb.rb +3 -1
  52. data/proto_docs/google/cloud/aiplatform/v1/content.rb +11 -47
  53. data/proto_docs/google/cloud/aiplatform/v1/dataset.rb +4 -0
  54. data/proto_docs/google/cloud/aiplatform/v1/dataset_service.rb +17 -0
  55. data/proto_docs/google/cloud/aiplatform/v1/dataset_version.rb +4 -0
  56. data/proto_docs/google/cloud/aiplatform/v1/deployment_resource_pool.rb +23 -0
  57. data/proto_docs/google/cloud/aiplatform/v1/feature.rb +3 -0
  58. data/proto_docs/google/cloud/aiplatform/v1/feature_group.rb +3 -3
  59. data/proto_docs/google/cloud/aiplatform/v1/feature_online_store.rb +4 -0
  60. data/proto_docs/google/cloud/aiplatform/v1/featurestore_online_service.rb +24 -0
  61. data/proto_docs/google/cloud/aiplatform/v1/index.rb +23 -3
  62. data/proto_docs/google/cloud/aiplatform/v1/index_service.rb +17 -2
  63. data/proto_docs/google/cloud/aiplatform/v1/match_service.rb +18 -1
  64. data/proto_docs/google/cloud/aiplatform/v1/metadata_store.rb +13 -0
  65. data/proto_docs/google/cloud/aiplatform/v1/model.rb +1 -1
  66. data/proto_docs/google/cloud/aiplatform/v1/notebook_runtime.rb +15 -0
  67. data/proto_docs/google/cloud/aiplatform/v1/notebook_service.rb +18 -0
  68. data/proto_docs/google/cloud/aiplatform/v1/persistent_resource.rb +63 -0
  69. data/proto_docs/google/cloud/aiplatform/v1/pipeline_job.rb +3 -0
  70. data/proto_docs/google/cloud/aiplatform/v1/prediction_service.rb +4 -0
  71. data/proto_docs/google/cloud/aiplatform/v1/tool.rb +43 -5
  72. data/proto_docs/google/cloud/aiplatform/v1/tuning_job.rb +5 -0
  73. metadata +2 -2
@@ -7,12 +7,13 @@ require 'google/protobuf'
7
7
  require 'google/api/field_behavior_pb'
8
8
  require 'google/api/resource_pb'
9
9
  require 'google/cloud/aiplatform/v1/content_pb'
10
+ require 'google/cloud/aiplatform/v1/encryption_spec_pb'
10
11
  require 'google/cloud/aiplatform/v1/job_state_pb'
11
12
  require 'google/protobuf/timestamp_pb'
12
13
  require 'google/rpc/status_pb'
13
14
 
14
15
 
15
- descriptor_data = "\n+google/cloud/aiplatform/v1/tuning_job.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a(google/cloud/aiplatform/v1/content.proto\x1a*google/cloud/aiplatform/v1/job_state.proto\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x17google/rpc/status.proto\"\xe9\x07\n\tTuningJob\x12\x14\n\nbase_model\x18\x04 \x01(\tH\x00\x12R\n\x16supervised_tuning_spec\x18\x05 \x01(\x0b\x32\x30.google.cloud.aiplatform.v1.SupervisedTuningSpecH\x01\x12\x14\n\x04name\x18\x01 \x01(\tB\x06\xe0\x41\x08\xe0\x41\x03\x12%\n\x18tuned_model_display_name\x18\x02 \x01(\tB\x03\xe0\x41\x01\x12\x18\n\x0b\x64\x65scription\x18\x03 \x01(\tB\x03\xe0\x41\x01\x12\x38\n\x05state\x18\x06 \x01(\x0e\x32$.google.cloud.aiplatform.v1.JobStateB\x03\xe0\x41\x03\x12\x34\n\x0b\x63reate_time\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x33\n\nstart_time\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x31\n\x08\x65nd_time\x18\t \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x34\n\x0bupdate_time\x18\n \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12&\n\x05\x65rror\x18\x0b \x01(\x0b\x32\x12.google.rpc.StatusB\x03\xe0\x41\x03\x12\x46\n\x06labels\x18\x0c \x03(\x0b\x32\x31.google.cloud.aiplatform.v1.TuningJob.LabelsEntryB\x03\xe0\x41\x01\x12=\n\nexperiment\x18\r \x01(\tB)\xe0\x41\x03\xfa\x41#\n!aiplatform.googleapis.com/Context\x12@\n\x0btuned_model\x18\x0e \x01(\x0b\x32&.google.cloud.aiplatform.v1.TunedModelB\x03\xe0\x41\x03\x12K\n\x11tuning_data_stats\x18\x0f \x01(\x0b\x32+.google.cloud.aiplatform.v1.TuningDataStatsB\x03\xe0\x41\x03\x1a-\n\x0bLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01:\x80\x01\xea\x41}\n#aiplatform.googleapis.com/TuningJob\x12?projects/{project}/locations/{location}/tuningJobs/{tuning_job}*\ntuningJobs2\ttuningJobB\x0e\n\x0csource_modelB\r\n\x0btuning_spec\"\x82\x01\n\nTunedModel\x12\x36\n\x05model\x18\x01 \x01(\tB\'\xe0\x41\x03\xfa\x41!\n\x1f\x61iplatform.googleapis.com/Model\x12<\n\x08\x65ndpoint\x18\x02 \x01(\tB*\xe0\x41\x03\xfa\x41$\n\"aiplatform.googleapis.com/Endpoint\"\xd7\x02\n#SupervisedTuningDatasetDistribution\x12\x10\n\x03sum\x18\x01 \x01(\x03\x42\x03\xe0\x41\x03\x12\x10\n\x03min\x18\x02 \x01(\x01\x42\x03\xe0\x41\x03\x12\x10\n\x03max\x18\x03 \x01(\x01\x42\x03\xe0\x41\x03\x12\x11\n\x04mean\x18\x04 \x01(\x01\x42\x03\xe0\x41\x03\x12\x13\n\x06median\x18\x05 \x01(\x01\x42\x03\xe0\x41\x03\x12\x0f\n\x02p5\x18\x06 \x01(\x01\x42\x03\xe0\x41\x03\x12\x10\n\x03p95\x18\x07 \x01(\x01\x42\x03\xe0\x41\x03\x12\x63\n\x07\x62uckets\x18\x08 \x03(\x0b\x32M.google.cloud.aiplatform.v1.SupervisedTuningDatasetDistribution.DatasetBucketB\x03\xe0\x41\x03\x1aJ\n\rDatasetBucket\x12\x12\n\x05\x63ount\x18\x01 \x01(\x01\x42\x03\xe0\x41\x03\x12\x11\n\x04left\x18\x02 \x01(\x01\x42\x03\xe0\x41\x03\x12\x12\n\x05right\x18\x03 \x01(\x01\x42\x03\xe0\x41\x03\"\xd7\x04\n\x19SupervisedTuningDataStats\x12)\n\x1ctuning_dataset_example_count\x18\x01 \x01(\x03\x42\x03\xe0\x41\x03\x12)\n\x1ctotal_tuning_character_count\x18\x02 \x01(\x03\x42\x03\xe0\x41\x03\x12+\n\x1etotal_billable_character_count\x18\x03 \x01(\x03\x42\x03\xe0\x41\x03\x12\x1e\n\x11tuning_step_count\x18\x04 \x01(\x03\x42\x03\xe0\x41\x03\x12k\n\x1duser_input_token_distribution\x18\x05 \x01(\x0b\x32?.google.cloud.aiplatform.v1.SupervisedTuningDatasetDistributionB\x03\xe0\x41\x03\x12l\n\x1euser_output_token_distribution\x18\x06 \x01(\x0b\x32?.google.cloud.aiplatform.v1.SupervisedTuningDatasetDistributionB\x03\xe0\x41\x03\x12s\n%user_message_per_example_distribution\x18\x07 \x01(\x0b\x32?.google.cloud.aiplatform.v1.SupervisedTuningDatasetDistributionB\x03\xe0\x41\x03\x12G\n\x15user_dataset_examples\x18\x08 \x03(\x0b\x32#.google.cloud.aiplatform.v1.ContentB\x03\xe0\x41\x03\"\x85\x01\n\x0fTuningDataStats\x12]\n\x1csupervised_tuning_data_stats\x18\x01 \x01(\x0b\x32\x35.google.cloud.aiplatform.v1.SupervisedTuningDataStatsH\x00\x42\x13\n\x11tuning_data_stats\"\xc7\x02\n\x19SupervisedHyperParameters\x12\x18\n\x0b\x65poch_count\x18\x01 \x01(\x03\x42\x03\xe0\x41\x01\x12%\n\x18learning_rate_multiplier\x18\x02 \x01(\x01\x42\x03\xe0\x41\x01\x12\\\n\x0c\x61\x64\x61pter_size\x18\x03 \x01(\x0e\x32\x41.google.cloud.aiplatform.v1.SupervisedHyperParameters.AdapterSizeB\x03\xe0\x41\x01\"\x8a\x01\n\x0b\x41\x64\x61pterSize\x12\x1c\n\x18\x41\x44\x41PTER_SIZE_UNSPECIFIED\x10\x00\x12\x14\n\x10\x41\x44\x41PTER_SIZE_ONE\x10\x01\x12\x15\n\x11\x41\x44\x41PTER_SIZE_FOUR\x10\x02\x12\x16\n\x12\x41\x44\x41PTER_SIZE_EIGHT\x10\x03\x12\x18\n\x14\x41\x44\x41PTER_SIZE_SIXTEEN\x10\x04\"\xb4\x01\n\x14SupervisedTuningSpec\x12!\n\x14training_dataset_uri\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12#\n\x16validation_dataset_uri\x18\x02 \x01(\tB\x03\xe0\x41\x01\x12T\n\x10hyper_parameters\x18\x03 \x01(\x0b\x32\x35.google.cloud.aiplatform.v1.SupervisedHyperParametersB\x03\xe0\x41\x01\x42\xcc\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\x0eTuningJobProtoP\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"
16
+ descriptor_data = "\n+google/cloud/aiplatform/v1/tuning_job.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a(google/cloud/aiplatform/v1/content.proto\x1a\x30google/cloud/aiplatform/v1/encryption_spec.proto\x1a*google/cloud/aiplatform/v1/job_state.proto\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x17google/rpc/status.proto\"\xae\x08\n\tTuningJob\x12\x14\n\nbase_model\x18\x04 \x01(\tH\x00\x12R\n\x16supervised_tuning_spec\x18\x05 \x01(\x0b\x32\x30.google.cloud.aiplatform.v1.SupervisedTuningSpecH\x01\x12\x14\n\x04name\x18\x01 \x01(\tB\x06\xe0\x41\x08\xe0\x41\x03\x12%\n\x18tuned_model_display_name\x18\x02 \x01(\tB\x03\xe0\x41\x01\x12\x18\n\x0b\x64\x65scription\x18\x03 \x01(\tB\x03\xe0\x41\x01\x12\x38\n\x05state\x18\x06 \x01(\x0e\x32$.google.cloud.aiplatform.v1.JobStateB\x03\xe0\x41\x03\x12\x34\n\x0b\x63reate_time\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x33\n\nstart_time\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x31\n\x08\x65nd_time\x18\t \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x34\n\x0bupdate_time\x18\n \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12&\n\x05\x65rror\x18\x0b \x01(\x0b\x32\x12.google.rpc.StatusB\x03\xe0\x41\x03\x12\x46\n\x06labels\x18\x0c \x03(\x0b\x32\x31.google.cloud.aiplatform.v1.TuningJob.LabelsEntryB\x03\xe0\x41\x01\x12=\n\nexperiment\x18\r \x01(\tB)\xe0\x41\x03\xfa\x41#\n!aiplatform.googleapis.com/Context\x12@\n\x0btuned_model\x18\x0e \x01(\x0b\x32&.google.cloud.aiplatform.v1.TunedModelB\x03\xe0\x41\x03\x12K\n\x11tuning_data_stats\x18\x0f \x01(\x0b\x32+.google.cloud.aiplatform.v1.TuningDataStatsB\x03\xe0\x41\x03\x12\x43\n\x0f\x65ncryption_spec\x18\x10 \x01(\x0b\x32*.google.cloud.aiplatform.v1.EncryptionSpec\x1a-\n\x0bLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01:\x80\x01\xea\x41}\n#aiplatform.googleapis.com/TuningJob\x12?projects/{project}/locations/{location}/tuningJobs/{tuning_job}*\ntuningJobs2\ttuningJobB\x0e\n\x0csource_modelB\r\n\x0btuning_spec\"\x82\x01\n\nTunedModel\x12\x36\n\x05model\x18\x01 \x01(\tB\'\xe0\x41\x03\xfa\x41!\n\x1f\x61iplatform.googleapis.com/Model\x12<\n\x08\x65ndpoint\x18\x02 \x01(\tB*\xe0\x41\x03\xfa\x41$\n\"aiplatform.googleapis.com/Endpoint\"\xd7\x02\n#SupervisedTuningDatasetDistribution\x12\x10\n\x03sum\x18\x01 \x01(\x03\x42\x03\xe0\x41\x03\x12\x10\n\x03min\x18\x02 \x01(\x01\x42\x03\xe0\x41\x03\x12\x10\n\x03max\x18\x03 \x01(\x01\x42\x03\xe0\x41\x03\x12\x11\n\x04mean\x18\x04 \x01(\x01\x42\x03\xe0\x41\x03\x12\x13\n\x06median\x18\x05 \x01(\x01\x42\x03\xe0\x41\x03\x12\x0f\n\x02p5\x18\x06 \x01(\x01\x42\x03\xe0\x41\x03\x12\x10\n\x03p95\x18\x07 \x01(\x01\x42\x03\xe0\x41\x03\x12\x63\n\x07\x62uckets\x18\x08 \x03(\x0b\x32M.google.cloud.aiplatform.v1.SupervisedTuningDatasetDistribution.DatasetBucketB\x03\xe0\x41\x03\x1aJ\n\rDatasetBucket\x12\x12\n\x05\x63ount\x18\x01 \x01(\x01\x42\x03\xe0\x41\x03\x12\x11\n\x04left\x18\x02 \x01(\x01\x42\x03\xe0\x41\x03\x12\x12\n\x05right\x18\x03 \x01(\x01\x42\x03\xe0\x41\x03\"\xd7\x04\n\x19SupervisedTuningDataStats\x12)\n\x1ctuning_dataset_example_count\x18\x01 \x01(\x03\x42\x03\xe0\x41\x03\x12)\n\x1ctotal_tuning_character_count\x18\x02 \x01(\x03\x42\x03\xe0\x41\x03\x12+\n\x1etotal_billable_character_count\x18\x03 \x01(\x03\x42\x03\xe0\x41\x03\x12\x1e\n\x11tuning_step_count\x18\x04 \x01(\x03\x42\x03\xe0\x41\x03\x12k\n\x1duser_input_token_distribution\x18\x05 \x01(\x0b\x32?.google.cloud.aiplatform.v1.SupervisedTuningDatasetDistributionB\x03\xe0\x41\x03\x12l\n\x1euser_output_token_distribution\x18\x06 \x01(\x0b\x32?.google.cloud.aiplatform.v1.SupervisedTuningDatasetDistributionB\x03\xe0\x41\x03\x12s\n%user_message_per_example_distribution\x18\x07 \x01(\x0b\x32?.google.cloud.aiplatform.v1.SupervisedTuningDatasetDistributionB\x03\xe0\x41\x03\x12G\n\x15user_dataset_examples\x18\x08 \x03(\x0b\x32#.google.cloud.aiplatform.v1.ContentB\x03\xe0\x41\x03\"\x85\x01\n\x0fTuningDataStats\x12]\n\x1csupervised_tuning_data_stats\x18\x01 \x01(\x0b\x32\x35.google.cloud.aiplatform.v1.SupervisedTuningDataStatsH\x00\x42\x13\n\x11tuning_data_stats\"\xc7\x02\n\x19SupervisedHyperParameters\x12\x18\n\x0b\x65poch_count\x18\x01 \x01(\x03\x42\x03\xe0\x41\x01\x12%\n\x18learning_rate_multiplier\x18\x02 \x01(\x01\x42\x03\xe0\x41\x01\x12\\\n\x0c\x61\x64\x61pter_size\x18\x03 \x01(\x0e\x32\x41.google.cloud.aiplatform.v1.SupervisedHyperParameters.AdapterSizeB\x03\xe0\x41\x01\"\x8a\x01\n\x0b\x41\x64\x61pterSize\x12\x1c\n\x18\x41\x44\x41PTER_SIZE_UNSPECIFIED\x10\x00\x12\x14\n\x10\x41\x44\x41PTER_SIZE_ONE\x10\x01\x12\x15\n\x11\x41\x44\x41PTER_SIZE_FOUR\x10\x02\x12\x16\n\x12\x41\x44\x41PTER_SIZE_EIGHT\x10\x03\x12\x18\n\x14\x41\x44\x41PTER_SIZE_SIXTEEN\x10\x04\"\xb4\x01\n\x14SupervisedTuningSpec\x12!\n\x14training_dataset_uri\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12#\n\x16validation_dataset_uri\x18\x02 \x01(\tB\x03\xe0\x41\x01\x12T\n\x10hyper_parameters\x18\x03 \x01(\x0b\x32\x35.google.cloud.aiplatform.v1.SupervisedHyperParametersB\x03\xe0\x41\x01\x42\xcc\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\x0eTuningJobProtoP\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"
16
17
 
17
18
  pool = Google::Protobuf::DescriptorPool.generated_pool
18
19
 
@@ -29,6 +30,7 @@ rescue TypeError
29
30
  imports = [
30
31
  ["google.protobuf.Timestamp", "google/protobuf/timestamp.proto"],
31
32
  ["google.rpc.Status", "google/rpc/status.proto"],
33
+ ["google.cloud.aiplatform.v1.EncryptionSpec", "google/cloud/aiplatform/v1/encryption_spec.proto"],
32
34
  ["google.cloud.aiplatform.v1.Content", "google/cloud/aiplatform/v1/content.proto"],
33
35
  ]
34
36
  imports.each do |type_name, expected_filename|
@@ -150,6 +150,15 @@ module Google
150
150
  # The model needs to be prompted to output the appropriate response type,
151
151
  # otherwise the behavior is undefined.
152
152
  # This is a preview feature.
153
+ # @!attribute [rw] response_schema
154
+ # @return [::Google::Cloud::AIPlatform::V1::Schema]
155
+ # Optional. The `Schema` object allows the definition of input and output
156
+ # data types. These types can be objects, but also primitives and arrays.
157
+ # Represents a select subset of an [OpenAPI 3.0 schema
158
+ # object](https://spec.openapis.org/oas/v3.0.3#schema).
159
+ # If set, a compatible response_mime_type must also be set.
160
+ # Compatible mimetypes:
161
+ # `application/json`: Schema for JSON response.
153
162
  class GenerationConfig
154
163
  include ::Google::Protobuf::MessageExts
155
164
  extend ::Google::Protobuf::MessageExts::ClassMethods
@@ -360,51 +369,9 @@ module Google
360
369
  # The token generation was stopped as the response was flagged for
361
370
  # Sensitive Personally Identifiable Information (SPII) contents.
362
371
  SPII = 8
363
- end
364
- end
365
-
366
- # Segment of the content.
367
- # @!attribute [r] part_index
368
- # @return [::Integer]
369
- # Output only. The index of a Part object within its parent Content object.
370
- # @!attribute [r] start_index
371
- # @return [::Integer]
372
- # Output only. Start index in the given Part, measured in bytes. Offset from
373
- # the start of the Part, inclusive, starting at zero.
374
- # @!attribute [r] end_index
375
- # @return [::Integer]
376
- # Output only. End index in the given Part, measured in bytes. Offset from
377
- # the start of the Part, exclusive, starting at zero.
378
- class Segment
379
- include ::Google::Protobuf::MessageExts
380
- extend ::Google::Protobuf::MessageExts::ClassMethods
381
- end
382
-
383
- # Grounding attribution.
384
- # @!attribute [rw] web
385
- # @return [::Google::Cloud::AIPlatform::V1::GroundingAttribution::Web]
386
- # Optional. Attribution from the web.
387
- # @!attribute [r] segment
388
- # @return [::Google::Cloud::AIPlatform::V1::Segment]
389
- # Output only. Segment of the content this attribution belongs to.
390
- # @!attribute [r] confidence_score
391
- # @return [::Float]
392
- # Optional. Output only. Confidence score of the attribution. Ranges from 0
393
- # to 1. 1 is the most confident.
394
- class GroundingAttribution
395
- include ::Google::Protobuf::MessageExts
396
- extend ::Google::Protobuf::MessageExts::ClassMethods
397
372
 
398
- # Attribution from the web.
399
- # @!attribute [r] uri
400
- # @return [::String]
401
- # Output only. URI reference of the attribution.
402
- # @!attribute [r] title
403
- # @return [::String]
404
- # Output only. Title of the attribution.
405
- class Web
406
- include ::Google::Protobuf::MessageExts
407
- extend ::Google::Protobuf::MessageExts::ClassMethods
373
+ # The function call generated by the model is invalid.
374
+ MALFORMED_FUNCTION_CALL = 9
408
375
  end
409
376
  end
410
377
 
@@ -412,9 +379,6 @@ module Google
412
379
  # @!attribute [rw] web_search_queries
413
380
  # @return [::Array<::String>]
414
381
  # Optional. Web search queries for the following-up web search.
415
- # @!attribute [rw] grounding_attributions
416
- # @return [::Array<::Google::Cloud::AIPlatform::V1::GroundingAttribution>]
417
- # Optional. List of grounding attributions.
418
382
  # @!attribute [rw] search_entry_point
419
383
  # @return [::Google::Cloud::AIPlatform::V1::SearchEntryPoint]
420
384
  # Optional. Google search entry for the following-up web searches.
@@ -93,6 +93,10 @@ module Google
93
93
  # MetadataStore when creating the Dataset. The Artifact resource name pattern
94
94
  # is
95
95
  # `projects/{project}/locations/{location}/metadataStores/{metadata_store}/artifacts/{artifact}`.
96
+ # @!attribute [rw] model_reference
97
+ # @return [::String]
98
+ # Optional. Reference to the public base model last used by the dataset. Only
99
+ # set for prompt datasets.
96
100
  class Dataset
97
101
  include ::Google::Protobuf::MessageExts
98
102
  extend ::Google::Protobuf::MessageExts::ClassMethods
@@ -77,6 +77,23 @@ module Google
77
77
  extend ::Google::Protobuf::MessageExts::ClassMethods
78
78
  end
79
79
 
80
+ # Request message for
81
+ # {::Google::Cloud::AIPlatform::V1::DatasetService::Client#update_dataset_version DatasetService.UpdateDatasetVersion}.
82
+ # @!attribute [rw] dataset_version
83
+ # @return [::Google::Cloud::AIPlatform::V1::DatasetVersion]
84
+ # Required. The DatasetVersion which replaces the resource on the server.
85
+ # @!attribute [rw] update_mask
86
+ # @return [::Google::Protobuf::FieldMask]
87
+ # Required. The update mask applies to the resource.
88
+ # For the `FieldMask` definition, see
89
+ # {::Google::Protobuf::FieldMask google.protobuf.FieldMask}. Updatable fields:
90
+ #
91
+ # * `display_name`
92
+ class UpdateDatasetVersionRequest
93
+ include ::Google::Protobuf::MessageExts
94
+ extend ::Google::Protobuf::MessageExts::ClassMethods
95
+ end
96
+
80
97
  # Request message for
81
98
  # {::Google::Cloud::AIPlatform::V1::DatasetService::Client#list_datasets DatasetService.ListDatasets}.
82
99
  # @!attribute [rw] parent
@@ -46,6 +46,10 @@ module Google
46
46
  # @!attribute [r] metadata
47
47
  # @return [::Google::Protobuf::Value]
48
48
  # Required. Output only. Additional information about the DatasetVersion.
49
+ # @!attribute [r] model_reference
50
+ # @return [::String]
51
+ # Output only. Reference to the public base model last used by the dataset
52
+ # version. Only set for prompt dataset versions.
49
53
  class DatasetVersion
50
54
  include ::Google::Protobuf::MessageExts
51
55
  extend ::Google::Protobuf::MessageExts::ClassMethods
@@ -32,6 +32,29 @@ module Google
32
32
  # @return [::Google::Cloud::AIPlatform::V1::DedicatedResources]
33
33
  # Required. The underlying DedicatedResources that the DeploymentResourcePool
34
34
  # uses.
35
+ # @!attribute [rw] encryption_spec
36
+ # @return [::Google::Cloud::AIPlatform::V1::EncryptionSpec]
37
+ # Customer-managed encryption key spec for a DeploymentResourcePool. If set,
38
+ # this DeploymentResourcePool will be secured by this key. Endpoints and the
39
+ # DeploymentResourcePool they deploy in need to have the same EncryptionSpec.
40
+ # @!attribute [rw] service_account
41
+ # @return [::String]
42
+ # The service account that the DeploymentResourcePool's container(s) run as.
43
+ # Specify the email address of the service account. If this service account
44
+ # is not specified, the container(s) run as a service account that doesn't
45
+ # have access to the resource project.
46
+ #
47
+ # Users deploying the Models to this DeploymentResourcePool must have the
48
+ # `iam.serviceAccounts.actAs` permission on this service account.
49
+ # @!attribute [rw] disable_container_logging
50
+ # @return [::Boolean]
51
+ # If the DeploymentResourcePool is deployed with custom-trained Models or
52
+ # AutoML Tabular Models, the container(s) of the DeploymentResourcePool will
53
+ # send `stderr` and `stdout` streams to Cloud Logging by default.
54
+ # Please note that the logs incur cost, which are subject to [Cloud Logging
55
+ # pricing](https://cloud.google.com/logging/pricing).
56
+ #
57
+ # User can disable container logging by setting this flag to true.
35
58
  # @!attribute [r] create_time
36
59
  # @return [::Google::Protobuf::Timestamp]
37
60
  # Output only. Timestamp when this DeploymentResourcePool was created.
@@ -168,6 +168,9 @@ module Google
168
168
 
169
169
  # Used for Feature that is bytes.
170
170
  BYTES = 13
171
+
172
+ # Used for Feature that is struct.
173
+ STRUCT = 14
171
174
  end
172
175
  end
173
176
  end
@@ -25,9 +25,9 @@ module Google
25
25
  # @!attribute [rw] big_query
26
26
  # @return [::Google::Cloud::AIPlatform::V1::FeatureGroup::BigQuery]
27
27
  # Indicates that features for this group come from BigQuery Table/View.
28
- # By default treats the source as a sparse time series source, which is
29
- # required to have an entity_id and a feature_timestamp column in the
30
- # source.
28
+ # By default treats the source as a sparse time series source. The BigQuery
29
+ # source table or view must have at least one entity ID column and a column
30
+ # named `feature_timestamp`.
31
31
  # @!attribute [rw] name
32
32
  # @return [::String]
33
33
  # Identifier. Name of the FeatureGroup. Format:
@@ -70,6 +70,10 @@ module Google
70
70
  # @return [::Google::Cloud::AIPlatform::V1::FeatureOnlineStore::DedicatedServingEndpoint]
71
71
  # Optional. The dedicated serving endpoint for this FeatureOnlineStore, which
72
72
  # is different from common Vertex service endpoint.
73
+ # @!attribute [rw] encryption_spec
74
+ # @return [::Google::Cloud::AIPlatform::V1::EncryptionSpec]
75
+ # Optional. Customer-managed encryption key spec for data storage. If set,
76
+ # online store will be secured by this key.
73
77
  class FeatureOnlineStore
74
78
  include ::Google::Protobuf::MessageExts
75
79
  extend ::Google::Protobuf::MessageExts::ClassMethods
@@ -221,6 +221,9 @@ module Google
221
221
  # @!attribute [rw] bytes_value
222
222
  # @return [::String]
223
223
  # Bytes feature value.
224
+ # @!attribute [rw] struct_value
225
+ # @return [::Google::Cloud::AIPlatform::V1::StructValue]
226
+ # A struct type feature value.
224
227
  # @!attribute [rw] metadata
225
228
  # @return [::Google::Cloud::AIPlatform::V1::FeatureValue::Metadata]
226
229
  # Metadata of feature value.
@@ -243,6 +246,27 @@ module Google
243
246
  end
244
247
  end
245
248
 
249
+ # Struct (or object) type feature value.
250
+ # @!attribute [rw] values
251
+ # @return [::Array<::Google::Cloud::AIPlatform::V1::StructFieldValue>]
252
+ # A list of field values.
253
+ class StructValue
254
+ include ::Google::Protobuf::MessageExts
255
+ extend ::Google::Protobuf::MessageExts::ClassMethods
256
+ end
257
+
258
+ # One field of a Struct (or object) type feature value.
259
+ # @!attribute [rw] name
260
+ # @return [::String]
261
+ # Name of the field in the struct feature.
262
+ # @!attribute [rw] value
263
+ # @return [::Google::Cloud::AIPlatform::V1::FeatureValue]
264
+ # The value for this field.
265
+ class StructFieldValue
266
+ include ::Google::Protobuf::MessageExts
267
+ extend ::Google::Protobuf::MessageExts::ClassMethods
268
+ end
269
+
246
270
  # Container for list of values.
247
271
  # @!attribute [rw] values
248
272
  # @return [::Array<::Google::Cloud::AIPlatform::V1::FeatureValue>]
@@ -126,8 +126,11 @@ module Google
126
126
  # Required. Unique identifier of the datapoint.
127
127
  # @!attribute [rw] feature_vector
128
128
  # @return [::Array<::Float>]
129
- # Required. Feature embedding vector. An array of numbers with the length of
130
- # [NearestNeighborSearchConfig.dimensions].
129
+ # Required. Feature embedding vector for dense index. An array of numbers
130
+ # with the length of [NearestNeighborSearchConfig.dimensions].
131
+ # @!attribute [rw] sparse_embedding
132
+ # @return [::Google::Cloud::AIPlatform::V1::IndexDatapoint::SparseEmbedding]
133
+ # Optional. Feature embedding vector for sparse index.
131
134
  # @!attribute [rw] restricts
132
135
  # @return [::Array<::Google::Cloud::AIPlatform::V1::IndexDatapoint::Restriction>]
133
136
  # Optional. List of Restrict of the datapoint, used to perform "restricted
@@ -147,6 +150,20 @@ module Google
147
150
  include ::Google::Protobuf::MessageExts
148
151
  extend ::Google::Protobuf::MessageExts::ClassMethods
149
152
 
153
+ # Feature embedding vector for sparse index. An array of numbers whose values
154
+ # are located in the specified dimensions.
155
+ # @!attribute [rw] values
156
+ # @return [::Array<::Float>]
157
+ # Required. The list of embedding values of the sparse vector.
158
+ # @!attribute [rw] dimensions
159
+ # @return [::Array<::Integer>]
160
+ # Required. The list of indexes for the embedding values of the sparse
161
+ # vector.
162
+ class SparseEmbedding
163
+ include ::Google::Protobuf::MessageExts
164
+ extend ::Google::Protobuf::MessageExts::ClassMethods
165
+ end
166
+
150
167
  # Restriction of a datapoint which describe its attributes(tokens) from each
151
168
  # of several attribute categories(namespaces).
152
169
  # @!attribute [rw] namespace
@@ -233,7 +250,10 @@ module Google
233
250
  # Stats of the Index.
234
251
  # @!attribute [r] vectors_count
235
252
  # @return [::Integer]
236
- # Output only. The number of vectors in the Index.
253
+ # Output only. The number of dense vectors in the Index.
254
+ # @!attribute [r] sparse_vectors_count
255
+ # @return [::Integer]
256
+ # Output only. The number of sparse vectors in the Index.
237
257
  # @!attribute [r] shards_count
238
258
  # @return [::Integer]
239
259
  # Output only. The number of shards in the Index.
@@ -253,8 +253,8 @@ module Google
253
253
  # The embedding id is not valid.
254
254
  INVALID_EMBEDDING_ID = 5
255
255
 
256
- # The size of the embedding vectors does not match with the specified
257
- # dimension.
256
+ # The size of the dense embedding vectors does not match with the
257
+ # specified dimension.
258
258
  EMBEDDING_SIZE_MISMATCH = 6
259
259
 
260
260
  # The `namespace` field is missing.
@@ -279,8 +279,17 @@ module Google
279
279
  # File is not in UTF_8 format.
280
280
  INVALID_ENCODING = 13
281
281
 
282
+ # Error parsing sparse dimensions field.
283
+ INVALID_SPARSE_DIMENSIONS = 14
284
+
282
285
  # Token restrict value is invalid.
283
286
  INVALID_TOKEN_VALUE = 15
287
+
288
+ # Invalid sparse embedding.
289
+ INVALID_SPARSE_EMBEDDING = 16
290
+
291
+ # Invalid dense embedding.
292
+ INVALID_EMBEDDING = 17
284
293
  end
285
294
  end
286
295
 
@@ -298,6 +307,12 @@ module Google
298
307
  # The detail information of the partial failures encountered for those
299
308
  # invalid records that couldn't be parsed.
300
309
  # Up to 50 partial errors will be reported.
310
+ # @!attribute [rw] valid_sparse_record_count
311
+ # @return [::Integer]
312
+ # Number of sparse records in this file that were successfully processed.
313
+ # @!attribute [rw] invalid_sparse_record_count
314
+ # @return [::Integer]
315
+ # Number of sparse records in this file we skipped due to validate errors.
301
316
  class ContentValidationStats
302
317
  include ::Google::Protobuf::MessageExts
303
318
  extend ::Google::Protobuf::MessageExts::ClassMethods
@@ -51,6 +51,9 @@ module Google
51
51
 
52
52
  # A query to find a number of the nearest neighbors (most similar vectors)
53
53
  # of a vector.
54
+ # @!attribute [rw] rrf
55
+ # @return [::Google::Cloud::AIPlatform::V1::FindNeighborsRequest::Query::RRF]
56
+ # Optional. Represents RRF algorithm that combines search results.
54
57
  # @!attribute [rw] datapoint
55
58
  # @return [::Google::Cloud::AIPlatform::V1::IndexDatapoint]
56
59
  # Required. The datapoint/vector whose nearest neighbors should be searched
@@ -83,6 +86,17 @@ module Google
83
86
  class Query
84
87
  include ::Google::Protobuf::MessageExts
85
88
  extend ::Google::Protobuf::MessageExts::ClassMethods
89
+
90
+ # Parameters for RRF algorithm that combines search results.
91
+ # @!attribute [rw] alpha
92
+ # @return [::Float]
93
+ # Required. Users can provide an alpha value to give more weight to dense
94
+ # vs sparse results. For example, if the alpha is 0, we only return
95
+ # sparse and if the alpha is 1, we only return dense.
96
+ class RRF
97
+ include ::Google::Protobuf::MessageExts
98
+ extend ::Google::Protobuf::MessageExts::ClassMethods
99
+ end
86
100
  end
87
101
  end
88
102
 
@@ -104,7 +118,10 @@ module Google
104
118
  # fields are populated.
105
119
  # @!attribute [rw] distance
106
120
  # @return [::Float]
107
- # The distance between the neighbor and the query vector.
121
+ # The distance between the neighbor and the dense embedding query.
122
+ # @!attribute [rw] sparse_distance
123
+ # @return [::Float]
124
+ # The distance between the neighbor and the query sparse_embedding.
108
125
  class Neighbor
109
126
  include ::Google::Protobuf::MessageExts
110
127
  extend ::Google::Protobuf::MessageExts::ClassMethods
@@ -43,6 +43,9 @@ module Google
43
43
  # @!attribute [r] state
44
44
  # @return [::Google::Cloud::AIPlatform::V1::MetadataStore::MetadataStoreState]
45
45
  # Output only. State information of the MetadataStore.
46
+ # @!attribute [rw] dataplex_config
47
+ # @return [::Google::Cloud::AIPlatform::V1::MetadataStore::DataplexConfig]
48
+ # Optional. Dataplex integration settings.
46
49
  class MetadataStore
47
50
  include ::Google::Protobuf::MessageExts
48
51
  extend ::Google::Protobuf::MessageExts::ClassMethods
@@ -55,6 +58,16 @@ module Google
55
58
  include ::Google::Protobuf::MessageExts
56
59
  extend ::Google::Protobuf::MessageExts::ClassMethods
57
60
  end
61
+
62
+ # Represents Dataplex integration settings.
63
+ # @!attribute [rw] enabled_pipelines_lineage
64
+ # @return [::Boolean]
65
+ # Optional. Whether or not Data Lineage synchronization is enabled for
66
+ # Vertex Pipelines.
67
+ class DataplexConfig
68
+ include ::Google::Protobuf::MessageExts
69
+ extend ::Google::Protobuf::MessageExts::ClassMethods
70
+ end
58
71
  end
59
72
  end
60
73
  end
@@ -830,7 +830,7 @@ module Google
830
830
  # determine whether it is alive or ready to receive traffic.
831
831
  # @!attribute [rw] exec
832
832
  # @return [::Google::Cloud::AIPlatform::V1::Probe::ExecAction]
833
- # Exec specifies the action to take.
833
+ # ExecAction probes the health of a container by executing a command.
834
834
  # @!attribute [rw] period_seconds
835
835
  # @return [::Integer]
836
836
  # How often (in seconds) to perform the probe. Default to 10 seconds.
@@ -97,6 +97,9 @@ module Google
97
97
  # @return [::Array<::String>]
98
98
  # Optional. The Compute Engine tags to add to runtime (see [Tagging
99
99
  # instances](https://cloud.google.com/vpc/docs/add-remove-network-tags)).
100
+ # @!attribute [rw] encryption_spec
101
+ # @return [::Google::Cloud::AIPlatform::V1::EncryptionSpec]
102
+ # Customer-managed encryption key spec for the notebook runtime.
100
103
  class NotebookRuntimeTemplate
101
104
  include ::Google::Protobuf::MessageExts
102
105
  extend ::Google::Protobuf::MessageExts::ClassMethods
@@ -186,10 +189,22 @@ module Google
186
189
  # @!attribute [r] notebook_runtime_type
187
190
  # @return [::Google::Cloud::AIPlatform::V1::NotebookRuntimeType]
188
191
  # Output only. The type of the notebook runtime.
192
+ # @!attribute [r] idle_shutdown_config
193
+ # @return [::Google::Cloud::AIPlatform::V1::NotebookIdleShutdownConfig]
194
+ # Output only. The idle shutdown configuration of the notebook runtime.
189
195
  # @!attribute [rw] network_tags
190
196
  # @return [::Array<::String>]
191
197
  # Optional. The Compute Engine tags to add to runtime (see [Tagging
192
198
  # instances](https://cloud.google.com/vpc/docs/add-remove-network-tags)).
199
+ # @!attribute [r] encryption_spec
200
+ # @return [::Google::Cloud::AIPlatform::V1::EncryptionSpec]
201
+ # Output only. Customer-managed encryption key spec for the notebook runtime.
202
+ # @!attribute [r] satisfies_pzs
203
+ # @return [::Boolean]
204
+ # Output only. Reserved for future use.
205
+ # @!attribute [r] satisfies_pzi
206
+ # @return [::Boolean]
207
+ # Output only. Reserved for future use.
193
208
  class NotebookRuntime
194
209
  include ::Google::Protobuf::MessageExts
195
210
  extend ::Google::Protobuf::MessageExts::ClassMethods
@@ -147,6 +147,24 @@ module Google
147
147
  extend ::Google::Protobuf::MessageExts::ClassMethods
148
148
  end
149
149
 
150
+ # Request message for
151
+ # {::Google::Cloud::AIPlatform::V1::NotebookService::Client#update_notebook_runtime_template NotebookService.UpdateNotebookRuntimeTemplate}.
152
+ # @!attribute [rw] notebook_runtime_template
153
+ # @return [::Google::Cloud::AIPlatform::V1::NotebookRuntimeTemplate]
154
+ # Required. The NotebookRuntimeTemplate to update.
155
+ # @!attribute [rw] update_mask
156
+ # @return [::Google::Protobuf::FieldMask]
157
+ # Required. The update mask applies to the resource.
158
+ # For the `FieldMask` definition, see
159
+ # {::Google::Protobuf::FieldMask google.protobuf.FieldMask}. Input format:
160
+ # `{paths: "${updated_filed}"}` Updatable fields:
161
+ #
162
+ # * `encryption_spec.kms_key_name`
163
+ class UpdateNotebookRuntimeTemplateRequest
164
+ include ::Google::Protobuf::MessageExts
165
+ extend ::Google::Protobuf::MessageExts::ClassMethods
166
+ end
167
+
150
168
  # Request message for
151
169
  # {::Google::Cloud::AIPlatform::V1::NotebookService::Client#assign_notebook_runtime NotebookService.AssignNotebookRuntime}.
152
170
  # @!attribute [rw] parent
@@ -207,15 +207,69 @@ module Google
207
207
  # For experimental launch, Ray cluster creation and Persistent
208
208
  # cluster creation are 1:1 mapping: We will provision all the nodes within the
209
209
  # Persistent cluster as Ray nodes.
210
+ # @!attribute [rw] image_uri
211
+ # @return [::String]
212
+ # Optional. Default image for user to choose a preferred ML framework
213
+ # (for example, TensorFlow or Pytorch) by choosing from [Vertex prebuilt
214
+ # images](https://cloud.google.com/vertex-ai/docs/training/pre-built-containers).
215
+ # Either this or the resource_pool_images is required. Use this field if
216
+ # you need all the resource pools to have the same Ray image. Otherwise, use
217
+ # the \\{@code resource_pool_images} field.
218
+ # @!attribute [rw] resource_pool_images
219
+ # @return [::Google::Protobuf::Map{::String => ::String}]
220
+ # Optional. Required if image_uri isn't set. A map of resource_pool_id to
221
+ # prebuild Ray image if user need to use different images for different
222
+ # head/worker pools. This map needs to cover all the resource pool ids.
223
+ # Example:
224
+ # {
225
+ # "ray_head_node_pool": "head image"
226
+ # "ray_worker_node_pool1": "worker image"
227
+ # "ray_worker_node_pool2": "another worker image"
228
+ # }
229
+ # @!attribute [rw] head_node_resource_pool_id
230
+ # @return [::String]
231
+ # Optional. This will be used to indicate which resource pool will serve as
232
+ # the Ray head node(the first node within that pool). Will use the machine
233
+ # from the first workerpool as the head node by default if this field isn't
234
+ # set.
235
+ # @!attribute [rw] ray_metric_spec
236
+ # @return [::Google::Cloud::AIPlatform::V1::RayMetricSpec]
237
+ # Optional. Ray metrics configurations.
210
238
  class RaySpec
211
239
  include ::Google::Protobuf::MessageExts
212
240
  extend ::Google::Protobuf::MessageExts::ClassMethods
241
+
242
+ # @!attribute [rw] key
243
+ # @return [::String]
244
+ # @!attribute [rw] value
245
+ # @return [::String]
246
+ class ResourcePoolImagesEntry
247
+ include ::Google::Protobuf::MessageExts
248
+ extend ::Google::Protobuf::MessageExts::ClassMethods
249
+ end
213
250
  end
214
251
 
215
252
  # Persistent Cluster runtime information as output
253
+ # @!attribute [r] access_uris
254
+ # @return [::Google::Protobuf::Map{::String => ::String}]
255
+ # Output only. URIs for user to connect to the Cluster.
256
+ # Example:
257
+ # {
258
+ # "RAY_HEAD_NODE_INTERNAL_IP": "head-node-IP:10001"
259
+ # "RAY_DASHBOARD_URI": "ray-dashboard-address:8888"
260
+ # }
216
261
  class ResourceRuntime
217
262
  include ::Google::Protobuf::MessageExts
218
263
  extend ::Google::Protobuf::MessageExts::ClassMethods
264
+
265
+ # @!attribute [rw] key
266
+ # @return [::String]
267
+ # @!attribute [rw] value
268
+ # @return [::String]
269
+ class AccessUrisEntry
270
+ include ::Google::Protobuf::MessageExts
271
+ extend ::Google::Protobuf::MessageExts::ClassMethods
272
+ end
219
273
  end
220
274
 
221
275
  # Configuration for the use of custom service account to run the workloads.
@@ -242,6 +296,15 @@ module Google
242
296
  include ::Google::Protobuf::MessageExts
243
297
  extend ::Google::Protobuf::MessageExts::ClassMethods
244
298
  end
299
+
300
+ # Configuration for the Ray metrics.
301
+ # @!attribute [rw] disabled
302
+ # @return [::Boolean]
303
+ # Optional. Flag to disable the Ray metrics collection.
304
+ class RayMetricSpec
305
+ include ::Google::Protobuf::MessageExts
306
+ extend ::Google::Protobuf::MessageExts::ClassMethods
307
+ end
245
308
  end
246
309
  end
247
310
  end
@@ -126,6 +126,9 @@ module Google
126
126
  # @return [::String]
127
127
  # Output only. The schedule resource name.
128
128
  # Only returned if the Pipeline is created by Schedule API.
129
+ # @!attribute [rw] preflight_validations
130
+ # @return [::Boolean]
131
+ # Optional. Whether to do component level validations before job creation.
129
132
  class PipelineJob
130
133
  include ::Google::Protobuf::MessageExts
131
134
  extend ::Google::Protobuf::MessageExts::ClassMethods
@@ -485,6 +485,10 @@ module Google
485
485
  # A `Tool` is a piece of code that enables the system to interact with
486
486
  # external systems to perform an action, or set of actions, outside of
487
487
  # knowledge and scope of the model.
488
+ # @!attribute [rw] tool_config
489
+ # @return [::Google::Cloud::AIPlatform::V1::ToolConfig]
490
+ # Optional. Tool config. This config is shared for all tools provided in the
491
+ # request.
488
492
  # @!attribute [rw] safety_settings
489
493
  # @return [::Array<::Google::Cloud::AIPlatform::V1::SafetySetting>]
490
494
  # Optional. Per request settings for blocking unsafe content.