fix agent_execution_metadata (#15444)
This commit is contained in:
parent
b6b1903a37
commit
a58b990855
@ -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:
|
||||||
|
Loading…
Reference in New Issue
Block a user