estel/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix: handle EndpointSetupFailedError in BasePluginClient (#19613)

Browse Source
This commit is contained in:
Yeuoly
2025-05-15 13:59:43 +08:00
committed by GitHub
parent dc75a10989
commit ff0feaf34e
2 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
api/core/plugin/endpoint/exc.py Normal file
View File

@@ -0,0 +1,6 @@
class EndpointSetupFailedError(ValueError):
"""
Endpoint setup failed error
"""
pass

3
api/core/plugin/impl/base.py
View File

@@ -17,6 +17,7 @@ from core.model_runtime.errors.invoke import (
InvokeServerUnavailableError, InvokeServerUnavailableError,
) )
from core.model_runtime.errors.validate import CredentialsValidateFailedError from core.model_runtime.errors.validate import CredentialsValidateFailedError
from core.plugin.endpoint.exc import EndpointSetupFailedError
from core.plugin.entities.plugin_daemon import PluginDaemonBasicResponse, PluginDaemonError, PluginDaemonInnerError from core.plugin.entities.plugin_daemon import PluginDaemonBasicResponse, PluginDaemonError, PluginDaemonInnerError
from core.plugin.impl.exc import ( from core.plugin.impl.exc import (
PluginDaemonBadRequestError, PluginDaemonBadRequestError,
@@ -219,6 +220,8 @@ class BasePluginClient:
raise InvokeServerUnavailableError(description=args.get("description")) raise InvokeServerUnavailableError(description=args.get("description"))
case CredentialsValidateFailedError.__name__: case CredentialsValidateFailedError.__name__:
raise CredentialsValidateFailedError(error_object.get("message")) raise CredentialsValidateFailedError(error_object.get("message"))
case EndpointSetupFailedError.__name__:
raise EndpointSetupFailedError(error_object.get("message"))
case _: case _:
raise PluginInvokeError(description=message) raise PluginInvokeError(description=message)
case PluginDaemonInternalServerError.__name__: case PluginDaemonInternalServerError.__name__: