Files
dify-plugin-daemon/cmd/commandline/plugin/templates/python/model_provider.py
2024-11-13 20:38:47 +08:00

28 lines
891 B
Python

import logging
from collections.abc import Mapping
from dify_plugin import ModelProvider
from dify_plugin.entities.model import ModelType
from dify_plugin.errors.model import CredentialsValidateFailedError
logger = logging.getLogger(__name__)
class {{ .PluginName | SnakeToCamel }}ModelProvider(ModelProvider):
def validate_provider_credentials(self, credentials: Mapping) -> None:
"""
Validate provider credentials
if validate failed, raise exception
:param credentials: provider credentials, credentials form defined in `provider_credential_schema`.
"""
try:
pass
except CredentialsValidateFailedError as ex:
raise ex
except Exception as ex:
logger.exception(
f"{self.get_provider_schema().provider} credentials validate failed"
)
raise ex