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

Merge 581f7c24f4 into b82af5b826

Browse source
This commit is contained in:
Ruperto A. Martinez 2025-04-24 00:55:39 -07:00 • committed by GitHub
commit e211329642
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
5 changed files with 21 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

1
litellm/__init__.py
View file

@ -353,6 +353,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

@ -8585,6 +8585,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_reasoning": true, "supports_reasoning": true,
"supports_tool_choice": true "supports_tool_choice": true
}, },
@ -8703,6 +8704,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,
"supports_reasoning": true "supports_reasoning": true
}, },

2
model_prices_and_context_window.json
View file

@ -8585,6 +8585,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_reasoning": true, "supports_reasoning": true,
"supports_tool_choice": true "supports_tool_choice": true
}, },
@ -8703,6 +8704,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,
"supports_reasoning": true "supports_reasoning": true
}, },

16
tests/litellm_utils_tests/test_utils.py
View file

@ -2127,6 +2127,20 @@ def test_get_provider_audio_transcription_config():
) )
@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(model) == expected_bool
def test_get_valid_models_from_provider(): def test_get_valid_models_from_provider():
""" """
Test that get_valid_models returns the correct models for a given provider Test that get_valid_models returns the correct models for a given provider
@ -2178,5 +2192,5 @@ def test_get_valid_models_from_dynamic_api_key():
valid_models = get_valid_models(custom_llm_provider="anthropic", litellm_params=creds, check_provider_endpoint=True) valid_models = get_valid_models(custom_llm_provider="anthropic", litellm_params=creds, check_provider_endpoint=True)
assert len(valid_models) > 0 assert len(valid_models) > 0
assert "anthropic/claude-3-7-sonnet-20250219" in valid_models assert "anthropic/claude-3-7-sonnet-20250219" in valid_models

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