@sema-agent/server 1.315.0 → 1.317.0

package/dist/plugins/tidb-resume-anchor-store.js

@@ -1,23 +1,2 @@
-export class TiDBResumeAnchorStore {
-    pool;
-    constructor(pool) {
-        this.pool = pool;
-    }
-    async put(sessionId, eventId, entryId, owner) {
-        await this.pool.query("INSERT INTO resume_anchor (session_id, event_id, entry_id, owner, created_at) VALUES (?,?,?,?,?) " +
-            "ON DUPLICATE KEY UPDATE entry_id = VALUES(entry_id), owner = VALUES(owner)", [sessionId, eventId, entryId, owner, new Date()]);
-    }
-    async resolve(sessionId, eventId, owner) {
-        const [rows] = await this.pool.query("SELECT entry_id FROM resume_anchor WHERE session_id = ? AND event_id = ? AND owner <=> ?", [sessionId, eventId, owner]);
-        return rows[0] ? String(rows[0].entry_id) : undefined;
-    }
-    async listBySession(sessionId) {
-        const [rows] = await this.pool.query("SELECT event_id, entry_id, owner FROM resume_anchor WHERE session_id = ?", [sessionId]);
-        return rows.map((r) => ({ eventId: String(r.event_id), entryId: String(r.entry_id), owner: r.owner == null ? null : String(r.owner) }));
-    }
-    async deleteBySession(sessionId) {
-        const [r] = await this.pool.query("DELETE FROM resume_anchor WHERE session_id = ?", [sessionId]);
-        return r.affectedRows;
-    }
-}
+export { TiDBResumeAnchorStore } from "./resume-anchor-store-sql.js";
 //# sourceMappingURL=tidb-resume-anchor-store.js.map

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

@@ -1,77 +1,3 @@
-import type { TaskResult, TaskStatus } from "@sema-agent/core";
-import { type UsageRow } from "../usage-analytics.js";
-import type { Pool } from "mysql2/promise";
-import type { RunRecord, SessionSummary, RunEvent } from "./store-contracts.js";
+export { TiDBRunStore } from "./run-store-sql.js";
 export type { RunRecord, SessionSummary, RunEvent } from "./store-contracts.js";
-export declare class TiDBRunStore {
-    private readonly pool;
-    constructor(pool: Pool);
-    createRun(taskId: string, sessionId: string, owner: string | null, instanceId: string, meta?: {
-        jobId?: string | null;
-        source?: string | null;
-        objectivePreview?: string | null;
-    }): Promise<{
-        ok: true;
-    } | {
-        ok: false;
-        activeTaskId: string;
-    }>;
-    requestCancel(taskId: string, owner: string | null): Promise<boolean>;
-    isCancelRequested(taskId: string, owner: string | null): Promise<boolean>;
-    requestPreempt(taskId: string, owner: string | null): Promise<boolean>;
-    isPreemptRequested(taskId: string, owner: string | null): Promise<boolean>;
-    heartbeat(taskId: string, owner: string | null): Promise<void>;
-    appendEvent(taskId: string, seq: number, type: string, data: unknown): Promise<void>;
-    maxSeq(taskId: string): Promise<number>;
-    private tx;
-    setTerminal(taskId: string, status: TaskStatus, result: TaskResult | null, error: string | null): Promise<void>;
-    setSuspended(taskId: string): Promise<void>;
-    setNeedsReview(taskId: string): Promise<void>;
-    markResuming(taskId: string): Promise<boolean>;
-    getActiveTaskId(sessionId: string): Promise<string | undefined>;
-    private rowToRun;
-    getRun(taskId: string): Promise<RunRecord | undefined>;
-    listRuns(opts: {
-        status?: string;
-        jobId?: string;
-        source?: string;
-        owner?: string;
-        cursor?: {
-            createdAt: string;
-            taskId: string;
-        };
-        limit: number;
-    }): Promise<RunRecord[]>;
-    listSessions(opts: {
-        owner?: string;
-        cursor?: {
-            lastActivityAt: string;
-            sessionId: string;
-        };
-        limit: number;
-        q?: string;
-    }): Promise<SessionSummary[]>;
-    deleteBySession(sessionId: string, owner: string | null): Promise<{
-        removed: number;
-    } | {
-        active: string;
-    }>;
-    sourceSummary(sinceMs: number): Promise<Array<{
-        source: string | null;
-        status: string;
-        count: number;
-    }>>;
-    usageScan(fromMs: number, toMs: number, opts?: {
-        owner?: string;
-        limit?: number;
-    }): Promise<{
-        rows: UsageRow[];
-        truncated: boolean;
-    }>;
-    retainedFrom(taskId: string): Promise<number>;
-    getEvents(taskId: string, afterSeq: number): Promise<RunEvent[]>;
-    reapStale(olderThanMs: number): Promise<number>;
-    reapSuspended(olderThanMs: number): Promise<number>;
-    failSuspendedWithExpiredCheckpoint(): Promise<number>;
-}
 //# sourceMappingURL=tidb-run-store.d.ts.map

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

@@ -1,307 +1,2 @@
-import { projectUsageStats, USAGE_SCAN_LIMIT } from "../usage-analytics.js";
-import { escapeLike } from "./sql-escape.js";
-import { parseJsonLenient as parseJson, toIso as iso } from "./sql-row-helpers.js";
-function isDupKey(err) {
-    return Boolean(err) && err.code === "ER_DUP_ENTRY";
-}
-export class TiDBRunStore {
-    pool;
-    constructor(pool) {
-        this.pool = pool;
-    }
-    async createRun(taskId, sessionId, owner, instanceId, meta = {}) {
-        const conn = await this.pool.getConnection();
-        try {
-            await conn.beginTransaction();
-            const now = new Date();
-            try {
-                await conn.query("INSERT INTO task_active (session_id, task_id, since) VALUES (?,?,?)", [
-                    sessionId,
-                    taskId,
-                    now,
-                ]);
-            }
-            catch (err) {
-                if (isDupKey(err)) {
-                    await conn.rollback();
-                    const [active] = await conn.query("SELECT task_id FROM task_active WHERE session_id = ?", [sessionId]);
-                    return { ok: false, activeTaskId: String(active[0]?.task_id ?? "") };
-                }
-                throw err;
-            }
-            await conn.query("INSERT INTO task_run (task_id, session_id, owner, status, instance_id, job_id, source, objective_preview, created_at, updated_at) " +
-                "VALUES (?,?,?,'running',?,?,?,?,?,?)", [taskId, sessionId, owner, instanceId, meta.jobId ?? null, meta.source ?? null, meta.objectivePreview ?? null, now, now]);
-            await conn.commit();
-            return { ok: true };
-        }
-        catch (err) {
-            try {
-                await conn.rollback();
-            }
-            catch {
-            }
-            throw err;
-        }
-        finally {
-            conn.release();
-        }
-    }
-    async requestCancel(taskId, owner) {
-        const [r] = await this.pool.query("UPDATE task_run SET cancel_requested = 1 WHERE task_id = ? AND owner <=> ? AND status = 'running'", [taskId, owner]);
-        return r.affectedRows > 0;
-    }
-    async isCancelRequested(taskId, owner) {
-        const [rows] = await this.pool.query("SELECT cancel_requested FROM task_run WHERE task_id = ? AND owner <=> ?", [taskId, owner]);
-        return Boolean(rows[0]?.cancel_requested);
-    }
-    async requestPreempt(taskId, owner) {
-        const [r] = await this.pool.query("UPDATE task_run SET preempt_requested = 1 WHERE task_id = ? AND owner <=> ? AND status = 'running'", [taskId, owner]);
-        return r.affectedRows > 0;
-    }
-    async isPreemptRequested(taskId, owner) {
-        const [rows] = await this.pool.query("SELECT preempt_requested FROM task_run WHERE task_id = ? AND owner <=> ?", [taskId, owner]);
-        return Boolean(rows[0]?.preempt_requested);
-    }
-    async heartbeat(taskId, owner) {
-        await this.pool.query("UPDATE task_run SET updated_at = ? WHERE task_id = ? AND owner <=> ? AND status = 'running'", [
-            new Date(),
-            taskId,
-            owner,
-        ]);
-    }
-    async appendEvent(taskId, seq, type, data) {
-        await this.pool.query("INSERT INTO task_event (task_id, seq, type, data, ts) VALUES (?,?,?,?,?)", [
-            taskId,
-            seq,
-            type,
-            data == null ? null : JSON.stringify(data),
-            new Date(),
-        ]);
-    }
-    async maxSeq(taskId) {
-        const [rows] = await this.pool.query("SELECT MAX(seq) AS m FROM task_event WHERE task_id = ?", [taskId]);
-        const m = rows[0]?.m;
-        return m == null ? 0 : Number(m);
-    }
-    async tx(fn) {
-        const conn = await this.pool.getConnection();
-        try {
-            await conn.beginTransaction();
-            await fn(conn);
-            await conn.commit();
-        }
-        catch (err) {
-            try {
-                await conn.rollback();
-            }
-            catch {
-            }
-            throw err;
-        }
-        finally {
-            conn.release();
-        }
-    }
-    async setTerminal(taskId, status, result, error) {
-        const errorCode = result?.errorCode ?? null;
-        await this.tx(async (conn) => {
-            await conn.query("UPDATE task_run SET status = ?, result = ?, error = ?, error_code = ?, updated_at = ? WHERE task_id = ? AND status IN ('running','suspended','needs_review')", [status, result == null ? null : JSON.stringify(result), error, errorCode, new Date(), taskId]);
-            await conn.query("DELETE FROM task_active WHERE task_id = ?", [taskId]);
-        });
-    }
-    async setSuspended(taskId) {
-        await this.pool.query("UPDATE task_run SET status = 'suspended', updated_at = ? WHERE task_id = ? AND status IN ('running','suspended')", [new Date(), taskId]);
-    }
-    async setNeedsReview(taskId) {
-        await this.pool.query("UPDATE task_run SET status = 'needs_review', updated_at = ? WHERE task_id = ? AND status IN ('running','needs_review')", [new Date(), taskId]);
-    }
-    async markResuming(taskId) {
-        const [res] = await this.pool.query("UPDATE task_run SET status = 'running', cancel_requested = 0, preempt_requested = 0, updated_at = ? WHERE task_id = ? AND status IN ('suspended','needs_review')", [new Date(), taskId]);
-        return res.affectedRows === 1;
-    }
-    async getActiveTaskId(sessionId) {
-        const [rows] = await this.pool.query("SELECT task_id FROM task_active WHERE session_id = ?", [sessionId]);
-        return rows[0] ? String(rows[0].task_id) : undefined;
-    }
-    rowToRun(r) {
-        return {
-            taskId: String(r.task_id),
-            sessionId: String(r.session_id),
-            owner: r.owner ?? null,
-            status: r.status,
-            result: parseJson(r.result),
-            error: r.error ?? null,
-            errorCode: r.error_code ?? null,
-            jobId: r.job_id ?? null,
-            source: r.source ?? null,
-            instanceId: r.instance_id ?? null,
-            objectivePreview: r.objective_preview ?? null,
-            createdAt: iso(r.created_at),
-            updatedAt: iso(r.updated_at),
-        };
-    }
-    async getRun(taskId) {
-        const [rows] = await this.pool.query("SELECT task_id, session_id, owner, status, result, error, error_code, job_id, source, instance_id, objective_preview, created_at, updated_at FROM task_run WHERE task_id = ?", [taskId]);
-        return rows[0] ? this.rowToRun(rows[0]) : undefined;
-    }
-    async listRuns(opts) {
-        if (opts.cursor && Number.isNaN(Date.parse(opts.cursor.createdAt)))
-            return [];
-        const where = [];
-        const params = [];
-        if (opts.status) {
-            where.push("status = ?");
-            params.push(opts.status);
-        }
-        if (opts.jobId) {
-            where.push("job_id = ?");
-            params.push(opts.jobId);
-        }
-        if (opts.source) {
-            where.push("source = ?");
-            params.push(opts.source);
-        }
-        if (opts.owner) {
-            where.push("owner = ?");
-            params.push(opts.owner);
-        }
-        if (opts.cursor) {
-            where.push("(created_at < ? OR (created_at = ? AND task_id < ?))");
-            params.push(new Date(opts.cursor.createdAt), new Date(opts.cursor.createdAt), opts.cursor.taskId);
-        }
-        const sql = "SELECT task_id, session_id, owner, status, result, error, error_code, job_id, source, objective_preview, created_at, updated_at FROM task_run" +
-            (where.length ? " WHERE " + where.join(" AND ") : "") +
-            " ORDER BY created_at DESC, task_id DESC LIMIT ?";
-        params.push(opts.limit);
-        const [rows] = await this.pool.query(sql, params);
-        return rows.map((r) => this.rowToRun(r));
-    }
-    async listSessions(opts) {
-        const params = [];
-        let innerWhere = "";
-        if (opts.owner) {
-            innerWhere = " WHERE owner = ?";
-            params.push(opts.owner);
-        }
-        const outer = ["rn = 1"];
-        if (opts.q) {
-            outer.push("EXISTS (SELECT 1 FROM task_run tr WHERE tr.session_id = t.session_id AND LOWER(tr.objective_preview) LIKE LOWER(?) ESCAPE '\\\\')");
-            params.push(`%${escapeLike(opts.q)}%`);
-        }
-        if (opts.cursor) {
-            outer.push("(last_activity < ? OR (last_activity = ? AND session_id < ?))");
-            params.push(new Date(opts.cursor.lastActivityAt), new Date(opts.cursor.lastActivityAt), opts.cursor.sessionId);
-        }
-        const sql = "SELECT session_id, owner, task_id, last_activity, first_activity, run_count, objective_preview, status FROM (" +
-            "SELECT session_id, owner, task_id, objective_preview, status," +
-            " ROW_NUMBER() OVER (PARTITION BY session_id ORDER BY created_at DESC, task_id DESC) AS rn," +
-            " MAX(created_at) OVER (PARTITION BY session_id) AS last_activity," +
-            " MIN(created_at) OVER (PARTITION BY session_id) AS first_activity," +
-            " COUNT(*) OVER (PARTITION BY session_id) AS run_count" +
-            " FROM task_run" + innerWhere + ") t WHERE " + outer.join(" AND ") +
-            " ORDER BY last_activity DESC, session_id DESC LIMIT ?";
-        params.push(opts.limit);
-        const [rows] = await this.pool.query(sql, params);
-        return rows.map((r) => ({
-            sessionId: String(r.session_id),
-            owner: r.owner ?? null,
-            lastActivityAt: iso(r.last_activity),
-            firstActivityAt: iso(r.first_activity),
-            runCount: Number(r.run_count),
-            objectivePreview: r.objective_preview ?? null,
-            lastStatus: String(r.status),
-            lastRunId: r.task_id ?? null,
-            title: null,
-        }));
-    }
-    async deleteBySession(sessionId, owner) {
-        let result = { removed: 0 };
-        await this.tx(async (conn) => {
-            const [active] = await conn.query("SELECT task_id FROM task_active WHERE session_id = ? FOR UPDATE", [sessionId]);
-            if (active[0]) {
-                result = { active: String(active[0].task_id) };
-                return;
-            }
-            await conn.query("DELETE te FROM task_event te JOIN task_run tr ON te.task_id = tr.task_id WHERE tr.session_id = ? AND tr.owner <=> ?", [sessionId, owner]);
-            await conn.query("DELETE ta FROM task_active ta JOIN task_run tr ON ta.task_id = tr.task_id WHERE tr.session_id = ? AND tr.owner <=> ?", [sessionId, owner]);
-            const [res] = await conn.query("DELETE FROM task_run WHERE session_id = ? AND owner <=> ?", [sessionId, owner]);
-            result = { removed: res.affectedRows };
-        });
-        return result;
-    }
-    async sourceSummary(sinceMs) {
-        const [rows] = await this.pool.query("SELECT source, status, COUNT(*) AS c FROM task_run WHERE created_at >= ? GROUP BY source, status", [new Date(sinceMs)]);
-        return rows.map((r) => ({ source: r.source ?? null, status: String(r.status), count: Number(r.c) }));
-    }
-    async usageScan(fromMs, toMs, opts) {
-        const limit = Math.min(opts?.limit ?? USAGE_SCAN_LIMIT, USAGE_SCAN_LIMIT);
-        const where = ["created_at >= ?", "created_at < ?"];
-        const params = [new Date(fromMs), new Date(toMs)];
-        if (opts?.owner !== undefined) {
-            where.push("owner = ?");
-            params.push(opts.owner);
-        }
-        const [rows] = await this.pool.query(`SELECT owner, created_at, status, result FROM task_run WHERE ${where.join(" AND ")} ORDER BY created_at LIMIT ${limit + 1}`, params);
-        const truncated = rows.length > limit;
-        const page = truncated ? rows.slice(0, limit) : rows;
-        return {
-            truncated,
-            rows: page.map((r) => ({
-                owner: r.owner ?? null,
-                createdAtMs: new Date(r.created_at).getTime(),
-                status: String(r.status),
-                ...(r.result != null ? { stats: projectUsageStats(r.result) } : {}),
-            })),
-        };
-    }
-    async retainedFrom(taskId) {
-        const [rows] = await this.pool.query("SELECT MIN(seq) AS m FROM task_event WHERE task_id = ?", [taskId]);
-        const m = rows[0]?.m;
-        return m == null ? 0 : Number(m);
-    }
-    async getEvents(taskId, afterSeq) {
-        const [rows] = await this.pool.query("SELECT seq, type, data, ts FROM task_event WHERE task_id = ? AND seq > ? ORDER BY seq ASC", [taskId, afterSeq]);
-        return rows.map((r) => ({
-            seq: Number(r.seq),
-            type: String(r.type),
-            data: parseJson(r.data),
-            ts: iso(r.ts),
-        }));
-    }
-    async reapStale(olderThanMs) {
-        const cutoff = new Date(Date.now() - olderThanMs);
-        let reaped = 0;
-        await this.tx(async (conn) => {
-            const [res] = await conn.query("UPDATE task_run SET status = 'failed', error = 'run stalled (instance lost?)', updated_at = ? " +
-                "WHERE status = 'running' AND updated_at < ?", [new Date(), cutoff]);
-            reaped = res.affectedRows;
-            await conn.query("DELETE ta FROM task_active ta JOIN task_run tr ON ta.task_id = tr.task_id WHERE tr.status NOT IN ('running','suspended','needs_review')");
-        });
-        return reaped;
-    }
-    async reapSuspended(olderThanMs) {
-        const cutoff = new Date(Date.now() - olderThanMs);
-        let reaped = 0;
-        await this.tx(async (conn) => {
-            const [res] = await conn.query("UPDATE task_run tr SET tr.status = 'failed', tr.error = 'approval expired before decision', tr.error_code = 'approval.expired', tr.updated_at = ? " +
-                "WHERE tr.status = 'suspended' AND tr.updated_at < ? " +
-                "AND NOT EXISTS (SELECT 1 FROM checkpoint c WHERE c.session_id = tr.session_id AND c.status = 'pending')", [new Date(), cutoff]);
-            reaped = res.affectedRows;
-            await conn.query("DELETE ta FROM task_active ta JOIN task_run tr ON ta.task_id = tr.task_id WHERE tr.status NOT IN ('running','suspended','needs_review')");
-        });
-        return reaped;
-    }
-    async failSuspendedWithExpiredCheckpoint() {
-        let reaped = 0;
-        await this.tx(async (conn) => {
-            const [res] = await conn.query("UPDATE task_run tr SET tr.status = 'failed', tr.error = 'approval expired before decision', tr.error_code = 'approval.expired', tr.updated_at = ? " +
-                "WHERE tr.status IN ('suspended','needs_review') " +
-                "AND EXISTS (SELECT 1 FROM checkpoint ce WHERE ce.session_id = tr.session_id AND ce.status = 'expired') " +
-                "AND NOT EXISTS (SELECT 1 FROM checkpoint cp WHERE cp.session_id = tr.session_id AND cp.status = 'pending')", [new Date()]);
-            reaped = res.affectedRows;
-            await conn.query("DELETE ta FROM task_active ta JOIN task_run tr ON ta.task_id = tr.task_id WHERE tr.status NOT IN ('running','suspended','needs_review')");
-        });
-        return reaped;
-    }
-}
+export { TiDBRunStore } from "./run-store-sql.js";
 //# sourceMappingURL=tidb-run-store.js.map

package/dist/plugins/tidb-session-policy-store.d.ts

@@ -1,12 +1,2 @@
-import { type SessionPolicyStore, type SessionPermissionRules, type StoredSessionRules, type SessionRulesRecord, type PutRulesOptions } from "@sema-agent/core";
-import type { Pool } from "mysql2/promise";
-export declare class TiDBSessionPolicyStore implements SessionPolicyStore {
-    private readonly pool;
-    constructor(pool: Pool);
-    getRules(sessionId: string, principal?: string): Promise<StoredSessionRules | null>;
-    listBySession(sessionId: string): Promise<SessionRulesRecord[]>;
-    putRules(sessionId: string, principal: string | undefined, rules: SessionPermissionRules, opts?: PutRulesOptions): Promise<StoredSessionRules>;
-    deleteBySession(sessionId: string): Promise<void>;
-    private tx;
-}
+export { TiDBSessionPolicyStore } from "./session-policy-store-sql.js";
 //# sourceMappingURL=tidb-session-policy-store.d.ts.map

package/dist/plugins/tidb-session-policy-store.js

@@ -1,87 +1,2 @@
-import { createHash } from "node:crypto";
-import { loosenReasons, SessionPolicyError, normalizeRules, stripRev } from "@sema-agent/core";
-const policyKey = (sessionId, principal) => createHash("sha256").update(JSON.stringify([sessionId, principal ?? null])).digest("hex");
-const parseRules = (v) => (typeof v === "string" ? JSON.parse(v) : v);
-const isDupKey = (e) => e?.code === "ER_DUP_ENTRY" || e?.errno === 1062;
-export class TiDBSessionPolicyStore {
-    pool;
-    constructor(pool) {
-        this.pool = pool;
-    }
-    async getRules(sessionId, principal) {
-        const [rows] = await this.pool.query("SELECT rules, rev FROM session_policy WHERE policy_key = ?", [policyKey(sessionId, principal)]);
-        if (!rows[0])
-            return null;
-        return { ...parseRules(rows[0].rules), rev: Number(rows[0].rev) };
-    }
-    async listBySession(sessionId) {
-        const [rows] = await this.pool.query("SELECT principal, rules, rev FROM session_policy WHERE session_id = ?", [sessionId]);
-        return rows.map((r) => ({
-            principal: r.principal == null ? undefined : String(r.principal),
-            rules: { ...parseRules(r.rules), rev: Number(r.rev) },
-        }));
-    }
-    async putRules(sessionId, principal, rules, opts) {
-        const key = policyKey(sessionId, principal);
-        for (let attempt = 0;; attempt++) {
-            try {
-                return await this.tx(async (conn) => {
-                    const [rows] = await conn.query("SELECT rules, rev FROM session_policy WHERE policy_key = ? FOR UPDATE", [key]);
-                    const prior = rows[0] ? { rules: parseRules(rows[0].rules), rev: Number(rows[0].rev) } : undefined;
-                    const priorRev = prior?.rev ?? 0;
-                    if (opts?.expectedRev !== undefined && priorRev !== opts.expectedRev) {
-                        throw new SessionPolicyError("conflict", `session rules rev mismatch (expected ${opts.expectedRev}, have ${priorRev})`);
-                    }
-                    const clean = normalizeRules(stripRev(rules));
-                    if (prior && opts?.operator !== true) {
-                        const loosens = loosenReasons(prior.rules, clean);
-                        if (loosens.length > 0) {
-                            throw new SessionPolicyError("loosen_forbidden", `a non-operator session-rule write may only tighten; refused: ${loosens.join("; ")}`);
-                        }
-                    }
-                    const nextRev = priorRev + 1;
-                    const json = JSON.stringify(clean);
-                    const now = new Date();
-                    if (prior) {
-                        const [r] = await conn.query("UPDATE session_policy SET principal = ?, rules = ?, rev = ?, updated_at = ? WHERE policy_key = ? AND rev = ?", [principal ?? null, json, nextRev, now, key, priorRev]);
-                        if (r.affectedRows !== 1)
-                            throw new SessionPolicyError("conflict", `session rules rev mismatch (expected ${opts?.expectedRev ?? priorRev}, have ${priorRev})`);
-                    }
-                    else {
-                        await conn.query("INSERT INTO session_policy (policy_key, session_id, principal, rules, rev, created_at, updated_at) VALUES (?,?,?,?,?,?,?)", [key, sessionId, principal ?? null, json, nextRev, now, now]);
-                    }
-                    return structuredClone({ ...clean, rev: nextRev });
-                });
-            }
-            catch (e) {
-                if (isDupKey(e) && attempt < 4)
-                    continue;
-                throw e;
-            }
-        }
-    }
-    async deleteBySession(sessionId) {
-        await this.pool.query("DELETE FROM session_policy WHERE session_id = ?", [sessionId]);
-    }
-    async tx(fn) {
-        const conn = await this.pool.getConnection();
-        try {
-            await conn.beginTransaction();
-            const out = await fn(conn);
-            await conn.commit();
-            return out;
-        }
-        catch (err) {
-            try {
-                await conn.rollback();
-            }
-            catch {
-            }
-            throw err;
-        }
-        finally {
-            conn.release();
-        }
-    }
-}
+export { TiDBSessionPolicyStore } from "./session-policy-store-sql.js";
 //# sourceMappingURL=tidb-session-policy-store.js.map

