From c59f8de306c44a5172341178814f19145fbd4229 Mon Sep 17 00:00:00 2001 From: Eric-Terminal <121368508+Eric-Terminal@users.noreply.github.com> Date: Tue, 12 Aug 2025 20:21:25 +0800 Subject: [PATCH] fix(gemini): Correct MIME type for jpg images --- src/llm_models/model_client/gemini_client.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/llm_models/model_client/gemini_client.py b/src/llm_models/model_client/gemini_client.py index a74b466f1..ae0747bcb 100644 --- a/src/llm_models/model_client/gemini_client.py +++ b/src/llm_models/model_client/gemini_client.py @@ -82,8 +82,11 @@ def _convert_messages( content: List[Part] = [] for item in message.content: if isinstance(item, tuple): + image_format = item[0].lower() + if image_format == "jpg": + image_format = "jpeg" content.append( - Part.from_bytes(data=base64.b64decode(item[1]), mime_type=f"image/{item[0].lower()}") + Part.from_bytes(data=base64.b64decode(item[1]), mime_type=f"image/{image_format}") ) elif isinstance(item, str): content.append(Part.from_text(text=item))