dify/api/core/tools/errors.py

from core.tools.entities.tool_entities import ToolInvokeMeta


class ToolProviderNotFoundError(ValueError):
    pass


class ToolNotFoundError(ValueError):
    pass


class ToolParameterValidationError(ValueError):
    pass


class ToolProviderCredentialValidationError(ValueError):
    pass


class ToolNotSupportedError(ValueError):
    pass


class ToolInvokeError(ValueError):
    pass


class ToolApiSchemaError(ValueError):
    pass


class ToolSSRFError(ValueError):
    pass


class ToolCredentialPolicyViolationError(ValueError):
    pass


class ApiToolProviderNotFoundError(ValueError):
    error_code = "api_tool_provider_not_found"
    provider_name: str
    tenant_id: str

    def __init__(self, provider_name: str, tenant_id: str):
        self.provider_name = provider_name
        self.tenant_id = tenant_id
        super().__init__(f"api provider {provider_name} does not exist")


class ToolEngineInvokeError(Exception):
    meta: ToolInvokeMeta

    def __init__(self, meta, **kwargs):
        self.meta = meta
        super().__init__(**kwargs)