From 51a140feb70ef05a46788f6284a30ebffa7feb3e Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Tue, 29 Jul 2025 17:06:16 -0700 Subject: [PATCH] transform_image_generation_response --- litellm/llms/gemini/image_generation/transformation.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/litellm/llms/gemini/image_generation/transformation.py b/litellm/llms/gemini/image_generation/transformation.py index 72ba5bcf1e..14c6955772 100644 --- a/litellm/llms/gemini/image_generation/transformation.py +++ b/litellm/llms/gemini/image_generation/transformation.py @@ -190,8 +190,7 @@ class GoogleImageGenConfig(BaseImageGenerationConfig): predictions = response_data.get("predictions", []) for prediction in predictions: # Google AI returns base64 encoded images in the prediction - generated_images = prediction.get("generatedImages", []) - for image_data in generated_images: + for image_data in prediction: model_response.data.append(ImageObject( b64_json=image_data.get("bytesBase64Encoded", None), url=None, # Google AI returns base64, not URLs