Add before and after values to __interrupt metadata.

This commit is contained in:
Andrew Nguonly
2024-09-04 15:36:05 -07:00
parent dc10b3aa8e
commit d89fb9eef0
+3 -1
View File
@@ -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"