koishi-plugin-chatluna-think-viewer 2.2.5 → 2.2.6

package/index.js

@@ -8,8 +8,8 @@ const inject = {
   chatluna: { required: false },
 };
 
-const defaultForbidden = [
-  '<think>[\\s\\S]*?<\\/think>',
+const defaultForbidden = [
+  '<think>[\\s\\S]*?<\\/think>',
   '<status>[\\s\\S]*?<\\/status>',
   '<output>[\\s\\S]*?<\\/output>',
   '<analysis>[\\s\\S]*?<\\/analysis>',
@@ -28,7 +28,7 @@ const defaultForbidden = [
 const strictOutputPattern =
   '^\\s*<output>\\s*(<message>(?:<at>\\d+<\\/at>\\s*)?(?:<sticker>[^<]*<\\/sticker>|[^<]*)<\\/message>\\s*){1,5}<\\/output>\\s*$';
 
-const Config = Schema.intersect([
+const Config = Schema.intersect([
   Schema.object({
     command: Schema.string().default('think').description('\u67e5\u770b\u601d\u8003\u5185\u5bb9\u7684\u6307\u4ee4\u540d'),
     keywords: Schema.array(Schema.string()).default(['\u67e5\u770b\u601d\u8003', '\u4e0a\u6b21\u601d\u8003']).description('\u53ef\u65e0\u524d\u7f00\u89e6\u53d1\u7684\u5173\u952e\u8bcd'),
@@ -323,10 +323,42 @@ function apply(ctx, config) {
   });
 
   // \u5f02\u5e38\u8f93\u51fa\u81ea\u52a8\u5904\u7406
-  applyGuard(ctx, config);
-}
-
-module.exports = {
+  applyGuard(ctx, config);
+
+  // Hot-fix chatluna-character completionMessages trimming bug:
+  // it used to drop newest messages; we wrap getTemp to trim from the head.
+  const service = ctx.chatluna_character;
+  if (service && typeof service.getTemp === 'function') {
+    const originalGetTemp = service.getTemp.bind(service);
+    service.getTemp = async function patchedGetTemp(session) {
+      const temp = await originalGetTemp(session);
+      const limit = () => {
+        const c = service._config?.modelCompletionCount ?? 3;
+        return Math.max(2, c * 2);
+      };
+      if (Array.isArray(temp.completionMessages) && !temp.completionMessages._thinkViewerPatched) {
+        const arr = temp.completionMessages;
+        const originalPush = arr.push;
+        arr.push = function patchedPush(...args) {
+          const res = originalPush.apply(this, args);
+          const max = limit();
+          if (this.length > max) {
+            this.splice(0, this.length - max);
+          }
+          return res;
+        };
+        Object.defineProperty(arr, '_thinkViewerPatched', { value: true, enumerable: false });
+      }
+      return temp;
+    };
+
+    ctx.on('dispose', () => {
+      service.getTemp = originalGetTemp;
+    });
+  }
+}
+
+module.exports = {
   name,
   apply,
   Config,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "koishi-plugin-chatluna-think-viewer",
-  "version": "2.2.5",
+  "version": "2.2.6",
   "main": "index.js",
   "description": "通过命令/关键词查看 chatluna-character 最近一次回复的 <think> 思考内容，并在消息格式异常时自动拦截/撤回。",
   "license": "MIT",