@sanity/assist 1.1.4 → 1.2.1

package/dist/index.esm.js

@@ -1,13 +1,14 @@
 import { jsx, jsxs, Fragment } from 'react/jsx-runtime';
-import { useClient, typed, useSchema, useDocumentStore, useDocumentPresence, createPatchChannel, FormBuilder, fromMutationPatches, pathToString, isObjectSchemaType, useEditState, useCurrentUser, useValidationStatus, StatusButton, FormFieldHeaderText, PatchEvent, unset, PresenceOverlay, VirtualizerScrollInstanceProvider, stringToPath, isKeySegment, useColorSchemeValue, isArraySchemaType, useFormCallbacks, set, FormCallbacksProvider, FormInput, setIfMissing, insert, ObjectInputMember, defineType, defineField, defineArrayMember, getPublishedId, definePlugin } from 'sanity';
-import { Card, Stack, Box, Button, Spinner, Flex, Label, focusFirstDescendant, Text, useClickOutside, Popover, useLayer, useGlobalKeyDown, useToast, Dialog, Tooltip, TextArea, Container, Badge, useTheme, rgba, ThemeProvider, ErrorBoundary, Switch, MenuButton, Menu, MenuItem, Breadcrumbs, Autocomplete } from '@sanity/ui';
+import { useClient, typed, useSchema, useDocumentStore, useDocumentPresence, createPatchChannel, FormBuilder, fromMutationPatches, pathToString, isObjectSchemaType, stringToPath, isKeySegment, useEditState, useCurrentUser, useValidationStatus, StatusButton, FormFieldHeaderText, PatchEvent, unset, set, useFormCallbacks, FormCallbacksProvider, FormInput, setIfMissing, insert, PresenceOverlay, VirtualizerScrollInstanceProvider, useColorSchemeValue, isArraySchemaType, ObjectInputMember, defineType, defineField, defineArrayMember, getPublishedId, definePlugin } from 'sanity';
+import { Card, Stack, Box, Button, Spinner, Flex, Label, focusFirstDescendant, Text, useClickOutside, Popover, useLayer, useGlobalKeyDown, useToast, Dialog, Tooltip, TextArea, Container, Autocomplete, Breadcrumbs, Badge, useTheme, rgba, ThemeProvider, ErrorBoundary, Switch, MenuButton, Menu, MenuItem } from '@sanity/ui';
 import { useState, useRef, useEffect, useMemo, useCallback, createContext, useReducer, forwardRef, createElement, useContext, useId } from 'react';
-import { SyncIcon, DocumentIcon, LinkIcon, ImageIcon, BlockContentIcon, OlistIcon, BlockquoteIcon, StringIcon, ErrorOutlineIcon, CheckmarkCircleIcon, CloseCircleIcon, ClockIcon, PlayIcon, SparklesIcon, RetryIcon, ArrowRightIcon, CloseIcon, CheckmarkIcon, ArrowLeftIcon, icons, SearchIcon, TokenIcon, DocumentTextIcon, ThListIcon, CodeIcon, ComposeIcon, LockIcon, ControlsIcon } from '@sanity/icons';
+import { SyncIcon, DocumentIcon, LinkIcon, ImageIcon, BlockContentIcon, OlistIcon, BlockquoteIcon, StringIcon, ErrorOutlineIcon, CheckmarkCircleIcon, CloseCircleIcon, ClockIcon, PlayIcon, SparklesIcon, ArrowLeftIcon, SearchIcon, RetryIcon, ArrowRightIcon, CloseIcon, CheckmarkIcon, icons, TokenIcon, DocumentTextIcon, ThListIcon, CodeIcon, ComposeIcon, LockIcon, ControlsIcon } from '@sanity/icons';
 import { mergeMap, share, take, filter, distinctUntilChanged, catchError, tap } from 'rxjs/operators';
 import isEqual from 'react-fast-compare';
 import { throwError, of, partition, merge, switchMap, delay, defer } from 'rxjs';
 import { exhaustMapToWithTrailing } from 'rxjs-exhaustmap-with-trailing';
 import { useDocumentPane, usePaneRouter, DocumentInspectorHeader, DocumentPaneProvider } from 'sanity/desk';
+import { extractWithPath } from '@sanity/mutator';
 import styled, { keyframes } from 'styled-components';
 import { minutesToMilliseconds, isAfter, addSeconds } from 'date-fns';
 import formatDistanceToNow from 'date-fns/formatDistanceToNow';
@@ -506,7 +507,7 @@ function prepareRebaseEvent(event) {
 }
 const SelectedFieldContext = createContext(void 0);
 const SelectedFieldContextProvider = SelectedFieldContext.Provider;
-const maxDepth = 4;
+const maxDepth = 6;
 function getTypeIcon(schemaType) {
   let t = schemaType;
   while (t) {
@@ -542,19 +543,76 @@ function getFieldRefs(schemaType, parent) {
     const path = parent ? [...parent.path, field.name] : [field.name];
     const title = (_a = field.type.title) != null ? _a : field.name;
     const fieldRef = {
-      key: pathToString(path),
+      key: patchableKey(pathToString(path)),
       path,
       title: parent ? [parent.title, title].join(" / ") : title,
       schemaType: field.type,
       icon: getTypeIcon(field.type)
     };
     const fields = field.type.jsonType === "object" ? getFieldRefs(field.type, fieldRef, depth + 1) : [];
+    const syntheticFields = field.type.jsonType === "array" ? getSyntheticFields(field.type, fieldRef, depth + 1) : [];
     if (!isAssistSupported(field.type, true)) {
+      return [...fields, ...syntheticFields];
+    }
+    return [fieldRef, ...fields, ...syntheticFields];
+  });
+}
+function getSyntheticFields(schemaType, parent) {
+  let depth = arguments.length > 2 && arguments[2] !== undefined ? arguments[2] : 0;
+  if (depth >= maxDepth) {
+    return [];
+  }
+  return schemaType.of.filter(itemType => !isType(itemType, "block")).flatMap(itemType => {
+    var _a;
+    const segment = {
+      _key: itemType.name
+    };
+    const title = (_a = itemType.title) != null ? _a : itemType.name;
+    const path = parent ? [...parent.path, segment] : [segment];
+    const fieldRef = {
+      key: patchableKey(pathToString(path)),
+      path,
+      title: parent ? [parent.title, title].join(" / ") : title,
+      schemaType: itemType,
+      icon: getTypeIcon(itemType),
+      synthetic: true
+    };
+    const fields = itemType.jsonType === "object" ? getFieldRefs(itemType, fieldRef, depth + 1) : [];
+    if (!isAssistSupported(itemType, true)) {
       return fields;
     }
     return [fieldRef, ...fields];
   });
 }
+function getTypePath(doc, pathString) {
+  if (!pathString) {
+    return void 0;
+  }
+  const path = stringToPath(pathString);
+  const currentPath = [];
+  let valid = true;
+  const syntheticPath = path.map(segment => {
+    currentPath.push(segment);
+    if (isKeySegment(segment)) {
+      const match = extractWithPath(pathToString(currentPath), doc)[0];
+      const value = match == null ? void 0 : match.value;
+      if (match && value && typeof value === "object" && "_type" in value) {
+        return {
+          _key: value._type
+        };
+      }
+      valid = false;
+    }
+    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(() => {
@@ -562,9 +620,9 @@ function useSelectedField(documentSchemaType, path) {
   }, [selectableFields, path]);
 }
 function getFieldTitle(field) {
-  var _a, _b;
+  var _a, _b, _c;
   const schemaType = field == null ? void 0 : field.schemaType;
-  return (_b = (_a = schemaType == null ? void 0 : schemaType.title) != null ? _a : schemaType == null ? void 0 : schemaType.name) != null ? _b : "Untitled";
+  return (_c = (_b = (_a = field == null ? void 0 : field.title) != null ? _a : schemaType == null ? void 0 : schemaType.title) != null ? _b : schemaType == null ? void 0 : schemaType.name) != null ? _c : "Untitled";
 }
 function useAiPaneRouter() {
   const paneRouter = usePaneRouter();
@@ -1351,32 +1409,338 @@ function InspectorOnboarding(props) {
     })
   });
 }
