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

test: cover idempotent safety identifier enforcement

Browse source
This commit is contained in:
Dominic White 2026-09-07 12:09:43 +02:00
parent 0d5a3bf2c3
commit fcdff6ed11
1 changed files with 11 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
tests/proxy_unit_tests/test_safety_identifier.py
View file

@ -5,6 +5,7 @@ from unittest.mock import AsyncMock, MagicMock, patch
import pytest import pytest
import litellm import litellm
from litellm.litellm_core_utils.safety_identifier import enforce_safety_identifier
from litellm.llms.perplexity.responses.transformation import PerplexityResponsesConfig from litellm.llms.perplexity.responses.transformation import PerplexityResponsesConfig
from litellm.proxy._types import UserAPIKeyAuth from litellm.proxy._types import UserAPIKeyAuth
from litellm.proxy.common_request_processing import ProxyBaseLLMRequestProcessing from litellm.proxy.common_request_processing import ProxyBaseLLMRequestProcessing
@ -24,6 +25,16 @@ def test_enforce_safety_identifier_hashes_authenticated_user(monkeypatch: pytest
assert data["safety_identifier"] == hashlib.sha256(b"user-123").hexdigest() assert data["safety_identifier"] == hashlib.sha256(b"user-123").hexdigest()
def test_enforce_safety_identifier_is_idempotent():
safety_identifier = hashlib.sha256(b"user-123").hexdigest()
data = {"safety_identifier": safety_identifier}
modified = enforce_safety_identifier(data=data, user_id="user-123", enabled=True)
assert modified is False
assert data == {"safety_identifier": safety_identifier}
@pytest.mark.parametrize("setting", [None, "false"]) @pytest.mark.parametrize("setting", [None, "false"])
def test_enforce_safety_identifier_is_opt_in(monkeypatch: pytest.MonkeyPatch, setting: str | None): def test_enforce_safety_identifier_is_opt_in(monkeypatch: pytest.MonkeyPatch, setting: str | None):
if setting is None: if setting is None: