mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-26 01:12:21 +00:00
test(integration): reconcile concurrent cost tracking changes
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
commit
7d9a95462e
1 changed files with 2 additions and 1 deletions
|
|
@ -177,7 +177,7 @@ def poll_rollups(
|
|||
)
|
||||
if not all((key_rows, team_rows, user_rows, end_user_rows, daily_user_rows, daily_team_rows)):
|
||||
return None
|
||||
return Rollups(
|
||||
rollups: Final = Rollups(
|
||||
key_spend=float(key_rows[0]["spend"]),
|
||||
team_spend=float(team_rows[0]["spend"]),
|
||||
user_spend=float(user_rows[0]["spend"]),
|
||||
|
|
@ -185,6 +185,7 @@ def poll_rollups(
|
|||
daily_user=DailySpend.model_validate(daily_user_rows[0]),
|
||||
daily_team=DailySpend.model_validate(daily_team_rows[0]),
|
||||
)
|
||||
return rollups
|
||||
|
||||
result: Final = eventually(
|
||||
read,
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue