@sanity/assist 4.4.6 → 4.4.7

package/dist/index.mjs

@@ -1,10 +1,10 @@
 import { jsx, jsxs, Fragment } from "react/jsx-runtime";
-import { pathToString, getVersionFromId, getPublishedId, isVersionId, useEditState, useCurrentUser, useClient, typed, isObjectSchemaType, stringToPath, isKeySegment, isArraySchemaType, useSchema, FormFieldHeaderText, PatchEvent, unset, getVersionId, getDraftId, useColorSchemeValue, useFormCallbacks, useDocumentStore, useDocumentPresence, createPatchChannel, FormBuilder, fromMutationPatches, StatusButton, PresenceOverlay, VirtualizerScrollInstanceProvider, isDocumentSchemaType, useSyncState, set, useWorkspaceSchemaId, MemberFieldError, FormCallbacksProvider, FormInput, setIfMissing, insert, ObjectInputMember, isArrayOfObjectsSchemaType, defineType, defineField, defineArrayMember, definePlugin } from "sanity";
+import { pathToString, getVersionFromId, getPublishedId, isVersionId, useEditState, useCurrentUser, useClient, typed, FormFieldHeaderText, PatchEvent, unset, isObjectSchemaType, stringToPath, isKeySegment, useSchema, getVersionId, getDraftId, useColorSchemeValue, isArraySchemaType, useFormCallbacks, useDocumentStore, useDocumentPresence, createPatchChannel, FormBuilder, fromMutationPatches, StatusButton, PresenceOverlay, VirtualizerScrollInstanceProvider, isDocumentSchemaType, useSyncState, set, useWorkspaceSchemaId, MemberFieldError, FormCallbacksProvider, FormInput, setIfMissing, insert, ObjectInputMember, isArrayOfObjectsSchemaType, defineType, defineField, defineArrayMember, definePlugin } from "sanity";
 import { useToast, useLayer, Dialog, Stack, Flex, Tooltip, Text, TextArea, Button, Badge, Popover, Card, Box, ErrorBoundary, focusFirstDescendant, Spinner, Container, Autocomplete, Breadcrumbs, useClickOutside, useGlobalKeyDown, useTheme, rgba, Radio, Checkbox, ThemeProvider, MenuButton, Menu, MenuItem, Switch, Label } from "@sanity/ui";
 import { useRef, useState, useEffect, useMemo, createContext, useContext, useCallback, useId, forwardRef, createElement, useReducer } from "react";
 import { useDocumentPane, usePaneRouter, DocumentInspectorHeader, DocumentPaneProvider } from "sanity/structure";
 import { minutesToMilliseconds, isAfter, addSeconds, formatDistanceToNow } from "date-fns";
-import { DocumentIcon, LinkIcon, ImageIcon, BlockContentIcon, OlistIcon, BlockquoteIcon, StringIcon, PlayIcon, SparklesIcon, ArrowRightIcon, CheckmarkIcon, SearchIcon, SyncIcon, ErrorOutlineIcon, CheckmarkCircleIcon, ClockIcon, CloseCircleIcon, RetryIcon, CloseIcon, icons, TranslateIcon, LockIcon, ControlsIcon, ArrowLeftIcon, TokenIcon, DocumentTextIcon, ThListIcon, CodeIcon, ComposeIcon } from "@sanity/icons";
+import { PlayIcon, DocumentIcon, LinkIcon, ImageIcon, BlockContentIcon, OlistIcon, BlockquoteIcon, StringIcon, SparklesIcon, ArrowRightIcon, CheckmarkIcon, SearchIcon, SyncIcon, ErrorOutlineIcon, CheckmarkCircleIcon, ClockIcon, CloseCircleIcon, RetryIcon, CloseIcon, icons, TranslateIcon, LockIcon, ControlsIcon, ArrowLeftIcon, TokenIcon, DocumentTextIcon, ThListIcon, CodeIcon, ComposeIcon } from "@sanity/icons";
 import { extractWithPath } from "@sanity/mutator";
 import { keyframes, styled } from "styled-components";
 import { tap, mergeMap, share, take, filter, distinctUntilChanged, catchError } from "rxjs/operators";
@@ -392,221 +392,6 @@ function useAssistDocumentContext() {
     throw new Error("AssistDocumentContext value is missing");
   return context;
 }
-const SelectedFieldContext = createContext(void 0), SelectedFieldContextProvider = SelectedFieldContext.Provider, maxDepth = 6;
-function getTypeIcon(schemaType) {
-  let t = schemaType;
-  for (; t; ) {
-    if (t.icon) return t.icon;
-    t = t.type;
-  }
-  return isType(schemaType, "slug") ? LinkIcon : isType(schemaType, "image") ? ImageIcon : schemaType.jsonType === "array" && isPortableTextArray(schemaType) ? BlockContentIcon : schemaType.jsonType === "array" ? OlistIcon : schemaType.jsonType === "object" ? BlockquoteIcon : schemaType.jsonType === "string" ? StringIcon : DocumentIcon;
-}
-function asFieldRefsByTypePath(fieldRefs) {
-  return fieldRefs.reduce(
-    (acc, ref) => ({ ...acc, [ref.key]: ref }),
-    {}
-  );
-}
-function getFieldRefsWithDocument(schemaType) {
-  const fields = getFieldRefs(schemaType);
-  return [
-    {
-      key: documentRootKey,
-      icon: schemaType.icon ?? DocumentIcon,
-      title: "The entire document",
-      path: [],
-      schemaType
-    },
-    ...fields
-  ];
-}
-function getFieldRefs(schemaType, parent, depth = 0) {
-  return depth >= maxDepth ? [] : schemaType.fields.filter((f) => !f.name.startsWith("_")).flatMap((field) => {
-    const path = parent ? [...parent.path, field.name] : [field.name], title = field.type.title ?? field.name, fieldRef = {
-      key: patchableKey(pathToString(path)),
-      path,
-      title: parent ? [parent.title, title].join(" / ") : title,
-      schemaType: field.type,
-      icon: getTypeIcon(field.type)
-    }, fields = field.type.jsonType === "object" ? getFieldRefs(field.type, fieldRef, depth + 1) : [], syntheticFields = field.type.jsonType === "array" ? getSyntheticFields(field.type, fieldRef, depth + 1) : [];
-    return isAssistSupported(field.type) ? [fieldRef, ...fields, ...syntheticFields] : [...fields, ...syntheticFields];
-  });
-}
-function getSyntheticFields(schemaType, parent, depth = 0) {
-  return depth >= maxDepth ? [] : schemaType.of.filter((itemType) => !isType(itemType, "block")).flatMap((itemType) => {
-    const segment = { _key: itemType.name }, title = itemType.title ?? itemType.name, path = parent ? [...parent.path, segment] : [segment], fieldRef = {
-      key: patchableKey(pathToString(path)),
-      path,
-      title: parent ? [parent.title, title].join(" / ") : title,
-      schemaType: itemType,
-      icon: getTypeIcon(itemType),
-      synthetic: !0
-    }, fields = itemType.jsonType === "object" ? getFieldRefs(itemType, fieldRef, depth + 1) : [];
-    return isAssistSupported(itemType) ? [fieldRef, ...fields] : fields;
-  });
-}
-function getTypePath(doc, pathString) {
-  if (!pathString)
-    return;
-  const path = stringToPath(pathString), currentPath = [];
-  let valid = !0;
-  const syntheticPath = path.map((segment) => {
-    if (currentPath.push(segment), isKeySegment(segment)) {
-      const match = extractWithPath(pathToString(currentPath), doc)[0], value = match?.value;
-      if (match && value && typeof value == "object" && "_type" in value)
-        return { _key: value._type };
-      valid = !1;
-    }
-    return segment;
-  });
-  return valid ? patchableKey(pathToString(syntheticPath)) : void 0;
-}
-function patchableKey(pathKey) {
-  return pathKey.replace(/[=]=/g, ":").replace(/[[\]]/g, "|").replace(/"/g, "");
-}
-function useTypePath(doc, pathString) {
-  return useMemo(() => getTypePath(doc, pathString), [doc, pathString]);
-}
-function useSelectedField(documentSchemaType, path) {
-  const selectableFields = useMemo(
-    () => documentSchemaType && isObjectSchemaType(documentSchemaType) ? getFieldRefsWithDocument(documentSchemaType) : [],
-    [documentSchemaType]
-  );
-  return useMemo(() => path ? selectableFields?.find((f) => f.key === path) : void 0, [selectableFields, path]);
-}
-function getFieldTitle(field) {
-  const schemaType = field?.schemaType;
-  return field?.title ?? schemaType?.title ?? schemaType?.name ?? "Untitled";
-}
-function useAiPaneRouter() {
-  const paneRouter = usePaneRouter();
-  return useMemo(
-    () => ({ ...paneRouter, params: paneRouter.params ?? {} }),
-    [paneRouter]
-  );
-}
-const hiddenTypes = [
-  "any",
-  "array",
-  "block",
-  "boolean",
-  "crossDatasetReference",
-  "date",
-  "datetime",
-  "document",
-  "email",
-  "file",
-  "globalDocumentReference",
-  "image",
-  "number",
-  "object",
-  "reference",
-  "span",
-  "string",
-  "text",
-  "url",
-  "slug",
-  "geopoint",
-  "sanity.assetSourceData",
-  "sanity.imageAsset",
-  "sanity.fileAsset",
-  "sanity.imageCrop",
-  "sanity.imageHotspot",
-  "sanity.imageMetadata",
-  "sanity.imageDimensions",
-  "sanity.imagePalette",
-  "sanity.imagePaletteSwatch",
-  assistSerializedTypeName,
-  assistSerializedFieldTypeName,
-  "sanity-agent.job.document"
-], inlineTypes = ["document", "object", "image", "file"];
-function serializeSchema(schema, options2) {
-  const list = schema.getTypeNames().filter((t) => !(hiddenTypes.includes(t) || t.startsWith("sanity."))).map((t) => schema.get(t)).filter((t) => !!t).map((t) => getSchemaStub(t, schema, options2)).filter((t) => !("to" in t && t.to && !t.to.length || "of" in t && t.of && !t.of.length || "fields" in t && t.fields && !t.fields.length));
-  return list.sort((a, b) => (a?.name ?? "").localeCompare(b?.name ?? "")), list;
-}
-function getSchemaStub(schemaType, schema, options2) {
-  if (!schemaType.type?.name)
-    throw console.error("Missing type name", schemaType.type), new Error("Type is missing name!");
-  const baseSchema = {
-    // we dont need type or id when we send using POST, so leave these out to save bandwidth
-    ...options2?.leanFormat ? {} : { _id: `${assistSchemaIdPrefix}${schemaType.name}`, _type: assistSerializedTypeName },
-    name: schemaType.name,
-    title: schemaType.title,
-    type: schemaType.type.name,
-    ...getBaseFields(schema, schemaType, schemaType.type.name, options2)
-  };
-  return removeUndef(baseSchema);
-}
-function getBaseFields(schema, type, typeName, options2) {
-  const schemaOptions = removeUndef({
-    imagePromptField: type.options?.aiAssist?.imageInstructionField,
-    embeddingsIndex: type.options?.aiAssist?.embeddingsIndex
-  });
-  return removeUndef({
-    options: Object.keys(schemaOptions).length ? schemaOptions : void 0,
-    values: Array.isArray(type?.options?.list) ? type?.options?.list.map(
-      (v) => typeof v == "string" ? v : v.value ?? `${v.title}`
-    ) : void 0,
-    of: "of" in type && typeName === "array" ? arrayOf(type, schema, options2) : void 0,
-    to: "to" in type && typeName === "reference" ? refToTypeNames(type) : void 0,
-    fields: "fields" in type && inlineTypes.includes(typeName) ? serializeFields(schema, type, options2) : void 0,
-    annotations: typeName === "block" && "fields" in type ? serializeAnnotations(type, schema, options2) : void 0,
-    inlineOf: typeName === "block" && "fields" in type ? serializeInlineOf(type, schema, options2) : void 0,
-    hidden: typeof type.hidden == "function" ? "function" : type.hidden ? !0 : void 0,
-    readOnly: typeof type.readOnly == "function" ? "function" : type.readOnly ? !0 : void 0
-  });
-}
-function serializeFields(schema, schemaType, options2) {
-  return (schemaType.fieldsets ? schemaType.fieldsets.flatMap(
-    (fs) => fs.single ? fs.field : fs.fields.map((f) => ({
-      ...f,
-      type: {
-        ...f.type,
-        // if fieldset is (conditionally) hidden, the field must be considered the same way
-        // ie, if the field does not show up in conditionalMembers, it is hidden
-        // regardless of weather or not it is the field function or the fieldset function that hides it
-        hidden: typeof fs.hidden == "function" ? fs.hidden : fs.hidden ? !0 : f.type.hidden
-      }
-    }))
-  ) : schemaType.fields).filter((f) => !["sanity.imageHotspot", "sanity.imageCrop"].includes(f.type?.name ?? "")).filter((f) => isAssistSupported(f.type)).map((field) => serializeMember(schema, field.type, field.name, options2));
-}
-function serializeMember(schema, type, name, options2) {
-  const typeName = schema.get(type?.name) ? type.name : type.type?.name ?? "";
-  return removeUndef({
-    ...options2?.leanFormat ? {} : { _type: assistSerializedFieldTypeName },
-    name,
-    type: typeName,
-    title: type.title,
-    ...getBaseFields(schema, type, typeName, options2)
-  });
-}
-function serializeInlineOf(blockSchemaType, schema, options2) {
-  const childrenType = blockSchemaType.fields.find((f) => f.name === "children")?.type;
-  if (!(!childrenType || !isArraySchemaType(childrenType)))
-    return arrayOf(
-      {
-        of: childrenType.of.filter((t) => !isType(t, "span"))
-      },
-      schema,
-      options2
-    );
-}
-function serializeAnnotations(blockSchemaType, schema, options2) {
-  const marksType = blockSchemaType.fields.find((f) => f.name === "markDefs")?.type;
-  if (!(!marksType || !isArraySchemaType(marksType)))
-    return arrayOf(marksType, schema, options2);
-}
-function arrayOf(arrayType, schema, options2) {
-  return arrayType.of.filter((type) => isAssistSupported(type)).map((t) => serializeMember(schema, t, t.name, options2));
-}
-function refToTypeNames(type) {
-  return type.to.map((t) => ({
-    type: typed(t.name)
-  }));
-}
-function removeUndef(obj) {
-  return Object.keys(obj).forEach((key) => obj[key] === void 0 ? delete obj[key] : {}), obj;
-}
 const AiAssistanceConfigContext = createContext({});
 function useAiAssistanceConfig() {
   const context = useContext(AiAssistanceConfigContext);
@@ -617,33 +402,6 @@ function useAiAssistanceConfig() {
 function useSerializedTypes() {
   return useAiAssistanceConfig().serializedTypes;
 }
-function AiAssistanceConfigProvider(props) {
-  const [status, setStatus] = useState(), [error, setError] = useState(), apiClient = useApiClient(props.config?.__customApiClient), { getInstructStatus, loading: statusLoading } = useGetInstructStatus(apiClient), { initInstruct, loading: initLoading } = useInitInstruct(apiClient), schema = useSchema(), serializedTypes = useMemo(() => serializeSchema(schema, { leanFormat: !0 }), [schema]);
-  useEffect(() => {
-    getInstructStatus().then((s) => setStatus(s)).catch((e) => {
-      console.error(e), setError(e);
-    });
-  }, [getInstructStatus]);
-  const init = useCallback(async () => {
-    setError(void 0);
-    try {
-      await initInstruct();
-      const status2 = await getInstructStatus();
-      setStatus(status2);
-    } catch (e) {
-      console.error("Failed to init ai assistance", e), setError(e);
-    }
-  }, [initInstruct, getInstructStatus, setStatus]), { config, children } = props, context = useMemo(() => ({
-    config,
-    status,
-    statusLoading,
-    init,
-    initLoading,
-    error,
-    serializedTypes
-  }), [config, status, init, statusLoading, initLoading, error, serializedTypes]);
-  return /* @__PURE__ */ jsx(AiAssistanceConfigContext.Provider, { value: context, children });
-}
 const basePath = "/assist/tasks/instruction", API_VERSION_WITH_EXTENDED_TYPES = "2025-04-01";
 function canUseAssist(status) {
   return status?.enabled && status.initialized && status.validToken;
@@ -1022,19 +780,105 @@ function useDraftDelayedTask(args) {
     [setQueuedArgs, documentOnChange]
   );
 }
+const maxDepth = 6;
+function getTypeIcon(schemaType) {
+  let t = schemaType;
+  for (; t; ) {
+    if (t.icon) return t.icon;
+    t = t.type;
+  }
+  return isType(schemaType, "slug") ? LinkIcon : isType(schemaType, "image") ? ImageIcon : schemaType.jsonType === "array" && isPortableTextArray(schemaType) ? BlockContentIcon : schemaType.jsonType === "array" ? OlistIcon : schemaType.jsonType === "object" ? BlockquoteIcon : schemaType.jsonType === "string" ? StringIcon : DocumentIcon;
+}
+function asFieldRefsByTypePath(fieldRefs) {
+  return fieldRefs.reduce(
+    (acc, ref) => ({ ...acc, [ref.key]: ref }),
+    {}
+  );
+}
+function getDocumentFieldRef(schemaType) {
+  return {
+    key: documentRootKey,
+    icon: schemaType.icon ?? DocumentIcon,
+    title: "The entire document",
+    path: [],
+    schemaType
+  };
+}
+function getFieldRefs(schemaType, parent, depth = 0) {
+  return depth >= maxDepth ? [] : schemaType.fields.filter((f) => !f.name.startsWith("_")).flatMap((field) => {
+    const path = parent ? [...parent.path, field.name] : [field.name], title = field.type.title ?? field.name, fieldRef = {
+      key: patchableKey(pathToString(path)),
+      path,
+      title: parent ? [parent.title, title].join(" / ") : title,
+      schemaType: field.type,
+      icon: getTypeIcon(field.type)
+    }, fields = field.type.jsonType === "object" ? getFieldRefs(field.type, fieldRef, depth + 1) : [], syntheticFields = field.type.jsonType === "array" ? getSyntheticFields(field.type, fieldRef, depth + 1) : [];
+    return isAssistSupported(field.type) ? [fieldRef, ...fields, ...syntheticFields] : [...fields, ...syntheticFields];
+  });
+}
+function getSyntheticFields(schemaType, parent, depth = 0) {
+  return depth >= maxDepth ? [] : schemaType.of.filter((itemType) => !isType(itemType, "block")).flatMap((itemType) => {
+    const segment = { _key: itemType.name }, title = itemType.title ?? itemType.name, path = parent ? [...parent.path, segment] : [segment], fieldRef = {
+      key: patchableKey(pathToString(path)),
+      path,
+      title: parent ? [parent.title, title].join(" / ") : title,
+      schemaType: itemType,
+      icon: getTypeIcon(itemType),
+      synthetic: !0
+    }, fields = itemType.jsonType === "object" ? getFieldRefs(itemType, fieldRef, depth + 1) : [];
+    return isAssistSupported(itemType) ? [fieldRef, ...fields] : fields;
+  });
+}
+function getTypePath(doc, pathString) {
+  if (!pathString)
+    return;
+  const path = stringToPath(pathString), currentPath = [];
+  let valid = !0;
+  const syntheticPath = path.map((segment) => {
+    if (currentPath.push(segment), isKeySegment(segment)) {
+      const match = extractWithPath(pathToString(currentPath), doc)[0], value = match?.value;
+      if (match && value && typeof value == "object" && "_type" in value)
+        return { _key: value._type };
+      valid = !1;
+    }
+    return segment;
+  });
+  return valid ? patchableKey(pathToString(syntheticPath)) : void 0;
+}
+function patchableKey(pathKey) {
+  return pathKey.replace(/[=]=/g, ":").replace(/[[\]]/g, "|").replace(/"/g, "");
+}
+function useTypePath(doc, pathString) {
+  return useMemo(() => getTypePath(doc, pathString), [doc, pathString]);
+}
+function useSelectedField(documentSchemaType, path) {
+  const { getFieldRefs: getFieldRefs2 } = useAiAssistanceConfig(), selectableFields = useMemo(
+    () => documentSchemaType && isObjectSchemaType(documentSchemaType) ? [getDocumentFieldRef(documentSchemaType), ...getFieldRefs2(documentSchemaType.name)] : [],
+    [documentSchemaType, getFieldRefs2]
+  );
+  return useMemo(() => path ? selectableFields?.find((f) => f.key === path) : void 0, [selectableFields, path]);
+}
+function getFieldTitle(field) {
+  const schemaType = field?.schemaType;
+  return field?.title ?? schemaType?.title ?? schemaType?.name ?? "Untitled";
+}
+function useAiPaneRouter() {
+  const paneRouter = usePaneRouter();
+  return useMemo(
+    () => ({ ...paneRouter, params: paneRouter.params ?? {} }),
+    [paneRouter]
+  );
+}
 function useAssistDocumentContextValue(documentId, documentType) {
-  const schema = useSchema(), documentSchemaType = useMemo(() => {
+  const schema = useSchema(), { getFieldRefs: getFieldRefs2, getFieldRefsByTypePath } = useAiAssistanceConfig(), documentSchemaType = useMemo(() => {
     const schemaType = schema.get(documentType);
     if (!schemaType)
       throw new Error(`Schema type "${documentType}" not found`);
     return schemaType;
-  }, [documentType, schema]), { fieldRefs, fieldRefsByTypePath } = useMemo(() => {
-    const fieldRefs2 = getFieldRefs(documentSchemaType), fieldRefsByTypePath2 = asFieldRefsByTypePath(fieldRefs2);
-    return {
-      fieldRefs: fieldRefs2,
-      fieldRefsByTypePath: fieldRefsByTypePath2
-    };
-  }, [documentSchemaType]), {
+  }, [documentType, schema]), { fieldRefs, fieldRefsByTypePath } = useMemo(() => ({
+    fieldRefs: getFieldRefs2(documentType),
+    fieldRefsByTypePath: getFieldRefsByTypePath(documentType)
+  }), [getFieldRefs2, getFieldRefsByTypePath, documentType]), {
     openInspector,
     closeInspector,
     inspector,
@@ -1601,7 +1445,10 @@ function InspectorOnboarding(props) {
   ] }) }) });
 }
 function FieldAutocomplete(props) {
-  const { id, schemaType, fieldPath, onSelect, includeDocument, filter: filter2 } = props, fieldRefs = useMemo(() => includeDocument ? getFieldRefsWithDocument(schemaType) : getFieldRefs(schemaType), [schemaType, includeDocument]), currentField = useMemo(
+  const { id, schemaType, fieldPath, onSelect, includeDocument, filter: filter2 } = props, { getFieldRefs: getFieldRefs2 } = useAiAssistanceConfig(), fieldRefs = useMemo(() => {
+    const refs = getFieldRefs2(schemaType.name);
+    return includeDocument ? [getDocumentFieldRef(schemaType), ...refs] : refs;
+  }, [schemaType, includeDocument, getFieldRefs2]), currentField = useMemo(
     () => fieldRefs.find((f) => f.key === fieldPath),
     [fieldPath, fieldRefs]
   ), autocompleteOptions = useMemo(
@@ -2602,6 +2449,195 @@ function FromLanguageRadio(props) {
     /* @__PURE__ */ jsx(Text, { children: radioLanguage.title ?? radioLanguage.id })
   ] }, langId);
 }
