fix unit tests

langgenius · Dec 21, 2024 · 6c63dbe · 6c63dbe
1 parent 3daffb6
commit 6c63dbe
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/api/core/model_runtime/model_providers/google/llm/llm.py b/api/core/model_runtime/model_providers/google/llm/llm.py
@@ -144,7 +144,7 @@ def validate_credentials(self, model: str, credentials: dict) -> None:
         """
 
         try:
-            ping_message = SystemPromptMessage(content="ping")
+            ping_message = UserPromptMessage(content="ping")
             self._generate(model, credentials, [ping_message], {"max_output_tokens": 5})
 
         except Exception as ex: