tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-08 18:21:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

fixes: AI/ML API

Browse Source
This commit is contained in:
Ishaan Jaffer
2025-10-16 17:52:37 -07:00
parent 1c53a686d5
commit 1c21b2e285
1 changed files with 21 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/llms/aiml/image_generation/transformation.py
+21 -5
View File
@@ -172,16 +172,32 @@ class AimlImageGenerationConfig(BaseImageGenerationConfig):
if not model_response.data:
model_response.data = []
# AI/ML API can return images in two different formats:
# 1. output.choices array with image_base64
# 2. images array with url (and optional width, height, content_type)
# AI/ML API can return images in multiple formats:
# 1. Top-level data array with url (OpenAI-like format)
# 2. output.choices array with image_base64
# 3. images array with url (and optional width, height, content_type)
if "output" in response_data and "choices" in response_data["output"]:
if "data" in response_data and isinstance(response_data["data"], list):
# Handle OpenAI-like format: {"data": [{"url": "...", "width": 1024, "height": 768, "content_type": "image/jpeg"}]}
for image in response_data["data"]:
if "url" in image:
model_response.data.append(ImageObject(
b64_json=None,
url=image["url"],
revised_prompt=image.get("revised_prompt"),
))
elif "b64_json" in image or "image_base64" in image:
model_response.data.append(ImageObject(
b64_json=image.get("b64_json") or image.get("image_base64"),
url=None,
revised_prompt=image.get("revised_prompt"),
))
elif "output" in response_data and "choices" in response_data["output"]:
for choice in response_data["output"]["choices"]:
if "image_base64" in choice:
model_response.data.append(ImageObject(
b64_json=choice["image_base64"],
url=None, # AI/ML API returns base64, not URLs
url=None,
))
elif "url" in choice:
model_response.data.append(ImageObject(