scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-14 09:03:16 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 214

refactor(guardrails): drop the CollieAi wrapper's function docstrings

Browse source 
AGENTS.md keeps comments to complex logic and tool input, and these only restated what each function returns
This commit is contained in:
CollieAi 2026-11-04 21:57:48 +01:00
parent 39e4bbc191
commit 462b8f795a
1 changed files with 0 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
litellm/proxy/guardrails/guardrail_hooks/collieai/collieai.py
View file

@ -21,12 +21,10 @@ MISSING_PACKAGE_MESSAGE: Final = (
def has_readiness_guard(package: object) -> bool:
"""1.0.0 is the first release with the readiness guard, `require_guardrail`, that the package's block mode needs"""
return getattr(package, "require_guardrail", None) is not None
def package_guardrail(package: object) -> type[CustomGuardrail] | None:
"""The package's guardrail class, or None for a package too old or not shaped as one"""
guardrail: Final[object] = getattr(package, "CollieAIGuardrail", None)
if not has_readiness_guard(package) or not isinstance(guardrail, type):
return None
@ -34,8 +32,6 @@ def package_guardrail(package: object) -> type[CustomGuardrail] | None:
def missing_package(import_error: ImportError | None) -> type[CustomGuardrail]:
"""A stand-in that fails where the proxy builds the guardrail, naming what to install"""
class MissingPackage(CustomGuardrail):
def __init__(self, **kwargs: object) -> None: # kwargs-ok: mirrors the package constructor, only raises
raise ImportError(MISSING_PACKAGE_MESSAGE) from import_error