@automattic/jetpack-ai-client 0.31.2 → 0.32.1

package/CHANGELOG.md

@@ -5,6 +5,21 @@ 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.32.1] - 2025-06-24
+### Added
+- Improve error handling on Chrome AI events. [#44048]
+
+## [0.32.0] - 2025-06-23
+### Changed
+- AI Assistant: Add A/B test for Chrome AI API features. [#43690]
+- Scripts: Change imports for hosting checks. [#43972]
+- Update package dependencies. [#44020]
+
+### Fixed
+- Fix Chrome experiment assignment group. [#44018]
+- Fix async function call on Chrome experiment availability. [#44026]
+- Invert feature flag check. [#44011]
+
 ## [0.31.2] - 2025-06-16
 ### Changed
 - Update dependencies. [#43878]
@@ -635,6 +650,8 @@ 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.32.1]: https://github.com/Automattic/jetpack-ai-client/compare/v0.32.0...v0.32.1
+[0.32.0]: https://github.com/Automattic/jetpack-ai-client/compare/v0.31.2...v0.32.0
 [0.31.2]: https://github.com/Automattic/jetpack-ai-client/compare/v0.31.1...v0.31.2
 [0.31.1]: https://github.com/Automattic/jetpack-ai-client/compare/v0.31.0...v0.31.1
 [0.31.0]: https://github.com/Automattic/jetpack-ai-client/compare/v0.30.0...v0.31.0

package/build/chrome-ai/factory.js

@@ -1,6 +1,8 @@
+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.
  *
@@ -8,7 +10,8 @@ import ChromeAISuggestionsEventSource from "./suggestions.js";
  * @return ChromeAISuggestionsEventSource | bool
  */
 export default async function ChromeAIFactory(promptArg) {
-    if (!isChromeAIAvailable()) {
+    if (!(await isChromeAIAvailable())) {
+        debug('Chrome AI is not available');
         return false;
     }
     const context = {
@@ -18,6 +21,7 @@ export default async function ChromeAIFactory(promptArg) {
     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];
@@ -45,9 +49,11 @@ export default async function ChromeAIFactory(promptArg) {
             }
         }
     }
+    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
@@ -55,14 +61,17 @@ export default async function ChromeAIFactory(promptArg) {
     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')) {
@@ -70,6 +79,7 @@ export default async function ChromeAIFactory(promptArg) {
         if (!('Translator' in self) ||
             !self.Translator.create ||
             !self.Translator.availability) {
+            debug('Translator is not available');
             return false;
         }
         const languageOpts = {
@@ -87,8 +97,11 @@ export default async function ChromeAIFactory(promptArg) {
                 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({
@@ -100,11 +113,14 @@ export default async function ChromeAIFactory(promptArg) {
     }
     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) {
@@ -113,6 +129,7 @@ export default async function ChromeAIFactory(promptArg) {
             // 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;
             }
         }
@@ -120,11 +137,14 @@ export default async function ChromeAIFactory(promptArg) {
             tone: tone,
             wordCount: wordCount,
         };
-        return new ChromeAISuggestionsEventSource({
+        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

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

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

@@ -1,8 +1,10 @@
 /**
  * External dependencies
  */
-import { getJetpackExtensionAvailability } from '@automattic/jetpack-shared-extension-utils';
+import { initializeExPlat, loadExperimentAssignmentWithAuth } from '@automattic/jetpack-explat';
 import { select } from '@wordpress/data';
+import debugFactory from 'debug';
+const debug = debugFactory('ai-client:chrome-ai-availability');
 /**
  * Get the AI Assistant feature.
  *
@@ -17,9 +19,17 @@ function getAiAssistantFeature() {
  *
  * @return {boolean} Whether Chrome AI can be enabled.
  */
-export function isChromeAIAvailable() {
+export async function isChromeAIAvailable() {
     const { featuresControl } = getAiAssistantFeature();
-    return (featuresControl?.['chrome-ai']?.enabled !== false &&
-        getJetpackExtensionAvailability('ai-use-chrome-ai-sometimes').available !== false);
+    // 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();
+    debug('initialized explat');
+    const { variationName } = await loadExperimentAssignmentWithAuth('calypso_jetpack_ai_gemini_api_202503_v1');
+    debug('variationName', variationName);
+    return variationName === 'treatment';
 }
 export default isChromeAIAvailable;

package/build/chrome-ai/suggestions.js

@@ -1,7 +1,9 @@
+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;
@@ -20,6 +22,7 @@ export default class ChromeAISuggestionsEventSource extends EventTarget {
         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);
         }
@@ -32,6 +35,7 @@ export default class ChromeAISuggestionsEventSource extends EventTarget {
     checkForUnclearPrompt() { }
     processEvent(e) {
         let data;
+        debug('processEvent', e);
         try {
             data = JSON.parse(e.data);
         }
@@ -47,6 +51,7 @@ export default class ChromeAISuggestionsEventSource extends EventTarget {
         }
     }
     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, {
@@ -58,11 +63,25 @@ export default class ChromeAISuggestionsEventSource extends EventTarget {
         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 {
@@ -96,24 +115,32 @@ export default class ChromeAISuggestionsEventSource extends EventTarget {
     }
     // 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({
@@ -126,6 +153,7 @@ export default class ChromeAISuggestionsEventSource extends EventTarget {
             });
         }
         catch (error) {
+            debug('error', error);
             this.processErrorEvent(error);
         }
     }

package/build/components/ai-image/hooks/use-site-type.js

@@ -1,7 +1,7 @@
 /**
  * External dependencies
  */
-import { isAtomicSite, isSimpleSite } from '@automattic/jetpack-shared-extension-utils';
+import { isWoASite, isSimpleSite } from '@automattic/jetpack-script-data';
 import { useState } from '@wordpress/element';
 /**
  * Hook to get the type of site.
@@ -10,7 +10,7 @@ import { useState } from '@wordpress/element';
  */
 export default function useSiteType() {
     const getSiteType = () => {
-        if (isAtomicSite()) {
+        if (isWoASite()) {
             return 'atomic';
         }
         if (isSimpleSite()) {

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

@@ -2,7 +2,8 @@
  * External dependencies
  */
 import getRedirectUrl from '@automattic/jetpack-components/tools/jp-redirect';
-import { isAtomicSite, isSimpleSite, getSiteFragment, useAutosaveAndRedirect, } from '@automattic/jetpack-shared-extension-utils';
+import { isWpcomPlatformSite, isSimpleSite } from '@automattic/jetpack-script-data';
+import { getSiteFragment, useAutosaveAndRedirect, } from '@automattic/jetpack-shared-extension-utils';
 import useAiFeature from "../use-ai-feature/index.js";
 const getWPComRedirectToURL = () => {
     const searchParams = new URLSearchParams(window.location.search);
@@ -31,7 +32,7 @@ export default function useAICheckout() {
         site: getSiteFragment(),
         path: 'jetpack_ai_yearly',
     });
-    const checkoutUrl = isAtomicSite() || isSimpleSite() ? wpcomCheckoutUrl : jetpackCheckoutUrl;
+    const checkoutUrl = isWpcomPlatformSite() ? wpcomCheckoutUrl : jetpackCheckoutUrl;
     const { autosaveAndRedirect, isRedirecting } = useAutosaveAndRedirect(checkoutUrl);
     return {
         checkoutUrl,

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

@@ -3,12 +3,14 @@
  */
 import { useCallback, useEffect, useRef, useState } from '@wordpress/element';
 import { __ } from '@wordpress/i18n';
+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";
+const debug = debugFactory('ai-client:use-ai-suggestions');
 /**
  * Get the error data for a given error code.
  *
@@ -91,8 +93,10 @@ export default function useAiSuggestions({ prompt, autoRequest = false, askQuest
      * @return {void}
      */
     const handleSuggestion = useCallback((event) => {
+        debug('handleSuggestion', event);
         const partialSuggestion = removeLlamaArtifact(event?.detail);
         if (!partialSuggestion) {
+            debug('no partial suggestion');
             return;
         }
         setSuggestion(partialSuggestion);
@@ -138,6 +142,7 @@ export default function useAiSuggestions({ prompt, autoRequest = false, askQuest
         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;
@@ -147,6 +152,7 @@ export default function useAiSuggestions({ prompt, autoRequest = false, askQuest
             eventSourceRef.current = await askQuestion(promptArg, options);
         }
         if (!eventSourceRef?.current) {
+            debug('no event source');
             return;
         }
         // Alias

package/build/jwt/index.js

@@ -1,7 +1,7 @@
 /**
  * External dependencies
  */
-import { isSimpleSite } from '@automattic/jetpack-shared-extension-utils';
+import { isSimpleSite } from '@automattic/jetpack-script-data';
 import debugFactory from 'debug';
 /**
  * Internal dependencies

package/build/logo-generator/hooks/use-checkout.js

@@ -1,7 +1,8 @@
 /**
  * External dependencies
  */
-import { isAtomicSite, isSimpleSite, getSiteFragment, } from '@automattic/jetpack-shared-extension-utils';
+import { isWpcomPlatformSite } from '@automattic/jetpack-script-data';
+import { getSiteFragment } from '@automattic/jetpack-shared-extension-utils';
 import { useSelect } from '@wordpress/data';
 /**
  * Internal dependencies
@@ -15,7 +16,7 @@ export const useCheckout = () => {
             tierPlansEnabled: selectors.getAiAssistantFeature().tierPlansEnabled,
         };
     }, []);
-    const isJetpackSite = !isAtomicSite() && !isSimpleSite();
+    const isJetpackSite = !isWpcomPlatformSite();
     const redirectSource = isJetpackSite
         ? 'jetpack-ai-upgrade-url-for-jetpack-sites'
         : 'jetpack-ai-yearly-tier-upgrade-nudge';

package/package.json

@@ -1,7 +1,7 @@
 {
 	"private": false,
 	"name": "@automattic/jetpack-ai-client",
-	"version": "0.31.2",
+	"version": "0.32.1",
 	"description": "A JS client for consuming Jetpack AI services",
 	"homepage": "https://github.com/Automattic/jetpack/tree/HEAD/projects/js-packages/ai-client/#readme",
 	"bugs": {
@@ -31,8 +31,8 @@
 		"@storybook/react": "8.6.7",
 		"@types/markdown-it": "14.1.2",
 		"@types/turndown": "5.0.5",
-		"jest": "^29.6.2",
-		"jest-environment-jsdom": "29.7.0",
+		"jest": "30.0.0",
+		"jest-environment-jsdom": "30.0.0",
 		"storybook": "8.6.7",
 		"typescript": "5.8.3"
 	},
@@ -45,28 +45,30 @@
 	"main": "./build/index.js",
 	"types": "./build/index.d.ts",
 	"dependencies": {
-		"@automattic/jetpack-base-styles": "^1.0.0",
-		"@automattic/jetpack-components": "^1.1.2",
-		"@automattic/jetpack-connection": "^1.2.2",
-		"@automattic/jetpack-shared-extension-utils": "^1.1.0",
+		"@automattic/jetpack-base-styles": "^1.0.1",
+		"@automattic/jetpack-components": "^1.1.6",
+		"@automattic/jetpack-connection": "^1.2.6",
+		"@automattic/jetpack-script-data": "^0.4.4",
+		"@automattic/jetpack-explat": "workspace:*",
+		"@automattic/jetpack-shared-extension-utils": "^1.2.1",
 		"@microsoft/fetch-event-source": "2.0.1",
-		"@types/jest": "29.5.14",
+		"@types/jest": "30.0.0",
 		"@types/react": "18.3.23",
 		"@types/wordpress__block-editor": "11.5.16",
-		"@wordpress/api-fetch": "7.24.0",
-		"@wordpress/base-styles": "6.0.0",
-		"@wordpress/blob": "4.24.0",
-		"@wordpress/blocks": "14.13.0",
-		"@wordpress/block-editor": "14.19.0",
-		"@wordpress/components": "29.10.0",
-		"@wordpress/compose": "7.24.0",
-		"@wordpress/data": "10.24.0",
-		"@wordpress/editor": "14.24.0",
-		"@wordpress/element": "6.24.0",
-		"@wordpress/i18n": "5.24.0",
-		"@wordpress/icons": "10.24.0",
-		"@wordpress/primitives": "4.24.0",
-		"@wordpress/url": "4.24.0",
+		"@wordpress/api-fetch": "7.25.0",
+		"@wordpress/base-styles": "6.1.0",
+		"@wordpress/blob": "4.25.0",
+		"@wordpress/block-editor": "14.20.0",
+		"@wordpress/blocks": "14.14.0",
+		"@wordpress/components": "29.11.0",
+		"@wordpress/compose": "7.25.0",
+		"@wordpress/data": "10.25.0",
+		"@wordpress/editor": "14.25.0",
+		"@wordpress/element": "6.25.0",
+		"@wordpress/i18n": "5.25.0",
+		"@wordpress/icons": "10.25.0",
+		"@wordpress/primitives": "4.25.0",
+		"@wordpress/url": "4.25.0",
 		"clsx": "2.1.1",
 		"debug": "4.4.1",
 		"markdown-it": "14.1.0",

package/src/chrome-ai/factory.ts

@@ -1,8 +1,11 @@
+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;
@@ -18,7 +21,8 @@ interface PromptContext {
  * @return ChromeAISuggestionsEventSource | bool
  */
 export default async function ChromeAIFactory( promptArg: PromptProp ) {
-	if ( ! isChromeAIAvailable() ) {
+	if ( ! ( await isChromeAIAvailable() ) ) {
+		debug( 'Chrome AI is not available' );
 		return false;
 	}
 
@@ -31,6 +35,7 @@ export default async function ChromeAIFactory( promptArg: PromptProp ) {
 	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 ];
@@ -66,11 +71,13 @@ export default async function ChromeAIFactory( promptArg: PromptProp ) {
 		}
 	}
 
+	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;
 	}
 
@@ -81,16 +88,19 @@ export default async function ChromeAIFactory( promptArg: PromptProp ) {
 		! 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;
 	}
 
@@ -102,6 +112,7 @@ export default async function ChromeAIFactory( promptArg: PromptProp ) {
 			! self.Translator.create ||
 			! self.Translator.availability
 		) {
+			debug( 'Translator is not available' );
 			return false;
 		}
 
@@ -123,9 +134,12 @@ export default async function ChromeAIFactory( promptArg: PromptProp ) {
 			}
 		}
 
+		debug( 'languageOpts', languageOpts );
 		const translationAvailability = await self.Translator.availability( languageOpts );
 
+		debug( 'translationAvailability', translationAvailability );
 		if ( translationAvailability === 'unavailable' ) {
+			debug( 'Translator is unavailable' );
 			return false;
 		}
 
@@ -140,13 +154,16 @@ export default async function ChromeAIFactory( promptArg: PromptProp ) {
 
 	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
@@ -156,6 +173,7 @@ export default async function ChromeAIFactory( promptArg: PromptProp ) {
 			// 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;
 			}
 		}
@@ -165,11 +183,16 @@ export default async function ChromeAIFactory( promptArg: PromptProp ) {
 			wordCount: wordCount,
 		};
 
-		return new ChromeAISuggestionsEventSource( {
+		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,8 +1,11 @@
 /**
  * External dependencies
  */
-import { getJetpackExtensionAvailability } from '@automattic/jetpack-shared-extension-utils';
+import { initializeExPlat, loadExperimentAssignmentWithAuth } from '@automattic/jetpack-explat';
 import { select } from '@wordpress/data';
+import debugFactory from 'debug';
+
+const debug = debugFactory( 'ai-client:chrome-ai-availability' );
 
 /**
  * Types
@@ -33,12 +36,25 @@ function getAiAssistantFeature() {
  *
  * @return {boolean} Whether Chrome AI can be enabled.
  */
-export function isChromeAIAvailable() {
+export async function isChromeAIAvailable() {
 	const { featuresControl } = getAiAssistantFeature();
-	return (
-		featuresControl?.[ 'chrome-ai' ]?.enabled !== false &&
-		getJetpackExtensionAvailability( 'ai-use-chrome-ai-sometimes' ).available !== false
+
+	// 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();
+	debug( 'initialized explat' );
+
+	const { variationName } = await loadExperimentAssignmentWithAuth(
+		'calypso_jetpack_ai_gemini_api_202503_v1'
 	);
+
+	debug( 'variationName', variationName );
+
+	return variationName === 'treatment';
 }
 
 export default isChromeAIAvailable;

package/src/chrome-ai/suggestions.ts

@@ -1,4 +1,5 @@
 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';
@@ -36,6 +37,8 @@ type FunctionCallProps = {
 	arguments?: string;
 };
 
+const debug = debugFactory( 'ai-client:chrome-ai-suggestions' );
+
 export default class ChromeAISuggestionsEventSource extends EventTarget {
 	fullMessage: string;
 	fullFunctionCall: FunctionCallProps;
@@ -63,6 +66,7 @@ export default class ChromeAISuggestionsEventSource extends EventTarget {
 		promptType,
 		options = {},
 	}: ChromeAISuggestionsEventSourceConstructorArgs ) {
+		debug( 'initSource', content, promptType, options );
 		if ( promptType === PROMPT_TYPE_CHANGE_LANGUAGE ) {
 			this.translate( content, options.targetLanguage, options.sourceLanguage );
 		}
@@ -80,6 +84,7 @@ export default class ChromeAISuggestionsEventSource extends EventTarget {
 
 	processEvent( e: EventSourceMessage ) {
 		let data: ChromeAIEvent;
+		debug( 'processEvent', e );
 		try {
 			data = JSON.parse( e.data );
 		} catch ( err ) {
@@ -99,6 +104,7 @@ export default class ChromeAISuggestionsEventSource extends EventTarget {
 	}
 
 	processErrorEvent( e ) {
+		debug( 'processErrorEvent', e );
 		// Dispatch a generic network error event
 		this.dispatchEvent( new CustomEvent( ERROR_NETWORK, { detail: e } ) );
 		this.dispatchEvent(
@@ -114,12 +120,28 @@ export default class ChromeAISuggestionsEventSource extends EventTarget {
 			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;
 		}
 
@@ -161,6 +183,7 @@ export default class ChromeAISuggestionsEventSource extends EventTarget {
 
 	// use the Chrome AI summarizer
 	async summarize( text: string, tone?: string, wordCount?: number ) {
+		debug( 'summarize', text, tone, wordCount );
 		if ( ! ( 'Summarizer' in self ) ) {
 			return;
 		}
@@ -168,6 +191,9 @@ export default class ChromeAISuggestionsEventSource extends EventTarget {
 		const availability = await self.Summarizer.availability();
 
 		if ( availability === 'unavailable' ) {
+			this.processErrorEvent( {
+				data: { message: 'Summarizer is unavailable' },
+			} );
 			return;
 		}
 
@@ -176,17 +202,20 @@ export default class ChromeAISuggestionsEventSource extends EventTarget {
 		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 } );
 			}
 
@@ -199,6 +228,7 @@ export default class ChromeAISuggestionsEventSource extends EventTarget {
 				} ),
 			} );
 		} catch ( error ) {
+			debug( 'error', error );
 			this.processErrorEvent( error );
 		}
 	}

package/src/components/ai-image/hooks/use-site-type.ts

@@ -1,7 +1,7 @@
 /**
  * External dependencies
  */
-import { isAtomicSite, isSimpleSite } from '@automattic/jetpack-shared-extension-utils';
+import { isWoASite, isSimpleSite } from '@automattic/jetpack-script-data';
 import { useState } from '@wordpress/element';
 
 /**
@@ -11,7 +11,7 @@ import { useState } from '@wordpress/element';
  */
 export default function useSiteType() {
 	const getSiteType = () => {
-		if ( isAtomicSite() ) {
+		if ( isWoASite() ) {
 			return 'atomic';
 		}
 		if ( isSimpleSite() ) {

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

@@ -2,9 +2,8 @@
  * External dependencies
  */
 import getRedirectUrl from '@automattic/jetpack-components/tools/jp-redirect';
+import { isWpcomPlatformSite, isSimpleSite } from '@automattic/jetpack-script-data';
 import {
-	isAtomicSite,
-	isSimpleSite,
 	getSiteFragment,
 	useAutosaveAndRedirect,
 } from '@automattic/jetpack-shared-extension-utils';
@@ -53,7 +52,7 @@ export default function useAICheckout(): UseAICheckoutReturn {
 		path: 'jetpack_ai_yearly',
 	} );
 
-	const checkoutUrl = isAtomicSite() || isSimpleSite() ? wpcomCheckoutUrl : jetpackCheckoutUrl;
+	const checkoutUrl = isWpcomPlatformSite() ? wpcomCheckoutUrl : jetpackCheckoutUrl;
 
 	const { autosaveAndRedirect, isRedirecting } = useAutosaveAndRedirect( checkoutUrl );
 

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

@@ -3,6 +3,7 @@
  */
 import { useCallback, useEffect, useRef, useState } from '@wordpress/element';
 import { __ } from '@wordpress/i18n';
+import debugFactory from 'debug';
 /**
  * Internal dependencies
  */
@@ -31,6 +32,8 @@ import type {
 	AiModelTypeProp,
 } from '../../types.ts';
 
+const debug = debugFactory( 'ai-client:use-ai-suggestions' );
+
 export type RequestingErrorProps = {
 	/*
 	 * A string code to refer to the error.
@@ -252,9 +255,11 @@ export default function useAiSuggestions( {
 	 */
 	const handleSuggestion = useCallback(
 		( event: CustomEvent ) => {
+			debug( 'handleSuggestion', event );
 			const partialSuggestion = removeLlamaArtifact( event?.detail );
 
 			if ( ! partialSuggestion ) {
+				debug( 'no partial suggestion' );
 				return;
 			}
 
@@ -335,6 +340,7 @@ export default function useAiSuggestions( {
 
 			// 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 );
@@ -345,6 +351,7 @@ export default function useAiSuggestions( {
 			}
 
 			if ( ! eventSourceRef?.current ) {
+				debug( 'no event source' );
 				return;
 			}
 

package/src/jwt/index.ts

@@ -1,7 +1,7 @@
 /**
  * External dependencies
  */
-import { isSimpleSite } from '@automattic/jetpack-shared-extension-utils';
+import { isSimpleSite } from '@automattic/jetpack-script-data';
 import debugFactory from 'debug';
 /**
  * Internal dependencies

package/src/logo-generator/hooks/use-checkout.ts

@@ -1,11 +1,8 @@
 /**
  * External dependencies
  */
-import {
-	isAtomicSite,
-	isSimpleSite,
-	getSiteFragment,
-} from '@automattic/jetpack-shared-extension-utils';
+import { isWpcomPlatformSite } from '@automattic/jetpack-script-data';
+import { getSiteFragment } from '@automattic/jetpack-shared-extension-utils';
 import { useSelect } from '@wordpress/data';
 /**
  * Internal dependencies
@@ -25,7 +22,7 @@ export const useCheckout = () => {
 		};
 	}, [] );
 
-	const isJetpackSite = ! isAtomicSite() && ! isSimpleSite();
+	const isJetpackSite = ! isWpcomPlatformSite();
 	const redirectSource = isJetpackSite
 		? 'jetpack-ai-upgrade-url-for-jetpack-sites'
 		: 'jetpack-ai-yearly-tier-upgrade-nudge';