@sanity/embeddings-index-ui 1.0.1 → 1.0.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@sanity/embeddings-index-ui",
-  "version": "1.0.1",
+  "version": "1.0.2",
   "description": "Various Sanity Studio plugins for integrating with the embeddings index API",
   "keywords": [
     "sanity",

package/src/api/embeddingsApiHooks.ts

@@ -1,12 +1,17 @@
 import {SanityClient, useClient} from 'sanity'
 import {useMemo} from 'react'
 
-export function useApiClient(
-  customApiClient?: (defaultClient: SanityClient) => SanityClient,
-): SanityClient {
+export function useApiClient(): SanityClient {
   const client = useClient({apiVersion: 'vX'})
-  return useMemo(
-    () => (customApiClient ? customApiClient(client) : client),
-    [client, customApiClient],
-  )
+  return useMemo(() => {
+    const customHost = localStorage.getItem('embeddings-index-host')
+    if (customHost) {
+      return client.withConfig({
+        apiHost: customHost,
+        useProjectHostname: false,
+        withCredentials: false,
+      })
+    }
+    return client
+  }, [client])
 }

package/src/embeddingsIndexDashboard/EmbeddingsIndexTool.tsx

@@ -1,16 +1,11 @@
 import {Box, Button, Card, Flex, Heading, Spinner, Stack} from '@sanity/ui'
-import {useClient} from 'sanity'
-import {useCallback, useEffect, useMemo, useState} from 'react'
+import {useCallback, useEffect, useState} from 'react'
 import {AddIcon, UndoIcon} from '@sanity/icons'
 import {deleteIndex, getIndexes, IndexState, NamedIndex} from '../api/embeddingsApi'
 import {EditIndexDialog} from './IndexEditor'
 import {IndexList} from './IndexList'
 import {IndexInfo} from './IndexInfo'
-
-function useApiClient() {
-  const client = useClient({apiVersion: 'vX'})
-  return useMemo(() => client, [client])
-}
+import {useApiClient} from '../api/embeddingsApiHooks'
 
 export function EmbeddingsIndexTool() {
   return (

package/src/referenceInput/SemanticSearchReferenceInput.tsx

@@ -1,12 +1,4 @@
-import {
-  ObjectInputProps,
-  ReferenceSchemaType,
-  set,
-  setIfMissing,
-  typed,
-  unset,
-  useClient,
-} from 'sanity'
+import {ObjectInputProps, ReferenceSchemaType, set, setIfMissing, typed, unset} from 'sanity'
 import {Autocomplete, Box, Button, Flex, Text, AutocompleteOpenButtonProps} from '@sanity/ui'
 import {EarthGlobeIcon, LinkIcon} from '@sanity/icons'
 import {useCallback, useEffect, useId, useMemo, useRef, useState} from 'react'
@@ -14,6 +6,7 @@ import {DocumentPreview} from '../preview/DocumentPreview'
 import {useDocumentPane} from 'sanity/desk'
 import {queryIndex, QueryResult} from '../api/embeddingsApi'
 import {publicId} from '../utils/id'
+import {useApiClient} from '../api/embeddingsApiHooks'
 
 interface Option {
   result: QueryResult
@@ -60,11 +53,6 @@ function useDebouncedValue<T>(value: T, ms: number) {
   return debouncedValue
 }
 
-function useApiClient() {
-  const client = useClient({apiVersion: 'vX'})
-  return useMemo(() => client, [client])
-}
-
 function SemanticSearchInput(props: ObjectInputProps) {
   const {onPathFocus, onChange, readOnly, schemaType, value} = props