@automattic/jetpack-ai-client 0.33.32 → 0.34.0

package/CHANGELOG.md

@@ -5,6 +5,16 @@ 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.34.0] - 2026-02-10
+### Added
+- AI Client: Add explicit window interface with missing types. [#46938]
+
+### Changed
+- Update package dependencies. [#46905]
+
+### Removed
+- Removed Chrome AI built-in API integration code. [#46896]
+
 ## [0.33.32] - 2026-02-02
 ### Changed
 - Update package dependencies. [#46854]
@@ -790,6 +800,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.34.0]: https://github.com/Automattic/jetpack-ai-client/compare/v0.33.32...v0.34.0
 [0.33.32]: https://github.com/Automattic/jetpack-ai-client/compare/v0.33.31...v0.33.32
 [0.33.31]: https://github.com/Automattic/jetpack-ai-client/compare/v0.33.30...v0.33.31
 [0.33.30]: https://github.com/Automattic/jetpack-ai-client/compare/v0.33.29...v0.33.30

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

@@ -30,5 +30,4 @@ export default function useAiFeature(): {
         };
     };
     featuresControl?: import("@automattic/jetpack-shared-extension-utils/store/wordpress-com/types").FeaturesControl;
-    chromeAiTokens?: import("@automattic/jetpack-shared-extension-utils/store/wordpress-com/types").ChromeAiTokens;
 };

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

@@ -18,10 +18,10 @@ type useAiSuggestionsOptions = {
     askQuestionOptions?: AskQuestionOptionsArgProps;
     initialRequestingState?: RequestingStateProp;
     onSuggestion?: (suggestion: string) => void;
-    onDone?: (content: string, skipRequestCount?: boolean, modelUsed?: AiModelTypeProp) => void;
+    onDone?: (content: string, modelUsed?: AiModelTypeProp) => void;
     onStop?: () => void;
     onError?: (error: RequestingErrorProps) => void;
-    onAllErrors?: (error: RequestingErrorProps, skipRequestCount?: boolean) => void;
+    onAllErrors?: (error: RequestingErrorProps) => void;
 };
 type useAiSuggestionsProps = {
     suggestion: string;

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

@@ -8,8 +8,7 @@ import debugFactory from 'debug';
  * Internal dependencies
  */
 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, AI_MODEL_DEFAULT, AI_MODEL_GEMINI_NANO, } 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, } from "../../types.js";
 const debug = debugFactory('ai-client:use-ai-suggestions');
 /**
  * Get the error data for a given error code.
@@ -111,11 +110,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', modelRef.current);
+        onDone?.(fullSuggestion, modelRef.current);
         setRequestingState('done');
     }, [onDone]);
     const handleAnyError = useCallback((event) => {
-        onAllErrors?.(event?.detail, event?.detail?.source === 'chromeAI');
+        onAllErrors?.(event?.detail);
     }, [onAllErrors]);
     const handleError = useCallback((errorCode) => {
         eventSourceRef?.current?.close();
@@ -140,17 +139,8 @@ export default function useAiSuggestions({ prompt, autoRequest = false, askQuest
         setError(undefined);
         // Set the request status.
         setRequestingState('requesting');
-        // check if we can (or should) use Chrome AI
-        const chromeAI = await ChromeAIFactory(promptArg);
-        debug('chromeAI', chromeAI !== false);
-        if (chromeAI !== false) {
-            setModelAndRef(AI_MODEL_GEMINI_NANO);
-            eventSourceRef.current = chromeAI;
-        }
-        else {
-            setModelAndRef(AI_MODEL_DEFAULT);
-            eventSourceRef.current = await askQuestion(promptArg, options);
-        }
+        setModelAndRef(AI_MODEL_DEFAULT);
+        eventSourceRef.current = await askQuestion(promptArg, options);
         if (!eventSourceRef?.current) {
             debug('no event source');
             return;

package/build/index.d.ts

@@ -20,7 +20,3 @@ export * from './types.ts';
 export * from './libs/index.ts';
 export * from './constants.ts';
 export * from './logo-generator/index.ts';
-/**
- * Chrome AI
- */
-export * from './chrome-ai/index.ts';

package/build/index.js

@@ -47,7 +47,3 @@ export * from "./constants.js";
  * Logo Generator
  */
 export * from "./logo-generator/index.js";
-/**
- * Chrome AI
- */
-export * from "./chrome-ai/index.js";

package/build/types.d.ts

@@ -30,8 +30,7 @@ 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 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 AiModelTypeProp = typeof AI_MODEL_GPT_3_5_Turbo_16K | typeof AI_MODEL_GPT_4 | 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 +52,3 @@ export interface BlockEditorStore {
         [key in keyof typeof BlockEditorSelectors]: (typeof BlockEditorSelectors)[key];
     };
 }
