scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-14 23:21:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 156

test: cover thinking with tool use replay

Browse source
This commit is contained in:
Genmin 2026-05-01 07:32:20 -07:00
parent 8989a6667a
commit dcde4a7baa
1 changed files with 30 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
tests/test_litellm/llms/anthropic/experimental_pass_through/responses_adapters/test_responses_adapters_transformation.py
View file

@ -414,6 +414,36 @@ class TestTranslateMessagesToResponsesInput:
{"type": "output_text", "text": "Here is the answer."}
]
def test_assistant_thinking_with_tool_use_keeps_function_call(self):
"""Thinking is dropped while adjacent tool_use blocks are still replayed."""
messages = [
{
"role": "assistant",
"content": [
{
"type": "thinking",
"thinking": "I should call the weather tool.",
"signature": "sig_abc",
},
{
"type": "tool_use",
"id": "toolu_03",
"name": "get_weather",
"input": {"location": "Paris"},
},
],
}
]
result = _translate_messages(messages)
assert result == [
{
"type": "function_call",
"call_id": "toolu_03",
"name": "get_weather",
"arguments": json.dumps({"location": "Paris"}),
}
]
def test_assistant_only_thinking_block_skipped(self):
"""Assistant messages with only thinking blocks are omitted on replay."""
messages = [