@ai-stack/payloadcms 3.68.0 → 3.76.0-beta.1

package/dist/ui/Compose/hooks/useGenerate.js

@@ -1,14 +1,14 @@
 import { experimental_useObject as useObject } from '@ai-sdk/react';
 import { useEditorConfigContext } from '@payloadcms/richtext-lexical/client';
-import { toast, useConfig, useDocumentInfo, useField, useForm, useLocale } from '@payloadcms/ui';
+import { toast, useDocumentInfo, useField, useForm, useLocale } from '@payloadcms/ui';
 import { jsonSchema } from 'ai';
-import { useCallback, useEffect, useMemo, useRef, useState } from 'react';
-import { PLUGIN_AI_JOBS_TABLE, PLUGIN_API_ENDPOINT_GENERATE, PLUGIN_API_ENDPOINT_GENERATE_UPLOAD, PLUGIN_INSTRUCTIONS_TABLE, PLUGIN_NAME } from '../../../defaults.js';
+import { useCallback, useEffect, useRef } from 'react';
+import { PLUGIN_API_ENDPOINT_GENERATE } from '../../../defaults.js';
 import { useFieldProps } from '../../../providers/FieldProvider/useFieldProps.js';
-import { editorSchemaValidator } from '../../../utilities/editorSchemaValidator.js';
-import { fieldToJsonSchema } from '../../../utilities/fieldToJsonSchema.js';
 import { setSafeLexicalState } from '../../../utilities/setSafeLexicalState.js';
+import { useGenerateUpload } from './useGenerateUpload.js';
 import { useHistory } from './useHistory.js';
+import { useStreamingUpdate } from './useStreamingUpdate.js';
 export const useGenerate = ({ instructionId })=>{
     // Create a ref to hold the current instructionId
     const instructionIdRef = useRef(instructionId);
@@ -20,66 +20,13 @@ export const useGenerate = ({ instructionId })=>{
     ]);
     const { field, path: pathFromContext } = useFieldProps();
     const editorConfigContext = useEditorConfigContext();
-    const { editor } = editorConfigContext;
-    const { config } = useConfig();
-    const { routes: { api }, serverURL } = config;
     const { setValue } = useField({
         path: pathFromContext ?? ''
     });
     const { set: setHistory } = useHistory();
-    // Async job UI state
-    const [jobStatus, setJobStatus] = useState(undefined);
-    const [jobProgress, setJobProgress] = useState(0);
-    const [isJobActive, setIsJobActive] = useState(false);
     const { getData } = useForm();
-    const { id: documentId, collectionSlug } = useDocumentInfo();
+    const { id: documentId } = useDocumentInfo();
     const localFromContext = useLocale();
