scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-06 02:48:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

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