@mux/ai 0.4.0 → 0.4.2

package/dist/index.js

@@ -1338,57 +1338,57 @@ async function processConcurrently(items, processor, maxConcurrent = 5) {
   }
   return results;
 }
+async function moderateImageWithOpenAI(entry) {
+  "use step";
+  const apiKey = getApiKeyFromEnv("openai");
+  try {
+    const res = await fetch("https://api.openai.com/v1/moderations", {
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json",
+        "Authorization": `Bearer ${apiKey}`
+      },
+      body: JSON.stringify({
+        model: entry.model,
+        input: [
+          {
+            type: "image_url",
+            image_url: {
+              url: entry.image
+            }
+          }
+        ]
+      })
+    });
+    const json = await res.json();
+    if (!res.ok) {
+      throw new Error(
+        `OpenAI moderation error: ${res.status} ${res.statusText} - ${JSON.stringify(json)}`
+      );
+    }
+    const categoryScores = json.results?.[0]?.category_scores || {};
+    return {
+      url: entry.url,
+      sexual: categoryScores.sexual || 0,
+      violence: categoryScores.violence || 0,
+      error: false
+    };
+  } catch (error) {
+    console.error("OpenAI moderation failed:", error);
+    return {
+      url: entry.url,
+      sexual: 0,
+      violence: 0,
+      error: true
+    };
+  }
+}
 async function requestOpenAIModeration(imageUrls, model, maxConcurrent = 5, submissionMode = "url", downloadOptions) {
   "use step";
   const targetUrls = submissionMode === "base64" ? (await downloadImagesAsBase64(imageUrls, downloadOptions, maxConcurrent)).map(
     (img) => ({ url: img.url, image: img.base64Data, model })
   ) : imageUrls.map((url) => ({ url, image: url, model }));
-  const moderate = async (entry) => {
-    "use step";
-    const apiKey = getApiKeyFromEnv("openai");
-    try {
-      const res = await fetch("https://api.openai.com/v1/moderations", {
-        method: "POST",
-        headers: {
-          "Content-Type": "application/json",
-          "Authorization": `Bearer ${apiKey}`
-        },
-        body: JSON.stringify({
-          model: entry.model,
-          input: [
-            {
-              type: "image_url",
-              image_url: {
-                url: entry.image
-              }
-            }
-          ]
-        })
-      });
-      const json = await res.json();
-      if (!res.ok) {
-        throw new Error(
-          `OpenAI moderation error: ${res.status} ${res.statusText} - ${JSON.stringify(json)}`
-        );
-      }
-      const categoryScores = json.results?.[0]?.category_scores || {};
-      return {
-        url: entry.url,
-        sexual: categoryScores.sexual || 0,
-        violence: categoryScores.violence || 0,
-        error: false
-      };
-    } catch (error) {
-      console.error("OpenAI moderation failed:", error);
-      return {
-        url: entry.url,
-        sexual: 0,
-        violence: 0,
-        error: true
-      };
-    }
-  };
-  return processConcurrently(targetUrls, moderate, maxConcurrent);
+  return processConcurrently(targetUrls, moderateImageWithOpenAI, maxConcurrent);
 }
 function getHiveCategoryScores(classes, categoryNames) {
   const scoreMap = Object.fromEntries(
@@ -1397,6 +1397,51 @@ function getHiveCategoryScores(classes, categoryNames) {
   const scores = categoryNames.map((category) => scoreMap[category] || 0);
   return Math.max(...scores, 0);
 }
+async function moderateImageWithHive(entry) {
+  "use step";
+  const apiKey = getApiKeyFromEnv("hive");
+  try {
+    const formData = new FormData();
+    if (entry.source.kind === "url") {
+      formData.append("url", entry.source.value);
+    } else {
+      const extension = entry.source.contentType.split("/")[1] || "jpg";
+      const blob = new Blob([entry.source.buffer], {
+        type: entry.source.contentType
+      });
+      formData.append("media", blob, `thumbnail.${extension}`);
+    }
+    const res = await fetch(HIVE_ENDPOINT, {
+      method: "POST",
+      headers: {
+        Accept: "application/json",
+        Authorization: `Token ${apiKey}`
+      },
+      body: formData
+    });
+    const json = await res.json().catch(() => void 0);
+    if (!res.ok) {
+      throw new Error(
+        `Hive moderation error: ${res.status} ${res.statusText} - ${JSON.stringify(json)}`
+      );
+    }
+    const classes = json?.status?.[0]?.response?.output?.[0]?.classes || [];
+    return {
+      url: entry.url,
+      sexual: getHiveCategoryScores(classes, HIVE_SEXUAL_CATEGORIES),
+      violence: getHiveCategoryScores(classes, HIVE_VIOLENCE_CATEGORIES),
+      error: false
+    };
+  } catch (error) {
+    console.error("Hive moderation failed:", error);
+    return {
+      url: entry.url,
+      sexual: 0,
+      violence: 0,
+      error: true
+    };
+  }
+}
 async function requestHiveModeration(imageUrls, maxConcurrent = 5, submissionMode = "url", downloadOptions) {
   "use step";
   const targets = submissionMode === "base64" ? (await downloadImagesAsBase64(imageUrls, downloadOptions, maxConcurrent)).map((img) => ({
@@ -1410,52 +1455,7 @@ async function requestHiveModeration(imageUrls, maxConcurrent = 5, submissionMod
     url,
     source: { kind: "url", value: url }
   }));
-  const moderate = async (entry) => {
-    "use step";
-    const apiKey = getApiKeyFromEnv("hive");
-    try {
-      const formData = new FormData();
-      if (entry.source.kind === "url") {
-        formData.append("url", entry.source.value);
-      } else {
-        const extension = entry.source.contentType.split("/")[1] || "jpg";
-        const blob = new Blob([entry.source.buffer], {
-          type: entry.source.contentType
-        });
-        formData.append("media", blob, `thumbnail.${extension}`);
-      }
-      const res = await fetch(HIVE_ENDPOINT, {
-        method: "POST",
-        headers: {
-          Accept: "application/json",
-          Authorization: `Token ${apiKey}`
-        },
-        body: formData
-      });
-      const json = await res.json().catch(() => void 0);
-      if (!res.ok) {
-        throw new Error(
-          `Hive moderation error: ${res.status} ${res.statusText} - ${JSON.stringify(json)}`
-        );
-      }
-      const classes = json?.status?.[0]?.response?.output?.[0]?.classes || [];
-      return {
-        url: entry.url,
-        sexual: getHiveCategoryScores(classes, HIVE_SEXUAL_CATEGORIES),
-        violence: getHiveCategoryScores(classes, HIVE_VIOLENCE_CATEGORIES),
-        error: false
-      };
-    } catch (error) {
-      console.error("Hive moderation failed:", error);
-      return {
-        url: entry.url,
-        sexual: 0,
-        violence: 0,
-        error: true
-      };
-    }
-  };
-  return processConcurrently(targets, moderate, maxConcurrent);
+  return processConcurrently(targets, moderateImageWithHive, maxConcurrent);
 }
 async function getModerationScores(assetId, options = {}) {
   "use workflow";