| @@ -103,7 +103,7 @@ class BaichuanLarguageModel(LargeLanguageModel): | |||
| ], parameters={ | |||
| 'max_tokens': 1, | |||
| }, timeout=60) | |||
| except (InvalidAPIKeyError, InvalidAuthenticationError) as e: | |||
| except Exception as e: | |||
| raise CredentialsValidateFailedError(f"Invalid API key: {e}") | |||
| def _generate(self, model: str, credentials: dict, prompt_messages: list[PromptMessage], | |||