scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-18 04:44:37 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 123

style: apply ruff line-wrap formatting to singulr guardrail

Browse source
This commit is contained in:
aniket-kardile 2026-07-01 22:38:20 +05:30
parent 556a2d8dba
commit 25e9cee9e9
2 changed files with 9 additions and 19 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
litellm/proxy/guardrails/guardrail_hooks/singulr/__init__.py
View file

@ -21,10 +21,8 @@ def initialize_guardrail(
import litellm
_cb = SingulrGuardrail(
singulr_api_base=getattr(litellm_params, "singulr_api_base", None)
or litellm_params.api_base,
singulr_api_key=getattr(litellm_params, "singulr_api_key", None)
or litellm_params.api_key,
singulr_api_base=getattr(litellm_params, "singulr_api_base", None) or litellm_params.api_base,
singulr_api_key=getattr(litellm_params, "singulr_api_key", None) or litellm_params.api_key,
singulr_application_id=getattr(litellm_params, "singulr_application_id", None),
singulr_guardrail_id=getattr(litellm_params, "singulr_guardrail_id", None),
block_on_error=getattr(litellm_params, "block_on_error", None),

22
litellm/proxy/guardrails/guardrail_hooks/singulr/singulr.py
View file

@ -48,9 +48,9 @@ class SingulrGuardrail(CustomGuardrail):
) -> None:
self.singulr_api_key = singulr_api_key or os.environ.get("SINGULR_API_KEY")
self.singulr_api_base = (
singulr_api_base or os.environ.get("SINGULR_API_BASE") or _DEFAULT_API_BASE
).rstrip("/")
self.singulr_api_base = (singulr_api_base or os.environ.get("SINGULR_API_BASE") or _DEFAULT_API_BASE).rstrip(
"/"
)
parsed = urlparse(self.singulr_api_base)
if parsed.scheme == "http" and parsed.hostname not in (
@ -64,12 +64,8 @@ class SingulrGuardrail(CustomGuardrail):
self.singulr_api_base,
)
self.singulr_application_id = singulr_application_id or os.environ.get(
"SINGULR_ENFORCEMENT_ENTITY_ID"
)
self.singulr_guardrail_id = singulr_guardrail_id or os.environ.get(
"SINGULR_GUARDRAIL_ID"
)
self.singulr_application_id = singulr_application_id or os.environ.get("SINGULR_ENFORCEMENT_ENTITY_ID")
self.singulr_guardrail_id = singulr_guardrail_id or os.environ.get("SINGULR_GUARDRAIL_ID")
if block_on_error is None:
env = os.environ.get("SINGULR_BLOCK_ON_ERROR", "true")
@ -148,9 +144,7 @@ class SingulrGuardrail(CustomGuardrail):
if self.block_on_error:
raise GuardrailRaisedException(
guardrail_name=self.guardrail_name,
message=(
f"Singulr API returned HTTP {exc.response.status_code}: {exc.response.text}"
),
message=(f"Singulr API returned HTTP {exc.response.status_code}: {exc.response.text}"),
) from exc
return None
@ -164,9 +158,7 @@ class SingulrGuardrail(CustomGuardrail):
return None
except ValueError as exc:
verbose_proxy_logger.error(
"Singulr API returned non-JSON response: %s", str(exc)
)
verbose_proxy_logger.error("Singulr API returned non-JSON response: %s", str(exc))
if self.block_on_error:
raise GuardrailRaisedException(
guardrail_name=self.guardrail_name,