From ee6604442572e49111478a07e19dc55aa4b2a5af Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Tue, 4 Feb 2025 12:02:38 -0800 Subject: [PATCH] refactor: Update send_completion calls to use model method syntax --- tests/basic/test_sendchat.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/basic/test_sendchat.py b/tests/basic/test_sendchat.py index 6fe0d807e..27effc87f 100644 --- a/tests/basic/test_sendchat.py +++ b/tests/basic/test_sendchat.py @@ -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