mioku-plugin-chat 2.10.1 → 3.0.1

package/configs/base.ts

@@ -1,5 +1,5 @@
 export const BASE_CONFIG = {
-  maxContextTokens: 128,
+  maxContextTokens: 256,
   temperature: 0.8,
   historyCount: 100,
   maxIterations: 20,

package/context.ts

@@ -1,4 +1,4 @@
-import type { MiokiContext } from "mioki";
+import type { MiokuContext } from "mioku";
 import type { AIInstance, AIModelRole, AIService } from "mioku";
 import type { ChatConfig } from "./types";
 import type { ChatDatabase } from "./db";
@@ -35,7 +35,7 @@ import type { AudioServiceApi } from "mioku-service-audio";
 import type { ChatMessage } from "./types";
 
 export interface ChatPluginContext {
-  ctx: MiokiContext;
+  ctx: MiokuContext;
   defaultConfig: ChatConfig;
   configProvider: ChatConfigProvider;
   getConfig: (groupId?: number) => Promise<ChatConfig>;

package/core/base.ts

@@ -1,5 +1,5 @@
-import type { MiokiContext } from "mioki";
-import { logger } from "mioki";
+import type { Bot, MiokuContext } from "mioku";
+import { createGroupRef, logger } from "mioku";
 import type { SkillPermissionRole } from "mioku";
 import type { AIInstance, AIService } from "mioku";
 import type { ScreenshotService } from "mioku";
@@ -24,7 +24,7 @@ import {
 } from "./media/markdown-message";
 
 export interface SendAIResponseOptions {
-  ctx: MiokiContext;
+  ctx: MiokuContext;
   groupId: number;
   messages: string[];
   config: ChatConfig;
@@ -79,7 +79,7 @@ async function waitTypingDelay(
 export async function sendAIResponse(
   options: SendAIResponseOptions,
   selfId: number,
-): Promise<void> {
+): Promise<Array<string | undefined>> {
   const {
     ctx,
     groupId,
@@ -96,17 +96,25 @@ export async function sendAIResponse(
     ctx.logger.error(
       `[sendAIResponse] bot ${String(selfId)} not found, skip sending group message`,
     );
-    return;
+    return [];
   }
 
-  if (messages.length === 0) return;
+  if (messages.length === 0) return [];
+
+  const sentMessageIds: Array<string | undefined> = new Array(messages.length);
+  const recordSentId = (messageIndex: number, messageId?: string) => {
+    if (!messageId) return;
+    if (messageIndex >= 0 && messageIndex < messages.length) {
+      sentMessageIds[messageIndex] ??= messageId;
+    }
+  };
 
   for (let i = 0; i < messages.length; i++) {
     if (sentIndices?.has(i)) continue;
 
     const expandedLines = expandOutgoingLines(messages[i]);
 
-    let pendingReply: number | undefined;
+    let pendingReply: string | undefined;
     let lastDelayBasisText = "";
 
     for (let j = 0; j < expandedLines.length; j++) {
@@ -141,10 +149,7 @@ export async function sendAIResponse(
 
       if (pokeUsers.length > 0) {
         for (const pokeId of pokeUsers) {
-          await bot.api("group_poke", {
-            group_id: groupId,
-            user_id: pokeId,
-          });
+          await bot.pokeMember(String(groupId), String(pokeId));
         }
       }
 
@@ -152,12 +157,13 @@ export async function sendAIResponse(
 
       const finalQuoteId = pendingReply;
       if (finalQuoteId !== undefined) {
-        lineSegments.push({ type: "reply", id: String(finalQuoteId) });
+        lineSegments.push(ctx.segment.reply(finalQuoteId));
         pendingReply = undefined;
       }
 
       for (const atId of atUsers) {
-        lineSegments.push(ctx.segment.at(atId));
+        if (String(atId) === String(selfId)) continue;
+        lineSegments.push(ctx.segment.at(String(atId)));
       }
 
       if (markdownContent) {
@@ -178,13 +184,11 @@ export async function sendAIResponse(
             (imageSource?: string) => {
               const segments: any[] = [];
               if (finalQuoteIdForImage !== undefined) {
-                segments.push({
-                  type: "reply",
-                  id: String(finalQuoteIdForImage),
-                });
+                segments.push(ctx.segment.reply(finalQuoteIdForImage));
               }
               for (const atId of atUsers) {
-                segments.push(ctx.segment.at(atId));
+                if (String(atId) === String(selfId)) continue;
+                segments.push(ctx.segment.at(String(atId)));
               }
               segments.push(
                 ctx.segment.image(
@@ -194,7 +198,7 @@ export async function sendAIResponse(
               return segments;
             },
             imagePath,
-          );
+          ).then((sentId) => recordSentId(i, sentId));
           lastDelayBasisText = summarizeMarkdown(markdownContent);
           if (typingDelayEnabled && j < expandedLines.length - 1) {
             const delayMs = calculateTypingDelayMs(lastDelayBasisText || line);
@@ -217,11 +221,15 @@ export async function sendAIResponse(
         lineSegments.push(ctx.segment.text(fallbackText));
       }
       if (audioSource) {
-        lineSegments.push(ctx.segment.record(audioSource));
+        lineSegments.push(ctx.segment.raw("record", { file: audioSource }));
       }
 
       if (lineSegments.length > 0) {
-        await bot.sendGroupMsg(groupId, lineSegments);
+        const sent = await bot.sendMessage(
+          { type: "group", group_id: groupId },
+          lineSegments,
+        );
+        recordSentId(i, sent?.message_id);
         lastDelayBasisText = sendableText || fallbackText || audioText || line;
       }
 
@@ -238,10 +246,12 @@ export async function sendAIResponse(
 
     await onLineSent?.();
   }
+
+  return sentMessageIds;
 }
 
 export async function sendMessage(
-  ctx: MiokiContext,
+  ctx: MiokuContext,
   groupId: number | undefined,
   userId: number,
   text: string,
@@ -263,7 +273,7 @@ export async function sendMessage(
     // 应用错别字生成器
     const expandedLines = expandOutgoingLines(text);
 
-    let pendingReply: number | undefined;
+    let pendingReply: string | undefined;
     let lastDelayBasisText = "";
 
     for (let j = 0; j < expandedLines.length; j++) {
@@ -296,13 +306,9 @@ export async function sendMessage(
         continue;
       }
 
-      // 戳人 - 立即执行
       if (groupId && pokeUsers.length > 0) {
         for (const pokeId of pokeUsers) {
-          await bot.api("group_poke", {
-            group_id: groupId,
-            user_id: pokeId,
-          });
+          await bot.pokeMember(String(groupId), String(pokeId));
         }
       }
 
@@ -327,14 +333,11 @@ export async function sendMessage(
             (imageSource?: string) => {
               const segments: any[] = [];
               if (finalQuoteIdForImage !== undefined) {
-                segments.push({
-                  type: "reply",
-                  id: String(finalQuoteIdForImage),
-                });
+                segments.push(ctx.segment.reply(finalQuoteIdForImage));
               }
               for (const atId of atUsers) {
                 if (String(atId) !== String(selfId)) {
-                  segments.push(ctx.segment.at(atId));
+                  segments.push(ctx.segment.at(String(atId)));
                 }
               }
               segments.push(
@@ -366,13 +369,13 @@ export async function sendMessage(
         const sendableText = markdownContent ?? cleanText;
         const segments: any[] = [];
         if (pendingReply !== undefined) {
-          segments.push({ type: "reply", id: String(pendingReply) });
+          segments.push(ctx.segment.reply(pendingReply));
           pendingReply = undefined;
         }
         if (markdownContent) {
           for (const atId of atUsers) {
             if (String(atId) !== String(selfId)) {
-              segments.push(ctx.segment.at(atId));
+              segments.push(ctx.segment.at(String(atId)));
             }
           }
           if (sendableText) {
@@ -381,11 +384,14 @@ export async function sendMessage(
             segments.push(ctx.segment.text(fallbackText));
           }
           if (audioSource) {
-            segments.push(ctx.segment.record(audioSource));
+            segments.push(ctx.segment.raw("record", { file: audioSource }));
           }
 
           if (segments.length > 0 && groupId) {
-            await bot.sendGroupMsg(groupId, segments);
+            await bot.sendMessage(
+              { type: "group", group_id: groupId },
+              segments,
+            );
             lastDelayBasisText =
               sendableText || fallbackText || audioText || line;
           }
@@ -398,7 +404,7 @@ export async function sendMessage(
         // 有 @ 用户时，构建消息保持原始位置
         // 先将原始行按 @ 标记分割，然后重新构建
         let remaining = line;
-        const atPattern = /\[at:(\d+)\]/g;
+        const atPattern = /\[at: \d+\]/g;
 
         let lastIndex = 0;
         let match;
@@ -407,18 +413,18 @@ export async function sendMessage(
           const beforeAt = remaining.slice(lastIndex, match.index);
           if (beforeAt) {
             const cleaned = beforeAt
-              .replace(/\[reply:-?\d+\]/g, "")
+              .replace(/\[reply:[^\]\n]+\]/g, "")
               .replace(/\[poke:\d+\]/g, "")
               .replace(/\[audio:[^\]]+\]/gi, "")
               .trim();
             if (cleaned) {
-              segments.push({ type: "text", text: cleaned });
+              segments.push(ctx.segment.text(cleaned));
             }
           }
 
           const atId = match[1];
           if (String(atId) !== String(selfId)) {
-            segments.push(ctx.segment.at(atId));
+            segments.push(ctx.segment.at(String(atId)));
           }
 
           lastIndex = match.index + match[0].length;
@@ -428,25 +434,28 @@ export async function sendMessage(
         const afterAt = remaining.slice(lastIndex);
         if (afterAt) {
           const cleaned = afterAt
-            .replace(/\[reply:-?\d+\]/g, "")
+            .replace(/\[reply:[^\]\n]+\]/g, "")
             .replace(/\[poke:\d+\]/g, "")
             .replace(/\[audio:[^\]]+\]/gi, "")
             .trim();
           if (cleaned) {
-            segments.push({ type: "text", text: cleaned });
+            segments.push(ctx.segment.text(cleaned));
           }
         }
 
         if (audioSource) {
-          segments.push(ctx.segment.record(audioSource));
+          segments.push(ctx.segment.raw("record", { file: audioSource }));
         } else if (fallbackText) {
-          segments.push({ type: "text", text: fallbackText });
+          segments.push(ctx.segment.text(fallbackText));
         }
 
         // 发送消息
         if (segments.length > 0) {
           if (groupId) {
-            await bot.sendGroupMsg(groupId, segments);
+            await bot.sendMessage(
+              { type: "group", group_id: groupId },
+              segments,
+            );
             lastDelayBasisText =
               sendableText || fallbackText || audioText || line;
           }
@@ -462,7 +471,7 @@ export async function sendMessage(
         ) {
           const sendSegments: any[] = [];
           if (pendingReply !== undefined) {
-            sendSegments.push({ type: "reply", id: String(pendingReply) });
+            sendSegments.push(ctx.segment.reply(pendingReply));
             pendingReply = undefined;
           }
           if (sendableText) {
@@ -471,7 +480,7 @@ export async function sendMessage(
             sendSegments.push(ctx.segment.text(fallbackText));
           }
           if (audioSource) {
-            sendSegments.push(ctx.segment.record(audioSource));
+            sendSegments.push(ctx.segment.raw("record", { file: audioSource }));
           }
           if (sendSegments.length > 0) {
             await dispatchSegments(bot, groupId, userId, () => sendSegments);
@@ -521,13 +530,13 @@ function expandOutgoingLines(text: string): string[] {
 
 function normalizeActionLineBreaks(text: string): string {
   return String(text || "").replace(
-    /\\\s*(?=(?:\[\]|\[emotion:[^\]]+\]|\[audio:[^\]]+\]|\[reply:-?\d+\]))/gi,
+    /\\\s*(?=(?:\[\]|\[emotion:[^\]]+\]|\[audio:[^\]]+\]|\[reply:[^\]\n]+\]))/gi,
     "\n",
   );
 }
 
 async function buildMarkdownImage(
-  ctx: MiokiContext,
+  ctx: MiokuContext,
   markdownContent: string,
   screenshotService: ScreenshotService | undefined,
   enableMarkdownScreenshot: boolean,
@@ -550,9 +559,10 @@ async function dispatchSegments(
   userId: number | undefined,
   buildSegments: (imageSource?: string) => any[],
   fallbackImagePath?: string,
-): Promise<void> {
+): Promise<string | undefined> {
   try {
-    await sendByTarget(bot, groupId, userId, buildSegments());
+    const sent = await sendByTarget(bot, groupId, userId, buildSegments());
+    return sent?.message_id;
   } catch (error) {
     if (!fallbackImagePath || !isLocalFilePath(fallbackImagePath)) {
       throw error;
@@ -561,24 +571,34 @@ async function dispatchSegments(
     const fsPromises = await import("fs/promises");
     const buffer = await fsPromises.readFile(fallbackImagePath);
     const base64Image = `base64://${buffer.toString("base64")}`;
-    await sendByTarget(bot, groupId, userId, buildSegments(base64Image));
+    const sent = await sendByTarget(
+      bot,
+      groupId,
+      userId,
+      buildSegments(base64Image),
+    );
+    return sent?.message_id;
   }
 }
 
 async function sendByTarget(
-  bot: any,
+  bot: Bot,
   groupId: number | undefined,
   userId: number | undefined,
-  segments: any[],
-): Promise<void> {
+  segments: readonly any[],
+): Promise<import("mioku").SentMessage | undefined> {
   if (groupId) {
-    await bot.sendGroupMsg(groupId, segments);
-    return;
+    return await bot.sendMessage(
+      { type: "group", group_id: groupId },
+      segments,
+    );
   }
 
   if (userId) {
-    await bot.sendPrivateMsg(userId, segments);
-    return;
+    return await bot.sendMessage(
+      { type: "private", user_id: userId },
+      segments,
+    );
   }
 
   throw new Error("No valid message target");
@@ -612,11 +632,11 @@ function isLocalFilePath(value: string): boolean {
 }
 
 async function resolveAudioSource(
-  ctx: MiokiContext,
+  ctx: MiokuContext,
   options: {
     audioText?: string;
     config: ChatConfig;
-  audioService?: AudioServiceApi;
+    audioService?: AudioServiceApi;
   },
 ): Promise<string | null> {
   const trimmed = String(options.audioText || "").trim();
@@ -650,7 +670,7 @@ export interface GroupHistoryResult {
 export async function getGroupHistoryMessages(
   groupId: number,
   groupSessionId: string,
-  ctx: MiokiContext,
+  ctx: MiokuContext,
   historyCount: number,
   db: ChatDatabase,
   selfId: number,
@@ -689,7 +709,7 @@ export interface GroupInfoResult {
 }
 
 export async function getGroupInfoData(
-  ctx: MiokiContext,
+  ctx: MiokuContext,
   groupId: number,
   selfId: number,
   fallbackGroupName?: string,
@@ -698,9 +718,16 @@ export async function getGroupInfoData(
   let memberCount: number | undefined;
 
   try {
-    const groupInfo = await ctx.pickBot(selfId).getGroupInfo(groupId);
-    groupName = (groupInfo as any)?.group_name || fallbackGroupName;
-    memberCount = (groupInfo as any)?.member_count;
+    const bot = ctx.pickBot(selfId);
+    if (!bot) {
+      groupName = fallbackGroupName;
+      return { groupName, memberCount };
+    }
+    const groupInfo = await createGroupRef(bot, String(groupId)).getInfo();
+    if (groupInfo) {
+      groupName = groupInfo.group_name || fallbackGroupName;
+      memberCount = groupInfo.member_count;
+    }
   } catch {
     groupName = fallbackGroupName;
   }
@@ -739,12 +766,14 @@ export async function getHumanizeContexts(
     topicContext: topicContext || undefined,
     expressionContext: expressionContext || undefined,
   };
-  logger.info(`[getHumanizeContexts] session=${groupSessionId} user=${userName} triggerUserId=${triggerUserId} memoryContext=undefined topicContext="${topicContext?.slice(0, 100) ?? ""}" expressionContext="${expressionContext?.slice(0, 100) ?? ""}"`);
+  logger.info(
+    `[getHumanizeContexts] session=${groupSessionId} user=${userName} triggerUserId=${triggerUserId} memoryContext=undefined topicContext="${topicContext?.slice(0, 100) ?? ""}" expressionContext="${expressionContext?.slice(0, 100) ?? ""}"`,
+  );
   return result;
 }
 
 export interface BuildToolContextOptions {
-  ctx: MiokiContext;
+  ctx: MiokuContext;
   event: any;
   selfId: number;
   groupSessionId: string;
@@ -761,7 +790,7 @@ export interface BuildToolContextOptions {
 }
 
 function resolveTriggerSkillRole(
-  ctx: MiokiContext,
+  ctx: MiokuContext,
   event: any,
 ): SkillPermissionRole {
   const userId = event?.user_id || event?.sender?.user_id;
@@ -844,18 +873,20 @@ export function saveBotMessages(
   timestamp: number,
   config: ChatConfig,
   db: ChatDatabase,
-  ctx: MiokiContext,
-  selfId: number,
+  ctx: MiokuContext,
+  bot: Bot | undefined,
+  sentMessageIds?: Array<string | undefined>,
 ): void {
-  const bot = ctx.pickBot(selfId);
   const botNickname = config.nicknames[0] || (bot?.nickname ?? "Miku");
+  const selfId = bot ? Number(bot.bot_id) : 0;
 
   if (!bot) {
     ctx.logger.warn(`[saveBotMessages] bot ${selfId} not available`);
     return;
   }
 
-  for (const msg of messages) {
+  for (let i = 0; i < messages.length; i++) {
+    const msg = messages[i];
     const botMsg: ChatMessage = {
       sessionId: groupSessionId,
       role: "assistant",
@@ -865,29 +896,27 @@ export function saveBotMessages(
       userRole: "member",
       groupId,
       timestamp,
+      messageId: sentMessageIds?.[i],
     };
     db.saveMessage(botMsg);
   }
 }
 
 export async function sendEmoji(
-  ctx: MiokiContext,
+  ctx: MiokuContext,
   groupId: number,
   emojiPath: string | null | undefined,
-  selfId: number,
+  bot: Bot | undefined,
 ): Promise<void> {
   if (!emojiPath) return;
-  const bot = ctx.pickBot(selfId);
   if (!bot) {
-    ctx.logger.error(
-      `[sendEmoji] bot ${String(selfId)} not found, skip sending emoji`,
-    );
+    ctx.logger.error(`[sendEmoji] bot not available, skip sending emoji`);
     return;
   }
 
   try {
     const emojiSegment = ctx.segment.image(`file://${emojiPath}`);
-    await bot.sendGroupMsg(groupId, [emojiSegment]);
+    await bot.sendMessage({ type: "group", group_id: groupId }, [emojiSegment]);
   } catch (err) {
     try {
       const fsPromises = await import("fs/promises");
@@ -922,7 +951,9 @@ export async function sendEmoji(
 
       const base64DataUrl = `data:${mimeType};base64,${base64}`;
       const base64Segment = ctx.segment.image(base64DataUrl);
-      await bot.sendGroupMsg(groupId, [base64Segment]);
+      await bot.sendMessage({ type: "group", group_id: groupId }, [
+        base64Segment,
+      ]);
       ctx.logger.info(`[Emoji] Sent via base64: ${path.basename(emojiPath)}`);
     } catch (base64Err) {
       ctx.logger.error(`[Emoji] Base64 also failed: ${base64Err}`);