From 2a624274fd96714bbe7131ad4b7c4bd5cd19c2ff Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Thu, 29 Feb 2024 18:26:28 -0800 Subject: [PATCH] Draw end edge for cond edge without mapping --- langgraph/graph/graph.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/langgraph/graph/graph.py b/langgraph/graph/graph.py index b4ce617a8..14860dd9d 100644 --- a/langgraph/graph/graph.py +++ b/langgraph/graph/graph.py @@ -257,7 +257,10 @@ class CompiledGraph(Pregel): name, ) graph.add_edge(graph.nodes[start], graph.nodes[name]) - ends = branch.ends or {k: k for k in self.graph.nodes} + ends = branch.ends or { + **{k: k for k in self.graph.nodes}, + END: END, + } for label, end in ends.items(): graph.add_edge(graph.nodes[name], graph.nodes[end], label) if self.graph.entry_point_branch: