koishi-plugin-aka-ai-generator 0.6.0 → 0.6.1

package/lib/index.js

@@ -1580,15 +1580,22 @@ ${infoParts.join("\n")}`;
       if (images.length === 0) {
         return "图像处理失败：未能生成图片";
       }
+      await recordUserUsage(session, styleName, images.length);
       await session.send("图像处理完成！");
       for (let i = 0; i < images.length; i++) {
         if (checkTimeout && checkTimeout()) break;
-        await session.send(import_koishi2.h.image(images[i]));
+        try {
+          await Promise.race([
+            session.send(import_koishi2.h.image(images[i])),
+            new Promise((_, reject) => setTimeout(() => reject(new Error("SendTimeout")), 2e4))
+          ]);
+        } catch (err) {
+          logger.warn(`图片发送可能超时 (用户: ${userId}): ${err instanceof Error ? err.message : String(err)}`);
+        }
         if (images.length > 1 && i < images.length - 1) {
           await new Promise((resolve) => setTimeout(resolve, 1e3));
         }
       }
-      await recordUserUsage(session, styleName, images.length);
     } finally {
       userManager.endTask(userId);
     }
@@ -1742,15 +1749,22 @@ Prompt: ${prompt}`);
           if (resultImages.length === 0) {
             return "图片合成失败：未能生成图片";
           }
+          await recordUserUsage(session, COMMANDS.COMPOSE_IMAGE, resultImages.length);
           await session.send("图片合成完成！");
           for (let i = 0; i < resultImages.length; i++) {
             if (isTimeout) break;
-            await session.send(import_koishi2.h.image(resultImages[i]));
+            try {
+              await Promise.race([
+                session.send(import_koishi2.h.image(resultImages[i])),
+                new Promise((_, reject) => setTimeout(() => reject(new Error("SendTimeout")), 2e4))
+              ]);
+            } catch (err) {
+              logger.warn(`图片合成发送可能超时 (用户: ${userId}): ${err instanceof Error ? err.message : String(err)}`);
+            }
             if (resultImages.length > 1 && i < resultImages.length - 1) {
               await new Promise((resolve) => setTimeout(resolve, 1e3));
             }
           }
-          await recordUserUsage(session, COMMANDS.COMPOSE_IMAGE, resultImages.length);
         } finally {
           userManager.endTask(userId);
         }

package/package.json

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