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>
This commit is contained in:
yucheng 2026-09-30 23:10:40 +00:00
parent 27dcfc8c83
commit fdc44608c3

View file

@ -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: