AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-20 01:29:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: missing tool invoke messages

Browse Source
This commit is contained in:
Yeuoly 2024-11-28 19:09:04 +08:00
parent eead6abe85
commit 0452dfd029
No known key found for this signature in database
GPG Key ID: A66E7E320FB19F61
1 changed files with 13 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -7,6 +7,7 @@ from sqlalchemy.orm import Session
from core.callback_handler.workflow_tool_callback_handler import DifyWorkflowCallbackHandler from core.callback_handler.workflow_tool_callback_handler import DifyWorkflowCallbackHandler
from core.file import File, FileTransferMethod, FileType from core.file import File, FileTransferMethod, FileType
from core.plugin.manager.exc import PluginInvokeError
from core.tools.entities.tool_entities import ToolInvokeMessage, ToolParameter from core.tools.entities.tool_entities import ToolInvokeMessage, ToolParameter
from core.tools.tool_engine import ToolEngine from core.tools.tool_engine import ToolEngine
from core.tools.tool_manager import ToolManager from core.tools.tool_manager import ToolManager
@ -101,8 +102,18 @@ class ToolNode(BaseNode[ToolNodeData]):
) )
) )
# convert tool messages try:
yield from self._transform_message(message_stream, tool_info, parameters_for_log) # convert tool messages
yield from self._transform_message(message_stream, tool_info, parameters_for_log)
except PluginInvokeError as e:
yield RunCompletedEvent(
run_result=NodeRunResult(
status=WorkflowNodeExecutionStatus.FAILED,
inputs=parameters_for_log,
metadata={NodeRunMetadataKey.TOOL_INFO: tool_info},
error=f"Failed to transform tool message: {str(e)}",
)
)
def _generate_parameters( def _generate_parameters(
self, self,