scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-08-28 05:25:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 8

test(deepseek): exercise replay through request transform

Browse source
This commit is contained in:
IvanShang 2026-08-21 17:21:42 +08:00
parent 375ea2483e
commit 3376a8464d
1 changed files with 14 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
tests/test_litellm/llms/deepseek/chat/test_deepseek_chat_transformation.py
View file

@ -106,7 +106,7 @@ async def test_async_transform_request_strips_unsupported_tools_from_body():
assert body["tools"][0]["function"]["name"] == "shell"
def test_fill_reasoning_content_warns_once_per_request_and_preserves_history():
def test_transform_request_warns_once_per_replayed_history_and_preserves_history():
messages = [
{"role": "user", "content": "Use both tools."},
{
@ -130,17 +130,20 @@ def test_fill_reasoning_content_warns_once_per_request_and_preserves_history():
warning_path = "litellm.llms.deepseek.chat.transformation.litellm.verbose_logger.warning"
with patch(warning_path) as warning:
first_result = config._fill_reasoning_content(messages)
warning.assert_called_once()
results = [
config.transform_request(
model="deepseek-reasoner",
messages=messages,
optional_params={"thinking": {"type": "enabled"}},
litellm_params={},
headers={},
)
for _ in range(2)
]
with patch(warning_path) as warning:
second_result = config._fill_reasoning_content(messages)
warning.assert_called_once()
assert [first_result[index]["reasoning_content"] for index in (1, 3)] == [" ", " "]
assert [second_result[index]["reasoning_content"] for index in (1, 3)] == [" ", " "]
assert first_result[0] is messages[0]
assert second_result[4] is messages[4]
assert warning.call_count == 2
for result in results:
assert [result["messages"][index]["reasoning_content"] for index in (1, 3)] == [" ", " "]
assert messages == original_messages