from enum import Enum

class ConsultaType(str, Enum):
    HISTORICO = "HISTORICO"
    TIEMPO_REAL = "TIEMPO_REAL"

class ToolNames(str, Enum):
    OBTENER_FECHA_ACTUAL = "obtener_fecha_actual"
    CALCULAR_FECHA = "calcular_fecha"
    DETECTAR_TIPO_CONSULTA = "detectar_tipo_consulta"
    BUSCAR_EN_DOCUMENTOS = "buscar_en_documentos"
    GET_ARRIBOS = "getArribos"
    GET_SALIDAS = "getSalidas"
    GET_ITINERARIOS = "getItinerarios"
    BUSCAR_POR_NUMERO_VUELO = "buscarPorNumeroVuelo"
    EMITIR_BOLETO = "emitirBoleto"
    LEER_ITINERARIO_MENSUAL = "leer_itinerario_mensual"

class ErrorMessages(str, Enum):
    AGENT_NOT_INITIALIZED = "El agente no ha sido inicializado."
    MCP_CONNECTION_FAILED = "Fallo al conectar con servidor MCP."
    EMPTY_QUERY = "La consulta no puede estar vacía."
    TIMEOUT = "La consulta tardó demasiado tiempo en procesarse."