fix: run model serval ex not return (#915)

This commit is contained in:
takatost 2023-08-19 14:16:41 +08:00 committed by GitHub
parent 930f74c610
commit 2a27dad2fb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions

View File

@ -166,7 +166,7 @@ class AzureOpenAIModel(BaseLLM):
elif isinstance(ex, openai.error.RateLimitError): elif isinstance(ex, openai.error.RateLimitError):
return LLMRateLimitError('Azure ' + str(ex)) return LLMRateLimitError('Azure ' + str(ex))
elif isinstance(ex, openai.error.AuthenticationError): elif isinstance(ex, openai.error.AuthenticationError):
raise LLMAuthorizationError('Azure ' + str(ex)) return LLMAuthorizationError('Azure ' + str(ex))
elif isinstance(ex, openai.error.OpenAIError): elif isinstance(ex, openai.error.OpenAIError):
return LLMBadRequestError('Azure ' + ex.__class__.__name__ + ":" + str(ex)) return LLMBadRequestError('Azure ' + ex.__class__.__name__ + ":" + str(ex))
else: else:

View File

@ -185,7 +185,7 @@ class OpenAIModel(BaseLLM):
elif isinstance(ex, openai.error.RateLimitError): elif isinstance(ex, openai.error.RateLimitError):
return LLMRateLimitError(str(ex)) return LLMRateLimitError(str(ex))
elif isinstance(ex, openai.error.AuthenticationError): elif isinstance(ex, openai.error.AuthenticationError):
raise LLMAuthorizationError(str(ex)) return LLMAuthorizationError(str(ex))
elif isinstance(ex, openai.error.OpenAIError): elif isinstance(ex, openai.error.OpenAIError):
return LLMBadRequestError(ex.__class__.__name__ + ":" + str(ex)) return LLMBadRequestError(ex.__class__.__name__ + ":" + str(ex))
else: else: