diff --git a/matrix_pygmalion_bot/ai/runpod_pygmalion.py b/matrix_pygmalion_bot/ai/runpod_pygmalion.py index 92b723d..deacc12 100644 --- a/matrix_pygmalion_bot/ai/runpod_pygmalion.py +++ b/matrix_pygmalion_bot/ai/runpod_pygmalion.py @@ -201,7 +201,8 @@ async def generate_image(input_prompt: str, negative_prompt: str, api_url: str, if r.status_code == 200: status = r_json["status"] if status != 'IN_QUEUE': - raise ValueError(f"RETURN CODE {status}") + err_msg = r_json["error"] if "error" in r_json else "" + raise ValueError(f"RETURN CODE {status}: {err_msg}") job_id = r_json["id"] TIMEOUT = 360 DELAY = 5 @@ -220,7 +221,8 @@ async def generate_image(input_prompt: str, negative_prompt: str, api_url: str, output = r_json["output"] break else: - raise ValueError(f"RETURN CODE {status}") + err_msg = r_json["error"] if "error" in r_json else "" + raise ValueError(f"RETURN CODE {status}: {err_msg}") if not output: raise ValueError(f"") @@ -264,7 +266,7 @@ async def serverless_automatic_request(payload, cmd, api_url: str, api_key: str) } # Define your inputs - payload.update({"api_endpoint": cmd}) + #payload.update({"api_endpoint": cmd}) input_data = { "input": payload, "cmd": cmd, @@ -280,7 +282,8 @@ async def serverless_automatic_request(payload, cmd, api_url: str, api_key: str) if r.status_code == 200: status = r_json["status"] if status != 'IN_QUEUE': - raise ValueError(f"RETURN CODE {status}") + err_msg = r_json["error"] if "error" in r_json else "" + raise ValueError(f"RETURN CODE {status}: {err_msg}") job_id = r_json["id"] TIMEOUT = 360 DELAY = 5 @@ -299,7 +302,8 @@ async def serverless_automatic_request(payload, cmd, api_url: str, api_key: str) output = r_json["output"] break else: - raise ValueError(f"RETURN CODE {status}") + err_msg = r_json["error"] if "error" in r_json else "" + raise ValueError(f"RETURN CODE {status}: {err_msg}") if not output: raise ValueError(f" {status}")