@atlaskit/editor-plugin-autocomplete 9.0.0 → 9.1.0

package/CHANGELOG.md

@@ -1,5 +1,33 @@
 # @atlaskit/editor-plugin-autocomplete
 
+## 9.1.0
+
+### Minor Changes
+
+- [`a164c6d8e7de6`](https://bitbucket.org/atlassian/atlassian-frontend-monorepo/commits/a164c6d8e7de6) -
+  Offer the Confluence page the chat is open on to autocomplete as context, so words already written
+  on it carry an L1 session boost in the chat input. This reads the editor context the page editor
+  already publishes, so no host wiring is needed, and covers a page being edited; reaching a page
+  that is being viewed needs a host bridge that comes separately. The page text is held together
+  with the page it was read from, so navigating invalidates it in the same render that changes the
+  page rather than just after — the context store keeps the last published document until another
+  publisher replaces it, and that document does not say which page it came from.
+
+  Both the page and the chat transcript sit behind
+  `platform_editor_ai_autocomplete_rovo_chat_editor`, and the editor context subscription that reads
+  the page lives in a component the chat input only mounts once enrolled. An unenrolled session
+  therefore neither parses a page document for context it cannot use nor re-renders the input on a
+  publish, which the editor does on blur and on selection change.
+
+  Context text now waits for the vocabulary load to settle before it is applied. Session boosts were
+  previously spent against an empty trie whenever the page resolved first, which silently dropped
+  every word in it. What the session holds is readable from the console with
+  `__atlCtcDebug__.session()`.
+
+### Patch Changes
+
+- Updated dependencies
+
 ## 9.0.0
 
 ### Major Changes

package/dist/cjs/pm-plugins/autocomplete-plugin.js

@@ -516,7 +516,72 @@ var createAutocompletePlugin = exports.createAutocompletePlugin = function creat
   // Set when the plugin is torn down so in-flight getContext() resolutions don't
   // mutate the global text-predictor state after destruction.
   var destroyed = false;
-  var ingestedContextTexts = new Set();
+  // L1 ingestion cannot run before the vocabulary has loaded: `incrementSessionFreq`
+  // only touches trie nodes that already exist, so an empty trie silently rejects
+  // every word. Text that arrives first — which a page usually does, since the
+  // chat resolves it on focus in the same tick as the load starts — is held here
+  // and applied once the load has settled.
+  var pendingSessionContextTexts = new Set();
+  var sessionIngestedContextTexts = new Set();
+  var isVocabularyReady = false;
+  var vocabularyLoadPromise;
+  var addBoundedContextText = function addBoundedContextText(texts, text) {
+    texts.add(text);
+    // Evict oldest entries (Set preserves insertion order) to bound memory.
+    while (texts.size > MAX_INGESTED_CONTEXT_TEXTS) {
+      var oldest = texts.values().next().value;
+      if (oldest === undefined) {
+        break;
+      }
+      texts.delete(oldest);
+    }
+  };
+  var flushPendingSessionContext = function flushPendingSessionContext() {
+    if (!isVocabularyReady || destroyed) {
+      return;
+    }
+    var _iterator = _createForOfIteratorHelper(pendingSessionContextTexts),
+      _step;
+    try {
+      for (_iterator.s(); !(_step = _iterator.n()).done;) {
+        var text = _step.value;
+        if (!sessionIngestedContextTexts.has(text)) {
+          (0, _textPredictor.ingestDocumentPage)(text);
+          addBoundedContextText(sessionIngestedContextTexts, text);
+        }
+        pendingSessionContextTexts.delete(text);
+      }
+    } catch (err) {
+      _iterator.e(err);
+    } finally {
+      _iterator.f();
+    }
+  };
+  var ensureVocabularyReady = function ensureVocabularyReady() {
+    var _options$useLocalMode;
+    if (isVocabularyReady) {
+      flushPendingSessionContext();
+      return Promise.resolve();
+    }
+    vocabularyLoadPromise !== null && vocabularyLoadPromise !== void 0 ? vocabularyLoadPromise : vocabularyLoadPromise = (0, _textPredictor.loadDefaultVocabulary)({
+      isLocalLLM: (_options$useLocalMode = options === null || options === void 0 ? void 0 : options.useLocalModel) !== null && _options$useLocalMode !== void 0 ? _options$useLocalMode : false,
+      surface: surface
+    }).then(function () {
+      if (destroyed) {
+        return;
+      }
+      isVocabularyReady = true;
+      flushPendingSessionContext();
+    }).catch(function (error) {
+      // Do not consume the pending text on failure. A later focus retries the
+      // load and can still apply the original context exactly once.
+      vocabularyLoadPromise = undefined;
+      (0, _monitoring.logException)(error, {
+        location: 'editor-plugin-autocomplete/loadDefaultVocabulary'
+      });
+    });
+    return vocabularyLoadPromise;
+  };
   var logContextResolved = function logContextResolved(source, context) {
     var _context$parentCommen, _context$siblingComme4, _context$siblingComme5;
     if (!(0, _debugMode.isAutocompleteDebugEnabled)()) {
@@ -546,37 +611,29 @@ var createAutocompletePlugin = exports.createAutocompletePlugin = function creat
     }));
     resolvedContext = _objectSpread(_objectSpread({}, resolvedContext), definedContext);
     var ingestContextText = function ingestContextText(text) {
-      if (!text || ingestedContextTexts.has(text)) {
+      if (!text || sessionIngestedContextTexts.has(text)) {
         return;
       }
-      ingestedContextTexts.add(text);
-      // Evict oldest entries (Set preserves insertion order) to bound memory.
-      while (ingestedContextTexts.size > MAX_INGESTED_CONTEXT_TEXTS) {
-        var oldest = ingestedContextTexts.values().next().value;
-        if (oldest === undefined) {
-          break;
-        }
-        ingestedContextTexts.delete(oldest);
-      }
-      (0, _textPredictor.ingestDocumentPage)(text);
+      addBoundedContextText(pendingSessionContextTexts, text);
     };
     ingestContextText(context.fullPageContent);
     ingestContextText(context.parentCommentContent);
-    var _iterator = _createForOfIteratorHelper((_context$siblingComme6 = context.siblingCommentsContents) !== null && _context$siblingComme6 !== void 0 ? _context$siblingComme6 : []),
-      _step;
+    var _iterator2 = _createForOfIteratorHelper((_context$siblingComme6 = context.siblingCommentsContents) !== null && _context$siblingComme6 !== void 0 ? _context$siblingComme6 : []),
+      _step2;
     try {
-      for (_iterator.s(); !(_step = _iterator.n()).done;) {
-        var siblingCommentContent = _step.value;
+      for (_iterator2.s(); !(_step2 = _iterator2.n()).done;) {
+        var siblingCommentContent = _step2.value;
         ingestContextText(siblingCommentContent);
       }
-
-      // Context arrived after word boundaries may already have fired. Re-send
-      // slow-lane context immediately so the next inference includes the thread.
     } catch (err) {
-      _iterator.e(err);
+      _iterator2.e(err);
     } finally {
-      _iterator.f();
+      _iterator2.f();
     }
+    flushPendingSessionContext();
+
+    // Context arrived after word boundaries may already have fired. Re-send
+    // slow-lane context immediately so the next inference includes the thread.
     if (currentView) {
       slowLaneClient.updateContext(buildSlowLaneText(currentView.state.doc.textContent, resolvedContext));
     }
@@ -967,18 +1024,11 @@ var createAutocompletePlugin = exports.createAutocompletePlugin = function creat
           return accepted;
         },
         focus: function focus() {
-          var _options$useLocalMode, _options$useLocalMode2, _options$useLocalMode3;
+          var _options$useLocalMode2, _options$useLocalMode3;
           if (!isAutocompleteEnabled) {
             return false;
           }
-          (0, _textPredictor.loadDefaultVocabulary)({
-            isLocalLLM: (_options$useLocalMode = options === null || options === void 0 ? void 0 : options.useLocalModel) !== null && _options$useLocalMode !== void 0 ? _options$useLocalMode : false,
-            surface: surface
-          }).catch(function (error) {
-            (0, _monitoring.logException)(error, {
-              location: 'editor-plugin-autocomplete/loadDefaultVocabulary'
-            });
-          });
+          void ensureVocabularyReady();
           (0, _textPredictor.loadVectorsAsync)({
             isLocalLLM: (_options$useLocalMode2 = options === null || options === void 0 ? void 0 : options.useLocalModel) !== null && _options$useLocalMode2 !== void 0 ? _options$useLocalMode2 : false,
             surface: surface
@@ -1076,7 +1126,8 @@ var createAutocompletePlugin = exports.createAutocompletePlugin = function creat
           if (hasDestroy(slowLaneClient)) {
             slowLaneClient.destroy();
           }
-          ingestedContextTexts.clear();
+          pendingSessionContextTexts.clear();
+          sessionIngestedContextTexts.clear();
         }
       };
     }

package/dist/cjs/pm-plugins/debug-mode.js

@@ -3,7 +3,7 @@
 Object.defineProperty(exports, "__esModule", {
   value: true
 });
-exports.isAutocompleteDebugVerbose = exports.isAutocompleteDebugEnabled = exports.ctcTag = exports.ctcSection = exports.CTC_STYLES = void 0;
+exports.registerCtcSessionInspector = exports.isAutocompleteDebugVerbose = exports.isAutocompleteDebugEnabled = exports.ctcTag = exports.ctcSection = exports.CTC_STYLES = void 0;
 /**
  * Contextual Typeahead Completions (CTC) debug logging utility.
  *
@@ -96,7 +96,7 @@ var printLegend = function printLegend(verbose) {
   // eslint-disable-next-line no-console
   console.log('%cPlanes%c    [CTC:init] loads · [CTC:signal] semantic/network · [CTC:model] primes/exact · [CTC:model-cost] prefill/decode · [CTC:readiness] deadline progress', CTC_STYLES.section, CTC_STYLES.body);
   // eslint-disable-next-line no-console
-  console.log('%cInspect%c   __atlCtcDebug__.enable("verbose") · .disable()', CTC_STYLES.section, CTC_STYLES.body);
+  console.log('%cInspect%c   __atlCtcDebug__.enable("verbose") · .disable() · .session() (L1 boosts, narrow to a family with .session("poll"))', CTC_STYLES.section, CTC_STYLES.body);
   // eslint-disable-next-line no-console
   console.groupEnd();
 };
@@ -144,5 +144,20 @@ var isAutocompleteDebugVerbose = exports.isAutocompleteDebugVerbose = function i
   return (_getDebugApi$isVerbos = (_getDebugApi2 = getDebugApi()) === null || _getDebugApi2 === void 0 ? void 0 : _getDebugApi2.isVerbose()) !== null && _getDebugApi$isVerbos !== void 0 ? _getDebugApi$isVerbos : false;
 };
 
+/**
+ * Hang the L1 session-boost snapshot off the console API.
+ *
+ * Unlike the log helpers this is available whether or not debug is enabled:
+ * inspecting state on demand is not logging, and asking someone to turn on
+ * logging and retype to find out what the session already holds defeats the
+ * point of being able to ask.
+ */
+var registerCtcSessionInspector = exports.registerCtcSessionInspector = function registerCtcSessionInspector(inspect) {
+  var api = getDebugApi();
+  if (api) {
+    api.session = inspect;
+  }
+};
+
 // Eagerly install so the console API is available on load, regardless of call order.
 getDebugApi();