Преглед на файлове

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

takatost преди 1 година
родител
ревизия
2a27dad2fb
променени са 2 файла, в които са добавени 2 реда и са изтрити 2 реда
  1. 1 1
      api/core/model_providers/models/llm/azure_openai_model.py
  2. 1 1
      api/core/model_providers/models/llm/openai_model.py

+ 1 - 1
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:

+ 1 - 1
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: