diff --git a/libs/langgraph/langgraph/pregel/main.py b/libs/langgraph/langgraph/pregel/main.py index c22f9fa29..04d58f909 100644 --- a/libs/langgraph/langgraph/pregel/main.py +++ b/libs/langgraph/langgraph/pregel/main.py @@ -3105,11 +3105,11 @@ class Pregel( durability=durability, **kwargs, ): - # when subgraphs=True with a single stream_mode, stream() - # yields (namespace, payload) tuples — unwrap them - if subgraphs and isinstance(chunk, tuple): - chunk = chunk[-1] if stream_mode == "values": + # when subgraphs=True with a single stream_mode, stream() + # yields (namespace, payload) tuples — unwrap them + if subgraphs and isinstance(chunk, tuple): + chunk = chunk[-1] latest = chunk if ( isinstance(chunk, dict) @@ -3191,11 +3191,11 @@ class Pregel( durability=durability, **kwargs, ): - # when subgraphs=True with a single stream_mode, astream() - # yields (namespace, payload) tuples — unwrap them - if subgraphs and isinstance(chunk, tuple): - chunk = chunk[-1] if stream_mode == "values": + # when subgraphs=True with a single stream_mode, astream() + # yields (namespace, payload) tuples — unwrap them + if subgraphs and isinstance(chunk, tuple): + chunk = chunk[-1] latest = chunk if ( isinstance(chunk, dict)