Explorar o código

fix agent_execution_metadata (#15444)

heyszt hai 1 mes
pai
achega
a58b990855
Modificáronse 1 ficheiros con 3 adicións e 1 borrados
  1. 3 1
      api/core/workflow/nodes/tool/tool_node.py

+ 3 - 1
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: