diff --git a/libs/langgraph/langgraph/pregel/_algo.py b/libs/langgraph/langgraph/pregel/_algo.py index 08b5b3cc0..d8f061ff5 100644 --- a/libs/langgraph/langgraph/pregel/_algo.py +++ b/libs/langgraph/langgraph/pregel/_algo.py @@ -1121,14 +1121,10 @@ def sanitize_untracked_values_in_send( # Command return packet - sanitized_arg = dict(packet.arg) - - # top level keys should be the channel names - to_pop = set() - for k, v in sanitized_arg.items(): - if isinstance(channels.get(k), UntrackedValue): - to_pop.add(k) - for k in to_pop: - sanitized_arg.pop(k) + sanitized_arg = { + k: v + for k, v in packet.arg.items() + if not isinstance(channels.get(k), UntrackedValue) + } return Send(node=packet.node, arg=sanitized_arg)