@automattic/jetpack-ai-client 0.14.6 → 0.15.0

package/build/logo-generator/store/selectors.d.ts

@@ -0,0 +1,119 @@
+import type { AiFeatureProps, LogoGeneratorStateProp, Logo, RequestError } from './types.js';
+import type { SiteDetails } from '../types.js';
+declare const selectors: {
+    /**
+     * Return the AI Assistant feature.
+     * @param {LogoGeneratorStateProp} state - The app state tree.
+     * @returns {Partial<AiFeatureProps>}      The AI Assistant feature data.
+     */
+    getAiAssistantFeature(state: LogoGeneratorStateProp): Partial<AiFeatureProps>;
+    /**
+     * Return the site details.
+     * @param {LogoGeneratorStateProp} state       - The app state tree.
+     * @returns {Partial<SiteDetails> | undefined}   The site details.
+     */
+    getSiteDetails(state: LogoGeneratorStateProp): Partial<SiteDetails> | undefined;
+    /**
+     * Get the isRequesting flag for the AI Assistant feature.
+     * @param {LogoGeneratorStateProp} state - The app state tree.
+     * @returns {boolean}                      The isRequesting flag.
+     */
+    getIsRequestingAiAssistantFeature(state: LogoGeneratorStateProp): boolean;
+    /**
+     * Get the logos history.
+     * @param {LogoGeneratorStateProp} state - The app state tree.
+     * @returns {Array<Logo>}                  The logos history array.
+     */
+    getLogos(state: LogoGeneratorStateProp): Array<Logo>;
+    /**
+     * Get the selected logo index.
+     * @param {LogoGeneratorStateProp} state - The app state tree.
+     * @returns {number | null}                The selected logo index.
+     */
+    getSelectedLogoIndex(state: LogoGeneratorStateProp): number | null;
+    /**
+     * Get the selected logo.
+     * @param {LogoGeneratorStateProp} state - The app state tree.
+     * @returns {Logo}                         The selected logo.
+     */
+    getSelectedLogo(state: LogoGeneratorStateProp): Logo;
+    /**
+     * Get the isSavingToLibrary flag.
+     * @param {LogoGeneratorStateProp} state - The app state tree.
+     * @returns {boolean}                      The isSavingToLibrary flag.
+     */
+    getIsSavingLogoToLibrary(state: LogoGeneratorStateProp): boolean;
+    /**
+     * Get the isApplyingLogo flag.
+     * @param {LogoGeneratorStateProp} state - The app state tree.
+     * @returns {boolean}                      The isApplyingLogo flag.
+     */
+    getIsApplyingLogo(state: LogoGeneratorStateProp): boolean;
+    /**
+     * Get the isEnhancingPrompt flag.
+     * @param {LogoGeneratorStateProp} state - The app state tree.
+     * @returns {boolean}                      The isEnhancingPrompt flag.
+     */
+    getIsEnhancingPrompt(state: LogoGeneratorStateProp): boolean;
+    /**
+     * Get the isRequestingImage flag.
+     * @param {LogoGeneratorStateProp} state - The app state tree.
+     * @returns {boolean}                      The isRequestingImage flag.
+     */
+    getIsRequestingImage(state: LogoGeneratorStateProp): boolean;
+    /**
+     * Get an aggregated isBusy flag, based on the loading states of the app.
+     * @param {LogoGeneratorStateProp} state - The app state tree.
+     * @returns {boolean}                      The isBusy flag.
+     */
+    getIsBusy(state: LogoGeneratorStateProp): boolean;
+    /**
+     * Get the requireUpgrade value from aiAssistantFeature
+     * @param {LogoGeneratorStateProp} state - The app state tree.
+     * @returns {boolean}                      The requireUpgrade flag.
+     */
+    getRequireUpgrade(state: LogoGeneratorStateProp): boolean;
+    /**
+     * Get the featureFetchError value.
+     * @param {LogoGeneratorStateProp} state - The app state tree.
+     * @returns {RequestError}                 The featureFetchError value.
+     */
+    getFeatureFetchError(state: LogoGeneratorStateProp): RequestError;
+    /**
+     * Get the firstLogoPromptFetchError value.
+     * @param {LogoGeneratorStateProp} state - The app state tree.
+     * @returns {RequestError}                 The firstLogoPromptFetchError value.
+     */
+    getFirstLogoPromptFetchError(state: LogoGeneratorStateProp): RequestError;
+    /**
+     * Get the enhancePromptFetchError value.
+     * @param {LogoGeneratorStateProp} state - The app state tree.
+     * @returns {RequestError}                 The enhancePromptFetchError value.
+     */
+    getEnhancePromptFetchError(state: LogoGeneratorStateProp): RequestError;
+    /**
+     * Get the logoFetchError value.
+     * @param {LogoGeneratorStateProp} state - The app state tree.
+     * @returns {RequestError}                 The logoFetchError value.
+     */
+    getLogoFetchError(state: LogoGeneratorStateProp): RequestError;
+    /**
+     * Get the saveToLibraryError value.
+     * @param {LogoGeneratorStateProp} state - The app state tree.
+     * @returns {RequestError}                 The saveToLibraryError value.
+     */
+    getSaveToLibraryError(state: LogoGeneratorStateProp): RequestError;
+    /**
+     * Get the logoUpdateError value.
+     * @param {LogoGeneratorStateProp} state - The app state tree.
+     * @returns {RequestError}                 The logoUpdateError value.
+     */
+    getLogoUpdateError(state: LogoGeneratorStateProp): RequestError;
+    /**
+     * Get the context value.
+     * @param {LogoGeneratorStateProp} state - The app state tree.
+     * @returns {string}                       The context value.
+     */
+    getContext(state: LogoGeneratorStateProp): string;
+};
+export default selectors;

package/build/logo-generator/store/selectors.js

@@ -0,0 +1,173 @@
+/**
+ * Types
+ */
+import { DEFAULT_LOGO_COST } from '../constants.js';
+const selectors = {
+    /**
+     * Return the AI Assistant feature.
+     * @param {LogoGeneratorStateProp} state - The app state tree.
+     * @returns {Partial<AiFeatureProps>}      The AI Assistant feature data.
+     */
+    getAiAssistantFeature(state) {
+        // Clean up the _meta property.
+        const data = { ...state.features.aiAssistantFeature };
+        delete data._meta;
+        return data;
+    },
+    /**
+     * Return the site details.
+     * @param {LogoGeneratorStateProp} state       - The app state tree.
+     * @returns {Partial<SiteDetails> | undefined}   The site details.
+     */
+    getSiteDetails(state) {
+        return state.siteDetails;
+    },
+    /**
+     * Get the isRequesting flag for the AI Assistant feature.
+     * @param {LogoGeneratorStateProp} state - The app state tree.
+     * @returns {boolean}                      The isRequesting flag.
+     */
+    getIsRequestingAiAssistantFeature(state) {
+        return state.features.aiAssistantFeature?._meta?.isRequesting ?? false;
+    },
+    /**
+     * Get the logos history.
+     * @param {LogoGeneratorStateProp} state - The app state tree.
+     * @returns {Array<Logo>}                  The logos history array.
+     */
+    getLogos(state) {
+        return state.history ?? [];
+    },
+    /**
+     * Get the selected logo index.
+     * @param {LogoGeneratorStateProp} state - The app state tree.
+     * @returns {number | null}                The selected logo index.
+     */
+    getSelectedLogoIndex(state) {
+        return state.selectedLogoIndex ?? null;
+    },
+    /**
+     * Get the selected logo.
+     * @param {LogoGeneratorStateProp} state - The app state tree.
+     * @returns {Logo}                         The selected logo.
+     */
+    getSelectedLogo(state) {
+        return state.history?.[state.selectedLogoIndex] ?? null;
+    },
+    /**
+     * Get the isSavingToLibrary flag.
+     * @param {LogoGeneratorStateProp} state - The app state tree.
+     * @returns {boolean}                      The isSavingToLibrary flag.
+     */
+    getIsSavingLogoToLibrary(state) {
+        return state._meta?.isSavingLogoToLibrary ?? false;
+    },
+    /**
+     * Get the isApplyingLogo flag.
+     * @param {LogoGeneratorStateProp} state - The app state tree.
+     * @returns {boolean}                      The isApplyingLogo flag.
+     */
+    getIsApplyingLogo(state) {
+        return state._meta?.isApplyingLogo ?? false;
+    },
+    /**
+     * Get the isEnhancingPrompt flag.
+     * @param {LogoGeneratorStateProp} state - The app state tree.
+     * @returns {boolean}                      The isEnhancingPrompt flag.
+     */
+    getIsEnhancingPrompt(state) {
+        return state._meta?.isEnhancingPrompt ?? false;
+    },
+    /**
+     * Get the isRequestingImage flag.
+     * @param {LogoGeneratorStateProp} state - The app state tree.
+     * @returns {boolean}                      The isRequestingImage flag.
+     */
+    getIsRequestingImage(state) {
+        return state._meta?.isRequestingImage ?? false;
+    },
+    /**
+     * Get an aggregated isBusy flag, based on the loading states of the app.
+     * @param {LogoGeneratorStateProp} state - The app state tree.
+     * @returns {boolean}                      The isBusy flag.
+     */
+    getIsBusy(state) {
+        return (selectors.getIsApplyingLogo(state) ||
+            selectors.getIsSavingLogoToLibrary(state) ||
+            selectors.getIsRequestingImage(state) ||
+            selectors.getIsEnhancingPrompt(state));
+    },
+    /**
+     * Get the requireUpgrade value from aiAssistantFeature
+     * @param {LogoGeneratorStateProp} state - The app state tree.
+     * @returns {boolean}                      The requireUpgrade flag.
+     */
+    getRequireUpgrade(state) {
+        const feature = state.features.aiAssistantFeature;
+        const logoCost = feature?.costs?.['jetpack-ai-logo-generator']?.logo ?? DEFAULT_LOGO_COST;
+        const currentLimit = feature?.currentTier?.value || 0;
+        const currentUsage = feature?.usagePeriod?.requestsCount || 0;
+        const isUnlimited = currentLimit === 1;
+        const hasNoNextTier = !feature?.nextTier; // If there is no next tier, the user cannot upgrade.
+        // Add a local check on top of the feature flag, based on the current usage and logo cost.
+        return (state.features.aiAssistantFeature?.requireUpgrade ||
+            (!isUnlimited && !hasNoNextTier && currentLimit - currentUsage < logoCost));
+    },
+    /**
+     * Get the featureFetchError value.
+     * @param {LogoGeneratorStateProp} state - The app state tree.
+     * @returns {RequestError}                 The featureFetchError value.
+     */
+    getFeatureFetchError(state) {
+        return state._meta?.featureFetchError ?? null;
+    },
+    /**
+     * Get the firstLogoPromptFetchError value.
+     * @param {LogoGeneratorStateProp} state - The app state tree.
+     * @returns {RequestError}                 The firstLogoPromptFetchError value.
+     */
+    getFirstLogoPromptFetchError(state) {
+        return state._meta?.firstLogoPromptFetchError ?? null;
+    },
+    /**
+     * Get the enhancePromptFetchError value.
+     * @param {LogoGeneratorStateProp} state - The app state tree.
+     * @returns {RequestError}                 The enhancePromptFetchError value.
+     */
+    getEnhancePromptFetchError(state) {
+        return state._meta?.enhancePromptFetchError ?? null;
+    },
+    /**
+     * Get the logoFetchError value.
+     * @param {LogoGeneratorStateProp} state - The app state tree.
+     * @returns {RequestError}                 The logoFetchError value.
+     */
+    getLogoFetchError(state) {
+        return state._meta?.logoFetchError ?? null;
+    },
+    /**
+     * Get the saveToLibraryError value.
+     * @param {LogoGeneratorStateProp} state - The app state tree.
+     * @returns {RequestError}                 The saveToLibraryError value.
+     */
+    getSaveToLibraryError(state) {
+        return state._meta?.saveToLibraryError ?? null;
+    },
+    /**
+     * Get the logoUpdateError value.
+     * @param {LogoGeneratorStateProp} state - The app state tree.
+     * @returns {RequestError}                 The logoUpdateError value.
+     */
+    getLogoUpdateError(state) {
+        return state._meta?.logoUpdateError ?? null;
+    },
+    /**
+     * Get the context value.
+     * @param {LogoGeneratorStateProp} state - The app state tree.
+     * @returns {string}                       The context value.
+     */
+    getContext(state) {
+        return state._meta?.context ?? '';
+    },
+};
+export default selectors;

package/build/logo-generator/store/types.d.ts

@@ -0,0 +1,164 @@
+import type { SiteDetails } from '../types.js';
+/**
+ * Types for the AI Assistant feature.
+ */
+export type Plan = {
+    product_id: number;
+    product_name: string;
+    product_slug: string;
+};
+export type UpgradeTypeProp = 'vip' | 'default';
+export type TierUnlimitedProps = {
+    slug: 'ai-assistant-tier-unlimited';
+    limit: 999999999;
+    value: 1;
+    readableLimit: string;
+};
+export type TierFreeProps = {
+    slug: 'ai-assistant-tier-free';
+    limit: 20;
+    value: 0;
+};
+export type Tier100Props = {
+    slug: 'ai-assistant-tier-100';
+    limit: 100;
+    value: 100;
+};
+export type Tier200Props = {
+    slug: 'ai-assistant-tier-200';
+    limit: 200;
+    value: 200;
+};
+export type Tier500Props = {
+    slug: 'ai-assistant-tier-500';
+    limit: 500;
+    value: 500;
+};
+export type Tier750Props = {
+    slug: 'ai-assistant-tier-750';
+    limit: 750;
+    value: 750;
+};
+export type Tier1000Props = {
+    slug: 'ai-assistant-tier-1000';
+    limit: 1000;
+    value: 1000;
+};
+export type TierProp = {
+    slug: TierSlugProp;
+    limit: TierLimitProp;
+    value: TierValueProp;
+    readableLimit?: string;
+};
+export type TierLimitProp = TierUnlimitedProps['limit'] | TierFreeProps['limit'] | Tier100Props['limit'] | Tier200Props['limit'] | Tier500Props['limit'] | Tier750Props['limit'] | Tier1000Props['limit'];
+export type TierSlugProp = TierUnlimitedProps['slug'] | TierFreeProps['slug'] | Tier100Props['slug'] | Tier200Props['slug'] | Tier500Props['slug'] | Tier750Props['slug'] | Tier1000Props['slug'];
+export type TierValueProp = TierUnlimitedProps['value'] | TierFreeProps['value'] | Tier100Props['value'] | Tier200Props['value'] | Tier500Props['value'] | Tier750Props['value'] | Tier1000Props['value'];
+export type AiFeatureProps = {
+    hasFeature: boolean;
+    isOverLimit: boolean;
+    requestsCount: number;
+    requestsLimit: number;
+    requireUpgrade: boolean;
+    errorMessage?: string;
+    errorCode?: string;
+    upgradeType: UpgradeTypeProp;
+    currentTier?: TierProp;
+    usagePeriod?: {
+        currentStart: string;
+        nextStart: string;
+        requestsCount: number;
+    };
+    nextTier?: TierProp | null;
+    tierPlansEnabled?: boolean;
+    costs?: {
+        'jetpack-ai-logo-generator': {
+            logo: number;
+        };
+    };
+};
+export type AiFeatureStateProps = AiFeatureProps & {
+    _meta?: {
+        isRequesting: boolean;
+        asyncRequestCountdown: number;
+        asyncRequestTimerId: number;
+        isRequestingImage: boolean;
+    };
+};
+export type Logo = {
+    url: string;
+    description: string;
+    mediaId?: number;
+};
+export type RequestError = string | Error | null;
+export type LogoGeneratorStateProp = {
+    _meta?: {
+        isSavingLogoToLibrary: boolean;
+        isApplyingLogo: boolean;
+        isRequestingImage: boolean;
+        isEnhancingPrompt: boolean;
+        featureFetchError?: RequestError;
+        firstLogoPromptFetchError?: RequestError;
+        enhancePromptFetchError?: RequestError;
+        logoFetchError?: RequestError;
+        saveToLibraryError?: RequestError;
+        logoUpdateError?: RequestError;
+        context: string;
+    };
+    siteDetails?: SiteDetails | Record<string, never>;
+    features: {
+        aiAssistantFeature?: AiFeatureStateProps;
+    };
+    history: Array<Logo>;
+    selectedLogoIndex: number;
+};
+export type Selectors = {
+    getAiAssistantFeature(siteId?: string): Partial<AiFeatureProps>;
+    getIsRequestingAiAssistantFeature(): boolean;
+    getLogos(): Array<Logo>;
+    getSelectedLogoIndex(): number | null;
+    getSelectedLogo(): Logo;
+    getSiteDetails(): SiteDetails;
+    getIsSavingLogoToLibrary(): boolean;
+    getIsApplyingLogo(): boolean;
+    getIsRequestingImage(): boolean;
+    getIsEnhancingPrompt(): boolean;
+    getIsBusy(): boolean;
+    getRequireUpgrade(): boolean;
+    getFeatureFetchError(): RequestError;
+    getFirstLogoPromptFetchError(): RequestError;
+    getEnhancePromptFetchError(): RequestError;
+    getLogoFetchError(): RequestError;
+    getSaveToLibraryError(): RequestError;
+    getLogoUpdateError(): RequestError;
+    getContext(): string;
+};
+export type AiAssistantFeatureEndpointResponseProps = {
+    'is-enabled': boolean;
+    'has-feature': boolean;
+    'is-over-limit': boolean;
+    'requests-count': number;
+    'requests-limit': number;
+    'usage-period': {
+        'current-start': string;
+        'next-start': string;
+        'requests-count': number;
+    };
+    'site-require-upgrade': boolean;
+    'error-message'?: string;
+    'error-code'?: string;
+    'is-playground-visible'?: boolean;
+    'upgrade-type': UpgradeTypeProp;
+    'current-tier': TierProp;
+    'tier-plans': Array<TierProp>;
+    'next-tier'?: TierProp | null;
+    'tier-plans-enabled': boolean;
+    costs: {
+        'jetpack-ai-logo-generator': {
+            logo: number;
+        };
+    };
+};
+export type SaveLogo = (logo: Logo) => Promise<{
+    mediaId: number;
+    mediaURL: string;
+}>;

