config.py

import os
import logging

logger = logging.getLogger(__name__)
logger.setLevel(logging.INFO)
handler = logging.StreamHandler()
handler.setFormatter(logging.Formatter('%(asctime)s - %(levelname)s - %(message)s'))
logger.addHandler(handler)

HF_TOKEN = os.getenv('HF_TOKEN', '')
GROQ_API_KEY = os.getenv('GROQ_API_KEY', '')

GROQ_MODEL = 'llama-3.1-8b-instant'
GROQ_TEMPERATURE = 0.3
GROQ_MAX_TOKENS = 2048

TAILA_NAME = 'TAILA'
TAILA_VERSION = '1.0.0'
TAILA_AUTHOR = 'Ricardo Pino Peña'

MEMORY_DIR = '/data/memory'
CACHE_DIR = '/data/cache'

def get_config() -> dict:
    config = {
        'hf_token': HF_TOKEN,
        'groq_api_key': GROQ_API_KEY,
        'groq_model': GROQ_MODEL,
        'groq_temperature': GROQ_TEMPERATURE,
        'groq_max_tokens': GROQ_MAX_TOKENS,
        'taila_name': TAILA_NAME,
        'taila_version': TAILA_VERSION,
        'taila_author': TAILA_AUTHOR,
        'memory_dir': MEMORY_DIR,
        'cache_dir': CACHE_DIR
    }

    if HF_TOKEN:
        logger.info('✅ Hugging Face token cargado')
    else:
        logger.warning('⚠️ Hugging Face token no cargado')

    if GROQ_API_KEY:
        logger.info('✅ Groq API key cargada')
    else:
        logger.warning('⚠️ Groq API key no cargada')

    return config