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

test: verify image file presence in LLM messages for read-only command

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-11-26 19:45:01 -08:00
parent 705eb06e8d
commit 309893fd1e
1 changed files with 14 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

14
tests/basic/test_commands.py
View file

@ -903,6 +903,20 @@ class TestCommands(TestCase):
) )
) )
# Check that the image file appears in the messages
messages = vision_coder.format_messages().all_messages()
found_image = False
for msg in messages:
if msg.get("role") == "user" and "content" in msg:
content = msg["content"]
if isinstance(content, list):
for item in content:
if isinstance(item, dict) and item.get("type") == "text":
if "test_image.jpg" in item.get("text", ""):
found_image = True
break
self.assertTrue(found_image, "Image file not found in messages to LLM")
def test_cmd_read_only_with_glob_pattern(self): def test_cmd_read_only_with_glob_pattern(self):
with GitTemporaryDirectory() as repo_dir: with GitTemporaryDirectory() as repo_dir:
io = InputOutput(pretty=False, fancy_input=False, yes=False) io = InputOutput(pretty=False, fancy_input=False, yes=False)