package/build/logo-generator/store/types.js

@@ -0,0 +1 @@
+export {};

package/build/logo-generator/types.d.ts

@@ -0,0 +1,82 @@
+/**
+ * Types
+ */
+import type { Logo } from './store/types.js';
+export type SiteDetails = {
+    ID: number;
+    URL: string;
+    domain: string;
+    name: string;
+    description: string;
+};
+export interface GeneratorModalProps {
+    siteDetails?: SiteDetails;
+    isOpen: boolean;
+    onClose: () => void;
+    context: string;
+}
+export interface LogoPresenterProps {
+    logo?: Logo;
+    loading?: boolean;
+    onApplyLogo: () => void;
+    logoAccepted?: boolean;
+    siteId: string | number;
+}
+export type SaveToMediaLibraryProps = {
+    siteId: string | number;
+    url: string;
+    attrs?: {
+        caption?: string;
+        description?: string;
+        title?: string;
+        alt?: string;
+    };
+};
+export type SaveToMediaLibraryResponseProps = {
+    code: number;
+    media: [
+        {
+            ID: number;
+            URL: string;
+        }
+    ];
+};
+export type CheckMediaProps = {
+    siteId: string | number;
+    mediaId: Logo['mediaId'];
+};
+export type SetSiteLogoProps = {
+    siteId: string | number;
+    imageId: string | number;
+};
+export type SetSiteLogoResponseProps = {
+    id: number;
+    url: string;
+};
+export type RequestTokenOptions = {
+    siteDetails?: SiteDetails;
+    isJetpackSite?: boolean;
+    expirationTime?: number;
+};
+export type TokenDataProps = {
+    token: string;
+    blogId: string | undefined;
+    expire: number;
+};
+export type TokenDataEndpointResponseProps = {
+    token: string;
+    blog_id: string;
+};
+export type SaveToStorageProps = Logo & {
+    siteId: string;
+};
+export type UpdateInStorageProps = {
+    siteId: string;
+    url: Logo['url'];
+    newUrl: Logo['url'];
+    mediaId: Logo['mediaId'];
+};
+export type RemoveFromStorageProps = {
+    mediaId: Logo['mediaId'];
+    siteId: string;
+};

