@atlaskit/editor-plugin-autocomplete 3.6.0 → 3.6.2

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

@@ -4,12 +4,5 @@
   "module": "../../../dist/esm/entry-points/src-pm-plugins-autocomplete-plugin.js",
   "module:es2019": "../../../dist/es2019/entry-points/src-pm-plugins-autocomplete-plugin.js",
   "sideEffects": false,
-  "types": "../../../dist/types/entry-points/src-pm-plugins-autocomplete-plugin.d.ts",
-  "typesVersions": {
-    ">=4.5 <5.9": {
-      "*": [
-        "../../../dist/types-ts4.5/entry-points/src-pm-plugins-autocomplete-plugin.d.ts"
-      ]
-    }
-  }
+  "types": "../../../dist/types/entry-points/src-pm-plugins-autocomplete-plugin.d.ts"
 }

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

@@ -514,6 +514,19 @@ export const createAutocompletePlugin = (
 					return;
 				}
 
+				// Suppress suggestions when the cursor is mid-word — only offer
+				// completions when the cursor is at the trailing edge of a token.
+				// nodeAfter correctly handles inline atoms (mentions, emojis) where
+				// parentOffset and textContent indices diverge.
+				const nodeAfter = selection.$from.nodeAfter;
+				const charAfterCursor = nodeAfter?.isText ? nodeAfter.text?.[0] : 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).
+				if (charAfterCursor && /[\p{L}\p{N}_]/u.test(charAfterCursor)) {
+					return;
+				}
+
 				const textBefore = getTextBeforeCursor(state);
 
 				// Suppress re-showing the same suggestion the user just dismissed.
@@ -681,12 +694,15 @@ export const createAutocompletePlugin = (
 					return false;
 				},
 				focus: () => {
-					loadDefaultVocabulary().catch((error) => {
+					loadDefaultVocabulary({ isLocalLLM: options?.useLocalModel ?? false }).catch((error) => {
 						logException(error as Error, {
 							location: 'editor-plugin-autocomplete/loadDefaultVocabulary',
 						});
 					});
-					loadVectorsAsync({ getBinaryUrl: options?.getVectorsBinaryUrl }).catch((error) => {
+					loadVectorsAsync({
+						getBinaryUrl: options?.getVectorsBinaryUrl,
+						isLocalLLM: options?.useLocalModel ?? false,
+					}).catch((error) => {
 						logException(error as Error, {
 							location: 'editor-plugin-autocomplete/loadVectorsAsync',
 						});

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. */
@@ -367,16 +367,14 @@ 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 => {
+const unwrapJsonModule = <T>(mod: unknown, shape: 'object' | 'array'): T | null => {
 	if (mod == null || typeof mod !== 'object') {
 		return null;
 	}
 	const namespace = mod as Record<string, unknown> & { default?: unknown };
 
 	// Compute the named-export own-keys (strip synthetic markers).
-	const ownKeys = Object.keys(namespace).filter(
-		(k) => k !== 'default' && k !== '__esModule',
-	);
+	const ownKeys = Object.keys(namespace).filter((k) => k !== 'default' && k !== '__esModule');
 
 	// PREFER named exports when present — they always reflect the JSON's real
 	// top-level keys / indices, regardless of what `default` happens to be.
@@ -456,10 +454,7 @@ const loadBePayloadData = (): Promise<void> => {
 			}
 
 			firstTokenToWords = new Map(
-				Object.entries(firstTokenToWordsData).map(([tokenId, words]) => [
-					Number(tokenId),
-					words,
-				]),
+				Object.entries(firstTokenToWordsData).map(([tokenId, words]) => [Number(tokenId), words]),
 			);
 			l2Words = new Set(Object.keys(vocabularyData.words));
 			prefixMapTokenIds = Array.from(firstTokenToWords.keys());
@@ -519,7 +514,6 @@ export const computeBePayload = (
 	 */
 	validTokenIds: number[] = prefixMapTokenIds,
 ): Record<string, number> => {
-
 	// 1. Numerically-stable masked softmax over validTokenIds only.
 	let maxLogit = -Infinity;
 	for (const id of validTokenIds) {
@@ -773,7 +767,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') ||
@@ -1005,7 +1000,7 @@ 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,>(
+		const captureCompletionTime = <T>(
 			promise: Promise<T>,
 			onResolved: (resolvedAt: number) => void,
 		): Promise<T> =>
@@ -1042,8 +1037,7 @@ export const createLocalSlowLaneClient = (
 						max_tokens: 1,
 						temperature: 0,
 						logprobs: false,
-					})
-					,
+					}),
 					(resolvedAt) => {
 						tLmDone = resolvedAt;
 					},
@@ -1052,8 +1046,7 @@ export const createLocalSlowLaneClient = (
 					engine.embeddings.create({
 						model: LOCAL_MLC_EMBEDDING_MODEL_ID,
 						input: arcticInput,
-					})
-					,
+					}),
 					(resolvedAt) => {
 						tEmbDone = resolvedAt;
 					},

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

@@ -4,12 +4,5 @@
   "module": "../../../dist/esm/entry-points/src-pm-plugins-slow-lane-client.js",
   "module:es2019": "../../../dist/es2019/entry-points/src-pm-plugins-slow-lane-client.js",
   "sideEffects": false,
-  "types": "../../../dist/types/entry-points/src-pm-plugins-slow-lane-client.d.ts",
-  "typesVersions": {
-    ">=4.5 <5.9": {
-      "*": [
-        "../../../dist/types-ts4.5/entry-points/src-pm-plugins-slow-lane-client.d.ts"
-      ]
-    }
-  }
+  "types": "../../../dist/types/entry-points/src-pm-plugins-slow-lane-client.d.ts"
 }

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

@@ -4,12 +4,5 @@
   "module": "../../../dist/esm/entry-points/src-pm-plugins-text-predictor.js",
   "module:es2019": "../../../dist/es2019/entry-points/src-pm-plugins-text-predictor.js",
   "sideEffects": false,
-  "types": "../../../dist/types/entry-points/src-pm-plugins-text-predictor.d.ts",
-  "typesVersions": {
-    ">=4.5 <5.9": {
-      "*": [
-        "../../../dist/types-ts4.5/entry-points/src-pm-plugins-text-predictor.d.ts"
-      ]
-    }
-  }
+  "types": "../../../dist/types/entry-points/src-pm-plugins-text-predictor.d.ts"
 }

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,15 @@ 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;

package/dist/types-ts4.5/analytics/ufo.d.ts

@@ -1,38 +0,0 @@
-/**
- * UFO experience tracking helpers for `@atlaskit/editor-plugin-autocomplete`.
- *
- * Pattern follows `packages/linking-platform/smart-card/src/state/analytics/ufoExperiences.ts`
- * (ConcurrentExperience keyed per request/instance) plus a try/catch safety wrap
- * inspired by `packages/editor/collab-provider/src/analytics/ufo.ts` so a UFO
- * runtime error can never break the autocomplete feature.
- *
- * Experience names must be registered in DataPortal's `task` attribute before
- * they can power FE Reliability SLOs:
- * https://hello.atlassian.net/wiki/spaces/AA6/pages/3961393753
- */
-import { type CustomData } from '@atlaskit/ufo';
-/**
- * Experience name strings are surfaced downstream by the UFO pipeline as
- * `platform.fe.<type>.<platform.component>.<name>` (e.g.
- * `platform.fe.operation.editor-plugin-autocomplete.slow-lane-fetch`), so
- * names here should be short, dot-free, and must not repeat the component
- * name. This matches the convention used by `@atlaskit/emoji`'s
- * `ufoExperiences` map (e.g. `'emoji-rendered'`).
- */
-/**
- * Only experiences with meaningful async work and a real success/failure
- * outcome belong on UFO (which measures latency and success rate for FE
- * Reliability SLOs). Per-keystroke suggestion lifecycle counters (viewed /
- * inserted / dismissed) are tracked via analytics-next instead — they would
- * otherwise emit zero-duration events on every word boundary.
- */
-export declare const EXPERIENCE_NAME: {
-    readonly SLOW_LANE_FETCH: "slow-lane-fetch";
-    readonly LOAD_VOCABULARY: "load-vocabulary";
-    readonly LOAD_VECTORS: "load-vectors";
-};
-export type AutocompleteExperienceName = (typeof EXPERIENCE_NAME)[keyof typeof EXPERIENCE_NAME];
-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;

package/dist/types-ts4.5/autocompletePlugin.d.ts

@@ -1,2 +0,0 @@
-import type { AutocompletePlugin } from './autocompletePluginType';
-export declare const autocompletePlugin: AutocompletePlugin;

package/dist/types-ts4.5/autocompletePluginType.d.ts

@@ -1,10 +0,0 @@
-import type { NextEditorPlugin, OptionalPlugin } from '@atlaskit/editor-common/types';
-import type { AnalyticsPlugin } from '@atlaskit/editor-plugin-analytics';
-import type { AutocompletePluginOptions, AutocompletePluginState } from './pm-plugins/autocomplete-plugin';
-export type AutocompletePlugin = NextEditorPlugin<'autocomplete', {
-    pluginConfiguration?: AutocompletePluginOptions | undefined;
-    sharedState: AutocompletePluginState | undefined;
-    dependencies: [
-        OptionalPlugin<AnalyticsPlugin>
-    ];
-}>;

package/dist/types-ts4.5/entry-points/autocompletePlugin.d.ts

@@ -1 +0,0 @@
-export { autocompletePlugin } from '../autocompletePlugin';

package/dist/types-ts4.5/entry-points/autocompletePluginType.d.ts

@@ -1 +0,0 @@
-export type { AutocompletePlugin } from '../autocompletePluginType';

package/dist/types-ts4.5/entry-points/src-pm-plugins-autocomplete-plugin.d.ts

@@ -1,2 +0,0 @@
-export { autocompletePluginKey, createAutocompletePlugin } from '../pm-plugins/autocomplete-plugin';
-export type { AutocompleteContext, AutocompletePluginOptions, AutocompletePluginState, } from '../pm-plugins/autocomplete-plugin';

package/dist/types-ts4.5/entry-points/src-pm-plugins-slow-lane-client.d.ts

@@ -1,2 +0,0 @@
-export { createSlowLaneClient, getStoredContextVector, getStoredLmLogits, isWordBoundary, setDefaultSlowLaneClient, } from '../pm-plugins/slow-lane-client';
-export type { SlowLaneClientConfig, TypeaheadEncodingsRequest, TypeaheadEncodingsResponse, } from '../pm-plugins/slow-lane-client';

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

@@ -1,2 +0,0 @@
-export { getLastPredictionDebug, getPredictorStatus, incrementSessionFreq, ingestDocumentPage, initL3Vocabulary, initVectors, initVocabulary, loadDefaultVocabulary, loadVectorsAsync, predict, } from '../pm-plugins/text-predictor';
-export type { TenantVocabulary, WeightedTerm } from '../pm-plugins/text-predictor';

package/dist/types-ts4.5/index.d.ts

@@ -1,2 +0,0 @@
-export { autocompletePlugin } from './autocompletePlugin';
-export type { AutocompletePlugin } from './autocompletePluginType';

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

@@ -1,49 +0,0 @@
-import { SafePlugin } from '@atlaskit/editor-common/safe-plugin';
-import type { ExtractInjectionAPI } from '@atlaskit/editor-common/types';
-import { PluginKey } from '@atlaskit/editor-prosemirror/state';
-import { DecorationSet } from '@atlaskit/editor-prosemirror/view';
-import type { AutocompletePlugin } from '../autocompletePluginType';
-export declare const autocompletePluginKey: PluginKey;
-export interface AutocompletePluginState {
-    /** The decoration set containing the ghost text widget */
-    decorationSet: DecorationSet;
-    /** The document position where the ghost text should appear */
-    ghostPosition: number;
-    /** The predicted ghost text to display */
-    ghostText: string;
-}
-/**
- * Context provided to the autocomplete plugin on first editor focus.
- * Text fields are selectively ingested to boost word-frequency scoring for
- * predictions, giving words already present in the document/thread an L1
- * priority boost.
- */
-export interface AutocompleteContext {
-    /** Full page content as a string (e.g. markdown). */
-    fullPageContent?: string;
-    /** The currently selected text on the page, if any. */
-    pageSelectionContent?: string;
-    /** Content of the parent comment when the editor is in reply or edit mode. */
-    parentCommentContent?: string;
-    /** Contents of sibling comments when the editor is in reply or edit mode. */
-    siblingCommentsContents?: string[];
-}
-export interface AutocompletePluginOptions {
-    /**
-     * Async function called once on first editor focus to retrieve context for
-     * word-frequency boosting. Called lazily so the preset can remain synchronous.
-     */
-    getContext?: () => Promise<AutocompleteContext | undefined>;
-    /**
-     * Async function that resolves to a URL for the word vectors binary file.
-     * When provided, this takes precedence over the bundled asset URL.
-     * Use this to serve vectors from a CDN or media service in production.
-     */
-    getVectorsBinaryUrl?: () => Promise<string>;
-    /**
-     * When true, uses on-device inference via WebGPU (MLC WebLLM) instead of
-     * the network-based slow-lane backend. Defaults to false (network client).
-     */
-    useLocalModel?: boolean;
-}
-export declare const createAutocompletePlugin: (options?: AutocompletePluginOptions, api?: ExtractInjectionAPI<AutocompletePlugin>) => SafePlugin<AutocompletePluginState>;

package/dist/types-ts4.5/pm-plugins/debug-mode.d.ts

@@ -1,27 +0,0 @@
-/**
- * Contextual Typeahead Completions (CTC) debug logging utility.
- *
- * Logs are silent by default. Enable in any environment (dev, staging, prod):
- *
- *   // Live, for the current session (no reload required):
- *   __atlCtcDebug__.enable()
- *
- *   // To disable:
- *   __atlCtcDebug__.disable()
- *
- *   // From initial page load (survives reload) — append to the URL:
- *   ?atlCtcDebug=1
- *
- * Storage-free by design: avoids browser-storage consent controls (BSC), which can
- * block uncategorized localStorage/sessionStorage/cookie writes in some products.
- */
-declare global {
-    interface Window {
-        __atlCtcDebug__?: {
-            enable: () => void;
-            disable: () => void;
-            isEnabled: () => boolean;
-        };
-    }
-}
-export declare const isAutocompleteDebugEnabled: () => boolean;

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

@@ -1,7 +0,0 @@
-import type { EditorState } from '@atlaskit/editor-prosemirror/state';
-import { DecorationSet } from '@atlaskit/editor-prosemirror/view';
-/**
- * Creates a DecorationSet containing a ghost text widget at the given position.
- * The ghost text is rendered as a styled <span> that appears after the cursor.
- */
-export declare const createGhostTextDecorationSet: (state: EditorState, position: number, text: string) => DecorationSet;