encounter/adk-python
0
0
Fork 0
mirror of https://github.com/encounter/adk-python.git synced 2026-07-09 18:19:28 -07:00
Code Issues Packages Projects Releases Wiki Activity

fix: Remove redundant format field from LiteLLM content objects

Browse Source 
LiteLLM providers can extract the MIME type from the data URI. Removing the separate `format` field avoids redundancy and potential issues with backends that may reject requests containing this field.

Close #2017

Co-authored-by: George Weale <gweale@google.com>
PiperOrigin-RevId: 828014286
This commit is contained in:
George Weale
2025-11-04 09:44:38 -08:00
committed by Copybara-Service
parent 38ea749c9c
commit 489c39db01
2 changed files with 13 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/google/adk/models/lite_llm.py
+6 -13
View File
@@ -314,33 +314,28 @@ def _get_content(
):
base64_string = base64.b64encode(part.inline_data.data).decode("utf-8")
data_uri = f"data:{part.inline_data.mime_type};base64,{base64_string}"
# LiteLLM providers extract the MIME type from the data URI; avoid
# passing a separate `format` field that some backends reject.
if part.inline_data.mime_type.startswith("image"):
# Use full MIME type (e.g., "image/png") for providers that validate it
format_type = part.inline_data.mime_type
content_objects.append({
"type": "image_url",
"image_url": {"url": data_uri, "format": format_type},
"image_url": {"url": data_uri},
})
elif part.inline_data.mime_type.startswith("video"):
# Use full MIME type (e.g., "video/mp4") for providers that validate it
format_type = part.inline_data.mime_type
content_objects.append({
"type": "video_url",
"video_url": {"url": data_uri, "format": format_type},
"video_url": {"url": data_uri},
})
elif part.inline_data.mime_type.startswith("audio"):
# Use full MIME type (e.g., "audio/mpeg") for providers that validate it
format_type = part.inline_data.mime_type
content_objects.append({
"type": "audio_url",
"audio_url": {"url": data_uri, "format": format_type},
"audio_url": {"url": data_uri},
})
elif part.inline_data.mime_type == "application/pdf":
format_type = part.inline_data.mime_type
content_objects.append({
"type": "file",
"file": {"file_data": data_uri, "format": format_type},
"file": {"file_data": data_uri},
})
else:
raise ValueError("LiteLlm(BaseLlm) does not support this content part.")
@@ -348,8 +343,6 @@ def _get_content(
file_object: ChatCompletionFileUrlObject = {
"file_id": part.file_data.file_uri,
}
if part.file_data.mime_type:
file_object["format"] = part.file_data.mime_type
content_objects.append({
"type": "file",
"file": file_object,