mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-06 02:48:13 +00:00
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:
parent
27dcfc8c83
commit
fdc44608c3
1 changed files with 1 additions and 2 deletions
|
|
@ -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:
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue