sij/khoj
1
0
Fork 0
mirror of https://github.com/khoj-ai/khoj.git synced 2025-02-17 08:04:21 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

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
Show stats 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
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"]