add deepseek-coder-v2 in siliconflow (#6149)

2024-07-29 18:45:19 +08:00 · 2024-07-29 18:45:19 +08:00 · 3e18d32ce5
commit 3e18d32ce5
parent 94d68b6a08
13 changed files with 165 additions and 24 deletions
--- a/api/tests/integration_tests/model_runtime/siliconflow/test_provider.py
+++ b/api/tests/integration_tests/model_runtime/siliconflow/test_provider.py
@ -0,0 +1,21 @@
+import os
+
+import pytest
+
+from core.model_runtime.errors.validate import CredentialsValidateFailedError
+from core.model_runtime.model_providers.siliconflow.siliconflow import SiliconflowProvider
+
+
+def test_validate_provider_credentials():
+    provider = SiliconflowProvider()
+
+    with pytest.raises(CredentialsValidateFailedError):
+        provider.validate_provider_credentials(
+            credentials={}
+        )
+
+    provider.validate_provider_credentials(
+        credentials={
+            'api_key': os.environ.get('API_KEY')
+        }
+    )