scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-09 03:18:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 109

test(anthropic): expect the per-turn-control beta to reach Bedrock Invoke

Browse source 
Generated with AI

Co-Authored-By: Claude Code
This commit is contained in:
Scott Wilson 2026-09-28 16:24:23 -04:00
parent 71e538d413
commit 107a07f615
1 changed files with 4 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
tests/unit/llms/anthropic/pass_through/messages/test_anthropic_messages_per_turn_control.py
View file

@ -95,15 +95,16 @@ def test_added_per_turn_control_beta_survives_the_anthropic_allowlist():
assert PER_TURN_CONTROL in _betas(filtered)
@pytest.mark.parametrize("provider", ["bedrock", "bedrock_converse", "vertex_ai", "databricks"])
@pytest.mark.parametrize("provider", ["bedrock_converse", "vertex_ai", "databricks"])
def test_per_turn_control_beta_is_dropped_for_providers_without_it(provider):
filtered = update_headers_with_filtered_beta(headers={"anthropic-beta": PER_TURN_CONTROL}, provider=provider)
assert "anthropic-beta" not in filtered
def test_per_turn_control_beta_is_forwarded_for_azure_ai():
filtered = update_headers_with_filtered_beta(headers={"anthropic-beta": PER_TURN_CONTROL}, provider="azure_ai")
@pytest.mark.parametrize("provider", ["azure_ai", "bedrock"])
def test_per_turn_control_beta_is_forwarded(provider):
filtered = update_headers_with_filtered_beta(headers={"anthropic-beta": PER_TURN_CONTROL}, provider=provider)
assert _betas(filtered) == {PER_TURN_CONTROL}