google-cloud-ai_platform-v1 1.15.0 → 1.16.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- checksums.yaml +4 -4
- data/lib/google/cloud/ai_platform/v1/model_garden_service/client.rb +137 -0
- data/lib/google/cloud/ai_platform/v1/model_garden_service/operations.rb +821 -0
- data/lib/google/cloud/ai_platform/v1/model_garden_service/paths.rb +36 -0
- data/lib/google/cloud/ai_platform/v1/model_garden_service/rest/client.rb +130 -0
- data/lib/google/cloud/ai_platform/v1/model_garden_service/rest/operations.rb +4398 -0
- data/lib/google/cloud/ai_platform/v1/model_garden_service/rest/service_stub.rb +62 -0
- data/lib/google/cloud/ai_platform/v1/model_garden_service/rest.rb +1 -0
- data/lib/google/cloud/ai_platform/v1/model_garden_service.rb +1 -0
- data/lib/google/cloud/ai_platform/v1/version.rb +1 -1
- data/lib/google/cloud/ai_platform/v1/vertex_rag_data_service/client.rb +198 -0
- data/lib/google/cloud/ai_platform/v1/vertex_rag_data_service/paths.rb +17 -0
- data/lib/google/cloud/ai_platform/v1/vertex_rag_data_service/rest/client.rb +184 -0
- data/lib/google/cloud/ai_platform/v1/vertex_rag_data_service/rest/service_stub.rb +123 -0
- data/lib/google/cloud/aiplatform/v1/endpoint_pb.rb +3 -1
- data/lib/google/cloud/aiplatform/v1/model_garden_service_pb.rb +14 -1
- data/lib/google/cloud/aiplatform/v1/model_garden_service_services_pb.rb +2 -0
- data/lib/google/cloud/aiplatform/v1/service_networking_pb.rb +3 -1
- data/lib/google/cloud/aiplatform/v1/vertex_rag_data_pb.rb +6 -1
- data/lib/google/cloud/aiplatform/v1/vertex_rag_data_service_pb.rb +4 -1
- data/lib/google/cloud/aiplatform/v1/vertex_rag_data_service_services_pb.rb +4 -0
- data/proto_docs/google/cloud/aiplatform/v1/endpoint.rb +31 -6
- data/proto_docs/google/cloud/aiplatform/v1/model_garden_service.rb +153 -0
- data/proto_docs/google/cloud/aiplatform/v1/service_networking.rb +61 -5
- data/proto_docs/google/cloud/aiplatform/v1/vertex_rag_data.rb +67 -0
- data/proto_docs/google/cloud/aiplatform/v1/vertex_rag_data_service.rb +35 -0
- metadata +3 -1
|
@@ -473,6 +473,86 @@ module Google
|
|
|
473
473
|
end
|
|
474
474
|
end
|
|
475
475
|
|
|
476
|
+
##
|
|
477
|
+
# Baseline implementation for the update_rag_engine_config REST call
|
|
478
|
+
#
|
|
479
|
+
# @param request_pb [::Google::Cloud::AIPlatform::V1::UpdateRagEngineConfigRequest]
|
|
480
|
+
# A request object representing the call parameters. Required.
|
|
481
|
+
# @param options [::Gapic::CallOptions]
|
|
482
|
+
# Overrides the default settings for this call, e.g, timeout, retries etc. Optional.
|
|
483
|
+
#
|
|
484
|
+
# @yield [result, operation] Access the result along with the TransportOperation object
|
|
485
|
+
# @yieldparam result [::Google::Longrunning::Operation]
|
|
486
|
+
# @yieldparam operation [::Gapic::Rest::TransportOperation]
|
|
487
|
+
#
|
|
488
|
+
# @return [::Google::Longrunning::Operation]
|
|
489
|
+
# A result object deserialized from the server's reply
|
|
490
|
+
def update_rag_engine_config request_pb, options = nil
|
|
491
|
+
raise ::ArgumentError, "request must be provided" if request_pb.nil?
|
|
492
|
+
|
|
493
|
+
verb, uri, query_string_params, body = ServiceStub.transcode_update_rag_engine_config_request request_pb
|
|
494
|
+
query_string_params = if query_string_params.any?
|
|
495
|
+
query_string_params.to_h { |p| p.split "=", 2 }
|
|
496
|
+
else
|
|
497
|
+
{}
|
|
498
|
+
end
|
|
499
|
+
|
|
500
|
+
response = @client_stub.make_http_request(
|
|
501
|
+
verb,
|
|
502
|
+
uri: uri,
|
|
503
|
+
body: body || "",
|
|
504
|
+
params: query_string_params,
|
|
505
|
+
method_name: "update_rag_engine_config",
|
|
506
|
+
options: options
|
|
507
|
+
)
|
|
508
|
+
operation = ::Gapic::Rest::TransportOperation.new response
|
|
509
|
+
result = ::Google::Longrunning::Operation.decode_json response.body, ignore_unknown_fields: true
|
|
510
|
+
catch :response do
|
|
511
|
+
yield result, operation if block_given?
|
|
512
|
+
result
|
|
513
|
+
end
|
|
514
|
+
end
|
|
515
|
+
|
|
516
|
+
##
|
|
517
|
+
# Baseline implementation for the get_rag_engine_config REST call
|
|
518
|
+
#
|
|
519
|
+
# @param request_pb [::Google::Cloud::AIPlatform::V1::GetRagEngineConfigRequest]
|
|
520
|
+
# A request object representing the call parameters. Required.
|
|
521
|
+
# @param options [::Gapic::CallOptions]
|
|
522
|
+
# Overrides the default settings for this call, e.g, timeout, retries etc. Optional.
|
|
523
|
+
#
|
|
524
|
+
# @yield [result, operation] Access the result along with the TransportOperation object
|
|
525
|
+
# @yieldparam result [::Google::Cloud::AIPlatform::V1::RagEngineConfig]
|
|
526
|
+
# @yieldparam operation [::Gapic::Rest::TransportOperation]
|
|
527
|
+
#
|
|
528
|
+
# @return [::Google::Cloud::AIPlatform::V1::RagEngineConfig]
|
|
529
|
+
# A result object deserialized from the server's reply
|
|
530
|
+
def get_rag_engine_config request_pb, options = nil
|
|
531
|
+
raise ::ArgumentError, "request must be provided" if request_pb.nil?
|
|
532
|
+
|
|
533
|
+
verb, uri, query_string_params, body = ServiceStub.transcode_get_rag_engine_config_request request_pb
|
|
534
|
+
query_string_params = if query_string_params.any?
|
|
535
|
+
query_string_params.to_h { |p| p.split "=", 2 }
|
|
536
|
+
else
|
|
537
|
+
{}
|
|
538
|
+
end
|
|
539
|
+
|
|
540
|
+
response = @client_stub.make_http_request(
|
|
541
|
+
verb,
|
|
542
|
+
uri: uri,
|
|
543
|
+
body: body || "",
|
|
544
|
+
params: query_string_params,
|
|
545
|
+
method_name: "get_rag_engine_config",
|
|
546
|
+
options: options
|
|
547
|
+
)
|
|
548
|
+
operation = ::Gapic::Rest::TransportOperation.new response
|
|
549
|
+
result = ::Google::Cloud::AIPlatform::V1::RagEngineConfig.decode_json response.body, ignore_unknown_fields: true
|
|
550
|
+
catch :response do
|
|
551
|
+
yield result, operation if block_given?
|
|
552
|
+
result
|
|
553
|
+
end
|
|
554
|
+
end
|
|
555
|
+
|
|
476
556
|
##
|
|
477
557
|
# @private
|
|
478
558
|
#
|
|
@@ -686,6 +766,49 @@ module Google
|
|
|
686
766
|
)
|
|
687
767
|
transcoder.transcode request_pb
|
|
688
768
|
end
|
|
769
|
+
|
|
770
|
+
##
|
|
771
|
+
# @private
|
|
772
|
+
#
|
|
773
|
+
# GRPC transcoding helper method for the update_rag_engine_config REST call
|
|
774
|
+
#
|
|
775
|
+
# @param request_pb [::Google::Cloud::AIPlatform::V1::UpdateRagEngineConfigRequest]
|
|
776
|
+
# A request object representing the call parameters. Required.
|
|
777
|
+
# @return [Array(String, [String, nil], Hash{String => String})]
|
|
778
|
+
# Uri, Body, Query string parameters
|
|
779
|
+
def self.transcode_update_rag_engine_config_request request_pb
|
|
780
|
+
transcoder = Gapic::Rest::GrpcTranscoder.new
|
|
781
|
+
.with_bindings(
|
|
782
|
+
uri_method: :patch,
|
|
783
|
+
uri_template: "/v1/{rag_engine_config.name}",
|
|
784
|
+
body: "rag_engine_config",
|
|
785
|
+
matches: [
|
|
786
|
+
["rag_engine_config.name", %r{^projects/[^/]+/locations/[^/]+/ragEngineConfig/?$}, false]
|
|
787
|
+
]
|
|
788
|
+
)
|
|
789
|
+
transcoder.transcode request_pb
|
|
790
|
+
end
|
|
791
|
+
|
|
792
|
+
##
|
|
793
|
+
# @private
|
|
794
|
+
#
|
|
795
|
+
# GRPC transcoding helper method for the get_rag_engine_config REST call
|
|
796
|
+
#
|
|
797
|
+
# @param request_pb [::Google::Cloud::AIPlatform::V1::GetRagEngineConfigRequest]
|
|
798
|
+
# A request object representing the call parameters. Required.
|
|
799
|
+
# @return [Array(String, [String, nil], Hash{String => String})]
|
|
800
|
+
# Uri, Body, Query string parameters
|
|
801
|
+
def self.transcode_get_rag_engine_config_request request_pb
|
|
802
|
+
transcoder = Gapic::Rest::GrpcTranscoder.new
|
|
803
|
+
.with_bindings(
|
|
804
|
+
uri_method: :get,
|
|
805
|
+
uri_template: "/v1/{name}",
|
|
806
|
+
matches: [
|
|
807
|
+
["name", %r{^projects/[^/]+/locations/[^/]+/ragEngineConfig/?$}, false]
|
|
808
|
+
]
|
|
809
|
+
)
|
|
810
|
+
transcoder.transcode request_pb
|
|
811
|
+
end
|
|
689
812
|
end
|
|
690
813
|
end
|
|
691
814
|
end
|
|
@@ -15,7 +15,7 @@ require 'google/protobuf/duration_pb'
|
|
|
15
15
|
require 'google/protobuf/timestamp_pb'
|
|
16
16
|
|
|
17
17
|
|
|
18
|
-
descriptor_data = "\n)google/cloud/aiplatform/v1/endpoint.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a\x30google/cloud/aiplatform/v1/encryption_spec.proto\x1a,google/cloud/aiplatform/v1/explanation.proto\x1a#google/cloud/aiplatform/v1/io.proto\x1a\x32google/cloud/aiplatform/v1/machine_resources.proto\x1a\x33google/cloud/aiplatform/v1/service_networking.proto\x1a\x1egoogle/protobuf/duration.proto\x1a\x1fgoogle/protobuf/timestamp.proto\"\
|
|
18
|
+
descriptor_data = "\n)google/cloud/aiplatform/v1/endpoint.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a\x30google/cloud/aiplatform/v1/encryption_spec.proto\x1a,google/cloud/aiplatform/v1/explanation.proto\x1a#google/cloud/aiplatform/v1/io.proto\x1a\x32google/cloud/aiplatform/v1/machine_resources.proto\x1a\x33google/cloud/aiplatform/v1/service_networking.proto\x1a\x1egoogle/protobuf/duration.proto\x1a\x1fgoogle/protobuf/timestamp.proto\"\xe6\x0b\n\x08\x45ndpoint\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\x12G\n\x0f\x64\x65ployed_models\x18\x04 \x03(\x0b\x32).google.cloud.aiplatform.v1.DeployedModelB\x03\xe0\x41\x03\x12M\n\rtraffic_split\x18\x05 \x03(\x0b\x32\x36.google.cloud.aiplatform.v1.Endpoint.TrafficSplitEntry\x12\x0c\n\x04\x65tag\x18\x06 \x01(\t\x12@\n\x06labels\x18\x07 \x03(\x0b\x32\x30.google.cloud.aiplatform.v1.Endpoint.LabelsEntry\x12\x34\n\x0b\x63reate_time\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x34\n\x0bupdate_time\x18\t \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x43\n\x0f\x65ncryption_spec\x18\n \x01(\x0b\x32*.google.cloud.aiplatform.v1.EncryptionSpec\x12\x37\n\x07network\x18\r \x01(\tB&\xe0\x41\x01\xfa\x41 \n\x1e\x63ompute.googleapis.com/Network\x12*\n\x1e\x65nable_private_service_connect\x18\x11 \x01(\x08\x42\x02\x18\x01\x12\x64\n\x1eprivate_service_connect_config\x18\x15 \x01(\x0b\x32\x37.google.cloud.aiplatform.v1.PrivateServiceConnectConfigB\x03\xe0\x41\x01\x12g\n\x1fmodel_deployment_monitoring_job\x18\x0e \x01(\tB>\xe0\x41\x03\xfa\x41\x38\n6aiplatform.googleapis.com/ModelDeploymentMonitoringJob\x12p\n\'predict_request_response_logging_config\x18\x12 \x01(\x0b\x32?.google.cloud.aiplatform.v1.PredictRequestResponseLoggingConfig\x12\"\n\x1a\x64\x65\x64icated_endpoint_enabled\x18\x18 \x01(\x08\x12#\n\x16\x64\x65\x64icated_endpoint_dns\x18\x19 \x01(\tB\x03\xe0\x41\x03\x12T\n\x18\x63lient_connection_config\x18\x17 \x01(\x0b\x32\x32.google.cloud.aiplatform.v1.ClientConnectionConfig\x12\x1a\n\rsatisfies_pzs\x18\x1b \x01(\x08\x42\x03\xe0\x41\x03\x12\x1a\n\rsatisfies_pzi\x18\x1c \x01(\x08\x42\x03\xe0\x41\x03\x12\x65\n\x1fgen_ai_advanced_features_config\x18\x1d \x01(\x0b\x32\x37.google.cloud.aiplatform.v1.GenAiAdvancedFeaturesConfigB\x03\xe0\x41\x01\x1a\x33\n\x11TrafficSplitEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x05:\x02\x38\x01\x1a-\n\x0bLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01:\xb5\x01\xea\x41\xb1\x01\n\"aiplatform.googleapis.com/Endpoint\x12<projects/{project}/locations/{location}/endpoints/{endpoint}\x12Mprojects/{project}/locations/{location}/publishers/{publisher}/models/{model}\"\xf2\t\n\rDeployedModel\x12M\n\x13\x64\x65\x64icated_resources\x18\x07 \x01(\x0b\x32..google.cloud.aiplatform.v1.DedicatedResourcesH\x00\x12M\n\x13\x61utomatic_resources\x18\x08 \x01(\x0b\x32..google.cloud.aiplatform.v1.AutomaticResourcesH\x00\x12Q\n\x10shared_resources\x18\x11 \x01(\tB5\xfa\x41\x32\n0aiplatform.googleapis.com/DeploymentResourcePoolH\x00\x12\x0f\n\x02id\x18\x01 \x01(\tB\x03\xe0\x41\x05\x12\x33\n\x05model\x18\x02 \x01(\tB$\xfa\x41!\n\x1f\x61iplatform.googleapis.com/Model\x12\x1d\n\x10model_version_id\x18\x12 \x01(\tB\x03\xe0\x41\x03\x12\x14\n\x0c\x64isplay_name\x18\x03 \x01(\t\x12\x34\n\x0b\x63reate_time\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x45\n\x10\x65xplanation_spec\x18\t \x01(\x0b\x32+.google.cloud.aiplatform.v1.ExplanationSpec\x12\x1c\n\x14\x64isable_explanations\x18\x13 \x01(\x08\x12\x17\n\x0fservice_account\x18\x0b \x01(\t\x12!\n\x19\x64isable_container_logging\x18\x0f \x01(\x08\x12\x1d\n\x15\x65nable_access_logging\x18\r \x01(\x08\x12L\n\x11private_endpoints\x18\x0e \x01(\x0b\x32,.google.cloud.aiplatform.v1.PrivateEndpointsB\x03\xe0\x41\x03\x12T\n\x18\x66\x61ster_deployment_config\x18\x17 \x01(\x0b\x32\x32.google.cloud.aiplatform.v1.FasterDeploymentConfig\x12\x45\n\x06status\x18\x1a \x01(\x0b\x32\x30.google.cloud.aiplatform.v1.DeployedModel.StatusB\x03\xe0\x41\x03\x12R\n\rsystem_labels\x18\x1c \x03(\x0b\x32;.google.cloud.aiplatform.v1.DeployedModel.SystemLabelsEntry\x12\x15\n\rcheckpoint_id\x18\x1d \x01(\t\x12[\n\x19speculative_decoding_spec\x18\x1e \x01(\x0b\x32\x33.google.cloud.aiplatform.v1.SpeculativeDecodingSpecB\x03\xe0\x41\x01\x1a\x7f\n\x06Status\x12\x14\n\x07message\x18\x01 \x01(\tB\x03\xe0\x41\x03\x12\x39\n\x10last_update_time\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12$\n\x17\x61vailable_replica_count\x18\x03 \x01(\x05\x42\x03\xe0\x41\x03\x1a\x33\n\x11SystemLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x42\x16\n\x14prediction_resources\"\x8f\x01\n\x10PrivateEndpoints\x12\x1d\n\x10predict_http_uri\x18\x01 \x01(\tB\x03\xe0\x41\x03\x12\x1d\n\x10\x65xplain_http_uri\x18\x02 \x01(\tB\x03\xe0\x41\x03\x12\x1c\n\x0fhealth_http_uri\x18\x03 \x01(\tB\x03\xe0\x41\x03\x12\x1f\n\x12service_attachment\x18\x04 \x01(\tB\x03\xe0\x41\x03\"\x9c\x01\n#PredictRequestResponseLoggingConfig\x12\x0f\n\x07\x65nabled\x18\x01 \x01(\x08\x12\x15\n\rsampling_rate\x18\x02 \x01(\x01\x12M\n\x14\x62igquery_destination\x18\x03 \x01(\x0b\x32/.google.cloud.aiplatform.v1.BigQueryDestination\"N\n\x16\x43lientConnectionConfig\x12\x34\n\x11inference_timeout\x18\x01 \x01(\x0b\x32\x19.google.protobuf.Duration\"5\n\x16\x46\x61sterDeploymentConfig\x12\x1b\n\x13\x66\x61st_tryout_enabled\x18\x02 \x01(\x08\"\x95\x01\n\x1bGenAiAdvancedFeaturesConfig\x12U\n\nrag_config\x18\x01 \x01(\x0b\x32\x41.google.cloud.aiplatform.v1.GenAiAdvancedFeaturesConfig.RagConfig\x1a\x1f\n\tRagConfig\x12\x12\n\nenable_rag\x18\x01 \x01(\x08\"\x99\x03\n\x17SpeculativeDecodingSpec\x12l\n\x17\x64raft_model_speculation\x18\x02 \x01(\x0b\x32I.google.cloud.aiplatform.v1.SpeculativeDecodingSpec.DraftModelSpeculationH\x00\x12\x61\n\x11ngram_speculation\x18\x03 \x01(\x0b\x32\x44.google.cloud.aiplatform.v1.SpeculativeDecodingSpec.NgramSpeculationH\x00\x12\x1f\n\x17speculative_token_count\x18\x01 \x01(\x05\x1aU\n\x15\x44raftModelSpeculation\x12<\n\x0b\x64raft_model\x18\x01 \x01(\tB\'\xe0\x41\x02\xfa\x41!\n\x1f\x61iplatform.googleapis.com/Model\x1a&\n\x10NgramSpeculation\x12\x12\n\nngram_size\x18\x01 \x01(\x05\x42\r\n\x0bspeculationB\xcb\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\rEndpointProtoP\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
19
|
|
|
20
20
|
pool = Google::Protobuf::DescriptorPool.generated_pool
|
|
21
21
|
|
|
@@ -59,6 +59,8 @@ module Google
|
|
|
59
59
|
PredictRequestResponseLoggingConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.PredictRequestResponseLoggingConfig").msgclass
|
|
60
60
|
ClientConnectionConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ClientConnectionConfig").msgclass
|
|
61
61
|
FasterDeploymentConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.FasterDeploymentConfig").msgclass
|
|
62
|
+
GenAiAdvancedFeaturesConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.GenAiAdvancedFeaturesConfig").msgclass
|
|
63
|
+
GenAiAdvancedFeaturesConfig::RagConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.GenAiAdvancedFeaturesConfig.RagConfig").msgclass
|
|
62
64
|
SpeculativeDecodingSpec = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.SpeculativeDecodingSpec").msgclass
|
|
63
65
|
SpeculativeDecodingSpec::DraftModelSpeculation = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.SpeculativeDecodingSpec.DraftModelSpeculation").msgclass
|
|
64
66
|
SpeculativeDecodingSpec::NgramSpeculation = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.SpeculativeDecodingSpec.NgramSpeculation").msgclass
|
|
@@ -8,10 +8,14 @@ require 'google/api/annotations_pb'
|
|
|
8
8
|
require 'google/api/client_pb'
|
|
9
9
|
require 'google/api/field_behavior_pb'
|
|
10
10
|
require 'google/api/resource_pb'
|
|
11
|
+
require 'google/cloud/aiplatform/v1/machine_resources_pb'
|
|
12
|
+
require 'google/cloud/aiplatform/v1/model_pb'
|
|
13
|
+
require 'google/cloud/aiplatform/v1/operation_pb'
|
|
11
14
|
require 'google/cloud/aiplatform/v1/publisher_model_pb'
|
|
15
|
+
require 'google/longrunning/operations_pb'
|
|
12
16
|
|
|
13
17
|
|
|
14
|
-
descriptor_data = "\n5google/cloud/aiplatform/v1/model_garden_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\x30google/cloud/aiplatform/v1/publisher_model.proto\"\xfe\x01\n\x18GetPublisherModelRequest\x12>\n\x04name\x18\x01 \x01(\tB0\xe0\x41\x02\xfa\x41*\n(aiplatform.googleapis.com/PublisherModel\x12\x1a\n\rlanguage_code\x18\x02 \x01(\tB\x03\xe0\x41\x01\x12\x41\n\x04view\x18\x03 \x01(\x0e\x32..google.cloud.aiplatform.v1.PublisherModelViewB\x03\xe0\x41\x01\x12\"\n\x15is_hugging_face_model\x18\x05 \x01(\x08\x42\x03\xe0\x41\x01\x12\x1f\n\x12hugging_face_token\x18\x06 \x01(\tB\x03\xe0\x41\x01*\xa1\x01\n\x12PublisherModelView\x12$\n PUBLISHER_MODEL_VIEW_UNSPECIFIED\x10\x00\x12\x1e\n\x1aPUBLISHER_MODEL_VIEW_BASIC\x10\x01\x12\x1d\n\x19PUBLISHER_MODEL_VIEW_FULL\x10\x02\x12&\n\"PUBLISHER_MODEL_VERSION_VIEW_BASIC\x10\x03\x32\
|
|
18
|
+
descriptor_data = "\n5google/cloud/aiplatform/v1/model_garden_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\x32google/cloud/aiplatform/v1/machine_resources.proto\x1a&google/cloud/aiplatform/v1/model.proto\x1a*google/cloud/aiplatform/v1/operation.proto\x1a\x30google/cloud/aiplatform/v1/publisher_model.proto\x1a#google/longrunning/operations.proto\"\xfe\x01\n\x18GetPublisherModelRequest\x12>\n\x04name\x18\x01 \x01(\tB0\xe0\x41\x02\xfa\x41*\n(aiplatform.googleapis.com/PublisherModel\x12\x1a\n\rlanguage_code\x18\x02 \x01(\tB\x03\xe0\x41\x01\x12\x41\n\x04view\x18\x03 \x01(\x0e\x32..google.cloud.aiplatform.v1.PublisherModelViewB\x03\xe0\x41\x01\x12\"\n\x15is_hugging_face_model\x18\x05 \x01(\x08\x42\x03\xe0\x41\x01\x12\x1f\n\x12hugging_face_token\x18\x06 \x01(\tB\x03\xe0\x41\x01\"\xb2\x08\n\rDeployRequest\x12M\n\x14publisher_model_name\x18\x01 \x01(\tB-\xfa\x41*\n(aiplatform.googleapis.com/PublisherModelH\x00\x12\x1f\n\x15hugging_face_model_id\x18\x02 \x01(\tH\x00\x12>\n\x0b\x64\x65stination\x18\x04 \x01(\tB)\xe0\x41\x02\xfa\x41#\n!locations.googleapis.com/Location\x12P\n\x0cmodel_config\x18\x05 \x01(\x0b\x32\x35.google.cloud.aiplatform.v1.DeployRequest.ModelConfigB\x03\xe0\x41\x01\x12V\n\x0f\x65ndpoint_config\x18\x06 \x01(\x0b\x32\x38.google.cloud.aiplatform.v1.DeployRequest.EndpointConfigB\x03\xe0\x41\x01\x12R\n\rdeploy_config\x18\x07 \x01(\x0b\x32\x36.google.cloud.aiplatform.v1.DeployRequest.DeployConfigB\x03\xe0\x41\x01\x1a\xe6\x01\n\x0bModelConfig\x12\x18\n\x0b\x61\x63\x63\x65pt_eula\x18\x01 \x01(\x08\x42\x03\xe0\x41\x01\x12&\n\x19hugging_face_access_token\x18\x02 \x01(\tB\x03\xe0\x41\x01\x12\'\n\x1ahugging_face_cache_enabled\x18\x03 \x01(\x08\x42\x03\xe0\x41\x01\x12\x1f\n\x12model_display_name\x18\x04 \x01(\tB\x03\xe0\x41\x01\x12K\n\x0e\x63ontainer_spec\x18\x05 \x01(\x0b\x32..google.cloud.aiplatform.v1.ModelContainerSpecB\x03\xe0\x41\x01\x1a]\n\x0e\x45ndpointConfig\x12\"\n\x15\x65ndpoint_display_name\x18\x01 \x01(\tB\x03\xe0\x41\x01\x12\'\n\x1a\x64\x65\x64icated_endpoint_enabled\x18\x02 \x01(\x08\x42\x03\xe0\x41\x01\x1a\x9d\x02\n\x0c\x44\x65ployConfig\x12P\n\x13\x64\x65\x64icated_resources\x18\x01 \x01(\x0b\x32..google.cloud.aiplatform.v1.DedicatedResourcesB\x03\xe0\x41\x01\x12 \n\x13\x66\x61st_tryout_enabled\x18\x02 \x01(\x08\x42\x03\xe0\x41\x01\x12\x64\n\rsystem_labels\x18\x03 \x03(\x0b\x32H.google.cloud.aiplatform.v1.DeployRequest.DeployConfig.SystemLabelsEntryB\x03\xe0\x41\x01\x1a\x33\n\x11SystemLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x42\x0b\n\tartifacts\"\xd1\x01\n\x0e\x44\x65ployResponse\x12I\n\x0fpublisher_model\x18\x01 \x01(\tB0\xe0\x41\x03\xfa\x41*\n(aiplatform.googleapis.com/PublisherModel\x12<\n\x08\x65ndpoint\x18\x02 \x01(\tB*\xe0\x41\x03\xfa\x41$\n\"aiplatform.googleapis.com/Endpoint\x12\x36\n\x05model\x18\x03 \x01(\tB\'\xe0\x41\x03\xfa\x41!\n\x1f\x61iplatform.googleapis.com/Model\"\xa8\x02\n\x17\x44\x65ployOperationMetadata\x12N\n\x10generic_metadata\x18\x01 \x01(\x0b\x32\x34.google.cloud.aiplatform.v1.GenericOperationMetadata\x12I\n\x0fpublisher_model\x18\x02 \x01(\tB0\xe0\x41\x03\xfa\x41*\n(aiplatform.googleapis.com/PublisherModel\x12>\n\x0b\x64\x65stination\x18\x03 \x01(\tB)\xe0\x41\x03\xfa\x41#\n!locations.googleapis.com/Location\x12\x1b\n\x0eproject_number\x18\x04 \x01(\x03\x42\x03\xe0\x41\x03\x12\x15\n\x08model_id\x18\x05 \x01(\tB\x03\xe0\x41\x03*\xa1\x01\n\x12PublisherModelView\x12$\n PUBLISHER_MODEL_VIEW_UNSPECIFIED\x10\x00\x12\x1e\n\x1aPUBLISHER_MODEL_VIEW_BASIC\x10\x01\x12\x1d\n\x19PUBLISHER_MODEL_VIEW_FULL\x10\x02\x12&\n\"PUBLISHER_MODEL_VERSION_VIEW_BASIC\x10\x03\x32\xc9\x03\n\x12ModelGardenService\x12\xa6\x01\n\x11GetPublisherModel\x12\x34.google.cloud.aiplatform.v1.GetPublisherModelRequest\x1a*.google.cloud.aiplatform.v1.PublisherModel\"/\xda\x41\x04name\x82\xd3\xe4\x93\x02\"\x12 /v1/{name=publishers/*/models/*}\x12\xba\x01\n\x06\x44\x65ploy\x12).google.cloud.aiplatform.v1.DeployRequest\x1a\x1d.google.longrunning.Operation\"f\xca\x41)\n\x0e\x44\x65ployResponse\x12\x17\x44\x65ployOperationMetadata\x82\xd3\xe4\x93\x02\x34\"//v1/{destination=projects/*/locations/*}:deploy:\x01*\x1aM\xca\x41\x19\x61iplatform.googleapis.com\xd2\x41.https://www.googleapis.com/auth/cloud-platformB\xd5\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\x17ModelGardenServiceProtoP\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
19
|
|
|
16
20
|
pool = Google::Protobuf::DescriptorPool.generated_pool
|
|
17
21
|
|
|
@@ -26,6 +30,9 @@ rescue TypeError
|
|
|
26
30
|
file = pool.add_serialized_file(serialized)
|
|
27
31
|
warn "Warning: Protobuf detected an import path issue while loading generated file #{__FILE__}"
|
|
28
32
|
imports = [
|
|
33
|
+
["google.cloud.aiplatform.v1.ModelContainerSpec", "google/cloud/aiplatform/v1/model.proto"],
|
|
34
|
+
["google.cloud.aiplatform.v1.DedicatedResources", "google/cloud/aiplatform/v1/machine_resources.proto"],
|
|
35
|
+
["google.cloud.aiplatform.v1.GenericOperationMetadata", "google/cloud/aiplatform/v1/operation.proto"],
|
|
29
36
|
]
|
|
30
37
|
imports.each do |type_name, expected_filename|
|
|
31
38
|
import_file = pool.lookup(type_name).file_descriptor
|
|
@@ -42,6 +49,12 @@ module Google
|
|
|
42
49
|
module AIPlatform
|
|
43
50
|
module V1
|
|
44
51
|
GetPublisherModelRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.GetPublisherModelRequest").msgclass
|
|
52
|
+
DeployRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.DeployRequest").msgclass
|
|
53
|
+
DeployRequest::ModelConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.DeployRequest.ModelConfig").msgclass
|
|
54
|
+
DeployRequest::EndpointConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.DeployRequest.EndpointConfig").msgclass
|
|
55
|
+
DeployRequest::DeployConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.DeployRequest.DeployConfig").msgclass
|
|
56
|
+
DeployResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.DeployResponse").msgclass
|
|
57
|
+
DeployOperationMetadata = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.DeployOperationMetadata").msgclass
|
|
45
58
|
PublisherModelView = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.PublisherModelView").enummodule
|
|
46
59
|
end
|
|
47
60
|
end
|
|
@@ -35,6 +35,8 @@ module Google
|
|
|
35
35
|
|
|
36
36
|
# Gets a Model Garden publisher model.
|
|
37
37
|
rpc :GetPublisherModel, ::Google::Cloud::AIPlatform::V1::GetPublisherModelRequest, ::Google::Cloud::AIPlatform::V1::PublisherModel
|
|
38
|
+
# Deploys a model to a new endpoint.
|
|
39
|
+
rpc :Deploy, ::Google::Cloud::AIPlatform::V1::DeployRequest, ::Google::Longrunning::Operation
|
|
38
40
|
end
|
|
39
41
|
|
|
40
42
|
Stub = Service.rpc_stub_class
|
|
@@ -8,7 +8,7 @@ require 'google/api/field_behavior_pb'
|
|
|
8
8
|
require 'google/api/resource_pb'
|
|
9
9
|
|
|
10
10
|
|
|
11
|
-
descriptor_data = "\n3google/cloud/aiplatform/v1/service_networking.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\"
|
|
11
|
+
descriptor_data = "\n3google/cloud/aiplatform/v1/service_networking.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\"\xdb\x01\n\x13PSCAutomationConfig\x12\x17\n\nproject_id\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x14\n\x07network\x18\x02 \x01(\tB\x03\xe0\x41\x02\x12\x17\n\nip_address\x18\x03 \x01(\tB\x03\xe0\x41\x03\x12\x1c\n\x0f\x66orwarding_rule\x18\x04 \x01(\tB\x03\xe0\x41\x03\x12\x42\n\x05state\x18\x05 \x01(\x0e\x32..google.cloud.aiplatform.v1.PSCAutomationStateB\x03\xe0\x41\x03\x12\x1a\n\rerror_message\x18\x06 \x01(\tB\x03\xe0\x41\x03\"\xdc\x01\n\x1bPrivateServiceConnectConfig\x12+\n\x1e\x65nable_private_service_connect\x18\x01 \x01(\x08\x42\x03\xe0\x41\x02\x12\x19\n\x11project_allowlist\x18\x02 \x03(\t\x12T\n\x16psc_automation_configs\x18\x03 \x03(\x0b\x32/.google.cloud.aiplatform.v1.PSCAutomationConfigB\x03\xe0\x41\x01\x12\x1f\n\x12service_attachment\x18\x05 \x01(\tB\x03\xe0\x41\x03\"S\n\x15PscAutomatedEndpoints\x12\x12\n\nproject_id\x18\x01 \x01(\t\x12\x0f\n\x07network\x18\x02 \x01(\t\x12\x15\n\rmatch_address\x18\x03 \x01(\t\"\xb2\x01\n\x12PscInterfaceConfig\x12L\n\x12network_attachment\x18\x01 \x01(\tB0\xe0\x41\x01\xfa\x41*\n(compute.googleapis.com/NetworkAttachment\x12N\n\x13\x64ns_peering_configs\x18\x02 \x03(\x0b\x32,.google.cloud.aiplatform.v1.DnsPeeringConfigB\x03\xe0\x41\x01\"a\n\x10\x44nsPeeringConfig\x12\x13\n\x06\x64omain\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x1b\n\x0etarget_project\x18\x02 \x01(\tB\x03\xe0\x41\x02\x12\x1b\n\x0etarget_network\x18\x03 \x01(\tB\x03\xe0\x41\x02*\x80\x01\n\x12PSCAutomationState\x12$\n PSC_AUTOMATION_STATE_UNSPECIFIED\x10\x00\x12#\n\x1fPSC_AUTOMATION_STATE_SUCCESSFUL\x10\x01\x12\x1f\n\x1bPSC_AUTOMATION_STATE_FAILED\x10\x02\x42\xcd\x02\n\x1e\x63om.google.cloud.aiplatform.v1B\x16ServiceNetworkingProtoP\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::V1\xea\x41v\n(compute.googleapis.com/NetworkAttachment\x12Jprojects/{project}/regions/{region}/networkAttachments/{networkattachment}b\x06proto3"
|
|
12
12
|
|
|
13
13
|
pool = Google::Protobuf::DescriptorPool.generated_pool
|
|
14
14
|
|
|
@@ -42,6 +42,8 @@ module Google
|
|
|
42
42
|
PrivateServiceConnectConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.PrivateServiceConnectConfig").msgclass
|
|
43
43
|
PscAutomatedEndpoints = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.PscAutomatedEndpoints").msgclass
|
|
44
44
|
PscInterfaceConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.PscInterfaceConfig").msgclass
|
|
45
|
+
DnsPeeringConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.DnsPeeringConfig").msgclass
|
|
46
|
+
PSCAutomationState = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.PSCAutomationState").enummodule
|
|
45
47
|
end
|
|
46
48
|
end
|
|
47
49
|
end
|
|
@@ -12,7 +12,7 @@ require 'google/cloud/aiplatform/v1/io_pb'
|
|
|
12
12
|
require 'google/protobuf/timestamp_pb'
|
|
13
13
|
|
|
14
14
|
|
|
15
|
-
descriptor_data = "\n0google/cloud/aiplatform/v1/vertex_rag_data.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a)google/cloud/aiplatform/v1/api_auth.proto\x1a\x30google/cloud/aiplatform/v1/encryption_spec.proto\x1a#google/cloud/aiplatform/v1/io.proto\x1a\x1fgoogle/protobuf/timestamp.proto\"\xcf\x02\n\x17RagEmbeddingModelConfig\x12r\n\x1avertex_prediction_endpoint\x18\x01 \x01(\x0b\x32L.google.cloud.aiplatform.v1.RagEmbeddingModelConfig.VertexPredictionEndpointH\x00\x1a\xaf\x01\n\x18VertexPredictionEndpoint\x12<\n\x08\x65ndpoint\x18\x01 \x01(\tB*\xe0\x41\x02\xfa\x41$\n\"aiplatform.googleapis.com/Endpoint\x12\x36\n\x05model\x18\x02 \x01(\tB\'\xe0\x41\x03\xfa\x41!\n\x1f\x61iplatform.googleapis.com/Model\x12\x1d\n\x10model_version_id\x18\x03 \x01(\tB\x03\xe0\x41\x03\x42\x0e\n\x0cmodel_config\"\x94\x06\n\x11RagVectorDbConfig\x12T\n\x0erag_managed_db\x18\x01 \x01(\x0b\x32:.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDbH\x00\x12J\n\x08pinecone\x18\x03 \x01(\x0b\x32\x36.google.cloud.aiplatform.v1.RagVectorDbConfig.PineconeH\x00\x12`\n\x14vertex_vector_search\x18\x06 \x01(\x0b\x32@.google.cloud.aiplatform.v1.RagVectorDbConfig.VertexVectorSearchH\x00\x12\x35\n\x08\x61pi_auth\x18\x05 \x01(\x0b\x32#.google.cloud.aiplatform.v1.ApiAuth\x12_\n\x1arag_embedding_model_config\x18\x07 \x01(\x0b\x32\x33.google.cloud.aiplatform.v1.RagEmbeddingModelConfigB\x06\xe0\x41\x01\xe0\x41\x05\x1a\xf8\x01\n\x0cRagManagedDb\x12M\n\x03knn\x18\x01 \x01(\x0b\x32>.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNNH\x00\x12M\n\x03\x61nn\x18\x02 \x01(\x0b\x32>.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANNH\x00\x1a\x05\n\x03KNN\x1a-\n\x03\x41NN\x12\x12\n\ntree_depth\x18\x01 \x01(\x05\x12\x12\n\nleaf_count\x18\x02 \x01(\x05\x42\x14\n\x12retrieval_strategy\x1a\x1e\n\x08Pinecone\x12\x12\n\nindex_name\x18\x01 \x01(\t\x1a;\n\x12VertexVectorSearch\x12\x16\n\x0eindex_endpoint\x18\x01 \x01(\t\x12\r\n\x05index\x18\x02 \x01(\tB\x0b\n\tvector_db\"\xa0\x01\n\nFileStatus\x12@\n\x05state\x18\x01 \x01(\x0e\x32,.google.cloud.aiplatform.v1.FileStatus.StateB\x03\xe0\x41\x03\x12\x19\n\x0c\x65rror_status\x18\x02 \x01(\tB\x03\xe0\x41\x03\"5\n\x05State\x12\x15\n\x11STATE_UNSPECIFIED\x10\x00\x12\n\n\x06\x41\x43TIVE\x10\x01\x12\t\n\x05\x45RROR\x10\x02\".\n\x14VertexAiSearchConfig\x12\x16\n\x0eserving_config\x18\x01 \x01(\t\"\xab\x01\n\x0c\x43orpusStatus\x12\x42\n\x05state\x18\x01 \x01(\x0e\x32..google.cloud.aiplatform.v1.CorpusStatus.StateB\x03\xe0\x41\x03\x12\x19\n\x0c\x65rror_status\x18\x02 \x01(\tB\x03\xe0\x41\x03\"<\n\x05State\x12\x0b\n\x07UNKNOWN\x10\x00\x12\x0f\n\x0bINITIALIZED\x10\x01\x12\n\n\x06\x41\x43TIVE\x10\x02\x12\t\n\x05\x45RROR\x10\x03\"\x97\x05\n\tRagCorpus\x12Q\n\x10vector_db_config\x18\t \x01(\x0b\x32-.google.cloud.aiplatform.v1.RagVectorDbConfigB\x06\xe0\x41\x01\xe0\x41\x05H\x00\x12[\n\x17vertex_ai_search_config\x18\n \x01(\x0b\x32\x30.google.cloud.aiplatform.v1.VertexAiSearchConfigB\x06\xe0\x41\x01\xe0\x41\x05H\x00\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\x18\n\x0b\x64\x65scription\x18\x03 \x01(\tB\x03\xe0\x41\x01\x12\x34\n\x0b\x63reate_time\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x34\n\x0bupdate_time\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x44\n\rcorpus_status\x18\x08 \x01(\x0b\x32(.google.cloud.aiplatform.v1.CorpusStatusB\x03\xe0\x41\x03\x12K\n\x0f\x65ncryption_spec\x18\x0c \x01(\x0b\x32*.google.cloud.aiplatform.v1.EncryptionSpecB\x06\xe0\x41\x05\xe0\x41\x01:\x80\x01\xea\x41}\n#aiplatform.googleapis.com/RagCorpus\x12?projects/{project}/locations/{location}/ragCorpora/{rag_corpus}*\nragCorpora2\tragCorpusB\x10\n\x0e\x62\x61\x63kend_config\"\xdc\x06\n\x07RagFile\x12@\n\ngcs_source\x18\x08 \x01(\x0b\x32%.google.cloud.aiplatform.v1.GcsSourceB\x03\xe0\x41\x03H\x00\x12Q\n\x13google_drive_source\x18\t \x01(\x0b\x32-.google.cloud.aiplatform.v1.GoogleDriveSourceB\x03\xe0\x41\x03H\x00\x12S\n\x14\x64irect_upload_source\x18\n \x01(\x0b\x32..google.cloud.aiplatform.v1.DirectUploadSourceB\x03\xe0\x41\x03H\x00\x12?\n\x0cslack_source\x18\x0b \x01(\x0b\x32\'.google.cloud.aiplatform.v1.SlackSourceH\x00\x12=\n\x0bjira_source\x18\x0c \x01(\x0b\x32&.google.cloud.aiplatform.v1.JiraSourceH\x00\x12L\n\x13share_point_sources\x18\x0e \x01(\x0b\x32-.google.cloud.aiplatform.v1.SharePointSourcesH\x00\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\x18\n\x0b\x64\x65scription\x18\x03 \x01(\tB\x03\xe0\x41\x01\x12\x34\n\x0b\x63reate_time\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x34\n\x0bupdate_time\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12@\n\x0b\x66ile_status\x18\r \x01(\x0b\x32&.google.cloud.aiplatform.v1.FileStatusB\x03\xe0\x41\x03:\x8f\x01\xea\x41\x8b\x01\n!aiplatform.googleapis.com/RagFile\x12Sprojects/{project}/locations/{location}/ragCorpora/{rag_corpus}/ragFiles/{rag_file}*\x08ragFiles2\x07ragFileB\x11\n\x0frag_file_source\"\xa0\x01\n\x08RagChunk\x12\x0c\n\x04text\x18\x01 \x01(\t\x12\x45\n\tpage_span\x18\x02 \x01(\x0b\x32-.google.cloud.aiplatform.v1.RagChunk.PageSpanH\x00\x88\x01\x01\x1a\x31\n\x08PageSpan\x12\x12\n\nfirst_page\x18\x01 \x01(\x05\x12\x11\n\tlast_page\x18\x02 \x01(\x05\x42\x0c\n\n_page_span\"\xd4\x01\n\x15RagFileChunkingConfig\x12\x66\n\x15\x66ixed_length_chunking\x18\x03 \x01(\x0b\x32\x45.google.cloud.aiplatform.v1.RagFileChunkingConfig.FixedLengthChunkingH\x00\x1a@\n\x13\x46ixedLengthChunking\x12\x12\n\nchunk_size\x18\x01 \x01(\x05\x12\x15\n\rchunk_overlap\x18\x02 \x01(\x05\x42\x11\n\x0f\x63hunking_config\"r\n\x1bRagFileTransformationConfig\x12S\n\x18rag_file_chunking_config\x18\x01 \x01(\x0b\x32\x31.google.cloud.aiplatform.v1.RagFileChunkingConfig\"\xfe\x02\n\x14RagFileParsingConfig\x12V\n\rlayout_parser\x18\x04 \x01(\x0b\x32=.google.cloud.aiplatform.v1.RagFileParsingConfig.LayoutParserH\x00\x12P\n\nllm_parser\x18\x05 \x01(\x0b\x32:.google.cloud.aiplatform.v1.RagFileParsingConfig.LlmParserH\x00\x1aL\n\x0cLayoutParser\x12\x16\n\x0eprocessor_name\x18\x01 \x01(\t\x12$\n\x1cmax_parsing_requests_per_min\x18\x02 \x01(\x05\x1a\x64\n\tLlmParser\x12\x12\n\nmodel_name\x18\x01 \x01(\t\x12$\n\x1cmax_parsing_requests_per_min\x18\x02 \x01(\x05\x12\x1d\n\x15\x63ustom_parsing_prompt\x18\x03 \x01(\tB\x08\n\x06parser\"v\n\x13UploadRagFileConfig\x12_\n\x1erag_file_transformation_config\x18\x03 \x01(\x0b\x32\x37.google.cloud.aiplatform.v1.RagFileTransformationConfig\"\x87\x08\n\x14ImportRagFilesConfig\x12;\n\ngcs_source\x18\x02 \x01(\x0b\x32%.google.cloud.aiplatform.v1.GcsSourceH\x00\x12L\n\x13google_drive_source\x18\x03 \x01(\x0b\x32-.google.cloud.aiplatform.v1.GoogleDriveSourceH\x00\x12?\n\x0cslack_source\x18\x06 \x01(\x0b\x32\'.google.cloud.aiplatform.v1.SlackSourceH\x00\x12=\n\x0bjira_source\x18\x07 \x01(\x0b\x32&.google.cloud.aiplatform.v1.JiraSourceH\x00\x12L\n\x13share_point_sources\x18\r \x01(\x0b\x32-.google.cloud.aiplatform.v1.SharePointSourcesH\x00\x12R\n\x18partial_failure_gcs_sink\x18\x0b \x01(\x0b\x32*.google.cloud.aiplatform.v1.GcsDestinationB\x02\x18\x01H\x01\x12\\\n\x1dpartial_failure_bigquery_sink\x18\x0c \x01(\x0b\x32/.google.cloud.aiplatform.v1.BigQueryDestinationB\x02\x18\x01H\x01\x12L\n\x16import_result_gcs_sink\x18\x0e \x01(\x0b\x32*.google.cloud.aiplatform.v1.GcsDestinationH\x02\x12V\n\x1bimport_result_bigquery_sink\x18\x0f \x01(\x0b\x32/.google.cloud.aiplatform.v1.BigQueryDestinationH\x02\x12_\n\x1erag_file_transformation_config\x18\x10 \x01(\x0b\x32\x37.google.cloud.aiplatform.v1.RagFileTransformationConfig\x12V\n\x17rag_file_parsing_config\x18\x08 \x01(\x0b\x32\x30.google.cloud.aiplatform.v1.RagFileParsingConfigB\x03\xe0\x41\x01\x12+\n\x1emax_embedding_requests_per_min\x18\x05 \x01(\x05\x42\x03\xe0\x41\x01\x12\x19\n\x11rebuild_ann_index\x18\x13 \x01(\x08\x42\x0f\n\rimport_sourceB\x16\n\x14partial_failure_sinkB\x14\n\
|
|
15
|
+
descriptor_data = "\n0google/cloud/aiplatform/v1/vertex_rag_data.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a)google/cloud/aiplatform/v1/api_auth.proto\x1a\x30google/cloud/aiplatform/v1/encryption_spec.proto\x1a#google/cloud/aiplatform/v1/io.proto\x1a\x1fgoogle/protobuf/timestamp.proto\"\xcf\x02\n\x17RagEmbeddingModelConfig\x12r\n\x1avertex_prediction_endpoint\x18\x01 \x01(\x0b\x32L.google.cloud.aiplatform.v1.RagEmbeddingModelConfig.VertexPredictionEndpointH\x00\x1a\xaf\x01\n\x18VertexPredictionEndpoint\x12<\n\x08\x65ndpoint\x18\x01 \x01(\tB*\xe0\x41\x02\xfa\x41$\n\"aiplatform.googleapis.com/Endpoint\x12\x36\n\x05model\x18\x02 \x01(\tB\'\xe0\x41\x03\xfa\x41!\n\x1f\x61iplatform.googleapis.com/Model\x12\x1d\n\x10model_version_id\x18\x03 \x01(\tB\x03\xe0\x41\x03\x42\x0e\n\x0cmodel_config\"\x94\x06\n\x11RagVectorDbConfig\x12T\n\x0erag_managed_db\x18\x01 \x01(\x0b\x32:.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDbH\x00\x12J\n\x08pinecone\x18\x03 \x01(\x0b\x32\x36.google.cloud.aiplatform.v1.RagVectorDbConfig.PineconeH\x00\x12`\n\x14vertex_vector_search\x18\x06 \x01(\x0b\x32@.google.cloud.aiplatform.v1.RagVectorDbConfig.VertexVectorSearchH\x00\x12\x35\n\x08\x61pi_auth\x18\x05 \x01(\x0b\x32#.google.cloud.aiplatform.v1.ApiAuth\x12_\n\x1arag_embedding_model_config\x18\x07 \x01(\x0b\x32\x33.google.cloud.aiplatform.v1.RagEmbeddingModelConfigB\x06\xe0\x41\x01\xe0\x41\x05\x1a\xf8\x01\n\x0cRagManagedDb\x12M\n\x03knn\x18\x01 \x01(\x0b\x32>.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNNH\x00\x12M\n\x03\x61nn\x18\x02 \x01(\x0b\x32>.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANNH\x00\x1a\x05\n\x03KNN\x1a-\n\x03\x41NN\x12\x12\n\ntree_depth\x18\x01 \x01(\x05\x12\x12\n\nleaf_count\x18\x02 \x01(\x05\x42\x14\n\x12retrieval_strategy\x1a\x1e\n\x08Pinecone\x12\x12\n\nindex_name\x18\x01 \x01(\t\x1a;\n\x12VertexVectorSearch\x12\x16\n\x0eindex_endpoint\x18\x01 \x01(\t\x12\r\n\x05index\x18\x02 \x01(\tB\x0b\n\tvector_db\"\xa0\x01\n\nFileStatus\x12@\n\x05state\x18\x01 \x01(\x0e\x32,.google.cloud.aiplatform.v1.FileStatus.StateB\x03\xe0\x41\x03\x12\x19\n\x0c\x65rror_status\x18\x02 \x01(\tB\x03\xe0\x41\x03\"5\n\x05State\x12\x15\n\x11STATE_UNSPECIFIED\x10\x00\x12\n\n\x06\x41\x43TIVE\x10\x01\x12\t\n\x05\x45RROR\x10\x02\".\n\x14VertexAiSearchConfig\x12\x16\n\x0eserving_config\x18\x01 \x01(\t\"\xab\x01\n\x0c\x43orpusStatus\x12\x42\n\x05state\x18\x01 \x01(\x0e\x32..google.cloud.aiplatform.v1.CorpusStatus.StateB\x03\xe0\x41\x03\x12\x19\n\x0c\x65rror_status\x18\x02 \x01(\tB\x03\xe0\x41\x03\"<\n\x05State\x12\x0b\n\x07UNKNOWN\x10\x00\x12\x0f\n\x0bINITIALIZED\x10\x01\x12\n\n\x06\x41\x43TIVE\x10\x02\x12\t\n\x05\x45RROR\x10\x03\"\x97\x05\n\tRagCorpus\x12Q\n\x10vector_db_config\x18\t \x01(\x0b\x32-.google.cloud.aiplatform.v1.RagVectorDbConfigB\x06\xe0\x41\x01\xe0\x41\x05H\x00\x12[\n\x17vertex_ai_search_config\x18\n \x01(\x0b\x32\x30.google.cloud.aiplatform.v1.VertexAiSearchConfigB\x06\xe0\x41\x01\xe0\x41\x05H\x00\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\x18\n\x0b\x64\x65scription\x18\x03 \x01(\tB\x03\xe0\x41\x01\x12\x34\n\x0b\x63reate_time\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x34\n\x0bupdate_time\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x44\n\rcorpus_status\x18\x08 \x01(\x0b\x32(.google.cloud.aiplatform.v1.CorpusStatusB\x03\xe0\x41\x03\x12K\n\x0f\x65ncryption_spec\x18\x0c \x01(\x0b\x32*.google.cloud.aiplatform.v1.EncryptionSpecB\x06\xe0\x41\x05\xe0\x41\x01:\x80\x01\xea\x41}\n#aiplatform.googleapis.com/RagCorpus\x12?projects/{project}/locations/{location}/ragCorpora/{rag_corpus}*\nragCorpora2\tragCorpusB\x10\n\x0e\x62\x61\x63kend_config\"\xdc\x06\n\x07RagFile\x12@\n\ngcs_source\x18\x08 \x01(\x0b\x32%.google.cloud.aiplatform.v1.GcsSourceB\x03\xe0\x41\x03H\x00\x12Q\n\x13google_drive_source\x18\t \x01(\x0b\x32-.google.cloud.aiplatform.v1.GoogleDriveSourceB\x03\xe0\x41\x03H\x00\x12S\n\x14\x64irect_upload_source\x18\n \x01(\x0b\x32..google.cloud.aiplatform.v1.DirectUploadSourceB\x03\xe0\x41\x03H\x00\x12?\n\x0cslack_source\x18\x0b \x01(\x0b\x32\'.google.cloud.aiplatform.v1.SlackSourceH\x00\x12=\n\x0bjira_source\x18\x0c \x01(\x0b\x32&.google.cloud.aiplatform.v1.JiraSourceH\x00\x12L\n\x13share_point_sources\x18\x0e \x01(\x0b\x32-.google.cloud.aiplatform.v1.SharePointSourcesH\x00\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\x18\n\x0b\x64\x65scription\x18\x03 \x01(\tB\x03\xe0\x41\x01\x12\x34\n\x0b\x63reate_time\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x34\n\x0bupdate_time\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12@\n\x0b\x66ile_status\x18\r \x01(\x0b\x32&.google.cloud.aiplatform.v1.FileStatusB\x03\xe0\x41\x03:\x8f\x01\xea\x41\x8b\x01\n!aiplatform.googleapis.com/RagFile\x12Sprojects/{project}/locations/{location}/ragCorpora/{rag_corpus}/ragFiles/{rag_file}*\x08ragFiles2\x07ragFileB\x11\n\x0frag_file_source\"\xa0\x01\n\x08RagChunk\x12\x0c\n\x04text\x18\x01 \x01(\t\x12\x45\n\tpage_span\x18\x02 \x01(\x0b\x32-.google.cloud.aiplatform.v1.RagChunk.PageSpanH\x00\x88\x01\x01\x1a\x31\n\x08PageSpan\x12\x12\n\nfirst_page\x18\x01 \x01(\x05\x12\x11\n\tlast_page\x18\x02 \x01(\x05\x42\x0c\n\n_page_span\"\xd4\x01\n\x15RagFileChunkingConfig\x12\x66\n\x15\x66ixed_length_chunking\x18\x03 \x01(\x0b\x32\x45.google.cloud.aiplatform.v1.RagFileChunkingConfig.FixedLengthChunkingH\x00\x1a@\n\x13\x46ixedLengthChunking\x12\x12\n\nchunk_size\x18\x01 \x01(\x05\x12\x15\n\rchunk_overlap\x18\x02 \x01(\x05\x42\x11\n\x0f\x63hunking_config\"r\n\x1bRagFileTransformationConfig\x12S\n\x18rag_file_chunking_config\x18\x01 \x01(\x0b\x32\x31.google.cloud.aiplatform.v1.RagFileChunkingConfig\"\xfe\x02\n\x14RagFileParsingConfig\x12V\n\rlayout_parser\x18\x04 \x01(\x0b\x32=.google.cloud.aiplatform.v1.RagFileParsingConfig.LayoutParserH\x00\x12P\n\nllm_parser\x18\x05 \x01(\x0b\x32:.google.cloud.aiplatform.v1.RagFileParsingConfig.LlmParserH\x00\x1aL\n\x0cLayoutParser\x12\x16\n\x0eprocessor_name\x18\x01 \x01(\t\x12$\n\x1cmax_parsing_requests_per_min\x18\x02 \x01(\x05\x1a\x64\n\tLlmParser\x12\x12\n\nmodel_name\x18\x01 \x01(\t\x12$\n\x1cmax_parsing_requests_per_min\x18\x02 \x01(\x05\x12\x1d\n\x15\x63ustom_parsing_prompt\x18\x03 \x01(\tB\x08\n\x06parser\"v\n\x13UploadRagFileConfig\x12_\n\x1erag_file_transformation_config\x18\x03 \x01(\x0b\x32\x37.google.cloud.aiplatform.v1.RagFileTransformationConfig\"\x87\x08\n\x14ImportRagFilesConfig\x12;\n\ngcs_source\x18\x02 \x01(\x0b\x32%.google.cloud.aiplatform.v1.GcsSourceH\x00\x12L\n\x13google_drive_source\x18\x03 \x01(\x0b\x32-.google.cloud.aiplatform.v1.GoogleDriveSourceH\x00\x12?\n\x0cslack_source\x18\x06 \x01(\x0b\x32\'.google.cloud.aiplatform.v1.SlackSourceH\x00\x12=\n\x0bjira_source\x18\x07 \x01(\x0b\x32&.google.cloud.aiplatform.v1.JiraSourceH\x00\x12L\n\x13share_point_sources\x18\r \x01(\x0b\x32-.google.cloud.aiplatform.v1.SharePointSourcesH\x00\x12R\n\x18partial_failure_gcs_sink\x18\x0b \x01(\x0b\x32*.google.cloud.aiplatform.v1.GcsDestinationB\x02\x18\x01H\x01\x12\\\n\x1dpartial_failure_bigquery_sink\x18\x0c \x01(\x0b\x32/.google.cloud.aiplatform.v1.BigQueryDestinationB\x02\x18\x01H\x01\x12L\n\x16import_result_gcs_sink\x18\x0e \x01(\x0b\x32*.google.cloud.aiplatform.v1.GcsDestinationH\x02\x12V\n\x1bimport_result_bigquery_sink\x18\x0f \x01(\x0b\x32/.google.cloud.aiplatform.v1.BigQueryDestinationH\x02\x12_\n\x1erag_file_transformation_config\x18\x10 \x01(\x0b\x32\x37.google.cloud.aiplatform.v1.RagFileTransformationConfig\x12V\n\x17rag_file_parsing_config\x18\x08 \x01(\x0b\x32\x30.google.cloud.aiplatform.v1.RagFileParsingConfigB\x03\xe0\x41\x01\x12+\n\x1emax_embedding_requests_per_min\x18\x05 \x01(\x05\x42\x03\xe0\x41\x01\x12\x19\n\x11rebuild_ann_index\x18\x13 \x01(\x08\x42\x0f\n\rimport_sourceB\x16\n\x14partial_failure_sinkB\x14\n\x12import_result_sink\"\xa7\x02\n\x12RagManagedDbConfig\x12G\n\x06scaled\x18\x04 \x01(\x0b\x32\x35.google.cloud.aiplatform.v1.RagManagedDbConfig.ScaledH\x00\x12\x45\n\x05\x62\x61sic\x18\x02 \x01(\x0b\x32\x34.google.cloud.aiplatform.v1.RagManagedDbConfig.BasicH\x00\x12U\n\runprovisioned\x18\x03 \x01(\x0b\x32<.google.cloud.aiplatform.v1.RagManagedDbConfig.UnprovisionedH\x00\x1a\x08\n\x06Scaled\x1a\x07\n\x05\x42\x61sic\x1a\x0f\n\rUnprovisionedB\x06\n\x04tier\"\x81\x02\n\x0fRagEngineConfig\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x08\x12M\n\x15rag_managed_db_config\x18\x02 \x01(\x0b\x32..google.cloud.aiplatform.v1.RagManagedDbConfig:\x8b\x01\xea\x41\x87\x01\n)aiplatform.googleapis.com/RagEngineConfig\x12\x37projects/{project}/locations/{location}/ragEngineConfig*\x10ragEngineConfigs2\x0fragEngineConfigB\xd0\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\x12VertexRagDataProtoP\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
|
|
|
@@ -71,6 +71,11 @@ module Google
|
|
|
71
71
|
RagFileParsingConfig::LlmParser = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.RagFileParsingConfig.LlmParser").msgclass
|
|
72
72
|
UploadRagFileConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.UploadRagFileConfig").msgclass
|
|
73
73
|
ImportRagFilesConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ImportRagFilesConfig").msgclass
|
|
74
|
+
RagManagedDbConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.RagManagedDbConfig").msgclass
|
|
75
|
+
RagManagedDbConfig::Scaled = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.RagManagedDbConfig.Scaled").msgclass
|
|
76
|
+
RagManagedDbConfig::Basic = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.RagManagedDbConfig.Basic").msgclass
|
|
77
|
+
RagManagedDbConfig::Unprovisioned = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.RagManagedDbConfig.Unprovisioned").msgclass
|
|
78
|
+
RagEngineConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.RagEngineConfig").msgclass
|
|
74
79
|
end
|
|
75
80
|
end
|
|
76
81
|
end
|
|
@@ -15,7 +15,7 @@ require 'google/protobuf/empty_pb'
|
|
|
15
15
|
require 'google/rpc/status_pb'
|
|
16
16
|
|
|
17
17
|
|
|
18
|
-
descriptor_data = "\n8google/cloud/aiplatform/v1/vertex_rag_data_service.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1cgoogle/api/annotations.proto\x1a\x17google/api/client.proto\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a*google/cloud/aiplatform/v1/operation.proto\x1a\x30google/cloud/aiplatform/v1/vertex_rag_data.proto\x1a#google/longrunning/operations.proto\x1a\x1bgoogle/protobuf/empty.proto\x1a\x17google/rpc/status.proto\"\x93\x01\n\x16\x43reateRagCorpusRequest\x12\x39\n\x06parent\x18\x01 \x01(\tB)\xe0\x41\x02\xfa\x41#\n!locations.googleapis.com/Location\x12>\n\nrag_corpus\x18\x02 \x01(\x0b\x32%.google.cloud.aiplatform.v1.RagCorpusB\x03\xe0\x41\x02\"P\n\x13GetRagCorpusRequest\x12\x39\n\x04name\x18\x01 \x01(\tB+\xe0\x41\x02\xfa\x41%\n#aiplatform.googleapis.com/RagCorpus\"\x83\x01\n\x15ListRagCorporaRequest\x12\x39\n\x06parent\x18\x01 \x01(\tB)\xe0\x41\x02\xfa\x41#\n!locations.googleapis.com/Location\x12\x16\n\tpage_size\x18\x02 \x01(\x05\x42\x03\xe0\x41\x01\x12\x17\n\npage_token\x18\x03 \x01(\tB\x03\xe0\x41\x01\"m\n\x16ListRagCorporaResponse\x12:\n\x0brag_corpora\x18\x01 \x03(\x0b\x32%.google.cloud.aiplatform.v1.RagCorpus\x12\x17\n\x0fnext_page_token\x18\x02 \x01(\t\"g\n\x16\x44\x65leteRagCorpusRequest\x12\x39\n\x04name\x18\x01 \x01(\tB+\xe0\x41\x02\xfa\x41%\n#aiplatform.googleapis.com/RagCorpus\x12\x12\n\x05\x66orce\x18\x02 \x01(\x08\x42\x03\xe0\x41\x01\"\xe5\x01\n\x14UploadRagFileRequest\x12;\n\x06parent\x18\x01 \x01(\tB+\xe0\x41\x02\xfa\x41%\n#aiplatform.googleapis.com/RagCorpus\x12:\n\x08rag_file\x18\x02 \x01(\x0b\x32#.google.cloud.aiplatform.v1.RagFileB\x03\xe0\x41\x02\x12T\n\x16upload_rag_file_config\x18\x05 \x01(\x0b\x32/.google.cloud.aiplatform.v1.UploadRagFileConfigB\x03\xe0\x41\x02\"\x7f\n\x15UploadRagFileResponse\x12\x37\n\x08rag_file\x18\x01 \x01(\x0b\x32#.google.cloud.aiplatform.v1.RagFileH\x00\x12#\n\x05\x65rror\x18\x04 \x01(\x0b\x32\x12.google.rpc.StatusH\x00\x42\x08\n\x06result\"\xac\x01\n\x15ImportRagFilesRequest\x12;\n\x06parent\x18\x01 \x01(\tB+\xe0\x41\x02\xfa\x41%\n#aiplatform.googleapis.com/RagCorpus\x12V\n\x17import_rag_files_config\x18\x02 \x01(\x0b\x32\x30.google.cloud.aiplatform.v1.ImportRagFilesConfigB\x03\xe0\x41\x02\"\xe3\x01\n\x16ImportRagFilesResponse\x12#\n\x19partial_failures_gcs_path\x18\x04 \x01(\tH\x00\x12)\n\x1fpartial_failures_bigquery_table\x18\x05 \x01(\tH\x00\x12 \n\x18imported_rag_files_count\x18\x01 \x01(\x03\x12\x1e\n\x16\x66\x61iled_rag_files_count\x18\x02 \x01(\x03\x12\x1f\n\x17skipped_rag_files_count\x18\x03 \x01(\x03\x42\x16\n\x14partial_failure_sink\"L\n\x11GetRagFileRequest\x12\x37\n\x04name\x18\x01 \x01(\tB)\xe0\x41\x02\xfa\x41#\n!aiplatform.googleapis.com/RagFile\"\x83\x01\n\x13ListRagFilesRequest\x12;\n\x06parent\x18\x01 \x01(\tB+\xe0\x41\x02\xfa\x41%\n#aiplatform.googleapis.com/RagCorpus\x12\x16\n\tpage_size\x18\x02 \x01(\x05\x42\x03\xe0\x41\x01\x12\x17\n\npage_token\x18\x03 \x01(\tB\x03\xe0\x41\x01\"g\n\x14ListRagFilesResponse\x12\x36\n\trag_files\x18\x01 \x03(\x0b\x32#.google.cloud.aiplatform.v1.RagFile\x12\x17\n\x0fnext_page_token\x18\x02 \x01(\t\"O\n\x14\x44\x65leteRagFileRequest\x12\x37\n\x04name\x18\x01 \x01(\tB)\xe0\x41\x02\xfa\x41#\n!aiplatform.googleapis.com/RagFile\"r\n CreateRagCorpusOperationMetadata\x12N\n\x10generic_metadata\x18\x01 \x01(\x0b\x32\x34.google.cloud.aiplatform.v1.GenericOperationMetadata\"X\n\x16UpdateRagCorpusRequest\x12>\n\nrag_corpus\x18\x01 \x01(\x0b\x32%.google.cloud.aiplatform.v1.RagCorpusB\x03\xe0\x41\x02\"r\n UpdateRagCorpusOperationMetadata\x12N\n\x10generic_metadata\x18\x01 \x01(\x0b\x32\x34.google.cloud.aiplatform.v1.GenericOperationMetadata\"\xfd\x01\n\x1fImportRagFilesOperationMetadata\x12N\n\x10generic_metadata\x18\x01 \x01(\x0b\x32\x34.google.cloud.aiplatform.v1.GenericOperationMetadata\x12\x15\n\rrag_corpus_id\x18\x02 \x01(\x03\x12V\n\x17import_rag_files_config\x18\x03 \x01(\x0b\x32\x30.google.cloud.aiplatform.v1.ImportRagFilesConfigB\x03\xe0\x41\x03\x12\x1b\n\x13progress_percentage\x18\x04 \x01(\x05\x32\
|
|
18
|
+
descriptor_data = "\n8google/cloud/aiplatform/v1/vertex_rag_data_service.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1cgoogle/api/annotations.proto\x1a\x17google/api/client.proto\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a*google/cloud/aiplatform/v1/operation.proto\x1a\x30google/cloud/aiplatform/v1/vertex_rag_data.proto\x1a#google/longrunning/operations.proto\x1a\x1bgoogle/protobuf/empty.proto\x1a\x17google/rpc/status.proto\"\x93\x01\n\x16\x43reateRagCorpusRequest\x12\x39\n\x06parent\x18\x01 \x01(\tB)\xe0\x41\x02\xfa\x41#\n!locations.googleapis.com/Location\x12>\n\nrag_corpus\x18\x02 \x01(\x0b\x32%.google.cloud.aiplatform.v1.RagCorpusB\x03\xe0\x41\x02\"P\n\x13GetRagCorpusRequest\x12\x39\n\x04name\x18\x01 \x01(\tB+\xe0\x41\x02\xfa\x41%\n#aiplatform.googleapis.com/RagCorpus\"\x83\x01\n\x15ListRagCorporaRequest\x12\x39\n\x06parent\x18\x01 \x01(\tB)\xe0\x41\x02\xfa\x41#\n!locations.googleapis.com/Location\x12\x16\n\tpage_size\x18\x02 \x01(\x05\x42\x03\xe0\x41\x01\x12\x17\n\npage_token\x18\x03 \x01(\tB\x03\xe0\x41\x01\"m\n\x16ListRagCorporaResponse\x12:\n\x0brag_corpora\x18\x01 \x03(\x0b\x32%.google.cloud.aiplatform.v1.RagCorpus\x12\x17\n\x0fnext_page_token\x18\x02 \x01(\t\"g\n\x16\x44\x65leteRagCorpusRequest\x12\x39\n\x04name\x18\x01 \x01(\tB+\xe0\x41\x02\xfa\x41%\n#aiplatform.googleapis.com/RagCorpus\x12\x12\n\x05\x66orce\x18\x02 \x01(\x08\x42\x03\xe0\x41\x01\"\xe5\x01\n\x14UploadRagFileRequest\x12;\n\x06parent\x18\x01 \x01(\tB+\xe0\x41\x02\xfa\x41%\n#aiplatform.googleapis.com/RagCorpus\x12:\n\x08rag_file\x18\x02 \x01(\x0b\x32#.google.cloud.aiplatform.v1.RagFileB\x03\xe0\x41\x02\x12T\n\x16upload_rag_file_config\x18\x05 \x01(\x0b\x32/.google.cloud.aiplatform.v1.UploadRagFileConfigB\x03\xe0\x41\x02\"\x7f\n\x15UploadRagFileResponse\x12\x37\n\x08rag_file\x18\x01 \x01(\x0b\x32#.google.cloud.aiplatform.v1.RagFileH\x00\x12#\n\x05\x65rror\x18\x04 \x01(\x0b\x32\x12.google.rpc.StatusH\x00\x42\x08\n\x06result\"\xac\x01\n\x15ImportRagFilesRequest\x12;\n\x06parent\x18\x01 \x01(\tB+\xe0\x41\x02\xfa\x41%\n#aiplatform.googleapis.com/RagCorpus\x12V\n\x17import_rag_files_config\x18\x02 \x01(\x0b\x32\x30.google.cloud.aiplatform.v1.ImportRagFilesConfigB\x03\xe0\x41\x02\"\xe3\x01\n\x16ImportRagFilesResponse\x12#\n\x19partial_failures_gcs_path\x18\x04 \x01(\tH\x00\x12)\n\x1fpartial_failures_bigquery_table\x18\x05 \x01(\tH\x00\x12 \n\x18imported_rag_files_count\x18\x01 \x01(\x03\x12\x1e\n\x16\x66\x61iled_rag_files_count\x18\x02 \x01(\x03\x12\x1f\n\x17skipped_rag_files_count\x18\x03 \x01(\x03\x42\x16\n\x14partial_failure_sink\"L\n\x11GetRagFileRequest\x12\x37\n\x04name\x18\x01 \x01(\tB)\xe0\x41\x02\xfa\x41#\n!aiplatform.googleapis.com/RagFile\"\x83\x01\n\x13ListRagFilesRequest\x12;\n\x06parent\x18\x01 \x01(\tB+\xe0\x41\x02\xfa\x41%\n#aiplatform.googleapis.com/RagCorpus\x12\x16\n\tpage_size\x18\x02 \x01(\x05\x42\x03\xe0\x41\x01\x12\x17\n\npage_token\x18\x03 \x01(\tB\x03\xe0\x41\x01\"g\n\x14ListRagFilesResponse\x12\x36\n\trag_files\x18\x01 \x03(\x0b\x32#.google.cloud.aiplatform.v1.RagFile\x12\x17\n\x0fnext_page_token\x18\x02 \x01(\t\"O\n\x14\x44\x65leteRagFileRequest\x12\x37\n\x04name\x18\x01 \x01(\tB)\xe0\x41\x02\xfa\x41#\n!aiplatform.googleapis.com/RagFile\"r\n CreateRagCorpusOperationMetadata\x12N\n\x10generic_metadata\x18\x01 \x01(\x0b\x32\x34.google.cloud.aiplatform.v1.GenericOperationMetadata\"\\\n\x19GetRagEngineConfigRequest\x12?\n\x04name\x18\x01 \x01(\tB1\xe0\x41\x02\xfa\x41+\n)aiplatform.googleapis.com/RagEngineConfig\"X\n\x16UpdateRagCorpusRequest\x12>\n\nrag_corpus\x18\x01 \x01(\x0b\x32%.google.cloud.aiplatform.v1.RagCorpusB\x03\xe0\x41\x02\"r\n UpdateRagCorpusOperationMetadata\x12N\n\x10generic_metadata\x18\x01 \x01(\x0b\x32\x34.google.cloud.aiplatform.v1.GenericOperationMetadata\"\xfd\x01\n\x1fImportRagFilesOperationMetadata\x12N\n\x10generic_metadata\x18\x01 \x01(\x0b\x32\x34.google.cloud.aiplatform.v1.GenericOperationMetadata\x12\x15\n\rrag_corpus_id\x18\x02 \x01(\x03\x12V\n\x17import_rag_files_config\x18\x03 \x01(\x0b\x32\x30.google.cloud.aiplatform.v1.ImportRagFilesConfigB\x03\xe0\x41\x03\x12\x1b\n\x13progress_percentage\x18\x04 \x01(\x05\"k\n\x1cUpdateRagEngineConfigRequest\x12K\n\x11rag_engine_config\x18\x01 \x01(\x0b\x32+.google.cloud.aiplatform.v1.RagEngineConfigB\x03\xe0\x41\x02\"x\n&UpdateRagEngineConfigOperationMetadata\x12N\n\x10generic_metadata\x18\x01 \x01(\x0b\x32\x34.google.cloud.aiplatform.v1.GenericOperationMetadata2\xb0\x15\n\x14VertexRagDataService\x12\xed\x01\n\x0f\x43reateRagCorpus\x12\x32.google.cloud.aiplatform.v1.CreateRagCorpusRequest\x1a\x1d.google.longrunning.Operation\"\x86\x01\xca\x41-\n\tRagCorpus\x12 CreateRagCorpusOperationMetadata\xda\x41\x11parent,rag_corpus\x82\xd3\xe4\x93\x02<\"./v1/{parent=projects/*/locations/*}/ragCorpora:\nrag_corpus\x12\xf1\x01\n\x0fUpdateRagCorpus\x12\x32.google.cloud.aiplatform.v1.UpdateRagCorpusRequest\x1a\x1d.google.longrunning.Operation\"\x8a\x01\xca\x41-\n\tRagCorpus\x12 UpdateRagCorpusOperationMetadata\xda\x41\nrag_corpus\x82\xd3\xe4\x93\x02G29/v1/{rag_corpus.name=projects/*/locations/*/ragCorpora/*}:\nrag_corpus\x12\xa5\x01\n\x0cGetRagCorpus\x12/.google.cloud.aiplatform.v1.GetRagCorpusRequest\x1a%.google.cloud.aiplatform.v1.RagCorpus\"=\xda\x41\x04name\x82\xd3\xe4\x93\x02\x30\x12./v1/{name=projects/*/locations/*/ragCorpora/*}\x12\xb8\x01\n\x0eListRagCorpora\x12\x31.google.cloud.aiplatform.v1.ListRagCorporaRequest\x1a\x32.google.cloud.aiplatform.v1.ListRagCorporaResponse\"?\xda\x41\x06parent\x82\xd3\xe4\x93\x02\x30\x12./v1/{parent=projects/*/locations/*}/ragCorpora\x12\xd6\x01\n\x0f\x44\x65leteRagCorpus\x12\x32.google.cloud.aiplatform.v1.DeleteRagCorpusRequest\x1a\x1d.google.longrunning.Operation\"p\xca\x41\x30\n\x15google.protobuf.Empty\x12\x17\x44\x65leteOperationMetadata\xda\x41\x04name\x82\xd3\xe4\x93\x02\x30*./v1/{name=projects/*/locations/*/ragCorpora/*}\x12\xea\x01\n\rUploadRagFile\x12\x30.google.cloud.aiplatform.v1.UploadRagFileRequest\x1a\x31.google.cloud.aiplatform.v1.UploadRagFileResponse\"t\xda\x41&parent,rag_file,upload_rag_file_config\x82\xd3\xe4\x93\x02\x45\"@/v1/{parent=projects/*/locations/*/ragCorpora/*}/ragFiles:upload:\x01*\x12\x8d\x02\n\x0eImportRagFiles\x12\x31.google.cloud.aiplatform.v1.ImportRagFilesRequest\x1a\x1d.google.longrunning.Operation\"\xa8\x01\xca\x41\x39\n\x16ImportRagFilesResponse\x12\x1fImportRagFilesOperationMetadata\xda\x41\x1eparent,import_rag_files_config\x82\xd3\xe4\x93\x02\x45\"@/v1/{parent=projects/*/locations/*/ragCorpora/*}/ragFiles:import:\x01*\x12\xaa\x01\n\nGetRagFile\x12-.google.cloud.aiplatform.v1.GetRagFileRequest\x1a#.google.cloud.aiplatform.v1.RagFile\"H\xda\x41\x04name\x82\xd3\xe4\x93\x02;\x12\x39/v1/{name=projects/*/locations/*/ragCorpora/*/ragFiles/*}\x12\xbd\x01\n\x0cListRagFiles\x12/.google.cloud.aiplatform.v1.ListRagFilesRequest\x1a\x30.google.cloud.aiplatform.v1.ListRagFilesResponse\"J\xda\x41\x06parent\x82\xd3\xe4\x93\x02;\x12\x39/v1/{parent=projects/*/locations/*/ragCorpora/*}/ragFiles\x12\xdd\x01\n\rDeleteRagFile\x12\x30.google.cloud.aiplatform.v1.DeleteRagFileRequest\x1a\x1d.google.longrunning.Operation\"{\xca\x41\x30\n\x15google.protobuf.Empty\x12\x17\x44\x65leteOperationMetadata\xda\x41\x04name\x82\xd3\xe4\x93\x02;*9/v1/{name=projects/*/locations/*/ragCorpora/*/ragFiles/*}\x12\xa1\x02\n\x15UpdateRagEngineConfig\x12\x38.google.cloud.aiplatform.v1.UpdateRagEngineConfigRequest\x1a\x1d.google.longrunning.Operation\"\xae\x01\xca\x41\x39\n\x0fRagEngineConfig\x12&UpdateRagEngineConfigOperationMetadata\xda\x41\x11rag_engine_config\x82\xd3\xe4\x93\x02X2C/v1/{rag_engine_config.name=projects/*/locations/*/ragEngineConfig}:\x11rag_engine_config\x12\xba\x01\n\x12GetRagEngineConfig\x12\x35.google.cloud.aiplatform.v1.GetRagEngineConfigRequest\x1a+.google.cloud.aiplatform.v1.RagEngineConfig\"@\xda\x41\x04name\x82\xd3\xe4\x93\x02\x33\x12\x31/v1/{name=projects/*/locations/*/ragEngineConfig}\x1aM\xca\x41\x19\x61iplatform.googleapis.com\xd2\x41.https://www.googleapis.com/auth/cloud-platformB\xd7\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\x19VertexRagDataServiceProtoP\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
19
|
|
|
20
20
|
pool = Google::Protobuf::DescriptorPool.generated_pool
|
|
21
21
|
|
|
@@ -62,9 +62,12 @@ module Google
|
|
|
62
62
|
ListRagFilesResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ListRagFilesResponse").msgclass
|
|
63
63
|
DeleteRagFileRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.DeleteRagFileRequest").msgclass
|
|
64
64
|
CreateRagCorpusOperationMetadata = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.CreateRagCorpusOperationMetadata").msgclass
|
|
65
|
+
GetRagEngineConfigRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.GetRagEngineConfigRequest").msgclass
|
|
65
66
|
UpdateRagCorpusRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.UpdateRagCorpusRequest").msgclass
|
|
66
67
|
UpdateRagCorpusOperationMetadata = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.UpdateRagCorpusOperationMetadata").msgclass
|
|
67
68
|
ImportRagFilesOperationMetadata = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ImportRagFilesOperationMetadata").msgclass
|
|
69
|
+
UpdateRagEngineConfigRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.UpdateRagEngineConfigRequest").msgclass
|
|
70
|
+
UpdateRagEngineConfigOperationMetadata = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.UpdateRagEngineConfigOperationMetadata").msgclass
|
|
68
71
|
end
|
|
69
72
|
end
|
|
70
73
|
end
|
|
@@ -53,6 +53,10 @@ module Google
|
|
|
53
53
|
rpc :ListRagFiles, ::Google::Cloud::AIPlatform::V1::ListRagFilesRequest, ::Google::Cloud::AIPlatform::V1::ListRagFilesResponse
|
|
54
54
|
# Deletes a RagFile.
|
|
55
55
|
rpc :DeleteRagFile, ::Google::Cloud::AIPlatform::V1::DeleteRagFileRequest, ::Google::Longrunning::Operation
|
|
56
|
+
# Updates a RagEngineConfig.
|
|
57
|
+
rpc :UpdateRagEngineConfig, ::Google::Cloud::AIPlatform::V1::UpdateRagEngineConfigRequest, ::Google::Longrunning::Operation
|
|
58
|
+
# Gets a RagEngineConfig.
|
|
59
|
+
rpc :GetRagEngineConfig, ::Google::Cloud::AIPlatform::V1::GetRagEngineConfigRequest, ::Google::Cloud::AIPlatform::V1::RagEngineConfig
|
|
56
60
|
end
|
|
57
61
|
|
|
58
62
|
Stub = Service.rpc_stub_class
|
|
@@ -132,9 +132,10 @@ module Google
|
|
|
132
132
|
# @!attribute [r] dedicated_endpoint_dns
|
|
133
133
|
# @return [::String]
|
|
134
134
|
# Output only. DNS of the dedicated endpoint. Will only be populated if
|
|
135
|
-
# dedicated_endpoint_enabled is true.
|
|
136
|
-
#
|
|
137
|
-
#
|
|
135
|
+
# dedicated_endpoint_enabled is true. Depending on the features enabled, uid
|
|
136
|
+
# might be a random number or a string. For example, if fast_tryout is
|
|
137
|
+
# enabled, uid will be fasttryout. Format:
|
|
138
|
+
# `https://{endpoint_id}.{region}-{uid}.prediction.vertexai.goog`.
|
|
138
139
|
# @!attribute [rw] client_connection_config
|
|
139
140
|
# @return [::Google::Cloud::AIPlatform::V1::ClientConnectionConfig]
|
|
140
141
|
# Configurations that are applied to the endpoint for online prediction.
|
|
@@ -144,6 +145,11 @@ module Google
|
|
|
144
145
|
# @!attribute [r] satisfies_pzi
|
|
145
146
|
# @return [::Boolean]
|
|
146
147
|
# Output only. Reserved for future use.
|
|
148
|
+
# @!attribute [rw] gen_ai_advanced_features_config
|
|
149
|
+
# @return [::Google::Cloud::AIPlatform::V1::GenAiAdvancedFeaturesConfig]
|
|
150
|
+
# Optional. Configuration for GenAiAdvancedFeatures. If the endpoint is
|
|
151
|
+
# serving GenAI models, advanced features like native RAG integration can be
|
|
152
|
+
# configured. Currently, only Model Garden models are supported.
|
|
147
153
|
class Endpoint
|
|
148
154
|
include ::Google::Protobuf::MessageExts
|
|
149
155
|
extend ::Google::Protobuf::MessageExts::ClassMethods
|
|
@@ -195,9 +201,8 @@ module Google
|
|
|
195
201
|
# This value should be 1-10 characters, and valid characters are `/[0-9]/`.
|
|
196
202
|
# @!attribute [rw] model
|
|
197
203
|
# @return [::String]
|
|
198
|
-
#
|
|
199
|
-
#
|
|
200
|
-
# Endpoint.
|
|
204
|
+
# The resource name of the Model that this is the deployment of. Note that
|
|
205
|
+
# the Model may be in a different location than the DeployedModel's Endpoint.
|
|
201
206
|
#
|
|
202
207
|
# The resource name may contain version id or version alias to specify the
|
|
203
208
|
# version.
|
|
@@ -382,6 +387,26 @@ module Google
|
|
|
382
387
|
extend ::Google::Protobuf::MessageExts::ClassMethods
|
|
383
388
|
end
|
|
384
389
|
|
|
390
|
+
# Configuration for GenAiAdvancedFeatures.
|
|
391
|
+
# @!attribute [rw] rag_config
|
|
392
|
+
# @return [::Google::Cloud::AIPlatform::V1::GenAiAdvancedFeaturesConfig::RagConfig]
|
|
393
|
+
# Configuration for Retrieval Augmented Generation feature.
|
|
394
|
+
class GenAiAdvancedFeaturesConfig
|
|
395
|
+
include ::Google::Protobuf::MessageExts
|
|
396
|
+
extend ::Google::Protobuf::MessageExts::ClassMethods
|
|
397
|
+
|
|
398
|
+
# Configuration for Retrieval Augmented Generation feature.
|
|
399
|
+
# @!attribute [rw] enable_rag
|
|
400
|
+
# @return [::Boolean]
|
|
401
|
+
# If true, enable Retrieval Augmented Generation in ChatCompletion request.
|
|
402
|
+
# Once enabled, the endpoint will be identified as GenAI endpoint and
|
|
403
|
+
# Arthedain router will be used.
|
|
404
|
+
class RagConfig
|
|
405
|
+
include ::Google::Protobuf::MessageExts
|
|
406
|
+
extend ::Google::Protobuf::MessageExts::ClassMethods
|
|
407
|
+
end
|
|
408
|
+
end
|
|
409
|
+
|
|
385
410
|
# Configuration for Speculative Decoding.
|
|
386
411
|
# @!attribute [rw] draft_model_speculation
|
|
387
412
|
# @return [::Google::Cloud::AIPlatform::V1::SpeculativeDecodingSpec::DraftModelSpeculation]
|
|
@@ -47,6 +47,159 @@ module Google
|
|
|
47
47
|
extend ::Google::Protobuf::MessageExts::ClassMethods
|
|
48
48
|
end
|
|
49
49
|
|
|
50
|
+
# Request message for
|
|
51
|
+
# {::Google::Cloud::AIPlatform::V1::ModelGardenService::Client#deploy ModelGardenService.Deploy}.
|
|
52
|
+
# @!attribute [rw] publisher_model_name
|
|
53
|
+
# @return [::String]
|
|
54
|
+
# The Model Garden model to deploy.
|
|
55
|
+
# Format:
|
|
56
|
+
# `publishers/{publisher}/models/{publisher_model}@{version_id}`, or
|
|
57
|
+
# `publishers/hf-{hugging-face-author}/models/{hugging-face-model-name}@001`.
|
|
58
|
+
#
|
|
59
|
+
# Note: The following fields are mutually exclusive: `publisher_model_name`, `hugging_face_model_id`. If a field in that set is populated, all other fields in the set will automatically be cleared.
|
|
60
|
+
# @!attribute [rw] hugging_face_model_id
|
|
61
|
+
# @return [::String]
|
|
62
|
+
# The Hugging Face model to deploy.
|
|
63
|
+
# Format: Hugging Face model ID like `google/gemma-2-2b-it`.
|
|
64
|
+
#
|
|
65
|
+
# Note: The following fields are mutually exclusive: `hugging_face_model_id`, `publisher_model_name`. If a field in that set is populated, all other fields in the set will automatically be cleared.
|
|
66
|
+
# @!attribute [rw] destination
|
|
67
|
+
# @return [::String]
|
|
68
|
+
# Required. The resource name of the Location to deploy the model in.
|
|
69
|
+
# Format: `projects/{project}/locations/{location}`
|
|
70
|
+
# @!attribute [rw] model_config
|
|
71
|
+
# @return [::Google::Cloud::AIPlatform::V1::DeployRequest::ModelConfig]
|
|
72
|
+
# Optional. The model config to use for the deployment.
|
|
73
|
+
# If not specified, the default model config will be used.
|
|
74
|
+
# @!attribute [rw] endpoint_config
|
|
75
|
+
# @return [::Google::Cloud::AIPlatform::V1::DeployRequest::EndpointConfig]
|
|
76
|
+
# Optional. The endpoint config to use for the deployment.
|
|
77
|
+
# If not specified, the default endpoint config will be used.
|
|
78
|
+
# @!attribute [rw] deploy_config
|
|
79
|
+
# @return [::Google::Cloud::AIPlatform::V1::DeployRequest::DeployConfig]
|
|
80
|
+
# Optional. The deploy config to use for the deployment.
|
|
81
|
+
# If not specified, the default deploy config will be used.
|
|
82
|
+
class DeployRequest
|
|
83
|
+
include ::Google::Protobuf::MessageExts
|
|
84
|
+
extend ::Google::Protobuf::MessageExts::ClassMethods
|
|
85
|
+
|
|
86
|
+
# The model config to use for the deployment.
|
|
87
|
+
# @!attribute [rw] accept_eula
|
|
88
|
+
# @return [::Boolean]
|
|
89
|
+
# Optional. Whether the user accepts the End User License Agreement (EULA)
|
|
90
|
+
# for the model.
|
|
91
|
+
# @!attribute [rw] hugging_face_access_token
|
|
92
|
+
# @return [::String]
|
|
93
|
+
# Optional. The Hugging Face read access token used to access the model
|
|
94
|
+
# artifacts of gated models.
|
|
95
|
+
# @!attribute [rw] hugging_face_cache_enabled
|
|
96
|
+
# @return [::Boolean]
|
|
97
|
+
# Optional. If true, the model will deploy with a cached version instead of
|
|
98
|
+
# directly downloading the model artifacts from Hugging Face. This is
|
|
99
|
+
# suitable for VPC-SC users with limited internet access.
|
|
100
|
+
# @!attribute [rw] model_display_name
|
|
101
|
+
# @return [::String]
|
|
102
|
+
# Optional. The user-specified display name of the uploaded model. If not
|
|
103
|
+
# set, a default name will be used.
|
|
104
|
+
# @!attribute [rw] container_spec
|
|
105
|
+
# @return [::Google::Cloud::AIPlatform::V1::ModelContainerSpec]
|
|
106
|
+
# Optional. The specification of the container that is to be used when
|
|
107
|
+
# deploying. If not set, the default container spec will be used.
|
|
108
|
+
class ModelConfig
|
|
109
|
+
include ::Google::Protobuf::MessageExts
|
|
110
|
+
extend ::Google::Protobuf::MessageExts::ClassMethods
|
|
111
|
+
end
|
|
112
|
+
|
|
113
|
+
# The endpoint config to use for the deployment.
|
|
114
|
+
# @!attribute [rw] endpoint_display_name
|
|
115
|
+
# @return [::String]
|
|
116
|
+
# Optional. The user-specified display name of the endpoint. If not set, a
|
|
117
|
+
# default name will be used.
|
|
118
|
+
# @!attribute [rw] dedicated_endpoint_enabled
|
|
119
|
+
# @return [::Boolean]
|
|
120
|
+
# Optional. If true, the endpoint will be exposed through a dedicated
|
|
121
|
+
# DNS [Endpoint.dedicated_endpoint_dns]. Your request to the dedicated DNS
|
|
122
|
+
# will be isolated from other users' traffic and will have better
|
|
123
|
+
# performance and reliability. Note: Once you enabled dedicated endpoint,
|
|
124
|
+
# you won't be able to send request to the shared DNS
|
|
125
|
+
# \\{region}-aiplatform.googleapis.com. The limitations will be removed soon.
|
|
126
|
+
class EndpointConfig
|
|
127
|
+
include ::Google::Protobuf::MessageExts
|
|
128
|
+
extend ::Google::Protobuf::MessageExts::ClassMethods
|
|
129
|
+
end
|
|
130
|
+
|
|
131
|
+
# The deploy config to use for the deployment.
|
|
132
|
+
# @!attribute [rw] dedicated_resources
|
|
133
|
+
# @return [::Google::Cloud::AIPlatform::V1::DedicatedResources]
|
|
134
|
+
# Optional. The dedicated resources to use for the endpoint. If not set,
|
|
135
|
+
# the default resources will be used.
|
|
136
|
+
# @!attribute [rw] fast_tryout_enabled
|
|
137
|
+
# @return [::Boolean]
|
|
138
|
+
# Optional. If true, enable the QMT fast tryout feature for this model if
|
|
139
|
+
# possible.
|
|
140
|
+
# @!attribute [rw] system_labels
|
|
141
|
+
# @return [::Google::Protobuf::Map{::String => ::String}]
|
|
142
|
+
# Optional. System labels for Model Garden deployments.
|
|
143
|
+
# These labels are managed by Google and for tracking purposes only.
|
|
144
|
+
class DeployConfig
|
|
145
|
+
include ::Google::Protobuf::MessageExts
|
|
146
|
+
extend ::Google::Protobuf::MessageExts::ClassMethods
|
|
147
|
+
|
|
148
|
+
# @!attribute [rw] key
|
|
149
|
+
# @return [::String]
|
|
150
|
+
# @!attribute [rw] value
|
|
151
|
+
# @return [::String]
|
|
152
|
+
class SystemLabelsEntry
|
|
153
|
+
include ::Google::Protobuf::MessageExts
|
|
154
|
+
extend ::Google::Protobuf::MessageExts::ClassMethods
|
|
155
|
+
end
|
|
156
|
+
end
|
|
157
|
+
end
|
|
158
|
+
|
|
159
|
+
# Response message for
|
|
160
|
+
# {::Google::Cloud::AIPlatform::V1::ModelGardenService::Client#deploy ModelGardenService.Deploy}.
|
|
161
|
+
# @!attribute [r] publisher_model
|
|
162
|
+
# @return [::String]
|
|
163
|
+
# Output only. The name of the PublisherModel resource.
|
|
164
|
+
# Format:
|
|
165
|
+
# `publishers/{publisher}/models/{publisher_model}@{version_id}`, or
|
|
166
|
+
# `publishers/hf-{hugging-face-author}/models/{hugging-face-model-name}@001`
|
|
167
|
+
# @!attribute [r] endpoint
|
|
168
|
+
# @return [::String]
|
|
169
|
+
# Output only. The name of the Endpoint created.
|
|
170
|
+
# Format: `projects/{project}/locations/{location}/endpoints/{endpoint}`
|
|
171
|
+
# @!attribute [r] model
|
|
172
|
+
# @return [::String]
|
|
173
|
+
# Output only. The name of the Model created.
|
|
174
|
+
# Format: `projects/{project}/locations/{location}/models/{model}`
|
|
175
|
+
class DeployResponse
|
|
176
|
+
include ::Google::Protobuf::MessageExts
|
|
177
|
+
extend ::Google::Protobuf::MessageExts::ClassMethods
|
|
178
|
+
end
|
|
179
|
+
|
|
180
|
+
# Runtime operation information for
|
|
181
|
+
# {::Google::Cloud::AIPlatform::V1::ModelGardenService::Client#deploy ModelGardenService.Deploy}.
|
|
182
|
+
# @!attribute [rw] generic_metadata
|
|
183
|
+
# @return [::Google::Cloud::AIPlatform::V1::GenericOperationMetadata]
|
|
184
|
+
# The operation generic information.
|
|
185
|
+
# @!attribute [r] publisher_model
|
|
186
|
+
# @return [::String]
|
|
187
|
+
# Output only. The name of the model resource.
|
|
188
|
+
# @!attribute [r] destination
|
|
189
|
+
# @return [::String]
|
|
190
|
+
# Output only. The resource name of the Location to deploy the model in.
|
|
191
|
+
# Format: `projects/{project}/locations/{location}`
|
|
192
|
+
# @!attribute [r] project_number
|
|
193
|
+
# @return [::Integer]
|
|
194
|
+
# Output only. The project number where the deploy model request is sent.
|
|
195
|
+
# @!attribute [r] model_id
|
|
196
|
+
# @return [::String]
|
|
197
|
+
# Output only. The model id to be used at query time.
|
|
198
|
+
class DeployOperationMetadata
|
|
199
|
+
include ::Google::Protobuf::MessageExts
|
|
200
|
+
extend ::Google::Protobuf::MessageExts::ClassMethods
|
|
201
|
+
end
|
|
202
|
+
|
|
50
203
|
# View enumeration of PublisherModel.
|
|
51
204
|
module PublisherModelView
|
|
52
205
|
# The default / unset value. The API will default to the BASIC view.
|