google-cloud-ai_platform-v1 0.33.0 → 0.35.0

Sign up to get free protection for your applications and to get access to all the features.
Files changed (63) hide show
  1. checksums.yaml +4 -4
  2. data/lib/google/cloud/ai_platform/v1/feature_online_store_admin_service/client.rb +1796 -0
  3. data/lib/google/cloud/ai_platform/v1/feature_online_store_admin_service/credentials.rb +47 -0
  4. data/lib/google/cloud/ai_platform/v1/feature_online_store_admin_service/operations.rb +787 -0
  5. data/lib/google/cloud/ai_platform/v1/feature_online_store_admin_service/paths.rb +111 -0
  6. data/lib/google/cloud/ai_platform/v1/feature_online_store_admin_service.rb +51 -0
  7. data/lib/google/cloud/ai_platform/v1/feature_online_store_service/client.rb +431 -0
  8. data/lib/google/cloud/ai_platform/v1/feature_online_store_service/credentials.rb +47 -0
  9. data/lib/google/cloud/ai_platform/v1/feature_online_store_service/paths.rb +54 -0
  10. data/lib/google/cloud/ai_platform/v1/feature_online_store_service.rb +47 -0
  11. data/lib/google/cloud/ai_platform/v1/feature_registry_service/client.rb +1498 -0
  12. data/lib/google/cloud/ai_platform/v1/feature_registry_service/credentials.rb +47 -0
  13. data/lib/google/cloud/ai_platform/v1/feature_registry_service/operations.rb +787 -0
  14. data/lib/google/cloud/ai_platform/v1/feature_registry_service/paths.rb +139 -0
  15. data/lib/google/cloud/ai_platform/v1/feature_registry_service.rb +51 -0
  16. data/lib/google/cloud/ai_platform/v1/featurestore_service/client.rb +20 -7
  17. data/lib/google/cloud/ai_platform/v1/featurestore_service/paths.rb +53 -8
  18. data/lib/google/cloud/ai_platform/v1/llm_utility_service/client.rb +528 -0
  19. data/lib/google/cloud/ai_platform/v1/llm_utility_service/credentials.rb +47 -0
  20. data/lib/google/cloud/ai_platform/v1/llm_utility_service/paths.rb +78 -0
  21. data/lib/google/cloud/ai_platform/v1/llm_utility_service.rb +49 -0
  22. data/lib/google/cloud/ai_platform/v1/prediction_service/client.rb +374 -0
  23. data/lib/google/cloud/ai_platform/v1/version.rb +1 -1
  24. data/lib/google/cloud/ai_platform/v1.rb +5 -1
  25. data/lib/google/cloud/aiplatform/v1/custom_job_pb.rb +1 -1
  26. data/lib/google/cloud/aiplatform/v1/feature_group_pb.rb +50 -0
  27. data/lib/google/cloud/aiplatform/v1/feature_online_store_admin_service_pb.rb +79 -0
  28. data/lib/google/cloud/aiplatform/v1/feature_online_store_admin_service_services_pb.rb +71 -0
  29. data/lib/google/cloud/aiplatform/v1/feature_online_store_pb.rb +50 -0
  30. data/lib/google/cloud/aiplatform/v1/feature_online_store_service_pb.rb +56 -0
  31. data/lib/google/cloud/aiplatform/v1/feature_online_store_service_services_pb.rb +44 -0
  32. data/lib/google/cloud/aiplatform/v1/feature_pb.rb +1 -1
  33. data/lib/google/cloud/aiplatform/v1/feature_registry_service_pb.rb +66 -0
  34. data/lib/google/cloud/aiplatform/v1/feature_registry_service_services_pb.rb +64 -0
  35. data/lib/google/cloud/aiplatform/v1/feature_view_pb.rb +51 -0
  36. data/lib/google/cloud/aiplatform/v1/feature_view_sync_pb.rb +51 -0
  37. data/lib/google/cloud/aiplatform/v1/index_pb.rb +3 -1
  38. data/lib/google/cloud/aiplatform/v1/llm_utility_service_pb.rb +52 -0
  39. data/lib/google/cloud/aiplatform/v1/llm_utility_service_services_pb.rb +47 -0
  40. data/lib/google/cloud/aiplatform/v1/machine_resources_pb.rb +1 -1
  41. data/lib/google/cloud/aiplatform/v1/model_pb.rb +5 -1
  42. data/lib/google/cloud/aiplatform/v1/prediction_service_pb.rb +9 -1
  43. data/lib/google/cloud/aiplatform/v1/prediction_service_services_pb.rb +10 -0
  44. data/lib/google/cloud/aiplatform/v1/study_pb.rb +5 -1
  45. data/proto_docs/google/api/client.rb +13 -0
  46. data/proto_docs/google/cloud/aiplatform/v1/custom_job.rb +6 -0
  47. data/proto_docs/google/cloud/aiplatform/v1/feature.rb +21 -10
  48. data/proto_docs/google/cloud/aiplatform/v1/feature_group.rb +91 -0
  49. data/proto_docs/google/cloud/aiplatform/v1/feature_online_store.rb +124 -0
  50. data/proto_docs/google/cloud/aiplatform/v1/feature_online_store_admin_service.rb +458 -0
  51. data/proto_docs/google/cloud/aiplatform/v1/feature_online_store_service.rb +102 -0
  52. data/proto_docs/google/cloud/aiplatform/v1/feature_registry_service.rb +208 -0
  53. data/proto_docs/google/cloud/aiplatform/v1/feature_view.rb +131 -0
  54. data/proto_docs/google/cloud/aiplatform/v1/feature_view_sync.rb +48 -0
  55. data/proto_docs/google/cloud/aiplatform/v1/featurestore_service.rb +32 -7
  56. data/proto_docs/google/cloud/aiplatform/v1/index.rb +53 -0
  57. data/proto_docs/google/cloud/aiplatform/v1/index_endpoint.rb +5 -4
  58. data/proto_docs/google/cloud/aiplatform/v1/llm_utility_service.rb +64 -0
  59. data/proto_docs/google/cloud/aiplatform/v1/machine_resources.rb +4 -0
  60. data/proto_docs/google/cloud/aiplatform/v1/model.rb +61 -0
  61. data/proto_docs/google/cloud/aiplatform/v1/prediction_service.rb +140 -2
  62. data/proto_docs/google/cloud/aiplatform/v1/study.rb +75 -0
  63. metadata +40 -2
@@ -12,7 +12,7 @@ require 'google/protobuf/struct_pb'
12
12
  require 'google/protobuf/timestamp_pb'
13
13
 
14
14
 
15
- descriptor_data = "\n&google/cloud/aiplatform/v1/index.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a\x33google/cloud/aiplatform/v1/deployed_index_ref.proto\x1a\x30google/cloud/aiplatform/v1/encryption_spec.proto\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1fgoogle/protobuf/timestamp.proto\"\xec\x06\n\x05Index\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x03\x12\x19\n\x0c\x64isplay_name\x18\x02 \x01(\tB\x03\xe0\x41\x02\x12\x13\n\x0b\x64\x65scription\x18\x03 \x01(\t\x12 \n\x13metadata_schema_uri\x18\x04 \x01(\tB\x03\xe0\x41\x05\x12(\n\x08metadata\x18\x06 \x01(\x0b\x32\x16.google.protobuf.Value\x12K\n\x10\x64\x65ployed_indexes\x18\x07 \x03(\x0b\x32,.google.cloud.aiplatform.v1.DeployedIndexRefB\x03\xe0\x41\x03\x12\x0c\n\x04\x65tag\x18\x08 \x01(\t\x12=\n\x06labels\x18\t \x03(\x0b\x32-.google.cloud.aiplatform.v1.Index.LabelsEntry\x12\x34\n\x0b\x63reate_time\x18\n \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x34\n\x0bupdate_time\x18\x0b \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12@\n\x0bindex_stats\x18\x0e \x01(\x0b\x32&.google.cloud.aiplatform.v1.IndexStatsB\x03\xe0\x41\x03\x12U\n\x13index_update_method\x18\x10 \x01(\x0e\x32\x33.google.cloud.aiplatform.v1.Index.IndexUpdateMethodB\x03\xe0\x41\x05\x12H\n\x0f\x65ncryption_spec\x18\x11 \x01(\x0b\x32*.google.cloud.aiplatform.v1.EncryptionSpecB\x03\xe0\x41\x05\x1a-\n\x0bLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"]\n\x11IndexUpdateMethod\x12#\n\x1fINDEX_UPDATE_METHOD_UNSPECIFIED\x10\x00\x12\x10\n\x0c\x42\x41TCH_UPDATE\x10\x01\x12\x11\n\rSTREAM_UPDATE\x10\x02:]\xea\x41Z\n\x1f\x61iplatform.googleapis.com/Index\x12\x37projects/{project}/locations/{location}/indexes/{index}\"\xdf\x02\n\x0eIndexDatapoint\x12\x19\n\x0c\x64\x61tapoint_id\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x1b\n\x0e\x66\x65\x61ture_vector\x18\x02 \x03(\x02\x42\x03\xe0\x41\x02\x12N\n\trestricts\x18\x04 \x03(\x0b\x32\x36.google.cloud.aiplatform.v1.IndexDatapoint.RestrictionB\x03\xe0\x41\x01\x12Q\n\x0c\x63rowding_tag\x18\x05 \x01(\x0b\x32\x36.google.cloud.aiplatform.v1.IndexDatapoint.CrowdingTagB\x03\xe0\x41\x01\x1aG\n\x0bRestriction\x12\x11\n\tnamespace\x18\x01 \x01(\t\x12\x12\n\nallow_list\x18\x02 \x03(\t\x12\x11\n\tdeny_list\x18\x03 \x03(\t\x1a)\n\x0b\x43rowdingTag\x12\x1a\n\x12\x63rowding_attribute\x18\x01 \x01(\t\"C\n\nIndexStats\x12\x1a\n\rvectors_count\x18\x01 \x01(\x03\x42\x03\xe0\x41\x03\x12\x19\n\x0cshards_count\x18\x02 \x01(\x05\x42\x03\xe0\x41\x03\x42\xc8\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\nIndexProtoP\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"
15
+ descriptor_data = "\n&google/cloud/aiplatform/v1/index.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a\x33google/cloud/aiplatform/v1/deployed_index_ref.proto\x1a\x30google/cloud/aiplatform/v1/encryption_spec.proto\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1fgoogle/protobuf/timestamp.proto\"\xec\x06\n\x05Index\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x03\x12\x19\n\x0c\x64isplay_name\x18\x02 \x01(\tB\x03\xe0\x41\x02\x12\x13\n\x0b\x64\x65scription\x18\x03 \x01(\t\x12 \n\x13metadata_schema_uri\x18\x04 \x01(\tB\x03\xe0\x41\x05\x12(\n\x08metadata\x18\x06 \x01(\x0b\x32\x16.google.protobuf.Value\x12K\n\x10\x64\x65ployed_indexes\x18\x07 \x03(\x0b\x32,.google.cloud.aiplatform.v1.DeployedIndexRefB\x03\xe0\x41\x03\x12\x0c\n\x04\x65tag\x18\x08 \x01(\t\x12=\n\x06labels\x18\t \x03(\x0b\x32-.google.cloud.aiplatform.v1.Index.LabelsEntry\x12\x34\n\x0b\x63reate_time\x18\n \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x34\n\x0bupdate_time\x18\x0b \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12@\n\x0bindex_stats\x18\x0e \x01(\x0b\x32&.google.cloud.aiplatform.v1.IndexStatsB\x03\xe0\x41\x03\x12U\n\x13index_update_method\x18\x10 \x01(\x0e\x32\x33.google.cloud.aiplatform.v1.Index.IndexUpdateMethodB\x03\xe0\x41\x05\x12H\n\x0f\x65ncryption_spec\x18\x11 \x01(\x0b\x32*.google.cloud.aiplatform.v1.EncryptionSpecB\x03\xe0\x41\x05\x1a-\n\x0bLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"]\n\x11IndexUpdateMethod\x12#\n\x1fINDEX_UPDATE_METHOD_UNSPECIFIED\x10\x00\x12\x10\n\x0c\x42\x41TCH_UPDATE\x10\x01\x12\x11\n\rSTREAM_UPDATE\x10\x02:]\xea\x41Z\n\x1f\x61iplatform.googleapis.com/Index\x12\x37projects/{project}/locations/{location}/indexes/{index}\"\xf4\x05\n\x0eIndexDatapoint\x12\x19\n\x0c\x64\x61tapoint_id\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x1b\n\x0e\x66\x65\x61ture_vector\x18\x02 \x03(\x02\x42\x03\xe0\x41\x02\x12N\n\trestricts\x18\x04 \x03(\x0b\x32\x36.google.cloud.aiplatform.v1.IndexDatapoint.RestrictionB\x03\xe0\x41\x01\x12]\n\x11numeric_restricts\x18\x06 \x03(\x0b\x32=.google.cloud.aiplatform.v1.IndexDatapoint.NumericRestrictionB\x03\xe0\x41\x01\x12Q\n\x0c\x63rowding_tag\x18\x05 \x01(\x0b\x32\x36.google.cloud.aiplatform.v1.IndexDatapoint.CrowdingTagB\x03\xe0\x41\x01\x1aG\n\x0bRestriction\x12\x11\n\tnamespace\x18\x01 \x01(\t\x12\x12\n\nallow_list\x18\x02 \x03(\t\x12\x11\n\tdeny_list\x18\x03 \x03(\t\x1a\xb3\x02\n\x12NumericRestriction\x12\x13\n\tvalue_int\x18\x02 \x01(\x03H\x00\x12\x15\n\x0bvalue_float\x18\x03 \x01(\x02H\x00\x12\x16\n\x0cvalue_double\x18\x04 \x01(\x01H\x00\x12\x11\n\tnamespace\x18\x01 \x01(\t\x12R\n\x02op\x18\x05 \x01(\x0e\x32\x46.google.cloud.aiplatform.v1.IndexDatapoint.NumericRestriction.Operator\"i\n\x08Operator\x12\x18\n\x14OPERATOR_UNSPECIFIED\x10\x00\x12\x08\n\x04LESS\x10\x01\x12\x0e\n\nLESS_EQUAL\x10\x02\x12\t\n\x05\x45QUAL\x10\x03\x12\x11\n\rGREATER_EQUAL\x10\x04\x12\x0b\n\x07GREATER\x10\x05\x42\x07\n\x05Value\x1a)\n\x0b\x43rowdingTag\x12\x1a\n\x12\x63rowding_attribute\x18\x01 \x01(\t\"C\n\nIndexStats\x12\x1a\n\rvectors_count\x18\x01 \x01(\x03\x42\x03\xe0\x41\x03\x12\x19\n\x0cshards_count\x18\x02 \x01(\x05\x42\x03\xe0\x41\x03\x42\xc8\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\nIndexProtoP\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
16
 
17
17
  pool = Google::Protobuf::DescriptorPool.generated_pool
18
18
 
@@ -50,6 +50,8 @@ module Google
50
50
  Index::IndexUpdateMethod = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.Index.IndexUpdateMethod").enummodule
51
51
  IndexDatapoint = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.IndexDatapoint").msgclass
52
52
  IndexDatapoint::Restriction = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.IndexDatapoint.Restriction").msgclass
53
+ IndexDatapoint::NumericRestriction = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.IndexDatapoint.NumericRestriction").msgclass
54
+ IndexDatapoint::NumericRestriction::Operator = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.IndexDatapoint.NumericRestriction.Operator").enummodule
53
55
  IndexDatapoint::CrowdingTag = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.IndexDatapoint.CrowdingTag").msgclass
54
56
  IndexStats = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.IndexStats").msgclass
55
57
  end
@@ -0,0 +1,52 @@
1
+ # frozen_string_literal: true
2
+ # Generated by the protocol buffer compiler. DO NOT EDIT!
3
+ # source: google/cloud/aiplatform/v1/llm_utility_service.proto
4
+
5
+ require 'google/protobuf'
6
+
7
+ require 'google/api/annotations_pb'
8
+ require 'google/api/client_pb'
9
+ require 'google/api/field_behavior_pb'
10
+ require 'google/api/resource_pb'
11
+ require 'google/cloud/aiplatform/v1/prediction_service_pb'
12
+ require 'google/protobuf/struct_pb'
13
+
14
+
15
+ descriptor_data = "\n4google/cloud/aiplatform/v1/llm_utility_service.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1cgoogle/api/annotations.proto\x1a\x17google/api/client.proto\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a\x33google/cloud/aiplatform/v1/prediction_service.proto\x1a\x1cgoogle/protobuf/struct.proto\"\x84\x01\n\x14\x43omputeTokensRequest\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\"/\n\nTokensInfo\x12\x0e\n\x06tokens\x18\x01 \x03(\x0c\x12\x11\n\ttoken_ids\x18\x02 \x03(\x03\"T\n\x15\x43omputeTokensResponse\x12;\n\x0btokens_info\x18\x01 \x03(\x0b\x32&.google.cloud.aiplatform.v1.TokensInfo2\xac\x05\n\x11LlmUtilityService\x12\x9d\x02\n\x0b\x43ountTokens\x12..google.cloud.aiplatform.v1.CountTokensRequest\x1a/.google.cloud.aiplatform.v1.CountTokensResponse\"\xac\x01\x82\xd3\xe4\x93\x02\x90\x01\"=/v1/{endpoint=projects/*/locations/*/endpoints/*}:countTokens:\x01*ZL\"G/v1/{endpoint=projects/*/locations/*/publishers/*/models/*}:countTokens:\x01*\xda\x41\x12\x65ndpoint,instances\x12\xa7\x02\n\rComputeTokens\x12\x30.google.cloud.aiplatform.v1.ComputeTokensRequest\x1a\x31.google.cloud.aiplatform.v1.ComputeTokensResponse\"\xb0\x01\x82\xd3\xe4\x93\x02\x94\x01\"?/v1/{endpoint=projects/*/locations/*/endpoints/*}:computeTokens:\x01*ZN\"I/v1/{endpoint=projects/*/locations/*/publishers/*/models/*}:computeTokens:\x01*\xda\x41\x12\x65ndpoint,instances\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\x16LlmUtilityServiceProtoP\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
+ pool = Google::Protobuf::DescriptorPool.generated_pool
18
+
19
+ begin
20
+ pool.add_serialized_file(descriptor_data)
21
+ rescue TypeError => e
22
+ # Compatibility code: will be removed in the next major version.
23
+ require 'google/protobuf/descriptor_pb'
24
+ parsed = Google::Protobuf::FileDescriptorProto.decode(descriptor_data)
25
+ parsed.clear_dependency
26
+ serialized = parsed.class.encode(parsed)
27
+ file = pool.add_serialized_file(serialized)
28
+ warn "Warning: Protobuf detected an import path issue while loading generated file #{__FILE__}"
29
+ imports = [
30
+ ["google.protobuf.Value", "google/protobuf/struct.proto"],
31
+ ]
32
+ imports.each do |type_name, expected_filename|
33
+ import_file = pool.lookup(type_name).file_descriptor
34
+ if import_file.name != expected_filename
35
+ warn "- #{file.name} imports #{expected_filename}, but that import was loaded as #{import_file.name}"
36
+ end
37
+ end
38
+ warn "Each proto file must use a consistent fully-qualified name."
39
+ warn "This will become an error in the next major version."
40
+ end
41
+
42
+ module Google
43
+ module Cloud
44
+ module AIPlatform
45
+ module V1
46
+ ComputeTokensRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ComputeTokensRequest").msgclass
47
+ TokensInfo = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.TokensInfo").msgclass
48
+ ComputeTokensResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ComputeTokensResponse").msgclass
49
+ end
50
+ end
51
+ end
52
+ end
@@ -0,0 +1,47 @@
1
+ # Generated by the protocol buffer compiler. DO NOT EDIT!
2
+ # Source: google/cloud/aiplatform/v1/llm_utility_service.proto for package 'Google.Cloud.AIPlatform.V1'
3
+ # Original file comments:
4
+ # Copyright 2023 Google LLC
5
+ #
6
+ # Licensed under the Apache License, Version 2.0 (the "License");
7
+ # you may not use this file except in compliance with the License.
8
+ # You may obtain a copy of the License at
9
+ #
10
+ # http://www.apache.org/licenses/LICENSE-2.0
11
+ #
12
+ # Unless required by applicable law or agreed to in writing, software
13
+ # distributed under the License is distributed on an "AS IS" BASIS,
14
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15
+ # See the License for the specific language governing permissions and
16
+ # limitations under the License.
17
+ #
18
+
19
+ require 'grpc'
20
+ require 'google/cloud/aiplatform/v1/llm_utility_service_pb'
21
+
22
+ module Google
23
+ module Cloud
24
+ module AIPlatform
25
+ module V1
26
+ module LlmUtilityService
27
+ # Service for LLM related utility functions.
28
+ class Service
29
+
30
+ include ::GRPC::GenericService
31
+
32
+ self.marshal_class_method = :encode
33
+ self.unmarshal_class_method = :decode
34
+ self.service_name = 'google.cloud.aiplatform.v1.LlmUtilityService'
35
+
36
+ # Perform a token counting.
37
+ rpc :CountTokens, ::Google::Cloud::AIPlatform::V1::CountTokensRequest, ::Google::Cloud::AIPlatform::V1::CountTokensResponse
38
+ # Return a list of tokens based on the input text.
39
+ rpc :ComputeTokens, ::Google::Cloud::AIPlatform::V1::ComputeTokensRequest, ::Google::Cloud::AIPlatform::V1::ComputeTokensResponse
40
+ end
41
+
42
+ Stub = Service.rpc_stub_class
43
+ end
44
+ end
45
+ end
46
+ end
47
+ end
@@ -8,7 +8,7 @@ require 'google/api/field_behavior_pb'
8
8
  require 'google/cloud/aiplatform/v1/accelerator_type_pb'
9
9
 
10
10
 
