scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-08 03:08:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 115

fix(model_armor): decorate apply_guardrail with log_guardrail_information

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
joshua-berri 2026-09-11 09:07:06 +00:00 • committed by yucheng
parent f417d7f739
commit 39b916f13f
1 changed files with 1 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -1226,6 +1226,7 @@ class ModelArmorGuardrail(CustomGuardrail, VertexBase):
for chunk in all_chunks: for chunk in all_chunks:
yield chunk yield chunk
@log_guardrail_information
async def apply_guardrail( async def apply_guardrail(
self, self,
inputs: GenericGuardrailAPIInputs, inputs: GenericGuardrailAPIInputs,