stageflow-framework 0.1.2__tar.gz → 0.1.3__tar.gz

{stageflow_framework-0.1.2 → stageflow_framework-0.1.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: stageflow-framework
-Version: 0.1.2
+Version: 0.1.3
 Summary: StageFlow: Pipeline framework for stages
 Author: lethargy
 Requires-Python: >=3.10

{stageflow_framework-0.1.2 → stageflow_framework-0.1.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "stageflow-framework"
-version = "0.1.2"
+version = "0.1.3"
 description = "StageFlow: Pipeline framework for stages"
 readme = "README.md"
 authors = [

{stageflow_framework-0.1.2 → stageflow_framework-0.1.3}/stageflow/core/session.py

@@ -87,15 +87,20 @@ class Session:
             self._pending_inputs.setdefault(type_, []).append(entry)
         return entry
 
-    async def wait_input(self, type_: str, timeout: float | None = None):
-        # Deliver buffered input if it arrived before waiter.
+    def start_wait_input(self, type_: str) -> asyncio.Future:
         pending = self._pending_inputs.get(type_)
         if pending:
-            return pending.pop(0)
+            loop = asyncio.get_running_loop()
+            fut = loop.create_future()
+            fut.set_result(pending.pop(0))
+            return fut
         loop = asyncio.get_running_loop()
         fut = loop.create_future()
         self._waiting.setdefault(type_, []).append(fut)
         self.emit(Event(type="waiting_for_input", session_id=self.id, payload={"type": type_}))
+        return fut
+
+    async def finish_wait_input(self, type_: str, fut: asyncio.Future, timeout: float | None = None):
         try:
             return await asyncio.wait_for(fut, timeout=timeout)
         except asyncio.TimeoutError:
@@ -108,6 +113,10 @@ class Session:
                 if not waiters:
                     del self._waiting[type_]
 
+    async def wait_input(self, type_: str, timeout: float | None = None):
+        fut = self.start_wait_input(type_)
+        return await self.finish_wait_input(type_, fut, timeout=timeout)
+
     def last_input(self, type_: str | None = None):
         if not self.input_history:
             return None

{stageflow_framework-0.1.2 → stageflow_framework-0.1.3}/stageflow_framework.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: stageflow-framework
-Version: 0.1.2
+Version: 0.1.3
 Summary: StageFlow: Pipeline framework for stages
 Author: lethargy
 Requires-Python: >=3.10

{stageflow_framework-0.1.2 → stageflow_framework-0.1.3}/stageflow_framework.egg-info/SOURCES.txt

@@ -34,6 +34,7 @@ tests/test_full_pipeline.py
 tests/test_jsonlogic.py
 tests/test_payload_validation.py
 tests/test_pipeline_tester.py
+tests/test_session_wait_input.py
 tests/test_snapshot.py
 tests/test_std_stages.py
 tests/test_subpipeline.py

stageflow_framework-0.1.3/tests/test_session_wait_input.py

@@ -0,0 +1,40 @@
+import unittest
+
+from stageflow.core.context import Context
+from stageflow.core.node import TerminalNode
+from stageflow.core.pipeline import Pipeline
+from stageflow.core.session import Session
+
+
+def _make_pipeline():
+    end = TerminalNode(id="end", type="terminal")
+    return Pipeline(entry="end", nodes=[end])
+
+
+class SessionWaitInputTests(unittest.IsolatedAsyncioTestCase):
+    async def test_start_and_finish_wait_input_success(self):
+        session = Session(id="s1", pipeline=_make_pipeline(), context=Context())
+        waiter = session.start_wait_input("user")
+
+        await session.input("user", {"id": 1})
+        result = await session.finish_wait_input("user", waiter, timeout=1)
+
+        self.assertEqual(result["payload"]["id"], 1)
+        self.assertIn("waiting_for_input", [e.type for e in session.event_history])
+
+    async def test_start_finish_timeout_and_pending_consumption(self):
+        session = Session(id="s2", pipeline=_make_pipeline(), context=Context())
+        waiter = session.start_wait_input("later")
+        result = await session.finish_wait_input("later", waiter, timeout=0.01)
+
+        self.assertIsNone(result)
+        self.assertIn("input_timeout", [e.type for e in session.event_history])
+
+        await session.input("later", {"value": 2})
+        waiter2 = session.start_wait_input("later")
+        result2 = await session.finish_wait_input("later", waiter2, timeout=1)
+        self.assertEqual(result2["payload"]["value"], 2)
+
+
+if __name__ == "__main__":
+    unittest.main()