edgygraph 0.0.11__py3-none-any.whl

edgygraph/__init__.py

@@ -0,0 +1,14 @@
+from .edges import GraphEdge, START, END
+from .nodes import GraphNode
+from .states import GraphState, Stream
+from .graph import GraphExecutor
+
+__all__ = [
+    "GraphEdge",
+    "GraphNode",
+    "GraphState",
+    "Stream",
+    "GraphExecutor",
+    "START",
+    "END",
+]

edgygraph/edges.py

@@ -0,0 +1,22 @@
+from typing import Callable, Type, TypeVar, Generic
+from .states import GraphState
+from .nodes import GraphNode
+
+
+class START:
+    pass
+
+class END:
+    pass
+
+
+T = TypeVar('T', bound=GraphState)
+
+class GraphEdge(Generic[T]):
+
+    source: GraphNode[T] | Type[START]
+    next: Callable[[T], GraphNode[T] | Type[END]]
+
+    def __init__(self, source: GraphNode[T] | Type[START], next: Callable[[T], GraphNode[T] | Type[END]]):
+        self.source = source
+        self.next = next

edgygraph/graph.py

@@ -0,0 +1,35 @@
+from .edges import GraphEdge, START, END
+from .nodes import GraphNode
+from .states import GraphState
+from typing import Type, TypeVar, Generic
+
+T = TypeVar('T', bound=GraphState)
+
+class GraphExecutor(Generic[T]):
+
+    edges: list[GraphEdge[T]]
+
+    def __init__(self, edges: list[GraphEdge[T]]):
+        self.edges = edges
+
+    async def __call__(self, initial_state: T) -> T:
+        state: T = initial_state
+        current_node: GraphNode[T] | Type[START] = START
+
+        index_dict: dict[GraphNode[T] | Type[START], GraphEdge[T]] = {edge.source: edge for edge in self.edges}
+
+        while True:
+            # Find the edge corresponding to the current node
+            edge: GraphEdge[T] = index_dict[current_node]
+            # Determine the next node using the edge's next function
+            next_node = edge.next(state)
+
+            if next_node == END:
+                break
+            else:
+                assert isinstance(next_node, GraphNode)
+                # Run the current node to update the state
+                state: T = await next_node.run(state)
+                current_node = next_node
+
+        return state

edgygraph/nodes.py

@@ -0,0 +1,12 @@
+from abc import ABC, abstractmethod
+from .states import GraphState
+from typing import TypeVar, Generic
+
+
+T = TypeVar('T', bound=GraphState)
+
+class GraphNode(ABC, Generic[T]):
+    
+    @abstractmethod
+    async def run(self, state: T) -> T:
+        pass

edgygraph/states.py

@@ -0,0 +1,27 @@
+from pydantic import BaseModel, Field
+from typing import TypeVar, AsyncIterator, Protocol, Generic
+from types import TracebackType
+
+
+T = TypeVar('T', covariant=True, default=object)
+
+class Stream(AsyncIterator[T], Protocol):
+    async def aclose(self) -> None: ...
+
+    async def __aenter__(self) -> "Stream[T]":
+        return self
+
+    async def __aexit__(
+            self, exc_type: type[BaseException] | None, 
+            exc: BaseException | None, 
+            tb: TracebackType | None
+        ) -> None: # Not handling exceptions here -> returns None
+
+        await self.aclose()
+
+
+S = TypeVar('S', bound=Stream, contravariant=True, default=Stream)
+
+class GraphState(BaseModel, Generic[T]):
+    vars: dict[str, object] = Field(default_factory=dict)
+    streams: dict[str, T] = Field(default_factory=dict)

edgygraph-0.0.11.dist-info/METADATA

@@ -0,0 +1,6 @@
+Metadata-Version: 2.4
+Name: edgygraph
+Version: 0.0.11
+Summary: Graph-based Pipeline Builder
+Author: Mathis Siebert
+Requires-Python: >=3.13

edgygraph-0.0.11.dist-info/RECORD

@@ -0,0 +1,10 @@
+edgygraph/__init__.py,sha256=L4CHY_1tID4NTEO4XyH-8CNLPUWbZ9vh4u2mjaEIe8I,267
+edgygraph/edges.py,sha256=J_zxFfQaGuQuMg3TohuHlfcDlCo5lpoa3tHowe1d-Jk,476
+edgygraph/graph.py,sha256=cvDsyAXGCr4RfWIOcM7XFy9_L6gwa-qbRlan1Z7BPIU,1136
+edgygraph/nodes.py,sha256=tIWg59M1n3D7u7EvloiVt6SioZZj5GW0m6zAayksy3M,252
+edgygraph/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+edgygraph/states.py,sha256=CkYNtmgvmoN1witDFPEHYcc8bTyXVMrxQchS9gCQ10M,812
+edgygraph-0.0.11.dist-info/METADATA,sha256=HesifsK8qWhUpeaHEBcJK6GfKcSLVlNLduUcE38HNv4,139
+edgygraph-0.0.11.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+edgygraph-0.0.11.dist-info/top_level.txt,sha256=1aTnTyO0rVyypQxuzthXouPY8XY6ybCuRrRvS4FNLeY,10
+edgygraph-0.0.11.dist-info/RECORD,,

edgygraph-0.0.11.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (80.9.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

edgygraph-0.0.11.dist-info/top_level.txt

@@ -0,0 +1 @@
+edgygraph