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 199

fix(guardrails): require privaite>=0.7.2

Browse source 
0.7.2 turns off the ONNX Runtime telemetry that older releases send from the process hosting the engine, which here is the proxy itself
This commit is contained in:
crp4222 2026-10-05 16:12:36 +02:00
parent c6505635ec
commit 5939dda0aa
2 changed files with 10 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/proxy/guardrails/guardrail_hooks/privaite/privaite.py
View file

@ -372,7 +372,7 @@ class PrivaiteGuardrail(CustomGuardrail):
except ImportError as exc: # pragma: no cover
raise RuntimeError(
"The PrivAiTe guardrail requires the 'privaite' package. "
"Install it with: pip install 'privaite>=0.4.2'"
"Install it with: pip install 'privaite>=0.7.2'"
) from exc
if self.block_entities and "block_entities" not in PIIConfig.model_fields:

9
tests/unit/proxy/guardrails/guardrail_hooks/test_privaite.py
View file

@ -293,6 +293,15 @@ def test_event_hook_always_includes_pre_and_post():
assert "pre_call" in gr3.event_hook and "post_call" in gr3.event_hook
@pytest.mark.asyncio
async def test_engine_missing_privaite_recommends_safe_minimum(monkeypatch: pytest.MonkeyPatch) -> None:
guardrail = _make_guardrail()
monkeypatch.setitem(sys.modules, "privaite.config.schema", None)
with pytest.raises(RuntimeError, match=r"pip install 'privaite>=0\.7\.2'"):
await guardrail._engine_for(["en"])
@pytest.mark.asyncio
async def test_engine_is_cached():
gr = _make_guardrail()