|
@@ -79,8 +79,10 @@ class BaichuanLarguageModel(LargeLanguageModel):
|
|
|
try:
|
|
|
instance.generate(model=model, stream=False, messages=[
|
|
|
BaichuanMessage(content='ping', role='user')
|
|
|
- ], parameters={}, timeout=10)
|
|
|
- except InvalidAPIKeyError as e:
|
|
|
+ ], parameters={
|
|
|
+ 'max_tokens': 1,
|
|
|
+ }, timeout=60)
|
|
|
+ except (InvalidAPIKeyError, InvalidAuthenticationError) as e:
|
|
|
raise CredentialsValidateFailedError(f"Invalid API key: {e}")
|
|
|
|
|
|
def _generate(self, model: str, credentials: dict, prompt_messages: list[PromptMessage],
|