fix: credential verification of baichuan did not throw all errors (#2475)

pull/2481/head
takatost 2 years ago committed by GitHub
parent ca4aa340f6
commit 4cf475680d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -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],

Loading…
Cancel
Save