mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-25 06:44:59 +00:00
added assistant:ok turn after image message
This commit is contained in:
parent
2824acb5c2
commit
369d746f2f
2 changed files with 11 additions and 15 deletions
|
@ -23,8 +23,8 @@ from aider.io import InputOutput
|
||||||
from aider.repo import GitRepo
|
from aider.repo import GitRepo
|
||||||
from aider.repomap import RepoMap
|
from aider.repomap import RepoMap
|
||||||
from aider.sendchat import send_with_retries
|
from aider.sendchat import send_with_retries
|
||||||
|
|
||||||
from aider.utils import is_image_file
|
from aider.utils import is_image_file
|
||||||
|
|
||||||
from ..dump import dump # noqa: F401
|
from ..dump import dump # noqa: F401
|
||||||
|
|
||||||
|
|
||||||
|
@ -347,7 +347,10 @@ class Coder:
|
||||||
|
|
||||||
images_message = self.get_images_message()
|
images_message = self.get_images_message()
|
||||||
if images_message is not None:
|
if images_message is not None:
|
||||||
files_messages.append(images_message)
|
files_messages += [
|
||||||
|
images_message,
|
||||||
|
dict(role="assistant", content="Ok."),
|
||||||
|
]
|
||||||
|
|
||||||
return files_messages
|
return files_messages
|
||||||
|
|
||||||
|
@ -359,21 +362,14 @@ class Coder:
|
||||||
for fname, content in self.get_abs_fnames_content():
|
for fname, content in self.get_abs_fnames_content():
|
||||||
if is_image_file(fname):
|
if is_image_file(fname):
|
||||||
image_url = f"data:image/{Path(fname).suffix.lstrip('.')};base64,{content}"
|
image_url = f"data:image/{Path(fname).suffix.lstrip('.')};base64,{content}"
|
||||||
image_messages.append({
|
image_messages.append(
|
||||||
"type": "image_url",
|
{"type": "image_url", "image_url": {"url": image_url, "detail": "high"}}
|
||||||
"image_url": {
|
)
|
||||||
"url": image_url,
|
|
||||||
"detail": "high"
|
|
||||||
}
|
|
||||||
})
|
|
||||||
|
|
||||||
if not image_messages:
|
if not image_messages:
|
||||||
return None
|
return None
|
||||||
|
|
||||||
return {
|
return {"role": "user", "content": image_messages}
|
||||||
"role": "user",
|
|
||||||
"content": image_messages
|
|
||||||
}
|
|
||||||
|
|
||||||
def run(self, with_message=None):
|
def run(self, with_message=None):
|
||||||
while True:
|
while True:
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue