4idZddlmZddlZddlZddlmZmZddlmZddl m Z m Z e hdZ dZd ZGd d Zy) uSwarmSagaWriter — Saga Lifecycle Recorder for the Genesis Cold Ledger. Manages the full lifecycle of multi-agent swarm executions: 1. open_saga() — create a RUNNING saga record 2. record_proposed_delta() — append one agent's delta (SQL-level JSONB append) 3. close_saga() — finalise with resolved_state and terminal status 4. get_saga() — delegate to ColdLedger.get_saga() Design decisions: - record_proposed_delta uses a single UPDATE … SET proposed_deltas = proposed_deltas || ? so concurrent appends from multiple agents are safe without a read-modify-write race. - SwarmSaga is created via ColdLedger.write_saga() so the pool/commit/release pattern is centralised in one place. - All status validation happens here, not in the ledger — writer owns the lifecycle contract. Rules enforced (Genesis hardwired): - NO SQLite - All SQL uses parameterised %s placeholders — never f-strings - Connection pool getconn/putconn in try/finally (no leaks) — delegated to ledger ) annotationsN)datetimetimezone)Optional) ColdLedger SwarmSaga>FAILED COMPLETED PARTIAL_FAILz^UPDATE swarm_sagas SET proposed_deltas = proposed_deltas || %s::jsonb WHERE saga_id = %s::uuidzTUPDATE swarm_sagas SET resolved_state = %s, status = %s WHERE saga_id = %s::uuidcXeZdZdZddZd dZ d dZ d dZd dZy) SwarmSagaWriterzLifecycle manager for swarm sagas stored in the Genesis Cold Ledger. Args: cold_ledger: A fully initialised :class:`ColdLedger` instance. c||_y)N)ledger)self cold_ledgers 1/mnt/e/genesis-system/core/storage/saga_writer.py__init__zSwarmSagaWriter.__init__:s ! c ttj}tjt j }t|||gdd|}|jj||S)a)Create a new saga record with status RUNNING and empty proposed_deltas. Args: session_id: UUID string of the owning session. orchestrator_dag: DAG descriptor produced by the orchestrator. Returns: The new saga_id (UUID4 string). tzNRUNNING)saga_id session_idorchestrator_dagproposed_deltasresolved_statestatus created_at) struuiduuid4rnowrutcrr write_saga)rrrrr#sagas r open_sagazSwarmSagaWriter.open_sagaAs^djjl#llhll+!-  t$rctjtjj }t j |||dg}|jj} |j5}|jt||fddd|j|jj|y#1swY5xYw#|jj|wxYw)aDAppend one agent's proposed delta to the saga's proposed_deltas JSONB array. Uses a single SQL UPDATE with JSONB concatenation (``||``) so concurrent calls from multiple agents are safe without a client-side read-modify-write. Each stored element is: ``{"agent_id": , "delta": , "submitted_at": }`` Args: saga_id: UUID string of the target saga. agent_id: Identifier of the submitting agent. delta: Arbitrary JSON-serialisable dict describing the proposed change. r)agent_iddelta submitted_atN)rr#rr$ isoformatjsondumpsr_acquirecursorexecute_APPEND_DELTA_SQLcommit_release)rrr)r*r+elementconncurs rrecord_proposed_deltaz%SwarmSagaWriter.record_proposed_deltaZs  ||x||4>>@ **"UL Q R {{##% ' C# -/AB C KKM KK  &  C C KK  &s$(C8CCC CC.c|tvrtd|dtttj|}|j j } |j5}|jt|||fddd|j|j j|y#1swY5xYw#|j j|wxYw)aFinalise a saga with its resolved state and terminal status. Args: saga_id: UUID string of the saga to close. resolved_state: Final merged state dict (JSON-serialisable). status: One of ``"COMPLETED"``, ``"PARTIAL_FAIL"``, ``"FAILED"``. Raises: ValueError: If *status* is not one of the three valid values. zInvalid saga status z. Must be one of: N) _VALID_CLOSE_STATUSES ValueErrorsortedr-r.rr/r0r1_CLOSE_SAGA_SQLr3r4)rrrr resolved_jsonr6r7s r close_sagazSwarmSagaWriter.close_sagaws . .&vj1##)*?#@"AC   >2 {{##% ' O# OmVW-MN O KKM KK  &  O O KK  &s$C'B5C5B>:CCc8|jj|S)zRetrieve a saga by ID. Delegates to ColdLedger.get_saga(). Args: saga_id: UUID string of the saga to retrieve. Returns: A :class:`SwarmSaga` instance, or ``None`` if not found. )rget_saga)rrs rrAzSwarmSagaWriter.get_sagas{{##G,,rN)rrreturnNone)rr rdictrBr )rr r)r r*rDrBrC)rr rrDrr rBrC)rr rBzOptional[SwarmSaga]) __name__ __module__ __qualname____doc__rr'r8r?rArrr r 3s` "2''&)'26' ':'',0':=' ': -rr )rH __future__rr-r!rrtypingrcore.storage.cold_ledgerrr frozensetr:r2r=r rIrrrNsL*# ':""IJ   j-j-r