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(guardrails): inject AIDR HTTP client

Browse source
This commit is contained in:
abrekhov 2026-08-24 10:50:45 +00:00
parent e22eeab464
commit dcfc625a5f
1 changed files with 6 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
tests/test_litellm/proxy/guardrails/guardrail_hooks/test_crowdstrike_aidr.py
View file

@ -93,16 +93,13 @@ async def test_apply_guardrail_fails_open_on_4xx() -> None:
"structured_messages": [{"role": "user", "content": "core dump: raw bytes"}],
}
request_data = {"messages": inputs["structured_messages"]}
endpoint = f"{guardrail.api_base}/v1/guard_chat_completions"
with patch(
"litellm.llms.custom_httpx.http_handler.AsyncHTTPHandler.post",
return_value=httpx.Response(
status_code=400,
json={"error": "guard api error"},
request=httpx.Request(method="POST", url=endpoint),
),
):
transport = httpx.MockTransport(
lambda request: httpx.Response(status_code=400, json={"error": "guard api error"}, request=request)
)
async with httpx.AsyncClient(transport=transport) as client:
await guardrail.async_handler.close()
guardrail.async_handler.client = client
result = await guardrail.apply_guardrail(
inputs=inputs,
request_data=request_data,