-const inspectorOnboardingKey = "sanityStudio:assist:inspector:onboarding:dismissed";
-const fieldOnboardingKey = "sanityStudio:assist:field:onboarding:dismissed";
-function isFeatureOnboardingDismissed(featureKey) {
-  if (typeof localStorage === "undefined") {
-    return false;
-  }
-  const value = localStorage.getItem(featureKey);
-  return value === "true";
-}
-function dismissFeatureOnboarding(featureKey) {
-  if (typeof localStorage === "undefined") {
-    return;
-  }
-  localStorage.setItem(featureKey, "true");
-}
-function useOnboardingFeature(featureKey) {
-  const [showOnboarding, setShowOnboarding] = useState(() => !isFeatureOnboardingDismissed(featureKey));
-  const dismissOnboarding = useCallback(() => {
-    setShowOnboarding(false);
-    dismissFeatureOnboarding(featureKey);
-  }, [setShowOnboarding, featureKey]);
-  return {
-    showOnboarding,
-    dismissOnboarding
-  };
-}
+const inspectorOnboardingKey = "sanityStudio:assist:inspector:onboarding:dismissed";
+const fieldOnboardingKey = "sanityStudio:assist:field:onboarding:dismissed";
+function isFeatureOnboardingDismissed(featureKey) {
+  if (typeof localStorage === "undefined") {
+    return false;
+  }
+  const value = localStorage.getItem(featureKey);
+  return value === "true";
+}
+function dismissFeatureOnboarding(featureKey) {
+  if (typeof localStorage === "undefined") {
+    return;
+  }
+  localStorage.setItem(featureKey, "true");
+}
+function useOnboardingFeature(featureKey) {
+  const [showOnboarding, setShowOnboarding] = useState(() => !isFeatureOnboardingDismissed(featureKey));
+  const dismissOnboarding = useCallback(() => {
+    setShowOnboarding(false);
+    dismissFeatureOnboarding(featureKey);
+  }, [setShowOnboarding, featureKey]);
+  return {
+    showOnboarding,
+    dismissOnboarding
+  };
+}
+function BackToInstructionListLink() {
+  const {
+    openInspector
+  } = useDocumentPane();
+  const goBack = useCallback(() => openInspector(aiInspectorId, {
+    [instructionParam]: void 0
+  }), [openInspector]);
+  return /* @__PURE__ */jsx("div", {
+    children: /* @__PURE__ */jsx(Button, {
+      as: "a",
+      fontSize: 1,
+      icon: ArrowLeftIcon,
+      mode: "bleed",
+      padding: 1,
+      space: 2,
+      onClick: goBack,
+      text: "  Instructions",
+      textAlign: "left"
+    })
+  });
+}
+const EMPTY_FIELDS = [];
+const TypePathContext = createContext(void 0);
+function AssistDocumentForm(props) {
+  const {
+    onChange
+  } = props;
+  const value = props.value;
+  const id = value == null ? void 0 : value._id;
+  const fields = value == null ? void 0 : value.fields;
+  const targetDocumentType = useMemo(() => {
+    if (!id) {
+      return void 0;
+    }
+    return documentTypeFromAiDocumentId(id);
+  }, [id]);
+  const {
+    params,
+    setParams
+  } = useAiPaneRouter();
+  const pathKey = params[fieldPathParam];
+  const typePath = useContext(TypePathContext);
+  const instruction = params[instructionParam];
+  const activeKey = useMemo(() => {
+    var _a;
+    if (!typePath) {
+      return void 0;
+    }
+    return (_a = (fields != null ? fields : EMPTY_FIELDS).find(f => f.path === typePath)) == null ? void 0 : _a._key;
+  }, [fields, typePath]);
+  const activePath = useMemo(() => {
+    if (!activeKey) {
+      return void 0;
+    }
+    const base = ["fields", {
+      _key: activeKey
+    }];
+    return instruction ? [...base, "instructions", {
+      _key: instruction
+    }] : base;
+  }, [activeKey, instruction]);
+  const schema = useSchema();
+  const documentSchema = useMemo(() => {
+    if (!targetDocumentType) {
+      return void 0;
+    }
+    return schema.get(targetDocumentType);
+  }, [schema, targetDocumentType]);
+  const fieldSchema = useSelectedSchema(pathKey, documentSchema);
+  const context = useMemo(() => ({
+    documentSchema,
+    fieldSchema: fieldSchema != null ? fieldSchema : documentSchema
+  }), [fieldSchema, documentSchema]);
+  const title = value == null ? void 0 : value.title;
+  useEffect(() => {
+    var _a;
+    if (!title && documentSchema && !(id == null ? void 0 : id.startsWith("drafts."))) {
+      onChange(set((_a = documentSchema.title) != null ? _a : documentSchema.name, ["title"]));
+    }
+  }, [title, documentSchema, onChange, id]);
+  const fieldExists = !!(fields == null ? void 0 : fields.some(f => f._key === typePath));
+  const {
+    onPathOpen,
+    ...formCallbacks
+  } = useFormCallbacks();
+  const newCallbacks = useMemo(() => ({
+    ...formCallbacks,
+    onPathOpen: path => {
+      var _a;
+      if (!instruction && path.length === 4 && path[2] === "instructions") {
+        setParams(typed({
+          ...params,
+          [instructionParam]: (_a = path[3]) == null ? void 0 : _a._key
+        }));
+        onPathOpen([]);
+      } else {
+        setTimeout(() => onPathOpen(path), 0);
+      }
+    }
+  }), [formCallbacks, onPathOpen, params, setParams, instruction]);
+  useEffect(() => {
+    if (activePath && !instruction) {
+      onPathOpen([]);
+    }
+  }, [activePath, instruction, onPathOpen]);
+  return /* @__PURE__ */jsx(SelectedFieldContextProvider, {
+    value: context,
+    children: /* @__PURE__ */jsxs(Stack, {
+      space: 5,
+      children: [/* @__PURE__ */jsx(FieldsInitializer, {
+        pathKey: typePath,
+        activePath,
+        fieldExists,
+        onChange
+      }, typePath), instruction && /* @__PURE__ */jsx(BackToInstructionListLink, {}), activePath && /* @__PURE__ */jsx(FormCallbacksProvider, {
+        ...newCallbacks,
+        children: /* @__PURE__ */jsx("div", {
+          style: {
+            lineHeight: "1.25em"
+          },
+          children: /* @__PURE__ */jsx(FormInput, {
+            ...props,
+            absolutePath: activePath
+          })
+        })
+      }), !activePath && props.renderDefault(props)]
+    })
+  });
+}
+function useSelectedSchema(fieldPath, documentSchema) {
+  return useMemo(() => {
+    if (!fieldPath) {
+      return void 0;
+    }
+    if (fieldPath === documentRootKey) {
+      return documentSchema;
+    }
+    const path = stringToPath(fieldPath);
+    let currentSchema = documentSchema;
+    for (let i = 0; i < path.length; i++) {
+      const segment = path[i];
+      const field = currentSchema == null ? void 0 : currentSchema.fields.find(f => f.name === segment);
+      if (!field) {
+        return void 0;
+      }
+      if (i === path.length - 1) {
+        return field.type;
+      }
+      if (field.type.jsonType !== "object") {
+        return void 0;
+      }
+      currentSchema = field.type;
+    }
+    return currentSchema;
+  }, [documentSchema, fieldPath]);
+}
+function FieldsInitializer(_ref7) {
+  let {
+    pathKey,
+    activePath,
+    fieldExists,
+    onChange
+  } = _ref7;
+  const initialized = useRef(false);
+  useEffect(() => {
+    if (initialized.current || fieldExists || activePath || !pathKey) {
+      return;
+    }
+    onChange([setIfMissing([], ["fields"]), insert([typed({
+      _key: pathKey,
+      _type: assistFieldTypeName,
+      path: pathKey
+    })], "after", ["fields", -1])]);
+    initialized.current = true;
+  }, [activePath, onChange, pathKey, fieldExists]);
+  return null;
+}
+function FieldAutocomplete(props) {
+  const {
+    id,
+    schemaType,
+    fieldPath,
+    onSelect,
+    includeDocument,
+    filter
+  } = props;
+  const fieldRefs = useMemo(() => {
+    if (includeDocument) {
+      return getFieldRefsWithDocument(schemaType);
+    }
+    return getFieldRefs(schemaType);
+  }, [schemaType, includeDocument]);
+  const currentField = useMemo(() => fieldRefs.find(f => f.key === fieldPath), [fieldPath, fieldRefs]);
+  const autocompleteOptions = useMemo(() => fieldRefs.filter(field => filter ? filter(field) : true).map(field => ({
+    value: field.key,
+    field
+  })), [fieldRefs, filter]);
+  const renderOption = useCallback(option => {
+    const {
+      value,
+      field
+    } = option;
+    if (!value) {
+      return /* @__PURE__ */jsx(Card, {
+        as: "button",
+        padding: 3,
+        radius: 1,
+        children: /* @__PURE__ */jsx(Text, {
+          accent: true,
+          size: 1,
+          children: option.value
+        })
+      });
+    }
+    if (isType(field.schemaType, "document")) {
+      return /* @__PURE__ */jsx(Card, {
+        as: "button",
+        padding: 3,
+        radius: 1,
+        children: /* @__PURE__ */jsx(Text, {
+          size: 1,
+          weight: "semibold",
+          children: "The entire document"
+        })
+      });
+    }
+    return /* @__PURE__ */jsx(Card, {
+      as: "button",
+      padding: 3,
+      radius: 1,
+      children: /* @__PURE__ */jsxs(Flex, {
+        gap: 3,
+        children: [/* @__PURE__ */jsx(Text, {
+          size: 1,
+          children: createElement(field.icon)
+        }), /* @__PURE__ */jsx(FieldTitle, {
+          field
+        })]
+      })
+    });
+  }, []);
+  const renderValue = useCallback((value, option) => {
+    var _a;
+    return (_a = option == null ? void 0 : option.field.title) != null ? _a : value;
+  }, []);
+  const filterOption = useCallback((query, option) => {
+    var _a, _b, _c;
+    const lQuery = query.toLowerCase();
+    return ((_a = option == null ? void 0 : option.value) == null ? void 0 : _a.toLowerCase().includes(lQuery)) || ((_c = (_b = option == null ? void 0 : option.field) == null ? void 0 : _b.title) == null ? void 0 : _c.toLowerCase().includes(lQuery));
+  }, []);
+  return /* @__PURE__ */jsx(Autocomplete, {
+    fontSize: 1,
+    icon: currentField ? currentField.icon : SearchIcon,
+    onChange: onSelect,
+    openButton: true,
+    id,
+    options: autocompleteOptions,
+    placeholder: "Search for a field",
+    radius: 2,
+    renderOption,
+    renderValue,
+    value: currentField == null ? void 0 : currentField.key,
+    filterOption
+  });
+}
+function FieldTitle(props) {
+  const splitTitle = props.field.title.split("/");
+  return /* @__PURE__ */jsx(Box, {
+    flex: "none",
+    children: /* @__PURE__ */jsxs(Breadcrumbs, {
+      style: {
+        display: "flex",
+        flexWrap: "wrap",
+        alignItems: "center",
+        marginTop: "-4px"
+      },
+      separator: /* @__PURE__ */jsx(Box, {
+        marginTop: 1,
+        children: /* @__PURE__ */jsx(Text, {
+          muted: true,
+          size: 1,
+          children: "/"
+        })
+      }),
+      space: 1,
+      children: [splitTitle.slice(0, splitTitle.length - 1).map((pt, i) =>
+      // eslint-disable-next-line react/no-array-index-key
+      /* @__PURE__ */
+      jsx(Box, {
+        marginTop: 1,
+        children: /* @__PURE__ */jsx(Text, {
+          muted: true,
+          size: 1,
+          children: pt.trim()
+        })
+      }, i)), /* @__PURE__ */jsx(Box, {
+        marginTop: 1,
+        children: /* @__PURE__ */jsx(Text, {
+          size: 1,
+          weight: "medium",
+          children: splitTitle[splitTitle.length - 1]
+        })
+      })]
+    })
+  });
+}
 var __freeze$4 = Object.freeze;
 var __defProp$4 = Object.defineProperty;
 var __template$4 = (cooked, raw) => __freeze$4(__defProp$4(cooked, "raw", {
@@ -1514,6 +1878,7 @@ function AssistInspector(props) {
   const {
     documentId,
     documentType,
+    value: docValue,
     schemaType,
     onChange: documentOnChange
   } = documentPane;
@@ -1529,13 +1894,14 @@ function AssistInspector(props) {
     documentOnChange,
     isDocAssistable: isDocAssistable(schemaType, published, draft)
   });
-  const selectedField = useSelectedField(schemaType, params[fieldPathParam]);
+  const typePath = useTypePath(docValue, pathKey != null ? pathKey : "");
+  const selectedField = useSelectedField(schemaType, typePath);
   const aiDocId = assistDocumentId(documentType);
   const assistDocument = useStudioAssistDocument({
     documentId,
     schemaType
   });
-  const assistField = (_a2 = assistDocument == null ? void 0 : assistDocument.fields) == null ? void 0 : _a2.find(f => f.path === pathKey);
+  const assistField = (_a2 = assistDocument == null ? void 0 : assistDocument.fields) == null ? void 0 : _a2.find(f => f.path === typePath);
   const instruction = (_b2 = assistField == null ? void 0 : assistField.instructions) == null ? void 0 : _b2.find(i => i._key === instructionKey);
   const tasks = useMemo(() => {
     var _a3;
@@ -1567,12 +1933,13 @@ function AssistInspector(props) {
       type: assistDocumentTypeName
     }
   }), [aiDocId]);
-  const runCurrentInstruction = useCallback(() => instruction && pathKey && requestRunInstruction({
+  const runCurrentInstruction = useCallback(() => instruction && pathKey && typePath && requestRunInstruction({
     documentId: assistableDocId,
     path: pathKey,
+    typePath,
     assistDocumentId: assistDocumentId(documentType),
     instruction
-  }), [instruction, pathKey, documentType, assistableDocId, requestRunInstruction]);
+  }), [pathKey, instruction, typePath, documentType, assistableDocId, requestRunInstruction]);
   const Region = useCallback(_props => {
     return /* @__PURE__ */jsx("div", {
       ..._props,
@@ -1603,6 +1970,7 @@ function AssistInspector(props) {
     },
     children: [/* @__PURE__ */jsx(AiInspectorHeader, {
       onClose: props.onClose,
+      field: selectedField,
       fieldTitle: getFieldTitle(selectedField)
     }), /* @__PURE__ */jsx(Card, {
       as: Region,
@@ -1618,15 +1986,18 @@ function AssistInspector(props) {
           children: /* @__PURE__ */jsx(PresenceOverlay, {
             children: /* @__PURE__ */jsx(Box, {
               padding: 4,
-              children: selectedField && /* @__PURE__ */jsx(VirtualizerScrollInstanceProvider, {
-                scrollElement: boundary.current,
-                containerElement: boundary,
-                children: /* @__PURE__ */jsx(DocumentPaneProvider, {
-                  paneKey: documentPane.paneKey,
-                  index: documentPane.index,
-                  itemId: "ai",
-                  pane: paneNode,
-                  children: /* @__PURE__ */jsx(DocumentForm, {})
+              children: selectedField && /* @__PURE__ */jsx(TypePathContext.Provider, {
+                value: typePath,
+                children: /* @__PURE__ */jsx(VirtualizerScrollInstanceProvider, {
+                  scrollElement: boundary.current,
+                  containerElement: boundary,
+                  children: /* @__PURE__ */jsx(DocumentPaneProvider, {
+                    paneKey: documentPane.paneKey,
+                    index: documentPane.index,
+                    itemId: "ai",
+                    pane: paneNode,
+                    children: /* @__PURE__ */jsx(DocumentForm, {})
+                  })
                 })
               })
             })
@@ -1684,6 +2055,7 @@ function AssistInspector(props) {
 function AiInspectorHeader(props) {
   const {
     onClose,
+    field,
     fieldTitle
   } = props;
   const {
@@ -1704,18 +2076,25 @@ function AiInspectorHeader(props) {
         children: /* @__PURE__ */jsxs(Flex, {
           gap: 1,
           align: "center",
-          children: [/* @__PURE__ */jsx(Text, {
-            size: 1,
-            weight: "semibold",
-            children: "AI instructions for"
+          wrap: "wrap",
+          style: {
+            marginTop: "-4px"
+          },
+          children: [/* @__PURE__ */jsx(Box, {
+            marginTop: 1,
+            children: /* @__PURE__ */jsx(Text, {
+              size: 1,
+              weight: "semibold",
+              children: "AI instructions for"
+            })
           }), /* @__PURE__ */jsx(Card, {
             radius: 2,
             border: true,
             padding: 1,
-            style: {
-              margin: "-4px 0"
-            },
-            children: /* @__PURE__ */jsx(Text, {
+            marginTop: 1,
+            children: field ? /* @__PURE__ */jsx(FieldTitle, {
+              field
+            }) : /* @__PURE__ */jsx(Text, {
               size: 1,
               weight: "semibold",
               children: fieldTitle
@@ -1745,10 +2124,10 @@ const assistInspector = {
     showAsAction: false
   }),
   component: AssistInspectorWrapper,
-  onClose(_ref7) {
+  onClose(_ref8) {
     let {
       params
-    } = _ref7;
+    } = _ref8;
     return {
       params: typed({
         ...params,
@@ -1821,11 +2200,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(_ref8, ref) {
+const FadeInContent = forwardRef(function FadeInContent2(_ref9, ref) {
   let {
     children,
     durationMs = 250
-  } = _ref8;
+  } = _ref9;
   return /* @__PURE__ */jsx(FadeInDiv, {
     ref,
     style: {
@@ -2850,291 +3229,119 @@ function SafeValueInput(props) {
 function ErrorWrapper(props) {
   const {
     onChange
-  } = props;
-  const [error, setError] = useState();
-  const catchError = useCallback(params => {
-    setError(params.error);
-  }, [setError]);
-  const unsetValue = useCallback(() => onChange(PatchEvent.from(unset())), [onChange]);
-  const dismiss = useCallback(() => setError(void 0), []);
-  const catcher = /* @__PURE__ */jsx(ErrorBoundary, {
-    onCatch: catchError,
-    children: props.children
-  });
-  return error ? /* @__PURE__ */jsx(Card, {
-    border: true,
-    tone: "critical",
-    padding: 2,
-    contentEditable: false,
-    children: /* @__PURE__ */jsxs(Stack, {
-      space: 3,
-      children: [/* @__PURE__ */jsx(Text, {
-        muted: true,
-        weight: "semibold",
-        children: "An error occurred."
-      }), /* @__PURE__ */jsx(WrapPreCard, {
-        flex: 1,
-        padding: 2,
-        tone: "critical",
-        border: true,
-        children: catcher
-      }), /* @__PURE__ */jsxs(Flex, {
-        width: "fill",
-        flex: 1,
-        gap: 3,
-        children: [/* @__PURE__ */jsx(Box, {
-          flex: 1,
-          children: /* @__PURE__ */jsx(Button, {
-            text: "Dismiss",
-            onClick: dismiss,
-            tone: "primary"
-          })
-        }), /* @__PURE__ */jsx(Button, {
-          text: "Unset value",
-          onClick: unsetValue,
-          tone: "critical"
-        })]
-      })]
-    })
-  }) : catcher;
-}
-function PteValueFixer(props) {
-  const isPortableText = useMemo(() => isArraySchemaType(props.schemaType) && isPortableTextArray(props.schemaType), [props.schemaType]);
-  const value = props.value;
-  if (isPortableText && value && !value.length) {
-    return props.renderDefault({
-      ...props,
-      value: void 0
-    });
-  }
-  return props.renderDefault(props);
-}
-function AssistFormBlock(props) {
-  const presence = useAssistPresence(props.path, true);
-  const {
-    onChange
-  } = useFormCallbacks();
-  const key = props.value._key;
-  const localOnChange = useCallback(patchEvent => {
-    if (!key) {
-      return;
-    }
-    onChange(PatchEvent.from(patchEvent).prefixAll({
-      _key: key
-    }));
-  }, [onChange, key]);
-  return /* @__PURE__ */jsx(ErrorWrapper, {
-    onChange: localOnChange,
-    children: /* @__PURE__ */jsxs(Flex, {
-      align: "center",
-      justify: "space-between",
-      children: [/* @__PURE__ */jsx(Box, {
-        flex: 1,
-        children: props.renderDefault(props)
-      }), presence.map(pre => /* @__PURE__ */jsx(AiFieldPresence, {
-        presence: pre
-      }, pre.lastActiveAt))]
-    })
-  });
-}
-function AssistItem(props) {
-  const {
-    path
-  } = props;
-  const presence = useAssistPresence(path, true);
-  return /* @__PURE__ */jsxs(Flex, {
-    align: "center",
-    width: "fill",
-    style: {
-      position: "relative"
-    },
-    children: [/* @__PURE__ */jsx(Box, {
-      flex: 1,
-      children: props.renderDefault({
-        ...props
-      })
-    }), presence.map(pre => /* @__PURE__ */jsx(Box, {
-      style: {
-        position: "absolute",
-        right: 35
-      },
-      children: /* @__PURE__ */jsx(AiFieldPresence, {
-        presence: pre
-      })
-    }, pre.user.id))]
-  });
-}
-function BackToInstructionListLink() {
-  const {
-    openInspector
-  } = useDocumentPane();
-  const goBack = useCallback(() => openInspector(aiInspectorId, {
-    [instructionParam]: void 0
-  }), [openInspector]);
-  return /* @__PURE__ */jsx("div", {
-    children: /* @__PURE__ */jsx(Button, {
-      as: "a",
-      fontSize: 1,
-      icon: ArrowLeftIcon,
-      mode: "bleed",
-      padding: 1,
-      space: 2,
-      onClick: goBack,
-      text: "  Instructions",
-      textAlign: "left"
-    })
-  });
-}
-const EMPTY_FIELDS = [];
-function AssistDocumentForm(props) {
-  const {
-    onChange
-  } = props;
-  const value = props.value;
-  const id = value == null ? void 0 : value._id;
-  const fields = value == null ? void 0 : value.fields;
-  const targetDocumentType = useMemo(() => {
-    if (!id) {
-      return void 0;
-    }
-    return documentTypeFromAiDocumentId(id);
-  }, [id]);
-  const {
-    params,
-    setParams
-  } = useAiPaneRouter();
-  const pathKey = params[fieldPathParam];
-  const instruction = params[instructionParam];
-  const activeKey = useMemo(() => {
-    var _a;
-    return (_a = (fields != null ? fields : EMPTY_FIELDS).find(f => f.path === pathKey)) == null ? void 0 : _a._key;
-  }, [fields, pathKey]);
-  const activePath = useMemo(() => {
-    if (!activeKey) {
-      return void 0;
-    }
-    const base = ["fields", {
-      _key: activeKey
-    }];
-    return instruction ? [...base, "instructions", {
-      _key: instruction
-    }] : base;
-  }, [activeKey, instruction]);
-  const schema = useSchema();
-  const documentSchema = useMemo(() => {
-    if (!targetDocumentType) {
-      return void 0;
-    }
-    return schema.get(targetDocumentType);
-  }, [schema, targetDocumentType]);
-  const fieldSchema = useSelectedSchema(pathKey, documentSchema);
-  const context = useMemo(() => ({
-    documentSchema,
-    fieldSchema: fieldSchema != null ? fieldSchema : documentSchema
-  }), [fieldSchema, documentSchema]);
-  const title = value == null ? void 0 : value.title;
-  useEffect(() => {
-    var _a;
-    if (!title && documentSchema && !(id == null ? void 0 : id.startsWith("drafts."))) {
-      onChange(set((_a = documentSchema.title) != null ? _a : documentSchema.name, ["title"]));
-    }
-  }, [title, documentSchema, onChange, id]);
-  const fieldExists = !!(fields == null ? void 0 : fields.some(f => f._key === pathKey));
-  const {
-    onPathOpen,
-    ...formCallbacks
-  } = useFormCallbacks();
-  const newCallbacks = useMemo(() => ({
-    ...formCallbacks,
-    onPathOpen: path => {
-      var _a;
-      if (!instruction && path.length === 4 && path[2] === "instructions") {
-        setParams(typed({
-          ...params,
-          [instructionParam]: (_a = path[3]) == null ? void 0 : _a._key
-        }));
-        onPathOpen([]);
-      } else {
-        setTimeout(() => onPathOpen(path), 0);
-      }
-    }
-  }), [formCallbacks, onPathOpen, params, setParams, instruction]);
-  useEffect(() => {
-    if (activePath && !instruction) {
-      onPathOpen([]);
-    }
-  }, [activePath, instruction, onPathOpen]);
-  return /* @__PURE__ */jsx(SelectedFieldContextProvider, {
-    value: context,
+  } = props;
+  const [error, setError] = useState();
+  const catchError = useCallback(params => {
+    setError(params.error);
+  }, [setError]);
+  const unsetValue = useCallback(() => onChange(PatchEvent.from(unset())), [onChange]);
+  const dismiss = useCallback(() => setError(void 0), []);
+  const catcher = /* @__PURE__ */jsx(ErrorBoundary, {
+    onCatch: catchError,
+    children: props.children
+  });
+  return error ? /* @__PURE__ */jsx(Card, {
+    border: true,
+    tone: "critical",
+    padding: 2,
+    contentEditable: false,
     children: /* @__PURE__ */jsxs(Stack, {
-      space: 5,
-      children: [/* @__PURE__ */jsx(FieldsInitializer, {
-        pathKey,
-        activePath,
-        fieldExists,
-        onChange
-      }, pathKey), instruction && /* @__PURE__ */jsx(BackToInstructionListLink, {}), activePath && /* @__PURE__ */jsx(FormCallbacksProvider, {
-        ...newCallbacks,
-        children: /* @__PURE__ */jsx("div", {
-          style: {
-            lineHeight: "1.25em"
-          },
-          children: /* @__PURE__ */jsx(FormInput, {
-            ...props,
-            absolutePath: activePath
+      space: 3,
+      children: [/* @__PURE__ */jsx(Text, {
+        muted: true,
+        weight: "semibold",
+        children: "An error occurred."
+      }), /* @__PURE__ */jsx(WrapPreCard, {
+        flex: 1,
+        padding: 2,
+        tone: "critical",
+        border: true,
+        children: catcher
+      }), /* @__PURE__ */jsxs(Flex, {
+        width: "fill",
+        flex: 1,
+        gap: 3,
+        children: [/* @__PURE__ */jsx(Box, {
+          flex: 1,
+          children: /* @__PURE__ */jsx(Button, {
+            text: "Dismiss",
+            onClick: dismiss,
+            tone: "primary"
           })
-        })
-      }), !activePath && props.renderDefault(props)]
+        }), /* @__PURE__ */jsx(Button, {
+          text: "Unset value",
+          onClick: unsetValue,
+          tone: "critical"
+        })]
+      })]
     })
-  });
+  }) : catcher;
 }
-function useSelectedSchema(fieldPath, documentSchema) {
-  return useMemo(() => {
-    if (!fieldPath) {
-      return void 0;
-    }
-    if (fieldPath === documentRootKey) {
-      return documentSchema;
-    }
-    const path = stringToPath(fieldPath);
-    let currentSchema = documentSchema;
-    for (let i = 0; i < path.length; i++) {
-      const segment = path[i];
-      const field = currentSchema == null ? void 0 : currentSchema.fields.find(f => f.name === segment);
-      if (!field) {
-        return void 0;
-      }
-      if (i === path.length - 1) {
-        return field.type;
-      }
-      if (field.type.jsonType !== "object") {
-        return void 0;
-      }
-      currentSchema = field.type;
-    }
-    return currentSchema;
-  }, [documentSchema, fieldPath]);
+function PteValueFixer(props) {
+  const isPortableText = useMemo(() => isArraySchemaType(props.schemaType) && isPortableTextArray(props.schemaType), [props.schemaType]);
+  const value = props.value;
+  if (isPortableText && value && !value.length) {
+    return props.renderDefault({
+      ...props,
+      value: void 0
+    });
+  }
+  return props.renderDefault(props);
 }
-function FieldsInitializer(_ref9) {
-  let {
-    pathKey,
-    activePath,
-    fieldExists,
+function AssistFormBlock(props) {
+  const presence = useAssistPresence(props.path, true);
+  const {
     onChange
-  } = _ref9;
-  const initialized = useRef(false);
-  useEffect(() => {
-    if (initialized.current || fieldExists || activePath || !pathKey) {
+  } = useFormCallbacks();
+  const key = props.value._key;
+  const localOnChange = useCallback(patchEvent => {
+    if (!key) {
       return;
     }
-    onChange([setIfMissing([], ["fields"]), insert([typed({
-      _key: pathKey,
-      _type: assistFieldTypeName,
-      path: pathKey
-    })], "after", ["fields", -1])]);
-    initialized.current = true;
-  }, [activePath, onChange, pathKey, fieldExists]);
-  return null;
+    onChange(PatchEvent.from(patchEvent).prefixAll({
+      _key: key
+    }));
+  }, [onChange, key]);
+  return /* @__PURE__ */jsx(ErrorWrapper, {
+    onChange: localOnChange,
+    children: /* @__PURE__ */jsxs(Flex, {
+      align: "center",
+      justify: "space-between",
+      children: [/* @__PURE__ */jsx(Box, {
+        flex: 1,
+        children: props.renderDefault(props)
+      }), presence.map(pre => /* @__PURE__ */jsx(AiFieldPresence, {
+        presence: pre
+      }, pre.lastActiveAt))]
+    })
+  });
+}
+function AssistItem(props) {
+  const {
+    path
+  } = props;
+  const presence = useAssistPresence(path, true);
+  return /* @__PURE__ */jsxs(Flex, {
+    align: "center",
+    width: "fill",
+    style: {
+      position: "relative"
+    },
+    children: [/* @__PURE__ */jsx(Box, {
+      flex: 1,
+      children: props.renderDefault({
+        ...props
+      })
+    }), presence.map(pre => /* @__PURE__ */jsx(Box, {
+      style: {
+        position: "absolute",
+        right: 35
+      },
+      children: /* @__PURE__ */jsx(AiFieldPresence, {
+        presence: pre
+      })
+    }, pre.user.id))]
+  });
 }
 function findFieldMember(members, fieldName) {
   return members.find(m => m.kind === "field" && m.name === fieldName || m.kind === "error" && m.fieldName === fieldName);
@@ -3343,117 +3550,10 @@ function getIcon(iconName) {
     return key === iconName;
   })) == null ? void 0 : _a[1]) != null ? _b : icons.sparkles;
 }
-function FieldAutocomplete(props) {
-  const {
-    id,
-    schemaType,
-    fieldPath,
-    onSelect,
-    includeDocument
-  } = props;
-  const fieldNames = useMemo(() => {
-    if (includeDocument) {
-      return getFieldRefsWithDocument(schemaType);
-    }
-    return getFieldRefs(schemaType);
-  }, [schemaType, includeDocument]);
-  const currentField = useMemo(() => fieldNames.find(f => f.key === fieldPath), [fieldPath, fieldNames]);
-  const autocompleteOptions = useMemo(() => fieldNames.map(field => ({
-    value: field.key,
-    field
-  })), [fieldNames]);
-  const renderOption = useCallback(option => {
-    const {
-      value,
-      field
-    } = option;
-    if (!value) {
-      return /* @__PURE__ */jsx(Card, {
-        as: "button",
-        padding: 3,
-        radius: 1,
-        children: /* @__PURE__ */jsx(Text, {
-          accent: true,
-          size: 1,
-          children: option.value
-        })
-      });
-    }
-    if (isType(field.schemaType, "document")) {
-      return /* @__PURE__ */jsx(Card, {
-        as: "button",
-        padding: 3,
-        radius: 1,
-        children: /* @__PURE__ */jsx(Text, {
-          size: 1,
-          weight: "semibold",
-          children: "The entire document"
-        })
-      });
-    }
-    const splitTitle = field.title.split("/");
-    return /* @__PURE__ */jsx(Card, {
-      as: "button",
-      padding: 3,
-      radius: 1,
-      children: /* @__PURE__ */jsxs(Flex, {
-        gap: 3,
-        children: [/* @__PURE__ */jsx(Text, {
-          size: 1,
-          children: createElement(field.icon)
-        }), /* @__PURE__ */jsx(Box, {
-          flex: "none",
-          children: /* @__PURE__ */jsxs(Breadcrumbs, {
-            separator: /* @__PURE__ */jsx(Text, {
-              muted: true,
-              size: 1,
-              children: "/"
-            }),
-            space: 1,
-            children: [splitTitle.slice(0, splitTitle.length - 1).map((pt, i) =>
-            // eslint-disable-next-line react/no-array-index-key
-            /* @__PURE__ */
-            jsx(Text, {
-              muted: true,
-              size: 1,
-              children: pt.trim()
-            }, i)), /* @__PURE__ */jsx(Text, {
-              size: 1,
-              weight: "medium",
-              children: splitTitle[splitTitle.length - 1]
-            })]
-          })
-        })]
-      })
-    });
-  }, []);
-  const renderValue = useCallback((value, option) => {
-    var _a;
-    return (_a = option == null ? void 0 : option.field.title) != null ? _a : value;
-  }, []);
-  const filterOption = useCallback((query, option) => {
-    var _a, _b, _c;
-    const lQuery = query.toLowerCase();
-    return ((_a = option == null ? void 0 : option.value) == null ? void 0 : _a.toLowerCase().includes(lQuery)) || ((_c = (_b = option == null ? void 0 : option.field) == null ? void 0 : _b.title) == null ? void 0 : _c.toLowerCase().includes(lQuery));
-  }, []);
-  return /* @__PURE__ */jsx(Autocomplete, {
-    fontSize: 1,
-    icon: currentField ? currentField.icon : SearchIcon,
-    onChange: onSelect,
-    openButton: true,
-    id,
-    options: autocompleteOptions,
-    placeholder: "Search for a field",
-    radius: 2,
-    renderOption,
-    renderValue,
-    value: currentField == null ? void 0 : currentField.key,
-    filterOption
-  });
-}
 function FieldRefPathInput(props) {
   var _a;
   const documentSchema = (_a = useContext(SelectedFieldContext)) == null ? void 0 : _a.documentSchema;
+  const typePath = useContext(TypePathContext);
   const ref = useRef(null);
   const id = useId();
   const {
@@ -3464,20 +3564,29 @@ function FieldRefPathInput(props) {
     (_b = (_a2 = ref.current) == null ? void 0 : _a2.querySelector("input")) == null ? void 0 : _b.focus();
   }, []);
   const onSelect = useCallback(path => onChange(set(path)), [onChange]);
+  const filter = useCallback(field => {
+    if (!field.key.includes("|") || !typePath) {
+      return true;
+    }
+    const dotSplit = typePath.split(".");
+    const base = dotSplit.slice(0, dotSplit.length - 1).join(".");
+    return field.key.includes(base);
+  }, [typePath]);
   if (!documentSchema) {
     return props.renderDefault(props);
   }
   return /* @__PURE__ */jsx(Box, {
     flex: 1,
     style: {
-      minWidth: 200
+      minWidth: 300
     },
     ref,
     children: /* @__PURE__ */jsx(FieldAutocomplete, {
       id,
       schemaType: documentSchema,
       onSelect,
-      fieldPath: props.value
+      fieldPath: props.value,
+      filter
     })
   });
 }
@@ -3680,7 +3789,7 @@ function useOnlyInlineBlocks(props) {
 function InstructionsArrayField(props) {
   return props.renderDefault({
     ...props,
-    title: void 0
+    title: " "
   });
 }
 const fieldReference = defineType({
@@ -4020,7 +4129,7 @@ const fieldInstructions = defineType({
 });
 const assistDocumentSchema = defineType({
   //NOTE: this is a document type. Using object here ensures it does not appear in structure menus
-  type: "object",
+  type: "document",
   //workaround for using object and not document
   ...{
     liveEdit: true
@@ -4100,7 +4209,7 @@ const documentInstructionStatus = defineType({
 });
 const schemaTypes = [fieldInstructions, assistDocumentSchema, prompt, fieldReference, instruction, documentInstructionStatus, instructionTask, contextDocumentSchema, userInput, promptContext];
 function useAssistSupported(path, schemaType) {
-  return useMemo(() => path.every(p => typeof p === "string") && isAssistSupported(schemaType), [path, schemaType]);
+  return useMemo(() => isAssistSupported(schemaType), [schemaType]);
 }
 function useAssistDocumentContextValue(documentId, documentSchemaType) {
   const {
@@ -4271,7 +4380,6 @@ const assistFieldActions = {
     const {
       schemaType
     } = props;
-    const assistSupported = useAssistSupported(props.path, schemaType);
     const isDocumentLevel = props.path.length === 0;
     const {
       assistDocument,
@@ -4292,9 +4400,13 @@ const assistFieldActions = {
     useAssistDocumentContextValue(props.documentId, schemaType) :
     // eslint-disable-next-line react-hooks/rules-of-hooks
     useAssistDocumentContext();
+    const {
+      value: docValue
+    } = useDocumentPane();
     const currentUser = useCurrentUser();
     const isHidden = !assistDocument;
     const pathKey = usePathKey(props.path);
+    const typePath = useTypePath(docValue, pathKey);
     const assistDocumentId = assistDocument == null ? void 0 : assistDocument._id;
     const assistableDocId = getAssistableDocId(documentSchemaType, documentId);
     const {
@@ -4303,10 +4415,12 @@ const assistFieldActions = {
       documentOnChange,
       isDocAssistable: documentIsAssistable != null ? documentIsAssistable : false
     });
+    const isSelectable = !!useSelectedField(documentSchemaType, typePath);
+    const assistSupported = useAssistSupported(props.path, schemaType) && isSelectable;
     const fieldAssist = useMemo(() => {
       var _a;
-      return ((_a = assistDocument == null ? void 0 : assistDocument.fields) != null ? _a : []).find(f => f.path == pathKey);
-    }, [assistDocument == null ? void 0 : assistDocument.fields, pathKey]);
+      return ((_a = assistDocument == null ? void 0 : assistDocument.fields) != null ? _a : []).find(f => f.path === typePath || pathKey === documentRootKey && f.path === pathKey);
+    }, [assistDocument == null ? void 0 : assistDocument.fields, pathKey, typePath]);
     const fieldAssistKey = fieldAssist == null ? void 0 : fieldAssist._key;
     const isInspectorOpen = (inspector == null ? void 0 : inspector.name) === aiInspectorId;
     const isPathSelected = pathKey === selectedPath;
@@ -4324,9 +4438,10 @@ const assistFieldActions = {
         documentId: assistableDocId,
         assistDocumentId,
         path: pathKey,
+        typePath,
         instruction
       });
-    }, [requestRunInstruction, assistableDocId, pathKey, assistDocumentId, fieldAssistKey]);
+    }, [requestRunInstruction, assistableDocId, pathKey, typePath, assistDocumentId, fieldAssistKey]);
     const privateInstructions = useMemo(() => {
       var _a;
       return ((_a = fieldAssist == null ? void 0 : fieldAssist.instructions) == null ? void 0 : _a.filter(i => i.userId && i.userId === (currentUser == null ? void 0 : currentUser.id))) || [];