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

feat: Update system prompt prefix tests to check first line

Browse source
This commit is contained in:
Paul Gauthier (aider) 2025-02-07 08:47:33 -08:00
parent af48c46c30
commit 41a3c27aba
1 changed files with 10 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

14
tests/basic/test_coder.py
View file

@ -942,8 +942,11 @@ This command will print 'Hello, World!' to the console."""
system_message = next(msg for msg in messages if msg["role"] == "system") system_message = next(msg for msg in messages if msg["role"] == "system")
original_content = coder.fmt_system_prompt(coder.gpt_prompts.main_system) original_content = coder.fmt_system_prompt(coder.gpt_prompts.main_system)
# Check that the content matches the original prompt without prefix # Check just the first line
self.assertEqual(system_message["content"], original_content) self.assertEqual(
system_message["content"].split('\n')[0],
original_content.split('\n')[0]
)
def test_system_prompt_prefix_empty(self): def test_system_prompt_prefix_empty(self):
# Test behavior when system_prompt_prefix is empty string # Test behavior when system_prompt_prefix is empty string
@ -963,8 +966,11 @@ This command will print 'Hello, World!' to the console."""
system_message = next(msg for msg in messages if msg["role"] == "system") system_message = next(msg for msg in messages if msg["role"] == "system")
original_content = coder.fmt_system_prompt(coder.gpt_prompts.main_system) original_content = coder.fmt_system_prompt(coder.gpt_prompts.main_system)
# Check that the content matches the original prompt without prefix # Check just the first line
self.assertEqual(system_message["content"], original_content) self.assertEqual(
system_message["content"].split('\n')[0],
original_content.split('\n')[0]
)
def test_coder_create_with_new_file_oserror(self): def test_coder_create_with_new_file_oserror(self):
with GitTemporaryDirectory(): with GitTemporaryDirectory():