package/build/logo-generator/types.js

@@ -0,0 +1 @@
+export {};

package/build/types.d.ts

@@ -1,3 +1,4 @@
+import type * as BlockEditorSelectors from '@wordpress/block-editor/store/selectors.js';
 export declare const ERROR_SERVICE_UNAVAILABLE: "error_service_unavailable";
 export declare const ERROR_QUOTA_EXCEEDED: "error_quota_exceeded";
 export declare const ERROR_MODERATION: "error_moderation";
@@ -40,3 +41,8 @@ export type Block = {
 };
 export type TranscriptionState = RecordingState | 'validating' | 'processing' | 'error';
 export type { RenderHTMLRules } from './libs/index.js';
+export interface BlockEditorStore {
+    selectors: {
+        [key in keyof typeof BlockEditorSelectors]: (typeof BlockEditorSelectors)[key];
+    };
+}

package/package.json

@@ -1,7 +1,7 @@
 {
 	"private": false,
 	"name": "@automattic/jetpack-ai-client",
-	"version": "0.14.6",
+	"version": "0.15.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": {
@@ -43,11 +43,13 @@
 	"types": "./build/index.d.ts",
 	"dependencies": {
 		"@automattic/jetpack-base-styles": "^0.6.28",
-		"@automattic/jetpack-connection": "^0.33.19",
-		"@automattic/jetpack-shared-extension-utils": "^0.14.17",
+		"@automattic/jetpack-connection": "^0.34.1",
+		"@automattic/jetpack-shared-extension-utils": "^0.14.19",
 		"@microsoft/fetch-event-source": "2.0.1",
 		"@types/react": "18.3.1",
+		"@types/wordpress__block-editor": "11.5.14",
 		"@wordpress/api-fetch": "7.2.0",
+		"@wordpress/blob": "4.2.0",
 		"@wordpress/block-editor": "13.2.0",
 		"@wordpress/components": "28.2.0",
 		"@wordpress/compose": "7.2.0",

package/src/ask-question/sync.ts

@@ -12,13 +12,7 @@ import type { PromptProp } from '../types.js';
 /**
  * The response data from the AI assistant when doing a sync, not-streamed question.
  */
-export type ResponseData = {
-	choices: Array< {
-		message: {
-			content: string;
-		};
-	} >;
-};
+export type ResponseData = string;
 
 const debug = debugFactory( 'jetpack-ai-client:ask-question-sync' );
 
@@ -41,13 +35,9 @@ const debug = debugFactory( 'jetpack-ai-client:ask-question-sync' );
  */
 export default async function askQuestionSync(
 	question: PromptProp,
-	{ postId = null, feature, model }: AskQuestionOptionsArgProps = {}
+	options: AskQuestionOptionsArgProps = {}
 ): Promise< ResponseData > {
-	debug( 'Asking question with no streaming: %o. options: %o', question, {
-		postId,
-		feature,
-		model,
-	} );
+	debug( 'Asking question with no streaming: %o. options: %o', question, options );
 
 	/**
 	 * The URL to the AI assistant query endpoint.
@@ -63,12 +53,12 @@ export default async function askQuestionSync(
 		return Promise.reject( error );
 	}
 
+	const messages = Array.isArray( question ) ? { messages: question } : { question: question };
+
 	const body = {
-		question: question,
+		...messages,
+		...options,
 		stream: false,
-		postId,
-		feature,
-		model,
 	};
 
 	const headers = {
@@ -76,16 +66,21 @@ export default async function askQuestionSync(
 		'Content-Type': 'application/json',
 	};
 
-	const data = await fetch( URL, {
-		method: 'POST',
-		headers,
-		body: JSON.stringify( body ),
-	} ).then( response => response.json() );
+	try {
+		const data = await fetch( URL, {
+			method: 'POST',
+			headers,
+			body: JSON.stringify( body ),
+		} ).then( response => response.json() );
 
-	if ( data?.data?.status && data?.data?.status > 200 ) {
-		debug( 'Error generating prompt: %o', data );
-		return Promise.reject( data );
-	}
+		if ( data?.data?.status && data?.data?.status > 200 ) {
+			debug( 'Error generating prompt: %o', data );
+			return Promise.reject( data );
+		}
 
-	return data as ResponseData;
+		return data.choices?.[ 0 ]?.message?.content as string;
+	} catch ( error ) {
+		debug( 'Error asking question: %o', error );
+		return Promise.reject( error );
+	}
 }

package/src/hooks/use-image-generator/index.ts

@@ -150,7 +150,7 @@ const getStableDiffusionImageGenerationPrompt = async (
 	 */
 	const data = await askQuestionSync( prompt, { feature } );
 
-	return data.choices?.[ 0 ]?.message?.content;
+	return data;
 };
 
 const useImageGenerator = () => {