google-cloud-ai_platform-v1 0.57.0 → 0.59.0

data/lib/google/cloud/aiplatform/v1/vertex_rag_data_service_pb.rb

@@ -0,0 +1,71 @@
+# frozen_string_literal: true
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: google/cloud/aiplatform/v1/vertex_rag_data_service.proto
+
+require 'google/protobuf'
+
+require 'google/api/annotations_pb'
+require 'google/api/client_pb'
+require 'google/api/field_behavior_pb'
+require 'google/api/resource_pb'
+require 'google/cloud/aiplatform/v1/operation_pb'
+require 'google/cloud/aiplatform/v1/vertex_rag_data_pb'
+require 'google/longrunning/operations_pb'
+require 'google/protobuf/empty_pb'
+require 'google/rpc/status_pb'
+
+
+descriptor_data = "\n8google/cloud/aiplatform/v1/vertex_rag_data_service.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1cgoogle/api/annotations.proto\x1a\x17google/api/client.proto\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a*google/cloud/aiplatform/v1/operation.proto\x1a\x30google/cloud/aiplatform/v1/vertex_rag_data.proto\x1a#google/longrunning/operations.proto\x1a\x1bgoogle/protobuf/empty.proto\x1a\x17google/rpc/status.proto\"\x93\x01\n\x16\x43reateRagCorpusRequest\x12\x39\n\x06parent\x18\x01 \x01(\tB)\xe0\x41\x02\xfa\x41#\n!locations.googleapis.com/Location\x12>\n\nrag_corpus\x18\x02 \x01(\x0b\x32%.google.cloud.aiplatform.v1.RagCorpusB\x03\xe0\x41\x02\"P\n\x13GetRagCorpusRequest\x12\x39\n\x04name\x18\x01 \x01(\tB+\xe0\x41\x02\xfa\x41%\n#aiplatform.googleapis.com/RagCorpus\"\x83\x01\n\x15ListRagCorporaRequest\x12\x39\n\x06parent\x18\x01 \x01(\tB)\xe0\x41\x02\xfa\x41#\n!locations.googleapis.com/Location\x12\x16\n\tpage_size\x18\x02 \x01(\x05\x42\x03\xe0\x41\x01\x12\x17\n\npage_token\x18\x03 \x01(\tB\x03\xe0\x41\x01\"m\n\x16ListRagCorporaResponse\x12:\n\x0brag_corpora\x18\x01 \x03(\x0b\x32%.google.cloud.aiplatform.v1.RagCorpus\x12\x17\n\x0fnext_page_token\x18\x02 \x01(\t\"g\n\x16\x44\x65leteRagCorpusRequest\x12\x39\n\x04name\x18\x01 \x01(\tB+\xe0\x41\x02\xfa\x41%\n#aiplatform.googleapis.com/RagCorpus\x12\x12\n\x05\x66orce\x18\x02 \x01(\x08\x42\x03\xe0\x41\x01\"\xe5\x01\n\x14UploadRagFileRequest\x12;\n\x06parent\x18\x01 \x01(\tB+\xe0\x41\x02\xfa\x41%\n#aiplatform.googleapis.com/RagCorpus\x12:\n\x08rag_file\x18\x02 \x01(\x0b\x32#.google.cloud.aiplatform.v1.RagFileB\x03\xe0\x41\x02\x12T\n\x16upload_rag_file_config\x18\x05 \x01(\x0b\x32/.google.cloud.aiplatform.v1.UploadRagFileConfigB\x03\xe0\x41\x02\"\x7f\n\x15UploadRagFileResponse\x12\x37\n\x08rag_file\x18\x01 \x01(\x0b\x32#.google.cloud.aiplatform.v1.RagFileH\x00\x12#\n\x05\x65rror\x18\x04 \x01(\x0b\x32\x12.google.rpc.StatusH\x00\x42\x08\n\x06result\"\xac\x01\n\x15ImportRagFilesRequest\x12;\n\x06parent\x18\x01 \x01(\tB+\xe0\x41\x02\xfa\x41%\n#aiplatform.googleapis.com/RagCorpus\x12V\n\x17import_rag_files_config\x18\x02 \x01(\x0b\x32\x30.google.cloud.aiplatform.v1.ImportRagFilesConfigB\x03\xe0\x41\x02\"\xe3\x01\n\x16ImportRagFilesResponse\x12#\n\x19partial_failures_gcs_path\x18\x04 \x01(\tH\x00\x12)\n\x1fpartial_failures_bigquery_table\x18\x05 \x01(\tH\x00\x12 \n\x18imported_rag_files_count\x18\x01 \x01(\x03\x12\x1e\n\x16\x66\x61iled_rag_files_count\x18\x02 \x01(\x03\x12\x1f\n\x17skipped_rag_files_count\x18\x03 \x01(\x03\x42\x16\n\x14partial_failure_sink\"L\n\x11GetRagFileRequest\x12\x37\n\x04name\x18\x01 \x01(\tB)\xe0\x41\x02\xfa\x41#\n!aiplatform.googleapis.com/RagFile\"\x83\x01\n\x13ListRagFilesRequest\x12;\n\x06parent\x18\x01 \x01(\tB+\xe0\x41\x02\xfa\x41%\n#aiplatform.googleapis.com/RagCorpus\x12\x16\n\tpage_size\x18\x02 \x01(\x05\x42\x03\xe0\x41\x01\x12\x17\n\npage_token\x18\x03 \x01(\tB\x03\xe0\x41\x01\"g\n\x14ListRagFilesResponse\x12\x36\n\trag_files\x18\x01 \x03(\x0b\x32#.google.cloud.aiplatform.v1.RagFile\x12\x17\n\x0fnext_page_token\x18\x02 \x01(\t\"O\n\x14\x44\x65leteRagFileRequest\x12\x37\n\x04name\x18\x01 \x01(\tB)\xe0\x41\x02\xfa\x41#\n!aiplatform.googleapis.com/RagFile\"r\n CreateRagCorpusOperationMetadata\x12N\n\x10generic_metadata\x18\x01 \x01(\x0b\x32\x34.google.cloud.aiplatform.v1.GenericOperationMetadata\"X\n\x16UpdateRagCorpusRequest\x12>\n\nrag_corpus\x18\x01 \x01(\x0b\x32%.google.cloud.aiplatform.v1.RagCorpusB\x03\xe0\x41\x02\"r\n UpdateRagCorpusOperationMetadata\x12N\n\x10generic_metadata\x18\x01 \x01(\x0b\x32\x34.google.cloud.aiplatform.v1.GenericOperationMetadata\"\xfd\x01\n\x1fImportRagFilesOperationMetadata\x12N\n\x10generic_metadata\x18\x01 \x01(\x0b\x32\x34.google.cloud.aiplatform.v1.GenericOperationMetadata\x12\x15\n\rrag_corpus_id\x18\x02 \x01(\x03\x12V\n\x17import_rag_files_config\x18\x03 \x01(\x0b\x32\x30.google.cloud.aiplatform.v1.ImportRagFilesConfigB\x03\xe0\x41\x03\x12\x1b\n\x13progress_percentage\x18\x04 \x01(\x05\x32\xcf\x11\n\x14VertexRagDataService\x12\xed\x01\n\x0f\x43reateRagCorpus\x12\x32.google.cloud.aiplatform.v1.CreateRagCorpusRequest\x1a\x1d.google.longrunning.Operation\"\x86\x01\xca\x41-\n\tRagCorpus\x12 CreateRagCorpusOperationMetadata\xda\x41\x11parent,rag_corpus\x82\xd3\xe4\x93\x02<\"./v1/{parent=projects/*/locations/*}/ragCorpora:\nrag_corpus\x12\xf1\x01\n\x0fUpdateRagCorpus\x12\x32.google.cloud.aiplatform.v1.UpdateRagCorpusRequest\x1a\x1d.google.longrunning.Operation\"\x8a\x01\xca\x41-\n\tRagCorpus\x12 UpdateRagCorpusOperationMetadata\xda\x41\nrag_corpus\x82\xd3\xe4\x93\x02G29/v1/{rag_corpus.name=projects/*/locations/*/ragCorpora/*}:\nrag_corpus\x12\xa5\x01\n\x0cGetRagCorpus\x12/.google.cloud.aiplatform.v1.GetRagCorpusRequest\x1a%.google.cloud.aiplatform.v1.RagCorpus\"=\xda\x41\x04name\x82\xd3\xe4\x93\x02\x30\x12./v1/{name=projects/*/locations/*/ragCorpora/*}\x12\xb8\x01\n\x0eListRagCorpora\x12\x31.google.cloud.aiplatform.v1.ListRagCorporaRequest\x1a\x32.google.cloud.aiplatform.v1.ListRagCorporaResponse\"?\xda\x41\x06parent\x82\xd3\xe4\x93\x02\x30\x12./v1/{parent=projects/*/locations/*}/ragCorpora\x12\xd6\x01\n\x0f\x44\x65leteRagCorpus\x12\x32.google.cloud.aiplatform.v1.DeleteRagCorpusRequest\x1a\x1d.google.longrunning.Operation\"p\xca\x41\x30\n\x15google.protobuf.Empty\x12\x17\x44\x65leteOperationMetadata\xda\x41\x04name\x82\xd3\xe4\x93\x02\x30*./v1/{name=projects/*/locations/*/ragCorpora/*}\x12\xea\x01\n\rUploadRagFile\x12\x30.google.cloud.aiplatform.v1.UploadRagFileRequest\x1a\x31.google.cloud.aiplatform.v1.UploadRagFileResponse\"t\xda\x41&parent,rag_file,upload_rag_file_config\x82\xd3\xe4\x93\x02\x45\"@/v1/{parent=projects/*/locations/*/ragCorpora/*}/ragFiles:upload:\x01*\x12\x8d\x02\n\x0eImportRagFiles\x12\x31.google.cloud.aiplatform.v1.ImportRagFilesRequest\x1a\x1d.google.longrunning.Operation\"\xa8\x01\xca\x41\x39\n\x16ImportRagFilesResponse\x12\x1fImportRagFilesOperationMetadata\xda\x41\x1eparent,import_rag_files_config\x82\xd3\xe4\x93\x02\x45\"@/v1/{parent=projects/*/locations/*/ragCorpora/*}/ragFiles:import:\x01*\x12\xaa\x01\n\nGetRagFile\x12-.google.cloud.aiplatform.v1.GetRagFileRequest\x1a#.google.cloud.aiplatform.v1.RagFile\"H\xda\x41\x04name\x82\xd3\xe4\x93\x02;\x12\x39/v1/{name=projects/*/locations/*/ragCorpora/*/ragFiles/*}\x12\xbd\x01\n\x0cListRagFiles\x12/.google.cloud.aiplatform.v1.ListRagFilesRequest\x1a\x30.google.cloud.aiplatform.v1.ListRagFilesResponse\"J\xda\x41\x06parent\x82\xd3\xe4\x93\x02;\x12\x39/v1/{parent=projects/*/locations/*/ragCorpora/*}/ragFiles\x12\xdd\x01\n\rDeleteRagFile\x12\x30.google.cloud.aiplatform.v1.DeleteRagFileRequest\x1a\x1d.google.longrunning.Operation\"{\xca\x41\x30\n\x15google.protobuf.Empty\x12\x17\x44\x65leteOperationMetadata\xda\x41\x04name\x82\xd3\xe4\x93\x02;*9/v1/{name=projects/*/locations/*/ragCorpora/*/ragFiles/*}\x1aM\xca\x41\x19\x61iplatform.googleapis.com\xd2\x41.https://www.googleapis.com/auth/cloud-platformB\xd7\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\x19VertexRagDataServiceProtoP\x01Z>cloud.google.com/go/aiplatform/apiv1/aiplatformpb;aiplatformpb\xaa\x02\x1aGoogle.Cloud.AIPlatform.V1\xca\x02\x1aGoogle\\Cloud\\AIPlatform\\V1\xea\x02\x1dGoogle::Cloud::AIPlatform::V1b\x06proto3"
+
+pool = Google::Protobuf::DescriptorPool.generated_pool
+
+begin
+  pool.add_serialized_file(descriptor_data)
+rescue TypeError
+  # Compatibility code: will be removed in the next major version.
+  require 'google/protobuf/descriptor_pb'
+  parsed = Google::Protobuf::FileDescriptorProto.decode(descriptor_data)
+  parsed.clear_dependency
+  serialized = parsed.class.encode(parsed)
+  file = pool.add_serialized_file(serialized)
+  warn "Warning: Protobuf detected an import path issue while loading generated file #{__FILE__}"
+  imports = [
+    ["google.cloud.aiplatform.v1.RagCorpus", "google/cloud/aiplatform/v1/vertex_rag_data.proto"],
+    ["google.rpc.Status", "google/rpc/status.proto"],
+    ["google.cloud.aiplatform.v1.GenericOperationMetadata", "google/cloud/aiplatform/v1/operation.proto"],
+  ]
+  imports.each do |type_name, expected_filename|
+    import_file = pool.lookup(type_name).file_descriptor
+    if import_file.name != expected_filename
+      warn "- #{file.name} imports #{expected_filename}, but that import was loaded as #{import_file.name}"
+    end
+  end
+  warn "Each proto file must use a consistent fully-qualified name."
+  warn "This will become an error in the next major version."
+end
+
+module Google
+  module Cloud
+    module AIPlatform
+      module V1
+        CreateRagCorpusRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.CreateRagCorpusRequest").msgclass
+        GetRagCorpusRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.GetRagCorpusRequest").msgclass
+        ListRagCorporaRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ListRagCorporaRequest").msgclass
+        ListRagCorporaResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ListRagCorporaResponse").msgclass
+        DeleteRagCorpusRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.DeleteRagCorpusRequest").msgclass
+        UploadRagFileRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.UploadRagFileRequest").msgclass
+        UploadRagFileResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.UploadRagFileResponse").msgclass
+        ImportRagFilesRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ImportRagFilesRequest").msgclass
+        ImportRagFilesResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ImportRagFilesResponse").msgclass
+        GetRagFileRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.GetRagFileRequest").msgclass
+        ListRagFilesRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ListRagFilesRequest").msgclass
+        ListRagFilesResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ListRagFilesResponse").msgclass
+        DeleteRagFileRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.DeleteRagFileRequest").msgclass
+        CreateRagCorpusOperationMetadata = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.CreateRagCorpusOperationMetadata").msgclass
+        UpdateRagCorpusRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.UpdateRagCorpusRequest").msgclass
+        UpdateRagCorpusOperationMetadata = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.UpdateRagCorpusOperationMetadata").msgclass
+        ImportRagFilesOperationMetadata = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ImportRagFilesOperationMetadata").msgclass
+      end
+    end
+  end
+end

