@sanity/assist 3.2.2 → 4.0.1

package/dist/index.js

@@ -229,7 +229,7 @@ function isDisabled(type) {
   return !isSchemaAssistEnabled(type) || isUnsupportedType(type);
 }
 function isUnsupportedType(type) {
-  return type.jsonType === "number" || type.name === "sanity.imageCrop" || type.name === "sanity.imageHotspot" || isType(type, "globalDocumentReference") || isType(type, "reference") && !type?.options?.aiAssist?.embeddingsIndex || isType(type, "crossDatasetReference") || isType(type, "slug") || isType(type, "url") || isType(type, "date") || isType(type, "datetime") || isType(type, "file");
+  return type.name === "sanity.imageCrop" || type.name === "sanity.imageHotspot" || isType(type, "globalDocumentReference") || isType(type, "reference") && !type?.options?.aiAssist?.embeddingsIndex || isType(type, "crossDatasetReference") || isType(type, "file");
 }
 const FirstAssistedPathContext = react.createContext(void 0);
 function FirstAssistedPathProvider(props) {
@@ -459,6 +459,39 @@ function useAiPaneRouter() {
     [paneRouter]
   );
 }
+const AiAssistanceConfigContext = react.createContext({});
+function useAiAssistanceConfig() {
+  const context = react.useContext(AiAssistanceConfigContext);
+  if (!context)
+    throw new Error("Missing AiAssistanceConfigContext");
+  return context;
+}
+function AiAssistanceConfigProvider(props) {
+  const [status, setStatus] = react.useState(), [error, setError] = react.useState(), apiClient = useApiClient(props.config?.__customApiClient), { getInstructStatus, loading: statusLoading } = useGetInstructStatus(apiClient), { initInstruct, loading: initLoading } = useInitInstruct(apiClient);
+  react.useEffect(() => {
+    getInstructStatus().then((s) => setStatus(s)).catch((e) => {
+      console.error(e), setError(e);
+    });
+  }, [getInstructStatus]);
+  const init = react.useCallback(async () => {
+    setError(void 0);
+    try {
+      await initInstruct();
+      const status2 = await getInstructStatus();
+      setStatus(status2);
+    } catch (e) {
+      console.error("Failed to init ai assistance", e), setError(e);
+    }
+  }, [initInstruct, getInstructStatus, setStatus]), { config, children } = props, context = react.useMemo(() => ({
+    config,
+    status,
+    statusLoading,
+    init,
+    initLoading,
+    error
+  }), [config, status, init, statusLoading, initLoading, error]);
+  return /* @__PURE__ */ jsxRuntime.jsx(AiAssistanceConfigContext.Provider, { value: context, children });
+}
 const hiddenTypes = [
   "any",
   "array",
@@ -582,12 +615,12 @@ function refToTypeNames(type) {
 function removeUndef(obj) {
   return Object.keys(obj).forEach((key) => obj[key] === void 0 ? delete obj[key] : {}), obj;
 }
-const basePath = "/assist/tasks/instruction";
+const basePath = "/assist/tasks/instruction", API_VERSION_WITH_EXTENDED_TYPES = "2025-04-01";
 function canUseAssist(status) {
   return status?.enabled && status.initialized && status.validToken;
 }
 function useApiClient(customApiClient) {
-  const client = sanity.useClient({ apiVersion: "2023-06-05" });
+  const client = sanity.useClient({ apiVersion: API_VERSION_WITH_EXTENDED_TYPES });
   return react.useMemo(
     () => customApiClient ? customApiClient(client) : client,
     [client, customApiClient]
@@ -731,25 +764,40 @@ function useInitInstruct(apiClient) {
   };
 }
 function useRunInstructionApi(apiClient) {
-  const toast = ui.useToast(), [loading, setLoading] = react.useState(!1), user = sanity.useCurrentUser(), schema = sanity.useSchema(), types = react.useMemo(() => serializeSchema(schema, { leanFormat: !0 }), [schema]), runInstruction = react.useCallback(
-    (request) => (setLoading(!0), apiClient.request({
-      method: "POST",
-      url: `${basePath}/${apiClient.config().dataset}?projectId=${apiClient.config().projectId}`,
-      body: {
-        ...request,
-        types,
-        userId: user?.id
+  const toast = ui.useToast(), [loading, setLoading] = react.useState(!1), user = sanity.useCurrentUser(), schema = sanity.useSchema(), types = react.useMemo(() => serializeSchema(schema, { leanFormat: !0 }), [schema]), {
+    config: { assist: assistConfig }
+  } = useAiAssistanceConfig(), runInstruction = react.useCallback(
+    (request) => {
+      if (!user) {
+        toast.push({
+          status: "error",
+          title: "Unable to get user for instruction."
+        });
+        return;
       }
-    }).catch((e) => {
-      throw toast.push({
-        status: "error",
-        title: "Instruction failed",
-        description: e.message
-      }), e;
-    }).finally(() => {
-      setLoading(!1);
-    })),
-    [apiClient, types, user, toast]
+      setLoading(!0);
+      const { timeZone, locale } = Intl.DateTimeFormat().resolvedOptions(), defaultLocaleSettings = { timeZone, locale }, localeSettings = assistConfig?.localeSettings?.({ user, defaultSettings: defaultLocaleSettings }) ?? defaultLocaleSettings;
+      return apiClient.request({
+        method: "POST",
+        url: `${basePath}/${apiClient.config().dataset}?projectId=${apiClient.config().projectId}`,
+        body: {
+          ...request,
+          types,
+          userId: user?.id,
+          localeSettings,
+          maxPathDepth: assistConfig?.maxPathDepth
+        }
+      }).catch((e) => {
+        throw toast.push({
+          status: "error",
+          title: "Instruction failed",
+          description: e.message
+        }), e;
+      }).finally(() => {
+        setLoading(!1);
+      });
+    },
+    [apiClient, types, user, toast, assistConfig]
   );
   return react.useMemo(
     () => ({
@@ -759,39 +807,6 @@ function useRunInstructionApi(apiClient) {
     [runInstruction, loading]
   );
 }
-const AiAssistanceConfigContext = react.createContext({});
-function useAiAssistanceConfig() {
-  const context = react.useContext(AiAssistanceConfigContext);
-  if (!context)
-    throw new Error("Missing AiAssistanceConfigContext");
-  return context;
-}
-function AiAssistanceConfigProvider(props) {
-  const [status, setStatus] = react.useState(), [error, setError] = react.useState(), apiClient = useApiClient(props.config?.__customApiClient), { getInstructStatus, loading: statusLoading } = useGetInstructStatus(apiClient), { initInstruct, loading: initLoading } = useInitInstruct(apiClient);
-  react.useEffect(() => {
-    getInstructStatus().then((s) => setStatus(s)).catch((e) => {
-      console.error(e), setError(e);
-    });
-  }, [getInstructStatus]);
-  const init = react.useCallback(async () => {
-    setError(void 0);
-    try {
-      await initInstruct();
-      const status2 = await getInstructStatus();
-      setStatus(status2);
-    } catch (e) {
-      console.error("Failed to init ai assistance", e), setError(e);
-    }
-  }, [initInstruct, getInstructStatus, setStatus]), { config, children } = props, context = react.useMemo(() => ({
-    config,
-    status,
-    statusLoading,
-    init,
-    initLoading,
-    error
-  }), [config, status, init, statusLoading, initLoading, error]);
-  return /* @__PURE__ */ jsxRuntime.jsx(AiAssistanceConfigContext.Provider, { value: context, children });
-}
 const NO_INPUT = {}, RunInstructionContext = react.createContext({
   runInstruction: () => {
   },
@@ -960,10 +975,7 @@ function useAssistDocumentContextValue(documentId, documentType) {
     // @ts-ignore this is a valid option available in `corel` - Remove after corel is merged to next
     selectedReleaseId,
     editState
-  } = structure.useDocumentPane(), { draft, published, version } = editState || {};
-  let assistableDocumentId = version?._id || draft?._id || published?._id;
-  assistableDocumentId || (assistableDocumentId = selectedReleaseId ? sanity.getVersionId(documentId, selectedReleaseId) : documentSchemaType.liveEdit ? documentId : sanity.getDraftId(documentId));
-  const documentIsNew = selectedReleaseId ? !version?._id : !draft?._id && !published?._id, documentIsAssistable = selectedReleaseId ? !!version : isDocAssistable(documentSchemaType, published, draft), { params } = useAiPaneRouter(), selectedPath = params[fieldPathParam], assistDocument = useStudioAssistDocument({
+  } = structure.useDocumentPane(), { draft, published, version } = editState || {}, assistableDocumentId = selectedReleaseId ? sanity.getVersionId(documentId, selectedReleaseId) : documentSchemaType.liveEdit ? documentId : sanity.getDraftId(documentId), documentIsNew = selectedReleaseId ? !version?._id : !draft?._id && !published?._id, documentIsAssistable = selectedReleaseId ? !!version : isDocAssistable(documentSchemaType, published, draft), { params } = useAiPaneRouter(), selectedPath = params[fieldPathParam], assistDocument = useStudioAssistDocument({
     documentId: assistableDocumentId,
     schemaType: documentSchemaType
   });
@@ -1968,7 +1980,7 @@ function AssistInspector(props) {
               mode: "ghost",
               disabled: isEmptyPrompt || instructionLoading,
               fontSize: 1,
-              icon: instructionLoading ? /* @__PURE__ */ jsxRuntime.jsx(ui.Spinner, {}) : icons.PlayIcon,
+              icon: instructionLoading ? /* @__PURE__ */ jsxRuntime.jsx(ui.Spinner, { style: { marginTop: 3 } }) : icons.PlayIcon,
               onClick: runCurrentInstruction,
               padding: 3,
               text: "Run instruction"
@@ -3864,11 +3876,19 @@ const instructionForm = [
   instructionTask,
   contextDocumentSchema
 ], assist = sanity.definePlugin((config) => {
-  const configWithDefaults = config ?? {}, styleguide = configWithDefaults.translate?.styleguide || "";
+  const configWithDefaults = config ?? {}, styleguide = configWithDefaults.translate?.styleguide || "", maxPathDepth = configWithDefaults.assist?.maxPathDepth, temperature = configWithDefaults.assist?.temperature;
   if (styleguide.length > 2e3)
     throw new Error(
       `[${packageName}]: \`translate.styleguide\` value is too long. It must be 2000 characters or less, was ${styleguide.length} characters`
     );
+  if (maxPathDepth !== void 0 && (maxPathDepth < 1 || maxPathDepth > 12))
+    throw new Error(
+      `[${packageName}]: \`assist.maxPathDepth\` must be be in the range [1,12] inclusive, but was ${maxPathDepth}`
+    );
+  if (temperature !== void 0 && (temperature < 0 || temperature > 1))
+    throw new Error(
+      `[${packageName}]: \`assist.maxPathDepth\` must be be in the range [0,1] inclusive, but was ${temperature}`
+    );
   return {
     name: packageName,
     handlesGDR: !0,