@cortexkit/pi-magic-context 0.49.12 → 0.49.13

package/dist/auto-search-worker.js

@@ -18,11 +18,11 @@ var [
   { Database, registerSqliteDiagnosticSink },
   { log }
 ] = await Promise.all([
-  import("./embedding-backfill-14tqcfrn.js"),
+  import("./embedding-backfill-7kj16fh9.js"),
   import("./embedding-cache-7rp4gqzr.js"),
   import("./storage-memory-axnsk5t1.js"),
-  import("./project-embedding-registry-ymvne11v.js"),
-  import("./search-85e21wb8.js"),
+  import("./project-embedding-registry-52zwmtbn.js"),
+  import("./search-hffb8xht.js"),
   import("./embedding-activity-ew6dshfd.js"),
   import("./harness-9872e0ym.js"),
   import("./sqlite-adjpy399.js"),

package/dist/chunk-embedding-write-worker.js

@@ -0,0 +1,87 @@
+import {
+  setHarness2
+} from "./index-0h2mkab2.js";
+import {
+  Database2
+} from "./index-asrxgdt4.js";
+import {
+  configureContextDatabasePragmas2
+} from "./index-fapenpch.js";
+import"./index-g4jf772p.js";
+import"./index-7bqrfyny.js";
+import"./index-6dz2ed1v.js";
+import"./index-djxc8w8c.js";
+import {
+  replaceCompartmentChunkEmbeddings,
+  recordChunkEmbedBackoff,
+  loadUnembeddedSessionChunkCandidates,
+  countUnembeddedSessionCompartments
+} from "./index-7qc73yk5.js";
+import {
+  recordSessionProjectIdentity,
+  acquireGitSweepLease,
+  renewGitSweepLease,
+  releaseGitSweepLease
+} from "./index-1mrvw9en.js";
+import"./index-6ex901g3.js";
+import"./index-apgwdby5.js";
+import {
+  LATEST_SUPPORTED_VERSION,
+  enforceSchemaFence
+} from "./index-t27xk4r7.js";
+import"./index-3cnarwf8.js";
+import"./index-jrap43gc.js";
+import"./index-135e8hvg.js";
+import"./index-8976mezv.js";
+import"./index-d4dt19md.js";
+
+// ../plugin/src/features/magic-context/chunk-embedding-write-worker.ts
+import { parentPort, workerData } from "node:worker_threads";
+var port = parentPort;
+if (!port)
+  throw new Error("chunk embedding writer requires a parent port");
+setHarness2(workerData.harness);
+var db = new Database2(workerData.path);
+if (!enforceSchemaFence(db, workerData.path, LATEST_SUPPORTED_VERSION)) {
+  db.close();
+  throw new Error("chunk embedding writer rejected an unsupported store schema");
+}
+var timeout = Math.max(0, Math.floor(workerData.timeout));
+db.exec(`PRAGMA busy_timeout=${timeout}; PRAGMA foreign_keys=ON`);
+configureContextDatabasePragmas2(db);
+port.on("message", (input) => {
+  try {
+    let value;
+    switch (input.kind) {
+      case "replace":
+        replaceCompartmentChunkEmbeddings(db, input.rows, input.sourceKey);
+        break;
+      case "project":
+        recordSessionProjectIdentity(db, input.session, input.project);
+        break;
+      case "count":
+        value = countUnembeddedSessionCompartments(db, input.project, input.session, input.model, input.tokens);
+        break;
+      case "candidates":
+        value = loadUnembeddedSessionChunkCandidates(db, input.project, input.session, input.model, input.limit, input.excluded, input.tokens, true);
+        break;
+      case "backoff":
+        recordChunkEmbedBackoff(db, input.candidate, input.project, input.model);
+        break;
+      case "acquire":
+        value = acquireGitSweepLease(db, input.project, input.holder, {
+          ignoreCooldown: true
+        });
+        break;
+      case "renew":
+        value = renewGitSweepLease(db, input.project, input.holder);
+        break;
+      case "release":
+        releaseGitSweepLease(db, input.project, input.holder);
+        break;
+    }
+    port.postMessage({ value });
+  } catch (error) {
+    port.postMessage({ error: error instanceof Error ? error.message : String(error) });
+  }
+});

package/dist/{embedding-backfill-14tqcfrn.js → embedding-backfill-7kj16fh9.js}

@@ -1,11 +1,12 @@
 import {
   ensureMemoryEmbeddings2
-} from "./index-xpsht877.js";
-import"./index-f6e8jsrh.js";
+} from "./index-7wagp5jp.js";
+import"./index-nwbhe6qv.js";
 import"./index-9p40dgkm.js";
-import"./index-af9nb5db.js";
 import"./index-yhzhdp3m.js";
-import"./index-j24916mq.js";
+import"./index-1mrvw9en.js";
+import"./index-7qc73yk5.js";
+import"./index-9jvktm7k.js";
 import"./index-6dz2ed1v.js";
 import"./index-1k729e4f.js";
 import"./index-6ex901g3.js";

package/dist/embedding-worker.js

@@ -12,7 +12,7 @@ import {
   resolveLocalEmbeddingWorkerRuntime,
   getLocalEmbeddingNativeMemoryStats,
   LocalEmbeddingProvider
-} from "./index-j24916mq.js";
+} from "./index-9jvktm7k.js";
 import {
   setStoragePrivatePermissionEnforcement
 } from "./index-81nthk8j.js";

package/dist/historian-calibration-extension.js

@@ -2,7 +2,7 @@ import {
   subagentTemperature,
   promptFingerprint,
   readHostSystemPrompt
-} from "./index-xmdkde4v.js";
+} from "./index-y29y2j2w.js";
 import"./index-6x556c2q.js";
 import"./index-0h2mkab2.js";
 import"./index-7bqrfyny.js";

package/dist/index-1mrvw9en.js

@@ -0,0 +1,225 @@
+import {
+  getHarness2
+} from "./index-0h2mkab2.js";
+import {
+  isUserHomeDirectory
+} from "./index-6ex901g3.js";
+import {
+  logSlowWriteTransaction
+} from "./index-d4dt19md.js";
+
+// ../plugin/src/features/magic-context/session-project-storage.ts
+var SESSION_CHUNK_REPAIR_BATCH_SIZE = 100;
+var upsertSessionProjectStatements = new WeakMap;
+var repairSessionChunkProjectStatements = new WeakMap;
+var misScopedProjectChunkStatements = new WeakMap;
+var MIS_SCOPED_PROJECT_CHUNK_IDS_SQL = `
+    SELECT e.id FROM compartment_chunk_embeddings e
+    JOIN session_projects sp ON sp.session_id = e.session_id AND sp.harness = e.harness
+    WHERE e.project_path = ? AND sp.project_path <> e.project_path
+    UNION ALL
+    SELECT e.id FROM session_projects sp
+    JOIN compartment_chunk_embeddings e ON e.session_id = sp.session_id
+    WHERE sp.project_path = ? AND e.harness = sp.harness AND e.project_path <> sp.project_path`;
+function getUpsertSessionProjectStatement(db) {
+  let stmt = upsertSessionProjectStatements.get(db);
+  if (!stmt) {
+    stmt = db.prepare(`INSERT INTO session_projects (session_id, harness, project_path, updated_at)
+             VALUES (?, ?, ?, ?)
+             ON CONFLICT(session_id, harness) DO UPDATE SET
+                 project_path = excluded.project_path,
+                 updated_at = excluded.updated_at
+             WHERE session_projects.project_path <> excluded.project_path`);
+    upsertSessionProjectStatements.set(db, stmt);
+  }
+  return stmt;
+}
+function getRepairSessionChunkProjectStatement(db) {
+  let stmt = repairSessionChunkProjectStatements.get(db);
+  if (!stmt) {
+    stmt = db.prepare(`UPDATE compartment_chunk_embeddings
+             SET project_path = ?
+             WHERE id IN (
+                 SELECT id
+                 FROM compartment_chunk_embeddings
+                 WHERE session_id = ?
+                   AND harness = ?
+                   AND project_path <> ?
+                 LIMIT ?
+             )`);
+    repairSessionChunkProjectStatements.set(db, stmt);
+  }
+  return stmt;
+}
+function findMisScopedCompartmentChunkEmbeddingIdsForProject(db, projectPath) {
+  return db.prepare(`SELECT id FROM (${MIS_SCOPED_PROJECT_CHUNK_IDS_SQL}) LIMIT 25`).all(projectPath, projectPath).map(({ id }) => id);
+}
+function recordSessionProjectIdentity(db, sessionId, projectPath) {
+  if (!sessionId || !projectPath)
+    return;
+  if (!projectPath.startsWith("git:") && !projectPath.startsWith("dir:") && isUserHomeDirectory(projectPath))
+    return;
+  const harness = getHarness2();
+  const now = Date.now();
+  let previousProjectPath;
+  db.transaction(() => {
+    const existing = db.prepare("SELECT project_path FROM session_projects WHERE session_id = ? AND harness = ?").get(sessionId, harness);
+    previousProjectPath = existing?.project_path;
+    getUpsertSessionProjectStatement(db).run(sessionId, harness, projectPath, now);
+    getRepairSessionChunkProjectStatement(db).run(projectPath, sessionId, harness, projectPath, SESSION_CHUNK_REPAIR_BATCH_SIZE);
+  }).immediate();
+  return previousProjectPath === projectPath ? undefined : { previousProjectPath, projectPath };
+}
+function repairMisScopedCompartmentChunkEmbeddingsForProject(db, projectPath, ids = findMisScopedCompartmentChunkEmbeddingIdsForProject(db, projectPath)) {
+  if (!projectPath || ids.length === 0)
+    return 0;
+  return db.prepare(`UPDATE compartment_chunk_embeddings
+        SET project_path = (SELECT sp.project_path FROM session_projects sp
+            WHERE sp.session_id = compartment_chunk_embeddings.session_id
+              AND sp.harness = compartment_chunk_embeddings.harness)
+        WHERE id IN (${ids.map(() => "?").join(",")}) AND EXISTS (
+            SELECT 1 FROM session_projects sp
+            WHERE sp.session_id = compartment_chunk_embeddings.session_id
+              AND sp.harness = compartment_chunk_embeddings.harness
+              AND sp.project_path <> compartment_chunk_embeddings.project_path
+              AND (sp.project_path = ? OR compartment_chunk_embeddings.project_path = ?)
+        )`).run(...ids, projectPath, projectPath).changes;
+}
+
+// ../plugin/src/features/magic-context/git-commits/sweep-coordinator.ts
+var GIT_SWEEP_COOLDOWN_MS = 10 * 60 * 1000;
+var GIT_SWEEP_LEASE_TTL_MS = 5 * 60 * 1000;
+var GIT_SWEEP_NON_INDEXABLE_REPROBE_MS = 24 * 60 * 60 * 1000;
+var GIT_SWEEP_LEASE_RENEWAL_MS = 60 * 1000;
+function runImmediate(db, body) {
+  db.exec("BEGIN IMMEDIATE");
+  const transactionStartedAt = performance.now();
+  let committed = false;
+  try {
+    const result = body();
+    db.exec("COMMIT");
+    committed = true;
+    logSlowWriteTransaction("git_sweep_lease", transactionStartedAt);
+    return result;
+  } finally {
+    if (!committed) {
+      try {
+        db.exec("ROLLBACK");
+      } catch {}
+    }
+  }
+}
+function rowToState(row) {
+  return {
+    projectPath: row.project_path,
+    leaseHolder: row.lease_holder,
+    leaseExpiresAt: row.lease_expires_at,
+    lastSweptAt: row.last_swept_at
+  };
+}
+function getGitSweepCoordinatorState(db, projectPath) {
+  const row = db.prepare(`SELECT project_path, lease_holder, lease_expires_at, last_swept_at
+             FROM git_sweep_coordinator
+             WHERE project_path = ?`).get(projectPath);
+  return row ? rowToState(row) : null;
+}
+function acquireGitSweepLease(db, projectPath, holderId, options = {}) {
+  const cooldownMs = options.cooldownMs ?? GIT_SWEEP_COOLDOWN_MS;
+  const leaseTtlMs = options.leaseTtlMs ?? GIT_SWEEP_LEASE_TTL_MS;
+  return runImmediate(db, () => {
+    const now = Date.now();
+    const row = getGitSweepCoordinatorState(db, projectPath);
+    if (row?.leaseHolder && row.leaseExpiresAt !== null && row.leaseExpiresAt > now) {
+      return {
+        acquired: false,
+        projectPath,
+        reason: "lease_active",
+        leaseHolder: row.leaseHolder,
+        leaseExpiresAt: row.leaseExpiresAt,
+        lastSweptAt: row.lastSweptAt,
+        nextAllowedAt: null
+      };
+    }
+    if (!options.ignoreCooldown && row?.lastSweptAt !== null && row?.lastSweptAt !== undefined) {
+      const nextAllowedAt = row.lastSweptAt + cooldownMs;
+      if (nextAllowedAt > now) {
+        return {
+          acquired: false,
+          projectPath,
+          reason: "cooldown_active",
+          leaseHolder: row.leaseHolder,
+          leaseExpiresAt: row.leaseExpiresAt,
+          lastSweptAt: row.lastSweptAt,
+          nextAllowedAt
+        };
+      }
+    }
+    const leaseExpiresAt = now + leaseTtlMs;
+    db.prepare(`INSERT INTO git_sweep_coordinator (
+                 project_path,
+                 lease_holder,
+                 lease_expires_at,
+                 last_swept_at
+             ) VALUES (?, ?, ?, NULL)
+             ON CONFLICT(project_path) DO UPDATE SET
+                 lease_holder = excluded.lease_holder,
+                 lease_expires_at = excluded.lease_expires_at`).run(projectPath, holderId, leaseExpiresAt);
+    return {
+      acquired: true,
+      projectPath,
+      holderId,
+      acquiredAt: now,
+      leaseExpiresAt
+    };
+  });
+}
+function renewGitSweepLease(db, projectPath, holderId, leaseTtlMs = GIT_SWEEP_LEASE_TTL_MS) {
+  return runImmediate(db, () => {
+    const now = Date.now();
+    const leaseExpiresAt = now + leaseTtlMs;
+    const result = db.prepare(`UPDATE git_sweep_coordinator
+                 SET lease_expires_at = ?
+                 WHERE project_path = ?
+                   AND lease_holder = ?
+                   AND lease_expires_at > ?`).run(leaseExpiresAt, projectPath, holderId, now);
+    return result.changes === 1;
+  });
+}
+function markGitSweepSuccessAndRelease(db, projectPath, holderId) {
+  return runImmediate(db, () => {
+    const now = Date.now();
+    const result = db.prepare(`UPDATE git_sweep_coordinator
+                 SET lease_holder = NULL,
+                     lease_expires_at = NULL,
+                     last_swept_at = ?
+                 WHERE project_path = ?
+                   AND lease_holder = ?
+                   AND lease_expires_at > ?`).run(now, projectPath, holderId, now);
+    return result.changes === 1;
+  });
+}
+function parkGitSweepNonIndexable(db, projectPath, holderId, reprobeMs = GIT_SWEEP_NON_INDEXABLE_REPROBE_MS) {
+  return runImmediate(db, () => {
+    const now = Date.now();
+    const sweptAt = now + reprobeMs - GIT_SWEEP_COOLDOWN_MS;
+    const result = db.prepare(`UPDATE git_sweep_coordinator
+                 SET lease_holder = NULL,
+                     lease_expires_at = NULL,
+                     last_swept_at = ?
+                 WHERE project_path = ?
+                   AND lease_holder = ?
+                   AND lease_expires_at > ?`).run(sweptAt, projectPath, holderId, now);
+    return result.changes === 1;
+  });
+}
+function releaseGitSweepLease(db, projectPath, holderId) {
+  runImmediate(db, () => {
+    db.prepare(`UPDATE git_sweep_coordinator
+             SET lease_holder = NULL,
+                 lease_expires_at = NULL
+             WHERE project_path = ?
+               AND lease_holder = ?`).run(projectPath, holderId);
+  });
+}
+
+export { findMisScopedCompartmentChunkEmbeddingIdsForProject, recordSessionProjectIdentity, repairMisScopedCompartmentChunkEmbeddingsForProject, GIT_SWEEP_LEASE_RENEWAL_MS, acquireGitSweepLease, renewGitSweepLease, markGitSweepSuccessAndRelease, parkGitSweepNonIndexable, releaseGitSweepLease };

package/dist/{index-af9nb5db.js → index-7qc73yk5.js}

@@ -858,13 +858,17 @@ function loadUnembeddedShadowChunkCandidates(db, projectPath, primaryModelId, sh
   const rows = getShadowBackfillCandidateStatement(db).all(projectPath, projectPath, primaryModelId);
   return selectHashIncompleteChunkCandidates(db, projectPath, shadowModelId, mapBackfillCandidateRows(rows), Math.max(1, limit), shadowMaxInputTokens);
 }
-async function loadUnembeddedCompartmentChunkCandidatesPolite(db, projectPath, modelId, limit, maxInputTokens = DEFAULT_COMPARTMENT_CHUNK_MAX_INPUT_TOKENS, leaseHeldRenumber = false) {
+async function loadUnembeddedCompartmentChunkCandidatesPolite(db, projectPath, modelId, limit, maxInputTokens = DEFAULT_COMPARTMENT_CHUNK_MAX_INPUT_TOKENS, leaseHeldRenumber = false, isEligible) {
   const rows = getBackfillCandidateStatement(db).all(projectPath);
   const candidates = mapBackfillCandidateRows(rows);
   const missing = [];
   const stale = [];
   let sliceStarted = performance.now();
   for (const candidate of candidates) {
+    if (isEligible && !isEligible(candidate)) {
+      sliceStarted = await yieldCoverageSlice(sliceStarted);
+      continue;
+    }
     const classification = classifyChunkCoverageDefect(db, projectPath, modelId, candidate, maxInputTokens);
     if (classification.defect === "missing")
       missing.push(candidate);
@@ -1019,7 +1023,7 @@ function memoizedCoverageWindows(candidate, canonicalText, sourceKey, modelId, m
   return windows;
 }
 async function yieldCoverageSlice(started) {
-  if (performance.now() - started < 35)
+  if (performance.now() - started < 8)
     return started;
   await new Promise((resolve) => setTimeout(resolve, 0));
   return performance.now();
@@ -1093,6 +1097,47 @@ function selectHashIncompleteChunkCandidates(db, projectPath, modelId, candidate
   return [...missing, ...stale].slice(0, limit);
 }
 var sessionBackfillCandidateStatements = new WeakMap;
+function loadUnembeddedSessionChunkCandidates(db, projectPath, sessionId, modelId, limit, excludeIds, maxInputTokens = DEFAULT_COMPARTMENT_CHUNK_MAX_INPUT_TOKENS, leaseHeldRenumber = false) {
+  const exclusions = excludeIds && excludeIds.length > 0 ? excludeIds : [];
+  const exclusionSql = exclusions.length > 0 ? `AND c.id NOT IN (${exclusions.map(() => "?").join(", ")})` : "";
+  const stmt = exclusions.length > 0 ? db.prepare(`SELECT c.id AS id,
+                          c.session_id AS sessionId,
+                          c.start_message AS startMessage,
+                          c.end_message AS endMessage,
+                          c.title AS title
+                   FROM compartments c
+                   JOIN session_projects sp
+                     ON sp.session_id = c.session_id
+                    AND sp.harness = c.harness
+                    AND sp.project_path = ?
+                   WHERE c.session_id = ?
+                     AND c.start_message IS NOT NULL
+                     AND c.end_message IS NOT NULL
+                     ${exclusionSql}
+                   ORDER BY c.start_message ASC, c.id ASC`) : (() => {
+    let cached = sessionBackfillCandidateStatements.get(db);
+    if (!cached) {
+      cached = db.prepare(`SELECT c.id AS id,
+                                  c.session_id AS sessionId,
+                                  c.start_message AS startMessage,
+                                  c.end_message AS endMessage,
+                                  c.title AS title
+                           FROM compartments c
+                           JOIN session_projects sp
+                             ON sp.session_id = c.session_id
+                            AND sp.harness = c.harness
+                            AND sp.project_path = ?
+                           WHERE c.session_id = ?
+                             AND c.start_message IS NOT NULL
+                             AND c.end_message IS NOT NULL
+                           ORDER BY c.start_message ASC, c.id ASC`);
+      sessionBackfillCandidateStatements.set(db, cached);
+    }
+    return cached;
+  })();
+  const rows = stmt.all(projectPath, sessionId, ...exclusions);
+  return selectHashIncompleteChunkCandidates(db, projectPath, modelId, mapBackfillCandidateRows(rows), Math.max(1, limit), maxInputTokens, leaseHeldRenumber);
+}
 async function loadUnembeddedSessionChunkCandidatesPolite(db, projectPath, sessionId, modelId, limit, excludeIds = [], maxInputTokens = DEFAULT_COMPARTMENT_CHUNK_MAX_INPUT_TOKENS, leaseHeldRenumber = false) {
   const excluded = new Set(excludeIds);
   const missing = [];
@@ -1109,10 +1154,22 @@ async function loadUnembeddedSessionChunkCandidatesPolite(db, projectPath, sessi
     else if (leaseHeldRenumber) {
       applyLeaseHeldCoverageRepair(db, candidate, projectPath, modelId, classification);
     }
+    if (missing.length >= Math.max(1, limit))
+      return missing;
     started = await yieldCoverageSlice(started);
   }
   return [...missing, ...stale].slice(0, Math.max(1, limit));
 }
+function countUnembeddedSessionCompartments(db, projectPath, sessionId, modelId, maxInputTokens = DEFAULT_COMPARTMENT_CHUNK_MAX_INPUT_TOKENS) {
+  const rows = getSessionBackfillRows(db, projectPath, sessionId);
+  let count = 0;
+  for (const candidate of rows) {
+    const { defect } = classifyChunkCoverageDefect(db, projectPath, modelId, candidate, maxInputTokens);
+    if (defect === "missing" || defect === "stale")
+      count++;
+  }
+  return count;
+}
 function countSessionCompartmentEmbedCoverage(db, projectPath, sessionId, modelId, maxInputTokens = DEFAULT_COMPARTMENT_CHUNK_MAX_INPUT_TOKENS) {
   const candidates = getSessionBackfillRows(db, projectPath, sessionId);
   let embedded = 0;
@@ -1165,4 +1222,4 @@ async function countUnembeddedSessionCompartmentsPolite(db, projectPath, session
   return count;
 }
 
-export { DEFAULT_COMPARTMENT_CHUNK_MAX_INPUT_TOKENS, CHUNK_WINDOW_SAFETY_RATIO, normalizeCompartmentChunkMaxInputTokens, buildCanonicalChunkTextFromFts, buildCompartmentSummaryFallbackText, canonicalizeInMemoryChunkTextForEmbedding, chunkCanonicalText, chunkEmbeddingWindowsAreCurrent, replaceCompartmentChunkEmbeddings, loadCompartmentChunkEmbeddingsForSearch, loadUnembeddedShadowChunkCandidates, loadUnembeddedCompartmentChunkCandidatesPolite, deleteChunkEmbedBackoffForSession, recordChunkEmbedBackoff, chunkWindowSourceKey, loadUnembeddedSessionChunkCandidatesPolite, countSessionCompartmentEmbedCoverage, countSessionCompartmentEmbedCoveragePolite, countUnembeddedSessionCompartmentsPolite };
+export { DEFAULT_COMPARTMENT_CHUNK_MAX_INPUT_TOKENS, CHUNK_WINDOW_SAFETY_RATIO, normalizeCompartmentChunkMaxInputTokens, buildCanonicalChunkTextFromFts, buildCompartmentSummaryFallbackText, canonicalizeInMemoryChunkTextForEmbedding, chunkCanonicalText, chunkEmbeddingWindowsAreCurrent, replaceCompartmentChunkEmbeddings, loadCompartmentChunkEmbeddingsForSearch, loadUnembeddedShadowChunkCandidates, loadUnembeddedCompartmentChunkCandidatesPolite, deleteChunkEmbedBackoffForSession, recordChunkEmbedBackoff, chunkWindowSourceKey, loadUnembeddedSessionChunkCandidates, loadUnembeddedSessionChunkCandidatesPolite, countUnembeddedSessionCompartments, countSessionCompartmentEmbedCoverage, countSessionCompartmentEmbedCoveragePolite, countUnembeddedSessionCompartmentsPolite };

package/dist/{index-xpsht877.js → index-7wagp5jp.js}

@@ -10,12 +10,12 @@ import {
 } from "./index-6dz2ed1v.js";
 import {
   LocalEmbeddingProvider
-} from "./index-j24916mq.js";
+} from "./index-9jvktm7k.js";
 import {
   OpenAICompatibleEmbeddingProvider,
   getProjectEmbeddingSnapshot2,
   embedBatchForProject2
-} from "./index-f6e8jsrh.js";
+} from "./index-nwbhe6qv.js";
 import {
   isEmbeddingHostBusy2
 } from "./index-9p40dgkm.js";

package/dist/{index-zapgkmp7.js → index-aypx5h57.js}

@@ -103,7 +103,7 @@ import {
   clearLkgRecoveryFence,
   deleteSessionScopedRows,
   getLastIndexedOrdinal2
-} from "./index-6hex51s9.js";
+} from "./index-vv1j5cak.js";
 import {
   SubcClient,
   connectionFileExists,
@@ -114,9 +114,22 @@ import {
 import {
   sameProjectDirectory
 } from "./index-djxc8w8c.js";
+import {
+  invalidateAutoEmbedSession,
+  drainBackgroundBatches,
+  markProjectLoadUntrusted2,
+  drainProjectEmbeddingIdentityMaintenance2,
+  registerProjectEmbedding2,
+  registerProjectShadowEmbedding2,
+  registerProjectInObservationMode2,
+  unregisterProjectShadowEmbedding2,
+  unregisterProjectEmbedding2,
+  getProjectEmbeddingSnapshot2,
+  embedTextForProject2
+} from "./index-nwbhe6qv.js";
 import {
   getErrorMessage
-} from "./index-z5ftst3e.js";
+} from "./index-t27xk4r7.js";
 import {
   V2_MEMORY_CATEGORIES,
   CATEGORY_PRIORITY,
@@ -137,18 +150,6 @@ import {
   normalizeStoredProjectPath,
   storedPathBelongsToIdentity
 } from "./index-6ex901g3.js";
-import {
-  drainBackgroundBatches,
-  markProjectLoadUntrusted2,
-  drainProjectEmbeddingIdentityMaintenance2,
-  registerProjectEmbedding2,
-  registerProjectShadowEmbedding2,
-  registerProjectInObservationMode2,
-  unregisterProjectShadowEmbedding2,
-  unregisterProjectEmbedding2,
-  getProjectEmbeddingSnapshot2,
-  embedTextForProject2
-} from "./index-f6e8jsrh.js";
 import {
   getNoteByIdInScope,
   getNotes,
@@ -165,7 +166,7 @@ import {
   resolveMemoriesByIdsForSearch2,
   unifiedSearch2,
   formatSearchResults2
-} from "./index-gykw2zhn.js";
+} from "./index-wp9q91kn.js";
 import {
   computeCueContentHash,
   getMuralCueState,
@@ -4499,11 +4500,12 @@ class V2StoreReader {
     const head = Math.floor((maxChars - truncationMarker.length) / 2);
     const tail = maxChars - truncationMarker.length - head;
     return this.prepare(`WITH users AS (
-            SELECT seq, time_created, data FROM session_message
-            WHERE session_id = ? AND type = 'user' AND time_created ${before ? "<=" : ">"} ?
+            SELECT session_id, seq, time_created, data FROM session_message
+            WHERE session_id = ? AND type = 'user' AND time_created ${before ? "<=" : ">"}
+                ?
             ORDER BY time_created ${direction}, seq ${direction} LIMIT ?
         ), texts AS (
-            SELECT seq, time_created, CASE
+            SELECT session_id, seq, time_created, CASE
                 WHEN json_extract(data, '$.synthetic') = 1 OR json_extract(data, '$.ignored') = 1 THEN ''
                 WHEN json_type(data, '$.text') = 'text' THEN json_extract(data, '$.text')
                 ELSE COALESCE((SELECT group_concat(json_extract(value, '$.text'), char(10))
@@ -4513,7 +4515,19 @@ class V2StoreReader {
                       AND COALESCE(json_extract(value, '$.ignored'), 0) <> 1), '') END AS text
             FROM users WHERE json_valid(data)
         ) SELECT seq, time_created, CASE WHEN length(text) > ?
-            THEN substr(text, 1, ?) || ? || substr(text, -?) ELSE text END AS text
+            THEN substr(text, 1, ?) || ? || substr(text, -?) ELSE text END AS text,
+            EXISTS (
+                SELECT 1 FROM session_message assistant_reply
+                 WHERE assistant_reply.session_id = texts.session_id
+                   AND assistant_reply.type = 'assistant'
+                   AND assistant_reply.seq < texts.seq
+                   AND EXISTS (
+                       SELECT 1 FROM session_message prior_user
+                        WHERE prior_user.session_id = texts.session_id
+                          AND prior_user.type = 'user'
+                          AND prior_user.seq < assistant_reply.seq
+                   )
+            ) AS follows_assistant_reply
           FROM texts ORDER BY time_created ${direction}, seq ${direction}`).all(sessionID, boundaryMs, limit, maxChars, head, truncationMarker, tail);
   }
   storedRowsAfter(sessionID, after) {
@@ -4948,31 +4962,6 @@ function clearSession(db, sessionId, rustModuleCleanupAcknowledged = false, perm
   clearUnpublishedAutoSearchNoHints(sessionId);
   logSlowWriteTransaction("clear-session", transactionStartedAt);
 }
-// ../plugin/src/hooks/magic-context/embed-session-state.ts
-var embedPauseBySession = new Set;
-var embedRunStateBySession = new Map;
-var autoEmbedAttemptedBySession = new Set;
-var autoEmbedIdentityBySession = new Map;
-function invalidateAutoEmbedSession(sessionId) {
-  autoEmbedAttemptedBySession.delete(sessionId);
-  autoEmbedIdentityBySession.delete(sessionId);
-}
-function getEmbedDrainUiStatus(sessionId, progress) {
-  if (embedPauseBySession.has(sessionId)) {
-    return { status: "paused" };
-  }
-  const activeRun = embedRunStateBySession.get(sessionId);
-  if (activeRun && !activeRun.signal.aborted || progress?.kind === "embed" && progress.phase === "recomp") {
-    return { status: "running" };
-  }
-  if (progress?.kind === "embed" && (progress.phase === "failed" || progress.phase === "skipped") && progress.message) {
-    if (/provider/i.test(progress.message)) {
-      return { status: "stopped", detail: progress.message };
-    }
-  }
-  return { status: "idle" };
-}
-
 // ../plugin/src/features/magic-context/compartment-lease.ts
 var COMPARTMENT_LEASE_TTL_MS = 5 * 60 * 1000;
 var COMPARTMENT_LEASE_RENEWAL_MS = 60 * 1000;
@@ -15565,4 +15554,4 @@ function registerMagicContextTools(pi, opts) {
   }
 }
 
-export { COMPACTION_ENABLED_PATH, isDreamerRunnable, isCompactionEnabled, formatConfigParseStatusLine, formatConfigParseNotice, claimConfigParseFailuresOnce, parseJsoncRecovering, detectConfigFile, MIN_PLAUSIBLE_CONTEXT_LIMIT, catalogOutputReserve, deriveWindowGeometry, hasTrustedAbsoluteWall, applyProvenInputFloor, formatWindowDerivationLine, formatWindowSource, isSaneLimit, resolveOutputReserve, getSdkWindowGeometry, getSdkContextLimit, getSdkOutputLimit, getSdkInputLimit, cortexKitUserConfigBasePath, cortexKitProjectConfigBasePath, migrateMagicContextConfigLocations, getProtectedTokensTierOverrides, rememberUnpublishedAutoSearchNoHint, getUnpublishedAutoSearchNoHint, forgetUnpublishedAutoSearchNoHint, registerAutoSearchWork, cancelAutoSearchWork, readSessionMeta, getOrCreateSessionMeta, updateSessionMeta, advanceToolReclaimWatermark, retryPendingSessionCleanups, retryPendingRustSessionCleanupsForProject, embedPauseBySession, embedRunStateBySession, autoEmbedAttemptedBySession, autoEmbedIdentityBySession, invalidateAutoEmbedSession, getEmbedDrainUiStatus, COMPARTMENT_LEASE_RENEWAL_MS, acquireCompartmentLease, renewCompartmentLease, releaseCompartmentLease, releaseCompartmentLeaseBestEffort, isCompartmentLeaseHeld, isNoContentCompartment, HAS_COMPARTMENT_CONTENT_SQL, queueM0Mutation, getMaxM0MutationId, getCompartments, getLastCompartmentEndMessage, getLastCompartmentEndMessageId, getCompartmentsByEndMessageId, appendCompartments, saveRecompStagingPass, getRecompStaging, clearRecompStaging, getRecompPartialRange, setRecompPartialRange, escapeXmlAttr, escapeXmlContent, getUncoveredCompartmentEndThrough, isPartialCompartmentEnd, readProjectDocsCanonical, providerResponseFailed, isSuccessfulProviderCompletion, requestPendingMaterialization, pendingMaterializationOrigin, hasPendingMaterializationOrigin, clearPendingMaterialization, recordSystemRefreshOwnership, recordSystemPromptForRequest, recordHeldHeadForRequest, bindSystemPromptResponse, acknowledgeSystemPromptDelivery, deferDeliveredSystemRefresh, clearSystemPromptDelivery, saveLkgSlotToDb, loadPersistedLkgSlot, drainStaleLkgSlots, createDbLkgPersistence, replayLkg, registerLkgPersistence, LKG_SNAPSHOT_ARRAY, LKG_SNAPSHOT_OBJECT, LKG_SNAPSHOT_KEY, LKG_SNAPSHOT_STRING, LKG_SNAPSHOT_NUMBER, LKG_SNAPSHOT_BOOLEAN, LKG_SNAPSHOT_NULL, LKG_SNAPSHOT_UNDEFINED, contentSnapshotValue, signatureForFields, lkgContentFields, lkgContentDigestFromFields, isLegacyLkgDigest, legacyLkgDigestFromFields, lkgFieldsMatchDigest, exactReusablePrefix, incrementalLkgContentDigests, lkgSlotRejection, captureSlot, getInMemorySlot, getSlot, dropSlot, TOOL_SWEEP_SCOPED_MARKER, encodePiContentDecision, decodePiContentDecision, getPiContentDecisions, freezePiContentDecision, getNativeReplayState, saveNativeToolInputs, addNativeReasoningIds, copySessionStateForClone, externalHistorianFailureReason, subprocessExitHistorianFailureReason, countHistorianRuns, recordHistorianRun, summarizeImportance, tallyFactsByCategory, queueMemoryMutation, getMemoryMutationsForRender, getMemoryMutationsForRenderByProjects, getMaxMemoryMutationId, getMaxMemoryMutationIdForProjects, GLOBAL_USER_PROFILE_PROJECT_PATH, getProjectState, bumpProjectUserProfileVersion, saveSourceContent, getSourceContents, recordSubagentInvocation, getLatestHistorianInvocationId, noteTouchedAt, noteNudgePickIndex, formatNoteNudge, systemPromptFingerprint, createPromptSurfaceRuntime, createPromptSurfaceGuidanceEpochCache, SYNTH_USER_ID_PREFIX, resolvePiStableId, resolvePiRealStableId, isPiContentFallbackId, isPiIndexFallbackId, piContentFallbackIds, readPiSessionSnapshot, readPiSessionMessages, readPiSessionMessagePage, findRestartModelSeedFromBranch, iterateEntriesToRawMessageRange, iterateAsyncEntriesToRawMessageRange, convertEntriesToRawMessages, convertPiAssistantEntryById, convertEntriesToRawMessagePage, countPiRawMessages, USER_MEMORY_CANDIDATE_TTL_MS, insertUserMemoryCandidates, getUserMemoryCandidates, deleteUserMemoryCandidates, pruneExpiredUserMemoryCandidates, insertUserMemory, getActiveUserMemories, updateUserMemoryContent, dismissUserMemory, getTaskScheduleState, getFailingDreamTasks, getSkippedDreamTasks, getMostRecentTaskRunAt, pruneNonCanonicalTaskRows, deleteTaskScheduleRowsForProject, seedTaskScheduleState, writeTaskScheduleState, isRetrospectiveWindowProcessed, recordRetrospectiveWindowProcessed, curateCategoryForMemoryCategory, peekCurateCategoryScope, beginCurateCategoryRun, curateTaskStateAfterSuccess, MEMORY_MURAL_BLOCK, buildMagicContextSection, resolveMuralWire, updateCompactionMarkerAfterPublication, COMPARTMENT_RENDER_EPOCH, MEMORY_RENDER_FORMAT_EPOCH, encodeCachedM0UpgradeIdentity, decodeCachedM0UpgradeIdentity, withCachedM0MemoryIds, readCachedM0MemoryIds, renderBudgetIdentityChanged, renderedBudgetSnapshot, renderedBudgetShrinkReason, DEFAULT_HISTORY_BUDGET_TOKENS, renderCompartmentAtTier, renderDecayedCompartments, extractM0Block, TEMPORAL_MARKER_PATTERN, TEMPORAL_MARKER_REPLAY_PATTERN, temporalMarkerPrefix, renderUserProfileContent, clearInjectionCache, getVisibleMemoryIds, orderHistorianMemories, renderHistorianMemoryBlock, DEFAULT_MEMORY_BUDGET_TOKENS, DEFAULT_USER_PROFILE_BUDGET_TOKENS, trimMemoriesToBudgetV2, trimWorkspaceMemoriesToBudgetV2, trimUserMemoriesToBudget, renderMemoryBlockV2, stripMemoryMuralBlock, resetPrefixTrimFallbackState, isDirectiveShapedProjectRule, takeCurateSafetyRefusalCount, wakePlaneStatus, loadPiConfig, loadPiConfigDetailed, unregisterPiProjectEmbeddings, ensureProjectRegisteredFromPiDirectory, resolvePiHarnessDetection, resolvePiHarnessKind, resolvePiOrdinalAlignment, piRawOrdinalOffsetSource, resolvePiOrdinalAlignmentForContext, isPiOrdinalAlignmentUnanchored, clearPiOrdinalAlignmentSession, rememberTodowriteToolCallTodos, parseTodos, setTodoSnapshot, registerTodoOverlay, registerTodoStateLifecycle, syncCtxMemoryToolEnabled, registerMagicContextTools };
+export { COMPACTION_ENABLED_PATH, isDreamerRunnable, isCompactionEnabled, formatConfigParseStatusLine, formatConfigParseNotice, claimConfigParseFailuresOnce, parseJsoncRecovering, detectConfigFile, MIN_PLAUSIBLE_CONTEXT_LIMIT, catalogOutputReserve, deriveWindowGeometry, hasTrustedAbsoluteWall, applyProvenInputFloor, formatWindowDerivationLine, formatWindowSource, isSaneLimit, resolveOutputReserve, getSdkWindowGeometry, getSdkContextLimit, getSdkOutputLimit, getSdkInputLimit, cortexKitUserConfigBasePath, cortexKitProjectConfigBasePath, migrateMagicContextConfigLocations, getProtectedTokensTierOverrides, rememberUnpublishedAutoSearchNoHint, getUnpublishedAutoSearchNoHint, forgetUnpublishedAutoSearchNoHint, registerAutoSearchWork, cancelAutoSearchWork, readSessionMeta, getOrCreateSessionMeta, updateSessionMeta, advanceToolReclaimWatermark, retryPendingSessionCleanups, retryPendingRustSessionCleanupsForProject, COMPARTMENT_LEASE_RENEWAL_MS, acquireCompartmentLease, renewCompartmentLease, releaseCompartmentLease, releaseCompartmentLeaseBestEffort, isCompartmentLeaseHeld, isNoContentCompartment, HAS_COMPARTMENT_CONTENT_SQL, queueM0Mutation, getMaxM0MutationId, getCompartments, getLastCompartmentEndMessage, getLastCompartmentEndMessageId, getCompartmentsByEndMessageId, appendCompartments, saveRecompStagingPass, getRecompStaging, clearRecompStaging, getRecompPartialRange, setRecompPartialRange, escapeXmlAttr, escapeXmlContent, getUncoveredCompartmentEndThrough, isPartialCompartmentEnd, readProjectDocsCanonical, providerResponseFailed, isSuccessfulProviderCompletion, requestPendingMaterialization, pendingMaterializationOrigin, hasPendingMaterializationOrigin, clearPendingMaterialization, recordSystemRefreshOwnership, recordSystemPromptForRequest, recordHeldHeadForRequest, bindSystemPromptResponse, acknowledgeSystemPromptDelivery, deferDeliveredSystemRefresh, clearSystemPromptDelivery, saveLkgSlotToDb, loadPersistedLkgSlot, drainStaleLkgSlots, createDbLkgPersistence, replayLkg, registerLkgPersistence, LKG_SNAPSHOT_ARRAY, LKG_SNAPSHOT_OBJECT, LKG_SNAPSHOT_KEY, LKG_SNAPSHOT_STRING, LKG_SNAPSHOT_NUMBER, LKG_SNAPSHOT_BOOLEAN, LKG_SNAPSHOT_NULL, LKG_SNAPSHOT_UNDEFINED, contentSnapshotValue, signatureForFields, lkgContentFields, lkgContentDigestFromFields, isLegacyLkgDigest, legacyLkgDigestFromFields, lkgFieldsMatchDigest, exactReusablePrefix, incrementalLkgContentDigests, lkgSlotRejection, captureSlot, getInMemorySlot, getSlot, dropSlot, TOOL_SWEEP_SCOPED_MARKER, encodePiContentDecision, decodePiContentDecision, getPiContentDecisions, freezePiContentDecision, getNativeReplayState, saveNativeToolInputs, addNativeReasoningIds, copySessionStateForClone, externalHistorianFailureReason, subprocessExitHistorianFailureReason, countHistorianRuns, recordHistorianRun, summarizeImportance, tallyFactsByCategory, queueMemoryMutation, getMemoryMutationsForRender, getMemoryMutationsForRenderByProjects, getMaxMemoryMutationId, getMaxMemoryMutationIdForProjects, GLOBAL_USER_PROFILE_PROJECT_PATH, getProjectState, bumpProjectUserProfileVersion, saveSourceContent, getSourceContents, recordSubagentInvocation, getLatestHistorianInvocationId, noteTouchedAt, noteNudgePickIndex, formatNoteNudge, systemPromptFingerprint, createPromptSurfaceRuntime, createPromptSurfaceGuidanceEpochCache, SYNTH_USER_ID_PREFIX, resolvePiStableId, resolvePiRealStableId, isPiContentFallbackId, isPiIndexFallbackId, piContentFallbackIds, readPiSessionSnapshot, readPiSessionMessages, readPiSessionMessagePage, findRestartModelSeedFromBranch, iterateEntriesToRawMessageRange, iterateAsyncEntriesToRawMessageRange, convertEntriesToRawMessages, convertPiAssistantEntryById, convertEntriesToRawMessagePage, countPiRawMessages, USER_MEMORY_CANDIDATE_TTL_MS, insertUserMemoryCandidates, getUserMemoryCandidates, deleteUserMemoryCandidates, pruneExpiredUserMemoryCandidates, insertUserMemory, getActiveUserMemories, updateUserMemoryContent, dismissUserMemory, getTaskScheduleState, getFailingDreamTasks, getSkippedDreamTasks, getMostRecentTaskRunAt, pruneNonCanonicalTaskRows, deleteTaskScheduleRowsForProject, seedTaskScheduleState, writeTaskScheduleState, isRetrospectiveWindowProcessed, recordRetrospectiveWindowProcessed, curateCategoryForMemoryCategory, peekCurateCategoryScope, beginCurateCategoryRun, curateTaskStateAfterSuccess, MEMORY_MURAL_BLOCK, buildMagicContextSection, resolveMuralWire, updateCompactionMarkerAfterPublication, COMPARTMENT_RENDER_EPOCH, MEMORY_RENDER_FORMAT_EPOCH, encodeCachedM0UpgradeIdentity, decodeCachedM0UpgradeIdentity, withCachedM0MemoryIds, readCachedM0MemoryIds, renderBudgetIdentityChanged, renderedBudgetSnapshot, renderedBudgetShrinkReason, DEFAULT_HISTORY_BUDGET_TOKENS, renderCompartmentAtTier, renderDecayedCompartments, extractM0Block, TEMPORAL_MARKER_PATTERN, TEMPORAL_MARKER_REPLAY_PATTERN, temporalMarkerPrefix, renderUserProfileContent, clearInjectionCache, getVisibleMemoryIds, orderHistorianMemories, renderHistorianMemoryBlock, DEFAULT_MEMORY_BUDGET_TOKENS, DEFAULT_USER_PROFILE_BUDGET_TOKENS, trimMemoriesToBudgetV2, trimWorkspaceMemoriesToBudgetV2, trimUserMemoriesToBudget, renderMemoryBlockV2, stripMemoryMuralBlock, resetPrefixTrimFallbackState, isDirectiveShapedProjectRule, takeCurateSafetyRefusalCount, wakePlaneStatus, loadPiConfig, loadPiConfigDetailed, unregisterPiProjectEmbeddings, ensureProjectRegisteredFromPiDirectory, resolvePiHarnessDetection, resolvePiHarnessKind, resolvePiOrdinalAlignment, piRawOrdinalOffsetSource, resolvePiOrdinalAlignmentForContext, isPiOrdinalAlignmentUnanchored, clearPiOrdinalAlignmentSession, rememberTodowriteToolCallTodos, parseTodos, setTodoSnapshot, registerTodoOverlay, registerTodoStateLifecycle, syncCtxMemoryToolEnabled, registerMagicContextTools };