From 4448ab665c16f92f5dc5727de45a7d792ae5f7bb Mon Sep 17 00:00:00 2001 From: Debanjum Date: Fri, 20 Jun 2025 00:36:55 -0700 Subject: [PATCH] Improve google image generation configuration --- src/khoj/processor/image/generate.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/src/khoj/processor/image/generate.py b/src/khoj/processor/image/generate.py index 86f1cde1..124a9ce8 100644 --- a/src/khoj/processor/image/generate.py +++ b/src/khoj/processor/image/generate.py @@ -242,7 +242,13 @@ def generate_image_with_google( client = genai.Client(api_key=api_key) # Configure image generation settings - config = gtypes.GenerateImagesConfig(number_of_images=1) + config = gtypes.GenerateImagesConfig( + number_of_images=1, + safety_filter_level=gtypes.SafetyFilterLevel.BLOCK_LOW_AND_ABOVE, + person_generation=gtypes.PersonGeneration.ALLOW_ADULT, + include_rai_reason=True, + output_mime_type="image/png", + ) # Call the Gemini API to generate the image response = client.models.generate_images(model=text2image_model, prompt=improved_image_prompt, config=config)