diff --git a/tests/test_litellm/proxy/db/test_model_access_group_spend.py b/tests/test_litellm/proxy/db/test_model_access_group_spend.py index 79dffc29e7e..d2d079bb0e4 100644 --- a/tests/test_litellm/proxy/db/test_model_access_group_spend.py +++ b/tests/test_litellm/proxy/db/test_model_access_group_spend.py @@ -414,12 +414,14 @@ async def test_redis_buffer_requeues_access_group_transactions_as_queue_items(): daily_org_spend_update_transactions=None, daily_end_user_spend_update_transactions=None, daily_agent_spend_update_transactions=None, + window_spend_update_transactions=None, spend_update_queue=queue, daily_spend_update_queue=daily_queue, daily_team_spend_update_queue=daily_queue, daily_org_spend_update_queue=daily_queue, daily_end_user_spend_update_queue=daily_queue, daily_agent_spend_update_queue=daily_queue, + window_spend_update_queue=None, ) updates = await _drain(queue)