Add IBM Models Provider

api/core/model_runtime/model_providers/ibm/ibm.py

@@ -0,0 +1,28 @@
+import logging
+
+from ibm_watsonx_ai import APIClient, Credentials
+
+from core.model_runtime.errors.validate import CredentialsValidateFailedError
+from core.model_runtime.model_providers.__base.model_provider import ModelProvider
+
+logger = logging.getLogger(__name__)
+
+
+class IbmProvider(ModelProvider):
+    def validate_provider_credentials(self, credentials: dict) -> None:
+        """
+        Validate provider credentials
+
+        if validate failed, raise exception
+
+        :param credentials: provider credentials, credentials form defined in `provider_credential_schema`.
+        """
+        try:
+            credentials = Credentials(url=credentials.get("base_url"), api_key=credentials.get("api_key"))
+
+            client = APIClient(credentials)
+        except CredentialsValidateFailedError as ex:
+            raise ex
+        except Exception as ex:
+            logger.exception(f"{self.get_provider_schema().provider} credentials validate failed")
+            raise ex