klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-07 13:23:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

Log more descriptive error when image gen fails with Replicate

Browse Source
This commit is contained in:
Debanjum Singh Solanky
2024-10-24 01:01:48 -07:00
parent 5120597d4e
commit 2a32836d1a
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/khoj/processor/image/generate.py
View File

@@ -204,9 +204,10 @@ def generate_image_with_replicate(
# Raise exception if the image generation task fails # Raise exception if the image generation task fails
if status != "succeeded": if status != "succeeded":
error = get_prediction.get("error")
if retry_count >= 10: if retry_count >= 10:
raise requests.RequestException("Image generation timed out") 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 # Get the generated image
image_url = get_prediction["output"][0] if isinstance(get_prediction["output"], list) else get_prediction["output"] image_url = get_prediction["output"][0] if isinstance(get_prediction["output"], list) else get_prediction["output"]