mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-07 08:26:10 +00:00
fix: correct incorrect type annotation
This commit is contained in:
parent
4eecf59f86
commit
e5b06effda
1 changed files with 1 additions and 1 deletions
|
|
@ -638,7 +638,7 @@ class NomaGuardrail(CustomGuardrail):
|
|||
call_type: CallTypesLiteral,
|
||||
) -> Union[Exception, str, dict, None]:
|
||||
event_type: GuardrailEventHooks = GuardrailEventHooks.during_call
|
||||
if call_type == CallTypes.during_mcp_call.value:
|
||||
if call_type == CallTypes.call_mcp_tool.value:
|
||||
event_type = GuardrailEventHooks.pre_mcp_call
|
||||
|
||||
if self.should_run_guardrail(data=data, event_type=event_type) is not True:
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue