gardel/Mofox-Core
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #1177 from Eric-Terminal/fix/gemini-mime-type

Browse Source 
fix(gemini): Correct MIME type for jpg images
This commit is contained in:
UnCLAS-Prommer
2025-08-12 20:59:25 +08:00
committed by GitHub
parent 04bd05c1fe 52cbaca6c2
commit c0c1cea6dd
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/llm_models/model_client/gemini_client.py
View File

@@ -82,8 +82,9 @@ def _convert_messages(
content: List[Part] = [] content: List[Part] = []
for item in message.content: for item in message.content:
if isinstance(item, tuple): if isinstance(item, tuple):
image_format = "jpeg" if item[0].lower() == "jpg" else item[0].lower()
content.append( 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): elif isinstance(item, str):
content.append(Part.from_text(text=item)) content.append(Part.from_text(text=item))