@sema-agent/server 1.314.0 → 1.316.0

package/dist/plugins/sql-driver.js

@@ -0,0 +1,59 @@
+function mysqlResult(r) {
+    if (Array.isArray(r))
+        return { rows: r, affected: 0 };
+    const h = r;
+    return { rows: [], affected: Number(h?.affectedRows ?? 0) };
+}
+async function mysqlQuery(q, sql, params) {
+    const [r] = (await q.query(sql, params ?? []));
+    return mysqlResult(r);
+}
+export function mysqlDriver(pool) {
+    return {
+        dialect: "tidb",
+        query: (sql, params) => mysqlQuery(pool, sql, params),
+        async connect() {
+            const c = await pool.getConnection();
+            return {
+                query: (sql, params) => mysqlQuery(c, sql, params),
+                begin: () => c.beginTransaction(),
+                beginPessimistic: async () => {
+                    await c.query("BEGIN PESSIMISTIC");
+                },
+                commit: () => c.commit(),
+                rollback: () => c.rollback(),
+                release: () => c.release(),
+            };
+        },
+    };
+}
+async function pgQuery(q, sql, params) {
+    const res = await q.query(sql, params ?? []);
+    return { rows: (res.rows ?? []), affected: res.rowCount ?? 0 };
+}
+export function pgDriver(pool) {
+    return {
+        dialect: "pg",
+        query: (sql, params) => pgQuery(pool, sql, params),
+        async connect() {
+            const c = await pool.connect();
+            return {
+                query: (sql, params) => pgQuery(c, sql, params),
+                begin: async () => {
+                    await c.query("BEGIN");
+                },
+                beginPessimistic: async () => {
+                    await c.query("BEGIN");
+                },
+                commit: async () => {
+                    await c.query("COMMIT");
+                },
+                rollback: async () => {
+                    await c.query("ROLLBACK");
+                },
+                release: () => c.release(),
+            };
+        },
+    };
+}
+//# sourceMappingURL=sql-driver.js.map

package/dist/plugins/tidb-approval-store.d.ts

@@ -1,40 +1,2 @@
-import type { Pool } from "mysql2/promise";
-export type ApprovalStatus = "pending" | "approved" | "denied" | "expired";
-export interface ApprovalRow {
-    id: string;
-    taskId: string | null;
-    sessionId: string | null;
-    owner: string | null;
-    scope: string | null;
-    toolName: string;
-    args: unknown;
-    status: ApprovalStatus;
-    reason: string | null;
-    decidedBy: string | null;
-    createdAt: string;
-    decidedAt: string | null;
-}
-export declare class TiDBApprovalStore {
-    private readonly pool;
-    constructor(pool: Pool);
-    createPending(id: string, scope: string | null, ctx: {
-        taskId?: string | null;
-        sessionId?: string | null;
-        owner?: string | null;
-        toolName: string;
-        args: unknown;
-    }): Promise<void>;
-    getStatus(id: string, scope: string | null): Promise<{
-        status: ApprovalStatus;
-        reason: string | null;
-        decidedBy: string | null;
-    } | undefined>;
-    decide(id: string, scope: string | null, decision: "approved" | "denied" | "expired", reason: string | null, decidedBy: string | null): Promise<boolean>;
-    listPending(scope: string | null, owner?: string): Promise<ApprovalRow[]>;
-    get(id: string, scope: string | null): Promise<ApprovalRow | undefined>;
-    getById(id: string): Promise<ApprovalRow | undefined>;
-    listPendingAll(): Promise<ApprovalRow[]>;
-    expireStale(olderThanMs: number): Promise<number>;
-}
-export declare function mapRow(r: Record<string, unknown>): ApprovalRow;
+export { TiDBApprovalStore, mapRow, type ApprovalRow, type ApprovalStatus, } from "./approval-store-sql.js";
 //# sourceMappingURL=tidb-approval-store.d.ts.map

package/dist/plugins/tidb-approval-store.js

@@ -1,72 +1,2 @@
-import { parseJsonStrict as parseJson, toIsoOrNull as iso } from "./sql-row-helpers.js";
-export class TiDBApprovalStore {
-    pool;
-    constructor(pool) {
-        this.pool = pool;
-    }
-    async createPending(id, scope, ctx) {
-        await this.pool.query("INSERT INTO approval (id, task_id, session_id, owner, scope, tool_name, args, status, created_at) " +
-            "VALUES (?,?,?,?,?,?,?,'pending',?)", [
-            id,
-            ctx.taskId ?? null,
-            ctx.sessionId ?? null,
-            ctx.owner ?? null,
-            scope,
-            ctx.toolName,
-            ctx.args == null ? null : JSON.stringify(ctx.args),
-            new Date(),
-        ]);
-    }
-    async getStatus(id, scope) {
-        const [rows] = await this.pool.query("SELECT status, reason, decided_by FROM approval WHERE id = ? AND scope <=> ?", [id, scope]);
-        const r = rows[0];
-        if (!r)
-            return undefined;
-        return { status: r.status, reason: r.reason ?? null, decidedBy: r.decided_by ?? null };
-    }
-    async decide(id, scope, decision, reason, decidedBy) {
-        const [res] = await this.pool.query("UPDATE approval SET status = ?, reason = ?, decided_by = ?, decided_at = ? WHERE id = ? AND scope <=> ? AND status = 'pending'", [decision, reason, decidedBy, new Date(), id, scope]);
-        return res.affectedRows === 1;
-    }
-    async listPending(scope, owner) {
-        const [rows] = owner
-            ? await this.pool.query("SELECT * FROM approval WHERE status = 'pending' AND scope <=> ? AND owner = ? ORDER BY created_at ASC", [scope, owner])
-            : await this.pool.query("SELECT * FROM approval WHERE status = 'pending' AND scope <=> ? ORDER BY created_at ASC", [scope]);
-        return rows.map(mapRow);
-    }
-    async get(id, scope) {
-        const [rows] = await this.pool.query("SELECT * FROM approval WHERE id = ? AND scope <=> ?", [id, scope]);
-        return rows[0] ? mapRow(rows[0]) : undefined;
-    }
-    async getById(id) {
-        const [rows] = await this.pool.query("SELECT * FROM approval WHERE id = ?", [id]);
-        return rows[0] ? mapRow(rows[0]) : undefined;
-    }
-    async listPendingAll() {
-        const [rows] = await this.pool.query("SELECT * FROM approval WHERE status = 'pending' ORDER BY created_at ASC");
-        return rows.map(mapRow);
-    }
-    async expireStale(olderThanMs) {
-        const cutoff = new Date(Date.now() - olderThanMs);
-        const [res] = await this.pool.query("UPDATE approval SET status = 'expired', reason = 'timed out before decision', decided_at = ? " +
-            "WHERE status = 'pending' AND created_at < ?", [new Date(), cutoff]);
-        return res.affectedRows;
-    }
-}
-export function mapRow(r) {
-    return {
-        id: String(r.id),
-        taskId: r.task_id ?? null,
-        sessionId: r.session_id ?? null,
-        owner: r.owner ?? null,
-        scope: r.scope ?? null,
-        toolName: String(r.tool_name),
-        args: parseJson(r.args),
-        status: r.status,
-        reason: r.reason ?? null,
-        decidedBy: r.decided_by ?? null,
-        createdAt: iso(r.created_at),
-        decidedAt: iso(r.decided_at),
-    };
-}
+export { TiDBApprovalStore, mapRow, } from "./approval-store-sql.js";
 //# sourceMappingURL=tidb-approval-store.js.map

package/dist/plugins/tidb-breaker-state.d.ts

@@ -1,18 +1,2 @@
-import type { Pool } from "mysql2/promise";
-import type { BreakerSnapshot, BreakerState } from "@sema-agent/core";
-export declare class TiDBBreakerState implements BreakerState {
-    private readonly pool;
-    private readonly onWriteFail?;
-    private cache;
-    private timer?;
-    private readonly pending;
-    constructor(pool: Pool, onWriteFail?: ((streak: number) => void) | undefined);
-    private writeFailStreak;
-    get(key: string): BreakerSnapshot | undefined;
-    set(key: string, snap: BreakerSnapshot): void;
-    private writeThrough;
-    refresh(): Promise<void>;
-    startRefresh(intervalMs?: number): this;
-    stop(): void;
-}
+export { TiDBBreakerState } from "./breaker-state-sql.js";
 //# sourceMappingURL=tidb-breaker-state.d.ts.map

