@2515097216/jimeng-free-api 0.8.8 → 0.8.11

package/dist/{chunk-KHOQ4PCS.js → chunk-YJIZUPUG.js}

@@ -31,10 +31,29 @@ var Environment = class {
     this.package = _package;
   }
 };
+function loadPackageJson() {
+  const cwdPkgPath = path.join(process.cwd(), "package.json");
+  if (fs.pathExistsSync(cwdPkgPath)) {
+    return JSON.parse(fs.readFileSync(cwdPkgPath).toString());
+  }
+  const entryFile = process.argv[1];
+  const entryDir = entryFile ? path.dirname(entryFile) : process.cwd();
+  const candidates = [
+    path.resolve(entryDir, "../package.json"),
+    path.resolve(entryDir, "../../package.json"),
+    path.resolve(entryDir, "../../../package.json")
+  ];
+  for (const p of candidates) {
+    if (fs.pathExistsSync(p)) {
+      return JSON.parse(fs.readFileSync(p).toString());
+    }
+  }
+  return {};
+}
 var environment_default = new Environment({
   cmdArgs,
   envVars,
-  package: JSON.parse(fs.readFileSync(path.join(path.resolve(), "package.json")).toString())
+  package: loadPackageJson()
 });
 
 // src/lib/configs/service-config.ts
@@ -1177,7 +1196,7 @@ process.setMaxListeners(Infinity);
 process.on("uncaughtException", (err, origin) => {
   logger_default.error(`An unhandled error occurred: ${origin}`, err);
 });
-process.on("unhandledRejection", (_17, promise) => {
+process.on("unhandledRejection", (_18, promise) => {
   promise.catch((err) => logger_default.error("An unhandled rejection occurred:", err));
 });
 process.on("warning", (warning) => logger_default.warn("System warning: ", warning));
@@ -1580,13 +1599,16 @@ var Server = class {
 var server_default = new Server();
 
 // src/api/routes/index.ts
-import fs8 from "fs-extra";
+import fs9 from "fs-extra";
+import path7 from "path";
+import mime4 from "mime";
 
 // src/api/routes/images.ts
-import fs6 from "fs";
-import _13 from "lodash";
+import fs7 from "fs";
+import _14 from "lodash";
 
 // src/api/controllers/images.ts
+import _13 from "lodash";
 import crypto2 from "crypto";
 
 // src/lib/configs/model-config.ts
@@ -2384,7 +2406,7 @@ async function generateImageComposition(_model, prompt, imageUrls, {
         submit_id: submitId,
         metrics_extra: JSON.stringify({
           promptSource: "custom",
-          generateCount: 1,
+          generateCount: finalTargetImageCount,
           enterFrom: "click",
           sceneOptions: JSON.stringify([sceneOption]),
           generateId: submitId,
@@ -2456,7 +2478,7 @@ async function generateImageComposition(_model, prompt, imageUrls, {
                     }],
                     strength: 0.5
                   })),
-                  prompt_placeholder_info_list: uploadedImageIds.map((_17, index) => ({
+                  prompt_placeholder_info_list: uploadedImageIds.map((_18, index) => ({
                     type: "",
                     id: util_default.uuid(),
                     ability_index: index
@@ -2483,6 +2505,7 @@ async function generateImageComposition(_model, prompt, imageUrls, {
   logger_default.info(`\u56FE\u751F\u56FE\u4EFB\u52A1\u5DF2\u63D0\u4EA4\uFF0Chistory_id: ${historyId}\uFF0C\u7B49\u5F85\u751F\u6210\u5B8C\u6210...`);
   let status = 20, failCode, item_list = [];
   let pollCount = 0;
+  let missingRecordCount = 0;
   const maxPollCount = 600;
   while (pollCount < maxPollCount) {
     await new Promise((resolve) => setTimeout(resolve, 1e3));
@@ -2568,8 +2591,15 @@ async function generateImageComposition(_model, prompt, imageUrls, {
         }
       }
     });
-    if (!result[historyId])
-      throw new APIException(exceptions_default.API_IMAGE_GENERATION_FAILED, "\u8BB0\u5F55\u4E0D\u5B58\u5728");
+    if (!result[historyId]) {
+      missingRecordCount++;
+      logger_default.warn(`history_id ${historyId} not found in poll result, retry ${missingRecordCount}/10`);
+      if (missingRecordCount >= 10) {
+        throw new APIException(exceptions_default.API_IMAGE_GENERATION_FAILED, "\u8BB0\u5F55\u4E0D\u5B58\u5728");
+      }
+      continue;
+    }
+    missingRecordCount = 0;
     status = result[historyId].status;
     failCode = result[historyId].fail_code;
     item_list = result[historyId].item_list || [];
@@ -2607,13 +2637,18 @@ async function generateMultiImages(_model, prompt, {
   resolution = "2k",
   sampleStrength = 0.5,
   negativePrompt = "",
-  intelligentRatio = false
+  intelligentRatio = false,
+  targetImageCount
 }, refreshToken) {
   const model = getModel(_model);
   const resolutionResult = resolveResolution(resolution, ratio);
   const { width, height, imageRatio, resolutionType } = resolutionResult;
-  const targetImageCount = prompt.match(/(\d+)张/) ? parseInt(prompt.match(/(\d+)张/)[1]) : 4;
-  logger_default.info(`\u4F7F\u7528 ${_model} \u591A\u56FE\u751F\u6210: ${targetImageCount}\u5F20\u56FE\u7247 ${width}x${height} (${ratio}@${resolution}) \u7CBE\u7EC6\u5EA6: ${sampleStrength}`);
+  const finalTargetImageCount2 = _13.clamp(
+    targetImageCount ?? (prompt.match(/(\d+)张/) ? parseInt(prompt.match(/(\d+)张/)[1]) : 4),
+    1,
+    10
+  );
+  logger_default.info(`\u4F7F\u7528 ${_model} \u591A\u56FE\u751F\u6210: ${finalTargetImageCount2}\u5F20\u56FE\u7247 ${width}x${height} (${ratio}@${resolution}) \u7CBE\u7EC6\u5EA6: ${sampleStrength}`);
   const componentId = util_default.uuid();
   const submitId = util_default.uuid();
   const sceneOption = {
@@ -2702,7 +2737,7 @@ async function generateMultiImages(_model, prompt, {
                   gen_option: {
                     type: "",
                     id: util_default.uuid(),
-                    generate_all: false
+                    generate_all: finalTargetImageCount2 > 1
                   }
                 }
               }
@@ -2718,15 +2753,16 @@ async function generateMultiImages(_model, prompt, {
   const historyId = aigc_data == null ? void 0 : aigc_data.history_record_id;
   if (!historyId)
     throw new APIException(exceptions_default.API_IMAGE_GENERATION_FAILED, "\u8BB0\u5F55ID\u4E0D\u5B58\u5728");
-  logger_default.info(`\u591A\u56FE\u751F\u6210\u4EFB\u52A1\u5DF2\u63D0\u4EA4\uFF0Csubmit_id: ${submitId}, history_id: ${historyId}\uFF0C\u7B49\u5F85\u751F\u6210 ${targetImageCount} \u5F20\u56FE\u7247...`);
+  logger_default.info(`\u591A\u56FE\u751F\u6210\u4EFB\u52A1\u5DF2\u63D0\u4EA4\uFF0Csubmit_id: ${submitId}, history_id: ${historyId}\uFF0C\u7B49\u5F85\u751F\u6210 ${finalTargetImageCount2} \u5F20\u56FE\u7247...`);
   let status = 20, failCode, item_list = [];
   let pollCount = 0;
+  let missingRecordCount = 0;
   const maxPollCount = 600;
   while (pollCount < maxPollCount) {
     await new Promise((resolve) => setTimeout(resolve, 1e3));
     pollCount++;
     if (pollCount % 30 === 0) {
-      logger_default.info(`\u591A\u56FE\u751F\u6210\u8FDB\u5EA6: \u7B2C ${pollCount} \u6B21\u8F6E\u8BE2 (history_id: ${historyId})\uFF0C\u5F53\u524D\u72B6\u6001: ${status}\uFF0C\u5DF2\u751F\u6210: ${item_list.length}/${targetImageCount} \u5F20\u56FE\u7247...`);
+      logger_default.info(`\u591A\u56FE\u751F\u6210\u8FDB\u5EA6: \u7B2C ${pollCount} \u6B21\u8F6E\u8BE2 (history_id: ${historyId})\uFF0C\u5F53\u524D\u72B6\u6001: ${status}\uFF0C\u5DF2\u751F\u6210: ${item_list.length}/${finalTargetImageCount2} \u5F20\u56FE\u7247...`);
     }
     const result = await request("post", "/mweb/v1/get_history_by_ids", refreshToken, {
       data: {
@@ -2806,20 +2842,27 @@ async function generateMultiImages(_model, prompt, {
         }
       }
     });
-    if (!result[historyId])
-      throw new APIException(exceptions_default.API_IMAGE_GENERATION_FAILED, "\u8BB0\u5F55\u4E0D\u5B58\u5728");
+    if (!result[historyId]) {
+      missingRecordCount++;
+      logger_default.warn(`history_id ${historyId} not found in poll result, retry ${missingRecordCount}/10`);
+      if (missingRecordCount >= 10) {
+        throw new APIException(exceptions_default.API_IMAGE_GENERATION_FAILED, "\u8BB0\u5F55\u4E0D\u5B58\u5728");
+      }
+      continue;
+    }
+    missingRecordCount = 0;
     status = result[historyId].status;
     failCode = result[historyId].fail_code;
     item_list = result[historyId].item_list || [];
-    if (item_list.length >= targetImageCount) {
+    if (item_list.length >= finalTargetImageCount2) {
       logger_default.info(`\u591A\u56FE\u751F\u6210\u5B8C\u6210: \u72B6\u6001=${status}, \u5DF2\u751F\u6210 ${item_list.length} \u5F20\u56FE\u7247`);
       break;
     }
     if (pollCount % 60 === 0) {
       logger_default.info(`jimeng-4.0 \u8BE6\u7EC6\u72B6\u6001: status=${status}, item_list.length=${item_list.length}, failCode=${failCode || "none"}`);
     }
-    if (status === 10 && item_list.length < targetImageCount && pollCount % 30 === 0) {
-      logger_default.info(`jimeng-4.0 \u72B6\u6001\u5DF2\u5B8C\u6210\u4F46\u56FE\u7247\u6570\u91CF\u4E0D\u8DB3: \u72B6\u6001=${status}, \u5DF2\u751F\u6210 ${item_list.length}/${targetImageCount} \u5F20\u56FE\u7247\uFF0C\u7EE7\u7EED\u7B49\u5F85...`);
+    if (status === 10 && item_list.length < finalTargetImageCount2 && pollCount % 30 === 0) {
+      logger_default.info(`jimeng-4.0 \u72B6\u6001\u5DF2\u5B8C\u6210\u4F46\u56FE\u7247\u6570\u91CF\u4E0D\u8DB3: \u72B6\u6001=${status}, \u5DF2\u751F\u6210 ${item_list.length}/${finalTargetImageCount2} \u5F20\u56FE\u7247\uFF0C\u7EE7\u7EED\u7B49\u5F85...`);
     }
   }
   if (pollCount >= maxPollCount) {
@@ -2836,7 +2879,7 @@ async function generateMultiImages(_model, prompt, {
     if (!((_c = (_b = (_a = item == null ? void 0 : item.image) == null ? void 0 : _a.large_images) == null ? void 0 : _b[0]) == null ? void 0 : _c.image_url))
       return ((_d = item == null ? void 0 : item.common_attr) == null ? void 0 : _d.cover_url) || null;
     return item.image.large_images[0].image_url;
-  }).filter((url) => url !== null);
+  }).filter((url) => url !== null).slice(0, finalTargetImageCount2);
   logger_default.info(`\u591A\u56FE\u751F\u6210\u7ED3\u679C: \u6210\u529F\u751F\u6210 ${imageUrls.length} \u5F20\u56FE\u7247`);
   return imageUrls;
 }
@@ -2845,7 +2888,8 @@ async function generateImages(_model, prompt, {
   resolution = "2k",
   sampleStrength = 0.5,
   negativePrompt = "",
-  intelligentRatio = false
+  intelligentRatio = false,
+  n = 1
 }, refreshToken) {
   const model = getModel(_model);
   const resolutionResult = resolveResolution(resolution, ratio);
@@ -2854,9 +2898,15 @@ async function generateImages(_model, prompt, {
   const { totalCredit } = await getCredit(refreshToken);
   if (totalCredit <= 0)
     await receiveCredit(refreshToken);
-  const isMultiImageRequest = /jimeng-[45]\.[0-9]/.test(_model) && (prompt.includes("\u8FDE\u7EED") || prompt.includes("\u7ED8\u672C") || prompt.includes("\u6545\u4E8B") || /\d+张/.test(prompt));
+  const finalN = _13.clamp(Number(n) || 1, 1, 10);
+  const isMultiImageRequest = /jimeng-[45]\.[0-9]/.test(_model) && (finalN > 1 || prompt.includes("\u8FDE\u7EED") || prompt.includes("\u7ED8\u672C") || prompt.includes("\u6545\u4E8B") || /\d+张/.test(prompt));
   if (isMultiImageRequest) {
-    return await generateMultiImages(_model, prompt, { ratio, resolution, sampleStrength, negativePrompt, intelligentRatio }, refreshToken);
+    return await generateMultiImages(
+      _model,
+      prompt,
+      { ratio, resolution, sampleStrength, negativePrompt, intelligentRatio, targetImageCount: finalN > 1 ? finalN : void 0 },
+      refreshToken
+    );
   }
   const componentId = util_default.uuid();
   const submitId = util_default.uuid();
@@ -2961,6 +3011,7 @@ async function generateImages(_model, prompt, {
   logger_default.info(`\u6587\u751F\u56FE\u4EFB\u52A1\u5DF2\u63D0\u4EA4\uFF0Csubmit_id: ${submitId}, history_id: ${historyId}\uFF0C\u7B49\u5F85\u751F\u6210\u5B8C\u6210...`);
   let status = 20, failCode, item_list = [];
   let pollCount = 0;
+  let missingRecordCount = 0;
   const maxPollCount = 600;
   while (pollCount < maxPollCount) {
     await new Promise((resolve) => setTimeout(resolve, 1e3));
@@ -3067,8 +3118,15 @@ async function generateImages(_model, prompt, {
         }
       }
     });
-    if (!result[historyId])
-      throw new APIException(exceptions_default.API_IMAGE_GENERATION_FAILED, "\u8BB0\u5F55\u4E0D\u5B58\u5728");
+    if (!result[historyId]) {
+      missingRecordCount++;
+      logger_default.warn(`history_id ${historyId} not found in poll result, retry ${missingRecordCount}/10`);
+      if (missingRecordCount >= 10) {
+        throw new APIException(exceptions_default.API_IMAGE_GENERATION_FAILED, "\u8BB0\u5F55\u4E0D\u5B58\u5728");
+      }
+      continue;
+    }
+    missingRecordCount = 0;
     status = result[historyId].status;
     failCode = result[historyId].fail_code;
     item_list = result[historyId].item_list || [];
@@ -3102,6 +3160,55 @@ async function generateImages(_model, prompt, {
   return imageUrls;
 }
 
+// src/lib/local-assets.ts
+import path6 from "path";
+import axios3 from "axios";
+import fs6 from "fs-extra";
+function resolveExtension(sourceUrl, contentType, fallback = "bin") {
+  if (contentType) {
+    const ext = util_default.mimeToExtension(contentType.split(";")[0].trim());
+    if (ext) return ext;
+  }
+  const urlExt = util_default.extractURLExtension(sourceUrl);
+  if (urlExt) return urlExt;
+  return fallback;
+}
+function normalizePrefix(prefix = "") {
+  if (!prefix) return "";
+  return prefix.startsWith("/") ? prefix : `/${prefix}`;
+}
+function buildPublicBaseUrl(headers = {}) {
+  const forwardedProto = headers["x-forwarded-proto"];
+  const forwardedHost = headers["x-forwarded-host"];
+  const host = forwardedHost || headers.host;
+  if (!host) return null;
+  const protocol = forwardedProto || "http";
+  return `${protocol}://${host}${normalizePrefix(config_default.service.urlPrefix)}`;
+}
+async function saveRemoteAssetToLocalUrl(remoteUrl, assetType = "images", publicBaseUrl) {
+  const response = await axios3.get(remoteUrl, {
+    responseType: "arraybuffer",
+    timeout: 12e4
+  });
+  const ext = resolveExtension(
+    remoteUrl,
+    response.headers["content-type"],
+    assetType === "videos" ? "mp4" : "png"
+  );
+  const dateDir = util_default.getDateString("yyyyMMdd");
+  const relativePath = path6.posix.join("generated", assetType, dateDir, `${util_default.uuid(false)}.${ext}`);
+  const outputPath = path6.join(config_default.system.publicDirPath, ...relativePath.split("/"));
+  await fs6.ensureDir(path6.dirname(outputPath));
+  await fs6.writeFile(outputPath, response.data);
+  const baseUrl = publicBaseUrl || `${config_default.service.publicDirUrl.replace(/\/public$/, "")}${normalizePrefix(config_default.service.urlPrefix)}`;
+  const localUrl = `${baseUrl}/public/${relativePath}`;
+  logger_default.info(`asset localized: ${remoteUrl} -> ${localUrl}`);
+  return localUrl;
+}
+async function saveRemoteAssetsToLocalUrls(remoteUrls, assetType = "images", publicBaseUrl) {
+  return Promise.all(remoteUrls.map((url) => saveRemoteAssetToLocalUrl(url, assetType, publicBaseUrl)));
+}
+
 // src/api/routes/images.ts
 var images_default = {
   prefix: "/v1/images",
@@ -3117,9 +3224,9 @@ var images_default = {
       const contentType = request2.headers["content-type"] || "";
       const isMultiPart = contentType.startsWith("multipart/form-data");
       if (isMultiPart) {
-        request2.validate("body.model", (v) => _13.isUndefined(v) || _13.isString(v)).validate("body.prompt", _13.isString).validate("body.negative_prompt", (v) => _13.isUndefined(v) || _13.isString(v)).validate("body.ratio", (v) => _13.isUndefined(v) || _13.isString(v)).validate("body.resolution", (v) => _13.isUndefined(v) || _13.isString(v)).validate("body.intelligent_ratio", (v) => _13.isUndefined(v) || typeof v === "string" && (v === "true" || v === "false") || _13.isBoolean(v)).validate("body.sample_strength", (v) => _13.isUndefined(v) || typeof v === "string" && !isNaN(parseFloat(v)) || _13.isFinite(v)).validate("body.response_format", (v) => _13.isUndefined(v) || _13.isString(v)).validate("headers.authorization", _13.isString);
+        request2.validate("body.model", (v) => _14.isUndefined(v) || _14.isString(v)).validate("body.prompt", _14.isString).validate("body.negative_prompt", (v) => _14.isUndefined(v) || _14.isString(v)).validate("body.ratio", (v) => _14.isUndefined(v) || _14.isString(v)).validate("body.resolution", (v) => _14.isUndefined(v) || _14.isString(v)).validate("body.intelligent_ratio", (v) => _14.isUndefined(v) || typeof v === "string" && (v === "true" || v === "false") || _14.isBoolean(v)).validate("body.sample_strength", (v) => _14.isUndefined(v) || typeof v === "string" && !isNaN(parseFloat(v)) || _14.isFinite(v)).validate("body.n", (v) => _14.isUndefined(v) || typeof v === "string" && /^\d+$/.test(v) || _14.isFinite(v) && v >= 1 && v <= 10).validate("body.response_format", (v) => _14.isUndefined(v) || _14.isString(v)).validate("headers.authorization", _14.isString);
       } else {
-        request2.validate("body.model", (v) => _13.isUndefined(v) || _13.isString(v)).validate("body.prompt", _13.isString).validate("body.images", (v) => _13.isUndefined(v) || _13.isArray(v)).validate("body.negative_prompt", (v) => _13.isUndefined(v) || _13.isString(v)).validate("body.ratio", (v) => _13.isUndefined(v) || _13.isString(v)).validate("body.resolution", (v) => _13.isUndefined(v) || _13.isString(v)).validate("body.intelligent_ratio", (v) => _13.isUndefined(v) || _13.isBoolean(v)).validate("body.sample_strength", (v) => _13.isUndefined(v) || _13.isFinite(v)).validate("body.response_format", (v) => _13.isUndefined(v) || _13.isString(v)).validate("headers.authorization", _13.isString);
+        request2.validate("body.model", (v) => _14.isUndefined(v) || _14.isString(v)).validate("body.prompt", _14.isString).validate("body.images", (v) => _14.isUndefined(v) || _14.isArray(v)).validate("body.negative_prompt", (v) => _14.isUndefined(v) || _14.isString(v)).validate("body.ratio", (v) => _14.isUndefined(v) || _14.isString(v)).validate("body.resolution", (v) => _14.isUndefined(v) || _14.isString(v)).validate("body.intelligent_ratio", (v) => _14.isUndefined(v) || _14.isBoolean(v)).validate("body.sample_strength", (v) => _14.isUndefined(v) || _14.isFinite(v)).validate("body.n", (v) => _14.isUndefined(v) || _14.isFinite(v) && v >= 1 && v <= 10).validate("body.response_format", (v) => _14.isUndefined(v) || _14.isString(v)).validate("headers.authorization", _14.isString);
       }
       let images = null;
       if (isMultiPart) {
@@ -3130,7 +3237,7 @@ var images_default = {
             if (imageFiles.length > 10) {
               throw new Error("\u6700\u591A\u652F\u630110\u5F20\u8F93\u5165\u56FE\u7247");
             }
-            images = imageFiles.map((file) => fs6.readFileSync(file.filepath));
+            images = imageFiles.map((file) => fs7.readFileSync(file.filepath));
           }
         }
       } else {
@@ -3140,18 +3247,18 @@ var images_default = {
             throw new Error("\u6700\u591A\u652F\u630110\u5F20\u8F93\u5165\u56FE\u7247");
           }
           bodyImages.forEach((image, index) => {
-            if (!_13.isString(image) && !_13.isObject(image)) {
+            if (!_14.isString(image) && !_14.isObject(image)) {
               throw new Error(`\u56FE\u7247 ${index + 1} \u683C\u5F0F\u4E0D\u6B63\u786E\uFF1A\u5E94\u4E3AURL\u5B57\u7B26\u4E32\u6216\u5305\u542Burl\u5B57\u6BB5\u7684\u5BF9\u8C61`);
             }
-            if (_13.isObject(image) && !image.url) {
+            if (_14.isObject(image) && !image.url) {
               throw new Error(`\u56FE\u7247 ${index + 1} \u7F3A\u5C11url\u5B57\u6BB5`);
             }
           });
-          images = bodyImages.map((image) => _13.isString(image) ? image : image.url);
+          images = bodyImages.map((image) => _14.isString(image) ? image : image.url);
         }
       }
       const tokens = tokenSplit(request2.headers.authorization);
-      const token = _13.sample(tokens);
+      const token = _14.sample(tokens);
       const {
         model,
         prompt,
@@ -3160,11 +3267,14 @@ var images_default = {
         resolution,
         intelligent_ratio: intelligentRatio,
         sample_strength: sampleStrength,
+        n,
         response_format
       } = request2.body;
       const finalSampleStrength = isMultiPart && typeof sampleStrength === "string" ? parseFloat(sampleStrength) : sampleStrength;
       const finalIntelligentRatio = isMultiPart && typeof intelligentRatio === "string" ? intelligentRatio === "true" : intelligentRatio;
-      const responseFormat = _13.defaultTo(response_format, "url");
+      const finalN = isMultiPart && typeof n === "string" ? parseInt(n) : n;
+      const responseFormat = _14.defaultTo(response_format, "url");
+      const publicBaseUrl = buildPublicBaseUrl(request2.headers);
       let imageUrls;
       let resultData = {
         created: util_default.unixTimestamp()
@@ -3185,14 +3295,16 @@ var images_default = {
           resolution,
           sampleStrength: finalSampleStrength,
           negativePrompt,
-          intelligentRatio: finalIntelligentRatio
+          intelligentRatio: finalIntelligentRatio,
+          n: finalN
         }, token);
       }
       let data = [];
       if (responseFormat == "b64_json") {
         data = (await Promise.all(imageUrls.map((url) => util_default.fetchFileBASE64(url)))).map((b64) => ({ b64_json: b64 }));
       } else {
-        data = imageUrls.map((url) => ({
+        const localUrls = await saveRemoteAssetsToLocalUrls(imageUrls, "images", publicBaseUrl);
+        data = localUrls.map((url) => ({
           url
         }));
       }
@@ -3211,9 +3323,9 @@ var images_default = {
       const contentType = request2.headers["content-type"] || "";
       const isMultiPart = contentType.startsWith("multipart/form-data");
       if (isMultiPart) {
-        request2.validate("body.model", (v) => _13.isUndefined(v) || _13.isString(v)).validate("body.prompt", _13.isString).validate("body.negative_prompt", (v) => _13.isUndefined(v) || _13.isString(v)).validate("body.ratio", (v) => _13.isUndefined(v) || _13.isString(v)).validate("body.resolution", (v) => _13.isUndefined(v) || _13.isString(v)).validate("body.intelligent_ratio", (v) => _13.isUndefined(v) || typeof v === "string" && (v === "true" || v === "false") || _13.isBoolean(v)).validate("body.sample_strength", (v) => _13.isUndefined(v) || typeof v === "string" && !isNaN(parseFloat(v)) || _13.isFinite(v)).validate("body.response_format", (v) => _13.isUndefined(v) || _13.isString(v)).validate("headers.authorization", _13.isString);
+        request2.validate("body.model", (v) => _14.isUndefined(v) || _14.isString(v)).validate("body.prompt", _14.isString).validate("body.negative_prompt", (v) => _14.isUndefined(v) || _14.isString(v)).validate("body.ratio", (v) => _14.isUndefined(v) || _14.isString(v)).validate("body.resolution", (v) => _14.isUndefined(v) || _14.isString(v)).validate("body.intelligent_ratio", (v) => _14.isUndefined(v) || typeof v === "string" && (v === "true" || v === "false") || _14.isBoolean(v)).validate("body.sample_strength", (v) => _14.isUndefined(v) || typeof v === "string" && !isNaN(parseFloat(v)) || _14.isFinite(v)).validate("body.response_format", (v) => _14.isUndefined(v) || _14.isString(v)).validate("headers.authorization", _14.isString);
       } else {
-        request2.validate("body.model", (v) => _13.isUndefined(v) || _13.isString(v)).validate("body.prompt", _13.isString).validate("body.images", _13.isArray).validate("body.negative_prompt", (v) => _13.isUndefined(v) || _13.isString(v)).validate("body.ratio", (v) => _13.isUndefined(v) || _13.isString(v)).validate("body.resolution", (v) => _13.isUndefined(v) || _13.isString(v)).validate("body.intelligent_ratio", (v) => _13.isUndefined(v) || _13.isBoolean(v)).validate("body.sample_strength", (v) => _13.isUndefined(v) || _13.isFinite(v)).validate("body.response_format", (v) => _13.isUndefined(v) || _13.isString(v)).validate("headers.authorization", _13.isString);
+        request2.validate("body.model", (v) => _14.isUndefined(v) || _14.isString(v)).validate("body.prompt", _14.isString).validate("body.images", _14.isArray).validate("body.negative_prompt", (v) => _14.isUndefined(v) || _14.isString(v)).validate("body.ratio", (v) => _14.isUndefined(v) || _14.isString(v)).validate("body.resolution", (v) => _14.isUndefined(v) || _14.isString(v)).validate("body.intelligent_ratio", (v) => _14.isUndefined(v) || _14.isBoolean(v)).validate("body.sample_strength", (v) => _14.isUndefined(v) || _14.isFinite(v)).validate("body.response_format", (v) => _14.isUndefined(v) || _14.isString(v)).validate("headers.authorization", _14.isString);
       }
       let images = [];
       if (isMultiPart) {
@@ -3228,7 +3340,7 @@ var images_default = {
         if (imageFiles.length > 10) {
           throw new Error("\u6700\u591A\u652F\u630110\u5F20\u8F93\u5165\u56FE\u7247");
         }
-        images = imageFiles.map((file) => fs6.readFileSync(file.filepath));
+        images = imageFiles.map((file) => fs7.readFileSync(file.filepath));
       } else {
         const bodyImages = request2.body.images;
         if (!bodyImages || bodyImages.length === 0) {
@@ -3238,17 +3350,17 @@ var images_default = {
           throw new Error("\u6700\u591A\u652F\u630110\u5F20\u8F93\u5165\u56FE\u7247");
         }
         bodyImages.forEach((image, index) => {
-          if (!_13.isString(image) && !_13.isObject(image)) {
+          if (!_14.isString(image) && !_14.isObject(image)) {
             throw new Error(`\u56FE\u7247 ${index + 1} \u683C\u5F0F\u4E0D\u6B63\u786E\uFF1A\u5E94\u4E3AURL\u5B57\u7B26\u4E32\u6216\u5305\u542Burl\u5B57\u6BB5\u7684\u5BF9\u8C61`);
           }
-          if (_13.isObject(image) && !image.url) {
+          if (_14.isObject(image) && !image.url) {
             throw new Error(`\u56FE\u7247 ${index + 1} \u7F3A\u5C11url\u5B57\u6BB5`);
           }
         });
-        images = bodyImages.map((image) => _13.isString(image) ? image : image.url);
+        images = bodyImages.map((image) => _14.isString(image) ? image : image.url);
       }
       const tokens = tokenSplit(request2.headers.authorization);
-      const token = _13.sample(tokens);
+      const token = _14.sample(tokens);
       const {
         model,
         prompt,
@@ -3261,7 +3373,8 @@ var images_default = {
       } = request2.body;
       const finalSampleStrength = isMultiPart && typeof sampleStrength === "string" ? parseFloat(sampleStrength) : sampleStrength;
       const finalIntelligentRatio = isMultiPart && typeof intelligentRatio === "string" ? intelligentRatio === "true" : intelligentRatio;
-      const responseFormat = _13.defaultTo(response_format, "url");
+      const responseFormat = _14.defaultTo(response_format, "url");
+      const publicBaseUrl = buildPublicBaseUrl(request2.headers);
       const resultUrls = await generateImageComposition(model, prompt, images, {
         ratio,
         resolution,
@@ -3273,7 +3386,8 @@ var images_default = {
       if (responseFormat == "b64_json") {
         data = (await Promise.all(resultUrls.map((url) => util_default.fetchFileBASE64(url)))).map((b64) => ({ b64_json: b64 }));
       } else {
-        data = resultUrls.map((url) => ({
+        const localUrls = await saveRemoteAssetsToLocalUrls(resultUrls, "images", publicBaseUrl);
+        data = localUrls.map((url) => ({
           url
         }));
       }
@@ -3288,14 +3402,14 @@ var images_default = {
 };
 
 // src/api/routes/chat.ts
-import _14 from "lodash";
+import _15 from "lodash";
 
 // src/api/controllers/chat.ts
 import { PassThrough } from "stream";
 
 // src/api/controllers/videos.ts
 import crypto3 from "crypto";
-import fs7 from "fs";
+import fs8 from "fs";
 var DEFAULT_ASSISTANT_ID3 = 513695;
 var DEFAULT_MODEL2 = "jimeng-video-3.0";
 var DEFAULT_DRAFT_VERSION = "3.2.8";
@@ -3371,8 +3485,8 @@ var MATERIAL_TYPE_CODE = {
   audio: 3
 };
 function detectMaterialType(file) {
-  const mime4 = (file.mimetype || file.mimeType || "").toLowerCase();
-  if (mime4 && MIME_TO_MATERIAL_TYPE[mime4]) return MIME_TO_MATERIAL_TYPE[mime4];
+  const mime5 = (file.mimetype || file.mimeType || "").toLowerCase();
+  if (mime5 && MIME_TO_MATERIAL_TYPE[mime5]) return MIME_TO_MATERIAL_TYPE[mime5];
   const filename = (file.originalFilename || file.newFilename || "").toLowerCase();
   const dotIdx = filename.lastIndexOf(".");
   if (dotIdx >= 0) {
@@ -4058,7 +4172,7 @@ async function generateVideo(_model, prompt, {
       }
       try {
         logger_default.info(`\u5F00\u59CB\u4E0A\u4F20\u7B2C ${i + 1} \u4E2A\u6587\u4EF6: ${file.originalFilename || file.filepath}`);
-        const buffer = fs7.readFileSync(file.filepath);
+        const buffer = fs8.readFileSync(file.filepath);
         const imageUri = await uploadImageBufferForVideo(buffer, refreshToken);
         if (imageUri) {
           uploadIDs.push(imageUri);
@@ -4428,7 +4542,7 @@ async function generateSeedanceVideo(_model, prompt, {
       const materialType = detectMaterialType(file);
       try {
         logger_default.info(`Seedance: \u5F00\u59CB\u4E0A\u4F20\u7B2C ${i + 1} \u4E2A\u6587\u4EF6 (${materialType}): ${file.originalFilename || file.filepath}`);
-        const buffer = fs7.readFileSync(file.filepath);
+        const buffer = fs8.readFileSync(file.filepath);
         if (materialType === "image") {
           const imageUri = await uploadImageBufferForVideo(buffer, refreshToken);
           if (imageUri) {
@@ -4817,7 +4931,7 @@ var MAX_RETRY_COUNT = 3;
 var RETRY_DELAY = 5e3;
 function parseModel(model) {
   const [_model, size] = model.split(":");
-  const [_17, width, height] = /(\d+)[\W\w](\d+)/.exec(size) ?? [];
+  const [_18, width, height] = /(\d+)[\W\w](\d+)/.exec(size) ?? [];
   return {
     model: _model,
     width: size ? Math.ceil(parseInt(width) / 2) * 2 : 1024,
@@ -4827,7 +4941,7 @@ function parseModel(model) {
 function isVideoModel(model) {
   return model.startsWith("jimeng-video") || model.startsWith("seedance-");
 }
-async function createCompletion(messages, refreshToken, _model = DEFAULT_MODEL, retryCount = 0) {
+async function createCompletion(messages, refreshToken, _model = DEFAULT_MODEL, retryCount = 0, publicBaseUrl) {
   return (async () => {
     if (messages.length === 0)
       throw new APIException(exceptions_default.API_REQUEST_PARAMS_INVALID, "\u6D88\u606F\u4E0D\u80FD\u4E3A\u7A7A");
@@ -4886,7 +5000,8 @@ curl -X POST http://localhost:3000/v1/videos/generations \\
           },
           refreshToken
         );
-        logger_default.info(`\u89C6\u9891\u751F\u6210\u6210\u529F\uFF0CURL: ${videoUrl}`);
+        const localVideoUrl = await saveRemoteAssetToLocalUrl(videoUrl, "videos", publicBaseUrl);
+        logger_default.info(`\u89C6\u9891\u751F\u6210\u6210\u529F\uFF0CURL: ${localVideoUrl}`);
         return {
           id: util_default.uuid(),
           model: _model,
@@ -4896,7 +5011,7 @@ curl -X POST http://localhost:3000/v1/videos/generations \\
               index: 0,
               message: {
                 role: "assistant",
-                content: `![video](${videoUrl})
+                content: `![video](${localVideoUrl})
 `
               },
               finish_reason: "stop"
@@ -4940,6 +5055,7 @@ curl -X POST http://localhost:3000/v1/videos/generations \\
         },
         refreshToken
       );
+      const localImageUrls = await saveRemoteAssetsToLocalUrls(imageUrls, "images", publicBaseUrl);
       return {
         id: util_default.uuid(),
         model: _model || model,
@@ -4949,7 +5065,7 @@ curl -X POST http://localhost:3000/v1/videos/generations \\
             index: 0,
             message: {
               role: "assistant",
-              content: imageUrls.reduce(
+              content: localImageUrls.reduce(
                 (acc, url, i) => acc + `![image_${i}](${url})
 `,
                 ""
@@ -4968,13 +5084,13 @@ curl -X POST http://localhost:3000/v1/videos/generations \\
       logger_default.warn(`Try again after ${RETRY_DELAY / 1e3}s...`);
       return (async () => {
         await new Promise((resolve) => setTimeout(resolve, RETRY_DELAY));
-        return createCompletion(messages, refreshToken, _model, retryCount + 1);
+        return createCompletion(messages, refreshToken, _model, retryCount + 1, publicBaseUrl);
       })();
     }
     throw err;
   });
 }
-async function createCompletionStream(messages, refreshToken, _model = DEFAULT_MODEL, retryCount = 0) {
+async function createCompletionStream(messages, refreshToken, _model = DEFAULT_MODEL, retryCount = 0, publicBaseUrl) {
   return (async () => {
     const { model, width, height } = parseModel(_model);
     logger_default.info(messages);
@@ -5060,10 +5176,11 @@ async function createCompletionStream(messages, refreshToken, _model = DEFAULT_M
         messages[messages.length - 1].content,
         { ratio: "16:9", resolution: "720p" },
         refreshToken
-      ).then((videoUrl) => {
+      ).then(async (videoUrl) => {
         clearInterval(progressInterval);
         clearTimeout(timeoutId);
-        logger_default.info(`\u89C6\u9891\u751F\u6210\u6210\u529F\uFF0CURL: ${videoUrl}`);
+        const localVideoUrl = await saveRemoteAssetToLocalUrl(videoUrl, "videos", publicBaseUrl);
+        logger_default.info(`\u89C6\u9891\u751F\u6210\u6210\u529F\uFF0CURL: ${localVideoUrl}`);
         stream.write(
           "data: " + JSON.stringify({
             id: util_default.uuid(),
@@ -5078,11 +5195,11 @@ async function createCompletionStream(messages, refreshToken, _model = DEFAULT_M
 
 \u2705 \u89C6\u9891\u751F\u6210\u5B8C\u6210\uFF01
 
-![video](${videoUrl})
+![video](${localVideoUrl})
 
 \u60A8\u53EF\u4EE5\uFF1A
 1. \u76F4\u63A5\u67E5\u770B\u4E0A\u65B9\u89C6\u9891
-2. \u4F7F\u7528\u4EE5\u4E0B\u94FE\u63A5\u4E0B\u8F7D\u6216\u5206\u4EAB\uFF1A${videoUrl}`
+2. \u4F7F\u7528\u4EE5\u4E0B\u94FE\u63A5\u4E0B\u8F7D\u6216\u5206\u4EAB\uFF1A${localVideoUrl}`
                 },
                 finish_reason: null
               }
@@ -5168,9 +5285,10 @@ ${errorMessage}`
         messages[messages.length - 1].content,
         { width, height },
         refreshToken
-      ).then((imageUrls) => {
-        for (let i = 0; i < imageUrls.length; i++) {
-          const url = imageUrls[i];
+      ).then(async (imageUrls) => {
+        const localImageUrls = await saveRemoteAssetsToLocalUrls(imageUrls, "images", publicBaseUrl);
+        for (let i = 0; i < localImageUrls.length; i++) {
+          const url = localImageUrls[i];
           stream.write(
             "data: " + JSON.stringify({
               id: util_default.uuid(),
@@ -5184,7 +5302,7 @@ ${errorMessage}`
                     content: `![image_${i}](${url})
 `
                   },
-                  finish_reason: i < imageUrls.length - 1 ? null : "stop"
+                  finish_reason: i < localImageUrls.length - 1 ? null : "stop"
                 }
               ]
             }) + "\n\n"
@@ -5197,7 +5315,7 @@ ${errorMessage}`
             object: "chat.completion.chunk",
             choices: [
               {
-                index: imageUrls.length + 1,
+                index: localImageUrls.length + 1,
                 delta: {
                   role: "assistant",
                   content: "\u56FE\u50CF\u751F\u6210\u5B8C\u6210\uFF01"
@@ -5240,7 +5358,8 @@ ${errorMessage}`
           messages,
           refreshToken,
           _model,
-          retryCount + 1
+          retryCount + 1,
+          publicBaseUrl
         );
       })();
     }
@@ -5253,17 +5372,18 @@ var chat_default = {
   prefix: "/v1/chat",
   post: {
     "/completions": async (request2) => {
-      request2.validate("body.model", (v) => _14.isUndefined(v) || _14.isString(v)).validate("body.messages", _14.isArray).validate("headers.authorization", _14.isString);
+      request2.validate("body.model", (v) => _15.isUndefined(v) || _15.isString(v)).validate("body.messages", _15.isArray).validate("headers.authorization", _15.isString);
       const tokens = tokenSplit(request2.headers.authorization);
-      const token = _14.sample(tokens);
+      const token = _15.sample(tokens);
       const { model, messages, stream } = request2.body;
+      const publicBaseUrl = buildPublicBaseUrl(request2.headers);
       if (stream) {
-        const stream2 = await createCompletionStream(messages, token, model);
+        const stream2 = await createCompletionStream(messages, token, model, 0, publicBaseUrl);
         return new Response(stream2, {
           type: "text/event-stream"
         });
       } else
-        return await createCompletion(messages, token, model);
+        return await createCompletion(messages, token, model, 0, publicBaseUrl);
     }
   }
 };
@@ -5277,19 +5397,19 @@ var ping_default = {
 };
 
 // src/api/routes/token.ts
-import _15 from "lodash";
+import _16 from "lodash";
 var token_default = {
   prefix: "/token",
   post: {
     "/check": async (request2) => {
-      request2.validate("body.token", _15.isString);
+      request2.validate("body.token", _16.isString);
       const live = await getTokenLiveStatus(request2.body.token);
       return {
         live
       };
     },
     "/points": async (request2) => {
-      request2.validate("headers.authorization", _15.isString);
+      request2.validate("headers.authorization", _16.isString);
       const tokens = tokenSplit(request2.headers.authorization);
       const points = await Promise.all(tokens.map(async (token) => {
         return {
@@ -5453,7 +5573,7 @@ var models_default = {
 };
 
 // src/api/routes/videos.ts
-import _16 from "lodash";
+import _17 from "lodash";
 var videos_default = {
   prefix: "/v1/videos",
   post: {
@@ -5466,16 +5586,16 @@ var videos_default = {
       }
       const contentType = request2.headers["content-type"] || "";
       const isMultiPart = contentType.startsWith("multipart/form-data");
-      request2.validate("body.model", (v) => _16.isUndefined(v) || _16.isString(v)).validate("body.prompt", (v) => _16.isUndefined(v) || _16.isString(v)).validate("body.ratio", (v) => _16.isUndefined(v) || _16.isString(v)).validate("body.resolution", (v) => _16.isUndefined(v) || _16.isString(v)).validate("body.duration", (v) => {
-        if (_16.isUndefined(v)) return true;
+      request2.validate("body.model", (v) => _17.isUndefined(v) || _17.isString(v)).validate("body.prompt", (v) => _17.isUndefined(v) || _17.isString(v)).validate("body.ratio", (v) => _17.isUndefined(v) || _17.isString(v)).validate("body.resolution", (v) => _17.isUndefined(v) || _17.isString(v)).validate("body.duration", (v) => {
+        if (_17.isUndefined(v)) return true;
         if (isMultiPart && typeof v === "string") {
           const num = parseInt(v);
           return num >= 4 && num <= 15 || num === 5 || num === 10;
         }
-        return _16.isFinite(v) && (v >= 4 && v <= 15 || v === 5 || v === 10);
-      }).validate("body.file_paths", (v) => _16.isUndefined(v) || _16.isArray(v)).validate("body.filePaths", (v) => _16.isUndefined(v) || _16.isArray(v)).validate("body.response_format", (v) => _16.isUndefined(v) || _16.isString(v)).validate("headers.authorization", _16.isString);
+        return _17.isFinite(v) && (v >= 4 && v <= 15 || v === 5 || v === 10);
+      }).validate("body.file_paths", (v) => _17.isUndefined(v) || _17.isArray(v)).validate("body.filePaths", (v) => _17.isUndefined(v) || _17.isArray(v)).validate("body.response_format", (v) => _17.isUndefined(v) || _17.isString(v)).validate("headers.authorization", _17.isString);
       const tokens = tokenSplit(request2.headers.authorization);
-      const token = _16.sample(tokens);
+      const token = _17.sample(tokens);
       const {
         model = DEFAULT_MODEL2,
         prompt,
@@ -5528,10 +5648,12 @@ var videos_default = {
           }]
         };
       } else {
+        const publicBaseUrl = buildPublicBaseUrl(request2.headers);
+        const localVideoUrl = await saveRemoteAssetToLocalUrl(videoUrl, "videos", publicBaseUrl);
         return {
           created: util_default.unixTimestamp(),
           data: [{
-            url: videoUrl,
+            url: localVideoUrl,
             revised_prompt: prompt
           }]
         };
@@ -5551,13 +5673,32 @@ var routes_default = [
   {
     get: {
       "/": async () => {
-        const content = await fs8.readFile("public/welcome.html");
+        const content = await fs9.readFile("public/welcome.html");
         return new Response(content, {
           type: "html",
           headers: {
             Expires: "-1"
           }
         });
+      },
+      "/public/(.*)": async (request2) => {
+        const rawPath = decodeURIComponent(request2.params[0] || "").replace(/^[/\\]+/, "");
+        if (!rawPath) throw new Error("\u6587\u4EF6\u8DEF\u5F84\u4E0D\u80FD\u4E3A\u7A7A");
+        const root = path7.resolve(config_default.system.publicDirPath);
+        const filePath = path7.resolve(root, rawPath);
+        const relative = path7.relative(root, filePath);
+        if (relative.startsWith("..") || path7.isAbsolute(relative)) {
+          throw new Error("\u975E\u6CD5\u6587\u4EF6\u8DEF\u5F84");
+        }
+        if (!await fs9.pathExists(filePath)) {
+          throw new Error("\u6587\u4EF6\u4E0D\u5B58\u5728");
+        }
+        return new Response(fs9.createReadStream(filePath), {
+          type: mime4.getType(filePath) || "application/octet-stream",
+          headers: {
+            "Cache-Control": "public, max-age=31536000, immutable"
+          }
+        });
       }
     }
   },
@@ -5585,4 +5726,4 @@ export {
   generateSeedanceVideo,
   routes_default
 };
-//# sourceMappingURL=chunk-KHOQ4PCS.js.map
+//# sourceMappingURL=chunk-YJIZUPUG.js.map