@cortexkit/opencode-magic-context 0.49.19 → 0.49.21

package/dist/{index-ze7mvk54.js → index-zy67whjr.js}

@@ -4,54 +4,60 @@ import {
   withSqliteBackgroundWriter2,
   withoutSqliteTransformPass2,
   isTransientSqliteError2
-} from "./index-beht37jy.js";
+} from "./index-gzy5w8ya.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-wfybj1dw.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-m32esgmj.js";
 import {
   DEFAULT_LOCAL_EMBEDDING_MODEL
-} from "./index-wgypt3cs.js";
+} from "./index-nxrsns5x.js";
 import {
   embeddingModelsMatch,
   resolveEmbeddingTextPrefixes,
+  getCachedLocalEmbeddingPooling,
+  logLocalEmbeddingPoolingDecisionOnce,
+  logLocalEmbeddingPoolingUnavailableOnce,
   getEmbeddingProviderIdentity,
   dominantEmbeddingFailure,
   LocalEmbeddingProvider
-} from "./index-bjcnntmz.js";
+} from "./index-51rdvtkz.js";
 import {
   setBootQuietPeriodForTests
-} from "./index-2vxtcb95.js";
+} from "./index-kq5w68dp.js";
 import {
   isEmbeddingHostBusy2
 } from "./index-qt1qqasc.js";
@@ -62,7 +68,7 @@ import {
   acquireGitSweepLease,
   renewGitSweepLease,
   releaseGitSweepLease
-} from "./index-rr2bb45z.js";
+} from "./index-kfyb3k6n.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();
     }
   };
 }
@@ -1062,6 +1069,23 @@ var EMBED_SLICE_RETRY_BASE_MS = 250;
 var EMBED_SLOW_FAILURE_NO_RETRY_MS = 1e4;
 var MAX_CONSECUTIVE_FAILED_BATCHES = 3;
 var projectRegistrations = new Map;
+var projectContextUsers = new Map;
+function retainProjectEmbedding2(projectIdentity) {
+  projectContextUsers.set(projectIdentity, (projectContextUsers.get(projectIdentity) ?? 0) + 1);
+  let released = false;
+  return () => {
+    if (released)
+      return;
+    released = true;
+    const remaining = (projectContextUsers.get(projectIdentity) ?? 1) - 1;
+    if (remaining > 0) {
+      projectContextUsers.set(projectIdentity, remaining);
+    } else {
+      projectContextUsers.delete(projectIdentity);
+      unregisterProjectEmbedding2(projectIdentity);
+    }
+  };
+}
 var shadowRegistrations = new Map;
 var shadowQueue = [];
 var shadowWorker = null;
@@ -1229,7 +1253,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 +1274,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 +1287,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 +1375,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 +1701,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 +1907,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 +1915,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 +1924,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 +2262,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 +2310,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 +2445,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 +2703,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 +2729,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 +2813,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 +2841,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 +2915,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 +2958,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 +3008,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 +3036,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 +3060,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) {
@@ -3093,6 +3128,7 @@ function _resetProjectEmbeddingRegistryForTests2() {
     disposeProvider(registration.provider);
   }
   projectRegistrations.clear();
+  projectContextUsers.clear();
   shadowRegistrations.clear();
   shadowQueue.length = 0;
   pendingShadowBackfills.clear();
@@ -3108,4 +3144,4 @@ function _resetProjectEmbeddingRegistryForTests2() {
   setBootQuietPeriodForTests(null);
 }
 
-export { embedPauseBySession, embedRunStateBySession, activeEmbeddingSessions, autoEmbedAttemptedBySession, autoEmbedIdentityBySession, autoEmbedRetryAfterBySession, invalidateAutoEmbedSession, getEmbedDrainUiStatus, clearEmbedSessionState, drainBackgroundBatches, recordEmbeddingMeasurement, OpenAICompatibleEmbeddingProvider, saveCommitEmbedding, loadProjectCommitEmbeddings, loadUnembeddedCommits, countEmbeddedCommits, upsertCommits, getLatestIndexedCommitSha, enforceProjectCap, describeShadowBackfillWriteRefusal2, listShadowBackfillStalls2, getShadowBackfillStopReason2, markProjectLoadUntrusted2, contentSha2562, drainProjectEmbeddingIdentityMaintenance2, drainStaleEmbeddingIdentitiesForProject2, registerProjectEmbedding2, registerProjectShadowEmbedding2, getShadowEmbeddingMeasurementCohort2, getPrimaryEmbeddingMeasurementCohort2, embedShadowTextForProject2, enqueueShadowEmbeddingItems2, getShadowBackfillRemaining2, flushShadowEmbeddingBacklog2, registerProjectInObservationMode2, unregisterProjectShadowEmbedding2, unregisterProjectEmbedding2, getProjectEmbeddingSnapshot2, installProjectEmbeddingSearchBridge2, getProjectChunkEmbeddingModelId2, getProjectEmbeddingMaxInputTokens2, embedTextForProject2, embedBatchForProject2, embedItemsForProject2, embedUnembeddedMemoriesForProject2, embedUnembeddedMemoriesForProjectOutcome2, drainCommitBacklogForProject2, embedUnembeddedCompartmentChunksForProject2, embedSessionCompartmentChunks2, getEmbeddingCoverageStatus2, getAutoEmbeddingSessionCoverage2, sweepAllRegisteredProjects2, _setTestProviderFactoryForProject2, _resetProjectEmbeddingRegistryForTests2 };
+export { embedPauseBySession, embedRunStateBySession, activeEmbeddingSessions, autoEmbedAttemptedBySession, autoEmbedIdentityBySession, autoEmbedRetryAfterBySession, invalidateAutoEmbedSession, getEmbedDrainUiStatus, clearEmbedSessionState, drainBackgroundBatches, recordEmbeddingMeasurement, OpenAICompatibleEmbeddingProvider, saveCommitEmbedding, loadProjectCommitEmbeddings, loadUnembeddedCommits, countEmbeddedCommits, upsertCommits, getLatestIndexedCommitSha, enforceProjectCap, describeShadowBackfillWriteRefusal2, listShadowBackfillStalls2, retainProjectEmbedding2, getShadowBackfillStopReason2, markProjectLoadUntrusted2, contentSha2562, drainProjectEmbeddingIdentityMaintenance2, drainStaleEmbeddingIdentitiesForProject2, registerProjectEmbedding2, registerProjectShadowEmbedding2, getShadowEmbeddingMeasurementCohort2, getPrimaryEmbeddingMeasurementCohort2, embedShadowTextForProject2, enqueueShadowEmbeddingItems2, getShadowBackfillRemaining2, flushShadowEmbeddingBacklog2, registerProjectInObservationMode2, unregisterProjectShadowEmbedding2, unregisterProjectEmbedding2, getProjectEmbeddingSnapshot2, installProjectEmbeddingSearchBridge2, getProjectChunkEmbeddingModelId2, getProjectEmbeddingMaxInputTokens2, embedTextForProject2, embedBatchForProject2, embedItemsForProject2, embedUnembeddedMemoriesForProject2, embedUnembeddedMemoriesForProjectOutcome2, drainCommitBacklogForProject2, embedUnembeddedCompartmentChunksForProject2, embedSessionCompartmentChunks2, getEmbeddingCoverageStatus2, getAutoEmbeddingSessionCoverage2, sweepAllRegisteredProjects2, _setTestProviderFactoryForProject2, _resetProjectEmbeddingRegistryForTests2 };