scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-07 02:59:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

litellm_fix(test): fix Bedrock tool search header test regression (#20135)

Browse source
This commit is contained in:
shin-bot-litellm 2026-01-31 00:44:47 -08:00 • committed by GitHub
parent 013b4701f4
commit 14a5706131
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 9 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
tests/test_litellm/llms/bedrock/chat/invoke_transformations/test_bedrock_chat_invoke_transformations_anthropic_claude3_transformation.py
View file

@ -335,11 +335,11 @@ def test_advanced_tool_use_header_translation_for_opus_4_5():
def test_advanced_tool_use_header_filtered_for_non_opus_4_5():
"""
Test that advanced-tool-use-2025-11-20 header is filtered out for non-Opus 4.5 models
without adding Bedrock-specific headers.
Test that advanced-tool-use-2025-11-20 header is filtered out for models
that don't support tool search on Bedrock.
The translation to tool-search-tool-2025-10-19 and tool-examples-2025-10-29 should
only happen for Claude Opus 4.5.
Tool search is supported on: Claude Opus 4.5, Claude Sonnet 4.5
Tool search is NOT supported on: Claude 3.5 Sonnet and earlier
"""
from litellm.llms.bedrock.messages.invoke_transformations.anthropic_claude3_transformation import (
AmazonAnthropicClaudeMessagesConfig,
@ -360,9 +360,9 @@ def test_advanced_tool_use_header_filtered_for_non_opus_4_5():
"anthropic-beta": "advanced-tool-use-2025-11-20"
}
# Test with Claude Sonnet 4.5 (not Opus 4.5)
# Test with Claude 3.5 Sonnet (does NOT support tool search on Bedrock)
result = config.transform_anthropic_messages_request(
model="anthropic.claude-sonnet-4-5-20250929-v1:0",
model="anthropic.claude-3-5-sonnet-20241022-v2:0",
messages=messages,
anthropic_messages_optional_request_params=anthropic_messages_optional_request_params,
litellm_params={},
@ -374,11 +374,11 @@ def test_advanced_tool_use_header_filtered_for_non_opus_4_5():
assert "advanced-tool-use-2025-11-20" not in beta_headers, \
"advanced-tool-use header should be removed for Bedrock"
# Verify Bedrock-specific headers were NOT added (only for Opus 4.5)
# Verify Bedrock-specific headers were NOT added (only for Opus 4.5 and Sonnet 4.5)
assert "tool-search-tool-2025-10-19" not in beta_headers, \
"tool-search-tool should not be added for non-Opus 4.5 models"
"tool-search-tool should not be added for models without tool search support"
assert "tool-examples-2025-10-29" not in beta_headers, \
"tool-examples should not be added for non-Opus 4.5 models"
"tool-examples should not be added for models without tool search support"
def test_advanced_tool_use_header_translation_with_multiple_beta_headers():