Merge pull request #1610 from langchain-ai/an/04sept/metadata-interrupts

Return `before,after` for `__interrupt` metadata if both interrupts are configured for a node
This commit is contained in:
Nuno Campos
2024-09-04 17:29:07 -07:00
committed by GitHub
+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"