google-cloud-ai_platform-v1 1.14.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.
Files changed (41) hide show
  1. checksums.yaml +4 -4
  2. data/lib/google/cloud/ai_platform/v1/job_service/paths.rb +19 -0
  3. data/lib/google/cloud/ai_platform/v1/model_garden_service/client.rb +137 -0
  4. data/lib/google/cloud/ai_platform/v1/model_garden_service/operations.rb +821 -0
  5. data/lib/google/cloud/ai_platform/v1/model_garden_service/paths.rb +36 -0
  6. data/lib/google/cloud/ai_platform/v1/model_garden_service/rest/client.rb +130 -0
  7. data/lib/google/cloud/ai_platform/v1/model_garden_service/rest/operations.rb +4398 -0
  8. data/lib/google/cloud/ai_platform/v1/model_garden_service/rest/service_stub.rb +62 -0
  9. data/lib/google/cloud/ai_platform/v1/model_garden_service/rest.rb +1 -0
  10. data/lib/google/cloud/ai_platform/v1/model_garden_service.rb +1 -0
  11. data/lib/google/cloud/ai_platform/v1/persistent_resource_service/paths.rb +19 -0
  12. data/lib/google/cloud/ai_platform/v1/pipeline_service/paths.rb +19 -0
  13. data/lib/google/cloud/ai_platform/v1/schedule_service/paths.rb +19 -0
  14. data/lib/google/cloud/ai_platform/v1/version.rb +1 -1
  15. data/lib/google/cloud/ai_platform/v1/vertex_rag_data_service/client.rb +198 -0
  16. data/lib/google/cloud/ai_platform/v1/vertex_rag_data_service/paths.rb +17 -0
  17. data/lib/google/cloud/ai_platform/v1/vertex_rag_data_service/rest/client.rb +184 -0
  18. data/lib/google/cloud/ai_platform/v1/vertex_rag_data_service/rest/service_stub.rb +123 -0
  19. data/lib/google/cloud/aiplatform/v1/content_pb.rb +6 -1
  20. data/lib/google/cloud/aiplatform/v1/custom_job_pb.rb +3 -1
  21. data/lib/google/cloud/aiplatform/v1/endpoint_pb.rb +3 -1
  22. data/lib/google/cloud/aiplatform/v1/model_garden_service_pb.rb +14 -1
  23. data/lib/google/cloud/aiplatform/v1/model_garden_service_services_pb.rb +2 -0
  24. data/lib/google/cloud/aiplatform/v1/persistent_resource_pb.rb +3 -1
  25. data/lib/google/cloud/aiplatform/v1/pipeline_job_pb.rb +3 -1
  26. data/lib/google/cloud/aiplatform/v1/service_networking_pb.rb +4 -1
  27. data/lib/google/cloud/aiplatform/v1/tool_pb.rb +3 -2
  28. data/lib/google/cloud/aiplatform/v1/vertex_rag_data_pb.rb +8 -1
  29. data/lib/google/cloud/aiplatform/v1/vertex_rag_data_service_pb.rb +4 -1
  30. data/lib/google/cloud/aiplatform/v1/vertex_rag_data_service_services_pb.rb +4 -0
  31. data/proto_docs/google/cloud/aiplatform/v1/content.rb +96 -7
  32. data/proto_docs/google/cloud/aiplatform/v1/custom_job.rb +3 -0
  33. data/proto_docs/google/cloud/aiplatform/v1/endpoint.rb +31 -6
  34. data/proto_docs/google/cloud/aiplatform/v1/model_garden_service.rb +153 -0
  35. data/proto_docs/google/cloud/aiplatform/v1/persistent_resource.rb +3 -0
  36. data/proto_docs/google/cloud/aiplatform/v1/pipeline_job.rb +3 -0
  37. data/proto_docs/google/cloud/aiplatform/v1/service_networking.rb +76 -5
  38. data/proto_docs/google/cloud/aiplatform/v1/tool.rb +35 -0
  39. data/proto_docs/google/cloud/aiplatform/v1/vertex_rag_data.rb +73 -0
  40. data/proto_docs/google/cloud/aiplatform/v1/vertex_rag_data_service.rb +35 -0
  41. 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
@@ -9,10 +9,11 @@ require 'google/cloud/aiplatform/v1/openapi_pb'
9
9
  require 'google/cloud/aiplatform/v1/tool_pb'
10
10
  require 'google/cloud/aiplatform/v1/vertex_rag_data_pb'
11
11
  require 'google/protobuf/duration_pb'
12
+ require 'google/protobuf/struct_pb'
12
13
  require 'google/type/date_pb'
13
14
 
14
15
 
