@sanity/assist 1.2.15-lang.2 → 1.2.15-lang.4

package/dist/index.esm.js

@@ -137,6 +137,17 @@ function getCaptionFieldOption(schemaType) {
   }
   return getCaptionFieldOption(schemaType.type);
 }
+function getImagePromptFieldOption(schemaType) {
+  var _a;
+  if (!schemaType) {
+    return void 0;
+  }
+  const imagePromptField = (_a = schemaType.options) == null ? void 0 : _a.imagePromptField;
+  if (imagePromptField) {
+    return imagePromptField;
+  }
+  return getImagePromptFieldOption(schemaType.type);
+}
 function isSchemaAssistEnabled(type) {
   var _a, _b;
   return !((_b = (_a = type.options) == null ? void 0 : _a.aiWritingAssistance) == null ? void 0 : _b.exclude);
@@ -971,6 +982,7 @@ function useTranslate(apiClient) {
     let {
       documentId,
       languagePath,
+      translatePath,
       fieldLanguageMap
     } = _ref4;
     setLoading(true);
@@ -982,6 +994,7 @@ function useTranslate(apiClient) {
         types,
         languagePath,
         fieldLanguageMap,
+        translatePath: translatePath.length === 0 ? documentRootKey : pathToString(translatePath),
         userId: user == null ? void 0 : user.id
       }
     }).catch(e => {
@@ -1045,6 +1058,48 @@ function useGenerateCaption(apiClient) {
     loading
   }), [generateCaption, loading]);
 }
