AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-14 00:55:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

optimize baichuan invalid key error (#1996)

Browse Source
This commit is contained in:
Yeuoly 2024-01-11 13:35:24 +08:00 committed by GitHub
parent 3f25e7ec79
commit c9e4147b11
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
api/core/model_runtime/model_providers/baichuan/llm/llm.py
View File

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