# -*- coding: utf-8 -*-
# Generated by the protocol buffer compiler.  DO NOT EDIT!
# NO CHECKED-IN PROTOBUF GENCODE
# source: truffle/infer/model.proto
# Protobuf Python Version: 6.30.0
"""Generated protocol buffer code."""
from google.protobuf import descriptor as _descriptor
from google.protobuf import descriptor_pool as _descriptor_pool
from google.protobuf import runtime_version as _runtime_version
from google.protobuf import symbol_database as _symbol_database
from google.protobuf.internal import builder as _builder
_runtime_version.ValidateProtobufRuntimeVersion(
    _runtime_version.Domain.PUBLIC,
    6,
    30,
    0,
    '',
    'truffle/infer/model.proto'
)
# @@protoc_insertion_point(imports)

_sym_db = _symbol_database.Default()




DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x19truffle/infer/model.proto\x12\rtruffle.infer\"\xda\x01\n\x12\x45mbeddingModelInfo\x12\n\n\x02id\x18\x01 \x01(\x05\x12\x0c\n\x04name\x18\x02 \x01(\t\x12\x0f\n\x07version\x18\x03 \x01(\t\x12\x38\n\x06\x63onfig\x18\x04 \x01(\x0b\x32(.truffle.infer.EmbeddingModelInfo.Config\x12\x0c\n\x04uuid\x18\x05 \x01(\t\x1aQ\n\x06\x43onfig\x12\x18\n\x10max_input_length\x18\x01 \x01(\x03\x12\x16\n\x0emax_batch_size\x18\x02 \x01(\x03\x12\x15\n\rembedding_dim\x18\x03 \x01(\x03\"G\n\x12\x45mbeddingModelList\x12\x31\n\x06models\x18\x01 \x03(\x0b\x32!.truffle.infer.EmbeddingModelInfo\"\xba\x04\n\x0bModelConfig\x12=\n\x04info\x18\x01 \x01(\x0b\x32*.truffle.infer.ModelConfig.ModelConfigInfoH\x00\x88\x01\x01\x12\x1b\n\x0e\x63ontext_length\x18\x02 \x01(\rH\x01\x88\x01\x01\x12\x1b\n\x0emax_batch_size\x18\x03 \x01(\rH\x02\x88\x01\x01\x12\x16\n\tdata_type\x18\x04 \x01(\tH\x03\x88\x01\x01\x12\x13\n\x06loaded\x18\x05 \x01(\x08H\x04\x88\x01\x01\x1a\xbc\x02\n\x0fModelConfigInfo\x12 \n\x18\x63ontext_length_limit_max\x18\x01 \x01(\r\x12 \n\x18\x63ontext_length_limit_min\x18\x02 \x01(\r\x12\x1c\n\x14model_context_length\x18\n \x01(\r\x12\x1c\n\x14\x62\x61tch_size_limit_max\x18\x03 \x01(\r\x12\x1c\n\x14\x62\x61tch_size_limit_min\x18\x04 \x01(\r\x12\x1c\n\x14has_chain_of_thought\x18\x05 \x01(\x08\x12\x12\n\nis_agentic\x18\x06 \x01(\x08\x12\x1b\n\x13memory_usage_params\x18\x07 \x01(\x03\x12\x1e\n\x16memory_usage_inference\x18\x08 \x01(\x03\x12\x1c\n\x14\x61vailable_data_types\x18\t \x03(\tB\x07\n\x05_infoB\x11\n\x0f_context_lengthB\x11\n\x0f_max_batch_sizeB\x0c\n\n_data_typeB\t\n\x07_loaded\"\xa0\x02\n\x05Model\x12\x0c\n\x04uuid\x18\x01 \x01(\t\x12\x0c\n\x04name\x18\x02 \x01(\t\x12\x10\n\x08provider\x18\x03 \x01(\t\x12*\n\x06\x63onfig\x18\x04 \x01(\x0b\x32\x1a.truffle.infer.ModelConfig\x12.\n\x05state\x18\x05 \x01(\x0e\x32\x1f.truffle.infer.Model.ModelState\"\x8c\x01\n\nModelState\x12\x17\n\x13MODEL_STATE_INVALID\x10\x00\x12\x19\n\x15MODEL_STATE_AVAILABLE\x10\x01\x12\x17\n\x13MODEL_STATE_LOADING\x10\x02\x12\x19\n\x15MODEL_STATE_UNLOADING\x10\x03\x12\x16\n\x12MODEL_STATE_LOADED\x10\x04\"z\n\x10ModelStateUpdate\x12\x12\n\nmodel_uuid\x18\x01 \x01(\t\x12.\n\x05state\x18\x02 \x01(\x0e\x32\x1f.truffle.infer.Model.ModelState\x12\x15\n\x08progress\x18\x03 \x01(\x05H\x00\x88\x01\x01\x42\x0b\n\t_progress\"\\\n\tModelList\x12$\n\x06models\x18\x01 \x03(\x0b\x32\x14.truffle.infer.Model\x12\x14\n\x0ctotal_memory\x18\x02 \x01(\x04\x12\x13\n\x0bused_memory\x18\x03 \x01(\x04\"%\n\x0fGetModelRequest\x12\x12\n\nmodel_uuid\x18\x01 \x01(\t\"m\n\x13GetModelListRequest\x12\x12\n\nuse_filter\x18\x01 \x01(\x08\x12\x13\n\tavailable\x18\x02 \x01(\x08H\x00\x12\x10\n\x06loaded\x18\x03 \x01(\x08H\x00\x12\x11\n\x07\x61gentic\x18\x04 \x01(\x08H\x00\x42\x08\n\x06\x66ilter\"\x88\x02\n\x11SetModelsResponse\x12\x41\n\x04\x63ode\x18\x01 \x01(\x0e\x32\x33.truffle.infer.SetModelsResponse.SetModelsErrorCode\x12\x0f\n\x07message\x18\x02 \x01(\t\x12\x33\n\x0cupdated_list\x18\x03 \x01(\x0b\x32\x18.truffle.infer.ModelListH\x00\x88\x01\x01\"Y\n\x12SetModelsErrorCode\x12\x06\n\x02OK\x10\x00\x12\x12\n\x0eINVALID_CONFIG\x10\x01\x12\x15\n\x11NOT_ENOUGH_MEMORY\x10\x02\x12\x10\n\x0cMODEL_IN_USE\x10\x03\x42\x0f\n\r_updated_list\"\x9d\x01\n\x10SetModelsRequest\x12=\n\x07updates\x18\x01 \x03(\x0b\x32,.truffle.infer.SetModelsRequest.UpdatesEntry\x1aJ\n\x0cUpdatesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12)\n\x05value\x18\x02 \x01(\x0b\x32\x1a.truffle.infer.ModelConfig:\x02\x38\x01\x62\x06proto3')

_globals = globals()
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'truffle.infer.model_pb2', _globals)
if not _descriptor._USE_C_DESCRIPTORS:
  DESCRIPTOR._loaded_options = None
  _globals['_SETMODELSREQUEST_UPDATESENTRY']._loaded_options = None
  _globals['_SETMODELSREQUEST_UPDATESENTRY']._serialized_options = b'8\001'
  _globals['_EMBEDDINGMODELINFO']._serialized_start=45
  _globals['_EMBEDDINGMODELINFO']._serialized_end=263
  _globals['_EMBEDDINGMODELINFO_CONFIG']._serialized_start=182
  _globals['_EMBEDDINGMODELINFO_CONFIG']._serialized_end=263
  _globals['_EMBEDDINGMODELLIST']._serialized_start=265
  _globals['_EMBEDDINGMODELLIST']._serialized_end=336
  _globals['_MODELCONFIG']._serialized_start=339
  _globals['_MODELCONFIG']._serialized_end=909
  _globals['_MODELCONFIG_MODELCONFIGINFO']._serialized_start=521
  _globals['_MODELCONFIG_MODELCONFIGINFO']._serialized_end=837
  _globals['_MODEL']._serialized_start=912
  _globals['_MODEL']._serialized_end=1200
  _globals['_MODEL_MODELSTATE']._serialized_start=1060
  _globals['_MODEL_MODELSTATE']._serialized_end=1200
  _globals['_MODELSTATEUPDATE']._serialized_start=1202
  _globals['_MODELSTATEUPDATE']._serialized_end=1324
  _globals['_MODELLIST']._serialized_start=1326
  _globals['_MODELLIST']._serialized_end=1418
  _globals['_GETMODELREQUEST']._serialized_start=1420
  _globals['_GETMODELREQUEST']._serialized_end=1457
  _globals['_GETMODELLISTREQUEST']._serialized_start=1459
  _globals['_GETMODELLISTREQUEST']._serialized_end=1568
  _globals['_SETMODELSRESPONSE']._serialized_start=1571
  _globals['_SETMODELSRESPONSE']._serialized_end=1835
  _globals['_SETMODELSRESPONSE_SETMODELSERRORCODE']._serialized_start=1729
  _globals['_SETMODELSRESPONSE_SETMODELSERRORCODE']._serialized_end=1818
  _globals['_SETMODELSREQUEST']._serialized_start=1838
  _globals['_SETMODELSREQUEST']._serialized_end=1995
  _globals['_SETMODELSREQUEST_UPDATESENTRY']._serialized_start=1921
  _globals['_SETMODELSREQUEST_UPDATESENTRY']._serialized_end=1995
# @@protoc_insertion_point(module_scope)
