Update agent.py

agent.py

@@ -56,15 +56,24 @@ def wait_for_model():
                 headers=headers,
                 timeout=30
             )
-            # Si el modelo responde sin error, lo consideramos listo
+            
+            # Verificación de que el modelo esté listo
             if response.status_code == 200:
-                print(f"Modelo {MODEL_NAME} listo para recibir solicitudes.")
-                break
+                # 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.")
+                    break
             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
         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
+            time.sleep(10)  # Esperar 10 segundos antes de intentar nuevamente
 
 # Clase para representar la lógica del agente
 class Agente: