Add novita.ai as model provider (#4961)

2024-06-14 18:23:06 +08:00
parent c28d709d7f
commit b7ff765d8d
17 changed files with 575 additions and 1 deletions
--- a/api/tests/integration_tests/model_runtime/novita/test_provider.py
+++ b/api/tests/integration_tests/model_runtime/novita/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.novita.novita import NovitaProvider
+
+
+def test_validate_provider_credentials():
+    provider = NovitaProvider()
+
+    with pytest.raises(CredentialsValidateFailedError):
+        provider.validate_provider_credentials(
+            credentials={}
+        )
+
+    provider.validate_provider_credentials(
+        credentials={
+            'api_key': os.environ.get('NOVITA_API_KEY'),
+        }
+    )