scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-31 06:56:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 416

refactor(guardrails): type the shared constructor extension fields

Browse source
This commit is contained in:
albertbausili 2026-10-16 10:03:34 +02:00
parent b2b61d2199
commit 37d6a1cacd
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
litellm/integrations/custom_guardrail.py
View file

@ -256,8 +256,8 @@ class CustomGuardrail(CustomLogger):
scan_raw_request: bool = False,
only_scan_new_messages: bool = False,
timeout: float | None = None,
**kwargs,
):
**kwargs: object,
) -> None:
"""
Initialize the CustomGuardrail class
@ -303,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] = cast(object, kwargs.pop("stream_scope", None)) # cast-ok: config
stream_scope_arg: Final[object] = kwargs.pop("stream_scope", None)
self.apply_stream_scope(stream_scope_arg)
self.logging_only_scope = None
self.logging_only_continue_on_input_failure = False