Update src/llm_models/model_client/gemini_client.py

Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
2025-08-12 20:29:22 +08:00
parent c59f8de306
commit 52cbaca6c2
1 changed files with 1 additions and 3 deletions
--- a/src/llm_models/model_client/gemini_client.py
+++ b/src/llm_models/model_client/gemini_client.py
@@ -82,9 +82,7 @@ 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"
+                    image_format = "jpeg" if item[0].lower() == "jpg" else item[0].lower()
                    content.append(
                        Part.from_bytes(data=base64.b64decode(item[1]), mime_type=f"image/{image_format}")
                    )