@cortexkit/pi-magic-context 0.42.5 → 0.43.0

package/dist/message-time-backfill-dah83bcz.js

@@ -0,0 +1,108 @@
+import"./index-83wwtw1q.js";
+import {
+  logSlowWriteTransaction
+} from "./index-y33j7m6y.js";
+
+// ../plugin/src/features/magic-context/message-time-backfill.ts
+var MESSAGE_TIME_BACKFILL_BATCH_SIZE = 500;
+var BACKFILL_STATE_ID = 1;
+var activeBackfills = new WeakMap;
+function normalizedMessageTime(value) {
+  return typeof value === "number" && Number.isSafeInteger(value) && value >= 0 ? value : null;
+}
+function getBackfillState(db) {
+  const row = db.prepare(`SELECT cursor_session_id AS cursorSessionId,
+                    cursor_ordinal AS cursorOrdinal,
+                    completed
+               FROM message_time_backfill_state
+              WHERE id = ?`).get(BACKFILL_STATE_ID);
+  return {
+    processed: 0,
+    cursorSessionId: typeof row?.cursorSessionId === "string" ? row.cursorSessionId : "",
+    cursorOrdinal: typeof row?.cursorOrdinal === "number" && Number.isSafeInteger(row.cursorOrdinal) ? row.cursorOrdinal : 0,
+    completed: row?.completed === 1
+  };
+}
+function readMessageTimes(reader, rows) {
+  const times = new Map;
+  const bySession = new Map;
+  for (const row of rows) {
+    const list = bySession.get(row.sessionId) ?? [];
+    list.push(row);
+    bySession.set(row.sessionId, list);
+  }
+  for (const [sessionId, sessionRows] of bySession) {
+    const ordinals = sessionRows.map((row) => Number(row.messageOrdinal));
+    const firstOrdinal = Math.min(...ordinals);
+    const lastOrdinal = Math.max(...ordinals);
+    const messages = reader.readPage ? reader.readPage(sessionId, Math.max(0, firstOrdinal - 1), Math.max(1, lastOrdinal - firstOrdinal + 1), lastOrdinal) : reader(sessionId).filter((message) => message.ordinal >= firstOrdinal && message.ordinal <= lastOrdinal);
+    const wantedIds = new Set(sessionRows.map((row) => row.messageId));
+    for (const message of messages) {
+      const time = normalizedMessageTime(message.createdAt);
+      if (time !== null && wantedIds.has(message.id))
+        times.set(message.id, time);
+    }
+  }
+  return times;
+}
+function backfillMessageTimesBatch(db, reader, batchSize = MESSAGE_TIME_BACKFILL_BATCH_SIZE) {
+  const state = getBackfillState(db);
+  if (state.completed)
+    return state;
+  const boundedBatchSize = Math.max(1, Math.floor(batchSize));
+  const rows = db.prepare(`SELECT map.session_id AS sessionId,
+                    map.message_ordinal AS messageOrdinal,
+                    fts.message_id AS messageId
+               FROM message_fts_rowid_map AS map
+               JOIN message_history_fts AS fts ON fts.rowid = map.fts_rowid
+              WHERE map.message_time_ms IS NULL
+                AND (map.session_id > ?
+                     OR (map.session_id = ? AND map.message_ordinal > ?))
+              ORDER BY map.session_id ASC, map.message_ordinal ASC
+              LIMIT ?`).all(state.cursorSessionId, state.cursorSessionId, state.cursorOrdinal, boundedBatchSize);
+  const times = readMessageTimes(reader, rows);
+  const last = rows.at(-1);
+  const completed = rows.length < boundedBatchSize;
+  const cursorSessionId = last?.sessionId ?? state.cursorSessionId;
+  const cursorOrdinal = last ? Number(last.messageOrdinal) : state.cursorOrdinal;
+  const transactionStartedAt = performance.now();
+  db.transaction(() => {
+    const update = db.prepare(`UPDATE message_fts_rowid_map
+                SET message_time_ms = ?
+              WHERE session_id = ? AND message_ordinal = ? AND message_time_ms IS NULL`);
+    for (const row of rows) {
+      const time = times.get(row.messageId);
+      if (time !== undefined)
+        update.run(time, row.sessionId, Number(row.messageOrdinal));
+    }
+    db.prepare(`UPDATE message_time_backfill_state
+                SET cursor_session_id = ?, cursor_ordinal = ?, completed = ?, updated_at = ?
+              WHERE id = ?`).run(cursorSessionId, cursorOrdinal, completed ? 1 : 0, Date.now(), BACKFILL_STATE_ID);
+  })();
+  logSlowWriteTransaction("message_time_backfill", transactionStartedAt);
+  return {
+    processed: rows.length,
+    cursorSessionId,
+    cursorOrdinal,
+    completed
+  };
+}
+async function runMessageTimeBackfill(db, reader) {
+  for (;; ) {
+    const progress = backfillMessageTimesBatch(db, reader);
+    if (progress.completed)
+      return;
+    await new Promise((resolve) => setTimeout(resolve, 0));
+  }
+}
+function startMessageTimeBackfill(db, reader) {
+  const active = activeBackfills.get(db);
+  if (active)
+    return active;
+  const run = runMessageTimeBackfill(db, reader).finally(() => activeBackfills.delete(db));
+  activeBackfills.set(db, run);
+  return run;
+}
+export {
+  startMessageTimeBackfill
+};

package/dist/read-session-chunk-3zybnn0f.js

@@ -0,0 +1,47 @@
+import {
+  hasMeaningfulUserText2,
+  extractTexts2,
+  resolveHostServedBoundaryId2,
+  hasRawMessageProvider2,
+  setRawMessageProvider2,
+  withRawMessageProvider2,
+  cleanUserText2,
+  withRawSessionMessageCache2,
+  readRawSessionMessages2,
+  readRawSessionMessagePage2,
+  getRawSessionMessageOrdinalCount2,
+  readRawSessionMessageRange2,
+  primeTailRawMessageCache2,
+  getCachedAbsoluteMessageCount2,
+  primeInMemoryTailRawMessageCache2,
+  readRawSessionMessageIdOrdinalsForRange2,
+  hasRawSessionMessageById2,
+  getRawSessionMessageCount2,
+  getRawSessionTagKeysThrough2,
+  getLegacyProtectedTailStartOrdinal2,
+  readSessionChunk2
+} from "./index-vcjgpb6f.js";
+import"./index-83wwtw1q.js";
+export {
+  cleanUserText2 as cleanUserText,
+  extractTexts2 as extractTexts,
+  getCachedAbsoluteMessageCount2 as getCachedAbsoluteMessageCount,
+  getLegacyProtectedTailStartOrdinal2 as getLegacyProtectedTailStartOrdinal,
+  getRawSessionMessageCount2 as getRawSessionMessageCount,
+  getRawSessionMessageOrdinalCount2 as getRawSessionMessageOrdinalCount,
+  getRawSessionTagKeysThrough2 as getRawSessionTagKeysThrough,
+  hasMeaningfulUserText2 as hasMeaningfulUserText,
+  hasRawMessageProvider2 as hasRawMessageProvider,
+  hasRawSessionMessageById2 as hasRawSessionMessageById,
+  primeInMemoryTailRawMessageCache2 as primeInMemoryTailRawMessageCache,
+  primeTailRawMessageCache2 as primeTailRawMessageCache,
+  readRawSessionMessageIdOrdinalsForRange2 as readRawSessionMessageIdOrdinalsForRange,
+  readRawSessionMessagePage2 as readRawSessionMessagePage,
+  readRawSessionMessageRange2 as readRawSessionMessageRange,
+  readRawSessionMessages2 as readRawSessionMessages,
+  readSessionChunk2 as readSessionChunk,
+  resolveHostServedBoundaryId2 as resolveHostServedBoundaryId,
+  setRawMessageProvider2 as setRawMessageProvider,
+  withRawMessageProvider2 as withRawMessageProvider,
+  withRawSessionMessageCache2 as withRawSessionMessageCache
+};

package/dist/subagent-entry.js

@@ -1,14 +1,18 @@
 import {
   setHarness,
-  log,
-  resolveProjectIdentityForSession,
+  log
+} from "./index-83wwtw1q.js";
+import {
   setStoragePrivatePermissionEnforcement,
+  resolveProjectIdentityForSession,
   openDatabase,
   loadPiConfig,
   ensureProjectRegisteredFromPiDirectory,
   resolvePiHarnessKind,
   registerMagicContextTools
-} from "./index-a0960w4w.js";
+} from "./index-5tw61yhp.js";
+import"./index-vcjgpb6f.js";
+import"./index-y33j7m6y.js";
 
 // src/subagent-entry.ts
 var SUBAGENT_DREAMER_ACTIONS_FLAG = "magic-context-dreamer-actions";