data/lib/google/cloud/aiplatform/v1/vertex_rag_data_service_services_pb.rb

@@ -0,0 +1,63 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# Source: google/cloud/aiplatform/v1/vertex_rag_data_service.proto for package 'Google.Cloud.AIPlatform.V1'
+# Original file comments:
+# Copyright 2024 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require 'grpc'
+require 'google/cloud/aiplatform/v1/vertex_rag_data_service_pb'
+
+module Google
+  module Cloud
+    module AIPlatform
+      module V1
+        module VertexRagDataService
+          # A service for managing user data for RAG.
+          class Service
+
+            include ::GRPC::GenericService
+
+            self.marshal_class_method = :encode
+            self.unmarshal_class_method = :decode
+            self.service_name = 'google.cloud.aiplatform.v1.VertexRagDataService'
+
+            # Creates a RagCorpus.
+            rpc :CreateRagCorpus, ::Google::Cloud::AIPlatform::V1::CreateRagCorpusRequest, ::Google::Longrunning::Operation
+            # Updates a RagCorpus.
+            rpc :UpdateRagCorpus, ::Google::Cloud::AIPlatform::V1::UpdateRagCorpusRequest, ::Google::Longrunning::Operation
+            # Gets a RagCorpus.
+            rpc :GetRagCorpus, ::Google::Cloud::AIPlatform::V1::GetRagCorpusRequest, ::Google::Cloud::AIPlatform::V1::RagCorpus
+            # Lists RagCorpora in a Location.
+            rpc :ListRagCorpora, ::Google::Cloud::AIPlatform::V1::ListRagCorporaRequest, ::Google::Cloud::AIPlatform::V1::ListRagCorporaResponse
+            # Deletes a RagCorpus.
+            rpc :DeleteRagCorpus, ::Google::Cloud::AIPlatform::V1::DeleteRagCorpusRequest, ::Google::Longrunning::Operation
+            # Upload a file into a RagCorpus.
+            rpc :UploadRagFile, ::Google::Cloud::AIPlatform::V1::UploadRagFileRequest, ::Google::Cloud::AIPlatform::V1::UploadRagFileResponse
+            # Import files from Google Cloud Storage or Google Drive into a RagCorpus.
+            rpc :ImportRagFiles, ::Google::Cloud::AIPlatform::V1::ImportRagFilesRequest, ::Google::Longrunning::Operation
+            # Gets a RagFile.
+            rpc :GetRagFile, ::Google::Cloud::AIPlatform::V1::GetRagFileRequest, ::Google::Cloud::AIPlatform::V1::RagFile
+            # Lists RagFiles in a RagCorpus.
+            rpc :ListRagFiles, ::Google::Cloud::AIPlatform::V1::ListRagFilesRequest, ::Google::Cloud::AIPlatform::V1::ListRagFilesResponse
+            # Deletes a RagFile.
+            rpc :DeleteRagFile, ::Google::Cloud::AIPlatform::V1::DeleteRagFileRequest, ::Google::Longrunning::Operation
+          end
+
+          Stub = Service.rpc_stub_class
+        end
+      end
+    end
+  end
+end

data/lib/google/cloud/aiplatform/v1/vertex_rag_service_pb.rb

@@ -0,0 +1,65 @@
+# frozen_string_literal: true
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: google/cloud/aiplatform/v1/vertex_rag_service.proto
+
+require 'google/protobuf'
+
+require 'google/api/annotations_pb'
+require 'google/api/client_pb'
+require 'google/api/field_behavior_pb'
+require 'google/api/resource_pb'
+require 'google/cloud/aiplatform/v1/content_pb'
+require 'google/cloud/aiplatform/v1/tool_pb'
+
+
+descriptor_data = "\n3google/cloud/aiplatform/v1/vertex_rag_service.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1cgoogle/api/annotations.proto\x1a\x17google/api/client.proto\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a(google/cloud/aiplatform/v1/content.proto\x1a%google/cloud/aiplatform/v1/tool.proto\"{\n\x08RagQuery\x12\x13\n\x04text\x18\x01 \x01(\tB\x03\xe0\x41\x01H\x00\x12Q\n\x14rag_retrieval_config\x18\x06 \x01(\x0b\x32..google.cloud.aiplatform.v1.RagRetrievalConfigB\x03\xe0\x41\x01\x42\x07\n\x05query\"\xb4\x04\n\x17RetrieveContextsRequest\x12^\n\x10vertex_rag_store\x18\x02 \x01(\x0b\x32\x42.google.cloud.aiplatform.v1.RetrieveContextsRequest.VertexRagStoreH\x00\x12\x39\n\x06parent\x18\x01 \x01(\tB)\xe0\x41\x02\xfa\x41#\n!locations.googleapis.com/Location\x12\x38\n\x05query\x18\x03 \x01(\x0b\x32$.google.cloud.aiplatform.v1.RagQueryB\x03\xe0\x41\x02\x1a\xb4\x02\n\x0eVertexRagStore\x12j\n\rrag_resources\x18\x03 \x03(\x0b\x32N.google.cloud.aiplatform.v1.RetrieveContextsRequest.VertexRagStore.RagResourceB\x03\xe0\x41\x01\x12-\n\x19vector_distance_threshold\x18\x02 \x01(\x01\x42\x05\x18\x01\xe0\x41\x01H\x00\x88\x01\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\x1c\n\x1a_vector_distance_thresholdB\r\n\x0b\x64\x61ta_source\"\xb8\x01\n\x0bRagContexts\x12\x41\n\x08\x63ontexts\x18\x01 \x03(\x0b\x32/.google.cloud.aiplatform.v1.RagContexts.Context\x1a\x66\n\x07\x43ontext\x12\x12\n\nsource_uri\x18\x01 \x01(\t\x12\x1b\n\x13source_display_name\x18\x05 \x01(\t\x12\x0c\n\x04text\x18\x02 \x01(\t\x12\x12\n\x05score\x18\x06 \x01(\x01H\x00\x88\x01\x01\x42\x08\n\x06_score\"U\n\x18RetrieveContextsResponse\x12\x39\n\x08\x63ontexts\x18\x01 \x01(\x0b\x32\'.google.cloud.aiplatform.v1.RagContexts\"\xee\x02\n\x14\x41ugmentPromptRequest\x12K\n\x10vertex_rag_store\x18\x04 \x01(\x0b\x32*.google.cloud.aiplatform.v1.VertexRagStoreB\x03\xe0\x41\x01H\x00\x12\x39\n\x06parent\x18\x01 \x01(\tB)\xe0\x41\x02\xfa\x41#\n!locations.googleapis.com/Location\x12:\n\x08\x63ontents\x18\x02 \x03(\x0b\x32#.google.cloud.aiplatform.v1.ContentB\x03\xe0\x41\x01\x12J\n\x05model\x18\x03 \x01(\x0b\x32\x36.google.cloud.aiplatform.v1.AugmentPromptRequest.ModelB\x03\xe0\x41\x01\x1a\x37\n\x05Model\x12\x12\n\x05model\x18\x01 \x01(\tB\x03\xe0\x41\x01\x12\x1a\n\rmodel_version\x18\x02 \x01(\tB\x03\xe0\x41\x01\x42\r\n\x0b\x64\x61ta_source\"\x87\x01\n\x15\x41ugmentPromptResponse\x12=\n\x10\x61ugmented_prompt\x18\x01 \x03(\x0b\x32#.google.cloud.aiplatform.v1.Content\x12/\n\x05\x66\x61\x63ts\x18\x02 \x03(\x0b\x32 .google.cloud.aiplatform.v1.Fact\"\xe2\x02\n\x19\x43orroborateContentRequest\x12\x39\n\x06parent\x18\x01 \x01(\tB)\xe0\x41\x02\xfa\x41#\n!locations.googleapis.com/Location\x12>\n\x07\x63ontent\x18\x02 \x01(\x0b\x32#.google.cloud.aiplatform.v1.ContentB\x03\xe0\x41\x01H\x00\x88\x01\x01\x12\x34\n\x05\x66\x61\x63ts\x18\x03 \x03(\x0b\x32 .google.cloud.aiplatform.v1.FactB\x03\xe0\x41\x01\x12Y\n\nparameters\x18\x04 \x01(\x0b\x32@.google.cloud.aiplatform.v1.CorroborateContentRequest.ParametersB\x03\xe0\x41\x01\x1a-\n\nParameters\x12\x1f\n\x12\x63itation_threshold\x18\x01 \x01(\x01\x42\x03\xe0\x41\x01\x42\n\n\x08_content\"\x89\x01\n\x1a\x43orroborateContentResponse\x12 \n\x13\x63orroboration_score\x18\x01 \x01(\x02H\x00\x88\x01\x01\x12\x31\n\x06\x63laims\x18\x02 \x03(\x0b\x32!.google.cloud.aiplatform.v1.ClaimB\x16\n\x14_corroboration_score\"\xd2\x01\n\x04\x46\x61\x63t\x12\x12\n\x05query\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x12\n\x05title\x18\x02 \x01(\tH\x01\x88\x01\x01\x12\x10\n\x03uri\x18\x03 \x01(\tH\x02\x88\x01\x01\x12\x14\n\x07summary\x18\x04 \x01(\tH\x03\x88\x01\x01\x12 \n\x0fvector_distance\x18\x05 \x01(\x01\x42\x02\x18\x01H\x04\x88\x01\x01\x12\x12\n\x05score\x18\x06 \x01(\x01H\x05\x88\x01\x01\x42\x08\n\x06_queryB\x08\n\x06_titleB\x06\n\x04_uriB\n\n\x08_summaryB\x12\n\x10_vector_distanceB\x08\n\x06_score\"\x8b\x01\n\x05\x43laim\x12\x18\n\x0bstart_index\x18\x01 \x01(\x05H\x00\x88\x01\x01\x12\x16\n\tend_index\x18\x02 \x01(\x05H\x01\x88\x01\x01\x12\x14\n\x0c\x66\x61\x63t_indexes\x18\x03 \x03(\x05\x12\x12\n\x05score\x18\x04 \x01(\x02H\x02\x88\x01\x01\x42\x0e\n\x0c_start_indexB\x0c\n\n_end_indexB\x08\n\x06_score2\xe6\x05\n\x10VertexRagService\x12\xcd\x01\n\x10RetrieveContexts\x12\x33.google.cloud.aiplatform.v1.RetrieveContextsRequest\x1a\x34.google.cloud.aiplatform.v1.RetrieveContextsResponse\"N\xda\x41\x0cparent,query\x82\xd3\xe4\x93\x02\x39\"4/v1/{parent=projects/*/locations/*}:retrieveContexts:\x01*\x12\xd2\x01\n\rAugmentPrompt\x12\x30.google.cloud.aiplatform.v1.AugmentPromptRequest\x1a\x31.google.cloud.aiplatform.v1.AugmentPromptResponse\"\\\xda\x41\x1dparent,model,vertex_rag_store\x82\xd3\xe4\x93\x02\x36\"1/v1/{parent=projects/*/locations/*}:augmentPrompt:\x01*\x12\xdd\x01\n\x12\x43orroborateContent\x12\x35.google.cloud.aiplatform.v1.CorroborateContentRequest\x1a\x36.google.cloud.aiplatform.v1.CorroborateContentResponse\"X\xda\x41\x14parent,content,facts\x82\xd3\xe4\x93\x02;\"6/v1/{parent=projects/*/locations/*}:corroborateContent:\x01*\x1aM\xca\x41\x19\x61iplatform.googleapis.com\xd2\x41.https://www.googleapis.com/auth/cloud-platformB\xd3\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\x15VertexRagServiceProtoP\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"
+
+pool = Google::Protobuf::DescriptorPool.generated_pool
+
+begin
+  pool.add_serialized_file(descriptor_data)
+rescue TypeError
+  # Compatibility code: will be removed in the next major version.
+  require 'google/protobuf/descriptor_pb'
+  parsed = Google::Protobuf::FileDescriptorProto.decode(descriptor_data)
+  parsed.clear_dependency
+  serialized = parsed.class.encode(parsed)
+  file = pool.add_serialized_file(serialized)
+  warn "Warning: Protobuf detected an import path issue while loading generated file #{__FILE__}"
+  imports = [
+    ["google.cloud.aiplatform.v1.RagRetrievalConfig", "google/cloud/aiplatform/v1/tool.proto"],
+    ["google.cloud.aiplatform.v1.Content", "google/cloud/aiplatform/v1/content.proto"],
+  ]
+  imports.each do |type_name, expected_filename|
+    import_file = pool.lookup(type_name).file_descriptor
+    if import_file.name != expected_filename
+      warn "- #{file.name} imports #{expected_filename}, but that import was loaded as #{import_file.name}"
+    end
+  end
+  warn "Each proto file must use a consistent fully-qualified name."
+  warn "This will become an error in the next major version."
+end
+
+module Google
+  module Cloud
+    module AIPlatform
+      module V1
+        RagQuery = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.RagQuery").msgclass
+        RetrieveContextsRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.RetrieveContextsRequest").msgclass
+        RetrieveContextsRequest::VertexRagStore = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.RetrieveContextsRequest.VertexRagStore").msgclass
+        RetrieveContextsRequest::VertexRagStore::RagResource = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.RetrieveContextsRequest.VertexRagStore.RagResource").msgclass
+        RagContexts = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.RagContexts").msgclass
+        RagContexts::Context = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.RagContexts.Context").msgclass
+        RetrieveContextsResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.RetrieveContextsResponse").msgclass
+        AugmentPromptRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.AugmentPromptRequest").msgclass
+        AugmentPromptRequest::Model = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.AugmentPromptRequest.Model").msgclass
+        AugmentPromptResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.AugmentPromptResponse").msgclass
+        CorroborateContentRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.CorroborateContentRequest").msgclass
+        CorroborateContentRequest::Parameters = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.CorroborateContentRequest.Parameters").msgclass
+        CorroborateContentResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.CorroborateContentResponse").msgclass
+        Fact = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.Fact").msgclass
+        Claim = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.Claim").msgclass
+      end
+    end
+  end
+end

data/lib/google/cloud/aiplatform/v1/vertex_rag_service_services_pb.rb

@@ -0,0 +1,52 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# Source: google/cloud/aiplatform/v1/vertex_rag_service.proto for package 'Google.Cloud.AIPlatform.V1'
+# Original file comments:
+# Copyright 2024 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require 'grpc'
+require 'google/cloud/aiplatform/v1/vertex_rag_service_pb'
+
+module Google
+  module Cloud
+    module AIPlatform
+      module V1
+        module VertexRagService
+          # A service for retrieving relevant contexts.
+          class Service
+
+            include ::GRPC::GenericService
+
+            self.marshal_class_method = :encode
+            self.unmarshal_class_method = :decode
+            self.service_name = 'google.cloud.aiplatform.v1.VertexRagService'
+
+            # Retrieves relevant contexts for a query.
+            rpc :RetrieveContexts, ::Google::Cloud::AIPlatform::V1::RetrieveContextsRequest, ::Google::Cloud::AIPlatform::V1::RetrieveContextsResponse
+            # Given an input prompt, it returns augmented prompt from vertex rag store
+            #  to guide LLM towards generating grounded responses.
+            rpc :AugmentPrompt, ::Google::Cloud::AIPlatform::V1::AugmentPromptRequest, ::Google::Cloud::AIPlatform::V1::AugmentPromptResponse
+            # Given an input text, it returns a score that evaluates the factuality of
+            # the text. It also extracts and returns claims from the text and provides
+            # supporting facts.
+            rpc :CorroborateContent, ::Google::Cloud::AIPlatform::V1::CorroborateContentRequest, ::Google::Cloud::AIPlatform::V1::CorroborateContentResponse
+          end
+
+          Stub = Service.rpc_stub_class
+        end
+      end
+    end
+  end
+end

