@sanity/embeddings-index-ui 1.0.2 → 1.0.3

package/dist/index.esm.js

@@ -1,8 +1,8 @@
 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';
 
@@ -1895,11 +1895,56 @@ function useApiClient() {
     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, {
@@ -1908,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, {
@@ -2080,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 => {
@@ -2649,17 +2706,21 @@ function IndexStatus(_ref4) {
   });
 }
 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]
       })
     })
   });