added assistant:ok turn after image message

This commit is contained in:
Paul Gauthier 2024-01-02 09:40:03 -08:00
parent 2824acb5c2
commit 369d746f2f
2 changed files with 11 additions and 15 deletions

View file

@ -23,8 +23,8 @@ from aider.io import InputOutput
from aider.repo import GitRepo
from aider.repomap import RepoMap
from aider.sendchat import send_with_retries
from aider.utils import is_image_file
from ..dump import dump # noqa: F401
@ -347,7 +347,10 @@ class Coder:
images_message = self.get_images_message()
if images_message is not None:
files_messages.append(images_message)
files_messages += [
images_message,
dict(role="assistant", content="Ok."),
]
return files_messages
@ -359,21 +362,14 @@ class Coder:
for fname, content in self.get_abs_fnames_content():
if is_image_file(fname):
image_url = f"data:image/{Path(fname).suffix.lstrip('.')};base64,{content}"
image_messages.append({
"type": "image_url",
"image_url": {
"url": image_url,
"detail": "high"
}
})
image_messages.append(
{"type": "image_url", "image_url": {"url": image_url, "detail": "high"}}
)
if not image_messages:
return None
return {
"role": "user",
"content": image_messages
}
return {"role": "user", "content": image_messages}
def run(self, with_message=None):
while True: