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}"