phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-24 18:24:20 +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
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",

2
litellm/model_prices_and_context_window_backup.json
View file

@ -8585,6 +8585,7 @@
"supports_assistant_prefill": true,
"supports_prompt_caching": true,
"supports_response_schema": true,
"supports_pdf_input": true,
"supports_reasoning": true,
"supports_tool_choice": true
},
@ -8703,6 +8704,7 @@
"supports_assistant_prefill": true,
"supports_prompt_caching": true,
"supports_response_schema": true,
"supports_pdf_input": true,
"supports_tool_choice": true,
"supports_reasoning": true
},

2
model_prices_and_context_window.json
View file

@ -8585,6 +8585,7 @@
"supports_assistant_prefill": true,
"supports_prompt_caching": true,
"supports_response_schema": true,
"supports_pdf_input": true,
"supports_reasoning": true,
"supports_tool_choice": true
},
@ -8703,6 +8704,7 @@
"supports_assistant_prefill": true,
"supports_prompt_caching": true,
"supports_response_schema": true,
"supports_pdf_input": true,
"supports_tool_choice": 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():
"""
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)
assert len(valid_models) > 0
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
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