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

package/README.md

@@ -1,19 +1,18 @@
-# koishi-plugin-chatluna-think-viewer
+# koishi-plugin-chatluna-think-viewer
 
 通过命令或快捷关键词查看 `chatluna-character` 最近一次回复中的 `<think>` 思考内容，便于调试和复盘。
 
 ## 特性
-- 复用 `chatluna-character` 内存，不额外占用数据库。
+- 复用 `chatluna-character` 的内存，不额外占用数据库。
 - 支持命令与无前缀关键词触发。
 - 群聊可用，默认禁止私聊（可配置）。
 
 ## 安装
 ```bash
-# 使用 Koishi 控制台市场搜索「chatluna-think-viewer」安装
-# 或者 npm/yarn 安装：
+# Koishi 控制台市场搜索「chatluna-think-viewer」安装
+# 或者 npm/yarn 安装
 npm install koishi-plugin-chatluna-think-viewer
-# 或
-yarn add koishi-plugin-chatluna-think-viewer
+# yarn add koishi-plugin-chatluna-think-viewer
 ```
 
 ## 配置示例 (koishi.yml)
@@ -30,11 +29,11 @@ plugins:
 ```
 
 ## 使用
-- 群聊中发送 `think`（按你的命令前缀）或关键词“查看思考”/“上次思考”，返回上一条回复的 `<think>` 内容。
+- 在群聊中发送 `think`（根据你的命令前缀）或关键词 “查看思考 / 上次思考”，返回上一条回复的 `<think>` 内容。
 
 ## 依赖
 - koishi >= 4.18.0
 - koishi-plugin-chatluna-character >= 0.0.180
 
 ## 协议
-MIT
+MIT

package/index.js

@@ -8,11 +8,11 @@ const inject = {
 };
 
 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 将思考内容渲染为图片发送，失败则回退文本'),
+  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 将思考渲染为图片，失败时回退文本。'),
 });
 
 function extractText(content) {
@@ -37,7 +37,7 @@ function extractText(content) {
 }
 
 function extractThink(text) {
-  // 有些模型/中间件会在同一条消息里多次输出 <think>，取最后一段避免误用旧片段
+  // 某些模型/中间件会在同一条消息里多次出现 <think>，取最后一次出现的片段
   let last = '';
   const regex = /<think>([\s\S]*?)<\/think>/gi;
   let m;
@@ -49,12 +49,12 @@ function extractThink(text) {
 
 function formatThink(text) {
   if (!text) return text;
-  // 尝试 JSON 美化
+  // 尝试格式化 JSON，失败则做基础去空行/缩进美化
   try {
     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);
@@ -108,10 +108,24 @@ function getNthThink(messages, n = 1) {
   return null;
 }
 
+function getLatestRawThink(temp) {
+  if (!temp) return '';
+  const candidates = [
+    temp?.lastCompletion?.raw?.choices?.[0]?.message?.content,
+    temp?.lastCompletion?.raw?.content,
+    temp?.lastCompletion?.content,
+  ];
+  for (const c of candidates) {
+    const think = extractThink(extractText(c));
+    if (think) return think;
+  }
+  return '';
+}
+
 function apply(ctx, config) {
   const cmd = ctx
-    .command(`${config.command} [index:string]`, '获取上一条回复中的 <think> 内容（可指定倒数第 N 条）')
-    .usage('不带参数默认读取最近一条；例如 think 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 });
@@ -119,31 +133,35 @@ function apply(ctx, config) {
 
   cmd.action(async ({ session, args }, rawIndex) => {
     if (!config.allowPrivate && !session.guildId) {
-      return '仅支持在群聊中查询。';
+      return '不支持在私聊中查询。';
     }
 
     const service = ctx.chatluna_character;
-    if (!service) return 'chatluna-character 未启用。';
+    if (!service) return 'chatluna-character 未加载。';
 
     const temp = await service.getTemp(session);
-    const messages = temp?.completionMessages || [];
-    if (!messages.length) return config.emptyMessage;
-
     const targetIndex = parseIndex(rawIndex ?? args?.[0]);
 
-    const rawThink = getNthThink(messages, targetIndex);
-    const think = rawThink
-      ? formatThink(rawThink)
-      : formatThink(extractThink(extractText(getNthAiMessage(messages, targetIndex)?.content)));
-    if (!think) return config.emptyMessage;
+    // 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 消息再尝试抽取
+    const fallbackMsg = thinkFromRaw || thinkFromHistory ? null : getNthAiMessage(messages, targetIndex);
+    const think = thinkFromRaw || thinkFromHistory || extractThink(extractText(fallbackMsg?.content));
+    const formatted = formatThink(think);
+    if (!formatted) return config.emptyMessage;
 
     if (config.renderImage && ctx.chatluna?.renderer) {
       try {
         const title = `### 上一条思考（倒数第 ${targetIndex} 条）`;
