koishi-plugin-aka-ai-generator 0.8.6 → 0.8.7

package/lib/index.js

@@ -69,14 +69,35 @@ function sanitizeString(str) {
 __name(sanitizeString, "sanitizeString");
 async function downloadImageAsBase64(ctx, url, timeout, logger, maxSize = 10 * 1024 * 1024) {
   try {
-    const response = await ctx.http.get(url, {
-      responseType: "arraybuffer",
-      timeout: timeout * 1e3,
-      headers: {
-        "Accept": "image/*"
+    let buffer;
+    if (url.startsWith("internal:")) {
+      logger.debug("检测到 Koishi 内部协议 URL，使用 ctx.http.file() 获取", { url: url.substring(0, 80) });
+      try {
+        const fileResult = await ctx.http.file(url, { timeout: timeout * 1e3 });
+        buffer = Buffer.from(fileResult.data);
+        logger.info("Koishi 内部协议资源获取成功", {
+          url: url.substring(0, 80),
+          size: buffer.length,
+          type: fileResult.type
+        });
+      } catch (internalError) {
+        logger.error("Koishi 内部协议资源获取失败", {
+          url: url.substring(0, 80),
+          error: sanitizeError(internalError),
+          message: internalError?.message
+        });
+        throw new Error(`无法获取飞书/Lark 图片资源: ${internalError?.message || "未知错误"}`);
       }
-    });
-    const buffer = Buffer.from(response);
+    } else {
+      const response = await ctx.http.get(url, {
+        responseType: "arraybuffer",
+        timeout: timeout * 1e3,
+        headers: {
+          "Accept": "image/*"
+        }
+      });
+      buffer = Buffer.from(response);
+    }
     if (buffer.length > maxSize) {
       throw new Error(`图片大小超过限制 (${(maxSize / 1024 / 1024).toFixed(1)}MB)`);
     }

package/lib/providers/utils.d.ts

@@ -10,6 +10,7 @@ export declare function sanitizeString(str: string): string;
 /**
  * 下载图片并转换为 Base64
  * 包含 MIME 类型检测和大小限制
+ * 支持 Koishi 内部协议 URL (如 internal:lark/... internal:onebot/... 等)
  */
 export declare function downloadImageAsBase64(ctx: Context, url: string, timeout: number, logger: any, maxSize?: number): Promise<{
     data: string;

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-aka-ai-generator",
   "description": "自用AI生成插件(GPTGod & Yunwu)",
-  "version": "0.8.6",
+  "version": "0.8.7",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
   "files": [