scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-06 02:48:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

test(guardrails): assert no duplicate rows instead of exact row count after worker kill

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
yucheng 2026-09-30 23:48:31 +00:00
parent 71e087fa78
commit f3e9d34aa2
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tests/integration/observability/test_azure_content_safety_audit.py
View file

@ -962,10 +962,12 @@ def test_killing_a_worker_mid_burst_leaves_no_duplicate_rows(
assert set(scans) <= set(identities), scans
assert {identity for identity, status in outcomes if status == 200} <= set(scans), (outcomes, scans)
rows: Final = eventually(
lambda: read_rows('SELECT metadata FROM "LiteLLM_SpendLogs" WHERE model_group=%s', (model,)),
lambda values: len(values) == len(survivors),
lambda: read_rows('SELECT request_id, metadata FROM "LiteLLM_SpendLogs" WHERE model_group=%s', (model,)),
lambda values: len(values) >= 1,
seconds=70,
)
assert len(rows) <= len(survivors), (outcomes, rows)
assert len({row["request_id"] for row in rows}) == len(rows), rows
for row in rows:
entries: Final = object_value(row["metadata"])["guardrail_information"]
assert isinstance(entries, list) and len(entries) == 1, row