From bea74c144eb400dc399e286cb77903f20706b367 Mon Sep 17 00:00:00 2001
From: lei_lei <96427312+leilei3167@users.noreply.github.com>
Date: Sun, 30 Aug 2026 14:15:52 +0800
Subject: [PATCH] fix: use Python 3.10-compatible spend log timeout

---
 litellm/proxy/utils.py | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/litellm/proxy/utils.py b/litellm/proxy/utils.py
index 24895d8643d..cb9072761cc 100644
--- a/litellm/proxy/utils.py
+++ b/litellm/proxy/utils.py
@@ -6305,14 +6305,16 @@ async def update_spend_logs_job(
     logs_to_process: Final = await dequeue_spend_logs(prisma_client, MAX_LOGS_PER_INTERVAL)
 
     try:
-        async with asyncio.timeout(timeout):
-            await ProxyUpdateSpend.update_spend_logs(
+        await asyncio.wait_for(
+            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,
-            )
+            ),
+            timeout=timeout,
+        )
     except asyncio.CancelledError:
         await enqueue_spend_logs(prisma_client, logs_to_process, at_head=True)
         verbose_proxy_logger.warning(
@@ -6320,7 +6322,7 @@ async def update_spend_logs_job(
             len(logs_to_process),
         )
         raise
-    except TimeoutError:
+    except asyncio.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",