From cec5280e033e6ff57ed1a1da0016622ad34ae1e9 Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Fri, 28 Mar 2025 16:16:44 -0700
Subject: [PATCH] debugging for Redis TX management

---
 litellm/proxy/db/db_spend_update_writer.py | 27 ++++++++++++++--------
 1 file changed, 17 insertions(+), 10 deletions(-)

diff --git a/litellm/proxy/db/db_spend_update_writer.py b/litellm/proxy/db/db_spend_update_writer.py
index 8f175fc905..fea40e58e7 100644
--- a/litellm/proxy/db/db_spend_update_writer.py
+++ b/litellm/proxy/db/db_spend_update_writer.py
@@ -496,6 +496,9 @@ class DBSpendUpdateWriter:
 
         ### UPDATE USER TABLE ###
         user_list_transactions = db_spend_update_transactions["user_list_transactions"]
+        verbose_proxy_logger.debug(
+            "User Spend transactions: {}".format(user_list_transactions)
+        )
         if (
             user_list_transactions is not None
             and len(user_list_transactions.keys()) > 0
@@ -536,14 +539,12 @@ class DBSpendUpdateWriter:
                     )
 
         ### UPDATE END-USER TABLE ###
-        verbose_proxy_logger.debug(
-            "End-User Spend transactions: {}".format(
-                len(prisma_client.end_user_list_transactions.keys())
-            )
-        )
         end_user_list_transactions = db_spend_update_transactions[
             "end_user_list_transactions"
         ]
+        verbose_proxy_logger.debug(
+            "End-User Spend transactions: {}".format(end_user_list_transactions)
+        )
         if (
             end_user_list_transactions is not None
             and len(end_user_list_transactions.keys()) > 0
@@ -595,12 +596,10 @@ class DBSpendUpdateWriter:
                     )
 
         ### UPDATE TEAM TABLE ###
-        verbose_proxy_logger.debug(
-            "Team Spend transactions: {}".format(
-                len(prisma_client.team_list_transactions.keys())
-            )
-        )
         team_list_transactions = db_spend_update_transactions["team_list_transactions"]
+        verbose_proxy_logger.debug(
+            "Team Spend transactions: {}".format(team_list_transactions)
+        )
         if (
             team_list_transactions is not None
             and len(team_list_transactions.keys()) > 0
@@ -649,6 +648,11 @@ class DBSpendUpdateWriter:
         team_member_list_transactions = db_spend_update_transactions[
             "team_member_list_transactions"
         ]
+        verbose_proxy_logger.debug(
+            "Team Membership Spend transactions: {}".format(
+                team_member_list_transactions
+            )
+        )
         if (
             team_member_list_transactions is not None
             and len(team_member_list_transactions.keys()) > 0
@@ -694,6 +698,9 @@ class DBSpendUpdateWriter:
 
         ### UPDATE ORG TABLE ###
         org_list_transactions = db_spend_update_transactions["org_list_transactions"]
+        verbose_proxy_logger.debug(
+            "Org Spend transactions: {}".format(org_list_transactions)
+        )
         if org_list_transactions is not None and len(org_list_transactions.keys()) > 0:
             for i in range(n_retry_times + 1):
                 start_time = time.time()