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