From 56dafd230fbf626eea48bcbd7688b7cd05adf3a7 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Sat, 18 Jul 2026 22:26:53 +0000 Subject: [PATCH] test(spend): update daily spend aggregation fixtures for savings columns Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --- .../db/db_transaction_queue/test_daily_spend_update_queue.py | 4 ++++ 1 file changed, 4 insertions(+) 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