+const hiddenTypes = [
+  "any",
+  "array",
+  "block",
+  "boolean",
+  "crossDatasetReference",
+  "date",
+  "datetime",
+  "document",
+  "email",
+  "file",
+  "globalDocumentReference",
+  "image",
+  "number",
+  "object",
+  "reference",
+  "span",
+  "string",
+  "text",
+  "url",
+  "slug",
+  "geopoint",
+  "sanity.assetSourceData",
+  "sanity.imageAsset",
+  "sanity.fileAsset",
+  "sanity.imageCrop",
+  "sanity.imageHotspot",
+  "sanity.imageMetadata",
+  "sanity.imageDimensions",
+  "sanity.imagePalette",
+  "sanity.imagePaletteSwatch",
+  assistSerializedTypeName,
+  assistSerializedFieldTypeName,
+  "sanity-agent.job.document"
+], inlineTypes = ["document", "object", "image", "file"];
+function serializeSchema(schema, options2) {
+  const list = schema.getTypeNames().filter((t) => !(hiddenTypes.includes(t) || t.startsWith("sanity."))).map((t) => schema.get(t)).filter((t) => !!t).map((t) => getSchemaStub(t, schema, options2)).filter((t) => !("to" in t && t.to && !t.to.length || "of" in t && t.of && !t.of.length || "fields" in t && t.fields && !t.fields.length));
+  return list.sort((a, b) => (a?.name ?? "").localeCompare(b?.name ?? "")), list;
+}
+function getSchemaStub(schemaType, schema, options2) {
+  if (!schemaType.type?.name)
+    throw console.error("Missing type name", schemaType.type), new Error("Type is missing name!");
+  const baseSchema = {
+    // we dont need type or id when we send using POST, so leave these out to save bandwidth
+    ...options2?.leanFormat ? {} : { _id: `${assistSchemaIdPrefix}${schemaType.name}`, _type: assistSerializedTypeName },
+    name: schemaType.name,
+    title: schemaType.title,
+    type: schemaType.type.name,
+    ...getBaseFields(schema, schemaType, schemaType.type.name, options2)
+  };
+  return removeUndef(baseSchema);
+}
+function getBaseFields(schema, type, typeName, options2) {
+  const schemaOptions = removeUndef({
+    imagePromptField: type.options?.aiAssist?.imageInstructionField,
+    embeddingsIndex: type.options?.aiAssist?.embeddingsIndex
+  });
+  return removeUndef({
+    options: Object.keys(schemaOptions).length ? schemaOptions : void 0,
+    values: Array.isArray(type?.options?.list) ? type?.options?.list.map(
+      (v) => typeof v == "string" ? v : v.value ?? `${v.title}`
+    ) : void 0,
+    of: "of" in type && typeName === "array" ? arrayOf(type, schema, options2) : void 0,
+    to: "to" in type && typeName === "reference" ? refToTypeNames(type) : void 0,
+    fields: "fields" in type && inlineTypes.includes(typeName) ? serializeFields(schema, type, options2) : void 0,
+    annotations: typeName === "block" && "fields" in type ? serializeAnnotations(type, schema, options2) : void 0,
+    inlineOf: typeName === "block" && "fields" in type ? serializeInlineOf(type, schema, options2) : void 0,
+    hidden: typeof type.hidden == "function" ? "function" : type.hidden ? !0 : void 0,
+    readOnly: typeof type.readOnly == "function" ? "function" : type.readOnly ? !0 : void 0
+  });
+}
+function serializeFields(schema, schemaType, options2) {
+  return (schemaType.fieldsets ? schemaType.fieldsets.flatMap(
+    (fs) => fs.single ? fs.field : fs.fields.map((f) => ({
+      ...f,
+      type: {
+        ...f.type,
+        // if fieldset is (conditionally) hidden, the field must be considered the same way
+        // ie, if the field does not show up in conditionalMembers, it is hidden
+        // regardless of weather or not it is the field function or the fieldset function that hides it
+        hidden: typeof fs.hidden == "function" ? fs.hidden : fs.hidden ? !0 : f.type.hidden
+      }
+    }))
+  ) : schemaType.fields).filter((f) => !["sanity.imageHotspot", "sanity.imageCrop"].includes(f.type?.name ?? "")).filter((f) => isAssistSupported(f.type)).map((field) => serializeMember(schema, field.type, field.name, options2));
+}
+function serializeMember(schema, type, name, options2) {
+  const typeName = schema.get(type?.name) ? type.name : type.type?.name ?? "";
+  return removeUndef({
+    ...options2?.leanFormat ? {} : { _type: assistSerializedFieldTypeName },
+    name,
+    type: typeName,
+    title: type.title,
+    ...getBaseFields(schema, type, typeName, options2)
+  });
+}
+function serializeInlineOf(blockSchemaType, schema, options2) {
+  const childrenType = blockSchemaType.fields.find((f) => f.name === "children")?.type;
+  if (!(!childrenType || !isArraySchemaType(childrenType)))
+    return arrayOf(
+      {
+        of: childrenType.of.filter((t) => !isType(t, "span"))
+      },
+      schema,
+      options2
+    );
+}
+function serializeAnnotations(blockSchemaType, schema, options2) {
+  const marksType = blockSchemaType.fields.find((f) => f.name === "markDefs")?.type;
+  if (!(!marksType || !isArraySchemaType(marksType)))
+    return arrayOf(marksType, schema, options2);
+}
+function arrayOf(arrayType, schema, options2) {
+  return arrayType.of.filter((type) => isAssistSupported(type)).map((t) => serializeMember(schema, t, t.name, options2));
+}
+function refToTypeNames(type) {
+  return type.to.map((t) => ({
+    type: typed(t.name)
+  }));
+}
+function removeUndef(obj) {
+  return Object.keys(obj).forEach((key) => obj[key] === void 0 ? delete obj[key] : {}), obj;
+}
+function createFieldRefCache() {
+  const byType = {};
+  function getRefsForType(schemaType) {
+    const documentType = schemaType.name, cached = byType[documentType];
+    if (cached) return cached;
+    const fieldRefs = getFieldRefs(schemaType), fieldRefsByTypePath = asFieldRefsByTypePath(fieldRefs), refs = {
+      fieldRefs,
+      fieldRefsByTypePath
+    };
+    return byType[documentType] = refs, refs;
+  }
+  return getRefsForType;
+}
+function AiAssistanceConfigProvider(props) {
+  const [status, setStatus] = useState(), [error, setError] = useState(), apiClient = useApiClient(props.config?.__customApiClient), { getInstructStatus, loading: statusLoading } = useGetInstructStatus(apiClient), { initInstruct, loading: initLoading } = useInitInstruct(apiClient), schema = useSchema(), serializedTypes = useMemo(() => serializeSchema(schema, { leanFormat: !0 }), [schema]), { getFieldRefs: getFieldRefs2, getFieldRefsByTypePath } = useFieldRefGetters(schema);
+  useEffect(() => {
+    getInstructStatus().then((s) => setStatus(s)).catch((e) => {
+      console.error(e), setError(e);
+    });
+  }, [getInstructStatus]);
+  const init = useCallback(async () => {
+    setError(void 0);
+    try {
+      await initInstruct();
+      const status2 = await getInstructStatus();
+      setStatus(status2);
+    } catch (e) {
+      console.error("Failed to init ai assistance", e), setError(e);
+    }
+  }, [initInstruct, getInstructStatus, setStatus]), { config, children } = props, context = useMemo(() => ({
+    config,
+    status,
+    statusLoading,
+    init,
+    initLoading,
+    error,
+    serializedTypes,
+    getFieldRefs: getFieldRefs2,
+    getFieldRefsByTypePath
+  }), [
+    config,
+    status,
+    init,
+    statusLoading,
+    initLoading,
+    error,
+    serializedTypes,
+    getFieldRefs2,
+    getFieldRefsByTypePath
+  ]);
+  return /* @__PURE__ */ jsx(AiAssistanceConfigContext.Provider, { value: context, children });
+}
+function useFieldRefGetters(schema) {
+  return useMemo(() => {
+    const getForSchemaType = createFieldRefCache();
+    function getRefsForType(documentType) {
+      const schemaType = schema.get(documentType);
+      if (!schemaType)
+        throw new Error(`Schema type "${documentType}" not found`);
+      return getForSchemaType(schemaType);
+    }
+    return {
+      getFieldRefs: (documentType) => getRefsForType(documentType).fieldRefs,
+      getFieldRefsByTypePath: (documentType) => getRefsForType(documentType).fieldRefsByTypePath
+    };
+  }, [schema]);
+}
 function AssistLayout(props) {
   const [connectors, setConnectors] = useState([]);
   return /* @__PURE__ */ jsx(AiAssistanceConfigProvider, { config: props.config, children: /* @__PURE__ */ jsx(RunInstructionProvider, { children: /* @__PURE__ */ jsx(FieldTranslationProvider, { children: /* @__PURE__ */ jsxs(ConnectorsProvider, { onConnectorsChange: setConnectors, children: [
@@ -3187,7 +3223,7 @@ function BackToInstructionListLink() {
     }
   ) });
 }
-const EMPTY_FIELDS = [];
+const SelectedFieldContext = createContext(void 0), SelectedFieldContextProvider = SelectedFieldContext.Provider, EMPTY_FIELDS = [];
 function AssistDocumentForm(props) {
   return props.readOnly ? /* @__PURE__ */ jsx(Card, { border: !0, tone: "caution", padding: 2, children: /* @__PURE__ */ jsx(Text, { size: 1, children: " You do not have sufficient permissions to manage instructions." }) }) : /* @__PURE__ */ jsx(AssistDocumentFormEditable, { ...props });
 }
@@ -3687,19 +3723,25 @@ const contextDocumentSchema = defineType({
       components: {
         input: FieldRefPathInput
       },
-      validation: (rule) => rule.custom((value, context) => {
-        if (!value)
-          return "Please select a field";
-        try {
-          const docId = context.document?._id;
-          if (!docId)
-            return "Field reference cannot be used outside document inspector context. Could not resolve document id.";
-          const targetDocType = docId.replace(new RegExp(`^${assistDocumentIdPrefix}`), ""), schema = context.schema.get(targetDocType);
-          return schema ? getFieldRefsWithDocument(schema).find((r) => r.key === value) ? !0 : `Field with path "${value}" does not exist in the schema.` : `Field reference cannot be used outside document inspector context. Could not resolve schema: ${targetDocType}`;
-        } catch (e) {
-          return console.error("Failed to resolve field reference", e), "Invalid field reference.";
-        }
-      })
+      validation: (rule) => {
+        const getForSchemaType = createFieldRefCache();
+        return rule.custom((value, context) => {
+          if (!value)
+            return "Please select a field";
+          try {
+            const docId = context.document?._id;
+            if (!docId)
+              return "Field reference cannot be used outside document inspector context. Could not resolve document id.";
+            const targetDocType = docId.replace(new RegExp(`^${assistDocumentIdPrefix}`), ""), schema = context.schema.get(targetDocType);
+            if (!schema)
+              return `Field reference cannot be used outside document inspector context. Could not resolve schema: ${targetDocType}`;
+            const { fieldRefs } = getForSchemaType(schema);
+            return fieldRefs.find((r) => r.key === value) ? !0 : `Field with path "${value}" does not exist in the schema.`;
+          } catch (e) {
+            return console.error("Failed to resolve field reference", e), "Invalid field reference.";
+          }
+        });
+      }
     })
   ],
   preview: {