mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-06 08:16:43 +00:00
Fix gaurdrail code qa
This commit is contained in:
parent
eb7a6040c1
commit
84b908413f
1 changed files with 2 additions and 0 deletions
|
|
@ -25,6 +25,7 @@ from fastapi import HTTPException
|
|||
from litellm.integrations.custom_guardrail import (
|
||||
CustomGuardrail,
|
||||
ModifyResponseException,
|
||||
log_guardrail_information,
|
||||
)
|
||||
from litellm.types.guardrails import GuardrailEventHooks
|
||||
from litellm.types.proxy.guardrails.guardrail_hooks.base import GuardrailConfigModel
|
||||
|
|
@ -536,6 +537,7 @@ class BlockCodeExecutionGuardrail(CustomGuardrail):
|
|||
detection_info={"language": language},
|
||||
)
|
||||
|
||||
@log_guardrail_information
|
||||
async def apply_guardrail(
|
||||
self,
|
||||
inputs: GenericGuardrailAPIInputs,
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue