@automattic/jetpack-ai-client 0.28.0 → 0.29.0

package/CHANGELOG.md

@@ -5,6 +5,14 @@ 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.29.0] - 2025-05-19
+### Added
+- AI Assistant: Retrieve Chrome AI token from AI Features response and inject it from the frontend. [#43442]
+
+## [0.28.1] - 2025-05-15
+### Fixed
+- AI Assistant: Shorten AI excerpt if the built-in AI model doesn't respect the word count limit. [#43433]
+
 ## [0.28.0] - 2025-05-12
 ### Changed
 - AI Assistant: Propagate the AI model used in the AI requests. [#43390]
@@ -608,6 +616,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.29.0]: https://github.com/Automattic/jetpack-ai-client/compare/v0.28.1...v0.29.0
+[0.28.1]: https://github.com/Automattic/jetpack-ai-client/compare/v0.28.0...v0.28.1
 [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

package/build/chrome-ai/suggestions.js

@@ -82,7 +82,7 @@ export default class ChromeAISuggestionsEventSource extends EventTarget {
     }
     // Helper function to format summarizer options
     getSummarizerOptions(tone, wordCount) {
-        let sharedContext = `The summary you write should contain approximately ${wordCount ?? 50} words long. Strive for precision in word count without compromising clarity and significance`;
+        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`;
         }
@@ -99,8 +99,6 @@ export default class ChromeAISuggestionsEventSource extends EventTarget {
         if (!('Summarizer' in self)) {
             return;
         }
-        // eslint-disable-next-line no-console
-        console.log('Summarizer is available');
         const availability = await self.Summarizer.availability();
         if (availability === 'unavailable') {
             return;
@@ -112,7 +110,12 @@ export default class ChromeAISuggestionsEventSource extends EventTarget {
         }
         try {
             const context = `Write with a ${tone} tone.`;
-            const summary = await summarizer.summarize(text, { context: context });
+            let summary = await summarizer.summarize(text, { context: context });
+            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) {
+                summary = await summarizer.summarize(summary, { context: context });
+            }
             this.processEvent({
                 id: '',
                 event: 'summary',

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

@@ -30,4 +30,5 @@ 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-feature/index.js

@@ -3,7 +3,7 @@
  */
 import { PLAN_TYPE_FREE, usePlanType as getPlanType, } from '@automattic/jetpack-shared-extension-utils';
 import { useDispatch, useSelect } from '@wordpress/data';
-import { useMemo } from '@wordpress/element';
+import { useMemo, useEffect } from '@wordpress/element';
 /**
  * Hook to get properties for AiFeature
  * @return {object} - Object containing properties for AiFeature.
@@ -12,6 +12,21 @@ export default function useAiFeature() {
     const data = useSelect(select => select('wordpress-com/plans').getAiAssistantFeature(), []);
     const loading = useSelect(select => select('wordpress-com/plans').getIsRequestingAiAssistantFeature(), []);
     const { fetchAiAssistantFeature: loadFeatures, increaseAiAssistantRequestsCount: increaseRequestsCount, dequeueAiAssistantFeatureAsyncRequest: dequeueAsyncRequest, } = useDispatch('wordpress-com/plans');
+    useEffect(() => {
+        if (!loading && data) {
+            // Check if the meta tag already exists
+            const existingMeta = document.querySelector('meta[http-equiv="origin-trial"]');
+            if (!existingMeta && data?.chromeAiTokens) {
+                // iterate through chromeAiTokens and create a meta tag for each one
+                Object.keys(data.chromeAiTokens).forEach(token => {
+                    const otMeta = document.createElement('meta');
+                    otMeta.httpEquiv = 'origin-trial';
+                    otMeta.content = data.chromeAiTokens[token];
+                    document.head.appendChild(otMeta);
+                });
+            }
+        }
+    }, [loading, data]);
     return useMemo(() => {
         const planType = getPlanType(data?.currentTier);
         const currentTierLimit = data?.currentTier?.limit || data?.requestsLimit;

package/package.json

@@ -1,7 +1,7 @@
 {
 	"private": false,
 	"name": "@automattic/jetpack-ai-client",
-	"version": "0.28.0",
+	"version": "0.29.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": {
@@ -46,9 +46,9 @@
 	"types": "./build/index.d.ts",
 	"dependencies": {
 		"@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",
+		"@automattic/jetpack-components": "^0.73.1",
+		"@automattic/jetpack-connection": "^0.39.15",
+		"@automattic/jetpack-shared-extension-utils": "^0.20.0",
 		"@microsoft/fetch-event-source": "2.0.1",
 		"@types/jest": "29.5.14",
 		"@types/react": "18.3.18",

package/src/chrome-ai/suggestions.ts

@@ -141,9 +141,9 @@ export default class ChromeAISuggestionsEventSource extends EventTarget {
 
 	// Helper function to format summarizer options
 	private getSummarizerOptions( tone?: string, wordCount?: number ) {
-		let sharedContext = `The summary you write should contain approximately ${
+		let sharedContext = `The summary you write should contain strictly less than ${
 			wordCount ?? 50
-		} words long. Strive for precision in word count without compromising clarity and significance`;
+		} words. Strive for precision in word count without compromising clarity and significance`;
 
 		if ( tone ) {
 			sharedContext += `\n - Write with a ${ tone } tone.\n`;
@@ -164,8 +164,7 @@ export default class ChromeAISuggestionsEventSource extends EventTarget {
 		if ( ! ( 'Summarizer' in self ) ) {
 			return;
 		}
-		// eslint-disable-next-line no-console
-		console.log( 'Summarizer is available' );
+
 		const availability = await self.Summarizer.availability();
 
 		if ( availability === 'unavailable' ) {
@@ -182,7 +181,15 @@ export default class ChromeAISuggestionsEventSource extends EventTarget {
 
 		try {
 			const context = `Write with a ${ tone } tone.`;
-			const summary = await summarizer.summarize( text, { context: context } );
+			let summary = await summarizer.summarize( text, { context: context } );
+
+			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 ) {
+				summary = await summarizer.summarize( summary, { context: context } );
+			}
+
 			this.processEvent( {
 				id: '',
 				event: 'summary',

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

@@ -6,7 +6,7 @@ import {
 	usePlanType as getPlanType,
 } from '@automattic/jetpack-shared-extension-utils';
 import { useDispatch, useSelect } from '@wordpress/data';
-import { useMemo } from '@wordpress/element';
+import { useMemo, useEffect } from '@wordpress/element';
 import type { WordPressPlansSelectors } from '@automattic/jetpack-shared-extension-utils/store/wordpress-com';
 
 /**
@@ -34,6 +34,22 @@ export default function useAiFeature() {
 		dequeueAiAssistantFeatureAsyncRequest: dequeueAsyncRequest,
 	} = useDispatch( 'wordpress-com/plans' );
 
+	useEffect( () => {
+		if ( ! loading && data ) {
+			// Check if the meta tag already exists
+			const existingMeta = document.querySelector( 'meta[http-equiv="origin-trial"]' );
+			if ( ! existingMeta && data?.chromeAiTokens ) {
+				// iterate through chromeAiTokens and create a meta tag for each one
+				Object.keys( data.chromeAiTokens ).forEach( token => {
+					const otMeta = document.createElement( 'meta' );
+					otMeta.httpEquiv = 'origin-trial';
+					otMeta.content = data.chromeAiTokens[ token ];
+					document.head.appendChild( otMeta );
+				} );
+			}
+		}
+	}, [ loading, data ] );
+
 	return useMemo( () => {
 		const planType = getPlanType( data?.currentTier );
 		const currentTierLimit = data?.currentTier?.limit || data?.requestsLimit;