瀏覽代碼

fix: handle EndpointSetupFailedError in BasePluginClient (#19613)

tags/1.4.0
Yeuoly 5 月之前
父節點
當前提交
ff0feaf34e
沒有連結到貢獻者的電子郵件帳戶。
共有 2 個檔案被更改,包括 9 行新增0 行删除
  1. 6
    0
      api/core/plugin/endpoint/exc.py
  2. 3
    0
      api/core/plugin/impl/base.py

+ 6
- 0
api/core/plugin/endpoint/exc.py 查看文件

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

pass

+ 3
- 0
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__:

Loading…
取消
儲存