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

fix transform_image_generation_response

Browse source
This commit is contained in:
Ishaan Jaff 2025-07-29 17:08:44 -07:00
parent 51a140feb7
commit e938fa9e41
1 changed files with 4 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
litellm/llms/gemini/image_generation/transformation.py
View file

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