koishi-plugin-chatluna-think-viewer 1.0.17 → 1.0.19

package/README.md

@@ -1,39 +1,44 @@
-# koishi-plugin-chatluna-think-viewer
-
-通过命令或快捷关键词查看 `chatluna-character` 最近一次回复中的 `<think>` 思考内容，便于调试和复盘。
-
-## 特性
-- 复用 `chatluna-character` 的内存，不额外占用数据库。
-- 支持命令与无前缀关键词触发。
-- 群聊可用，默认禁止私聊（可配置）。
-
-## 安装
-```bash
-# Koishi 控制台市场搜索「chatluna-think-viewer」安装
-# 或者 npm/yarn 安装
-npm install koishi-plugin-chatluna-think-viewer
-# yarn add koishi-plugin-chatluna-think-viewer
-```
-
-## 配置示例 (koishi.yml)
-```yaml
-plugins:
-  chatluna-character: {}
-  chatluna-think-viewer:
-    command: think
-    keywords:
-      - 查看思考
-      - 上次思考
-    allowPrivate: false
-    emptyMessage: 暂时没有可用的思考记录。
-```
-
-## 使用
-- 在群聊中发送 `think`（根据你的命令前缀）或关键词 “查看思考 / 上次思考”，返回上一条回复的 `<think>` 内容。
-
-## 依赖
-- koishi >= 4.18.0
-- koishi-plugin-chatluna-character >= 0.0.180
-
-## 协议
+# koishi-plugin-chatluna-think-viewer
+
+????/????? `chatluna-character` ??????? `<think>` ???????????????????????
+
+## ??
+- ?? `chatluna-character` ??????????????????????
+- ??????????????????
+- **????????**?? bot ???????????????? `<think>` ????? JSON ???????????????????
+
+## ??
+```bash
+# Koishi ??????? chatluna-think-viewer ??
+npm install koishi-plugin-chatluna-think-viewer
+```
+
+## ???? (koishi.yml)
+```yaml
+plugins:
+  chatluna-character: {}
+  chatluna-think-viewer:
+    command: think
+    keywords:
+      - ????
+      - ????
+    allowPrivate: false
+    emptyMessage: ????????????
+    # ??????
+    guardEnabled: true
+    guardMode: recall   # recall | block
+    guardForbiddenPatterns:
+      - '<think>[\s\S]*?<\/think>'
+      - '"role"\s*:\s*"assistant"'
+```
+
+## ??
+- ????? `think` ????????????? `<think>` ???`think 2` ????? 2 ??
+- ???????????? bot ???????????????????????????????
+
+## ??
+- koishi >= 4.18.0
+- koishi-plugin-chatluna-character >= 0.0.180
+
+## ??
 MIT

package/index.js

@@ -1,4 +1,5 @@
-const { Schema } = require('koishi');
+const { Schema, h } = require('koishi');
+const { Bot } = require('@satorijs/core');
 
 const name = 'chatluna-think-viewer';
 
@@ -7,37 +8,59 @@ const inject = {
   chatluna: { required: false },
 };
 
