scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-08 03:08:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 114

fix comments

Browse source
This commit is contained in:
aniket-kardile 2026-06-23 19:10:52 +05:30
parent ffb91bb129
commit 38fefa49be
2 changed files with 27 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -36,12 +36,14 @@ if TYPE_CHECKING:
from litellm.types.proxy.guardrails.guardrail_hooks.base import (
GuardrailConfigModel,
)
import httpx
_DEFAULT_API_BASE = "http://localhost:8000"
_GUARD_ENDPOINT = "/api/v1/ai-platform/controller/singulr-guardrails-litellm"
class SingulrMissingCredentials(Exception):
"""Custom exception for missing Singulr secrets."""
pass
@ -139,6 +141,7 @@ class SingulrGuardrail(CustomGuardrail):
endpoint,
)
try:
response = await self.async_handler.post(
url=endpoint,
@ -148,13 +151,34 @@ class SingulrGuardrail(CustomGuardrail):
)
response.raise_for_status()
result = response.json()
except Exception as exc:
verbose_proxy_logger.error("Singulr API error: %s", str(exc))
except httpx.HTTPStatusError as exc:
verbose_proxy_logger.error(
"Singulr API returned HTTP %s: %s",
exc.response.status_code,
str(exc),
)
if self.block_on_error:
raise GuardrailRaisedException(
guardrail_name=self.guardrail_name,
message=(
f"Singulr API returned HTTP "
f"{exc.response.status_code}: {exc.response.text}"
),
) from exc
return inputs
except (httpx.ConnectError, httpx.TimeoutException, httpx.NetworkError) as exc:
verbose_proxy_logger.error("Singulr API unreachable: %s", str(exc))
if self.block_on_error:
raise GuardrailRaisedException(
guardrail_name=self.guardrail_name,
message=f"Singulr API unreachable (block_on_error=True): {exc}",
) from exc
return inputs
should_block = result.get("should_block", False)

2
tests/test_litellm/proxy/guardrails/guardrail_hooks/test_singulr.py
View file

@ -155,7 +155,7 @@ class TestSingulrRequestPayload:
)
call_kwargs = mock_post.call_args
url = call_kwargs.kwargs["url"]
assert url == "https://api.test.singulr.ai/api/v1/ai-platform/controller/singulr-guardrails-sdk"
assert url == "https://api.test.singulr.ai/api/v1/ai-platform/controller/singulr-guardrails-litellm"
# ---------------------------------------------------------------------------
# Config model