@atlaskit/editor-plugin-autocomplete 4.0.3 → 4.0.4

package/CHANGELOG.md

@@ -1,5 +1,20 @@
 # @atlaskit/editor-plugin-autocomplete
 
+## 4.0.4
+
+### Patch Changes
+
+- [`ac12ed0b41ef4`](https://bitbucket.org/atlassian/atlassian-frontend-monorepo/commits/ac12ed0b41ef4) -
+  Refactor contextual autocomplete wiring so the conversation-store-aware ChatInput owns the
+  autocomplete context bridge (getContext + subscribeToContextUpdates) via a new
+  useAutocompleteEditorContext hook, and the shared RovoChatPromptInput simply forwards a single
+  `autocomplete` prop to the editor preset. Removes the duplicated keying / listener-set plumbing
+  from the reusable input.
+
+  Adds an optional `subscribeToContextUpdates` option to the autocomplete plugin so host surfaces
+  that stream context after mount (e.g. Rovo chat messages) can push refreshes, complementing the
+  existing word-boundary retry that only covers a one-time, still-loading comment thread.
+
 ## 4.0.3
 
 ### Patch Changes

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

@@ -249,6 +249,7 @@ var createAutocompletePlugin = exports.createAutocompletePlugin = function creat
    * user has already typed several words before the promise resolved.
    */
   var currentView = null;
+  var unsubscribeFromContextUpdates;
   /**
    * Set after accepting a suggestion so the next doc-change update
    * skips scheduling a new prediction for the just-inserted text.
@@ -670,7 +671,21 @@ var createAutocompletePlugin = exports.createAutocompletePlugin = function creat
         }
       }
     },
-    view: function view() {
+    view: function view(editorView) {
+      // Capture up front so a subscription notification before the first PM
+      // transaction can still drive slowLaneClient.updateContext (gated on currentView).
+      currentView = editorView;
+
+      // Push channel for hosts that keep producing context after mount (e.g. Rovo chat).
+      if (isAutocompleteEnabled && options !== null && options !== void 0 && options.subscribeToContextUpdates) {
+        unsubscribeFromContextUpdates = options.subscribeToContextUpdates(function () {
+          // Bypass throttle for freshness; the in-flight guard prevents overlap.
+          refreshContext({
+            source: 'subscription',
+            allowThrottle: false
+          });
+        });
+      }
       return {
         update: function update(view, prevState) {
           if (!isAutocompleteEnabled) {
@@ -699,7 +714,9 @@ var createAutocompletePlugin = exports.createAutocompletePlugin = function creat
               // thread still loading). Retry on word boundaries until we have
               // the parent comment, throttled so we don't refetch constantly
               // and capped so non-comment editors stop retrying entirely.
-              if (!((_resolvedContext = resolvedContext) !== null && _resolvedContext !== void 0 && _resolvedContext.parentCommentContent) && wordBoundaryRefreshAttempts < MAX_CONTEXT_REFRESH_ATTEMPTS) {
+              // Skipped for push-channel hosts (subscribeToContextUpdates): they
+              // never grow parentCommentContent, so polling only burns the budget.
+              if (!(options !== null && options !== void 0 && options.subscribeToContextUpdates) && !((_resolvedContext = resolvedContext) !== null && _resolvedContext !== void 0 && _resolvedContext.parentCommentContent) && wordBoundaryRefreshAttempts < MAX_CONTEXT_REFRESH_ATTEMPTS) {
                 // Only count the attempt when a fetch actually started, so an
                 // in-flight or throttled no-op doesn't burn the retry budget.
                 if (refreshContext({
@@ -713,8 +730,11 @@ var createAutocompletePlugin = exports.createAutocompletePlugin = function creat
           }
         },
         destroy: function destroy() {
+          var _unsubscribeFromConte;
           destroyed = true;
           currentView = null;
+          (_unsubscribeFromConte = unsubscribeFromContextUpdates) === null || _unsubscribeFromConte === void 0 || _unsubscribeFromConte();
+          unsubscribeFromContextUpdates = undefined;
           if (debounceTimer) {
             clearTimeout(debounceTimer);
           }

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

@@ -264,7 +264,7 @@ var bePayloadDataPromise;
  * :param shape: `'object'` if the source JSON is `{...}`, `'array'` if `[...]`.
  * :returns: The parsed JSON value, or `null` if neither interop mode applies.
  */
-var unwrapJsonModule = function unwrapJsonModule(mod, shape) {
+function unwrapJsonModule(mod, shape) {
   if (mod == null || (0, _typeof2.default)(mod) !== 'object') {
     return null;
   }
@@ -316,7 +316,7 @@ var unwrapJsonModule = function unwrapJsonModule(mod, shape) {
     return namespace.default;
   }
   return null;
-};
+}
 
 /**
  * Lazily load and build the BE-parity lookup tables from their JSON payloads.
@@ -964,6 +964,12 @@ var createLocalSlowLaneClient = exports.createLocalSlowLaneClient = function cre
       return _regenerator.default.wrap(function (_context4) {
         while (1) switch (_context4.prev = _context4.next) {
           case 0:
+            captureCompletionTime = function _captureCompletionTim(promise, onResolved) {
+              return promise.then(function (value) {
+                onResolved(performance.now());
+                return value;
+              });
+            };
             if (!(!engine || destroyed)) {
               _context4.next = 1;
               break;
@@ -983,12 +989,6 @@ var createLocalSlowLaneClient = exports.createLocalSlowLaneClient = function cre
             lmText = truncateToLastNWords(text, BE_PARITY.MAX_CONTEXT_TOKENS);
             semanticText = truncateToLastNWords(text, BE_PARITY.MAX_CONTEXT_WORDS);
             arcticInput = wrapForArctic(semanticText);
-            captureCompletionTime = function captureCompletionTime(promise, onResolved) {
-              return promise.then(function (value) {
-                onResolved(performance.now());
-                return value;
-              });
-            };
             if ((0, _debugMode.isAutocompleteDebugEnabled)()) {
               // eslint-disable-next-line no-console
               console.log("%c[LocalSlowLane] %c\uD83D\uDD22 Arctic input (".concat(arcticInput.length, " chars, ").concat(splitOnWhitespace(semanticText).length, " words): \"").concat(arcticInput.length > 100 ? "".concat(arcticInput.slice(0, 100), "\u2026") : arcticInput, "\""), 'color: #9c27b0; font-weight: bold;', 'color: #009688;');

package/dist/cjs/pm-plugins/text-predictor.js

@@ -694,7 +694,7 @@ var predict = exports.predict = function predict(textBefore) {
  * array and a sparse numeric-keyed object are emitted identically as named
  * exports. Kept in lock-step with the matching helper in local-slow-lane-client.ts.
  */
-var unwrapJsonModule = function unwrapJsonModule(mod, shape) {
+function unwrapJsonModule(mod, shape) {
   if (mod == null || (0, _typeof2.default)(mod) !== 'object') {
     return null;
   }
@@ -730,7 +730,7 @@ var unwrapJsonModule = function unwrapJsonModule(mod, shape) {
     return namespace.default;
   }
   return null;
-};
+}
 var loadVectorsAsync = exports.loadVectorsAsync = /*#__PURE__*/function () {
   var _ref6 = (0, _asyncToGenerator2.default)( /*#__PURE__*/_regenerator.default.mark(function _callee(options) {
     var _options$isLocalLLM;

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

@@ -237,6 +237,7 @@ export const createAutocompletePlugin = (options, api) => {
    * user has already typed several words before the promise resolved.
    */
   let currentView = null;
+  let unsubscribeFromContextUpdates;
   /**
    * Set after accepting a suggestion so the next doc-change update
    * skips scheduling a new prediction for the just-inserted text.
@@ -657,59 +658,80 @@ export const createAutocompletePlugin = (options, api) => {
         }
       }
     },
-    view: () => ({
-      update: (view, prevState) => {
-        if (!isAutocompleteEnabled) {
-          return;
-        }
-        currentView = view;
-        if (!prevState.doc.eq(view.state.doc)) {
-          if (justAccepted) {
-            justAccepted = false;
+    view: editorView => {
+      // Capture up front so a subscription notification before the first PM
+      // transaction can still drive slowLaneClient.updateContext (gated on currentView).
+      currentView = editorView;
 
-            // Snapshot the post-acceptance text so follow-up transactions hit
-            // the dismissedContext guard and abort until the user types again.
-            dismissedContext = getTextBeforeCursor(view.state);
-            if (debounceTimer) {
-              clearTimeout(debounceTimer);
-            }
+      // Push channel for hosts that keep producing context after mount (e.g. Rovo chat).
+      if (isAutocompleteEnabled && options !== null && options !== void 0 && options.subscribeToContextUpdates) {
+        unsubscribeFromContextUpdates = options.subscribeToContextUpdates(() => {
+          // Bypass throttle for freshness; the in-flight guard prevents overlap.
+          refreshContext({
+            source: 'subscription',
+            allowThrottle: false
+          });
+        });
+      }
+      return {
+        update: (view, prevState) => {
+          if (!isAutocompleteEnabled) {
             return;
           }
-          maybeUpdateSessionFrequency(view, prevState);
-          const textBefore = getTextBeforeCursor(view.state);
-          if (isWordBoundary(textBefore)) {
-            var _resolvedContext;
-            slowLaneClient.updateContext(buildSlowLaneText(view.state.doc.textContent, resolvedContext));
+          currentView = view;
+          if (!prevState.doc.eq(view.state.doc)) {
+            if (justAccepted) {
+              justAccepted = false;
+
+              // Snapshot the post-acceptance text so follow-up transactions hit
+              // the dismissedContext guard and abort until the user types again.
+              dismissedContext = getTextBeforeCursor(view.state);
+              if (debounceTimer) {
+                clearTimeout(debounceTimer);
+              }
+              return;
+            }
+            maybeUpdateSessionFrequency(view, prevState);
+            const textBefore = getTextBeforeCursor(view.state);
+            if (isWordBoundary(textBefore)) {
+              var _resolvedContext;
+              slowLaneClient.updateContext(buildSlowLaneText(view.state.doc.textContent, resolvedContext));
 
-            // Context may not have resolved on first focus (e.g. comment
-            // thread still loading). Retry on word boundaries until we have
-            // the parent comment, throttled so we don't refetch constantly
-            // and capped so non-comment editors stop retrying entirely.
-            if (!((_resolvedContext = resolvedContext) !== null && _resolvedContext !== void 0 && _resolvedContext.parentCommentContent) && wordBoundaryRefreshAttempts < MAX_CONTEXT_REFRESH_ATTEMPTS) {
-              // Only count the attempt when a fetch actually started, so an
-              // in-flight or throttled no-op doesn't burn the retry budget.
-              if (refreshContext({
-                source: 'word-boundary'
-              })) {
-                wordBoundaryRefreshAttempts++;
+              // Context may not have resolved on first focus (e.g. comment
+              // thread still loading). Retry on word boundaries until we have
+              // the parent comment, throttled so we don't refetch constantly
+              // and capped so non-comment editors stop retrying entirely.
+              // Skipped for push-channel hosts (subscribeToContextUpdates): they
+              // never grow parentCommentContent, so polling only burns the budget.
+              if (!(options !== null && options !== void 0 && options.subscribeToContextUpdates) && !((_resolvedContext = resolvedContext) !== null && _resolvedContext !== void 0 && _resolvedContext.parentCommentContent) && wordBoundaryRefreshAttempts < MAX_CONTEXT_REFRESH_ATTEMPTS) {
+                // Only count the attempt when a fetch actually started, so an
+                // in-flight or throttled no-op doesn't burn the retry budget.
+                if (refreshContext({
+                  source: 'word-boundary'
+                })) {
+                  wordBoundaryRefreshAttempts++;
+                }
               }
             }
+            schedulePrediction(view);
           }
-          schedulePrediction(view);
-        }
-      },
-      destroy: () => {
-        destroyed = true;
-        currentView = null;
-        if (debounceTimer) {
-          clearTimeout(debounceTimer);
-        }
-        if (hasDestroy(slowLaneClient)) {
-          slowLaneClient.destroy();
+        },
+        destroy: () => {
+          var _unsubscribeFromConte;
+          destroyed = true;
+          currentView = null;
+          (_unsubscribeFromConte = unsubscribeFromContextUpdates) === null || _unsubscribeFromConte === void 0 ? void 0 : _unsubscribeFromConte();
+          unsubscribeFromContextUpdates = undefined;
+          if (debounceTimer) {
+            clearTimeout(debounceTimer);
+          }
+          if (hasDestroy(slowLaneClient)) {
+            slowLaneClient.destroy();
+          }
+          ingestedContextTexts.clear();
+          setDefaultSlowLaneClient(null);
         }
-        ingestedContextTexts.clear();
-        setDefaultSlowLaneClient(null);
-      }
-    })
+      };
+    }
   });
 };

package/dist/es2019/pm-plugins/local-slow-lane-client.js

@@ -247,7 +247,7 @@ let bePayloadDataPromise;
  * :param shape: `'object'` if the source JSON is `{...}`, `'array'` if `[...]`.
  * :returns: The parsed JSON value, or `null` if neither interop mode applies.
  */
-const unwrapJsonModule = (mod, shape) => {
+function unwrapJsonModule(mod, shape) {
   if (mod == null || typeof mod !== 'object') {
     return null;
   }
@@ -288,7 +288,7 @@ const unwrapJsonModule = (mod, shape) => {
     return namespace.default;
   }
   return null;
-};
+}
 
 /**
  * Lazily load and build the BE-parity lookup tables from their JSON payloads.
@@ -764,10 +764,12 @@ export const createLocalSlowLaneClient = (config = {}) => {
     const lmText = truncateToLastNWords(text, BE_PARITY.MAX_CONTEXT_TOKENS);
     const semanticText = truncateToLastNWords(text, BE_PARITY.MAX_CONTEXT_WORDS);
     const arcticInput = wrapForArctic(semanticText);
-    const captureCompletionTime = (promise, onResolved) => promise.then(value => {
-      onResolved(performance.now());
-      return value;
-    });
+    function captureCompletionTime(promise, onResolved) {
+      return promise.then(value => {
+        onResolved(performance.now());
+        return value;
+      });
+    }
     if (isAutocompleteDebugEnabled()) {
       // eslint-disable-next-line no-console
       console.log(`%c[LocalSlowLane] %c🔢 Arctic input (${arcticInput.length} chars, ${splitOnWhitespace(semanticText).length} words): "${arcticInput.length > 100 ? `${arcticInput.slice(0, 100)}…` : arcticInput}"`, 'color: #9c27b0; font-weight: bold;', 'color: #009688;');

package/dist/es2019/pm-plugins/text-predictor.js

@@ -560,7 +560,7 @@ export const predict = textBefore => {
  * array and a sparse numeric-keyed object are emitted identically as named
  * exports. Kept in lock-step with the matching helper in local-slow-lane-client.ts.
  */
-const unwrapJsonModule = (mod, shape) => {
+function unwrapJsonModule(mod, shape) {
   if (mod == null || typeof mod !== 'object') {
     return null;
   }
@@ -585,7 +585,7 @@ const unwrapJsonModule = (mod, shape) => {
     return namespace.default;
   }
   return null;
-};
+}
 export const loadVectorsAsync = async options => {
   var _options$isLocalLLM;
   if (vectorStore || vectorsLoadStarted) {

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

@@ -242,6 +242,7 @@ export var createAutocompletePlugin = function createAutocompletePlugin(options,
    * user has already typed several words before the promise resolved.
    */
   var currentView = null;
+  var unsubscribeFromContextUpdates;
   /**
    * Set after accepting a suggestion so the next doc-change update
    * skips scheduling a new prediction for the just-inserted text.
@@ -663,7 +664,21 @@ export var createAutocompletePlugin = function createAutocompletePlugin(options,
         }
       }
     },
-    view: function view() {
+    view: function view(editorView) {
+      // Capture up front so a subscription notification before the first PM
+      // transaction can still drive slowLaneClient.updateContext (gated on currentView).
+      currentView = editorView;
+
+      // Push channel for hosts that keep producing context after mount (e.g. Rovo chat).
+      if (isAutocompleteEnabled && options !== null && options !== void 0 && options.subscribeToContextUpdates) {
+        unsubscribeFromContextUpdates = options.subscribeToContextUpdates(function () {
+          // Bypass throttle for freshness; the in-flight guard prevents overlap.
+          refreshContext({
+            source: 'subscription',
+            allowThrottle: false
+          });
+        });
+      }
       return {
         update: function update(view, prevState) {
           if (!isAutocompleteEnabled) {
@@ -692,7 +707,9 @@ export var createAutocompletePlugin = function createAutocompletePlugin(options,
               // thread still loading). Retry on word boundaries until we have
               // the parent comment, throttled so we don't refetch constantly
               // and capped so non-comment editors stop retrying entirely.
-              if (!((_resolvedContext = resolvedContext) !== null && _resolvedContext !== void 0 && _resolvedContext.parentCommentContent) && wordBoundaryRefreshAttempts < MAX_CONTEXT_REFRESH_ATTEMPTS) {
+              // Skipped for push-channel hosts (subscribeToContextUpdates): they
+              // never grow parentCommentContent, so polling only burns the budget.
+              if (!(options !== null && options !== void 0 && options.subscribeToContextUpdates) && !((_resolvedContext = resolvedContext) !== null && _resolvedContext !== void 0 && _resolvedContext.parentCommentContent) && wordBoundaryRefreshAttempts < MAX_CONTEXT_REFRESH_ATTEMPTS) {
                 // Only count the attempt when a fetch actually started, so an
                 // in-flight or throttled no-op doesn't burn the retry budget.
                 if (refreshContext({
@@ -706,8 +723,11 @@ export var createAutocompletePlugin = function createAutocompletePlugin(options,
           }
         },
         destroy: function destroy() {
+          var _unsubscribeFromConte;
           destroyed = true;
           currentView = null;
+          (_unsubscribeFromConte = unsubscribeFromContextUpdates) === null || _unsubscribeFromConte === void 0 || _unsubscribeFromConte();
+          unsubscribeFromContextUpdates = undefined;
           if (debounceTimer) {
             clearTimeout(debounceTimer);
           }

package/dist/esm/pm-plugins/local-slow-lane-client.js

@@ -258,7 +258,7 @@ var bePayloadDataPromise;
  * :param shape: `'object'` if the source JSON is `{...}`, `'array'` if `[...]`.
  * :returns: The parsed JSON value, or `null` if neither interop mode applies.
  */
-var unwrapJsonModule = function unwrapJsonModule(mod, shape) {
+function unwrapJsonModule(mod, shape) {
   if (mod == null || _typeof(mod) !== 'object') {
     return null;
   }
@@ -310,7 +310,7 @@ var unwrapJsonModule = function unwrapJsonModule(mod, shape) {
     return namespace.default;
   }
   return null;
-};
+}
 
 /**
  * Lazily load and build the BE-parity lookup tables from their JSON payloads.
@@ -952,6 +952,12 @@ export var createLocalSlowLaneClient = function createLocalSlowLaneClient() {
       return _regeneratorRuntime.wrap(function (_context4) {
         while (1) switch (_context4.prev = _context4.next) {
           case 0:
+            captureCompletionTime = function _captureCompletionTim(promise, onResolved) {
+              return promise.then(function (value) {
+                onResolved(performance.now());
+                return value;
+              });
+            };
             if (!(!engine || destroyed)) {
               _context4.next = 1;
               break;
@@ -971,12 +977,6 @@ export var createLocalSlowLaneClient = function createLocalSlowLaneClient() {
             lmText = truncateToLastNWords(text, BE_PARITY.MAX_CONTEXT_TOKENS);
             semanticText = truncateToLastNWords(text, BE_PARITY.MAX_CONTEXT_WORDS);
             arcticInput = wrapForArctic(semanticText);
-            captureCompletionTime = function captureCompletionTime(promise, onResolved) {
-              return promise.then(function (value) {
-                onResolved(performance.now());
-                return value;
-              });
-            };
             if (isAutocompleteDebugEnabled()) {
               // eslint-disable-next-line no-console
               console.log("%c[LocalSlowLane] %c\uD83D\uDD22 Arctic input (".concat(arcticInput.length, " chars, ").concat(splitOnWhitespace(semanticText).length, " words): \"").concat(arcticInput.length > 100 ? "".concat(arcticInput.slice(0, 100), "\u2026") : arcticInput, "\""), 'color: #9c27b0; font-weight: bold;', 'color: #009688;');

package/dist/esm/pm-plugins/text-predictor.js

@@ -690,7 +690,7 @@ export var predict = function predict(textBefore) {
  * array and a sparse numeric-keyed object are emitted identically as named
  * exports. Kept in lock-step with the matching helper in local-slow-lane-client.ts.
  */
-var unwrapJsonModule = function unwrapJsonModule(mod, shape) {
+function unwrapJsonModule(mod, shape) {
   if (mod == null || _typeof(mod) !== 'object') {
     return null;
   }
@@ -726,7 +726,7 @@ var unwrapJsonModule = function unwrapJsonModule(mod, shape) {
     return namespace.default;
   }
   return null;
-};
+}
 export var loadVectorsAsync = /*#__PURE__*/function () {
   var _ref6 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime.mark(function _callee(options) {
     var _options$isLocalLLM;

package/dist/types/index.d.ts

@@ -1,2 +1,3 @@
 export { autocompletePlugin } from './autocompletePlugin';
 export type { AutocompletePlugin } from './autocompletePluginType';
+export type { AutocompleteContext, AutocompletePluginOptions, } from './pm-plugins/autocomplete-plugin';

package/dist/types/pm-plugins/autocomplete-plugin.d.ts

@@ -40,6 +40,12 @@ export interface AutocompletePluginOptions {
      * Use this to serve vectors from a CDN or media service in production.
      */
     getVectorsBinaryUrl?: () => Promise<string>;
+    /**
+     * Subscription for hosts with live external context (e.g. Rovo chat). The plugin
+     * re-fetches via getContext() on each notification and unsubscribes on destroy.
+     * Only meaningful alongside `getContext`; without it each notification is a no-op.
+     */
+    subscribeToContextUpdates?: (onContextUpdated: () => void) => () => void;
     /**
      * User locale used to determine whether autocomplete should run.
      * Defaults to browser locale when omitted.

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@atlaskit/editor-plugin-autocomplete",
-	"version": "4.0.3",
+	"version": "4.0.4",
 	"description": "Client-side text autocomplete plugin for @atlaskit/editor-core",
 	"author": "Atlassian Pty Ltd",
 	"license": "Apache-2.0",
@@ -27,7 +27,7 @@
 		"wink-nlp": "^2.4.0"
 	},
 	"peerDependencies": {
-		"@atlaskit/editor-common": "^116.6.0",
+		"@atlaskit/editor-common": "^116.15.0",
 		"@atlaskit/editor-plugin-analytics": "^12.0.0",
 		"react": "^18.2.0"
 	},

package/src/index.ts

@@ -3,3 +3,7 @@
 
 export { autocompletePlugin } from './autocompletePlugin';
 export type { AutocompletePlugin } from './autocompletePluginType';
+export type {
+	AutocompleteContext,
+	AutocompletePluginOptions,
+} from './pm-plugins/autocomplete-plugin';

package/src/pm-plugins/autocomplete-plugin.ts

@@ -214,6 +214,12 @@ export interface AutocompletePluginOptions {
 	 * Use this to serve vectors from a CDN or media service in production.
 	 */
 	getVectorsBinaryUrl?: () => Promise<string>;
+	/**
+	 * Subscription for hosts with live external context (e.g. Rovo chat). The plugin
+	 * re-fetches via getContext() on each notification and unsubscribes on destroy.
+	 * Only meaningful alongside `getContext`; without it each notification is a no-op.
+	 */
+	subscribeToContextUpdates?: (onContextUpdated: () => void) => () => void;
 	/**
 	 * User locale used to determine whether autocomplete should run.
 	 * Defaults to browser locale when omitted.
@@ -329,6 +335,7 @@ export const createAutocompletePlugin = (
 	 * user has already typed several words before the promise resolved.
 	 */
 	let currentView: EditorView | null = null;
+	let unsubscribeFromContextUpdates: (() => void) | undefined;
 	/**
 	 * Set after accepting a suggestion so the next doc-change update
 	 * skips scheduling a new prediction for the just-inserted text.
@@ -582,7 +589,9 @@ export const createAutocompletePlugin = (
 				// nodeAfter correctly handles inline atoms (mentions, emojis) where
 				// parentOffset and textContent indices diverge.
 				const nodeAfter = selection.$from.nodeAfter;
-				const charAfterCursor = nodeAfter?.isText ? getLeadingTextCharacter(nodeAfter.text) : undefined;
+				const charAfterCursor = nodeAfter?.isText
+					? getLeadingTextCharacter(nodeAfter.text)
+					: undefined;
 				// Only suppress for Unicode letters, digits, and underscore — hyphens,
 				// apostrophes, and similar punctuation are valid left-edge boundaries
 				// and should not block suggestions (e.g. cursor before '-' in compound-word).
@@ -790,66 +799,85 @@ export const createAutocompletePlugin = (
 			},
 		},
 
-		view: () => ({
-			update: (view: EditorView, prevState: EditorState) => {
-				if (!isAutocompleteEnabled) {
-					return;
-				}
+		view: (editorView: EditorView) => {
+			// Capture up front so a subscription notification before the first PM
+			// transaction can still drive slowLaneClient.updateContext (gated on currentView).
+			currentView = editorView;
 
-				currentView = view;
-				if (!prevState.doc.eq(view.state.doc)) {
-					if (justAccepted) {
-						justAccepted = false;
-
-						// Snapshot the post-acceptance text so follow-up transactions hit
-						// the dismissedContext guard and abort until the user types again.
-						dismissedContext = getTextBeforeCursor(view.state);
+			// Push channel for hosts that keep producing context after mount (e.g. Rovo chat).
+			if (isAutocompleteEnabled && options?.subscribeToContextUpdates) {
+				unsubscribeFromContextUpdates = options.subscribeToContextUpdates(() => {
+					// Bypass throttle for freshness; the in-flight guard prevents overlap.
+					refreshContext({ source: 'subscription', allowThrottle: false });
+				});
+			}
 
-						if (debounceTimer) {
-							clearTimeout(debounceTimer);
-						}
+			return {
+				update: (view: EditorView, prevState: EditorState) => {
+					if (!isAutocompleteEnabled) {
 						return;
 					}
 
-					maybeUpdateSessionFrequency(view, prevState);
-
-					const textBefore = getTextBeforeCursor(view.state);
-					if (isWordBoundary(textBefore)) {
-						slowLaneClient.updateContext(
-							buildSlowLaneText(view.state.doc.textContent, resolvedContext),
-						);
-
-						// Context may not have resolved on first focus (e.g. comment
-						// thread still loading). Retry on word boundaries until we have
-						// the parent comment, throttled so we don't refetch constantly
-						// and capped so non-comment editors stop retrying entirely.
-						if (
-							!resolvedContext?.parentCommentContent &&
-							wordBoundaryRefreshAttempts < MAX_CONTEXT_REFRESH_ATTEMPTS
-						) {
-							// Only count the attempt when a fetch actually started, so an
-							// in-flight or throttled no-op doesn't burn the retry budget.
-							if (refreshContext({ source: 'word-boundary' })) {
-								wordBoundaryRefreshAttempts++;
+					currentView = view;
+					if (!prevState.doc.eq(view.state.doc)) {
+						if (justAccepted) {
+							justAccepted = false;
+
+							// Snapshot the post-acceptance text so follow-up transactions hit
+							// the dismissedContext guard and abort until the user types again.
+							dismissedContext = getTextBeforeCursor(view.state);
+
+							if (debounceTimer) {
+								clearTimeout(debounceTimer);
 							}
+							return;
 						}
-					}
 
-					schedulePrediction(view);
-				}
-			},
-			destroy: () => {
-				destroyed = true;
-				currentView = null;
-				if (debounceTimer) {
-					clearTimeout(debounceTimer);
-				}
-				if (hasDestroy(slowLaneClient)) {
-					slowLaneClient.destroy();
-				}
-				ingestedContextTexts.clear();
-				setDefaultSlowLaneClient(null);
-			},
-		}),
+						maybeUpdateSessionFrequency(view, prevState);
+
+						const textBefore = getTextBeforeCursor(view.state);
+						if (isWordBoundary(textBefore)) {
+							slowLaneClient.updateContext(
+								buildSlowLaneText(view.state.doc.textContent, resolvedContext),
+							);
+
+							// Context may not have resolved on first focus (e.g. comment
+							// thread still loading). Retry on word boundaries until we have
+							// the parent comment, throttled so we don't refetch constantly
+							// and capped so non-comment editors stop retrying entirely.
+							// Skipped for push-channel hosts (subscribeToContextUpdates): they
+							// never grow parentCommentContent, so polling only burns the budget.
+							if (
+								!options?.subscribeToContextUpdates &&
+								!resolvedContext?.parentCommentContent &&
+								wordBoundaryRefreshAttempts < MAX_CONTEXT_REFRESH_ATTEMPTS
+							) {
+								// Only count the attempt when a fetch actually started, so an
+								// in-flight or throttled no-op doesn't burn the retry budget.
+								if (refreshContext({ source: 'word-boundary' })) {
+									wordBoundaryRefreshAttempts++;
+								}
+							}
+						}
+
+						schedulePrediction(view);
+					}
+				},
+				destroy: () => {
+					destroyed = true;
+					currentView = null;
+					unsubscribeFromContextUpdates?.();
+					unsubscribeFromContextUpdates = undefined;
+					if (debounceTimer) {
+						clearTimeout(debounceTimer);
+					}
+					if (hasDestroy(slowLaneClient)) {
+						slowLaneClient.destroy();
+					}
+					ingestedContextTexts.clear();
+					setDefaultSlowLaneClient(null);
+				},
+			};
+		},
 	});
 };

package/src/pm-plugins/local-slow-lane-client.ts

@@ -369,7 +369,7 @@ let bePayloadDataPromise: Promise<void> | undefined;
  * :param shape: `'object'` if the source JSON is `{...}`, `'array'` if `[...]`.
  * :returns: The parsed JSON value, or `null` if neither interop mode applies.
  */
-const unwrapJsonModule = <T>(mod: unknown, shape: 'object' | 'array'): T | null => {
+function unwrapJsonModule<T>(mod: unknown, shape: 'object' | 'array'): T | null {
 	if (mod == null || typeof mod !== 'object') {
 		return null;
 	}
@@ -411,7 +411,7 @@ const unwrapJsonModule = <T>(mod: unknown, shape: 'object' | 'array'): T | null
 	}
 
 	return null;
-};
+}
 
 /**
  * Lazily load and build the BE-parity lookup tables from their JSON payloads.
@@ -1004,14 +1004,15 @@ export const createLocalSlowLaneClient = (
 		const lmText = truncateToLastNWords(text, BE_PARITY.MAX_CONTEXT_TOKENS);
 		const semanticText = truncateToLastNWords(text, BE_PARITY.MAX_CONTEXT_WORDS);
 		const arcticInput = wrapForArctic(semanticText);
-		const captureCompletionTime = <T>(
+		function captureCompletionTime<T>(
 			promise: Promise<T>,
 			onResolved: (resolvedAt: number) => void,
-		): Promise<T> =>
-			promise.then((value: T) => {
+		): Promise<T> {
+			return promise.then((value: T) => {
 				onResolved(performance.now());
 				return value;
 			});
+		}
 
 		if (isAutocompleteDebugEnabled()) {
 			// eslint-disable-next-line no-console

package/src/pm-plugins/text-predictor.ts

@@ -715,7 +715,7 @@ interface VocabularyJson {
  * array and a sparse numeric-keyed object are emitted identically as named
  * exports. Kept in lock-step with the matching helper in local-slow-lane-client.ts.
  */
-const unwrapJsonModule = <T>(mod: unknown, shape: 'object' | 'array'): T | null => {
+function unwrapJsonModule<T>(mod: unknown, shape: 'object' | 'array'): T | null {
 	if (mod == null || typeof mod !== 'object') {
 		return null;
 	}
@@ -743,7 +743,7 @@ const unwrapJsonModule = <T>(mod: unknown, shape: 'object' | 'array'): T | null
 	}
 
 	return null;
-};
+}
 
 export const loadVectorsAsync = async (options?: {
 	getBinaryUrl?: () => Promise<string>;