@atlaskit/editor-plugin-autocomplete 4.0.2 → 4.0.4

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/analytics/ufo.js

@@ -1,4 +1,6 @@
 import _defineProperty from "@babel/runtime/helpers/defineProperty";
+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) { _defineProperty(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; }
 /**
  * UFO experience tracking helpers for `@atlaskit/editor-plugin-autocomplete`.
  *
@@ -89,15 +91,21 @@ export var failExp = function failExp(name, id, metadata) {
     // UFO errors must never break the plugin
   }
 };
-export var abortExp = function abortExp(name, id, reason) {
+export var abortExp = function abortExp(name, id, reason, metadata) {
   if (!isUfoEnabled()) {
     return;
   }
   try {
-    experiences[name].getInstance(id).abort(reason ? {
-      metadata: {
-        reason: reason
-      }
+    // `reason` is the dedicated, canonical channel for the abort reason and
+    // intentionally takes precedence over a `reason` key in `metadata`. Callers
+    // must pass the reason via the param, not inside `metadata` (the `reason`
+    // key there is reserved and would be overwritten).
+    var merged = _objectSpread(_objectSpread({}, metadata), reason ? {
+      reason: reason
+    } : {});
+    var hasMetadata = Object.keys(merged).length > 0;
+    experiences[name].getInstance(id).abort(hasMetadata ? {
+      metadata: merged
     } : undefined);
   } catch (_unused4) {
     // UFO errors must never break the plugin

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

@@ -48,6 +48,7 @@ function _arrayLikeToArray(r, a) { (null == a || a > r.length) && (a = r.length)
  * asynchronously after each updateContext() call.
  */
 
+import { abortExp, EXPERIENCE_NAME, failExp, startExp, succeedExp } from '../analytics/ufo';
 import { isAutocompleteDebugEnabled } from './debug-mode';
 import { isWordBoundary } from './slow-lane-client';
 
@@ -257,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;
   }
@@ -309,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.
@@ -947,17 +948,23 @@ export var createLocalSlowLaneClient = function createLocalSlowLaneClient() {
    */
   var runInference = /*#__PURE__*/function () {
     var _ref0 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime.mark(function _callee4(text, requestId) {
-      var lmText, semanticText, arcticInput, captureCompletionTime, _data, tStart, tLmDone, tEmbDone, _yield$Promise$all5, _yield$Promise$all6, embeddingResponse, rawLogits, payload, embedding, sumSq, i, topTokens, errorMsg, _t5;
+      var experienceId, lmText, semanticText, arcticInput, captureCompletionTime, _data, tStart, tLmDone, tEmbDone, _yield$Promise$all5, _yield$Promise$all6, embeddingResponse, rawLogits, payload, embedding, sumSq, i, topTokens, errorMsg, _t5;
       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;
             }
             return _context4.abrupt("return");
           case 1:
-            // Clear the capture buffer so we read only this pass's logits. The engine
+            experienceId = String(requestId); // Clear the capture buffer so we read only this pass's logits. The engine
             // serialises per-model requests and updateContext is debounced, so the
             // latest request's decode step is the last to populate `captured` before
             // we read it below; stale requests bail on the latestRequestId guard.
@@ -970,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;');
@@ -983,6 +984,16 @@ export var createLocalSlowLaneClient = function createLocalSlowLaneClient() {
               console.log("%c[LocalSlowLane] %c\uD83E\uDDE0 LM input (".concat(lmText.length, " chars, ").concat(splitOnWhitespace(lmText).length, " words): \"").concat(lmText.length > 100 ? "".concat(lmText.slice(0, 100), "\u2026") : lmText, "\""), 'color: #9c27b0; font-weight: bold;', 'color: #2196f3;');
             }
             _context4.prev = 2;
+            // Reuse the network slow-lane-fetch UFO experience (tagged isLocalLLM:true,
+            // matching LOAD_VECTORS/LOAD_VOCABULARY) so on-device inference
+            // latency/success-rate feeds the same FE Reliability SLO. Started inside
+            // the try — post engine-init (parity with the network fetch, which
+            // excludes the one-time model load) and so the catch below always
+            // terminates the experience, even on a synchronous throw.
+            startExp(EXPERIENCE_NAME.SLOW_LANE_FETCH, experienceId, {
+              textLength: text.length,
+              isLocalLLM: true
+            });
             tStart = performance.now();
             tLmDone = 0;
             tEmbDone = 0;
@@ -1015,6 +1026,9 @@ export var createLocalSlowLaneClient = function createLocalSlowLaneClient() {
               _context4.next = 4;
               break;
             }
+            abortExp(EXPERIENCE_NAME.SLOW_LANE_FETCH, experienceId, destroyed ? 'destroyed' : 'superseded', {
+              isLocalLLM: true
+            });
             return _context4.abrupt("return");
           case 4:
             // ── LM logits: whole-word BE-parity payload ──────────────────
@@ -1066,6 +1080,12 @@ export var createLocalSlowLaneClient = function createLocalSlowLaneClient() {
               // eslint-disable-next-line no-console
               console.groupEnd();
             }
+            succeedExp(EXPERIENCE_NAME.SLOW_LANE_FETCH, experienceId, {
+              textLength: text.length,
+              hasVector: storedContextVector !== null,
+              hasLmLogits: storedLmLogits !== null,
+              isLocalLLM: true
+            });
             onUpdate === null || onUpdate === void 0 || onUpdate({
               textLength: text.length,
               hasVector: storedContextVector !== null,
@@ -1080,10 +1100,17 @@ export var createLocalSlowLaneClient = function createLocalSlowLaneClient() {
               _context4.next = 6;
               break;
             }
+            abortExp(EXPERIENCE_NAME.SLOW_LANE_FETCH, experienceId, destroyed ? 'destroyed' : 'superseded', {
+              isLocalLLM: true
+            });
             return _context4.abrupt("return");
           case 6:
             storedContextVector = null;
             storedLmLogits = null;
+            failExp(EXPERIENCE_NAME.SLOW_LANE_FETCH, experienceId, {
+              errorType: 'inference',
+              isLocalLLM: true
+            });
             onUpdate === null || onUpdate === void 0 || onUpdate({
               textLength: text.length,
               hasVector: false,

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

@@ -81,7 +81,8 @@ export var createSlowLaneClient = function createSlowLaneClient(config) {
               session_id: sessionId
             };
             startExp(EXPERIENCE_NAME.SLOW_LANE_FETCH, requestId, {
-              textLength: text.length
+              textLength: text.length,
+              isLocalLLM: false
             });
             if (isAutocompleteDebugEnabled()) {
               // eslint-disable-next-line no-console
@@ -110,7 +111,8 @@ export var createSlowLaneClient = function createSlowLaneClient(config) {
             storedLmLogits = null;
             failExp(EXPERIENCE_NAME.SLOW_LANE_FETCH, requestId, {
               status: res.status,
-              errorType: 'http_error'
+              errorType: 'http_error',
+              isLocalLLM: false
             });
             if (isAutocompleteDebugEnabled()) {
               // eslint-disable-next-line no-console
@@ -145,7 +147,8 @@ export var createSlowLaneClient = function createSlowLaneClient(config) {
             succeedExp(EXPERIENCE_NAME.SLOW_LANE_FETCH, requestId, {
               textLength: text.length,
               hasVector: storedContextVector !== null,
-              hasLmLogits: storedLmLogits !== null
+              hasLmLogits: storedLmLogits !== null,
+              isLocalLLM: false
             });
             onUpdate === null || onUpdate === void 0 || onUpdate({
               textLength: text.length,
@@ -161,7 +164,8 @@ export var createSlowLaneClient = function createSlowLaneClient(config) {
             storedContextVector = null;
             storedLmLogits = null;
             failExp(EXPERIENCE_NAME.SLOW_LANE_FETCH, requestId, {
-              errorType: 'network'
+              errorType: 'network',
+              isLocalLLM: false
             });
             if (isAutocompleteDebugEnabled()) {
               // eslint-disable-next-line no-console
@@ -191,7 +195,9 @@ export var createSlowLaneClient = function createSlowLaneClient(config) {
     debounceTimer = setTimeout(function () {
       debounceTimer = null;
       if (inflightRequestId !== null) {
-        abortExp(EXPERIENCE_NAME.SLOW_LANE_FETCH, inflightRequestId, 'superseded');
+        abortExp(EXPERIENCE_NAME.SLOW_LANE_FETCH, inflightRequestId, 'superseded', {
+          isLocalLLM: false
+        });
       }
       var requestId = String(++requestSeq);
       inflightRequestId = requestId;

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/analytics/ufo.d.ts

@@ -35,4 +35,4 @@ export type AutocompleteExperienceName = (typeof EXPERIENCE_NAME)[keyof typeof E
 export declare const startExp: (name: AutocompleteExperienceName, id: string, metadata?: CustomData) => void;
 export declare const succeedExp: (name: AutocompleteExperienceName, id: string, metadata?: CustomData) => void;
 export declare const failExp: (name: AutocompleteExperienceName, id: string, metadata?: CustomData) => void;
-export declare const abortExp: (name: AutocompleteExperienceName, id: string, reason?: string) => void;
+export declare const abortExp: (name: AutocompleteExperienceName, id: string, reason?: string, metadata?: CustomData) => void;

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.2",
+	"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.4.0",
+		"@atlaskit/editor-common": "^116.15.0",
 		"@atlaskit/editor-plugin-analytics": "^12.0.0",
 		"react": "^18.2.0"
 	},

package/src/analytics/ufo.ts

@@ -120,12 +120,23 @@ export const failExp = (
 	}
 };
 
-export const abortExp = (name: AutocompleteExperienceName, id: string, reason?: string): void => {
+export const abortExp = (
+	name: AutocompleteExperienceName,
+	id: string,
+	reason?: string,
+	metadata?: CustomData,
+): void => {
 	if (!isUfoEnabled()) {
 		return;
 	}
 	try {
-		experiences[name].getInstance(id).abort(reason ? { metadata: { reason } } : undefined);
+		// `reason` is the dedicated, canonical channel for the abort reason and
+		// intentionally takes precedence over a `reason` key in `metadata`. Callers
+		// must pass the reason via the param, not inside `metadata` (the `reason`
+		// key there is reserved and would be overwritten).
+		const merged: CustomData = { ...metadata, ...(reason ? { reason } : {}) };
+		const hasMetadata = Object.keys(merged).length > 0;
+		experiences[name].getInstance(id).abort(hasMetadata ? { metadata: merged } : undefined);
 	} catch {
 		// UFO errors must never break the plugin
 	}

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);
+				},
+			};
+		},
 	});
 };