-const Config = Schema.object({
-  command: Schema.string().default('think').description('命令名。'),
-  keywords: Schema.array(Schema.string()).default(['查看思考', '上次思考']).description('可无前缀触发的关键词。'),
-  allowPrivate: Schema.boolean().default(false).description('是否允许在私聊中使用。'),
-  emptyMessage: Schema.string().default('暂时没有可用的思考记录。').description('没有记录时的提示文本。'),
-  renderImage: Schema.boolean().default(false).description('是否用 ChatLuna 的 image renderer 将思考渲染为图片，失败时回退文本。'),
-});
+const defaultForbidden = [
+  '<think>[\\s\\S]*?<\\/think>',
+  '```\\s*think[\\s\\S]*?```',
+  '"role"\\s*:\\s*"assistant"',
+  '"analysis"\\s*:',
+  '"thought"\\s*:',
+];
+
+const Config = Schema.intersect([
+  Schema.object({
+    command: Schema.string().default('think').description('查看思考内容的指令名'),
+    keywords: Schema.array(Schema.string()).default(['查看思考', '上次思考']).description('可无前缀触发的关键词'),
+    allowPrivate: Schema.boolean().default(false).description('是否允许在私聊中使用'),
+    emptyMessage: Schema.string().default('暂时没有可用的思考记录。').description('没有记录时的提示文本'),
+    renderImage: Schema.boolean().default(false).description('是否通过 ChatLuna image renderer 将思考渲染为图片，失败时回退文本'),
+  }).description('思考查看配置'),
+  Schema.object({
+    guardEnabled: Schema.boolean().default(true).description('异常输出自动拦截开关'),
+    guardMode: Schema.union(['recall', 'block']).default('recall').description('recall=先发送后撤回，block=直接阻止发送'),
+    guardDelay: Schema.number().default(1).min(0).max(60).description('撤回延迟（秒）'),
+    guardAllowPrivate: Schema.boolean().default(true).description('是否在私聊中也启用拦截'),
+    guardGroups: Schema.array(Schema.string()).default([]).description('只在这些群生效，留空表示全部'),
+    guardForbiddenPatterns: Schema.array(Schema.string())
+      .default(defaultForbidden)
+      .description('命中即视为异常的模式，用于避免思考泄露或 JSON 生出'),
+    guardAllowedPatterns: Schema.array(Schema.string())
+      .default(['[\\s\\S]+'])
+      .description('可选白名单，至少匹配一个才算正常'),
+    guardLog: Schema.boolean().default(true).description('是否在日志记录异常原因和内容'),
+    guardContentPreview: Schema.number().default(80).min(10).max(500).description('日志内容预览长度'),
+  }).description('异常输出自动处理'),
+]);
 
 function extractText(content) {
-  if (typeof content === 'string') return content;
-  if (Array.isArray(content)) {
-    return content
-      .map((part) => {
-        if (typeof part === 'string') return part;
-        if (part && typeof part === 'object') {
-          return part.text || part.content || part.value || '';
-        }
-        return '';
-      })
-      .join('');
-  }
-  if (content && typeof content === 'object') {
-    if (typeof content.text === 'string') return content.text;
-    if (typeof content.content === 'string') return content.content;
-    if (typeof content.value === 'string') return content.value;
+  if (content == null) return '';
+  const normalized = h.normalize(content);
+  const parts = [];
+  for (const el of normalized) {
+    if (typeof el === 'string') {
+      parts.push(el);
+      continue;
+    }
+    if (Array.isArray(el.children) && el.children.length) {
+      parts.push(extractText(el.children));
+    }
+    const textLike = el.attrs?.content ?? el.attrs?.text ?? el.children?.join?.('') ?? '';
+    if (textLike) parts.push(textLike);
   }
-  return '';
+  return parts.join('');
 }
 
 function extractThink(text) {
-  // 某些模型/中间件会在同一条消息里多次出现 <think>，取最后一次出现的片段
+  // 某些模型/中间件会在同一条消息里多次出现 <think>，取最后一次
   let last = '';
   const regex = /<think>([\s\S]*?)<\/think>/gi;
   let m;
@@ -54,7 +77,6 @@ function formatThink(text) {
     const parsed = JSON.parse(text);
     return JSON.stringify(parsed, null, 2);
   } catch {
-    // 保留原文，去掉多余空行并统一缩进
     const lines = text.split('\n').map((l) => l.trimEnd());
     const filtered = lines.filter((l, idx, arr) => !(l === '' && arr[idx - 1] === ''));
     const nonEmpty = filtered.filter((l) => l.trim().length > 0);
@@ -122,10 +144,95 @@ function getLatestRawThink(temp) {
   return '';
 }
 
+function compileRegex(list) {
+  return (list || [])
+    .map((p) => {
+      try {
+        return new RegExp(p, 'i');
+      } catch (err) {
+        return null;
+      }
+    })
+    .filter(Boolean);
+}
+
+function detectAbnormal(text, forbidden, allowed) {
+  if (!text) return null;
+  for (const re of forbidden) {
+    if (re.test(text)) return `命中禁止模式: /${re.source}/`;
+  }
+  if (allowed.length && !allowed.some((re) => re.test(text))) {
+    return '未匹配任何允许模式';
+  }
+  return null;
+}
+
+function shorten(text, limit = 80) {
+  if (!text) return '';
+  if (text.length <= limit) return text;
+  return `${text.slice(0, limit)}...(${text.length} chars)`;
+}
+
+function shouldGuard(config, options) {
+  const session = options?.session;
+  if (!session) return false;
+  const guildId = session.guildId || session.event?.guild?.id;
+  const isGroup = !!guildId;
+  if (!config.guardAllowPrivate && !isGroup) return false;
+  if (config.guardGroups?.length && isGroup && !config.guardGroups.includes(guildId)) return false;
+  return true;
+}
+
+function applyGuard(ctx, config) {
+  if (!config.guardEnabled) return;
+  const logger = ctx.logger(`${name}:guard`);
+  const forbidden = compileRegex(config.guardForbiddenPatterns);
+  const allowed = compileRegex(config.guardAllowedPatterns);
+  const original = Bot.prototype.sendMessage;
+
+  Bot.prototype.sendMessage = async function patched(channelId, content, referrer, options = {}) {
+    if (!shouldGuard(config, options)) {
+      return original.call(this, channelId, content, referrer, options);
+    }
+
+    const text = extractText(content);
+    const reason = detectAbnormal(text, forbidden, allowed);
+
+    if (!reason) {
+      return original.call(this, channelId, content, referrer, options);
+    }
+
+    const preview = shorten(text, config.guardContentPreview);
+    if (config.guardMode === 'block') {
+      if (config.guardLog) logger.warn(`[block] ${reason} | content: ${preview}`);
+      return [];
+    }
+
+    const ids = await original.call(this, channelId, content, referrer, options);
+    if (config.guardLog) logger.warn(`[recall] ${reason} | content: ${preview}`);
+    const delay = Math.max(0, config.guardDelay) * 1000;
+    if (Array.isArray(ids) && ids.length && typeof this.deleteMessage === 'function') {
+      setTimeout(() => {
+        for (const id of ids) {
+          this.deleteMessage(channelId, id).catch((err) => {
+            logger.warn(`[recall-failed] id=${id} reason=${err?.message || err}`);
+          });
+        }
+      }, delay);
+    }
+    return ids;
+  };
+
+  ctx.on('dispose', () => {
+    Bot.prototype.sendMessage = original;
+  });
+}
+
 function apply(ctx, config) {
+  // 思考查看指令
   const cmd = ctx
-    .command(`${config.command} [index:string]`, '读取上一条回复里的 <think> 内容，可指定倒数第 N 条')
-    .usage('不带参数默认读取最新一条；示例：think 2 读取倒数第 2 条 AI 回复的思考');
+    .command(`${config.command} [index:string]`, '读取上一条含 <think> 的内容，可指定倒数第 N 条')
+    .usage('不带参数默认最新；示例：think 2 查询倒数第 2 条 AI 回复的思考');
 
   for (const keyword of config.keywords || []) {
     cmd.shortcut(keyword, { prefix: false });
@@ -142,14 +249,14 @@ function apply(ctx, config) {
     const temp = await service.getTemp(session);
     const targetIndex = parseIndex(rawIndex ?? args?.[0]);
 
-    // 1) 优先读取最新一次原始响应（通常仍含 <think>），只对第 1 条有效
+    // 1) 优先读取最新一次原始响应（通常仍含 <think>），仅对第 1 条有效
     const thinkFromRaw = targetIndex === 1 ? getLatestRawThink(temp) : '';
 
     // 2) 历史 completionMessages 中真正带 <think> 的 AI 消息
     const messages = temp?.completionMessages || [];
     const thinkFromHistory = thinkFromRaw ? '' : getNthThink(messages, targetIndex);
 
-    // 3) 回退：第 N 条 AI 消息再尝试抽取
+    // 3) 回退：第 N 条 AI 消息再尝试抽取 <think>
     const fallbackMsg = thinkFromRaw || thinkFromHistory ? null : getNthAiMessage(messages, targetIndex);
     const think = thinkFromRaw || thinkFromHistory || extractThink(extractText(fallbackMsg?.content));
     const formatted = formatThink(think);
@@ -161,7 +268,6 @@ function apply(ctx, config) {
         const markdown = `<div align="center">\n${title}\n</div>\n\n<div align="left">\n${formatted}\n</div>`;
         const rendered = await ctx.chatluna.renderer.render(
           {
-            // 居中标题、左对齐正文，保持 renderer 兼容
             content: [{ type: 'text', text: markdown }],
           },
           { type: 'image', session },
@@ -174,6 +280,9 @@ function apply(ctx, config) {
 
     return `上一条思考（倒数第 ${targetIndex} 条）\n${formatted}`;
   });
+
+  // 异常输出自动处理
+  applyGuard(ctx, config);
 }
 
 module.exports = {
@@ -181,4 +290,4 @@ module.exports = {
   apply,
   Config,
   inject,
-};
+};

package/package.json

@@ -1,14 +1,17 @@
-{
+{
   "name": "koishi-plugin-chatluna-think-viewer",
-  "version": "1.0.17",
+  "version": "1.0.19",
   "main": "index.js",
-  "description": "Expose a command/shortcut to read the latest <think> block from chatluna-character.",
+  "description": "View chatluna <think> blocks and auto recall abnormal formatted replies.",
   "license": "MIT",
   "keywords": [
     "koishi",
     "chatluna",
     "character",
-    "think"
+    "think",
+    "recall",
+    "guard",
+    "moderation"
   ],
   "homepage": "https://github.com/sCR0WN-s/koishi-plugin-chatluna-think-viewer",
   "repository": {
@@ -24,8 +27,8 @@
   },
   "koishi": {
     "description": {
-      "zh": "通过命令/关键词查看 chatluna-character 最近一次回复的 <think> 思考内容。",
-      "en": "Expose a command/shortcut to read the last <think> block from chatluna-character."
+      "zh": "?? chatluna-character ? <think> ??????????????????/???",
+      "en": "View <think> blocks from chatluna-character and auto recall malformed replies."
     },
     "service": {
       "required": [
@@ -38,3 +41,4 @@
     "README.md"
   ]
 }
+