mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-25 14:55:00 +00:00
fix: Update test assertions to match actual model completion call parameters
This commit is contained in:
parent
016aa87e34
commit
11a233da84
1 changed files with 2 additions and 6 deletions
|
@ -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)
|
||||||
|
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue