diff --git a/api/core/model_runtime/model_providers/baichuan/llm/llm.py b/api/core/model_runtime/model_providers/baichuan/llm/llm.py
index 768f527c6e..8b646cc765 100644
--- a/api/core/model_runtime/model_providers/baichuan/llm/llm.py
+++ b/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],