ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

transform_image_generation_response

Browse Source
This commit is contained in:
Ishaan Jaff 2025-07-29 17:06:16 -07:00
parent 33f65114b6
commit 51a140feb7
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
litellm/llms/gemini/image_generation/transformation.py
View File

@ -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