@atlaskit/editor-plugin-autocomplete 4.0.2 → 4.0.4

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

@@ -37,6 +37,8 @@
 
 import type { MLCEngine, InitProgressReport, AppConfig, LogitProcessor } from '@mlc-ai/web-llm';
 
+import { abortExp, EXPERIENCE_NAME, failExp, startExp, succeedExp } from '../analytics/ufo';
+
 import { isAutocompleteDebugEnabled } from './debug-mode';
 import { isWordBoundary } from './slow-lane-client';
 
@@ -367,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;
 	}
@@ -409,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.
@@ -987,6 +989,8 @@ export const createLocalSlowLaneClient = (
 			return;
 		}
 
+		const 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
@@ -1000,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
@@ -1025,6 +1030,17 @@ export const createLocalSlowLaneClient = (
 		}
 
 		try {
+			// 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,
+			});
+
 			const tStart = performance.now();
 			let tLmDone = 0;
 			let tEmbDone = 0;
@@ -1064,6 +1080,12 @@ export const createLocalSlowLaneClient = (
 
 			// Discard stale results
 			if (requestId < latestRequestId || destroyed) {
+				abortExp(
+					EXPERIENCE_NAME.SLOW_LANE_FETCH,
+					experienceId,
+					destroyed ? 'destroyed' : 'superseded',
+					{ isLocalLLM: true },
+				);
 				return;
 			}
 
@@ -1126,6 +1148,13 @@ export const createLocalSlowLaneClient = (
 				console.groupEnd();
 			}
 
+			succeedExp(EXPERIENCE_NAME.SLOW_LANE_FETCH, experienceId, {
+				textLength: text.length,
+				hasVector: storedContextVector !== null,
+				hasLmLogits: storedLmLogits !== null,
+				isLocalLLM: true,
+			});
+
 			onUpdate?.({
 				textLength: text.length,
 				hasVector: storedContextVector !== null,
@@ -1134,11 +1163,21 @@ export const createLocalSlowLaneClient = (
 		} catch (err) {
 			// Discard errors for stale requests or after teardown
 			if (requestId < latestRequestId || destroyed) {
+				abortExp(
+					EXPERIENCE_NAME.SLOW_LANE_FETCH,
+					experienceId,
+					destroyed ? 'destroyed' : 'superseded',
+					{ isLocalLLM: true },
+				);
 				return;
 			}
 
 			storedContextVector = null;
 			storedLmLogits = null;
+			failExp(EXPERIENCE_NAME.SLOW_LANE_FETCH, experienceId, {
+				errorType: 'inference',
+				isLocalLLM: true,
+			});
 			onUpdate?.({ textLength: text.length, hasVector: false, hasLmLogits: false });
 
 			const errorMsg = err instanceof Error ? err.message : String(err);

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

@@ -101,7 +101,10 @@ export const createSlowLaneClient = (
 			session_id: sessionId,
 		};
 
-		startExp(EXPERIENCE_NAME.SLOW_LANE_FETCH, requestId, { textLength: text.length });
+		startExp(EXPERIENCE_NAME.SLOW_LANE_FETCH, requestId, {
+			textLength: text.length,
+			isLocalLLM: false,
+		});
 
 		if (isAutocompleteDebugEnabled()) {
 			// eslint-disable-next-line no-console
@@ -131,6 +134,7 @@ export const createSlowLaneClient = (
 				failExp(EXPERIENCE_NAME.SLOW_LANE_FETCH, requestId, {
 					status: res.status,
 					errorType: 'http_error',
+					isLocalLLM: false,
 				});
 				if (isAutocompleteDebugEnabled()) {
 					// eslint-disable-next-line no-console
@@ -184,6 +188,7 @@ export const createSlowLaneClient = (
 				textLength: text.length,
 				hasVector: storedContextVector !== null,
 				hasLmLogits: storedLmLogits !== null,
+				isLocalLLM: false,
 			});
 
 			onUpdate?.({
@@ -195,7 +200,10 @@ export const createSlowLaneClient = (
 		} catch (e) {
 			storedContextVector = null;
 			storedLmLogits = null;
-			failExp(EXPERIENCE_NAME.SLOW_LANE_FETCH, requestId, { errorType: 'network' });
+			failExp(EXPERIENCE_NAME.SLOW_LANE_FETCH, requestId, {
+				errorType: 'network',
+				isLocalLLM: false,
+			});
 			if (isAutocompleteDebugEnabled()) {
 				// eslint-disable-next-line no-console
 				console.log(
@@ -219,7 +227,9 @@ export const createSlowLaneClient = (
 		debounceTimer = setTimeout(() => {
 			debounceTimer = null;
 			if (inflightRequestId !== null) {
-				abortExp(EXPERIENCE_NAME.SLOW_LANE_FETCH, inflightRequestId, 'superseded');
+				abortExp(EXPERIENCE_NAME.SLOW_LANE_FETCH, inflightRequestId, 'superseded', {
+					isLocalLLM: false,
+				});
 			}
 			const requestId = String(++requestSeq);
 			inflightRequestId = requestId;

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>;