package/dist/plugins/tidb-tool-result-store.d.ts

@@ -1,15 +1,3 @@
-import type { ToolResultStore, ToolResultSlice } from "@sema-agent/core";
-import type { Pool } from "mysql2/promise";
+export { TiDBToolResultStore, SqlToolResultStore } from "./tool-result-store-sql.js";
 export { escapeLike } from "./sql-escape.js";
-export declare class TiDBToolResultStore implements ToolResultStore {
-    private readonly pool;
-    constructor(pool: Pool);
-    put(ref: string, content: string): Promise<void>;
-    get(ref: string, opts?: {
-        offset?: number;
-        limit?: number;
-    }): Promise<ToolResultSlice | undefined>;
-    reapOlderThan(cutoffMs: number): Promise<number>;
-    deleteBySession(sessionId: string): Promise<number>;
-}
 //# sourceMappingURL=tidb-tool-result-store.d.ts.map

package/dist/plugins/tidb-tool-result-store.js

@@ -1,40 +1,3 @@
-import { escapeLike } from "./sql-escape.js";
+export { TiDBToolResultStore, SqlToolResultStore } from "./tool-result-store-sql.js";
 export { escapeLike } from "./sql-escape.js";
-export class TiDBToolResultStore {
-    pool;
-    constructor(pool) {
-        this.pool = pool;
-    }
-    async put(ref, content) {
-        const safe = Buffer.from(content, "utf8").toString("utf8");
-        await this.pool.query("INSERT IGNORE INTO tool_result (ref, content, created_at) VALUES (?,?,?)", [ref, safe, new Date()]);
-    }
-    async get(ref, opts = {}) {
-        const offset = Math.max(0, Math.floor(opts.offset ?? 0));
-        const hasLimit = opts.limit !== undefined && Number.isFinite(opts.limit);
-        const limit = hasLimit ? Math.max(0, Math.floor(opts.limit)) : 0;
-        const sql = hasLimit
-            ? "SELECT SUBSTRING(content, ?, ?) AS slice, CHAR_LENGTH(content) AS total FROM tool_result WHERE ref = ?"
-            : "SELECT SUBSTRING(content, ?) AS slice, CHAR_LENGTH(content) AS total FROM tool_result WHERE ref = ?";
-        const params = hasLimit ? [offset + 1, limit, ref] : [offset + 1, ref];
-        const [rows] = await this.pool.query(sql, params);
-        if (rows.length === 0)
-            return undefined;
-        return { content: String(rows[0].slice ?? ""), offset, totalChars: Number(rows[0].total) };
-    }
-    async reapOlderThan(cutoffMs) {
-        try {
-            const [res] = await this.pool.query("DELETE FROM tool_result WHERE created_at < ?", [new Date(cutoffMs)]);
-            return res.affectedRows;
-        }
-        catch {
-            return 0;
-        }
-    }
-    async deleteBySession(sessionId) {
-        const prefix = `${escapeLike(`tr_${sessionId}_`)}%`;
-        const [res] = await this.pool.query("DELETE FROM tool_result WHERE ref LIKE ? ESCAPE '\\\\'", [prefix]);
-        return res.affectedRows;
-    }
-}
 //# sourceMappingURL=tidb-tool-result-store.js.map

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

@@ -1,21 +1,2 @@
-import type { Pool } from "mysql2/promise";
-import { type WorkflowJournalEntry, type WorkflowJournalStore } from "@sema-agent/core";
-export declare class TiDBWorkflowJournalStore implements WorkflowJournalStore {
-    private readonly pool;
-    constructor(pool: Pool);
-    append(runId: string, scope: string, entry: WorkflowJournalEntry): Promise<void>;
-    load(runId: string, scope: string): Promise<WorkflowJournalEntry[]>;
-    locator(runId: string, _scope: string): string;
-    loadPage(runId: string, scope: string, opts: {
-        offset: number;
-        limit: number;
-        maxResultBytes: number;
-    }): Promise<Array<{
-        callKey: string;
-        resultJson: string | null;
-        resultBytes: number;
-    }>>;
-    deleteByRun(runId: string): Promise<number>;
-    reapExpired(now: number, maxAgeMs: number): Promise<number>;
-}
+export { TiDBWorkflowJournalStore } from "./workflow-journal-store-sql.js";
 //# sourceMappingURL=tidb-workflow-journal-store.d.ts.map

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

@@ -1,35 +1,2 @@
-import { callKeyOrdinal } from "@sema-agent/core";
-import { oversizeJournalResult } from "./workflow-journal-limits.js";
-export class TiDBWorkflowJournalStore {
-    pool;
-    constructor(pool) {
-        this.pool = pool;
-    }
-    async append(runId, scope, entry) {
-        const serialized = JSON.stringify(entry.result);
-        if (oversizeJournalResult(serialized))
-            return;
-        await this.pool.query("INSERT INTO workflow_journal (run_id, ordinal, scope, call_key, result, created_at) VALUES (?,?,?,?,?,?) " +
-            "ON DUPLICATE KEY UPDATE call_key = VALUES(call_key), result = VALUES(result)", [runId, callKeyOrdinal(entry.callKey), scope, entry.callKey, serialized, Date.now()]);
-    }
-    async load(runId, scope) {
-        const [rows] = await this.pool.query("SELECT call_key, result FROM workflow_journal WHERE run_id = ? AND scope = ? ORDER BY ordinal ASC", [runId, scope]);
-        return rows.map((r) => ({ callKey: String(r.call_key), result: JSON.parse(String(r.result)) }));
-    }
-    locator(runId, _scope) {
-        return `/v1/workflows/${runId}/journal`;
-    }
-    async loadPage(runId, scope, opts) {
-        const [rows] = await this.pool.query("SELECT call_key, CASE WHEN LENGTH(result) > ? THEN NULL ELSE result END AS result_json, LENGTH(result) AS result_bytes FROM workflow_journal WHERE run_id = ? AND scope = ? ORDER BY ordinal ASC LIMIT ? OFFSET ?", [opts.maxResultBytes, runId, scope, opts.limit, opts.offset]);
-        return rows.map((r) => ({ callKey: String(r.call_key), resultJson: r.result_json === null ? null : String(r.result_json), resultBytes: Number(r.result_bytes) }));
-    }
-    async deleteByRun(runId) {
-        const [r] = await this.pool.query("DELETE FROM workflow_journal WHERE run_id = ?", [runId]);
-        return r.affectedRows;
-    }
-    async reapExpired(now, maxAgeMs) {
-        const [r] = await this.pool.query("DELETE FROM workflow_journal WHERE created_at < ?", [now - maxAgeMs]);
-        return r.affectedRows;
-    }
-}
+export { TiDBWorkflowJournalStore } from "./workflow-journal-store-sql.js";
 //# sourceMappingURL=tidb-workflow-journal-store.js.map