diff --git a/bin/rate_update.py b/bin/rate_update.py index 06f09bd..e5865ef 100755 --- a/bin/rate_update.py +++ b/bin/rate_update.py @@ -85,6 +85,7 @@ def check_install_pycountry(): # Directorio base del proyecto BASE_DIR = Path(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) CONFIG_DIR = BASE_DIR / 'bin' / 'config' +RATES_DIR = CONFIG_DIR / 'rates' # Nueva ubicación para archivos de tarifas # Archivo de configuración para el modelo de IA MODEL_CONFIG_FILE = CONFIG_DIR / 'rate_model.ai' @@ -447,9 +448,10 @@ def generate_prompt(programmer_type, region_code): 'it': 'Italia' } - # Archivos especiales como kdevs.rate se manejan de forma diferente - if programmer_type == 'kdevs': - # Simplemente mantener el valor actual para kdevs.rate + # Ya no tenemos archivos especiales como kdevs.rate + # Este bloque solo debe ejecutarse para tipos especiales que no siguen el patrón normal + # Como ya no tenemos kdevs.rate, esta sección no se ejecutará + if programmer_type in ['kdevs', 'special_case']: return None programmer_map = { @@ -566,8 +568,11 @@ def update_rate_files(): api_available = False logger.warning("API de Perplexity no disponible. Se usarán valores de respaldo.") - # Buscar todos los archivos .rate - rate_files = glob.glob(str(CONFIG_DIR / '*.rate')) + # Crear la carpeta rates si no existe + os.makedirs(RATES_DIR, exist_ok=True) + + # Buscar todos los archivos .rate en la nueva ubicación + rate_files = glob.glob(str(RATES_DIR / '*.rate')) logger.info(f"Encontrados {len(rate_files)} archivos de tarifas para actualizar.") # Control de errores para limitar los intentos de API