Procházet zdrojové kódy

fix agent_execution_metadata (#15444)

heyszt před 1 měsícem
rodič
revize
a58b990855
1 změnil soubory, kde provedl 3 přidání a 1 odebrání
  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:
                 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: