@cortexkit/opencode-magic-context 0.49.20 → 0.49.21

package/dist/{index-616rsjrv.js → index-qgh1qwjj.js}

@@ -7,7 +7,7 @@ import {
 import {
   inRootAsyncContext2,
   getSqliteDatabasePath2
-} from "./index-1zx2pa5d.js";
+} from "./index-gzy5w8ya.js";
 
 // src/features/magic-context/compaction-marker-candidates.ts
 import { existsSync, realpathSync } from "node:fs";

package/dist/{index-jr5na10t.js → index-rxqr723x.js}

@@ -1,21 +1,21 @@
 import {
   DEFAULT_LOCAL_EMBEDDING_MODEL
-} from "./index-wgypt3cs.js";
+} from "./index-nxrsns5x.js";
 import {
   log2
 } from "./index-h7tgn49j.js";
 import {
   isSynapseEmbeddingTruncated,
   SynapseEmbeddingProvider
-} from "./index-adk2z5d8.js";
+} from "./index-wfybj1dw.js";
 import {
   LocalEmbeddingProvider
-} from "./index-0wthn7fp.js";
+} from "./index-51rdvtkz.js";
 import {
   OpenAICompatibleEmbeddingProvider,
   getProjectEmbeddingSnapshot2,
   embedBatchForProject2
-} from "./index-2yhtnfcf.js";
+} from "./index-zy67whjr.js";
 import {
   isEmbeddingHostBusy2
 } from "./index-qt1qqasc.js";

package/dist/{index-dc8hy95c.js → index-syaazvy2.js}

@@ -6,7 +6,7 @@ import {
   StaleRouteHandleError,
   SubcClient,
   isConsumerReconnectTransient
-} from "./index-adk2z5d8.js";
+} from "./index-wfybj1dw.js";
 import {
   getHarness2
 } from "./index-pavzsdc6.js";
@@ -18,13 +18,13 @@ import {
 } from "./index-h7tgn49j.js";
 import {
   inRootAsyncContext2
-} from "./index-1zx2pa5d.js";
+} from "./index-gzy5w8ya.js";
 import {
   buildCanonicalChunkTextFromFts2,
   loadCompartmentChunkEmbeddingsForSearch2,
   peekCompartmentChunkSearchPool2,
   warmCompartmentChunkSearchPool2
-} from "./index-153r2q6z.js";
+} from "./index-m32esgmj.js";
 import {
   recordEmbeddingMeasurement,
   saveCommitEmbedding,
@@ -41,7 +41,7 @@ import {
   enqueueShadowEmbeddingItems2,
   getProjectEmbeddingSnapshot2,
   embedItemsForProject2
-} from "./index-2yhtnfcf.js";
+} from "./index-zy67whjr.js";
 import {
   getProjectEmbeddings2
 } from "./index-4dhv7mqz.js";
@@ -50,13 +50,13 @@ import {
   isEmbeddingEnabled,
   embedText,
   ensureMemoryEmbeddings2
-} from "./index-jr5na10t.js";
+} from "./index-rxqr723x.js";
 import {
   ModuleMemoryAuthorityError2,
   getMemoriesByProject2,
   getMemoriesByProjects2,
   updateMemoryRetrievalCount2
-} from "./index-y4a3nhtk.js";
+} from "./index-jjvbk92s.js";
 import {
   USER_FACING_FAILURES,
   getNotes,
@@ -65,27 +65,27 @@ import {
   sanitizeFtsQuery,
   searchMemoriesFTS,
   searchMemoriesFTSUnion
-} from "./index-1pefp8ar.js";
+} from "./index-3svaejkh.js";
 import {
   cleanSearchText,
   keywordRelevance,
   matchedSearchSnippet,
   getIndexedMessageCorpusSize2
-} from "./index-48stjd56.js";
+} from "./index-mnnkgq0q.js";
 import {
   MESSAGE_FTS_SESSION_FILTER_SQL,
   withMessageFtsSessionFilter
 } from "./index-kaqs6h31.js";
 import {
   isRecord
-} from "./index-305y133n.js";
+} from "./index-whcgcfrv.js";
 import {
   resolveWorkspaceShareCategories,
   resolveWorkspaceIdentitySet,
   expandWorkspaceIdentitySetWithAliases,
   resolveStoredPathWorkspaceIdentity,
   sourceNameForMemory
-} from "./index-pnjwa6qa.js";
+} from "./index-9q1g3c5q.js";
 import {
   isKnownAutocommit
 } from "./index-8ajstryz.js";
