Refactor/index async to then

src/AI/AI.ts

@@ -1,12 +1,12 @@
-import { ImageManager } from "./image";
+import { Image, ImageManager } from "./image";
 import { ConfigManager } from "../config/config";
-import { replyToSender } from "../utils/utils";
-import { endStream, pollStream, sendChatRequest, startStream } from "./service";
+import { replyToSender, transformMsgId } from "../utils/utils";
+import { endStream, pollStream, sendChatRequest, startStream } from "../service";
 import { Context } from "./context";
 import { Memory } from "./memory";
 import { handleMessages, parseBody } from "../utils/utils_message";
 import { ToolManager } from "../tool/tool";
-import { logger } from "./logger";
+import { logger } from "../logger";
 import { checkRepeat, handleReply } from "../utils/utils_string";
 import { checkContextUpdate } from "../utils/utils_update";
 
@@ -85,7 +85,24 @@ export class AI {
         this.tool.toolCallCount = 0;
     }
 
-    async chat(ctx: seal.MsgContext, msg: seal.Message): Promise<void> {
+    async handleReceipt(ctx: seal.MsgContext, msg: seal.Message, ai: AI, message: string, CQTypes: string[]) {
+        // 图片偷取，以及图片转文字
+        let images: Image[] = [];
+        if (CQTypes.includes('image')) {
+            const result = await ImageManager.handleImageMessage(ctx, message);
+            message = result.message;
+            images = result.images;
+            if (ai.imageManager.stealStatus) {
+                ai.imageManager.updateStolenImages(images);
+            }
+        }
+
+        await ai.context.addMessage(ctx, msg, ai, message, images, 'user', transformMsgId(msg.rawId));
+    }
+
+    async chat(ctx: seal.MsgContext, msg: seal.Message, reason: string = ''): Promise<void> {
+        logger.info('触发回复:', reason || '未知原因');
+
         const { bucketLimit, fillInterval } = ConfigManager.received;
         // 补充并检查触发次数
         if (Date.now() - this.bucket.lastTime > fillInterval * 1000) {
@@ -113,6 +130,7 @@ export class AI {
         }
         if (stream) {
             await this.chatStream(ctx, msg);
+            AIManager.saveAI(this.id);
             return;
         }
 
@@ -168,6 +186,7 @@ export class AI {
         }
 
         clearTimeout(timeout);
+        AIManager.saveAI(this.id);
     }
 
     async chatStream(ctx: seal.MsgContext, msg: seal.Message): Promise<void> {

src/AI/context.ts

@@ -4,7 +4,7 @@ import { Image, ImageManager } from "./image";
 import { createCtx, createMsg } from "../utils/utils_seal";
 import { levenshteinDistance } from "../utils/utils_string";
 import { AI, AIManager } from "./AI";
-import { logger } from "./logger";
+import { logger } from "../logger";
 import { transformMsgId } from "../utils/utils";
 
 export interface Message {
@@ -194,7 +194,7 @@ export class Context {
         this.messages.push(message);
     }
 
-    async limitMessages(maxRounds: number) {
+    limitMessages(maxRounds: number) {
         const messages = this.messages;
         let round = 0;
         for (let i = messages.length - 1; i >= 0; i--) {

src/AI/image.ts

@@ -1,7 +1,7 @@
 import { ConfigManager } from "../config/config";
-import { sendITTRequest } from "./service";
+import { sendITTRequest } from "../service";
 import { generateId } from "../utils/utils";
-import { logger } from "./logger";
+import { logger } from "../logger";
 
 export class Image {
     id: string;

src/AI/memory.ts

@@ -3,8 +3,8 @@ import { ConfigManager } from "../config/config";
 import { AI, AIManager } from "./AI";
 import { Context, Message } from "./context";
 import { generateId } from "../utils/utils";
-import { logger } from "./logger";
-import { fetchData } from "./service";
+import { logger } from "../logger";
+import { fetchData } from "../service";
 import { parseBody } from "../utils/utils_message";
 import { ToolManager } from "../tool/tool";