model_provider.py 891 B

12345678910111213141516171819202122232425262728
  1. import logging
  2. from collections.abc import Mapping
  3. from dify_plugin import ModelProvider
  4. from dify_plugin.entities.model import ModelType
  5. from dify_plugin.errors.model import CredentialsValidateFailedError
  6. logger = logging.getLogger(__name__)
  7. class {{ .PluginName | SnakeToCamel }}ModelProvider(ModelProvider):
  8. def validate_provider_credentials(self, credentials: Mapping) -> None:
  9. """
  10. Validate provider credentials
  11. if validate failed, raise exception
  12. :param credentials: provider credentials, credentials form defined in `provider_credential_schema`.
  13. """
  14. try:
  15. pass
  16. except CredentialsValidateFailedError as ex:
  17. raise ex
  18. except Exception as ex:
  19. logger.exception(
  20. f"{self.get_provider_schema().provider} credentials validate failed"
  21. )
  22. raise ex