phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 03:04:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add supports_pdf_input: true to Claude 3.7 bedrock models

Browse source
This commit is contained in:
RupertoXTI 2025-04-11 14:40:30 -04:00
parent 8ecd9ede81
commit 31867f64c6
5 changed files with 21 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
litellm/__init__.py
View file

@ -350,6 +350,7 @@ project = None
config_path = None config_path = None
vertex_ai_safety_settings: Optional[dict] = None vertex_ai_safety_settings: Optional[dict] = None
BEDROCK_CONVERSE_MODELS = [ BEDROCK_CONVERSE_MODELS = [
"anthropic.claude-3-7-sonnet-20250219-v1:0",
"anthropic.claude-3-5-haiku-20241022-v1:0", "anthropic.claude-3-5-haiku-20241022-v1:0",
"anthropic.claude-3-5-sonnet-20241022-v2:0", "anthropic.claude-3-5-sonnet-20241022-v2:0",
"anthropic.claude-3-5-sonnet-20240620-v1:0", "anthropic.claude-3-5-sonnet-20240620-v1:0",

2
litellm/model_prices_and_context_window_backup.json
View file

@ -7780,6 +7780,7 @@
"supports_assistant_prefill": true, "supports_assistant_prefill": true,
"supports_prompt_caching": true, "supports_prompt_caching": true,
"supports_response_schema": true, "supports_response_schema": true,
"supports_pdf_input": true,
"supports_tool_choice": true "supports_tool_choice": true
}, },
"anthropic.claude-3-5-sonnet-20241022-v2:0": { "anthropic.claude-3-5-sonnet-20241022-v2:0": {
@ -7897,6 +7898,7 @@
"supports_assistant_prefill": true, "supports_assistant_prefill": true,
"supports_prompt_caching": true, "supports_prompt_caching": true,
"supports_response_schema": true, "supports_response_schema": true,
"supports_pdf_input": true,
"supports_tool_choice": true "supports_tool_choice": true
}, },
"us.anthropic.claude-3-haiku-20240307-v1:0": { "us.anthropic.claude-3-haiku-20240307-v1:0": {

2
model_prices_and_context_window.json
View file

@ -7780,6 +7780,7 @@
"supports_assistant_prefill": true, "supports_assistant_prefill": true,
"supports_prompt_caching": true, "supports_prompt_caching": true,
"supports_response_schema": true, "supports_response_schema": true,
"supports_pdf_input": true,
"supports_tool_choice": true "supports_tool_choice": true
}, },
"anthropic.claude-3-5-sonnet-20241022-v2:0": { "anthropic.claude-3-5-sonnet-20241022-v2:0": {
@ -7897,6 +7898,7 @@
"supports_assistant_prefill": true, "supports_assistant_prefill": true,
"supports_prompt_caching": true, "supports_prompt_caching": true,
"supports_response_schema": true, "supports_response_schema": true,
"supports_pdf_input": true,
"supports_tool_choice": true "supports_tool_choice": true
}, },
"us.anthropic.claude-3-haiku-20240307-v1:0": { "us.anthropic.claude-3-haiku-20240307-v1:0": {

15
tests/litellm_utils_tests/test_utils.py
View file

@ -2102,3 +2102,18 @@ def test_get_provider_audio_transcription_config():
config = ProviderConfigManager.get_provider_audio_transcription_config( config = ProviderConfigManager.get_provider_audio_transcription_config(
model="whisper-1", provider=provider 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

1
tests/local_testing/whitelisted_bedrock_models.txt
View file

@ -27,6 +27,7 @@ anthropic.claude-3-5-haiku-20241022-v1:0
anthropic.claude-3-opus-20240229-v1:0 anthropic.claude-3-opus-20240229-v1:0
us.anthropic.claude-3-sonnet-20240229-v1:0 us.anthropic.claude-3-sonnet-20240229-v1:0
us.anthropic.claude-3-5-sonnet-20240620-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-5-sonnet-20241022-v2:0
us.anthropic.claude-3-haiku-20240307-v1:0 us.anthropic.claude-3-haiku-20240307-v1:0
us.anthropic.claude-3-5-haiku-20241022-v1:0 us.anthropic.claude-3-5-haiku-20241022-v1:0