From 11a233da84bc050f7eb4ed7f0f1b59039c02e267 Mon Sep 17 00:00:00 2001
From: "Paul Gauthier (aider)" <paul@aider.chat>
Date: Thu, 6 Feb 2025 08:20:38 -0800
Subject: [PATCH] fix: Update test assertions to match actual model completion
 call parameters

---
 tests/basic/test_models.py | 8 ++------
 1 file changed, 2 insertions(+), 6 deletions(-)

diff --git a/tests/basic/test_models.py b/tests/basic/test_models.py
index 3532fd31f..66bcf3095 100644
--- a/tests/basic/test_models.py
+++ b/tests/basic/test_models.py
@@ -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)