# frozen_string_literal: true
# Generated by the protocol buffer compiler.  DO NOT EDIT!
# source: google/cloud/aiplatform/v1/tool.proto

require 'google/protobuf'

require 'google/api/field_behavior_pb'
require 'google/api/resource_pb'
require 'google/cloud/aiplatform/v1/openapi_pb'
require 'google/protobuf/struct_pb'
require 'google/type/latlng_pb'


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\"\xf3\t\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\x12@\n\x0bgoogle_maps\x18\x05 \x01(\x0b\x32&.google.cloud.aiplatform.v1.GoogleMapsB\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\x12G\n\x0c\x63omputer_use\x18\x0b \x01(\x0b\x32,.google.cloud.aiplatform.v1.Tool.ComputerUseB\x03\xe0\x41\x01\x1a\xa1\x01\n\x0cGoogleSearch\x12\x1c\n\x0f\x65xclude_domains\x18\x03 \x03(\tB\x03\xe0\x41\x01\x12[\n\x13\x62locking_confidence\x18\x04 \x01(\x0e\x32\x34.google.cloud.aiplatform.v1.Tool.PhishBlockThresholdB\x03\xe0\x41\x01H\x00\x88\x01\x01\x42\x16\n\x14_blocking_confidence\x1a\x0f\n\rCodeExecution\x1a\xa6\x01\n\x0b\x43omputerUse\x12R\n\x0b\x65nvironment\x18\x01 \x01(\x0e\x32\x38.google.cloud.aiplatform.v1.Tool.ComputerUse.EnvironmentB\x03\xe0\x41\x02\"C\n\x0b\x45nvironment\x12\x1b\n\x17\x45NVIRONMENT_UNSPECIFIED\x10\x00\x12\x17\n\x13\x45NVIRONMENT_BROWSER\x10\x01\"\xe5\x01\n\x13PhishBlockThreshold\x12%\n!PHISH_BLOCK_THRESHOLD_UNSPECIFIED\x10\x00\x12\x17\n\x13\x42LOCK_LOW_AND_ABOVE\x10\x1e\x12\x1a\n\x16\x42LOCK_MEDIUM_AND_ABOVE\x10(\x12\x18\n\x14\x42LOCK_HIGH_AND_ABOVE\x10\x32\x12\x1a\n\x16\x42LOCK_HIGHER_AND_ABOVE\x10\x37\x12\x1d\n\x19\x42LOCK_VERY_HIGH_AND_ABOVE\x10<\x12\x1d\n\x19\x42LOCK_ONLY_EXTREMELY_HIGH\x10\x64\"\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\"(\n\nGoogleMaps\x12\x1a\n\renable_widget\x18\x01 \x01(\x08\x42\x03\xe0\x41\x01\"\xa8\x01\n\x13\x45nterpriseWebSearch\x12\x1c\n\x0f\x65xclude_domains\x18\x01 \x03(\tB\x03\xe0\x41\x01\x12[\n\x13\x62locking_confidence\x18\x02 \x01(\x0e\x32\x34.google.cloud.aiplatform.v1.Tool.PhishBlockThresholdB\x03\xe0\x41\x01H\x00\x88\x01\x01\x42\x16\n\x14_blocking_confidence\"\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"

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.Schema", "google/cloud/aiplatform/v1/openapi.proto"],
    ["google.protobuf.Value", "google/protobuf/struct.proto"],
    ["google.type.LatLng", "google/type/latlng.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
        Tool = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.Tool").msgclass
        Tool::GoogleSearch = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.Tool.GoogleSearch").msgclass
        Tool::CodeExecution = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.Tool.CodeExecution").msgclass
        Tool::ComputerUse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.Tool.ComputerUse").msgclass
        Tool::ComputerUse::Environment = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.Tool.ComputerUse.Environment").enummodule
        Tool::PhishBlockThreshold = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.Tool.PhishBlockThreshold").enummodule
        UrlContext = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.UrlContext").msgclass
        FunctionDeclaration = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.FunctionDeclaration").msgclass
        FunctionCall = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.FunctionCall").msgclass
        FunctionResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.FunctionResponse").msgclass
        ExecutableCode = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ExecutableCode").msgclass
        ExecutableCode::Language = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ExecutableCode.Language").enummodule
        CodeExecutionResult = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.CodeExecutionResult").msgclass
        CodeExecutionResult::Outcome = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.CodeExecutionResult.Outcome").enummodule
        Retrieval = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.Retrieval").msgclass
        VertexRagStore = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.VertexRagStore").msgclass
        VertexRagStore::RagResource = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.VertexRagStore.RagResource").msgclass
        VertexAISearch = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.VertexAISearch").msgclass
        VertexAISearch::DataStoreSpec = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.VertexAISearch.DataStoreSpec").msgclass
        GoogleSearchRetrieval = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.GoogleSearchRetrieval").msgclass
        GoogleMaps = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.GoogleMaps").msgclass
        EnterpriseWebSearch = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.EnterpriseWebSearch").msgclass
        DynamicRetrievalConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.DynamicRetrievalConfig").msgclass
        DynamicRetrievalConfig::Mode = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.DynamicRetrievalConfig.Mode").enummodule
        ToolConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ToolConfig").msgclass
        FunctionCallingConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.FunctionCallingConfig").msgclass
        FunctionCallingConfig::Mode = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.FunctionCallingConfig.Mode").enummodule
        RetrievalConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.RetrievalConfig").msgclass
        RagRetrievalConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.RagRetrievalConfig").msgclass
        RagRetrievalConfig::Filter = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.RagRetrievalConfig.Filter").msgclass
        RagRetrievalConfig::Ranking = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.RagRetrievalConfig.Ranking").msgclass
        RagRetrievalConfig::Ranking::RankService = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.RagRetrievalConfig.Ranking.RankService").msgclass
        RagRetrievalConfig::Ranking::LlmRanker = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.RagRetrievalConfig.Ranking.LlmRanker").msgclass
      end
    end
  end
end