-        const markdown = `<div align="center">\n${title}\n</div>\n\n<div align="left">\n${think}\n</div>`;
+        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 },
@@ -154,7 +172,7 @@ function apply(ctx, config) {
       }
     }
 
-    return `上一条思考：\n${think}`;
+    return `上一条思考（倒数第 ${targetIndex} 条）\n${formatted}`;
   });
 }
 

package/package.json

@@ -1,40 +1,40 @@
-{
-  "name": "koishi-plugin-chatluna-think-viewer",
-  "version": "1.0.15",
-  "main": "index.js",
-  "description": "Expose a command/shortcut to read the latest <think> block from chatluna-character.",
-  "license": "MIT",
-  "keywords": [
-    "koishi",
-    "chatluna",
-    "character",
-    "think"
-  ],
-  "homepage": "https://github.com/sCR0WN-s/koishi-plugin-chatluna-think-viewer",
-  "repository": {
-    "type": "git",
-    "url": "git+https://github.com/sCR0WN-s/koishi-plugin-chatluna-think-viewer.git"
-  },
-  "contributors": [
-    "sCR0WN-s <2892511968@qq.com>"
-  ],
-  "peerDependencies": {
-    "koishi": "^4.18.0",
-    "koishi-plugin-chatluna-character": "^0.0.180"
-  },
-  "koishi": {
-    "description": {
-      "zh": "通过命令/关键词查看 chatluna-character 最近一次回复的 <think> 思考内容。",
-      "en": "Expose a command/shortcut to read the last <think> block from chatluna-character."
-    },
-    "service": {
-      "required": [
-        "chatluna_character"
-      ]
-    }
-  },
-  "files": [
-    "index.js",
-    "README.md"
-  ]
-}
+{
+  "name": "koishi-plugin-chatluna-think-viewer",
+  "version": "1.0.17",
+  "main": "index.js",
+  "description": "Expose a command/shortcut to read the latest <think> block from chatluna-character.",
+  "license": "MIT",
+  "keywords": [
+    "koishi",
+    "chatluna",
+    "character",
+    "think"
+  ],
+  "homepage": "https://github.com/sCR0WN-s/koishi-plugin-chatluna-think-viewer",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/sCR0WN-s/koishi-plugin-chatluna-think-viewer.git"
+  },
+  "contributors": [
+    "sCR0WN-s <2892511968@qq.com>"
+  ],
+  "peerDependencies": {
+    "koishi": "^4.18.0",
+    "koishi-plugin-chatluna-character": "^0.0.180"
+  },
+  "koishi": {
+    "description": {
+      "zh": "通过命令/关键词查看 chatluna-character 最近一次回复的 <think> 思考内容。",
+      "en": "Expose a command/shortcut to read the last <think> block from chatluna-character."
+    },
+    "service": {
+      "required": [
+        "chatluna_character"
+      ]
+    }
+  },
+  "files": [
+    "index.js",
+    "README.md"
+  ]
+}