@sema-agent/server 1.321.0 → 1.322.0

package/dist/plugins/pg-pool.js

@@ -156,6 +156,13 @@ export const PG_SCHEMA_STATEMENTS = [
      result     TEXT         NOT NULL,
      created_at BIGINT       NOT NULL,
      PRIMARY KEY (run_id, ordinal)
+   )`,
+    `CREATE TABLE IF NOT EXISTS workflow_resume_claim (
+     source_run_id VARCHAR(191) NOT NULL,
+     scope         VARCHAR(190) NOT NULL,
+     new_run_id    VARCHAR(191) NOT NULL,
+     claimed_at    BIGINT       NOT NULL,
+     PRIMARY KEY (source_run_id, scope)
    )`,
     `CREATE TABLE IF NOT EXISTS workflow_run (
      id         VARCHAR(191) NOT NULL,

package/dist/plugins/tidb-pool.js

@@ -312,6 +312,13 @@ export const SCHEMA_STATEMENTS = [
      result     MEDIUMTEXT   NOT NULL,
      created_at BIGINT       NOT NULL,
      PRIMARY KEY (run_id, ordinal)
+   )`,
+    `CREATE TABLE IF NOT EXISTS workflow_resume_claim (
+     source_run_id VARCHAR(191) NOT NULL,
+     scope         VARCHAR(190) NOT NULL,
+     new_run_id    VARCHAR(191) NOT NULL,
+     claimed_at    BIGINT       NOT NULL,
+     PRIMARY KEY (source_run_id, scope)
    )`,
     `CREATE TABLE IF NOT EXISTS workflow_run (
      id         VARCHAR(191) NOT NULL,

package/dist/plugins/workflow-journal-store-sql.d.ts

@@ -2,9 +2,13 @@ import type { Pool as MySqlPool } from "mysql2/promise";
 import type { Pool as PgPool } from "pg";
 import { type WorkflowJournalEntry, type WorkflowJournalStore } from "@sema-agent/core";
 import { type SqlDriver } from "./sql-driver.js";
+export interface SqlWorkflowJournalStoreOptions {
+    resumeClaimTtlMs?: number;
+}
 export declare class SqlWorkflowJournalStore implements WorkflowJournalStore {
     private readonly db;
-    constructor(db: SqlDriver);
+    private readonly resumeClaimTtlMs;
+    constructor(db: SqlDriver, opts?: SqlWorkflowJournalStoreOptions);
     private q;
     append(runId: string, scope: string, entry: WorkflowJournalEntry): Promise<void>;
     load(runId: string, scope: string): Promise<WorkflowJournalEntry[]>;
@@ -19,12 +23,25 @@ export declare class SqlWorkflowJournalStore implements WorkflowJournalStore {
         resultBytes: number;
     }>>;
     deleteByRun(runId: string): Promise<number>;
+    resumeClaim(input: {
+        sourceRunId: string;
+        newRunId: string;
+        scope: string;
+    }): Promise<{
+        granted: boolean;
+        holder?: string;
+    }>;
+    releaseResumeClaim(input: {
+        sourceRunId: string;
+        newRunId: string;
+        scope: string;
+    }): Promise<void>;
     reapExpired(now: number, maxAgeMs: number): Promise<number>;
 }
 export declare class TiDBWorkflowJournalStore extends SqlWorkflowJournalStore {
-    constructor(pool: MySqlPool);
+    constructor(pool: MySqlPool, opts?: SqlWorkflowJournalStoreOptions);
 }
 export declare class PgWorkflowJournalStore extends SqlWorkflowJournalStore {
-    constructor(pool: PgPool);
+    constructor(pool: PgPool, opts?: SqlWorkflowJournalStoreOptions);
 }
 //# sourceMappingURL=workflow-journal-store-sql.d.ts.map

package/dist/plugins/workflow-journal-store-sql.js

@@ -1,10 +1,13 @@
 import { callKeyOrdinal } from "@sema-agent/core";
 import { oversizeJournalResult } from "./workflow-journal-limits.js";
 import { mysqlDriver, pgDriver } from "./sql-driver.js";
+const DEFAULT_RESUME_CLAIM_TTL_MS = 15 * 60 * 1000;
 export class SqlWorkflowJournalStore {
     db;
-    constructor(db) {
+    resumeClaimTtlMs;
+    constructor(db, opts) {
         this.db = db;
+        this.resumeClaimTtlMs = opts?.resumeClaimTtlMs ?? DEFAULT_RESUME_CLAIM_TTL_MS;
     }
     q(tidb, pg) {
         return this.db.dialect === "tidb" ? tidb : pg;
@@ -35,19 +38,40 @@ export class SqlWorkflowJournalStore {
         const res = await this.db.query(this.q("DELETE FROM workflow_journal WHERE run_id = ?", "DELETE FROM workflow_journal WHERE run_id = $1"), [runId]);
         return res.affected;
     }
+    async resumeClaim(input) {
+        const now = Date.now();
+        const ins = await this.db.query(this.q("INSERT IGNORE INTO workflow_resume_claim (source_run_id, scope, new_run_id, claimed_at) VALUES (?, ?, ?, ?)", "INSERT INTO workflow_resume_claim (source_run_id, scope, new_run_id, claimed_at) VALUES ($1, $2, $3, $4) ON CONFLICT (source_run_id, scope) DO NOTHING"), [input.sourceRunId, input.scope, input.newRunId, now]);
+        if (ins.affected === 1)
+            return { granted: true };
+        const refresh = await this.db.query(this.q("UPDATE workflow_resume_claim SET claimed_at = ? WHERE source_run_id = ? AND scope = ? AND new_run_id = ?", "UPDATE workflow_resume_claim SET claimed_at = $1 WHERE source_run_id = $2 AND scope = $3 AND new_run_id = $4"), [now, input.sourceRunId, input.scope, input.newRunId]);
+        if (refresh.affected >= 1)
+            return { granted: true };
+        const takeover = await this.db.query(this.q("UPDATE workflow_resume_claim SET new_run_id = ?, claimed_at = ? WHERE source_run_id = ? AND scope = ? AND claimed_at < ?", "UPDATE workflow_resume_claim SET new_run_id = $1, claimed_at = $2 WHERE source_run_id = $3 AND scope = $4 AND claimed_at < $5"), [input.newRunId, now, input.sourceRunId, input.scope, now - this.resumeClaimTtlMs]);
+        if (takeover.affected >= 1)
+            return { granted: true };
+        const holder = await this.db.query(this.q("SELECT new_run_id FROM workflow_resume_claim WHERE source_run_id = ? AND scope = ?", "SELECT new_run_id FROM workflow_resume_claim WHERE source_run_id = $1 AND scope = $2"), [input.sourceRunId, input.scope]);
+        const row = holder.rows[0];
+        const holderId = row?.new_run_id === undefined ? undefined : String(row.new_run_id);
+        if (holderId === input.newRunId)
+            return { granted: true };
+        return { granted: false, holder: holderId };
+    }
+    async releaseResumeClaim(input) {
+        await this.db.query(this.q("DELETE FROM workflow_resume_claim WHERE source_run_id = ? AND scope = ? AND new_run_id = ?", "DELETE FROM workflow_resume_claim WHERE source_run_id = $1 AND scope = $2 AND new_run_id = $3"), [input.sourceRunId, input.scope, input.newRunId]);
+    }
     async reapExpired(now, maxAgeMs) {
         const res = await this.db.query(this.q("DELETE FROM workflow_journal WHERE created_at < ?", "DELETE FROM workflow_journal WHERE created_at < $1"), [now - maxAgeMs]);
         return res.affected;
     }
 }
 export class TiDBWorkflowJournalStore extends SqlWorkflowJournalStore {
-    constructor(pool) {
-        super(mysqlDriver(pool));
+    constructor(pool, opts) {
+        super(mysqlDriver(pool), opts);
     }
 }
 export class PgWorkflowJournalStore extends SqlWorkflowJournalStore {
-    constructor(pool) {
-        super(pgDriver(pool));
+    constructor(pool, opts) {
+        super(pgDriver(pool), opts);
     }
 }
 //# sourceMappingURL=workflow-journal-store-sql.js.map

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@sema-agent/server",
-  "version": "1.321.0",
+  "version": "1.322.0",
   "description": "Sema Server — the server/API implementation layer for Sema, wiring core, registry, model providers, and cloud agent execution. Built on @sema-agent/core.",
   "type": "module",
   "license": "BUSL-1.1",