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