data/proto_docs/google/cloud/aiplatform/v1/api_auth.rb

@@ -0,0 +1,45 @@
+# frozen_string_literal: true
+
+# Copyright 2024 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# Auto-generated by gapic-generator-ruby. DO NOT EDIT!
+
+
+module Google
+  module Cloud
+    module AIPlatform
+      module V1
+        # The generic reusable api auth config.
+        # @!attribute [rw] api_key_config
+        #   @return [::Google::Cloud::AIPlatform::V1::ApiAuth::ApiKeyConfig]
+        #     The API secret.
+        class ApiAuth
+          include ::Google::Protobuf::MessageExts
+          extend ::Google::Protobuf::MessageExts::ClassMethods
+
+          # The API secret.
+          # @!attribute [rw] api_key_secret_version
+          #   @return [::String]
+          #     Required. The SecretManager secret version resource name storing API key.
+          #     e.g. projects/\\{project}/secrets/\\{secret}/versions/\\{version}
+          class ApiKeyConfig
+            include ::Google::Protobuf::MessageExts
+            extend ::Google::Protobuf::MessageExts::ClassMethods
+          end
+        end
+      end
+    end
+  end
+end

data/proto_docs/google/cloud/aiplatform/v1/custom_job.rb

@@ -331,10 +331,10 @@ module Google
         # All parameters related to queuing and scheduling of custom jobs.
         # @!attribute [rw] timeout
         #   @return [::Google::Protobuf::Duration]
-        #     The maximum job running time. The default is 7 days.
+        #     Optional. The maximum job running time. The default is 7 days.
         # @!attribute [rw] restart_job_on_worker_restart
         #   @return [::Boolean]
-        #     Restarts the entire CustomJob if a worker gets restarted.
+        #     Optional. Restarts the entire CustomJob if a worker gets restarted.
         #     This feature can be used by distributed training jobs that are not
         #     resilient to workers leaving and joining a job.
         # @!attribute [rw] strategy

data/proto_docs/google/cloud/aiplatform/v1/dataset.rb

@@ -206,7 +206,8 @@ module Google
         # @!attribute [rw] saved_query_id
         #   @return [::String]
         #     The ID of a SavedQuery (annotation set) under the Dataset specified by
-        #     [dataset_id][] used for filtering Annotations for training.
+        #     {::Google::Cloud::AIPlatform::V1::ExportDataRequest#name ExportDataRequest.name}
+        #     used for filtering Annotations for training.
         #
         #     Only used for custom training data export use cases.
         #     Only applicable to Datasets that have SavedQueries.
@@ -233,7 +234,8 @@ module Google
         #     gs://google-cloud-aiplatform/schema/dataset/annotation/, note that the
         #     chosen schema must be consistent with
         #     {::Google::Cloud::AIPlatform::V1::Dataset#metadata_schema_uri metadata} of the
