mioku-plugin-chat 3.0.2 → 3.0.4

package/core/chat-turn.ts

@@ -1,12 +1,8 @@
 import type { MiokuContext } from "mioku";
-import type { AITool, ChatRuntimePromptInjection } from "mioku";
+import type { AITool, Bot, ChatRuntimePromptInjection } from "mioku";
 import type { ChatPluginContext, ChatRuntimeState } from "../context";
 import type { ChatConfig, ChatMessage, TargetMessage } from "../types";
-import {
-  getGroupHistory,
-  getBotRole,
-  getQuotedContent,
-} from "../utils";
+import { getGroupHistory, getBotRole, getQuotedContent } from "../utils";
 import { buildStructuredUserInputFromTarget } from "../manage/group-structured-history";
 
 export type RuntimeReplyContextType =
@@ -53,7 +49,8 @@ function buildRuntimeTargetMessageContent(
   overrideContent?: string,
 ): string {
   if (overrideContent?.trim()) return overrideContent.trim();
-  if (!event || !event.message || !Array.isArray(event.message)) return NO_NEW_MESSAGE;
+  if (!event || !event.message || !Array.isArray(event.message))
+    return NO_NEW_MESSAGE;
   return ctx.text(event)?.trim() || NO_NEW_MESSAGE;
 }
 