+function useGenerateImage(apiClient) {
+  const [loading, setLoading] = useState(false);
+  const user = useCurrentUser();
+  const schema = useSchema();
+  const types = useMemo(() => serializeSchema(schema, {
+    leanFormat: true
+  }), [schema]);
+  const toast = useToast();
+  const generateImage = useCallback(_ref6 => {
+    let {
+      path,
+      documentId
+    } = _ref6;
+    setLoading(true);
+    return apiClient.request({
+      method: "POST",
+      url: "/assist/tasks/generate-image/".concat(apiClient.config().dataset, "?projectId=").concat(apiClient.config().projectId),
+      body: {
+        path,
+        documentId,
+        types,
+        userId: user == null ? void 0 : user.id
+      }
+    }).catch(e => {
+      toast.push({
+        status: "error",
+        title: "Generate image from prompt failed",
+        description: e.message
+      });
+      setLoading(false);
+      throw e;
+    }).finally(() => {
+      setTimeout(() => {
+        setLoading(false);
+      }, 2e3);
+    });
+  }, [setLoading, apiClient, toast, user, types]);
+  return useMemo(() => ({
+    generateImage,
+    loading
+  }), [generateImage, loading]);
+}
 function useGetInstructStatus(apiClient) {
   const [loading, setLoading] = useState(true);
   const getInstructStatus = useCallback(async () => {
@@ -1234,8 +1289,8 @@ function RunInstructionProvider(props) {
       runInstructionRequest({
         ...request,
         instructionKey: instruction._key,
-        userTexts: Object.entries(inputs).map(_ref6 => {
-          let [key, value] = _ref6;
+        userTexts: Object.entries(inputs).map(_ref7 => {
+          let [key, value] = _ref7;
           return {
             blockKey: key,
             userInput: value
@@ -1248,8 +1303,8 @@ function RunInstructionProvider(props) {
   const open = !!runRequest;
   const runDisabled = useMemo(() => {
     var _a2, _b;
-    return ((_b = (_a2 = runRequest == null ? void 0 : runRequest.userInputBlocks) == null ? void 0 : _a2.length) != null ? _b : 0) > Object.entries(inputs).filter(_ref7 => {
-      let [, value] = _ref7;
+    return ((_b = (_a2 = runRequest == null ? void 0 : runRequest.userInputBlocks) == null ? void 0 : _a2.length) != null ? _b : 0) > Object.entries(inputs).filter(_ref8 => {
+      let [, value] = _ref8;
       return !!value;
     }).length;
   }, [runRequest == null ? void 0 : runRequest.userInputBlocks, inputs]);
@@ -1641,13 +1696,13 @@ function useSelectedSchema(fieldPath, documentSchema) {
     return currentSchema;
   }, [documentSchema, fieldPath]);
 }
-function FieldsInitializer(_ref8) {
+function FieldsInitializer(_ref9) {
   let {
     pathKey,
     activePath,
     fieldExists,
     onChange
-  } = _ref8;
+  } = _ref9;
   const initialized = useRef(false);
   useEffect(() => {
     if (initialized.current || fieldExists || activePath || !pathKey) {
@@ -2175,10 +2230,10 @@ const assistInspector = {
     showAsAction: false
   }),
   component: AssistInspectorWrapper,
-  onClose(_ref9) {
+  onClose(_ref10) {
     let {
       params
-    } = _ref9;
+    } = _ref10;
     return {
       params: typed({
         ...params,
@@ -2251,11 +2306,11 @@ var __template$3 = (cooked, raw) => __freeze$3(__defProp$3(cooked, "raw", {
 var _a$3, _b$1;
 const fadeIn = keyframes(_a$3 || (_a$3 = __template$3(["\n  0% {\n    opacity: 0;\n    transform: scale(0.75);\n  }\n  40% {\n    opacity: 0;\n    transform: scale(0.75);\n  }\n  100% {\n    opacity: 1;\n    transform: scale(1);\n  }\n"])));
 const FadeInDiv = styled.div(_b$1 || (_b$1 = __template$3(["\n  animation-name: ", ";\n  animation-timing-function: ease-in-out;\n"])), fadeIn);
-const FadeInContent = forwardRef(function FadeInContent2(_ref10, ref) {
+const FadeInContent = forwardRef(function FadeInContent2(_ref11, ref) {
   let {
     children,
     durationMs = 250
-  } = _ref10;
+  } = _ref11;
   return /* @__PURE__ */jsx(FadeInDiv, {
     ref,
     style: {
@@ -3316,7 +3371,7 @@ const defaultLanguageOutputs = function (member, enclosingType, translateFromLan
   }
   return void 0;
 };
-function getTranslationMap(documentSchema, documentMembers, translateFromLanguageId, outputLanguageIds, langFn) {
+function getFieldLanguageMap(documentSchema, documentMembers, translateFromLanguageId, outputLanguageIds, langFn) {
   var _a, _b, _c;
   const translationMaps = [];
   for (const member of documentMembers) {
@@ -4545,16 +4600,29 @@ const getLanguageParams = (select, document) => {
   }
   return selectedValue;
 };
+const toFieldLanguagesKeyPrefix = "sanityStudio:assist:field-languages:from:";
+function getPreferredToFieldLanguages(fromLanguageId) {
+  if (typeof localStorage === "undefined") {
+    return [];
+  }
+  const value = localStorage.getItem("".concat(toFieldLanguagesKeyPrefix).concat(fromLanguageId));
+  return value ? JSON.parse(value) : [];
+}
+function setPreferredToFieldLanguages(fromLanguageId, languageIds) {
+  if (typeof localStorage === "undefined") {
+    return;
+  }
+  localStorage.setItem("".concat(toFieldLanguagesKeyPrefix).concat(fromLanguageId), JSON.stringify(languageIds));
+}
 const FieldTranslationContext = createContext({
   openFieldTranslation: () => {},
   translationLoading: false
-  //loadLanguages: () => {},
 });
 function useFieldTranslation() {
   return useContext(FieldTranslationContext);
 }
 function FieldTranslationProvider(props) {
-  var _a, _b;
+  var _a, _b, _c;
   const {
     config: assistConfig
   } = useAiAssistanceConfig();
@@ -4564,64 +4632,63 @@ function FieldTranslationProvider(props) {
     translate: runTranslate
   } = useTranslate(apiClient);
   const [dialogOpen, setDialogOpen] = useState(false);
-  const [document, setDocument] = useState();
-  const [documentSchema, setDocumentSchema] = useState();
+  const [fieldTranslationParams, setFieldTranslationParams] = useState();
   const [languages, setLanguages] = useState();
   const [fromLanguage, setFromLanguage] = useState(void 0);
   const [toLanguages, setToLanguages] = useState(void 0);
-  const [translationMap, setTranslationMap] = useState();
+  const [fieldLanguageMaps, setFieldLanguageMaps] = useState();
   const close = useCallback(() => {
     setDialogOpen(false);
     setLanguages(void 0);
-    setDocument(void 0);
-    setDocument(void 0);
+    setFieldTranslationParams(void 0);
   }, []);
   const languageClient = useClient({
     apiVersion: (_b = config == null ? void 0 : config.apiVersion) != null ? _b : "2022-11-27"
   });
-  const documentId = document == null ? void 0 : document._id;
+  const documentId = (_c = fieldTranslationParams == null ? void 0 : fieldTranslationParams.document) == null ? void 0 : _c._id;
   const id = useId();
-  const selectFromLanguage = useCallback((from, languages2, document2, documentSchema2) => {
+  const selectFromLanguage = useCallback((from, languages2, params) => {
     var _a2, _b2;
+    const {
+      document,
+      documentSchema
+    } = params != null ? params : {};
     setFromLanguage(from);
-    if (!document2 || !documentSchema2 || !languages2) {
-      setTranslationMap(void 0);
+    if (!document || !documentSchema || !params || !languages2) {
+      setFieldLanguageMaps(void 0);
       return;
     }
-    const to = languages2.filter(l => l.id !== (from == null ? void 0 : from.id));
+    const preferred = getPreferredToFieldLanguages(from.id);
+    const to = languages2.filter(l => l.id !== (from == null ? void 0 : from.id)).filter(l => !preferred.length || preferred.includes(l.id));
     setToLanguages(to);
     const fromId = from == null ? void 0 : from.id;
     const toIds = (_a2 = to == null ? void 0 : to.map(l => l.id)) != null ? _a2 : [];
-    const docMembers = getDocumentMembersFlat(document2, documentSchema2);
+    const docMembers = getDocumentMembersFlat(document, documentSchema);
     if (fromId && (toIds == null ? void 0 : toIds.length)) {
-      const transMap = getTranslationMap(documentSchema2, docMembers, fromId, toIds, (_b2 = config == null ? void 0 : config.translationOutputs) != null ? _b2 : defaultLanguageOutputs);
-      setTranslationMap(transMap);
+      const transMap = getFieldLanguageMap(documentSchema, docMembers, fromId, toIds, (_b2 = config == null ? void 0 : config.translationOutputs) != null ? _b2 : defaultLanguageOutputs);
+      setFieldLanguageMaps(transMap);
     } else {
-      setTranslationMap(void 0);
+      setFieldLanguageMaps(void 0);
     }
   }, [config]);
   const toggleToLanguage = useCallback((toggledLang, toLanguages2, languages2) => {
-    if (!languages2) {
+    if (!languages2 || !fromLanguage) {
       return;
     }
     const wasSelected = !!(toLanguages2 == null ? void 0 : toLanguages2.find(l => l.id === toggledLang.id));
     const newToLangs = languages2.filter(anyLang => !!(toLanguages2 == null ? void 0 : toLanguages2.find(selectedLang => toggledLang.id !== selectedLang.id && selectedLang.id === anyLang.id)) || toggledLang.id === anyLang.id && !wasSelected);
     setToLanguages(newToLangs);
-  }, []);
-  const openFieldTranslation = useCallback(async _ref11 => {
-    let {
-      document: document2,
-      documentSchema: documentSchema2
-    } = _ref11;
+    setPreferredToFieldLanguages(fromLanguage.id, newToLangs.map(l => l.id));
+  }, [fromLanguage]);
+  const openFieldTranslation = useCallback(async params => {
     setDialogOpen(true);
-    const languageParams = getLanguageParams(config == null ? void 0 : config.selectLanguageParams, document2);
+    const languageParams = getLanguageParams(config == null ? void 0 : config.selectLanguageParams, params.document);
     const languages2 = await (typeof (config == null ? void 0 : config.languages) === "function" ? config == null ? void 0 : config.languages(languageClient, languageParams) : Promise.resolve(config == null ? void 0 : config.languages));
     setLanguages(languages2);
-    setDocument(document2);
-    setDocumentSchema(documentSchema2);
+    setFieldTranslationParams(params);
     const fromLanguage2 = languages2 == null ? void 0 : languages2[0];
     if (fromLanguage2) {
-      selectFromLanguage(fromLanguage2, languages2, document2, documentSchema2);
+      selectFromLanguage(fromLanguage2, languages2, params);
     } else {
       console.error("No languages available for selected language params", languageParams);
     }
@@ -4632,12 +4699,14 @@ function FieldTranslationProvider(props) {
       translationLoading: false
     };
   }, [openFieldTranslation]);
-  const runDisabled = !fromLanguage || !(toLanguages == null ? void 0 : toLanguages.length) || !(translationMap == null ? void 0 : translationMap.length) || !documentId;
+  const runDisabled = !fromLanguage || !(toLanguages == null ? void 0 : toLanguages.length) || !(fieldLanguageMaps == null ? void 0 : fieldLanguageMaps.length) || !documentId;
   const onRunTranslation = useCallback(() => {
-    if (translationMap && documentId) {
+    const translatePath = fieldTranslationParams == null ? void 0 : fieldTranslationParams.translatePath;
+    if (fieldLanguageMaps && documentId && translatePath) {
       runTranslate({
         documentId,
-        fieldLanguageMap: translationMap.map(map => ({
+        translatePath: fieldTranslationParams == null ? void 0 : fieldTranslationParams.translatePath,
+        fieldLanguageMap: fieldLanguageMaps.map(map => ({
           ...map,
           // eslint-disable-next-line max-nested-callbacks
           outputs: map.outputs.filter(out => !!(toLanguages == null ? void 0 : toLanguages.find(l => l.id === out.id)))
@@ -4645,7 +4714,7 @@ function FieldTranslationProvider(props) {
       });
     }
     close();
-  }, [translationMap, documentId, runTranslate, close, toLanguages]);
+  }, [fieldLanguageMaps, documentId, runTranslate, close, toLanguages, fieldTranslationParams == null ? void 0 : fieldTranslationParams.translatePath]);
   const runButton = /* @__PURE__ */jsx(Button, {
     text: "Translate",
     tone: "primary",
@@ -4704,7 +4773,7 @@ function FieldTranslationProvider(props) {
                 name: "fromLang",
                 value: l.id,
                 checked: l.id === (fromLanguage == null ? void 0 : fromLanguage.id),
-                onClick: () => selectFromLanguage(l, languages, document, documentSchema)
+                onClick: () => selectFromLanguage(l, languages, fieldTranslationParams)
               }), /* @__PURE__ */jsx(Text, {
                 children: (_a2 = l.title) != null ? _a2 : l.id
               })]
@@ -4727,8 +4796,7 @@ function FieldTranslationProvider(props) {
                 name: "toLang",
                 value: l.id,
                 checked: !!(toLanguages == null ? void 0 : toLanguages.find(tl => tl.id === l.id)),
-                onClick: () => toggleToLanguage(l, toLanguages, languages),
-                disabled: !(translationMap == null ? void 0 : translationMap.find(tm => tm.outputs.find(o => o.id === l.id)))
+                onClick: () => toggleToLanguage(l, toLanguages, languages)
               }), /* @__PURE__ */jsx(Text, {
                 children: (_a2 = l.title) != null ? _a2 : l.id
               })]
@@ -4793,7 +4861,10 @@ function ErrorWrapper(props) {
   const catchError = useCallback(params => {
     setError(params.error);
   }, [setError]);
-  const unsetValue = useCallback(() => onChange(PatchEvent.from(unset())), [onChange]);
+  const unsetValue = useCallback(() => {
+    onChange(PatchEvent.from(unset()));
+    setError(void 0);
+  }, [onChange]);
   const dismiss = useCallback(() => setError(void 0), []);
   const catcher = /* @__PURE__ */jsx(ErrorBoundary, {
     onCatch: catchError,
@@ -6101,7 +6172,7 @@ function PrivateIcon() {
     children: /* @__PURE__ */jsx(LockIcon, {})
   });
 }
-const ImageContext = createContext(void 0);
+const ImageContext = createContext({});
 function ImageContextProvider(props) {
   var _a;
   const {
@@ -6137,17 +6208,19 @@ function ImageContextProvider(props) {
   }, [schemaType, path, assetRef, assetRefState, documentId, generateCaption, isSyncing]);
   const context = useMemo(() => {
     const captionField = getCaptionFieldOption(schemaType);
-    return captionField ? {
-      captionPath: pathToString([...path, captionField]),
+    const imagePromptField = getImagePromptFieldOption(schemaType);
+    return {
+      captionPath: captionField ? pathToString([...path, captionField]) : void 0,
+      imagePromptPath: imagePromptField ? pathToString([...path, imagePromptField]) : void 0,
       assetRef
-    } : void 0;
+    };
   }, [schemaType, path, assetRef]);
   return /* @__PURE__ */jsx(ImageContext.Provider, {
     value: context,
     children: props.renderDefault(props)
   });
 }
-function node$2(node2) {
+function node$3(node2) {
   return node2;
 }
 const generateCaptionsActions = {
@@ -6168,7 +6241,7 @@ const generateCaptionsActions = {
         documentId
       } = useAssistDocumentContext();
       return useMemo(() => {
-        return node$2({
+        return node$3({
           type: "action",
           icon: loading ? () => /* @__PURE__ */jsx(Box, {
             style: {
@@ -6199,27 +6272,31 @@ const generateCaptionsActions = {
     return void 0;
   }
 };
-function node$1(node2) {
+function node$2(node2) {
   return node2;
 }
 const translateActions = {
   name: "sanity-assist-translate",
   useAction(props) {
-    var _a, _b, _c, _d, _e, _f, _g;
+    var _a, _b, _c, _d, _e, _f, _g, _h;
     const {
       config
     } = useAiAssistanceConfig();
     const apiClient = useApiClient(config == null ? void 0 : config.__customApiClient);
-    const isDocumentLevel = props.path.length === 0;
     const {
-      schemaType,
+      schemaType: fieldSchemaType,
+      path,
       documentId,
+      documentSchemaType,
       documentIsAssistable
     } = props;
-    const fieldTransEnabled = (_c = (_b = (_a = config.translate) == null ? void 0 : _a.field) == null ? void 0 : _b.documentTypes) == null ? void 0 : _c.includes(schemaType.name);
-    const docTransTypes = (_e = (_d = config.translate) == null ? void 0 : _d.document) == null ? void 0 : _e.documentTypes;
-    const documentTranslationEnabled = isDocumentLevel && (!docTransTypes && isAssistSupported(schemaType) || (docTransTypes == null ? void 0 : docTransTypes.includes(schemaType.name)));
-    if (documentTranslationEnabled || fieldTransEnabled) {
+    const isDocumentLevel = path.length === 0;
+    const docTransTypes = (_b = (_a = config.translate) == null ? void 0 : _a.document) == null ? void 0 : _b.documentTypes;
+    const options = fieldSchemaType == null ? void 0 : fieldSchemaType.options;
+    const addFieldAction = isDocumentLevel || ((_c = options == null ? void 0 : options.aiWritingAssistance) == null ? void 0 : _c.translateAction);
+    const fieldTransEnabled = addFieldAction && documentSchemaType && ((_f = (_e = (_d = config.translate) == null ? void 0 : _d.field) == null ? void 0 : _e.documentTypes) == null ? void 0 : _f.includes(documentSchemaType.name));
+    const documentTranslationEnabled = addFieldAction && documentSchemaType && (!docTransTypes && isAssistSupported(fieldSchemaType) || (docTransTypes == null ? void 0 : docTransTypes.includes(documentSchemaType.name)));
+    if (documentSchemaType && (documentTranslationEnabled || fieldTransEnabled)) {
       const {
         value: documentValue,
         onChange: documentOnChange
@@ -6232,39 +6309,52 @@ const translateActions = {
         task: translationApi.translate
       });
       docRef.current = documentValue;
-      const languagePath = (_g = (_f = config.translate) == null ? void 0 : _f.document) == null ? void 0 : _g.languageField;
-      const translateDocumentAction = useMemo(() => languagePath && documentTranslationEnabled ? node$1({
-        type: "action",
-        icon: translationApi.loading ? () => /* @__PURE__ */jsx(Box, {
-          style: {
-            height: 17
-          },
-          children: /* @__PURE__ */jsx(Spinner, {
+      const languagePath = (_h = (_g = config.translate) == null ? void 0 : _g.document) == null ? void 0 : _h.languageField;
+      const translateDocumentAction = useMemo(() => {
+        var _a2;
+        if (!languagePath || !documentTranslationEnabled) {
+          return void 0;
+        }
+        const {
+          value: languageId
+        } = (_a2 = extractWithPath(languagePath, documentValue)[0]) != null ? _a2 : {};
+        const title = path.length ? "Translate" : "Translate document";
+        return node$2({
+          type: "action",
+          icon: translationApi.loading ? () => /* @__PURE__ */jsx(Box, {
             style: {
-              transform: "translateY(6px)"
+              height: 17
+            },
+            children: /* @__PURE__ */jsx(Spinner, {
+              style: {
+                transform: "translateY(6px)"
+              }
+            })
+          }) : TranslateIcon,
+          title,
+          onAction: () => {
+            if (translationApi.loading || !languagePath || !documentId) {
+              return;
             }
-          })
-        }) : TranslateIcon,
-        title: "Translate document",
-        onAction: () => {
-          if (translationApi.loading || !languagePath || !documentId) {
-            return;
-          }
-          translate({
-            languagePath,
-            documentId: documentId != null ? documentId : ""
-          });
-        },
-        renderAsButton: true,
-        disabled: translationApi.loading
-      }) : void 0, [languagePath, translate, documentId, translationApi.loading, documentTranslationEnabled]);
+            translate({
+              languagePath,
+              translatePath: path,
+              documentId: documentId != null ? documentId : ""
+            });
+          },
+          renderAsButton: true,
+          disabled: translationApi.loading || !languageId ? {
+            reason: "Language is not set for this document"
+          } : void 0
+        });
+      }, [languagePath, translate, documentId, translationApi.loading, documentTranslationEnabled, path]);
       const fieldTranslate = useFieldTranslation();
       const openFieldTranslation = useDraftDelayedTask({
         documentOnChange,
         isDocAssistable: documentIsAssistable != null ? documentIsAssistable : false,
         task: fieldTranslate.openFieldTranslation
       });
-      const translateFieldsAction = useMemo(() => fieldTransEnabled ? node$1({
+      const translateFieldsAction = useMemo(() => fieldTransEnabled ? node$2({
         type: "action",
         icon: fieldTranslate.translationLoading ? () => /* @__PURE__ */jsx(Box, {
           style: {
@@ -6276,24 +6366,25 @@ const translateActions = {
             }
           })
         }) : TranslateIcon,
-        title: "Translate fields",
+        title: "Translate fields...",
         onAction: () => {
           if (fieldTranslate.translationLoading || !documentId) {
             return;
           }
           openFieldTranslation({
             document: docRef.current,
-            documentSchema: schemaType
+            documentSchema: documentSchemaType,
+            translatePath: path
           });
         },
         renderAsButton: true,
         disabled: fieldTranslate.translationLoading
-      }) : void 0, [openFieldTranslation, schemaType, documentId, fieldTranslate.translationLoading, fieldTransEnabled]);
+      }) : void 0, [openFieldTranslation, documentSchemaType, documentId, fieldTranslate.translationLoading, fieldTransEnabled, path]);
       return useMemo(() => {
-        return node$1({
+        return node$2({
           type: "group",
           icon: () => null,
-          title: "Translate",
+          title: "Translation",
           children: [translateDocumentAction, translateFieldsAction].filter(c => !!c),
           expanded: true
         });
@@ -6302,6 +6393,58 @@ const translateActions = {
     return void 0;
   }
 };
+function node$1(node2) {
+  return node2;
+}
+const generateImagActions = {
+  name: "sanity-assist-generate-image",
+  useAction(props) {
+    const pathKey = usePathKey(props.path);
+    const {
+      config
+    } = useAiAssistanceConfig();
+    const apiClient = useApiClient(config == null ? void 0 : config.__customApiClient);
+    const {
+      generateImage,
+      loading
+    } = useGenerateImage(apiClient);
+    const imageContext = useContext(ImageContext);
+    if (imageContext && pathKey === (imageContext == null ? void 0 : imageContext.imagePromptPath)) {
+      const {
+        documentId
+      } = useAssistDocumentContext();
+      return useMemo(() => {
+        return node$1({
+          type: "action",
+          icon: loading ? () => /* @__PURE__ */jsx(Box, {
+            style: {
+              height: 17
+            },
+            children: /* @__PURE__ */jsx(Spinner, {
+              style: {
+                transform: "translateY(6px)"
+              }
+            })
+          }) : ImageIcon,
+          title: "Generate image from prompt",
+          onAction: () => {
+            if (loading) {
+              return;
+            }
+            generateImage({
+              path: pathKey,
+              documentId: documentId != null ? documentId : ""
+            });
+          },
+          renderAsButton: true,
+          disabled: loading,
+          hidden: !imageContext.assetRef
+        });
+      }, [generateImage, pathKey, documentId, loading, imageContext]);
+    }
+    return void 0;
+  }
+};
 function node(node2) {
   return node2;
 }
@@ -6358,10 +6501,12 @@ const assistFieldActions = {
     const isPathSelected = pathKey === selectedPath;
     const isSelected = isInspectorOpen && isPathSelected;
     const imageCaptionAction = generateCaptionsActions.useAction(props);
+    const imageGenAction = generateImagActions.useAction(props);
     const translateAction = translateActions.useAction(typed({
       ...props,
       documentId: assistableDocumentId,
-      documentIsAssistable
+      documentIsAssistable,
+      documentSchemaType
     }));
     const manageInstructions = useCallback(() => isSelected ? closeInspector(aiInspectorId) : openInspector(aiInspectorId, {
       [fieldPathParam]: pathKey,
@@ -6389,7 +6534,7 @@ const assistFieldActions = {
     }, [fieldAssist == null ? void 0 : fieldAssist.instructions]);
     const instructions = useMemo(() => [...privateInstructions, ...sharedInstructions], [privateInstructions, sharedInstructions]);
     const runInstructionsGroup = useMemo(() => {
-      return (instructions == null ? void 0 : instructions.length) || imageCaptionAction || translateAction ? node({
+      return (instructions == null ? void 0 : instructions.length) || imageCaptionAction || translateAction || imageGenAction ? node({
         type: "group",
         icon: () => null,
         title: "Run instructions",
@@ -6400,10 +6545,10 @@ const assistFieldActions = {
           hidden: isHidden,
           documentIsNew: !!documentIsNew,
           assistSupported
-        }))), imageCaptionAction].filter(a => !!a),
+        }))), imageCaptionAction, imageGenAction].filter(a => !!a),
         expanded: true
       }) : void 0;
-    }, [instructions, currentUser == null ? void 0 : currentUser.id, onInstructionAction, isHidden, documentIsNew, assistSupported, imageCaptionAction, translateAction]);
+    }, [instructions, currentUser == null ? void 0 : currentUser.id, onInstructionAction, isHidden, documentIsNew, assistSupported, imageCaptionAction, translateAction, imageGenAction]);
     const instructionsLength = (instructions == null ? void 0 : instructions.length) || 0;
     const manageInstructionsItem = useMemo(() => node({
       type: "action",
@@ -6416,13 +6561,13 @@ const assistFieldActions = {
       type: "group",
       icon: SparklesIcon,
       title: pluginTitleShort,
-      children: [runInstructionsGroup, translateAction, assistSupported && manageInstructionsItem].filter(c => !!c),
+      children: [runInstructionsGroup, translateAction, assistSupported && manageInstructionsItem].filter(c => !!c).filter(c => c.type === "group" ? c.children.length : true),
       expanded: false,
       renderAsButton: true,
-      hidden: !assistSupported && !imageCaptionAction && !translateAction
+      hidden: !assistSupported && !imageCaptionAction && !translateAction && !imageGenAction
     }), [
     //documentIsNew,
-    runInstructionsGroup, manageInstructionsItem, assistSupported, imageCaptionAction, translateAction]);
+    runInstructionsGroup, manageInstructionsItem, assistSupported, imageCaptionAction, translateAction, imageGenAction]);
     const emptyAction = useMemo(() => node({
       type: "action",
       hidden: !assistSupported,
@@ -6432,7 +6577,7 @@ const assistFieldActions = {
       title: pluginTitleShort,
       selected: isSelected
     }), [assistSupported, manageInstructions, isSelected]);
-    if (instructionsLength === 0 && !imageCaptionAction && !translateAction) {
+    if (instructionsLength === 0 && !imageCaptionAction && !translateAction && !imageGenAction) {
       return emptyAction;
     }
     return group;
@@ -6551,12 +6696,28 @@ function AssistDocumentInput(_ref23) {
     ...props
   } = _ref23;
   useInstructionToaster(documentId, props.schemaType);
+  const schemaType = useMemo(() => {
+    if (props.schemaType.name !== assistDocumentTypeName) {
+      return props.schemaType;
+    }
+    return {
+      ...props.schemaType,
+      type: {
+        ...props.schemaType.type,
+        // compatability with i18nArrays plugin that requires this to be document
+        name: "document"
+      }
+    };
+  }, [props.schemaType]);
   return /* @__PURE__ */jsx(FirstAssistedPathProvider, {
     members: props.members,
     children: /* @__PURE__ */jsx(AssistDocumentContextProvider, {
-      schemaType: props.schemaType,
+      schemaType,
       documentId,
-      children: props.renderDefault(props)
+      children: props.renderDefault({
+        ...props,
+        schemaType
+      })
     })
   });
 }
@@ -6632,7 +6793,8 @@ const assist = definePlugin(config => {
     },
     document: {
       inspectors: (prev, context) => {
-        const docSchema = context.schema.get(context.documentType);
+        const documentType = context.documentType;
+        const docSchema = context.schema.get(documentType);
         if (docSchema && isSchemaAssistEnabled(docSchema)) {
           return [...prev, assistInspector];
         }
@@ -6643,6 +6805,9 @@ const assist = definePlugin(config => {
           documentType,
           schema
         } = _ref24;
+        if (documentType === assistDocumentTypeName) {
+          return [];
+        }
         const docSchema = schema.get(documentType);
         if (docSchema && isSchemaAssistEnabled(docSchema)) {
           return [...prev, assistFieldActions];
@@ -6655,6 +6820,9 @@ const assist = definePlugin(config => {
           schema,
           schemaType
         } = _ref25;
+        if (schemaType === assistDocumentTypeName) {
+          return [];
+        }
         const docSchema = schema.get(schemaType);
         if (docSchema && isObjectSchemaType(docSchema) && isSchemaAssistEnabled(docSchema)) {
           return [...prev, createAssistDocumentPresence(documentId, docSchema)];