heyszt
7 months ago
1 changed files with 3 additions and 1 deletions
+ 3
- 1
api/core/workflow/nodes/tool/tool_node.py
View File
| if self.node_type == NodeType.AGENT: | if self.node_type == NodeType.AGENT: | ||||
| msg_metadata = message.message.json_object.pop("execution_metadata", {}) | msg_metadata = message.message.json_object.pop("execution_metadata", {}) | ||||
| agent_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) | json.append(message.message.json_object) | ||||
| elif message.type == ToolInvokeMessage.MessageType.LINK: | elif message.type == ToolInvokeMessage.MessageType.LINK: |
Loading…