15
- descriptor_data = "\n(google/cloud/aiplatform/v1/content.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a(google/cloud/aiplatform/v1/openapi.proto\x1a%google/cloud/aiplatform/v1/tool.proto\x1a\x30google/cloud/aiplatform/v1/vertex_rag_data.proto\x1a\x1egoogle/protobuf/duration.proto\x1a\x16google/type/date.proto\"R\n\x07\x43ontent\x12\x11\n\x04role\x18\x01 \x01(\tB\x03\xe0\x41\x01\x12\x34\n\x05parts\x18\x02 \x03(\x0b\x32 .google.cloud.aiplatform.v1.PartB\x03\xe0\x41\x02\"\xb2\x04\n\x04Part\x12\x13\n\x04text\x18\x01 \x01(\tB\x03\xe0\x41\x01H\x00\x12<\n\x0binline_data\x18\x02 \x01(\x0b\x32 .google.cloud.aiplatform.v1.BlobB\x03\xe0\x41\x01H\x00\x12>\n\tfile_data\x18\x03 \x01(\x0b\x32$.google.cloud.aiplatform.v1.FileDataB\x03\xe0\x41\x01H\x00\x12\x46\n\rfunction_call\x18\x05 \x01(\x0b\x32(.google.cloud.aiplatform.v1.FunctionCallB\x03\xe0\x41\x01H\x00\x12N\n\x11\x66unction_response\x18\x06 \x01(\x0b\x32,.google.cloud.aiplatform.v1.FunctionResponseB\x03\xe0\x41\x01H\x00\x12J\n\x0f\x65xecutable_code\x18\x08 \x01(\x0b\x32*.google.cloud.aiplatform.v1.ExecutableCodeB\x03\xe0\x41\x01H\x00\x12U\n\x15\x63ode_execution_result\x18\t \x01(\x0b\x32/.google.cloud.aiplatform.v1.CodeExecutionResultB\x03\xe0\x41\x01H\x00\x12H\n\x0evideo_metadata\x18\x04 \x01(\x0b\x32).google.cloud.aiplatform.v1.VideoMetadataB\x03\xe0\x41\x01H\x01\x42\x06\n\x04\x64\x61taB\n\n\x08metadata\"1\n\x04\x42lob\x12\x16\n\tmime_type\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x11\n\x04\x64\x61ta\x18\x02 \x01(\x0c\x42\x03\xe0\x41\x02\"9\n\x08\x46ileData\x12\x16\n\tmime_type\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x15\n\x08\x66ile_uri\x18\x02 \x01(\tB\x03\xe0\x41\x02\"y\n\rVideoMetadata\x12\x34\n\x0cstart_offset\x18\x01 \x01(\x0b\x32\x19.google.protobuf.DurationB\x03\xe0\x41\x01\x12\x32\n\nend_offset\x18\x02 \x01(\x0b\x32\x19.google.protobuf.DurationB\x03\xe0\x41\x01\"\xe5\x0b\n\x10GenerationConfig\x12\x1d\n\x0btemperature\x18\x01 \x01(\x02\x42\x03\xe0\x41\x01H\x00\x88\x01\x01\x12\x17\n\x05top_p\x18\x02 \x01(\x02\x42\x03\xe0\x41\x01H\x01\x88\x01\x01\x12\x17\n\x05top_k\x18\x03 \x01(\x02\x42\x03\xe0\x41\x01H\x02\x88\x01\x01\x12!\n\x0f\x63\x61ndidate_count\x18\x04 \x01(\x05\x42\x03\xe0\x41\x01H\x03\x88\x01\x01\x12#\n\x11max_output_tokens\x18\x05 \x01(\x05\x42\x03\xe0\x41\x01H\x04\x88\x01\x01\x12\x1b\n\x0estop_sequences\x18\x06 \x03(\tB\x03\xe0\x41\x01\x12#\n\x11response_logprobs\x18\x12 \x01(\x08\x42\x03\xe0\x41\x01H\x05\x88\x01\x01\x12\x1a\n\x08logprobs\x18\x07 \x01(\x05\x42\x03\xe0\x41\x01H\x06\x88\x01\x01\x12\"\n\x10presence_penalty\x18\x08 \x01(\x02\x42\x03\xe0\x41\x01H\x07\x88\x01\x01\x12#\n\x11\x66requency_penalty\x18\t \x01(\x02\x42\x03\xe0\x41\x01H\x08\x88\x01\x01\x12\x16\n\x04seed\x18\x0c \x01(\x05\x42\x03\xe0\x41\x01H\t\x88\x01\x01\x12\x1f\n\x12response_mime_type\x18\r \x01(\tB\x03\xe0\x41\x01\x12\x45\n\x0fresponse_schema\x18\x10 \x01(\x0b\x32\".google.cloud.aiplatform.v1.SchemaB\x03\xe0\x41\x01H\n\x88\x01\x01\x12\\\n\x0erouting_config\x18\x11 \x01(\x0b\x32:.google.cloud.aiplatform.v1.GenerationConfig.RoutingConfigB\x03\xe0\x41\x01H\x0b\x88\x01\x01\x12Y\n\x0fthinking_config\x18\x19 \x01(\x0b\x32;.google.cloud.aiplatform.v1.GenerationConfig.ThinkingConfigB\x03\xe0\x41\x01\x1a\xc2\x04\n\rRoutingConfig\x12_\n\tauto_mode\x18\x01 \x01(\x0b\x32J.google.cloud.aiplatform.v1.GenerationConfig.RoutingConfig.AutoRoutingModeH\x00\x12\x63\n\x0bmanual_mode\x18\x02 \x01(\x0b\x32L.google.cloud.aiplatform.v1.GenerationConfig.RoutingConfig.ManualRoutingModeH\x00\x1a\x9b\x02\n\x0f\x41utoRoutingMode\x12\x88\x01\n\x18model_routing_preference\x18\x01 \x01(\x0e\x32\x61.google.cloud.aiplatform.v1.GenerationConfig.RoutingConfig.AutoRoutingMode.ModelRoutingPreferenceH\x00\x88\x01\x01\"`\n\x16ModelRoutingPreference\x12\x0b\n\x07UNKNOWN\x10\x00\x12\x16\n\x12PRIORITIZE_QUALITY\x10\x01\x12\x0c\n\x08\x42\x41LANCED\x10\x02\x12\x13\n\x0fPRIORITIZE_COST\x10\x03\x42\x1b\n\x19_model_routing_preference\x1a;\n\x11ManualRoutingMode\x12\x17\n\nmodel_name\x18\x01 \x01(\tH\x00\x88\x01\x01\x42\r\n\x0b_model_nameB\x10\n\x0erouting_config\x1aG\n\x0eThinkingConfig\x12!\n\x0fthinking_budget\x18\x03 \x01(\x05\x42\x03\xe0\x41\x01H\x00\x88\x01\x01\x42\x12\n\x10_thinking_budgetB\x0e\n\x0c_temperatureB\x08\n\x06_top_pB\x08\n\x06_top_kB\x12\n\x10_candidate_countB\x14\n\x12_max_output_tokensB\x14\n\x12_response_logprobsB\x0b\n\t_logprobsB\x13\n\x11_presence_penaltyB\x14\n\x12_frequency_penaltyB\x07\n\x05_seedB\x12\n\x10_response_schemaB\x11\n\x0f_routing_config\"\xeb\x03\n\rSafetySetting\x12?\n\x08\x63\x61tegory\x18\x01 \x01(\x0e\x32(.google.cloud.aiplatform.v1.HarmCategoryB\x03\xe0\x41\x02\x12T\n\tthreshold\x18\x02 \x01(\x0e\x32<.google.cloud.aiplatform.v1.SafetySetting.HarmBlockThresholdB\x03\xe0\x41\x02\x12N\n\x06method\x18\x04 \x01(\x0e\x32\x39.google.cloud.aiplatform.v1.SafetySetting.HarmBlockMethodB\x03\xe0\x41\x01\"\x9d\x01\n\x12HarmBlockThreshold\x12$\n HARM_BLOCK_THRESHOLD_UNSPECIFIED\x10\x00\x12\x17\n\x13\x42LOCK_LOW_AND_ABOVE\x10\x01\x12\x1a\n\x16\x42LOCK_MEDIUM_AND_ABOVE\x10\x02\x12\x13\n\x0f\x42LOCK_ONLY_HIGH\x10\x03\x12\x0e\n\nBLOCK_NONE\x10\x04\x12\x07\n\x03OFF\x10\x05\"S\n\x0fHarmBlockMethod\x12!\n\x1dHARM_BLOCK_METHOD_UNSPECIFIED\x10\x00\x12\x0c\n\x08SEVERITY\x10\x01\x12\x0f\n\x0bPROBABILITY\x10\x02\"\xbf\x04\n\x0cSafetyRating\x12?\n\x08\x63\x61tegory\x18\x01 \x01(\x0e\x32(.google.cloud.aiplatform.v1.HarmCategoryB\x03\xe0\x41\x03\x12R\n\x0bprobability\x18\x02 \x01(\x0e\x32\x38.google.cloud.aiplatform.v1.SafetyRating.HarmProbabilityB\x03\xe0\x41\x03\x12\x1e\n\x11probability_score\x18\x05 \x01(\x02\x42\x03\xe0\x41\x03\x12L\n\x08severity\x18\x06 \x01(\x0e\x32\x35.google.cloud.aiplatform.v1.SafetyRating.HarmSeverityB\x03\xe0\x41\x03\x12\x1b\n\x0eseverity_score\x18\x07 \x01(\x02\x42\x03\xe0\x41\x03\x12\x14\n\x07\x62locked\x18\x03 \x01(\x08\x42\x03\xe0\x41\x03\"b\n\x0fHarmProbability\x12 \n\x1cHARM_PROBABILITY_UNSPECIFIED\x10\x00\x12\x0e\n\nNEGLIGIBLE\x10\x01\x12\x07\n\x03LOW\x10\x02\x12\n\n\x06MEDIUM\x10\x03\x12\x08\n\x04HIGH\x10\x04\"\x94\x01\n\x0cHarmSeverity\x12\x1d\n\x19HARM_SEVERITY_UNSPECIFIED\x10\x00\x12\x1c\n\x18HARM_SEVERITY_NEGLIGIBLE\x10\x01\x12\x15\n\x11HARM_SEVERITY_LOW\x10\x02\x12\x18\n\x14HARM_SEVERITY_MEDIUM\x10\x03\x12\x16\n\x12HARM_SEVERITY_HIGH\x10\x04\"P\n\x10\x43itationMetadata\x12<\n\tcitations\x18\x01 \x03(\x0b\x32$.google.cloud.aiplatform.v1.CitationB\x03\xe0\x41\x03\"\xaa\x01\n\x08\x43itation\x12\x18\n\x0bstart_index\x18\x01 \x01(\x05\x42\x03\xe0\x41\x03\x12\x16\n\tend_index\x18\x02 \x01(\x05\x42\x03\xe0\x41\x03\x12\x10\n\x03uri\x18\x03 \x01(\tB\x03\xe0\x41\x03\x12\x12\n\x05title\x18\x04 \x01(\tB\x03\xe0\x41\x03\x12\x14\n\x07license\x18\x05 \x01(\tB\x03\xe0\x41\x03\x12\x30\n\x10publication_date\x18\x06 \x01(\x0b\x32\x11.google.type.DateB\x03\xe0\x41\x03\"\xfc\x05\n\tCandidate\x12\x12\n\x05index\x18\x01 \x01(\x05\x42\x03\xe0\x41\x03\x12\x39\n\x07\x63ontent\x18\x02 \x01(\x0b\x32#.google.cloud.aiplatform.v1.ContentB\x03\xe0\x41\x03\x12\x12\n\x05score\x18\x08 \x01(\x01\x42\x03\xe0\x41\x03\x12\x19\n\x0c\x61vg_logprobs\x18\t \x01(\x01\x42\x03\xe0\x41\x03\x12H\n\x0flogprobs_result\x18\n \x01(\x0b\x32*.google.cloud.aiplatform.v1.LogprobsResultB\x03\xe0\x41\x03\x12N\n\rfinish_reason\x18\x03 \x01(\x0e\x32\x32.google.cloud.aiplatform.v1.Candidate.FinishReasonB\x03\xe0\x41\x03\x12\x45\n\x0esafety_ratings\x18\x04 \x03(\x0b\x32(.google.cloud.aiplatform.v1.SafetyRatingB\x03\xe0\x41\x03\x12 \n\x0e\x66inish_message\x18\x05 \x01(\tB\x03\xe0\x41\x03H\x00\x88\x01\x01\x12L\n\x11\x63itation_metadata\x18\x06 \x01(\x0b\x32,.google.cloud.aiplatform.v1.CitationMetadataB\x03\xe0\x41\x03\x12N\n\x12grounding_metadata\x18\x07 \x01(\x0b\x32-.google.cloud.aiplatform.v1.GroundingMetadataB\x03\xe0\x41\x03\"\xbc\x01\n\x0c\x46inishReason\x12\x1d\n\x19\x46INISH_REASON_UNSPECIFIED\x10\x00\x12\x08\n\x04STOP\x10\x01\x12\x0e\n\nMAX_TOKENS\x10\x02\x12\n\n\x06SAFETY\x10\x03\x12\x0e\n\nRECITATION\x10\x04\x12\t\n\x05OTHER\x10\x05\x12\r\n\tBLOCKLIST\x10\x06\x12\x16\n\x12PROHIBITED_CONTENT\x10\x07\x12\x08\n\x04SPII\x10\x08\x12\x1b\n\x17MALFORMED_FUNCTION_CALL\x10\tB\x11\n\x0f_finish_message\"\x8f\x03\n\x0eLogprobsResult\x12P\n\x0etop_candidates\x18\x01 \x03(\x0b\x32\x38.google.cloud.aiplatform.v1.LogprobsResult.TopCandidates\x12O\n\x11\x63hosen_candidates\x18\x02 \x03(\x0b\x32\x34.google.cloud.aiplatform.v1.LogprobsResult.Candidate\x1a\x7f\n\tCandidate\x12\x12\n\x05token\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x15\n\x08token_id\x18\x03 \x01(\x05H\x01\x88\x01\x01\x12\x1c\n\x0flog_probability\x18\x02 \x01(\x02H\x02\x88\x01\x01\x42\x08\n\x06_tokenB\x0b\n\t_token_idB\x12\n\x10_log_probability\x1aY\n\rTopCandidates\x12H\n\ncandidates\x18\x01 \x03(\x0b\x32\x34.google.cloud.aiplatform.v1.LogprobsResult.Candidate\"g\n\x07Segment\x12\x17\n\npart_index\x18\x01 \x01(\x05\x42\x03\xe0\x41\x03\x12\x18\n\x0bstart_index\x18\x02 \x01(\x05\x42\x03\xe0\x41\x03\x12\x16\n\tend_index\x18\x03 \x01(\x05\x42\x03\xe0\x41\x03\x12\x11\n\x04text\x18\x04 \x01(\tB\x03\xe0\x41\x03\"\xad\x03\n\x0eGroundingChunk\x12=\n\x03web\x18\x01 \x01(\x0b\x32..google.cloud.aiplatform.v1.GroundingChunk.WebH\x00\x12X\n\x11retrieved_context\x18\x02 \x01(\x0b\x32;.google.cloud.aiplatform.v1.GroundingChunk.RetrievedContextH\x00\x1a=\n\x03Web\x12\x10\n\x03uri\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x12\n\x05title\x18\x02 \x01(\tH\x01\x88\x01\x01\x42\x06\n\x04_uriB\x08\n\x06_title\x1a\xb4\x01\n\x10RetrievedContext\x12\x39\n\trag_chunk\x18\x04 \x01(\x0b\x32$.google.cloud.aiplatform.v1.RagChunkH\x00\x12\x10\n\x03uri\x18\x01 \x01(\tH\x01\x88\x01\x01\x12\x12\n\x05title\x18\x02 \x01(\tH\x02\x88\x01\x01\x12\x11\n\x04text\x18\x03 \x01(\tH\x03\x88\x01\x01\x42\x11\n\x0f\x63ontext_detailsB\x06\n\x04_uriB\x08\n\x06_titleB\x07\n\x05_textB\x0c\n\nchunk_type\"\x95\x01\n\x10GroundingSupport\x12\x39\n\x07segment\x18\x01 \x01(\x0b\x32#.google.cloud.aiplatform.v1.SegmentH\x00\x88\x01\x01\x12\x1f\n\x17grounding_chunk_indices\x18\x02 \x03(\x05\x12\x19\n\x11\x63onfidence_scores\x18\x03 \x03(\x02\x42\n\n\x08_segment\"\xa3\x03\n\x11GroundingMetadata\x12\x1f\n\x12web_search_queries\x18\x01 \x03(\tB\x03\xe0\x41\x01\x12R\n\x12search_entry_point\x18\x04 \x01(\x0b\x32,.google.cloud.aiplatform.v1.SearchEntryPointB\x03\xe0\x41\x01H\x00\x88\x01\x01\x12\x44\n\x10grounding_chunks\x18\x05 \x03(\x0b\x32*.google.cloud.aiplatform.v1.GroundingChunk\x12M\n\x12grounding_supports\x18\x06 \x03(\x0b\x32,.google.cloud.aiplatform.v1.GroundingSupportB\x03\xe0\x41\x01\x12V\n\x12retrieval_metadata\x18\x07 \x01(\x0b\x32-.google.cloud.aiplatform.v1.RetrievalMetadataB\x06\xe0\x41\x01\xe0\x41\x03H\x01\x88\x01\x01\x42\x15\n\x13_search_entry_pointB\x15\n\x13_retrieval_metadata\"H\n\x10SearchEntryPoint\x12\x1d\n\x10rendered_content\x18\x01 \x01(\tB\x03\xe0\x41\x01\x12\x15\n\x08sdk_blob\x18\x02 \x01(\x0c\x42\x03\xe0\x41\x01\"G\n\x11RetrievalMetadata\x12\x32\n%google_search_dynamic_retrieval_score\x18\x02 \x01(\x02\x42\x03\xe0\x41\x01\"a\n\x12ModalityTokenCount\x12\x36\n\x08modality\x18\x01 \x01(\x0e\x32$.google.cloud.aiplatform.v1.Modality\x12\x13\n\x0btoken_count\x18\x02 \x01(\x05*\xdb\x01\n\x0cHarmCategory\x12\x1d\n\x19HARM_CATEGORY_UNSPECIFIED\x10\x00\x12\x1d\n\x19HARM_CATEGORY_HATE_SPEECH\x10\x01\x12#\n\x1fHARM_CATEGORY_DANGEROUS_CONTENT\x10\x02\x12\x1c\n\x18HARM_CATEGORY_HARASSMENT\x10\x03\x12#\n\x1fHARM_CATEGORY_SEXUALLY_EXPLICIT\x10\x04\x12%\n\x1dHARM_CATEGORY_CIVIC_INTEGRITY\x10\x05\x1a\x02\x08\x01*]\n\x08Modality\x12\x18\n\x14MODALITY_UNSPECIFIED\x10\x00\x12\x08\n\x04TEXT\x10\x01\x12\t\n\x05IMAGE\x10\x02\x12\t\n\x05VIDEO\x10\x03\x12\t\n\x05\x41UDIO\x10\x04\x12\x0c\n\x08\x44OCUMENT\x10\x05\x42\xca\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\x0c\x43ontentProtoP\x01Z>cloud.google.com/go/aiplatform/apiv1/aiplatformpb;aiplatformpb\xaa\x02\x1aGoogle.Cloud.AIPlatform.V1\xca\x02\x1aGoogle\\Cloud\\AIPlatform\\V1\xea\x02\x1dGoogle::Cloud::AIPlatform::V1b\x06proto3"
16
+ descriptor_data = "\n(google/cloud/aiplatform/v1/content.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a(google/cloud/aiplatform/v1/openapi.proto\x1a%google/cloud/aiplatform/v1/tool.proto\x1a\x30google/cloud/aiplatform/v1/vertex_rag_data.proto\x1a\x1egoogle/protobuf/duration.proto\x1a\x1cgoogle/protobuf/struct.proto\x1a\x16google/type/date.proto\"R\n\x07\x43ontent\x12\x11\n\x04role\x18\x01 \x01(\tB\x03\xe0\x41\x01\x12\x34\n\x05parts\x18\x02 \x03(\x0b\x32 .google.cloud.aiplatform.v1.PartB\x03\xe0\x41\x02\"\xec\x04\n\x04Part\x12\x13\n\x04text\x18\x01 \x01(\tB\x03\xe0\x41\x01H\x00\x12<\n\x0binline_data\x18\x02 \x01(\x0b\x32 .google.cloud.aiplatform.v1.BlobB\x03\xe0\x41\x01H\x00\x12>\n\tfile_data\x18\x03 \x01(\x0b\x32$.google.cloud.aiplatform.v1.FileDataB\x03\xe0\x41\x01H\x00\x12\x46\n\rfunction_call\x18\x05 \x01(\x0b\x32(.google.cloud.aiplatform.v1.FunctionCallB\x03\xe0\x41\x01H\x00\x12N\n\x11\x66unction_response\x18\x06 \x01(\x0b\x32,.google.cloud.aiplatform.v1.FunctionResponseB\x03\xe0\x41\x01H\x00\x12J\n\x0f\x65xecutable_code\x18\x08 \x01(\x0b\x32*.google.cloud.aiplatform.v1.ExecutableCodeB\x03\xe0\x41\x01H\x00\x12U\n\x15\x63ode_execution_result\x18\t \x01(\x0b\x32/.google.cloud.aiplatform.v1.CodeExecutionResultB\x03\xe0\x41\x01H\x00\x12\x16\n\x07thought\x18\n \x01(\x08\x42\x03\xe0\x41\x01H\x00\x12 \n\x11thought_signature\x18\x0b \x01(\x0c\x42\x03\xe0\x41\x01H\x00\x12H\n\x0evideo_metadata\x18\x04 \x01(\x0b\x32).google.cloud.aiplatform.v1.VideoMetadataB\x03\xe0\x41\x01H\x01\x42\x06\n\x04\x64\x61taB\n\n\x08metadata\"1\n\x04\x42lob\x12\x16\n\tmime_type\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x11\n\x04\x64\x61ta\x18\x02 \x01(\x0c\x42\x03\xe0\x41\x02\"9\n\x08\x46ileData\x12\x16\n\tmime_type\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x15\n\x08\x66ile_uri\x18\x02 \x01(\tB\x03\xe0\x41\x02\"y\n\rVideoMetadata\x12\x34\n\x0cstart_offset\x18\x01 \x01(\x0b\x32\x19.google.protobuf.DurationB\x03\xe0\x41\x01\x12\x32\n\nend_offset\x18\x02 \x01(\x0b\x32\x19.google.protobuf.DurationB\x03\xe0\x41\x01\"\xf8\x0c\n\x10GenerationConfig\x12\x1d\n\x0btemperature\x18\x01 \x01(\x02\x42\x03\xe0\x41\x01H\x00\x88\x01\x01\x12\x17\n\x05top_p\x18\x02 \x01(\x02\x42\x03\xe0\x41\x01H\x01\x88\x01\x01\x12\x17\n\x05top_k\x18\x03 \x01(\x02\x42\x03\xe0\x41\x01H\x02\x88\x01\x01\x12!\n\x0f\x63\x61ndidate_count\x18\x04 \x01(\x05\x42\x03\xe0\x41\x01H\x03\x88\x01\x01\x12#\n\x11max_output_tokens\x18\x05 \x01(\x05\x42\x03\xe0\x41\x01H\x04\x88\x01\x01\x12\x1b\n\x0estop_sequences\x18\x06 \x03(\tB\x03\xe0\x41\x01\x12#\n\x11response_logprobs\x18\x12 \x01(\x08\x42\x03\xe0\x41\x01H\x05\x88\x01\x01\x12\x1a\n\x08logprobs\x18\x07 \x01(\x05\x42\x03\xe0\x41\x01H\x06\x88\x01\x01\x12\"\n\x10presence_penalty\x18\x08 \x01(\x02\x42\x03\xe0\x41\x01H\x07\x88\x01\x01\x12#\n\x11\x66requency_penalty\x18\t \x01(\x02\x42\x03\xe0\x41\x01H\x08\x88\x01\x01\x12\x16\n\x04seed\x18\x0c \x01(\x05\x42\x03\xe0\x41\x01H\t\x88\x01\x01\x12\x1f\n\x12response_mime_type\x18\r \x01(\tB\x03\xe0\x41\x01\x12\x45\n\x0fresponse_schema\x18\x10 \x01(\x0b\x32\".google.cloud.aiplatform.v1.SchemaB\x03\xe0\x41\x01H\n\x88\x01\x01\x12>\n\x14response_json_schema\x18\x1c \x01(\x0b\x32\x16.google.protobuf.ValueB\x03\xe0\x41\x01H\x0b\x88\x01\x01\x12\\\n\x0erouting_config\x18\x11 \x01(\x0b\x32:.google.cloud.aiplatform.v1.GenerationConfig.RoutingConfigB\x03\xe0\x41\x01H\x0c\x88\x01\x01\x12Y\n\x0fthinking_config\x18\x19 \x01(\x0b\x32;.google.cloud.aiplatform.v1.GenerationConfig.ThinkingConfigB\x03\xe0\x41\x01\x1a\xc2\x04\n\rRoutingConfig\x12_\n\tauto_mode\x18\x01 \x01(\x0b\x32J.google.cloud.aiplatform.v1.GenerationConfig.RoutingConfig.AutoRoutingModeH\x00\x12\x63\n\x0bmanual_mode\x18\x02 \x01(\x0b\x32L.google.cloud.aiplatform.v1.GenerationConfig.RoutingConfig.ManualRoutingModeH\x00\x1a\x9b\x02\n\x0f\x41utoRoutingMode\x12\x88\x01\n\x18model_routing_preference\x18\x01 \x01(\x0e\x32\x61.google.cloud.aiplatform.v1.GenerationConfig.RoutingConfig.AutoRoutingMode.ModelRoutingPreferenceH\x00\x88\x01\x01\"`\n\x16ModelRoutingPreference\x12\x0b\n\x07UNKNOWN\x10\x00\x12\x16\n\x12PRIORITIZE_QUALITY\x10\x01\x12\x0c\n\x08\x42\x41LANCED\x10\x02\x12\x13\n\x0fPRIORITIZE_COST\x10\x03\x42\x1b\n\x19_model_routing_preference\x1a;\n\x11ManualRoutingMode\x12\x17\n\nmodel_name\x18\x01 \x01(\tH\x00\x88\x01\x01\x42\r\n\x0b_model_nameB\x10\n\x0erouting_config\x1a\x80\x01\n\x0eThinkingConfig\x12\"\n\x10include_thoughts\x18\x01 \x01(\x08\x42\x03\xe0\x41\x01H\x00\x88\x01\x01\x12!\n\x0fthinking_budget\x18\x03 \x01(\x05\x42\x03\xe0\x41\x01H\x01\x88\x01\x01\x42\x13\n\x11_include_thoughtsB\x12\n\x10_thinking_budgetB\x0e\n\x0c_temperatureB\x08\n\x06_top_pB\x08\n\x06_top_kB\x12\n\x10_candidate_countB\x14\n\x12_max_output_tokensB\x14\n\x12_response_logprobsB\x0b\n\t_logprobsB\x13\n\x11_presence_penaltyB\x14\n\x12_frequency_penaltyB\x07\n\x05_seedB\x12\n\x10_response_schemaB\x17\n\x15_response_json_schemaB\x11\n\x0f_routing_config\"\xeb\x03\n\rSafetySetting\x12?\n\x08\x63\x61tegory\x18\x01 \x01(\x0e\x32(.google.cloud.aiplatform.v1.HarmCategoryB\x03\xe0\x41\x02\x12T\n\tthreshold\x18\x02 \x01(\x0e\x32<.google.cloud.aiplatform.v1.SafetySetting.HarmBlockThresholdB\x03\xe0\x41\x02\x12N\n\x06method\x18\x04 \x01(\x0e\x32\x39.google.cloud.aiplatform.v1.SafetySetting.HarmBlockMethodB\x03\xe0\x41\x01\"\x9d\x01\n\x12HarmBlockThreshold\x12$\n HARM_BLOCK_THRESHOLD_UNSPECIFIED\x10\x00\x12\x17\n\x13\x42LOCK_LOW_AND_ABOVE\x10\x01\x12\x1a\n\x16\x42LOCK_MEDIUM_AND_ABOVE\x10\x02\x12\x13\n\x0f\x42LOCK_ONLY_HIGH\x10\x03\x12\x0e\n\nBLOCK_NONE\x10\x04\x12\x07\n\x03OFF\x10\x05\"S\n\x0fHarmBlockMethod\x12!\n\x1dHARM_BLOCK_METHOD_UNSPECIFIED\x10\x00\x12\x0c\n\x08SEVERITY\x10\x01\x12\x0f\n\x0bPROBABILITY\x10\x02\"\xbf\x04\n\x0cSafetyRating\x12?\n\x08\x63\x61tegory\x18\x01 \x01(\x0e\x32(.google.cloud.aiplatform.v1.HarmCategoryB\x03\xe0\x41\x03\x12R\n\x0bprobability\x18\x02 \x01(\x0e\x32\x38.google.cloud.aiplatform.v1.SafetyRating.HarmProbabilityB\x03\xe0\x41\x03\x12\x1e\n\x11probability_score\x18\x05 \x01(\x02\x42\x03\xe0\x41\x03\x12L\n\x08severity\x18\x06 \x01(\x0e\x32\x35.google.cloud.aiplatform.v1.SafetyRating.HarmSeverityB\x03\xe0\x41\x03\x12\x1b\n\x0eseverity_score\x18\x07 \x01(\x02\x42\x03\xe0\x41\x03\x12\x14\n\x07\x62locked\x18\x03 \x01(\x08\x42\x03\xe0\x41\x03\"b\n\x0fHarmProbability\x12 \n\x1cHARM_PROBABILITY_UNSPECIFIED\x10\x00\x12\x0e\n\nNEGLIGIBLE\x10\x01\x12\x07\n\x03LOW\x10\x02\x12\n\n\x06MEDIUM\x10\x03\x12\x08\n\x04HIGH\x10\x04\"\x94\x01\n\x0cHarmSeverity\x12\x1d\n\x19HARM_SEVERITY_UNSPECIFIED\x10\x00\x12\x1c\n\x18HARM_SEVERITY_NEGLIGIBLE\x10\x01\x12\x15\n\x11HARM_SEVERITY_LOW\x10\x02\x12\x18\n\x14HARM_SEVERITY_MEDIUM\x10\x03\x12\x16\n\x12HARM_SEVERITY_HIGH\x10\x04\"P\n\x10\x43itationMetadata\x12<\n\tcitations\x18\x01 \x03(\x0b\x32$.google.cloud.aiplatform.v1.CitationB\x03\xe0\x41\x03\"\xaa\x01\n\x08\x43itation\x12\x18\n\x0bstart_index\x18\x01 \x01(\x05\x42\x03\xe0\x41\x03\x12\x16\n\tend_index\x18\x02 \x01(\x05\x42\x03\xe0\x41\x03\x12\x10\n\x03uri\x18\x03 \x01(\tB\x03\xe0\x41\x03\x12\x12\n\x05title\x18\x04 \x01(\tB\x03\xe0\x41\x03\x12\x14\n\x07license\x18\x05 \x01(\tB\x03\xe0\x41\x03\x12\x30\n\x10publication_date\x18\x06 \x01(\x0b\x32\x11.google.type.DateB\x03\xe0\x41\x03\"\xcf\x06\n\tCandidate\x12\x12\n\x05index\x18\x01 \x01(\x05\x42\x03\xe0\x41\x03\x12\x39\n\x07\x63ontent\x18\x02 \x01(\x0b\x32#.google.cloud.aiplatform.v1.ContentB\x03\xe0\x41\x03\x12\x12\n\x05score\x18\x08 \x01(\x01\x42\x03\xe0\x41\x03\x12\x19\n\x0c\x61vg_logprobs\x18\t \x01(\x01\x42\x03\xe0\x41\x03\x12H\n\x0flogprobs_result\x18\n \x01(\x0b\x32*.google.cloud.aiplatform.v1.LogprobsResultB\x03\xe0\x41\x03\x12N\n\rfinish_reason\x18\x03 \x01(\x0e\x32\x32.google.cloud.aiplatform.v1.Candidate.FinishReasonB\x03\xe0\x41\x03\x12\x45\n\x0esafety_ratings\x18\x04 \x03(\x0b\x32(.google.cloud.aiplatform.v1.SafetyRatingB\x03\xe0\x41\x03\x12 \n\x0e\x66inish_message\x18\x05 \x01(\tB\x03\xe0\x41\x03H\x00\x88\x01\x01\x12L\n\x11\x63itation_metadata\x18\x06 \x01(\x0b\x32,.google.cloud.aiplatform.v1.CitationMetadataB\x03\xe0\x41\x03\x12N\n\x12grounding_metadata\x18\x07 \x01(\x0b\x32-.google.cloud.aiplatform.v1.GroundingMetadataB\x03\xe0\x41\x03\x12Q\n\x14url_context_metadata\x18\x0b \x01(\x0b\x32..google.cloud.aiplatform.v1.UrlContextMetadataB\x03\xe0\x41\x03\"\xbc\x01\n\x0c\x46inishReason\x12\x1d\n\x19\x46INISH_REASON_UNSPECIFIED\x10\x00\x12\x08\n\x04STOP\x10\x01\x12\x0e\n\nMAX_TOKENS\x10\x02\x12\n\n\x06SAFETY\x10\x03\x12\x0e\n\nRECITATION\x10\x04\x12\t\n\x05OTHER\x10\x05\x12\r\n\tBLOCKLIST\x10\x06\x12\x16\n\x12PROHIBITED_CONTENT\x10\x07\x12\x08\n\x04SPII\x10\x08\x12\x1b\n\x17MALFORMED_FUNCTION_CALL\x10\tB\x11\n\x0f_finish_message\"X\n\x12UrlContextMetadata\x12\x42\n\x0curl_metadata\x18\x01 \x03(\x0b\x32\'.google.cloud.aiplatform.v1.UrlMetadataB\x03\xe0\x41\x03\"\xfc\x01\n\x0bUrlMetadata\x12\x15\n\rretrieved_url\x18\x01 \x01(\t\x12X\n\x14url_retrieval_status\x18\x02 \x01(\x0e\x32:.google.cloud.aiplatform.v1.UrlMetadata.UrlRetrievalStatus\"|\n\x12UrlRetrievalStatus\x12$\n URL_RETRIEVAL_STATUS_UNSPECIFIED\x10\x00\x12 \n\x1cURL_RETRIEVAL_STATUS_SUCCESS\x10\x01\x12\x1e\n\x1aURL_RETRIEVAL_STATUS_ERROR\x10\x02\"\x8f\x03\n\x0eLogprobsResult\x12P\n\x0etop_candidates\x18\x01 \x03(\x0b\x32\x38.google.cloud.aiplatform.v1.LogprobsResult.TopCandidates\x12O\n\x11\x63hosen_candidates\x18\x02 \x03(\x0b\x32\x34.google.cloud.aiplatform.v1.LogprobsResult.Candidate\x1a\x7f\n\tCandidate\x12\x12\n\x05token\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x15\n\x08token_id\x18\x03 \x01(\x05H\x01\x88\x01\x01\x12\x1c\n\x0flog_probability\x18\x02 \x01(\x02H\x02\x88\x01\x01\x42\x08\n\x06_tokenB\x0b\n\t_token_idB\x12\n\x10_log_probability\x1aY\n\rTopCandidates\x12H\n\ncandidates\x18\x01 \x03(\x0b\x32\x34.google.cloud.aiplatform.v1.LogprobsResult.Candidate\"g\n\x07Segment\x12\x17\n\npart_index\x18\x01 \x01(\x05\x42\x03\xe0\x41\x03\x12\x18\n\x0bstart_index\x18\x02 \x01(\x05\x42\x03\xe0\x41\x03\x12\x16\n\tend_index\x18\x03 \x01(\x05\x42\x03\xe0\x41\x03\x12\x11\n\x04text\x18\x04 \x01(\tB\x03\xe0\x41\x03\"\xad\x03\n\x0eGroundingChunk\x12=\n\x03web\x18\x01 \x01(\x0b\x32..google.cloud.aiplatform.v1.GroundingChunk.WebH\x00\x12X\n\x11retrieved_context\x18\x02 \x01(\x0b\x32;.google.cloud.aiplatform.v1.GroundingChunk.RetrievedContextH\x00\x1a=\n\x03Web\x12\x10\n\x03uri\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x12\n\x05title\x18\x02 \x01(\tH\x01\x88\x01\x01\x42\x06\n\x04_uriB\x08\n\x06_title\x1a\xb4\x01\n\x10RetrievedContext\x12\x39\n\trag_chunk\x18\x04 \x01(\x0b\x32$.google.cloud.aiplatform.v1.RagChunkH\x00\x12\x10\n\x03uri\x18\x01 \x01(\tH\x01\x88\x01\x01\x12\x12\n\x05title\x18\x02 \x01(\tH\x02\x88\x01\x01\x12\x11\n\x04text\x18\x03 \x01(\tH\x03\x88\x01\x01\x42\x11\n\x0f\x63ontext_detailsB\x06\n\x04_uriB\x08\n\x06_titleB\x07\n\x05_textB\x0c\n\nchunk_type\"\x95\x01\n\x10GroundingSupport\x12\x39\n\x07segment\x18\x01 \x01(\x0b\x32#.google.cloud.aiplatform.v1.SegmentH\x00\x88\x01\x01\x12\x1f\n\x17grounding_chunk_indices\x18\x02 \x03(\x05\x12\x19\n\x11\x63onfidence_scores\x18\x03 \x03(\x02\x42\n\n\x08_segment\"\xa3\x03\n\x11GroundingMetadata\x12\x1f\n\x12web_search_queries\x18\x01 \x03(\tB\x03\xe0\x41\x01\x12R\n\x12search_entry_point\x18\x04 \x01(\x0b\x32,.google.cloud.aiplatform.v1.SearchEntryPointB\x03\xe0\x41\x01H\x00\x88\x01\x01\x12\x44\n\x10grounding_chunks\x18\x05 \x03(\x0b\x32*.google.cloud.aiplatform.v1.GroundingChunk\x12M\n\x12grounding_supports\x18\x06 \x03(\x0b\x32,.google.cloud.aiplatform.v1.GroundingSupportB\x03\xe0\x41\x01\x12V\n\x12retrieval_metadata\x18\x07 \x01(\x0b\x32-.google.cloud.aiplatform.v1.RetrievalMetadataB\x06\xe0\x41\x01\xe0\x41\x03H\x01\x88\x01\x01\x42\x15\n\x13_search_entry_pointB\x15\n\x13_retrieval_metadata\"H\n\x10SearchEntryPoint\x12\x1d\n\x10rendered_content\x18\x01 \x01(\tB\x03\xe0\x41\x01\x12\x15\n\x08sdk_blob\x18\x02 \x01(\x0c\x42\x03\xe0\x41\x01\"G\n\x11RetrievalMetadata\x12\x32\n%google_search_dynamic_retrieval_score\x18\x02 \x01(\x02\x42\x03\xe0\x41\x01\"a\n\x12ModalityTokenCount\x12\x36\n\x08modality\x18\x01 \x01(\x0e\x32$.google.cloud.aiplatform.v1.Modality\x12\x13\n\x0btoken_count\x18\x02 \x01(\x05*\xdb\x01\n\x0cHarmCategory\x12\x1d\n\x19HARM_CATEGORY_UNSPECIFIED\x10\x00\x12\x1d\n\x19HARM_CATEGORY_HATE_SPEECH\x10\x01\x12#\n\x1fHARM_CATEGORY_DANGEROUS_CONTENT\x10\x02\x12\x1c\n\x18HARM_CATEGORY_HARASSMENT\x10\x03\x12#\n\x1fHARM_CATEGORY_SEXUALLY_EXPLICIT\x10\x04\x12%\n\x1dHARM_CATEGORY_CIVIC_INTEGRITY\x10\x05\x1a\x02\x08\x01*]\n\x08Modality\x12\x18\n\x14MODALITY_UNSPECIFIED\x10\x00\x12\x08\n\x04TEXT\x10\x01\x12\t\n\x05IMAGE\x10\x02\x12\t\n\x05VIDEO\x10\x03\x12\t\n\x05\x41UDIO\x10\x04\x12\x0c\n\x08\x44OCUMENT\x10\x05\x42\xca\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\x0c\x43ontentProtoP\x01Z>cloud.google.com/go/aiplatform/apiv1/aiplatformpb;aiplatformpb\xaa\x02\x1aGoogle.Cloud.AIPlatform.V1\xca\x02\x1aGoogle\\Cloud\\AIPlatform\\V1\xea\x02\x1dGoogle::Cloud::AIPlatform::V1b\x06proto3"
16
17
 
17
18
  pool = Google::Protobuf::DescriptorPool.generated_pool
18
19
 
@@ -30,6 +31,7 @@ rescue TypeError
30
31
  ["google.cloud.aiplatform.v1.FunctionCall", "google/cloud/aiplatform/v1/tool.proto"],
31
32
  ["google.protobuf.Duration", "google/protobuf/duration.proto"],
32
33
  ["google.cloud.aiplatform.v1.Schema", "google/cloud/aiplatform/v1/openapi.proto"],
34
+ ["google.protobuf.Value", "google/protobuf/struct.proto"],
33
35
  ["google.type.Date", "google/type/date.proto"],
34
36
  ["google.cloud.aiplatform.v1.RagChunk", "google/cloud/aiplatform/v1/vertex_rag_data.proto"],
35
37
  ]
@@ -68,6 +70,9 @@ module Google
68
70
  Citation = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.Citation").msgclass
69
71
  Candidate = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.Candidate").msgclass
70
72
  Candidate::FinishReason = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.Candidate.FinishReason").enummodule
73
+ UrlContextMetadata = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.UrlContextMetadata").msgclass
74
+ UrlMetadata = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.UrlMetadata").msgclass
75
+ UrlMetadata::UrlRetrievalStatus = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.UrlMetadata.UrlRetrievalStatus").enummodule
71
76
  LogprobsResult = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.LogprobsResult").msgclass
72
77
  LogprobsResult::Candidate = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.LogprobsResult.Candidate").msgclass
73
78
  LogprobsResult::TopCandidates = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.LogprobsResult.TopCandidates").msgclass
@@ -11,12 +11,13 @@ require 'google/cloud/aiplatform/v1/env_var_pb'
11
11
  require 'google/cloud/aiplatform/v1/io_pb'
12
12
  require 'google/cloud/aiplatform/v1/job_state_pb'
13
13
  require 'google/cloud/aiplatform/v1/machine_resources_pb'
14
+ require 'google/cloud/aiplatform/v1/service_networking_pb'
14
15
  require 'google/protobuf/duration_pb'
15
16
  require 'google/protobuf/timestamp_pb'
16
17
  require 'google/rpc/status_pb'
17
18
 
18
19
 
19
- descriptor_data = "\n+google/cloud/aiplatform/v1/custom_job.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/env_var.proto\x1a#google/cloud/aiplatform/v1/io.proto\x1a*google/cloud/aiplatform/v1/job_state.proto\x1a\x32google/cloud/aiplatform/v1/machine_resources.proto\x1a\x1egoogle/protobuf/duration.proto\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x17google/rpc/status.proto\"\x99\x07\n\tCustomJob\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\x08job_spec\x18\x04 \x01(\x0b\x32).google.cloud.aiplatform.v1.CustomJobSpecB\x03\xe0\x41\x02\x12\x38\n\x05state\x18\x05 \x01(\x0e\x32$.google.cloud.aiplatform.v1.JobStateB\x03\xe0\x41\x03\x12\x34\n\x0b\x63reate_time\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\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\x34\n\x0bupdate_time\x18\t \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12&\n\x05\x65rror\x18\n \x01(\x0b\x32\x12.google.rpc.StatusB\x03\xe0\x41\x03\x12\x41\n\x06labels\x18\x0b \x03(\x0b\x32\x31.google.cloud.aiplatform.v1.CustomJob.LabelsEntry\x12\x43\n\x0f\x65ncryption_spec\x18\x0c \x01(\x0b\x32*.google.cloud.aiplatform.v1.EncryptionSpec\x12V\n\x0fweb_access_uris\x18\x10 \x03(\x0b\x32\x38.google.cloud.aiplatform.v1.CustomJob.WebAccessUrisEntryB\x03\xe0\x41\x03\x12\x1a\n\rsatisfies_pzs\x18\x12 \x01(\x08\x42\x03\xe0\x41\x03\x12\x1a\n\rsatisfies_pzi\x18\x13 \x01(\x08\x42\x03\xe0\x41\x03\x1a-\n\x0bLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x1a\x34\n\x12WebAccessUrisEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01:i\xea\x41\x66\n#aiplatform.googleapis.com/CustomJob\x12?projects/{project}/locations/{location}/customJobs/{custom_job}\"\x98\x06\n\rCustomJobSpec\x12T\n\x16persistent_resource_id\x18\x0e \x01(\tB4\xe0\x41\x01\xfa\x41.\n,aiplatform.googleapis.com/PersistentResource\x12J\n\x11worker_pool_specs\x18\x01 \x03(\x0b\x32*.google.cloud.aiplatform.v1.WorkerPoolSpecB\x03\xe0\x41\x02\x12:\n\nscheduling\x18\x03 \x01(\x0b\x32&.google.cloud.aiplatform.v1.Scheduling\x12\x17\n\x0fservice_account\x18\x04 \x01(\t\x12\x37\n\x07network\x18\x05 \x01(\tB&\xe0\x41\x01\xfa\x41 \n\x1e\x63ompute.googleapis.com/Network\x12\x1f\n\x12reserved_ip_ranges\x18\r \x03(\tB\x03\xe0\x41\x01\x12I\n\x15\x62\x61se_output_directory\x18\x06 \x01(\x0b\x32*.google.cloud.aiplatform.v1.GcsDestination\x12&\n\x1eprotected_artifact_location_id\x18\x13 \x01(\t\x12\x42\n\x0btensorboard\x18\x07 \x01(\tB-\xe0\x41\x01\xfa\x41\'\n%aiplatform.googleapis.com/Tensorboard\x12\x1e\n\x11\x65nable_web_access\x18\n \x01(\x08\x42\x03\xe0\x41\x01\x12$\n\x17\x65nable_dashboard_access\x18\x10 \x01(\x08\x42\x03\xe0\x41\x01\x12=\n\nexperiment\x18\x11 \x01(\tB)\xe0\x41\x01\xfa\x41#\n!aiplatform.googleapis.com/Context\x12\x41\n\x0e\x65xperiment_run\x18\x12 \x01(\tB)\xe0\x41\x01\xfa\x41#\n!aiplatform.googleapis.com/Context\x12\x37\n\x06models\x18\x14 \x03(\tB\'\xe0\x41\x01\xfa\x41!\n\x1f\x61iplatform.googleapis.com/Model\"\x86\x03\n\x0eWorkerPoolSpec\x12\x43\n\x0e\x63ontainer_spec\x18\x06 \x01(\x0b\x32).google.cloud.aiplatform.v1.ContainerSpecH\x00\x12L\n\x13python_package_spec\x18\x07 \x01(\x0b\x32-.google.cloud.aiplatform.v1.PythonPackageSpecH\x00\x12\x45\n\x0cmachine_spec\x18\x01 \x01(\x0b\x32\'.google.cloud.aiplatform.v1.MachineSpecB\x06\xe0\x41\x01\xe0\x41\x05\x12\x1a\n\rreplica_count\x18\x02 \x01(\x03\x42\x03\xe0\x41\x01\x12=\n\nnfs_mounts\x18\x04 \x03(\x0b\x32$.google.cloud.aiplatform.v1.NfsMountB\x03\xe0\x41\x01\x12\x37\n\tdisk_spec\x18\x05 \x01(\x0b\x32$.google.cloud.aiplatform.v1.DiskSpecB\x06\n\x04task\"w\n\rContainerSpec\x12\x16\n\timage_uri\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x0f\n\x07\x63ommand\x18\x02 \x03(\t\x12\x0c\n\x04\x61rgs\x18\x03 \x03(\t\x12/\n\x03\x65nv\x18\x04 \x03(\x0b\x32\".google.cloud.aiplatform.v1.EnvVar\"\xaa\x01\n\x11PythonPackageSpec\x12\x1f\n\x12\x65xecutor_image_uri\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x19\n\x0cpackage_uris\x18\x02 \x03(\tB\x03\xe0\x41\x02\x12\x1a\n\rpython_module\x18\x03 \x01(\tB\x03\xe0\x41\x02\x12\x0c\n\x04\x61rgs\x18\x04 \x03(\t\x12/\n\x03\x65nv\x18\x05 \x03(\x0b\x32\".google.cloud.aiplatform.v1.EnvVar\"\xf3\x02\n\nScheduling\x12*\n\x07timeout\x18\x01 \x01(\x0b\x32\x19.google.protobuf.Duration\x12%\n\x1drestart_job_on_worker_restart\x18\x03 \x01(\x08\x12\x46\n\x08strategy\x18\x04 \x01(\x0e\x32/.google.cloud.aiplatform.v1.Scheduling.StrategyB\x03\xe0\x41\x01\x12\x1c\n\x0f\x64isable_retries\x18\x05 \x01(\x08\x42\x03\xe0\x41\x01\x12\x39\n\x11max_wait_duration\x18\x06 \x01(\x0b\x32\x19.google.protobuf.DurationB\x03\xe0\x41\x01\"q\n\x08Strategy\x12\x18\n\x14STRATEGY_UNSPECIFIED\x10\x00\x12\x11\n\tON_DEMAND\x10\x01\x1a\x02\x08\x01\x12\x10\n\x08LOW_COST\x10\x02\x1a\x02\x08\x01\x12\x0c\n\x08STANDARD\x10\x03\x12\x08\n\x04SPOT\x10\x04\x12\x0e\n\nFLEX_START\x10\x06\x42\xcc\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\x0e\x43ustomJobProtoP\x01Z>cloud.google.com/go/aiplatform/apiv1/aiplatformpb;aiplatformpb\xaa\x02\x1aGoogle.Cloud.AIPlatform.V1\xca\x02\x1aGoogle\\Cloud\\AIPlatform\\V1\xea\x02\x1dGoogle::Cloud::AIPlatform::V1b\x06proto3"
20
+ descriptor_data = "\n+google/cloud/aiplatform/v1/custom_job.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/env_var.proto\x1a#google/cloud/aiplatform/v1/io.proto\x1a*google/cloud/aiplatform/v1/job_state.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\x1a\x17google/rpc/status.proto\"\x99\x07\n\tCustomJob\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\x08job_spec\x18\x04 \x01(\x0b\x32).google.cloud.aiplatform.v1.CustomJobSpecB\x03\xe0\x41\x02\x12\x38\n\x05state\x18\x05 \x01(\x0e\x32$.google.cloud.aiplatform.v1.JobStateB\x03\xe0\x41\x03\x12\x34\n\x0b\x63reate_time\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\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\x34\n\x0bupdate_time\x18\t \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12&\n\x05\x65rror\x18\n \x01(\x0b\x32\x12.google.rpc.StatusB\x03\xe0\x41\x03\x12\x41\n\x06labels\x18\x0b \x03(\x0b\x32\x31.google.cloud.aiplatform.v1.CustomJob.LabelsEntry\x12\x43\n\x0f\x65ncryption_spec\x18\x0c \x01(\x0b\x32*.google.cloud.aiplatform.v1.EncryptionSpec\x12V\n\x0fweb_access_uris\x18\x10 \x03(\x0b\x32\x38.google.cloud.aiplatform.v1.CustomJob.WebAccessUrisEntryB\x03\xe0\x41\x03\x12\x1a\n\rsatisfies_pzs\x18\x12 \x01(\x08\x42\x03\xe0\x41\x03\x12\x1a\n\rsatisfies_pzi\x18\x13 \x01(\x08\x42\x03\xe0\x41\x03\x1a-\n\x0bLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x1a\x34\n\x12WebAccessUrisEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01:i\xea\x41\x66\n#aiplatform.googleapis.com/CustomJob\x12?projects/{project}/locations/{location}/customJobs/{custom_job}\"\xeb\x06\n\rCustomJobSpec\x12T\n\x16persistent_resource_id\x18\x0e \x01(\tB4\xe0\x41\x01\xfa\x41.\n,aiplatform.googleapis.com/PersistentResource\x12J\n\x11worker_pool_specs\x18\x01 \x03(\x0b\x32*.google.cloud.aiplatform.v1.WorkerPoolSpecB\x03\xe0\x41\x02\x12:\n\nscheduling\x18\x03 \x01(\x0b\x32&.google.cloud.aiplatform.v1.Scheduling\x12\x17\n\x0fservice_account\x18\x04 \x01(\t\x12\x37\n\x07network\x18\x05 \x01(\tB&\xe0\x41\x01\xfa\x41 \n\x1e\x63ompute.googleapis.com/Network\x12\x1f\n\x12reserved_ip_ranges\x18\r \x03(\tB\x03\xe0\x41\x01\x12Q\n\x14psc_interface_config\x18\x15 \x01(\x0b\x32..google.cloud.aiplatform.v1.PscInterfaceConfigB\x03\xe0\x41\x01\x12I\n\x15\x62\x61se_output_directory\x18\x06 \x01(\x0b\x32*.google.cloud.aiplatform.v1.GcsDestination\x12&\n\x1eprotected_artifact_location_id\x18\x13 \x01(\t\x12\x42\n\x0btensorboard\x18\x07 \x01(\tB-\xe0\x41\x01\xfa\x41\'\n%aiplatform.googleapis.com/Tensorboard\x12\x1e\n\x11\x65nable_web_access\x18\n \x01(\x08\x42\x03\xe0\x41\x01\x12$\n\x17\x65nable_dashboard_access\x18\x10 \x01(\x08\x42\x03\xe0\x41\x01\x12=\n\nexperiment\x18\x11 \x01(\tB)\xe0\x41\x01\xfa\x41#\n!aiplatform.googleapis.com/Context\x12\x41\n\x0e\x65xperiment_run\x18\x12 \x01(\tB)\xe0\x41\x01\xfa\x41#\n!aiplatform.googleapis.com/Context\x12\x37\n\x06models\x18\x14 \x03(\tB\'\xe0\x41\x01\xfa\x41!\n\x1f\x61iplatform.googleapis.com/Model\"\x86\x03\n\x0eWorkerPoolSpec\x12\x43\n\x0e\x63ontainer_spec\x18\x06 \x01(\x0b\x32).google.cloud.aiplatform.v1.ContainerSpecH\x00\x12L\n\x13python_package_spec\x18\x07 \x01(\x0b\x32-.google.cloud.aiplatform.v1.PythonPackageSpecH\x00\x12\x45\n\x0cmachine_spec\x18\x01 \x01(\x0b\x32\'.google.cloud.aiplatform.v1.MachineSpecB\x06\xe0\x41\x01\xe0\x41\x05\x12\x1a\n\rreplica_count\x18\x02 \x01(\x03\x42\x03\xe0\x41\x01\x12=\n\nnfs_mounts\x18\x04 \x03(\x0b\x32$.google.cloud.aiplatform.v1.NfsMountB\x03\xe0\x41\x01\x12\x37\n\tdisk_spec\x18\x05 \x01(\x0b\x32$.google.cloud.aiplatform.v1.DiskSpecB\x06\n\x04task\"w\n\rContainerSpec\x12\x16\n\timage_uri\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x0f\n\x07\x63ommand\x18\x02 \x03(\t\x12\x0c\n\x04\x61rgs\x18\x03 \x03(\t\x12/\n\x03\x65nv\x18\x04 \x03(\x0b\x32\".google.cloud.aiplatform.v1.EnvVar\"\xaa\x01\n\x11PythonPackageSpec\x12\x1f\n\x12\x65xecutor_image_uri\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x19\n\x0cpackage_uris\x18\x02 \x03(\tB\x03\xe0\x41\x02\x12\x1a\n\rpython_module\x18\x03 \x01(\tB\x03\xe0\x41\x02\x12\x0c\n\x04\x61rgs\x18\x04 \x03(\t\x12/\n\x03\x65nv\x18\x05 \x03(\x0b\x32\".google.cloud.aiplatform.v1.EnvVar\"\xf3\x02\n\nScheduling\x12*\n\x07timeout\x18\x01 \x01(\x0b\x32\x19.google.protobuf.Duration\x12%\n\x1drestart_job_on_worker_restart\x18\x03 \x01(\x08\x12\x46\n\x08strategy\x18\x04 \x01(\x0e\x32/.google.cloud.aiplatform.v1.Scheduling.StrategyB\x03\xe0\x41\x01\x12\x1c\n\x0f\x64isable_retries\x18\x05 \x01(\x08\x42\x03\xe0\x41\x01\x12\x39\n\x11max_wait_duration\x18\x06 \x01(\x0b\x32\x19.google.protobuf.DurationB\x03\xe0\x41\x01\"q\n\x08Strategy\x12\x18\n\x14STRATEGY_UNSPECIFIED\x10\x00\x12\x11\n\tON_DEMAND\x10\x01\x1a\x02\x08\x01\x12\x10\n\x08LOW_COST\x10\x02\x1a\x02\x08\x01\x12\x0c\n\x08STANDARD\x10\x03\x12\x08\n\x04SPOT\x10\x04\x12\x0e\n\nFLEX_START\x10\x06\x42\xcc\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\x0e\x43ustomJobProtoP\x01Z>cloud.google.com/go/aiplatform/apiv1/aiplatformpb;aiplatformpb\xaa\x02\x1aGoogle.Cloud.AIPlatform.V1\xca\x02\x1aGoogle\\Cloud\\AIPlatform\\V1\xea\x02\x1dGoogle::Cloud::AIPlatform::V1b\x06proto3"
20
21
 
21
22
  pool = Google::Protobuf::DescriptorPool.generated_pool
22
23
 
@@ -34,6 +35,7 @@ rescue TypeError
34
35
  ["google.protobuf.Timestamp", "google/protobuf/timestamp.proto"],
35
36
  ["google.rpc.Status", "google/rpc/status.proto"],
36
37
  ["google.cloud.aiplatform.v1.EncryptionSpec", "google/cloud/aiplatform/v1/encryption_spec.proto"],
38
+ ["google.cloud.aiplatform.v1.PscInterfaceConfig", "google/cloud/aiplatform/v1/service_networking.proto"],
37
39
  ["google.cloud.aiplatform.v1.GcsDestination", "google/cloud/aiplatform/v1/io.proto"],
38
40
  ["google.cloud.aiplatform.v1.MachineSpec", "google/cloud/aiplatform/v1/machine_resources.proto"],
39
41
  ["google.cloud.aiplatform.v1.EnvVar", "google/cloud/aiplatform/v1/env_var.proto"],
@@ -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\"\xff\n\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\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}\"\xf5\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\x36\n\x05model\x18\x02 \x01(\tB\'\xe0\x41\x02\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\"\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"
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\x8c\x02\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/*}\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"
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,11 +8,12 @@ require 'google/api/field_behavior_pb'
8
8
  require 'google/api/resource_pb'
9
9
  require 'google/cloud/aiplatform/v1/encryption_spec_pb'
10
10
  require 'google/cloud/aiplatform/v1/machine_resources_pb'
11
+ require 'google/cloud/aiplatform/v1/service_networking_pb'
11
12
  require 'google/protobuf/timestamp_pb'
12
13
  require 'google/rpc/status_pb'
13
14
 
14
15
 
15
- descriptor_data = "\n4google/cloud/aiplatform/v1/persistent_resource.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\x32google/cloud/aiplatform/v1/machine_resources.proto\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x17google/rpc/status.proto\"\xde\x08\n\x12PersistentResource\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x05\x12\x19\n\x0c\x64isplay_name\x18\x02 \x01(\tB\x03\xe0\x41\x01\x12\x45\n\x0eresource_pools\x18\x04 \x03(\x0b\x32(.google.cloud.aiplatform.v1.ResourcePoolB\x03\xe0\x41\x02\x12H\n\x05state\x18\x05 \x01(\x0e\x32\x34.google.cloud.aiplatform.v1.PersistentResource.StateB\x03\xe0\x41\x03\x12&\n\x05\x65rror\x18\x06 \x01(\x0b\x32\x12.google.rpc.StatusB\x03\xe0\x41\x03\x12\x34\n\x0b\x63reate_time\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x33\n\nstart_time\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x34\n\x0bupdate_time\x18\t \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12O\n\x06labels\x18\n \x03(\x0b\x32:.google.cloud.aiplatform.v1.PersistentResource.LabelsEntryB\x03\xe0\x41\x01\x12\x37\n\x07network\x18\x0b \x01(\tB&\xe0\x41\x01\xfa\x41 \n\x1e\x63ompute.googleapis.com/Network\x12H\n\x0f\x65ncryption_spec\x18\x0c \x01(\x0b\x32*.google.cloud.aiplatform.v1.EncryptionSpecB\x03\xe0\x41\x01\x12S\n\x15resource_runtime_spec\x18\r \x01(\x0b\x32/.google.cloud.aiplatform.v1.ResourceRuntimeSpecB\x03\xe0\x41\x01\x12J\n\x10resource_runtime\x18\x0e \x01(\x0b\x32+.google.cloud.aiplatform.v1.ResourceRuntimeB\x03\xe0\x41\x03\x12\x1f\n\x12reserved_ip_ranges\x18\x0f \x03(\tB\x03\xe0\x41\x01\x1a-\n\x0bLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"s\n\x05State\x12\x15\n\x11STATE_UNSPECIFIED\x10\x00\x12\x10\n\x0cPROVISIONING\x10\x01\x12\x0b\n\x07RUNNING\x10\x03\x12\x0c\n\x08STOPPING\x10\x04\x12\t\n\x05\x45RROR\x10\x05\x12\r\n\tREBOOTING\x10\x06\x12\x0c\n\x08UPDATING\x10\x07:\x85\x01\xea\x41\x81\x01\n,aiplatform.googleapis.com/PersistentResource\x12Qprojects/{project}/locations/{location}/persistentResources/{persistent_resource}\"\xdb\x03\n\x0cResourcePool\x12\x0f\n\x02id\x18\x01 \x01(\tB\x03\xe0\x41\x05\x12\x45\n\x0cmachine_spec\x18\x02 \x01(\x0b\x32\'.google.cloud.aiplatform.v1.MachineSpecB\x06\xe0\x41\x02\xe0\x41\x05\x12\x1f\n\rreplica_count\x18\x03 \x01(\x03\x42\x03\xe0\x41\x01H\x00\x88\x01\x01\x12<\n\tdisk_spec\x18\x04 \x01(\x0b\x32$.google.cloud.aiplatform.v1.DiskSpecB\x03\xe0\x41\x01\x12\x1f\n\x12used_replica_count\x18\x06 \x01(\x03\x42\x03\xe0\x41\x03\x12W\n\x10\x61utoscaling_spec\x18\x07 \x01(\x0b\x32\x38.google.cloud.aiplatform.v1.ResourcePool.AutoscalingSpecB\x03\xe0\x41\x01\x1a\x87\x01\n\x0f\x41utoscalingSpec\x12#\n\x11min_replica_count\x18\x01 \x01(\x03\x42\x03\xe0\x41\x01H\x00\x88\x01\x01\x12#\n\x11max_replica_count\x18\x02 \x01(\x03\x42\x03\xe0\x41\x01H\x01\x88\x01\x01\x42\x14\n\x12_min_replica_countB\x14\n\x12_max_replica_countB\x10\n\x0e_replica_count\"\xa4\x01\n\x13ResourceRuntimeSpec\x12Q\n\x14service_account_spec\x18\x02 \x01(\x0b\x32..google.cloud.aiplatform.v1.ServiceAccountSpecB\x03\xe0\x41\x01\x12:\n\x08ray_spec\x18\x01 \x01(\x0b\x32#.google.cloud.aiplatform.v1.RaySpecB\x03\xe0\x41\x01\"\xf3\x02\n\x07RaySpec\x12\x16\n\timage_uri\x18\x01 \x01(\tB\x03\xe0\x41\x01\x12^\n\x14resource_pool_images\x18\x06 \x03(\x0b\x32;.google.cloud.aiplatform.v1.RaySpec.ResourcePoolImagesEntryB\x03\xe0\x41\x01\x12\'\n\x1ahead_node_resource_pool_id\x18\x07 \x01(\tB\x03\xe0\x41\x01\x12G\n\x0fray_metric_spec\x18\x08 \x01(\x0b\x32).google.cloud.aiplatform.v1.RayMetricSpecB\x03\xe0\x41\x01\x12\x43\n\rray_logs_spec\x18\n \x01(\x0b\x32\'.google.cloud.aiplatform.v1.RayLogsSpecB\x03\xe0\x41\x01\x1a\x39\n\x17ResourcePoolImagesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\x9b\x01\n\x0fResourceRuntime\x12U\n\x0b\x61\x63\x63\x65ss_uris\x18\x01 \x03(\x0b\x32;.google.cloud.aiplatform.v1.ResourceRuntime.AccessUrisEntryB\x03\xe0\x41\x03\x1a\x31\n\x0f\x41\x63\x63\x65ssUrisEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"^\n\x12ServiceAccountSpec\x12*\n\x1d\x65nable_custom_service_account\x18\x01 \x01(\x08\x42\x03\xe0\x41\x02\x12\x1c\n\x0fservice_account\x18\x02 \x01(\tB\x03\xe0\x41\x01\"&\n\rRayMetricSpec\x12\x15\n\x08\x64isabled\x18\x01 \x01(\x08\x42\x03\xe0\x41\x01\"$\n\x0bRayLogsSpec\x12\x15\n\x08\x64isabled\x18\x01 \x01(\x08\x42\x03\xe0\x41\x01\x42\xd5\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\x17PersistentResourceProtoP\x01Z>cloud.google.com/go/aiplatform/apiv1/aiplatformpb;aiplatformpb\xaa\x02\x1aGoogle.Cloud.AIPlatform.V1\xca\x02\x1aGoogle\\Cloud\\AIPlatform\\V1\xea\x02\x1dGoogle::Cloud::AIPlatform::V1b\x06proto3"
16
+ descriptor_data = "\n4google/cloud/aiplatform/v1/persistent_resource.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\x32google/cloud/aiplatform/v1/machine_resources.proto\x1a\x33google/cloud/aiplatform/v1/service_networking.proto\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x17google/rpc/status.proto\"\xb1\t\n\x12PersistentResource\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x05\x12\x19\n\x0c\x64isplay_name\x18\x02 \x01(\tB\x03\xe0\x41\x01\x12\x45\n\x0eresource_pools\x18\x04 \x03(\x0b\x32(.google.cloud.aiplatform.v1.ResourcePoolB\x03\xe0\x41\x02\x12H\n\x05state\x18\x05 \x01(\x0e\x32\x34.google.cloud.aiplatform.v1.PersistentResource.StateB\x03\xe0\x41\x03\x12&\n\x05\x65rror\x18\x06 \x01(\x0b\x32\x12.google.rpc.StatusB\x03\xe0\x41\x03\x12\x34\n\x0b\x63reate_time\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x33\n\nstart_time\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x34\n\x0bupdate_time\x18\t \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12O\n\x06labels\x18\n \x03(\x0b\x32:.google.cloud.aiplatform.v1.PersistentResource.LabelsEntryB\x03\xe0\x41\x01\x12\x37\n\x07network\x18\x0b \x01(\tB&\xe0\x41\x01\xfa\x41 \n\x1e\x63ompute.googleapis.com/Network\x12Q\n\x14psc_interface_config\x18\x11 \x01(\x0b\x32..google.cloud.aiplatform.v1.PscInterfaceConfigB\x03\xe0\x41\x01\x12H\n\x0f\x65ncryption_spec\x18\x0c \x01(\x0b\x32*.google.cloud.aiplatform.v1.EncryptionSpecB\x03\xe0\x41\x01\x12S\n\x15resource_runtime_spec\x18\r \x01(\x0b\x32/.google.cloud.aiplatform.v1.ResourceRuntimeSpecB\x03\xe0\x41\x01\x12J\n\x10resource_runtime\x18\x0e \x01(\x0b\x32+.google.cloud.aiplatform.v1.ResourceRuntimeB\x03\xe0\x41\x03\x12\x1f\n\x12reserved_ip_ranges\x18\x0f \x03(\tB\x03\xe0\x41\x01\x1a-\n\x0bLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"s\n\x05State\x12\x15\n\x11STATE_UNSPECIFIED\x10\x00\x12\x10\n\x0cPROVISIONING\x10\x01\x12\x0b\n\x07RUNNING\x10\x03\x12\x0c\n\x08STOPPING\x10\x04\x12\t\n\x05\x45RROR\x10\x05\x12\r\n\tREBOOTING\x10\x06\x12\x0c\n\x08UPDATING\x10\x07:\x85\x01\xea\x41\x81\x01\n,aiplatform.googleapis.com/PersistentResource\x12Qprojects/{project}/locations/{location}/persistentResources/{persistent_resource}\"\xdb\x03\n\x0cResourcePool\x12\x0f\n\x02id\x18\x01 \x01(\tB\x03\xe0\x41\x05\x12\x45\n\x0cmachine_spec\x18\x02 \x01(\x0b\x32\'.google.cloud.aiplatform.v1.MachineSpecB\x06\xe0\x41\x02\xe0\x41\x05\x12\x1f\n\rreplica_count\x18\x03 \x01(\x03\x42\x03\xe0\x41\x01H\x00\x88\x01\x01\x12<\n\tdisk_spec\x18\x04 \x01(\x0b\x32$.google.cloud.aiplatform.v1.DiskSpecB\x03\xe0\x41\x01\x12\x1f\n\x12used_replica_count\x18\x06 \x01(\x03\x42\x03\xe0\x41\x03\x12W\n\x10\x61utoscaling_spec\x18\x07 \x01(\x0b\x32\x38.google.cloud.aiplatform.v1.ResourcePool.AutoscalingSpecB\x03\xe0\x41\x01\x1a\x87\x01\n\x0f\x41utoscalingSpec\x12#\n\x11min_replica_count\x18\x01 \x01(\x03\x42\x03\xe0\x41\x01H\x00\x88\x01\x01\x12#\n\x11max_replica_count\x18\x02 \x01(\x03\x42\x03\xe0\x41\x01H\x01\x88\x01\x01\x42\x14\n\x12_min_replica_countB\x14\n\x12_max_replica_countB\x10\n\x0e_replica_count\"\xa4\x01\n\x13ResourceRuntimeSpec\x12Q\n\x14service_account_spec\x18\x02 \x01(\x0b\x32..google.cloud.aiplatform.v1.ServiceAccountSpecB\x03\xe0\x41\x01\x12:\n\x08ray_spec\x18\x01 \x01(\x0b\x32#.google.cloud.aiplatform.v1.RaySpecB\x03\xe0\x41\x01\"\xf3\x02\n\x07RaySpec\x12\x16\n\timage_uri\x18\x01 \x01(\tB\x03\xe0\x41\x01\x12^\n\x14resource_pool_images\x18\x06 \x03(\x0b\x32;.google.cloud.aiplatform.v1.RaySpec.ResourcePoolImagesEntryB\x03\xe0\x41\x01\x12\'\n\x1ahead_node_resource_pool_id\x18\x07 \x01(\tB\x03\xe0\x41\x01\x12G\n\x0fray_metric_spec\x18\x08 \x01(\x0b\x32).google.cloud.aiplatform.v1.RayMetricSpecB\x03\xe0\x41\x01\x12\x43\n\rray_logs_spec\x18\n \x01(\x0b\x32\'.google.cloud.aiplatform.v1.RayLogsSpecB\x03\xe0\x41\x01\x1a\x39\n\x17ResourcePoolImagesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\x9b\x01\n\x0fResourceRuntime\x12U\n\x0b\x61\x63\x63\x65ss_uris\x18\x01 \x03(\x0b\x32;.google.cloud.aiplatform.v1.ResourceRuntime.AccessUrisEntryB\x03\xe0\x41\x03\x1a\x31\n\x0f\x41\x63\x63\x65ssUrisEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"^\n\x12ServiceAccountSpec\x12*\n\x1d\x65nable_custom_service_account\x18\x01 \x01(\x08\x42\x03\xe0\x41\x02\x12\x1c\n\x0fservice_account\x18\x02 \x01(\tB\x03\xe0\x41\x01\"&\n\rRayMetricSpec\x12\x15\n\x08\x64isabled\x18\x01 \x01(\x08\x42\x03\xe0\x41\x01\"$\n\x0bRayLogsSpec\x12\x15\n\x08\x64isabled\x18\x01 \x01(\x08\x42\x03\xe0\x41\x01\x42\xd5\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\x17PersistentResourceProtoP\x01Z>cloud.google.com/go/aiplatform/apiv1/aiplatformpb;aiplatformpb\xaa\x02\x1aGoogle.Cloud.AIPlatform.V1\xca\x02\x1aGoogle\\Cloud\\AIPlatform\\V1\xea\x02\x1dGoogle::Cloud::AIPlatform::V1b\x06proto3"
16
17
 
17
18
  pool = Google::Protobuf::DescriptorPool.generated_pool
18
19
 
@@ -29,6 +30,7 @@ rescue TypeError
29
30
  imports = [
30
31
  ["google.rpc.Status", "google/rpc/status.proto"],
31
32
  ["google.protobuf.Timestamp", "google/protobuf/timestamp.proto"],
33
+ ["google.cloud.aiplatform.v1.PscInterfaceConfig", "google/cloud/aiplatform/v1/service_networking.proto"],
32
34
  ["google.cloud.aiplatform.v1.EncryptionSpec", "google/cloud/aiplatform/v1/encryption_spec.proto"],
33
35
  ["google.cloud.aiplatform.v1.MachineSpec", "google/cloud/aiplatform/v1/machine_resources.proto"],
34
36
  ]
@@ -12,13 +12,14 @@ require 'google/cloud/aiplatform/v1/encryption_spec_pb'
12
12
  require 'google/cloud/aiplatform/v1/execution_pb'
13
13
  require 'google/cloud/aiplatform/v1/pipeline_failure_policy_pb'
14
14
  require 'google/cloud/aiplatform/v1/pipeline_state_pb'
15
+ require 'google/cloud/aiplatform/v1/service_networking_pb'
15
16
  require 'google/cloud/aiplatform/v1/value_pb'
16
17
  require 'google/protobuf/struct_pb'
17
18
  require 'google/protobuf/timestamp_pb'
18
19
  require 'google/rpc/status_pb'
19
20
 
20
21
 
21
- descriptor_data = "\n-google/cloud/aiplatform/v1/pipeline_job.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a)google/cloud/aiplatform/v1/artifact.proto\x1a(google/cloud/aiplatform/v1/context.proto\x1a\x30google/cloud/aiplatform/v1/encryption_spec.proto\x1a*google/cloud/aiplatform/v1/execution.proto\x1a\x38google/cloud/aiplatform/v1/pipeline_failure_policy.proto\x1a/google/cloud/aiplatform/v1/pipeline_state.proto\x1a&google/cloud/aiplatform/v1/value.proto\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x17google/rpc/status.proto\"\xf4\x0e\n\x0bPipelineJob\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x03\x12\x14\n\x0c\x64isplay_name\x18\x02 \x01(\t\x12\x34\n\x0b\x63reate_time\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x33\n\nstart_time\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x31\n\x08\x65nd_time\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x34\n\x0bupdate_time\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12.\n\rpipeline_spec\x18\x07 \x01(\x0b\x32\x17.google.protobuf.Struct\x12=\n\x05state\x18\x08 \x01(\x0e\x32).google.cloud.aiplatform.v1.PipelineStateB\x03\xe0\x41\x03\x12\x46\n\njob_detail\x18\t \x01(\x0b\x32-.google.cloud.aiplatform.v1.PipelineJobDetailB\x03\xe0\x41\x03\x12&\n\x05\x65rror\x18\n \x01(\x0b\x32\x12.google.rpc.StatusB\x03\xe0\x41\x03\x12\x43\n\x06labels\x18\x0b \x03(\x0b\x32\x33.google.cloud.aiplatform.v1.PipelineJob.LabelsEntry\x12M\n\x0eruntime_config\x18\x0c \x01(\x0b\x32\x35.google.cloud.aiplatform.v1.PipelineJob.RuntimeConfig\x12\x43\n\x0f\x65ncryption_spec\x18\x10 \x01(\x0b\x32*.google.cloud.aiplatform.v1.EncryptionSpec\x12\x17\n\x0fservice_account\x18\x11 \x01(\t\x12\x34\n\x07network\x18\x12 \x01(\tB#\xfa\x41 \n\x1e\x63ompute.googleapis.com/Network\x12\x1a\n\x12reserved_ip_ranges\x18\x19 \x03(\t\x12\x14\n\x0ctemplate_uri\x18\x13 \x01(\t\x12T\n\x11template_metadata\x18\x14 \x01(\x0b\x32\x34.google.cloud.aiplatform.v1.PipelineTemplateMetadataB\x03\xe0\x41\x03\x12\x1a\n\rschedule_name\x18\x16 \x01(\tB\x03\xe0\x41\x03\x12\"\n\x15preflight_validations\x18\x1a \x01(\x08\x42\x03\xe0\x41\x01\x1a\xf8\x05\n\rRuntimeConfig\x12]\n\nparameters\x18\x01 \x03(\x0b\x32\x45.google.cloud.aiplatform.v1.PipelineJob.RuntimeConfig.ParametersEntryB\x02\x18\x01\x12!\n\x14gcs_output_directory\x18\x02 \x01(\tB\x03\xe0\x41\x02\x12\x64\n\x10parameter_values\x18\x03 \x03(\x0b\x32J.google.cloud.aiplatform.v1.PipelineJob.RuntimeConfig.ParameterValuesEntry\x12I\n\x0e\x66\x61ilure_policy\x18\x04 \x01(\x0e\x32\x31.google.cloud.aiplatform.v1.PipelineFailurePolicy\x12\x62\n\x0finput_artifacts\x18\x05 \x03(\x0b\x32I.google.cloud.aiplatform.v1.PipelineJob.RuntimeConfig.InputArtifactsEntry\x1a.\n\rInputArtifact\x12\x15\n\x0b\x61rtifact_id\x18\x01 \x01(\tH\x00\x42\x06\n\x04kind\x1aT\n\x0fParametersEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x30\n\x05value\x18\x02 \x01(\x0b\x32!.google.cloud.aiplatform.v1.Value:\x02\x38\x01\x1aN\n\x14ParameterValuesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12%\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.Value:\x02\x38\x01\x1az\n\x13InputArtifactsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12R\n\x05value\x18\x02 \x01(\x0b\x32\x43.google.cloud.aiplatform.v1.PipelineJob.RuntimeConfig.InputArtifact:\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:o\xea\x41l\n%aiplatform.googleapis.com/PipelineJob\x12\x43projects/{project}/locations/{location}/pipelineJobs/{pipeline_job}\"+\n\x18PipelineTemplateMetadata\x12\x0f\n\x07version\x18\x03 \x01(\t\"\xea\x01\n\x11PipelineJobDetail\x12\x42\n\x10pipeline_context\x18\x01 \x01(\x0b\x32#.google.cloud.aiplatform.v1.ContextB\x03\xe0\x41\x03\x12\x46\n\x14pipeline_run_context\x18\x02 \x01(\x0b\x32#.google.cloud.aiplatform.v1.ContextB\x03\xe0\x41\x03\x12I\n\x0ctask_details\x18\x03 \x03(\x0b\x32..google.cloud.aiplatform.v1.PipelineTaskDetailB\x03\xe0\x41\x03\"\x9d\x0b\n\x12PipelineTaskDetail\x12\x14\n\x07task_id\x18\x01 \x01(\x03\x42\x03\xe0\x41\x03\x12\x1b\n\x0eparent_task_id\x18\x0c \x01(\x03\x42\x03\xe0\x41\x03\x12\x16\n\ttask_name\x18\x02 \x01(\tB\x03\xe0\x41\x03\x12\x34\n\x0b\x63reate_time\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x33\n\nstart_time\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x31\n\x08\x65nd_time\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12T\n\x0f\x65xecutor_detail\x18\x06 \x01(\x0b\x32\x36.google.cloud.aiplatform.v1.PipelineTaskExecutorDetailB\x03\xe0\x41\x03\x12H\n\x05state\x18\x07 \x01(\x0e\x32\x34.google.cloud.aiplatform.v1.PipelineTaskDetail.StateB\x03\xe0\x41\x03\x12=\n\texecution\x18\x08 \x01(\x0b\x32%.google.cloud.aiplatform.v1.ExecutionB\x03\xe0\x41\x03\x12&\n\x05\x65rror\x18\t \x01(\x0b\x32\x12.google.rpc.StatusB\x03\xe0\x41\x03\x12\x64\n\x14pipeline_task_status\x18\r \x03(\x0b\x32\x41.google.cloud.aiplatform.v1.PipelineTaskDetail.PipelineTaskStatusB\x03\xe0\x41\x03\x12O\n\x06inputs\x18\n \x03(\x0b\x32:.google.cloud.aiplatform.v1.PipelineTaskDetail.InputsEntryB\x03\xe0\x41\x03\x12Q\n\x07outputs\x18\x0b \x03(\x0b\x32;.google.cloud.aiplatform.v1.PipelineTaskDetail.OutputsEntryB\x03\xe0\x41\x03\x1a\xbc\x01\n\x12PipelineTaskStatus\x12\x34\n\x0bupdate_time\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12H\n\x05state\x18\x02 \x01(\x0e\x32\x34.google.cloud.aiplatform.v1.PipelineTaskDetail.StateB\x03\xe0\x41\x03\x12&\n\x05\x65rror\x18\x03 \x01(\x0b\x32\x12.google.rpc.StatusB\x03\xe0\x41\x03\x1aL\n\x0c\x41rtifactList\x12<\n\tartifacts\x18\x01 \x03(\x0b\x32$.google.cloud.aiplatform.v1.ArtifactB\x03\xe0\x41\x03\x1aj\n\x0bInputsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12J\n\x05value\x18\x02 \x01(\x0b\x32;.google.cloud.aiplatform.v1.PipelineTaskDetail.ArtifactList:\x02\x38\x01\x1ak\n\x0cOutputsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12J\n\x05value\x18\x02 \x01(\x0b\x32;.google.cloud.aiplatform.v1.PipelineTaskDetail.ArtifactList:\x02\x38\x01\"\xa6\x01\n\x05State\x12\x15\n\x11STATE_UNSPECIFIED\x10\x00\x12\x0b\n\x07PENDING\x10\x01\x12\x0b\n\x07RUNNING\x10\x02\x12\r\n\tSUCCEEDED\x10\x03\x12\x12\n\x0e\x43\x41NCEL_PENDING\x10\x04\x12\x0e\n\nCANCELLING\x10\x05\x12\r\n\tCANCELLED\x10\x06\x12\n\n\x06\x46\x41ILED\x10\x07\x12\x0b\n\x07SKIPPED\x10\x08\x12\x11\n\rNOT_TRIGGERED\x10\t\"\xcb\x04\n\x1aPipelineTaskExecutorDetail\x12g\n\x10\x63ontainer_detail\x18\x01 \x01(\x0b\x32\x46.google.cloud.aiplatform.v1.PipelineTaskExecutorDetail.ContainerDetailB\x03\xe0\x41\x03H\x00\x12h\n\x11\x63ustom_job_detail\x18\x02 \x01(\x0b\x32\x46.google.cloud.aiplatform.v1.PipelineTaskExecutorDetail.CustomJobDetailB\x03\xe0\x41\x03H\x00\x1a\xe7\x01\n\x0f\x43ontainerDetail\x12=\n\x08main_job\x18\x01 \x01(\tB+\xe0\x41\x03\xfa\x41%\n#aiplatform.googleapis.com/CustomJob\x12J\n\x15pre_caching_check_job\x18\x02 \x01(\tB+\xe0\x41\x03\xfa\x41%\n#aiplatform.googleapis.com/CustomJob\x12\x1d\n\x10\x66\x61iled_main_jobs\x18\x03 \x03(\tB\x03\xe0\x41\x03\x12*\n\x1d\x66\x61iled_pre_caching_check_jobs\x18\x04 \x03(\tB\x03\xe0\x41\x03\x1a\x65\n\x0f\x43ustomJobDetail\x12\x38\n\x03job\x18\x01 \x01(\tB+\xe0\x41\x03\xfa\x41%\n#aiplatform.googleapis.com/CustomJob\x12\x18\n\x0b\x66\x61iled_jobs\x18\x03 \x03(\tB\x03\xe0\x41\x03\x42\t\n\x07\x64\x65tailsB\x97\x02\n\x1e\x63om.google.cloud.aiplatform.v1B\x08PipelineP\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\x41N\n\x1e\x63ompute.googleapis.com/Network\x12,projects/{project}/global/networks/{network}b\x06proto3"
22
+ descriptor_data = "\n-google/cloud/aiplatform/v1/pipeline_job.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a)google/cloud/aiplatform/v1/artifact.proto\x1a(google/cloud/aiplatform/v1/context.proto\x1a\x30google/cloud/aiplatform/v1/encryption_spec.proto\x1a*google/cloud/aiplatform/v1/execution.proto\x1a\x38google/cloud/aiplatform/v1/pipeline_failure_policy.proto\x1a/google/cloud/aiplatform/v1/pipeline_state.proto\x1a\x33google/cloud/aiplatform/v1/service_networking.proto\x1a&google/cloud/aiplatform/v1/value.proto\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x17google/rpc/status.proto\"\xc7\x0f\n\x0bPipelineJob\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x03\x12\x14\n\x0c\x64isplay_name\x18\x02 \x01(\t\x12\x34\n\x0b\x63reate_time\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x33\n\nstart_time\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x31\n\x08\x65nd_time\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x34\n\x0bupdate_time\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12.\n\rpipeline_spec\x18\x07 \x01(\x0b\x32\x17.google.protobuf.Struct\x12=\n\x05state\x18\x08 \x01(\x0e\x32).google.cloud.aiplatform.v1.PipelineStateB\x03\xe0\x41\x03\x12\x46\n\njob_detail\x18\t \x01(\x0b\x32-.google.cloud.aiplatform.v1.PipelineJobDetailB\x03\xe0\x41\x03\x12&\n\x05\x65rror\x18\n \x01(\x0b\x32\x12.google.rpc.StatusB\x03\xe0\x41\x03\x12\x43\n\x06labels\x18\x0b \x03(\x0b\x32\x33.google.cloud.aiplatform.v1.PipelineJob.LabelsEntry\x12M\n\x0eruntime_config\x18\x0c \x01(\x0b\x32\x35.google.cloud.aiplatform.v1.PipelineJob.RuntimeConfig\x12\x43\n\x0f\x65ncryption_spec\x18\x10 \x01(\x0b\x32*.google.cloud.aiplatform.v1.EncryptionSpec\x12\x17\n\x0fservice_account\x18\x11 \x01(\t\x12\x34\n\x07network\x18\x12 \x01(\tB#\xfa\x41 \n\x1e\x63ompute.googleapis.com/Network\x12\x1a\n\x12reserved_ip_ranges\x18\x19 \x03(\t\x12Q\n\x14psc_interface_config\x18\x1f \x01(\x0b\x32..google.cloud.aiplatform.v1.PscInterfaceConfigB\x03\xe0\x41\x01\x12\x14\n\x0ctemplate_uri\x18\x13 \x01(\t\x12T\n\x11template_metadata\x18\x14 \x01(\x0b\x32\x34.google.cloud.aiplatform.v1.PipelineTemplateMetadataB\x03\xe0\x41\x03\x12\x1a\n\rschedule_name\x18\x16 \x01(\tB\x03\xe0\x41\x03\x12\"\n\x15preflight_validations\x18\x1a \x01(\x08\x42\x03\xe0\x41\x01\x1a\xf8\x05\n\rRuntimeConfig\x12]\n\nparameters\x18\x01 \x03(\x0b\x32\x45.google.cloud.aiplatform.v1.PipelineJob.RuntimeConfig.ParametersEntryB\x02\x18\x01\x12!\n\x14gcs_output_directory\x18\x02 \x01(\tB\x03\xe0\x41\x02\x12\x64\n\x10parameter_values\x18\x03 \x03(\x0b\x32J.google.cloud.aiplatform.v1.PipelineJob.RuntimeConfig.ParameterValuesEntry\x12I\n\x0e\x66\x61ilure_policy\x18\x04 \x01(\x0e\x32\x31.google.cloud.aiplatform.v1.PipelineFailurePolicy\x12\x62\n\x0finput_artifacts\x18\x05 \x03(\x0b\x32I.google.cloud.aiplatform.v1.PipelineJob.RuntimeConfig.InputArtifactsEntry\x1a.\n\rInputArtifact\x12\x15\n\x0b\x61rtifact_id\x18\x01 \x01(\tH\x00\x42\x06\n\x04kind\x1aT\n\x0fParametersEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x30\n\x05value\x18\x02 \x01(\x0b\x32!.google.cloud.aiplatform.v1.Value:\x02\x38\x01\x1aN\n\x14ParameterValuesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12%\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.Value:\x02\x38\x01\x1az\n\x13InputArtifactsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12R\n\x05value\x18\x02 \x01(\x0b\x32\x43.google.cloud.aiplatform.v1.PipelineJob.RuntimeConfig.InputArtifact:\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:o\xea\x41l\n%aiplatform.googleapis.com/PipelineJob\x12\x43projects/{project}/locations/{location}/pipelineJobs/{pipeline_job}\"+\n\x18PipelineTemplateMetadata\x12\x0f\n\x07version\x18\x03 \x01(\t\"\xea\x01\n\x11PipelineJobDetail\x12\x42\n\x10pipeline_context\x18\x01 \x01(\x0b\x32#.google.cloud.aiplatform.v1.ContextB\x03\xe0\x41\x03\x12\x46\n\x14pipeline_run_context\x18\x02 \x01(\x0b\x32#.google.cloud.aiplatform.v1.ContextB\x03\xe0\x41\x03\x12I\n\x0ctask_details\x18\x03 \x03(\x0b\x32..google.cloud.aiplatform.v1.PipelineTaskDetailB\x03\xe0\x41\x03\"\x9d\x0b\n\x12PipelineTaskDetail\x12\x14\n\x07task_id\x18\x01 \x01(\x03\x42\x03\xe0\x41\x03\x12\x1b\n\x0eparent_task_id\x18\x0c \x01(\x03\x42\x03\xe0\x41\x03\x12\x16\n\ttask_name\x18\x02 \x01(\tB\x03\xe0\x41\x03\x12\x34\n\x0b\x63reate_time\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x33\n\nstart_time\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x31\n\x08\x65nd_time\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12T\n\x0f\x65xecutor_detail\x18\x06 \x01(\x0b\x32\x36.google.cloud.aiplatform.v1.PipelineTaskExecutorDetailB\x03\xe0\x41\x03\x12H\n\x05state\x18\x07 \x01(\x0e\x32\x34.google.cloud.aiplatform.v1.PipelineTaskDetail.StateB\x03\xe0\x41\x03\x12=\n\texecution\x18\x08 \x01(\x0b\x32%.google.cloud.aiplatform.v1.ExecutionB\x03\xe0\x41\x03\x12&\n\x05\x65rror\x18\t \x01(\x0b\x32\x12.google.rpc.StatusB\x03\xe0\x41\x03\x12\x64\n\x14pipeline_task_status\x18\r \x03(\x0b\x32\x41.google.cloud.aiplatform.v1.PipelineTaskDetail.PipelineTaskStatusB\x03\xe0\x41\x03\x12O\n\x06inputs\x18\n \x03(\x0b\x32:.google.cloud.aiplatform.v1.PipelineTaskDetail.InputsEntryB\x03\xe0\x41\x03\x12Q\n\x07outputs\x18\x0b \x03(\x0b\x32;.google.cloud.aiplatform.v1.PipelineTaskDetail.OutputsEntryB\x03\xe0\x41\x03\x1a\xbc\x01\n\x12PipelineTaskStatus\x12\x34\n\x0bupdate_time\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12H\n\x05state\x18\x02 \x01(\x0e\x32\x34.google.cloud.aiplatform.v1.PipelineTaskDetail.StateB\x03\xe0\x41\x03\x12&\n\x05\x65rror\x18\x03 \x01(\x0b\x32\x12.google.rpc.StatusB\x03\xe0\x41\x03\x1aL\n\x0c\x41rtifactList\x12<\n\tartifacts\x18\x01 \x03(\x0b\x32$.google.cloud.aiplatform.v1.ArtifactB\x03\xe0\x41\x03\x1aj\n\x0bInputsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12J\n\x05value\x18\x02 \x01(\x0b\x32;.google.cloud.aiplatform.v1.PipelineTaskDetail.ArtifactList:\x02\x38\x01\x1ak\n\x0cOutputsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12J\n\x05value\x18\x02 \x01(\x0b\x32;.google.cloud.aiplatform.v1.PipelineTaskDetail.ArtifactList:\x02\x38\x01\"\xa6\x01\n\x05State\x12\x15\n\x11STATE_UNSPECIFIED\x10\x00\x12\x0b\n\x07PENDING\x10\x01\x12\x0b\n\x07RUNNING\x10\x02\x12\r\n\tSUCCEEDED\x10\x03\x12\x12\n\x0e\x43\x41NCEL_PENDING\x10\x04\x12\x0e\n\nCANCELLING\x10\x05\x12\r\n\tCANCELLED\x10\x06\x12\n\n\x06\x46\x41ILED\x10\x07\x12\x0b\n\x07SKIPPED\x10\x08\x12\x11\n\rNOT_TRIGGERED\x10\t\"\xcb\x04\n\x1aPipelineTaskExecutorDetail\x12g\n\x10\x63ontainer_detail\x18\x01 \x01(\x0b\x32\x46.google.cloud.aiplatform.v1.PipelineTaskExecutorDetail.ContainerDetailB\x03\xe0\x41\x03H\x00\x12h\n\x11\x63ustom_job_detail\x18\x02 \x01(\x0b\x32\x46.google.cloud.aiplatform.v1.PipelineTaskExecutorDetail.CustomJobDetailB\x03\xe0\x41\x03H\x00\x1a\xe7\x01\n\x0f\x43ontainerDetail\x12=\n\x08main_job\x18\x01 \x01(\tB+\xe0\x41\x03\xfa\x41%\n#aiplatform.googleapis.com/CustomJob\x12J\n\x15pre_caching_check_job\x18\x02 \x01(\tB+\xe0\x41\x03\xfa\x41%\n#aiplatform.googleapis.com/CustomJob\x12\x1d\n\x10\x66\x61iled_main_jobs\x18\x03 \x03(\tB\x03\xe0\x41\x03\x12*\n\x1d\x66\x61iled_pre_caching_check_jobs\x18\x04 \x03(\tB\x03\xe0\x41\x03\x1a\x65\n\x0f\x43ustomJobDetail\x12\x38\n\x03job\x18\x01 \x01(\tB+\xe0\x41\x03\xfa\x41%\n#aiplatform.googleapis.com/CustomJob\x12\x18\n\x0b\x66\x61iled_jobs\x18\x03 \x03(\tB\x03\xe0\x41\x03\x42\t\n\x07\x64\x65tailsB\x97\x02\n\x1e\x63om.google.cloud.aiplatform.v1B\x08PipelineP\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\x41N\n\x1e\x63ompute.googleapis.com/Network\x12,projects/{project}/global/networks/{network}b\x06proto3"
22
23
 
23
24
  pool = Google::Protobuf::DescriptorPool.generated_pool
24
25
 
@@ -37,6 +38,7 @@ rescue TypeError
37
38
  ["google.protobuf.Struct", "google/protobuf/struct.proto"],
38
39
  ["google.rpc.Status", "google/rpc/status.proto"],
39
40
  ["google.cloud.aiplatform.v1.EncryptionSpec", "google/cloud/aiplatform/v1/encryption_spec.proto"],
41
+ ["google.cloud.aiplatform.v1.PscInterfaceConfig", "google/cloud/aiplatform/v1/service_networking.proto"],
40
42
  ["google.cloud.aiplatform.v1.Value", "google/cloud/aiplatform/v1/value.proto"],
41
43
  ["google.cloud.aiplatform.v1.Context", "google/cloud/aiplatform/v1/context.proto"],
42
44
  ["google.cloud.aiplatform.v1.Execution", "google/cloud/aiplatform/v1/execution.proto"],
@@ -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\"D\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\"\x86\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\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(\tB\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"
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
 
@@ -41,6 +41,9 @@ module Google
41
41
  PSCAutomationConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.PSCAutomationConfig").msgclass
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
+ 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
44
47
  end
45
48
  end
46
49
  end
@@ -11,7 +11,7 @@ require 'google/protobuf/struct_pb'
11
11
  require 'google/type/latlng_pb'
12
12
 
13
13
 
14
- descriptor_data = "\n%google/cloud/aiplatform/v1/tool.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a(google/cloud/aiplatform/v1/openapi.proto\x1a\x1cgoogle/protobuf/struct.proto\x1a\x18google/type/latlng.proto\"\x81\x04\n\x04Tool\x12S\n\x15\x66unction_declarations\x18\x01 \x03(\x0b\x32/.google.cloud.aiplatform.v1.FunctionDeclarationB\x03\xe0\x41\x01\x12=\n\tretrieval\x18\x02 \x01(\x0b\x32%.google.cloud.aiplatform.v1.RetrievalB\x03\xe0\x41\x01\x12I\n\rgoogle_search\x18\x07 \x01(\x0b\x32-.google.cloud.aiplatform.v1.Tool.GoogleSearchB\x03\xe0\x41\x01\x12W\n\x17google_search_retrieval\x18\x03 \x01(\x0b\x32\x31.google.cloud.aiplatform.v1.GoogleSearchRetrievalB\x03\xe0\x41\x01\x12S\n\x15\x65nterprise_web_search\x18\x06 \x01(\x0b\x32/.google.cloud.aiplatform.v1.EnterpriseWebSearchB\x03\xe0\x41\x01\x12K\n\x0e\x63ode_execution\x18\x04 \x01(\x0b\x32..google.cloud.aiplatform.v1.Tool.CodeExecutionB\x03\xe0\x41\x01\x1a\x0e\n\x0cGoogleSearch\x1a\x0f\n\rCodeExecution\"\xba\x01\n\x13\x46unctionDeclaration\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x18\n\x0b\x64\x65scription\x18\x02 \x01(\tB\x03\xe0\x41\x01\x12;\n\nparameters\x18\x03 \x01(\x0b\x32\".google.cloud.aiplatform.v1.SchemaB\x03\xe0\x41\x01\x12\x39\n\x08response\x18\x04 \x01(\x0b\x32\".google.cloud.aiplatform.v1.SchemaB\x03\xe0\x41\x01\"M\n\x0c\x46unctionCall\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12*\n\x04\x61rgs\x18\x02 \x01(\x0b\x32\x17.google.protobuf.StructB\x03\xe0\x41\x01\"U\n\x10\x46unctionResponse\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12.\n\x08response\x18\x02 \x01(\x0b\x32\x17.google.protobuf.StructB\x03\xe0\x41\x02\"\xa1\x01\n\x0e\x45xecutableCode\x12J\n\x08language\x18\x01 \x01(\x0e\x32\x33.google.cloud.aiplatform.v1.ExecutableCode.LanguageB\x03\xe0\x41\x02\x12\x11\n\x04\x63ode\x18\x02 \x01(\tB\x03\xe0\x41\x02\"0\n\x08Language\x12\x18\n\x14LANGUAGE_UNSPECIFIED\x10\x00\x12\n\n\x06PYTHON\x10\x01\"\xe0\x01\n\x13\x43odeExecutionResult\x12M\n\x07outcome\x18\x01 \x01(\x0e\x32\x37.google.cloud.aiplatform.v1.CodeExecutionResult.OutcomeB\x03\xe0\x41\x02\x12\x13\n\x06output\x18\x02 \x01(\tB\x03\xe0\x41\x01\"e\n\x07Outcome\x12\x17\n\x13OUTCOME_UNSPECIFIED\x10\x00\x12\x0e\n\nOUTCOME_OK\x10\x01\x12\x12\n\x0eOUTCOME_FAILED\x10\x02\x12\x1d\n\x19OUTCOME_DEADLINE_EXCEEDED\x10\x03\"\xc9\x01\n\tRetrieval\x12\x46\n\x10vertex_ai_search\x18\x02 \x01(\x0b\x32*.google.cloud.aiplatform.v1.VertexAISearchH\x00\x12\x46\n\x10vertex_rag_store\x18\x04 \x01(\x0b\x32*.google.cloud.aiplatform.v1.VertexRagStoreH\x00\x12\"\n\x13\x64isable_attribution\x18\x03 \x01(\x08\x42\x05\x18\x01\xe0\x41\x01\x42\x08\n\x06source\"\xaa\x03\n\x0eVertexRagStore\x12R\n\rrag_resources\x18\x04 \x03(\x0b\x32\x36.google.cloud.aiplatform.v1.VertexRagStore.RagResourceB\x03\xe0\x41\x01\x12$\n\x10similarity_top_k\x18\x02 \x01(\x05\x42\x05\x18\x01\xe0\x41\x01H\x00\x88\x01\x01\x12-\n\x19vector_distance_threshold\x18\x03 \x01(\x01\x42\x05\x18\x01\xe0\x41\x01H\x01\x88\x01\x01\x12Q\n\x14rag_retrieval_config\x18\x06 \x01(\x0b\x32..google.cloud.aiplatform.v1.RagRetrievalConfigB\x03\xe0\x41\x01\x1ai\n\x0bRagResource\x12?\n\nrag_corpus\x18\x01 \x01(\tB+\xe0\x41\x01\xfa\x41%\n#aiplatform.googleapis.com/RagCorpus\x12\x19\n\x0crag_file_ids\x18\x02 \x03(\tB\x03\xe0\x41\x01\x42\x13\n\x11_similarity_top_kB\x1c\n\x1a_vector_distance_threshold\"\xfa\x01\n\x0eVertexAISearch\x12\x16\n\tdatastore\x18\x01 \x01(\tB\x03\xe0\x41\x01\x12\x13\n\x06\x65ngine\x18\x02 \x01(\tB\x03\xe0\x41\x01\x12\x18\n\x0bmax_results\x18\x03 \x01(\x05\x42\x03\xe0\x41\x01\x12\x13\n\x06\x66ilter\x18\x04 \x01(\tB\x03\xe0\x41\x01\x12R\n\x10\x64\x61ta_store_specs\x18\x05 \x03(\x0b\x32\x38.google.cloud.aiplatform.v1.VertexAISearch.DataStoreSpec\x1a\x38\n\rDataStoreSpec\x12\x12\n\ndata_store\x18\x01 \x01(\t\x12\x13\n\x06\x66ilter\x18\x02 \x01(\tB\x03\xe0\x41\x01\"m\n\x15GoogleSearchRetrieval\x12T\n\x18\x64ynamic_retrieval_config\x18\x02 \x01(\x0b\x32\x32.google.cloud.aiplatform.v1.DynamicRetrievalConfig\"\x15\n\x13\x45nterpriseWebSearch\"\xca\x01\n\x16\x44ynamicRetrievalConfig\x12\x45\n\x04mode\x18\x01 \x01(\x0e\x32\x37.google.cloud.aiplatform.v1.DynamicRetrievalConfig.Mode\x12#\n\x11\x64ynamic_threshold\x18\x02 \x01(\x02\x42\x03\xe0\x41\x01H\x00\x88\x01\x01\".\n\x04Mode\x12\x14\n\x10MODE_UNSPECIFIED\x10\x00\x12\x10\n\x0cMODE_DYNAMIC\x10\x01\x42\x14\n\x12_dynamic_threshold\"\xb1\x01\n\nToolConfig\x12W\n\x17\x66unction_calling_config\x18\x01 \x01(\x0b\x32\x31.google.cloud.aiplatform.v1.FunctionCallingConfigB\x03\xe0\x41\x01\x12J\n\x10retrieval_config\x18\x02 \x01(\x0b\x32+.google.cloud.aiplatform.v1.RetrievalConfigB\x03\xe0\x41\x01\"\xc2\x01\n\x15\x46unctionCallingConfig\x12I\n\x04mode\x18\x01 \x01(\x0e\x32\x36.google.cloud.aiplatform.v1.FunctionCallingConfig.ModeB\x03\xe0\x41\x01\x12#\n\x16\x61llowed_function_names\x18\x02 \x03(\tB\x03\xe0\x41\x01\"9\n\x04Mode\x12\x14\n\x10MODE_UNSPECIFIED\x10\x00\x12\x08\n\x04\x41UTO\x10\x01\x12\x07\n\x03\x41NY\x10\x02\x12\x08\n\x04NONE\x10\x03\"v\n\x0fRetrievalConfig\x12)\n\x07lat_lng\x18\x01 \x01(\x0b\x32\x13.google.type.LatLngH\x00\x88\x01\x01\x12\x1a\n\rlanguage_code\x18\x02 \x01(\tH\x01\x88\x01\x01\x42\n\n\x08_lat_lngB\x10\n\x0e_language_code\"\xaa\x05\n\x12RagRetrievalConfig\x12\x12\n\x05top_k\x18\x01 \x01(\x05\x42\x03\xe0\x41\x01\x12J\n\x06\x66ilter\x18\x03 \x01(\x0b\x32\x35.google.cloud.aiplatform.v1.RagRetrievalConfig.FilterB\x03\xe0\x41\x01\x12L\n\x07ranking\x18\x04 \x01(\x0b\x32\x36.google.cloud.aiplatform.v1.RagRetrievalConfig.RankingB\x03\xe0\x41\x01\x1a\x93\x01\n\x06\x46ilter\x12(\n\x19vector_distance_threshold\x18\x03 \x01(\x01\x42\x03\xe0\x41\x01H\x00\x12*\n\x1bvector_similarity_threshold\x18\x04 \x01(\x01\x42\x03\xe0\x41\x01H\x00\x12\x1c\n\x0fmetadata_filter\x18\x02 \x01(\tB\x03\xe0\x41\x01\x42\x15\n\x13vector_db_threshold\x1a\xcf\x02\n\x07Ranking\x12_\n\x0crank_service\x18\x01 \x01(\x0b\x32\x42.google.cloud.aiplatform.v1.RagRetrievalConfig.Ranking.RankServiceB\x03\xe0\x41\x01H\x00\x12[\n\nllm_ranker\x18\x03 \x01(\x0b\x32@.google.cloud.aiplatform.v1.RagRetrievalConfig.Ranking.LlmRankerB\x03\xe0\x41\x01H\x00\x1a:\n\x0bRankService\x12\x1c\n\nmodel_name\x18\x01 \x01(\tB\x03\xe0\x41\x01H\x00\x88\x01\x01\x42\r\n\x0b_model_name\x1a\x38\n\tLlmRanker\x12\x1c\n\nmodel_name\x18\x01 \x01(\tB\x03\xe0\x41\x01H\x00\x88\x01\x01\x42\r\n\x0b_model_nameB\x10\n\x0eranking_configB\xc7\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\tToolProtoP\x01Z>cloud.google.com/go/aiplatform/apiv1/aiplatformpb;aiplatformpb\xaa\x02\x1aGoogle.Cloud.AIPlatform.V1\xca\x02\x1aGoogle\\Cloud\\AIPlatform\\V1\xea\x02\x1dGoogle::Cloud::AIPlatform::V1b\x06proto3"
14
+ descriptor_data = "\n%google/cloud/aiplatform/v1/tool.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a(google/cloud/aiplatform/v1/openapi.proto\x1a\x1cgoogle/protobuf/struct.proto\x1a\x18google/type/latlng.proto\"\xc3\x04\n\x04Tool\x12S\n\x15\x66unction_declarations\x18\x01 \x03(\x0b\x32/.google.cloud.aiplatform.v1.FunctionDeclarationB\x03\xe0\x41\x01\x12=\n\tretrieval\x18\x02 \x01(\x0b\x32%.google.cloud.aiplatform.v1.RetrievalB\x03\xe0\x41\x01\x12I\n\rgoogle_search\x18\x07 \x01(\x0b\x32-.google.cloud.aiplatform.v1.Tool.GoogleSearchB\x03\xe0\x41\x01\x12W\n\x17google_search_retrieval\x18\x03 \x01(\x0b\x32\x31.google.cloud.aiplatform.v1.GoogleSearchRetrievalB\x03\xe0\x41\x01\x12S\n\x15\x65nterprise_web_search\x18\x06 \x01(\x0b\x32/.google.cloud.aiplatform.v1.EnterpriseWebSearchB\x03\xe0\x41\x01\x12K\n\x0e\x63ode_execution\x18\x04 \x01(\x0b\x32..google.cloud.aiplatform.v1.Tool.CodeExecutionB\x03\xe0\x41\x01\x12@\n\x0burl_context\x18\x08 \x01(\x0b\x32&.google.cloud.aiplatform.v1.UrlContextB\x03\xe0\x41\x01\x1a\x0e\n\x0cGoogleSearch\x1a\x0f\n\rCodeExecution\"\x0c\n\nUrlContext\"\xb2\x02\n\x13\x46unctionDeclaration\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x18\n\x0b\x64\x65scription\x18\x02 \x01(\tB\x03\xe0\x41\x01\x12;\n\nparameters\x18\x03 \x01(\x0b\x32\".google.cloud.aiplatform.v1.SchemaB\x03\xe0\x41\x01\x12;\n\x16parameters_json_schema\x18\x05 \x01(\x0b\x32\x16.google.protobuf.ValueB\x03\xe0\x41\x01\x12\x39\n\x08response\x18\x04 \x01(\x0b\x32\".google.cloud.aiplatform.v1.SchemaB\x03\xe0\x41\x01\x12\x39\n\x14response_json_schema\x18\x06 \x01(\x0b\x32\x16.google.protobuf.ValueB\x03\xe0\x41\x01\"M\n\x0c\x46unctionCall\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12*\n\x04\x61rgs\x18\x02 \x01(\x0b\x32\x17.google.protobuf.StructB\x03\xe0\x41\x01\"U\n\x10\x46unctionResponse\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12.\n\x08response\x18\x02 \x01(\x0b\x32\x17.google.protobuf.StructB\x03\xe0\x41\x02\"\xa1\x01\n\x0e\x45xecutableCode\x12J\n\x08language\x18\x01 \x01(\x0e\x32\x33.google.cloud.aiplatform.v1.ExecutableCode.LanguageB\x03\xe0\x41\x02\x12\x11\n\x04\x63ode\x18\x02 \x01(\tB\x03\xe0\x41\x02\"0\n\x08Language\x12\x18\n\x14LANGUAGE_UNSPECIFIED\x10\x00\x12\n\n\x06PYTHON\x10\x01\"\xe0\x01\n\x13\x43odeExecutionResult\x12M\n\x07outcome\x18\x01 \x01(\x0e\x32\x37.google.cloud.aiplatform.v1.CodeExecutionResult.OutcomeB\x03\xe0\x41\x02\x12\x13\n\x06output\x18\x02 \x01(\tB\x03\xe0\x41\x01\"e\n\x07Outcome\x12\x17\n\x13OUTCOME_UNSPECIFIED\x10\x00\x12\x0e\n\nOUTCOME_OK\x10\x01\x12\x12\n\x0eOUTCOME_FAILED\x10\x02\x12\x1d\n\x19OUTCOME_DEADLINE_EXCEEDED\x10\x03\"\xc9\x01\n\tRetrieval\x12\x46\n\x10vertex_ai_search\x18\x02 \x01(\x0b\x32*.google.cloud.aiplatform.v1.VertexAISearchH\x00\x12\x46\n\x10vertex_rag_store\x18\x04 \x01(\x0b\x32*.google.cloud.aiplatform.v1.VertexRagStoreH\x00\x12\"\n\x13\x64isable_attribution\x18\x03 \x01(\x08\x42\x05\x18\x01\xe0\x41\x01\x42\x08\n\x06source\"\xaa\x03\n\x0eVertexRagStore\x12R\n\rrag_resources\x18\x04 \x03(\x0b\x32\x36.google.cloud.aiplatform.v1.VertexRagStore.RagResourceB\x03\xe0\x41\x01\x12$\n\x10similarity_top_k\x18\x02 \x01(\x05\x42\x05\x18\x01\xe0\x41\x01H\x00\x88\x01\x01\x12-\n\x19vector_distance_threshold\x18\x03 \x01(\x01\x42\x05\x18\x01\xe0\x41\x01H\x01\x88\x01\x01\x12Q\n\x14rag_retrieval_config\x18\x06 \x01(\x0b\x32..google.cloud.aiplatform.v1.RagRetrievalConfigB\x03\xe0\x41\x01\x1ai\n\x0bRagResource\x12?\n\nrag_corpus\x18\x01 \x01(\tB+\xe0\x41\x01\xfa\x41%\n#aiplatform.googleapis.com/RagCorpus\x12\x19\n\x0crag_file_ids\x18\x02 \x03(\tB\x03\xe0\x41\x01\x42\x13\n\x11_similarity_top_kB\x1c\n\x1a_vector_distance_threshold\"\xfa\x01\n\x0eVertexAISearch\x12\x16\n\tdatastore\x18\x01 \x01(\tB\x03\xe0\x41\x01\x12\x13\n\x06\x65ngine\x18\x02 \x01(\tB\x03\xe0\x41\x01\x12\x18\n\x0bmax_results\x18\x03 \x01(\x05\x42\x03\xe0\x41\x01\x12\x13\n\x06\x66ilter\x18\x04 \x01(\tB\x03\xe0\x41\x01\x12R\n\x10\x64\x61ta_store_specs\x18\x05 \x03(\x0b\x32\x38.google.cloud.aiplatform.v1.VertexAISearch.DataStoreSpec\x1a\x38\n\rDataStoreSpec\x12\x12\n\ndata_store\x18\x01 \x01(\t\x12\x13\n\x06\x66ilter\x18\x02 \x01(\tB\x03\xe0\x41\x01\"m\n\x15GoogleSearchRetrieval\x12T\n\x18\x64ynamic_retrieval_config\x18\x02 \x01(\x0b\x32\x32.google.cloud.aiplatform.v1.DynamicRetrievalConfig\"\x15\n\x13\x45nterpriseWebSearch\"\xca\x01\n\x16\x44ynamicRetrievalConfig\x12\x45\n\x04mode\x18\x01 \x01(\x0e\x32\x37.google.cloud.aiplatform.v1.DynamicRetrievalConfig.Mode\x12#\n\x11\x64ynamic_threshold\x18\x02 \x01(\x02\x42\x03\xe0\x41\x01H\x00\x88\x01\x01\".\n\x04Mode\x12\x14\n\x10MODE_UNSPECIFIED\x10\x00\x12\x10\n\x0cMODE_DYNAMIC\x10\x01\x42\x14\n\x12_dynamic_threshold\"\xb1\x01\n\nToolConfig\x12W\n\x17\x66unction_calling_config\x18\x01 \x01(\x0b\x32\x31.google.cloud.aiplatform.v1.FunctionCallingConfigB\x03\xe0\x41\x01\x12J\n\x10retrieval_config\x18\x02 \x01(\x0b\x32+.google.cloud.aiplatform.v1.RetrievalConfigB\x03\xe0\x41\x01\"\xc2\x01\n\x15\x46unctionCallingConfig\x12I\n\x04mode\x18\x01 \x01(\x0e\x32\x36.google.cloud.aiplatform.v1.FunctionCallingConfig.ModeB\x03\xe0\x41\x01\x12#\n\x16\x61llowed_function_names\x18\x02 \x03(\tB\x03\xe0\x41\x01\"9\n\x04Mode\x12\x14\n\x10MODE_UNSPECIFIED\x10\x00\x12\x08\n\x04\x41UTO\x10\x01\x12\x07\n\x03\x41NY\x10\x02\x12\x08\n\x04NONE\x10\x03\"v\n\x0fRetrievalConfig\x12)\n\x07lat_lng\x18\x01 \x01(\x0b\x32\x13.google.type.LatLngH\x00\x88\x01\x01\x12\x1a\n\rlanguage_code\x18\x02 \x01(\tH\x01\x88\x01\x01\x42\n\n\x08_lat_lngB\x10\n\x0e_language_code\"\xaa\x05\n\x12RagRetrievalConfig\x12\x12\n\x05top_k\x18\x01 \x01(\x05\x42\x03\xe0\x41\x01\x12J\n\x06\x66ilter\x18\x03 \x01(\x0b\x32\x35.google.cloud.aiplatform.v1.RagRetrievalConfig.FilterB\x03\xe0\x41\x01\x12L\n\x07ranking\x18\x04 \x01(\x0b\x32\x36.google.cloud.aiplatform.v1.RagRetrievalConfig.RankingB\x03\xe0\x41\x01\x1a\x93\x01\n\x06\x46ilter\x12(\n\x19vector_distance_threshold\x18\x03 \x01(\x01\x42\x03\xe0\x41\x01H\x00\x12*\n\x1bvector_similarity_threshold\x18\x04 \x01(\x01\x42\x03\xe0\x41\x01H\x00\x12\x1c\n\x0fmetadata_filter\x18\x02 \x01(\tB\x03\xe0\x41\x01\x42\x15\n\x13vector_db_threshold\x1a\xcf\x02\n\x07Ranking\x12_\n\x0crank_service\x18\x01 \x01(\x0b\x32\x42.google.cloud.aiplatform.v1.RagRetrievalConfig.Ranking.RankServiceB\x03\xe0\x41\x01H\x00\x12[\n\nllm_ranker\x18\x03 \x01(\x0b\x32@.google.cloud.aiplatform.v1.RagRetrievalConfig.Ranking.LlmRankerB\x03\xe0\x41\x01H\x00\x1a:\n\x0bRankService\x12\x1c\n\nmodel_name\x18\x01 \x01(\tB\x03\xe0\x41\x01H\x00\x88\x01\x01\x42\r\n\x0b_model_name\x1a\x38\n\tLlmRanker\x12\x1c\n\nmodel_name\x18\x01 \x01(\tB\x03\xe0\x41\x01H\x00\x88\x01\x01\x42\r\n\x0b_model_nameB\x10\n\x0eranking_configB\xc7\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\tToolProtoP\x01Z>cloud.google.com/go/aiplatform/apiv1/aiplatformpb;aiplatformpb\xaa\x02\x1aGoogle.Cloud.AIPlatform.V1\xca\x02\x1aGoogle\\Cloud\\AIPlatform\\V1\xea\x02\x1dGoogle::Cloud::AIPlatform::V1b\x06proto3"
15
15
 
16
16
  pool = Google::Protobuf::DescriptorPool.generated_pool
17
17
 
@@ -27,7 +27,7 @@ rescue TypeError
27
27
  warn "Warning: Protobuf detected an import path issue while loading generated file #{__FILE__}"
28
28
  imports = [
29
29
  ["google.cloud.aiplatform.v1.Schema", "google/cloud/aiplatform/v1/openapi.proto"],
30
- ["google.protobuf.Struct", "google/protobuf/struct.proto"],
30
+ ["google.protobuf.Value", "google/protobuf/struct.proto"],
31
31
  ["google.type.LatLng", "google/type/latlng.proto"],
32
32
  ]
33
33
  imports.each do |type_name, expected_filename|
@@ -47,6 +47,7 @@ module Google
47
47
  Tool = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.Tool").msgclass
48
48
  Tool::GoogleSearch = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.Tool.GoogleSearch").msgclass
49
49
  Tool::CodeExecution = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.Tool.CodeExecution").msgclass
50
+ UrlContext = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.UrlContext").msgclass
50
51
  FunctionDeclaration = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.FunctionDeclaration").msgclass
51
52
  FunctionCall = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.FunctionCall").msgclass
52
53
  FunctionResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.FunctionResponse").msgclass