@sanity/embeddings-index-ui 1.0.1 → 1.0.3

package/dist/index.js

@@ -1890,11 +1890,72 @@ function deleteIndex(indexName, client) {
 function publicId(id) {
   return id.replace("drafts.", "");
 }
+function useApiClient() {
+  const client = sanity.useClient({
+    apiVersion: "vX"
+  });
+  return require$$0.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 = 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, {
@@ -1903,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, {
@@ -1930,12 +1994,6 @@ function useDebouncedValue(value, ms) {
   }, [value, ms]);
   return debouncedValue;
 }
-function useApiClient$2() {
-  const client = sanity.useClient({
-    apiVersion: "vX"
-  });
-  return require$$0.useMemo(() => client, [client]);
-}
 function SemanticSearchInput(props) {
   const {
     onPathFocus,
@@ -1956,7 +2014,7 @@ function SemanticSearchInput(props) {
   const prevDebouncedQuery = require$$0.useRef(debouncedQuery);
   const [searching, setSearching] = require$$0.useState(false);
   const [options, setOptions] = require$$0.useState(NO_OPTIONS);
-  const client = useApiClient$2();
+  const client = useApiClient();
   require$$0.useEffect(() => {
     docRef.current = currentDocument;
   }, [currentDocument]);
@@ -2081,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 => {
@@ -2095,12 +2162,6 @@ const embeddingsIndexReferenceInput = sanity.definePlugin({
     }
   }
 });
-function useApiClient$1(customApiClient) {
-  const client = sanity.useClient({
-    apiVersion: "vX"
-  });
-  return require$$0.useMemo(() => customApiClient ? customApiClient(client) : client, [client, customApiClient]);
-}
 const defaultProjection = "{...}";
 function useDefaultIndex(schema, dataset) {
   const defaultFilter = require$$0.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]);
@@ -2217,7 +2278,7 @@ function IndexEditor(props) {
     index: selectedIndex,
     onSubmit
   } = props;
-  const client = useApiClient$1();
+  const client = useApiClient();
   const schema = sanity.useSchema();
   const defaultIndex = useDefaultIndex(schema, (_a = client.config().dataset) != null ? _a : "");
   const [errors, setErrors] = require$$0.useState();
@@ -2437,7 +2498,7 @@ function QueryIndex(props) {
   const [query, setQuery] = require$$0.useState("");
   const [searching, setSearching] = require$$0.useState(false);
   const [results, setResults] = require$$0.useState(NO_RESULTS);
-  const client = useApiClient$1();
+  const client = useApiClient();
   const search = require$$0.useCallback(queryString => {
     setSearching(true);
     return queryIndex({
@@ -2655,24 +2716,22 @@ function IndexStatus(_ref4) {
     })]
   });
 }
-function useApiClient() {
-  const client = sanity.useClient({
-    apiVersion: "vX"
-  });
-  return require$$0.useMemo(() => client, [client]);
-}
 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]
       })
     })
   });