diff --git a/tests/test_litellm/proxy/db/db_transaction_queue/test_daily_spend_update_queue.py b/tests/test_litellm/proxy/db/db_transaction_queue/test_daily_spend_update_queue.py index d8a937d30fa..c7e7ef5d469 100644 --- a/tests/test_litellm/proxy/db/db_transaction_queue/test_daily_spend_update_queue.py +++ b/tests/test_litellm/proxy/db/db_transaction_queue/test_daily_spend_update_queue.py @@ -207,6 +207,8 @@ async def test_get_aggregated_daily_spend_update_transactions_same_key(): "cache_creation_input_tokens": 0, "cache_read_input_tokens": 0, "compression_saved_tokens": 0, + "compression_savings_spend": 0, + "prompt_caching_savings_spend": 0, } updates = [{test_key: test_transaction1}, {test_key: test_transaction2}] @@ -255,6 +257,8 @@ async def test_flush_and_get_aggregated_daily_spend_update_transactions( "cache_creation_input_tokens": 0, "cache_read_input_tokens": 0, "compression_saved_tokens": 0, + "compression_savings_spend": 0, + "prompt_caching_savings_spend": 0, } # Add updates to queue