@sixb/sqlite 0.1.1 → 0.1.3

package/src/agents/rows.ts

@@ -1,10 +1,12 @@
 import type { Database } from "bun:sqlite"
-import { type AgentMessagePart, type Principal, SYSTEM_PRINCIPAL } from "@sixb/core"
+import type { AgentMessagePart, Principal } from "@sixb/core"
+import { parseSixbFailure } from "@sixb/core/internal/errors"
 import {
+  AGENT_RUN_FAILURE_CODES,
+  type AgentContextCheckpointRecord,
   type AgentMessageRecord,
   type AgentRunDiagnostic,
   type AgentRunRecord,
-  type AgentRunUsage,
   type AgentThreadRecord,
   coerceAgentRunFinishReason,
 } from "@sixb/core/storage"
@@ -30,21 +32,14 @@ export interface AgentThreadRow {
 export interface AgentRunRow {
   project_id: string
   id: string
+  execution_id: string
   thread_id: string
   agent_id: string
   trigger_message_id: string
-  requested_by_principal_type: Principal["type"] | null
-  requested_by_principal_id: string | null
-  execution_principal_type: "serviceAccount" | null
-  execution_principal_id: string | null
+  requester_group_ids: string
   status: AgentRunRecord["status"]
   model_id: string | null
   finish_reason: string | null
-  usage_input_tokens: number | null
-  usage_output_tokens: number | null
-  usage_total_tokens: number | null
-  usage_reasoning_tokens: number | null
-  usage_cached_input_tokens: number | null
   error: string | null
   diagnostics: string | null
   attempt: number
@@ -71,6 +66,23 @@ export interface AgentMessageRow {
   completed_at: string | null
 }
 
+export interface AgentContextCheckpointRow {
+  project_id: string
+  id: string
+  thread_id: string
+  created_by_run_id: string
+  previous_checkpoint_id: string | null
+  reason: AgentContextCheckpointRecord["reason"]
+  summary: string
+  summary_format_version: number
+  summarized_through_seq: number
+  observed_head_seq: number
+  estimated_input_tokens_before: number
+  estimated_input_tokens_after: number
+  summary_model_id: string
+  created_at: string
+}
+
 // ── row → record mappers ────────────────────────────────────────────────────────────────────────
 
 export function rowToThreadRecord(row: AgentThreadRow): AgentThreadRecord {
@@ -93,21 +105,15 @@ export function rowToRunRecord(row: AgentRunRow): AgentRunRecord {
   return {
     id: row.id,
     projectId: row.project_id,
+    executionId: row.execution_id,
     threadId: row.thread_id,
     agentId: row.agent_id,
     triggerMessageId: row.trigger_message_id,
-    requestedByPrincipal:
-      principalFromColumns(row.requested_by_principal_type, row.requested_by_principal_id) ??
-      SYSTEM_PRINCIPAL,
-    executionPrincipal: serviceAccountPrincipalFromColumns(
-      row.execution_principal_type,
-      row.execution_principal_id
-    ),
+    requesterGroupIds: JSON.parse(row.requester_group_ids) as string[],
     status: row.status,
     modelId: row.model_id ?? undefined,
     finishReason: coerceAgentRunFinishReason(row.finish_reason),
-    usage: rowToUsage(row),
-    error: row.error ?? undefined,
+    error: row.error === null ? undefined : parseSixbFailure(row.error, AGENT_RUN_FAILURE_CODES),
     diagnostics:
       row.diagnostics === null ? undefined : (JSON.parse(row.diagnostics) as AgentRunDiagnostic[]),
     attempt: row.attempt,
@@ -143,6 +149,32 @@ export function rowToMessageRecord(row: AgentMessageRow): AgentMessageRecord {
   }
 }
 
+export function rowToContextCheckpointRecord(
+  row: AgentContextCheckpointRow
+): AgentContextCheckpointRecord {
+  if (row.summary_format_version !== 1) {
+    throw new Error(
+      `[SixbSqlite] Unsupported agent context checkpoint summary format version '${row.summary_format_version}'.`
+    )
+  }
+  return {
+    id: row.id,
+    projectId: row.project_id,
+    threadId: row.thread_id,
+    createdByRunId: row.created_by_run_id,
+    previousCheckpointId: row.previous_checkpoint_id ?? undefined,
+    reason: row.reason,
+    summary: row.summary,
+    summaryFormatVersion: 1,
+    summarizedThroughSeq: row.summarized_through_seq,
+    observedHeadSeq: row.observed_head_seq,
+    estimatedInputTokensBefore: row.estimated_input_tokens_before,
+    estimatedInputTokensAfter: row.estimated_input_tokens_after,
+    summaryModelId: row.summary_model_id,
+    createdAt: new Date(row.created_at),
+  }
+}
+
 function principalFromColumns(
   type: Principal["type"] | null | undefined,
   id: string | null | undefined
@@ -150,26 +182,6 @@ function principalFromColumns(
   return type && id ? { type, id } : undefined
 }
 
-function serviceAccountPrincipalFromColumns(
-  type: "serviceAccount" | null | undefined,
-  id: string | null | undefined
-): Extract<Principal, { readonly type: "serviceAccount" }> | undefined {
-  return type === "serviceAccount" && id ? { type, id } : undefined
-}
-
-function rowToUsage(row: AgentRunRow): AgentRunUsage | undefined {
-  const usage: AgentRunUsage = {
-    ...(row.usage_input_tokens === null ? {} : { inputTokens: row.usage_input_tokens }),
-    ...(row.usage_output_tokens === null ? {} : { outputTokens: row.usage_output_tokens }),
-    ...(row.usage_total_tokens === null ? {} : { totalTokens: row.usage_total_tokens }),
-    ...(row.usage_reasoning_tokens === null ? {} : { reasoningTokens: row.usage_reasoning_tokens }),
-    ...(row.usage_cached_input_tokens === null
-      ? {}
-      : { cachedInputTokens: row.usage_cached_input_tokens }),
-  }
-  return Object.keys(usage).length === 0 ? undefined : usage
-}
-
 // ── offset pagination (threads / messages order by their own column, not started_at) ─────────────
 
 export function queryAgentList<TRow>(input: {

package/src/agents/runs.ts

@@ -1,10 +1,15 @@
 import type { Database } from "bun:sqlite"
+import { assertAgentRunExecution } from "@sixb/core/internal/agent-run-storage-provider"
+import { normalizeRequesterGroupIds } from "@sixb/core/internal/auth"
+import { serializeSixbFailure } from "@sixb/core/internal/errors"
 import {
+  AGENT_RUN_FAILURE_CODES,
   type AgentRunRecord,
   type AgentRunStore,
   AgentStorageError,
   type ConfirmAgentRunExecutionOwnershipInput,
   type CreateAgentRunInput,
+  type ExecutionStorage,
   type FinishAgentRunInput,
   type FinishQueuedAgentRunInput,
   type ListAgentRunsInput,
@@ -24,10 +29,20 @@ import { type AgentRunRow, rowToRunRecord } from "./rows"
 const SQLITE_RUN_ID_BATCH_SIZE = 500
 
 export class SqliteAgentRunStore implements AgentRunStore {
-  constructor(private readonly db: Database) {}
+  constructor(
+    private readonly db: Database,
+    private readonly executions: ExecutionStorage
+  ) {}
 
   async create(input: CreateAgentRunInput): Promise<AgentRunRecord> {
     const createdAt = input.createdAt ?? new Date()
+    await assertAgentRunExecution({
+      executions: this.executions,
+      projectId: input.projectId,
+      executionId: input.executionId,
+      runId: input.id,
+      agentId: input.agentId,
+    })
 
     return this.db.transaction(() => {
       const thread = this.db
@@ -55,11 +70,11 @@ export class SqliteAgentRunStore implements AgentRunStore {
             INSERT INTO agent_runs (
               project_id,
               id,
+              execution_id,
               thread_id,
               agent_id,
               trigger_message_id,
-              requested_by_principal_type,
-              requested_by_principal_id,
+              requester_group_ids,
               status,
               attempt,
               created_at
@@ -69,11 +84,11 @@ export class SqliteAgentRunStore implements AgentRunStore {
           .run(
             input.projectId,
             input.id,
+            input.executionId,
             input.threadId,
             input.agentId,
             input.triggerMessageId,
-            input.requestedByPrincipal.type,
-            input.requestedByPrincipal.id,
+            JSON.stringify(normalizeRequesterGroupIds(input.requesterGroupIds)),
             createdAt.toISOString()
           )
       } catch (error) {
@@ -107,8 +122,6 @@ export class SqliteAgentRunStore implements AgentRunStore {
           UPDATE agent_runs
           SET
             status = 'running',
-            execution_principal_type = ?,
-            execution_principal_id = ?,
             model_id = ?,
             attempt = 1,
             execution_token = ?,
@@ -118,8 +131,6 @@ export class SqliteAgentRunStore implements AgentRunStore {
         `
         )
         .run(
-          input.executionPrincipal?.type ?? null,
-          input.executionPrincipal?.id ?? null,
           input.modelId ?? null,
           input.execution.token,
           input.execution.queueLeaseExpiresAt.toISOString(),
@@ -146,7 +157,9 @@ export class SqliteAgentRunStore implements AgentRunStore {
         )
         .run(
           input.status,
-          input.error ?? null,
+          input.error === undefined
+            ? null
+            : serializeSixbFailure(input.error, AGENT_RUN_FAILURE_CODES),
           completedAt.toISOString(),
           input.projectId,
           input.id
@@ -208,7 +221,10 @@ export class SqliteAgentRunStore implements AgentRunStore {
 
   async finish(input: FinishAgentRunInput): Promise<AgentRunRecord> {
     const completedAt = input.completedAt ?? new Date()
-    const errorValue = input.status === "succeeded" ? null : (input.error ?? null)
+    const errorValue =
+      input.status === "succeeded" || input.error === undefined
+        ? null
+        : serializeSixbFailure(input.error, AGENT_RUN_FAILURE_CODES)
 
     return this.db.transaction(() => {
       const run = this.requireRunning(input.projectId, input.id)
@@ -228,11 +244,6 @@ export class SqliteAgentRunStore implements AgentRunStore {
             status = ?,
             model_id = COALESCE(?, model_id),
             finish_reason = ?,
-            usage_input_tokens = ?,
-            usage_output_tokens = ?,
-            usage_total_tokens = ?,
-            usage_reasoning_tokens = ?,
-            usage_cached_input_tokens = ?,
             error = ?,
             diagnostics = ?,
             execution_token = NULL,
@@ -245,11 +256,6 @@ export class SqliteAgentRunStore implements AgentRunStore {
           input.status,
           input.modelId ?? null,
           input.finishReason ?? null,
-          input.usage?.inputTokens ?? null,
-          input.usage?.outputTokens ?? null,
-          input.usage?.totalTokens ?? null,
-          input.usage?.reasoningTokens ?? null,
-          input.usage?.cachedInputTokens ?? null,
           errorValue,
           input.diagnostics === undefined ? null : JSON.stringify(input.diagnostics),
           completedAt.toISOString(),