Update agent.py

agent.py

@@ -48,7 +48,8 @@ def wait_for_model():
         "Content-Type": "application/json"
     }
 
-    # Intentamos hacer un request a la API del modelo cada 10 segundos
+    print(f"Esperando que el modelo {MODEL_NAME} esté completamente cargado...")
+
     while True:
         try:
             response = requests.get(
@@ -56,24 +57,21 @@ def wait_for_model():
                 headers=headers,
                 timeout=30
             )
-            
-            # Verificación de que el modelo esté listo
+
             if response.status_code == 200:
-                # Podrías agregar más validación aquí dependiendo de lo que devuelva el modelo.
-                # Algunos modelos pueden devolver un "ready" en el cuerpo de la respuesta
-                response_data = response.json()
-                if "error" in response_data:
-                    print(f"El modelo aún no está listo: {response_data.get('error')}")
-                    time.sleep(10)  # Esperar más si el modelo no está listo
-                else:
-                    print(f"Modelo {MODEL_NAME} listo para recibir solicitudes.")
+                json_response = response.json()
+                if isinstance(json_response, dict) and json_response.get("status") == "loaded":
+                    print(f"✅ Modelo {MODEL_NAME} completamente cargado.")
                     break
+                else:
+                    print(f"⏳ Modelo aún cargando... Estado actual: {json_response.get('status', 'desconocido')}")
             else:
-                print(f"Esperando que el modelo {MODEL_NAME} esté listo... (estado {response.status_code})")
-                time.sleep(10)  # Esperar 10 segundos antes de intentar nuevamente
+                print(f"❌ Estado inesperado: {response.status_code}")
         except requests.exceptions.RequestException as e:
-            print(f"Error al verificar el estado del modelo: {e}")
-            time.sleep(10)  # Esperar 10 segundos antes de intentar nuevamente
+            print(f"🌐 Error de red: {e}")
+
+        time.sleep(10)  # Espera 10 segundos antes del siguiente intento
+
 
 # Clase para representar la lógica del agente
 class Agente: