fix: Update test assertions to match actual model completion call parameters

This commit is contained in:
Paul Gauthier (aider) 2025-02-06 08:20:38 -08:00
parent 016aa87e34
commit 11a233da84

View file

@ -224,10 +224,8 @@ class TestModels(unittest.TestCase):
model=model.name, model=model.name,
messages=messages, messages=messages,
stream=False, stream=False,
temperature=ANY, temperature=0,
num_ctx=expected_ctx, num_ctx=expected_ctx,
tools=None,
tool_choice=None,
) )
@patch("aider.models.litellm.completion") @patch("aider.models.litellm.completion")
@ -261,9 +259,7 @@ class TestModels(unittest.TestCase):
model=model.name, model=model.name,
messages=messages, messages=messages,
stream=False, stream=False,
temperature=ANY, temperature=0,
tools=None,
tool_choice=None,
) )
self.assertNotIn("num_ctx", mock_completion.call_args.kwargs) self.assertNotIn("num_ctx", mock_completion.call_args.kwargs)