package/dist/plugins/tidb-breaker-state.js

@@ -1,75 +1,2 @@
-export class TiDBBreakerState {
-    pool;
-    onWriteFail;
-    cache = new Map();
-    timer;
-    pending = new Set();
-    constructor(pool, onWriteFail) {
-        this.pool = pool;
-        this.onWriteFail = onWriteFail;
-    }
-    writeFailStreak = 0;
-    get(key) {
-        return this.cache.get(key);
-    }
-    set(key, snap) {
-        this.cache.set(key, snap);
-        this.pending.add(key);
-        void this.writeThrough(key, snap);
-    }
-    async writeThrough(key, snap) {
-        try {
-            await this.pool.query("INSERT INTO circuit_breaker (breaker_key, phase, failures, opened_at, updated_at) VALUES (?,?,?,?,?) " +
-                "ON DUPLICATE KEY UPDATE phase=VALUES(phase), failures=VALUES(failures), opened_at=VALUES(opened_at), updated_at=VALUES(updated_at)", [key, snap.phase, snap.failures, snap.openedAt != null ? new Date(snap.openedAt) : null, new Date()]);
-            this.pending.delete(key);
-            if (this.writeFailStreak > 0) {
-                this.writeFailStreak = 0;
-                try {
-                    this.onWriteFail?.(0);
-                }
-                catch { }
-            }
-        }
-        catch {
-            try {
-                this.onWriteFail?.(++this.writeFailStreak);
-            }
-            catch { }
-        }
-    }
-    async refresh() {
-        try {
-            const pendingAtStart = new Set(this.pending);
-            const guarded = (key) => this.pending.has(key) || pendingAtStart.has(key);
-            const [rows] = await this.pool.query("SELECT breaker_key, phase, failures, opened_at FROM circuit_breaker");
-            const seen = new Set();
-            for (const r of rows) {
-                const key = String(r.breaker_key);
-                seen.add(key);
-                if (guarded(key))
-                    continue;
-                this.cache.set(key, {
-                    phase: r.phase,
-                    failures: Number(r.failures),
-                    openedAt: r.opened_at != null ? new Date(r.opened_at).getTime() : undefined,
-                });
-            }
-            for (const key of this.cache.keys())
-                if (!seen.has(key) && !guarded(key))
-                    this.cache.delete(key);
-        }
-        catch {
-        }
-    }
-    startRefresh(intervalMs = 5_000) {
-        this.timer ??= setInterval(() => void this.refresh(), intervalMs);
-        this.timer.unref?.();
-        return this;
-    }
-    stop() {
-        if (this.timer)
-            clearInterval(this.timer);
-        this.timer = undefined;
-    }
-}
+export { TiDBBreakerState } from "./breaker-state-sql.js";
 //# sourceMappingURL=tidb-breaker-state.js.map

package/dist/plugins/tidb-checkpoint-store.d.ts

@@ -1,50 +1,2 @@
-import type { Pool } from "mysql2/promise";
-import { type Checkpoint, type CheckpointStore, type CheckpointSummary, type CheckpointToken, type ResumeOutcome, type ResolveExpectation, type ReopenReason, type RiskDescriptor } from "@sema-agent/core";
-export declare const TERMINAL_BACKSTOP_MS: number;
-export declare const TERMINAL_GRACE_MS = 3600000;
-export declare function tokenFingerprint(token: string): string;
-export interface PendingCheckpoint {
-    sessionId: string;
-    scope: string;
-    toolName: string | null;
-    toolCallId: string | null;
-    boundCallId: string | null;
-    boundInputHash: string | null;
-    taskId: string | null;
-    input: unknown;
-    createdAt: number;
-    deadline: number | null;
-    riskDescriptor: RiskDescriptor | null;
-}
-export declare function boundedToolInput(args: unknown): unknown;
-export declare class TiDBCheckpointStore implements CheckpointStore {
-    private readonly pool;
-    private readonly logger?;
-    constructor(pool: Pool, logger?: {
-        info?(msg: string, meta?: unknown): void;
-    } | undefined);
-    put(token: CheckpointToken, cp: Checkpoint): Promise<void>;
-    get(token: CheckpointToken): Promise<Checkpoint | null>;
-    resolve(token: CheckpointToken, scope: string, outcome: ResumeOutcome, expect?: ResolveExpectation): Promise<boolean>;
-    reopen(token: CheckpointToken, scope: string, reason: ReopenReason): Promise<boolean>;
-    setPendingSteer(token: CheckpointToken, scope: string, steer: {
-        text: string;
-        trusted: boolean;
-    }): Promise<boolean>;
-    expire(token: CheckpointToken, scope: string): Promise<boolean>;
-    reap(scope: string, cutoff: number): Promise<number>;
-    reapExpired(cutoff: number): Promise<number>;
-    listExpiredApprovalGates(cutoff: number, limit?: number): Promise<Array<{
-        sessionId: string;
-        scope: string;
-    }>>;
-    listPending(scope?: string): Promise<PendingCheckpoint[]>;
-    listByScope(scope: string): Promise<CheckpointSummary[]>;
-    findPendingTokenBySession(sessionId: string, scope?: string): Promise<CheckpointToken | null>;
-    peekPendingScope(sessionId: string): Promise<string | null | undefined>;
-    putCtx(sessionId: string, ctx: unknown): Promise<void>;
-    getCtx<T = unknown>(sessionId: string): Promise<T | null>;
-    reapCtx(cutoff: number): Promise<number>;
-    deleteBySession(sessionId: string, owner: string | null): Promise<number>;
-}
+export { TiDBCheckpointStore, SqlCheckpointStore, TERMINAL_BACKSTOP_MS, TERMINAL_GRACE_MS, tokenFingerprint, boundedToolInput, type PendingCheckpoint, } from "./checkpoint-store-sql.js";
 //# sourceMappingURL=tidb-checkpoint-store.d.ts.map

package/dist/plugins/tidb-checkpoint-store.js

@@ -1,192 +1,2 @@
-import { createHash } from "node:crypto";
-import { CheckpointError, validatePendingSteer, checkpointVersionOf, winnerFromOutcome, summarizeCheckpoint, MAX_SUPPORTED_CHECKPOINT_VERSION } from "@sema-agent/core";
-import { redactDeep } from "../trace/redact.js";
-import { parseJsonStrict as parseJson } from "./sql-row-helpers.js";
-const MAX_TOOL_INPUT_CHARS = 8192;
-export const TERMINAL_BACKSTOP_MS = Math.max(60_000, Number(process.env.APPROVAL_TERMINAL_BACKSTOP_MS) || 30 * 86_400_000);
-export const TERMINAL_GRACE_MS = 3_600_000;
-export function tokenFingerprint(token) {
-    return "sha256:" + createHash("sha256").update(token).digest("hex").slice(0, 12);
-}
-export function boundedToolInput(args) {
-    if (args === undefined)
-        return null;
-    const redacted = redactDeep(args);
-    const json = JSON.stringify(redacted);
-    if (json === undefined)
-        return null;
-    if (json.length > MAX_TOOL_INPUT_CHARS)
-        return { truncated: true, bytes: json.length };
-    return redacted;
-}
-const DUP_ENTRY = 1062;
-export class TiDBCheckpointStore {
-    pool;
-    logger;
-    constructor(pool, logger) {
-        this.pool = pool;
-        this.logger = logger;
-    }
-    async put(token, cp) {
-        const pa = cp.pendingAction;
-        const version = checkpointVersionOf(cp);
-        const toolInput = boundedToolInput(pa?.args);
-        try {
-            await this.pool.query("INSERT INTO checkpoint (token, scope, session_id, version, status, tool_name, tool_call_id, tool_input, checkpoint, deadline, created_at, terminal_at, gate_kind, bound_input_hash, risk_descriptor) " +
-                "VALUES (?,?,?,?,'pending',?,?,?,?,?,?,?,?,?,?)", [
-                token,
-                cp.scope,
-                cp.sessionId,
-                version,
-                pa?.toolName ?? null,
-                pa?.toolCallId ?? null,
-                toolInput === null ? null : JSON.stringify(toolInput),
-                JSON.stringify(cp),
-                cp.deadline ?? null,
-                cp.createdAt,
-                Math.max(cp.createdAt + TERMINAL_BACKSTOP_MS, (cp.deadline ?? 0) + TERMINAL_GRACE_MS),
-                cp.gate?.kind ?? null,
-                pa?.boundInputHash ?? null,
-                ((g) => (g?.riskDescriptor ? JSON.stringify(g.riskDescriptor) : null))(cp.gate),
-            ]);
-        }
-        catch (e) {
-            if (e?.errno === DUP_ENTRY) {
-                throw new CheckpointError("checkpoint.already_exists", `checkpoint token already exists`);
-            }
-            throw e;
-        }
-    }
-    async get(token) {
-        const [rows] = await this.pool.query("SELECT version, status, checkpoint, outcome, reopen_reason, rev, pending_steer FROM checkpoint WHERE token = ?", [token]);
-        const r = rows[0];
-        if (!r)
-            return null;
-        const version = Number(r.version);
-        if (version > MAX_SUPPORTED_CHECKPOINT_VERSION) {
-            throw new Error(`checkpoint format version ${version} is newer than this build supports (max ${MAX_SUPPORTED_CHECKPOINT_VERSION}) — upgrade the service to resume it`);
-        }
-        const cp = parseJson(r.checkpoint);
-        cp.status = r.status;
-        const outcomeRaw = parseJson(r.outcome);
-        if (outcomeRaw)
-            cp.resolvedOutcome = winnerFromOutcome(outcomeRaw);
-        cp.reopenReason = r.reopen_reason ?? undefined;
-        cp.rev = r.rev == null ? 0 : Number(r.rev);
-        const steerRaw = parseJson(r.pending_steer);
-        if (steerRaw && cp.state)
-            cp.state.pendingSteer = steerRaw;
-        this.logger?.info?.("checkpoint_get", {
-            tokenFp: tokenFingerprint(token),
-            status: cp.status,
-            version,
-            hasWorkspaceHandle: !!cp.state?.workspaceHandle,
-            snapshotId: cp.state?.workspaceHandle?.snapshotId,
-        });
-        return cp;
-    }
-    async resolve(token, scope, outcome, expect) {
-        const params = [JSON.stringify(outcome), Date.now(), token, scope];
-        if (expect)
-            params.push(expect.rev);
-        const [res] = await this.pool.query("UPDATE checkpoint SET status = 'resolved', outcome = ?, decided_at = ?, rev = rev + 1, reopen_reason = NULL WHERE token = ? AND scope = ? AND status = 'pending'" +
-            (expect ? " AND rev = ?" : ""), params);
-        return res.affectedRows === 1;
-    }
-    async reopen(token, scope, reason) {
-        const [res] = await this.pool.query("UPDATE checkpoint SET status = 'pending', reopen_reason = ?, rev = rev + 1 WHERE token = ? AND scope = ? AND status = 'resolved'", [reason, token, scope]);
-        return res.affectedRows === 1;
-    }
-    async setPendingSteer(token, scope, steer) {
-        const clean = validatePendingSteer(steer);
-        const [res] = await this.pool.query("UPDATE checkpoint SET pending_steer = ? WHERE token = ? AND scope = ? AND status = 'pending'", [JSON.stringify(clean), token, scope]);
-        return res.affectedRows > 0;
-    }
-    async expire(token, scope) {
-        const [res] = await this.pool.query("UPDATE checkpoint SET status = 'expired', decided_at = ? WHERE token = ? AND scope = ? AND status = 'pending'", [Date.now(), token, scope]);
-        return res.affectedRows === 1;
-    }
-    async reap(scope, cutoff) {
-        const [res] = await this.pool.query("UPDATE checkpoint SET status = 'expired', decided_at = ? WHERE scope = ? AND status = 'pending' AND deadline IS NOT NULL AND deadline <= ?", [Date.now(), scope, cutoff]);
-        return res.affectedRows;
-    }
-    async reapExpired(cutoff) {
-        const [res] = await this.pool.query("UPDATE checkpoint SET status = 'expired', decided_at = ? " +
-            "WHERE status = 'pending' AND ((deadline IS NOT NULL AND deadline <= ? AND (gate_kind IS NULL OR gate_kind NOT IN ('human','irreversible_ask') OR COALESCE(tool_name,'') = 'AskUserQuestion')) OR (terminal_at IS NOT NULL AND terminal_at <= ?))", [Date.now(), cutoff, cutoff]);
-        return res.affectedRows;
-    }
-    async listExpiredApprovalGates(cutoff, limit = 100) {
-        const [rows] = await this.pool.query("SELECT session_id, scope FROM checkpoint WHERE status = 'pending' AND gate_kind IN ('human','irreversible_ask') " +
-            "AND COALESCE(tool_name,'') <> 'AskUserQuestion' AND deadline IS NOT NULL AND deadline <= ? ORDER BY deadline ASC LIMIT ?", [cutoff, limit]);
-        return rows.map((r) => ({ sessionId: String(r.session_id), scope: String(r.scope) }));
-    }
-    async listPending(scope) {
-        const base = "SELECT c.session_id, c.scope, c.tool_name, c.tool_call_id, c.tool_input, c.bound_input_hash, c.risk_descriptor, c.created_at, c.deadline, ta.task_id " +
-            "FROM checkpoint c LEFT JOIN task_active ta ON ta.session_id = c.session_id WHERE c.status='pending'";
-        const [rows] = scope
-            ? await this.pool.query(`${base} AND c.scope=? ORDER BY c.created_at ASC`, [scope])
-            : await this.pool.query(`${base} ORDER BY c.created_at ASC`);
-        const out = rows.map((r) => {
-            const toolCallId = r.tool_call_id ?? null;
-            return {
-                sessionId: String(r.session_id),
-                scope: String(r.scope),
-                toolName: r.tool_name ?? null,
-                toolCallId,
-                boundCallId: toolCallId,
-                boundInputHash: r.bound_input_hash ?? null,
-                taskId: r.task_id ?? null,
-                input: r.tool_input ?? null,
-                createdAt: Number(r.created_at),
-                deadline: r.deadline == null ? null : Number(r.deadline),
-                riskDescriptor: parseJson(r.risk_descriptor),
-            };
-        });
-        return out.sort((a, b) => (b.riskDescriptor?.severity ?? 0) - (a.riskDescriptor?.severity ?? 0) || a.createdAt - b.createdAt);
-    }
-    async listByScope(scope) {
-        const [rows] = await this.pool.query("SELECT checkpoint, status FROM checkpoint WHERE status = 'pending' AND scope = ? ORDER BY created_at ASC LIMIT 500", [scope]);
-        const out = [];
-        for (const r of rows) {
-            const cp = parseJson(r.checkpoint);
-            if (!cp)
-                continue;
-            cp.status = r.status;
-            out.push(summarizeCheckpoint(cp));
-        }
-        return out;
-    }
-    async findPendingTokenBySession(sessionId, scope) {
-        const [rows] = scope
-            ? await this.pool.query("SELECT token FROM checkpoint WHERE session_id=? AND scope=? AND status='pending' LIMIT 1", [sessionId, scope])
-            : await this.pool.query("SELECT token FROM checkpoint WHERE session_id=? AND status='pending' LIMIT 1", [sessionId]);
-        return rows[0] ? String(rows[0].token) : null;
-    }
-    async peekPendingScope(sessionId) {
-        const [rows] = await this.pool.query("SELECT scope FROM checkpoint WHERE session_id=? AND status='pending' LIMIT 1", [sessionId]);
-        if (!rows[0])
-            return undefined;
-        return rows[0].scope ?? null;
-    }
-    async putCtx(sessionId, ctx) {
-        await this.pool.query("INSERT INTO checkpoint_ctx (session_id, ctx, updated_at) VALUES (?,?,?) " +
-            "ON DUPLICATE KEY UPDATE ctx = VALUES(ctx), updated_at = VALUES(updated_at)", [sessionId, JSON.stringify(ctx), Date.now()]);
-    }
-    async getCtx(sessionId) {
-        const [rows] = await this.pool.query("SELECT ctx FROM checkpoint_ctx WHERE session_id=?", [sessionId]);
-        return rows[0] ? parseJson(rows[0].ctx) : null;
-    }
-    async reapCtx(cutoff) {
-        const [res] = await this.pool.query("DELETE FROM checkpoint_ctx WHERE updated_at < ? " +
-            "AND session_id NOT IN (SELECT session_id FROM checkpoint WHERE status='pending') " +
-            "AND session_id NOT IN (SELECT session_id FROM task_active)", [cutoff]);
-        return res.affectedRows;
-    }
-    async deleteBySession(sessionId, owner) {
-        const ownerGuard = "EXISTS (SELECT 1 FROM session_meta sm WHERE sm.session_id = ? AND sm.owner <=> ?)";
-        const [res] = await this.pool.query(`DELETE FROM checkpoint WHERE session_id = ? AND ${ownerGuard}`, [sessionId, sessionId, owner]);
-        await this.pool.query(`DELETE FROM checkpoint_ctx WHERE session_id = ? AND ${ownerGuard}`, [sessionId, sessionId, owner]);
-        return res.affectedRows;
-    }
-}
+export { TiDBCheckpointStore, SqlCheckpointStore, TERMINAL_BACKSTOP_MS, TERMINAL_GRACE_MS, tokenFingerprint, boundedToolInput, } from "./checkpoint-store-sql.js";
 //# sourceMappingURL=tidb-checkpoint-store.js.map

