From b27531e962e5f037852b62bc148935307532d637 Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Mon, 19 Feb 2024 13:19:49 -0800 Subject: [PATCH] Lint --- langgraph/graph/graph.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/langgraph/graph/graph.py b/langgraph/graph/graph.py index 22eba86ad..16e3551f5 100644 --- a/langgraph/graph/graph.py +++ b/langgraph/graph/graph.py @@ -196,10 +196,10 @@ class Graph: class CompiledGraph(Pregel): graph: Graph - def get_graph(self, config: RunnableConfig | None = None) -> RunnableGraph: + def get_graph(self, config: Optional[RunnableConfig] = None) -> RunnableGraph: graph = RunnableGraph() - graph.add_node(self.get_input_schema(), "__start__") - graph.add_node(self.get_output_schema(), END) + graph.add_node(self.get_input_schema(config), "__start__") + graph.add_node(self.get_output_schema(config), END) for key, node in self.graph.nodes.items(): graph.add_node(node, key)