From e65b708fa2d2ff425143da10970be96c8bfc277a Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Thu, 27 Mar 2025 19:15:30 -0700 Subject: [PATCH] get_all_update_transactions_from_redis --- litellm/proxy/db/redis_update_buffer.py | 29 +++++++++++++++++++++++-- 1 file changed, 27 insertions(+), 2 deletions(-) diff --git a/litellm/proxy/db/redis_update_buffer.py b/litellm/proxy/db/redis_update_buffer.py index 843ec445a3..d3591032ff 100644 --- a/litellm/proxy/db/redis_update_buffer.py +++ b/litellm/proxy/db/redis_update_buffer.py @@ -106,5 +106,30 @@ class RedisUpdateBuffer: value=transaction_amount, ) - async def get_all_update_transactions_from_redis(self): - pass + async def get_all_update_transactions_from_redis( + self, + ) -> Optional[DBSpendUpdateTransactions]: + """ + Gets all the update transactions from Redis + """ + if self.redis_cache is None: + return None + expected_keys = [ + "user_list_transactons", + "end_user_list_transactons", + "key_list_transactons", + "team_list_transactons", + "team_member_list_transactons", + "org_list_transactons", + ] + result = await self.redis_cache.async_batch_get_cache(expected_keys) + if result is None: + return None + return DBSpendUpdateTransactions( + user_list_transactons=result.get("user_list_transactons", {}), + end_user_list_transactons=result.get("end_user_list_transactons", {}), + key_list_transactons=result.get("key_list_transactons", {}), + team_list_transactons=result.get("team_list_transactons", {}), + team_member_list_transactons=result.get("team_member_list_transactons", {}), + org_list_transactons=result.get("org_list_transactons", {}), + )