Ver código fonte

feat: optimize openai error raise (#459)

tags/0.3.6
John Wang 2 anos atrás
pai
commit
df5763be37
Nenhuma conta vinculada ao e-mail do autor do commit
1 arquivos alterados com 8 adições e 4 exclusões
  1. 8
    4
      api/core/llm/error_handle_wraps.py

+ 8
- 4
api/core/llm/error_handle_wraps.py Ver arquivo

raise LLMBadRequestError(str(e)) raise LLMBadRequestError(str(e))
except openai.error.APIConnectionError as e: except openai.error.APIConnectionError as e:
logging.exception("Failed to connect to OpenAI API.") 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: except (openai.error.APIError, openai.error.ServiceUnavailableError, openai.error.Timeout) as e:
logging.exception("OpenAI service unavailable.") logging.exception("OpenAI service unavailable.")
raise LLMAPIUnavailableError(str(e))
raise LLMAPIUnavailableError(e.__class__.__name__ + ":" + str(e))
except openai.error.RateLimitError as e: except openai.error.RateLimitError as e:
raise LLMRateLimitError(str(e)) raise LLMRateLimitError(str(e))
except openai.error.AuthenticationError as e: except openai.error.AuthenticationError as e:
raise LLMAuthorizationError(str(e)) raise LLMAuthorizationError(str(e))
except openai.error.OpenAIError as e:
raise LLMBadRequestError(e.__class__.__name__ + ":" + str(e))


return wrapper return wrapper


raise LLMBadRequestError(str(e)) raise LLMBadRequestError(str(e))
except openai.error.APIConnectionError as e: except openai.error.APIConnectionError as e:
logging.exception("Failed to connect to OpenAI API.") 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: except (openai.error.APIError, openai.error.ServiceUnavailableError, openai.error.Timeout) as e:
logging.exception("OpenAI service unavailable.") logging.exception("OpenAI service unavailable.")
raise LLMAPIUnavailableError(str(e))
raise LLMAPIUnavailableError(e.__class__.__name__ + ":" + str(e))
except openai.error.RateLimitError as e: except openai.error.RateLimitError as e:
raise LLMRateLimitError(str(e)) raise LLMRateLimitError(str(e))
except openai.error.AuthenticationError as e: except openai.error.AuthenticationError as e:
raise LLMAuthorizationError(str(e)) raise LLMAuthorizationError(str(e))
except openai.error.OpenAIError as e:
raise LLMBadRequestError(e.__class__.__name__ + ":" + str(e))


return wrapper return wrapper

Carregando…
Cancelar
Salvar