scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-11 22:51:28 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 123

fix ModelArmorGuardrail

Browse source
This commit is contained in:
Ishaan Jaffer 2025-10-27 17:47:19 -07:00
parent d5f48c7e23
commit 0e23f89eb7
1 changed files with 8 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
litellm/proxy/guardrails/guardrail_hooks/model_armor/model_armor.py
View file

@ -56,6 +56,14 @@ class ModelArmorGuardrail(CustomGuardrail, VertexBase):
api_endpoint: Optional[str] = None,
**kwargs,
):
# Set supported event hooks if not already provided
if "event_hook" not in kwargs:
kwargs["event_hook"] = [
GuardrailEventHooks.pre_call,
GuardrailEventHooks.during_call,
GuardrailEventHooks.post_call,
]
# Initialize parent classes first
super().__init__(**kwargs)
VertexBase.__init__(self)