@@ -1429,6 +1429,7 @@ function sha256(value) {
 var RERANK_MODEL = "gte-reranker-modernbert-base-f32";
 var RERANK_LANE_LIMIT = 25;
 var DEADLINE_MS = 1500;
+var DOCUMENT_BYTES = 1024;
 function isProseQuery(query) {
   const text = query.trim();
   if (!text || /[\n;]|^["'`]|["'`]$/.test(text))
@@ -1508,7 +1509,7 @@ function candidateText(location, title, body) {
 ${truncateUtf8(title.replaceAll(`
 `, " "), 120)}
 `;
-  return header + truncateUtf8(body, 1024 - Buffer.byteLength(header));
+  return header + truncateUtf8(body, DOCUMENT_BYTES - Buffer.byteLength(header));
 }
 
 class RerankFailure extends Error {
@@ -1578,7 +1579,79 @@ async function connect(connectionFile) {
   }
   return bounded(connection);
 }
-async function prepareReranker(projectRoot, sessionId, connectionFile, injectedClient) {
+async function remoteScores(config, query, documents, signal) {
+  const controller = new AbortController;
+  let timer;
+  let onAbort;
+  try {
+    const cancelled = new Promise((_, reject) => {
+      const cancel = (reason) => {
+        reject(new RerankFailure(reason));
+        controller.abort();
+      };
+      onAbort = () => cancel("aborted");
+      if (signal?.aborted)
+        onAbort();
+      else
+        signal?.addEventListener("abort", onAbort, { once: true });
+      timer = setTimeout(() => cancel("timeout"), Math.max(50, Math.min(15000, config.timeout_ms ?? DEADLINE_MS)));
+    });
+    const request = async () => {
+      const tei = config.endpoint?.startsWith("tei+");
+      const endpoint = (tei ? config.endpoint?.slice(4) : config.endpoint)?.replace(/\/+$/, "");
+      if (!endpoint || !/^https?:\/\//.test(endpoint))
+        throw new RerankFailure("invalid_endpoint");
+      const response = await fetch(`${endpoint}/rerank`, {
+        method: "POST",
+        headers: {
+          "Content-Type": "application/json",
+          ...config.api_key ? { Authorization: `Bearer ${config.api_key}` } : {}
+        },
+        body: JSON.stringify(tei ? { query, texts: documents } : {
+          model: config.model,
+          query,
+          documents,
+          top_n: documents.length
+        }),
+        signal: controller.signal,
+        redirect: "error"
+      });
+      if (!response.ok)
+        throw new RerankFailure(`http_${response.status}`);
+      const body = await response.json();
+      const rows = tei ? body : body?.results;
+      if (!Array.isArray(rows) || rows.length !== documents.length)
+        throw new RerankFailure("invalid_reply");
+      const scores = new Array(documents.length);
+      const seen = new Set;
+      for (const row of rows) {
+        const index = row?.index;
+        const score = tei ? row?.score : row?.relevance_score;
+        if (!Number.isInteger(index) || index < 0 || index >= documents.length || seen.has(index) || typeof score !== "number" || !Number.isFinite(score))
+          throw new RerankFailure("invalid_reply");
+        seen.add(index);
+        scores[index] = score;
+      }
+      return scores;
+    };
+    return await Promise.race([cancelled, request()]);
+  } finally {
+    clearTimeout(timer);
+    if (onAbort)
+      signal?.removeEventListener("abort", onAbort);
+  }
+}
+function remoteRerankSkipped(reason, onSkipped) {
+  const safe = /^(?:invalid_reply|invalid_endpoint|validation|timeout|aborted|candidate_validation|http_\d{3})$/.test(reason) ? reason : "remote_error";
+  const line = `Reranking skipped: ${safe}`;
+  logRerankFallback(safe);
+  onSkipped?.(line);
+}
+async function prepareReranker(projectRoot, sessionId, connectionFile, injectedClient, config) {
+  if (config?.provider === "off")
+    return null;
+  if (config?.provider === "remote")
+    return { remote: config };
   try {
     if (injectedClient === null)
       throw new RerankFailure("missing_synapse");
@@ -1595,36 +1668,43 @@ async function prepareReranker(projectRoot, sessionId, connectionFile, injectedC
     return null;
   }
 }
-async function rerankSearchResults(session, query, baseline, pool, texts, limit) {
+async function rerankSearchResults(session, query, baseline, pool, texts, limit, signal, onSkipped) {
   if (!pool.length)
     return baseline;
   try {
-    if (pool.length > 20 || texts.length !== pool.length || texts.some((text) => Buffer.byteLength(text) > 1024))
-      throw new RerankFailure("validation");
-    const body = unwrap(await bounded(session.client.call("synapse", "rerank.score", {
-      model: RERANK_MODEL,
-      query,
-      candidates: texts,
-      required_fingerprint: session.lane.fingerprint,
-      allow_equivalent: false,
-      accept_declared: false,
-      max_queue_ms: 100,
-      deadline_ms: DEADLINE_MS
-    }, {
-      timeoutMs: DEADLINE_MS,
-      targetKind: "management_surface",
-      identity: session.identity
-    })));
-    if (body.fingerprint !== session.lane.fingerprint)
-      throw new RerankFailure("fingerprint_mismatch");
-    if (body.table_epoch !== session.lane.tableEpoch)
-      throw new RerankFailure("epoch_mismatch");
-    if (!Array.isArray(body.scores) || body.scores.length !== pool.length || body.scores.some((score) => typeof score !== "number" || !Number.isFinite(score)))
+    if (pool.length > 20 || texts.length !== pool.length || texts.some((text) => Buffer.byteLength(text) > DOCUMENT_BYTES))
       throw new RerankFailure("validation");
-    const scores = body.scores;
+    let scores;
+    if ("remote" in session) {
+      scores = await remoteScores(session.remote, query, texts, signal);
+    } else {
+      const body = unwrap(await bounded(session.client.call("synapse", "rerank.score", {
+        model: RERANK_MODEL,
+        query,
+        candidates: texts,
+        required_fingerprint: session.lane.fingerprint,
+        allow_equivalent: false,
+        max_queue_ms: 100,
+        deadline_ms: DEADLINE_MS
+      }, {
+        timeoutMs: DEADLINE_MS,
+        targetKind: "management_surface",
+        identity: session.identity
+      })));
+      if (body.fingerprint !== session.lane.fingerprint)
+        throw new RerankFailure("fingerprint_mismatch");
+      if (body.table_epoch !== session.lane.tableEpoch)
+        throw new RerankFailure("epoch_mismatch");
+      if (!Array.isArray(body.scores) || body.scores.length !== pool.length || body.scores.some((score) => typeof score !== "number" || !Number.isFinite(score)))
+        throw new RerankFailure("validation");
+      scores = body.scores;
+    }
     return pool.map((hit, index) => ({ hit, index, score: scores[index] })).sort((a, b) => b.score - a.score || a.index - b.index).slice(0, limit).map((row) => row.hit);
   } catch (error) {
-    logRerankFallback(reasonOf(error));
+    if ("remote" in session)
+      remoteRerankSkipped(error instanceof RerankFailure ? error.reason : "remote_error", onSkipped);
+    else
+      logRerankFallback(reasonOf(error));
     return baseline;
   }
 }
@@ -2945,7 +3025,7 @@ async function unifiedSearch2(db, sessionId, projectPath, query, options = {}) {
     return [];
   }
   const limit = normalizeLimit(options.limit);
-  const reranker = options.explicitSearch && options.rerank !== false && isProseQuery(trimmedQuery) ? await prepareReranker(projectPath, sessionId, options.rerankConnectionFile, options.rerankClient) : null;
+  const reranker = options.explicitSearch && options.rerank !== false && isProseQuery(trimmedQuery) ? await prepareReranker(projectPath, sessionId, options.rerankConnectionFile, options.rerankClient, options.rerankConfig) : null;
   const lanes = reranker || options.captureRerankLanes ? emptySearchLanes() : undefined;
   const dateRange = normalizeDateRange(options.from, options.to);
   const tierLimit = Math.max(limit * 3, DEFAULT_UNIFIED_SEARCH_LIMIT);
@@ -3115,11 +3195,14 @@ async function unifiedSearch2(db, sessionId, projectPath, query, options = {}) {
     options.captureRerankLanes?.(lanes);
   }
   if (reranker && lanes) {
-    const pool = fuseSearchLanes(lanes);
+    const pool = fuseSearchLanes(lanes).slice(0, "remote" in reranker ? reranker.remote.top_n : 20);
     try {
-      results = await rerankSearchResults(reranker, trimmedQuery, results, pool, candidateTexts(db, sessionId, projectPath, pool), limit);
+      results = await rerankSearchResults(reranker, trimmedQuery, results, pool, candidateTexts(db, sessionId, projectPath, pool), limit, options.signal, options.onRerankSkipped);
     } catch {
-      logRerankFallback("candidate_validation");
+      if ("remote" in reranker)
+        remoteRerankSkipped("candidate_validation", options.onRerankSkipped);
+      else
+        logRerankFallback("candidate_validation");
     }
   }
   if (!options.measurementDisabled) {

package/dist/{index-adk2z5d8.js → index-wfybj1dw.js}

@@ -9,7 +9,7 @@ import {
 } from "./index-h7tgn49j.js";
 import {
   inRootAsyncContext2
-} from "./index-1zx2pa5d.js";
+} from "./index-gzy5w8ya.js";
 
 // ../../node_modules/.bun/@cortexkit+subc-client@0.11.1/node_modules/@cortexkit/subc-client/dist/client.js
 import { promises as fs2 } from "node:fs";
@@ -1786,7 +1786,6 @@ function getSynapseBatchRequestKey(args) {
     required_fingerprint: args.fingerprint,
     required_epoch: args.tableEpoch,
     allow_equivalent: false,
-    accept_declared: false,
     ids: args.items.map((item) => item.id),
     content_sha256: args.items.map((item) => item.contentSha256),
     ...args.purpose === "query" ? { purpose: args.purpose } : {}
@@ -2175,8 +2174,7 @@ class SynapseEmbeddingProvider {
       model: metadata.model,
       required_fingerprint: metadata.fingerprint,
       required_epoch: metadata.table_epoch,
-      allow_equivalent: false,
-      accept_declared: false
+      allow_equivalent: false
     };
   }
   batchRequest(items, requestKey, purpose) {

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

@@ -14,7 +14,7 @@ import {
 import {
   getTagIdentityWriteGeneration2,
   getSqliteDatabasePath2
-} from "./index-1zx2pa5d.js";
+} from "./index-gzy5w8ya.js";
 import {
   tagIdentityRevisionKey,
   tagOrderConstraintIndex,
@@ -88,13 +88,16 @@ function stripDanglingTagNotationGlobally(value) {
 function stripTagSectionCharacters(value) {
   return value.replace(STRAY_SECTION_CHAR_REGEX, "");
 }
-function stripPersistedAssistantText(value) {
+function stripPersistedAssistantText(value, acceptsEmpty = true) {
+  if (!acceptsEmpty && !value.includes("§") && !value.trim())
+    return value;
   let text = stripWellFormedLeadingTagPrefix(value);
   text = stripCompleteTagPairsGlobally(text);
   text = stripMalformedTagNotationGlobally(text);
   text = stripDanglingTagNotationGlobally(text);
   text = stripTagSectionCharacters(text);
-  return text.trim();
+  const stripped = text.trim();
+  return !acceptsEmpty && value.trim().length > 0 && stripped.length === 0 ? value : stripped;
 }
 function byteSize(value) {
   return encoder.encode(value).length;

package/dist/{index-jmycsey7.js → index-x6cbrv9h.js}

@@ -2,10 +2,10 @@ import {
   getSqliteDatabasePath2,
   withSqliteBackgroundWriter2,
   withoutSqliteTransformPass2
-} from "./index-1zx2pa5d.js";
+} from "./index-gzy5w8ya.js";
 import {
   runSessionWarmupWorker
-} from "./index-35qc6606.js";
+} from "./index-jfem3dgz.js";
 import {
   logSlowWriteTransaction
 } from "./index-t5pck3ss.js";

package/dist/{index-v7r4td4c.js → index-y76rhsnr.js}

@@ -97,7 +97,7 @@ import {
   getRawSessionTagKeysThrough2,
   getLegacyProtectedTailStartOrdinal2,
   readSessionChunk2
-} from "./index-x4k3f4mf.js";
+} from "./index-mnhcdfxg.js";
 import {
   getHarness2,
   harnessOwnsOpenCodeStore2
@@ -147,7 +147,7 @@ import {
   ConfigProfilesSchema,
   MagicContextConfigSchema,
   LIVE_RELOAD_CONFIG_PATHS
-} from "./index-wgypt3cs.js";
+} from "./index-nxrsns5x.js";
 import {
   providerMass,
   calibrationForModelKey,
@@ -164,7 +164,7 @@ import {
   LATEST_SUPPORTED_VERSION,
   getPersistedSchemaVersion,
   openDatabase
-} from "./index-rmd0069m.js";
+} from "./index-a9zvn5fe.js";
 import {
   inRootAsyncContext2,
   Database2,
@@ -172,11 +172,11 @@ import {
   withSqliteBackgroundWriter2,
   withoutSqliteTransformPass2,
   beginSqliteWriterAsync2
-} from "./index-1zx2pa5d.js";
+} from "./index-gzy5w8ya.js";
 import {
   shouldSkipHomeProjectMemory,
   resolveProjectIdentity
-} from "./index-vkc1ca9p.js";
+} from "./index-kq5w68dp.js";
 import {
   isPidAlive
 } from "./index-c7phnbvk.js";
@@ -185,7 +185,7 @@ import {
   noteRawSessionSummaryRowWritten2,
   RawOrdinalsNotReady2,
   runSessionWarmupWorker
-} from "./index-35qc6606.js";
+} from "./index-jfem3dgz.js";
 import {
   layoutLkgPrefix,
   assembleLkgPrefix,
@@ -196,7 +196,7 @@ import {
   resolveStoredPathWorkspaceIdentity,
   sourceNameForMemory,
   computeWorkspaceEpochFingerprint
-} from "./index-pnjwa6qa.js";
+} from "./index-9q1g3c5q.js";
 import {
   getTableColumnNames
 } from "./index-x9k3mqrf.js";
@@ -222,7 +222,7 @@ import {
   getProjectChunkEmbeddingModelId2,
   getProjectEmbeddingMaxInputTokens2,
   embedItemsForProject2
-} from "./index-2yhtnfcf.js";
+} from "./index-zy67whjr.js";
 import {
   buildCanonicalChunkTextFromFts2,
   buildCompartmentSummaryFallbackText2,
@@ -232,7 +232,7 @@ import {
   replaceCompartmentChunkEmbeddings2,
   deleteChunkEmbedBackoffForSession2,
   chunkWindowSourceKey2
-} from "./index-153r2q6z.js";
+} from "./index-m32esgmj.js";
 import {
   BoundedSessionMap,
   clearCompressionDepth,
@@ -249,7 +249,7 @@ import {
   toMessageIndexSource2,
   clearIndexedMessagesInTransaction2,
   indexSourcesAfterOrdinal2
-} from "./index-48stjd56.js";
+} from "./index-mnnkgq0q.js";
 import {
   stripWellFormedLeadingTagPrefix,
   stripTagPrefix,
@@ -262,12 +262,12 @@ import {
   foldShrunkPartTags2,
   getTagNumberByMessageId2,
   getActiveTagsBySession2
-} from "./index-305y133n.js";
+} from "./index-whcgcfrv.js";
 import {
   openCodeDbExists,
   withReadOnlySessionDb,
   getMessageTimesFromOpenCodeDb
-} from "./index-11zf2wwc.js";
+} from "./index-ksq50rg0.js";
 import {
   renderUserFacingFailure,
   userFacingFailureCode,
@@ -276,7 +276,7 @@ import {
   insertPrimerCandidates,
   promoteSessionFactsDurable,
   embedPromotedFacts
-} from "./index-1pefp8ar.js";
+} from "./index-3svaejkh.js";
 import {
   pushNotification2
 } from "./index-x8a7jmkh.js";
@@ -287,16 +287,16 @@ import {
   getMemoriesByProject2,
   getMemoriesByProjects2,
   getMaxMemoryIdForProjects2
-} from "./index-y4a3nhtk.js";
+} from "./index-jjvbk92s.js";
 import {
   isMarkerCandidateBasisCurrent2,
   invalidateMarkerCandidateBasis2,
   readMarkerCandidateBasis2,
   prepareMarkerCandidateBasis2
-} from "./index-616rsjrv.js";
+} from "./index-qgh1qwjj.js";
 import {
   sendStatusNotification2
-} from "./index-kc9pn495.js";
+} from "./index-66eby7hx.js";
 import {
   isCompactionEnabled,
   migrateLegacyAgentEnabledInMemory
@@ -1702,6 +1702,17 @@ function stripUnsafeProjectConfigFields(projectRaw) {
       warnings.push(`Ignoring ${field} from project config (security: daemon routing and developer-only embedding traffic are user-level settings).`);
     }
   }
+  if ("rerank" in projectRaw) {
+    const rerank = projectRaw.rerank;
+    const off = isPlainObject(rerank) && rerank.provider === "off";
+    const removed = !isPlainObject(rerank) || Object.keys(rerank).some((key) => key !== "provider") || !off;
+    if (off)
+      projectRaw.rerank = { provider: "off" };
+    else
+      delete projectRaw.rerank;
+    if (removed)
+      warnings.push("Ignoring rerank settings from project config (security: projects may only set rerank.provider to off; hosted routing, credentials and limits are user-level only).");
+  }
   const embedding = projectRaw.embedding;
   if (isPlainObject(embedding)) {
     const removed = [];
@@ -9372,7 +9383,7 @@ async function promptSyncWithValidatedOutputRetry(client, args, options) {
       throwWithPromptFailure(failure.error, failedAttempts, args, timeoutMs, options.transport);
     }
     if (!cooledPrimary && !quota)
-      log2(`[${callContext}] primary (${explicitPrimaryLabel}) failed validation/prompt: ${shortErr(failure.error)}; trying ${fallbacks.length} fallback(s)`);
+      log2(`[${callContext}] primary (${explicitPrimaryLabel}) failed validation/prompt: ${shortErr(failure.error)}; ${fallbacks.length} fallback(s) configured, subject to remaining budget`);
   }
   for (let i = 0;i < fallbacks.length; i += 1) {
     const fallback = toModelEntry(fallbacks[i]);

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

@@ -4,7 +4,7 @@ import {
   withSqliteBackgroundWriter2,
   withoutSqliteTransformPass2,
   isTransientSqliteError2
-} from "./index-1zx2pa5d.js";
+} from "./index-gzy5w8ya.js";
 import {
   estimateTokens
 } from "./index-b88dwwrh.js";
@@ -20,7 +20,7 @@ import {
   isSynapseEmbeddingTruncated,
   getSynapseBatchRequestKey,
   SynapseEmbeddingProvider
-} from "./index-adk2z5d8.js";
+} from "./index-wfybj1dw.js";
 import {
   createWorkerShutdown
 } from "./index-cpjk5jd2.js";
@@ -41,10 +41,10 @@ import {
   countSessionCompartmentEmbedCoverage2,
   countSessionCompartmentEmbedCoveragePolite2,
   countUnembeddedSessionCompartmentsPolite2
-} from "./index-153r2q6z.js";
+} from "./index-m32esgmj.js";
 import {
   DEFAULT_LOCAL_EMBEDDING_MODEL
-} from "./index-wgypt3cs.js";
+} from "./index-nxrsns5x.js";
 import {
   embeddingModelsMatch,
   resolveEmbeddingTextPrefixes,
@@ -54,10 +54,10 @@ import {
   getEmbeddingProviderIdentity,
   dominantEmbeddingFailure,
   LocalEmbeddingProvider
-} from "./index-0wthn7fp.js";
+} from "./index-51rdvtkz.js";
 import {
   setBootQuietPeriodForTests
-} from "./index-vkc1ca9p.js";
+} from "./index-kq5w68dp.js";
 import {
   isEmbeddingHostBusy2
 } from "./index-qt1qqasc.js";
@@ -68,7 +68,7 @@ import {
   acquireGitSweepLease,
   renewGitSweepLease,
   releaseGitSweepLease
-} from "./index-c4b76d4t.js";
+} from "./index-kfyb3k6n.js";
 import {
   saveEmbeddingIfHashMatches,
   getMemoryEmbedCoverage,
@@ -1069,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;
@@ -3111,6 +3128,7 @@ function _resetProjectEmbeddingRegistryForTests2() {
     disposeProvider(registration.provider);
   }
   projectRegistrations.clear();
+  projectContextUsers.clear();
   shadowRegistrations.clear();
   shadowQueue.length = 0;
   pendingShadowBackfills.clear();
@@ -3126,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 };