From e938fa9e416bee76a0a250206f5d240c5577c1ac Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Tue, 29 Jul 2025 17:08:44 -0700 Subject: [PATCH] fix transform_image_generation_response --- litellm/llms/gemini/image_generation/transformation.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/litellm/llms/gemini/image_generation/transformation.py b/litellm/llms/gemini/image_generation/transformation.py index 14c6955772c..e57364fd288 100644 --- a/litellm/llms/gemini/image_generation/transformation.py +++ b/litellm/llms/gemini/image_generation/transformation.py @@ -190,10 +190,9 @@ class GoogleImageGenConfig(BaseImageGenerationConfig): predictions = response_data.get("predictions", []) for prediction in predictions: # Google AI returns base64 encoded images in the prediction - 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 - )) + model_response.data.append(ImageObject( + b64_json=prediction.get("bytesBase64Encoded", None), + url=None, # Google AI returns base64, not URLs + )) return model_response \ No newline at end of file