WEBDIENSTE/langgraph
1
0
Fork 0
mirror of https://github.com/langchain-ai/langgraph.git synced 2026-09-29 13:05:15 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix

Browse Source
This commit is contained in:
Nuno Campos
2024-10-30 14:17:50 -07:00
parent 5cb6ff79ca
commit 59e8d59b7c
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
libs/langgraph/langgraph/pregel/remote.py
+4 -4
View File
@@ -593,8 +593,6 @@ class RemoteGraph(PregelProtocol):
ns = caller_ns + ns
if stream is not None and chunk.event in stream.modes:
stream((ns, mode, chunk.data))
if chunk.event not in stream_modes:
continue
if chunk.event.startswith("updates"):
if isinstance(chunk.data, dict) and INTERRUPT in chunk.data:
raise GraphInterrupt(chunk.data[INTERRUPT])
@@ -602,6 +600,8 @@ class RemoteGraph(PregelProtocol):
continue
elif chunk.event.startswith("error"):
raise RemoteException(chunk.data)
if chunk.event.split("|", 1)[0] not in stream_modes:
continue
if subgraphs:
if "|" in chunk.event:
mode, ns_ = chunk.event.split("|", 1)
@@ -675,8 +675,6 @@ class RemoteGraph(PregelProtocol):
ns = caller_ns + ns
if stream is not None and chunk.event in stream.modes:
stream((ns, mode, chunk.data))
if chunk.event not in stream_modes:
continue
if chunk.event.startswith("updates"):
if isinstance(chunk.data, dict) and INTERRUPT in chunk.data:
raise GraphInterrupt(chunk.data[INTERRUPT])
@@ -684,6 +682,8 @@ class RemoteGraph(PregelProtocol):
continue
elif chunk.event.startswith("error"):
raise RemoteException(chunk.data)
if chunk.event.split("|", 1)[0] not in stream_modes:
continue
if subgraphs:
if req_single:
yield ns, chunk.data