WEBDIENSTE/langgraph
1
0
Fork 0
mirror of https://github.com/langchain-ai/langgraph.git synced 2026-09-06 01:37:49 +02:00
Code Issues Packages Projects Releases Wiki Activity

better

Browse Source
This commit is contained in:
Sydney Runkle
2026-02-27 15:40:43 -05:00
parent cd4be9241e
commit 71c0500caa
1 changed files with 12 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
libs/langgraph/langgraph/pregel/main.py
+12
View File
@@ -3091,6 +3091,8 @@ class Pregel(
chunks: list[dict[str, Any] | Any] = []
interrupts: list[Interrupt] = []
subgraphs = kwargs.get("subgraphs", False)
for chunk in self.stream(
input,
config,
@@ -3103,6 +3105,10 @@ 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":
latest = chunk
if (
@@ -3171,6 +3177,8 @@ class Pregel(
chunks: list[dict[str, Any] | Any] = []
interrupts: list[Interrupt] = []
subgraphs = kwargs.get("subgraphs", False)
async for chunk in self.astream(
input,
config,
@@ -3183,6 +3191,10 @@ 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":
latest = chunk
if (