@mux/ai 0.7.4 → 0.7.6

package/dist/workflows/index.js

@@ -138,6 +138,10 @@ var EnvSchema = z.object({
   ),
   MUX_SIGNING_KEY: optionalString("Mux signing key ID for signed playback URLs.", "Used to sign playback URLs"),
   MUX_PRIVATE_KEY: optionalString("Mux signing private key for signed playback URLs.", "Used to sign playback URLs"),
+  MUX_IMAGE_URL_OVERRIDE: optionalString(
+    "Override for Mux image base URL (defaults to https://image.mux.com).",
+    "Mux image URL override"
+  ),
   // Test-only helpers (used by this repo's integration tests)
   MUX_TEST_ASSET_ID: optionalString("Mux asset ID used by integration tests.", "Mux test asset id"),
   MUX_TEST_ASSET_ID_CHAPTERS: optionalString("Mux asset ID used by integration tests for chapters.", "Mux test asset id for chapters"),
@@ -424,12 +428,13 @@ function readString(record, key) {
 function resolveDirectMuxCredentials(record) {
   const tokenId = readString(record, "muxTokenId");
   const tokenSecret = readString(record, "muxTokenSecret");
+  const authorizationToken = readString(record, "muxAuthorizationToken");
   const signingKey = readString(record, "muxSigningKey");
   const privateKey = readString(record, "muxPrivateKey");
-  if (!tokenId && !tokenSecret && !signingKey && !privateKey) {
+  if (!tokenId && !tokenSecret && !authorizationToken && !signingKey && !privateKey) {
     return void 0;
   }
-  if (!tokenId || !tokenSecret) {
+  if ((!tokenId || !tokenSecret) && !authorizationToken) {
     throw new Error(
       "Both muxTokenId and muxTokenSecret are required when passing direct Mux workflow credentials."
     );
@@ -437,6 +442,7 @@ function resolveDirectMuxCredentials(record) {
   return {
     tokenId,
     tokenSecret,
+    authorizationToken,
     signingKey,
     privateKey
   };
@@ -447,7 +453,8 @@ function createWorkflowMuxClient(options) {
       const { default: MuxClient } = await import("@mux/mux-node");
       return new MuxClient({
         tokenId: options.tokenId,
-        tokenSecret: options.tokenSecret
+        tokenSecret: options.tokenSecret,
+        authorizationToken: options.authorizationToken
       });
     },
     getSigningKey() {
@@ -832,6 +839,44 @@ async function withRetry(fn, {
   throw lastError || new Error("Retry failed with unknown error");
 }
 
+// src/lib/mux-image-url.ts
+var DEFAULT_MUX_IMAGE_ORIGIN = "https://image.mux.com";
+function normalizeMuxImageOrigin(value) {
+  const trimmed = value.trim();
+  const candidate = trimmed.includes("://") ? trimmed : `https://${trimmed}`;
+  let parsed;
+  try {
+    parsed = new URL(candidate);
+  } catch {
+    throw new Error(
+      `Invalid MUX_IMAGE_URL_OVERRIDE. Provide a hostname like "image.example.mux.com" (or a URL origin such as "https://image.example.mux.com").`
+    );
+  }
+  if (parsed.username || parsed.password || parsed.search || parsed.hash || parsed.pathname && parsed.pathname !== "/") {
+    throw new Error(
+      "Invalid MUX_IMAGE_URL_OVERRIDE. Only a hostname/origin is allowed (no credentials, query params, hash fragments, or path)."
+    );
+  }
+  return parsed.origin;
+}
+function getMuxImageOrigin() {
+  const override = env_default.MUX_IMAGE_URL_OVERRIDE;
+  if (!override) {
+    return DEFAULT_MUX_IMAGE_ORIGIN;
+  }
+  return normalizeMuxImageOrigin(override);
+}
+function getMuxImageBaseUrl(playbackId, assetType) {
+  const origin = getMuxImageOrigin();
+  return `${origin}/${playbackId}/${assetType}.png`;
+}
+function getMuxStoryboardBaseUrl(playbackId) {
+  return getMuxImageBaseUrl(playbackId, "storyboard");
+}
+function getMuxThumbnailBaseUrl(playbackId) {
+  return getMuxImageBaseUrl(playbackId, "thumbnail");
+}
+
 // src/lib/url-signing.ts
 async function createSigningClient(context) {
   const { default: MuxClient } = await import("@mux/mux-node");
@@ -856,9 +901,9 @@ async function signPlaybackId(playbackId, context, type = "video", params) {
     params: stringParams
   });
 }
-async function signUrl(url, playbackId, context, type = "video", params, credentials) {
+async function signUrl(url, playbackId, type = "video", params, credentials) {
   "use step";
-  const resolvedContext = context ?? await resolveMuxSigningContext(credentials);
+  const resolvedContext = await resolveMuxSigningContext(credentials);
   if (!resolvedContext) {
     throw new Error(
       "Signed playback ID requires signing credentials. Provide muxSigningKey and muxPrivateKey via workflow credentials or set MUX_SIGNING_KEY and MUX_PRIVATE_KEY environment variables."
@@ -873,9 +918,9 @@ async function signUrl(url, playbackId, context, type = "video", params, credent
 var DEFAULT_STORYBOARD_WIDTH = 640;
 async function getStoryboardUrl(playbackId, width = DEFAULT_STORYBOARD_WIDTH, shouldSign = false, credentials) {
   "use step";
-  const baseUrl = `https://image.mux.com/${playbackId}/storyboard.png`;
+  const baseUrl = getMuxStoryboardBaseUrl(playbackId);
   if (shouldSign) {
-    return signUrl(baseUrl, playbackId, void 0, "storyboard", { width }, credentials);
+    return signUrl(baseUrl, playbackId, "storyboard", { width }, credentials);
   }
   return `${baseUrl}?width=${width}`;
 }
@@ -992,7 +1037,7 @@ async function buildTranscriptUrl(playbackId, trackId, shouldSign = false, crede
   "use step";
   const baseUrl = `https://stream.mux.com/${playbackId}/text/${trackId}.vtt`;
   if (shouldSign) {
-    return signUrl(baseUrl, playbackId, void 0, "video", void 0, credentials);
+    return signUrl(baseUrl, playbackId, "video", void 0, credentials);
   }
   return baseUrl;
 }
@@ -2082,10 +2127,10 @@ async function getThumbnailUrls(playbackId, duration, options = {}) {
     }
     timestamps = newTimestamps;
   }
-  const baseUrl = `https://image.mux.com/${playbackId}/thumbnail.png`;
+  const baseUrl = getMuxThumbnailBaseUrl(playbackId);
   const urlPromises = timestamps.map(async (time) => {
     if (shouldSign) {
-      return signUrl(baseUrl, playbackId, void 0, "thumbnail", { time, width }, credentials);
+      return signUrl(baseUrl, playbackId, "thumbnail", { time, width }, credentials);
     }
     return `${baseUrl}?time=${time}&width=${width}`;
   });
@@ -2101,11 +2146,10 @@ var DEFAULT_PROVIDER2 = "openai";
 var HIVE_ENDPOINT = "https://api.thehive.ai/api/v2/task/sync";
 var HIVE_SEXUAL_CATEGORIES = [
   "general_nsfw",
-  "general_suggestive",
   "yes_sexual_activity",
-  "sex_toys",
-  "nudity_female",
-  "nudity_male"
+  "yes_sex_toy",
+  "yes_female_nudity",
+  "yes_male_nudity"
 ];
 var HIVE_VIOLENCE_CATEGORIES = [
   "gun_in_hand",
@@ -2116,10 +2160,8 @@ var HIVE_VIOLENCE_CATEGORIES = [
   "hanging",
   "noose",
   "human_corpse",
-  "emaciated_body",
-  "self_harm",
-  "animal_abuse",
-  "fights",
+  "yes_emaciated_body",
+  "yes_self_harm",
   "garm_death_injury_or_military_conflict"
 ];
 async function processConcurrently(items, processor, maxConcurrent = 5) {
@@ -2263,6 +2305,12 @@ function getHiveCategoryScores(classes, categoryNames) {
   const scoreMap = Object.fromEntries(
     classes.map((c) => [c.class, c.score])
   );
+  const missingCategories = categoryNames.filter((category) => !(category in scoreMap));
+  if (missingCategories.length > 0) {
+    console.warn(
+      `Hive response missing expected categories: ${missingCategories.join(", ")}`
+    );
+  }
   const scores = categoryNames.map((category) => scoreMap[category] || 0);
   return Math.max(...scores, 0);
 }
@@ -2356,11 +2404,11 @@ async function requestHiveModeration(imageUrls, maxConcurrent = 5, submissionMod
 async function getThumbnailUrlsFromTimestamps(playbackId, timestampsMs, options) {
   "use step";
   const { width, shouldSign, credentials } = options;
-  const baseUrl = `https://image.mux.com/${playbackId}/thumbnail.png`;
+  const baseUrl = getMuxThumbnailBaseUrl(playbackId);
   const urlPromises = timestampsMs.map(async (tsMs) => {
     const time = Number((tsMs / 1e3).toFixed(2));
     if (shouldSign) {
-      return signUrl(baseUrl, playbackId, void 0, "thumbnail", { time, width }, credentials);
+      return signUrl(baseUrl, playbackId, "thumbnail", { time, width }, credentials);
     }
     return `${baseUrl}?time=${time}&width=${width}`;
   });
@@ -3633,7 +3681,7 @@ async function translateAudio(assetId, toLanguageCode, options = {}) {
   }
   let audioUrl = `https://stream.mux.com/${playbackId}/audio.m4a`;
   if (policy === "signed") {
-    audioUrl = await signUrl(audioUrl, playbackId, void 0, "video", void 0, credentials);
+    audioUrl = await signUrl(audioUrl, playbackId, "video", void 0, credentials);
   }
   console.warn("\u{1F399}\uFE0F Fetching audio from Mux...");
   let audioBuffer;
@@ -4013,6 +4061,8 @@ async function translateCaptions(assetId, fromLanguageCode, toLanguageCode, opti
   };
 }
 export {
+  HIVE_SEXUAL_CATEGORIES,
+  HIVE_VIOLENCE_CATEGORIES,
   SUMMARY_KEYWORD_LIMIT,
   askQuestions,
   burnedInCaptionsSchema,