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

2a27dad2 · takatost · GitHub · 930f74c6 · 2a27dad2 · 2a27dad2
Unverified Commit 2a27dad2 authored Aug 19, 2023 by takatost Committed by GitHub Aug 19, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

azure_openai_model.py api/core/model_providers/models/llm/azure_openai_model.py +1 -1

openai_model.py api/core/model_providers/models/llm/openai_model.py +1 -1

No files found.
--- a/api/core/model_providers/models/llm/azure_openai_model.py
+++ b/api/core/model_providers/models/llm/azure_openai_model.py
@@ -166,7 +166,7 @@ class AzureOpenAIModel(BaseLLM):
        elif isinstance(ex, openai.error.RateLimitError):
            return LLMRateLimitError('Azure ' + str(ex))
        elif isinstance(ex, openai.error.AuthenticationError):
-            raise LLMAuthorizationError('Azure ' + str(ex))
+            return LLMAuthorizationError('Azure ' + str(ex))
        elif isinstance(ex, openai.error.OpenAIError):
            return LLMBadRequestError('Azure ' + ex.__class__.__name__ + ":" + str(ex))
        else:

--- a/api/core/model_providers/models/llm/openai_model.py
+++ b/api/core/model_providers/models/llm/openai_model.py
@@ -185,7 +185,7 @@ class OpenAIModel(BaseLLM):
        elif isinstance(ex, openai.error.RateLimitError):
            return LLMRateLimitError(str(ex))
        elif isinstance(ex, openai.error.AuthenticationError):
-            raise LLMAuthorizationError(str(ex))
+            return LLMAuthorizationError(str(ex))
        elif isinstance(ex, openai.error.OpenAIError):
            return LLMBadRequestError(ex.__class__.__name__ + ":" + str(ex))
        else: