koishi-plugin-chatluna-think-viewer 1.0.8 → 1.0.10

package/index.js

@@ -1,4 +1,4 @@
-const { Schema } = require('koishi');
+const { Schema } = require('koishi');
 
 const name = 'chatluna-think-viewer';
 
@@ -53,12 +53,26 @@ function formatThink(text) {
     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))
+      ? Math.min(
+          ...nonEmpty.map((l) => {
+            const m = l.match(/^(\s*)/);
+            return m ? m[1].length : 0;
+          }),
+        )
       : 0;
     return filtered.map((l) => l.slice(minIndent)).join('\n');
   }
 }
 
+function parseIndex(rawIndex) {
+  if (!rawIndex) return 1;
+  if (typeof rawIndex === 'number' && Number.isFinite(rawIndex) && rawIndex > 0) return Math.floor(rawIndex);
+  const match = String(rawIndex).match(/\d+/);
+  if (!match) return 1;
+  const num = parseInt(match[0], 10);
+  return Number.isFinite(num) && num > 0 ? num : 1;
+}
+
 function getNthAiMessage(messages, n = 1) {
   if (!Array.isArray(messages) || n < 1) return null;
   let count = 0;
@@ -66,7 +80,7 @@ function getNthAiMessage(messages, n = 1) {
     const msg = messages[i];
     const type = typeof msg?._getType === 'function' ? msg._getType() : msg?.type || msg?.role;
     if (type === 'ai' || type === 'assistant') {
-      count++;
+      count += 1;
       if (count === n) return msg;
     }
   }
@@ -75,14 +89,14 @@ function getNthAiMessage(messages, n = 1) {
 
 function apply(ctx, config) {
   const cmd = ctx
-    .command(`${config.command} [index:number]`, '获取上一条回复中的 <think> 内容（可指定倒数第 N 条）')
-    .usage('不带参数默认读取最近一条；例如 think 2 读取倒数第二条 AI 回复的思考。');
+    .command(`${config.command} [index:string]`, '获取上一条回复中的 <think> 内容（可指定倒数第 N 条）')
+    .usage('不带参数默认读取最近一条；例如 think 2 读取倒数第二条 AI 回复的思考');
 
   for (const keyword of config.keywords || []) {
     cmd.shortcut(keyword, { prefix: false });
   }
 
-  cmd.action(async ({ session }, rawIndex) => {
+  cmd.action(async ({ session, args }, rawIndex) => {
     if (!config.allowPrivate && !session.guildId) {
       return '仅支持在群聊中查询。';
     }
@@ -94,26 +108,23 @@ function apply(ctx, config) {
     const messages = temp?.completionMessages || [];
     if (!messages.length) return config.emptyMessage;
 
-    let targetIndex = parseInt(rawIndex, 10);
-    if (!Number.isFinite(targetIndex) || targetIndex < 1) targetIndex = 1;
+    const targetIndex = parseIndex(rawIndex ?? args?.[0]);
 
-    const targetAi = getNthAiMessage(messages, targetIndex);
-    if (!targetAi) return `找不到倒数第 ${targetIndex} 条 AI 回复的记录。`;
+    const targetMessage = getNthAiMessage(messages, targetIndex);
+    if (!targetMessage) return config.emptyMessage;
 
-    const text = extractText(targetAi.content);
-    if (!text) return '未找到可解析的回复内容。';
-
-    const think = formatThink(extractThink(text));
-    if (!think) return config.emptyMessage || '上一次回复中没有 <think> 字段。';
+    const think = formatThink(extractThink(extractText(targetMessage.content)));
+    if (!think) return config.emptyMessage;
 
     if (config.renderImage && ctx.chatluna?.renderer) {
       try {
         const title = `### 上一条思考（倒数第 ${targetIndex} 条）`;
-        const rendered = await ctx.chatluna.renderer.render({
-          content: [
-            { type: 'text', text: `${title}\n\n\`\`\`\n${think}\n\`\`\`` },
-          ],
-        }, { type: 'image', session });
+        const rendered = await ctx.chatluna.renderer.render(
+          {
+            content: [{ type: 'text', text: `${title}\n\n\`\`\`\n${think}\n\`\`\`` }],
+          },
+          { type: 'image', session },
+        );
         if (rendered?.length) return rendered.map((r) => r.element);
       } catch (err) {
         ctx.logger?.warn?.('[think-viewer] image render failed, fallback text', err);
@@ -129,4 +140,4 @@ module.exports = {
   apply,
   Config,
   inject,
-};
+};

package/package.json

@@ -1,8 +1,8 @@
-{
+{
   "name": "koishi-plugin-chatluna-think-viewer",
-  "version": "1.0.8",
+  "version": "1.0.10",
   "main": "index.js",
-  "description": "通过命令/关键词查阅 chatluna-character 最近一次回复中的 <think> 思考内容。",
+  "description": "通过命令/关键词查看 chatluna-character 最近一次回复中的 <think> 思考内容。",
   "license": "MIT",
   "keywords": [
     "koishi",
@@ -24,7 +24,7 @@
   },
   "koishi": {
     "description": {
-      "zh": "通过命令/关键词查阅 chatluna-character 最近一次回复的 <think> 思考内容。",
+      "zh": "通过命令/关键词查看 chatluna-character 最近一次回复的 <think> 思考内容。",
       "en": "Expose a command/shortcut to read the last <think> block from chatluna-character."
     },
     "service": {
@@ -37,4 +37,4 @@
     "index.js",
     "README.md"
   ]
-}
+}