Преглед на файлове

fix agent_execution_metadata (#15444)

heyszt преди 1 месец
родител
ревизия
a58b990855
променени са 1 файла, в които са добавени 3 реда и са изтрити 1 реда
  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: