From d89fb9eef0d0bb1751c6a8f72fb407b93a4acedf Mon Sep 17 00:00:00 2001 From: Andrew Nguonly Date: Wed, 4 Sep 2024 15:36:05 -0700 Subject: [PATCH] Add before and after values to __interrupt metadata. --- libs/langgraph/langgraph/graph/graph.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/libs/langgraph/langgraph/graph/graph.py b/libs/langgraph/langgraph/graph/graph.py index 21868bf7c..b04594bdd 100644 --- a/libs/langgraph/langgraph/graph/graph.py +++ b/libs/langgraph/langgraph/graph/graph.py @@ -512,7 +512,9 @@ class CompiledGraph(Pregel): for key, n in self.builder.nodes.items(): node = n.runnable metadata = n.metadata or {} - if key in self.interrupt_before_nodes: + if key in self.interrupt_before_nodes and key in self.interrupt_after_nodes: + metadata["__interrupt"] = "before,after" + elif key in self.interrupt_before_nodes: metadata["__interrupt"] = "before" elif key in self.interrupt_after_nodes: metadata["__interrupt"] = "after"