@sixb/sqlite 0.1.3 → 0.1.5

package/dist/agents/checkpoints.d.ts

@@ -8,6 +8,10 @@ export declare class SqliteAgentContextCheckpointStore implements AgentContextCh
         readonly projectId: string;
         readonly threadId: string;
     }): Promise<AgentContextCheckpointRecord | null>;
+    getByRunIds(input: {
+        readonly projectId: string;
+        readonly runIds: readonly string[];
+    }): Promise<readonly AgentContextCheckpointRecord[]>;
     private findLatest;
     private requireById;
 }

package/dist/agents/checkpoints.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"checkpoints.d.ts","sourceRoot":"","sources":["../../src/agents/checkpoints.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAA;AAQ1C,OAAO,EACL,KAAK,4BAA4B,EACjC,KAAK,2BAA2B,EAEhC,KAAK,iCAAiC,EACvC,MAAM,oBAAoB,CAAA;AAY3B,qBAAa,iCAAkC,YAAW,2BAA2B;IACvE,OAAO,CAAC,QAAQ,CAAC,EAAE;gBAAF,EAAE,EAAE,QAAQ;IAEnC,MAAM,CAAC,KAAK,EAAE,iCAAiC,GAAG,OAAO,CAAC,4BAA4B,CAAC;IAiHvF,SAAS,CAAC,KAAK,EAAE;QACrB,QAAQ,CAAC,SAAS,EAAE,MAAM,CAAA;QAC1B,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAA;KAC1B,GAAG,OAAO,CAAC,4BAA4B,GAAG,IAAI,CAAC;IAIhD,OAAO,CAAC,UAAU;IAclB,OAAO,CAAC,WAAW;CAYpB"}
+{"version":3,"file":"checkpoints.d.ts","sourceRoot":"","sources":["../../src/agents/checkpoints.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAA;AAQ1C,OAAO,EACL,KAAK,4BAA4B,EACjC,KAAK,2BAA2B,EAEhC,KAAK,iCAAiC,EACvC,MAAM,oBAAoB,CAAA;AAc3B,qBAAa,iCAAkC,YAAW,2BAA2B;IACvE,OAAO,CAAC,QAAQ,CAAC,EAAE;gBAAF,EAAE,EAAE,QAAQ;IAEnC,MAAM,CAAC,KAAK,EAAE,iCAAiC,GAAG,OAAO,CAAC,4BAA4B,CAAC;IAiHvF,SAAS,CAAC,KAAK,EAAE;QACrB,QAAQ,CAAC,SAAS,EAAE,MAAM,CAAA;QAC1B,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAA;KAC1B,GAAG,OAAO,CAAC,4BAA4B,GAAG,IAAI,CAAC;IAI1C,WAAW,CAAC,KAAK,EAAE;QACvB,QAAQ,CAAC,SAAS,EAAE,MAAM,CAAA;QAC1B,QAAQ,CAAC,MAAM,EAAE,SAAS,MAAM,EAAE,CAAA;KACnC,GAAG,OAAO,CAAC,SAAS,4BAA4B,EAAE,CAAC;IAyBpD,OAAO,CAAC,UAAU;IAclB,OAAO,CAAC,WAAW;CAYpB"}

package/dist/ai-usage-storage.d.ts

@@ -1,4 +1,4 @@
-import { type AiUsageExecutionSummary, type AiUsageStorage, type RecordAiModelCallInput, type RecordAiModelCallResult, type SummarizeAiUsageExecutionInput, type SummarizeAiUsageExecutionsInput } from "@sixb/core/storage";
+import { type AiModelCallUsageRecord, type AiUsageExecutionSummary, type AiUsageStorage, type GetLatestAiModelCallForExecutionInput, type RecordAiModelCallInput, type RecordAiModelCallResult, type SummarizeAiUsageExecutionInput, type SummarizeAiUsageExecutionsInput } from "@sixb/core/storage";
 import { type SqliteStoreConnection } from "./transactions";
 export interface SqliteAiUsageStorageOptions {
     readonly path?: string;
@@ -10,6 +10,7 @@ export declare class SqliteAiUsageStorage implements AiUsageStorage {
     constructor(options?: SqliteAiUsageStorageOptions);
     recordModelCall(input: RecordAiModelCallInput): Promise<RecordAiModelCallResult>;
     summarizeExecution(input: SummarizeAiUsageExecutionInput): Promise<AiUsageExecutionSummary>;
+    getLatestForExecution(input: GetLatestAiModelCallForExecutionInput): Promise<AiModelCallUsageRecord | null>;
     summarizeExecutions(input: SummarizeAiUsageExecutionsInput): Promise<readonly AiUsageExecutionSummary[]>;
     close(): void;
     private insertRecord;

package/dist/ai-usage-storage.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ai-usage-storage.d.ts","sourceRoot":"","sources":["../src/ai-usage-storage.ts"],"names":[],"mappings":"AACA,OAAO,EAGL,KAAK,uBAAuB,EAC5B,KAAK,cAAc,EAKnB,KAAK,sBAAsB,EAC3B,KAAK,uBAAuB,EAC5B,KAAK,8BAA8B,EACnC,KAAK,+BAA+B,EACrC,MAAM,oBAAoB,CAAA;AAG3B,OAAO,EAIL,KAAK,qBAAqB,EAC3B,MAAM,gBAAgB,CAAA;AAEvB,MAAM,WAAW,2BAA2B;IAC1C,QAAQ,CAAC,IAAI,CAAC,EAAE,MAAM,CAAA;IACtB,QAAQ,CAAC,UAAU,CAAC,EAAE,qBAAqB,CAAA;CAC5C;AAED,yDAAyD;AACzD,qBAAa,oBAAqB,YAAW,cAAc;IACzD,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAuB;gBAEtC,OAAO,GAAE,2BAAgC;IAO/C,eAAe,CAAC,KAAK,EAAE,sBAAsB,GAAG,OAAO,CAAC,uBAAuB,CAAC;IAoChF,kBAAkB,CACtB,KAAK,EAAE,8BAA8B,GACpC,OAAO,CAAC,uBAAuB,CAAC;IAQ7B,mBAAmB,CACvB,KAAK,EAAE,+BAA+B,GACrC,OAAO,CAAC,SAAS,uBAAuB,EAAE,CAAC;IAS9C,KAAK,IAAI,IAAI;IAIb,OAAO,CAAC,YAAY;IAwDpB,OAAO,CAAC,oBAAoB;IA2B5B,OAAO,CAAC,WAAW;IAQnB,OAAO,CAAC,WAAW;IAmCnB,OAAO,CAAC,aAAa;IAkBrB,OAAO,CAAC,qBAAqB;CAU9B"}
+{"version":3,"file":"ai-usage-storage.d.ts","sourceRoot":"","sources":["../src/ai-usage-storage.ts"],"names":[],"mappings":"AACA,OAAO,EAEL,KAAK,sBAAsB,EAC3B,KAAK,uBAAuB,EAC5B,KAAK,cAAc,EAInB,KAAK,qCAAqC,EAE1C,KAAK,sBAAsB,EAC3B,KAAK,uBAAuB,EAC5B,KAAK,8BAA8B,EACnC,KAAK,+BAA+B,EACrC,MAAM,oBAAoB,CAAA;AAG3B,OAAO,EAIL,KAAK,qBAAqB,EAC3B,MAAM,gBAAgB,CAAA;AAEvB,MAAM,WAAW,2BAA2B;IAC1C,QAAQ,CAAC,IAAI,CAAC,EAAE,MAAM,CAAA;IACtB,QAAQ,CAAC,UAAU,CAAC,EAAE,qBAAqB,CAAA;CAC5C;AAED,yDAAyD;AACzD,qBAAa,oBAAqB,YAAW,cAAc;IACzD,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAuB;gBAEtC,OAAO,GAAE,2BAAgC;IAO/C,eAAe,CAAC,KAAK,EAAE,sBAAsB,GAAG,OAAO,CAAC,uBAAuB,CAAC;IAoChF,kBAAkB,CACtB,KAAK,EAAE,8BAA8B,GACpC,OAAO,CAAC,uBAAuB,CAAC;IAQ7B,qBAAqB,CACzB,KAAK,EAAE,qCAAqC,GAC3C,OAAO,CAAC,sBAAsB,GAAG,IAAI,CAAC;IAgBnC,mBAAmB,CACvB,KAAK,EAAE,+BAA+B,GACrC,OAAO,CAAC,SAAS,uBAAuB,EAAE,CAAC;IAS9C,KAAK,IAAI,IAAI;IAIb,OAAO,CAAC,YAAY;IAwDpB,OAAO,CAAC,oBAAoB;IA2B5B,OAAO,CAAC,WAAW;IAQnB,OAAO,CAAC,WAAW;IAmCnB,OAAO,CAAC,aAAa;IAkBrB,OAAO,CAAC,qBAAqB;CAU9B"}

package/dist/index.js

@@ -3844,6 +3844,8 @@ function queryAgentList(input) {
 }
 
 // src/agents/checkpoints.ts
+var SQLITE_CHECKPOINT_RUN_ID_BATCH_SIZE = 500;
+
 class SqliteAgentContextCheckpointStore {
   db;
   constructor(db) {
@@ -3914,6 +3916,22 @@ class SqliteAgentContextCheckpointStore {
   async getLatest(input) {
     return this.findLatest(input.projectId, input.threadId);
   }
+  async getByRunIds(input) {
+    const runIds = [...new Set(input.runIds)];
+    if (runIds.length === 0)
+      return [];
+    const rows = [];
+    for (let offset = 0;offset < runIds.length; offset += SQLITE_CHECKPOINT_RUN_ID_BATCH_SIZE) {
+      const batch = runIds.slice(offset, offset + SQLITE_CHECKPOINT_RUN_ID_BATCH_SIZE);
+      const placeholders = batch.map(() => "?").join(", ");
+      rows.push(...this.db.query(`SELECT * FROM agent_context_checkpoints WHERE project_id = ? AND created_by_run_id IN (${placeholders})`).all(input.projectId, ...batch));
+    }
+    const byRunId = new Map(rows.map((row) => [row.created_by_run_id, rowToContextCheckpointRecord(row)]));
+    return runIds.flatMap((runId) => {
+      const checkpoint = byRunId.get(runId);
+      return checkpoint ? [checkpoint] : [];
+    });
+  }
   findLatest(projectId, threadId) {
     const row = this.db.query(`
           SELECT * FROM agent_context_checkpoints
@@ -4974,6 +4992,17 @@ class SqliteAiUsageStorage {
     });
     return summary;
   }
+  async getLatestForExecution(input) {
+    assertNonBlankProjectId(input.projectId);
+    assertAiUsageExecutionId(input.executionId);
+    const row = this.connection.db.query(`
+          SELECT * FROM ai_model_call_usage
+          WHERE project_id = ? AND execution_id = ?
+          ORDER BY occurred_at DESC, id DESC
+          LIMIT 1
+        `).get(input.projectId, input.executionId);
+    return row ? this.rowToRecord(row) : null;
+  }
   async summarizeExecutions(input) {
     assertNonBlankProjectId(input.projectId);
     for (const executionId of input.executionIds)
@@ -7592,6 +7621,9 @@ function isDuplicateExecutionError(error) {
   return error instanceof Error && error.message.includes("UNIQUE constraint failed: executions.project_id, executions.id");
 }
 
+// src/object-storage.ts
+import { linkBatchKey, objectBatchKey } from "@sixb/core/storage";
+
 // src/object-query-compiler.ts
 import {
   ObjectQueryExecutionError
@@ -7607,6 +7639,8 @@ function compileObjectQueryInternal(projectId, query, ctx) {
   switch (query.kind) {
     case "start":
       return compileStart(projectId, query);
+    case "refs":
+      return compileRefs(projectId, query);
     case "filter":
       return compileFilter(projectId, query.input, query.predicate);
     case "sort":
@@ -7654,6 +7688,50 @@ function compileStart(projectId, query) {
     }
   };
 }
+function compileRefs(projectId, query) {
+  if (query.refs.length === 0) {
+    throw new Error("[Sixb] SQLite object storage requires at least one ref");
+  }
+  const order = compileOrder(identityOrderFields());
+  const selectedOrder = compileOrder(identityOrderFields(), "selected");
+  const refsJson = JSON.stringify(query.refs);
+  const requested = `
+    SELECT DISTINCT
+      json_extract(ref.value, '$.objectTypeId') AS object_type_id,
+      json_extract(ref.value, '$.primaryId') AS primary_id
+    FROM json_each(?) AS ref
+  `;
+  const sql = `
+    WITH requested AS (${requested})
+    SELECT selected.*, selected.properties AS _cursor_properties
+    FROM requested
+    JOIN objects AS selected
+      ON selected.project_id = ?
+     AND selected.object_type_id = requested.object_type_id
+     AND selected.primary_id = requested.primary_id
+    ORDER BY ${selectedOrder.sql}
+  `;
+  return {
+    sql,
+    args: [refsJson, projectId, ...selectedOrder.args],
+    totalSql: `
+      WITH requested AS (${requested})
+      SELECT COUNT(*) AS total
+      FROM requested
+      JOIN objects AS selected
+        ON selected.project_id = ?
+       AND selected.object_type_id = requested.object_type_id
+       AND selected.primary_id = requested.primary_id
+    `,
+    totalArgs: [refsJson, projectId],
+    order,
+    hasMore: () => false,
+    trimRows: identityRows,
+    nextPageToken: () => {
+      return;
+    }
+  };
+}
 function compileFilter(projectId, inputQuery, predicateNode) {
   const predicate = compilePredicate(predicateNode);
   return compileWhere(projectId, inputQuery, predicate);
@@ -8397,6 +8475,7 @@ var SQLITE_OBJECT_QUERY_CAPABILITIES = {
   facetObjects: true,
   nodes: {
     start: true,
+    refs: true,
     filter: true,
     text: true,
     sort: true,
@@ -8448,7 +8527,7 @@ var SQLITE_OBJECT_QUERY_CAPABILITIES = {
     stablePageTokens: true
   },
   notes: [
-    "SQLite object query pushdown supports start/filter/text/sort/limit/page/traverse/set/project/expand over JSON properties and object links.",
+    "SQLite object query pushdown supports start/refs/filter/text/sort/limit/page/traverse/set/project/expand over JSON properties and object links.",
     "expand hydrates linked objects in-database (top-N per parent via row_number() + json_group_array); core resolves each expansion's cardinality before pushdown, and a mixed/unresolved one stays on the fallback.",
     "Ordered decimal predicates and sorting use the bounded core fallback because SQLite has no native exact decimal type; canonical decimal equality remains pushdown-safe.",
     "Relevance sorting, vector search, and unresolved start.includeSubtypes remain planner fallback or rejection cases."
@@ -8527,12 +8606,16 @@ class SqliteObjectStorage {
     const result = new Map;
     if (params.items.length === 0)
       return result;
-    const stmt = this.db.query("SELECT * FROM objects WHERE project_id = ? AND object_type_id = ? AND primary_id = ?");
-    for (const item of params.items) {
-      const row = stmt.get(params.projectId, item.objectTypeId, item.primaryId);
-      if (row) {
-        result.set(`${item.objectTypeId}:${item.primaryId}`, this.rowToObject(row));
-      }
+    const rows = this.db.query(`
+          SELECT object.*
+          FROM json_each(?) AS requested
+          JOIN objects AS object
+            ON object.project_id = ?
+           AND object.object_type_id = json_extract(requested.value, '$.objectTypeId')
+           AND object.primary_id = json_extract(requested.value, '$.primaryId')
+        `).all(JSON.stringify(params.items), params.projectId);
+    for (const row of rows) {
+      result.set(objectBatchKey(row.object_type_id, row.primary_id), this.rowToObject(row));
     }
     return result;
   }
@@ -8544,25 +8627,95 @@ class SqliteObjectStorage {
     for (const item of params.items) {
       const rows = stmt.all(params.projectId, item.objectTypeId, item.objectId, item.linkId);
       if (rows.length > 0) {
-        result.set(`${item.objectTypeId}:${item.objectId}:${item.linkId}`, rows.map((r) => this.rowToLink(r)));
+        result.set(linkBatchKey(item.objectTypeId, item.objectId, item.linkId), rows.map((r) => this.rowToLink(r)));
       }
     }
     return result;
   }
-  async listIncidentLinksBatch(params) {
-    const deduped = new Map;
-    for (const item of params.items) {
-      const rows = await this.listLinks({
-        projectId: params.projectId,
-        objectTypeId: item.objectTypeId,
-        objectId: item.objectId,
-        direction: "both"
-      });
-      for (const row of rows) {
-        deduped.set(`${row.sourceTypeId}:${row.sourceId}:${row.linkId}:${row.targetTypeId}:${row.targetId}`, row);
-      }
+  async queryLinks(params) {
+    assertLinkQueryLimit(params.limit);
+    if (params.objectRefs.length === 0 || params.endpointObjectTypeIds?.length === 0) {
+      return { links: [], hasMore: false };
+    }
+    const sourceJoin = `
+      SELECT link.*
+      FROM links AS link
+      JOIN requested
+        ON requested.object_type_id = link.source_type_id
+       AND requested.object_id = link.source_id
+      WHERE link.project_id = ?
+    `;
+    const targetJoin = `
+      SELECT link.*
+      FROM links AS link
+      JOIN requested
+        ON requested.object_type_id = link.target_type_id
+       AND requested.object_id = link.target_id
+      WHERE link.project_id = ?
+    `;
+    const incidentSql = params.direction === "outgoing" ? sourceJoin : params.direction === "incoming" ? targetJoin : `${sourceJoin} UNION ${targetJoin}`;
+    const args = [JSON.stringify(params.objectRefs), params.projectId];
+    if (params.direction === "both")
+      args.push(params.projectId);
+    const predicates = [];
+    if (params.linkId !== undefined) {
+      predicates.push("link_id = ?");
+      args.push(params.linkId);
     }
-    return [...deduped.values()];
+    if (params.endpointObjectTypeIds !== undefined) {
+      const allowedTypes = JSON.stringify([...new Set(params.endpointObjectTypeIds)]);
+      predicates.push("source_type_id IN (SELECT value FROM json_each(?))", "target_type_id IN (SELECT value FROM json_each(?))");
+      args.push(allowedTypes, allowedTypes);
+    }
+    if (params.after) {
+      predicates.push("(source_type_id, source_id, link_id, target_type_id, target_id) > (?, ?, ?, ?, ?)");
+      args.push(...params.after);
+    }
+    args.push(params.limit + 1);
+    const whereSql = predicates.length > 0 ? `WHERE ${predicates.join(" AND ")}` : "";
+    const rows = this.db.query(`
+          WITH requested AS (
+            SELECT DISTINCT
+              json_extract(value, '$.objectTypeId') AS object_type_id,
+              json_extract(value, '$.primaryId') AS object_id
+            FROM json_each(?)
+          ), incident AS (${incidentSql})
+          SELECT *
+          FROM incident
+          ${whereSql}
+          ORDER BY source_type_id, source_id, link_id, target_type_id, target_id
+          LIMIT ?
+        `).all(...args);
+    return {
+      links: rows.slice(0, params.limit).map((row) => this.rowToLink(row)),
+      hasMore: rows.length > params.limit
+    };
+  }
+  async listIncidentLinksBatch(params) {
+    if (params.items.length === 0)
+      return [];
+    const rows = this.db.query(`
+          WITH requested AS (
+            SELECT
+              json_extract(value, '$.objectTypeId') AS object_type_id,
+              json_extract(value, '$.objectId') AS object_id
+            FROM json_each(?)
+          )
+          SELECT link.*
+          FROM links AS link
+          JOIN requested
+            ON requested.object_type_id = link.source_type_id
+           AND requested.object_id = link.source_id
+          WHERE link.project_id = ?
+          UNION
+          SELECT link.*
+          FROM links AS link
+          JOIN requested
+            ON requested.object_type_id = link.target_type_id
+           AND requested.object_id = link.target_id
+          WHERE link.project_id = ?
+        `).all(JSON.stringify(params.items), params.projectId, params.projectId);
+    return rows.map((row) => this.rowToLink(row));
   }
   async listByPrimaryIdPage(params) {
     assertReconciliationPageLimit(params.limit);
@@ -8674,6 +8827,11 @@ function assertReconciliationPageLimit(limit) {
     throw new Error("Object reconciliation page limit must be a positive safe integer.");
   }
 }
+function assertLinkQueryLimit(limit) {
+  if (!Number.isSafeInteger(limit) || limit <= 0) {
+    throw new Error("Object link query limit must be a positive safe integer.");
+  }
+}
 function readTotal(db, compiled) {
   const row = db.query(compiled.totalSql).get(...compiled.totalArgs);
   return row.total;
@@ -13913,4 +14071,4 @@ export {
   SqliteStorage
 };
 
-//# debugId=3913310C430C57B764756E2164756E21
+//# debugId=2D24EFD822513C9E64756E2164756E21