koishi-plugin-chatluna-think-viewer 1.0.2 → 1.0.4

package/index.js

@@ -48,32 +48,41 @@ function formatThink(text) {
     const parsed = JSON.parse(text);
     return JSON.stringify(parsed, null, 2);
   } catch {
-    // 保留原文，简单压缩多余空行
-    return text
-      .split('\n')
-      .map((l) => l.trimEnd())
-      .filter((l, idx, arr) => !(l === '' && arr[idx - 1] === ''))
-      .join('\n');
+    // 保留原文，去掉多余空行与统一左侧缩进
+    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);
+    const minIndent = nonEmpty.length
+      ? Math.min(...nonEmpty.map((l) => l.match(/^(\s*)/)?.[1]?.length ?? 0))
+      : 0;
+    return filtered.map((l) => l.slice(minIndent)).join('\n');
   }
 }
 
-function getLastAiMessage(messages) {
+function getNthAiMessage(messages, n = 1) {
+  if (!Array.isArray(messages) || n < 1) return null;
+  let count = 0;
   for (let i = messages.length - 1; i >= 0; i--) {
     const msg = messages[i];
     const type = typeof msg?._getType === 'function' ? msg._getType() : msg?.type || msg?.role;
-    if (type === 'ai' || type === 'assistant') return msg;
+    if (type === 'ai' || type === 'assistant') {
+      count++;
+      if (count === n) return msg;
+    }
   }
   return null;
 }
 
 function apply(ctx, config) {
-  const cmd = ctx.command(config.command, '获取上一条回复中的 <think> 内容');
+  const cmd = ctx
+    .command(`${config.command} [index:number]`, '获取上一条回复中的 <think> 内容（可指定倒数第 N 条）')
+    .usage('不带参数默认读取最近一条；例如 think 2 读取倒数第二条 AI 回复的思考。');
 
   for (const keyword of config.keywords || []) {
     cmd.shortcut(keyword, { prefix: false });
   }
 
-  cmd.action(async ({ session }) => {
+  cmd.action(async ({ session }, rawIndex) => {
     if (!config.allowPrivate && !session.guildId) {
       return '仅支持在群聊中查询。';
     }
@@ -85,10 +94,13 @@ function apply(ctx, config) {
     const messages = temp?.completionMessages || [];
     if (!messages.length) return config.emptyMessage;
 
-    const lastAi = getLastAiMessage(messages);
-    if (!lastAi) return config.emptyMessage;
+    let targetIndex = parseInt(rawIndex, 10);
+    if (!Number.isFinite(targetIndex) || targetIndex < 1) targetIndex = 1;
+
+    const targetAi = getNthAiMessage(messages, targetIndex);
+    if (!targetAi) return `找不到倒数第 ${targetIndex} 条 AI 回复的记录。`;
 
-    const text = extractText(lastAi.content);
+    const text = extractText(targetAi.content);
     if (!text) return '未找到可解析的回复内容。';
 
     const think = formatThink(extractThink(text));
@@ -96,10 +108,10 @@ function apply(ctx, config) {
 
     if (config.renderImage && ctx.chatluna?.renderer) {
       try {
+        const title = `### 上一条思考（倒数第 ${targetIndex} 条）`;
         const rendered = await ctx.chatluna.renderer.render({
           content: [
-            { type: 'text', text: '上一条思考：\n' },
-            { type: 'text', text: '```\n' + think + '\n```' },
+            { type: 'text', text: `${title}\n\n\`\`\`\n${think}\n\`\`\`` },
           ],
         }, { type: 'image', session });
         if (rendered?.length) return rendered.map((r) => r.element);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "koishi-plugin-chatluna-think-viewer",
-  "version": "1.0.2",
+  "version": "1.0.4",
   "main": "index.js",
   "description": "通过命令/关键词查看 chatluna-character 最近一次回复中的 <think> 思考内容。",
   "license": "MIT",