From 68362bec13506ea3edbac43e57ed0d8398846e35 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Rapha=C3=ABl=20MANSUY?= <raphael.mansuy@gmail.com>
Date: Thu, 4 Dec 2025 19:19:22 +0800
Subject: [PATCH] cherry-pick 8994c70f

---
 lightrag/llm/openai.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/lightrag/llm/openai.py b/lightrag/llm/openai.py
index 948ae270..68def79c 100644
--- a/lightrag/llm/openai.py
+++ b/lightrag/llm/openai.py
@@ -248,6 +248,10 @@ async def openai_complete_if_cache(
             response = await openai_async_client.chat.completions.create(
                 model=model, messages=messages, **kwargs
             )
+    except APITimeoutError as e:
+        logger.error(f"OpenAI API Timeout Error: {e}")
+        await openai_async_client.close()  # Ensure client is closed
+        raise
     except APIConnectionError as e:
         logger.error(f"OpenAI API Connection Error: {e}")
         await openai_async_client.close()  # Ensure client is closed
@@ -256,10 +260,6 @@ async def openai_complete_if_cache(
         logger.error(f"OpenAI API Rate Limit Error: {e}")
         await openai_async_client.close()  # Ensure client is closed
         raise
-    except APITimeoutError as e:
-        logger.error(f"OpenAI API Timeout Error: {e}")
-        await openai_async_client.close()  # Ensure client is closed
-        raise
     except Exception as e:
         logger.error(
             f"OpenAI API Call Failed,\nModel: {model},\nParams: {kwargs}, Got: {e}"