Feat/add 360-zhinao provider (#7069)

2024-08-08 06:23:08 +00:00
parent 8e23e24bd5
commit 34a9dbe826
16 changed files with 348 additions and 1 deletions
--- a/api/tests/integration_tests/model_runtime/zhinao/test_provider.py
+++ b/api/tests/integration_tests/model_runtime/zhinao/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.zhinao.zhinao import ZhinaoProvider
+
+
+def test_validate_provider_credentials():
+    provider = ZhinaoProvider()
+
+    with pytest.raises(CredentialsValidateFailedError):
+        provider.validate_provider_credentials(
+            credentials={}
+        )
+
+    provider.validate_provider_credentials(
+        credentials={
+            'api_key': os.environ.get('ZHINAO_API_KEY')
+        }
+    )