@sanity/embeddings-index-ui 1.0.0 → 1.0.2

package/README.md

@@ -1,7 +1,7 @@
 # @sanity/embeddings-index-ui
 > This package contains plugins for **Sanity Studio v3**.
 
-`@sanity/embeddings-index-ui` Sanity Studio plugins that interacts with the `/embeddings-index` HTTP API.
+Sanity Studio plugins that interacts with the `/embeddings-index` HTTP API.
 
 The embeddings index API allows the creation of named embeddings vector indexes.
 An embeddings index contains embeddings for all Sanity documents matching a configured GROQ filter in a dataset.

package/dist/index.d.ts

@@ -42,7 +42,7 @@ export declare interface QueryResult {
   score: number
   value: {
     documentId: string
-    type?: string
+    type: string
   }
 }
 

package/dist/index.esm.js

@@ -1,4 +1,4 @@
-import { useClient, useSchema, DefaultPreview, useDocumentPreviewStore, SanityDefaultPreview, getPreviewValueWithFallback, getPreviewStateObservable, typed, unset, setIfMissing, set, definePlugin, isObjectInputProps } from 'sanity';
+import { useSchema, DefaultPreview, useDocumentPreviewStore, SanityDefaultPreview, getPreviewValueWithFallback, getPreviewStateObservable, useClient, typed, unset, setIfMissing, set, definePlugin, isObjectInputProps } from 'sanity';
 import { jsx, jsxs, Fragment } from 'react/jsx-runtime';
 import { Card, Button, Box, Flex, Autocomplete, Text, Stack, Label, TextInput, TextArea, Dialog, Spinner, Heading, MenuButton, Menu, MenuItem } from '@sanity/ui';
 import { ErrorOutlineIcon, EarthGlobeIcon, LinkIcon, AddIcon, SearchIcon, EllipsisVerticalIcon, TrashIcon, UndoIcon } from '@sanity/icons';