-declare global {
-    interface Window {
-        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>;
-        };
-        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

@@ -34,4 +34,3 @@ 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/declarations.d.ts

@@ -0,0 +1,5 @@
+import type { ConnectionScriptData } from '@automattic/jetpack-connection';
+
+interface Window {
+	JP_CONNECTION_INITIAL_STATE: ConnectionScriptData;
+}

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@automattic/jetpack-ai-client",
-	"version": "0.33.32",
+	"version": "0.34.0",
 	"private": false,
 	"description": "A JS client for consuming Jetpack AI services",
 	"homepage": "https://github.com/Automattic/jetpack/tree/HEAD/projects/js-packages/ai-client/#readme",
@@ -34,11 +34,11 @@
 	},
 	"dependencies": {
 		"@automattic/jetpack-base-styles": "^1.0.15",
-		"@automattic/jetpack-components": "^1.4.9",
-		"@automattic/jetpack-connection": "^1.4.30",
+		"@automattic/jetpack-components": "^1.4.12",
+		"@automattic/jetpack-connection": "^1.4.33",
 		"@automattic/jetpack-explat": "workspace:*",
 		"@automattic/jetpack-script-data": "^0.5.4",
-		"@automattic/jetpack-shared-extension-utils": "^1.4.8",
+		"@automattic/jetpack-shared-extension-utils": "^1.4.11",
 		"@microsoft/fetch-event-source": "2.0.1",
 		"@types/jest": "30.0.0",
 		"@types/react": "18.3.26",
@@ -65,13 +65,13 @@
 		"turndown": "7.1.2"
 	},
 	"devDependencies": {
-		"@storybook/addon-docs": "10.1.10",
-		"@storybook/react": "10.1.10",
+		"@storybook/addon-docs": "10.2.3",
+		"@storybook/react": "10.2.3",
 		"@testing-library/dom": "10.4.1",
 		"@types/markdown-it": "14.1.2",
 		"@types/turndown": "5.0.6",
 		"jest": "30.2.0",
-		"storybook": "10.1.10",
+		"storybook": "10.2.3",
 		"typescript": "5.9.3"
 	}
 }

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

@@ -8,7 +8,6 @@ import debugFactory from 'debug';
  * Internal dependencies
  */
 import askQuestion from '../../ask-question/index.ts';
-import ChromeAIFactory from '../../chrome-ai/factory.ts';
 import {
 	ERROR_CONTEXT_TOO_LARGE,
 	ERROR_MODERATION,
@@ -18,7 +17,6 @@ import {
 	ERROR_UNCLEAR_PROMPT,
 	ERROR_RESPONSE,
 	AI_MODEL_DEFAULT,
-	AI_MODEL_GEMINI_NANO,
 } from '../../types.ts';
 /**
  * Types & constants
@@ -80,7 +78,7 @@ type useAiSuggestionsOptions = {
 	/*
 	 * onDone callback.
 	 */
-	onDone?: ( content: string, skipRequestCount?: boolean, modelUsed?: AiModelTypeProp ) => void;
+	onDone?: ( content: string, modelUsed?: AiModelTypeProp ) => void;
 
 	/*
 	 * onStop callback.
@@ -95,7 +93,7 @@ type useAiSuggestionsOptions = {
 	/*
 	 * Error callback common for all errors.
 	 */
-	onAllErrors?: ( error: RequestingErrorProps, skipRequestCount?: boolean ) => void;
+	onAllErrors?: ( error: RequestingErrorProps ) => void;
 };
 
 type useAiSuggestionsProps = {
@@ -281,7 +279,7 @@ export default function useAiSuggestions( {
 
 			const fullSuggestion = removeLlamaArtifact( event?.detail?.message ?? event?.detail );
 
-			onDone?.( fullSuggestion, event?.detail?.source === 'chromeAI', modelRef.current );
+			onDone?.( fullSuggestion, modelRef.current );
 			setRequestingState( 'done' );
 		},
 		[ onDone ]
@@ -289,7 +287,7 @@ export default function useAiSuggestions( {
 
 	const handleAnyError = useCallback(
 		( event: CustomEvent ) => {
-			onAllErrors?.( event?.detail, event?.detail?.source === 'chromeAI' );
+			onAllErrors?.( event?.detail );
 		},
 		[ onAllErrors ]
 	);
@@ -338,17 +336,8 @@ export default function useAiSuggestions( {
 			// Set the request status.
 			setRequestingState( 'requesting' );
 
-			// check if we can (or should) use Chrome AI
-			const chromeAI = await ChromeAIFactory( promptArg );
-			debug( 'chromeAI', chromeAI !== false );
-
-			if ( chromeAI !== false ) {
-				setModelAndRef( AI_MODEL_GEMINI_NANO );
-				eventSourceRef.current = chromeAI;
-			} else {
-				setModelAndRef( AI_MODEL_DEFAULT );
-				eventSourceRef.current = await askQuestion( promptArg, options );
-			}
+			setModelAndRef( AI_MODEL_DEFAULT );
+			eventSourceRef.current = await askQuestion( promptArg, options );
 
 			if ( ! eventSourceRef?.current ) {
 				debug( 'no event source' );

package/src/index.ts

@@ -55,8 +55,3 @@ export * from './constants.ts';
  * Logo Generator
  */
 export * from './logo-generator/index.ts';
-
-/**
- * Chrome AI
- */
-export * from './chrome-ai/index.ts';

package/src/types.ts

@@ -94,12 +94,9 @@ 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
-	| typeof AI_MODEL_GEMINI_NANO
 	| typeof AI_MODEL_DEFAULT;
 
 /*
@@ -138,46 +135,3 @@ export interface BlockEditorStore {
 		[ key in keyof typeof BlockEditorSelectors ]: ( typeof BlockEditorSelectors )[ key ];
 	};
 }
-
-declare global {
-	interface Window {
-		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 >;
-		};
-		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/chrome-ai/factory.d.ts

@@ -1,9 +0,0 @@
-import { PromptProp } from '../types.ts';
-import ChromeAISuggestionsEventSource from './suggestions.ts';
-/**
- * This will return an instance of ChromeAISuggestionsEventSource or false.
- *
- * @param promptArg - The messages array of the prompt.
- * @return ChromeAISuggestionsEventSource | bool
- */
-export default function ChromeAIFactory(promptArg: PromptProp): Promise<false | ChromeAISuggestionsEventSource>;

package/build/chrome-ai/factory.js

@@ -1,150 +0,0 @@
-import debugFactory from 'debug';
-import { PROMPT_TYPE_CHANGE_LANGUAGE, PROMPT_TYPE_SUMMARIZE } from "../constants.js";
-import { isChromeAIAvailable } from "./get-availability.js";
-import ChromeAISuggestionsEventSource from "./suggestions.js";
-const debug = debugFactory('ai-client:chrome-ai-factory');
-/**
- * This will return an instance of ChromeAISuggestionsEventSource or false.
- *
- * @param promptArg - The messages array of the prompt.
- * @return ChromeAISuggestionsEventSource | bool
- */
-export default async function ChromeAIFactory(promptArg) {
-    if (!(await isChromeAIAvailable())) {
-        debug('Chrome AI is not available');
-        return false;
-    }
-    const context = {
-        content: '',
-        language: '',
-    };
-    let promptType = '';
-    let tone = null;
-    let wordCount = null;
-    debug('promptArg', promptArg);
-    if (Array.isArray(promptArg)) {
-        for (let i = 0; i < promptArg.length; i++) {
-            const prompt = promptArg[i];
-            if (prompt.content) {
-                context.content = prompt.content;
-            }
-            if (!('context' in prompt)) {
-                continue;
-            }
-            const promptContext = prompt.context;
-            if (promptContext.type) {
-                promptType = promptContext.type;
-            }
-            if (promptContext.language) {
-                context.language = promptContext.language;
-            }
-            if (promptContext.content) {
-                context.content = promptContext.content;
-            }
-            if (promptContext.tone) {
-                tone = promptContext.tone;
-            }
-            if (promptContext.words) {
-                wordCount = promptContext.words;
-            }
-        }
-    }
-    debug('promptType', promptType);
-    // Early return if the prompt type is not supported.
-    if (!promptType.startsWith('ai-assistant-change-language') &&
-        !promptType.startsWith('ai-content-lens')) {
-        debug('promptType is not supported');
-        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) {
-        debug('LanguageDetector is not available');
-        return false;
-    }
-    const languageDetectorAvailability = await self.LanguageDetector.availability();
-    if (languageDetectorAvailability === 'unavailable') {
-        debug('LanguageDetector is unavailable');
-        return false;
-    }
-    const detector = await self.LanguageDetector.create();
-    if (languageDetectorAvailability !== 'available') {
-        debug('awaiting detector ready');
-        await detector.ready;
-    }
-    if (promptType.startsWith('ai-assistant-change-language')) {
-        const [language] = context.language.split(' ');
-        if (!('Translator' in self) ||
-            !self.Translator.create ||
-            !self.Translator.availability) {
-            debug('Translator is not available');
-            return false;
-        }
-        const languageOpts = {
-            sourceLanguage: 'en',
-            targetLanguage: language,
-        };
-        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;
-            }
-        }
-        debug('languageOpts', languageOpts);
-        const translationAvailability = await self.Translator.availability(languageOpts);
-        debug('translationAvailability', translationAvailability);
-        if (translationAvailability === 'unavailable') {
-            debug('Translator is unavailable');
-            return false;
-        }
-        const chromeAI = new ChromeAISuggestionsEventSource({
-            content: context.content,
-            promptType: PROMPT_TYPE_CHANGE_LANGUAGE,
-            options: languageOpts,
-        });
-        return chromeAI;
-    }
-    if (promptType.startsWith('ai-content-lens')) {
-        if (!('Summarizer' in self)) {
-            debug('Summarizer is not available');
-            return false;
-        }
-        if (context.language && context.language !== 'en (English)') {
-            debug('Summary is not English');
-            return false;
-        }
-        debug('awaiting detector detect');
-        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') {
-                debug('Confidence for non-English content');
-                return false;
-            }
-        }
-        const summaryOpts = {
-            tone: tone,
-            wordCount: wordCount,
-        };
-        debug('summaryOpts', summaryOpts);
-        const chromeAiEventSourceOpts = {
-            content: context.content,
-            promptType: PROMPT_TYPE_SUMMARIZE,
-            options: summaryOpts,
-        };
-        debug('chromeAiEventSourceOpts', chromeAiEventSourceOpts);
-        return new ChromeAISuggestionsEventSource(chromeAiEventSourceOpts);
-    }
-    return false;
-}

package/build/chrome-ai/get-availability.d.ts

@@ -1,7 +0,0 @@
-/**
- * Check if Chrome AI can be enabled.
- *
- * @return {boolean} Whether Chrome AI can be enabled.
- */
-export declare function isChromeAIAvailable(): Promise<boolean>;
-export default isChromeAIAvailable;

package/build/chrome-ai/get-availability.js

@@ -1,67 +0,0 @@
-/**
- * External dependencies
- */
-import { initializeExPlat, createExPlatClient } from '@automattic/jetpack-explat';
-import { select } from '@wordpress/data';
-import { addQueryArgs } from '@wordpress/url';
-import debugFactory from 'debug';
-/**
- * Internal dependencies
- */
-import apiFetch from "../api-fetch/index.js";
-const debug = debugFactory('ai-client:chrome-ai-availability');
-/**
- * Get the AI Assistant feature.
- *
- * @return {object} The AI Assistant feature.
- */
-function getAiAssistantFeature() {
-    const { getAiAssistantFeature: getFeature } = select('wordpress-com/plans');
-    return getFeature();
-}
-/**
- * Fetch an experiment assignment.
- *
- * @param {boolean} asConnectedUser - Whether the user is connected.
- * @return {Function} A function that fetches an experiment assignment.
- */
-const fetchExperimentAssignmentWithConnectedUser = async ({ experimentName, }) => {
-    const params = {
-        experiment_name: experimentName,
-        anon_id: undefined,
-        as_connected_user: true,
-    };
-    debug('params', params);
-    const assignmentsRequestUrl = addQueryArgs('https://public-api.wordpress.com/wpcom/v2/experiments/0.1.0/assignments/jetpack', params);
-    debug('assignmentsRequestUrl', assignmentsRequestUrl);
-    return apiFetch({
-        url: assignmentsRequestUrl,
-        credentials: 'include',
-        mode: 'cors',
-        global: true,
-    });
-};
-/**
- * Check if Chrome AI can be enabled.
- *
- * @return {boolean} Whether Chrome AI can be enabled.
- */
-export async function isChromeAIAvailable() {
-    const { featuresControl } = getAiAssistantFeature();
-    // Extra check if we want to control this via the feature flag for now
-    if (featuresControl?.['chrome-ai']?.enabled !== true) {
-        debug('feature is disabled for this site/user');
-        return false;
-    }
-    initializeExPlat();
-    const { loadExperimentAssignment: loadExperimentAssignmentWithAuth } = createExPlatClient({
-        fetchExperimentAssignment: fetchExperimentAssignmentWithConnectedUser,
-        getAnonId: async () => null,
-        logError: debug,
-        isDevelopmentMode: false,
-    });
-    const { variationName } = await loadExperimentAssignmentWithAuth('calypso_jetpack_ai_gemini_api_202503_v2');
-    debug('variationName', variationName);
-    return variationName === 'treatment';
-}
-export default isChromeAIAvailable;

package/build/chrome-ai/index.d.ts

@@ -1,2 +0,0 @@
-export { default as ChromeAIFactory } from './factory.ts';
-export { default as ChromeAISuggestionsEventSource } from './suggestions.ts';

package/build/chrome-ai/index.js

@@ -1,2 +0,0 @@
-export { default as ChromeAIFactory } from "./factory.js";
-export { default as ChromeAISuggestionsEventSource } from "./suggestions.js";

package/build/chrome-ai/suggestions.d.ts

@@ -1,38 +0,0 @@
-import { EventSourceMessage } from '@microsoft/fetch-event-source';
-import { AiModelTypeProp } from '../types.ts';
-type ChromeAISuggestionsEventSourceConstructorArgs = {
-    content: string;
-    promptType: string;
-    options?: {
-        postId?: number | string;
-        feature?: 'ai-assistant-experimental' | string | undefined;
-        sourceLanguage?: string;
-        targetLanguage?: string;
-        tone?: string;
-        wordCount?: number;
-        functions?: Array<object>;
-        model?: AiModelTypeProp;
-    };
-};
-type FunctionCallProps = {
-    name?: string;
-    arguments?: string;
-};
-export default class ChromeAISuggestionsEventSource extends EventTarget {
-    fullMessage: string;
-    fullFunctionCall: FunctionCallProps;
-    isPromptClear: boolean;
-    controller: AbortController;
-    errorUnclearPromptTriggered: boolean;
-    constructor(data: ChromeAISuggestionsEventSourceConstructorArgs);
-    initSource({ content, promptType, options, }: ChromeAISuggestionsEventSourceConstructorArgs): void;
-    initEventSource(): Promise<void>;
-    close(): void;
-    checkForUnclearPrompt(): void;
-    processEvent(e: EventSourceMessage): void;
-    processErrorEvent(e: any): void;
-    translate(text: string, target: string, source?: string): Promise<void>;
-    private getSummarizerOptions;
-    summarize(text: string, tone?: string, wordCount?: number): Promise<void>;
-}
-export {};

package/build/chrome-ai/suggestions.js

@@ -1,160 +0,0 @@
-import debugFactory from 'debug';
-import { PROMPT_TYPE_CHANGE_LANGUAGE, PROMPT_TYPE_SUMMARIZE } from "../constants.js";
-import { getErrorData } from "../hooks/use-ai-suggestions/index.js";
-import { renderHTMLFromMarkdown, renderMarkdownFromHTML } from "../libs/markdown/index.js";
-import { ERROR_RESPONSE, ERROR_NETWORK } from "../types.js";
-const debug = debugFactory('ai-client:chrome-ai-suggestions');
-export default class ChromeAISuggestionsEventSource extends EventTarget {
-    fullMessage;
-    fullFunctionCall;
-    isPromptClear;
-    controller;
-    errorUnclearPromptTriggered;
-    constructor(data) {
-        super();
-        this.fullMessage = '';
-        this.fullFunctionCall = {
-            name: '',
-            arguments: '',
-        };
-        this.isPromptClear = false;
-        this.controller = new AbortController();
-        this.initSource(data);
-    }
-    initSource({ content, promptType, options = {}, }) {
-        debug('initSource', content, promptType, options);
-        if (promptType === PROMPT_TYPE_CHANGE_LANGUAGE) {
-            this.translate(content, options.targetLanguage, options.sourceLanguage);
-        }
-        if (promptType === PROMPT_TYPE_SUMMARIZE) {
-            this.summarize(content, options.tone, options.wordCount);
-        }
-    }
-    async initEventSource() { }
-    close() { }
-    checkForUnclearPrompt() { }
-    processEvent(e) {
-        let data;
-        debug('processEvent', e);
-        try {
-            data = JSON.parse(e.data);
-        }
-        catch (err) {
-            this.processErrorEvent(err);
-            return;
-        }
-        if (e.event === 'translation' || e.event === 'summary') {
-            this.dispatchEvent(new CustomEvent('suggestion', { detail: data.message }));
-        }
-        if (data.complete) {
-            this.dispatchEvent(new CustomEvent('done', { detail: { message: data.message, source: 'chromeAI' } }));
-        }
-    }
-    processErrorEvent(e) {
-        debug('processErrorEvent', e);
-        // Dispatch a generic network error event
-        this.dispatchEvent(new CustomEvent(ERROR_NETWORK, { detail: e }));
-        this.dispatchEvent(new CustomEvent(ERROR_RESPONSE, {
-            detail: getErrorData(ERROR_NETWORK),
-        }));
-    }
-    // use the Chrome AI translator
-    async translate(text, target, source = '') {
-        if (!('Translator' in self)) {
-            return;
-        }
-        const translatorAvailability = await self.Translator.availability({
-            sourceLanguage: source,
-            targetLanguage: target,
-        });
-        if (translatorAvailability === 'unavailable') {
-            debug('awaiting translator ready');
-            this.processErrorEvent({
-                message: 'Translator is unavailable',
-            });
-            return;
-        }
-        const translator = await self.Translator.create({
-            sourceLanguage: source,
-            targetLanguage: target,
-        });
-        if (!translator) {
-            this.processErrorEvent({
-                message: 'Translator failed to initialize',
-            });
-            return;
-        }
-        try {
-            const translation = await translator.translate(renderHTMLFromMarkdown({ content: text }));
-            this.processEvent({
-                id: '',
-                event: 'translation',
-                data: JSON.stringify({
-                    message: renderMarkdownFromHTML({ content: translation }),
-                    complete: true,
-                }),
-            });
-        }
-        catch (error) {
-            this.processErrorEvent(error);
-        }
-    }
-    // Helper function to format summarizer options
-    getSummarizerOptions(tone, wordCount) {
-        let sharedContext = `The summary you write should contain strictly less than ${wordCount ?? 50} words. Strive for precision in word count without compromising clarity and significance`;
-        if (tone) {
-            sharedContext += `\n - Write with a ${tone} tone.\n`;
-        }
-        const options = {
-            sharedContext: sharedContext,
-            type: 'teaser',
-            format: 'plain-text',
-            length: 'medium',
-        };
-        return options;
-    }
-    // use the Chrome AI summarizer
-    async summarize(text, tone, wordCount) {
-        debug('summarize', text, tone, wordCount);
-        if (!('Summarizer' in self)) {
-            return;
-        }
-        const availability = await self.Summarizer.availability();
-        if (availability === 'unavailable') {
-            this.processErrorEvent({
-                data: { message: 'Summarizer is unavailable' },
-            });
-            return;
-        }
-        const summarizerOptions = this.getSummarizerOptions(tone, wordCount);
-        const summarizer = await self.Summarizer.create(summarizerOptions);
-        if (availability !== 'available') {
-            debug('awaiting summarizer ready');
-            await summarizer.ready;
-        }
-        try {
-            const context = `Write with a ${tone} tone.`;
-            debug('context', context);
-            let summary = await summarizer.summarize(text, { context: context });
-            debug('summary', summary);
-            wordCount = wordCount ?? 50;
-            // gemini-nano has a tendency to exceed the word count, so we need to check and summarize again if necessary
-            if (summary.split(' ').length > wordCount) {
-                debug('summary exceeds word count');
-                summary = await summarizer.summarize(summary, { context: context });
-            }
-            this.processEvent({
-                id: '',
-                event: 'summary',
-                data: JSON.stringify({
-                    message: summary,
-                    complete: true,
-                }),
-            });
-        }
-        catch (error) {
-            debug('error', error);
-            this.processErrorEvent(error);
-        }
-    }
-}

package/src/chrome-ai/factory.ts

@@ -1,199 +0,0 @@
-import debugFactory from 'debug';
-import { PROMPT_TYPE_CHANGE_LANGUAGE, PROMPT_TYPE_SUMMARIZE } from '../constants.ts';
-import { PromptProp, PromptItemProps } from '../types.ts';
-import { isChromeAIAvailable } from './get-availability.ts';
-import ChromeAISuggestionsEventSource from './suggestions.ts';
-
-const debug = debugFactory( 'ai-client:chrome-ai-factory' );
-
-interface PromptContext {
-	type?: string;
-	content?: string;
-	language?: string;
-	tone?: string;
-	words?: number;
-}
-
-/**
- * This will return an instance of ChromeAISuggestionsEventSource or false.
- *
- * @param promptArg - The messages array of the prompt.
- * @return ChromeAISuggestionsEventSource | bool
- */
-export default async function ChromeAIFactory( promptArg: PromptProp ) {
-	if ( ! ( await isChromeAIAvailable() ) ) {
-		debug( 'Chrome AI is not available' );
-		return false;
-	}
-
-	const context = {
-		content: '',
-		language: '',
-	};
-
-	let promptType = '';
-	let tone = null;
-	let wordCount = null;
-
-	debug( 'promptArg', promptArg );
-	if ( Array.isArray( promptArg ) ) {
-		for ( let i = 0; i < promptArg.length; i++ ) {
-			const prompt: PromptItemProps = promptArg[ i ];
-			if ( prompt.content ) {
-				context.content = prompt.content;
-			}
-
-			if ( ! ( 'context' in prompt ) ) {
-				continue;
-			}
-
-			const promptContext: PromptContext = prompt.context;
-
-			if ( promptContext.type ) {
-				promptType = promptContext.type;
-			}
-
-			if ( promptContext.language ) {
-				context.language = promptContext.language;
-			}
-
-			if ( promptContext.content ) {
-				context.content = promptContext.content;
-			}
-
-			if ( promptContext.tone ) {
-				tone = promptContext.tone;
-			}
-
-			if ( promptContext.words ) {
-				wordCount = promptContext.words;
-			}
-		}
-	}
-
-	debug( 'promptType', promptType );
-	// Early return if the prompt type is not supported.
-	if (
-		! promptType.startsWith( 'ai-assistant-change-language' ) &&
-		! promptType.startsWith( 'ai-content-lens' )
-	) {
-		debug( 'promptType is not supported' );
-		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
-	) {
-		debug( 'LanguageDetector is not available' );
-		return false;
-	}
-
-	const languageDetectorAvailability = await self.LanguageDetector.availability();
-	if ( languageDetectorAvailability === 'unavailable' ) {
-		debug( 'LanguageDetector is unavailable' );
-		return false;
-	}
-
-	const detector = await self.LanguageDetector.create();
-	if ( languageDetectorAvailability !== 'available' ) {
-		debug( 'awaiting detector ready' );
-		await detector.ready;
-	}
-
-	if ( promptType.startsWith( 'ai-assistant-change-language' ) ) {
-		const [ language ] = context.language.split( ' ' );
-
-		if (
-			! ( 'Translator' in self ) ||
-			! self.Translator.create ||
-			! self.Translator.availability
-		) {
-			debug( 'Translator is not available' );
-			return false;
-		}
-
-		const languageOpts = {
-			sourceLanguage: 'en',
-			targetLanguage: language,
-		};
-
-		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;
-			}
-		}
-
-		debug( 'languageOpts', languageOpts );
-		const translationAvailability = await self.Translator.availability( languageOpts );
-
-		debug( 'translationAvailability', translationAvailability );
-		if ( translationAvailability === 'unavailable' ) {
-			debug( 'Translator is unavailable' );
-			return false;
-		}
-
-		const chromeAI = new ChromeAISuggestionsEventSource( {
-			content: context.content,
-			promptType: PROMPT_TYPE_CHANGE_LANGUAGE,
-			options: languageOpts,
-		} );
-
-		return chromeAI;
-	}
-
-	if ( promptType.startsWith( 'ai-content-lens' ) ) {
-		if ( ! ( 'Summarizer' in self ) ) {
-			debug( 'Summarizer is not available' );
-			return false;
-		}
-
-		if ( context.language && context.language !== 'en (English)' ) {
-			debug( 'Summary is not English' );
-			return false;
-		}
-
-		debug( 'awaiting detector detect' );
-		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' ) {
-				debug( 'Confidence for non-English content' );
-				return false;
-			}
-		}
-
-		const summaryOpts = {
-			tone: tone,
-			wordCount: wordCount,
-		};
-
-		debug( 'summaryOpts', summaryOpts );
-
-		const chromeAiEventSourceOpts = {
-			content: context.content,
-			promptType: PROMPT_TYPE_SUMMARIZE,
-			options: summaryOpts,
-		};
-
-		debug( 'chromeAiEventSourceOpts', chromeAiEventSourceOpts );
-		return new ChromeAISuggestionsEventSource( chromeAiEventSourceOpts );
-	}
-
-	return false;
-}

package/src/chrome-ai/get-availability.ts

@@ -1,105 +0,0 @@
-/**
- * External dependencies
- */
-import { initializeExPlat, createExPlatClient } from '@automattic/jetpack-explat';
-import { select } from '@wordpress/data';
-import { addQueryArgs } from '@wordpress/url';
-import debugFactory from 'debug';
-/**
- * Internal dependencies
- */
-import apiFetch from '../api-fetch/index.ts';
-
-/**
- * Types
- */
-type FeatureControl = {
-	enabled: boolean;
-};
-
-type PlansSelect = {
-	getAiAssistantFeature: () => {
-		currentTier?: { value: number };
-		featuresControl?: Record< string, FeatureControl >;
-	};
-};
-
-const debug = debugFactory( 'ai-client:chrome-ai-availability' );
-
-/**
- * Get the AI Assistant feature.
- *
- * @return {object} The AI Assistant feature.
- */
-function getAiAssistantFeature() {
-	const { getAiAssistantFeature: getFeature } = select( 'wordpress-com/plans' ) as PlansSelect;
-	return getFeature();
-}
-
-/**
- * Fetch an experiment assignment.
- *
- * @param {boolean} asConnectedUser - Whether the user is connected.
- * @return {Function} A function that fetches an experiment assignment.
- */
-const fetchExperimentAssignmentWithConnectedUser = async ( {
-	experimentName,
-}: {
-	experimentName: string;
-} ): Promise< unknown > => {
-	const params = {
-		experiment_name: experimentName,
-		anon_id: undefined,
-		as_connected_user: true,
-	};
-
-	debug( 'params', params );
-
-	const assignmentsRequestUrl = addQueryArgs(
-		'https://public-api.wordpress.com/wpcom/v2/experiments/0.1.0/assignments/jetpack',
-		params
-	);
-
-	debug( 'assignmentsRequestUrl', assignmentsRequestUrl );
-
-	return apiFetch( {
-		url: assignmentsRequestUrl,
-		credentials: 'include',
-		mode: 'cors',
-		global: true,
-	} );
-};
-
-/**
- * Check if Chrome AI can be enabled.
- *
- * @return {boolean} Whether Chrome AI can be enabled.
- */
-export async function isChromeAIAvailable() {
-	const { featuresControl } = getAiAssistantFeature();
-
-	// Extra check if we want to control this via the feature flag for now
-	if ( featuresControl?.[ 'chrome-ai' ]?.enabled !== true ) {
-		debug( 'feature is disabled for this site/user' );
-		return false;
-	}
-
-	initializeExPlat();
-
-	const { loadExperimentAssignment: loadExperimentAssignmentWithAuth } = createExPlatClient( {
-		fetchExperimentAssignment: fetchExperimentAssignmentWithConnectedUser,
-		getAnonId: async () => null,
-		logError: debug,
-		isDevelopmentMode: false,
-	} );
-
-	const { variationName } = await loadExperimentAssignmentWithAuth(
-		'calypso_jetpack_ai_gemini_api_202503_v2'
-	);
-
-	debug( 'variationName', variationName );
-
-	return variationName === 'treatment';
-}
-
-export default isChromeAIAvailable;

