# 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