11
- descriptor_data = "\n2google/cloud/aiplatform/v1/machine_resources.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a\x31google/cloud/aiplatform/v1/accelerator_type.proto\"\x8f\x01\n\x0bMachineSpec\x12\x19\n\x0cmachine_type\x18\x01 \x01(\tB\x03\xe0\x41\x05\x12J\n\x10\x61\x63\x63\x65lerator_type\x18\x02 \x01(\x0e\x32+.google.cloud.aiplatform.v1.AcceleratorTypeB\x03\xe0\x41\x05\x12\x19\n\x11\x61\x63\x63\x65lerator_count\x18\x03 \x01(\x05\"\xf8\x01\n\x12\x44\x65\x64icatedResources\x12\x45\n\x0cmachine_spec\x18\x01 \x01(\x0b\x32\'.google.cloud.aiplatform.v1.MachineSpecB\x06\xe0\x41\x02\xe0\x41\x05\x12!\n\x11min_replica_count\x18\x02 \x01(\x05\x42\x06\xe0\x41\x02\xe0\x41\x05\x12\x1e\n\x11max_replica_count\x18\x03 \x01(\x05\x42\x03\xe0\x41\x05\x12X\n\x18\x61utoscaling_metric_specs\x18\x04 \x03(\x0b\x32\x31.google.cloud.aiplatform.v1.AutoscalingMetricSpecB\x03\xe0\x41\x05\"T\n\x12\x41utomaticResources\x12\x1e\n\x11min_replica_count\x18\x01 \x01(\x05\x42\x03\xe0\x41\x05\x12\x1e\n\x11max_replica_count\x18\x02 \x01(\x05\x42\x03\xe0\x41\x05\"\xa5\x01\n\x17\x42\x61tchDedicatedResources\x12\x45\n\x0cmachine_spec\x18\x01 \x01(\x0b\x32\'.google.cloud.aiplatform.v1.MachineSpecB\x06\xe0\x41\x02\xe0\x41\x05\x12#\n\x16starting_replica_count\x18\x02 \x01(\x05\x42\x03\xe0\x41\x05\x12\x1e\n\x11max_replica_count\x18\x03 \x01(\x05\x42\x03\xe0\x41\x05\"/\n\x11ResourcesConsumed\x12\x1a\n\rreplica_hours\x18\x01 \x01(\x01\x42\x03\xe0\x41\x03\"=\n\x08\x44iskSpec\x12\x16\n\x0e\x62oot_disk_type\x18\x01 \x01(\t\x12\x19\n\x11\x62oot_disk_size_gb\x18\x02 \x01(\x05\"=\n\x12PersistentDiskSpec\x12\x11\n\tdisk_type\x18\x01 \x01(\t\x12\x14\n\x0c\x64isk_size_gb\x18\x02 \x01(\x03\"L\n\x08NfsMount\x12\x13\n\x06server\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x11\n\x04path\x18\x02 \x01(\tB\x03\xe0\x41\x02\x12\x18\n\x0bmount_point\x18\x03 \x01(\tB\x03\xe0\x41\x02\"A\n\x15\x41utoscalingMetricSpec\x12\x18\n\x0bmetric_name\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x0e\n\x06target\x18\x02 \x01(\x05\x42\xd3\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\x15MachineResourcesProtoP\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"
11
+ descriptor_data = "\n2google/cloud/aiplatform/v1/machine_resources.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a\x31google/cloud/aiplatform/v1/accelerator_type.proto\"\xaa\x01\n\x0bMachineSpec\x12\x19\n\x0cmachine_type\x18\x01 \x01(\tB\x03\xe0\x41\x05\x12J\n\x10\x61\x63\x63\x65lerator_type\x18\x02 \x01(\x0e\x32+.google.cloud.aiplatform.v1.AcceleratorTypeB\x03\xe0\x41\x05\x12\x19\n\x11\x61\x63\x63\x65lerator_count\x18\x03 \x01(\x05\x12\x19\n\x0ctpu_topology\x18\x04 \x01(\tB\x03\xe0\x41\x05\"\xf8\x01\n\x12\x44\x65\x64icatedResources\x12\x45\n\x0cmachine_spec\x18\x01 \x01(\x0b\x32\'.google.cloud.aiplatform.v1.MachineSpecB\x06\xe0\x41\x02\xe0\x41\x05\x12!\n\x11min_replica_count\x18\x02 \x01(\x05\x42\x06\xe0\x41\x02\xe0\x41\x05\x12\x1e\n\x11max_replica_count\x18\x03 \x01(\x05\x42\x03\xe0\x41\x05\x12X\n\x18\x61utoscaling_metric_specs\x18\x04 \x03(\x0b\x32\x31.google.cloud.aiplatform.v1.AutoscalingMetricSpecB\x03\xe0\x41\x05\"T\n\x12\x41utomaticResources\x12\x1e\n\x11min_replica_count\x18\x01 \x01(\x05\x42\x03\xe0\x41\x05\x12\x1e\n\x11max_replica_count\x18\x02 \x01(\x05\x42\x03\xe0\x41\x05\"\xa5\x01\n\x17\x42\x61tchDedicatedResources\x12\x45\n\x0cmachine_spec\x18\x01 \x01(\x0b\x32\'.google.cloud.aiplatform.v1.MachineSpecB\x06\xe0\x41\x02\xe0\x41\x05\x12#\n\x16starting_replica_count\x18\x02 \x01(\x05\x42\x03\xe0\x41\x05\x12\x1e\n\x11max_replica_count\x18\x03 \x01(\x05\x42\x03\xe0\x41\x05\"/\n\x11ResourcesConsumed\x12\x1a\n\rreplica_hours\x18\x01 \x01(\x01\x42\x03\xe0\x41\x03\"=\n\x08\x44iskSpec\x12\x16\n\x0e\x62oot_disk_type\x18\x01 \x01(\t\x12\x19\n\x11\x62oot_disk_size_gb\x18\x02 \x01(\x05\"=\n\x12PersistentDiskSpec\x12\x11\n\tdisk_type\x18\x01 \x01(\t\x12\x14\n\x0c\x64isk_size_gb\x18\x02 \x01(\x03\"L\n\x08NfsMount\x12\x13\n\x06server\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x11\n\x04path\x18\x02 \x01(\tB\x03\xe0\x41\x02\x12\x18\n\x0bmount_point\x18\x03 \x01(\tB\x03\xe0\x41\x02\"A\n\x15\x41utoscalingMetricSpec\x12\x18\n\x0bmetric_name\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x0e\n\x06target\x18\x02 \x01(\x05\x42\xd3\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\x15MachineResourcesProtoP\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"
12
12
 
13
13
  pool = Google::Protobuf::DescriptorPool.generated_pool
14
14
 
@@ -10,11 +10,12 @@ require 'google/cloud/aiplatform/v1/deployed_model_ref_pb'
10
10
  require 'google/cloud/aiplatform/v1/encryption_spec_pb'
11
11
  require 'google/cloud/aiplatform/v1/env_var_pb'
12
12
  require 'google/cloud/aiplatform/v1/explanation_pb'
13
+ require 'google/protobuf/duration_pb'
13
14
  require 'google/protobuf/struct_pb'
14
15
  require 'google/protobuf/timestamp_pb'
15
16
 
16
17
 
