From 2a32836d1aaf89c5bf3100eb89b65ffc5b7c5c33 Mon Sep 17 00:00:00 2001 From: Debanjum Singh Solanky Date: Thu, 24 Oct 2024 01:01:48 -0700 Subject: [PATCH] Log more descriptive error when image gen fails with Replicate --- src/khoj/processor/image/generate.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/khoj/processor/image/generate.py b/src/khoj/processor/image/generate.py index 5dd51a60..343e44b3 100644 --- a/src/khoj/processor/image/generate.py +++ b/src/khoj/processor/image/generate.py @@ -204,9 +204,10 @@ def generate_image_with_replicate( # Raise exception if the image generation task fails if status != "succeeded": + error = get_prediction.get("error") if retry_count >= 10: raise requests.RequestException("Image generation timed out") - raise requests.RequestException(f"Image generation failed with status: {status}") + raise requests.RequestException(f"Image generation failed with status: {status}, message: {error}") # Get the generated image image_url = get_prediction["output"][0] if isinstance(get_prediction["output"], list) else get_prediction["output"]