diff --git a/litellm/tests/test_provider_specific_config.py b/litellm/tests/test_provider_specific_config.py index 5e5d19c786..dcb4dcb4c7 100644 --- a/litellm/tests/test_provider_specific_config.py +++ b/litellm/tests/test_provider_specific_config.py @@ -53,7 +53,7 @@ def claude_test_completion(): try: # OVERRIDE WITH DYNAMIC MAX TOKENS response_1 = litellm.completion( - model="claude-instant-1", + model="claude-instant-1.2", messages=[{"content": "Hello, how are you?", "role": "user"}], max_tokens=10, ) @@ -63,7 +63,7 @@ def claude_test_completion(): # USE CONFIG TOKENS response_2 = litellm.completion( - model="claude-instant-1", + model="claude-instant-1.2", messages=[{"content": "Hello, how are you?", "role": "user"}], ) # Add any assertions here to check the response @@ -74,7 +74,7 @@ def claude_test_completion(): try: response_3 = litellm.completion( - model="claude-instant-1", + model="claude-instant-1.2", messages=[{"content": "Hello, how are you?", "role": "user"}], n=2, ) diff --git a/litellm/tests/test_router.py b/litellm/tests/test_router.py index 127caf223b..7c182ee686 100644 --- a/litellm/tests/test_router.py +++ b/litellm/tests/test_router.py @@ -933,7 +933,7 @@ def test_router_anthropic_key_dynamic(): { "model_name": "anthropic-claude", "litellm_params": { - "model": "claude-instant-1", + "model": "claude-instant-1.2", "api_key": anthropic_api_key, }, } diff --git a/litellm/tests/test_router_timeout.py b/litellm/tests/test_router_timeout.py index 139914f6df..3816c649e9 100644 --- a/litellm/tests/test_router_timeout.py +++ b/litellm/tests/test_router_timeout.py @@ -35,7 +35,7 @@ def test_router_timeouts(): { "model_name": "anthropic-claude-instant-1.2", "litellm_params": { - "model": "claude-instant-1", + "model": "claude-instant-1.2", "api_key": "os.environ/ANTHROPIC_API_KEY", }, "tpm": 20000,