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

feat: add get_system_info method to Coder base class

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-09-05 12:59:12 -07:00
parent b57468a963
commit 45855bd96c
2 changed files with 8 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

6
aider/coders/base_coder.py
View file

@ -919,6 +919,12 @@ class Coder:
return platform_text return platform_text
def get_system_info(self):
system_info = self.get_platform_info()
if self.chat_language:
system_info += f"- Chat language: {self.chat_language}\n"
return system_info
def fmt_system_prompt(self, prompt): def fmt_system_prompt(self, prompt):
lazy_prompt = self.gpt_prompts.lazy_prompt if self.main_model.lazy else "" lazy_prompt = self.gpt_prompts.lazy_prompt if self.main_model.lazy else ""
platform_text = self.get_platform_info() platform_text = self.get_platform_info()

3
tests/basic/test_main.py
View file

@ -628,7 +628,8 @@ class TestMain(TestCase):
output=DummyOutput(), output=DummyOutput(),
return_coder=True, return_coder=True,
) )
self.assertIn("Spanish", coder.get_system_info()) system_info = coder.get_system_info()
self.assertIn("Chat language: Spanish", system_info)
@patch("git.Repo.init") @patch("git.Repo.init")
def test_main_exit_with_git_command_not_found(self, mock_git_init): def test_main_exit_with_git_command_not_found(self, mock_git_init):