From a58b990855cd3ca4d301aeeb9466b3a7f549d412 Mon Sep 17 00:00:00 2001 From: heyszt <36215648+hieheihei@users.noreply.github.com> Date: Tue, 11 Mar 2025 14:35:08 +0800 Subject: [PATCH] fix agent_execution_metadata (#15444) --- api/core/workflow/nodes/tool/tool_node.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/api/core/workflow/nodes/tool/tool_node.py b/api/core/workflow/nodes/tool/tool_node.py index 1e31a2c324..7ec092cfdd 100644 --- a/api/core/workflow/nodes/tool/tool_node.py +++ b/api/core/workflow/nodes/tool/tool_node.py @@ -270,7 +270,9 @@ class ToolNode(BaseNode[ToolNodeData]): if self.node_type == NodeType.AGENT: msg_metadata = message.message.json_object.pop("execution_metadata", {}) agent_execution_metadata = { - key: value for key, value in msg_metadata.items() if key in NodeRunMetadataKey + key: value + for key, value in msg_metadata.items() + if key in NodeRunMetadataKey.__members__.values() } json.append(message.message.json_object) elif message.type == ToolInvokeMessage.MessageType.LINK: