@sanity/embeddings-index-ui 1.0.2 → 1.0.3

package/dist/index.js

@@ -1906,11 +1906,56 @@ function useApiClient() {
     return client;
   }, [client]);
 }
+const featureName = "embeddingsIndexApi";
+const FeatureEnabledContext = require$$0.createContext("loading");
+function useIsFeatureEnabled() {
+  const client = sanity.useClient({
+    apiVersion: "2023-09-01"
+  });
+  const [status, setStatus] = require$$0.useState("loading");
+  require$$0.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__ */jsxRuntime.jsx(FeatureEnabledContext.Provider, {
+    value: status,
+    children: props.children
+  });
+}
+function useIsFeatureEnabledContext() {
+  return require$$0.useContext(FeatureEnabledContext);
+}
+function FeatureDisabledNotice() {
+  return /* @__PURE__ */jsxRuntime.jsx(ui.Card, {
+    tone: "primary",
+    border: true,
+    padding: 2,
+    children: /* @__PURE__ */jsxRuntime.jsxs(ui.Text, {
+      size: 1,
+      children: ["Embeddings index APIs are only available on the", " ", /* @__PURE__ */jsxRuntime.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] = require$$0.useState(defaultEnabled);
   const toggleSemantic = require$$0.useCallback(() => setSemantic(current => !current), []);
   return /* @__PURE__ */jsxRuntime.jsxs(ui.Flex, {
@@ -1919,13 +1964,16 @@ function SemanticSearchReferenceInput(props) {
     style: {
       width: "100%"
     },
-    children: [/* @__PURE__ */jsxRuntime.jsx(ui.Box, {
+    children: [semantic && featureState == "loading" ? /* @__PURE__ */jsxRuntime.jsx(ui.Box, {
+      padding: 2,
+      children: /* @__PURE__ */jsxRuntime.jsx(ui.Spinner, {})
+    }) : null, semantic && featureState == "disabled" ? /* @__PURE__ */jsxRuntime.jsx(FeatureDisabledNotice, {}) : null, /* @__PURE__ */jsxRuntime.jsx(ui.Box, {
       flex: 1,
       style: {
         maxHeight: 36,
         overflow: "hidden"
       },
-      children: semantic ? /* @__PURE__ */jsxRuntime.jsx(SemanticSearchInput, {
+      children: semantic && featureState == "enabled" ? /* @__PURE__ */jsxRuntime.jsx(SemanticSearchInput, {
         ...props
       }) : props.renderDefault(props)
     }), /* @__PURE__ */jsxRuntime.jsx(ui.Button, {
@@ -2091,6 +2139,15 @@ function isType(schemaType, typeName) {
 }
 const embeddingsIndexReferenceInput = sanity.definePlugin({
   name: "@sanity/embeddings-index-reference-input",
+  studio: {
+    components: {
+      layout: props => {
+        return /* @__PURE__ */jsxRuntime.jsx(FeatureEnabledProvider, {
+          children: props.renderDefault(props)
+        });
+      }
+    }
+  },
   form: {
     components: {
       input: props => {
@@ -2660,17 +2717,21 @@ function IndexStatus(_ref4) {
   });
 }
 function EmbeddingsIndexTool() {
+  const featureState = useIsFeatureEnabled();
   return /* @__PURE__ */jsxRuntime.jsx(ui.Card, {
     children: /* @__PURE__ */jsxRuntime.jsx(ui.Flex, {
       justify: "center",
       flex: 1,
-      children: /* @__PURE__ */jsxRuntime.jsx(ui.Card, {
+      children: /* @__PURE__ */jsxRuntime.jsxs(ui.Card, {
         flex: 1,
         style: {
           maxWidth: 1200
         },
         padding: 5,
-        children: /* @__PURE__ */jsxRuntime.jsx(Indexes, {})
+        children: [featureState == "loading" ? /* @__PURE__ */jsxRuntime.jsx(ui.Box, {
+          padding: 2,
+          children: /* @__PURE__ */jsxRuntime.jsx(ui.Spinner, {})
+        }) : null, featureState == "disabled" ? /* @__PURE__ */jsxRuntime.jsx(FeatureDisabledNotice, {}) : null, featureState == "enabled" ? /* @__PURE__ */jsxRuntime.jsx(Indexes, {}) : null]
       })
     })
   });