@sixb/sqlite 0.1.1 → 0.1.3

package/src/action-run-storage.ts

@@ -1,5 +1,10 @@
 import type { Database } from "bun:sqlite"
 import type { ActionSubject, JsonValue } from "@sixb/core"
+import {
+  parseActionRunFailure,
+  serializeActionRunFailure,
+} from "@sixb/core/internal/action-run-storage"
+import { assertActionRunExecution } from "@sixb/core/internal/action-run-storage-provider"
 import type {
   ActionRunEffectsRecord,
   ActionRunFailure,
@@ -9,6 +14,7 @@ import type {
   ActionRunStorage,
   ActionRunWritebackRecord,
   EnterActionRunPhaseInput,
+  ExecutionStorage,
   FinishActionRunInput,
   ListActionRunsInput,
   ListActionRunsResult,
@@ -33,6 +39,8 @@ import {
 } from "./transactions"
 
 export interface SqliteActionRunStorageOptions {
+  /** Execution lookup sharing the same provider transaction. */
+  executions: ExecutionStorage
   /** Path to SQLite database file. Defaults to ':memory:' for in-memory database. */
   path?: string
   /** Internal shared connection used by bundled SqliteStorage. */
@@ -43,9 +51,12 @@ export class SqliteActionRunStorage implements ActionRunStorage {
   private readonly connection: SqliteStoreConnection
   private readonly db: Database
 
-  constructor(options: SqliteActionRunStorageOptions = {}) {
+  private readonly executions: ExecutionStorage
+
+  constructor(options: SqliteActionRunStorageOptions) {
     this.connection = openSqliteStoreConnection(options)
     this.db = this.connection.db
+    this.executions = options.executions
 
     if (this.connection.installFreshSchema) {
       installFreshSqliteSchema(this.db)
@@ -85,6 +96,13 @@ export class SqliteActionRunStorage implements ActionRunStorage {
 
   async queue(input: QueueActionRunInput): Promise<ActionRunRecord> {
     const queuedAt = input.queuedAt ?? new Date()
+    await assertActionRunExecution({
+      executions: this.executions,
+      projectId: input.projectId,
+      executionId: input.executionId,
+      runId: input.id,
+      actionId: input.actionId,
+    })
 
     try {
       this.db
@@ -93,6 +111,7 @@ export class SqliteActionRunStorage implements ActionRunStorage {
           INSERT INTO action_runs (
             project_id,
             id,
+            execution_id,
             action_id,
             subject_kind,
             object_type_id,
@@ -107,28 +126,23 @@ export class SqliteActionRunStorage implements ActionRunStorage {
             writeback_status,
             writeback_completed_at,
             writeback_result,
-            writeback_error_name,
-            writeback_error_message,
-            writeback_error_phase,
+            writeback_error,
             effects_status,
             effects_completed_at,
-            effects_error_name,
-            effects_error_message,
-            effects_error_phase,
-            error_name,
-            error_message,
-            error_phase
+            effects_error,
+            error
           ) VALUES (
-            ?, ?, ?, ?, ?, ?, ?, ?, ?, NULL, NULL, ?, ?,
-            NULL, NULL, NULL, NULL, NULL, NULL,
-            NULL, NULL, NULL, NULL, NULL,
-            NULL, NULL, NULL
+            ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, NULL, NULL, ?, ?,
+            NULL, NULL, NULL, NULL,
+            NULL, NULL, NULL,
+            NULL
           )
         `
         )
         .run(
           input.projectId,
           input.id,
+          input.executionId,
           input.actionId,
           input.subject.kind,
           input.subject.kind === "object" ? input.subject.objectTypeId : null,
@@ -188,17 +202,11 @@ export class SqliteActionRunStorage implements ActionRunStorage {
             writeback_status = NULL,
             writeback_completed_at = NULL,
             writeback_result = NULL,
-            writeback_error_name = NULL,
-            writeback_error_message = NULL,
-            writeback_error_phase = NULL,
+            writeback_error = NULL,
             effects_status = NULL,
             effects_completed_at = NULL,
-            effects_error_name = NULL,
-            effects_error_message = NULL,
-            effects_error_phase = NULL,
-            error_name = NULL,
-            error_message = NULL,
-            error_phase = NULL
+            effects_error = NULL,
+            error = NULL
           WHERE project_id = ? AND id = ?
         `
         )
@@ -238,9 +246,7 @@ export class SqliteActionRunStorage implements ActionRunStorage {
             status = ?,
             phase = ?,
             started_at = ?,
-            error_name = NULL,
-            error_message = NULL,
-            error_phase = NULL
+            error = NULL
           WHERE project_id = ? AND id = ?
         `
         )
@@ -307,9 +313,7 @@ export class SqliteActionRunStorage implements ActionRunStorage {
             writeback_status = ?,
             writeback_completed_at = ?,
             writeback_result = ?,
-            writeback_error_name = ?,
-            writeback_error_message = ?,
-            writeback_error_phase = ?
+            writeback_error = ?
           WHERE project_id = ? AND id = ?
         `
         )
@@ -318,9 +322,7 @@ export class SqliteActionRunStorage implements ActionRunStorage {
           input.status,
           nextWriteback.completedAt.toISOString(),
           input.status === "succeeded" ? serializeJsonValue(input.result) : null,
-          input.status === "failed" ? (input.error.name ?? null) : null,
-          input.status === "failed" ? input.error.message : null,
-          input.status === "failed" ? (input.error.phase ?? "writeback") : null,
+          input.status === "failed" ? serializeActionRunFailure(input.error, "writeback") : null,
           input.projectId,
           input.id
         )
@@ -357,9 +359,7 @@ export class SqliteActionRunStorage implements ActionRunStorage {
             phase = ?,
             effects_status = ?,
             effects_completed_at = ?,
-            effects_error_name = ?,
-            effects_error_message = ?,
-            effects_error_phase = ?
+            effects_error = ?
           WHERE project_id = ? AND id = ?
         `
         )
@@ -367,9 +367,7 @@ export class SqliteActionRunStorage implements ActionRunStorage {
           "effects",
           input.status,
           nextEffects.completedAt.toISOString(),
-          input.status === "failed" ? (input.error.name ?? null) : null,
-          input.status === "failed" ? input.error.message : null,
-          input.status === "failed" ? (input.error.phase ?? "effects") : null,
+          input.status === "failed" ? serializeActionRunFailure(input.error, "effects") : null,
           input.projectId,
           input.id
         )
@@ -410,9 +408,7 @@ export class SqliteActionRunStorage implements ActionRunStorage {
             status = ?,
             phase = ?,
             finished_at = ?,
-            error_name = ?,
-            error_message = ?,
-            error_phase = ?
+            error = ?
           WHERE project_id = ? AND id = ?
         `
         )
@@ -420,9 +416,7 @@ export class SqliteActionRunStorage implements ActionRunStorage {
           input.status,
           phase,
           (input.finishedAt ?? new Date()).toISOString(),
-          input.status === "succeeded" ? null : (input.error?.name ?? null),
-          input.status === "succeeded" ? null : (input.error?.message ?? null),
-          input.status === "succeeded" ? null : (input.error?.phase ?? phase),
+          input.status === "succeeded" ? null : serializeActionRunFailure(input.error),
           input.projectId,
           input.id
         )
@@ -583,23 +577,14 @@ function serializeJsonValue(value: JsonValue): string {
 }
 
 function toActionRunFailure(row: DatabaseRow): ActionRunFailure | undefined {
-  return toFailure(row.error_name, row.error_message, row.error_phase)
+  return row.error === null ? undefined : parseActionRunFailure(row.error)
 }
 
-function toFailure(
-  name: string | null,
-  message: string | null,
-  phase: ActionRunPhase | null
-): ActionRunFailure | undefined {
-  if (!message) {
-    return undefined
-  }
-
-  return {
-    name: name ?? undefined,
-    message,
-    phase: phase ?? undefined,
-  }
+function parsePhaseFailure<TPhase extends Extract<ActionRunPhase, "writeback" | "effects">>(
+  value: string | null,
+  expectedPhase: TPhase
+): ActionRunFailure<TPhase> {
+  return parseActionRunFailure(value, expectedPhase)
 }
 
 function toActionRunWritebackRecord(row: DatabaseRow): ActionRunWritebackRecord | undefined {
@@ -623,11 +608,7 @@ function toActionRunWritebackRecord(row: DatabaseRow): ActionRunWritebackRecord
   return {
     status: "failed",
     completedAt,
-    error: toFailure(
-      row.writeback_error_name,
-      row.writeback_error_message,
-      row.writeback_error_phase
-    ),
+    error: parsePhaseFailure(row.writeback_error, "writeback"),
   }
 }
 
@@ -650,7 +631,7 @@ function toActionRunEffectsRecord(row: DatabaseRow): ActionRunEffectsRecord | un
   return {
     status: "failed",
     completedAt,
-    error: toFailure(row.effects_error_name, row.effects_error_message, row.effects_error_phase),
+    error: parsePhaseFailure(row.effects_error, "effects"),
   }
 }
 
@@ -695,6 +676,7 @@ function rowToActionRunRecord(row: DatabaseRow): ActionRunRecord {
   return {
     id: row.id,
     projectId: row.project_id,
+    executionId: row.execution_id,
     actionId: row.action_id,
     subject: rowToActionSubject(row),
     status: row.status,
@@ -733,6 +715,7 @@ function isUniqueConstraintError(error: unknown): boolean {
 interface DatabaseRow {
   project_id: string
   id: string
+  execution_id: string
   action_id: string
   subject_kind: ActionSubject["kind"]
   object_type_id: string | null
@@ -747,15 +730,9 @@ interface DatabaseRow {
   writeback_status: ActionRunWritebackRecord["status"] | null
   writeback_completed_at: string | null
   writeback_result: string | null
-  writeback_error_name: string | null
-  writeback_error_message: string | null
-  writeback_error_phase: ActionRunPhase | null
+  writeback_error: string | null
   effects_status: ActionRunEffectsRecord["status"] | null
   effects_completed_at: string | null
-  effects_error_name: string | null
-  effects_error_message: string | null
-  effects_error_phase: ActionRunPhase | null
-  error_name: string | null
-  error_message: string | null
-  error_phase: ActionRunPhase | null
+  effects_error: string | null
+  error: string | null
 }

package/src/agents/agent-storage.ts

@@ -1,10 +1,11 @@
-import type { AgentStorage } from "@sixb/core/storage"
+import type { AgentStorage, ExecutionStorage } from "@sixb/core/storage"
 import { installFreshSqliteSchema } from "../migrations"
 import {
   closeSqliteStoreConnection,
   openSqliteStoreConnection,
   type SqliteStoreConnection,
 } from "../transactions"
+import { SqliteAgentContextCheckpointStore } from "./checkpoints"
 import { SqliteAgentMessageStore } from "./messages"
 import { SqliteAgentRunStore } from "./runs"
 import { SqliteAgentThreadStore } from "./threads"
@@ -14,21 +15,24 @@ export interface SqliteAgentStorageOptions {
   path?: string
   /** Internal shared connection used by bundled SqliteStorage. */
   connection?: SqliteStoreConnection
+  /** Execution ledger used to validate immutable run ownership. */
+  executions: ExecutionStorage
 }
 
 /**
- * SQLite-backed agent persistence: thread / run / message sub-stores sharing one connection. The
- * cross-table writes (reserve + finish touch the thread anchor, append bumps thread stats) run in
- * SQLite transactions; the bundled {@link SqliteStorage} serializes all of them on one connection.
+ * SQLite-backed agent persistence: thread, run, message, and context-checkpoint sub-stores sharing
+ * one connection. Cross-table writes run in SQLite transactions; the bundled {@link SqliteStorage}
+ * serializes them for single-flight and checkpoint compare-and-swap safety.
  */
 export class SqliteAgentStorage implements AgentStorage {
   readonly threads: SqliteAgentThreadStore
   readonly runs: SqliteAgentRunStore
   readonly messages: SqliteAgentMessageStore
+  readonly checkpoints: SqliteAgentContextCheckpointStore
 
   private readonly connection: SqliteStoreConnection
 
-  constructor(options: SqliteAgentStorageOptions = {}) {
+  constructor(options: SqliteAgentStorageOptions) {
     this.connection = openSqliteStoreConnection(options)
 
     if (this.connection.installFreshSchema) {
@@ -36,8 +40,9 @@ export class SqliteAgentStorage implements AgentStorage {
     }
 
     this.threads = new SqliteAgentThreadStore(this.connection.db)
-    this.runs = new SqliteAgentRunStore(this.connection.db)
+    this.runs = new SqliteAgentRunStore(this.connection.db, options.executions)
     this.messages = new SqliteAgentMessageStore(this.connection.db)
+    this.checkpoints = new SqliteAgentContextCheckpointStore(this.connection.db)
   }
 
   close(): void {

package/src/agents/checkpoints.ts

@@ -0,0 +1,175 @@
+import type { Database } from "bun:sqlite"
+import {
+  agentContextCheckpointMatchesCreateInput,
+  assertAgentContextCheckpointAnchors,
+  assertAgentContextCheckpointAuthority,
+  assertAgentContextCheckpointReplayState,
+  assertCreateAgentContextCheckpointInput,
+} from "@sixb/core/internal/agent-run-storage-provider"
+import {
+  type AgentContextCheckpointRecord,
+  type AgentContextCheckpointStore,
+  AgentStorageError,
+  type CreateAgentContextCheckpointInput,
+} from "@sixb/core/storage"
+import {
+  type AgentContextCheckpointRow,
+  type AgentMessageRow,
+  type AgentRunRow,
+  type AgentThreadRow,
+  rowToContextCheckpointRecord,
+  rowToMessageRecord,
+  rowToRunRecord,
+  rowToThreadRecord,
+} from "./rows"
+
+export class SqliteAgentContextCheckpointStore implements AgentContextCheckpointStore {
+  constructor(private readonly db: Database) {}
+
+  async create(input: CreateAgentContextCheckpointInput): Promise<AgentContextCheckpointRecord> {
+    assertCreateAgentContextCheckpointInput(input, "SixbSqlite")
+
+    return this.db.transaction(() => {
+      const runRow = this.db
+        .query("SELECT * FROM agent_runs WHERE project_id = ? AND id = ?")
+        .get(input.projectId, input.createdByRunId) as AgentRunRow | null
+      const threadRow = this.db
+        .query("SELECT * FROM agent_threads WHERE project_id = ? AND id = ?")
+        .get(input.projectId, input.threadId) as AgentThreadRow | null
+      assertAgentContextCheckpointAuthority({
+        create: input,
+        run: runRow ? rowToRunRecord(runRow) : null,
+        thread: threadRow ? rowToThreadRecord(threadRow) : null,
+        prefix: "SixbSqlite",
+      })
+
+      const latest = this.findLatest(input.projectId, input.threadId)
+      const headRow = this.db
+        .query(
+          "SELECT COALESCE(MAX(seq), 0) AS seq FROM agent_messages WHERE project_id = ? AND thread_id = ?"
+        )
+        .get(input.projectId, input.threadId) as { seq: number }
+
+      const existingForRun = this.db
+        .query(
+          "SELECT * FROM agent_context_checkpoints WHERE project_id = ? AND created_by_run_id = ?"
+        )
+        .get(input.projectId, input.createdByRunId) as AgentContextCheckpointRow | null
+      if (existingForRun) {
+        const record = rowToContextCheckpointRecord(existingForRun)
+        if (agentContextCheckpointMatchesCreateInput(record, input)) {
+          assertAgentContextCheckpointReplayState({
+            create: input,
+            existing: record,
+            latest,
+            headSeq: headRow.seq,
+            prefix: "SixbSqlite",
+          })
+          return record
+        }
+        throw new AgentStorageError(
+          "invalid_state",
+          `[SixbSqlite] Agent run '${input.createdByRunId}' already created a different context checkpoint.`
+        )
+      }
+
+      const duplicateId = this.db
+        .query("SELECT 1 FROM agent_context_checkpoints WHERE project_id = ? AND id = ?")
+        .get(input.projectId, input.id)
+      if (duplicateId) {
+        throw new AgentStorageError(
+          "duplicate_id",
+          `[SixbSqlite] Agent context checkpoint '${input.id}' already exists for project '${input.projectId}'.`
+        )
+      }
+
+      const firstRetainedRow = this.db
+        .query(
+          "SELECT * FROM agent_messages WHERE project_id = ? AND thread_id = ? AND seq > ? ORDER BY seq ASC LIMIT 1"
+        )
+        .get(input.projectId, input.threadId, input.summarizedThroughSeq) as AgentMessageRow | null
+      assertAgentContextCheckpointAnchors({
+        create: input,
+        latest,
+        headSeq: headRow.seq,
+        firstRetained: firstRetainedRow ? rowToMessageRecord(firstRetainedRow) : null,
+        prefix: "SixbSqlite",
+      })
+
+      const createdAt = input.createdAt ?? new Date()
+      this.db
+        .query(
+          `
+            INSERT INTO agent_context_checkpoints (
+              project_id,
+              id,
+              thread_id,
+              created_by_run_id,
+              previous_checkpoint_id,
+              reason,
+              summary,
+              summary_format_version,
+              summarized_through_seq,
+              observed_head_seq,
+              estimated_input_tokens_before,
+              estimated_input_tokens_after,
+              summary_model_id,
+              created_at
+            ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+          `
+        )
+        .run(
+          input.projectId,
+          input.id,
+          input.threadId,
+          input.createdByRunId,
+          input.expectedPreviousCheckpointId,
+          input.reason,
+          input.summary,
+          input.summaryFormatVersion,
+          input.summarizedThroughSeq,
+          input.observedHeadSeq,
+          input.estimatedInputTokensBefore,
+          input.estimatedInputTokensAfter,
+          input.summaryModelId,
+          createdAt.toISOString()
+        )
+
+      return this.requireById(input.projectId, input.id)
+    })()
+  }
+
+  async getLatest(input: {
+    readonly projectId: string
+    readonly threadId: string
+  }): Promise<AgentContextCheckpointRecord | null> {
+    return this.findLatest(input.projectId, input.threadId)
+  }
+
+  private findLatest(projectId: string, threadId: string): AgentContextCheckpointRecord | null {
+    const row = this.db
+      .query(
+        `
+          SELECT * FROM agent_context_checkpoints
+          WHERE project_id = ? AND thread_id = ?
+          ORDER BY summarized_through_seq DESC, created_at DESC, id DESC
+          LIMIT 1
+        `
+      )
+      .get(projectId, threadId) as AgentContextCheckpointRow | null
+    return row ? rowToContextCheckpointRecord(row) : null
+  }
+
+  private requireById(projectId: string, id: string): AgentContextCheckpointRecord {
+    const row = this.db
+      .query("SELECT * FROM agent_context_checkpoints WHERE project_id = ? AND id = ?")
+      .get(projectId, id) as AgentContextCheckpointRow | null
+    if (!row) {
+      throw new AgentStorageError(
+        "invalid_state",
+        `[SixbSqlite] Failed to load agent context checkpoint '${id}' for project '${projectId}'.`
+      )
+    }
+    return rowToContextCheckpointRecord(row)
+  }
+}

package/src/agents/index.ts

@@ -1,5 +1,6 @@
 export type { SqliteAgentStorageOptions } from "./agent-storage"
 export { SqliteAgentStorage } from "./agent-storage"
+export { SqliteAgentContextCheckpointStore } from "./checkpoints"
 export { SqliteAgentMessageStore } from "./messages"
 export { SqliteAgentRunStore } from "./runs"
 export { SqliteAgentThreadStore } from "./threads"

package/src/agents/messages.ts

@@ -5,6 +5,7 @@ import {
   type AgentMessageStore,
   AgentStorageError,
   type AppendAgentMessageInput,
+  type DeleteAgentMessagesByRunInput,
   type ListAgentMessagesInput,
   type ListAgentMessagesResult,
 } from "@sixb/core/storage"
@@ -110,6 +111,59 @@ export class SqliteAgentMessageStore implements AgentMessageStore {
     })()
   }
 
+  async deleteByRunId(input: DeleteAgentMessagesByRunInput): Promise<number> {
+    return this.db.transaction(() => {
+      const thread = this.db
+        .query("SELECT 1 AS present FROM agent_threads WHERE project_id = ? AND id = ?")
+        .get(input.projectId, input.threadId) as { present: number } | null
+
+      if (!thread) {
+        throw new AgentStorageError(
+          "thread_not_found",
+          `[SixbSqlite] Agent thread '${input.threadId}' not found for project '${input.projectId}'.`
+        )
+      }
+
+      const deleted = this.db
+        .query("DELETE FROM agent_messages WHERE project_id = ? AND thread_id = ? AND run_id = ?")
+        .run(input.projectId, input.threadId, input.runId)
+      if (deleted.changes === 0) {
+        return 0
+      }
+
+      const updatedAt = new Date().toISOString()
+      this.db
+        .query(
+          `
+          UPDATE agent_threads
+          SET
+            message_count = (
+              SELECT COUNT(*) FROM agent_messages WHERE project_id = ? AND thread_id = ?
+            ),
+            last_message_at = (
+              SELECT created_at FROM agent_messages
+              WHERE project_id = ? AND thread_id = ?
+              ORDER BY seq DESC
+              LIMIT 1
+            ),
+            updated_at = ?
+          WHERE project_id = ? AND id = ?
+        `
+        )
+        .run(
+          input.projectId,
+          input.threadId,
+          input.projectId,
+          input.threadId,
+          updatedAt,
+          input.projectId,
+          input.threadId
+        )
+
+      return deleted.changes
+    })()
+  }
+
   async getById(params: { projectId: string; id: string }): Promise<AgentMessageRecord | null> {
     const row = this.db
       .query("SELECT * FROM agent_messages WHERE project_id = ? AND id = ?")
@@ -126,6 +180,11 @@ export class SqliteAgentMessageStore implements AgentMessageStore {
     const whereClauses = ["project_id = ?", "thread_id = ?"]
     const args: SqliteValue[] = [input.projectId, input.threadId]
 
+    if (input.afterSeq !== undefined) {
+      whereClauses.push("seq > ?")
+      args.push(input.afterSeq)
+    }
+
     if (input.roles) {
       whereClauses.push(`role IN (${input.roles.map(() => "?").join(", ")})`)
       args.push(...input.roles)