From cc2af50ce9794d66aeff1473fe87b579c255fe0c Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Sat, 6 Jan 2024 14:53:22 -0800 Subject: [PATCH] Lint --- permchain/langgraph/__init__.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/permchain/langgraph/__init__.py b/permchain/langgraph/__init__.py index e082ee8ca..02573bfc7 100644 --- a/permchain/langgraph/__init__.py +++ b/permchain/langgraph/__init__.py @@ -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)