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

Fix formatting in graph

Browse Source
This commit is contained in:
Angel Igareta
2024-04-17 18:03:22 +02:00
parent 35fe16b615
commit 4e8e51126d
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
langgraph/graph/graph.py
+6 -2
View File
@@ -364,7 +364,11 @@ class CompiledGraph(Pregel):
self.nodes[end].channels.append(channel_name)
def get_graph(
self, config: Optional[RunnableConfig] = None, *, xray: bool = False, add_condition_nodes: bool = True
self,
config: Optional[RunnableConfig] = None,
*,
xray: bool = False,
add_condition_nodes: bool = True,
) -> RunnableGraph:
"""Returns a drawable representation of the computation graph."""
graph = RunnableGraph()
@@ -405,7 +409,7 @@ class CompiledGraph(Pregel):
END: END,
}
if add_condition_nodes is True:
if add_condition_nodes is True:
cond = graph.add_node(branch.condition, name)
graph.add_edge(start_nodes[start], cond)
for label, end in ends.items():