narrowed down runtime check of type Next

Author: mathisxy

src/edgygraph/graph/types.py

@@ -36,7 +36,11 @@ class Types[T: StateProtocol, S: SharedProtocol]:
     def is_next(cls, x: Any) -> TypeGuard[Next[T, S]]:
         return (
             cls.is_resolved_next(x) or
-            callable(x)
+            (callable(x) and not (
+                cls.is_any_source(x) or
+                cls.is_resolved_next(x) or
+                x is END
+            ))
         )
 
     @classmethod

src/edgygraph/nodes.py

@@ -1,4 +1,6 @@
+from __future__ import annotations
 from abc import ABC, abstractmethod
+from copy import copy
 
 from .states import StateProtocol, SharedProtocol
 
@@ -33,6 +35,19 @@ async def __call__(self, state: T, shared: S) -> None:
         """
         pass
 
+    
+    def copy(self) -> Node[T, S]:
+        """
+        Creates a copy of the node. This can be used when the same node is used multiple times in the graph.
+
+        Args:
+            node: The node to copy.
+
+        Returns:
+            A copy of the node.
+        """
+        return copy(self)
+
 
 
 class START: