@sanity/assist 4.4.5 → 4.4.6

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@sanity/assist",
-  "version": "4.4.5",
+  "version": "4.4.6",
   "description": "You create the instructions; Sanity AI Assist does the rest.",
   "keywords": [
     "sanity",

package/src/assistDocument/AssistDocumentContext.tsx

@@ -1,7 +1,7 @@
 import {createContext, useContext} from 'react'
 import {DocumentInspector, ObjectSchemaType, PatchEvent} from 'sanity'
 
-import {InstructionTask, SerializedSchemaType, StudioAssistDocument} from '../types'
+import {InstructionTask, StudioAssistDocument} from '../types'
 import {FieldRef} from '../assistInspector/helpers'
 
 export type AssistDocumentContextValue = (
@@ -36,8 +36,6 @@ export type AssistDocumentContextValue = (
 
   fieldRefs: FieldRef[]
   fieldRefsByTypePath: Record<string, FieldRef | undefined>
-
-  serializedTypes: SerializedSchemaType[]
 }
 
 export const AssistDocumentContext = createContext<AssistDocumentContextValue | undefined>(

package/src/assistDocument/hooks/useAssistDocumentContextValue.tsx

@@ -4,14 +4,9 @@ import {useDocumentPane} from 'sanity/structure'
 
 import {asFieldRefsByTypePath, getFieldRefs, useAiPaneRouter} from '../../assistInspector/helpers'
 import {fieldPathParam, InstructionTask} from '../../types'
-import {AssistDocumentContextValue, useAssistDocumentContext} from '../AssistDocumentContext'
+import {AssistDocumentContextValue} from '../AssistDocumentContext'
 import {isDocAssistable} from '../RequestRunInstructionProvider'
 import {useStudioAssistDocument} from './useStudioAssistDocument'
-import {serializeSchema} from '../../schemas/serialize/serializeSchema'
-
-export function useSerializedTypes() {
-  return useAssistDocumentContext().serializedTypes
-}
 
 export function useAssistDocumentContextValue(documentId: string, documentType: string) {
   const schema = useSchema()
@@ -24,8 +19,6 @@ export function useAssistDocumentContextValue(documentId: string, documentType:
     return schemaType
   }, [documentType, schema])
 
-  const serializedTypes = useMemo(() => serializeSchema(schema, {leanFormat: true}), [schema])
-
   const {fieldRefs, fieldRefsByTypePath} = useMemo(() => {
     const fieldRefs = getFieldRefs(documentSchemaType)
     const fieldRefsByTypePath = asFieldRefsByTypePath(fieldRefs)
@@ -86,7 +79,6 @@ export function useAssistDocumentContextValue(documentId: string, documentType:
       removeSyntheticTask,
       fieldRefs,
       fieldRefsByTypePath,
-      serializedTypes,
     }
     if (!assistDocument) {
       return {...base, loading: true, assistDocument: undefined}
@@ -112,7 +104,6 @@ export function useAssistDocumentContextValue(documentId: string, documentType:
     removeSyntheticTask,
     fieldRefs,
     fieldRefsByTypePath,
-    serializedTypes,
   ])
 
   return value

package/src/assistLayout/AiAssistanceConfigContext.tsx

@@ -10,6 +10,9 @@ import {
 
 import {AssistPluginConfig} from '../plugin'
 import {InstructStatus, useApiClient, useGetInstructStatus, useInitInstruct} from '../useApiClient'
+import {SerializedSchemaType} from '../types'
+import {serializeSchema} from '../schemas/serialize/serializeSchema'
+import {useSchema} from 'sanity'
 
 export interface AiAssistanceConfigContextValue {
   config: AssistPluginConfig
@@ -18,6 +21,7 @@ export interface AiAssistanceConfigContextValue {
   initLoading: boolean
   init: () => void
   error?: Error
+  serializedTypes: SerializedSchemaType[]
 }
 
 export const AiAssistanceConfigContext = createContext<AiAssistanceConfigContextValue>({} as any)
@@ -30,6 +34,10 @@ export function useAiAssistanceConfig() {
   return context
 }
 
+export function useSerializedTypes() {
+  return useAiAssistanceConfig().serializedTypes
+}
+
 export function AiAssistanceConfigProvider(props: {
   children?: ReactNode
   config: AssistPluginConfig
@@ -41,6 +49,9 @@ export function AiAssistanceConfigProvider(props: {
   const {getInstructStatus, loading: statusLoading} = useGetInstructStatus(apiClient)
   const {initInstruct, loading: initLoading} = useInitInstruct(apiClient)
 
+  const schema = useSchema()
+  const serializedTypes = useMemo(() => serializeSchema(schema, {leanFormat: true}), [schema])
+
   useEffect(() => {
     getInstructStatus()
       .then((s) => setStatus(s))
@@ -71,8 +82,9 @@ export function AiAssistanceConfigProvider(props: {
       init,
       initLoading,
       error,
+      serializedTypes,
     }
-  }, [config, status, init, statusLoading, initLoading, error])
+  }, [config, status, init, statusLoading, initLoading, error, serializedTypes])
 
   return (
     <AiAssistanceConfigContext.Provider value={context}>

package/src/useApiClient.ts

@@ -3,11 +3,10 @@ import {useToast} from '@sanity/ui'
 import {useCallback, useMemo, useState} from 'react'
 import {Path, pathToString, useClient, useCurrentUser} from 'sanity'
 
-import {useAiAssistanceConfig} from './assistLayout/AiAssistanceConfigContext'
+import {useAiAssistanceConfig, useSerializedTypes} from './assistLayout/AiAssistanceConfigContext'
 import {ConditionalMemberState} from './helpers/conditionalMembers'
 import {FieldLanguageMap} from './translate/paths'
 import {documentRootKey} from './types'
-import {useSerializedTypes} from './assistDocument/hooks/useAssistDocumentContextValue'
 
 export interface UserTextInstance {
   blockKey: string