| @@ -17,14 +17,16 @@ def handle_llm_exceptions(func): | |||
| raise LLMBadRequestError(str(e)) | |||
| except openai.error.APIConnectionError as e: | |||
| logging.exception("Failed to connect to OpenAI API.") | |||
| raise LLMAPIConnectionError(str(e)) | |||
| raise LLMAPIConnectionError(e.__class__.__name__ + ":" + str(e)) | |||
| except (openai.error.APIError, openai.error.ServiceUnavailableError, openai.error.Timeout) as e: | |||
| logging.exception("OpenAI service unavailable.") | |||
| raise LLMAPIUnavailableError(str(e)) | |||
| raise LLMAPIUnavailableError(e.__class__.__name__ + ":" + str(e)) | |||
| except openai.error.RateLimitError as e: | |||
| raise LLMRateLimitError(str(e)) | |||
| except openai.error.AuthenticationError as e: | |||
| raise LLMAuthorizationError(str(e)) | |||
| except openai.error.OpenAIError as e: | |||
| raise LLMBadRequestError(e.__class__.__name__ + ":" + str(e)) | |||
| return wrapper | |||
| @@ -39,13 +41,15 @@ def handle_llm_exceptions_async(func): | |||
| raise LLMBadRequestError(str(e)) | |||
| except openai.error.APIConnectionError as e: | |||
| logging.exception("Failed to connect to OpenAI API.") | |||
| raise LLMAPIConnectionError(str(e)) | |||
| raise LLMAPIConnectionError(e.__class__.__name__ + ":" + str(e)) | |||
| except (openai.error.APIError, openai.error.ServiceUnavailableError, openai.error.Timeout) as e: | |||
| logging.exception("OpenAI service unavailable.") | |||
| raise LLMAPIUnavailableError(str(e)) | |||
| raise LLMAPIUnavailableError(e.__class__.__name__ + ":" + str(e)) | |||
| except openai.error.RateLimitError as e: | |||
| raise LLMRateLimitError(str(e)) | |||
| except openai.error.AuthenticationError as e: | |||
| raise LLMAuthorizationError(str(e)) | |||
| except openai.error.OpenAIError as e: | |||
| raise LLMBadRequestError(e.__class__.__name__ + ":" + str(e)) | |||
| return wrapper | |||