package/dist/plugins/tidb-file-snapshot-store.d.ts

@@ -1,28 +1,2 @@
-import { type FileSnapshotBounds, type FileSnapshotStore, type FileSnapshotResult, type ExecutionEnv } from "@sema-agent/core";
-import type { Pool } from "mysql2/promise";
-import { type BlobBackend } from "./blob-backend.js";
-export declare class TiDBFileSnapshotStore implements FileSnapshotStore {
-    private readonly pool;
-    private readonly bounds;
-    private readonly inFlight;
-    private readonly blobs;
-    onSkippedVanished: ((count: number) => void | Promise<void>) | undefined;
-    constructor(pool: Pool, blobBackend?: BlobBackend, bounds?: FileSnapshotBounds);
-    private putBlobRow;
-    putBlob(hash: string, bytes: Uint8Array): Promise<FileSnapshotResult>;
-    snapshot(scope: string, key: string, env: ExecutionEnv, root: string, signal?: AbortSignal): Promise<FileSnapshotResult>;
-    importManifest(scope: string, key: string, manifest: Map<string, string>, srcGetBlob: (hash: string) => Promise<Uint8Array | undefined>): Promise<FileSnapshotResult>;
-    restore(scope: string, key: string, env: ExecutionEnv, root: string, signal?: AbortSignal): Promise<FileSnapshotResult>;
-    has(scope: string, key: string): Promise<boolean>;
-    listKeys(scope: string): Promise<string[]>;
-    exportManifest(scope: string, key: string): Promise<Map<string, string> | null>;
-    getBlob(hash: string): Promise<Uint8Array | undefined>;
-    hasBlobs(hashes: string[]): Promise<Set<string>>;
-    blobSizes(hashes: string[]): Promise<Map<string, number>>;
-    reap(scope: string, keepKeys: string[]): Promise<number>;
-    deleteBySession(scope: string): Promise<number>;
-    private gcOrphanBlobs;
-    sweepOrphanBlobs(): Promise<number>;
-    private tx;
-}
+export { TiDBFileSnapshotStore } from "./file-snapshot-store-sql.js";
 //# sourceMappingURL=tidb-file-snapshot-store.d.ts.map