This commit is contained in:
Nuno Campos
2024-02-19 13:19:49 -08:00
parent 4f1980df7c
commit b27531e962
+3 -3
View File
@@ -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)