@sanity/assist 4.4.4 → 4.4.6

package/package.json

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

package/src/assistDocument/hooks/useAssistDocumentContextValue.tsx

@@ -4,7 +4,7 @@ import {useDocumentPane} from 'sanity/structure'
 
 import {asFieldRefsByTypePath, getFieldRefs, useAiPaneRouter} from '../../assistInspector/helpers'
 import {fieldPathParam, InstructionTask} from '../../types'
-import type {AssistDocumentContextValue} from '../AssistDocumentContext'
+import {AssistDocumentContextValue} from '../AssistDocumentContext'
 import {isDocAssistable} from '../RequestRunInstructionProvider'
 import {useStudioAssistDocument} from './useStudioAssistDocument'
 

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

@@ -1,11 +1,10 @@
 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 {useAiAssistanceConfig, useSerializedTypes} from './assistLayout/AiAssistanceConfigContext'
 import {ConditionalMemberState} from './helpers/conditionalMembers'
-import {serializeSchema} from './schemas/serialize/serializeSchema'
 import {FieldLanguageMap} from './translate/paths'
 import {documentRootKey} from './types'
 
@@ -58,8 +57,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 +124,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 +176,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 +276,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},