diff --git a/litellm/__init__.py b/litellm/__init__.py index 1252ee3d6f..d1633430f8 100644 --- a/litellm/__init__.py +++ b/litellm/__init__.py @@ -350,6 +350,7 @@ project = None config_path = None vertex_ai_safety_settings: Optional[dict] = None BEDROCK_CONVERSE_MODELS = [ + "anthropic.claude-3-7-sonnet-20250219-v1:0", "anthropic.claude-3-5-haiku-20241022-v1:0", "anthropic.claude-3-5-sonnet-20241022-v2:0", "anthropic.claude-3-5-sonnet-20240620-v1:0", diff --git a/litellm/model_prices_and_context_window_backup.json b/litellm/model_prices_and_context_window_backup.json index e7485f7378..0db03fcf73 100644 --- a/litellm/model_prices_and_context_window_backup.json +++ b/litellm/model_prices_and_context_window_backup.json @@ -7780,6 +7780,7 @@ "supports_assistant_prefill": true, "supports_prompt_caching": true, "supports_response_schema": true, + "supports_pdf_input": true, "supports_tool_choice": true }, "anthropic.claude-3-5-sonnet-20241022-v2:0": { @@ -7897,6 +7898,7 @@ "supports_assistant_prefill": true, "supports_prompt_caching": true, "supports_response_schema": true, + "supports_pdf_input": true, "supports_tool_choice": true }, "us.anthropic.claude-3-haiku-20240307-v1:0": { diff --git a/model_prices_and_context_window.json b/model_prices_and_context_window.json index e7485f7378..0db03fcf73 100644 --- a/model_prices_and_context_window.json +++ b/model_prices_and_context_window.json @@ -7780,6 +7780,7 @@ "supports_assistant_prefill": true, "supports_prompt_caching": true, "supports_response_schema": true, + "supports_pdf_input": true, "supports_tool_choice": true }, "anthropic.claude-3-5-sonnet-20241022-v2:0": { @@ -7897,6 +7898,7 @@ "supports_assistant_prefill": true, "supports_prompt_caching": true, "supports_response_schema": true, + "supports_pdf_input": true, "supports_tool_choice": true }, "us.anthropic.claude-3-haiku-20240307-v1:0": { diff --git a/tests/litellm_utils_tests/test_utils.py b/tests/litellm_utils_tests/test_utils.py index 3088fa250f..579a611de1 100644 --- a/tests/litellm_utils_tests/test_utils.py +++ b/tests/litellm_utils_tests/test_utils.py @@ -2102,3 +2102,18 @@ def test_get_provider_audio_transcription_config(): config = ProviderConfigManager.get_provider_audio_transcription_config( model="whisper-1", provider=provider ) + + +@pytest.mark.parametrize( + "model, expected_bool", + [ + ("anthropic.claude-3-7-sonnet-20250219-v1:0", True), + ("us.anthropic.claude-3-7-sonnet-20250219-v1:0", True), + ], +) + +def test_claude_3_7_sonnet_supports_pdf_input(model, expected_bool): + from litellm.utils import supports_pdf_input + + assert supports_pdf_input("anthropic.claude-3-7-sonnet-20250219-v1:0") == True + assert supports_pdf_input("us.anthropic.claude-3-7-sonnet-20250219-v1:0") == True diff --git a/tests/local_testing/whitelisted_bedrock_models.txt b/tests/local_testing/whitelisted_bedrock_models.txt index 8ad500b4c5..c529fa0160 100644 --- a/tests/local_testing/whitelisted_bedrock_models.txt +++ b/tests/local_testing/whitelisted_bedrock_models.txt @@ -27,6 +27,7 @@ anthropic.claude-3-5-haiku-20241022-v1:0 anthropic.claude-3-opus-20240229-v1:0 us.anthropic.claude-3-sonnet-20240229-v1:0 us.anthropic.claude-3-5-sonnet-20240620-v1:0 +us.anthropic.claude-3-7-sonnet-20250219-v1:0 us.anthropic.claude-3-5-sonnet-20241022-v2:0 us.anthropic.claude-3-haiku-20240307-v1:0 us.anthropic.claude-3-5-haiku-20241022-v1:0