Эх сурвалжийг харах

optimize baichuan invalid key error (#1996)

Yeuoly 1 жил өмнө
parent
commit
c9e4147b11

+ 4 - 2
api/core/model_runtime/model_providers/baichuan/llm/llm.py

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