@atlaskit/editor-plugin-autocomplete 8.0.0 → 9.1.0

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

@@ -1,6 +1,7 @@
 import _defineProperty from "@babel/runtime/helpers/defineProperty";
 import _asyncToGenerator from "@babel/runtime/helpers/asyncToGenerator";
 import _slicedToArray from "@babel/runtime/helpers/slicedToArray";
+import _typeof from "@babel/runtime/helpers/typeof";
 function _createForOfIteratorHelper(r, e) { var t = "undefined" != typeof Symbol && r[Symbol.iterator] || r["@@iterator"]; if (!t) { if (Array.isArray(r) || (t = _unsupportedIterableToArray(r)) || e && r && "number" == typeof r.length) { t && (r = t); var _n = 0, F = function F() {}; return { s: F, n: function n() { return _n >= r.length ? { done: !0 } : { done: !1, value: r[_n++] }; }, e: function e(r) { throw r; }, f: F }; } throw new TypeError("Invalid attempt to iterate non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); } var o, a = !0, u = !1; return { s: function s() { t = t.call(r); }, n: function n() { var r = t.next(); return a = r.done, r; }, e: function e(r) { u = !0, o = r; }, f: function f() { try { a || null == t.return || t.return(); } finally { if (u) throw o; } } }; }
 function _unsupportedIterableToArray(r, a) { if (r) { if ("string" == typeof r) return _arrayLikeToArray(r, a); var t = {}.toString.call(r).slice(8, -1); return "Object" === t && r.constructor && (t = r.constructor.name), "Map" === t || "Set" === t ? Array.from(r) : "Arguments" === t || /^(?:Ui|I)nt(?:8|16|32)(?:Clamped)?Array$/.test(t) ? _arrayLikeToArray(r, a) : void 0; } }
 function _arrayLikeToArray(r, a) { (null == a || a > r.length) && (a = r.length); for (var e = 0, n = Array(a); e < a; e++) n[e] = r[e]; return n; }
@@ -23,6 +24,11 @@ import { isAutocompleteDebugEnabled } from './debug-mode';
 
 // ─── 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 ──────────────────────────────────────
@@ -30,9 +36,25 @@ import { isAutocompleteDebugEnabled } from './debug-mode';
 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;
 export var STAGE1_WEIGHT = 0.35;
 export var STAGE2_WEIGHT = 0.65;
 export var 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.
+ */
+export var 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
@@ -51,13 +73,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 || _typeof(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 || _typeof(payload) !== 'object') {
+    return false;
+  }
+  var transitions = payload.transitions;
+  if (transitions == null || _typeof(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));
@@ -111,9 +158,27 @@ export var loadGrammarDataAsync = function loadGrammarDataAsync(options) {
           }
           throw new Error("[scoring-pipeline] ".concat(ARTIFACT_NAME.GRAMMAR_TRANSITIONS, " is missing its `transitions` wrapper"));
         case 3:
+          if (isGrammarTransitions(grammarTransitionsData)) {
+            _context.next = 4;
+            break;
+          }
+          throw new Error("[scoring-pipeline] ".concat(ARTIFACT_NAME.GRAMMAR_TRANSITIONS, " payload shape was invalid"));
+        case 4:
+          if (isPosTagsRecord(posTagsData)) {
+            _context.next = 5;
+            break;
+          }
+          throw new Error("[scoring-pipeline] ".concat(ARTIFACT_NAME.POS_TAGS, " payload shape was invalid"));
+        case 5:
+          if (isPosTagsRecord(ghostPosTagsData)) {
+            _context.next = 6;
+            break;
+          }
+          throw new Error("[scoring-pipeline] ".concat(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
           });
           succeedExp(EXPERIENCE_NAME.LOAD_GRAMMAR, 'singleton', _objectSpread({
@@ -131,10 +196,10 @@ export var loadGrammarDataAsync = function loadGrammarDataAsync(options) {
               transitionCount: precomputedAllowedByPos.size
             });
           }
-          _context.next = 5;
+          _context.next = 8;
           break;
-        case 4:
-          _context.prev = 4;
+        case 7:
+          _context.prev = 7;
           _t = _context["catch"](1);
           failExp(EXPERIENCE_NAME.LOAD_GRAMMAR, 'singleton', _objectSpread({
             isLocalLLM: isLocalLLM,
@@ -145,11 +210,11 @@ export var loadGrammarDataAsync = function loadGrammarDataAsync(options) {
           // 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;
 };
@@ -248,7 +313,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) {
@@ -296,19 +370,34 @@ function getLmScore(word, lmLogits) {
 
 // ─── Public API ─────────────────────────────────────────────
 
-export function rankCandidates(candidates, contextVector, getWordVector, lmLogits, maxTenantFreq, previousWord) {
+/**
+ * Apply Stage 1, grammar, and nullable context-local LM evidence to candidates.
+ */
+export 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 = [];
@@ -341,26 +430,82 @@ export function rankCandidates(candidates, contextVector, getWordVector, lmLogit
     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/esm/pm-plugins/slow-lane-client.js

@@ -14,7 +14,8 @@ function _objectSpread(e) { for (var r = 1; r < arguments.length; r++) { var t =
  */
 
 import { abortExp, EXPERIENCE_NAME, failExp, startExp, succeedExp } from '../analytics/ufo';
-import { isAutocompleteDebugEnabled } from './debug-mode';
+import { buildAutocompleteGatewayUrl } from './artifact-loader';
+import { CTC_STYLES, ctcTag } from './debug-mode';
 
 // ─── Types ───────────────────────────────────────────────────────────────────
 
@@ -26,8 +27,6 @@ import { isAutocompleteDebugEnabled } from './debug-mode';
 
 // 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;
 
 /**
@@ -64,6 +63,7 @@ export var createSlowLaneClient = function createSlowLaneClient(config) {
   };
   var debounceTimer = null;
   var lastRequestedText = '';
+  var storedContextInput = null;
   var storedContextVector = null;
   var storedLmLogits = null;
   var requestSeq = 0;
@@ -80,7 +80,7 @@ export var createSlowLaneClient = function createSlowLaneClient(config) {
             }
             return _context.abrupt("return");
           case 1:
-            url = "".concat(baseUrl.replace(TRAILING_SLASH_REGEX, '')).concat(endpoint);
+            url = buildAutocompleteGatewayUrl(endpoint, baseUrl);
             payload = {
               text: text,
               session_id: sessionId
@@ -91,16 +91,7 @@ export var createSlowLaneClient = function createSlowLaneClient(config) {
             }, surface ? {
               surface: surface
             } : {}));
-            if (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();
-            }
+            ctcTag('signal', "\uD83D\uDCE4 network context sent \xB7 ".concat(text.length, " chars"), CTC_STYLES.section);
             _context.prev = 2;
             _context.next = 3;
             return fetchFn(url, {
@@ -114,6 +105,7 @@ export var createSlowLaneClient = function createSlowLaneClient(config) {
               _context.next = 4;
               break;
             }
+            storedContextInput = null;
             storedContextVector = null;
             storedLmLogits = null;
             failExp(EXPERIENCE_NAME.SLOW_LANE_FETCH, requestId, _objectSpread({
@@ -123,10 +115,7 @@ export var createSlowLaneClient = function createSlowLaneClient(config) {
             }, surface ? {
               surface: surface
             } : {}));
-            if (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;');
-            }
+            ctcTag('signal', "\u274C network request failed (".concat(res.status, ")"), CTC_STYLES.bad);
             return _context.abrupt("return");
           case 4:
             _context.next = 5;
@@ -135,24 +124,17 @@ export var createSlowLaneClient = function createSlowLaneClient(config) {
             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 && _typeof(data.lm_logits) === 'object') {
               storedLmLogits = data.lm_logits;
             } else {
               storedLmLogits = null;
             }
-            if (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();
-            }
+            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 ❌'), CTC_STYLES.section);
             succeedExp(EXPERIENCE_NAME.SLOW_LANE_FETCH, requestId, _objectSpread({
               textLength: text.length,
               hasVector: storedContextVector !== null,
@@ -172,6 +154,7 @@ export var createSlowLaneClient = function createSlowLaneClient(config) {
           case 6:
             _context.prev = 6;
             _t = _context["catch"](2);
+            storedContextInput = null;
             storedContextVector = null;
             storedLmLogits = null;
             failExp(EXPERIENCE_NAME.SLOW_LANE_FETCH, requestId, _objectSpread({
@@ -180,10 +163,7 @@ export var createSlowLaneClient = function createSlowLaneClient(config) {
             }, surface ? {
               surface: surface
             } : {}));
-            if (isAutocompleteDebugEnabled()) {
-              // eslint-disable-next-line no-console
-              console.log('%c[SlowLane] %c❌ Network error — context cleared', 'color: #9c27b0; font-weight: bold;', 'color: #f44336;');
-            }
+            ctcTag('signal', '❌ network error — context cleared', CTC_STYLES.bad);
           case 7:
             _context.prev = 7;
             if (inflightRequestId === requestId) {
@@ -221,6 +201,9 @@ export var createSlowLaneClient = function createSlowLaneClient(config) {
   };
   return {
     updateContext: updateContextDebounced,
+    getContextInput: function getContextInput() {
+      return storedContextInput;
+    },
     getContextVector: function getContextVector() {
       return storedContextVector;
     },
@@ -228,6 +211,7 @@ export var createSlowLaneClient = function createSlowLaneClient(config) {
       return storedLmLogits;
     },
     setContextVector: function setContextVector(vector) {
+      storedContextInput = null;
       storedContextVector = vector;
     },
     setLmLogits: function setLmLogits(logits) {
@@ -237,14 +221,93 @@ export var createSlowLaneClient = function createSlowLaneClient(config) {
   };
 };
 var defaultSlowLaneClient = null;
-export var 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';
+};
+export var 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.
+ */
+export var clearDefaultSlowLaneClient = function clearDefaultSlowLaneClient(client) {
+  if (defaultSlowLaneClient !== client) {
+    return false;
+  }
+  defaultSlowLaneClient = null;
+  defaultSlowLaneClientRegistration = null;
+  return true;
+};
+export var 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
+  };
 };
 export var 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;
+};
+export var 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;
 };
 export var 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;
+};
+export var 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;
+};
+export var 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;
+};
+export var 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;
+};
+export var 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;
+};
+export var primeBoundaryLm = function primeBoundaryLm(input) {
+  var _defaultSlowLaneClien25, _defaultSlowLaneClien26;
+  (_defaultSlowLaneClien25 = defaultSlowLaneClient) === null || _defaultSlowLaneClien25 === void 0 || (_defaultSlowLaneClien26 = _defaultSlowLaneClien25.primeBoundaryLm) === null || _defaultSlowLaneClien26 === void 0 || _defaultSlowLaneClien26.call(_defaultSlowLaneClien25, input);
+};
+export var requestProgressiveSurfaceScores = function requestProgressiveSurfaceScores(input) {
+  var _defaultSlowLaneClien27, _defaultSlowLaneClien28;
+  (_defaultSlowLaneClien27 = defaultSlowLaneClient) === null || _defaultSlowLaneClien27 === void 0 || (_defaultSlowLaneClien28 = _defaultSlowLaneClien27.requestProgressiveSurfaceScores) === null || _defaultSlowLaneClien28 === void 0 || _defaultSlowLaneClien28.call(_defaultSlowLaneClien27, input);
+};
+export var 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.
+ */
+export var 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;
 };