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

refactor(guardrails): accept AIDR mappings

Browse source
This commit is contained in:
abrekhov 2026-08-24 10:33:51 +00:00
parent ebccc004b7
commit e22eeab464
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
litellm/proxy/guardrails/guardrail_hooks/crowdstrike_aidr/crowdstrike_aidr.py
View file

@ -274,7 +274,7 @@ class CrowdStrikeAIDRHandler(CustomGuardrail):
)
async def _call_crowdstrike_aidr_guard(
self, payload: dict[str, Any], hook_name: str
self, payload: Mapping[str, Any], hook_name: str
) -> _GuardChatCompletionsResult:
"""
Makes the API call to the CrowdStrike AIDR AI Guard endpoint.
@ -364,7 +364,7 @@ class CrowdStrikeAIDRHandler(CustomGuardrail):
tail: Final = guard_output.messages[-num_assistant_messages:] if num_assistant_messages > 0 else []
return [_extract_text_from_message(msg) for msg in tail]
async def _call_or_fail_open(self, payload: dict[str, Any], hook_name: str) -> _GuardChatCompletionsResult:
async def _call_or_fail_open(self, payload: Mapping[str, Any], hook_name: str) -> _GuardChatCompletionsResult:
try:
return await self._call_crowdstrike_aidr_guard(payload, hook_name)
except HTTPException: