HubBackedCheckpointStore

Persists task checkpoints through a :class:Hub or :class:HubClient.

Satisfies the framework-core :class:CheckpointStore Protocol by delegating to checkpoint_task / read_task_checkpoint on the supplied backend. Pass a :class:Hub for in-process durability or a :class:HubClient to route checkpoints to a remote hub over the wire. Deployments that need different durability supply another :class:CheckpointStore — the Protocol is the seam.

41
42
43

def __init__(self, hub: "Hub | HubClient") -> None:
    # __init__ stores params; no side effects.
    self._hub = hub

write(task_id, state)

45
46

async def write(self, task_id: str, state: dict[str, Any]) -> None:
    await self._hub.checkpoint_task(task_id, dict(state))

read(task_id)

48
49

async def read(self, task_id: str) -> dict[str, Any] | None:
    return await self._hub.read_task_checkpoint(task_id)