|
@@ -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:
|