17
- descriptor_data = "\n&google/cloud/aiplatform/v1/model.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a\x33google/cloud/aiplatform/v1/deployed_model_ref.proto\x1a\x30google/cloud/aiplatform/v1/encryption_spec.proto\x1a(google/cloud/aiplatform/v1/env_var.proto\x1a,google/cloud/aiplatform/v1/explanation.proto\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1fgoogle/protobuf/timestamp.proto\"\xdf\x10\n\x05Model\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x1a\n\nversion_id\x18\x1c \x01(\tB\x06\xe0\x41\x05\xe0\x41\x03\x12\x17\n\x0fversion_aliases\x18\x1d \x03(\t\x12<\n\x13version_create_time\x18\x1f \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12<\n\x13version_update_time\x18 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x19\n\x0c\x64isplay_name\x18\x02 \x01(\tB\x03\xe0\x41\x02\x12\x13\n\x0b\x64\x65scription\x18\x03 \x01(\t\x12\x1b\n\x13version_description\x18\x1e \x01(\t\x12\x45\n\x10predict_schemata\x18\x04 \x01(\x0b\x32+.google.cloud.aiplatform.v1.PredictSchemata\x12 \n\x13metadata_schema_uri\x18\x05 \x01(\tB\x03\xe0\x41\x05\x12-\n\x08metadata\x18\x06 \x01(\x0b\x32\x16.google.protobuf.ValueB\x03\xe0\x41\x05\x12U\n\x18supported_export_formats\x18\x14 \x03(\x0b\x32..google.cloud.aiplatform.v1.Model.ExportFormatB\x03\xe0\x41\x03\x12M\n\x11training_pipeline\x18\x07 \x01(\tB2\xe0\x41\x03\xfa\x41,\n*aiplatform.googleapis.com/TrainingPipeline\x12\x43\n\x0cpipeline_job\x18/ \x01(\tB-\xe0\x41\x01\xfa\x41\'\n%aiplatform.googleapis.com/PipelineJob\x12K\n\x0e\x63ontainer_spec\x18\t \x01(\x0b\x32..google.cloud.aiplatform.v1.ModelContainerSpecB\x03\xe0\x41\x04\x12\x19\n\x0c\x61rtifact_uri\x18\x1a \x01(\tB\x03\xe0\x41\x05\x12l\n$supported_deployment_resources_types\x18\n \x03(\x0e\x32\x39.google.cloud.aiplatform.v1.Model.DeploymentResourcesTypeB\x03\xe0\x41\x03\x12,\n\x1fsupported_input_storage_formats\x18\x0b \x03(\tB\x03\xe0\x41\x03\x12-\n supported_output_storage_formats\x18\x0c \x03(\tB\x03\xe0\x41\x03\x12\x34\n\x0b\x63reate_time\x18\r \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x34\n\x0bupdate_time\x18\x0e \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12J\n\x0f\x64\x65ployed_models\x18\x0f \x03(\x0b\x32,.google.cloud.aiplatform.v1.DeployedModelRefB\x03\xe0\x41\x03\x12\x45\n\x10\x65xplanation_spec\x18\x17 \x01(\x0b\x32+.google.cloud.aiplatform.v1.ExplanationSpec\x12\x0c\n\x04\x65tag\x18\x10 \x01(\t\x12=\n\x06labels\x18\x11 \x03(\x0b\x32-.google.cloud.aiplatform.v1.Model.LabelsEntry\x12\x43\n\x0f\x65ncryption_spec\x18\x18 \x01(\x0b\x32*.google.cloud.aiplatform.v1.EncryptionSpec\x12K\n\x11model_source_info\x18& \x01(\x0b\x32+.google.cloud.aiplatform.v1.ModelSourceInfoB\x03\xe0\x41\x03\x12U\n\x13original_model_info\x18\" \x01(\x0b\x32\x33.google.cloud.aiplatform.v1.Model.OriginalModelInfoB\x03\xe0\x41\x03\x12\x1e\n\x11metadata_artifact\x18, \x01(\tB\x03\xe0\x41\x03\x1a\xd5\x01\n\x0c\x45xportFormat\x12\x0f\n\x02id\x18\x01 \x01(\tB\x03\xe0\x41\x03\x12\x62\n\x13\x65xportable_contents\x18\x02 \x03(\x0e\x32@.google.cloud.aiplatform.v1.Model.ExportFormat.ExportableContentB\x03\xe0\x41\x03\"P\n\x11\x45xportableContent\x12\"\n\x1e\x45XPORTABLE_CONTENT_UNSPECIFIED\x10\x00\x12\x0c\n\x08\x41RTIFACT\x10\x01\x12\t\n\x05IMAGE\x10\x02\x1aK\n\x11OriginalModelInfo\x12\x36\n\x05model\x18\x01 \x01(\tB\'\xe0\x41\x03\xfa\x41!\n\x1f\x61iplatform.googleapis.com/Model\x1a-\n\x0bLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\x8c\x01\n\x17\x44\x65ploymentResourcesType\x12)\n%DEPLOYMENT_RESOURCES_TYPE_UNSPECIFIED\x10\x00\x12\x17\n\x13\x44\x45\x44ICATED_RESOURCES\x10\x01\x12\x17\n\x13\x41UTOMATIC_RESOURCES\x10\x02\x12\x14\n\x10SHARED_RESOURCES\x10\x03:\\\xea\x41Y\n\x1f\x61iplatform.googleapis.com/Model\x12\x36projects/{project}/locations/{location}/models/{model}\"(\n\x13LargeModelReference\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x02\"{\n\x0fPredictSchemata\x12 \n\x13instance_schema_uri\x18\x01 \x01(\tB\x03\xe0\x41\x05\x12\"\n\x15parameters_schema_uri\x18\x02 \x01(\tB\x03\xe0\x41\x05\x12\"\n\x15prediction_schema_uri\x18\x03 \x01(\tB\x03\xe0\x41\x05\"\xfb\x01\n\x12ModelContainerSpec\x12\x19\n\timage_uri\x18\x01 \x01(\tB\x06\xe0\x41\x02\xe0\x41\x05\x12\x14\n\x07\x63ommand\x18\x02 \x03(\tB\x03\xe0\x41\x05\x12\x11\n\x04\x61rgs\x18\x03 \x03(\tB\x03\xe0\x41\x05\x12\x34\n\x03\x65nv\x18\x04 \x03(\x0b\x32\".google.cloud.aiplatform.v1.EnvVarB\x03\xe0\x41\x05\x12\x34\n\x05ports\x18\x05 \x03(\x0b\x32 .google.cloud.aiplatform.v1.PortB\x03\xe0\x41\x05\x12\x1a\n\rpredict_route\x18\x06 \x01(\tB\x03\xe0\x41\x05\x12\x19\n\x0chealth_route\x18\x07 \x01(\tB\x03\xe0\x41\x05\"\x1e\n\x04Port\x12\x16\n\x0e\x63ontainer_port\x18\x03 \x01(\x05\"\xe6\x01\n\x0fModelSourceInfo\x12P\n\x0bsource_type\x18\x01 \x01(\x0e\x32;.google.cloud.aiplatform.v1.ModelSourceInfo.ModelSourceType\x12\x0c\n\x04\x63opy\x18\x02 \x01(\x08\"s\n\x0fModelSourceType\x12!\n\x1dMODEL_SOURCE_TYPE_UNSPECIFIED\x10\x00\x12\n\n\x06\x41UTOML\x10\x01\x12\n\n\x06\x43USTOM\x10\x02\x12\x08\n\x04\x42QML\x10\x03\x12\x10\n\x0cMODEL_GARDEN\x10\x04\x12\t\n\x05GENIE\x10\x05\x42\xc8\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\nModelProtoP\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"
18
+ descriptor_data = "\n&google/cloud/aiplatform/v1/model.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a\x33google/cloud/aiplatform/v1/deployed_model_ref.proto\x1a\x30google/cloud/aiplatform/v1/encryption_spec.proto\x1a(google/cloud/aiplatform/v1/env_var.proto\x1a,google/cloud/aiplatform/v1/explanation.proto\x1a\x1egoogle/protobuf/duration.proto\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1fgoogle/protobuf/timestamp.proto\"\xdf\x10\n\x05Model\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x1a\n\nversion_id\x18\x1c \x01(\tB\x06\xe0\x41\x05\xe0\x41\x03\x12\x17\n\x0fversion_aliases\x18\x1d \x03(\t\x12<\n\x13version_create_time\x18\x1f \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12<\n\x13version_update_time\x18 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x19\n\x0c\x64isplay_name\x18\x02 \x01(\tB\x03\xe0\x41\x02\x12\x13\n\x0b\x64\x65scription\x18\x03 \x01(\t\x12\x1b\n\x13version_description\x18\x1e \x01(\t\x12\x45\n\x10predict_schemata\x18\x04 \x01(\x0b\x32+.google.cloud.aiplatform.v1.PredictSchemata\x12 \n\x13metadata_schema_uri\x18\x05 \x01(\tB\x03\xe0\x41\x05\x12-\n\x08metadata\x18\x06 \x01(\x0b\x32\x16.google.protobuf.ValueB\x03\xe0\x41\x05\x12U\n\x18supported_export_formats\x18\x14 \x03(\x0b\x32..google.cloud.aiplatform.v1.Model.ExportFormatB\x03\xe0\x41\x03\x12M\n\x11training_pipeline\x18\x07 \x01(\tB2\xe0\x41\x03\xfa\x41,\n*aiplatform.googleapis.com/TrainingPipeline\x12\x43\n\x0cpipeline_job\x18/ \x01(\tB-\xe0\x41\x01\xfa\x41\'\n%aiplatform.googleapis.com/PipelineJob\x12K\n\x0e\x63ontainer_spec\x18\t \x01(\x0b\x32..google.cloud.aiplatform.v1.ModelContainerSpecB\x03\xe0\x41\x04\x12\x19\n\x0c\x61rtifact_uri\x18\x1a \x01(\tB\x03\xe0\x41\x05\x12l\n$supported_deployment_resources_types\x18\n \x03(\x0e\x32\x39.google.cloud.aiplatform.v1.Model.DeploymentResourcesTypeB\x03\xe0\x41\x03\x12,\n\x1fsupported_input_storage_formats\x18\x0b \x03(\tB\x03\xe0\x41\x03\x12-\n supported_output_storage_formats\x18\x0c \x03(\tB\x03\xe0\x41\x03\x12\x34\n\x0b\x63reate_time\x18\r \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x34\n\x0bupdate_time\x18\x0e \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12J\n\x0f\x64\x65ployed_models\x18\x0f \x03(\x0b\x32,.google.cloud.aiplatform.v1.DeployedModelRefB\x03\xe0\x41\x03\x12\x45\n\x10\x65xplanation_spec\x18\x17 \x01(\x0b\x32+.google.cloud.aiplatform.v1.ExplanationSpec\x12\x0c\n\x04\x65tag\x18\x10 \x01(\t\x12=\n\x06labels\x18\x11 \x03(\x0b\x32-.google.cloud.aiplatform.v1.Model.LabelsEntry\x12\x43\n\x0f\x65ncryption_spec\x18\x18 \x01(\x0b\x32*.google.cloud.aiplatform.v1.EncryptionSpec\x12K\n\x11model_source_info\x18& \x01(\x0b\x32+.google.cloud.aiplatform.v1.ModelSourceInfoB\x03\xe0\x41\x03\x12U\n\x13original_model_info\x18\" \x01(\x0b\x32\x33.google.cloud.aiplatform.v1.Model.OriginalModelInfoB\x03\xe0\x41\x03\x12\x1e\n\x11metadata_artifact\x18, \x01(\tB\x03\xe0\x41\x03\x1a\xd5\x01\n\x0c\x45xportFormat\x12\x0f\n\x02id\x18\x01 \x01(\tB\x03\xe0\x41\x03\x12\x62\n\x13\x65xportable_contents\x18\x02 \x03(\x0e\x32@.google.cloud.aiplatform.v1.Model.ExportFormat.ExportableContentB\x03\xe0\x41\x03\"P\n\x11\x45xportableContent\x12\"\n\x1e\x45XPORTABLE_CONTENT_UNSPECIFIED\x10\x00\x12\x0c\n\x08\x41RTIFACT\x10\x01\x12\t\n\x05IMAGE\x10\x02\x1aK\n\x11OriginalModelInfo\x12\x36\n\x05model\x18\x01 \x01(\tB\'\xe0\x41\x03\xfa\x41!\n\x1f\x61iplatform.googleapis.com/Model\x1a-\n\x0bLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\x8c\x01\n\x17\x44\x65ploymentResourcesType\x12)\n%DEPLOYMENT_RESOURCES_TYPE_UNSPECIFIED\x10\x00\x12\x17\n\x13\x44\x45\x44ICATED_RESOURCES\x10\x01\x12\x17\n\x13\x41UTOMATIC_RESOURCES\x10\x02\x12\x14\n\x10SHARED_RESOURCES\x10\x03:\\\xea\x41Y\n\x1f\x61iplatform.googleapis.com/Model\x12\x36projects/{project}/locations/{location}/models/{model}\"(\n\x13LargeModelReference\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x02\"{\n\x0fPredictSchemata\x12 \n\x13instance_schema_uri\x18\x01 \x01(\tB\x03\xe0\x41\x05\x12\"\n\x15parameters_schema_uri\x18\x02 \x01(\tB\x03\xe0\x41\x05\x12\"\n\x15prediction_schema_uri\x18\x03 \x01(\tB\x03\xe0\x41\x05\"\x93\x04\n\x12ModelContainerSpec\x12\x19\n\timage_uri\x18\x01 \x01(\tB\x06\xe0\x41\x02\xe0\x41\x05\x12\x14\n\x07\x63ommand\x18\x02 \x03(\tB\x03\xe0\x41\x05\x12\x11\n\x04\x61rgs\x18\x03 \x03(\tB\x03\xe0\x41\x05\x12\x34\n\x03\x65nv\x18\x04 \x03(\x0b\x32\".google.cloud.aiplatform.v1.EnvVarB\x03\xe0\x41\x05\x12\x34\n\x05ports\x18\x05 \x03(\x0b\x32 .google.cloud.aiplatform.v1.PortB\x03\xe0\x41\x05\x12\x1a\n\rpredict_route\x18\x06 \x01(\tB\x03\xe0\x41\x05\x12\x19\n\x0chealth_route\x18\x07 \x01(\tB\x03\xe0\x41\x05\x12\x39\n\ngrpc_ports\x18\t \x03(\x0b\x32 .google.cloud.aiplatform.v1.PortB\x03\xe0\x41\x05\x12:\n\x12\x64\x65ployment_timeout\x18\n \x01(\x0b\x32\x19.google.protobuf.DurationB\x03\xe0\x41\x05\x12\"\n\x15shared_memory_size_mb\x18\x0b \x01(\x03\x42\x03\xe0\x41\x05\x12=\n\rstartup_probe\x18\x0c \x01(\x0b\x32!.google.cloud.aiplatform.v1.ProbeB\x03\xe0\x41\x05\x12<\n\x0chealth_probe\x18\r \x01(\x0b\x32!.google.cloud.aiplatform.v1.ProbeB\x03\xe0\x41\x05\"\x1e\n\x04Port\x12\x16\n\x0e\x63ontainer_port\x18\x03 \x01(\x05\"\xe6\x01\n\x0fModelSourceInfo\x12P\n\x0bsource_type\x18\x01 \x01(\x0e\x32;.google.cloud.aiplatform.v1.ModelSourceInfo.ModelSourceType\x12\x0c\n\x04\x63opy\x18\x02 \x01(\x08\"s\n\x0fModelSourceType\x12!\n\x1dMODEL_SOURCE_TYPE_UNSPECIFIED\x10\x00\x12\n\n\x06\x41UTOML\x10\x01\x12\n\n\x06\x43USTOM\x10\x02\x12\x08\n\x04\x42QML\x10\x03\x12\x10\n\x0cMODEL_GARDEN\x10\x04\x12\t\n\x05GENIE\x10\x05\"\xa3\x01\n\x05Probe\x12<\n\x04\x65xec\x18\x01 \x01(\x0b\x32,.google.cloud.aiplatform.v1.Probe.ExecActionH\x00\x12\x16\n\x0eperiod_seconds\x18\x02 \x01(\x05\x12\x17\n\x0ftimeout_seconds\x18\x03 \x01(\x05\x1a\x1d\n\nExecAction\x12\x0f\n\x07\x63ommand\x18\x01 \x03(\tB\x0c\n\nprobe_typeB\xc8\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\nModelProtoP\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"
18
19
 
19
20
  pool = Google::Protobuf::DescriptorPool.generated_pool
20
21
 
@@ -35,6 +36,7 @@ rescue TypeError => e
35
36
  ["google.cloud.aiplatform.v1.ExplanationSpec", "google/cloud/aiplatform/v1/explanation.proto"],
36
37
  ["google.cloud.aiplatform.v1.EncryptionSpec", "google/cloud/aiplatform/v1/encryption_spec.proto"],
37
38
  ["google.cloud.aiplatform.v1.EnvVar", "google/cloud/aiplatform/v1/env_var.proto"],
39
+ ["google.protobuf.Duration", "google/protobuf/duration.proto"],
38
40
  ]
39
41
  imports.each do |type_name, expected_filename|
40
42
  import_file = pool.lookup(type_name).file_descriptor
@@ -61,6 +63,8 @@ module Google
61
63
  Port = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.Port").msgclass
62
64
  ModelSourceInfo = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ModelSourceInfo").msgclass
63
65
  ModelSourceInfo::ModelSourceType = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ModelSourceInfo.ModelSourceType").enummodule
66
+ Probe = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.Probe").msgclass
67
+ Probe::ExecAction = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.Probe.ExecAction").msgclass
64
68
  end
65
69
  end
66
70
  end
@@ -14,7 +14,7 @@ require 'google/cloud/aiplatform/v1/types_pb'
14
14
  require 'google/protobuf/struct_pb'
15
15
 
16
16
 
17
- 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/explanation.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\"\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\"\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.Value2\x8f\t\n\x11PredictionService\x12\x94\x02\n\x07Predict\x12*.google.cloud.aiplatform.v1.PredictRequest\x1a+.google.cloud.aiplatform.v1.PredictResponse\"\xaf\x01\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*\xda\x41\x1d\x65ndpoint,instances,parameters\x12\xfe\x01\n\nRawPredict\x12-.google.cloud.aiplatform.v1.RawPredictRequest\x1a\x14.google.api.HttpBody\"\xaa\x01\x82\xd3\xe4\x93\x02\x8e\x01\"</v1/{endpoint=projects/*/locations/*/endpoints/*}:rawPredict:\x01*ZK\"F/v1/{endpoint=projects/*/locations/*/publishers/*/models/*}:rawPredict:\x01*\xda\x41\x12\x65ndpoint,http_body\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\xda\x01\n\x07\x45xplain\x12*.google.cloud.aiplatform.v1.ExplainRequest\x1a+.google.cloud.aiplatform.v1.ExplainResponse\"v\x82\xd3\xe4\x93\x02>\"9/v1/{endpoint=projects/*/locations/*/endpoints/*}:explain:\x01*\xda\x41/endpoint,instances,parameters,deployed_model_id\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"
17
+ 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/explanation.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\"\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\"\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\"\x82\x01\n\x12\x43ountTokensRequest\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\"N\n\x13\x43ountTokensResponse\x12\x14\n\x0ctotal_tokens\x18\x01 \x01(\x05\x12!\n\x19total_billable_characters\x18\x02 \x01(\x05\x32\xb6\x0e\n\x11PredictionService\x12\x94\x02\n\x07Predict\x12*.google.cloud.aiplatform.v1.PredictRequest\x1a+.google.cloud.aiplatform.v1.PredictResponse\"\xaf\x01\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*\xda\x41\x1d\x65ndpoint,instances,parameters\x12\xfe\x01\n\nRawPredict\x12-.google.cloud.aiplatform.v1.RawPredictRequest\x1a\x14.google.api.HttpBody\"\xaa\x01\x82\xd3\xe4\x93\x02\x8e\x01\"</v1/{endpoint=projects/*/locations/*/endpoints/*}:rawPredict:\x01*ZK\"F/v1/{endpoint=projects/*/locations/*/publishers/*/models/*}:rawPredict:\x01*\xda\x41\x12\x65ndpoint,http_body\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\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\x82\xd3\xe4\x93\x02>\"9/v1/{endpoint=projects/*/locations/*/endpoints/*}:explain:\x01*\xda\x41/endpoint,instances,parameters,deployed_model_id\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"
18
18
 
19
19
  pool = Google::Protobuf::DescriptorPool.generated_pool
20
20
 
@@ -51,10 +51,18 @@ module Google
51
51
  PredictRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.PredictRequest").msgclass
52
52
  PredictResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.PredictResponse").msgclass
53
53
  RawPredictRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.RawPredictRequest").msgclass
54
+ DirectPredictRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.DirectPredictRequest").msgclass
55
+ DirectPredictResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.DirectPredictResponse").msgclass
56
+ DirectRawPredictRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.DirectRawPredictRequest").msgclass
57
+ DirectRawPredictResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.DirectRawPredictResponse").msgclass
54
58
  StreamingPredictRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.StreamingPredictRequest").msgclass
55
59
  StreamingPredictResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.StreamingPredictResponse").msgclass
60
+ StreamingRawPredictRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.StreamingRawPredictRequest").msgclass
61
+ StreamingRawPredictResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.StreamingRawPredictResponse").msgclass
56
62
  ExplainRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ExplainRequest").msgclass
57
63
  ExplainResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ExplainResponse").msgclass
64
+ CountTokensRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.CountTokensRequest").msgclass
65
+ CountTokensResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.CountTokensResponse").msgclass
58
66
  end
59
67
  end
60
68
  end
@@ -47,9 +47,19 @@ module Google
47
47
  # [DeployedModel][google.cloud.aiplatform.v1.DeployedModel] that served this
48
48
  # prediction.
49
49
  rpc :RawPredict, ::Google::Cloud::AIPlatform::V1::RawPredictRequest, ::Google::Api::HttpBody
50
+ # Perform an unary online prediction request for Vertex first-party products
51
+ # and frameworks.
52
+ rpc :DirectPredict, ::Google::Cloud::AIPlatform::V1::DirectPredictRequest, ::Google::Cloud::AIPlatform::V1::DirectPredictResponse
53
+ # Perform an online prediction request through gRPC.
54
+ rpc :DirectRawPredict, ::Google::Cloud::AIPlatform::V1::DirectRawPredictRequest, ::Google::Cloud::AIPlatform::V1::DirectRawPredictResponse
55
+ # Perform a streaming online prediction request for Vertex first-party
56
+ # products and frameworks.
57
+ rpc :StreamingPredict, stream(::Google::Cloud::AIPlatform::V1::StreamingPredictRequest), stream(::Google::Cloud::AIPlatform::V1::StreamingPredictResponse)
50
58
  # Perform a server-side streaming online prediction request for Vertex
51
59
  # LLM streaming.
52
60
  rpc :ServerStreamingPredict, ::Google::Cloud::AIPlatform::V1::StreamingPredictRequest, stream(::Google::Cloud::AIPlatform::V1::StreamingPredictResponse)
61
+ # Perform a streaming online prediction request through gRPC.
62
+ rpc :StreamingRawPredict, stream(::Google::Cloud::AIPlatform::V1::StreamingRawPredictRequest), stream(::Google::Cloud::AIPlatform::V1::StreamingRawPredictResponse)
53
63
  # Perform an online explanation.
54
64
  #
55
65
  # If
@@ -9,9 +9,10 @@ require 'google/api/resource_pb'
9
9
  require 'google/protobuf/duration_pb'
10
10
  require 'google/protobuf/struct_pb'
11
11
  require 'google/protobuf/timestamp_pb'
12
+ require 'google/protobuf/wrappers_pb'
12
13
 
13
14
 
14
- descriptor_data = "\n&google/cloud/aiplatform/v1/study.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a\x1egoogle/protobuf/duration.proto\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1fgoogle/protobuf/timestamp.proto\"\xae\x03\n\x05Study\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x03\x12\x19\n\x0c\x64isplay_name\x18\x02 \x01(\tB\x03\xe0\x41\x02\x12>\n\nstudy_spec\x18\x03 \x01(\x0b\x32%.google.cloud.aiplatform.v1.StudySpecB\x03\xe0\x41\x02\x12;\n\x05state\x18\x04 \x01(\x0e\x32\'.google.cloud.aiplatform.v1.Study.StateB\x03\xe0\x41\x03\x12\x34\n\x0b\x63reate_time\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x1c\n\x0finactive_reason\x18\x06 \x01(\tB\x03\xe0\x41\x03\"G\n\x05State\x12\x15\n\x11STATE_UNSPECIFIED\x10\x00\x12\n\n\x06\x41\x43TIVE\x10\x01\x12\x0c\n\x08INACTIVE\x10\x02\x12\r\n\tCOMPLETED\x10\x03:]\xea\x41Z\n\x1f\x61iplatform.googleapis.com/Study\x12\x37projects/{project}/locations/{location}/studies/{study}\"\xd0\x07\n\x05Trial\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x03\x12\x0f\n\x02id\x18\x02 \x01(\tB\x03\xe0\x41\x03\x12;\n\x05state\x18\x03 \x01(\x0e\x32\'.google.cloud.aiplatform.v1.Trial.StateB\x03\xe0\x41\x03\x12\x44\n\nparameters\x18\x04 \x03(\x0b\x32+.google.cloud.aiplatform.v1.Trial.ParameterB\x03\xe0\x41\x03\x12G\n\x11\x66inal_measurement\x18\x05 \x01(\x0b\x32\'.google.cloud.aiplatform.v1.MeasurementB\x03\xe0\x41\x03\x12\x42\n\x0cmeasurements\x18\x06 \x03(\x0b\x32\'.google.cloud.aiplatform.v1.MeasurementB\x03\xe0\x41\x03\x12\x33\n\nstart_time\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x31\n\x08\x65nd_time\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x16\n\tclient_id\x18\t \x01(\tB\x03\xe0\x41\x03\x12\x1e\n\x11infeasible_reason\x18\n \x01(\tB\x03\xe0\x41\x03\x12?\n\ncustom_job\x18\x0b \x01(\tB+\xe0\x41\x03\xfa\x41%\n#aiplatform.googleapis.com/CustomJob\x12R\n\x0fweb_access_uris\x18\x0c \x03(\x0b\x32\x34.google.cloud.aiplatform.v1.Trial.WebAccessUrisEntryB\x03\xe0\x41\x03\x1aR\n\tParameter\x12\x19\n\x0cparameter_id\x18\x01 \x01(\tB\x03\xe0\x41\x03\x12*\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueB\x03\xe0\x41\x03\x1a\x34\n\x12WebAccessUrisEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"f\n\x05State\x12\x15\n\x11STATE_UNSPECIFIED\x10\x00\x12\r\n\tREQUESTED\x10\x01\x12\n\n\x06\x41\x43TIVE\x10\x02\x12\x0c\n\x08STOPPING\x10\x03\x12\r\n\tSUCCEEDED\x10\x04\x12\x0e\n\nINFEASIBLE\x10\x05:l\xea\x41i\n\x1f\x61iplatform.googleapis.com/Trial\x12\x46projects/{project}/locations/{location}/studies/{study}/trials/{trial}\"d\n\x0cTrialContext\x12\x13\n\x0b\x64\x65scription\x18\x01 \x01(\t\x12?\n\nparameters\x18\x02 \x03(\x0b\x32+.google.cloud.aiplatform.v1.Trial.Parameter\"\xfe\x1c\n\tStudySpec\x12j\n\x19\x64\x65\x63\x61y_curve_stopping_spec\x18\x04 \x01(\x0b\x32\x45.google.cloud.aiplatform.v1.StudySpec.DecayCurveAutomatedStoppingSpecH\x00\x12k\n\x1emedian_automated_stopping_spec\x18\x05 \x01(\x0b\x32\x41.google.cloud.aiplatform.v1.StudySpec.MedianAutomatedStoppingSpecH\x00\x12k\n\x1e\x63onvex_automated_stopping_spec\x18\t \x01(\x0b\x32\x41.google.cloud.aiplatform.v1.StudySpec.ConvexAutomatedStoppingSpecH\x00\x12\x46\n\x07metrics\x18\x01 \x03(\x0b\x32\x30.google.cloud.aiplatform.v1.StudySpec.MetricSpecB\x03\xe0\x41\x02\x12L\n\nparameters\x18\x02 \x03(\x0b\x32\x33.google.cloud.aiplatform.v1.StudySpec.ParameterSpecB\x03\xe0\x41\x02\x12\x42\n\talgorithm\x18\x03 \x01(\x0e\x32/.google.cloud.aiplatform.v1.StudySpec.Algorithm\x12Q\n\x11observation_noise\x18\x06 \x01(\x0e\x32\x36.google.cloud.aiplatform.v1.StudySpec.ObservationNoise\x12\x62\n\x1ameasurement_selection_type\x18\x07 \x01(\x0e\x32>.google.cloud.aiplatform.v1.StudySpec.MeasurementSelectionType\x1a\xad\x03\n\nMetricSpec\x12\x16\n\tmetric_id\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12L\n\x04goal\x18\x02 \x01(\x0e\x32\x39.google.cloud.aiplatform.v1.StudySpec.MetricSpec.GoalTypeB\x03\xe0\x41\x02\x12_\n\rsafety_config\x18\x03 \x01(\x0b\x32\x43.google.cloud.aiplatform.v1.StudySpec.MetricSpec.SafetyMetricConfigH\x00\x88\x01\x01\x1a\x82\x01\n\x12SafetyMetricConfig\x12\x18\n\x10safety_threshold\x18\x01 \x01(\x01\x12-\n desired_min_safe_trials_fraction\x18\x02 \x01(\x01H\x00\x88\x01\x01\x42#\n!_desired_min_safe_trials_fraction\"A\n\x08GoalType\x12\x19\n\x15GOAL_TYPE_UNSPECIFIED\x10\x00\x12\x0c\n\x08MAXIMIZE\x10\x01\x12\x0c\n\x08MINIMIZE\x10\x02\x42\x10\n\x0e_safety_config\x1a\xcd\x0e\n\rParameterSpec\x12`\n\x11\x64ouble_value_spec\x18\x02 \x01(\x0b\x32\x43.google.cloud.aiplatform.v1.StudySpec.ParameterSpec.DoubleValueSpecH\x00\x12\x62\n\x12integer_value_spec\x18\x03 \x01(\x0b\x32\x44.google.cloud.aiplatform.v1.StudySpec.ParameterSpec.IntegerValueSpecH\x00\x12j\n\x16\x63\x61tegorical_value_spec\x18\x04 \x01(\x0b\x32H.google.cloud.aiplatform.v1.StudySpec.ParameterSpec.CategoricalValueSpecH\x00\x12\x64\n\x13\x64iscrete_value_spec\x18\x05 \x01(\x0b\x32\x45.google.cloud.aiplatform.v1.StudySpec.ParameterSpec.DiscreteValueSpecH\x00\x12\x19\n\x0cparameter_id\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12Q\n\nscale_type\x18\x06 \x01(\x0e\x32=.google.cloud.aiplatform.v1.StudySpec.ParameterSpec.ScaleType\x12q\n\x1b\x63onditional_parameter_specs\x18\n \x03(\x0b\x32L.google.cloud.aiplatform.v1.StudySpec.ParameterSpec.ConditionalParameterSpec\x1ao\n\x0f\x44oubleValueSpec\x12\x16\n\tmin_value\x18\x01 \x01(\x01\x42\x03\xe0\x41\x02\x12\x16\n\tmax_value\x18\x02 \x01(\x01\x42\x03\xe0\x41\x02\x12\x1a\n\rdefault_value\x18\x04 \x01(\x01H\x00\x88\x01\x01\x42\x10\n\x0e_default_value\x1ap\n\x10IntegerValueSpec\x12\x16\n\tmin_value\x18\x01 \x01(\x03\x42\x03\xe0\x41\x02\x12\x16\n\tmax_value\x18\x02 \x01(\x03\x42\x03\xe0\x41\x02\x12\x1a\n\rdefault_value\x18\x04 \x01(\x03H\x00\x88\x01\x01\x42\x10\n\x0e_default_value\x1aY\n\x14\x43\x61tegoricalValueSpec\x12\x13\n\x06values\x18\x01 \x03(\tB\x03\xe0\x41\x02\x12\x1a\n\rdefault_value\x18\x03 \x01(\tH\x00\x88\x01\x01\x42\x10\n\x0e_default_value\x1aV\n\x11\x44iscreteValueSpec\x12\x13\n\x06values\x18\x01 \x03(\x01\x42\x03\xe0\x41\x02\x12\x1a\n\rdefault_value\x18\x03 \x01(\x01H\x00\x88\x01\x01\x42\x10\n\x0e_default_value\x1a\xa4\x05\n\x18\x43onditionalParameterSpec\x12\x85\x01\n\x16parent_discrete_values\x18\x02 \x01(\x0b\x32\x63.google.cloud.aiplatform.v1.StudySpec.ParameterSpec.ConditionalParameterSpec.DiscreteValueConditionH\x00\x12{\n\x11parent_int_values\x18\x03 \x01(\x0b\x32^.google.cloud.aiplatform.v1.StudySpec.ParameterSpec.ConditionalParameterSpec.IntValueConditionH\x00\x12\x8b\x01\n\x19parent_categorical_values\x18\x04 \x01(\x0b\x32\x66.google.cloud.aiplatform.v1.StudySpec.ParameterSpec.ConditionalParameterSpec.CategoricalValueConditionH\x00\x12P\n\x0eparameter_spec\x18\x01 \x01(\x0b\x32\x33.google.cloud.aiplatform.v1.StudySpec.ParameterSpecB\x03\xe0\x41\x02\x1a-\n\x16\x44iscreteValueCondition\x12\x13\n\x06values\x18\x01 \x03(\x01\x42\x03\xe0\x41\x02\x1a(\n\x11IntValueCondition\x12\x13\n\x06values\x18\x01 \x03(\x03\x42\x03\xe0\x41\x02\x1a\x30\n\x19\x43\x61tegoricalValueCondition\x12\x13\n\x06values\x18\x01 \x03(\tB\x03\xe0\x41\x02\x42\x18\n\x16parent_value_condition\"n\n\tScaleType\x12\x1a\n\x16SCALE_TYPE_UNSPECIFIED\x10\x00\x12\x15\n\x11UNIT_LINEAR_SCALE\x10\x01\x12\x12\n\x0eUNIT_LOG_SCALE\x10\x02\x12\x1a\n\x16UNIT_REVERSE_LOG_SCALE\x10\x03\x42\x16\n\x14parameter_value_spec\x1a?\n\x1f\x44\x65\x63\x61yCurveAutomatedStoppingSpec\x12\x1c\n\x14use_elapsed_duration\x18\x01 \x01(\x08\x1a;\n\x1bMedianAutomatedStoppingSpec\x12\x1c\n\x14use_elapsed_duration\x18\x01 \x01(\x08\x1a\xf6\x01\n\x1b\x43onvexAutomatedStoppingSpec\x12\x16\n\x0emax_step_count\x18\x01 \x01(\x03\x12\x16\n\x0emin_step_count\x18\x02 \x01(\x03\x12\x1d\n\x15min_measurement_count\x18\x03 \x01(\x03\x12$\n\x1clearning_rate_parameter_name\x18\x04 \x01(\t\x12\x1c\n\x14use_elapsed_duration\x18\x05 \x01(\x08\x12&\n\x19update_all_stopped_trials\x18\x06 \x01(\x08H\x00\x88\x01\x01\x42\x1c\n\x1a_update_all_stopped_trials\"J\n\tAlgorithm\x12\x19\n\x15\x41LGORITHM_UNSPECIFIED\x10\x00\x12\x0f\n\x0bGRID_SEARCH\x10\x02\x12\x11\n\rRANDOM_SEARCH\x10\x03\"H\n\x10ObservationNoise\x12!\n\x1dOBSERVATION_NOISE_UNSPECIFIED\x10\x00\x12\x07\n\x03LOW\x10\x01\x12\x08\n\x04HIGH\x10\x02\"r\n\x18MeasurementSelectionType\x12*\n&MEASUREMENT_SELECTION_TYPE_UNSPECIFIED\x10\x00\x12\x14\n\x10LAST_MEASUREMENT\x10\x01\x12\x14\n\x10\x42\x45ST_MEASUREMENT\x10\x02\x42\x19\n\x17\x61utomated_stopping_spec\"\xdc\x01\n\x0bMeasurement\x12\x38\n\x10\x65lapsed_duration\x18\x01 \x01(\x0b\x32\x19.google.protobuf.DurationB\x03\xe0\x41\x03\x12\x17\n\nstep_count\x18\x02 \x01(\x03\x42\x03\xe0\x41\x03\x12\x44\n\x07metrics\x18\x03 \x03(\x0b\x32..google.cloud.aiplatform.v1.Measurement.MetricB\x03\xe0\x41\x03\x1a\x34\n\x06Metric\x12\x16\n\tmetric_id\x18\x01 \x01(\tB\x03\xe0\x41\x03\x12\x12\n\x05value\x18\x02 \x01(\x01\x42\x03\xe0\x41\x03\x42\xc8\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\nStudyProtoP\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"
15
+ descriptor_data = "\n&google/cloud/aiplatform/v1/study.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a\x1egoogle/protobuf/duration.proto\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x1egoogle/protobuf/wrappers.proto\"\xae\x03\n\x05Study\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x03\x12\x19\n\x0c\x64isplay_name\x18\x02 \x01(\tB\x03\xe0\x41\x02\x12>\n\nstudy_spec\x18\x03 \x01(\x0b\x32%.google.cloud.aiplatform.v1.StudySpecB\x03\xe0\x41\x02\x12;\n\x05state\x18\x04 \x01(\x0e\x32\'.google.cloud.aiplatform.v1.Study.StateB\x03\xe0\x41\x03\x12\x34\n\x0b\x63reate_time\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x1c\n\x0finactive_reason\x18\x06 \x01(\tB\x03\xe0\x41\x03\"G\n\x05State\x12\x15\n\x11STATE_UNSPECIFIED\x10\x00\x12\n\n\x06\x41\x43TIVE\x10\x01\x12\x0c\n\x08INACTIVE\x10\x02\x12\r\n\tCOMPLETED\x10\x03:]\xea\x41Z\n\x1f\x61iplatform.googleapis.com/Study\x12\x37projects/{project}/locations/{location}/studies/{study}\"\xd0\x07\n\x05Trial\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x03\x12\x0f\n\x02id\x18\x02 \x01(\tB\x03\xe0\x41\x03\x12;\n\x05state\x18\x03 \x01(\x0e\x32\'.google.cloud.aiplatform.v1.Trial.StateB\x03\xe0\x41\x03\x12\x44\n\nparameters\x18\x04 \x03(\x0b\x32+.google.cloud.aiplatform.v1.Trial.ParameterB\x03\xe0\x41\x03\x12G\n\x11\x66inal_measurement\x18\x05 \x01(\x0b\x32\'.google.cloud.aiplatform.v1.MeasurementB\x03\xe0\x41\x03\x12\x42\n\x0cmeasurements\x18\x06 \x03(\x0b\x32\'.google.cloud.aiplatform.v1.MeasurementB\x03\xe0\x41\x03\x12\x33\n\nstart_time\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x31\n\x08\x65nd_time\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x16\n\tclient_id\x18\t \x01(\tB\x03\xe0\x41\x03\x12\x1e\n\x11infeasible_reason\x18\n \x01(\tB\x03\xe0\x41\x03\x12?\n\ncustom_job\x18\x0b \x01(\tB+\xe0\x41\x03\xfa\x41%\n#aiplatform.googleapis.com/CustomJob\x12R\n\x0fweb_access_uris\x18\x0c \x03(\x0b\x32\x34.google.cloud.aiplatform.v1.Trial.WebAccessUrisEntryB\x03\xe0\x41\x03\x1aR\n\tParameter\x12\x19\n\x0cparameter_id\x18\x01 \x01(\tB\x03\xe0\x41\x03\x12*\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueB\x03\xe0\x41\x03\x1a\x34\n\x12WebAccessUrisEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"f\n\x05State\x12\x15\n\x11STATE_UNSPECIFIED\x10\x00\x12\r\n\tREQUESTED\x10\x01\x12\n\n\x06\x41\x43TIVE\x10\x02\x12\x0c\n\x08STOPPING\x10\x03\x12\r\n\tSUCCEEDED\x10\x04\x12\x0e\n\nINFEASIBLE\x10\x05:l\xea\x41i\n\x1f\x61iplatform.googleapis.com/Trial\x12\x46projects/{project}/locations/{location}/studies/{study}/trials/{trial}\"d\n\x0cTrialContext\x12\x13\n\x0b\x64\x65scription\x18\x01 \x01(\t\x12?\n\nparameters\x18\x02 \x03(\x0b\x32+.google.cloud.aiplatform.v1.Trial.Parameter\"\x86\x01\n\x13StudyTimeConstraint\x12\x31\n\x0cmax_duration\x18\x01 \x01(\x0b\x32\x19.google.protobuf.DurationH\x00\x12.\n\x08\x65nd_time\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00\x42\x0c\n\nconstraint\"\xd7!\n\tStudySpec\x12j\n\x19\x64\x65\x63\x61y_curve_stopping_spec\x18\x04 \x01(\x0b\x32\x45.google.cloud.aiplatform.v1.StudySpec.DecayCurveAutomatedStoppingSpecH\x00\x12k\n\x1emedian_automated_stopping_spec\x18\x05 \x01(\x0b\x32\x41.google.cloud.aiplatform.v1.StudySpec.MedianAutomatedStoppingSpecH\x00\x12k\n\x1e\x63onvex_automated_stopping_spec\x18\t \x01(\x0b\x32\x41.google.cloud.aiplatform.v1.StudySpec.ConvexAutomatedStoppingSpecH\x00\x12\x46\n\x07metrics\x18\x01 \x03(\x0b\x32\x30.google.cloud.aiplatform.v1.StudySpec.MetricSpecB\x03\xe0\x41\x02\x12L\n\nparameters\x18\x02 \x03(\x0b\x32\x33.google.cloud.aiplatform.v1.StudySpec.ParameterSpecB\x03\xe0\x41\x02\x12\x42\n\talgorithm\x18\x03 \x01(\x0e\x32/.google.cloud.aiplatform.v1.StudySpec.Algorithm\x12Q\n\x11observation_noise\x18\x06 \x01(\x0e\x32\x36.google.cloud.aiplatform.v1.StudySpec.ObservationNoise\x12\x62\n\x1ameasurement_selection_type\x18\x07 \x01(\x0e\x32>.google.cloud.aiplatform.v1.StudySpec.MeasurementSelectionType\x12]\n\x15study_stopping_config\x18\x0b \x01(\x0b\x32\x39.google.cloud.aiplatform.v1.StudySpec.StudyStoppingConfigH\x01\x88\x01\x01\x1a\xad\x03\n\nMetricSpec\x12\x16\n\tmetric_id\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12L\n\x04goal\x18\x02 \x01(\x0e\x32\x39.google.cloud.aiplatform.v1.StudySpec.MetricSpec.GoalTypeB\x03\xe0\x41\x02\x12_\n\rsafety_config\x18\x03 \x01(\x0b\x32\x43.google.cloud.aiplatform.v1.StudySpec.MetricSpec.SafetyMetricConfigH\x00\x88\x01\x01\x1a\x82\x01\n\x12SafetyMetricConfig\x12\x18\n\x10safety_threshold\x18\x01 \x01(\x01\x12-\n desired_min_safe_trials_fraction\x18\x02 \x01(\x01H\x00\x88\x01\x01\x42#\n!_desired_min_safe_trials_fraction\"A\n\x08GoalType\x12\x19\n\x15GOAL_TYPE_UNSPECIFIED\x10\x00\x12\x0c\n\x08MAXIMIZE\x10\x01\x12\x0c\n\x08MINIMIZE\x10\x02\x42\x10\n\x0e_safety_config\x1a\xcd\x0e\n\rParameterSpec\x12`\n\x11\x64ouble_value_spec\x18\x02 \x01(\x0b\x32\x43.google.cloud.aiplatform.v1.StudySpec.ParameterSpec.DoubleValueSpecH\x00\x12\x62\n\x12integer_value_spec\x18\x03 \x01(\x0b\x32\x44.google.cloud.aiplatform.v1.StudySpec.ParameterSpec.IntegerValueSpecH\x00\x12j\n\x16\x63\x61tegorical_value_spec\x18\x04 \x01(\x0b\x32H.google.cloud.aiplatform.v1.StudySpec.ParameterSpec.CategoricalValueSpecH\x00\x12\x64\n\x13\x64iscrete_value_spec\x18\x05 \x01(\x0b\x32\x45.google.cloud.aiplatform.v1.StudySpec.ParameterSpec.DiscreteValueSpecH\x00\x12\x19\n\x0cparameter_id\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12Q\n\nscale_type\x18\x06 \x01(\x0e\x32=.google.cloud.aiplatform.v1.StudySpec.ParameterSpec.ScaleType\x12q\n\x1b\x63onditional_parameter_specs\x18\n \x03(\x0b\x32L.google.cloud.aiplatform.v1.StudySpec.ParameterSpec.ConditionalParameterSpec\x1ao\n\x0f\x44oubleValueSpec\x12\x16\n\tmin_value\x18\x01 \x01(\x01\x42\x03\xe0\x41\x02\x12\x16\n\tmax_value\x18\x02 \x01(\x01\x42\x03\xe0\x41\x02\x12\x1a\n\rdefault_value\x18\x04 \x01(\x01H\x00\x88\x01\x01\x42\x10\n\x0e_default_value\x1ap\n\x10IntegerValueSpec\x12\x16\n\tmin_value\x18\x01 \x01(\x03\x42\x03\xe0\x41\x02\x12\x16\n\tmax_value\x18\x02 \x01(\x03\x42\x03\xe0\x41\x02\x12\x1a\n\rdefault_value\x18\x04 \x01(\x03H\x00\x88\x01\x01\x42\x10\n\x0e_default_value\x1aY\n\x14\x43\x61tegoricalValueSpec\x12\x13\n\x06values\x18\x01 \x03(\tB\x03\xe0\x41\x02\x12\x1a\n\rdefault_value\x18\x03 \x01(\tH\x00\x88\x01\x01\x42\x10\n\x0e_default_value\x1aV\n\x11\x44iscreteValueSpec\x12\x13\n\x06values\x18\x01 \x03(\x01\x42\x03\xe0\x41\x02\x12\x1a\n\rdefault_value\x18\x03 \x01(\x01H\x00\x88\x01\x01\x42\x10\n\x0e_default_value\x1a\xa4\x05\n\x18\x43onditionalParameterSpec\x12\x85\x01\n\x16parent_discrete_values\x18\x02 \x01(\x0b\x32\x63.google.cloud.aiplatform.v1.StudySpec.ParameterSpec.ConditionalParameterSpec.DiscreteValueConditionH\x00\x12{\n\x11parent_int_values\x18\x03 \x01(\x0b\x32^.google.cloud.aiplatform.v1.StudySpec.ParameterSpec.ConditionalParameterSpec.IntValueConditionH\x00\x12\x8b\x01\n\x19parent_categorical_values\x18\x04 \x01(\x0b\x32\x66.google.cloud.aiplatform.v1.StudySpec.ParameterSpec.ConditionalParameterSpec.CategoricalValueConditionH\x00\x12P\n\x0eparameter_spec\x18\x01 \x01(\x0b\x32\x33.google.cloud.aiplatform.v1.StudySpec.ParameterSpecB\x03\xe0\x41\x02\x1a-\n\x16\x44iscreteValueCondition\x12\x13\n\x06values\x18\x01 \x03(\x01\x42\x03\xe0\x41\x02\x1a(\n\x11IntValueCondition\x12\x13\n\x06values\x18\x01 \x03(\x03\x42\x03\xe0\x41\x02\x1a\x30\n\x19\x43\x61tegoricalValueCondition\x12\x13\n\x06values\x18\x01 \x03(\tB\x03\xe0\x41\x02\x42\x18\n\x16parent_value_condition\"n\n\tScaleType\x12\x1a\n\x16SCALE_TYPE_UNSPECIFIED\x10\x00\x12\x15\n\x11UNIT_LINEAR_SCALE\x10\x01\x12\x12\n\x0eUNIT_LOG_SCALE\x10\x02\x12\x1a\n\x16UNIT_REVERSE_LOG_SCALE\x10\x03\x42\x16\n\x14parameter_value_spec\x1a?\n\x1f\x44\x65\x63\x61yCurveAutomatedStoppingSpec\x12\x1c\n\x14use_elapsed_duration\x18\x01 \x01(\x08\x1a;\n\x1bMedianAutomatedStoppingSpec\x12\x1c\n\x14use_elapsed_duration\x18\x01 \x01(\x08\x1a\xf6\x01\n\x1b\x43onvexAutomatedStoppingSpec\x12\x16\n\x0emax_step_count\x18\x01 \x01(\x03\x12\x16\n\x0emin_step_count\x18\x02 \x01(\x03\x12\x1d\n\x15min_measurement_count\x18\x03 \x01(\x03\x12$\n\x1clearning_rate_parameter_name\x18\x04 \x01(\t\x12\x1c\n\x14use_elapsed_duration\x18\x05 \x01(\x08\x12&\n\x19update_all_stopped_trials\x18\x06 \x01(\x08H\x00\x88\x01\x01\x42\x1c\n\x1a_update_all_stopped_trials\x1a\xdd\x03\n\x13StudyStoppingConfig\x12\x34\n\x10should_stop_asap\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.BoolValue\x12S\n\x1aminimum_runtime_constraint\x18\x02 \x01(\x0b\x32/.google.cloud.aiplatform.v1.StudyTimeConstraint\x12S\n\x1amaximum_runtime_constraint\x18\x03 \x01(\x0b\x32/.google.cloud.aiplatform.v1.StudyTimeConstraint\x12\x33\n\x0emin_num_trials\x18\x04 \x01(\x0b\x32\x1b.google.protobuf.Int32Value\x12\x33\n\x0emax_num_trials\x18\x05 \x01(\x0b\x32\x1b.google.protobuf.Int32Value\x12?\n\x1amax_num_trials_no_progress\x18\x06 \x01(\x0b\x32\x1b.google.protobuf.Int32Value\x12;\n\x18max_duration_no_progress\x18\x07 \x01(\x0b\x32\x19.google.protobuf.Duration\"J\n\tAlgorithm\x12\x19\n\x15\x41LGORITHM_UNSPECIFIED\x10\x00\x12\x0f\n\x0bGRID_SEARCH\x10\x02\x12\x11\n\rRANDOM_SEARCH\x10\x03\"H\n\x10ObservationNoise\x12!\n\x1dOBSERVATION_NOISE_UNSPECIFIED\x10\x00\x12\x07\n\x03LOW\x10\x01\x12\x08\n\x04HIGH\x10\x02\"r\n\x18MeasurementSelectionType\x12*\n&MEASUREMENT_SELECTION_TYPE_UNSPECIFIED\x10\x00\x12\x14\n\x10LAST_MEASUREMENT\x10\x01\x12\x14\n\x10\x42\x45ST_MEASUREMENT\x10\x02\x42\x19\n\x17\x61utomated_stopping_specB\x18\n\x16_study_stopping_config\"\xdc\x01\n\x0bMeasurement\x12\x38\n\x10\x65lapsed_duration\x18\x01 \x01(\x0b\x32\x19.google.protobuf.DurationB\x03\xe0\x41\x03\x12\x17\n\nstep_count\x18\x02 \x01(\x03\x42\x03\xe0\x41\x03\x12\x44\n\x07metrics\x18\x03 \x03(\x0b\x32..google.cloud.aiplatform.v1.Measurement.MetricB\x03\xe0\x41\x03\x1a\x34\n\x06Metric\x12\x16\n\tmetric_id\x18\x01 \x01(\tB\x03\xe0\x41\x03\x12\x12\n\x05value\x18\x02 \x01(\x01\x42\x03\xe0\x41\x03\x42\xc8\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\nStudyProtoP\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"
15
16
 
