@sanity/assist 4.4.3 → 4.4.5

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@sanity/assist",
-  "version": "4.4.3",
+  "version": "4.4.5",
   "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, StudioAssistDocument} from '../types'
+import {InstructionTask, SerializedSchemaType, StudioAssistDocument} from '../types'
 import {FieldRef} from '../assistInspector/helpers'
 
 export type AssistDocumentContextValue = (
@@ -36,6 +36,8 @@ 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,9 +4,14 @@ import {useDocumentPane} from 'sanity/structure'
 
 import {asFieldRefsByTypePath, getFieldRefs, useAiPaneRouter} from '../../assistInspector/helpers'
 import {fieldPathParam, InstructionTask} from '../../types'
-import type {AssistDocumentContextValue} from '../AssistDocumentContext'
+import {AssistDocumentContextValue, useAssistDocumentContext} 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()
@@ -19,6 +24,17 @@ 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)
+    return {
+      fieldRefs,
+      fieldRefsByTypePath,
+    }
+  }, [documentSchemaType])
+
   const {
     openInspector,
     closeInspector,
@@ -54,9 +70,6 @@ export function useAssistDocumentContextValue(documentId: string, documentType:
   const {syntheticTasks, addSyntheticTask, removeSyntheticTask} =
     useSyntheticTasks(assistableDocumentId)
 
-  const fieldRefs = getFieldRefs(documentSchemaType)
-  const fieldRefsByTypePath = asFieldRefsByTypePath(fieldRefs)
-
   const value: AssistDocumentContextValue = useMemo(() => {
     const base = {
       assistableDocumentId,
@@ -73,6 +86,7 @@ export function useAssistDocumentContextValue(documentId: string, documentType:
       removeSyntheticTask,
       fieldRefs,
       fieldRefsByTypePath,
+      serializedTypes,
     }
     if (!assistDocument) {
       return {...base, loading: true, assistDocument: undefined}
@@ -96,6 +110,9 @@ export function useAssistDocumentContextValue(documentId: string, documentType:
     syntheticTasks,
     addSyntheticTask,
     removeSyntheticTask,
+    fieldRefs,
+    fieldRefsByTypePath,
+    serializedTypes,
   ])
 
   return value

package/src/useApiClient.ts

@@ -1,13 +1,13 @@
 import type {SanityClient} from '@sanity/client'
 import {useToast} from '@sanity/ui'
 import {useCallback, useMemo, useState} from 'react'
-import {Path, pathToString, useClient, useCurrentUser, useSchema} from 'sanity'
+import {Path, pathToString, useClient, useCurrentUser} from 'sanity'
 
 import {useAiAssistanceConfig} from './assistLayout/AiAssistanceConfigContext'
 import {ConditionalMemberState} from './helpers/conditionalMembers'
-import {serializeSchema} from './schemas/serialize/serializeSchema'
 import {FieldLanguageMap} from './translate/paths'
 import {documentRootKey} from './types'
+import {useSerializedTypes} from './assistDocument/hooks/useAssistDocumentContextValue'
 
 export interface UserTextInstance {
   blockKey: string
@@ -58,8 +58,7 @@ export function useApiClient(customApiClient?: (defaultClient: SanityClient) =>
 export function useTranslate(apiClient: SanityClient) {
   const [loading, setLoading] = useState(false)
   const user = useCurrentUser()
-  const schema = useSchema()
-  const types = useMemo(() => serializeSchema(schema, {leanFormat: true}), [schema])
+  const types = useSerializedTypes()
   const toast = useToast()
 
   const translate = useCallback(
@@ -126,8 +125,7 @@ export function useTranslate(apiClient: SanityClient) {
 export function useGenerateCaption(apiClient: SanityClient) {
   const [loading, setLoading] = useState(false)
   const user = useCurrentUser()
-  const schema = useSchema()
-  const types = useMemo(() => serializeSchema(schema, {leanFormat: true}), [schema])
+  const types = useSerializedTypes()
   const toast = useToast()
 
   const generateCaption = useCallback(
@@ -179,8 +177,7 @@ export function useGenerateCaption(apiClient: SanityClient) {
 export function useGenerateImage(apiClient: SanityClient) {
   const [loading, setLoading] = useState(false)
   const user = useCurrentUser()
-  const schema = useSchema()
-  const types = useMemo(() => serializeSchema(schema, {leanFormat: true}), [schema])
+  const types = useSerializedTypes()
   const toast = useToast()
 
   const generateImage = useCallback(
@@ -280,8 +277,7 @@ export function useRunInstructionApi(apiClient: SanityClient) {
   const toast = useToast()
   const [loading, setLoading] = useState(false)
   const user = useCurrentUser()
-  const schema = useSchema()
-  const types = useMemo(() => serializeSchema(schema, {leanFormat: true}), [schema])
+  const types = useSerializedTypes()
 
   const {
     config: {assist: assistConfig},