dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-06-02 02:34:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

feat: add image file support for read-only files

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-11-26 19:48:42 -08:00
parent 34b8c3f47c
commit 4be3728273
1 changed files with 13 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

15
aider/coders/base_coder.py
View file

@ -665,6 +665,8 @@ class Coder:
def get_readonly_files_messages(self): def get_readonly_files_messages(self):
readonly_messages = [] readonly_messages = []
# Handle non-image files
read_only_content = self.get_read_only_files_content() read_only_content = self.get_read_only_files_content()
if read_only_content: if read_only_content:
readonly_messages += [ readonly_messages += [
@ -676,6 +678,15 @@ class Coder:
content="Ok, I will use these files as references.", content="Ok, I will use these files as references.",
), ),
] ]
# Handle image files
images_message = self.get_images_message(self.abs_read_only_fnames)
if images_message is not None:
readonly_messages += [
images_message,
dict(role="assistant", content="Ok, I will use these images as references."),
]
return readonly_messages return readonly_messages
def get_chat_files_messages(self): def get_chat_files_messages(self):
@ -697,7 +708,7 @@ class Coder:
dict(role="assistant", content=files_reply), dict(role="assistant", content=files_reply),
] ]
images_message = self.get_images_message() images_message = self.get_images_message(self.abs_fnames)
if images_message is not None: if images_message is not None:
chat_files_messages += [ chat_files_messages += [
images_message, images_message,
@ -706,7 +717,7 @@ class Coder:
return chat_files_messages return chat_files_messages
def get_images_message(self): def get_images_message(self, fnames):
supports_images = self.main_model.info.get("supports_vision") supports_images = self.main_model.info.get("supports_vision")
supports_pdfs = self.main_model.info.get("supports_pdf_input") supports_pdfs = self.main_model.info.get("supports_pdf_input")