Browse Source

fix: credentials validation of ababa (#2183)

Yeuoly 1 year ago
parent
commit
dda32c6880

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

@@ -55,7 +55,7 @@ class MinimaxLargeLanguageModel(LargeLanguageModel):
                 stream=False,
                 user=''
             )
-        except InvalidAuthenticationError as e:
+        except (InvalidAuthenticationError, InsufficientAccountBalanceError) as e:
             raise CredentialsValidateFailedError(f"Invalid API key: {e}")
 
     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
         except Exception as ex:
             logger.exception(f'{self.get_provider_schema().provider} credentials validate failed')
-            raise ex
+            raise CredentialsValidateFailedError(f'{ex}')