From 52510c7da59d56e21b070bea24830cbbd7415d59 Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Fri, 30 May 2025 15:07:29 -0700 Subject: [PATCH] test: Update OpenRouter default model expectations --- tests/basic/test_onboarding.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/tests/basic/test_onboarding.py b/tests/basic/test_onboarding.py index 5c90a5648..ad32d06f8 100644 --- a/tests/basic/test_onboarding.py +++ b/tests/basic/test_onboarding.py @@ -94,7 +94,7 @@ class TestOnboarding(unittest.TestCase): def test_try_select_default_model_openrouter_free(self, mock_check_tier): """Test OpenRouter free model selection.""" self.assertEqual( - try_to_select_default_model(), "openrouter/google/gemini-2.5-pro-exp-03-25:free" + try_to_select_default_model(), "openrouter/deepseek/deepseek-r1:free" ) mock_check_tier.assert_called_once_with("or_key") @@ -102,7 +102,7 @@ class TestOnboarding(unittest.TestCase): @patch.dict(os.environ, {"OPENROUTER_API_KEY": "or_key"}, clear=True) def test_try_select_default_model_openrouter_paid(self, mock_check_tier): """Test OpenRouter paid model selection.""" - self.assertEqual(try_to_select_default_model(), "openrouter/anthropic/claude-3.7-sonnet") + self.assertEqual(try_to_select_default_model(), "openrouter/anthropic/claude-sonnet-4") mock_check_tier.assert_called_once_with("or_key") @patch("aider.onboarding.check_openrouter_tier") @@ -146,7 +146,7 @@ class TestOnboarding(unittest.TestCase): ) def test_try_select_default_model_priority_openrouter(self, mock_check_tier): """Test OpenRouter key takes priority.""" - self.assertEqual(try_to_select_default_model(), "openrouter/anthropic/claude-3.7-sonnet") + self.assertEqual(try_to_select_default_model(), "openrouter/anthropic/claude-sonnet-4") mock_check_tier.assert_called_once_with("or_key") @patch("aider.onboarding.check_openrouter_tier") @@ -346,7 +346,7 @@ class TestOnboarding(unittest.TestCase): @patch( "aider.onboarding.try_to_select_default_model", - side_effect=[None, "openrouter/google/gemini-2.5-pro-exp-03-25:free"], + side_effect=[None, "openrouter/deepseek/deepseek-r1:free"], ) # Fails first, succeeds after oauth @patch( "aider.onboarding.offer_openrouter_oauth", return_value=True @@ -360,7 +360,7 @@ class TestOnboarding(unittest.TestCase): selected_model = select_default_model(args, io_mock, analytics_mock) - self.assertEqual(selected_model, "openrouter/google/gemini-2.5-pro-exp-03-25:free") + self.assertEqual(selected_model, "openrouter/deepseek/deepseek-r1:free") self.assertEqual(mock_try_select.call_count, 2) # Called before and after oauth mock_offer_oauth.assert_called_once_with(io_mock, analytics_mock) # Only one warning is expected: "No LLM model..."