@atlaskit/editor-plugin-autocomplete 3.6.0 → 3.6.1

package/CHANGELOG.md

@@ -1,5 +1,16 @@
 # @atlaskit/editor-plugin-autocomplete
 
+## 3.6.1
+
+### Patch Changes
+
+- [`2d57d65205edd`](https://bitbucket.org/atlassian/atlassian-frontend-monorepo/commits/2d57d65205edd) -
+  [ASIMO] Add `isLocalLLM` attribute to `load-vectors` and `load-vocabulary` UFO experiences.
+
+  Both experiences now include an `isLocalLLM: boolean` attribute on every event (start, succeed,
+  fail) so analysts can segment loading performance and reliability by whether the user is on the
+  local on-device model setup vs the standard network-based slow-lane backend.
+
 ## 3.6.0
 
 ### Minor Changes

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

@@ -583,13 +583,17 @@ var createAutocompletePlugin = exports.createAutocompletePlugin = function creat
           return false;
         },
         focus: function focus() {
-          (0, _textPredictor.loadDefaultVocabulary)().catch(function (error) {
+          var _options$useLocalMode, _options$useLocalMode2;
+          (0, _textPredictor.loadDefaultVocabulary)({
+            isLocalLLM: (_options$useLocalMode = options === null || options === void 0 ? void 0 : options.useLocalModel) !== null && _options$useLocalMode !== void 0 ? _options$useLocalMode : false
+          }).catch(function (error) {
             (0, _monitoring.logException)(error, {
               location: 'editor-plugin-autocomplete/loadDefaultVocabulary'
             });
           });
           (0, _textPredictor.loadVectorsAsync)({
-            getBinaryUrl: options === null || options === void 0 ? void 0 : options.getVectorsBinaryUrl
+            getBinaryUrl: options === null || options === void 0 ? void 0 : options.getVectorsBinaryUrl,
+            isLocalLLM: (_options$useLocalMode2 = options === null || options === void 0 ? void 0 : options.useLocalModel) !== null && _options$useLocalMode2 !== void 0 ? _options$useLocalMode2 : false
           }).catch(function (error) {
             (0, _monitoring.logException)(error, {
               location: 'editor-plugin-autocomplete/loadVectorsAsync'

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

@@ -765,7 +765,9 @@ var createLocalSlowLaneClient = exports.createLocalSlowLaneClient = function cre
     if (lower.includes('loading chunk') || lower.includes('dynamically imported module') || lower.includes('dynamic import')) {
       return 'module_load_failed';
     }
-    if (lower.includes('out of memory') || /\boom\b/.test(lower) || lower.includes('allocation') || lower.includes('exceeds') || lower.includes('buffer size') || lower.includes('not enough memory')) {
+    if (lower.includes('out of memory') ||
+    // eslint-disable-next-line require-unicode-regexp
+    /\boom\b/.test(lower) || lower.includes('allocation') || lower.includes('exceeds') || lower.includes('buffer size') || lower.includes('not enough memory')) {
       return 'insufficient_memory';
     }
     // Pre-flight already returns missing_shader_f16 when the feature is absent,

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

@@ -733,7 +733,8 @@ var unwrapJsonModule = function unwrapJsonModule(mod, shape) {
 };
 var loadVectorsAsync = exports.loadVectorsAsync = /*#__PURE__*/function () {
   var _ref6 = (0, _asyncToGenerator2.default)( /*#__PURE__*/_regenerator.default.mark(function _callee(options) {
-    var url, _wordIndexOuter$index, res, buffer, float32, wordIndexModule, wordIndexOuter, wordIndex, nWords, dim, _t, _t2;
+    var _options$isLocalLLM;
+    var isLocalLLM, url, _wordIndexOuter$index, res, buffer, float32, wordIndexModule, wordIndexOuter, wordIndex, nWords, dim, _t, _t2;
     return _regenerator.default.wrap(function (_context) {
       while (1) switch (_context.prev = _context.next) {
         case 0:
@@ -751,8 +752,11 @@ var loadVectorsAsync = exports.loadVectorsAsync = /*#__PURE__*/function () {
           console.warn('[text-predictor] loadVectorsAsync called without a getBinaryUrl — vectors will not load. Pass getVectorsBinaryUrl via plugin options.');
           return _context.abrupt("return");
         case 2:
+          isLocalLLM = (_options$isLocalLLM = options === null || options === void 0 ? void 0 : options.isLocalLLM) !== null && _options$isLocalLLM !== void 0 ? _options$isLocalLLM : false;
           vectorsLoadStarted = true;
-          (0, _ufo.startExp)(_ufo.EXPERIENCE_NAME.LOAD_VECTORS, 'singleton');
+          (0, _ufo.startExp)(_ufo.EXPERIENCE_NAME.LOAD_VECTORS, 'singleton', {
+            isLocalLLM: isLocalLLM
+          });
           _context.prev = 3;
           _context.next = 4;
           return options.getBinaryUrl();
@@ -765,6 +769,7 @@ var loadVectorsAsync = exports.loadVectorsAsync = /*#__PURE__*/function () {
           _t = _context["catch"](3);
           vectorsLoadStarted = false;
           (0, _ufo.failExp)(_ufo.EXPERIENCE_NAME.LOAD_VECTORS, 'singleton', {
+            isLocalLLM: isLocalLLM,
             errorType: 'resolve_url'
           });
           // eslint-disable-next-line no-console
@@ -782,6 +787,7 @@ var loadVectorsAsync = exports.loadVectorsAsync = /*#__PURE__*/function () {
           }
           vectorsLoadStarted = false;
           (0, _ufo.failExp)(_ufo.EXPERIENCE_NAME.LOAD_VECTORS, 'singleton', {
+            isLocalLLM: isLocalLLM,
             status: res.status,
             errorType: 'http_error'
           });
@@ -816,6 +822,7 @@ var loadVectorsAsync = exports.loadVectorsAsync = /*#__PURE__*/function () {
             dim: dim
           };
           (0, _ufo.succeedExp)(_ufo.EXPERIENCE_NAME.LOAD_VECTORS, 'singleton', {
+            isLocalLLM: isLocalLLM,
             wordCount: nWords,
             dim: dim,
             sizeBytes: float32.byteLength
@@ -835,6 +842,7 @@ var loadVectorsAsync = exports.loadVectorsAsync = /*#__PURE__*/function () {
           _t2 = _context["catch"](6);
           vectorsLoadStarted = false;
           (0, _ufo.failExp)(_ufo.EXPERIENCE_NAME.LOAD_VECTORS, 'singleton', {
+            isLocalLLM: isLocalLLM,
             errorType: 'network'
           });
           // eslint-disable-next-line no-console
@@ -853,19 +861,23 @@ var initVectors = exports.initVectors = function initVectors(store) {
   vectorStore = store;
 };
 var vocabularyLoadPromise;
-var loadDefaultVocabulary = exports.loadDefaultVocabulary = function loadDefaultVocabulary() {
+var loadDefaultVocabulary = exports.loadDefaultVocabulary = function loadDefaultVocabulary(options) {
+  var _options$isLocalLLM2;
   if (isInitialized) {
     return Promise.resolve();
   }
   if (vocabularyLoadPromise) {
     return vocabularyLoadPromise;
   }
+  var isLocalLLM = (_options$isLocalLLM2 = options === null || options === void 0 ? void 0 : options.isLocalLLM) !== null && _options$isLocalLLM2 !== void 0 ? _options$isLocalLLM2 : false;
   vocabularyLoadPromise = (0, _asyncToGenerator2.default)( /*#__PURE__*/_regenerator.default.mark(function _callee2() {
     var _yield$Promise$all, _yield$Promise$all2, vocabularyModule, l3VocabularyModule, vocabularyData, l3VocabularyData, terms, _t3;
     return _regenerator.default.wrap(function (_context2) {
       while (1) switch (_context2.prev = _context2.next) {
         case 0:
-          (0, _ufo.startExp)(_ufo.EXPERIENCE_NAME.LOAD_VOCABULARY, 'singleton');
+          (0, _ufo.startExp)(_ufo.EXPERIENCE_NAME.LOAD_VOCABULARY, 'singleton', {
+            isLocalLLM: isLocalLLM
+          });
           _context2.prev = 1;
           _context2.next = 2;
           return Promise.all([Promise.resolve().then(function () {
@@ -904,6 +916,7 @@ var loadDefaultVocabulary = exports.loadDefaultVocabulary = function loadDefault
             terms: terms
           });
           (0, _ufo.succeedExp)(_ufo.EXPERIENCE_NAME.LOAD_VOCABULARY, 'singleton', {
+            isLocalLLM: isLocalLLM,
             l2WordCount: terms.length,
             l3WordCount: l3VocabularyData.length
           });
@@ -913,6 +926,7 @@ var loadDefaultVocabulary = exports.loadDefaultVocabulary = function loadDefault
           _context2.prev = 4;
           _t3 = _context2["catch"](1);
           (0, _ufo.failExp)(_ufo.EXPERIENCE_NAME.LOAD_VOCABULARY, 'singleton', {
+            isLocalLLM: isLocalLLM,
             errorType: 'parse_error'
           });
           // Allow a later call to retry the load rather than caching the failure.

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

@@ -570,13 +570,17 @@ export const createAutocompletePlugin = (options, api) => {
           return false;
         },
         focus: () => {
-          loadDefaultVocabulary().catch(error => {
+          var _options$useLocalMode, _options$useLocalMode2;
+          loadDefaultVocabulary({
+            isLocalLLM: (_options$useLocalMode = options === null || options === void 0 ? void 0 : options.useLocalModel) !== null && _options$useLocalMode !== void 0 ? _options$useLocalMode : false
+          }).catch(error => {
             logException(error, {
               location: 'editor-plugin-autocomplete/loadDefaultVocabulary'
             });
           });
           loadVectorsAsync({
-            getBinaryUrl: options === null || options === void 0 ? void 0 : options.getVectorsBinaryUrl
+            getBinaryUrl: options === null || options === void 0 ? void 0 : options.getVectorsBinaryUrl,
+            isLocalLLM: (_options$useLocalMode2 = options === null || options === void 0 ? void 0 : options.useLocalModel) !== null && _options$useLocalMode2 !== void 0 ? _options$useLocalMode2 : false
           }).catch(error => {
             logException(error, {
               location: 'editor-plugin-autocomplete/loadVectorsAsync'

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

@@ -586,7 +586,9 @@ export const createLocalSlowLaneClient = (config = {}) => {
     if (lower.includes('loading chunk') || lower.includes('dynamically imported module') || lower.includes('dynamic import')) {
       return 'module_load_failed';
     }
-    if (lower.includes('out of memory') || /\boom\b/u.test(lower) || lower.includes('allocation') || lower.includes('exceeds') || lower.includes('buffer size') || lower.includes('not enough memory')) {
+    if (lower.includes('out of memory') ||
+    // eslint-disable-next-line require-unicode-regexp
+    /\boom\b/.test(lower) || lower.includes('allocation') || lower.includes('exceeds') || lower.includes('buffer size') || lower.includes('not enough memory')) {
       return 'insufficient_memory';
     }
     // Pre-flight already returns missing_shader_f16 when the feature is absent,

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

@@ -587,6 +587,7 @@ const unwrapJsonModule = (mod, shape) => {
   return null;
 };
 export const loadVectorsAsync = async options => {
+  var _options$isLocalLLM;
   if (vectorStore || vectorsLoadStarted) {
     return;
   }
@@ -595,14 +596,18 @@ export const loadVectorsAsync = async options => {
     console.warn('[text-predictor] loadVectorsAsync called without a getBinaryUrl — vectors will not load. Pass getVectorsBinaryUrl via plugin options.');
     return;
   }
+  const isLocalLLM = (_options$isLocalLLM = options === null || options === void 0 ? void 0 : options.isLocalLLM) !== null && _options$isLocalLLM !== void 0 ? _options$isLocalLLM : false;
   vectorsLoadStarted = true;
-  startExp(EXPERIENCE_NAME.LOAD_VECTORS, 'singleton');
+  startExp(EXPERIENCE_NAME.LOAD_VECTORS, 'singleton', {
+    isLocalLLM
+  });
   let url;
   try {
     url = await options.getBinaryUrl();
   } catch (e) {
     vectorsLoadStarted = false;
     failExp(EXPERIENCE_NAME.LOAD_VECTORS, 'singleton', {
+      isLocalLLM,
       errorType: 'resolve_url'
     });
     // eslint-disable-next-line no-console
@@ -615,6 +620,7 @@ export const loadVectorsAsync = async options => {
     if (!res.ok) {
       vectorsLoadStarted = false;
       failExp(EXPERIENCE_NAME.LOAD_VECTORS, 'singleton', {
+        isLocalLLM,
         status: res.status,
         errorType: 'http_error'
       });
@@ -643,6 +649,7 @@ export const loadVectorsAsync = async options => {
       dim
     };
     succeedExp(EXPERIENCE_NAME.LOAD_VECTORS, 'singleton', {
+      isLocalLLM,
       wordCount: nWords,
       dim,
       sizeBytes: float32.byteLength
@@ -658,6 +665,7 @@ export const loadVectorsAsync = async options => {
   } catch (e) {
     vectorsLoadStarted = false;
     failExp(EXPERIENCE_NAME.LOAD_VECTORS, 'singleton', {
+      isLocalLLM,
       errorType: 'network'
     });
     // eslint-disable-next-line no-console
@@ -668,15 +676,19 @@ export const initVectors = store => {
   vectorStore = store;
 };
 let vocabularyLoadPromise;
-export const loadDefaultVocabulary = () => {
+export const loadDefaultVocabulary = options => {
+  var _options$isLocalLLM2;
   if (isInitialized) {
     return Promise.resolve();
   }
   if (vocabularyLoadPromise) {
     return vocabularyLoadPromise;
   }
+  const isLocalLLM = (_options$isLocalLLM2 = options === null || options === void 0 ? void 0 : options.isLocalLLM) !== null && _options$isLocalLLM2 !== void 0 ? _options$isLocalLLM2 : false;
   vocabularyLoadPromise = (async () => {
-    startExp(EXPERIENCE_NAME.LOAD_VOCABULARY, 'singleton');
+    startExp(EXPERIENCE_NAME.LOAD_VOCABULARY, 'singleton', {
+      isLocalLLM
+    });
     try {
       // The L2 vocabulary and L3 word list are code-split into their own async
       // chunks so they stay out of the editor's main bundle.
@@ -701,11 +713,13 @@ export const loadDefaultVocabulary = () => {
         terms
       });
       succeedExp(EXPERIENCE_NAME.LOAD_VOCABULARY, 'singleton', {
+        isLocalLLM,
         l2WordCount: terms.length,
         l3WordCount: l3VocabularyData.length
       });
     } catch (e) {
       failExp(EXPERIENCE_NAME.LOAD_VOCABULARY, 'singleton', {
+        isLocalLLM,
         errorType: 'parse_error'
       });
       // Allow a later call to retry the load rather than caching the failure.

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

@@ -576,13 +576,17 @@ export var createAutocompletePlugin = function createAutocompletePlugin(options,
           return false;
         },
         focus: function focus() {
-          loadDefaultVocabulary().catch(function (error) {
+          var _options$useLocalMode, _options$useLocalMode2;
+          loadDefaultVocabulary({
+            isLocalLLM: (_options$useLocalMode = options === null || options === void 0 ? void 0 : options.useLocalModel) !== null && _options$useLocalMode !== void 0 ? _options$useLocalMode : false
+          }).catch(function (error) {
             logException(error, {
               location: 'editor-plugin-autocomplete/loadDefaultVocabulary'
             });
           });
           loadVectorsAsync({
-            getBinaryUrl: options === null || options === void 0 ? void 0 : options.getVectorsBinaryUrl
+            getBinaryUrl: options === null || options === void 0 ? void 0 : options.getVectorsBinaryUrl,
+            isLocalLLM: (_options$useLocalMode2 = options === null || options === void 0 ? void 0 : options.useLocalModel) !== null && _options$useLocalMode2 !== void 0 ? _options$useLocalMode2 : false
           }).catch(function (error) {
             logException(error, {
               location: 'editor-plugin-autocomplete/loadVectorsAsync'

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

@@ -755,7 +755,9 @@ export var createLocalSlowLaneClient = function createLocalSlowLaneClient() {
     if (lower.includes('loading chunk') || lower.includes('dynamically imported module') || lower.includes('dynamic import')) {
       return 'module_load_failed';
     }
-    if (lower.includes('out of memory') || /\boom\b/.test(lower) || lower.includes('allocation') || lower.includes('exceeds') || lower.includes('buffer size') || lower.includes('not enough memory')) {
+    if (lower.includes('out of memory') ||
+    // eslint-disable-next-line require-unicode-regexp
+    /\boom\b/.test(lower) || lower.includes('allocation') || lower.includes('exceeds') || lower.includes('buffer size') || lower.includes('not enough memory')) {
       return 'insufficient_memory';
     }
     // Pre-flight already returns missing_shader_f16 when the feature is absent,

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

@@ -729,7 +729,8 @@ var unwrapJsonModule = function unwrapJsonModule(mod, shape) {
 };
 export var loadVectorsAsync = /*#__PURE__*/function () {
   var _ref6 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime.mark(function _callee(options) {
-    var url, _wordIndexOuter$index, res, buffer, float32, wordIndexModule, wordIndexOuter, wordIndex, nWords, dim, _t, _t2;
+    var _options$isLocalLLM;
+    var isLocalLLM, url, _wordIndexOuter$index, res, buffer, float32, wordIndexModule, wordIndexOuter, wordIndex, nWords, dim, _t, _t2;
     return _regeneratorRuntime.wrap(function (_context) {
       while (1) switch (_context.prev = _context.next) {
         case 0:
@@ -747,8 +748,11 @@ export var loadVectorsAsync = /*#__PURE__*/function () {
           console.warn('[text-predictor] loadVectorsAsync called without a getBinaryUrl — vectors will not load. Pass getVectorsBinaryUrl via plugin options.');
           return _context.abrupt("return");
         case 2:
+          isLocalLLM = (_options$isLocalLLM = options === null || options === void 0 ? void 0 : options.isLocalLLM) !== null && _options$isLocalLLM !== void 0 ? _options$isLocalLLM : false;
           vectorsLoadStarted = true;
-          startExp(EXPERIENCE_NAME.LOAD_VECTORS, 'singleton');
+          startExp(EXPERIENCE_NAME.LOAD_VECTORS, 'singleton', {
+            isLocalLLM: isLocalLLM
+          });
           _context.prev = 3;
           _context.next = 4;
           return options.getBinaryUrl();
@@ -761,6 +765,7 @@ export var loadVectorsAsync = /*#__PURE__*/function () {
           _t = _context["catch"](3);
           vectorsLoadStarted = false;
           failExp(EXPERIENCE_NAME.LOAD_VECTORS, 'singleton', {
+            isLocalLLM: isLocalLLM,
             errorType: 'resolve_url'
           });
           // eslint-disable-next-line no-console
@@ -778,6 +783,7 @@ export var loadVectorsAsync = /*#__PURE__*/function () {
           }
           vectorsLoadStarted = false;
           failExp(EXPERIENCE_NAME.LOAD_VECTORS, 'singleton', {
+            isLocalLLM: isLocalLLM,
             status: res.status,
             errorType: 'http_error'
           });
@@ -810,6 +816,7 @@ export var loadVectorsAsync = /*#__PURE__*/function () {
             dim: dim
           };
           succeedExp(EXPERIENCE_NAME.LOAD_VECTORS, 'singleton', {
+            isLocalLLM: isLocalLLM,
             wordCount: nWords,
             dim: dim,
             sizeBytes: float32.byteLength
@@ -829,6 +836,7 @@ export var loadVectorsAsync = /*#__PURE__*/function () {
           _t2 = _context["catch"](6);
           vectorsLoadStarted = false;
           failExp(EXPERIENCE_NAME.LOAD_VECTORS, 'singleton', {
+            isLocalLLM: isLocalLLM,
             errorType: 'network'
           });
           // eslint-disable-next-line no-console
@@ -847,19 +855,23 @@ export var initVectors = function initVectors(store) {
   vectorStore = store;
 };
 var vocabularyLoadPromise;
-export var loadDefaultVocabulary = function loadDefaultVocabulary() {
+export var loadDefaultVocabulary = function loadDefaultVocabulary(options) {
+  var _options$isLocalLLM2;
   if (isInitialized) {
     return Promise.resolve();
   }
   if (vocabularyLoadPromise) {
     return vocabularyLoadPromise;
   }
+  var isLocalLLM = (_options$isLocalLLM2 = options === null || options === void 0 ? void 0 : options.isLocalLLM) !== null && _options$isLocalLLM2 !== void 0 ? _options$isLocalLLM2 : false;
   vocabularyLoadPromise = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime.mark(function _callee2() {
     var _yield$Promise$all, _yield$Promise$all2, vocabularyModule, l3VocabularyModule, vocabularyData, l3VocabularyData, terms, _t3;
     return _regeneratorRuntime.wrap(function (_context2) {
       while (1) switch (_context2.prev = _context2.next) {
         case 0:
-          startExp(EXPERIENCE_NAME.LOAD_VOCABULARY, 'singleton');
+          startExp(EXPERIENCE_NAME.LOAD_VOCABULARY, 'singleton', {
+            isLocalLLM: isLocalLLM
+          });
           _context2.prev = 1;
           _context2.next = 2;
           return Promise.all([import( /* webpackChunkName: "@atlaskit-internal_editor-plugin-autocomplete-vocabulary-10k" */'./data/vocabulary_10k.json'), import( /* webpackChunkName: "@atlaskit-internal_editor-plugin-autocomplete-l3-vocabulary" */'./data/l3_vocabulary.json')]);
@@ -894,6 +906,7 @@ export var loadDefaultVocabulary = function loadDefaultVocabulary() {
             terms: terms
           });
           succeedExp(EXPERIENCE_NAME.LOAD_VOCABULARY, 'singleton', {
+            isLocalLLM: isLocalLLM,
             l2WordCount: terms.length,
             l3WordCount: l3VocabularyData.length
           });
@@ -903,6 +916,7 @@ export var loadDefaultVocabulary = function loadDefaultVocabulary() {
           _context2.prev = 4;
           _t3 = _context2["catch"](1);
           failExp(EXPERIENCE_NAME.LOAD_VOCABULARY, 'singleton', {
+            isLocalLLM: isLocalLLM,
             errorType: 'parse_error'
           });
           // Allow a later call to retry the load rather than caching the failure.

package/dist/types/pm-plugins/local-slow-lane-client.d.ts

@@ -96,10 +96,10 @@ export interface LocalSlowLaneClient {
 export type LocalSlowLaneLoadErrorReason = 'webgpu_unavailable' | 'webgpu_no_adapter' | 'missing_shader_f16' | 'insufficient_memory' | 'model_download_failed' | 'module_load_failed' | 'init_failed';
 /** Snapshot of the machine's WebGPU support, used to explain hardware limits. */
 export interface WebGpuCapabilities {
-    /** GPU architecture reported by the adapter (e.g. "metal-3", "rdna2"). */
-    architecture?: string;
     /** Whether `navigator.gpu.requestAdapter()` returned a usable adapter. */
     adapterAvailable?: boolean;
+    /** GPU architecture reported by the adapter (e.g. "metal-3", "rdna2"). */
+    architecture?: string;
     /** Whether `navigator.gpu` exists at all. */
     available: boolean;
     /** Largest single GPU buffer the adapter allows, in MB. */

package/dist/types/pm-plugins/text-predictor.d.ts

@@ -82,7 +82,10 @@ export declare const ingestDocumentPage: (pageContent: string | undefined) => vo
 export declare const predict: (textBefore: string) => string | null;
 export declare const loadVectorsAsync: (options?: {
     getBinaryUrl?: () => Promise<string>;
+    isLocalLLM?: boolean;
 }) => Promise<void>;
 export declare const initVectors: (store: VectorStore) => void;
-export declare const loadDefaultVocabulary: () => Promise<void>;
+export declare const loadDefaultVocabulary: (options?: {
+    isLocalLLM?: boolean;
+}) => Promise<void>;
 export {};

package/dist/types-ts4.5/pm-plugins/local-slow-lane-client.d.ts

@@ -96,10 +96,10 @@ export interface LocalSlowLaneClient {
 export type LocalSlowLaneLoadErrorReason = 'webgpu_unavailable' | 'webgpu_no_adapter' | 'missing_shader_f16' | 'insufficient_memory' | 'model_download_failed' | 'module_load_failed' | 'init_failed';
 /** Snapshot of the machine's WebGPU support, used to explain hardware limits. */
 export interface WebGpuCapabilities {
-    /** GPU architecture reported by the adapter (e.g. "metal-3", "rdna2"). */
-    architecture?: string;
     /** Whether `navigator.gpu.requestAdapter()` returned a usable adapter. */
     adapterAvailable?: boolean;
+    /** GPU architecture reported by the adapter (e.g. "metal-3", "rdna2"). */
+    architecture?: string;
     /** Whether `navigator.gpu` exists at all. */
     available: boolean;
     /** Largest single GPU buffer the adapter allows, in MB. */

package/dist/types-ts4.5/pm-plugins/text-predictor.d.ts

@@ -82,7 +82,10 @@ export declare const ingestDocumentPage: (pageContent: string | undefined) => vo
 export declare const predict: (textBefore: string) => string | null;
 export declare const loadVectorsAsync: (options?: {
     getBinaryUrl?: () => Promise<string>;
+    isLocalLLM?: boolean;
 }) => Promise<void>;
 export declare const initVectors: (store: VectorStore) => void;
-export declare const loadDefaultVocabulary: () => Promise<void>;
+export declare const loadDefaultVocabulary: (options?: {
+    isLocalLLM?: boolean;
+}) => Promise<void>;
 export {};

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@atlaskit/editor-plugin-autocomplete",
-	"version": "3.6.0",
+	"version": "3.6.1",
 	"description": "Client-side text autocomplete plugin for @atlaskit/editor-core",
 	"author": "Atlassian Pty Ltd",
 	"license": "Apache-2.0",

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

@@ -680,17 +680,20 @@ export const createAutocompletePlugin = (
 					}
 					return false;
 				},
-				focus: () => {
-					loadDefaultVocabulary().catch((error) => {
-						logException(error as Error, {
-							location: 'editor-plugin-autocomplete/loadDefaultVocabulary',
-						});
+			focus: () => {
+				loadDefaultVocabulary({ isLocalLLM: options?.useLocalModel ?? false }).catch((error) => {
+					logException(error as Error, {
+						location: 'editor-plugin-autocomplete/loadDefaultVocabulary',
 					});
-					loadVectorsAsync({ getBinaryUrl: options?.getVectorsBinaryUrl }).catch((error) => {
-						logException(error as Error, {
-							location: 'editor-plugin-autocomplete/loadVectorsAsync',
-						});
+				});
+				loadVectorsAsync({
+					getBinaryUrl: options?.getVectorsBinaryUrl,
+					isLocalLLM: options?.useLocalModel ?? false,
+				}).catch((error) => {
+					logException(error as Error, {
+						location: 'editor-plugin-autocomplete/loadVectorsAsync',
 					});
+				});
 					if (!hasIngestedPage) {
 						hasIngestedPage = true;
 						refreshContext({

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

@@ -114,10 +114,10 @@ export type LocalSlowLaneLoadErrorReason =
 
 /** Snapshot of the machine's WebGPU support, used to explain hardware limits. */
 export interface WebGpuCapabilities {
-	/** GPU architecture reported by the adapter (e.g. "metal-3", "rdna2"). */
-	architecture?: string;
 	/** Whether `navigator.gpu.requestAdapter()` returned a usable adapter. */
 	adapterAvailable?: boolean;
+	/** GPU architecture reported by the adapter (e.g. "metal-3", "rdna2"). */
+	architecture?: string;
 	/** Whether `navigator.gpu` exists at all. */
 	available: boolean;
 	/** Largest single GPU buffer the adapter allows, in MB. */
@@ -773,7 +773,8 @@ export const createLocalSlowLaneClient = (
 		}
 		if (
 			lower.includes('out of memory') ||
-			/\boom\b/u.test(lower) ||
+			// eslint-disable-next-line require-unicode-regexp
+			/\boom\b/.test(lower) ||
 			lower.includes('allocation') ||
 			lower.includes('exceeds') ||
 			lower.includes('buffer size') ||

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

@@ -747,6 +747,7 @@ const unwrapJsonModule = <T>(mod: unknown, shape: 'object' | 'array'): T | null
 
 export const loadVectorsAsync = async (options?: {
 	getBinaryUrl?: () => Promise<string>;
+	isLocalLLM?: boolean;
 }): Promise<void> => {
 	if (vectorStore || vectorsLoadStarted) {
 		return;
@@ -758,15 +759,16 @@ export const loadVectorsAsync = async (options?: {
 		);
 		return;
 	}
+	const isLocalLLM = options?.isLocalLLM ?? false;
 	vectorsLoadStarted = true;
-	startExp(EXPERIENCE_NAME.LOAD_VECTORS, 'singleton');
+	startExp(EXPERIENCE_NAME.LOAD_VECTORS, 'singleton', { isLocalLLM });
 
 	let url: string;
 	try {
 		url = await options.getBinaryUrl();
 	} catch (e) {
 		vectorsLoadStarted = false;
-		failExp(EXPERIENCE_NAME.LOAD_VECTORS, 'singleton', { errorType: 'resolve_url' });
+		failExp(EXPERIENCE_NAME.LOAD_VECTORS, 'singleton', { isLocalLLM, errorType: 'resolve_url' });
 		// eslint-disable-next-line no-console
 		console.warn('[text-predictor] Failed to resolve vectors URL:', e);
 		return;
@@ -777,6 +779,7 @@ export const loadVectorsAsync = async (options?: {
 		if (!res.ok) {
 			vectorsLoadStarted = false;
 			failExp(EXPERIENCE_NAME.LOAD_VECTORS, 'singleton', {
+				isLocalLLM,
 				status: res.status,
 				errorType: 'http_error',
 			});
@@ -809,6 +812,7 @@ export const loadVectorsAsync = async (options?: {
 
 		vectorStore = { float32, wordIndex, dim };
 		succeedExp(EXPERIENCE_NAME.LOAD_VECTORS, 'singleton', {
+			isLocalLLM,
 			wordCount: nWords,
 			dim,
 			sizeBytes: float32.byteLength,
@@ -823,7 +827,7 @@ export const loadVectorsAsync = async (options?: {
 		}
 	} catch (e) {
 		vectorsLoadStarted = false;
-		failExp(EXPERIENCE_NAME.LOAD_VECTORS, 'singleton', { errorType: 'network' });
+		failExp(EXPERIENCE_NAME.LOAD_VECTORS, 'singleton', { isLocalLLM, errorType: 'network' });
 		// eslint-disable-next-line no-console
 		console.warn('[text-predictor] Failed to load vectors:', e);
 	}
@@ -835,7 +839,7 @@ export const initVectors = (store: VectorStore): void => {
 
 let vocabularyLoadPromise: Promise<void> | undefined;
 
-export const loadDefaultVocabulary = (): Promise<void> => {
+export const loadDefaultVocabulary = (options?: { isLocalLLM?: boolean }): Promise<void> => {
 	if (isInitialized) {
 		return Promise.resolve();
 	}
@@ -843,8 +847,9 @@ export const loadDefaultVocabulary = (): Promise<void> => {
 		return vocabularyLoadPromise;
 	}
 
+	const isLocalLLM = options?.isLocalLLM ?? false;
 	vocabularyLoadPromise = (async () => {
-		startExp(EXPERIENCE_NAME.LOAD_VOCABULARY, 'singleton');
+		startExp(EXPERIENCE_NAME.LOAD_VOCABULARY, 'singleton', { isLocalLLM });
 
 		try {
 			// The L2 vocabulary and L3 word list are code-split into their own async
@@ -879,11 +884,12 @@ export const loadDefaultVocabulary = (): Promise<void> => {
 			initVocabulary({ terms });
 
 			succeedExp(EXPERIENCE_NAME.LOAD_VOCABULARY, 'singleton', {
+				isLocalLLM,
 				l2WordCount: terms.length,
 				l3WordCount: l3VocabularyData.length,
 			});
 		} catch (e) {
-			failExp(EXPERIENCE_NAME.LOAD_VOCABULARY, 'singleton', { errorType: 'parse_error' });
+			failExp(EXPERIENCE_NAME.LOAD_VOCABULARY, 'singleton', { isLocalLLM, errorType: 'parse_error' });
 			// Allow a later call to retry the load rather than caching the failure.
 			vocabularyLoadPromise = undefined;
 			throw e;