@sanity/embeddings-index-ui 1.0.1 → 1.0.3

package/dist/index.esm.js

@@ -1,8 +1,8 @@
-import { useSchema, DefaultPreview, useDocumentPreviewStore, SanityDefaultPreview, getPreviewValueWithFallback, getPreviewStateObservable, typed, unset, setIfMissing, set, useClient, 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 { Card, Button, Box, Text, Flex, Spinner, Autocomplete, Stack, Label, TextInput, TextArea, Dialog, Heading, MenuButton, Menu, MenuItem } from '@sanity/ui';
 import { ErrorOutlineIcon, EarthGlobeIcon, LinkIcon, AddIcon, SearchIcon, EllipsisVerticalIcon, TrashIcon, UndoIcon } from '@sanity/icons';
-import require$$0, { useMemo, useRef, useEffect, useState, useCallback, useId } from 'react';
+import require$$0, { useMemo, useRef, useEffect, createContext, useState, useContext, useCallback, useId } from 'react';
 import { useIntentLink } from 'sanity/router';
 import { useDocumentPane } from 'sanity/desk';
 
@@ -1879,11 +1879,72 @@ 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 featureName = "embeddingsIndexApi";
+const FeatureEnabledContext = createContext("loading");
+function useIsFeatureEnabled() {
+  const client = useClient({
+    apiVersion: "2023-09-01"
+  });
+  const [status, setStatus] = useState("loading");
+  useEffect(() => {
+    client.request({
+      method: "GET",
+      url: "/projects/".concat(client.config().projectId, "/features/").concat(featureName)
+    }).then(isEnabled => {
+      setStatus(isEnabled === "true" || isEnabled === true ? "enabled" : "disabled");
+    }).catch(err => {
+      console.error(err);
+      setStatus("disabled");
+    });
+  }, [client]);
+  return status;
+}
+function FeatureEnabledProvider(props) {
+  const status = useIsFeatureEnabled();
+  return /* @__PURE__ */jsx(FeatureEnabledContext.Provider, {
+    value: status,
+    children: props.children
+  });
+}
+function useIsFeatureEnabledContext() {
+  return useContext(FeatureEnabledContext);
+}
+function FeatureDisabledNotice() {
+  return /* @__PURE__ */jsx(Card, {
+    tone: "primary",
+    border: true,
+    padding: 2,
+    children: /* @__PURE__ */jsxs(Text, {
+      size: 1,
+      children: ["Embeddings index APIs are only available on the", " ", /* @__PURE__ */jsx("a", {
+        href: "https://sanity.io/pricing",
+        children: "Team tier and above"
+      }), ". Please upgrade to enable access."]
+    })
+  });
+}
 const NO_OPTIONS = [];
 const NO_FILTER = () => true;
 function SemanticSearchReferenceInput(props) {
   var _a, _b, _c;
   const defaultEnabled = ((_c = (_b = (_a = props.schemaType) == null ? void 0 : _a.options) == null ? void 0 : _b.embeddingsIndex) == null ? void 0 : _c.searchMode) === "embeddings";
+  const featureState = useIsFeatureEnabledContext();
   const [semantic, setSemantic] = useState(defaultEnabled);
   const toggleSemantic = useCallback(() => setSemantic(current => !current), []);
   return /* @__PURE__ */jsxs(Flex, {
@@ -1892,13 +1953,16 @@ function SemanticSearchReferenceInput(props) {
     style: {
       width: "100%"
     },
-    children: [/* @__PURE__ */jsx(Box, {
+    children: [semantic && featureState == "loading" ? /* @__PURE__ */jsx(Box, {
+      padding: 2,
+      children: /* @__PURE__ */jsx(Spinner, {})
+    }) : null, semantic && featureState == "disabled" ? /* @__PURE__ */jsx(FeatureDisabledNotice, {}) : null, /* @__PURE__ */jsx(Box, {
       flex: 1,
       style: {
         maxHeight: 36,
         overflow: "hidden"
       },
-      children: semantic ? /* @__PURE__ */jsx(SemanticSearchInput, {
+      children: semantic && featureState == "enabled" ? /* @__PURE__ */jsx(SemanticSearchInput, {
         ...props
       }) : props.renderDefault(props)
     }), /* @__PURE__ */jsx(Button, {
@@ -1919,12 +1983,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,
@@ -1945,7 +2003,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]);
@@ -2070,6 +2128,15 @@ function isType(schemaType, typeName) {
 }
 const embeddingsIndexReferenceInput = definePlugin({
   name: "@sanity/embeddings-index-reference-input",
+  studio: {
+    components: {
+      layout: props => {
+        return /* @__PURE__ */jsx(FeatureEnabledProvider, {
+          children: props.renderDefault(props)
+        });
+      }
+    }
+  },
   form: {
     components: {
       input: props => {
@@ -2084,12 +2151,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]);
@@ -2206,7 +2267,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();
@@ -2426,7 +2487,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({
@@ -2644,24 +2705,22 @@ function IndexStatus(_ref4) {
     })]
   });
 }
-function useApiClient() {
-  const client = useClient({
-    apiVersion: "vX"
-  });
-  return useMemo(() => client, [client]);
-}
 function EmbeddingsIndexTool() {
+  const featureState = useIsFeatureEnabled();
   return /* @__PURE__ */jsx(Card, {
     children: /* @__PURE__ */jsx(Flex, {
       justify: "center",
       flex: 1,
-      children: /* @__PURE__ */jsx(Card, {
+      children: /* @__PURE__ */jsxs(Card, {
         flex: 1,
         style: {
           maxWidth: 1200
         },
         padding: 5,
-        children: /* @__PURE__ */jsx(Indexes, {})
+        children: [featureState == "loading" ? /* @__PURE__ */jsx(Box, {
+          padding: 2,
+          children: /* @__PURE__ */jsx(Spinner, {})
+        }) : null, featureState == "disabled" ? /* @__PURE__ */jsx(FeatureDisabledNotice, {}) : null, featureState == "enabled" ? /* @__PURE__ */jsx(Indexes, {}) : null]
       })
     })
   });