16
17
  pool = Google::Protobuf::DescriptorPool.generated_pool
17
18
 
@@ -29,6 +30,7 @@ rescue TypeError => e
29
30
  ["google.protobuf.Timestamp", "google/protobuf/timestamp.proto"],
30
31
  ["google.protobuf.Value", "google/protobuf/struct.proto"],
31
32
  ["google.protobuf.Duration", "google/protobuf/duration.proto"],
33
+ ["google.protobuf.BoolValue", "google/protobuf/wrappers.proto"],
32
34
  ]
33
35
  imports.each do |type_name, expected_filename|
34
36
  import_file = pool.lookup(type_name).file_descriptor
@@ -50,6 +52,7 @@ module Google
50
52
  Trial::Parameter = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.Trial.Parameter").msgclass
51
53
  Trial::State = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.Trial.State").enummodule
52
54
  TrialContext = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.TrialContext").msgclass
55
+ StudyTimeConstraint = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.StudyTimeConstraint").msgclass
53
56
  StudySpec = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.StudySpec").msgclass
54
57
  StudySpec::MetricSpec = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.StudySpec.MetricSpec").msgclass
55
58
  StudySpec::MetricSpec::SafetyMetricConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.StudySpec.MetricSpec.SafetyMetricConfig").msgclass
@@ -67,6 +70,7 @@ module Google
67
70
  StudySpec::DecayCurveAutomatedStoppingSpec = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.StudySpec.DecayCurveAutomatedStoppingSpec").msgclass
68
71
  StudySpec::MedianAutomatedStoppingSpec = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.StudySpec.MedianAutomatedStoppingSpec").msgclass
69
72
  StudySpec::ConvexAutomatedStoppingSpec = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.StudySpec.ConvexAutomatedStoppingSpec").msgclass
73
+ StudySpec::StudyStoppingConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.StudySpec.StudyStoppingConfig").msgclass
70
74
  StudySpec::Algorithm = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.StudySpec.Algorithm").enummodule
71
75
  StudySpec::ObservationNoise = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.StudySpec.ObservationNoise").enummodule
72
76
  StudySpec::MeasurementSelectionType = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.StudySpec.MeasurementSelectionType").enummodule
@@ -304,6 +304,19 @@ module Google
304
304
  # seconds: 360 # 6 minutes
305
305
  # total_poll_timeout:
306
306
  # seconds: 54000 # 90 minutes
307
+ # @!attribute [rw] auto_populated_fields
308
+ # @return [::Array<::String>]
309
+ # List of top-level fields of the request message, that should be
310
+ # automatically populated by the client libraries based on their
311
+ # (google.api.field_info).format. Currently supported format: UUID4.
312
+ #
313
+ # Example of a YAML configuration:
314
+ #
315
+ # publishing:
316
+ # method_settings:
317
+ # - selector: google.example.v1.ExampleService.CreateExample
318
+ # auto_populated_fields:
319
+ # - request_id
307
320
  class MethodSettings
308
321
  include ::Google::Protobuf::MessageExts
309
322
  extend ::Google::Protobuf::MessageExts::ClassMethods
@@ -172,6 +172,12 @@ module Google
172
172
  # * AIP_MODEL_DIR = `<base_output_directory>/<trial_id>/model/`
173
173
  # * AIP_CHECKPOINT_DIR = `<base_output_directory>/<trial_id>/checkpoints/`
174
174
  # * AIP_TENSORBOARD_LOG_DIR = `<base_output_directory>/<trial_id>/logs/`
175
+ # @!attribute [rw] protected_artifact_location_id
176
+ # @return [::String]
177
+ # The ID of the location to store protected artifacts. e.g. us-central1.
178
+ # Populate only when the location is different than CustomJob location.
179
+ # List of supported locations:
180
+ # https://cloud.google.com/vertex-ai/docs/general/locations
175
181
  # @!attribute [rw] tensorboard
176
182
  # @return [::String]
177
183
  # Optional. The name of a Vertex AI
@@ -21,14 +21,14 @@ module Google
21
21
  module Cloud
22
22
  module AIPlatform
23
23
  module V1
24
- # Feature Metadata information that describes an attribute of an entity type.
25
- # For example, apple is an entity type, and color is a feature that describes
26
- # apple.
24
+ # Feature Metadata information.
25
+ # For example, color is a feature that describes an apple.
27
26
  # @!attribute [rw] name
28
27
  # @return [::String]
29
28
  # Immutable. Name of the Feature.
30
29
  # Format:
31
30
  # `projects/{project}/locations/{location}/featurestores/{featurestore}/entityTypes/{entity_type}/features/{feature}`
31
+ # `projects/{project}/locations/{location}/featureGroups/{feature_group}/features/{feature}`
32
32
  #
33
33
  # The last part feature is assigned by the client. The feature can be up to
34
34
  # 64 characters long and can consist only of ASCII Latin letters A-Z and a-z,
@@ -39,13 +39,16 @@ module Google
39
39
  # Description of the Feature.
40
40
  # @!attribute [rw] value_type
41
41
  # @return [::Google::Cloud::AIPlatform::V1::Feature::ValueType]
42
- # Required. Immutable. Type of Feature value.
42
+ # Immutable. Only applicable for Vertex AI Feature Store (Legacy).
43
+ # Type of Feature value.
43
44
  # @!attribute [r] create_time
44
45
  # @return [::Google::Protobuf::Timestamp]
45
- # Output only. Timestamp when this EntityType was created.
46
+ # Output only. Only applicable for Vertex AI Feature Store (Legacy).
47
+ # Timestamp when this EntityType was created.
46
48
  # @!attribute [r] update_time
47
49
  # @return [::Google::Protobuf::Timestamp]
48
- # Output only. Timestamp when this EntityType was most recently updated.
50
+ # Output only. Only applicable for Vertex AI Feature Store (Legacy).
51
+ # Timestamp when this EntityType was most recently updated.
49
52
  # @!attribute [rw] labels
50
53
  # @return [::Google::Protobuf::Map{::String => ::String}]
51
54
  # Optional. The labels with user-defined metadata to organize your Features.
@@ -65,8 +68,10 @@ module Google
65
68
  # "overwrite" update happens.
66
69
  # @!attribute [rw] disable_monitoring
67
70
  # @return [::Boolean]
68
- # Optional. If not set, use the monitoring_config defined for the EntityType
69
- # this Feature belongs to. Only Features with type
71
+ # Optional. Only applicable for Vertex AI Feature Store (Legacy).
72
+ # If not set, use the monitoring_config defined for the EntityType this
73
+ # Feature belongs to.
74
+ # Only Features with type
70
75
  # ({::Google::Cloud::AIPlatform::V1::Feature::ValueType Feature.ValueType}) BOOL,
71
76
  # STRING, DOUBLE or INT64 can enable monitoring.
72
77
  #
@@ -74,8 +79,13 @@ module Google
74
79
  # config on EntityType.
75
80
  # @!attribute [r] monitoring_stats_anomalies
76
81
  # @return [::Array<::Google::Cloud::AIPlatform::V1::Feature::MonitoringStatsAnomaly>]
77
- # Output only. The list of historical stats and anomalies with specified
78
- # objectives.
82
+ # Output only. Only applicable for Vertex AI Feature Store (Legacy).
83
+ # The list of historical stats and anomalies with specified objectives.
84
+ # @!attribute [rw] version_column_name
85
+ # @return [::String]
86
+ # Only applicable for Vertex AI Feature Store.
87
+ # The name of the BigQuery Table/View columnn hosting data for this version.
88
+ # If no value is provided, will use feature_id.
79
89
  class Feature
80
90
  include ::Google::Protobuf::MessageExts
81
91
  extend ::Google::Protobuf::MessageExts::ClassMethods
@@ -122,6 +132,7 @@ module Google
122
132
  extend ::Google::Protobuf::MessageExts::ClassMethods
123
133
  end
124
134
 
135
+ # Only applicable for Vertex AI Legacy Feature Store.
125
136
  # An enum representing the value type of a feature.
126
137
  module ValueType
127
138
  # The value type is unspecified.
@@ -0,0 +1,91 @@
1
+ # frozen_string_literal: true
2
+
3
+ # Copyright 2023 Google LLC
4
+ #
5
+ # Licensed under the Apache License, Version 2.0 (the "License");
6
+ # you may not use this file except in compliance with the License.
7
+ # You may obtain a copy of the License at
8
+ #
9
+ # https://www.apache.org/licenses/LICENSE-2.0
10
+ #
11
+ # Unless required by applicable law or agreed to in writing, software
12
+ # distributed under the License is distributed on an "AS IS" BASIS,
13
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14
+ # See the License for the specific language governing permissions and
15
+ # limitations under the License.
16
+
17
+ # Auto-generated by gapic-generator-ruby. DO NOT EDIT!
18
+
19
+
20
+ module Google
21
+ module Cloud
22
+ module AIPlatform
23
+ module V1
24
+ # Vertex AI Feature Group.
25
+ # @!attribute [rw] big_query
26
+ # @return [::Google::Cloud::AIPlatform::V1::FeatureGroup::BigQuery]
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.
31
+ # @!attribute [r] name
32
+ # @return [::String]
33
+ # Output only. Name of the FeatureGroup. Format:
34
+ # `projects/{project}/locations/{location}/featureGroups/{featureGroup}`
35
+ # @!attribute [r] create_time
36
+ # @return [::Google::Protobuf::Timestamp]
37
+ # Output only. Timestamp when this FeatureGroup was created.
38
+ # @!attribute [r] update_time
39
+ # @return [::Google::Protobuf::Timestamp]
40
+ # Output only. Timestamp when this FeatureGroup was last updated.
41
+ # @!attribute [rw] etag
42
+ # @return [::String]
43
+ # Optional. Used to perform consistent read-modify-write updates. If not set,
44
+ # a blind "overwrite" update happens.
45
+ # @!attribute [rw] labels
46
+ # @return [::Google::Protobuf::Map{::String => ::String}]
47
+ # Optional. The labels with user-defined metadata to organize your
48
+ # FeatureGroup.
49
+ #
50
+ # Label keys and values can be no longer than 64 characters
51
+ # (Unicode codepoints), can only contain lowercase letters, numeric
52
+ # characters, underscores and dashes. International characters are allowed.
53
+ #
54
+ # See https://goo.gl/xmQnxf for more information on and examples of labels.
55
+ # No more than 64 user labels can be associated with one
56
+ # FeatureGroup(System labels are excluded)." System reserved label keys
57
+ # are prefixed with "aiplatform.googleapis.com/" and are immutable.
58
+ # @!attribute [rw] description
59
+ # @return [::String]
60
+ # Optional. Description of the FeatureGroup.
61
+ class FeatureGroup
62
+ include ::Google::Protobuf::MessageExts
63
+ extend ::Google::Protobuf::MessageExts::ClassMethods
64
+
65
+ # Input source type for BigQuery Tables and Views.
66
+ # @!attribute [rw] big_query_source
67
+ # @return [::Google::Cloud::AIPlatform::V1::BigQuerySource]
68
+ # Required. Immutable. The BigQuery source URI that points to either a
69
+ # BigQuery Table or View.
70
+ # @!attribute [rw] entity_id_columns
71
+ # @return [::Array<::String>]
72
+ # Optional. Columns to construct entity_id / row keys. Currently only
73
+ # supports 1 entity_id_column. If not provided defaults to `entity_id`.
74
+ class BigQuery
75
+ include ::Google::Protobuf::MessageExts
76
+ extend ::Google::Protobuf::MessageExts::ClassMethods
77
+ end
78
+
79
+ # @!attribute [rw] key
80
+ # @return [::String]
81
+ # @!attribute [rw] value
82
+ # @return [::String]
83
+ class LabelsEntry
84
+ include ::Google::Protobuf::MessageExts
85
+ extend ::Google::Protobuf::MessageExts::ClassMethods
86
+ end
87
+ end
88
+ end
89
+ end
90
+ end
91
+ end