From 44171417e3cd2ce064586eb0d0ecf7d1baa38527 Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Thu, 6 Feb 2025 11:46:39 -0800 Subject: [PATCH] fix: Update test assertions to include timeout parameter --- tests/basic/test_models.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tests/basic/test_models.py b/tests/basic/test_models.py index f54cbca6c..6d156261e 100644 --- a/tests/basic/test_models.py +++ b/tests/basic/test_models.py @@ -391,6 +391,7 @@ And this text should remain""" stream=False, temperature=0, num_ctx=expected_ctx, + timeout=600, ) @patch("aider.models.litellm.completion") @@ -408,6 +409,7 @@ And this text should remain""" stream=False, temperature=0, num_ctx=4096, + timeout=600, ) @patch("aider.models.litellm.completion") @@ -423,6 +425,7 @@ And this text should remain""" messages=messages, stream=False, temperature=0, + timeout=600, ) self.assertNotIn("num_ctx", mock_completion.call_args.kwargs) @@ -452,6 +455,7 @@ And this text should remain""" messages=messages, stream=False, temperature=0, + timeout=600, ) # Test use_temperature=False doesn't send temperature