From 5163dd38e5d1f02ee155d2909bd94a5e285df573 Mon Sep 17 00:00:00 2001 From: takatost Date: Sat, 19 Aug 2023 14:35:16 +0800 Subject: [PATCH] fix: run extra model serval ex not return (#916) --- .../model_providers/models/embedding/azure_openai_embedding.py | 2 +- api/core/model_providers/models/embedding/openai_embedding.py | 2 +- api/core/model_providers/models/moderation/openai_moderation.py | 2 +- api/core/model_providers/models/speech2text/openai_whisper.py | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) diff --git a/api/core/model_providers/models/embedding/azure_openai_embedding.py b/api/core/model_providers/models/embedding/azure_openai_embedding.py index 81f08784b2..127bf640ee 100644 --- a/api/core/model_providers/models/embedding/azure_openai_embedding.py +++ b/api/core/model_providers/models/embedding/azure_openai_embedding.py @@ -71,7 +71,7 @@ class AzureOpenAIEmbedding(BaseEmbedding): 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: diff --git a/api/core/model_providers/models/embedding/openai_embedding.py b/api/core/model_providers/models/embedding/openai_embedding.py index 1d7af94fdb..3ab65c291d 100644 --- a/api/core/model_providers/models/embedding/openai_embedding.py +++ b/api/core/model_providers/models/embedding/openai_embedding.py @@ -65,7 +65,7 @@ class OpenAIEmbedding(BaseEmbedding): 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: diff --git a/api/core/model_providers/models/moderation/openai_moderation.py b/api/core/model_providers/models/moderation/openai_moderation.py index c1e792966b..e7012a0438 100644 --- a/api/core/model_providers/models/moderation/openai_moderation.py +++ b/api/core/model_providers/models/moderation/openai_moderation.py @@ -41,7 +41,7 @@ class OpenAIModeration(BaseProviderModel): 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: diff --git a/api/core/model_providers/models/speech2text/openai_whisper.py b/api/core/model_providers/models/speech2text/openai_whisper.py index 8bca2aaa6d..a1fdc484dd 100644 --- a/api/core/model_providers/models/speech2text/openai_whisper.py +++ b/api/core/model_providers/models/speech2text/openai_whisper.py @@ -40,7 +40,7 @@ class OpenAIWhisper(BaseSpeech2Text): 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: