edgygraph 0.0.7__tar.gz

edgygraph-0.0.7/PKG-INFO

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

edgygraph-0.0.7/README.md

@@ -0,0 +1 @@
+# Graph-based Pipeline Builder

edgygraph-0.0.7/edgygraph/__init__.py

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

edgygraph-0.0.7/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-0.0.7/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 = 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: GraphState = await next_node.run(state)
+                current_node = next_node
+
+        return state

edgygraph-0.0.7/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-0.0.7/edgygraph/states.py

@@ -0,0 +1,28 @@
+from pydantic import BaseModel, Field
+from typing import TypeVar, AsyncIterator, Protocol, Generic
+from types import TracebackType
+
+class GraphState(BaseModel):
+    vars: dict[str, object] = Field(default_factory=dict)
+
+
+T = TypeVar('T', covariant=True)
+
+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()
+
+
+class GraphStateStream(GraphState, Generic[T]):
+    
+    stream: Stream[T] | None = None

edgygraph-0.0.7/edgygraph.egg-info/PKG-INFO

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

edgygraph-0.0.7/edgygraph.egg-info/SOURCES.txt

@@ -0,0 +1,12 @@
+README.md
+pyproject.toml
+edgygraph/__init__.py
+edgygraph/edges.py
+edgygraph/graph.py
+edgygraph/nodes.py
+edgygraph/py.typed
+edgygraph/states.py
+edgygraph.egg-info/PKG-INFO
+edgygraph.egg-info/SOURCES.txt
+edgygraph.egg-info/dependency_links.txt
+edgygraph.egg-info/top_level.txt

edgygraph-0.0.7/edgygraph.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

edgygraph-0.0.7/edgygraph.egg-info/top_level.txt

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

edgygraph-0.0.7/pyproject.toml

@@ -0,0 +1,12 @@
+[project]
+name = "edgygraph"
+version = "0.0.7"
+description = "Graph-based Pipeline Builder"
+authors = [{ name="Mathis Siebert" }]
+requires-python = ">=3.11"
+dependencies = []
+
+[build-system]
+requires = ["setuptools", "wheel"]
+build-backend = "setuptools.build_meta"
+

edgygraph-0.0.7/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+