sij/khoj
1
0
Fork 0
mirror of https://github.com/khoj-ai/khoj.git synced 2024-11-23 15:38:55 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

Improve user visibility into errors during image generation

Browse source
This commit is contained in:
Debanjum Singh Solanky 2024-02-29 13:19:13 +05:30
parent 0bbb5cff85
commit 2d61591c22
2 changed files with 15 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
src/khoj/routers/api_chat.py
View file

@ -297,11 +297,7 @@ async def chat(
)
image, status_code, improved_image_prompt = await text_to_image(q, meta_log, location_data=location)
if image is None:
content_obj = {
"image": image,
"intentType": "text-to-image",
"detail": "Failed to generate image. Make sure your image generation configuration is set.",
}
content_obj = {"image": image, "intentType": "text-to-image", "detail": improved_image_prompt}
return Response(content=json.dumps(content_obj), media_type="application/json", status_code=status_code)
await sync_to_async(save_to_conversation_log)(
q,

17
src/khoj/routers/helpers.py
View file

@ -381,11 +381,14 @@ async def text_to_image(
) -> Tuple[Optional[str], int, Optional[str]]:
status_code = 200
image = None
response = None
text_to_image_config = await ConversationAdapters.aget_text_to_image_model_config()
if not text_to_image_config:
# If the user has not configured a text to image model, return an unsupported on server error
status_code = 501
message = "Failed to generate image. Setup image generation on the server."
return image, status_code, message
elif state.openai_client and text_to_image_config.model_type == TextToImageModelConfig.ModelType.OPENAI:
text2image_model = text_to_image_config.model_name
chat_history = ""
@ -399,9 +402,17 @@ async def text_to_image(
prompt=improved_image_prompt, model=text2image_model, response_format="b64_json"
)
image = response.data[0].b64_json
except openai.OpenAIError as e:
logger.error(f"Image Generation failed with {e}", exc_info=True)
status_code = 500
except openai.OpenAIError or openai.BadRequestError as e:
if "content_policy_violation" in e.message:
logger.error(f"Image Generation blocked by OpenAI: {e}")
status_code = e.status_code
message = f"Image generation blocked by OpenAI: {e.message}"
return image, status_code, message
else:
logger.error(f"Image Generation failed with {e}", exc_info=True)
message = f"Image generation failed with OpenAI error: {e.message}"
status_code = e.status_code
return image, status_code, message
return image, status_code, improved_image_prompt