scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-13 08:52:02 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 174

test(messages): expect rust to drop a null thinking signature on retry (#51461)

Browse source
This commit is contained in:
Nate Armstrong 2026-11-12 14:04:53 -08:00 • committed by GitHub
parent 6cc56fcc26
commit 4c9d72ded6
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 8 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
tests/integration/messages_endpoint/providers/anthropic/test_anthropic_thinking_signature_retry_wire.py
View file

@ -46,6 +46,13 @@ def test_missing_thinking_signature_400_retries_once_without_thinking_blocks_and
"role": "assistant",
"content": [{"type": "tool_use", "id": "call-1", "name": "lookup", "input": {"key": "value"}}],
}
unsigned_thinking_turn: Final = {
"role": "assistant",
"content": [
{"type": "thinking", "thinking": "replayed from another provider"},
{"type": "tool_use", "id": "call-1", "name": "lookup", "input": {"key": "value"}},
],
}
def respond(request: Request) -> Reply:
assert request.method == "POST" and request.target == "/v1/messages"
@ -54,7 +61,7 @@ def test_missing_thinking_signature_400_retries_once_without_thinking_blocks_and
assert body["model"] == MODEL
assert body["tools"] == list(TOOLS), body
if body["messages"][3]["content"][0]["type"] == "thinking":
assert body["messages"] == list(history), body
assert body["messages"] == [*history[:3], unsigned_thinking_turn, history[4]], body
assert body["thinking"] == {"type": "enabled", "budget_tokens": 1024}, body
return Reply(status=400, body=SIGNATURE_ERROR)
assert body["messages"] == [*history[:3], tool_use_only_turn, history[4]], body