@@ -74,7 +71,8 @@ function resolveRuntimeContext(
       selfId: event.self_id,
       sessionId: groupId ? `group:${groupId}` : `personal:${userId}`,
       personalSessionId: groupId ? `personal:${userId}` : undefined,
-      senderName: event.sender?.card || event.sender?.nickname || String(userId),
+      senderName:
+        event.sender?.card || event.sender?.nickname || String(userId),
       userRole: event.sender?.role || "member",
       userTitle: event.sender?.title || undefined,
       groupName: event.group_name,
@@ -85,7 +83,10 @@ function resolveRuntimeContext(
   if (typeof options.selfId !== "number") {
     throw new Error("Chat runtime requires either event or selfId");
   }
-  if (typeof options.groupId !== "number" && typeof options.userId !== "number") {
+  if (
+    typeof options.groupId !== "number" &&
+    typeof options.userId !== "number"
+  ) {
     throw new Error("Chat runtime requires groupId or userId");
   }
 
@@ -112,7 +113,9 @@ function resolveRuntimeContext(
     groupId: options.groupId,
     userId,
     selfId: options.selfId,
-    sessionId: options.groupId ? `group:${options.groupId}` : `personal:${userId}`,
+    sessionId: options.groupId
+      ? `group:${options.groupId}`
+      : `personal:${userId}`,
     personalSessionId:
       options.groupId && userId ? `personal:${userId}` : undefined,
     senderName: options.groupId ? "system" : String(userId),
@@ -144,39 +147,90 @@ export async function finalizeChatTurn(
     return;
   }
   const { ctx } = pluginCtx;
-  const { groupId, groupSessionId, userId, selfId, cfg, toolCtx, result, event } = args;
+  const {
+    groupId,
+    groupSessionId,
+    userId,
+    selfId,
+    cfg,
+    toolCtx,
+    result,
+    event,
+  } = args;
+
+  const actingBot = (selfId ? ctx.pickBot(selfId) : undefined) ?? event?.bot;
+  const eventBotId = event?.bot ? String(event.bot.bot_id) : undefined;
+  const actingIsEventBot =
+    eventBotId == null ||
+    (actingBot != null && String(actingBot.bot_id) === eventBotId);
 
   if (groupId) {
     const sentMessageIds = await pluginCtx.sendAIResponse(
-      { ctx, groupId, messages: result.messages, config: cfg, sentIndices: toolCtx.sentMessageIndices, audioService: pluginCtx.audioService },
+      {
+        ctx,
+        groupId,
+        messages: result.messages,
+        config: cfg,
+        sentIndices: toolCtx.sentMessageIndices,
+        audioService: pluginCtx.audioService,
+      },
       selfId,
     );
-    await pluginCtx.sendEmoji(ctx, groupId, result.emojiPath, event?.bot);
+    await pluginCtx.sendEmoji(ctx, groupId, result.emojiPath, actingBot);
     const now = Date.now();
-    pluginCtx.saveBotMessages(groupId, groupSessionId, result.messages, now, cfg, pluginCtx.db, ctx, event?.bot, sentMessageIds);
+    pluginCtx.saveBotMessages(
+      groupId,
+      groupSessionId,
+      result.messages,
+      now,
+      cfg,
+      pluginCtx.db,
+      ctx,
+      actingBot,
+      sentMessageIds,
+    );
     if (args.isLive) {
-      pluginCtx.idleCheckManager.recordBotMessages(groupSessionId, result.messages.length, selfId);
+      pluginCtx.idleCheckManager.recordBotMessages(
+        groupSessionId,
+        result.messages.length,
+        selfId,
+      );
     }
-    pluginCtx.cooldownManager.startCooldownTimer(groupSessionId, groupId, selfId);
+    pluginCtx.cooldownManager.startCooldownTimer(
+      groupSessionId,
+      groupId,
+      selfId,
+    );
   } else {
     const sentIndices = toolCtx.sentMessageIndices;
     for (let i = 0; i < result.messages.length; i++) {
       if (sentIndices?.has(i)) continue;
-      await pluginCtx.sendMessage(ctx, undefined, userId, result.messages[i], cfg, selfId, pluginCtx.audioService);
+      await pluginCtx.sendMessage(
+        ctx,
+        undefined,
+        userId,
+        result.messages[i],
+        cfg,
+        selfId,
+        pluginCtx.audioService,
+      );
     }
     if (result.emojiPath) {
       try {
         const emojiSegment = ctx.segment.image(`file://${result.emojiPath}`);
-        if (args.isLive) {
+        if (args.isLive && actingIsEventBot && event?.reply) {
           await event.reply([emojiSegment]);
         } else {
-          const bot = args.event?.bot;
-          if (!bot) throw new Error(`bot ${selfId} not found`);
-          await bot.sendMessage({ type: "private", user_id: userId}, [emojiSegment]);
+          if (!actingBot) throw new Error(`bot ${selfId} not found`);
+          await actingBot.sendMessage({ type: "private", user_id: userId }, [
+            emojiSegment,
+          ]);
         }
       } catch (err) {
         ctx.logger.warn(
-          args.isLive ? `[Emoticon] Send failed: ${err}` : `[chat-runtime] Send emoji failed: ${err}`,
+          args.isLive
+            ? `[Emoticon] Send failed: ${err}`
+            : `[chat-runtime] Send emoji failed: ${err}`,
         );
       }
     }
@@ -189,12 +243,13 @@ export async function processChat(
   e: any,
   pluginCtx: ChatPluginContext,
   runtimeState: ChatRuntimeState,
+  options: { replyBot?: Bot } = {},
 ): Promise<void> {
   const { ctx } = pluginCtx;
   const isGroup = e.message_type === "group";
   const groupId: number | undefined = isGroup ? e.group_id : undefined;
   const userId: number = e.user_id || e.sender?.user_id;
-  const selfId = e.self_id;
+  const selfId = options.replyBot ? Number(options.replyBot.bot_id) : e.self_id;
   const cfg = await pluginCtx.getConfig(groupId);
 
   const personalSessionId = `personal:${userId}`;
@@ -212,7 +267,11 @@ export async function processChat(
       groupId ?? userId,
     );
     if (groupId) {
-      pluginCtx.sessionManager.getOrCreate(personalSessionId, "personal", userId);
+      pluginCtx.sessionManager.getOrCreate(
+        personalSessionId,
+        "personal",
+        userId,
+      );
     }
 
     const quotedInfo = await getQuotedContent(e, ctx);
@@ -242,7 +301,7 @@ export async function processChat(
           groupId,
           ctx,
           cfg.historyCount,
-          e.self_id,
+          selfId,
           pluginCtx.db,
           pluginCtx.buildHistoryMediaOptions(pluginCtx.aiInstance, cfg),
         )
@@ -260,15 +319,18 @@ export async function processChat(
     }));
 
     const botNickname =
-      cfg.nicknames[0] || e.bot?.nickname || "Bot";
-    const botRole = groupId ? await getBotRole(groupId, ctx, e.self_id) : "member";
+      cfg.nicknames[0] ||
+      options.replyBot?.nickname ||
+      e.bot?.nickname ||
+      "Bot";
+    const botRole = groupId ? await getBotRole(groupId, ctx, selfId) : "member";
     let groupName: string | undefined;
     let memberCount: number | undefined;
     if (groupId) {
       const groupInfo = await pluginCtx.getGroupInfoData(
         ctx,
         groupId,
-        e.self_id,
+        selfId,
         e.group_name,
       );
       groupName = groupInfo.groupName;
@@ -311,7 +373,7 @@ export async function processChat(
       pendingImageUrls: imageUrls,
       humanize: pluginCtx.humanize,
       targetMessage,
-      selfId: e.self_id,
+      selfId,
       audioService: pluginCtx.audioService,
     });
 
@@ -351,7 +413,12 @@ export async function processChat(
               }
             : undefined,
         ),
-      { userId, groupId, label: isGroup ? "group-chat" : "private-chat", skipRetryOnRateLimit: true },
+      {
+        userId,
+        groupId,
+        label: isGroup ? "group-chat" : "private-chat",
+        skipRetryOnRateLimit: true,
+      },
     );
 
     if (!result) {
@@ -366,7 +433,7 @@ export async function processChat(
       groupId,
       groupSessionId,
       userId,
-      selfId: e.self_id,
+      selfId,
       toolCtx,
       send: true,
       isLive: true,
@@ -435,7 +502,13 @@ async function executeChatRuntimeRequestNow(
   }
 
   const rawHistory = groupId
-    ? await getGroupHistory(groupId, pluginCtx.ctx, cfg.historyCount, selfId, pluginCtx.db)
+    ? await getGroupHistory(
+        groupId,
+        pluginCtx.ctx,
+        cfg.historyCount,
+        selfId,
+        pluginCtx.db,
+      )
     : [];
   const history: ChatMessage[] = rawHistory.map((msg) => ({
     sessionId,
@@ -449,9 +522,10 @@ async function executeChatRuntimeRequestNow(
     messageId: msg.messageId,
   }));
 
-  const botRole = groupId ? await getBotRole(groupId, pluginCtx.ctx, selfId) : "member";
-  const botNickname =
-    cfg.nicknames[0] || event?.bot?.nickname || "Bot";
+  const botRole = groupId
+    ? await getBotRole(groupId, pluginCtx.ctx, selfId)
+    : "member";
+  const botNickname = cfg.nicknames[0] || event?.bot?.nickname || "Bot";
 
   let groupName: string | undefined;
   let memberCount: number | undefined;

package/handlers/idle-debug.ts

@@ -11,7 +11,7 @@ export async function handleIdleCheckDebug(
   cfg: ChatConfig,
 ): Promise<void> {
   const ctx: MiokuContext = pluginCtx.ctx;
-  const isOwner = ctx.isOwner?.(e) ?? false;
+  const isOwner = ctx.isMaster?.(e) ?? false;
   if (!isOwner) {
     await e.reply("只有主人才能使用这个指令~");
     return;

package/handlers/message.ts

@@ -22,7 +22,6 @@ import {
   summarizeHistoryVideo,
 } from "../core/media/history-media";
 import { handleIdleCheckDebug } from "./idle-debug";
-import { handleTtsCommand } from "./tts-command";
 import { processChat } from "../core/chat-turn";
 
 const POKE_COOLDOWN_MS = 10 * 60_000;
@@ -55,25 +54,6 @@ export function createMessageHandler(
       return;
     }
 
-    if (text.startsWith("/tts")) {
-      const payload = text.replace(/^\/tts\s*/, "");
-      await handleTtsCommand(pluginCtx, e, payload);
-      return;
-    }
-
-    if (text === "/重置会话") {
-      if (groupId) {
-        pluginCtx.sessionManager.resetBotMessages(`group:${groupId}`);
-        pluginCtx.groupStructuredHistory.clear(`group:${groupId}`);
-        await e.reply("已清除本群会话中 AI 发送的消息~");
-      } else {
-        pluginCtx.sessionManager.resetBotMessages(`personal:${userId}`);
-        pluginCtx.groupStructuredHistory.clear(`personal:${userId}`);
-        await e.reply("已清除你的个人会话中 AI 发送的消息~");
-      }
-      return;
-    }
-
     if (groupId && !isGroupAllowed(groupId, cfg)) return;
 
     // 媒体分析
@@ -222,6 +202,10 @@ export function createMessageHandler(
     }
 
     const atBot = shouldTrigger(e, text, cfg, ctx);
+    const replyBot = ctx.pickReplyBot(e);
+    const actorSelfId = replyBot
+      ? Number(replyBot.bot_id)
+      : Number(e.self_id || 0);
     const quotedBot = isGroup ? await isQuotingBot(e, ctx) : null;
     const mentionedNickname =
       cfg.nicknames.length > 0 &&
@@ -275,14 +259,14 @@ export function createMessageHandler(
               groupSessionId,
               groupId,
               delayInfo.delayMs,
-              Number(e.self_id || 0),
+              actorSelfId,
             );
             return;
           }
         }
 
         pluginCtx.rateLimiter.record(userId, groupId, text);
-        await processChat(e, pluginCtx, runtimeState);
+        await processChat(e, pluginCtx, runtimeState, { replyBot });
         return;
       }
 
@@ -294,10 +278,11 @@ export function createMessageHandler(
           ctx,
           cfg.historyCount,
           pluginCtx.db,
-          Number(e.self_id || 0),
+          actorSelfId,
           pluginCtx.buildHistoryMediaOptions(pluginCtx.aiInstance, cfg),
         );
-        const botNickname = cfg.nicknames[0] || e.bot?.nickname || "Bot";
+        const botNickname =
+          cfg.nicknames[0] || replyBot?.nickname || e.bot?.nickname || "Bot";
         const planResult = await pluginCtx.humanize.actionPlanner.plan(
           groupSessionId,
           botNickname,
@@ -307,7 +292,7 @@ export function createMessageHandler(
         if (planResult.action !== "reply") return;
         if (!pluginCtx.rateLimiter.canProcess(userId, groupId, text)) return;
         pluginCtx.rateLimiter.record(userId, groupId, text);
-        await processChat(e, pluginCtx, runtimeState);
+        await processChat(e, pluginCtx, runtimeState, { replyBot });
       }
     };
 
@@ -319,7 +304,7 @@ export function createMessageHandler(
           pluginCtx.rateLimiter.recordInteraction(groupId, userId);
           pluginCtx.queueProcessor.scheduleQueuedMessages(
             groupSessionId,
-            Number(e.self_id || 0),
+            actorSelfId,
           );
         }
         return;

package/handlers/tts-command.ts

@@ -15,7 +15,7 @@ export async function handleTtsCommand(
   payload: string,
 ): Promise<void> {
   const { ctx } = pluginCtx;
-  const isOwner = ctx.isOwner?.(e) ?? false;
+  const isOwner = ctx.isMaster?.(e) ?? false;
   // if (!isOwner) {
   //   await e.reply("只有主人才能使用 /tts 指令~");
   //   return;

package/index.ts

@@ -43,6 +43,7 @@ import { RateLimitGuard } from "./manage/rate-limit-guard";
 import { createChatRuntime } from "./runtime/chat-runtime";
 import { createMessageHandler } from "./handlers/message";
 import { createPokeHandler } from "./handlers/poke";
+import { handleTtsCommand } from "./handlers/tts-command";
 import { buildHistoryMediaOptions } from "./core/media/segment";
 import type { ChatConfigProvider } from "./humanize";
 import type { ChatPluginContext, ChatHandlerState } from "./context";
@@ -115,8 +116,6 @@ function resolveRoleInstances(aiService: AIService): {
 
 export default definePlugin({
   name: "chat",
-  version: "1.1.0",
-  description: "AI 智能聊天插件",
   async setup(ctx: MiokuContext) {
     ctx.logger.info("聊天插件正在初始化...");
 
@@ -511,6 +510,59 @@ roleModels = resolved.models;
     const runtime = createChatRuntime(pluginCtx, getConfig);
     aiService.registerChatRuntime(runtime);
 
+    ctx.command({
+      name: "/重置会话",
+      match: /^\/重置会话$/,
+      prefixes: false,
+      description: "重置自己的AI聊天记录",
+      handler: async ({ event }) => {
+        if (event.user_id === event.self_id) return;
+        const groupId =
+          event.message_type === "group" ? Number(event.group_id) : undefined;
+        if (groupId) {
+          pluginCtx.sessionManager.resetBotMessages(`group:${groupId}`);
+          pluginCtx.groupStructuredHistory.clear(`group:${groupId}`);
+          await event.reply("已清除本群会话中 AI 发送的消息~");
+        } else {
+          const userId = Number(event.user_id || event.sender?.user_id || 0);
+          pluginCtx.sessionManager.resetBotMessages(`personal:${userId}`);
+          pluginCtx.groupStructuredHistory.clear(`personal:${userId}`);
+          await event.reply("已清除你的个人会话中 AI 发送的消息~");
+        }
+      },
+    });
+    ctx.command({
+      name: "/重置群会话",
+      match: /^\/重置群会话$/,
+      prefixes: false,
+      permission: "admin",
+      description: "重置当前群的AI聊天记录",
+      handler: async ({ event }) => {
+        if (event.user_id === event.self_id) return;
+        const groupId =
+          event.message_type === "group" ? Number(event.group_id) : undefined;
+        if (!groupId) {
+          await event.reply("在群里使用试试看吧~", true);
+          return;
+        }
+        pluginCtx.sessionManager.resetBotMessages(`group:${groupId}`);
+        pluginCtx.groupStructuredHistory.clear(`group:${groupId}`);
+        await event.reply("已清除本群会话中 AI 发送的消息~");
+      },
+    });
+    ctx.command({
+      name: "/tts",
+      match: /^\/tts\s*(.*)$/,
+      prefixes: false,
+      permission: "owner",
+      description: "TTS 推理",
+      usage: "/tts <文本>",
+      handler: async ({ event, match }) => {
+        if (event.user_id === event.self_id) return;
+        await handleTtsCommand(pluginCtx, event, String(match?.[1] ?? "").trim());
+      },
+    });
+
     ctx.handle("message", createMessageHandler(pluginCtx, handlerState));
     ctx.handle("notice.group.poke", createPokeHandler(pluginCtx, handlerState));
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "mioku-plugin-chat",
-  "version": "3.0.2",
+  "version": "3.0.4",
   "description": "AI 智能聊天插件",
   "main": "index.ts",
   "type": "module",
@@ -17,35 +17,7 @@
       "config",
       "help",
       "screenshot"
-    ],
-    "accessHooks": [
-      {
-        "id": "戳一戳触发对话",
-        "event": "notice.group.poke",
-        "description": "群内戳一戳 bot"
-      }
-    ],
-    "help": {
-      "title": "AI 聊天",
-      "description": "智能 AI 聊天插件",
-      "commands": [
-        {
-          "cmd": "/重置会话",
-          "desc": "重置自己的AI聊天记录",
-          "role": "member"
-        },
-        {
-          "cmd": "/重置群会话",
-          "desc": "重置当前群的AI聊天记录",
-          "role": "admin"
-        },
-        {
-          "cmd": "/tts <文本>",
-          "desc": "TTS 推理",
-          "role": "owner"
-        }
-      ]
-    }
+    ]
   },
   "dependencies": {
     "openai": "^4.0.0",

package/utils/message.ts

@@ -27,11 +27,7 @@ export function shouldTrigger(
 ): boolean {
   if (e.message_type === "private") return false;
 
-  const atSeg = e.message?.find((seg: any) => seg.type === "at");
-  if (!atSeg) return false;
-  const data = (atSeg.data ?? {}) as { qq?: unknown; target?: unknown };
-  const qq = data.qq ?? data.target;
-  return String(qq ?? "") === String(e.self_id ?? "");
+  return ctx.mentionedBots(e).length > 0;
 }
 
 /**
@@ -45,8 +41,13 @@ export async function isQuotingBot(
   if (e.quote_id) {
     try {
       const quoteMsg = await fetchQuotedMessage(e, ctx);
-      if (quoteMsg && String(quoteMsg.sender?.user_id) === String(e.self_id)) {
-        const quotedText = quoteMsg.message
+      const quotedUserId = String(quoteMsg?.sender?.user_id ?? "");
+      const quotingAnyBot =
+        quoteMsg != null &&
+        quotedUserId !== "" &&
+        ctx.bots.some((bot) => String(bot.bot_id) === quotedUserId);
+      if (quotingAnyBot) {
+        const quotedText = quoteMsg!.message
           ?.filter((s: any) => s.type === "text")
           .map((s: any) => String(s.data?.text ?? s.text ?? ""))
           .join("");
@@ -130,15 +131,9 @@ export function extractContent(
     text = ctx.text(e) || "";
   } catch {}
 
-  // If text is empty but user @'d the bot, describe the action
+  // If text is empty but the message @'d a connected bot, describe the action
   if (!text.trim() && e.message) {
-    const hasAt = e.message.some((seg: any) => {
-      if (seg.type !== "at") return false;
-      const data = (seg.data ?? {}) as { qq?: unknown; target?: unknown };
-      const qq = data.qq ?? data.target;
-      return String(qq ?? "") === String(e.self_id ?? "");
-    });
-    if (hasAt) {
+    if (ctx.mentionedBots(e).length > 0) {
       text = "[@you with no text]";
     }
   }