@sanity/assist 3.0.9 → 3.1.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@sanity/assist",
-  "version": "3.0.9",
+  "version": "3.1.1",
   "description": "You create the instructions; Sanity AI Assist does the rest.",
   "keywords": [
     "sanity",
@@ -29,7 +29,6 @@
   },
   "main": "./dist/index.js",
   "module": "./dist/index.esm.js",
-  "source": "./src/index.ts",
   "types": "./dist/index.d.ts",
   "files": [
     "dist",
@@ -82,10 +81,10 @@
     "react": "^18.2.0",
     "react-dom": "^18.2.0",
     "rimraf": "^5.0.5",
-    "sanity": "^3.37.2",
+    "sanity": "^3.74.1",
     "semantic-release": "^23.0.8",
     "styled-components": "^6.1.8",
-    "typescript": "5.4.2",
+    "typescript": "^5.7.2",
     "vitest": "^1.4.0"
   },
   "peerDependencies": {

package/src/assistDocument/AssistDocumentContext.tsx

@@ -8,9 +8,11 @@ export type AssistDocumentContextValue = (
   | {assistDocument: undefined; loading: true}
 ) & {
   documentIsNew: boolean
+  /**
+   * This is the _actual_ id of the current document (ie the document loaded in the pane); it contains draft. versions. prefix ect depending on context
+   */
   assistableDocumentId: string
   documentIsAssistable: boolean
-  documentId: string
   documentSchemaType: ObjectSchemaType
   openInspector: (inspectorName: string, paneParams?: Record<string, string>) => void
   closeInspector: (inspectorName?: string) => void

package/src/assistDocument/AssistDocumentContextProvider.tsx

@@ -1,17 +1,16 @@
 import {PropsWithChildren} from 'react'
-import {ObjectSchemaType} from 'sanity'
 
 import {AssistDocumentContext} from './AssistDocumentContext'
 import {useAssistDocumentContextValue} from './hooks/useAssistDocumentContextValue'
 
 export interface AIDocumentInputProps {
   documentId: string
-  schemaType: ObjectSchemaType
+  documentType: string
 }
 
 export function AssistDocumentContextProvider(props: PropsWithChildren<AIDocumentInputProps>) {
-  const {documentId, schemaType} = props
-  const value = useAssistDocumentContextValue(documentId, schemaType)
+  const {documentId, documentType} = props
+  const value = useAssistDocumentContextValue(documentId, documentType)
   return (
     <AssistDocumentContext.Provider value={value}>{props.children}</AssistDocumentContext.Provider>
   )

package/src/assistDocument/AssistDocumentInput.tsx

@@ -9,7 +9,6 @@ import {usePathKey} from '../helpers/misc'
 import {isType} from '../helpers/typeUtils'
 import {FirstAssistedPathProvider} from '../onboarding/FirstAssistedPathProvider'
 import {assistDocumentTypeName} from '../types'
-import {AssistDocumentContextProvider} from './AssistDocumentContextProvider'
 import {useInstructionToaster} from './hooks/useInstructionToaster'
 
 export function AssistDocumentInputWrapper(props: InputProps) {
@@ -44,12 +43,7 @@ function AssistDocumentInput({documentId, ...props}: ObjectInputProps & {documen
 
   return (
     <FirstAssistedPathProvider members={props.members}>
-      <AssistDocumentContextProvider schemaType={schemaType} documentId={documentId}>
-        {props.renderDefault({
-          ...props,
-          schemaType,
-        })}
-      </AssistDocumentContextProvider>
+      {props.renderDefault({...props, schemaType})}
     </FirstAssistedPathProvider>
   )
 }

package/src/assistDocument/AssistDocumentLayout.tsx

@@ -0,0 +1,12 @@
+import {DocumentLayoutProps} from 'sanity'
+
+import {AssistDocumentContextProvider} from './AssistDocumentContextProvider'
+
+export function AssistDocumentLayout(props: DocumentLayoutProps) {
+  const {documentId, documentType} = props
+  return (
+    <AssistDocumentContextProvider documentType={documentType} documentId={documentId}>
+      {props.renderDefault(props)}
+    </AssistDocumentContextProvider>
+  )
+}

package/src/assistDocument/RequestRunInstructionProvider.tsx

@@ -2,7 +2,6 @@ import {useCallback, useEffect, useState} from 'react'
 import {ObjectSchemaType, PatchEvent, SanityDocument, unset} from 'sanity'
 
 import {useRunInstruction} from '../assistLayout/RunInstructionProvider'
-import {publicId} from '../helpers/ids'
 
 export interface DraftDelayedTaskArgs<T> {
   documentOnChange: (event: PatchEvent) => void
@@ -19,11 +18,6 @@ export function isDocAssistable(
   return !!(documentSchemaType.liveEdit ? published : draft)
 }
 
-export function getAssistableDocId(documentSchemaType: ObjectSchemaType, documentId: string) {
-  const baseId = publicId(documentId)
-  return documentSchemaType.liveEdit ? baseId : `drafts.${baseId}`
-}
-
 export function useRequestRunInstruction(args: {
   documentOnChange: (event: PatchEvent) => void
   // indicates if the document is a draft or liveEditable currently

package/src/assistDocument/components/FieldRefPreview.tsx

@@ -7,22 +7,20 @@ import {useSelectedField} from '../../assistInspector/helpers'
 import {SelectedFieldContext} from './SelectedFieldContext'
 
 export function FieldRefPreview(props: PreviewProps & {path?: string}) {
+  const {actions} = props
   const documentSchema = useContext(SelectedFieldContext)?.documentSchema
   const path = (useContext(InlineBlockValueContext) as {path?: string})?.path ?? props.path
   const selectedField = useSelectedField(documentSchema, path)
   return (
     <Flex gap={2} align="center" style={{width: '100%'}}>
       <Flex flex={1} gap={2} align="center" paddingY={3} paddingX={1}>
-        {/*<Box>
-          <Text>{selectedField?.icon ? createElement(selectedField?.icon) : <CodeIcon />}</Text>
-        </Box>*/}
         <Box>
           <Text size={1} textOverflow="ellipsis">
             {selectedField?.title ?? 'Select field'}
           </Text>
         </Box>
       </Flex>
-      <>{props.actions}</>
+      {actions as any}
     </Flex>
   )
 }

package/src/assistDocument/hooks/useAssistDocumentContextValue.tsx

@@ -1,5 +1,5 @@
 import {useMemo} from 'react'
-import {getPublishedId, type ObjectSchemaType, useEditState} from 'sanity'
+import {getDraftId, getVersionId, type ObjectSchemaType, useSchema} from 'sanity'
 import {useDocumentPane} from 'sanity/structure'
 
 import {useAiPaneRouter} from '../../assistInspector/helpers'
@@ -8,31 +8,55 @@ import type {AssistDocumentContextValue} from '../AssistDocumentContext'
 import {isDocAssistable} from '../RequestRunInstructionProvider'
 import {useStudioAssistDocument} from './useStudioAssistDocument'
 
-export function useAssistDocumentContextValue(
-  documentId: string,
-  documentSchemaType: ObjectSchemaType,
-) {
-  const {published, draft} = useEditState(
-    getPublishedId(documentId),
-    documentSchemaType.name,
-    'low',
-  )
-  const assistableDocumentId = draft?._id || published?._id || documentId
-  const documentIsNew = Boolean(!draft?._id && !published?._id)
-  const documentIsAssistable = isDocAssistable(documentSchemaType, published, draft)
+export function useAssistDocumentContextValue(documentId: string, documentType: string) {
+  const schema = useSchema()
+
+  const documentSchemaType = useMemo(() => {
+    const schemaType = schema.get(documentType) as ObjectSchemaType | undefined
+    if (!schemaType) {
+      throw new Error(`Schema type "${documentType}" not found`)
+    }
+    return schemaType
+  }, [documentType, schema])
+
+  const {
+    openInspector,
+    closeInspector,
+    inspector,
+    onChange: documentOnChange,
+    // eslint-disable-next-line @typescript-eslint/ban-ts-comment
+    // @ts-ignore this is a valid option available in `corel` - Remove after corel is merged to next
+    selectedReleaseId,
+    editState,
+  } = useDocumentPane()
+  // eslint-disable-next-line @typescript-eslint/ban-ts-comment
+  // @ts-ignore this is a valid option available in `corel` - Remove after corel is merged to next
+  const {draft, published, version} = editState || {}
+
+  let assistableDocumentId = version?._id || draft?._id || published?._id
+  if (!assistableDocumentId) {
+    assistableDocumentId = selectedReleaseId
+      ? getVersionId(documentId, selectedReleaseId)
+      : documentSchemaType.liveEdit
+        ? documentId
+        : getDraftId(documentId)
+  }
+
+  const documentIsNew = selectedReleaseId ? !version?._id : !draft?._id && !published?._id
+  const documentIsAssistable = selectedReleaseId
+    ? !!version
+    : isDocAssistable(documentSchemaType, published, draft)
 
   const {params} = useAiPaneRouter()
   const selectedPath = params[fieldPathParam]
-  const {openInspector, closeInspector, inspector, onChange: documentOnChange} = useDocumentPane()
 
   const assistDocument = useStudioAssistDocument({
-    documentId,
+    documentId: assistableDocumentId,
     schemaType: documentSchemaType,
   })
 
   const value: AssistDocumentContextValue = useMemo(() => {
     const base = {
-      documentId,
       assistableDocumentId,
       documentSchemaType,
       documentIsNew,
@@ -54,7 +78,6 @@ export function useAssistDocumentContextValue(
   }, [
     assistDocument,
     documentIsAssistable,
-    documentId,
     assistableDocumentId,
     documentSchemaType,
     documentIsNew,

package/src/assistInspector/AssistInspector.tsx

@@ -4,7 +4,6 @@ import {useCallback, useMemo, useRef} from 'react'
 import {
   type DocumentInspectorProps,
   PresenceOverlay,
-  useEditState,
   VirtualizerScrollInstanceProvider,
 } from 'sanity'
 import {
@@ -18,11 +17,7 @@ import {styled} from 'styled-components'
 import {DocumentForm} from '../_lib/form'
 import {AssistTypeContext} from '../assistDocument/components/AssistTypeContext'
 import {useStudioAssistDocument} from '../assistDocument/hooks/useStudioAssistDocument'
-import {
-  getAssistableDocId,
-  isDocAssistable,
-  useRequestRunInstruction,
-} from '../assistDocument/RequestRunInstructionProvider'
+import {useRequestRunInstruction} from '../assistDocument/RequestRunInstructionProvider'
 import {useAiAssistanceConfig} from '../assistLayout/AiAssistanceConfigContext'
 import {giveFeedbackUrl, pluginTitle, releaseAnnouncementUrl, salesUrl} from '../constants'
 import {getConditionalMembers} from '../helpers/conditionalMembers'
@@ -39,6 +34,7 @@ import {
   useTypePath,
 } from './helpers'
 import {InstructionTaskHistoryButton} from './InstructionTaskHistoryButton'
+import {useAssistDocumentContext} from '../assistDocument/AssistDocumentContext'
 
 const CardWithShadowBelow = styled(Card)`
   position: relative;
@@ -209,15 +205,15 @@ export function AssistInspector(props: DocumentInspectorProps) {
     onChange: documentOnChange,
     formState,
   } = documentPane
-  const {published, draft} = useEditState(documentId, documentType, 'low')
+
+  const {assistableDocumentId, documentIsAssistable} = useAssistDocumentContext()
 
   const formStateRef = useRef(formState)
   formStateRef.current = formState
 
-  const assistableDocId = getAssistableDocId(schemaType, documentId)
   const {instructionLoading, requestRunInstruction} = useRequestRunInstruction({
     documentOnChange,
-    isDocAssistable: isDocAssistable(schemaType, published, draft),
+    isDocAssistable: documentIsAssistable,
   })
 
   const typePath = useTypePath(docValue, pathKey ?? '')
@@ -268,14 +264,14 @@ export function AssistInspector(props: DocumentInspectorProps) {
       pathKey &&
       typePath &&
       requestRunInstruction({
-        documentId: assistableDocId,
+        documentId: assistableDocumentId,
         path: pathKey,
         typePath,
         assistDocumentId: assistDocumentId(documentType),
         instruction,
         conditionalMembers: formStateRef.current ? getConditionalMembers(formStateRef.current) : [],
       }),
-    [pathKey, instruction, typePath, documentType, assistableDocId, requestRunInstruction],
+    [pathKey, instruction, typePath, documentType, assistableDocumentId, requestRunInstruction],
   )
 
   const Region = useCallback((_props: any) => {
@@ -331,6 +327,13 @@ export function AssistInspector(props: DocumentInspectorProps) {
                         index={documentPane.index}
                         itemId="ai"
                         pane={paneNode}
+                        // eslint-disable-next-line @typescript-eslint/ban-ts-comment
+                        //@ts-ignore this is a valid option available in `corel` - Remove after corel is merged to next
+                        forcedVersion={{
+                          isReleaseLocked: false,
+                          selectedPerspectiveName: 'published',
+                          selectedReleaseId: undefined,
+                        }}
                       >
                         <DocumentForm />
                       </DocumentPaneProvider>
@@ -373,7 +376,7 @@ export function AssistInspector(props: DocumentInspectorProps) {
           )}
 
           <InstructionTaskHistoryButton
-            documentId={assistableDocId}
+            documentId={assistableDocumentId}
             tasks={tasks}
             instructions={instructions}
             showTitles={!instructionKey}

package/src/components/ImageContext.tsx

@@ -1,10 +1,9 @@
 import {createContext, useEffect, useMemo, useState} from 'react'
-import {type InputProps, pathToString, useSyncState} from 'sanity'
-import {usePaneRouter} from 'sanity/structure'
+import {getPublishedId, type InputProps, pathToString, useSyncState} from 'sanity'
+import {useDocumentPane, usePaneRouter} from 'sanity/structure'
 
 import {useAssistDocumentContext} from '../assistDocument/AssistDocumentContext'
 import {useAiAssistanceConfig} from '../assistLayout/AiAssistanceConfigContext'
-import {publicId} from '../helpers/ids'
 import {getDescriptionFieldOption, getImageInstructionFieldOption} from '../helpers/typeUtils'
 import {canUseAssist, useApiClient, useGenerateCaption} from '../useApiClient'
 
@@ -19,14 +18,19 @@ export const ImageContext = createContext<ImageContextValue>({})
 export function ImageContextProvider(props: InputProps) {
   const {schemaType, path, value, readOnly} = props
   const assetRef = (value as any)?.asset?._ref
+  const {selectedReleaseId} = useDocumentPane()
   const [assetRefState, setAssetRefState] = useState<string | undefined>(assetRef)
 
-  const {documentId, documentSchemaType} = useAssistDocumentContext()
+  const {assistableDocumentId, documentSchemaType} = useAssistDocumentContext()
   const {config, status} = useAiAssistanceConfig()
   const apiClient = useApiClient(config?.__customApiClient)
   const {generateCaption} = useGenerateCaption(apiClient)
 
-  const {isSyncing} = useSyncState(publicId(documentId), documentSchemaType.name)
+  const {isSyncing} = useSyncState(
+    getPublishedId(assistableDocumentId),
+    documentSchemaType.name,
+    selectedReleaseId,
+  )
 
   const router = usePaneRouter()
   const isShowingOlderRevision = !!router.params?.rev
@@ -35,7 +39,7 @@ export function ImageContextProvider(props: InputProps) {
     const descriptionField = getDescriptionFieldOption(schemaType)
     if (
       assetRef &&
-      documentId &&
+      assistableDocumentId &&
       descriptionField &&
       assetRef !== assetRefState &&
       !isSyncing &&
@@ -44,7 +48,10 @@ export function ImageContextProvider(props: InputProps) {
     ) {
       setAssetRefState(assetRef)
       if (canUseAssist(status)) {
-        generateCaption({path: pathToString([...path, descriptionField]), documentId: documentId})
+        generateCaption({
+          path: pathToString([...path, descriptionField]),
+          documentId: assistableDocumentId,
+        })
       }
     }
   }, [
@@ -52,7 +59,7 @@ export function ImageContextProvider(props: InputProps) {
     path,
     assetRef,
     assetRefState,
-    documentId,
+    assistableDocumentId,
     generateCaption,
     isSyncing,
     status,

package/src/fieldActions/assistFieldActions.tsx

@@ -4,7 +4,6 @@ import {
   type DocumentFieldAction,
   type DocumentFieldActionGroup,
   type DocumentFieldActionItem,
-  type ObjectSchemaType,
   typed,
   useCurrentUser,
 } from 'sanity'
@@ -12,11 +11,7 @@ import {useDocumentPane} from 'sanity/structure'
 
 import {useAssistDocumentContext} from '../assistDocument/AssistDocumentContext'
 import {getIcon} from '../assistDocument/components/instruction/appearance/IconInput'
-import {useAssistDocumentContextValue} from '../assistDocument/hooks/useAssistDocumentContextValue'
-import {
-  getAssistableDocId,
-  useRequestRunInstruction,
-} from '../assistDocument/RequestRunInstructionProvider'
+import {useRequestRunInstruction} from '../assistDocument/RequestRunInstructionProvider'
 import {aiInspectorId} from '../assistInspector/constants'
 import {useSelectedField, useTypePath} from '../assistInspector/helpers'
 import {pluginTitleShort} from '../constants'
@@ -39,8 +34,6 @@ export const assistFieldActions: DocumentFieldAction = {
   useAction(props) {
     const {schemaType} = props
 
-    const isDocumentLevel = props.path.length === 0
-
     const {
       assistDocument,
       documentIsNew,
@@ -50,17 +43,9 @@ export const assistFieldActions: DocumentFieldAction = {
       inspector,
       documentOnChange,
       documentSchemaType,
-      documentId,
       selectedPath,
       assistableDocumentId,
-    } =
-      // document field actions do not have access to the document context
-      // conditional hook _should_ be safe here since the logical path will be stable
-      isDocumentLevel
-        ? // eslint-disable-next-line react-hooks/rules-of-hooks
-          useAssistDocumentContextValue(props.documentId, schemaType as ObjectSchemaType)
-        : // eslint-disable-next-line react-hooks/rules-of-hooks
-          useAssistDocumentContext()
+    } = useAssistDocumentContext()
 
     const {value: docValue, formState} = useDocumentPane()
     const formStateRef = useRef(formState)
@@ -72,7 +57,6 @@ export const assistFieldActions: DocumentFieldAction = {
     const typePath = useTypePath(docValue, pathKey)
     const assistDocumentId = assistDocument?._id
 
-    const assistableDocId = getAssistableDocId(documentSchemaType, documentId)
     const {requestRunInstruction} = useRequestRunInstruction({
       documentOnChange,
       isDocAssistable: documentIsAssistable ?? false,
@@ -121,11 +105,11 @@ export const assistFieldActions: DocumentFieldAction = {
 
     const onInstructionAction = useCallback(
       (instruction: StudioInstruction) => {
-        if (!pathKey || !fieldAssistKey || !assistDocumentId || !assistableDocId) {
+        if (!pathKey || !fieldAssistKey || !assistDocumentId || !assistableDocumentId) {
           return
         }
         requestRunInstruction({
-          documentId: assistableDocId,
+          documentId: assistableDocumentId,
           assistDocumentId,
           path: pathKey,
           typePath,
@@ -135,7 +119,14 @@ export const assistFieldActions: DocumentFieldAction = {
             : [],
         })
       },
-      [requestRunInstruction, assistableDocId, pathKey, typePath, assistDocumentId, fieldAssistKey],
+      [
+        requestRunInstruction,
+        assistableDocumentId,
+        pathKey,
+        typePath,
+        assistDocumentId,
+        fieldAssistKey,
+      ],
     )
 
     const privateInstructions = useMemo(

package/src/fieldActions/generateCaptionActions.tsx

@@ -30,7 +30,7 @@ export const generateCaptionsActions: DocumentFieldAction = {
     if (imageContext && pathKey === imageContext?.imageDescriptionPath) {
       //if this is true, it is stable, and not breaking rules of hooks
       // eslint-disable-next-line react-hooks/rules-of-hooks
-      const {documentId} = useAssistDocumentContext()
+      const {assistableDocumentId} = useAssistDocumentContext()
       // eslint-disable-next-line react-hooks/rules-of-hooks
       return useMemo(() => {
         return node({
@@ -54,14 +54,21 @@ export const generateCaptionsActions: DocumentFieldAction = {
               })
               return
             }
-
-            generateCaption({path: pathKey, documentId: documentId ?? ''})
+            generateCaption({path: pathKey, documentId: assistableDocumentId})
           },
           renderAsButton: true,
           disabled: loading,
           hidden: !imageContext.assetRef,
         })
-      }, [generateCaption, pathKey, documentId, loading, imageContext, status, openInspector])
+      }, [
+        generateCaption,
+        pathKey,
+        assistableDocumentId,
+        loading,
+        imageContext,
+        status,
+        openInspector,
+      ])
     }
 
     // works but not supported by types

package/src/fieldActions/generateImageActions.tsx

@@ -27,7 +27,7 @@ export const generateImagActions: DocumentFieldAction = {
     if (imageContext && pathKey === imageContext?.imageInstructionPath) {
       //if this is true, it is stable, and not breaking rules of hooks
       // eslint-disable-next-line react-hooks/rules-of-hooks
-      const {documentId} = useAssistDocumentContext()
+      const {assistableDocumentId} = useAssistDocumentContext()
       // eslint-disable-next-line react-hooks/rules-of-hooks
       return useMemo(() => {
         return node({
@@ -44,12 +44,12 @@ export const generateImagActions: DocumentFieldAction = {
             if (loading) {
               return
             }
-            generateImage({path: pathKey, documentId: documentId ?? ''})
+            generateImage({path: pathKey, documentId: assistableDocumentId})
           },
           renderAsButton: true,
           disabled: loading,
         })
-      }, [generateImage, pathKey, documentId, loading])
+      }, [generateImage, pathKey, assistableDocumentId, loading])
     }
 
     // works but not supported by types

package/src/helpers/ids.test.ts

@@ -1,6 +1,6 @@
 import {describe, expect, test} from 'vitest'
 
-import {assistDocumentId} from './ids'
+import {assistDocumentId, assistTasksStatusId} from './ids'
 
 describe('ids', () => {
   test('assistDocumentId should replace illegal id chars with _', () => {
@@ -14,4 +14,15 @@ describe('ids', () => {
     const expected = testCases.map((testCase) => testCase.assistId)
     expect(outputs).toEqual(expected)
   })
+
+  test.each([
+    {documentId: 'foo', assistId: 'sanity.assist.status.foo'},
+    {documentId: 'drafts.foo', assistId: 'sanity.assist.status.foo'},
+    {documentId: 'versions.r12332.foo', assistId: 'sanity.assist.status.r12332.foo'},
+  ])(
+    "assistTasksStatusId should return the documentId with 'sanity.assist.status' prefix for $documentId",
+    ({documentId, assistId}) => {
+      expect(assistTasksStatusId(documentId)).toEqual(assistId)
+    },
+  )
 })

package/src/helpers/ids.ts

@@ -1,3 +1,5 @@
+import {getPublishedId, getVersionFromId, isVersionId} from 'sanity'
+
 import {assistDocumentIdPrefix, assistDocumentStatusIdPrefix} from '../types'
 
 const illegalIdChars = /[^a-zA-Z0-9._-]/g
@@ -11,5 +13,11 @@ export function assistDocumentId(documentType: string) {
 }
 
 export function assistTasksStatusId(documentId: string) {
-  return `${assistDocumentStatusIdPrefix}${publicId(documentId)}`
+  if (isVersionId(documentId)) {
+    // Creates an id: sanity.assist.status.<versionName>.<documentId>
+    return `${assistDocumentStatusIdPrefix}${getVersionFromId(documentId)}.${getPublishedId(documentId)}`
+  }
+
+  // Creates an id: sanity.assist.status<documentId>
+  return `${assistDocumentStatusIdPrefix}${getPublishedId(documentId)}`
 }

package/src/plugin.tsx

@@ -17,6 +17,7 @@ import {isImage} from './helpers/typeUtils'
 import {ImageContextProvider} from './components/ImageContext'
 import {TranslationConfig} from './translate/types'
 import {assistDocumentTypeName, AssistPreset} from './types'
+import {AssistDocumentLayout} from './assistDocument/AssistDocumentLayout'
 
 export interface AssistPluginConfig {
   translate?: TranslationConfig
@@ -69,10 +70,13 @@ export const assist = definePlugin<AssistPluginConfig | void>((config) => {
         }
         const docSchema = schema.get(schemaType)
         if (docSchema && isObjectSchemaType(docSchema) && isSchemaAssistEnabled(docSchema)) {
-          return [...prev, createAssistDocumentPresence(documentId, docSchema)]
+          return [...prev, createAssistDocumentPresence(documentId)]
         }
         return prev
       },
+      components: {
+        unstable_layout: AssistDocumentLayout,
+      },
     },
 
     studio: {

package/src/presence/AssistDocumentPresence.tsx

@@ -1,28 +1,19 @@
 import {Card, Flex} from '@sanity/ui'
 import {useMemo} from 'react'
-import {ObjectSchemaType} from 'sanity'
 
-import {useAssistDocumentContextValue} from '../assistDocument/hooks/useAssistDocumentContextValue'
+import {useAssistDocumentContext} from '../assistDocument/AssistDocumentContext'
 import {documentRootKey, fieldPresenceTypeName} from '../types'
 import {AiFieldPresence} from './AiFieldPresence'
 import {aiPresence} from './useAssistPresence'
 
-export function createAssistDocumentPresence(
-  documentId: string | undefined,
-  schemaType: ObjectSchemaType,
-) {
+export function createAssistDocumentPresence(documentId: string | undefined) {
   return function AssistDocumentPresenceWrapper() {
-    return documentId ? (
-      <AssistDocumentPresence documentId={documentId} schemaType={schemaType} />
-    ) : null
+    return documentId ? <AssistDocumentPresence /> : null
   }
 }
 
-function AssistDocumentPresence(props: {documentId: string; schemaType: ObjectSchemaType}) {
-  const {assistDocument} = useAssistDocumentContextValue(
-    props.documentId,
-    props.schemaType as ObjectSchemaType,
-  )
+function AssistDocumentPresence() {
+  const {assistDocument} = useAssistDocumentContext()
   const anyPresence = useMemo(() => {
     const anyPresence = assistDocument?.tasks
       ?.filter((run) => !run.ended && !run.reason)