This commit is contained in:
Nuno Campos
2024-01-06 14:53:22 -08:00
parent d399074d05
commit cc2af50ce9
+2 -1
View File
@@ -26,7 +26,6 @@ class Branch(NamedTuple):
return Channel.write_to(self.ends[result])
START = "__start__"
END = "__end__"
@@ -39,6 +38,8 @@ class Graph:
def add_node(self, key: str, action: RunnableLike) -> None:
if key in self.nodes:
raise ValueError(f"Node `{key}` already present.")
if key == END:
raise ValueError(f"Node `{key}` is reserved.")
self.nodes[key] = coerce_to_runnable(action)