-        #     Dataset specified by [dataset_id][].
+        #     Dataset specified by
+        #     {::Google::Cloud::AIPlatform::V1::ExportDataRequest#name ExportDataRequest.name}.
         #
         #     Only used for custom training data export use cases.
         #     Only applicable to Datasets that have DataItems and Annotations.

data/proto_docs/google/cloud/aiplatform/v1/dataset_service.rb

@@ -47,6 +47,7 @@ module Google
 
         # Request message for
         # {::Google::Cloud::AIPlatform::V1::DatasetService::Client#get_dataset DatasetService.GetDataset}.
+        # Next ID: 4
         # @!attribute [rw] name
         #   @return [::String]
         #     Required. The name of the Dataset resource.
@@ -285,6 +286,7 @@ module Google
 
         # Request message for
         # {::Google::Cloud::AIPlatform::V1::DatasetService::Client#get_dataset_version DatasetService.GetDatasetVersion}.
+        # Next ID: 4
         # @!attribute [rw] name
         #   @return [::String]
         #     Required. The resource name of the Dataset version to delete.

data/proto_docs/google/cloud/aiplatform/v1/endpoint.rb

@@ -135,6 +135,9 @@ module Google
         #     dedicated_endpoint_enabled is true.
         #     Format:
         #     `https://{endpoint_id}.{region}-{project_number}.prediction.vertexai.goog`.
