scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-26 01:12:21 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 99

test(bedrock): assert safeguard_results survive message_start and check the beta by membership

Browse source
This commit is contained in:
mateo-berri 2026-09-21 12:30:11 -07:00
parent 0e16050100
commit deca6aea79
1 changed files with 20 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
tests/test_litellm/llms/bedrock/messages/invoke_transformations/test_anthropic_claude3_transformation.py
View file

@ -1672,7 +1672,7 @@ def test_bedrock_messages_forwards_safeguards_with_dangerous_tool_use_beta(local
)
assert result["safeguards"] == safeguards
assert result["anthropic_beta"] == ["dangerous-tool-use-2026-09-03"]
assert "dangerous-tool-use-2026-09-03" in result["anthropic_beta"]
def test_bedrock_messages_stream_decoder_keeps_safeguard_results():
@ -1681,6 +1681,25 @@ def test_bedrock_messages_stream_decoder_keeps_safeguard_results():
tool_verdicts = {"toolu_01": {"type": "evaluated", "outcome": "not_flagged"}}
safeguard_results = [{"type": "dangerous_tool_use", "status": {"type": "available", "tool_uses": tool_verdicts}}]
message_start = decoder._chunk_parser(
{
"type": "message_start",
"message": {
"id": "msg_01",
"type": "message",
"role": "assistant",
"model": "claude-sonnet-5",
"content": [],
"stop_reason": None,
"usage": {"input_tokens": 3, "output_tokens": 0},
"safeguard_results": safeguard_results,
},
}
)
assert isinstance(message_start, dict)
assert message_start["message"]["safeguard_results"] == safeguard_results
message_delta = decoder._chunk_parser(
{
"type": "message_delta",