flo-python 0.1.0.dev3__tar.gz → 0.1.1__tar.gz

{flo_python-0.1.0.dev3 → flo_python-0.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.4
+Metadata-Version: 2.5
 Name: flo-python
-Version: 0.1.0.dev3
+Version: 0.1.1
 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
@@ -35,7 +35,7 @@ Python SDK for the [Flo](https://github.com/floruntime) distributed systems plat
 ## Installation
 
 ```bash
-pip install flo
+pip install flo-python
 ```
 
 ## Quick Start

{flo_python-0.1.0.dev3 → flo_python-0.1.1}/README.md

@@ -5,7 +5,7 @@ Python SDK for the [Flo](https://github.com/floruntime) distributed systems plat
 ## Installation
 
 ```bash
-pip install flo
+pip install flo-python
 ```
 
 ## Quick Start

{flo_python-0.1.0.dev3 → flo_python-0.1.1}/examples/worker.py

@@ -158,56 +158,39 @@ async def generate_report(ctx: ActionContext) -> bytes:
 
 
 async def main():
-    # Create and connect client (shared configuration)
-    client = FloClient(
-        os.getenv("FLO_ENDPOINT", "localhost:3000"),
+    async with FloClient(
+        os.getenv("FLO_ENDPOINT", "localhost:4455"),
         namespace=os.getenv("FLO_NAMESPACE", "myapp"),
         debug=os.getenv("FLO_DEBUG", "").lower() in ("1", "true"),
-    )
-    await client.connect()
-
-    # Create a worker from the client
-    worker = client.new_action_worker(
-        concurrency=5,
-        action_timeout=300,  # 5 minutes
-    )
+    ) as client:
+        worker = client.new_action_worker(
+            concurrency=5,
+            action_timeout=300,  # 5 minutes
+        )
 
-    # Register action handlers using register_action()
-    worker.register_action("process-order", process_order)
-    worker.register_action("send-notification", send_notification)
-    worker.register_action("generate-report", generate_report)
-
-    # Register using decorator syntax
-    @worker.action("health-check")
-    async def health_check(ctx: ActionContext) -> bytes:
-        """Simple health check action."""
-        return ctx.to_bytes({
-            "status": "healthy",
-            "worker_id": ctx.task_id,
-            "timestamp": ctx.created_at,
-        })
-
-    # Handle shutdown signals
-    stop_event = asyncio.Event()
-
-    def signal_handler():
-        logger.info("Received shutdown signal")
-        worker.stop()
-        stop_event.set()
-
-    loop = asyncio.get_running_loop()
-    for sig in (signal.SIGINT, signal.SIGTERM):
-        loop.add_signal_handler(sig, signal_handler)
-
-    # Start worker (blocks until stopped)
-    logger.info("Starting worker...")
-    try:
-        await worker.start()
-    except KeyboardInterrupt:
-        logger.info("Interrupted")
-    finally:
-        await worker.close()
-        await client.close()
+        # Register action handlers
+        worker.register_action("process-order", process_order)
+        worker.register_action("send-notification", send_notification)
+        worker.register_action("generate-report", generate_report)
+
+        # Register using decorator syntax
+        @worker.action("health-check")
+        async def health_check(ctx: ActionContext) -> bytes:
+            return ctx.to_bytes({"status": "healthy", "worker_id": ctx.task_id})
+
+        # Handle shutdown signals
+        def signal_handler():
+            logger.info("Received shutdown signal")
+            worker.stop()
+
+        loop = asyncio.get_running_loop()
+        for sig in (signal.SIGINT, signal.SIGTERM):
+            loop.add_signal_handler(sig, signal_handler)
+
+        # Start worker (blocks until stopped)
+        logger.info("Starting worker...")
+        async with worker:
+            await worker.start()
         logger.info("Worker shutdown complete")
 
 

{flo_python-0.1.0.dev3 → flo_python-0.1.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "flo-python"
-version = "0.1.0.dev3"
+version = "0.1.1"
 description = "Python SDK for the Flo distributed systems platform"
 readme = "README.md"
 license = "MIT"

{flo_python-0.1.0.dev3 → flo_python-0.1.1}/src/flo/__init__.py

@@ -60,6 +60,7 @@ from .exceptions import (
     ValueTooLargeError,
     is_connection_error,
 )
+from .kv_txn import Transaction, TxnFinishedError, TxnUnsupportedOpError
 from .processing import ProcessingOperations
 from .types import (
     AckOptions,
@@ -79,14 +80,25 @@ from .types import (
     DlqListOptions,
     DlqRequeueOptions,
     EnqueueOptions,
+    GetJsonResult,
     GetOptions,
+    GetResult,
     HistoryOptions,
+    KVBeginResult,
+    KVCommitResult,
     KVEntry,
+    KVExistsOptions,
+    KVIncrOptions,
+    KVJsonOptions,
+    KVMGetOptions,
+    KVTouchOptions,
     Message,
+    MGetEntry,
     NackOptions,
     OpCode,
     OptionTag,
     PeekOptions,
+    PendingEntry,
     ProcessingCancelOptions,
     ProcessingListEntry,
     ProcessingListOptions,
@@ -100,12 +112,14 @@ from .types import (
     ProcessingSyncOptions,
     ProcessingSyncResult,
     PutOptions,
+    PutResult,
     ScanOptions,
     ScanResult,
     StatusCode,
     StorageTier,
     StreamAppendOptions,
     StreamAppendResult,
+    StreamClaimResult,
     StreamGroupAckOptions,
     StreamGroupJoinOptions,
     StreamGroupNackOptions,
@@ -156,9 +170,23 @@ from .worker import (
 )
 from .workflows import WorkflowOperations
 
-__version__ = "0.1.0"
+__version__ = "0.1.1"
 
 __all__ = [
+    "Transaction",
+    "TxnFinishedError",
+    "TxnUnsupportedOpError",
+    "KVBeginResult",
+    "KVCommitResult",
+    "KVExistsOptions",
+    "KVIncrOptions",
+    "KVJsonOptions",
+    "KVMGetOptions",
+    "KVTouchOptions",
+    "GetResult",
+    "GetJsonResult",
+    "MGetEntry",
+    "PutResult",
     # Client
     "FloClient",
     # High-level Worker API
@@ -229,6 +257,8 @@ __all__ = [
     "StreamRecord",
     "StreamAppendResult",
     "StreamReadResult",
+    "StreamClaimResult",
+    "PendingEntry",
     "StreamInfo",
     # Stream options
     "StreamAppendOptions",

{flo_python-0.1.0.dev3 → flo_python-0.1.1}/src/flo/client.py

@@ -83,7 +83,27 @@ class FloClient:
 
     @staticmethod
     def _parse_endpoint(endpoint: str) -> tuple[str, int]:
-        """Parse endpoint string into host and port."""
+        """Parse endpoint string into host and port.
+
+        Accepts an optional URI scheme: ``flo://`` (plaintext) is stripped and
+        bare ``host:port`` remains valid. ``flos://`` (TLS) is reserved but not
+        yet implemented, and any other scheme is rejected so it fails here
+        rather than as a confusing DNS error at connect time.
+        """
+        if "://" in endpoint:
+            scheme, _, rest = endpoint.partition("://")
+            scheme = scheme.lower()
+            if scheme == "flo":
+                # Drop scheme and any trailing path/query (reserved for future use).
+                endpoint = rest.split("/", 1)[0]
+            elif scheme == "flos":
+                raise InvalidEndpointError(f"flos:// (TLS) is not yet supported: {endpoint}")
+            else:
+                raise InvalidEndpointError(
+                    f"Unsupported scheme {scheme}:// (expected flo:// or bare "
+                    f"host:port): {endpoint}"
+                )
+
         parts = endpoint.rsplit(":", 1)
         if len(parts) != 2:
             raise InvalidEndpointError(f"Invalid endpoint format: {endpoint} (expected host:port)")
@@ -372,11 +392,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
 
@@ -400,6 +422,8 @@ class FloClient:
         batch_size: int = 10,
         block_ms: int = 30000,
         message_timeout: float = 300.0,
+        redeliver_pending_on_reconnect: bool = True,
+        claim_min_idle_ms: int = 0,
     ) -> "StreamWorker":
         """Create a new StreamWorker from this client.
 
@@ -417,6 +441,14 @@ class FloClient:
             batch_size: Number of records to fetch per poll.
             block_ms: Timeout for blocking read in milliseconds.
             message_timeout: Timeout for record handlers in seconds.
+            redeliver_pending_on_reconnect: Drain this consumer's pending
+                (delivered-but-unacked) entries via group_claim after a
+                reconnect, before resuming normal reads. Defaults to True
+                (required for at-least-once across reconnects). Set False for
+                at-most-once.
+            claim_min_idle_ms: Minimum idle time (ms) before the reconnect drain
+                claims an entry. 0 drains everything this consumer owns; > 0
+                also steals entries abandoned by dead consumers.
 
         Returns:
             A new StreamWorker instance ready to start.
@@ -444,4 +476,6 @@ class FloClient:
             batch_size=batch_size,
             block_ms=block_ms,
             message_timeout=message_timeout,
+            redeliver_pending_on_reconnect=redeliver_pending_on_reconnect,
+            claim_min_idle_ms=claim_min_idle_ms,
         )