From 6b10d551aba60ca2471a26d68e6e9bcfe0883683 Mon Sep 17 00:00:00 2001
From: lei_lei <96427312+leilei3167@users.noreply.github.com>
Date: Sat, 29 Aug 2026 13:05:57 +0000
Subject: [PATCH] fix: time out spend drain writes without parking a coroutine

---
 litellm/proxy/utils.py | 21 +++++++++------------
 1 file changed, 9 insertions(+), 12 deletions(-)

diff --git a/litellm/proxy/utils.py b/litellm/proxy/utils.py
index 0599da86dcc..99260a930a7 100644
--- a/litellm/proxy/utils.py
+++ b/litellm/proxy/utils.py
@@ -6305,17 +6305,14 @@ async def update_spend_logs_job(
     logs_to_process: Final = await dequeue_spend_logs(prisma_client, MAX_LOGS_PER_INTERVAL)
 
     try:
-        update_spend_logs = ProxyUpdateSpend.update_spend_logs(
-            n_retry_times=n_retry_times,
-            prisma_client=prisma_client,
-            proxy_logging_obj=proxy_logging_obj,
-            db_writer_client=db_writer_client,
-            logs_to_process=logs_to_process,
-        )
-        if timeout is None:
-            await update_spend_logs
-        else:
-            await asyncio.wait_for(update_spend_logs, timeout=timeout)
+        async with asyncio.timeout(timeout):
+            await ProxyUpdateSpend.update_spend_logs(
+                n_retry_times=n_retry_times,
+                prisma_client=prisma_client,
+                proxy_logging_obj=proxy_logging_obj,
+                db_writer_client=db_writer_client,
+                logs_to_process=logs_to_process,
+            )
     except asyncio.CancelledError:
         await enqueue_spend_logs(prisma_client, logs_to_process, at_head=True)
         verbose_proxy_logger.warning(
@@ -6323,7 +6320,7 @@ async def update_spend_logs_job(
             len(logs_to_process),
         )
         raise
-    except asyncio.TimeoutError:
+    except TimeoutError:
         await enqueue_spend_logs(prisma_client, logs_to_process, at_head=True)
         verbose_proxy_logger.warning(
             "Spend tracking - spend log write timed out after %.2f seconds; requeued %d rows",