Merge pull request #1089 from mihaic/patch-1

Fix StateGraph.add_edge end_key check
2026-09-09 19:27:54 +02:00 · 2024-07-23 13:52:33 -07:00
parent 9711f39995 3f43e0d945
commit 4fa5c8bab0
1 changed files with 2 additions and 2 deletions
@@ -358,8 +358,8 @@ class StateGraph(Graph):
                raise ValueError("END cannot be a start node")
            if start not in self.nodes:
                raise ValueError(f"Need to add_node `{start}` first")
-        if end_key == END:
-            raise ValueError("END cannot be an end node")
+        if end_key == START:
+            raise ValueError("START cannot be an end node")
        if end_key not in self.nodes:
            raise ValueError(f"Need to add_node `{end_key}` first")