flo-python 0.1.0.dev3__py3-none-any.whl → 0.1.0.dev4__py3-none-any.whl

flo/client.py

@@ -372,11 +372,13 @@ class FloClient:
         Returns:
             A new ActionWorker instance ready to register actions and start.
 
-        Example:
+        Example::
+
             async with FloClient("localhost:3000", namespace="myapp") as client:
                 worker = client.new_action_worker(concurrency=5)
                 worker.register_action("process-order", process_order)
-                await worker.start()
+                async with worker:
+                    await worker.start()
         """
         from .worker import ActionWorker
 

flo/worker.py

@@ -3,7 +3,8 @@
 Provides ActionWorker for executing actions and StreamWorker for
 processing stream records via consumer groups.
 
-Example:
+Example::
+
     from flo import FloClient, ActionContext
 
     async def process_order(ctx: ActionContext) -> bytes:
@@ -14,8 +15,9 @@ Example:
     async def main():
         async with FloClient("localhost:3000", namespace="myapp") as client:
             worker = client.new_action_worker(concurrency=5)
-            worker.action("process-order")(process_order)
-            await worker.start()
+            worker.register_action("process-order", process_order)
+            async with worker:
+                await worker.start()
 """
 
 import asyncio
@@ -626,6 +628,19 @@ class ActionWorker:
         if self._client:
             await self._client.close()
 
+    async def __aenter__(self) -> "ActionWorker":
+        """Async context manager entry."""
+        return self
+
+    async def __aexit__(
+        self,
+        exc_type: type[BaseException] | None,
+        exc_val: BaseException | None,
+        exc_tb: object,
+    ) -> None:
+        """Async context manager exit — closes the worker."""
+        await self.close()
+
 
 # =============================================================================
 # Stream Worker
@@ -977,3 +992,16 @@ class StreamWorker:
         self.stop()
         if self._client:
             await self._client.close()
+
+    async def __aenter__(self) -> "StreamWorker":
+        """Async context manager entry."""
+        return self
+
+    async def __aexit__(
+        self,
+        exc_type: type[BaseException] | None,
+        exc_val: BaseException | None,
+        exc_tb: object,
+    ) -> None:
+        """Async context manager exit — closes the stream worker."""
+        await self.close()

{flo_python-0.1.0.dev3.dist-info → flo_python-0.1.0.dev4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flo-python
-Version: 0.1.0.dev3
+Version: 0.1.0.dev4
 Summary: Python SDK for the Flo distributed systems platform
 Project-URL: Homepage, https://github.com/floruntime/flo-python
 Project-URL: Documentation, https://github.com/floruntime/flo-python#readme

{flo_python-0.1.0.dev3.dist-info → flo_python-0.1.0.dev4.dist-info}/RECORD

@@ -1,6 +1,6 @@
 flo/__init__.py,sha256=7nQrfC6ge__E4scjMx5x-VwAiAKhsiRL-JEjC7cf0ZA,7039
 flo/actions.py,sha256=b4cd5AAe6TRAU-_UZmenbek0h--T-C_PiiDFCHyYDTs,12258
-flo/client.py,sha256=a6RpbUZD1FdInmuoZJVdwocC-vkG5efcwqevIX2_NEU,15018
+flo/client.py,sha256=hS64RG4WA9qK_ISdIqDxmV-7DDTn3pAYvYGFRxOjq7U,15059
 flo/exceptions.py,sha256=-3HUWvRGLe_4mSWxwZXk8CMpcfrRjFv0rRkTzu4kqP4,5868
 flo/kv.py,sha256=2PgvoZylU8Clsx8yaYdTLzdCZmD8nCZgtrmJHhAAZlM,7344
 flo/processing.py,sha256=_jOQZiSgyArOPOdnoARNy9x6QgzcYvt1R0wa0vLQfoc,10530
@@ -8,9 +8,9 @@ flo/queue.py,sha256=Nn6CkgUkbXgUuH9BOCWlRkjxTZLKRs0_iMyRmwkYK6k,10888
 flo/streams.py,sha256=G_27-_cq6hru712olwImd4Yxuo8x4l8XuKljUk0Y1JA,12965
 flo/types.py,sha256=g8yikYiiYxsjkUlSCZxKpLMdBv7jg-cWAGO-cnEnaEo,27808
 flo/wire.py,sha256=TLCMXeA8VnUTzyyORM7G3zrsSZhyZymSGxU9jhCxmmI,29373
-flo/worker.py,sha256=AUQX5Hd6-pqm5ouCoJLUiGhRbfE6hzrACvaY0OyWNhw,34604
+flo/worker.py,sha256=flSLz7FIV_zmvLhyHjR39KqUXk5cr8MQVHFkswHv78o,35395
 flo/workflows.py,sha256=8Z5mhq2UOAnjV_Ef261mcWlj4Z5yGCwrFbxjey0ICoU,15050
-flo_python-0.1.0.dev3.dist-info/METADATA,sha256=LjTltaQ-avls3GgUlmp-ysA43IrVpx6uIwZZoVZ907E,14638
-flo_python-0.1.0.dev3.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
-flo_python-0.1.0.dev3.dist-info/licenses/LICENSE,sha256=mADRSuVhPxFNpZiy7zO0gBjGrNO24OtWO9VCX6wmCVU,1065
-flo_python-0.1.0.dev3.dist-info/RECORD,,
+flo_python-0.1.0.dev4.dist-info/METADATA,sha256=oufJNR5tLKj7oUBiAoMsXx6eYzD7avnehWUAmC5o3ZA,14638
+flo_python-0.1.0.dev4.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
+flo_python-0.1.0.dev4.dist-info/licenses/LICENSE,sha256=mADRSuVhPxFNpZiy7zO0gBjGrNO24OtWO9VCX6wmCVU,1065
+flo_python-0.1.0.dev4.dist-info/RECORD,,