소스 검색

fix: credentials validation of ababa (#2183)

Yeuoly 1 년 전
부모
커밋
dda32c6880
2개의 변경된 파일2개의 추가작업 그리고 2개의 파일을 삭제
  1. 1 1
      api/core/model_runtime/model_providers/minimax/llm/llm.py
  2. 1 1
      api/core/model_runtime/model_providers/minimax/minimax.py

+ 1 - 1
api/core/model_runtime/model_providers/minimax/llm/llm.py

@@ -55,7 +55,7 @@ class MinimaxLargeLanguageModel(LargeLanguageModel):
                 stream=False,
                 stream=False,
                 user=''
                 user=''
             )
             )
-        except InvalidAuthenticationError as e:
+        except (InvalidAuthenticationError, InsufficientAccountBalanceError) as e:
             raise CredentialsValidateFailedError(f"Invalid API key: {e}")
             raise CredentialsValidateFailedError(f"Invalid API key: {e}")
 
 
     def get_num_tokens(self, model: str, credentials: dict, prompt_messages: list[PromptMessage],
     def get_num_tokens(self, model: str, credentials: dict, prompt_messages: list[PromptMessage],

+ 1 - 1
api/core/model_runtime/model_providers/minimax/minimax.py

@@ -27,4 +27,4 @@ class MinimaxProvider(ModelProvider):
             raise ex
             raise ex
         except Exception as ex:
         except Exception as ex:
             logger.exception(f'{self.get_provider_schema().provider} credentials validate failed')
             logger.exception(f'{self.get_provider_schema().provider} credentials validate failed')
-            raise ex
+            raise CredentialsValidateFailedError(f'{ex}')