WEBDIENSTE/langgraph
1
0
Fork 0
mirror of https://github.com/langchain-ai/langgraph.git synced 2026-08-20 22:52:29 +02:00
Code Issues Packages Projects Releases Wiki Activity

Remove labels if redundant

Browse Source
This commit is contained in:
Nuno Campos
2024-04-29 10:44:45 -07:00
parent a83fd19b29
commit 4a28f19922
1 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
langgraph/graph/graph.py
+10 -2
View File
@@ -450,13 +450,21 @@ class CompiledGraph(Pregel):
)
graph.add_edge(start_nodes[start], cond)
for label, end in ends.items():
graph.add_edge(cond, end_nodes[end], label, conditional=True)
graph.add_edge(
cond,
end_nodes[end],
label if label != end else None,
conditional=True,
)
if branch.then is not None:
graph.add_edge(start_nodes[end], end_nodes[branch.then])
else:
for label, end in ends.items():
graph.add_edge(
start_nodes[start], end_nodes[end], label, conditional=True
start_nodes[start],
end_nodes[end],
label if label != end else None,
conditional=True,
)
if branch.then is not None:
graph.add_edge(start_nodes[end], end_nodes[branch.then])