From ff0feaf34ea76c91165c548970ae265d7d30e972 Mon Sep 17 00:00:00 2001
From: Yeuoly <45712896+Yeuoly@users.noreply.github.com>
Date: Thu, 15 May 2025 13:59:43 +0800
Subject: [PATCH] fix: handle EndpointSetupFailedError in BasePluginClient
 (#19613)

---
 api/core/plugin/endpoint/exc.py | 6 ++++++
 api/core/plugin/impl/base.py    | 3 +++
 2 files changed, 9 insertions(+)
 create mode 100644 api/core/plugin/endpoint/exc.py

diff --git a/api/core/plugin/endpoint/exc.py b/api/core/plugin/endpoint/exc.py
new file mode 100644
index 0000000000..aa29f1e9a1
--- /dev/null
+++ b/api/core/plugin/endpoint/exc.py
@@ -0,0 +1,6 @@
+class EndpointSetupFailedError(ValueError):
+    """
+    Endpoint setup failed error
+    """
+
+    pass
diff --git a/api/core/plugin/impl/base.py b/api/core/plugin/impl/base.py
index 4f1d808a3e..591e7b0525 100644
--- a/api/core/plugin/impl/base.py
+++ b/api/core/plugin/impl/base.py
@@ -17,6 +17,7 @@ from core.model_runtime.errors.invoke import (
     InvokeServerUnavailableError,
 )
 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.impl.exc import (
     PluginDaemonBadRequestError,
@@ -219,6 +220,8 @@ class BasePluginClient:
                         raise InvokeServerUnavailableError(description=args.get("description"))
                     case CredentialsValidateFailedError.__name__:
                         raise CredentialsValidateFailedError(error_object.get("message"))
+                    case EndpointSetupFailedError.__name__:
+                        raise EndpointSetupFailedError(error_object.get("message"))
                     case _:
                         raise PluginInvokeError(description=message)
             case PluginDaemonInternalServerError.__name__: