dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-24 14:25:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 9

Implemented a more robust method to determine the image file type using the mimetypes module in the get_images_message function.

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-07-03 16:15:07 -03:00 • committed by Paul Gauthier
parent 0ef842dd53
commit 790c2f0330
1 changed files with 7 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

12
aider/coders/base_coder.py
View file

@ -2,6 +2,7 @@
import hashlib
import json
import mimetypes
import os
import re
import sys
@ -575,11 +576,12 @@ class Coder:
image_messages = []
for fname, content in self.get_abs_fnames_content():
if is_image_file(fname):
# todo: use a proper tool to convert fname to media type
image_url = f"data:image/{Path(fname).suffix.lstrip('.')};base64,{content}"
image_messages.append(
{"type": "image_url", "image_url": {"url": image_url, "detail": "high"}}
)
mime_type, _ = mimetypes.guess_type(fname)
if mime_type and mime_type.startswith("image/"):
image_url = f"data:{mime_type};base64,{content}"
image_messages.append(
{"type": "image_url", "image_url": {"url": image_url, "detail": "high"}}
)
if not image_messages:
return None