-    // Reuse config from above instead of calling useConfig again
-    const { collections } = config;
-    const collection = collections.find((collection)=>collection.slug === PLUGIN_INSTRUCTIONS_TABLE);
-    const { custom: { [PLUGIN_NAME]: { editorConfig = {} } = {} } = {} } = collection?.admin ?? {};
-    const { schema: editorSchema = {} } = editorConfig;
-    const memoizedValidator = useMemo(()=>{
-        return editorSchemaValidator(editorSchema);
-    }, [
-        editorSchema
-    ]);
-    const memoizedSchema = useMemo(()=>jsonSchema(editorSchema, {
-            validate: (value)=>{
-                const isValid = memoizedValidator(value);
-                if (isValid) {
-                    return {
-                        success: true,
-                        value
-                    };
-                } else {
-                    return {
-                        error: new Error('Invalid schema'),
-                        success: false
-                    };
-                }
-            }
-        }), [
-        memoizedValidator
-    ]);
-    // Active JSON schema for useObject based on field type
-    const activeSchema = useMemo(()=>{
-        const f = field;
-        const fieldType = f?.type;
-        if (fieldType === 'richText') {
-            return memoizedSchema;
-        }
-        if (f && f.name && fieldType) {
-            const schemaJson = fieldToJsonSchema(f);
-            if (schemaJson && Object.keys(schemaJson).length > 0) {
-                return jsonSchema(schemaJson);
-            }
-        }
-        return undefined;
-    }, [
-        field,
-        memoizedSchema
-    ]);
     const { isLoading: loadingObject, object, stop: objectStop, submit } = useObject({
         api: `/api${PLUGIN_API_ENDPOINT_GENERATE}`,
         onError: (error)=>{
@@ -90,8 +37,8 @@ export const useGenerate = ({ instructionId })=>{
             if (result.object && field) {
                 if (field.type === 'richText') {
                     setHistory(result.object);
-                    setValue(result.object);
-                } else if ('name' in field) {
+                    setSafeLexicalState(result.object, editor);
+                } else if ('name' in field && result.object[field.name]) {
                     setHistory(result.object[field.name]);
                     setValue(result.object[field.name]);
                 }
@@ -99,25 +46,23 @@ export const useGenerate = ({ instructionId })=>{
                 console.log('onFinish: result, field ', result, field);
             }
         },
-        schema: activeSchema
+        schema: jsonSchema({
+            type: 'object',
+            additionalProperties: true,
+            properties: {}
+        })
     });
-    useEffect(()=>{
-        if (!object) {
-            return;
-        }
-        requestAnimationFrame(()=>{
-            if (field?.type === 'richText') {
-                setSafeLexicalState(object, editor);
-            } else if (field && 'name' in field && object[field.name]) {
-                setValue(object[field.name]);
-            }
-        });
-    }, [
-        object,
+    const { editor } = editorConfigContext;
+    // Hook: Handle high-frequency streaming updates
+    useStreamingUpdate({
         editor,
-        field,
-        setValue
-    ]);
+        isLoading: loadingObject,
+        object
+    });
+    // Hook 2: Handle Upload generation and polling
+    const { generateUpload, isJobActive, jobProgress, jobStatus } = useGenerateUpload({
+        instructionIdRef
+    });
     const streamObject = useCallback(({ action = 'Compose', params })=>{
         const doc = getData();
         const currentInstructionId = instructionIdRef.current;
@@ -138,103 +83,10 @@ export const useGenerate = ({ instructionId })=>{
     }, [
         localFromContext?.code,
         instructionIdRef,
-        documentId
-    ]);
-    const generateUpload = useCallback(async ()=>{
-        const doc = getData();
-        const currentInstructionId = instructionIdRef.current;
-        return fetch(`${serverURL}${api}${PLUGIN_API_ENDPOINT_GENERATE_UPLOAD}`, {
-            body: JSON.stringify({
-                collectionSlug: collectionSlug ?? '',
-                doc,
-                documentId,
-                locale: localFromContext?.code,
-                options: {
-                    instructionId: currentInstructionId
-                }
-            }),
-            credentials: 'include',
-            headers: {
-                'Content-Type': 'application/json'
-            },
-            method: 'POST'
-        }).then(async (uploadResponse)=>{
-            if (uploadResponse.ok) {
-                const json = await uploadResponse.json();
-                const { job, result } = json || {};
-                if (result) {
-                    // Set the upload ID
-                    setValue(result?.id);
-                    setHistory(result?.id);
-                    // Show toast to prompt user to save
-                    toast.success('Image generated successfully! Click Save to see the preview.');
-                    return uploadResponse;
-                }
-                // Async job: poll AI Jobs collection for status/progress/result_id
-                if (job && job.id) {
-                    setIsJobActive(true);
-                    const cancelled = false;
-                    let attempts = 0;
-                    const maxAttempts = 600 // up to ~10 minutes @ 1s
-                    ;
-                    // Basic in-hook state via closure variables; UI will re-render off fetches below
-                    const poll = async ()=>{
-                        if (cancelled) {
-                            return;
-                        }
-                        try {
-                            const res = await fetch(`${serverURL}${api}/${PLUGIN_AI_JOBS_TABLE}/${job.id}`, {
-                                credentials: 'include'
-                            });
-                            if (res.ok) {
-                                const jobDoc = await res.json();
-                                const { progress, result_id, status } = jobDoc || {};
-                                setJobStatus(status);
-                                setJobProgress(progress ?? 0);
-                                // When result present, set field and finish
-                                if (status === 'completed' && result_id) {
-                                    // Force upload field to refetch by clearing then setting the ID
-                                    setValue(null);
-                                    setTimeout(()=>{
-                                        setValue(result_id);
-                                    }, 0);
-                                    setHistory(result_id);
-                                    setIsJobActive(false);
-                                    return;
-                                }
-                                if (status === 'failed') {
-                                    setIsJobActive(false);
-                                    throw new Error('Video generation failed');
-                                }
-                            }
-                        } catch (e) {
-                        // silent retry
-                        }
-                        attempts += 1;
-                        if (!cancelled && attempts < maxAttempts) {
-                            setTimeout(poll, 1000);
-                        }
-                    };
-                    setTimeout(poll, 1000);
-                    return uploadResponse;
-                }
-                throw new Error('generateUpload: Unexpected response');
-            } else {
-                const { errors = [] } = await uploadResponse.json();
-                const errStr = errors.map((error)=>error.message).join(', ');
-                throw new Error(errStr);
-            }
-        }).catch((error)=>{
-            toast.error(`Failed to generate: ${error.message}`);
-            console.error('Error generating or setting your upload, please set it manually if its saved in your media files.', error);
-        });
-    }, [
-        getData,
-        localFromContext?.code,
-        instructionIdRef,
-        setValue,
         documentId,
-        collectionSlug
+        getData,
+        submit,
+        editor
     ]);
     const generate = useCallback(async (options)=>{
         if (field?.type === 'upload') {

package/dist/ui/Compose/hooks/useGenerate.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../../src/ui/Compose/hooks/useGenerate.ts"],"sourcesContent":["import { experimental_useObject as useObject } from '@ai-sdk/react'\nimport { useEditorConfigContext } from '@payloadcms/richtext-lexical/client'\nimport { toast, useConfig, useDocumentInfo, useField, useForm, useLocale } from '@payloadcms/ui'\nimport { jsonSchema } from 'ai'\nimport { useCallback, useEffect, useMemo, useRef, useState } from 'react'\n\nimport type { ActionMenuItems, GenerateTextarea } from '../../../types.js'\n\nimport {\n  PLUGIN_AI_JOBS_TABLE,\n  PLUGIN_API_ENDPOINT_GENERATE,\n  PLUGIN_API_ENDPOINT_GENERATE_UPLOAD,\n  PLUGIN_INSTRUCTIONS_TABLE,\n  PLUGIN_NAME,\n} from '../../../defaults.js'\nimport { useFieldProps } from '../../../providers/FieldProvider/useFieldProps.js'\nimport { editorSchemaValidator } from '../../../utilities/editorSchemaValidator.js'\nimport { fieldToJsonSchema } from '../../../utilities/fieldToJsonSchema.js'\nimport { setSafeLexicalState } from '../../../utilities/setSafeLexicalState.js'\nimport { useHistory } from './useHistory.js'\n\ntype ActionCallbackParams = { action: ActionMenuItems; params?: unknown }\n\nexport const useGenerate = ({ instructionId }: { instructionId: string }) => {\n  // Create a ref to hold the current instructionId\n  const instructionIdRef = useRef(instructionId)\n\n  // Update the ref whenever instructionId changes\n  useEffect(() => {\n    instructionIdRef.current = instructionId\n  }, [instructionId])\n\n  const { field, path: pathFromContext } = useFieldProps()\n  const editorConfigContext = useEditorConfigContext()\n\n  const { editor } = editorConfigContext\n\n  const { config } = useConfig()\n  const {\n    routes: { api },\n    serverURL,\n  } = config\n\n  const { setValue } = useField<any>({\n    path: pathFromContext ?? '',\n  })\n\n  const { set: setHistory } = useHistory()\n\n  // Async job UI state\n  const [jobStatus, setJobStatus] = useState<string | undefined>(undefined)\n  const [jobProgress, setJobProgress] = useState<number>(0)\n  const [isJobActive, setIsJobActive] = useState<boolean>(false)\n\n  const { getData } = useForm()\n  const { id: documentId, collectionSlug } = useDocumentInfo()\n\n  const localFromContext = useLocale()\n  \n  // Reuse config from above instead of calling useConfig again\n  const { collections } = config\n\n  const collection = collections.find((collection) => collection.slug === PLUGIN_INSTRUCTIONS_TABLE)\n  const { custom: { [PLUGIN_NAME]: { editorConfig = {} } = {} } = {} } = collection?.admin ?? {}\n  const { schema: editorSchema = {} } = editorConfig\n\n  const memoizedValidator = useMemo(() => {\n    return editorSchemaValidator(editorSchema)\n  }, [editorSchema])\n\n  const memoizedSchema = useMemo(\n    () =>\n      jsonSchema(editorSchema, {\n        validate: (value) => {\n          const isValid = memoizedValidator(value)\n\n          if (isValid) {\n            return {\n              success: true,\n              value,\n            }\n          } else {\n            return {\n              error: new Error('Invalid schema'),\n              success: false,\n            }\n          }\n        },\n      }),\n    [memoizedValidator],\n  )\n\n  // Active JSON schema for useObject based on field type\n  const activeSchema = useMemo(() => {\n    const f = field as any\n    const fieldType = f?.type as string | undefined\n    if (fieldType === 'richText') {\n      return memoizedSchema\n    }\n    if (f && f.name && fieldType) {\n      const schemaJson = fieldToJsonSchema(f)\n      if (schemaJson && Object.keys(schemaJson).length > 0) {\n        return jsonSchema(schemaJson)\n      }\n    }\n    return undefined\n  }, [field, memoizedSchema])\n\n  const {\n    isLoading: loadingObject,\n    object,\n    stop: objectStop,\n    submit,\n  } = useObject({\n    api: `/api${PLUGIN_API_ENDPOINT_GENERATE}`,\n    onError: (error: any) => {\n      toast.error(`Failed to generate: ${error.message}`)\n      console.error('Error generating object:', error)\n    },\n    onFinish: (result) => {\n      if (result.object && field) {\n        if (field.type === 'richText') {\n          setHistory(result.object)\n          setValue(result.object)\n        } else if ('name' in field) {\n          setHistory(result.object[field.name])\n          setValue(result.object[field.name])\n        }\n      } else {\n        console.log('onFinish: result, field ', result, field)\n      }\n    },\n    schema: activeSchema as any,\n  })\n\n  useEffect(() => {\n    if (!object) {\n      return\n    }\n\n    requestAnimationFrame(() => {\n      if (field?.type === 'richText') {\n      setSafeLexicalState(object, editor)\n      } else if (field && 'name' in field && object[field.name]) {\n        setValue(object[field.name])\n      }\n    })\n  }, [object, editor, field, setValue])\n\n  const streamObject = useCallback(\n    ({ action = 'Compose', params }: ActionCallbackParams) => {\n      const doc = getData()\n\n      const currentInstructionId = instructionIdRef.current\n\n      const options = {\n        action,\n        actionParams: params,\n        instructionId: currentInstructionId,\n      }\n\n      submit({\n        allowedEditorNodes: Array.from(editor?._nodes?.keys() || []),\n        doc: {\n          ...doc,\n          id: documentId,\n        },\n        locale: localFromContext?.code,\n        options,\n      })\n    },\n    [localFromContext?.code, instructionIdRef, documentId],\n  )\n\n  const generateUpload = useCallback(async () => {\n    const doc = getData()\n    const currentInstructionId = instructionIdRef.current\n\n    return fetch(`${serverURL}${api}${PLUGIN_API_ENDPOINT_GENERATE_UPLOAD}`, {\n      body: JSON.stringify({\n        collectionSlug: collectionSlug ?? '',\n        doc,\n        documentId,\n        locale: localFromContext?.code,\n        options: {\n          instructionId: currentInstructionId,\n        },\n      } satisfies Parameters<GenerateTextarea>[0]),\n      credentials: 'include',\n      headers: {\n        'Content-Type': 'application/json',\n      },\n      method: 'POST',\n    })\n      .then(async (uploadResponse) => {\n        if (uploadResponse.ok) {\n          const json = await uploadResponse.json()\n          const { job, result } = json || {}\n          if (result) {\n            // Set the upload ID\n            setValue(result?.id)\n            setHistory(result?.id)\n            \n            // Show toast to prompt user to save\n            toast.success('Image generated successfully! Click Save to see the preview.')\n            \n            return uploadResponse\n          }\n\n          // Async job: poll AI Jobs collection for status/progress/result_id\n          if (job && job.id) {\n            setIsJobActive(true)\n            const cancelled = false\n            let attempts = 0\n            const maxAttempts = 600 // up to ~10 minutes @ 1s\n\n            // Basic in-hook state via closure variables; UI will re-render off fetches below\n            const poll = async (): Promise<void> => {\n              if (cancelled) {\n                return\n              }\n              try {\n                const res = await fetch(\n                  `${serverURL}${api}/${PLUGIN_AI_JOBS_TABLE}/${job.id}`,\n                  { credentials: 'include' },\n                )\n                if (res.ok) {\n                  const jobDoc = await res.json()\n                  const { progress, result_id, status } = jobDoc || {}\n                  setJobStatus(status)\n                  setJobProgress(progress ?? 0)\n                  // When result present, set field and finish\n                  if (status === 'completed' && result_id) {\n                    // Force upload field to refetch by clearing then setting the ID\n                    setValue(null)\n                    setTimeout(() => {\n                      setValue(result_id)\n                    }, 0)\n                    setHistory(result_id)\n                    setIsJobActive(false)\n                    return\n                  }\n                  if (status === 'failed') {\n                    setIsJobActive(false)\n                    throw new Error('Video generation failed')\n                  }\n                }\n              } catch (e) {\n                // silent retry\n              }\n\n              attempts += 1\n              if (!cancelled && attempts < maxAttempts) {\n                setTimeout(poll, 1000)\n              }\n            }\n            setTimeout(poll, 1000)\n            return uploadResponse\n          }\n\n          throw new Error('generateUpload: Unexpected response')\n        } else {\n          const { errors = [] } = await uploadResponse.json()\n          const errStr = errors.map((error: any) => error.message).join(', ')\n          throw new Error(errStr)\n        }\n      })\n      .catch((error) => {\n        toast.error(`Failed to generate: ${error.message}`)\n        console.error(\n          'Error generating or setting your upload, please set it manually if its saved in your media files.',\n          error,\n        )\n      })\n  }, [getData, localFromContext?.code, instructionIdRef, setValue, documentId, collectionSlug])\n\n  const generate = useCallback(\n    async (options?: ActionCallbackParams) => {\n      if ((field as any)?.type === 'upload') {\n        return generateUpload()\n      }\n      // All supported types use structured object generation when schema is provided server-side\n      return streamObject(options ?? { action: 'Compose' })\n    },\n    [generateUpload, streamObject, field],\n  )\n\n  const stop = useCallback(() => {\n    console.log('Stopping...')\n    objectStop()\n  }, [objectStop])\n\n  return {\n    generate,\n    isJobActive,\n    isLoading: loadingObject,\n    jobProgress,\n    jobStatus,\n    stop,\n  }\n}\n"],"names":["experimental_useObject","useObject","useEditorConfigContext","toast","useConfig","useDocumentInfo","useField","useForm","useLocale","jsonSchema","useCallback","useEffect","useMemo","useRef","useState","PLUGIN_AI_JOBS_TABLE","PLUGIN_API_ENDPOINT_GENERATE","PLUGIN_API_ENDPOINT_GENERATE_UPLOAD","PLUGIN_INSTRUCTIONS_TABLE","PLUGIN_NAME","useFieldProps","editorSchemaValidator","fieldToJsonSchema","setSafeLexicalState","useHistory","useGenerate","instructionId","instructionIdRef","current","field","path","pathFromContext","editorConfigContext","editor","config","routes","api","serverURL","setValue","set","setHistory","jobStatus","setJobStatus","undefined","jobProgress","setJobProgress","isJobActive","setIsJobActive","getData","id","documentId","collectionSlug","localFromContext","collections","collection","find","slug","custom","editorConfig","admin","schema","editorSchema","memoizedValidator","memoizedSchema","validate","value","isValid","success","error","Error","activeSchema","f","fieldType","type","name","schemaJson","Object","keys","length","isLoading","loadingObject","object","stop","objectStop","submit","onError","message","console","onFinish","result","log","requestAnimationFrame","streamObject","action","params","doc","currentInstructionId","options","actionParams","allowedEditorNodes","Array","from","_nodes","locale","code","generateUpload","fetch","body","JSON","stringify","credentials","headers","method","then","uploadResponse","ok","json","job","cancelled","attempts","maxAttempts","poll","res","jobDoc","progress","result_id","status","setTimeout","e","errors","errStr","map","join","catch","generate"],"mappings":"AAAA,SAASA,0BAA0BC,SAAS,QAAQ,gBAAe;AACnE,SAASC,sBAAsB,QAAQ,sCAAqC;AAC5E,SAASC,KAAK,EAAEC,SAAS,EAAEC,eAAe,EAAEC,QAAQ,EAAEC,OAAO,EAAEC,SAAS,QAAQ,iBAAgB;AAChG,SAASC,UAAU,QAAQ,KAAI;AAC/B,SAASC,WAAW,EAAEC,SAAS,EAAEC,OAAO,EAAEC,MAAM,EAAEC,QAAQ,QAAQ,QAAO;AAIzE,SACEC,oBAAoB,EACpBC,4BAA4B,EAC5BC,mCAAmC,EACnCC,yBAAyB,EACzBC,WAAW,QACN,uBAAsB;AAC7B,SAASC,aAAa,QAAQ,oDAAmD;AACjF,SAASC,qBAAqB,QAAQ,8CAA6C;AACnF,SAASC,iBAAiB,QAAQ,0CAAyC;AAC3E,SAASC,mBAAmB,QAAQ,4CAA2C;AAC/E,SAASC,UAAU,QAAQ,kBAAiB;AAI5C,OAAO,MAAMC,cAAc,CAAC,EAAEC,aAAa,EAA6B;IACtE,iDAAiD;IACjD,MAAMC,mBAAmBd,OAAOa;IAEhC,gDAAgD;IAChDf,UAAU;QACRgB,iBAAiBC,OAAO,GAAGF;IAC7B,GAAG;QAACA;KAAc;IAElB,MAAM,EAAEG,KAAK,EAAEC,MAAMC,eAAe,EAAE,GAAGX;IACzC,MAAMY,sBAAsB9B;IAE5B,MAAM,EAAE+B,MAAM,EAAE,GAAGD;IAEnB,MAAM,EAAEE,MAAM,EAAE,GAAG9B;IACnB,MAAM,EACJ+B,QAAQ,EAAEC,GAAG,EAAE,EACfC,SAAS,EACV,GAAGH;IAEJ,MAAM,EAAEI,QAAQ,EAAE,GAAGhC,SAAc;QACjCwB,MAAMC,mBAAmB;IAC3B;IAEA,MAAM,EAAEQ,KAAKC,UAAU,EAAE,GAAGhB;IAE5B,qBAAqB;IACrB,MAAM,CAACiB,WAAWC,aAAa,GAAG5B,SAA6B6B;IAC/D,MAAM,CAACC,aAAaC,eAAe,GAAG/B,SAAiB;IACvD,MAAM,CAACgC,aAAaC,eAAe,GAAGjC,SAAkB;IAExD,MAAM,EAAEkC,OAAO,EAAE,GAAGzC;IACpB,MAAM,EAAE0C,IAAIC,UAAU,EAAEC,cAAc,EAAE,GAAG9C;IAE3C,MAAM+C,mBAAmB5C;IAEzB,6DAA6D;IAC7D,MAAM,EAAE6C,WAAW,EAAE,GAAGnB;IAExB,MAAMoB,aAAaD,YAAYE,IAAI,CAAC,CAACD,aAAeA,WAAWE,IAAI,KAAKtC;IACxE,MAAM,EAAEuC,QAAQ,EAAE,CAACtC,YAAY,EAAE,EAAEuC,eAAe,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,EAAE,GAAGJ,YAAYK,SAAS,CAAC;IAC7F,MAAM,EAAEC,QAAQC,eAAe,CAAC,CAAC,EAAE,GAAGH;IAEtC,MAAMI,oBAAoBlD,QAAQ;QAChC,OAAOS,sBAAsBwC;IAC/B,GAAG;QAACA;KAAa;IAEjB,MAAME,iBAAiBnD,QACrB,IACEH,WAAWoD,cAAc;YACvBG,UAAU,CAACC;gBACT,MAAMC,UAAUJ,kBAAkBG;gBAElC,IAAIC,SAAS;oBACX,OAAO;wBACLC,SAAS;wBACTF;oBACF;gBACF,OAAO;oBACL,OAAO;wBACLG,OAAO,IAAIC,MAAM;wBACjBF,SAAS;oBACX;gBACF;YACF;QACF,IACF;QAACL;KAAkB;IAGrB,uDAAuD;IACvD,MAAMQ,eAAe1D,QAAQ;QAC3B,MAAM2D,IAAI1C;QACV,MAAM2C,YAAYD,GAAGE;QACrB,IAAID,cAAc,YAAY;YAC5B,OAAOT;QACT;QACA,IAAIQ,KAAKA,EAAEG,IAAI,IAAIF,WAAW;YAC5B,MAAMG,aAAarD,kBAAkBiD;YACrC,IAAII,cAAcC,OAAOC,IAAI,CAACF,YAAYG,MAAM,GAAG,GAAG;gBACpD,OAAOrE,WAAWkE;YACpB;QACF;QACA,OAAOhC;IACT,GAAG;QAACd;QAAOkC;KAAe;IAE1B,MAAM,EACJgB,WAAWC,aAAa,EACxBC,MAAM,EACNC,MAAMC,UAAU,EAChBC,MAAM,EACP,GAAGnF,UAAU;QACZmC,KAAK,CAAC,IAAI,EAAEpB,6BAA6B,CAAC;QAC1CqE,SAAS,CAACjB;YACRjE,MAAMiE,KAAK,CAAC,CAAC,oBAAoB,EAAEA,MAAMkB,OAAO,CAAC,CAAC;YAClDC,QAAQnB,KAAK,CAAC,4BAA4BA;QAC5C;QACAoB,UAAU,CAACC;YACT,IAAIA,OAAOR,MAAM,IAAIpD,OAAO;gBAC1B,IAAIA,MAAM4C,IAAI,KAAK,YAAY;oBAC7BjC,WAAWiD,OAAOR,MAAM;oBACxB3C,SAASmD,OAAOR,MAAM;gBACxB,OAAO,IAAI,UAAUpD,OAAO;oBAC1BW,WAAWiD,OAAOR,MAAM,CAACpD,MAAM6C,IAAI,CAAC;oBACpCpC,SAASmD,OAAOR,MAAM,CAACpD,MAAM6C,IAAI,CAAC;gBACpC;YACF,OAAO;gBACLa,QAAQG,GAAG,CAAC,4BAA4BD,QAAQ5D;YAClD;QACF;QACA+B,QAAQU;IACV;IAEA3D,UAAU;QACR,IAAI,CAACsE,QAAQ;YACX;QACF;QAEAU,sBAAsB;YACpB,IAAI9D,OAAO4C,SAAS,YAAY;gBAChClD,oBAAoB0D,QAAQhD;YAC5B,OAAO,IAAIJ,SAAS,UAAUA,SAASoD,MAAM,CAACpD,MAAM6C,IAAI,CAAC,EAAE;gBACzDpC,SAAS2C,MAAM,CAACpD,MAAM6C,IAAI,CAAC;YAC7B;QACF;IACF,GAAG;QAACO;QAAQhD;QAAQJ;QAAOS;KAAS;IAEpC,MAAMsD,eAAelF,YACnB,CAAC,EAAEmF,SAAS,SAAS,EAAEC,MAAM,EAAwB;QACnD,MAAMC,MAAM/C;QAEZ,MAAMgD,uBAAuBrE,iBAAiBC,OAAO;QAErD,MAAMqE,UAAU;YACdJ;YACAK,cAAcJ;YACdpE,eAAesE;QACjB;QAEAZ,OAAO;YACLe,oBAAoBC,MAAMC,IAAI,CAACpE,QAAQqE,QAAQzB,UAAU,EAAE;YAC3DkB,KAAK;gBACH,GAAGA,GAAG;gBACN9C,IAAIC;YACN;YACAqD,QAAQnD,kBAAkBoD;YAC1BP;QACF;IACF,GACA;QAAC7C,kBAAkBoD;QAAM7E;QAAkBuB;KAAW;IAGxD,MAAMuD,iBAAiB/F,YAAY;QACjC,MAAMqF,MAAM/C;QACZ,MAAMgD,uBAAuBrE,iBAAiBC,OAAO;QAErD,OAAO8E,MAAM,CAAC,EAAErE,UAAU,EAAED,IAAI,EAAEnB,oCAAoC,CAAC,EAAE;YACvE0F,MAAMC,KAAKC,SAAS,CAAC;gBACnB1D,gBAAgBA,kBAAkB;gBAClC4C;gBACA7C;gBACAqD,QAAQnD,kBAAkBoD;gBAC1BP,SAAS;oBACPvE,eAAesE;gBACjB;YACF;YACAc,aAAa;YACbC,SAAS;gBACP,gBAAgB;YAClB;YACAC,QAAQ;QACV,GACGC,IAAI,CAAC,OAAOC;YACX,IAAIA,eAAeC,EAAE,EAAE;gBACrB,MAAMC,OAAO,MAAMF,eAAeE,IAAI;gBACtC,MAAM,EAAEC,GAAG,EAAE5B,MAAM,EAAE,GAAG2B,QAAQ,CAAC;gBACjC,IAAI3B,QAAQ;oBACV,oBAAoB;oBACpBnD,SAASmD,QAAQxC;oBACjBT,WAAWiD,QAAQxC;oBAEnB,oCAAoC;oBACpC9C,MAAMgE,OAAO,CAAC;oBAEd,OAAO+C;gBACT;gBAEA,mEAAmE;gBACnE,IAAIG,OAAOA,IAAIpE,EAAE,EAAE;oBACjBF,eAAe;oBACf,MAAMuE,YAAY;oBAClB,IAAIC,WAAW;oBACf,MAAMC,cAAc,IAAI,yBAAyB;;oBAEjD,iFAAiF;oBACjF,MAAMC,OAAO;wBACX,IAAIH,WAAW;4BACb;wBACF;wBACA,IAAI;4BACF,MAAMI,MAAM,MAAMhB,MAChB,CAAC,EAAErE,UAAU,EAAED,IAAI,CAAC,EAAErB,qBAAqB,CAAC,EAAEsG,IAAIpE,EAAE,CAAC,CAAC,EACtD;gCAAE6D,aAAa;4BAAU;4BAE3B,IAAIY,IAAIP,EAAE,EAAE;gCACV,MAAMQ,SAAS,MAAMD,IAAIN,IAAI;gCAC7B,MAAM,EAAEQ,QAAQ,EAAEC,SAAS,EAAEC,MAAM,EAAE,GAAGH,UAAU,CAAC;gCACnDjF,aAAaoF;gCACbjF,eAAe+E,YAAY;gCAC3B,4CAA4C;gCAC5C,IAAIE,WAAW,eAAeD,WAAW;oCACvC,gEAAgE;oCAChEvF,SAAS;oCACTyF,WAAW;wCACTzF,SAASuF;oCACX,GAAG;oCACHrF,WAAWqF;oCACX9E,eAAe;oCACf;gCACF;gCACA,IAAI+E,WAAW,UAAU;oCACvB/E,eAAe;oCACf,MAAM,IAAIsB,MAAM;gCAClB;4BACF;wBACF,EAAE,OAAO2D,GAAG;wBACV,eAAe;wBACjB;wBAEAT,YAAY;wBACZ,IAAI,CAACD,aAAaC,WAAWC,aAAa;4BACxCO,WAAWN,MAAM;wBACnB;oBACF;oBACAM,WAAWN,MAAM;oBACjB,OAAOP;gBACT;gBAEA,MAAM,IAAI7C,MAAM;YAClB,OAAO;gBACL,MAAM,EAAE4D,SAAS,EAAE,EAAE,GAAG,MAAMf,eAAeE,IAAI;gBACjD,MAAMc,SAASD,OAAOE,GAAG,CAAC,CAAC/D,QAAeA,MAAMkB,OAAO,EAAE8C,IAAI,CAAC;gBAC9D,MAAM,IAAI/D,MAAM6D;YAClB;QACF,GACCG,KAAK,CAAC,CAACjE;YACNjE,MAAMiE,KAAK,CAAC,CAAC,oBAAoB,EAAEA,MAAMkB,OAAO,CAAC,CAAC;YAClDC,QAAQnB,KAAK,CACX,qGACAA;QAEJ;IACJ,GAAG;QAACpB;QAASI,kBAAkBoD;QAAM7E;QAAkBW;QAAUY;QAAYC;KAAe;IAE5F,MAAMmF,WAAW5H,YACf,OAAOuF;QACL,IAAI,AAACpE,OAAe4C,SAAS,UAAU;YACrC,OAAOgC;QACT;QACA,2FAA2F;QAC3F,OAAOb,aAAaK,WAAW;YAAEJ,QAAQ;QAAU;IACrD,GACA;QAACY;QAAgBb;QAAc/D;KAAM;IAGvC,MAAMqD,OAAOxE,YAAY;QACvB6E,QAAQG,GAAG,CAAC;QACZP;IACF,GAAG;QAACA;KAAW;IAEf,OAAO;QACLmD;QACAxF;QACAiC,WAAWC;QACXpC;QACAH;QACAyC;IACF;AACF,EAAC"}
+{"version":3,"sources":["../../../../src/ui/Compose/hooks/useGenerate.ts"],"sourcesContent":["import { experimental_useObject as useObject } from '@ai-sdk/react'\nimport { useEditorConfigContext } from '@payloadcms/richtext-lexical/client'\nimport { toast, useDocumentInfo, useField, useForm, useLocale } from '@payloadcms/ui'\nimport { jsonSchema } from 'ai'\nimport { useCallback, useEffect, useRef } from 'react'\n\nimport type { ActionMenuItems } from '../../../types.js'\n\nimport { PLUGIN_API_ENDPOINT_GENERATE } from '../../../defaults.js'\nimport { useFieldProps } from '../../../providers/FieldProvider/useFieldProps.js'\nimport { setSafeLexicalState } from '../../../utilities/setSafeLexicalState.js'\nimport { useGenerateUpload } from './useGenerateUpload.js'\nimport { useHistory } from './useHistory.js'\nimport { useStreamingUpdate } from './useStreamingUpdate.js'\n\ntype ActionCallbackParams = { action: ActionMenuItems; params?: unknown }\n\nexport const useGenerate = ({ instructionId }: { instructionId: string }) => {\n  // Create a ref to hold the current instructionId\n  const instructionIdRef = useRef(instructionId)\n\n  // Update the ref whenever instructionId changes\n  useEffect(() => {\n    instructionIdRef.current = instructionId\n  }, [instructionId])\n\n  const { field, path: pathFromContext } = useFieldProps()\n  const editorConfigContext = useEditorConfigContext()\n  const { setValue } = useField<any>({\n    path: pathFromContext ?? '',\n  })\n  const { set: setHistory } = useHistory()\n  const { getData } = useForm()\n  const { id: documentId } = useDocumentInfo()\n  const localFromContext = useLocale()\n\n  const {\n    isLoading: loadingObject,\n    object,\n    stop: objectStop,\n    submit,\n  } = useObject({\n    api: `/api${PLUGIN_API_ENDPOINT_GENERATE}`,\n    onError: (error: any) => {\n      toast.error(`Failed to generate: ${error.message}`)\n      console.error('Error generating object:', error)\n    },\n    onFinish: (result) => {\n      if (result.object && field) {\n        if (field.type === 'richText') {\n          setHistory(result.object)\n          setSafeLexicalState(result.object, editor)\n        } else if ('name' in field && result.object[field.name]) {\n          setHistory(result.object[field.name])\n          setValue(result.object[field.name])\n        }\n      } else {\n        console.log('onFinish: result, field ', result, field)\n      }\n    },\n    schema: jsonSchema({\n      type: 'object',\n      additionalProperties: true,\n      properties: {},\n    }) as any,\n  })\n\n  const { editor } = editorConfigContext\n\n  // Hook: Handle high-frequency streaming updates\n  useStreamingUpdate({\n    editor,\n    isLoading: loadingObject,\n    object,\n  })\n\n  // Hook 2: Handle Upload generation and polling\n  const { generateUpload, isJobActive, jobProgress, jobStatus } = useGenerateUpload({\n    instructionIdRef,\n  })\n\n  const streamObject = useCallback(\n    ({ action = 'Compose', params }: ActionCallbackParams) => {\n      const doc = getData()\n\n      const currentInstructionId = instructionIdRef.current\n\n      const options = {\n        action,\n        actionParams: params,\n        instructionId: currentInstructionId,\n      }\n\n      submit({\n        allowedEditorNodes: Array.from(editor?._nodes?.keys() || []),\n        doc: {\n          ...doc,\n          id: documentId,\n        },\n        locale: localFromContext?.code,\n        options,\n      })\n    },\n    [localFromContext?.code, instructionIdRef, documentId, getData, submit, editor],\n  )\n\n  const generate = useCallback(\n    async (options?: ActionCallbackParams) => {\n      if ((field as any)?.type === 'upload') {\n        return generateUpload()\n      }\n      // All supported types use structured object generation when schema is provided server-side\n      return streamObject(options ?? { action: 'Compose' })\n    },\n    [generateUpload, streamObject, field],\n  )\n\n  const stop = useCallback(() => {\n    console.log('Stopping...')\n    objectStop()\n  }, [objectStop])\n\n  return {\n    generate,\n    isJobActive,\n    isLoading: loadingObject,\n    jobProgress,\n    jobStatus,\n    stop,\n  }\n}\n"],"names":["experimental_useObject","useObject","useEditorConfigContext","toast","useDocumentInfo","useField","useForm","useLocale","jsonSchema","useCallback","useEffect","useRef","PLUGIN_API_ENDPOINT_GENERATE","useFieldProps","setSafeLexicalState","useGenerateUpload","useHistory","useStreamingUpdate","useGenerate","instructionId","instructionIdRef","current","field","path","pathFromContext","editorConfigContext","setValue","set","setHistory","getData","id","documentId","localFromContext","isLoading","loadingObject","object","stop","objectStop","submit","api","onError","error","message","console","onFinish","result","type","editor","name","log","schema","additionalProperties","properties","generateUpload","isJobActive","jobProgress","jobStatus","streamObject","action","params","doc","currentInstructionId","options","actionParams","allowedEditorNodes","Array","from","_nodes","keys","locale","code","generate"],"mappings":"AAAA,SAASA,0BAA0BC,SAAS,QAAQ,gBAAe;AACnE,SAASC,sBAAsB,QAAQ,sCAAqC;AAC5E,SAASC,KAAK,EAAEC,eAAe,EAAEC,QAAQ,EAAEC,OAAO,EAAEC,SAAS,QAAQ,iBAAgB;AACrF,SAASC,UAAU,QAAQ,KAAI;AAC/B,SAASC,WAAW,EAAEC,SAAS,EAAEC,MAAM,QAAQ,QAAO;AAItD,SAASC,4BAA4B,QAAQ,uBAAsB;AACnE,SAASC,aAAa,QAAQ,oDAAmD;AACjF,SAASC,mBAAmB,QAAQ,4CAA2C;AAC/E,SAASC,iBAAiB,QAAQ,yBAAwB;AAC1D,SAASC,UAAU,QAAQ,kBAAiB;AAC5C,SAASC,kBAAkB,QAAQ,0BAAyB;AAI5D,OAAO,MAAMC,cAAc,CAAC,EAAEC,aAAa,EAA6B;IACtE,iDAAiD;IACjD,MAAMC,mBAAmBT,OAAOQ;IAEhC,gDAAgD;IAChDT,UAAU;QACRU,iBAAiBC,OAAO,GAAGF;IAC7B,GAAG;QAACA;KAAc;IAElB,MAAM,EAAEG,KAAK,EAAEC,MAAMC,eAAe,EAAE,GAAGX;IACzC,MAAMY,sBAAsBvB;IAC5B,MAAM,EAAEwB,QAAQ,EAAE,GAAGrB,SAAc;QACjCkB,MAAMC,mBAAmB;IAC3B;IACA,MAAM,EAAEG,KAAKC,UAAU,EAAE,GAAGZ;IAC5B,MAAM,EAAEa,OAAO,EAAE,GAAGvB;IACpB,MAAM,EAAEwB,IAAIC,UAAU,EAAE,GAAG3B;IAC3B,MAAM4B,mBAAmBzB;IAEzB,MAAM,EACJ0B,WAAWC,aAAa,EACxBC,MAAM,EACNC,MAAMC,UAAU,EAChBC,MAAM,EACP,GAAGrC,UAAU;QACZsC,KAAK,CAAC,IAAI,EAAE3B,6BAA6B,CAAC;QAC1C4B,SAAS,CAACC;YACRtC,MAAMsC,KAAK,CAAC,CAAC,oBAAoB,EAAEA,MAAMC,OAAO,CAAC,CAAC;YAClDC,QAAQF,KAAK,CAAC,4BAA4BA;QAC5C;QACAG,UAAU,CAACC;YACT,IAAIA,OAAOV,MAAM,IAAIb,OAAO;gBAC1B,IAAIA,MAAMwB,IAAI,KAAK,YAAY;oBAC7BlB,WAAWiB,OAAOV,MAAM;oBACxBrB,oBAAoB+B,OAAOV,MAAM,EAAEY;gBACrC,OAAO,IAAI,UAAUzB,SAASuB,OAAOV,MAAM,CAACb,MAAM0B,IAAI,CAAC,EAAE;oBACvDpB,WAAWiB,OAAOV,MAAM,CAACb,MAAM0B,IAAI,CAAC;oBACpCtB,SAASmB,OAAOV,MAAM,CAACb,MAAM0B,IAAI,CAAC;gBACpC;YACF,OAAO;gBACLL,QAAQM,GAAG,CAAC,4BAA4BJ,QAAQvB;YAClD;QACF;QACA4B,QAAQ1C,WAAW;YACjBsC,MAAM;YACNK,sBAAsB;YACtBC,YAAY,CAAC;QACf;IACF;IAEA,MAAM,EAAEL,MAAM,EAAE,GAAGtB;IAEnB,gDAAgD;IAChDR,mBAAmB;QACjB8B;QACAd,WAAWC;QACXC;IACF;IAEA,+CAA+C;IAC/C,MAAM,EAAEkB,cAAc,EAAEC,WAAW,EAAEC,WAAW,EAAEC,SAAS,EAAE,GAAGzC,kBAAkB;QAChFK;IACF;IAEA,MAAMqC,eAAehD,YACnB,CAAC,EAAEiD,SAAS,SAAS,EAAEC,MAAM,EAAwB;QACnD,MAAMC,MAAM/B;QAEZ,MAAMgC,uBAAuBzC,iBAAiBC,OAAO;QAErD,MAAMyC,UAAU;YACdJ;YACAK,cAAcJ;YACdxC,eAAe0C;QACjB;QAEAvB,OAAO;YACL0B,oBAAoBC,MAAMC,IAAI,CAACnB,QAAQoB,QAAQC,UAAU,EAAE;YAC3DR,KAAK;gBACH,GAAGA,GAAG;gBACN9B,IAAIC;YACN;YACAsC,QAAQrC,kBAAkBsC;YAC1BR;QACF;IACF,GACA;QAAC9B,kBAAkBsC;QAAMlD;QAAkBW;QAAYF;QAASS;QAAQS;KAAO;IAGjF,MAAMwB,WAAW9D,YACf,OAAOqD;QACL,IAAI,AAACxC,OAAewB,SAAS,UAAU;YACrC,OAAOO;QACT;QACA,2FAA2F;QAC3F,OAAOI,aAAaK,WAAW;YAAEJ,QAAQ;QAAU;IACrD,GACA;QAACL;QAAgBI;QAAcnC;KAAM;IAGvC,MAAMc,OAAO3B,YAAY;QACvBkC,QAAQM,GAAG,CAAC;QACZZ;IACF,GAAG;QAACA;KAAW;IAEf,OAAO;QACLkC;QACAjB;QACArB,WAAWC;QACXqB;QACAC;QACApB;IACF;AACF,EAAC"}

package/dist/ui/Compose/hooks/useGenerateUpload.d.ts

@@ -0,0 +1,11 @@
+import { type RefObject } from 'react';
+type UseGenerateUploadParams = {
+    instructionIdRef: RefObject<string>;
+};
+export declare const useGenerateUpload: ({ instructionIdRef }: UseGenerateUploadParams) => {
+    generateUpload: () => Promise<void | Response>;
+    isJobActive: boolean;
+    jobProgress: number;
+    jobStatus: string | undefined;
+};
+export {};

package/dist/ui/Compose/hooks/useGenerateUpload.js

@@ -0,0 +1,156 @@
+import { toast, useConfig, useDocumentInfo, useField, useForm, useLocale } from '@payloadcms/ui';
+import { useCallback, useState } from 'react';
+import { PLUGIN_AI_JOBS_TABLE, PLUGIN_API_ENDPOINT_GENERATE_UPLOAD } from '../../../defaults.js';
+import { useFieldProps } from '../../../providers/FieldProvider/useFieldProps.js';
+import { useHistory } from './useHistory.js';
+export const useGenerateUpload = ({ instructionIdRef })=>{
+    const { config } = useConfig();
+    const { routes: { api }, serverURL } = config;
+    const { id: documentId, collectionSlug } = useDocumentInfo();
+    const localFromContext = useLocale();
+    const { getData } = useForm();
+    const { set: setHistory } = useHistory();
+    const { field, path: pathFromContext } = useFieldProps();
+    const { setValue } = useField({
+        path: pathFromContext ?? ''
+    });
+    // Async job UI state
+    const [jobStatus, setJobStatus] = useState(undefined);
+    const [jobProgress, setJobProgress] = useState(0);
+    const [isJobActive, setIsJobActive] = useState(false);
+    const generateUpload = useCallback(async ()=>{
+        const doc = getData();
+        const currentInstructionId = instructionIdRef.current;
+        return fetch(`${serverURL}${api}${PLUGIN_API_ENDPOINT_GENERATE_UPLOAD}`, {
+            body: JSON.stringify({
+                collectionSlug: collectionSlug ?? '',
+                doc,
+                documentId,
+                locale: localFromContext?.code,
+                options: {
+                    instructionId: currentInstructionId
+                }
+            }),
+            credentials: 'include',
+            headers: {
+                'Content-Type': 'application/json'
+            },
+            method: 'POST'
+        }).then(async (uploadResponse)=>{
+            if (uploadResponse.ok) {
+                const json = await uploadResponse.json();
+                const { job, result } = json || {};
+                if (result) {
+                    if (Array.isArray(result)) {
+                        const ids = result.map((r)=>r.id);
+                        setValue(ids);
+                        setHistory(ids);
+                    } else {
+                        setValue(result?.id);
+                        setHistory(result?.id);
+                    }
+                    // Show toast to prompt user to save
+                    toast.success('Image generated successfully! Click Save to see the preview.');
+                    return uploadResponse;
+                }
+                // Async job: poll AI Jobs collection for status/progress/result_id
+                if (job && job.id) {
+                    setIsJobActive(true);
+                    const cancelled = false;
+                    let attempts = 0;
+                    const maxAttempts = 600 // up to ~10 minutes @ 1s
+                    ;
+                    // Basic in-hook state via closure variables; UI will re-render off fetches below
+                    const poll = async ()=>{
+                        if (cancelled) {
+                            return;
+                        }
+                        try {
+                            const res = await fetch(`${serverURL}${api}/${PLUGIN_AI_JOBS_TABLE}/${job.id}`, {
+                                credentials: 'include'
+                            });
+                            if (res.ok) {
+                                const jobDoc = await res.json();
+                                const { progress, result_id, status } = jobDoc || {};
+                                setJobStatus(status);
+                                setJobProgress(progress ?? 0);
+                                // When result present, set field and finish
+                                if (status === 'completed' && result_id) {
+                                    let valueToSet = result_id;
+                                    // Attempt to fetch full document for immediate preview
+                                    if (field && 'relationTo' in field && typeof field.relationTo === 'string') {
+                                        let attempts = 0;
+                                        const maxAttempts = 3;
+                                        while(attempts < maxAttempts){
+                                            try {
+                                                const docRes = await fetch(`${serverURL}${api}/${field.relationTo}/${result_id}`, {
+                                                    credentials: 'include'
+                                                });
+                                                if (docRes.ok) {
+                                                    const doc = await docRes.json();
+                                                    // Verify we have a URL for preview
+                                                    if (doc && doc.url) {
+                                                        valueToSet = doc;
+                                                        break;
+                                                    }
+                                                }
+                                            } catch (e) {
+                                                console.error('Failed to fetch generated document for preview:', e);
+                                            }
+                                            attempts++;
+                                            if (attempts < maxAttempts) {
+                                                await new Promise((resolve)=>setTimeout(resolve, 500));
+                                            }
+                                        }
+                                    }
+                                    setValue(valueToSet);
+                                    setHistory(result_id);
+                                    setIsJobActive(false);
+                                    return;
+                                }
+                                if (status === 'failed') {
+                                    setIsJobActive(false);
+                                    throw new Error('Video generation failed');
+                                }
+                            }
+                        } catch (_) {
+                        // silent retry
+                        }
+                        attempts += 1;
+                        if (!cancelled && attempts < maxAttempts) {
+                            setTimeout(poll, 1000);
+                        }
+                    };
+                    setTimeout(poll, 1000);
+                    return uploadResponse;
+                }
+                throw new Error('generateUpload: Unexpected response');
+            } else {
+                const { errors = [] } = await uploadResponse.json();
+                const errStr = errors.map((error)=>error.message).join(', ');
+                throw new Error(errStr);
+            }
+        }).catch((error)=>{
+            toast.error(`Failed to generate: ${error.message}`);
+            console.error('Error generating or setting your upload, please set it manually if its saved in your media files.', error);
+        });
+    }, [
+        getData,
+        localFromContext?.code,
+        instructionIdRef,
+        // setValue,
+        documentId,
+        collectionSlug,
+        serverURL,
+        api,
+        setHistory
+    ]);
+    return {
+        generateUpload,
+        isJobActive,
+        jobProgress,
+        jobStatus
+    };
+};
+
+//# sourceMappingURL=useGenerateUpload.js.map

package/dist/ui/Compose/hooks/useGenerateUpload.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../../src/ui/Compose/hooks/useGenerateUpload.ts"],"sourcesContent":["import { toast, useConfig, useDocumentInfo, useField, useForm, useLocale } from '@payloadcms/ui'\nimport { type RefObject, useCallback, useState } from 'react'\n\nimport type { GenerateTextarea } from '../../../types.js'\n\nimport { PLUGIN_AI_JOBS_TABLE, PLUGIN_API_ENDPOINT_GENERATE_UPLOAD } from '../../../defaults.js'\nimport { useFieldProps } from '../../../providers/FieldProvider/useFieldProps.js'\nimport { useHistory } from './useHistory.js'\n\ntype UseGenerateUploadParams = {\n  instructionIdRef: RefObject<string>\n}\n\nexport const useGenerateUpload = ({ instructionIdRef }: UseGenerateUploadParams) => {\n  const { config } = useConfig()\n  const {\n    routes: { api },\n    serverURL,\n  } = config\n  const { id: documentId, collectionSlug } = useDocumentInfo()\n  const localFromContext = useLocale()\n  const { getData } = useForm()\n  const { set: setHistory } = useHistory()\n\n  const { field, path: pathFromContext } = useFieldProps()\n  const { setValue } = useField<any>({\n    path: pathFromContext ?? '',\n  })\n\n  // Async job UI state\n  const [jobStatus, setJobStatus] = useState<string | undefined>(undefined)\n  const [jobProgress, setJobProgress] = useState<number>(0)\n  const [isJobActive, setIsJobActive] = useState<boolean>(false)\n\n  const generateUpload = useCallback(async () => {\n    const doc = getData()\n    const currentInstructionId = instructionIdRef.current\n\n    return fetch(`${serverURL}${api}${PLUGIN_API_ENDPOINT_GENERATE_UPLOAD}`, {\n      body: JSON.stringify({\n        collectionSlug: collectionSlug ?? '',\n        doc,\n        documentId,\n        locale: localFromContext?.code,\n        options: {\n          instructionId: currentInstructionId,\n        },\n      } satisfies Parameters<GenerateTextarea>[0]),\n      credentials: 'include',\n      headers: {\n        'Content-Type': 'application/json',\n      },\n      method: 'POST',\n    })\n      .then(async (uploadResponse) => {\n        if (uploadResponse.ok) {\n          const json = await uploadResponse.json()\n          const { job, result } = json || {}\n          if (result) {\n            if (Array.isArray(result)) {\n              const ids = result.map((r: any) => r.id)\n              setValue(ids)\n              setHistory(ids)\n            } else {\n              setValue(result?.id)\n              setHistory(result?.id)\n            }\n\n            // Show toast to prompt user to save\n            toast.success('Image generated successfully! Click Save to see the preview.')\n\n            return uploadResponse\n          }\n\n          // Async job: poll AI Jobs collection for status/progress/result_id\n          if (job && job.id) {\n            setIsJobActive(true)\n            const cancelled = false\n            let attempts = 0\n            const maxAttempts = 600 // up to ~10 minutes @ 1s\n\n            // Basic in-hook state via closure variables; UI will re-render off fetches below\n            const poll = async (): Promise<void> => {\n              if (cancelled) {\n                return\n              }\n              try {\n                const res = await fetch(`${serverURL}${api}/${PLUGIN_AI_JOBS_TABLE}/${job.id}`, {\n                  credentials: 'include',\n                })\n                if (res.ok) {\n                  const jobDoc = await res.json()\n                  const { progress, result_id, status } = jobDoc || {}\n                  setJobStatus(status)\n                  setJobProgress(progress ?? 0)\n                  // When result present, set field and finish\n                  if (status === 'completed' && result_id) {\n                    let valueToSet = result_id\n\n                    // Attempt to fetch full document for immediate preview\n                    if (field && 'relationTo' in field && typeof field.relationTo === 'string') {\n                      let attempts = 0\n                      const maxAttempts = 3\n                      while (attempts < maxAttempts) {\n                        try {\n                          const docRes = await fetch(\n                            `${serverURL}${api}/${field.relationTo}/${result_id}`,\n                            {\n                              credentials: 'include',\n                            },\n                          )\n                          if (docRes.ok) {\n                            const doc = await docRes.json()\n                            // Verify we have a URL for preview\n                            if (doc && doc.url) {\n                              valueToSet = doc\n                              break\n                            }\n                          }\n                        } catch (e) {\n                          console.error('Failed to fetch generated document for preview:', e)\n                        }\n                        attempts++\n                        if (attempts < maxAttempts) {\n                          await new Promise((resolve) => setTimeout(resolve, 500))\n                        }\n                      }\n                    }\n\n                    setValue(valueToSet)\n                    setHistory(result_id)\n                    setIsJobActive(false)\n                    return\n                  }\n                  if (status === 'failed') {\n                    setIsJobActive(false)\n                    throw new Error('Video generation failed')\n                  }\n                }\n              } catch (_) {\n                // silent retry\n              }\n\n              attempts += 1\n              if (!cancelled && attempts < maxAttempts) {\n                setTimeout(poll, 1000)\n              }\n            }\n            setTimeout(poll, 1000)\n            return uploadResponse\n          }\n\n          throw new Error('generateUpload: Unexpected response')\n        } else {\n          const { errors = [] } = await uploadResponse.json()\n          const errStr = errors.map((error: any) => error.message).join(', ')\n          throw new Error(errStr)\n        }\n      })\n      .catch((error) => {\n        toast.error(`Failed to generate: ${error.message}`)\n        console.error(\n          'Error generating or setting your upload, please set it manually if its saved in your media files.',\n          error,\n        )\n      })\n  }, [\n    getData,\n    localFromContext?.code,\n    instructionIdRef,\n    // setValue,\n    documentId,\n    collectionSlug,\n    serverURL,\n    api,\n    setHistory,\n  ])\n\n  return {\n    generateUpload,\n    isJobActive,\n    jobProgress,\n    jobStatus,\n  }\n}\n"],"names":["toast","useConfig","useDocumentInfo","useField","useForm","useLocale","useCallback","useState","PLUGIN_AI_JOBS_TABLE","PLUGIN_API_ENDPOINT_GENERATE_UPLOAD","useFieldProps","useHistory","useGenerateUpload","instructionIdRef","config","routes","api","serverURL","id","documentId","collectionSlug","localFromContext","getData","set","setHistory","field","path","pathFromContext","setValue","jobStatus","setJobStatus","undefined","jobProgress","setJobProgress","isJobActive","setIsJobActive","generateUpload","doc","currentInstructionId","current","fetch","body","JSON","stringify","locale","code","options","instructionId","credentials","headers","method","then","uploadResponse","ok","json","job","result","Array","isArray","ids","map","r","success","cancelled","attempts","maxAttempts","poll","res","jobDoc","progress","result_id","status","valueToSet","relationTo","docRes","url","e","console","error","Promise","resolve","setTimeout","Error","_","errors","errStr","message","join","catch"],"mappings":"AAAA,SAASA,KAAK,EAAEC,SAAS,EAAEC,eAAe,EAAEC,QAAQ,EAAEC,OAAO,EAAEC,SAAS,QAAQ,iBAAgB;AAChG,SAAyBC,WAAW,EAAEC,QAAQ,QAAQ,QAAO;AAI7D,SAASC,oBAAoB,EAAEC,mCAAmC,QAAQ,uBAAsB;AAChG,SAASC,aAAa,QAAQ,oDAAmD;AACjF,SAASC,UAAU,QAAQ,kBAAiB;AAM5C,OAAO,MAAMC,oBAAoB,CAAC,EAAEC,gBAAgB,EAA2B;IAC7E,MAAM,EAAEC,MAAM,EAAE,GAAGb;IACnB,MAAM,EACJc,QAAQ,EAAEC,GAAG,EAAE,EACfC,SAAS,EACV,GAAGH;IACJ,MAAM,EAAEI,IAAIC,UAAU,EAAEC,cAAc,EAAE,GAAGlB;IAC3C,MAAMmB,mBAAmBhB;IACzB,MAAM,EAAEiB,OAAO,EAAE,GAAGlB;IACpB,MAAM,EAAEmB,KAAKC,UAAU,EAAE,GAAGb;IAE5B,MAAM,EAAEc,KAAK,EAAEC,MAAMC,eAAe,EAAE,GAAGjB;IACzC,MAAM,EAAEkB,QAAQ,EAAE,GAAGzB,SAAc;QACjCuB,MAAMC,mBAAmB;IAC3B;IAEA,qBAAqB;IACrB,MAAM,CAACE,WAAWC,aAAa,GAAGvB,SAA6BwB;IAC/D,MAAM,CAACC,aAAaC,eAAe,GAAG1B,SAAiB;IACvD,MAAM,CAAC2B,aAAaC,eAAe,GAAG5B,SAAkB;IAExD,MAAM6B,iBAAiB9B,YAAY;QACjC,MAAM+B,MAAMf;QACZ,MAAMgB,uBAAuBzB,iBAAiB0B,OAAO;QAErD,OAAOC,MAAM,CAAC,EAAEvB,UAAU,EAAED,IAAI,EAAEP,oCAAoC,CAAC,EAAE;YACvEgC,MAAMC,KAAKC,SAAS,CAAC;gBACnBvB,gBAAgBA,kBAAkB;gBAClCiB;gBACAlB;gBACAyB,QAAQvB,kBAAkBwB;gBAC1BC,SAAS;oBACPC,eAAeT;gBACjB;YACF;YACAU,aAAa;YACbC,SAAS;gBACP,gBAAgB;YAClB;YACAC,QAAQ;QACV,GACGC,IAAI,CAAC,OAAOC;YACX,IAAIA,eAAeC,EAAE,EAAE;gBACrB,MAAMC,OAAO,MAAMF,eAAeE,IAAI;gBACtC,MAAM,EAAEC,GAAG,EAAEC,MAAM,EAAE,GAAGF,QAAQ,CAAC;gBACjC,IAAIE,QAAQ;oBACV,IAAIC,MAAMC,OAAO,CAACF,SAAS;wBACzB,MAAMG,MAAMH,OAAOI,GAAG,CAAC,CAACC,IAAWA,EAAE3C,EAAE;wBACvCU,SAAS+B;wBACTnC,WAAWmC;oBACb,OAAO;wBACL/B,SAAS4B,QAAQtC;wBACjBM,WAAWgC,QAAQtC;oBACrB;oBAEA,oCAAoC;oBACpClB,MAAM8D,OAAO,CAAC;oBAEd,OAAOV;gBACT;gBAEA,mEAAmE;gBACnE,IAAIG,OAAOA,IAAIrC,EAAE,EAAE;oBACjBiB,eAAe;oBACf,MAAM4B,YAAY;oBAClB,IAAIC,WAAW;oBACf,MAAMC,cAAc,IAAI,yBAAyB;;oBAEjD,iFAAiF;oBACjF,MAAMC,OAAO;wBACX,IAAIH,WAAW;4BACb;wBACF;wBACA,IAAI;4BACF,MAAMI,MAAM,MAAM3B,MAAM,CAAC,EAAEvB,UAAU,EAAED,IAAI,CAAC,EAAER,qBAAqB,CAAC,EAAE+C,IAAIrC,EAAE,CAAC,CAAC,EAAE;gCAC9E8B,aAAa;4BACf;4BACA,IAAImB,IAAId,EAAE,EAAE;gCACV,MAAMe,SAAS,MAAMD,IAAIb,IAAI;gCAC7B,MAAM,EAAEe,QAAQ,EAAEC,SAAS,EAAEC,MAAM,EAAE,GAAGH,UAAU,CAAC;gCACnDtC,aAAayC;gCACbtC,eAAeoC,YAAY;gCAC3B,4CAA4C;gCAC5C,IAAIE,WAAW,eAAeD,WAAW;oCACvC,IAAIE,aAAaF;oCAEjB,uDAAuD;oCACvD,IAAI7C,SAAS,gBAAgBA,SAAS,OAAOA,MAAMgD,UAAU,KAAK,UAAU;wCAC1E,IAAIT,WAAW;wCACf,MAAMC,cAAc;wCACpB,MAAOD,WAAWC,YAAa;4CAC7B,IAAI;gDACF,MAAMS,SAAS,MAAMlC,MACnB,CAAC,EAAEvB,UAAU,EAAED,IAAI,CAAC,EAAES,MAAMgD,UAAU,CAAC,CAAC,EAAEH,UAAU,CAAC,EACrD;oDACEtB,aAAa;gDACf;gDAEF,IAAI0B,OAAOrB,EAAE,EAAE;oDACb,MAAMhB,MAAM,MAAMqC,OAAOpB,IAAI;oDAC7B,mCAAmC;oDACnC,IAAIjB,OAAOA,IAAIsC,GAAG,EAAE;wDAClBH,aAAanC;wDACb;oDACF;gDACF;4CACF,EAAE,OAAOuC,GAAG;gDACVC,QAAQC,KAAK,CAAC,mDAAmDF;4CACnE;4CACAZ;4CACA,IAAIA,WAAWC,aAAa;gDAC1B,MAAM,IAAIc,QAAQ,CAACC,UAAYC,WAAWD,SAAS;4CACrD;wCACF;oCACF;oCAEApD,SAAS4C;oCACThD,WAAW8C;oCACXnC,eAAe;oCACf;gCACF;gCACA,IAAIoC,WAAW,UAAU;oCACvBpC,eAAe;oCACf,MAAM,IAAI+C,MAAM;gCAClB;4BACF;wBACF,EAAE,OAAOC,GAAG;wBACV,eAAe;wBACjB;wBAEAnB,YAAY;wBACZ,IAAI,CAACD,aAAaC,WAAWC,aAAa;4BACxCgB,WAAWf,MAAM;wBACnB;oBACF;oBACAe,WAAWf,MAAM;oBACjB,OAAOd;gBACT;gBAEA,MAAM,IAAI8B,MAAM;YAClB,OAAO;gBACL,MAAM,EAAEE,SAAS,EAAE,EAAE,GAAG,MAAMhC,eAAeE,IAAI;gBACjD,MAAM+B,SAASD,OAAOxB,GAAG,CAAC,CAACkB,QAAeA,MAAMQ,OAAO,EAAEC,IAAI,CAAC;gBAC9D,MAAM,IAAIL,MAAMG;YAClB;QACF,GACCG,KAAK,CAAC,CAACV;YACN9E,MAAM8E,KAAK,CAAC,CAAC,oBAAoB,EAAEA,MAAMQ,OAAO,CAAC,CAAC;YAClDT,QAAQC,KAAK,CACX,qGACAA;QAEJ;IACJ,GAAG;QACDxD;QACAD,kBAAkBwB;QAClBhC;QACA,YAAY;QACZM;QACAC;QACAH;QACAD;QACAQ;KACD;IAED,OAAO;QACLY;QACAF;QACAF;QACAH;IACF;AACF,EAAC"}

package/dist/ui/Compose/hooks/useHistory.d.ts

@@ -1,7 +1,6 @@
 export declare const useHistory: () => {
     canRedo: boolean;
     canUndo: boolean;
-    currentValue: any;
     redo: () => any;
     set: (data: any) => any;
     undo: () => any;