Skip to content

D
dify
Unverified Commit c9e4147b authored by Yeuoly's avatar Yeuoly Committed by GitHub
Browse files
Options

optimize baichuan invalid key error (#1996)

parent 3f25e7ec
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 2 deletions
api/core/model_runtime/model_providers/baichuan/llm/llm.py
View file @ c9e4147b
...@@ -79,8 +79,10 @@ class BaichuanLarguageModel(LargeLanguageModel): ...@@ -79,8 +79,10 @@ class BaichuanLarguageModel(LargeLanguageModel):
try: try:
instance.generate(model=model, stream=False, messages=[ instance.generate(model=model, stream=False, messages=[
BaichuanMessage(content='ping', role='user') BaichuanMessage(content='ping', role='user')
], parameters={}, timeout=10) ], parameters={
except InvalidAPIKeyError as e: 'max_tokens': 1,
}, timeout=60)
except (InvalidAPIKeyError, InvalidAuthenticationError) as e:
raise CredentialsValidateFailedError(f"Invalid API key: {e}") raise CredentialsValidateFailedError(f"Invalid API key: {e}")
def _generate(self, model: str, credentials: dict, prompt_messages: list[PromptMessage], 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