Unverified Commit c9e4147b authored by Yeuoly's avatar Yeuoly Committed by GitHub

optimize baichuan invalid key error (#1996)

parent 3f25e7ec
......@@ -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],
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment