From fdc44608c353776a38625fe7027683bf84e65ce6 Mon Sep 17 00:00:00 2001 From: yucheng Date: Wed, 30 Sep 2026 23:10:40 +0000 Subject: [PATCH] test(guardrails): inline spend row count in the concurrency audit cell Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --- .../observability/test_azure_content_safety_audit.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tests/integration/observability/test_azure_content_safety_audit.py b/tests/integration/observability/test_azure_content_safety_audit.py index 376412b04d2..3c274f893b1 100644 --- a/tests/integration/observability/test_azure_content_safety_audit.py +++ b/tests/integration/observability/test_azure_content_safety_audit.py @@ -849,12 +849,11 @@ def test_concurrent_mixed_requests_scan_each_prompt_once( assert sorted(scans) == sorted(expected), scans assert len(_provider_calls(provider)) == 30 for model_group in (chat_model, messages_model, responses_model): - wanted: Final = 10 rows: Final = eventually( lambda group=model_group: read_rows( 'SELECT metadata FROM "LiteLLM_SpendLogs" WHERE model_group=%s', (group,) ), - lambda values: len(values) == wanted, + lambda values: len(values) == 10, seconds=70, ) for row in rows: