From dd48b740f975e3bff713ca0e816ba6cc66cc7b2d Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Tue, 26 Nov 2024 12:33:58 -0800 Subject: [PATCH] test: update Claude model name tests to use 3.5 version --- tests/basic/test_models.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/basic/test_models.py b/tests/basic/test_models.py index 8bf2ca44e..6aff2be7f 100644 --- a/tests/basic/test_models.py +++ b/tests/basic/test_models.py @@ -110,10 +110,10 @@ class TestModels(unittest.TestCase): self.assertEqual(model.name, "gpt-3.5-turbo") model = Model("sonnet") - self.assertEqual(model.name, "claude-3-sonnet-20241022") + self.assertEqual(model.name, "claude-3-5-sonnet-20241022") model = Model("haiku") - self.assertEqual(model.name, "claude-3-haiku-20241022") + self.assertEqual(model.name, "claude-3-5-haiku-20241022") model = Model("opus") self.assertEqual(model.name, "claude-3-opus-20240229")