scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-30 06:43:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 383

test(guardrails): verify disconnect inspection retains tool context

Browse source
This commit is contained in:
albertbausili 2026-10-16 10:00:45 +02:00
parent c99cc59dda
commit b2b61d2199
1 changed files with 6 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
tests/unit/proxy/guardrails/guardrail_hooks/unified_guardrails/test_unified_guardrail.py
View file

@ -2706,7 +2706,12 @@ class TestStreamingClientDisconnectScan:
assert [call.function.name for call in released[0].choices[0].delta.tool_calls] == ["get_weather"]
assert guardrail.received_texts == [["BLOCKME"]], guardrail.received_texts
assert guardrail.received_tool_calls == [], guardrail.received_tool_calls
assert guardrail.received_tool_calls == [[{
"id": "call_1",
"type": "function",
"index": 0,
"function": {"name": "get_weather", "arguments": '{"city": "Paris"}'},
}]]
@pytest.mark.asyncio
async def test_closing_after_a_released_tool_call_under_incremental_diff_scans_no_held_back_text(self):