针对gemini进行图像格式转换

2025-08-22 03:54:12 +08:00 · 2025-08-22 03:54:12 +08:00 · 3c9c83e00e
parent 05f6fdabe4
commit 3c9c83e00e
1 changed files with 1 additions and 2 deletions
--- a/src/llm_models/model_client/openai_client.py
+++ b/src/llm_models/model_client/openai_client.py
@ -60,8 +60,7 @@ def _convert_messages(messages: list[Message]) -> list[ChatCompletionMessagePara
            content = []
            for item in message.content:
                if isinstance(item, tuple):
-                    # 修复图像格式处理
-                    image_format = "jpeg" if item[0].lower() == "jpg" else item[0].lower()
+                    image_format = "jpeg" if "gemini-" in model_info.model_identifier and item[0].lower() == "jpg" else item[0].lower()
                    content.append(
                        {
                            "type": "image_url",