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

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

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(
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