package/src/chrome-ai/index.ts

@@ -1,2 +0,0 @@
-export { default as ChromeAIFactory } from './factory.ts';
-export { default as ChromeAISuggestionsEventSource } from './suggestions.ts';

package/src/chrome-ai/suggestions.ts

@@ -1,235 +0,0 @@
-import { EventSourceMessage } from '@microsoft/fetch-event-source';
-import debugFactory from 'debug';
-import { PROMPT_TYPE_CHANGE_LANGUAGE, PROMPT_TYPE_SUMMARIZE } from '../constants.ts';
-import { getErrorData } from '../hooks/use-ai-suggestions/index.ts';
-import { renderHTMLFromMarkdown, renderMarkdownFromHTML } from '../libs/markdown/index.ts';
-import { AiModelTypeProp, ERROR_RESPONSE, ERROR_NETWORK } from '../types.ts';
-
-type ChromeAISuggestionsEventSourceConstructorArgs = {
-	content: string;
-	promptType: string;
-	options?: {
-		postId?: number | string;
-		feature?: 'ai-assistant-experimental' | string | undefined;
-
-		// translation
-		sourceLanguage?: string;
-		targetLanguage?: string;
-
-		// summarization
-		tone?: string;
-		wordCount?: number;
-
-		// not sure if we need these
-		functions?: Array< object >;
-		model?: AiModelTypeProp;
-	};
-};
-
-type ChromeAIEvent = {
-	type: string;
-	message: string;
-	complete?: boolean;
-};
-
-type FunctionCallProps = {
-	name?: string;
-	arguments?: string;
-};
-
-const debug = debugFactory( 'ai-client:chrome-ai-suggestions' );
-
-export default class ChromeAISuggestionsEventSource extends EventTarget {
-	fullMessage: string;
-	fullFunctionCall: FunctionCallProps;
-	isPromptClear: boolean;
-	controller: AbortController;
-
-	errorUnclearPromptTriggered: boolean;
-
-	constructor( data: ChromeAISuggestionsEventSourceConstructorArgs ) {
-		super();
-		this.fullMessage = '';
-		this.fullFunctionCall = {
-			name: '',
-			arguments: '',
-		};
-		this.isPromptClear = false;
-
-		this.controller = new AbortController();
-
-		this.initSource( data );
-	}
-
-	initSource( {
-		content,
-		promptType,
-		options = {},
-	}: ChromeAISuggestionsEventSourceConstructorArgs ) {
-		debug( 'initSource', content, promptType, options );
-		if ( promptType === PROMPT_TYPE_CHANGE_LANGUAGE ) {
-			this.translate( content, options.targetLanguage, options.sourceLanguage );
-		}
-
-		if ( promptType === PROMPT_TYPE_SUMMARIZE ) {
-			this.summarize( content, options.tone, options.wordCount );
-		}
-	}
-
-	async initEventSource() {}
-
-	close() {}
-
-	checkForUnclearPrompt() {}
-
-	processEvent( e: EventSourceMessage ) {
-		let data: ChromeAIEvent;
-		debug( 'processEvent', e );
-		try {
-			data = JSON.parse( e.data );
-		} catch ( err ) {
-			this.processErrorEvent( err );
-			return;
-		}
-
-		if ( e.event === 'translation' || e.event === 'summary' ) {
-			this.dispatchEvent( new CustomEvent( 'suggestion', { detail: data.message } ) );
-		}
-
-		if ( data.complete ) {
-			this.dispatchEvent(
-				new CustomEvent( 'done', { detail: { message: data.message, source: 'chromeAI' } } )
-			);
-		}
-	}
-
-	processErrorEvent( e ) {
-		debug( 'processErrorEvent', e );
-		// Dispatch a generic network error event
-		this.dispatchEvent( new CustomEvent( ERROR_NETWORK, { detail: e } ) );
-		this.dispatchEvent(
-			new CustomEvent( ERROR_RESPONSE, {
-				detail: getErrorData( ERROR_NETWORK ),
-			} )
-		);
-	}
-
-	// use the Chrome AI translator
-	async translate( text: string, target: string, source: string = '' ) {
-		if ( ! ( 'Translator' in self ) ) {
-			return;
-		}
-
-		const translatorAvailability = await self.Translator.availability( {
-			sourceLanguage: source,
-			targetLanguage: target,
-		} );
-
-		if ( translatorAvailability === 'unavailable' ) {
-			debug( 'awaiting translator ready' );
-			this.processErrorEvent( {
-				message: 'Translator is unavailable',
-			} );
-			return;
-		}
-
-		const translator = await self.Translator.create( {
-			sourceLanguage: source,
-			targetLanguage: target,
-		} );
-
-		if ( ! translator ) {
-			this.processErrorEvent( {
-				message: 'Translator failed to initialize',
-			} );
-			return;
-		}
-
-		try {
-			const translation = await translator.translate( renderHTMLFromMarkdown( { content: text } ) );
-
-			this.processEvent( {
-				id: '',
-				event: 'translation',
-				data: JSON.stringify( {
-					message: renderMarkdownFromHTML( { content: translation } ),
-					complete: true,
-				} ),
-			} );
-		} catch ( error ) {
-			this.processErrorEvent( error );
-		}
-	}
-
-	// Helper function to format summarizer options
-	private getSummarizerOptions( tone?: string, wordCount?: number ) {
-		let sharedContext = `The summary you write should contain strictly less than ${
-			wordCount ?? 50
-		} words. Strive for precision in word count without compromising clarity and significance`;
-
-		if ( tone ) {
-			sharedContext += `\n - Write with a ${ tone } tone.\n`;
-		}
-
-		const options = {
-			sharedContext: sharedContext,
-			type: 'teaser',
-			format: 'plain-text',
-			length: 'medium',
-		};
-
-		return options;
-	}
-
-	// use the Chrome AI summarizer
-	async summarize( text: string, tone?: string, wordCount?: number ) {
-		debug( 'summarize', text, tone, wordCount );
-		if ( ! ( 'Summarizer' in self ) ) {
-			return;
-		}
-
-		const availability = await self.Summarizer.availability();
-
-		if ( availability === 'unavailable' ) {
-			this.processErrorEvent( {
-				data: { message: 'Summarizer is unavailable' },
-			} );
-			return;
-		}
-
-		const summarizerOptions = this.getSummarizerOptions( tone, wordCount );
-
-		const summarizer = await self.Summarizer.create( summarizerOptions );
-
-		if ( availability !== 'available' ) {
-			debug( 'awaiting summarizer ready' );
-			await summarizer.ready;
-		}
-
-		try {
-			const context = `Write with a ${ tone } tone.`;
-			debug( 'context', context );
-			let summary = await summarizer.summarize( text, { context: context } );
-			debug( 'summary', summary );
-			wordCount = wordCount ?? 50;
-
-			// gemini-nano has a tendency to exceed the word count, so we need to check and summarize again if necessary
-			if ( summary.split( ' ' ).length > wordCount ) {
-				debug( 'summary exceeds word count' );
-				summary = await summarizer.summarize( summary, { context: context } );
-			}
-
-			this.processEvent( {
-				id: '',
-				event: 'summary',
-				data: JSON.stringify( {
-					message: summary,
-					complete: true,
-				} ),
-			} );
-		} catch ( error ) {
-			debug( 'error', error );
-			this.processErrorEvent( error );
-		}
-	}
-}