@atlaskit/editor-plugin-autocomplete 8.0.0 → 9.1.0

package/dist/cjs/pm-plugins/scoring-pipeline.js

@@ -4,12 +4,13 @@ var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefau
 Object.defineProperty(exports, "__esModule", {
   value: true
 });
-exports.loadGrammarDataAsync = exports.isGrammarDataLoaded = exports.STAGE2_WEIGHT = exports.STAGE1_WEIGHT = exports.MIN_STAGE1_SCORE = void 0;
+exports.loadGrammarDataAsync = exports.isGrammarDataLoaded = exports.STAGE2_WEIGHT = exports.STAGE1_WEIGHT = exports.MIN_WINNER_MARGIN = exports.MIN_STAGE1_SCORE = void 0;
 exports.rankCandidates = rankCandidates;
 var _regenerator = _interopRequireDefault(require("@babel/runtime/regenerator"));
 var _defineProperty2 = _interopRequireDefault(require("@babel/runtime/helpers/defineProperty"));
 var _asyncToGenerator2 = _interopRequireDefault(require("@babel/runtime/helpers/asyncToGenerator"));
 var _slicedToArray2 = _interopRequireDefault(require("@babel/runtime/helpers/slicedToArray"));
+var _typeof2 = _interopRequireDefault(require("@babel/runtime/helpers/typeof"));
 var _ufo = require("../analytics/ufo");
 var _artifactsManifest = require("./artifacts-manifest");
 var _debugMode = require("./debug-mode");
@@ -28,6 +29,11 @@ function _objectSpread(e) { for (var r = 1; r < arguments.length; r++) { var t =
  */
 // ─── Types ──────────────────────────────────────────────────
 
+/**
+ * A completion term is `(term, termType)`. `word` is the v1 default; `bigram`
+ * (2 tokens) and `phrase` (3 tokens) are multi-word units mined by the producer.
+ */
+
 /** Metadata returned by the grammar filter for debug logging in the caller. */
 
 // ─── Scoring Constants ──────────────────────────────────────
@@ -35,9 +41,25 @@ function _objectSpread(e) { for (var r = 1; r < arguments.length; r++) { var t =
 var ALPHA = 0.5;
 var BETA = 0.5;
 var NEUTRAL_SCORE = 0.5;
+// Multiplicative Stage-1 boost applied to a multi-word term proportional to its
+// (clamped [0,1]) NPMI association, so a strongly-collocated phrase outranks a
+// coincidental word sequence of comparable freq/semantics. Tunable.
+var ASSOCIATION_BOOST = 0.5;
 var STAGE1_WEIGHT = exports.STAGE1_WEIGHT = 0.35;
 var STAGE2_WEIGHT = exports.STAGE2_WEIGHT = 0.65;
 var MIN_STAGE1_SCORE = exports.MIN_STAGE1_SCORE = 0.35;
+/**
+ * How far the leading candidate must sit above the runner-up on the blended
+ * score before a ghost is committed.
+ *
+ * Lives beside the weights it is measured in, because it is only meaningful on
+ * that scale — and because both the arbitration that applies it and the
+ * scheduler that decides when reading more tokens can no longer change the
+ * outcome have to use the same number. When those two drifted apart, the
+ * scheduler stopped early on a margin the arbitration did not recognise and the
+ * abstention that followed was invisible from either side.
+ */
+var MIN_WINNER_MARGIN = exports.MIN_WINNER_MARGIN = 0.08;
 var L1_SESSION_CAP = 1.2;
 // Minimum prefix-payload max LM probability before Stage 2 activates.
 // Below this threshold the LM signal is too weak to suppress Stage 1 — finalScore
@@ -56,13 +78,38 @@ var posTags = new Map();
  * without suggesting these words to the user.
  */
 var ghostPosTags = {};
+var grammarDataPromise;
 
 /**
  * Precomputed map from each POS tag to the set of allowed next POS tags, so
  * applyGrammarFilter never re-iterates the transition rules per call.
  */
 var precomputedAllowedByPos = new Map();
-var grammarDataPromise;
+var isPosTagsRecord = function isPosTagsRecord(payload) {
+  if (payload == null || (0, _typeof2.default)(payload) !== 'object') {
+    return false;
+  }
+  return Object.values(payload).every(function (value) {
+    return Array.isArray(value) && value.every(function (entry) {
+      return typeof entry === 'string';
+    });
+  });
+};
+var isGrammarTransitions = function isGrammarTransitions(payload) {
+  if (payload == null || (0, _typeof2.default)(payload) !== 'object') {
+    return false;
+  }
+  var transitions = payload.transitions;
+  if (transitions == null || (0, _typeof2.default)(transitions) !== 'object') {
+    return false;
+  }
+  return Object.values(transitions).every(function (value) {
+    var allowed = value === null || value === void 0 ? void 0 : value.allowed;
+    return Array.isArray(allowed) && allowed.every(function (entry) {
+      return typeof entry === 'string';
+    });
+  });
+};
 var setGrammarData = function setGrammarData(data) {
   var _data$grammarTransiti;
   posTags = new Map(Object.entries(data.posTags));
@@ -116,9 +163,27 @@ var loadGrammarDataAsync = exports.loadGrammarDataAsync = function loadGrammarDa
           }
           throw new Error("[scoring-pipeline] ".concat(_artifactsManifest.ARTIFACT_NAME.GRAMMAR_TRANSITIONS, " is missing its `transitions` wrapper"));
         case 3:
+          if (isGrammarTransitions(grammarTransitionsData)) {
+            _context.next = 4;
+            break;
+          }
+          throw new Error("[scoring-pipeline] ".concat(_artifactsManifest.ARTIFACT_NAME.GRAMMAR_TRANSITIONS, " payload shape was invalid"));
+        case 4:
+          if (isPosTagsRecord(posTagsData)) {
+            _context.next = 5;
+            break;
+          }
+          throw new Error("[scoring-pipeline] ".concat(_artifactsManifest.ARTIFACT_NAME.POS_TAGS, " payload shape was invalid"));
+        case 5:
+          if (isPosTagsRecord(ghostPosTagsData)) {
+            _context.next = 6;
+            break;
+          }
+          throw new Error("[scoring-pipeline] ".concat(_artifactsManifest.ARTIFACT_NAME.GHOST_POS_TAGS, " payload shape was invalid"));
+        case 6:
           setGrammarData({
-            posTags: posTagsData !== null && posTagsData !== void 0 ? posTagsData : {},
-            ghostPosTags: ghostPosTagsData !== null && ghostPosTagsData !== void 0 ? ghostPosTagsData : {},
+            posTags: posTagsData,
+            ghostPosTags: ghostPosTagsData,
             grammarTransitions: grammarTransitionsData
           });
           (0, _ufo.succeedExp)(_ufo.EXPERIENCE_NAME.LOAD_GRAMMAR, 'singleton', _objectSpread({
@@ -136,10 +201,10 @@ var loadGrammarDataAsync = exports.loadGrammarDataAsync = function loadGrammarDa
               transitionCount: precomputedAllowedByPos.size
             });
           }
-          _context.next = 5;
+          _context.next = 8;
           break;
-        case 4:
-          _context.prev = 4;
+        case 7:
+          _context.prev = 7;
           _t = _context["catch"](1);
           (0, _ufo.failExp)(_ufo.EXPERIENCE_NAME.LOAD_GRAMMAR, 'singleton', _objectSpread({
             isLocalLLM: isLocalLLM,
@@ -150,11 +215,11 @@ var loadGrammarDataAsync = exports.loadGrammarDataAsync = function loadGrammarDa
           // Allow a later call to retry the load rather than caching the failure.
           grammarDataPromise = undefined;
           throw _t;
-        case 5:
+        case 8:
         case "end":
           return _context.stop();
       }
-    }, _callee, null, [[1, 4]]);
+    }, _callee, null, [[1, 7]]);
   }))();
   return grammarDataPromise;
 };