@@ -1740,20 +1740,12 @@ function DocumentPreview(_ref) {
   let {
     documentId,
     style,
+    schemaTypeName,
     ...buttonProps
   } = _ref;
-  const client = useClient({
-    apiVersion: "2023-06-06"
-  });
-  const [type, setType] = useState(void 0);
   const schema = useSchema();
-  const schemaType = type ? schema.get(type) : void 0;
-  useEffect(() => {
-    client.fetch("* [_id==$id]._type", {
-      id: documentId
-    }).then(setType);
-  }, [documentId, client]);
-  if (!type) {
+  const schemaType = schemaTypeName ? schema.get(schemaTypeName) : void 0;
+  if (!schemaTypeName) {
     return /* @__PURE__ */jsx(Card, {
       style: {
         minHeight: "36px"
@@ -1775,7 +1767,7 @@ function DocumentPreview(_ref) {
         media: () => /* @__PURE__ */jsx(ErrorOutlineIcon, {}),
         title: /* @__PURE__ */jsxs(Fragment, {
           children: ["Unknown type ", /* @__PURE__ */jsx("code", {
-            children: type != null ? type : "N/A"
+            children: schemaTypeName != null ? schemaTypeName : "N/A"
           }), " for ", documentId]
         })
       })
@@ -1783,6 +1775,7 @@ function DocumentPreview(_ref) {
   }
   return /* @__PURE__ */jsx(DocumentPreviewInner, {
     documentId,
+    schemaTypeName,
     schemaType,
     style,
     ...buttonProps
@@ -1886,6 +1879,22 @@ function deleteIndex(indexName, client) {
 function publicId(id) {
   return id.replace("drafts.", "");
 }
+function useApiClient() {
+  const client = useClient({
+    apiVersion: "vX"
+  });
+  return useMemo(() => {
+    const customHost = localStorage.getItem("embeddings-index-host");
+    if (customHost) {
+      return client.withConfig({
+        apiHost: customHost,
+        useProjectHostname: false,
+        withCredentials: false
+      });
+    }
+    return client;
+  }, [client]);
+}
 const NO_OPTIONS = [];
 const NO_FILTER = () => true;
 function SemanticSearchReferenceInput(props) {
@@ -1926,12 +1935,6 @@ function useDebouncedValue(value, ms) {
   }, [value, ms]);
   return debouncedValue;
 }
-function useApiClient$2() {
-  const client = useClient({
-    apiVersion: "vX"
-  });
-  return useMemo(() => client, [client]);
-}
 function SemanticSearchInput(props) {
   const {
     onPathFocus,
@@ -1952,7 +1955,7 @@ function SemanticSearchInput(props) {
   const prevDebouncedQuery = useRef(debouncedQuery);
   const [searching, setSearching] = useState(false);
   const [options, setOptions] = useState(NO_OPTIONS);
-  const client = useApiClient$2();
+  const client = useApiClient();
   useEffect(() => {
     docRef.current = currentDocument;
   }, [currentDocument]);
@@ -1965,7 +1968,7 @@ function SemanticSearchInput(props) {
   const handleFocus = useCallback(() => onPathFocus(["_ref"]), [onPathFocus]);
   const handleBlur = useCallback(() => onPathFocus([]), [onPathFocus]);
   const runIndexQuery = useCallback(queryString => {
-    var _a, _b, _c, _d;
+    var _a, _b, _c, _d, _e;
     setSearching(true);
     const refSchema = schemaType;
     const indexName = (_b = (_a = refSchema.options) == null ? void 0 : _a.embeddingsIndex) == null ? void 0 : _b.indexName;
@@ -1975,7 +1978,7 @@ function SemanticSearchInput(props) {
       throw new Error("Reference option embeddingsIndex.indexName is required, but was missing in type ".concat(refSchema.name));
     }
     queryIndex({
-      query: queryString,
+      query: queryString.trim().length ? queryString : (_e = JSON.stringify(docRef.current)) != null ? _e : "",
       indexName,
       maxResults,
       filter: {
@@ -2024,7 +2027,7 @@ function SemanticSearchInput(props) {
     id,
     ref: autocompleteRef,
     "data-testid": "semantic-autocomplete",
-    placeholder: "Type to search",
+    placeholder: "Type to search...",
     openButton: openButtonConfig,
     onFocus: handleFocus,
     onChange: handleChange,
@@ -2038,6 +2041,7 @@ function SemanticSearchInput(props) {
   });
 }
 function AutocompleteOption(props) {
+  const value = props.result.value;
   return /* @__PURE__ */jsx(Button, {
     mode: "bleed",
     padding: 1,
@@ -2050,7 +2054,8 @@ function AutocompleteOption(props) {
       children: [/* @__PURE__ */jsx(Box, {
         flex: 1,
         children: /* @__PURE__ */jsx(DocumentPreview, {
-          documentId: props.result.value.documentId
+          documentId: value.documentId,
+          schemaTypeName: value.type
         })
       }), /* @__PURE__ */jsx(Box, {
         padding: 2,
@@ -2089,12 +2094,6 @@ const embeddingsIndexReferenceInput = definePlugin({
     }
   }
 });
-function useApiClient$1(customApiClient) {
-  const client = useClient({
-    apiVersion: "vX"
-  });
-  return useMemo(() => customApiClient ? customApiClient(client) : client, [client, customApiClient]);
-}
 const defaultProjection = "{...}";
 function useDefaultIndex(schema, dataset) {
   const defaultFilter = useMemo(() => "_type in [".concat(schema.getTypeNames().map(n => schema.get(n)).filter(schemaType => Boolean(schemaType && isType(schemaType, "document"))).filter(documentType => !documentType.name.startsWith("sanity.") && !documentType.name.startsWith("assist.") && documentType.name !== "document").map(documentType => '"'.concat(documentType.name, '"')).join(",\n"), "]"), [schema]);
@@ -2211,7 +2210,7 @@ function IndexEditor(props) {
     index: selectedIndex,
     onSubmit
   } = props;
-  const client = useApiClient$1();
+  const client = useApiClient();
   const schema = useSchema();
   const defaultIndex = useDefaultIndex(schema, (_a = client.config().dataset) != null ? _a : "");
   const [errors, setErrors] = useState();
@@ -2431,7 +2430,7 @@ function QueryIndex(props) {
   const [query, setQuery] = useState("");
   const [searching, setSearching] = useState(false);
   const [results, setResults] = useState(NO_RESULTS);
-  const client = useApiClient$1();
+  const client = useApiClient();
   const search = useCallback(queryString => {
     setSearching(true);
     return queryIndex({
@@ -2501,13 +2500,15 @@ function ResultList(props) {
   });
 }
 function ResultEntry(props) {
+  const value = props.result.value;
   return /* @__PURE__ */jsxs(Flex, {
     gap: 4,
     align: "center",
     children: [/* @__PURE__ */jsx(Box, {
       flex: 1,
       children: /* @__PURE__ */jsx(DocumentPreview, {
-        documentId: props.result.value.documentId,
+        documentId: value.documentId,
+        schemaTypeName: value.type,
         button: true
       })
     }), /* @__PURE__ */jsx(Box, {
@@ -2647,12 +2648,6 @@ function IndexStatus(_ref4) {
     })]
   });
 }
-function useApiClient() {
-  const client = useClient({
-    apiVersion: "vX"
-  });
-  return useMemo(() => client, [client]);
-}
 function EmbeddingsIndexTool() {
   return /* @__PURE__ */jsx(Card, {
     children: /* @__PURE__ */jsx(Flex, {