Underground-Digital/Workflow-Engine

1import logging2 3from core.model_runtime.entities.model_entities import ModelType4from core.model_runtime.errors.validate import CredentialsValidateFailedError5from core.model_runtime.model_providers.__base.model_provider import ModelProvider6 7logger = logging.getLogger(__name__)8 9 10class MixedBreadProvider(ModelProvider):11    def validate_provider_credentials(self, credentials: dict) -> None:12        """13        Validate provider credentials14        if validate failed, raise exception15 16        :param credentials: provider credentials, credentials form defined in `provider_credential_schema`.17        """18        try:19            model_instance = self.get_model_instance(ModelType.TEXT_EMBEDDING)20 21            # Use `mxbai-embed-large-v1` model for validate,22            model_instance.validate_credentials(model="mxbai-embed-large-v1", credentials=credentials)23        except CredentialsValidateFailedError as ex:24            raise ex25        except Exception as ex:26            logger.exception(f"{self.get_provider_schema().provider} credentials validate failed")27            raise ex28