@automattic/jetpack-ai-client 0.27.10 → 0.28.0

package/CHANGELOG.md

@@ -5,6 +5,10 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/)
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [0.28.0] - 2025-05-12
+### Changed
+- AI Assistant: Propagate the AI model used in the AI requests. [#43390]
+
 ## [0.27.10] - 2025-05-05
 ### Changed
 - Update package dependencies. [#43326]
@@ -604,6 +608,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 - AI Client: stop using smart document visibility handling on the fetchEventSource library, so it does not restart the completion when changing tabs. [#32004]
 - Updated package dependencies. [#31468] [#31659] [#31785]
 
+[0.28.0]: https://github.com/Automattic/jetpack-ai-client/compare/v0.27.10...v0.28.0
 [0.27.10]: https://github.com/Automattic/jetpack-ai-client/compare/v0.27.9...v0.27.10
 [0.27.9]: https://github.com/Automattic/jetpack-ai-client/compare/v0.27.8...v0.27.9
 [0.27.8]: https://github.com/Automattic/jetpack-ai-client/compare/v0.27.7...v0.27.8

package/build/chrome-ai/factory.js

@@ -53,34 +53,50 @@ export default async function ChromeAIFactory(promptArg) {
             }
         }
     }
+    // Early return if the prompt type is not supported.
+    if (!promptType.startsWith('ai-assistant-change-language') &&
+        !promptType.startsWith('ai-content-lens')) {
+        return false;
+    }
+    // If the languageDetector is not available, we can't use the translation or summary features—it's safer to fall back
+    // to the default AI model than to risk an unexpected error.
+    if (!('LanguageDetector' in self) ||
+        !self.LanguageDetector.create ||
+        !self.LanguageDetector.availability) {
+        return false;
+    }
+    const languageDetectorAvailability = await self.LanguageDetector.availability();
+    if (languageDetectorAvailability === 'unavailable') {
+        return false;
+    }
+    const detector = await self.LanguageDetector.create();
+    if (languageDetectorAvailability !== 'available') {
+        await detector.ready;
+    }
     if (promptType.startsWith('ai-assistant-change-language')) {
         const [language] = context.language.split(' ');
-        if (!('translation' in self) ||
-            !self.translation.createTranslator ||
-            !self.translation.canTranslate) {
+        if (!('Translator' in self) ||
+            !self.Translator.create ||
+            !self.Translator.availability) {
             return false;
         }
         const languageOpts = {
             sourceLanguage: 'en',
             targetLanguage: language,
         };
-        // see if we can detect the source language
-        if ('ai' in self && self.ai.languageDetector) {
-            const detector = await self.ai.languageDetector.create();
-            const confidences = await detector.detect(context.content);
-            for (const confidence of confidences) {
-                // 75% confidence is just a value that was picked. Generally
-                // 80% of higher is pretty safe, but the source language is
-                // required for the translator to work at all, which is also
-                // why en is the default language.
-                if (confidence.confidence > 0.75) {
-                    languageOpts.sourceLanguage = confidence.detectedLanguage;
-                    break;
-                }
+        const confidences = await detector.detect(context.content);
+        for (const confidence of confidences) {
+            // 75% confidence is just a value that was picked. Generally
+            // 80% of higher is pretty safe, but the source language is
+            // required for the translator to work at all, which is also
+            // why en is the default language.
+            if (confidence.confidence > 0.75) {
+                languageOpts.sourceLanguage = confidence.detectedLanguage;
+                break;
             }
         }
-        const canTranslate = await self.translation.canTranslate(languageOpts);
-        if (canTranslate === 'no') {
+        const translationAvailability = await self.Translator.availability(languageOpts);
+        if (translationAvailability === 'unavailable') {
             return false;
         }
         const chromeAI = new ChromeAISuggestionsEventSource({
@@ -90,13 +106,28 @@ export default async function ChromeAIFactory(promptArg) {
         });
         return chromeAI;
     }
-    // TODO: consider also using ChromeAI for ai-assistant-summarize
     if (promptType.startsWith('ai-content-lens')) {
+        if (!('Summarizer' in self)) {
+            return false;
+        }
+        if (context.language && context.language !== 'en (English)') {
+            return false;
+        }
+        const confidences = await detector.detect(context.content);
+        // if it doesn't look like the content is in English, we can't use the summary feature
+        for (const confidence of confidences) {
+            // 75% confidence is just a value that was picked. Generally
+            // 80% of higher is pretty safe, but the source language is
+            // required for the translator to work at all, which is also
+            // why en is the default language.
+            if (confidence.confidence > 0.75 && confidence.detectedLanguage !== 'en') {
+                return false;
+            }
+        }
         const summaryOpts = {
             tone: tone,
             wordCount: wordCount,
         };
-        // TODO: detect if the content is in English and fallback if it's not
         return new ChromeAISuggestionsEventSource({
             content: context.content,
             promptType: PROMPT_TYPE_SUMMARIZE,

package/build/chrome-ai/suggestions.js

@@ -55,10 +55,10 @@ export default class ChromeAISuggestionsEventSource extends EventTarget {
     }
     // use the Chrome AI translator
     async translate(text, target, source = '') {
-        if (!('translation' in self)) {
+        if (!('Translator' in self)) {
             return;
         }
-        const translator = await self.translation.createTranslator({
+        const translator = await self.Translator.create({
             sourceLanguage: source,
             targetLanguage: target,
         });
@@ -96,16 +96,18 @@ export default class ChromeAISuggestionsEventSource extends EventTarget {
     }
     // use the Chrome AI summarizer
     async summarize(text, tone, wordCount) {
-        if (!('ai' in self) || !('summarizer' in self.ai)) {
+        if (!('Summarizer' in self)) {
             return;
         }
-        const available = (await self.ai.summarizer.capabilities()).available;
-        if (available === 'no') {
+        // eslint-disable-next-line no-console
+        console.log('Summarizer is available');
+        const availability = await self.Summarizer.availability();
+        if (availability === 'unavailable') {
             return;
         }
-        const options = this.getSummarizerOptions(tone, wordCount);
-        const summarizer = await self.ai.summarizer.create(options);
-        if (available === 'after-download') {
+        const summarizerOptions = this.getSummarizerOptions(tone, wordCount);
+        const summarizer = await self.Summarizer.create(summarizerOptions);
+        if (availability !== 'available') {
             await summarizer.ready;
         }
         try {

package/build/hooks/use-ai-suggestions/index.d.ts

@@ -3,7 +3,7 @@
  */
 import type { AskQuestionOptionsArgProps } from '../../ask-question/index.ts';
 import type SuggestionsEventSource from '../../suggestions-event-source/index.ts';
-import type { PromptProp, SuggestionErrorCode, RequestingStateProp } from '../../types.ts';
+import type { PromptProp, SuggestionErrorCode, RequestingStateProp, AiModelTypeProp } from '../../types.ts';
 export type RequestingErrorProps = {
     code: SuggestionErrorCode;
     message: string;
@@ -18,13 +18,14 @@ type useAiSuggestionsOptions = {
     askQuestionOptions?: AskQuestionOptionsArgProps;
     initialRequestingState?: RequestingStateProp;
     onSuggestion?: (suggestion: string) => void;
-    onDone?: (content: string, skipRequestCount?: boolean) => void;
+    onDone?: (content: string, skipRequestCount?: boolean, modelUsed?: AiModelTypeProp) => void;
     onStop?: () => void;
     onError?: (error: RequestingErrorProps) => void;
-    onAllErrors?: (error: RequestingErrorProps) => void;
+    onAllErrors?: (error: RequestingErrorProps, skipRequestCount?: boolean) => void;
 };
 type useAiSuggestionsProps = {
     suggestion: string;
+    model: AiModelTypeProp;
     error: RequestingErrorProps | undefined;
     requestingState: RequestingStateProp;
     eventSource: SuggestionsEventSource | undefined;

package/build/hooks/use-ai-suggestions/index.js

@@ -8,7 +8,7 @@ import { __ } from '@wordpress/i18n';
  */
 import askQuestion from "../../ask-question/index.js";
 import ChromeAIFactory from "../../chrome-ai/factory.js";
-import { ERROR_CONTEXT_TOO_LARGE, ERROR_MODERATION, ERROR_NETWORK, ERROR_QUOTA_EXCEEDED, ERROR_SERVICE_UNAVAILABLE, ERROR_UNCLEAR_PROMPT, ERROR_RESPONSE, } from "../../types.js";
+import { ERROR_CONTEXT_TOO_LARGE, ERROR_MODERATION, ERROR_NETWORK, ERROR_QUOTA_EXCEEDED, ERROR_SERVICE_UNAVAILABLE, ERROR_UNCLEAR_PROMPT, ERROR_RESPONSE, AI_MODEL_DEFAULT, AI_MODEL_GEMINI_NANO, } from "../../types.js";
 /**
  * Get the error data for a given error code.
  *
@@ -75,6 +75,12 @@ export function removeLlamaArtifact(suggestion) {
 export default function useAiSuggestions({ prompt, autoRequest = false, askQuestionOptions = {}, initialRequestingState = 'init', onSuggestion, onDone, onStop, onError, onAllErrors, } = {}) {
     const [requestingState, setRequestingState] = useState(initialRequestingState);
     const [suggestion, setSuggestion] = useState('');
+    const [model, setModel] = useState(AI_MODEL_DEFAULT);
+    const modelRef = useRef(AI_MODEL_DEFAULT);
+    const setModelAndRef = (value) => {
+        setModel(value);
+        modelRef.current = value;
+    };
     const [error, setError] = useState();
     // Store the event source in a ref, so we can handle it if needed.
     const eventSourceRef = useRef(undefined);
@@ -101,11 +107,11 @@ export default function useAiSuggestions({ prompt, autoRequest = false, askQuest
     const handleDone = useCallback((event) => {
         closeEventSource();
         const fullSuggestion = removeLlamaArtifact(event?.detail?.message ?? event?.detail);
-        onDone?.(fullSuggestion, event?.detail?.source === 'chromeAI');
+        onDone?.(fullSuggestion, event?.detail?.source === 'chromeAI', modelRef.current);
         setRequestingState('done');
     }, [onDone]);
     const handleAnyError = useCallback((event) => {
-        onAllErrors?.(event?.detail);
+        onAllErrors?.(event?.detail, event?.detail?.source === 'chromeAI');
     }, [onAllErrors]);
     const handleError = useCallback((errorCode) => {
         eventSourceRef?.current?.close();
@@ -133,9 +139,11 @@ export default function useAiSuggestions({ prompt, autoRequest = false, askQuest
         // check if we can (or should) use Chrome AI
         const chromeAI = await ChromeAIFactory(promptArg);
         if (chromeAI !== false) {
+            setModelAndRef(AI_MODEL_GEMINI_NANO);
             eventSourceRef.current = chromeAI;
         }
         else {
+            setModelAndRef(AI_MODEL_DEFAULT);
             eventSourceRef.current = await askQuestion(promptArg, options);
         }
         if (!eventSourceRef?.current) {
@@ -231,6 +239,7 @@ export default function useAiSuggestions({ prompt, autoRequest = false, askQuest
     return {
         // Data
         suggestion,
+        model,
         error,
         requestingState,
         // Requests handlers

package/build/types.d.ts

@@ -29,7 +29,9 @@ export declare const REQUESTING_STATES: readonly ["init", "requesting", "suggest
 export type RequestingStateProp = (typeof REQUESTING_STATES)[number];
 export declare const AI_MODEL_GPT_3_5_Turbo_16K: "gpt-3.5-turbo-16k";
 export declare const AI_MODEL_GPT_4: "gpt-4";
-export type AiModelTypeProp = typeof AI_MODEL_GPT_3_5_Turbo_16K | typeof AI_MODEL_GPT_4;
+export declare const AI_MODEL_DEFAULT: "default";
+export declare const AI_MODEL_GEMINI_NANO: "gemini-nano";
+export type AiModelTypeProp = typeof AI_MODEL_GPT_3_5_Turbo_16K | typeof AI_MODEL_GPT_4 | typeof AI_MODEL_GEMINI_NANO | typeof AI_MODEL_DEFAULT;
 export type { RecordingState } from './hooks/use-media-recording/index.ts';
 export type CancelablePromise<T = void> = Promise<T> & {
     canceled?: boolean;
@@ -53,43 +55,41 @@ export interface BlockEditorStore {
 }
 declare global {
     interface Window {
-        translation?: {
-            canTranslate: (options: {
-                sourceLanguage: string;
-                targetLanguage: string;
-            }) => Promise<'no' | 'yes' | string>;
-            createTranslator: (options: {
+        LanguageDetector?: {
+            create: () => Promise<{
+                detect: (text: string) => Promise<{
+                    detectedLanguage: string;
+                    confidence: number;
+                }[]>;
+                ready: Promise<void>;
+            }>;
+            availability: () => Promise<'unavailable' | 'available' | 'downloadable' | 'downloading' | string>;
+        };
+        Translator?: {
+            create: (options: {
                 sourceLanguage: string;
                 targetLanguage: string;
             }) => Promise<{
                 translate: (text: string) => Promise<string>;
             }>;
+            availability: (options: {
+                sourceLanguage: string;
+                targetLanguage: string;
+            }) => Promise<'unavailable' | 'available' | 'downloadable' | 'downloading' | string>;
         };
-        ai?: {
-            languageDetector: {
-                create: () => Promise<{
-                    detect: (text: string) => Promise<{
-                        detectedLanguage: string;
-                        confidence: number;
-                    }[]>;
-                }>;
-            };
-            summarizer?: {
-                capabilities: () => Promise<{
-                    available: 'no' | 'yes' | 'after-download';
-                }>;
-                create: (options: {
-                    sharedContext?: string;
-                    type?: string;
-                    format?: string;
-                    length?: string;
-                }) => Promise<{
-                    ready: Promise<void>;
-                    summarize: (text: string, summarizeOptions?: {
-                        context?: string;
-                    }) => Promise<string>;
-                }>;
-            };
+        Summarizer?: {
+            availability: () => Promise<'unavailable' | 'available' | 'downloadable' | 'downloading' | string>;
+            create: (options: {
+                sharedContext?: string;
+                type?: string;
+                format?: string;
+                length?: string;
+            }) => Promise<{
+                ready: Promise<void>;
+                summarize: (text: string, summarizeOptions?: {
+                    context?: string;
+                }) => Promise<string>;
+            }>;
         };
     }
 }

package/build/types.js

@@ -33,3 +33,5 @@ export const REQUESTING_STATES = [
  */
 export const AI_MODEL_GPT_3_5_Turbo_16K = 'gpt-3.5-turbo-16k';
 export const AI_MODEL_GPT_4 = 'gpt-4';
+export const AI_MODEL_DEFAULT = 'default';
+export const AI_MODEL_GEMINI_NANO = 'gemini-nano';

package/package.json

@@ -1,7 +1,7 @@
 {
 	"private": false,
 	"name": "@automattic/jetpack-ai-client",
-	"version": "0.27.10",
+	"version": "0.28.0",
 	"description": "A JS client for consuming Jetpack AI services",
 	"homepage": "https://github.com/Automattic/jetpack/tree/HEAD/projects/js-packages/ai-client/#readme",
 	"bugs": {
@@ -45,10 +45,10 @@
 	"main": "./build/index.js",
 	"types": "./build/index.d.ts",
 	"dependencies": {
-		"@automattic/jetpack-base-styles": "^0.7.3",
-		"@automattic/jetpack-components": "^0.72.6",
-		"@automattic/jetpack-connection": "^0.39.13",
-		"@automattic/jetpack-shared-extension-utils": "^0.19.2",
+		"@automattic/jetpack-base-styles": "^0.7.4",
+		"@automattic/jetpack-components": "^0.73.0",
+		"@automattic/jetpack-connection": "^0.39.14",
+		"@automattic/jetpack-shared-extension-utils": "^0.19.3",
 		"@microsoft/fetch-event-source": "2.0.1",
 		"@types/jest": "29.5.14",
 		"@types/react": "18.3.18",

package/src/chrome-ai/factory.ts

@@ -75,13 +75,41 @@ export default async function ChromeAIFactory( promptArg: PromptProp ) {
 		}
 	}
 
+	// Early return if the prompt type is not supported.
+	if (
+		! promptType.startsWith( 'ai-assistant-change-language' ) &&
+		! promptType.startsWith( 'ai-content-lens' )
+	) {
+		return false;
+	}
+
+	// If the languageDetector is not available, we can't use the translation or summary features—it's safer to fall back
+	// to the default AI model than to risk an unexpected error.
+	if (
+		! ( 'LanguageDetector' in self ) ||
+		! self.LanguageDetector.create ||
+		! self.LanguageDetector.availability
+	) {
+		return false;
+	}
+
+	const languageDetectorAvailability = await self.LanguageDetector.availability();
+	if ( languageDetectorAvailability === 'unavailable' ) {
+		return false;
+	}
+
+	const detector = await self.LanguageDetector.create();
+	if ( languageDetectorAvailability !== 'available' ) {
+		await detector.ready;
+	}
+
 	if ( promptType.startsWith( 'ai-assistant-change-language' ) ) {
 		const [ language ] = context.language.split( ' ' );
 
 		if (
-			! ( 'translation' in self ) ||
-			! self.translation.createTranslator ||
-			! self.translation.canTranslate
+			! ( 'Translator' in self ) ||
+			! self.Translator.create ||
+			! self.Translator.availability
 		) {
 			return false;
 		}
@@ -91,26 +119,22 @@ export default async function ChromeAIFactory( promptArg: PromptProp ) {
 			targetLanguage: language,
 		};
 
-		// see if we can detect the source language
-		if ( 'ai' in self && self.ai.languageDetector ) {
-			const detector = await self.ai.languageDetector.create();
-			const confidences = await detector.detect( context.content );
-
-			for ( const confidence of confidences ) {
-				// 75% confidence is just a value that was picked. Generally
-				// 80% of higher is pretty safe, but the source language is
-				// required for the translator to work at all, which is also
-				// why en is the default language.
-				if ( confidence.confidence > 0.75 ) {
-					languageOpts.sourceLanguage = confidence.detectedLanguage;
-					break;
-				}
+		const confidences = await detector.detect( context.content );
+
+		for ( const confidence of confidences ) {
+			// 75% confidence is just a value that was picked. Generally
+			// 80% of higher is pretty safe, but the source language is
+			// required for the translator to work at all, which is also
+			// why en is the default language.
+			if ( confidence.confidence > 0.75 ) {
+				languageOpts.sourceLanguage = confidence.detectedLanguage;
+				break;
 			}
 		}
 
-		const canTranslate = await self.translation.canTranslate( languageOpts );
+		const translationAvailability = await self.Translator.availability( languageOpts );
 
-		if ( canTranslate === 'no' ) {
+		if ( translationAvailability === 'unavailable' ) {
 			return false;
 		}
 
@@ -123,14 +147,33 @@ export default async function ChromeAIFactory( promptArg: PromptProp ) {
 		return chromeAI;
 	}
 
-	// TODO: consider also using ChromeAI for ai-assistant-summarize
 	if ( promptType.startsWith( 'ai-content-lens' ) ) {
+		if ( ! ( 'Summarizer' in self ) ) {
+			return false;
+		}
+
+		if ( context.language && context.language !== 'en (English)' ) {
+			return false;
+		}
+
+		const confidences = await detector.detect( context.content );
+
+		// if it doesn't look like the content is in English, we can't use the summary feature
+		for ( const confidence of confidences ) {
+			// 75% confidence is just a value that was picked. Generally
+			// 80% of higher is pretty safe, but the source language is
+			// required for the translator to work at all, which is also
+			// why en is the default language.
+			if ( confidence.confidence > 0.75 && confidence.detectedLanguage !== 'en' ) {
+				return false;
+			}
+		}
+
 		const summaryOpts = {
 			tone: tone,
 			wordCount: wordCount,
 		};
 
-		// TODO: detect if the content is in English and fallback if it's not
 		return new ChromeAISuggestionsEventSource( {
 			content: context.content,
 			promptType: PROMPT_TYPE_SUMMARIZE,

package/src/chrome-ai/suggestions.ts

@@ -110,11 +110,11 @@ export default class ChromeAISuggestionsEventSource extends EventTarget {
 
 	// use the Chrome AI translator
 	async translate( text: string, target: string, source: string = '' ) {
-		if ( ! ( 'translation' in self ) ) {
+		if ( ! ( 'Translator' in self ) ) {
 			return;
 		}
 
-		const translator = await self.translation.createTranslator( {
+		const translator = await self.Translator.create( {
 			sourceLanguage: source,
 			targetLanguage: target,
 		} );
@@ -125,6 +125,7 @@ export default class ChromeAISuggestionsEventSource extends EventTarget {
 
 		try {
 			const translation = await translator.translate( renderHTMLFromMarkdown( { content: text } ) );
+
 			this.processEvent( {
 				id: '',
 				event: 'translation',
@@ -160,20 +161,22 @@ export default class ChromeAISuggestionsEventSource extends EventTarget {
 
 	// use the Chrome AI summarizer
 	async summarize( text: string, tone?: string, wordCount?: number ) {
-		if ( ! ( 'ai' in self ) || ! ( 'summarizer' in self.ai ) ) {
+		if ( ! ( 'Summarizer' in self ) ) {
 			return;
 		}
-		const available = ( await self.ai.summarizer.capabilities() ).available;
+		// eslint-disable-next-line no-console
+		console.log( 'Summarizer is available' );
+		const availability = await self.Summarizer.availability();
 
-		if ( available === 'no' ) {
+		if ( availability === 'unavailable' ) {
 			return;
 		}
 
-		const options = this.getSummarizerOptions( tone, wordCount );
+		const summarizerOptions = this.getSummarizerOptions( tone, wordCount );
 
-		const summarizer = await self.ai.summarizer.create( options );
+		const summarizer = await self.Summarizer.create( summarizerOptions );
 
-		if ( available === 'after-download' ) {
+		if ( availability !== 'available' ) {
 			await summarizer.ready;
 		}
 

package/src/hooks/use-ai-suggestions/index.ts

@@ -16,13 +16,20 @@ import {
 	ERROR_SERVICE_UNAVAILABLE,
 	ERROR_UNCLEAR_PROMPT,
 	ERROR_RESPONSE,
+	AI_MODEL_DEFAULT,
+	AI_MODEL_GEMINI_NANO,
 } from '../../types.ts';
 /**
  * Types & constants
  */
 import type { AskQuestionOptionsArgProps } from '../../ask-question/index.ts';
 import type SuggestionsEventSource from '../../suggestions-event-source/index.ts';
-import type { PromptProp, SuggestionErrorCode, RequestingStateProp } from '../../types.ts';
+import type {
+	PromptProp,
+	SuggestionErrorCode,
+	RequestingStateProp,
+	AiModelTypeProp,
+} from '../../types.ts';
 
 export type RequestingErrorProps = {
 	/*
@@ -70,7 +77,7 @@ type useAiSuggestionsOptions = {
 	/*
 	 * onDone callback.
 	 */
-	onDone?: ( content: string, skipRequestCount?: boolean ) => void;
+	onDone?: ( content: string, skipRequestCount?: boolean, modelUsed?: AiModelTypeProp ) => void;
 
 	/*
 	 * onStop callback.
@@ -85,7 +92,7 @@ type useAiSuggestionsOptions = {
 	/*
 	 * Error callback common for all errors.
 	 */
-	onAllErrors?: ( error: RequestingErrorProps ) => void;
+	onAllErrors?: ( error: RequestingErrorProps, skipRequestCount?: boolean ) => void;
 };
 
 type useAiSuggestionsProps = {
@@ -94,6 +101,11 @@ type useAiSuggestionsProps = {
 	 */
 	suggestion: string;
 
+	/*
+	 * The model.
+	 */
+	model: AiModelTypeProp;
+
 	/*
 	 * The error.
 	 */
@@ -221,6 +233,12 @@ export default function useAiSuggestions( {
 	const [ requestingState, setRequestingState ] =
 		useState< RequestingStateProp >( initialRequestingState );
 	const [ suggestion, setSuggestion ] = useState< string >( '' );
+	const [ model, setModel ] = useState< AiModelTypeProp >( AI_MODEL_DEFAULT );
+	const modelRef = useRef< AiModelTypeProp >( AI_MODEL_DEFAULT );
+	const setModelAndRef = ( value: AiModelTypeProp ) => {
+		setModel( value );
+		modelRef.current = value;
+	};
 	const [ error, setError ] = useState< RequestingErrorProps >();
 
 	// Store the event source in a ref, so we can handle it if needed.
@@ -258,7 +276,7 @@ export default function useAiSuggestions( {
 
 			const fullSuggestion = removeLlamaArtifact( event?.detail?.message ?? event?.detail );
 
-			onDone?.( fullSuggestion, event?.detail?.source === 'chromeAI' );
+			onDone?.( fullSuggestion, event?.detail?.source === 'chromeAI', modelRef.current );
 			setRequestingState( 'done' );
 		},
 		[ onDone ]
@@ -266,7 +284,7 @@ export default function useAiSuggestions( {
 
 	const handleAnyError = useCallback(
 		( event: CustomEvent ) => {
-			onAllErrors?.( event?.detail );
+			onAllErrors?.( event?.detail, event?.detail?.source === 'chromeAI' );
 		},
 		[ onAllErrors ]
 	);
@@ -319,8 +337,10 @@ export default function useAiSuggestions( {
 			const chromeAI = await ChromeAIFactory( promptArg );
 
 			if ( chromeAI !== false ) {
+				setModelAndRef( AI_MODEL_GEMINI_NANO );
 				eventSourceRef.current = chromeAI;
 			} else {
+				setModelAndRef( AI_MODEL_DEFAULT );
 				eventSourceRef.current = await askQuestion( promptArg, options );
 			}
 
@@ -436,6 +456,7 @@ export default function useAiSuggestions( {
 	return {
 		// Data
 		suggestion,
+		model,
 		error,
 		requestingState,
 

package/src/types.ts

@@ -93,8 +93,14 @@ export type RequestingStateProp = ( typeof REQUESTING_STATES )[ number ];
  */
 export const AI_MODEL_GPT_3_5_Turbo_16K = 'gpt-3.5-turbo-16k' as const;
 export const AI_MODEL_GPT_4 = 'gpt-4' as const;
+export const AI_MODEL_DEFAULT = 'default' as const;
+export const AI_MODEL_GEMINI_NANO = 'gemini-nano' as const;
 
-export type AiModelTypeProp = typeof AI_MODEL_GPT_3_5_Turbo_16K | typeof AI_MODEL_GPT_4;
+export type AiModelTypeProp =
+	| typeof AI_MODEL_GPT_3_5_Turbo_16K
+	| typeof AI_MODEL_GPT_4
+	| typeof AI_MODEL_GEMINI_NANO
+	| typeof AI_MODEL_DEFAULT;
 
 /*
  * Media recording types
@@ -135,43 +141,43 @@ export interface BlockEditorStore {
 
 declare global {
 	interface Window {
-		translation?: {
-			canTranslate: ( options: {
+		LanguageDetector?: {
+			create: () => Promise< {
+				detect: ( text: string ) => Promise<
+					{
+						detectedLanguage: string;
+						confidence: number;
+					}[]
+				>;
+				ready: Promise< void >;
+			} >;
+			availability: () => Promise<
+				'unavailable' | 'available' | 'downloadable' | 'downloading' | string
+			>;
+		};
+		Translator?: {
+			create: ( options: {
 				sourceLanguage: string;
 				targetLanguage: string;
-			} ) => Promise< 'no' | 'yes' | string >;
-			createTranslator: ( options: {
+			} ) => Promise< { translate: ( text: string ) => Promise< string > } >;
+			availability: ( options: {
 				sourceLanguage: string;
 				targetLanguage: string;
+			} ) => Promise< 'unavailable' | 'available' | 'downloadable' | 'downloading' | string >;
+		};
+		Summarizer?: {
+			availability: () => Promise<
+				'unavailable' | 'available' | 'downloadable' | 'downloading' | string
+			>;
+			create: ( options: {
+				sharedContext?: string;
+				type?: string;
+				format?: string;
+				length?: string;
 			} ) => Promise< {
-				translate: ( text: string ) => Promise< string >;
+				ready: Promise< void >;
+				summarize: ( text: string, summarizeOptions?: { context?: string } ) => Promise< string >;
 			} >;
 		};
-		ai?: {
-			languageDetector: {
-				create: () => Promise< {
-					detect: ( text: string ) => Promise<
-						{
-							detectedLanguage: string;
-							confidence: number;
-						}[]
-					>;
-				} >;
-			};
-			summarizer?: {
-				capabilities: () => Promise< {
-					available: 'no' | 'yes' | 'after-download';
-				} >;
-				create: ( options: {
-					sharedContext?: string;
-					type?: string;
-					format?: string;
-					length?: string;
-				} ) => Promise< {
-					ready: Promise< void >;
-					summarize: ( text: string, summarizeOptions?: { context?: string } ) => Promise< string >;
-				} >;
-			};
-		};
 	}
 }