From 3be505db48535a7e5cf5826a9cfd75db1921e791 Mon Sep 17 00:00:00 2001 From: Debanjum Singh Solanky Date: Tue, 22 Oct 2024 20:00:06 -0700 Subject: [PATCH] Only show type of error when image generation fails to clients Rather than showing raw error message from the underlying service as it could contain sensitive information --- src/khoj/processor/image/generate.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/khoj/processor/image/generate.py b/src/khoj/processor/image/generate.py index ee39bdc5..5dd51a60 100644 --- a/src/khoj/processor/image/generate.py +++ b/src/khoj/processor/image/generate.py @@ -87,18 +87,18 @@ async def text_to_image( if "content_policy_violation" in e.message: logger.error(f"Image Generation blocked by OpenAI: {e}") status_code = e.status_code # type: ignore - message = f"Image generation blocked by OpenAI: {e.message}" # type: ignore + message = f"Image generation blocked by OpenAI due to policy violation" # type: ignore yield image_url or image, status_code, message, intent_type.value return else: logger.error(f"Image Generation failed with {e}", exc_info=True) - message = f"Image generation failed with OpenAI error: {e.message}" # type: ignore + message = f"Image generation failed using OpenAI" # type: ignore status_code = e.status_code # type: ignore yield image_url or image, status_code, message, intent_type.value return except requests.RequestException as e: logger.error(f"Image Generation failed with {e}", exc_info=True) - message = f"Image generation using {text2image_model} via {text_to_image_config.model_type} failed with error: {e}" + message = f"Image generation using {text2image_model} via {text_to_image_config.model_type} failed due to a network error." status_code = 502 yield image_url or image, status_code, message, intent_type.value return