+        # @!attribute [rw] client_connection_config
+        #   @return [::Google::Cloud::AIPlatform::V1::ClientConnectionConfig]
+        #     Configurations that are applied to the endpoint for online prediction.
         # @!attribute [r] satisfies_pzs
         #   @return [::Boolean]
         #     Output only. Reserved for future use.
@@ -266,6 +269,12 @@ module Google
         #     requests directly to the deployed model services running on Cloud via
         #     private services access. This field is populated if
         #     {::Google::Cloud::AIPlatform::V1::Endpoint#network network} is configured.
+        # @!attribute [rw] faster_deployment_config
+        #   @return [::Google::Cloud::AIPlatform::V1::FasterDeploymentConfig]
+        #     Configuration for faster model deployment.
+        # @!attribute [r] status
+        #   @return [::Google::Cloud::AIPlatform::V1::DeployedModel::Status]
+        #     Output only. Runtime status of the deployed model.
         # @!attribute [rw] system_labels
         #   @return [::Google::Protobuf::Map{::String => ::String}]
         #     System labels to apply to Model Garden deployments.
@@ -274,6 +283,21 @@ module Google
           include ::Google::Protobuf::MessageExts
           extend ::Google::Protobuf::MessageExts::ClassMethods
 
+          # Runtime status of the deployed model.
+          # @!attribute [r] message
+          #   @return [::String]
+          #     Output only. The latest deployed model's status message (if any).
+          # @!attribute [r] last_update_time
+          #   @return [::Google::Protobuf::Timestamp]
+          #     Output only. The time at which the status was last updated.
+          # @!attribute [r] available_replica_count
+          #   @return [::Integer]
+          #     Output only. The number of available replicas of the deployed model.
+          class Status
+            include ::Google::Protobuf::MessageExts
+            extend ::Google::Protobuf::MessageExts::ClassMethods
+          end
+
           # @!attribute [rw] key
           #   @return [::String]
           # @!attribute [rw] value
@@ -327,6 +351,24 @@ module Google
           include ::Google::Protobuf::MessageExts
           extend ::Google::Protobuf::MessageExts::ClassMethods
         end
+
+        # Configuration for faster model deployment.
+        # @!attribute [rw] fast_tryout_enabled
+        #   @return [::Boolean]
+        #     If true, enable fast tryout feature for this deployed model.
+        class FasterDeploymentConfig
+          include ::Google::Protobuf::MessageExts
+          extend ::Google::Protobuf::MessageExts::ClassMethods
+        end
+
+        # Configurations (e.g. inference timeout) that are applied on your endpoints.
+        # @!attribute [rw] inference_timeout
+        #   @return [::Google::Protobuf::Duration]
+        #     Customizable online prediction request timeout.
+        class ClientConnectionConfig
+          include ::Google::Protobuf::MessageExts
+          extend ::Google::Protobuf::MessageExts::ClassMethods
+        end
       end
     end
   end

data/proto_docs/google/cloud/aiplatform/v1/endpoint_service.rb

@@ -160,6 +160,28 @@ module Google
           extend ::Google::Protobuf::MessageExts::ClassMethods
         end
 
+        # Request message for
+        # {::Google::Cloud::AIPlatform::V1::EndpointService::Client#update_endpoint_long_running EndpointService.UpdateEndpointLongRunning}.
+        # @!attribute [rw] endpoint
+        #   @return [::Google::Cloud::AIPlatform::V1::Endpoint]
+        #     Required. The Endpoint which replaces the resource on the server. Currently
+        #     we only support updating the `client_connection_config` field, all the
+        #     other fields' update will be blocked.
+        class UpdateEndpointLongRunningRequest
+          include ::Google::Protobuf::MessageExts
+          extend ::Google::Protobuf::MessageExts::ClassMethods
+        end
+
+        # Runtime operation information for
+        # {::Google::Cloud::AIPlatform::V1::EndpointService::Client#update_endpoint_long_running EndpointService.UpdateEndpointLongRunning}.
+        # @!attribute [rw] generic_metadata
+        #   @return [::Google::Cloud::AIPlatform::V1::GenericOperationMetadata]
+        #     The operation generic information.
+        class UpdateEndpointOperationMetadata
+          include ::Google::Protobuf::MessageExts
+          extend ::Google::Protobuf::MessageExts::ClassMethods
+        end
+
         # Request message for
         # {::Google::Cloud::AIPlatform::V1::EndpointService::Client#delete_endpoint EndpointService.DeleteEndpoint}.
         # @!attribute [rw] name

data/proto_docs/google/cloud/aiplatform/v1/evaluated_annotation.rb

@@ -38,7 +38,7 @@ module Google
         #
         #     For false positive, there is one and only one prediction, which doesn't
         #     match any ground truth annotation of the corresponding
-        #     [data_item_view_id][EvaluatedAnnotation.data_item_view_id].
+        #     {::Google::Cloud::AIPlatform::V1::EvaluatedAnnotation#evaluated_data_item_view_id data_item_view_id}.
         #
         #     For false negative, there are zero or more predictions which are similar to
         #     the only ground truth annotation in