@@ -253,7 +318,16 @@ function applyGrammarFilter(candidates, previousWord) {
   try {
     for (_iterator2.s(); !(_step2 = _iterator2.n()).done;) {
       var entry = _step2.value;
-      var candidateTags = posTags.get(entry.candidate.word.toLowerCase());
+      // The producer validates a multi-word term's *internal* grammar offline
+      // (boundary-clean, noun-headed, AUP-passed — spec §9), but nothing judges
+      // the transition into it. Exempting multi-word terms therefore let a noun
+      // this filter had just rejected re-enter wrapped in a bigram — "be" [AUX]
+      // dropped "quality" and then displayed "quality assurance". Only the term's
+      // first word takes part in the transition, so gate every term on that.
+      var surface = entry.candidate.word;
+      var firstSpace = surface.indexOf(' ');
+      var transitionWord = firstSpace === -1 ? surface : surface.slice(0, firstSpace);
+      var candidateTags = posTags.get(transitionWord.toLowerCase());
 
       // If candidate has no tags (unknown word), let it pass to be safe
       if (!candidateTags || candidateTags.length === 0) {
@@ -301,19 +375,34 @@ function getLmScore(word, lmLogits) {
 
 // ─── Public API ─────────────────────────────────────────────
 
-function rankCandidates(candidates, contextVector, getWordVector, lmLogits, maxTenantFreq, previousWord) {
+/**
+ * Apply Stage 1, grammar, and nullable context-local LM evidence to candidates.
+ */
+function rankCandidates(candidates, contextVector, getWordVector, lmLogits, maxFreqByType, previousWord, getCanonicalLmEvidence) {
   var _grammarMeta$dropped;
   // Stage 1
   var stage1Results = candidates.map(function (candidate) {
-    var _scoreStage = scoreStage1(candidate, contextVector, getWordVector, maxTenantFreq),
+    var _candidate$termType;
+    var termType = (_candidate$termType = candidate.termType) !== null && _candidate$termType !== void 0 ? _candidate$termType : 'word';
+    // Normalize freq within the candidate's own term-type population so a
+    // phrase (whose absolute counts are far below top unigrams) isn't crushed
+    // by the unigram max.
+    var maxFreq = maxFreqByType[termType] || 1;
+    var _scoreStage = scoreStage1(candidate, contextVector, getWordVector, maxFreq),
       semanticScore = _scoreStage.semanticScore,
       freqScore = _scoreStage.freqScore,
       stage1Score = _scoreStage.stage1Score;
+    // Multi-word units get an NPMI association boost (Stage-1 only).
+    var adjustedStage1 = stage1Score;
+    if (termType !== 'word' && typeof candidate.association === 'number') {
+      var assoc = Math.max(0, Math.min(1, candidate.association));
+      adjustedStage1 = stage1Score * (1 + ASSOCIATION_BOOST * assoc);
+    }
     return {
       candidate: candidate,
       semanticScore: semanticScore,
       freqScore: freqScore,
-      stage1Score: stage1Score
+      stage1Score: adjustedStage1
     };
   });
   var stage1Survivors = [];
@@ -346,26 +435,82 @@ function rankCandidates(candidates, contextVector, getWordVector, lmLogits, maxT
     var values = Object.values(lmLogits);
     if (values.length > 0) lmMax = Math.max.apply(Math, values);
   }
+  var canonicalEvidenceBySurface = new Map();
+  var maxLmRankingValueByTierAndContext = new Map();
+  var _iterator4 = _createForOfIteratorHelper(filtered),
+    _step4;
+  try {
+    for (_iterator4.s(); !(_step4 = _iterator4.n()).done;) {
+      var _getCanonicalLmEviden;
+      var _entry = _step4.value;
+      var evidence = (_getCanonicalLmEviden = getCanonicalLmEvidence === null || getCanonicalLmEvidence === void 0 ? void 0 : getCanonicalLmEvidence(_entry.candidate.word)) !== null && _getCanonicalLmEviden !== void 0 ? _getCanonicalLmEviden : null;
+      if (!evidence) {
+        continue;
+      }
+      canonicalEvidenceBySurface.set(_entry.candidate.word, evidence);
+      var rankingValue = evidence.source === 'canonical-full-surface' ? evidence.meanTokenLogProbability : evidence.rawLogit;
+      if (rankingValue !== undefined && Number.isFinite(rankingValue)) {
+        var _maxLmRankingValueByT;
+        var key = "".concat(evidence.source, "\0").concat(evidence.contextKey);
+        var current = (_maxLmRankingValueByT = maxLmRankingValueByTierAndContext.get(key)) !== null && _maxLmRankingValueByT !== void 0 ? _maxLmRankingValueByT : -Infinity;
+        if (rankingValue > current) {
+          maxLmRankingValueByTierAndContext.set(key, rankingValue);
+        }
+      }
+    }
+  } catch (err) {
+    _iterator4.e(err);
+  } finally {
+    _iterator4.f();
+  }
   var scored = filtered.map(function (entry) {
+    var _entry$candidate$term, _canonicalEvidenceByS;
+    var termType = (_entry$candidate$term = entry.candidate.termType) !== null && _entry$candidate$term !== void 0 ? _entry$candidate$term : 'word';
+    var isMultiWord = termType !== 'word';
     var lmScore = 0;
     var finalScore = entry.stage1Score;
-    if (lmLogits && lmMax >= LM_GATE_THRESHOLD) {
+    var lmEvidence = (_canonicalEvidenceByS = canonicalEvidenceBySurface.get(entry.candidate.word)) !== null && _canonicalEvidenceByS !== void 0 ? _canonicalEvidenceByS : null;
+    if (lmEvidence) {
+      var rankingValue = lmEvidence.source === 'canonical-full-surface' ? lmEvidence.meanTokenLogProbability : lmEvidence.rawLogit;
+      var maxRankingValue = maxLmRankingValueByTierAndContext.get("".concat(lmEvidence.source, "\0").concat(lmEvidence.contextKey));
+      if (rankingValue !== undefined && maxRankingValue !== undefined) {
+        lmEvidence = _objectSpread(_objectSpread({}, lmEvidence), {}, {
+          score: Math.exp(rankingValue - maxRankingValue)
+        });
+      }
+    }
+    if (lmEvidence) {
+      lmScore = lmEvidence.score;
+      finalScore = STAGE1_WEIGHT * entry.stage1Score + STAGE2_WEIGHT * lmScore;
+    } else if (!isMultiWord && lmLogits && lmMax >= LM_GATE_THRESHOLD) {
+      // Backward-compatible network slow-lane path.
       var rawLm = getLmScore(entry.candidate.word, lmLogits);
       if (rawLm !== 0) {
         var logitDiff = Math.log(rawLm) - Math.log(lmMax);
         lmScore = Math.exp(logitDiff);
+        lmEvidence = {
+          contextKey: 'network',
+          score: lmScore,
+          separatorKind: 'whitespace',
+          source: 'network-logit',
+          totalSurfaceCharCount: entry.candidate.word.length,
+          // The legacy BE word map does not report canonical token depth.
+          totalTokenCount: 0,
+          verifiedCharCount: 0,
+          verifiedTokenCount: 0
+        };
+        finalScore = STAGE1_WEIGHT * entry.stage1Score + STAGE2_WEIGHT * lmScore;
       }
-      // Words absent from the prefix-filtered payload get lmScore = 0,
-      // not 0.05, so they don't outrank genuine LM predictions.
-
-      finalScore = STAGE1_WEIGHT * entry.stage1Score + STAGE2_WEIGHT * lmScore;
     }
     return {
       word: entry.candidate.word,
       freqScore: entry.freqScore,
       semanticScore: entry.semanticScore,
       lmScore: lmScore,
-      finalScore: finalScore
+      lmEvidence: lmEvidence,
+      finalScore: finalScore,
+      stage1Score: entry.stage1Score,
+      termType: termType
     };
   });
   scored.sort(function (a, b) {

package/dist/cjs/pm-plugins/slow-lane-client.js

@@ -4,12 +4,13 @@ var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefau
 Object.defineProperty(exports, "__esModule", {
   value: true
 });
-exports.setDefaultSlowLaneClient = exports.isWordBoundary = exports.getStoredLmLogits = exports.getStoredContextVector = exports.createSlowLaneClient = void 0;
+exports.setDefaultSlowLaneClient = exports.requestProgressiveSurfaceScores = exports.primeBoundaryLm = exports.isWordBoundary = exports.isCanonicalSurfaceScoringSupported = exports.getSurfaceScore = exports.getStoredLmLogits = exports.getStoredContextVector = exports.getStoredContextInput = exports.getProgressiveSurfaceEvidence = exports.getDefaultSlowLaneClientStatus = exports.getCanonicalSurfaceTokenIds = exports.getCanonicalSurfaceCount = exports.getBoundaryLmState = exports.createSlowLaneClient = exports.clearDefaultSlowLaneClient = void 0;
 var _regenerator = _interopRequireDefault(require("@babel/runtime/regenerator"));
 var _typeof2 = _interopRequireDefault(require("@babel/runtime/helpers/typeof"));
 var _defineProperty2 = _interopRequireDefault(require("@babel/runtime/helpers/defineProperty"));
 var _asyncToGenerator2 = _interopRequireDefault(require("@babel/runtime/helpers/asyncToGenerator"));
 var _ufo = require("../analytics/ufo");
+var _artifactLoader = require("./artifact-loader");
 var _debugMode = require("./debug-mode");
 function ownKeys(e, r) { var t = Object.keys(e); if (Object.getOwnPropertySymbols) { var o = Object.getOwnPropertySymbols(e); r && (o = o.filter(function (r) { return Object.getOwnPropertyDescriptor(e, r).enumerable; })), t.push.apply(t, o); } return t; }
 function _objectSpread(e) { for (var r = 1; r < arguments.length; r++) { var t = null != arguments[r] ? arguments[r] : {}; r % 2 ? ownKeys(Object(t), !0).forEach(function (r) { (0, _defineProperty2.default)(e, r, t[r]); }) : Object.getOwnPropertyDescriptors ? Object.defineProperties(e, Object.getOwnPropertyDescriptors(t)) : ownKeys(Object(t)).forEach(function (r) { Object.defineProperty(e, r, Object.getOwnPropertyDescriptor(t, r)); }); } return e; } /**
@@ -30,8 +31,6 @@ function _objectSpread(e) { for (var r = 1; r < arguments.length; r++) { var t =
 
 // eslint-disable-next-line require-unicode-regexp
 var WORD_BOUNDARY_CHARS = /[\s.,;:!?]/;
-// eslint-disable-next-line require-unicode-regexp
-var TRAILING_SLASH_REGEX = /\/$/;
 var DEFAULT_DEBOUNCE_MS = 300;
 
 /**
@@ -68,6 +67,7 @@ var createSlowLaneClient = exports.createSlowLaneClient = function createSlowLan
   };
   var debounceTimer = null;
   var lastRequestedText = '';
+  var storedContextInput = null;
   var storedContextVector = null;
   var storedLmLogits = null;
   var requestSeq = 0;
@@ -84,7 +84,7 @@ var createSlowLaneClient = exports.createSlowLaneClient = function createSlowLan
             }
             return _context.abrupt("return");
           case 1:
-            url = "".concat(baseUrl.replace(TRAILING_SLASH_REGEX, '')).concat(endpoint);
+            url = (0, _artifactLoader.buildAutocompleteGatewayUrl)(endpoint, baseUrl);
             payload = {
               text: text,
               session_id: sessionId
@@ -95,16 +95,7 @@ var createSlowLaneClient = exports.createSlowLaneClient = function createSlowLan
             }, surface ? {
               surface: surface
             } : {}));
-            if ((0, _debugMode.isAutocompleteDebugEnabled)()) {
-              // eslint-disable-next-line no-console
-              console.groupCollapsed("%c[SlowLane] %c\uD83D\uDCE4 Sending context | ".concat(text.length, " chars"), 'color: #9c27b0; font-weight: bold;', 'color: inherit;');
-              text.split('\n').forEach(function (line, i, arr) {
-                // eslint-disable-next-line no-console
-                console.log("  ".concat(i === arr.length - 1 ? '▶' : ' ', " ").concat(line));
-              });
-              // eslint-disable-next-line no-console
-              console.groupEnd();
-            }
+            (0, _debugMode.ctcTag)('signal', "\uD83D\uDCE4 network context sent \xB7 ".concat(text.length, " chars"), _debugMode.CTC_STYLES.section);
             _context.prev = 2;
             _context.next = 3;
             return fetchFn(url, {
@@ -118,6 +109,7 @@ var createSlowLaneClient = exports.createSlowLaneClient = function createSlowLan
               _context.next = 4;
               break;
             }
+            storedContextInput = null;
             storedContextVector = null;
             storedLmLogits = null;
             (0, _ufo.failExp)(_ufo.EXPERIENCE_NAME.SLOW_LANE_FETCH, requestId, _objectSpread({
@@ -127,10 +119,7 @@ var createSlowLaneClient = exports.createSlowLaneClient = function createSlowLan
             }, surface ? {
               surface: surface
             } : {}));
-            if ((0, _debugMode.isAutocompleteDebugEnabled)()) {
-              // eslint-disable-next-line no-console
-              console.log("%c[SlowLane] %c\u274C Request failed (".concat(res.status, ")"), 'color: #9c27b0; font-weight: bold;', 'color: #f44336;');
-            }
+            (0, _debugMode.ctcTag)('signal', "\u274C network request failed (".concat(res.status, ")"), _debugMode.CTC_STYLES.bad);
             return _context.abrupt("return");
           case 4:
             _context.next = 5;
@@ -139,24 +128,17 @@ var createSlowLaneClient = exports.createSlowLaneClient = function createSlowLan
             data = _context.sent;
             if (data.semantic_vector && Array.isArray(data.semantic_vector)) {
               storedContextVector = new Float32Array(data.semantic_vector);
+              storedContextInput = text;
             } else {
               storedContextVector = null;
+              storedContextInput = null;
             }
             if (data.lm_logits && (0, _typeof2.default)(data.lm_logits) === 'object') {
               storedLmLogits = data.lm_logits;
             } else {
               storedLmLogits = null;
             }
-            if ((0, _debugMode.isAutocompleteDebugEnabled)()) {
-              // eslint-disable-next-line no-console
-              console.groupCollapsed("%c[SlowLane] %c\uD83D\uDCE5 Response received", 'color: #9c27b0; font-weight: bold;', 'color: inherit;');
-              // eslint-disable-next-line no-console
-              console.log(storedContextVector ? "\u2705 semantic_vector: ".concat(storedContextVector.length, " dims") : '❌ No semantic_vector');
-              // eslint-disable-next-line no-console
-              console.log(storedLmLogits ? "\u2705 lm_logits: ".concat(Object.keys(storedLmLogits).length, " tokens") : '❌ No lm_logits');
-              // eslint-disable-next-line no-console
-              console.groupEnd();
-            }
+            (0, _debugMode.ctcTag)('signal', "\uD83D\uDCE5 network response \xB7 ".concat(storedContextVector ? "semantic \u2705 ".concat(storedContextVector.length, "d") : 'semantic ❌', " \xB7 ").concat(storedLmLogits ? "logits \u2705 ".concat(Object.keys(storedLmLogits).length, "tok") : 'logits ❌'), _debugMode.CTC_STYLES.section);
             (0, _ufo.succeedExp)(_ufo.EXPERIENCE_NAME.SLOW_LANE_FETCH, requestId, _objectSpread({
               textLength: text.length,
               hasVector: storedContextVector !== null,
@@ -176,6 +158,7 @@ var createSlowLaneClient = exports.createSlowLaneClient = function createSlowLan
           case 6:
             _context.prev = 6;
             _t = _context["catch"](2);
+            storedContextInput = null;
             storedContextVector = null;
             storedLmLogits = null;
             (0, _ufo.failExp)(_ufo.EXPERIENCE_NAME.SLOW_LANE_FETCH, requestId, _objectSpread({
@@ -184,10 +167,7 @@ var createSlowLaneClient = exports.createSlowLaneClient = function createSlowLan
             }, surface ? {
               surface: surface
             } : {}));
-            if ((0, _debugMode.isAutocompleteDebugEnabled)()) {
-              // eslint-disable-next-line no-console
-              console.log('%c[SlowLane] %c❌ Network error — context cleared', 'color: #9c27b0; font-weight: bold;', 'color: #f44336;');
-            }
+            (0, _debugMode.ctcTag)('signal', '❌ network error — context cleared', _debugMode.CTC_STYLES.bad);
           case 7:
             _context.prev = 7;
             if (inflightRequestId === requestId) {
@@ -225,6 +205,9 @@ var createSlowLaneClient = exports.createSlowLaneClient = function createSlowLan
   };
   return {
     updateContext: updateContextDebounced,
+    getContextInput: function getContextInput() {
+      return storedContextInput;
+    },
     getContextVector: function getContextVector() {
       return storedContextVector;
     },
@@ -232,6 +215,7 @@ var createSlowLaneClient = exports.createSlowLaneClient = function createSlowLan
       return storedLmLogits;
     },
     setContextVector: function setContextVector(vector) {
+      storedContextInput = null;
       storedContextVector = vector;
     },
     setLmLogits: function setLmLogits(logits) {
@@ -241,14 +225,93 @@ var createSlowLaneClient = exports.createSlowLaneClient = function createSlowLan
   };
 };
 var defaultSlowLaneClient = null;
-var setDefaultSlowLaneClient = exports.setDefaultSlowLaneClient = function setDefaultSlowLaneClient(client) {
+var defaultSlowLaneClientRegistration = null;
+var supportsCanonicalSurfaceScoring = function supportsCanonicalSurfaceScoring(client) {
+  return typeof (client === null || client === void 0 ? void 0 : client.primeBoundaryLm) === 'function' && typeof (client === null || client === void 0 ? void 0 : client.requestProgressiveSurfaceScores) === 'function';
+};
+var setDefaultSlowLaneClient = exports.setDefaultSlowLaneClient = function setDefaultSlowLaneClient(client, registration) {
   defaultSlowLaneClient = client;
+  defaultSlowLaneClientRegistration = client ? registration !== null && registration !== void 0 ? registration : {
+    clientId: 'untracked',
+    kind: supportsCanonicalSurfaceScoring(client) ? 'localLlm' : 'server'
+  } : null;
+};
+
+/**
+ * Clear the shared client only when the caller still owns the registration.
+ *
+ * Multiple editor instances can briefly overlap during a React/editor remount.
+ * An older instance must not disconnect the newer instance when its delayed
+ * teardown runs after the newer client has registered.
+ */
+var clearDefaultSlowLaneClient = exports.clearDefaultSlowLaneClient = function clearDefaultSlowLaneClient(client) {
+  if (defaultSlowLaneClient !== client) {
+    return false;
+  }
+  defaultSlowLaneClient = null;
+  defaultSlowLaneClientRegistration = null;
+  return true;
+};
+var getDefaultSlowLaneClientStatus = exports.getDefaultSlowLaneClientStatus = function getDefaultSlowLaneClientStatus() {
+  var _defaultSlowLaneClien, _defaultSlowLaneClien2, _defaultSlowLaneClien3, _defaultSlowLaneClien4, _defaultSlowLaneClien5, _defaultSlowLaneClien6, _defaultSlowLaneClien7;
+  var canonicalScoringSupported = supportsCanonicalSurfaceScoring(defaultSlowLaneClient);
+  var isLocalClient = ((_defaultSlowLaneClien = defaultSlowLaneClientRegistration) === null || _defaultSlowLaneClien === void 0 ? void 0 : _defaultSlowLaneClien.kind) === 'localLlm';
+  return {
+    registered: defaultSlowLaneClient !== null,
+    clientId: (_defaultSlowLaneClien2 = (_defaultSlowLaneClien3 = defaultSlowLaneClientRegistration) === null || _defaultSlowLaneClien3 === void 0 ? void 0 : _defaultSlowLaneClien3.clientId) !== null && _defaultSlowLaneClien2 !== void 0 ? _defaultSlowLaneClien2 : null,
+    clientKind: (_defaultSlowLaneClien4 = (_defaultSlowLaneClien5 = defaultSlowLaneClientRegistration) === null || _defaultSlowLaneClien5 === void 0 ? void 0 : _defaultSlowLaneClien5.kind) !== null && _defaultSlowLaneClien4 !== void 0 ? _defaultSlowLaneClien4 : null,
+    canonicalScoringSupported: canonicalScoringSupported,
+    localModelReady: isLocalClient && typeof ((_defaultSlowLaneClien6 = defaultSlowLaneClient) === null || _defaultSlowLaneClien6 === void 0 ? void 0 : _defaultSlowLaneClien6.isReady) === 'function' ? defaultSlowLaneClient.isReady() : null,
+    canonicalSurfaceCount: canonicalScoringSupported && typeof ((_defaultSlowLaneClien7 = defaultSlowLaneClient) === null || _defaultSlowLaneClien7 === void 0 ? void 0 : _defaultSlowLaneClien7.getCanonicalSurfaceCount) === 'function' ? defaultSlowLaneClient.getCanonicalSurfaceCount() : null
+  };
 };
 var getStoredContextVector = exports.getStoredContextVector = function getStoredContextVector() {
-  var _defaultSlowLaneClien, _defaultSlowLaneClien2;
-  return (_defaultSlowLaneClien = (_defaultSlowLaneClien2 = defaultSlowLaneClient) === null || _defaultSlowLaneClien2 === void 0 ? void 0 : _defaultSlowLaneClien2.getContextVector()) !== null && _defaultSlowLaneClien !== void 0 ? _defaultSlowLaneClien : null;
+  var _defaultSlowLaneClien8, _defaultSlowLaneClien9;
+  return (_defaultSlowLaneClien8 = (_defaultSlowLaneClien9 = defaultSlowLaneClient) === null || _defaultSlowLaneClien9 === void 0 ? void 0 : _defaultSlowLaneClien9.getContextVector()) !== null && _defaultSlowLaneClien8 !== void 0 ? _defaultSlowLaneClien8 : null;
+};
+var getStoredContextInput = exports.getStoredContextInput = function getStoredContextInput() {
+  var _defaultSlowLaneClien0, _defaultSlowLaneClien1, _defaultSlowLaneClien10;
+  return (_defaultSlowLaneClien0 = (_defaultSlowLaneClien1 = defaultSlowLaneClient) === null || _defaultSlowLaneClien1 === void 0 || (_defaultSlowLaneClien10 = _defaultSlowLaneClien1.getContextInput) === null || _defaultSlowLaneClien10 === void 0 ? void 0 : _defaultSlowLaneClien10.call(_defaultSlowLaneClien1)) !== null && _defaultSlowLaneClien0 !== void 0 ? _defaultSlowLaneClien0 : null;
 };
 var getStoredLmLogits = exports.getStoredLmLogits = function getStoredLmLogits() {
-  var _defaultSlowLaneClien3, _defaultSlowLaneClien4;
-  return (_defaultSlowLaneClien3 = (_defaultSlowLaneClien4 = defaultSlowLaneClient) === null || _defaultSlowLaneClien4 === void 0 ? void 0 : _defaultSlowLaneClien4.getLmLogits()) !== null && _defaultSlowLaneClien3 !== void 0 ? _defaultSlowLaneClien3 : null;
+  var _defaultSlowLaneClien11, _defaultSlowLaneClien12;
+  return (_defaultSlowLaneClien11 = (_defaultSlowLaneClien12 = defaultSlowLaneClient) === null || _defaultSlowLaneClien12 === void 0 ? void 0 : _defaultSlowLaneClien12.getLmLogits()) !== null && _defaultSlowLaneClien11 !== void 0 ? _defaultSlowLaneClien11 : null;
+};
+var getBoundaryLmState = exports.getBoundaryLmState = function getBoundaryLmState(contextKey) {
+  var _defaultSlowLaneClien13, _defaultSlowLaneClien14, _defaultSlowLaneClien15;
+  return (_defaultSlowLaneClien13 = (_defaultSlowLaneClien14 = defaultSlowLaneClient) === null || _defaultSlowLaneClien14 === void 0 || (_defaultSlowLaneClien15 = _defaultSlowLaneClien14.getBoundaryLmState) === null || _defaultSlowLaneClien15 === void 0 ? void 0 : _defaultSlowLaneClien15.call(_defaultSlowLaneClien14, contextKey)) !== null && _defaultSlowLaneClien13 !== void 0 ? _defaultSlowLaneClien13 : null;
+};
+var getCanonicalSurfaceTokenIds = exports.getCanonicalSurfaceTokenIds = function getCanonicalSurfaceTokenIds(surface) {
+  var _defaultSlowLaneClien16, _defaultSlowLaneClien17, _defaultSlowLaneClien18;
+  return (_defaultSlowLaneClien16 = (_defaultSlowLaneClien17 = defaultSlowLaneClient) === null || _defaultSlowLaneClien17 === void 0 || (_defaultSlowLaneClien18 = _defaultSlowLaneClien17.getCanonicalSurfaceTokenIds) === null || _defaultSlowLaneClien18 === void 0 ? void 0 : _defaultSlowLaneClien18.call(_defaultSlowLaneClien17, surface)) !== null && _defaultSlowLaneClien16 !== void 0 ? _defaultSlowLaneClien16 : null;
+};
+var getSurfaceScore = exports.getSurfaceScore = function getSurfaceScore(contextKey, surface) {
+  var _defaultSlowLaneClien19, _defaultSlowLaneClien20, _defaultSlowLaneClien21;
+  return (_defaultSlowLaneClien19 = (_defaultSlowLaneClien20 = defaultSlowLaneClient) === null || _defaultSlowLaneClien20 === void 0 || (_defaultSlowLaneClien21 = _defaultSlowLaneClien20.getSurfaceScore) === null || _defaultSlowLaneClien21 === void 0 ? void 0 : _defaultSlowLaneClien21.call(_defaultSlowLaneClien20, contextKey, surface)) !== null && _defaultSlowLaneClien19 !== void 0 ? _defaultSlowLaneClien19 : null;
+};
+var getProgressiveSurfaceEvidence = exports.getProgressiveSurfaceEvidence = function getProgressiveSurfaceEvidence(contextKey, surface) {
+  var _defaultSlowLaneClien22, _defaultSlowLaneClien23, _defaultSlowLaneClien24;
+  return (_defaultSlowLaneClien22 = (_defaultSlowLaneClien23 = defaultSlowLaneClient) === null || _defaultSlowLaneClien23 === void 0 || (_defaultSlowLaneClien24 = _defaultSlowLaneClien23.getProgressiveSurfaceEvidence) === null || _defaultSlowLaneClien24 === void 0 ? void 0 : _defaultSlowLaneClien24.call(_defaultSlowLaneClien23, contextKey, surface)) !== null && _defaultSlowLaneClien22 !== void 0 ? _defaultSlowLaneClien22 : null;
+};
+var primeBoundaryLm = exports.primeBoundaryLm = function primeBoundaryLm(input) {
+  var _defaultSlowLaneClien25, _defaultSlowLaneClien26;
+  (_defaultSlowLaneClien25 = defaultSlowLaneClient) === null || _defaultSlowLaneClien25 === void 0 || (_defaultSlowLaneClien26 = _defaultSlowLaneClien25.primeBoundaryLm) === null || _defaultSlowLaneClien26 === void 0 || _defaultSlowLaneClien26.call(_defaultSlowLaneClien25, input);
+};
+var requestProgressiveSurfaceScores = exports.requestProgressiveSurfaceScores = function requestProgressiveSurfaceScores(input) {
+  var _defaultSlowLaneClien27, _defaultSlowLaneClien28;
+  (_defaultSlowLaneClien27 = defaultSlowLaneClient) === null || _defaultSlowLaneClien27 === void 0 || (_defaultSlowLaneClien28 = _defaultSlowLaneClien27.requestProgressiveSurfaceScores) === null || _defaultSlowLaneClien28 === void 0 || _defaultSlowLaneClien28.call(_defaultSlowLaneClien27, input);
+};
+var isCanonicalSurfaceScoringSupported = exports.isCanonicalSurfaceScoringSupported = function isCanonicalSurfaceScoringSupported() {
+  return supportsCanonicalSurfaceScoring(defaultSlowLaneClient);
+};
+
+/**
+ * How many surfaces the canonical token map currently holds. Goes from zero to
+ * its final size once the artifact lands, which is the only point at which
+ * `getCanonicalSurfaceTokenIds` starts answering — so callers memoising over
+ * that lookup can use this to detect the transition.
+ */
+var getCanonicalSurfaceCount = exports.getCanonicalSurfaceCount = function getCanonicalSurfaceCount() {
+  var _defaultSlowLaneClien29, _defaultSlowLaneClien30, _defaultSlowLaneClien31;
+  return (_defaultSlowLaneClien29 = (_defaultSlowLaneClien30 = defaultSlowLaneClient) === null || _defaultSlowLaneClien30 === void 0 || (_defaultSlowLaneClien31 = _defaultSlowLaneClien30.getCanonicalSurfaceCount) === null || _defaultSlowLaneClien31 === void 0 ? void 0 : _defaultSlowLaneClien31.call(_defaultSlowLaneClien30)) !== null && _defaultSlowLaneClien29 !== void 0 ? _defaultSlowLaneClien29 : 0;
 };