scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-29 06:33:28 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 201

Merge branch 'litellm_guardrail_stream_inspection' into albertbausili/fix-pr-failed-workflows-fork

Browse source
This commit is contained in:
albertbausili 2026-10-16 10:12:15 +02:00
commit cb62ed353f
1 changed files with 4 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
litellm/integrations/custom_guardrail.py
View file

@ -256,10 +256,8 @@ class CustomGuardrail(CustomLogger):
scan_raw_request: bool = False,
only_scan_new_messages: bool = False,
timeout: float | None = None,
turn_off_message_logging: bool = False,
message_logging: bool = True,
**kwargs: object,
) -> None:
**kwargs,
):
"""
Initialize the CustomGuardrail class
@ -305,7 +303,7 @@ class CustomGuardrail(CustomLogger):
self.run_in_parallel: bool = run_in_parallel
self.scan_raw_request: bool = scan_raw_request
self.only_scan_new_messages: bool = only_scan_new_messages
stream_scope_arg: Final[object] = kwargs.pop("stream_scope", None)
stream_scope_arg: Final[object] = cast(object, kwargs.pop("stream_scope", None)) # cast-ok: config
self.apply_stream_scope(stream_scope_arg)
self.logging_only_scope = None
self.logging_only_continue_on_input_failure = False
@ -325,11 +323,7 @@ class CustomGuardrail(CustomLogger):
"(default) to enforce validation and fail at startup.",
validation_error,
)
super().__init__(
turn_off_message_logging=turn_off_message_logging,
message_logging=message_logging,
**kwargs,
)
super().__init__(**kwargs)
def render_violation_message(self, default: str, context: Mapping[str, object] | None = None) -> str:
"""Return a custom violation message if template is configured."""