@braze/sanity-plugin-ai-translation 0.1.7 → 0.1.76

package/dist/index.js

@@ -1106,9 +1106,10 @@ const ICON_SIZE = 64, progressIconMap = {
 }), fetchTranslationPrompts = async ({
   client,
   sourceLanguage,
+  sourceDocument,
   targetLanguages
 }) => {
-  const translationSettings2 = await client.fetch(
+  const sourceDocumentString = JSON.stringify(sourceDocument), translationSettings2 = await client.fetch(
     `*[_id == "translation-settings"][0] {
       globalPrompt,
       localePrompts[locale in $targetLanguages],
@@ -1118,19 +1119,23 @@ const ICON_SIZE = 64, progressIconMap = {
   ), translationGlossaryTerms = await client.fetch(
     `*[
         _type == "translationGlossaryTerm"
-        && count(translations[locale in $targetLanguages]) > 0
-        && count(translations[locale == $sourceLanguage]) > 0
+        && count(translations[_key in $targetLanguages]) > 0
+        && count(translations[_key == $sourceLanguage]) > 0
       ].translations`,
     { targetLanguages, sourceLanguage }
-  ) || [], glossaryTermsByLocale = {};
+  ) || [];
+  console.log("translationGlossaryTerms", translationGlossaryTerms);
+  const glossaryTermsByLocale = {};
   for (const targetLanguage of targetLanguages) {
-    const relevantTranslations = translationGlossaryTerms.filter(
+    let relevantTranslations = translationGlossaryTerms.filter(
       (termTranslations) => termTranslations.some((translation) => translation._key === targetLanguage)
     ).map((termTranslations) => {
       const from = termTranslations.find((translation) => translation._key === sourceLanguage)?.value, to = termTranslations.find((translation) => translation._key === targetLanguage)?.value;
       return !from || !to ? null : { from, to };
     }).filter((item) => item !== null);
-    glossaryTermsByLocale[targetLanguage] = relevantTranslations;
+    console.log("relevantTranslations", relevantTranslations), sourceDocument && (relevantTranslations = relevantTranslations.filter(
+      ({ from }) => termAppearsInDocument(from, sourceDocumentString)
+    ), console.log("relevantTranslations", relevantTranslations)), glossaryTermsByLocale[targetLanguage] = relevantTranslations;
   }
   return {
     globalPrompt: translationSettings2?.globalPrompt,
@@ -1145,12 +1150,17 @@ const ICON_SIZE = 64, progressIconMap = {
       ])
     )
   };
+}, termAppearsInDocument = (term, documentString) => {
+  const baseTerm = term.split(":")[0].trim();
+  return documentString.includes(baseTerm.toLowerCase());
 }, formatGlossaryTermInstruction = (terms) => {
   const instructions = [
-    "Translate the following terms exactly as described below. When a term includes a colon, the suffix following the colon describes how the term should be translated for the female gender."
+    "## Glossary Terms",
+    "Translate the following terms exactly as described below.",
+    "When a term includes a colon, the suffix following the colon describes how the term should be translated for the female gender."
   ];
   return terms.forEach(({ from, to }) => {
-    instructions.push(`"${from}" -> "${to}"`);
+    instructions.push(`Always translate "${from}" to "${to}".`);
   }), instructions.join(`
 `);
 }, findKeyValues = (obj, targetKey) => {
@@ -1383,7 +1393,11 @@ const setupTranslationJobStatuses = (dispatch, sourceDocuments, targetLanguages)
       }
     });
   } catch (error) {
-    setJobStatus(dispatch, sourceDocument._id, targetLanguage, "error"), console.error("Error translating document:", error);
+    const errorMessage = error instanceof Error ? error.message : String(error);
+    setJobStatus(dispatch, sourceDocument._id, targetLanguage, "error", {
+      errorMessage
+    }), console.error("Error translating document:", error);
+    return;
   }
 }, assembleLanguageSpecs = ({
   sourceDocument,
@@ -1479,11 +1493,15 @@ const setupTranslationJobStatuses = (dispatch, sourceDocuments, targetLanguages)
       newDocumentRev: newDocument._rev
     }), devLog.timeEnd("translateEntireDocument"), devLog.groupEnd(), newDocument;
   } catch (error) {
-    throw devLog.error(
+    const errorMessage = error instanceof Error ? error.message : String(error);
+    devLog.error(
       "translateEntireDocument failed",
       error instanceof Error ? error : new Error(String(error)),
       logContext
-    ), devLog.timeEnd("translateEntireDocument"), devLog.groupEnd(), error;
+    ), setJobStatus(dispatch, sourceDocument._id, targetLanguage, "error", {
+      errorMessage
+    }), devLog.timeEnd("translateEntireDocument"), devLog.groupEnd();
+    return;
   }
 }, updateTranslationMetadata = async ({
   client,
@@ -1644,7 +1662,11 @@ const setupTranslationJobStatuses = (dispatch, sourceDocuments, targetLanguages)
       newDocumentId: newDocument._id
     }), devLog.timeEnd("translateDocument"), devLog.groupEnd(), newDocument._id;
   } catch (error) {
-    throw devLog.error("Translation failed", error instanceof Error ? error : new Error(String(error)), logContext), setJobStatus(dispatch, sourceDocument._id, targetLanguage, "error"), devLog.timeEnd("translateDocument"), devLog.groupEnd(), error;
+    const errorMessage = error instanceof Error ? error.message : String(error);
+    devLog.error("Translation failed", error instanceof Error ? error : new Error(String(error)), logContext), setJobStatus(dispatch, sourceDocument._id, targetLanguage, "error", {
+      errorMessage
+    }), devLog.timeEnd("translateDocument"), devLog.groupEnd();
+    return;
   }
 }, patchTranslationTarget = async ({
   client,
@@ -1719,36 +1741,54 @@ const setupTranslationJobStatuses = (dispatch, sourceDocuments, targetLanguages)
     const translationPrompts = await fetchTranslationPrompts({
       client,
       sourceLanguage,
-      targetLanguages
+      targetLanguages,
+      sourceDocument
     });
     devLog.info("Translation prompts", translationPrompts), setupTranslationJobStatuses(dispatch, sourceDocuments, targetLanguages);
     const translationTasks = sourceDocumentIds.flatMap(
       (documentId) => targetLanguages.map(async (targetLanguage) => {
-        await translationSemaphore.withPermit(async () => {
-          const document = await client.withConfig({
-            // This enables us to detect documents that might be stale due to unpublished changes
-            perspective: "raw"
-          }).fetch("*[_id == $id][0]", {
-            id: documentId
-          });
-          if (document) {
-            const targetDocumentId = await translateDocument({
-              dispatch,
-              client,
-              sourceDocument: document,
-              targetLanguage,
-              languageFieldName,
-              shouldOverwriteFreshTranslations,
-              schemaId,
-              shouldTranslateDiffs,
-              translationPrompts
+        try {
+          await translationSemaphore.withPermit(async () => {
+            const document = await client.withConfig({
+              // This enables us to detect documents that might be stale due to unpublished changes
+              perspective: "raw"
+            }).fetch("*[_id == $id][0]", {
+              id: documentId
             });
-            targetDocumentId && targetDocumentIds.push(targetDocumentId);
-          }
-        });
+            if (document) {
+              const targetDocumentId = await translateDocument({
+                dispatch,
+                client,
+                sourceDocument: document,
+                targetLanguage,
+                languageFieldName,
+                shouldOverwriteFreshTranslations,
+                schemaId,
+                shouldTranslateDiffs,
+                translationPrompts
+              });
+              targetDocumentId && targetDocumentIds.push(targetDocumentId);
+            }
+          });
+        } catch (error) {
+          devLog.error(
+            "Individual translation task failed",
+            error instanceof Error ? error : new Error(String(error)),
+            {
+              ...logContext,
+              failedDocumentId: documentId,
+              failedTargetLanguage: targetLanguage
+            }
+          );
+        }
       })
-    );
-    await Promise.all(translationTasks), devLog.info("All translation tasks completed", {
+    ), results = await Promise.allSettled(translationTasks), succeeded = results.filter((r) => r.status === "fulfilled").length, failed = results.filter((r) => r.status === "rejected").length;
+    devLog.info("Translation tasks summary", {
+      ...logContext,
+      total: results.length,
+      succeeded,
+      failed
+    }), devLog.info("All translation tasks completed", {
       ...logContext,
       targetDocumentIds: targetDocumentIds.length
     }), dispatch({
@@ -1765,9 +1805,19 @@ const setupTranslationJobStatuses = (dispatch, sourceDocuments, targetLanguages)
           languageFieldName
         });
       })
-    ), devLog.info("Document set translation completed successfully", logContext), devLog.timeEnd("translateDocumentSet"), devLog.groupEnd(), dispatch({ type: "SET_OVERALL_STATUS", payload: "completed" });
+    );
+    const hasAnyFailures = failed > 0;
+    hasAnyFailures && succeeded === 0 ? (devLog.error("All translation tasks failed", void 0, logContext), dispatch({ type: "SET_OVERALL_STATUS", payload: "error" })) : hasAnyFailures ? (devLog.warn("Translation completed with some failures", {
+      ...logContext,
+      succeeded,
+      failed
+    }), dispatch({ type: "SET_OVERALL_STATUS", payload: "completed" })) : (devLog.info("Document set translation completed successfully", logContext), dispatch({ type: "SET_OVERALL_STATUS", payload: "completed" })), devLog.timeEnd("translateDocumentSet"), devLog.groupEnd();
   } catch (error) {
-    return devLog.error("translateDocumentSet failed", error instanceof Error ? error : new Error(String(error)), logContext), devLog.timeEnd("translateDocumentSet"), devLog.groupEnd(), dispatch({ type: "SET_OVERALL_STATUS", payload: "error" }), Promise.reject(error);
+    throw devLog.error(
+      "translateDocumentSet failed with unexpected error",
+      error instanceof Error ? error : new Error(String(error)),
+      logContext
+    ), devLog.timeEnd("translateDocumentSet"), devLog.groupEnd(), dispatch({ type: "SET_OVERALL_STATUS", payload: "error" }), error;
   }
 }, ActionButtonBar = ({ setIsDialogOpen }) => {
   const {
@@ -2086,7 +2136,7 @@ const setupTranslationJobStatuses = (dispatch, sourceDocuments, targetLanguages)
       /* @__PURE__ */ jsx(Label, { children: "Translations" }),
       /* @__PURE__ */ jsx(Card, { padding: 2, tone: "transparent", children: /* @__PURE__ */ jsx(Stack, { space: 2, children: /* @__PURE__ */ jsx("ul", { style: { listStyleType: "none", paddingInlineStart: "0" }, children: jobs.map((job) => /* @__PURE__ */ jsx("li", { children: /* @__PURE__ */ jsx(Box, { padding: 3, children: /* @__PURE__ */ jsxs(Flex, { justify: "space-between", align: "center", children: [
         /* @__PURE__ */ jsxs(Inline, { space: 2, children: [
-          /* @__PURE__ */ jsx(Flex, { align: "center", justify: "center", style: { width: "24px", height: "24px" }, children: ["completed", "skipped"].includes(job.status) ? /* @__PURE__ */ jsx(CheckmarkIcon, { fontSize: 24 }) : job.status === "queued" ? /* @__PURE__ */ jsx(CircleIcon, { fontSize: 24 }) : /* @__PURE__ */ jsx(Spinner, {}) }),
+          /* @__PURE__ */ jsx(Flex, { align: "center", justify: "center", style: { width: "24px", height: "24px" }, children: ["completed", "skipped"].includes(job.status) ? /* @__PURE__ */ jsx(CheckmarkIcon, { fontSize: 24 }) : job.status === "error" ? /* @__PURE__ */ jsx(Box, { style: { color: "var(--card-critical-fg-color)" }, children: /* @__PURE__ */ jsx(ErrorOutlineIcon, { fontSize: 24 }) }) : job.status === "queued" ? /* @__PURE__ */ jsx(CircleIcon, { fontSize: 24 }) : /* @__PURE__ */ jsx(Spinner, {}) }),
           /* @__PURE__ */ jsxs(Stack, { space: 3, children: [
             /* @__PURE__ */ jsxs(Inline, { space: 2, children: [
               /* @__PURE__ */ jsx(Text, { size: 1, weight: "semibold", children: job.title || "Untitled Document" }),
@@ -2097,11 +2147,16 @@ const setupTranslationJobStatuses = (dispatch, sourceDocuments, targetLanguages)
               job._type,
               " \u2014 ",
               job._id.slice(-6)
-            ] })
+            ] }),
+            job.status === "error" && job.errorMessage ? /* @__PURE__ */ jsx(Card, { tone: "critical", padding: 2, radius: 2, children: /* @__PURE__ */ jsxs(Text, { size: 1, children: [
+              "Error: ",
+              job.errorMessage
+            ] }) }) : null
           ] })
         ] }),
         /* @__PURE__ */ jsxs(Flex, { gap: 2, align: "center", children: [
           job.status === "skipped" ? /* @__PURE__ */ jsx(SkippedBadge, {}) : null,
+          job.status === "error" ? /* @__PURE__ */ jsx(ErrorBadge, {}) : null,
           job.targetId ? /* @__PURE__ */ jsx(
             TranslationOpenerButton,
             {
@@ -2165,6 +2220,13 @@ const setupTranslationJobStatuses = (dispatch, sourceDocuments, targetLanguages)
     content: /* @__PURE__ */ jsx(Box, { padding: 1, children: /* @__PURE__ */ jsx(Text, { size: 1, muted: !0, children: "There was already a fresh translation for this document and language." }) }),
     children: /* @__PURE__ */ jsx(Badge, { tone: "neutral", children: "Skipped" })
   }
+), ErrorBadge = () => /* @__PURE__ */ jsx(
+  Tooltip,
+  {
+    portal: !0,
+    content: /* @__PURE__ */ jsx(Box, { padding: 1, children: /* @__PURE__ */ jsx(Text, { size: 1, muted: !0, children: "This translation failed. Check the error message for details." }) }),
+    children: /* @__PURE__ */ jsx(Badge, { tone: "critical", children: "Failed" })
+  }
 ), STATES_IN_WHICH_TO_SHOW_CONFIG = ["fetching_metadata", "idle"], STATES_IN_WHICH_TO_SHOW_TRANSLATION_PROGRESS = [
   "translating",
   "linking_references",