From 569055539449409e95c4760c71f7c77361f6ed3f Mon Sep 17 00:00:00 2001 From: vbarda Date: Tue, 8 Apr 2025 12:51:36 -0400 Subject: [PATCH] langgraph: raise GraphInterrupt only if used as a subgraph --- libs/langgraph/langgraph/pregel/remote.py | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) diff --git a/libs/langgraph/langgraph/pregel/remote.py b/libs/langgraph/langgraph/pregel/remote.py index f58d0a4d9..07d44cb66 100644 --- a/libs/langgraph/langgraph/pregel/remote.py +++ b/libs/langgraph/langgraph/pregel/remote.py @@ -654,9 +654,10 @@ class RemoteGraph(PregelProtocol): # raise interrupt or errors if chunk.event.startswith("updates"): if isinstance(chunk.data, dict) and INTERRUPT in chunk.data: - raise GraphInterrupt( - [Interrupt(**i) for i in chunk.data[INTERRUPT]] - ) + if caller_ns: + raise GraphInterrupt( + [Interrupt(**i) for i in chunk.data[INTERRUPT]] + ) elif chunk.event.startswith("error"): raise RemoteException(chunk.data) # filter for what was actually requested @@ -748,9 +749,10 @@ class RemoteGraph(PregelProtocol): # raise interrupt or errors if chunk.event.startswith("updates"): if isinstance(chunk.data, dict) and INTERRUPT in chunk.data: - raise GraphInterrupt( - [Interrupt(**i) for i in chunk.data[INTERRUPT]] - ) + if caller_ns: + raise GraphInterrupt( + [Interrupt(**i) for i in chunk.data[INTERRUPT]] + ) elif chunk.event.startswith("error"): raise RemoteException(chunk.data) # filter for what was actually requested