phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-24 18:24:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test_daily_spend_update_queue.py

Browse source
This commit is contained in:
Ishaan Jaff 2025-04-15 15:27:03 -07:00
parent e7ac71fc57
commit 544d2c8b9f
1 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
tests/litellm/proxy/db/db_transaction_queue/test_daily_spend_update_queue.py
View file

@ -204,6 +204,8 @@ async def test_get_aggregated_daily_spend_update_transactions_same_key():
"api_requests": 2, # 1 + 1
"successful_requests": 2, # 1 + 1
"failed_requests": 0, # 0 + 0
"cache_creation_input_tokens": 0,
"cache_read_input_tokens": 0,
}
updates = [{test_key: test_transaction1}, {test_key: test_transaction2}]
@ -249,6 +251,8 @@ async def test_flush_and_get_aggregated_daily_spend_update_transactions(
"api_requests": 2, # 1 + 1
"successful_requests": 2, # 1 + 1
"failed_requests": 0, # 0 + 0
"cache_creation_input_tokens": 0,
"cache_read_input_tokens": 0,
}
# Add updates to queue