dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-31 17:55:01 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 9

refactor: Update send_completion calls to use model method syntax

Browse source
This commit is contained in:
Paul Gauthier (aider) 2025-02-04 12:02:38 -08:00
parent 30d56e1af0
commit ee66044425
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/basic/test_sendchat.py
View file

@ -48,8 +48,8 @@ class TestSendChat(unittest.TestCase):
mock_completion.return_value = mock_response
# Test basic send_completion
hash_obj, response = send_completion(
self.mock_model, self.mock_messages, functions=None, stream=False
hash_obj, response = Model(self.mock_model).send_completion(
self.mock_messages, functions=None, stream=False
)
assert response == mock_response
@ -59,8 +59,8 @@ class TestSendChat(unittest.TestCase):
def test_send_completion_with_functions(self, mock_completion):
mock_function = {"name": "test_function", "parameters": {"type": "object"}}
hash_obj, response = send_completion(
self.mock_model, self.mock_messages, functions=[mock_function], stream=False
hash_obj, response = Model(self.mock_model).send_completion(
self.mock_messages, functions=[mock_function], stream=False
)
# Verify function was properly included in tools