google-cloud-ai_platform-v1 0.40.0 → 0.41.0

Sign up to get free protection for your applications and to get access to all the features.
Files changed (70) 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 +70 -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 +70 -0
  7. data/lib/google/cloud/ai_platform/v1/endpoint_service/rest/operations.rb +70 -0
  8. data/lib/google/cloud/ai_platform/v1/feature_online_store_admin_service/rest/operations.rb +70 -0
  9. data/lib/google/cloud/ai_platform/v1/feature_registry_service/rest/operations.rb +70 -0
  10. data/lib/google/cloud/ai_platform/v1/featurestore_service/rest/operations.rb +70 -0
  11. data/lib/google/cloud/ai_platform/v1/index_endpoint_service/rest/operations.rb +70 -0
  12. data/lib/google/cloud/ai_platform/v1/index_service/rest/operations.rb +70 -0
  13. data/lib/google/cloud/ai_platform/v1/job_service/rest/operations.rb +70 -0
  14. data/lib/google/cloud/ai_platform/v1/metadata_service/rest/operations.rb +70 -0
  15. data/lib/google/cloud/ai_platform/v1/migration_service/rest/operations.rb +70 -0
  16. data/lib/google/cloud/ai_platform/v1/model_service/rest/operations.rb +70 -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 +70 -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 +70 -0
  22. data/lib/google/cloud/ai_platform/v1/pipeline_service/rest/operations.rb +70 -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/prediction_service/rest/service_stub.rb +16 -0
  27. data/lib/google/cloud/ai_platform/v1/schedule_service/rest/operations.rb +70 -0
  28. data/lib/google/cloud/ai_platform/v1/specialist_pool_service/rest/operations.rb +70 -0
  29. data/lib/google/cloud/ai_platform/v1/tensorboard_service/rest/operations.rb +70 -0
  30. data/lib/google/cloud/ai_platform/v1/version.rb +1 -1
  31. data/lib/google/cloud/ai_platform/v1/vizier_service/rest/operations.rb +70 -0
  32. data/lib/google/cloud/aiplatform/v1/content_pb.rb +3 -4
  33. data/lib/google/cloud/aiplatform/v1/dataset_pb.rb +1 -1
  34. data/lib/google/cloud/aiplatform/v1/dataset_service_pb.rb +3 -2
  35. data/lib/google/cloud/aiplatform/v1/dataset_service_services_pb.rb +2 -0
  36. data/lib/google/cloud/aiplatform/v1/dataset_version_pb.rb +1 -1
  37. data/lib/google/cloud/aiplatform/v1/deployment_resource_pool_pb.rb +3 -1
  38. data/lib/google/cloud/aiplatform/v1/feature_online_store_pb.rb +3 -1
  39. data/lib/google/cloud/aiplatform/v1/feature_pb.rb +1 -1
  40. data/lib/google/cloud/aiplatform/v1/featurestore_online_service_pb.rb +3 -1
  41. data/lib/google/cloud/aiplatform/v1/index_pb.rb +2 -1
  42. data/lib/google/cloud/aiplatform/v1/index_service_pb.rb +1 -1
  43. data/lib/google/cloud/aiplatform/v1/match_service_pb.rb +2 -1
  44. data/lib/google/cloud/aiplatform/v1/metadata_store_pb.rb +2 -1
  45. data/lib/google/cloud/aiplatform/v1/notebook_runtime_pb.rb +3 -1
  46. data/lib/google/cloud/aiplatform/v1/notebook_service_pb.rb +2 -1
  47. data/lib/google/cloud/aiplatform/v1/notebook_service_services_pb.rb +2 -0
  48. data/lib/google/cloud/aiplatform/v1/persistent_resource_pb.rb +2 -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/proto_docs/google/cloud/aiplatform/v1/content.rb +9 -48
  52. data/proto_docs/google/cloud/aiplatform/v1/dataset.rb +4 -0
  53. data/proto_docs/google/cloud/aiplatform/v1/dataset_service.rb +17 -0
  54. data/proto_docs/google/cloud/aiplatform/v1/dataset_version.rb +4 -0
  55. data/proto_docs/google/cloud/aiplatform/v1/deployment_resource_pool.rb +23 -0
  56. data/proto_docs/google/cloud/aiplatform/v1/feature.rb +3 -0
  57. data/proto_docs/google/cloud/aiplatform/v1/feature_group.rb +3 -3
  58. data/proto_docs/google/cloud/aiplatform/v1/feature_online_store.rb +4 -0
  59. data/proto_docs/google/cloud/aiplatform/v1/featurestore_online_service.rb +24 -0
  60. data/proto_docs/google/cloud/aiplatform/v1/index.rb +23 -3
  61. data/proto_docs/google/cloud/aiplatform/v1/index_service.rb +17 -2
  62. data/proto_docs/google/cloud/aiplatform/v1/match_service.rb +18 -1
  63. data/proto_docs/google/cloud/aiplatform/v1/metadata_store.rb +13 -0
  64. data/proto_docs/google/cloud/aiplatform/v1/model.rb +1 -1
  65. data/proto_docs/google/cloud/aiplatform/v1/notebook_runtime.rb +15 -0
  66. data/proto_docs/google/cloud/aiplatform/v1/notebook_service.rb +18 -0
  67. data/proto_docs/google/cloud/aiplatform/v1/persistent_resource.rb +63 -0
  68. data/proto_docs/google/cloud/aiplatform/v1/prediction_service.rb +4 -0
  69. data/proto_docs/google/cloud/aiplatform/v1/tool.rb +43 -5
  70. metadata +2 -2
@@ -16,7 +16,7 @@ require 'google/cloud/aiplatform/v1/types_pb'
16
16
  require 'google/protobuf/struct_pb'
17
17
 
18
18
 
19
- descriptor_data = "\n3google/cloud/aiplatform/v1/prediction_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/httpbody.proto\x1a\x19google/api/resource.proto\x1a(google/cloud/aiplatform/v1/content.proto\x1a,google/cloud/aiplatform/v1/explanation.proto\x1a%google/cloud/aiplatform/v1/tool.proto\x1a&google/cloud/aiplatform/v1/types.proto\x1a\x1cgoogle/protobuf/struct.proto\"\xaa\x01\n\x0ePredictRequest\x12<\n\x08\x65ndpoint\x18\x01 \x01(\tB*\xe0\x41\x02\xfa\x41$\n\"aiplatform.googleapis.com/Endpoint\x12.\n\tinstances\x18\x02 \x03(\x0b\x32\x16.google.protobuf.ValueB\x03\xe0\x41\x02\x12*\n\nparameters\x18\x03 \x01(\x0b\x32\x16.google.protobuf.Value\"\x80\x02\n\x0fPredictResponse\x12+\n\x0bpredictions\x18\x01 \x03(\x0b\x32\x16.google.protobuf.Value\x12\x19\n\x11\x64\x65ployed_model_id\x18\x02 \x01(\t\x12\x36\n\x05model\x18\x03 \x01(\tB\'\xe0\x41\x03\xfa\x41!\n\x1f\x61iplatform.googleapis.com/Model\x12\x1d\n\x10model_version_id\x18\x05 \x01(\tB\x03\xe0\x41\x03\x12\x1f\n\x12model_display_name\x18\x04 \x01(\tB\x03\xe0\x41\x03\x12-\n\x08metadata\x18\x06 \x01(\x0b\x32\x16.google.protobuf.ValueB\x03\xe0\x41\x03\"z\n\x11RawPredictRequest\x12<\n\x08\x65ndpoint\x18\x01 \x01(\tB*\xe0\x41\x02\xfa\x41$\n\"aiplatform.googleapis.com/Endpoint\x12\'\n\thttp_body\x18\x02 \x01(\x0b\x32\x14.google.api.HttpBody\"\x80\x01\n\x17StreamRawPredictRequest\x12<\n\x08\x65ndpoint\x18\x01 \x01(\tB*\xe0\x41\x02\xfa\x41$\n\"aiplatform.googleapis.com/Endpoint\x12\'\n\thttp_body\x18\x02 \x01(\x0b\x32\x14.google.api.HttpBody\"\xc0\x01\n\x14\x44irectPredictRequest\x12<\n\x08\x65ndpoint\x18\x01 \x01(\tB*\xe0\x41\x02\xfa\x41$\n\"aiplatform.googleapis.com/Endpoint\x12\x32\n\x06inputs\x18\x02 \x03(\x0b\x32\".google.cloud.aiplatform.v1.Tensor\x12\x36\n\nparameters\x18\x03 \x01(\x0b\x32\".google.cloud.aiplatform.v1.Tensor\"\x84\x01\n\x15\x44irectPredictResponse\x12\x33\n\x07outputs\x18\x01 \x03(\x0b\x32\".google.cloud.aiplatform.v1.Tensor\x12\x36\n\nparameters\x18\x02 \x01(\x0b\x32\".google.cloud.aiplatform.v1.Tensor\"{\n\x17\x44irectRawPredictRequest\x12<\n\x08\x65ndpoint\x18\x01 \x01(\tB*\xe0\x41\x02\xfa\x41$\n\"aiplatform.googleapis.com/Endpoint\x12\x13\n\x0bmethod_name\x18\x02 \x01(\t\x12\r\n\x05input\x18\x03 \x01(\x0c\"*\n\x18\x44irectRawPredictResponse\x12\x0e\n\x06output\x18\x01 \x01(\x0c\"\xd0\x01\n\x1aStreamDirectPredictRequest\x12<\n\x08\x65ndpoint\x18\x01 \x01(\tB*\xe0\x41\x02\xfa\x41$\n\"aiplatform.googleapis.com/Endpoint\x12\x37\n\x06inputs\x18\x02 \x03(\x0b\x32\".google.cloud.aiplatform.v1.TensorB\x03\xe0\x41\x01\x12;\n\nparameters\x18\x03 \x01(\x0b\x32\".google.cloud.aiplatform.v1.TensorB\x03\xe0\x41\x01\"\x8a\x01\n\x1bStreamDirectPredictResponse\x12\x33\n\x07outputs\x18\x01 \x03(\x0b\x32\".google.cloud.aiplatform.v1.Tensor\x12\x36\n\nparameters\x18\x02 \x01(\x0b\x32\".google.cloud.aiplatform.v1.Tensor\"\x8b\x01\n\x1dStreamDirectRawPredictRequest\x12<\n\x08\x65ndpoint\x18\x01 \x01(\tB*\xe0\x41\x02\xfa\x41$\n\"aiplatform.googleapis.com/Endpoint\x12\x18\n\x0bmethod_name\x18\x02 \x01(\tB\x03\xe0\x41\x01\x12\x12\n\x05input\x18\x03 \x01(\x0c\x42\x03\xe0\x41\x01\"0\n\x1eStreamDirectRawPredictResponse\x12\x0e\n\x06output\x18\x01 \x01(\x0c\"\xc3\x01\n\x17StreamingPredictRequest\x12<\n\x08\x65ndpoint\x18\x01 \x01(\tB*\xe0\x41\x02\xfa\x41$\n\"aiplatform.googleapis.com/Endpoint\x12\x32\n\x06inputs\x18\x02 \x03(\x0b\x32\".google.cloud.aiplatform.v1.Tensor\x12\x36\n\nparameters\x18\x03 \x01(\x0b\x32\".google.cloud.aiplatform.v1.Tensor\"\x87\x01\n\x18StreamingPredictResponse\x12\x33\n\x07outputs\x18\x01 \x03(\x0b\x32\".google.cloud.aiplatform.v1.Tensor\x12\x36\n\nparameters\x18\x02 \x01(\x0b\x32\".google.cloud.aiplatform.v1.Tensor\"~\n\x1aStreamingRawPredictRequest\x12<\n\x08\x65ndpoint\x18\x01 \x01(\tB*\xe0\x41\x02\xfa\x41$\n\"aiplatform.googleapis.com/Endpoint\x12\x13\n\x0bmethod_name\x18\x02 \x01(\t\x12\r\n\x05input\x18\x03 \x01(\x0c\"-\n\x1bStreamingRawPredictResponse\x12\x0e\n\x06output\x18\x01 \x01(\x0c\"\x9d\x02\n\x0e\x45xplainRequest\x12<\n\x08\x65ndpoint\x18\x01 \x01(\tB*\xe0\x41\x02\xfa\x41$\n\"aiplatform.googleapis.com/Endpoint\x12.\n\tinstances\x18\x02 \x03(\x0b\x32\x16.google.protobuf.ValueB\x03\xe0\x41\x02\x12*\n\nparameters\x18\x04 \x01(\x0b\x32\x16.google.protobuf.Value\x12V\n\x19\x65xplanation_spec_override\x18\x05 \x01(\x0b\x32\x33.google.cloud.aiplatform.v1.ExplanationSpecOverride\x12\x19\n\x11\x64\x65ployed_model_id\x18\x03 \x01(\t\"\x98\x01\n\x0f\x45xplainResponse\x12=\n\x0c\x65xplanations\x18\x01 \x03(\x0b\x32\'.google.cloud.aiplatform.v1.Explanation\x12\x19\n\x11\x64\x65ployed_model_id\x18\x02 \x01(\t\x12+\n\x0bpredictions\x18\x03 \x03(\x0b\x32\x16.google.protobuf.Value\"\xd2\x01\n\x12\x43ountTokensRequest\x12<\n\x08\x65ndpoint\x18\x01 \x01(\tB*\xe0\x41\x02\xfa\x41$\n\"aiplatform.googleapis.com/Endpoint\x12\x12\n\x05model\x18\x03 \x01(\tB\x03\xe0\x41\x02\x12.\n\tinstances\x18\x02 \x03(\x0b\x32\x16.google.protobuf.ValueB\x03\xe0\x41\x02\x12:\n\x08\x63ontents\x18\x04 \x03(\x0b\x32#.google.cloud.aiplatform.v1.ContentB\x03\xe0\x41\x02\"N\n\x13\x43ountTokensResponse\x12\x14\n\x0ctotal_tokens\x18\x01 \x01(\x05\x12!\n\x19total_billable_characters\x18\x02 \x01(\x05\"\x97\x03\n\x16GenerateContentRequest\x12\x12\n\x05model\x18\x05 \x01(\tB\x03\xe0\x41\x02\x12:\n\x08\x63ontents\x18\x02 \x03(\x0b\x32#.google.cloud.aiplatform.v1.ContentB\x03\xe0\x41\x02\x12I\n\x12system_instruction\x18\x08 \x01(\x0b\x32#.google.cloud.aiplatform.v1.ContentB\x03\xe0\x41\x01H\x00\x88\x01\x01\x12\x34\n\x05tools\x18\x06 \x03(\x0b\x32 .google.cloud.aiplatform.v1.ToolB\x03\xe0\x41\x01\x12G\n\x0fsafety_settings\x18\x03 \x03(\x0b\x32).google.cloud.aiplatform.v1.SafetySettingB\x03\xe0\x41\x01\x12L\n\x11generation_config\x18\x04 \x01(\x0b\x32,.google.cloud.aiplatform.v1.GenerationConfigB\x03\xe0\x41\x01\x42\x15\n\x13_system_instruction\"\xd7\x05\n\x17GenerateContentResponse\x12>\n\ncandidates\x18\x02 \x03(\x0b\x32%.google.cloud.aiplatform.v1.CandidateB\x03\xe0\x41\x03\x12`\n\x0fprompt_feedback\x18\x03 \x01(\x0b\x32\x42.google.cloud.aiplatform.v1.GenerateContentResponse.PromptFeedbackB\x03\xe0\x41\x03\x12Y\n\x0eusage_metadata\x18\x04 \x01(\x0b\x32\x41.google.cloud.aiplatform.v1.GenerateContentResponse.UsageMetadata\x1a\xd6\x02\n\x0ePromptFeedback\x12k\n\x0c\x62lock_reason\x18\x01 \x01(\x0e\x32P.google.cloud.aiplatform.v1.GenerateContentResponse.PromptFeedback.BlockedReasonB\x03\xe0\x41\x03\x12\x45\n\x0esafety_ratings\x18\x02 \x03(\x0b\x32(.google.cloud.aiplatform.v1.SafetyRatingB\x03\xe0\x41\x03\x12!\n\x14\x62lock_reason_message\x18\x03 \x01(\tB\x03\xe0\x41\x03\"m\n\rBlockedReason\x12\x1e\n\x1a\x42LOCKED_REASON_UNSPECIFIED\x10\x00\x12\n\n\x06SAFETY\x10\x01\x12\t\n\x05OTHER\x10\x02\x12\r\n\tBLOCKLIST\x10\x03\x12\x16\n\x12PROHIBITED_CONTENT\x10\x04\x1a\x66\n\rUsageMetadata\x12\x1a\n\x12prompt_token_count\x18\x01 \x01(\x05\x12\x1e\n\x16\x63\x61ndidates_token_count\x18\x02 \x01(\x05\x12\x19\n\x11total_token_count\x18\x03 \x01(\x05\x32\xe0\x17\n\x11PredictionService\x12\x94\x02\n\x07Predict\x12*.google.cloud.aiplatform.v1.PredictRequest\x1a+.google.cloud.aiplatform.v1.PredictResponse\"\xaf\x01\xda\x41\x1d\x65ndpoint,instances,parameters\x82\xd3\xe4\x93\x02\x88\x01\"9/v1/{endpoint=projects/*/locations/*/endpoints/*}:predict:\x01*ZH\"C/v1/{endpoint=projects/*/locations/*/publishers/*/models/*}:predict:\x01*\x12\xfe\x01\n\nRawPredict\x12-.google.cloud.aiplatform.v1.RawPredictRequest\x1a\x14.google.api.HttpBody\"\xaa\x01\xda\x41\x12\x65ndpoint,http_body\x82\xd3\xe4\x93\x02\x8e\x01\"</v1/{endpoint=projects/*/locations/*/endpoints/*}:rawPredict:\x01*ZK\"F/v1/{endpoint=projects/*/locations/*/publishers/*/models/*}:rawPredict:\x01*\x12\x98\x02\n\x10StreamRawPredict\x12\x33.google.cloud.aiplatform.v1.StreamRawPredictRequest\x1a\x14.google.api.HttpBody\"\xb6\x01\xda\x41\x12\x65ndpoint,http_body\x82\xd3\xe4\x93\x02\x9a\x01\"B/v1/{endpoint=projects/*/locations/*/endpoints/*}:streamRawPredict:\x01*ZQ\"L/v1/{endpoint=projects/*/locations/*/publishers/*/models/*}:streamRawPredict:\x01*0\x01\x12\xc0\x01\n\rDirectPredict\x12\x30.google.cloud.aiplatform.v1.DirectPredictRequest\x1a\x31.google.cloud.aiplatform.v1.DirectPredictResponse\"J\x82\xd3\xe4\x93\x02\x44\"?/v1/{endpoint=projects/*/locations/*/endpoints/*}:directPredict:\x01*\x12\xcc\x01\n\x10\x44irectRawPredict\x12\x33.google.cloud.aiplatform.v1.DirectRawPredictRequest\x1a\x34.google.cloud.aiplatform.v1.DirectRawPredictResponse\"M\x82\xd3\xe4\x93\x02G\"B/v1/{endpoint=projects/*/locations/*/endpoints/*}:directRawPredict:\x01*\x12\x8c\x01\n\x13StreamDirectPredict\x12\x36.google.cloud.aiplatform.v1.StreamDirectPredictRequest\x1a\x37.google.cloud.aiplatform.v1.StreamDirectPredictResponse\"\x00(\x01\x30\x01\x12\x95\x01\n\x16StreamDirectRawPredict\x12\x39.google.cloud.aiplatform.v1.StreamDirectRawPredictRequest\x1a:.google.cloud.aiplatform.v1.StreamDirectRawPredictResponse\"\x00(\x01\x30\x01\x12\x83\x01\n\x10StreamingPredict\x12\x33.google.cloud.aiplatform.v1.StreamingPredictRequest\x1a\x34.google.cloud.aiplatform.v1.StreamingPredictResponse\"\x00(\x01\x30\x01\x12\xb5\x02\n\x16ServerStreamingPredict\x12\x33.google.cloud.aiplatform.v1.StreamingPredictRequest\x1a\x34.google.cloud.aiplatform.v1.StreamingPredictResponse\"\xad\x01\x82\xd3\xe4\x93\x02\xa6\x01\"H/v1/{endpoint=projects/*/locations/*/endpoints/*}:serverStreamingPredict:\x01*ZW\"R/v1/{endpoint=projects/*/locations/*/publishers/*/models/*}:serverStreamingPredict:\x01*0\x01\x12\x8c\x01\n\x13StreamingRawPredict\x12\x36.google.cloud.aiplatform.v1.StreamingRawPredictRequest\x1a\x37.google.cloud.aiplatform.v1.StreamingRawPredictResponse\"\x00(\x01\x30\x01\x12\xda\x01\n\x07\x45xplain\x12*.google.cloud.aiplatform.v1.ExplainRequest\x1a+.google.cloud.aiplatform.v1.ExplainResponse\"v\xda\x41/endpoint,instances,parameters,deployed_model_id\x82\xd3\xe4\x93\x02>\"9/v1/{endpoint=projects/*/locations/*/endpoints/*}:explain:\x01*\x12\xa7\x02\n\x0fGenerateContent\x12\x32.google.cloud.aiplatform.v1.GenerateContentRequest\x1a\x33.google.cloud.aiplatform.v1.GenerateContentResponse\"\xaa\x01\xda\x41\x0emodel,contents\x82\xd3\xe4\x93\x02\x92\x01\">/v1/{model=projects/*/locations/*/endpoints/*}:generateContent:\x01*ZM\"H/v1/{model=projects/*/locations/*/publishers/*/models/*}:generateContent:\x01*\x12\xbb\x02\n\x15StreamGenerateContent\x12\x32.google.cloud.aiplatform.v1.GenerateContentRequest\x1a\x33.google.cloud.aiplatform.v1.GenerateContentResponse\"\xb6\x01\xda\x41\x0emodel,contents\x82\xd3\xe4\x93\x02\x9e\x01\"D/v1/{model=projects/*/locations/*/endpoints/*}:streamGenerateContent:\x01*ZS\"N/v1/{model=projects/*/locations/*/publishers/*/models/*}:streamGenerateContent:\x01*0\x01\x1aM\xca\x41\x19\x61iplatform.googleapis.com\xd2\x41.https://www.googleapis.com/auth/cloud-platformB\xd4\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\x16PredictionServiceProtoP\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"
19
+ descriptor_data = "\n3google/cloud/aiplatform/v1/prediction_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/httpbody.proto\x1a\x19google/api/resource.proto\x1a(google/cloud/aiplatform/v1/content.proto\x1a,google/cloud/aiplatform/v1/explanation.proto\x1a%google/cloud/aiplatform/v1/tool.proto\x1a&google/cloud/aiplatform/v1/types.proto\x1a\x1cgoogle/protobuf/struct.proto\"\xaa\x01\n\x0ePredictRequest\x12<\n\x08\x65ndpoint\x18\x01 \x01(\tB*\xe0\x41\x02\xfa\x41$\n\"aiplatform.googleapis.com/Endpoint\x12.\n\tinstances\x18\x02 \x03(\x0b\x32\x16.google.protobuf.ValueB\x03\xe0\x41\x02\x12*\n\nparameters\x18\x03 \x01(\x0b\x32\x16.google.protobuf.Value\"\x80\x02\n\x0fPredictResponse\x12+\n\x0bpredictions\x18\x01 \x03(\x0b\x32\x16.google.protobuf.Value\x12\x19\n\x11\x64\x65ployed_model_id\x18\x02 \x01(\t\x12\x36\n\x05model\x18\x03 \x01(\tB\'\xe0\x41\x03\xfa\x41!\n\x1f\x61iplatform.googleapis.com/Model\x12\x1d\n\x10model_version_id\x18\x05 \x01(\tB\x03\xe0\x41\x03\x12\x1f\n\x12model_display_name\x18\x04 \x01(\tB\x03\xe0\x41\x03\x12-\n\x08metadata\x18\x06 \x01(\x0b\x32\x16.google.protobuf.ValueB\x03\xe0\x41\x03\"z\n\x11RawPredictRequest\x12<\n\x08\x65ndpoint\x18\x01 \x01(\tB*\xe0\x41\x02\xfa\x41$\n\"aiplatform.googleapis.com/Endpoint\x12\'\n\thttp_body\x18\x02 \x01(\x0b\x32\x14.google.api.HttpBody\"\x80\x01\n\x17StreamRawPredictRequest\x12<\n\x08\x65ndpoint\x18\x01 \x01(\tB*\xe0\x41\x02\xfa\x41$\n\"aiplatform.googleapis.com/Endpoint\x12\'\n\thttp_body\x18\x02 \x01(\x0b\x32\x14.google.api.HttpBody\"\xc0\x01\n\x14\x44irectPredictRequest\x12<\n\x08\x65ndpoint\x18\x01 \x01(\tB*\xe0\x41\x02\xfa\x41$\n\"aiplatform.googleapis.com/Endpoint\x12\x32\n\x06inputs\x18\x02 \x03(\x0b\x32\".google.cloud.aiplatform.v1.Tensor\x12\x36\n\nparameters\x18\x03 \x01(\x0b\x32\".google.cloud.aiplatform.v1.Tensor\"\x84\x01\n\x15\x44irectPredictResponse\x12\x33\n\x07outputs\x18\x01 \x03(\x0b\x32\".google.cloud.aiplatform.v1.Tensor\x12\x36\n\nparameters\x18\x02 \x01(\x0b\x32\".google.cloud.aiplatform.v1.Tensor\"{\n\x17\x44irectRawPredictRequest\x12<\n\x08\x65ndpoint\x18\x01 \x01(\tB*\xe0\x41\x02\xfa\x41$\n\"aiplatform.googleapis.com/Endpoint\x12\x13\n\x0bmethod_name\x18\x02 \x01(\t\x12\r\n\x05input\x18\x03 \x01(\x0c\"*\n\x18\x44irectRawPredictResponse\x12\x0e\n\x06output\x18\x01 \x01(\x0c\"\xd0\x01\n\x1aStreamDirectPredictRequest\x12<\n\x08\x65ndpoint\x18\x01 \x01(\tB*\xe0\x41\x02\xfa\x41$\n\"aiplatform.googleapis.com/Endpoint\x12\x37\n\x06inputs\x18\x02 \x03(\x0b\x32\".google.cloud.aiplatform.v1.TensorB\x03\xe0\x41\x01\x12;\n\nparameters\x18\x03 \x01(\x0b\x32\".google.cloud.aiplatform.v1.TensorB\x03\xe0\x41\x01\"\x8a\x01\n\x1bStreamDirectPredictResponse\x12\x33\n\x07outputs\x18\x01 \x03(\x0b\x32\".google.cloud.aiplatform.v1.Tensor\x12\x36\n\nparameters\x18\x02 \x01(\x0b\x32\".google.cloud.aiplatform.v1.Tensor\"\x8b\x01\n\x1dStreamDirectRawPredictRequest\x12<\n\x08\x65ndpoint\x18\x01 \x01(\tB*\xe0\x41\x02\xfa\x41$\n\"aiplatform.googleapis.com/Endpoint\x12\x18\n\x0bmethod_name\x18\x02 \x01(\tB\x03\xe0\x41\x01\x12\x12\n\x05input\x18\x03 \x01(\x0c\x42\x03\xe0\x41\x01\"0\n\x1eStreamDirectRawPredictResponse\x12\x0e\n\x06output\x18\x01 \x01(\x0c\"\xc3\x01\n\x17StreamingPredictRequest\x12<\n\x08\x65ndpoint\x18\x01 \x01(\tB*\xe0\x41\x02\xfa\x41$\n\"aiplatform.googleapis.com/Endpoint\x12\x32\n\x06inputs\x18\x02 \x03(\x0b\x32\".google.cloud.aiplatform.v1.Tensor\x12\x36\n\nparameters\x18\x03 \x01(\x0b\x32\".google.cloud.aiplatform.v1.Tensor\"\x87\x01\n\x18StreamingPredictResponse\x12\x33\n\x07outputs\x18\x01 \x03(\x0b\x32\".google.cloud.aiplatform.v1.Tensor\x12\x36\n\nparameters\x18\x02 \x01(\x0b\x32\".google.cloud.aiplatform.v1.Tensor\"~\n\x1aStreamingRawPredictRequest\x12<\n\x08\x65ndpoint\x18\x01 \x01(\tB*\xe0\x41\x02\xfa\x41$\n\"aiplatform.googleapis.com/Endpoint\x12\x13\n\x0bmethod_name\x18\x02 \x01(\t\x12\r\n\x05input\x18\x03 \x01(\x0c\"-\n\x1bStreamingRawPredictResponse\x12\x0e\n\x06output\x18\x01 \x01(\x0c\"\x9d\x02\n\x0e\x45xplainRequest\x12<\n\x08\x65ndpoint\x18\x01 \x01(\tB*\xe0\x41\x02\xfa\x41$\n\"aiplatform.googleapis.com/Endpoint\x12.\n\tinstances\x18\x02 \x03(\x0b\x32\x16.google.protobuf.ValueB\x03\xe0\x41\x02\x12*\n\nparameters\x18\x04 \x01(\x0b\x32\x16.google.protobuf.Value\x12V\n\x19\x65xplanation_spec_override\x18\x05 \x01(\x0b\x32\x33.google.cloud.aiplatform.v1.ExplanationSpecOverride\x12\x19\n\x11\x64\x65ployed_model_id\x18\x03 \x01(\t\"\x98\x01\n\x0f\x45xplainResponse\x12=\n\x0c\x65xplanations\x18\x01 \x03(\x0b\x32\'.google.cloud.aiplatform.v1.Explanation\x12\x19\n\x11\x64\x65ployed_model_id\x18\x02 \x01(\t\x12+\n\x0bpredictions\x18\x03 \x03(\x0b\x32\x16.google.protobuf.Value\"\xd2\x01\n\x12\x43ountTokensRequest\x12<\n\x08\x65ndpoint\x18\x01 \x01(\tB*\xe0\x41\x02\xfa\x41$\n\"aiplatform.googleapis.com/Endpoint\x12\x12\n\x05model\x18\x03 \x01(\tB\x03\xe0\x41\x02\x12.\n\tinstances\x18\x02 \x03(\x0b\x32\x16.google.protobuf.ValueB\x03\xe0\x41\x02\x12:\n\x08\x63ontents\x18\x04 \x03(\x0b\x32#.google.cloud.aiplatform.v1.ContentB\x03\xe0\x41\x02\"N\n\x13\x43ountTokensResponse\x12\x14\n\x0ctotal_tokens\x18\x01 \x01(\x05\x12!\n\x19total_billable_characters\x18\x02 \x01(\x05\"\xd9\x03\n\x16GenerateContentRequest\x12\x12\n\x05model\x18\x05 \x01(\tB\x03\xe0\x41\x02\x12:\n\x08\x63ontents\x18\x02 \x03(\x0b\x32#.google.cloud.aiplatform.v1.ContentB\x03\xe0\x41\x02\x12I\n\x12system_instruction\x18\x08 \x01(\x0b\x32#.google.cloud.aiplatform.v1.ContentB\x03\xe0\x41\x01H\x00\x88\x01\x01\x12\x34\n\x05tools\x18\x06 \x03(\x0b\x32 .google.cloud.aiplatform.v1.ToolB\x03\xe0\x41\x01\x12@\n\x0btool_config\x18\x07 \x01(\x0b\x32&.google.cloud.aiplatform.v1.ToolConfigB\x03\xe0\x41\x01\x12G\n\x0fsafety_settings\x18\x03 \x03(\x0b\x32).google.cloud.aiplatform.v1.SafetySettingB\x03\xe0\x41\x01\x12L\n\x11generation_config\x18\x04 \x01(\x0b\x32,.google.cloud.aiplatform.v1.GenerationConfigB\x03\xe0\x41\x01\x42\x15\n\x13_system_instruction\"\xd7\x05\n\x17GenerateContentResponse\x12>\n\ncandidates\x18\x02 \x03(\x0b\x32%.google.cloud.aiplatform.v1.CandidateB\x03\xe0\x41\x03\x12`\n\x0fprompt_feedback\x18\x03 \x01(\x0b\x32\x42.google.cloud.aiplatform.v1.GenerateContentResponse.PromptFeedbackB\x03\xe0\x41\x03\x12Y\n\x0eusage_metadata\x18\x04 \x01(\x0b\x32\x41.google.cloud.aiplatform.v1.GenerateContentResponse.UsageMetadata\x1a\xd6\x02\n\x0ePromptFeedback\x12k\n\x0c\x62lock_reason\x18\x01 \x01(\x0e\x32P.google.cloud.aiplatform.v1.GenerateContentResponse.PromptFeedback.BlockedReasonB\x03\xe0\x41\x03\x12\x45\n\x0esafety_ratings\x18\x02 \x03(\x0b\x32(.google.cloud.aiplatform.v1.SafetyRatingB\x03\xe0\x41\x03\x12!\n\x14\x62lock_reason_message\x18\x03 \x01(\tB\x03\xe0\x41\x03\"m\n\rBlockedReason\x12\x1e\n\x1a\x42LOCKED_REASON_UNSPECIFIED\x10\x00\x12\n\n\x06SAFETY\x10\x01\x12\t\n\x05OTHER\x10\x02\x12\r\n\tBLOCKLIST\x10\x03\x12\x16\n\x12PROHIBITED_CONTENT\x10\x04\x1a\x66\n\rUsageMetadata\x12\x1a\n\x12prompt_token_count\x18\x01 \x01(\x05\x12\x1e\n\x16\x63\x61ndidates_token_count\x18\x02 \x01(\x05\x12\x19\n\x11total_token_count\x18\x03 \x01(\x05\x32\x97\x1c\n\x11PredictionService\x12\x94\x02\n\x07Predict\x12*.google.cloud.aiplatform.v1.PredictRequest\x1a+.google.cloud.aiplatform.v1.PredictResponse\"\xaf\x01\xda\x41\x1d\x65ndpoint,instances,parameters\x82\xd3\xe4\x93\x02\x88\x01\"9/v1/{endpoint=projects/*/locations/*/endpoints/*}:predict:\x01*ZH\"C/v1/{endpoint=projects/*/locations/*/publishers/*/models/*}:predict:\x01*\x12\xfe\x01\n\nRawPredict\x12-.google.cloud.aiplatform.v1.RawPredictRequest\x1a\x14.google.api.HttpBody\"\xaa\x01\xda\x41\x12\x65ndpoint,http_body\x82\xd3\xe4\x93\x02\x8e\x01\"</v1/{endpoint=projects/*/locations/*/endpoints/*}:rawPredict:\x01*ZK\"F/v1/{endpoint=projects/*/locations/*/publishers/*/models/*}:rawPredict:\x01*\x12\x98\x02\n\x10StreamRawPredict\x12\x33.google.cloud.aiplatform.v1.StreamRawPredictRequest\x1a\x14.google.api.HttpBody\"\xb6\x01\xda\x41\x12\x65ndpoint,http_body\x82\xd3\xe4\x93\x02\x9a\x01\"B/v1/{endpoint=projects/*/locations/*/endpoints/*}:streamRawPredict:\x01*ZQ\"L/v1/{endpoint=projects/*/locations/*/publishers/*/models/*}:streamRawPredict:\x01*0\x01\x12\x92\x02\n\rDirectPredict\x12\x30.google.cloud.aiplatform.v1.DirectPredictRequest\x1a\x31.google.cloud.aiplatform.v1.DirectPredictResponse\"\x9b\x01\x82\xd3\xe4\x93\x02\x94\x01\"?/v1/{endpoint=projects/*/locations/*/endpoints/*}:directPredict:\x01*ZN\"I/v1/{endpoint=projects/*/locations/*/publishers/*/models/*}:directPredict:\x01*\x12\xa1\x02\n\x10\x44irectRawPredict\x12\x33.google.cloud.aiplatform.v1.DirectRawPredictRequest\x1a\x34.google.cloud.aiplatform.v1.DirectRawPredictResponse\"\xa1\x01\x82\xd3\xe4\x93\x02\x9a\x01\"B/v1/{endpoint=projects/*/locations/*/endpoints/*}:directRawPredict:\x01*ZQ\"L/v1/{endpoint=projects/*/locations/*/publishers/*/models/*}:directRawPredict:\x01*\x12\xb4\x02\n\x13StreamDirectPredict\x12\x36.google.cloud.aiplatform.v1.StreamDirectPredictRequest\x1a\x37.google.cloud.aiplatform.v1.StreamDirectPredictResponse\"\xa7\x01\x82\xd3\xe4\x93\x02\xa0\x01\"E/v1/{endpoint=projects/*/locations/*/endpoints/*}:streamDirectPredict:\x01*ZT\"O/v1/{endpoint=projects/*/locations/*/publishers/*/models/*}:streamDirectPredict:\x01*(\x01\x30\x01\x12\xc3\x02\n\x16StreamDirectRawPredict\x12\x39.google.cloud.aiplatform.v1.StreamDirectRawPredictRequest\x1a:.google.cloud.aiplatform.v1.StreamDirectRawPredictResponse\"\xad\x01\x82\xd3\xe4\x93\x02\xa6\x01\"H/v1/{endpoint=projects/*/locations/*/endpoints/*}:streamDirectRawPredict:\x01*ZW\"R/v1/{endpoint=projects/*/locations/*/publishers/*/models/*}:streamDirectRawPredict:\x01*(\x01\x30\x01\x12\x83\x01\n\x10StreamingPredict\x12\x33.google.cloud.aiplatform.v1.StreamingPredictRequest\x1a\x34.google.cloud.aiplatform.v1.StreamingPredictResponse\"\x00(\x01\x30\x01\x12\xb5\x02\n\x16ServerStreamingPredict\x12\x33.google.cloud.aiplatform.v1.StreamingPredictRequest\x1a\x34.google.cloud.aiplatform.v1.StreamingPredictResponse\"\xad\x01\x82\xd3\xe4\x93\x02\xa6\x01\"H/v1/{endpoint=projects/*/locations/*/endpoints/*}:serverStreamingPredict:\x01*ZW\"R/v1/{endpoint=projects/*/locations/*/publishers/*/models/*}:serverStreamingPredict:\x01*0\x01\x12\x8c\x01\n\x13StreamingRawPredict\x12\x36.google.cloud.aiplatform.v1.StreamingRawPredictRequest\x1a\x37.google.cloud.aiplatform.v1.StreamingRawPredictResponse\"\x00(\x01\x30\x01\x12\xda\x01\n\x07\x45xplain\x12*.google.cloud.aiplatform.v1.ExplainRequest\x1a+.google.cloud.aiplatform.v1.ExplainResponse\"v\xda\x41/endpoint,instances,parameters,deployed_model_id\x82\xd3\xe4\x93\x02>\"9/v1/{endpoint=projects/*/locations/*/endpoints/*}:explain:\x01*\x12\xa7\x02\n\x0fGenerateContent\x12\x32.google.cloud.aiplatform.v1.GenerateContentRequest\x1a\x33.google.cloud.aiplatform.v1.GenerateContentResponse\"\xaa\x01\xda\x41\x0emodel,contents\x82\xd3\xe4\x93\x02\x92\x01\">/v1/{model=projects/*/locations/*/endpoints/*}:generateContent:\x01*ZM\"H/v1/{model=projects/*/locations/*/publishers/*/models/*}:generateContent:\x01*\x12\xbb\x02\n\x15StreamGenerateContent\x12\x32.google.cloud.aiplatform.v1.GenerateContentRequest\x1a\x33.google.cloud.aiplatform.v1.GenerateContentResponse\"\xb6\x01\xda\x41\x0emodel,contents\x82\xd3\xe4\x93\x02\x9e\x01\"D/v1/{model=projects/*/locations/*/endpoints/*}:streamGenerateContent:\x01*ZS\"N/v1/{model=projects/*/locations/*/publishers/*/models/*}:streamGenerateContent:\x01*0\x01\x1a\x86\x01\xca\x41\x19\x61iplatform.googleapis.com\xd2\x41ghttps://www.googleapis.com/auth/cloud-platform,https://www.googleapis.com/auth/cloud-platform.read-onlyB\xd4\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\x16PredictionServiceProtoP\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"
20
20
 
21
21
  pool = Google::Protobuf::DescriptorPool.generated_pool
22
22
 
@@ -10,7 +10,7 @@ require 'google/cloud/aiplatform/v1/openapi_pb'
10
10
  require 'google/protobuf/struct_pb'
11
11
 
12
12
 
13
- descriptor_data = "\n%google/cloud/aiplatform/v1/tool.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a(google/cloud/aiplatform/v1/openapi.proto\x1a\x1cgoogle/protobuf/struct.proto\"\xf3\x01\n\x04Tool\x12S\n\x15\x66unction_declarations\x18\x01 \x03(\x0b\x32/.google.cloud.aiplatform.v1.FunctionDeclarationB\x03\xe0\x41\x01\x12=\n\tretrieval\x18\x02 \x01(\x0b\x32%.google.cloud.aiplatform.v1.RetrievalB\x03\xe0\x41\x01\x12W\n\x17google_search_retrieval\x18\x03 \x01(\x0b\x32\x31.google.cloud.aiplatform.v1.GoogleSearchRetrievalB\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\"\x7f\n\tRetrieval\x12\x46\n\x10vertex_ai_search\x18\x02 \x01(\x0b\x32*.google.cloud.aiplatform.v1.VertexAISearchH\x00\x12 \n\x13\x64isable_attribution\x18\x03 \x01(\x08\x42\x03\xe0\x41\x01\x42\x08\n\x06source\"(\n\x0eVertexAISearch\x12\x16\n\tdatastore\x18\x01 \x01(\tB\x03\xe0\x41\x02\"9\n\x15GoogleSearchRetrieval\x12 \n\x13\x64isable_attribution\x18\x01 \x01(\x08\x42\x03\xe0\x41\x01\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"
13
+ descriptor_data = "\n%google/cloud/aiplatform/v1/tool.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a(google/cloud/aiplatform/v1/openapi.proto\x1a\x1cgoogle/protobuf/struct.proto\"\xf3\x01\n\x04Tool\x12S\n\x15\x66unction_declarations\x18\x01 \x03(\x0b\x32/.google.cloud.aiplatform.v1.FunctionDeclarationB\x03\xe0\x41\x01\x12=\n\tretrieval\x18\x02 \x01(\x0b\x32%.google.cloud.aiplatform.v1.RetrievalB\x03\xe0\x41\x01\x12W\n\x17google_search_retrieval\x18\x03 \x01(\x0b\x32\x31.google.cloud.aiplatform.v1.GoogleSearchRetrievalB\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\"\x7f\n\tRetrieval\x12\x46\n\x10vertex_ai_search\x18\x02 \x01(\x0b\x32*.google.cloud.aiplatform.v1.VertexAISearchH\x00\x12 \n\x13\x64isable_attribution\x18\x03 \x01(\x08\x42\x03\xe0\x41\x01\x42\x08\n\x06source\"(\n\x0eVertexAISearch\x12\x16\n\tdatastore\x18\x01 \x01(\tB\x03\xe0\x41\x02\"\x17\n\x15GoogleSearchRetrieval\"e\n\nToolConfig\x12W\n\x17\x66unction_calling_config\x18\x01 \x01(\x0b\x32\x31.google.cloud.aiplatform.v1.FunctionCallingConfigB\x03\xe0\x41\x01\"\xc2\x01\n\x15\x46unctionCallingConfig\x12I\n\x04mode\x18\x01 \x01(\x0e\x32\x36.google.cloud.aiplatform.v1.FunctionCallingConfig.ModeB\x03\xe0\x41\x01\x12#\n\x16\x61llowed_function_names\x18\x02 \x03(\tB\x03\xe0\x41\x01\"9\n\x04Mode\x12\x14\n\x10MODE_UNSPECIFIED\x10\x00\x12\x08\n\x04\x41UTO\x10\x01\x12\x07\n\x03\x41NY\x10\x02\x12\x08\n\x04NONE\x10\x03\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"
14
14
 
15
15
  pool = Google::Protobuf::DescriptorPool.generated_pool
16
16
 
@@ -49,6 +49,9 @@ module Google
49
49
  Retrieval = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.Retrieval").msgclass
50
50
  VertexAISearch = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.VertexAISearch").msgclass
51
51
  GoogleSearchRetrieval = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.GoogleSearchRetrieval").msgclass
52
+ ToolConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ToolConfig").msgclass
53
+ FunctionCallingConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.FunctionCallingConfig").msgclass
54
+ FunctionCallingConfig::Mode = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.FunctionCallingConfig.Mode").enummodule
52
55
  end
53
56
  end
54
57
  end
@@ -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
@@ -363,58 +372,10 @@ module Google
363
372
  end
364
373
  end
365
374
 
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
-
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
408
- end
409
- end
410
-
411
375
  # Metadata returned to client when grounding is enabled.
412
376
  # @!attribute [rw] web_search_queries
413
377
  # @return [::Array<::String>]
414
378
  # 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
379
  # @!attribute [rw] search_entry_point
419
380
  # @return [::Google::Cloud::AIPlatform::V1::SearchEntryPoint]
420
381
  # 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