@cortexkit/opencode-magic-context 0.49.19 → 0.49.20

package/dist/{index-3h4dp9wb.js → index-1pefp8ar.js}

@@ -1,23 +1,23 @@
 import {
   sanitizeDiagnosticText,
   sessionLog2
-} from "./index-j75v2476.js";
+} from "./index-h7tgn49j.js";
 import {
   getHarness2
 } from "./index-pavzsdc6.js";
 import {
   isTransientSqliteError2
-} from "./index-beht37jy.js";
+} from "./index-1zx2pa5d.js";
 import {
   PROMOTABLE_CATEGORIES,
   CATEGORY_DEFAULT_TTL
 } from "./index-fr0wbhdn.js";
 import {
   managedAuthorityNoteRow
-} from "./index-63k48rv9.js";
+} from "./index-pnjwa6qa.js";
 import {
   embedTextForProject2
-} from "./index-ze7mvk54.js";
+} from "./index-2yhtnfcf.js";
 import {
   saveEmbeddingIfHashMatches
 } from "./index-4dhv7mqz.js";
@@ -30,7 +30,7 @@ import {
   buildWorkspaceMemorySqlFilter2,
   getMemoryById2,
   updateMemorySeenCount2
-} from "./index-n37sebck.js";
+} from "./index-y4a3nhtk.js";
 import {
   computeNormalizedHash
 } from "./index-2bvekkfy.js";

package/dist/{index-beht37jy.js → index-1zx2pa5d.js}

@@ -752,7 +752,7 @@ async function withAsyncPrivilegedWriter2(db, operation, options = {}) {
     } catch (error) {
       if (entered || !(error instanceof SqliteAcquisitionBusyError2))
         throw error;
-      options.beforeRetry?.(error);
+      await options.beforeRetry?.(error);
       const elapsed = performance.now() - started;
       if (elapsed >= budgetMs) {
         reportSqliteDiagnostic(`sqlite writer site=privileged_writer lane=foreground acquire_ms=${Math.round(elapsed)} attempts=${attempts} outcome=busy`);

package/dist/{index-ze7mvk54.js → index-2yhtnfcf.js}

@@ -4,54 +4,60 @@ import {
   withSqliteBackgroundWriter2,
   withoutSqliteTransformPass2,
   isTransientSqliteError2
-} from "./index-beht37jy.js";
+} from "./index-1zx2pa5d.js";
 import {
   estimateTokens
-} from "./index-c5pytzne.js";
+} from "./index-b88dwwrh.js";
 import {
   getHarness2
 } from "./index-pavzsdc6.js";
 import {
   sanitizeDiagnosticText,
   log2
-} from "./index-j75v2476.js";
+} from "./index-h7tgn49j.js";
 import {
   SYNAPSE_DEFAULT_MODEL,
   isSynapseEmbeddingTruncated,
   getSynapseBatchRequestKey,
   SynapseEmbeddingProvider
-} from "./index-6x57ar45.js";
+} from "./index-adk2z5d8.js";
 import {
-  DEFAULT_COMPARTMENT_CHUNK_MAX_INPUT_TOKENS,
-  CHUNK_WINDOW_SAFETY_RATIO,
-  normalizeCompartmentChunkMaxInputTokens,
-  buildCanonicalChunkTextFromFts,
-  buildCompartmentSummaryFallbackText,
-  chunkCanonicalText,
-  chunkEmbeddingWindowsAreCurrent,
-  replaceCompartmentChunkEmbeddings,
-  loadUnembeddedShadowChunkCandidates,
-  loadUnembeddedCompartmentChunkCandidatesPolite,
-  recordChunkEmbedBackoff,
-  chunkWindowSourceKey,
-  loadUnembeddedSessionChunkCandidatesPolite,
-  countSessionCompartmentEmbedCoverage,
-  countSessionCompartmentEmbedCoveragePolite,
-  countUnembeddedSessionCompartmentsPolite
-} from "./index-b5vqckbw.js";
+  createWorkerShutdown
+} from "./index-cpjk5jd2.js";
+import {
+  DEFAULT_COMPARTMENT_CHUNK_MAX_INPUT_TOKENS2,
+  CHUNK_WINDOW_SAFETY_RATIO2,
+  normalizeCompartmentChunkMaxInputTokens2,
+  buildCanonicalChunkTextFromFts2,
+  buildCompartmentSummaryFallbackText2,
+  chunkCanonicalText2,
+  chunkEmbeddingWindowsAreCurrent2,
+  replaceCompartmentChunkEmbeddings2,
+  loadUnembeddedShadowChunkCandidates2,
+  loadUnembeddedCompartmentChunkCandidatesPolite2,
+  recordChunkEmbedBackoff2,
+  chunkWindowSourceKey2,
+  loadUnembeddedSessionChunkCandidatesPolite2,
+  countSessionCompartmentEmbedCoverage2,
+  countSessionCompartmentEmbedCoveragePolite2,
+  countUnembeddedSessionCompartmentsPolite2
+} from "./index-153r2q6z.js";
 import {
   DEFAULT_LOCAL_EMBEDDING_MODEL
 } from "./index-wgypt3cs.js";
 import {
   embeddingModelsMatch,
   resolveEmbeddingTextPrefixes,
+  getCachedLocalEmbeddingPooling,
+  logLocalEmbeddingPoolingDecisionOnce,
+  logLocalEmbeddingPoolingUnavailableOnce,
   getEmbeddingProviderIdentity,
   dominantEmbeddingFailure,
   LocalEmbeddingProvider
-} from "./index-bjcnntmz.js";
+} from "./index-0wthn7fp.js";
 import {
   setBootQuietPeriodForTests
-} from "./index-2vxtcb95.js";
+} from "./index-vkc1ca9p.js";
 import {
   isEmbeddingHostBusy2
 } from "./index-qt1qqasc.js";
@@ -62,7 +68,7 @@ import {
   acquireGitSweepLease,
   renewGitSweepLease,
   releaseGitSweepLease
-} from "./index-rr2bb45z.js";
+} from "./index-c4b76d4t.js";
 import {
   saveEmbeddingIfHashMatches,
   getMemoryEmbedCoverage,
@@ -70,7 +76,7 @@ import {
 } from "./index-4dhv7mqz.js";
 import {
   logSlowWriteTransaction
-} from "./index-k29z4aqh.js";
+} from "./index-t5pck3ss.js";
 
 // src/hooks/magic-context/embed-session-state.ts
 var embedPauseBySession = new Set;
@@ -262,7 +268,7 @@ function responseSlots(items, inputCount) {
   return slots;
 }
 function capEmbeddingInput(input, maxInputTokens, purpose) {
-  const ratio = purpose === "query" ? CHUNK_WINDOW_SAFETY_RATIO : 1;
+  const ratio = purpose === "query" ? CHUNK_WINDOW_SAFETY_RATIO2 : 1;
   const budget = Math.max(1, Math.floor(maxInputTokens * ratio));
   if (estimateTokens(input) <= budget)
     return input;
@@ -372,12 +378,12 @@ class OpenAICompatibleEmbeddingProvider {
     this.truncate = options.truncate?.trim() ?? "";
     this.rateLimitKey = rateLimitStateKey(this.endpoint, this.apiKey);
     const maxInputTokensSet = typeof options.maxInputTokens === "number" && Number.isFinite(options.maxInputTokens);
-    this.maxInputTokens = maxInputTokensSet ? Math.max(1, Math.floor(options.maxInputTokens)) : DEFAULT_COMPARTMENT_CHUNK_MAX_INPUT_TOKENS;
+    this.maxInputTokens = maxInputTokensSet ? Math.max(1, Math.floor(options.maxInputTokens)) : DEFAULT_COMPARTMENT_CHUNK_MAX_INPUT_TOKENS2;
     if (!maxInputTokensSet && this.endpoint && this.model) {
       const warningKey = `${this.endpoint}\x00${this.model}`;
       if (!maxInputTokensWarnings.has(warningKey)) {
         maxInputTokensWarnings.add(warningKey);
-        log2(`[magic-context] embedding.max_input_tokens is not set for openai-compatible model "${sanitizeDiagnosticText(this.model)}"; chunk embeddings use the ${DEFAULT_COMPARTMENT_CHUNK_MAX_INPUT_TOKENS}-token default, which is sized for the bundled local model and splits history into small pieces. Set embedding.max_input_tokens to the model's input limit (changing it re-embeds stored chunks once).`);
+        log2(`[magic-context] embedding.max_input_tokens is not set for openai-compatible model "${sanitizeDiagnosticText(this.model)}"; chunk embeddings use the ${DEFAULT_COMPARTMENT_CHUNK_MAX_INPUT_TOKENS2}-token default, which is sized for the bundled local model and splits history into small pieces. Set embedding.max_input_tokens to the model's input limit (changing it re-embeds stored chunks once).`);
       }
     }
     this.modelId = getEmbeddingProviderIdentity({
@@ -721,6 +727,7 @@ function createChunkEmbeddingWriter(db) {
   const entry = new URL(import.meta.url.endsWith(".ts") ? "./chunk-embedding-write-worker.ts" : "./chunk-embedding-write-worker.js", import.meta.url);
   const timeout = db.prepare("PRAGMA busy_timeout").get().timeout;
   const worker = inRootAsyncContext2(() => new Worker(entry, { workerData: { path, harness: getHarness2(), timeout } }));
+  const shutdown = createWorkerShutdown(worker, "chunk-embedding-writer", timeout + 2000);
   let pending;
   let failure;
   let deadline;
@@ -751,7 +758,7 @@ function createChunkEmbeddingWriter(db) {
         pending = { resolve, reject };
         deadline = inRootAsyncContext2(() => setTimeout(() => {
           fail(new Error("chunk embedding database operation timed out; coverage will be rechecked"));
-          worker.terminate();
+          shutdown.close();
         }, 30000));
         deadline.unref();
         worker.postMessage(message);
@@ -783,7 +790,7 @@ function createChunkEmbeddingWriter(db) {
     release: (project, holder) => request({ kind: "release", project, holder }),
     async close() {
       clearTimeout(deadline);
-      await worker.terminate();
+      await shutdown.close();
     }
   };
 }
@@ -1229,7 +1236,7 @@ function resolveEmbeddingConfig(config) {
       model: config?.model?.trim() || DEFAULT_LOCAL_EMBEDDING_MODEL,
       local_runtime: config?.local_runtime ?? "auto",
       ...config?.max_input_tokens ? {
-        max_input_tokens: normalizeCompartmentChunkMaxInputTokens(config.max_input_tokens)
+        max_input_tokens: normalizeCompartmentChunkMaxInputTokens2(config.max_input_tokens)
       } : {},
       ...config?.local_dtype ? { local_dtype: config.local_dtype } : {}
     };
@@ -1250,7 +1257,7 @@ function resolveEmbeddingConfig(config) {
       ...config.document_prefix !== undefined ? { document_prefix: config.document_prefix } : {},
       ...truncate ? { truncate } : {},
       ...config.max_input_tokens ? {
-        max_input_tokens: normalizeCompartmentChunkMaxInputTokens(config.max_input_tokens)
+        max_input_tokens: normalizeCompartmentChunkMaxInputTokens2(config.max_input_tokens)
       } : {}
     };
   }
@@ -1263,7 +1270,7 @@ function resolveEmbeddingConfig(config) {
     return {
       provider: "synapse",
       model: synapse.model?.trim() || "gte-modernbert-base-f16",
-      max_input_tokens: normalizeCompartmentChunkMaxInputTokens(descriptor?.max_tokens ?? synapse.max_input_tokens),
+      max_input_tokens: normalizeCompartmentChunkMaxInputTokens2(descriptor?.max_tokens ?? synapse.max_input_tokens),
       ...synapse.synapse_connection_file ? { synapse_connection_file: synapse.synapse_connection_file } : {},
       ...synapse.synapse_fingerprint ? { synapse_fingerprint: synapse.synapse_fingerprint } : {},
       ...typeof synapse.synapse_table_epoch === "number" ? { synapse_table_epoch: synapse.synapse_table_epoch } : {},
@@ -1351,7 +1358,7 @@ function getChunkEmbeddingModelId(config, providerIdentity) {
   const chunkIdentity = {
     providerIdentity,
     chunkerVersion: 2,
-    maxInputTokens: normalizeCompartmentChunkMaxInputTokens("max_input_tokens" in config ? config.max_input_tokens : undefined),
+    maxInputTokens: normalizeCompartmentChunkMaxInputTokens2("max_input_tokens" in config ? config.max_input_tokens : undefined),
     truncate: config.provider === "openai-compatible" ? config.truncate ?? "" : ""
   };
   return `${providerIdentity}:chunk:${sha256Prefix(stableStringify(chunkIdentity))}`;
@@ -1677,6 +1684,17 @@ async function drainStaleEmbeddingIdentitiesForProject2(db, projectIdentity, bud
 }
 function registerProjectEmbedding2(db, projectIdentity, config, features, sourceDirectory) {
   const resolvedConfig = resolveEmbeddingConfig(config);
+  if (resolvedConfig.provider === "local") {
+    const pooling = getCachedLocalEmbeddingPooling(resolvedConfig.model?.trim() || DEFAULT_LOCAL_EMBEDDING_MODEL);
+    if (pooling?.status === "unresolved" || !pooling) {
+      logLocalEmbeddingPoolingUnavailableOnce(resolvedConfig.model?.trim() || DEFAULT_LOCAL_EMBEDDING_MODEL, pooling?.reason ?? "pooling metadata has not been resolved by the background embedding bootstrap", log2);
+    } else {
+      logLocalEmbeddingPoolingDecisionOnce(resolvedConfig.model?.trim() || DEFAULT_LOCAL_EMBEDDING_MODEL, pooling, log2);
+    }
+    if (pooling?.status === "unsupported") {
+      return registerProjectInObservationMode2(db, projectIdentity, sourceDirectory, resolvedConfig, pooling ? `local model pooling is unsupported: ${pooling.reason}` : "local model pooling metadata has not been resolved");
+    }
+  }
   const providerIdentity = getEmbeddingProviderIdentity(resolvedConfig);
   const runtimeFingerprint = getRuntimeFingerprint(resolvedConfig);
   const chunkModelId = getChunkEmbeddingModelId(resolvedConfig, providerIdentity);
@@ -1872,7 +1890,7 @@ function shadowBackfillMissingBase(scope, primaryModelId, shadowModelId, project
 }
 function shadowBackfillMissingIds(db, projectIdentity, scope, primaryModelId, shadowModelId, limit, shadowMaxInputTokens) {
   if (scope === "chunk") {
-    return loadUnembeddedShadowChunkCandidates(db, projectIdentity, primaryModelId, shadowModelId, limit, shadowMaxInputTokens).map((candidate) => String(candidate.id));
+    return loadUnembeddedShadowChunkCandidates2(db, projectIdentity, primaryModelId, shadowModelId, limit, shadowMaxInputTokens).map((candidate) => String(candidate.id));
   }
   const { sql, params, orderBy } = shadowBackfillMissingBase(scope, primaryModelId, shadowModelId, projectIdentity);
   const rows = db.prepare(`${sql}${orderBy} LIMIT ?`).all(...params, limit);
@@ -1880,7 +1898,7 @@ function shadowBackfillMissingIds(db, projectIdentity, scope, primaryModelId, sh
 }
 function countShadowBackfillMissing(db, projectIdentity, scope, primaryModelId, shadowModelId, shadowMaxInputTokens) {
   if (scope === "chunk") {
-    return loadUnembeddedShadowChunkCandidates(db, projectIdentity, primaryModelId, shadowModelId, Number.MAX_SAFE_INTEGER, shadowMaxInputTokens).length;
+    return loadUnembeddedShadowChunkCandidates2(db, projectIdentity, primaryModelId, shadowModelId, Number.MAX_SAFE_INTEGER, shadowMaxInputTokens).length;
   }
   const { sql, params } = shadowBackfillMissingBase(scope, primaryModelId, shadowModelId, projectIdentity);
   return db.prepare(`SELECT COUNT(*) AS count FROM (${sql})`).get(...params).count;
@@ -1889,7 +1907,7 @@ function shadowModelIdForScope(registration, scope) {
   return scope === "chunk" ? registration.chunkModelId : registration.modelId;
 }
 function shadowMaxInputTokensFor(registration) {
-  return normalizeCompartmentChunkMaxInputTokens("max_input_tokens" in registration.config ? registration.config.max_input_tokens : undefined);
+  return normalizeCompartmentChunkMaxInputTokens2("max_input_tokens" in registration.config ? registration.config.max_input_tokens : undefined);
 }
 function shadowBackfillCohortFingerprint(db, projectIdentity, scope, primaryModelId) {
   if (scope === "memory") {
@@ -2227,14 +2245,14 @@ async function processShadowQueueItem(item) {
   let ftsMappingIncomplete = false;
   let emptyCanonicalText = false;
   for (const candidate of candidates) {
-    const mappedText = buildCanonicalChunkTextFromFts(db, candidate.session_id, candidate.start_message, candidate.end_message);
+    const mappedText = buildCanonicalChunkTextFromFts2(db, candidate.session_id, candidate.start_message, candidate.end_message);
     if (mappedText === null) {
       ftsMappingIncomplete = true;
     } else {
-      const text = mappedText || buildCompartmentSummaryFallbackText(db, candidate.id);
+      const text = mappedText || buildCompartmentSummaryFallbackText2(db, candidate.id);
       const shadowMaxInputTokens = shadowMaxInputTokensFor(registration);
-      const windows = chunkCanonicalText(text, candidate.start_message, candidate.end_message, shadowMaxInputTokens);
-      const windowSourceKey = chunkWindowSourceKey(text, candidate.start_message, candidate.end_message, shadowMaxInputTokens);
+      const windows = chunkCanonicalText2(text, candidate.start_message, candidate.end_message, shadowMaxInputTokens);
+      const windowSourceKey = chunkWindowSourceKey2(text, candidate.start_message, candidate.end_message, shadowMaxInputTokens);
       if (windows.length > 0)
         prepared.push({ candidate, windows, windowSourceKey });
       else
@@ -2275,7 +2293,7 @@ async function processShadowQueueItem(item) {
       ] : [];
     });
     if (rows.length === item.windows.length) {
-      replaceCompartmentChunkEmbeddings(db, rows, item.windowSourceKey);
+      replaceCompartmentChunkEmbeddings2(db, rows, item.windowSourceKey);
       writes += 1;
     } else {
       partialVectorSet = true;
@@ -2410,7 +2428,7 @@ function getProjectChunkEmbeddingModelId2(projectIdentity) {
 function getProjectEmbeddingMaxInputTokens2(projectIdentity) {
   const registration = projectRegistrations.get(projectIdentity);
   const configMax = registration?.config && "max_input_tokens" in registration.config ? registration.config.max_input_tokens : undefined;
-  return normalizeCompartmentChunkMaxInputTokens(registration?.provider?.maxInputTokens ?? configMax);
+  return normalizeCompartmentChunkMaxInputTokens2(registration?.provider?.maxInputTokens ?? configMax);
 }
 function getOrCreateProjectProvider(registration) {
   if (registration.providerIdentity === OFF_PROVIDER_IDENTITY || registration.observationMode) {
@@ -2668,7 +2686,7 @@ async function embedCompartmentChunkBatch(db, projectIdentity, batchSize, signal
   const priorModels = db.prepare(`SELECT COUNT(*) AS total,
         COALESCE(MAX(model_id = ?), 0) AS current
         FROM compartment_chunk_embeddings WHERE compartment_id = ?`);
-  const candidates = await loadUnembeddedCompartmentChunkCandidatesPolite(db, projectIdentity, snapshot.chunkModelId, batchSize, getProjectEmbeddingMaxInputTokens2(projectIdentity), true, (candidate) => {
+  const candidates = await loadUnembeddedCompartmentChunkCandidatesPolite2(db, projectIdentity, snapshot.chunkModelId, batchSize, getProjectEmbeddingMaxInputTokens2(projectIdentity), true, (candidate) => {
     const activity = activeEmbeddingSessions.get(candidate.sessionId);
     if (activity?.project === projectIdentity && activity.model === snapshot.chunkModelId && !embedPauseBySession.has(candidate.sessionId))
       return true;
@@ -2694,27 +2712,27 @@ async function embedCandidateChunkBatch(db, projectIdentity, modelId, candidates
   const prepared = [];
   for (const candidate of candidates) {
     await new Promise((resolve) => setTimeout(resolve, 0));
-    const mappedText = buildCanonicalChunkTextFromFts(db, candidate.sessionId, candidate.startMessage, candidate.endMessage);
+    const mappedText = buildCanonicalChunkTextFromFts2(db, candidate.sessionId, candidate.startMessage, candidate.endMessage);
     if (mappedText === null)
       continue;
-    const canonicalText = mappedText || buildCompartmentSummaryFallbackText(db, candidate.id);
+    const canonicalText = mappedText || buildCompartmentSummaryFallbackText2(db, candidate.id);
     if (canonicalText.length === 0) {
       if (writer)
         await writer.backoff(candidate, projectIdentity, modelId);
       else
-        recordChunkEmbedBackoff(db, candidate, projectIdentity, modelId);
+        recordChunkEmbedBackoff2(db, candidate, projectIdentity, modelId);
       noWork.push(candidate.id);
       continue;
     }
-    const windows = chunkCanonicalText(canonicalText, candidate.startMessage, candidate.endMessage, maxInputTokens);
-    if (windows.length === 0 || chunkEmbeddingWindowsAreCurrent(db, candidate.id, modelId, windows, projectIdentity)) {
+    const windows = chunkCanonicalText2(canonicalText, candidate.startMessage, candidate.endMessage, maxInputTokens);
+    if (windows.length === 0 || chunkEmbeddingWindowsAreCurrent2(db, candidate.id, modelId, windows, projectIdentity)) {
       noWork.push(candidate.id);
       continue;
     }
     prepared.push({
       candidate,
       windows,
-      windowSourceKey: chunkWindowSourceKey(canonicalText, candidate.startMessage, candidate.endMessage, maxInputTokens)
+      windowSourceKey: chunkWindowSourceKey2(canonicalText, candidate.startMessage, candidate.endMessage, maxInputTokens)
     });
   }
   if (prepared.length === 0)
@@ -2778,7 +2796,7 @@ async function embedCandidateChunkBatch(db, projectIdentity, modelId, candidates
           if (writer)
             await writer.replace(rows, item.windowSourceKey);
           else
-            replaceCompartmentChunkEmbeddings(db, rows, item.windowSourceKey);
+            replaceCompartmentChunkEmbeddings2(db, rows, item.windowSourceKey);
           persistedIds.add(item.candidate.id);
           enqueueShadowEmbeddingItems2(projectIdentity, "chunk", [
             String(item.candidate.id)
@@ -2806,7 +2824,7 @@ async function embedCandidateChunkBatch(db, projectIdentity, modelId, candidates
           if (writer)
             await writer.backoff(item.candidate, projectIdentity, modelId);
           else
-            recordChunkEmbedBackoff(db, item.candidate, projectIdentity, modelId);
+            recordChunkEmbedBackoff2(db, item.candidate, projectIdentity, modelId);
           failed.push(item.candidate.id);
         }
       }
@@ -2880,7 +2898,7 @@ async function embedSessionCompartmentChunks2(db, projectIdentity, sessionId, op
   else
     recordSessionProjectIdentity(db, sessionId, projectIdentity);
   const maxInputTokens = getProjectEmbeddingMaxInputTokens2(projectIdentity);
-  const total = writer ? await writer.count(projectIdentity, sessionId, snapshot.chunkModelId, maxInputTokens) : await countUnembeddedSessionCompartmentsPolite(db, projectIdentity, sessionId, snapshot.chunkModelId, maxInputTokens);
+  const total = writer ? await writer.count(projectIdentity, sessionId, snapshot.chunkModelId, maxInputTokens) : await countUnembeddedSessionCompartmentsPolite2(db, projectIdentity, sessionId, snapshot.chunkModelId, maxInputTokens);
   const holderId = `session-embed-${randomUUID()}`;
   const lease = writer ? await writer.acquire(projectIdentity, holderId) : acquireGitSweepLease(db, projectIdentity, holderId, { ignoreCooldown: true });
   if (!lease.acquired)
@@ -2923,7 +2941,7 @@ async function embedSessionCompartmentChunks2(db, projectIdentity, sessionId, op
           break;
         }
       }
-      const candidates = writer ? await writer.candidates(projectIdentity, sessionId, snapshot.chunkModelId, batchSize, [...skipIds, ...failedIds], maxInputTokens) : await loadUnembeddedSessionChunkCandidatesPolite(db, projectIdentity, sessionId, snapshot.chunkModelId, batchSize, [...skipIds, ...failedIds], maxInputTokens, true);
+      const candidates = writer ? await writer.candidates(projectIdentity, sessionId, snapshot.chunkModelId, batchSize, [...skipIds, ...failedIds], maxInputTokens) : await loadUnembeddedSessionChunkCandidatesPolite2(db, projectIdentity, sessionId, snapshot.chunkModelId, batchSize, [...skipIds, ...failedIds], maxInputTokens, true);
       if (candidates.length === 0)
         break;
       const {
@@ -2973,7 +2991,7 @@ async function embedSessionCompartmentChunks2(db, projectIdentity, sessionId, op
   if (aborted)
     return { status: "aborted", embedded, total, failed: failedIds.length };
   if (providerDown || failedIds.length > 0) {
-    const remaining = Math.max(failedIds.length, (writer ? await writer.count(projectIdentity, sessionId, snapshot.chunkModelId, maxInputTokens) : await countUnembeddedSessionCompartmentsPolite(db, projectIdentity, sessionId, snapshot.chunkModelId, maxInputTokens)) - skipIds.length);
+    const remaining = Math.max(failedIds.length, (writer ? await writer.count(projectIdentity, sessionId, snapshot.chunkModelId, maxInputTokens) : await countUnembeddedSessionCompartmentsPolite2(db, projectIdentity, sessionId, snapshot.chunkModelId, maxInputTokens)) - skipIds.length);
     if (remaining > 0) {
       return {
         status: "stalled",
@@ -3001,7 +3019,7 @@ function getEmbeddingCoverageStatus2(db, projectIdentity, sessionId) {
       shadowBackfillStalls: listShadowBackfillStalls2(db, projectIdentity)
     };
   }
-  const session = countSessionCompartmentEmbedCoverage(db, projectIdentity, sessionId, snapshot.chunkModelId, getProjectEmbeddingMaxInputTokens2(projectIdentity));
+  const session = countSessionCompartmentEmbedCoverage2(db, projectIdentity, sessionId, snapshot.chunkModelId, getProjectEmbeddingMaxInputTokens2(projectIdentity));
   const memories = snapshot.enabled ? getMemoryEmbedCoverage(db, projectIdentity, snapshot.modelId) : { embedded: 0, total: 0, memoryEnabled: false };
   const gitEnabled = snapshot.gitCommitEnabled;
   const commits = gitEnabled ? {
@@ -3025,7 +3043,7 @@ async function getAutoEmbeddingSessionCoverage2(db, projectIdentity, sessionId)
   if (!snapshot?.historyEnabled || snapshot.chunkModelId === "off") {
     return { enabled: false, embedded: 0, total: 0 };
   }
-  const coverage = await countSessionCompartmentEmbedCoveragePolite(db, projectIdentity, sessionId, snapshot.chunkModelId, getProjectEmbeddingMaxInputTokens2(projectIdentity));
+  const coverage = await countSessionCompartmentEmbedCoveragePolite2(db, projectIdentity, sessionId, snapshot.chunkModelId, getProjectEmbeddingMaxInputTokens2(projectIdentity));
   return { enabled: true, ...coverage };
 }
 async function sweepAllRegisteredProjects2(db, batchSize = 10) {

package/dist/{index-cf74k5ak.js → index-305y133n.js}

@@ -4,23 +4,23 @@ import {
 } from "./index-1zx87sdm.js";
 import {
   estimateTokens
-} from "./index-c5pytzne.js";
+} from "./index-b88dwwrh.js";
 import {
   getHarness2
 } from "./index-pavzsdc6.js";
 import {
   sessionLog2
-} from "./index-j75v2476.js";
+} from "./index-h7tgn49j.js";
 import {
   getTagIdentityWriteGeneration2,
   getSqliteDatabasePath2
-} from "./index-beht37jy.js";
+} from "./index-1zx2pa5d.js";
 import {
   tagIdentityRevisionKey,
   tagOrderConstraintIndex,
   stableStringify,
   stableStringifyHash
-} from "./index-bp5083x2.js";
+} from "./index-x9k3mqrf.js";
 import {
   DEFAULT_PROTECTED_TOOLS,
   normalizeProtectedToolName,

package/dist/{index-s734jcsr.js → index-35qc6606.js}

@@ -3,12 +3,15 @@ import {
 } from "./index-v0p37hm2.js";
 import {
   log2
-} from "./index-j75v2476.js";
+} from "./index-h7tgn49j.js";
 import {
   inRootAsyncContext2,
   getSqliteDatabasePath2,
   releaseIndexOwner2
-} from "./index-beht37jy.js";
+} from "./index-1zx2pa5d.js";
+import {
+  createWorkerShutdown
+} from "./index-cpjk5jd2.js";
 
 // src/hooks/magic-context/raw-ordinal-warmup.ts
 import { Worker } from "node:worker_threads";
@@ -1127,6 +1130,7 @@ function readRawSessionBoundaryEndpointsFromDb2(db, sessionId, ids, blockIds = [
 // src/hooks/magic-context/raw-ordinal-warmup.ts
 var ACCEPT_TIMEOUT_MS = 2000;
 var RESULT_TIMEOUT_MS = 10 * 60000;
+var WARMUP_CLOSE_DEADLINE_MS = 2000;
 var FAILURE_BACKOFF_MS = 5 * 60000;
 var inFlight = new Map;
 var connectionIds = new WeakMap;
@@ -1155,6 +1159,7 @@ function runSessionWarmupWorker(input, options) {
   return new Promise((resolve) => {
     const openCancellation = new Int32Array(new SharedArrayBuffer(Int32Array.BYTES_PER_ELEMENT));
     let worker;
+    let shutdown;
     let deadline;
     let settled = false;
     const finish = (result) => {
@@ -1166,7 +1171,7 @@ function runSessionWarmupWorker(input, options) {
       options.signal?.removeEventListener("abort", abort);
       if (worker) {
         activeWorkers.delete(worker);
-        const stopped = worker.terminate();
+        const stopped = shutdown?.close() ?? worker.terminate();
         worker.unref();
         stopped.finally(() => {
           releaseIndexOwner2(input);
@@ -1192,6 +1197,7 @@ function runSessionWarmupWorker(input, options) {
       worker = inRootAsyncContext2(() => new Worker(options.entry ?? defaultEntry(), {
         workerData: { ...input, openCancellation: openCancellation.buffer }
       }));
+      shutdown = createWorkerShutdown(worker, "raw-ordinal-warmup", WARMUP_CLOSE_DEADLINE_MS);
     } catch (error) {
       finish({ failure: "start-failed", detail: String(error) });
       return;