youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-05 09:58:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: unify error handling

Browse Source
This commit is contained in:
Yeuoly
2024-11-28 20:44:06 +08:00
parent 772009115d
commit 8c5928da2f
5 changed files with 58 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
api/core/workflow/nodes/tool/tool_node.py
View File

@ -7,7 +7,7 @@ from sqlalchemy.orm import Session
from core.callback_handler.workflow_tool_callback_handler import DifyWorkflowCallbackHandler
from core.file import File, FileTransferMethod, FileType
from core.plugin.manager.exc import PluginInvokeError
from core.plugin.manager.exc import PluginDaemonClientSideError
from core.tools.entities.tool_entities import ToolInvokeMessage, ToolParameter
from core.tools.tool_engine import ToolEngine
from core.tools.tool_manager import ToolManager
@ -105,7 +105,7 @@ class ToolNode(BaseNode[ToolNodeData]):
try:
# convert tool messages
yield from self._transform_message(message_stream, tool_info, parameters_for_log)
except PluginInvokeError as e:
except PluginDaemonClientSideError as e:
yield RunCompletedEvent(
run_result=NodeRunResult(
status=WorkflowNodeExecutionStatus.FAILED,