flo-python 0.1.0.dev4__py3-none-any.whl → 0.1.2__py3-none-any.whl

flo/__init__.py

@@ -34,6 +34,7 @@ Example:
 from .client import FloClient
 from .exceptions import (
     BadRequestError,
+    BlockTooLongError,
     ConflictError,
     ConnectionFailedError,
     FloError,
@@ -52,15 +53,19 @@ from .exceptions import (
     PayloadTooLargeError,
     ProtocolError,
     RateLimitedError,
+    RequestTimeoutError,
     ServerError,
+    TableMismatchError,
     UnauthorizedError,
+    UnavailableError,
     UnexpectedEofError,
-    UnsupportedVersionError,
     ValidationError,
     ValueTooLargeError,
     is_connection_error,
 )
+from .kv_txn import Transaction, TxnFinishedError, TxnUnsupportedOpError
 from .processing import ProcessingOperations
+from .retry import DEFAULT_RETRY_POLICY, RetryPolicy
 from .types import (
     AckOptions,
     ActionDeleteOptions,
@@ -73,26 +78,35 @@ from .types import (
     ActionRunStatus,
     ActionStatusOptions,
     ActionType,
+    Condition,
     DeleteOptions,
     DequeueOptions,
     DequeueResult,
     DlqListOptions,
     DlqRequeueOptions,
     EnqueueOptions,
+    GetJsonResult,
     GetOptions,
+    GetResult,
     HistoryOptions,
-    KVEntry,
+    IncrResult,
+    KVBeginResult,
+    KVCommitResult,
+    KVExistsOptions,
+    KVExpireOptions,
+    KVIncrOptions,
+    KVJsonOptions,
+    KVMGetOptions,
     Message,
+    MGetEntry,
     NackOptions,
     OpCode,
     OptionTag,
     PeekOptions,
+    PendingEntry,
     ProcessingCancelOptions,
     ProcessingListEntry,
     ProcessingListOptions,
-    ProcessingRescaleOptions,
-    ProcessingRestoreOptions,
-    ProcessingSavepointOptions,
     ProcessingStatusOptions,
     ProcessingStatusResult,
     ProcessingStopOptions,
@@ -100,12 +114,17 @@ from .types import (
     ProcessingSyncOptions,
     ProcessingSyncResult,
     PutOptions,
+    PutResult,
+    Ran,
+    Reason,
+    RunStatus,
     ScanOptions,
     ScanResult,
     StatusCode,
     StorageTier,
     StreamAppendOptions,
     StreamAppendResult,
+    StreamClaimResult,
     StreamGroupAckOptions,
     StreamGroupJoinOptions,
     StreamGroupNackOptions,
@@ -117,9 +136,10 @@ from .types import (
     StreamReadResult,
     StreamRecord,
     StreamTrimOptions,
+    StreamTrimResult,
     TaskAssignment,
-    TouchOptions,
     VersionEntry,
+    WaitFor,
     WorkerAwaitOptions,
     WorkerAwaitResult,
     WorkerCompleteOptions,
@@ -128,8 +148,9 @@ from .types import (
     WorkerListOptions,
     WorkerListResult,
     WorkerRegisterOptions,
-    WorkerTask,
+    WorkerStatus,
     WorkerTouchOptions,
+    WorkerType,
     WorkflowCancelOptions,
     WorkflowCreateOptions,
     WorkflowDisableOptions,
@@ -156,11 +177,28 @@ from .worker import (
 )
 from .workflows import WorkflowOperations
 
-__version__ = "0.1.0"
+__version__ = "0.1.2"
 
 __all__ = [
+    "Transaction",
+    "TxnFinishedError",
+    "TxnUnsupportedOpError",
+    "KVBeginResult",
+    "KVCommitResult",
+    "KVExistsOptions",
+    "KVIncrOptions",
+    "KVJsonOptions",
+    "KVMGetOptions",
+    "KVExpireOptions",
+    "GetResult",
+    "GetJsonResult",
+    "MGetEntry",
+    "PutResult",
+    "IncrResult",
     # Client
     "FloClient",
+    "RetryPolicy",
+    "DEFAULT_RETRY_POLICY",
     # High-level Worker API
     "ActionWorker",
     "ActionWorkerOptions",
@@ -179,9 +217,10 @@ __all__ = [
     "ConnectionFailedError",
     "InvalidEndpointError",
     "UnexpectedEofError",
+    "RequestTimeoutError",
     "ProtocolError",
     "InvalidMagicError",
-    "UnsupportedVersionError",
+    "TableMismatchError",
     "InvalidChecksumError",
     "PayloadTooLargeError",
     "IncompleteResponseError",
@@ -189,28 +228,33 @@ __all__ = [
     "NamespaceTooLargeError",
     "KeyTooLargeError",
     "ValueTooLargeError",
+    "BlockTooLongError",
     "ServerError",
     "NotFoundError",
     "BadRequestError",
     "ConflictError",
     "UnauthorizedError",
     "OverloadedError",
+    "UnavailableError",
     "RateLimitedError",
     "InternalServerError",
     "GenericServerError",
     "is_connection_error",
     # Types
     "OpCode",
+    "Ran",
+    "Reason",
     "StatusCode",
     "OptionTag",
     # Result types
-    "KVEntry",
     "ScanResult",
     "VersionEntry",
     "Message",
     "DequeueResult",
     # Options
+    "Condition",
     "GetOptions",
+    "WaitFor",
     "PutOptions",
     "DeleteOptions",
     "ScanOptions",
@@ -222,18 +266,20 @@ __all__ = [
     "DlqListOptions",
     "DlqRequeueOptions",
     "PeekOptions",
-    "TouchOptions",
     # Stream types
     "StreamID",
     "StorageTier",
     "StreamRecord",
     "StreamAppendResult",
     "StreamReadResult",
+    "StreamClaimResult",
+    "PendingEntry",
     "StreamInfo",
     # Stream options
     "StreamAppendOptions",
     "StreamReadOptions",
     "StreamTrimOptions",
+    "StreamTrimResult",
     "StreamInfoOptions",
     "StreamGroupJoinOptions",
     "StreamGroupReadOptions",
@@ -241,6 +287,7 @@ __all__ = [
     "StreamGroupNackOptions",
     # Action types
     "ActionType",
+    "RunStatus",
     "ActionInfo",
     "ActionRunStatus",
     "ActionInvokeResult",
@@ -253,10 +300,11 @@ __all__ = [
     "ActionDeleteOptions",
     # Worker types
     "TaskAssignment",
-    "WorkerTask",
     "WorkerAwaitResult",
     "WorkerInfo",
     "WorkerListResult",
+    "WorkerStatus",
+    "WorkerType",
     # Worker options
     "WorkerRegisterOptions",
     "WorkerAwaitOptions",
@@ -286,9 +334,6 @@ __all__ = [
     "ProcessingListOptions",
     "ProcessingStopOptions",
     "ProcessingCancelOptions",
-    "ProcessingSavepointOptions",
-    "ProcessingRestoreOptions",
-    "ProcessingRescaleOptions",
     "ProcessingSyncOptions",
     "ProcessingStatusResult",
     "ProcessingListEntry",

flo/actions.py

@@ -9,6 +9,12 @@ For a higher-level API, see `Worker` in worker.py.
 
 from typing import TYPE_CHECKING
 
+from .kv_body import (
+    decode_action_invoke_answer,
+    decode_action_status_answer,
+    decode_worker_list_answer,
+    encode_worker_list_body,
+)
 from .types import (
     ActionDeleteOptions,
     ActionInvokeOptions,
@@ -39,7 +45,6 @@ from .wire import (
     serialize_worker_await_value,
     serialize_worker_complete_value,
     serialize_worker_fail_value,
-    serialize_worker_list_value,
     serialize_worker_register_value,
     serialize_worker_touch_value,
 )
@@ -78,7 +83,7 @@ class ActionOperations:
             action_type=action_type,
             timeout_ms=opts.timeout_ms,
             max_retries=opts.max_retries,
-            description=opts.description,
+            owner=opts.owner,
         )
 
         await self._client._send_and_check(
@@ -109,8 +114,7 @@ class ActionOperations:
 
         value = serialize_action_invoke_value(
             input_data=input_data,
-            priority=opts.priority,
-            idempotency_key=opts.idempotency_key,
+            labels=opts.labels,
         )
 
         response = await self._client._send_and_check(
@@ -120,9 +124,7 @@ class ActionOperations:
             value,
         )
 
-        # Response contains run_id as string
-        run_id = response.data.decode("utf-8") if response.data else ""
-        return ActionInvokeResult(run_id=run_id)
+        return ActionInvokeResult(run_id=decode_action_invoke_answer(response.data))
 
     async def status(
         self,
@@ -148,13 +150,7 @@ class ActionOperations:
             b"",
         )
 
-        # Parse status response - format TBD based on server implementation
-        # For now return basic status
-        return ActionRunStatus(
-            run_id=run_id,
-            status="unknown",
-            result=response.data if response.data else None,
-        )
+        return decode_action_status_answer(run_id, response.data)
 
     async def list(
         self,
@@ -280,7 +276,7 @@ class WorkerOperations:
         Args:
             worker_id: Worker identifier.
             task_types: Task types to listen for.
-            options: Optional await options (block_ms, timeout_ms).
+            options: Optional await options (block_ms).
 
         Returns:
             WorkerAwaitResult with task if available.
@@ -290,12 +286,9 @@ class WorkerOperations:
 
         value = serialize_worker_await_value(task_types)
 
-        # Build options for block_ms and timeout_ms
         options_builder = OptionsBuilder()
         if opts.block_ms is not None:
             options_builder.add_u32(OptionTag.BLOCK_MS, opts.block_ms)
-        if opts.timeout_ms is not None:
-            options_builder.add_u32(OptionTag.TIMEOUT_MS, opts.timeout_ms)
 
         response = await self._client._send_and_check(
             OpCode.ACTION_AWAIT,
@@ -402,25 +395,25 @@ class WorkerOperations:
         self,
         options: WorkerListOptions | None = None,
     ) -> WorkerListResult:
-        """List registered workers.
+        """List a page of the namespace's workers, in id order.
 
         Args:
-            options: Optional list options.
+            options: The page's limit and cursor (the last page's ``cursor``;
+                None starts).
 
         Returns:
-            WorkerListResult with list of workers.
+            WorkerListResult with the page's workers and the next page's
+            cursor, None on the last page.
         """
         opts = options or WorkerListOptions()
         namespace = self._client.get_namespace(opts.namespace)
 
-        value = serialize_worker_list_value(opts.limit)
-
-        await self._client._send_and_check(
+        response = await self._client._send_and_check(
             OpCode.WORKER_LIST,
             namespace,
             b"",
-            value,
+            encode_worker_list_body(opts.limit, opts.cursor),
         )
 
-        # Parse list response - format TBD based on server implementation
-        return WorkerListResult(workers=[])
+        workers, cursor = decode_worker_list_answer(response.data)
+        return WorkerListResult(workers=workers, cursor=cursor)