koishi-plugin-chatluna 1.3.4 → 1.3.6

package/lib/chains/chain.d.ts

@@ -46,7 +46,7 @@ export declare class ChainMiddleware {
     constructor(name: string, execute: ChainMiddlewareFunction, graph: ChatChainDependencyGraph);
     before<T extends keyof ChainMiddlewareName>(name: T): this;
     after<T extends keyof ChainMiddlewareName>(name: T): this;
-    run(session: Session, options: ChainMiddlewareContext): Promise<string | ChainMiddlewareRunStatus | h[] | h[][]>;
+    run(session: Session, options: ChainMiddlewareContext): Promise<string | h[] | ChainMiddlewareRunStatus | h[][]>;
 }
 export interface ChainMiddlewareContext {
     config: Config;

package/lib/index.cjs

@@ -1046,9 +1046,13 @@ var ChatLunaChatChain = class _ChatLunaChatChain extends import_base.ChatLunaLLM
     requests["variables"] = Object.assign(variables ?? {}, {
       prompt: (0, import_string.getMessageContent)(message.content)
     });
+    requests["variables"]["built"] = {
+      conversationId
+    };
     requests["variables_hide"] = requests["variables"];
     requests["configurable"] = {
-      session
+      session,
+      conversationId
     };
     requests["id"] = conversationId;
     const response = await (0, import_base.callChatLunaChain)(
@@ -1185,12 +1189,16 @@ var ChatLunaPluginChain = class _ChatLunaPluginChain extends import_base2.ChatLu
     requests["variables"] = Object.assign(variables ?? {}, {
       prompt: (0, import_string2.getMessageContent)(message.content)
     });
+    requests["variables"]["built"] = {
+      conversationId
+    };
     requests["after_user_message"] = new import_messages2.HumanMessage(
       AGENT_AFTER_USER_PROMPT
     );
     requests["variables_hide"] = requests["variables"];
     requests["configurable"] = {
-      session
+      session,
+      conversationId
     };
     this._toolsRef.update(session, messages.concat(message));
     const preset = this.preset.value;
@@ -5816,6 +5824,7 @@ var import_crypto7 = require("crypto");
 var logger11;
 function apply56(ctx, config, chain) {
   logger11 = (0, import_logger7.createLogger)(ctx);
+  const selectRoomForSession = /* @__PURE__ */ __name(async (session, joinedRooms) => pickContextualRoom(ctx, session, config, joinedRooms), "selectRoomForSession");
   chain.middleware("resolve_room", async (session, context) => {
     let joinRoom = await queryJoinedConversationRoom(
       ctx,
@@ -5858,32 +5867,16 @@ function apply56(ctx, config, chain) {
         session
       );
       if (joinedRooms.length > 0) {
-        joinRoom = // 优先加入自己创建的房间
-        joinedRooms.find(
-          (room) => room.visibility === "private" && room.roomMasterId === session.userId
-        ) ?? // 优先加入自己创建的房间
-        joinedRooms.find(
-          (room) => room.visibility === "template_clone" && room.roomMasterId === session.userId
+        joinRoom = await selectRoomForSession(session, joinedRooms);
+      }
+      if (joinRoom != null) {
+        await switchConversationRoom(ctx, session, joinRoom.roomId);
+        logger11.success(
+          session.text("chatluna.room.auto_switch", [
+            session.userId,
+            joinRoom.roomName
+          ])
         );
-        if (config.autoCreateRoomFromUser !== true && joinRoom == null) {
-          joinRoom = // 优先加入模版克隆房间
-          joinedRooms.find(
-            (room) => room.visibility === "template_clone"
-          ) ?? joinedRooms[Math.floor(Math.random() * joinedRooms.length)];
-        }
-        if (joinRoom != null) {
-          await switchConversationRoom(
-            ctx,
-            session,
-            joinRoom.roomId
-          );
-          logger11.success(
-            session.text("chatluna.room.auto_switch", [
-              session.userId,
-              joinRoom.roomName
-            ])
-          );
-        }
       }
     }
     if (joinRoom == null && config.autoCreateRoomFromUser !== true && !session.isDirect && (context.command?.length ?? 0) < 1) {
@@ -5979,6 +5972,34 @@ function apply56(ctx, config, chain) {
   }).after("lifecycle-prepare");
 }
 __name(apply56, "apply");
+async function pickContextualRoom(ctx, session, config, joinedRooms) {
+  if (joinedRooms.length === 0) {
+    return void 0;
+  }
+  if (!session.isDirect && config.autoCreateRoomFromUser !== true) {
+    const groupRooms = await ctx.database.get("chathub_room_group_member", {
+      groupId: session.guildId,
+      roomId: { $in: joinedRooms.map((room) => room.roomId) }
+    });
+    if (groupRooms.length > 0) {
+      const scopedRoomIds = new Set(groupRooms.map((room) => room.roomId));
+      const findScopedRoom = /* @__PURE__ */ __name((matcher) => joinedRooms.find((room) => {
+        if (!scopedRoomIds.has(room.roomId)) {
+          return false;
+        }
+        return matcher ? matcher(room) : true;
+      }), "findScopedRoom");
+      return findScopedRoom(
+        (room) => room.visibility === "template_clone"
+      ) ?? findScopedRoom();
+    }
+    return void 0;
+  }
+  return joinedRooms.find(
+    (room) => room.visibility === "private" && room.roomMasterId === session.userId
+  ) ?? joinedRooms.find((room) => room.visibility === "private") ?? joinedRooms.find((room) => room.visibility !== "template_clone") ?? joinedRooms[0];
+}
+__name(pickContextualRoom, "pickContextualRoom");
 
 // src/middlewares/room/room_info.ts
 function apply57(ctx, config, chain) {

package/lib/index.mjs

@@ -1021,9 +1021,13 @@ var ChatLunaChatChain = class _ChatLunaChatChain extends ChatLunaLLMChainWrapper
     requests["variables"] = Object.assign(variables ?? {}, {
       prompt: getMessageContent(message.content)
     });
+    requests["variables"]["built"] = {
+      conversationId
+    };
     requests["variables_hide"] = requests["variables"];
     requests["configurable"] = {
-      session
+      session,
+      conversationId
     };
     requests["id"] = conversationId;
     const response = await callChatLunaChain(
@@ -1168,12 +1172,16 @@ var ChatLunaPluginChain = class _ChatLunaPluginChain extends ChatLunaLLMChainWra
     requests["variables"] = Object.assign(variables ?? {}, {
       prompt: getMessageContent2(message.content)
     });
+    requests["variables"]["built"] = {
+      conversationId
+    };
     requests["after_user_message"] = new HumanMessage(
       AGENT_AFTER_USER_PROMPT
     );
     requests["variables_hide"] = requests["variables"];
     requests["configurable"] = {
-      session
+      session,
+      conversationId
     };
     this._toolsRef.update(session, messages.concat(message));
     const preset = this.preset.value;
@@ -5826,6 +5834,7 @@ import { randomUUID as randomUUID4 } from "crypto";
 var logger11;
 function apply56(ctx, config, chain) {
   logger11 = createLogger7(ctx);
+  const selectRoomForSession = /* @__PURE__ */ __name(async (session, joinedRooms) => pickContextualRoom(ctx, session, config, joinedRooms), "selectRoomForSession");
   chain.middleware("resolve_room", async (session, context) => {
     let joinRoom = await queryJoinedConversationRoom(
       ctx,
@@ -5868,32 +5877,16 @@ function apply56(ctx, config, chain) {
         session
       );
       if (joinedRooms.length > 0) {
-        joinRoom = // 优先加入自己创建的房间
-        joinedRooms.find(
-          (room) => room.visibility === "private" && room.roomMasterId === session.userId
-        ) ?? // 优先加入自己创建的房间
-        joinedRooms.find(
-          (room) => room.visibility === "template_clone" && room.roomMasterId === session.userId
+        joinRoom = await selectRoomForSession(session, joinedRooms);
+      }
+      if (joinRoom != null) {
+        await switchConversationRoom(ctx, session, joinRoom.roomId);
+        logger11.success(
+          session.text("chatluna.room.auto_switch", [
+            session.userId,
+            joinRoom.roomName
+          ])
         );
-        if (config.autoCreateRoomFromUser !== true && joinRoom == null) {
-          joinRoom = // 优先加入模版克隆房间
-          joinedRooms.find(
-            (room) => room.visibility === "template_clone"
-          ) ?? joinedRooms[Math.floor(Math.random() * joinedRooms.length)];
-        }
-        if (joinRoom != null) {
-          await switchConversationRoom(
-            ctx,
-            session,
-            joinRoom.roomId
-          );
-          logger11.success(
-            session.text("chatluna.room.auto_switch", [
-              session.userId,
-              joinRoom.roomName
-            ])
-          );
-        }
       }
     }
     if (joinRoom == null && config.autoCreateRoomFromUser !== true && !session.isDirect && (context.command?.length ?? 0) < 1) {
@@ -5989,6 +5982,34 @@ function apply56(ctx, config, chain) {
   }).after("lifecycle-prepare");
 }
 __name(apply56, "apply");
+async function pickContextualRoom(ctx, session, config, joinedRooms) {
+  if (joinedRooms.length === 0) {
+    return void 0;
+  }
+  if (!session.isDirect && config.autoCreateRoomFromUser !== true) {
+    const groupRooms = await ctx.database.get("chathub_room_group_member", {
+      groupId: session.guildId,
+      roomId: { $in: joinedRooms.map((room) => room.roomId) }
+    });
+    if (groupRooms.length > 0) {
+      const scopedRoomIds = new Set(groupRooms.map((room) => room.roomId));
+      const findScopedRoom = /* @__PURE__ */ __name((matcher) => joinedRooms.find((room) => {
+        if (!scopedRoomIds.has(room.roomId)) {
+          return false;
+        }
+        return matcher ? matcher(room) : true;
+      }), "findScopedRoom");
+      return findScopedRoom(
+        (room) => room.visibility === "template_clone"
+      ) ?? findScopedRoom();
+    }
+    return void 0;
+  }
+  return joinedRooms.find(
+    (room) => room.visibility === "private" && room.roomMasterId === session.userId
+  ) ?? joinedRooms.find((room) => room.visibility === "private") ?? joinedRooms.find((room) => room.visibility !== "template_clone") ?? joinedRooms[0];
+}
+__name(pickContextualRoom, "pickContextualRoom");
 
 // src/middlewares/room/room_info.ts
 function apply57(ctx, config, chain) {

package/lib/llm-core/agent/openai/index.d.ts

@@ -22,4 +22,4 @@ export type CreateOpenAIAgentParams = {
 };
 export declare function createOpenAIAgent({ llm, tools, prompt }: CreateOpenAIAgentParams): RunnableSequence<{
     steps: AgentStep[];
-}, AgentAction | AgentAction[] | AgentFinish>;
+}, AgentAction | AgentFinish | AgentAction[]>;

package/lib/llm-core/chat/app.cjs

@@ -188,12 +188,40 @@ var InfiniteContextManager = class {
     import_koishi_plugin_chatluna.logger.info(
       `[InfiniteContext] Start compression with total tokens: ${totalTokens}, threshold: ${threshold}`
     );
+    const filteredMessages = messages.filter(
+      (message) => !this._isToolRelatedMessage(message)
+    );
+    if (filteredMessages.length === 0) {
+      return;
+    }
+    const filteredStats = await this._calculateMessageTokenStats(
+      model,
+      filteredMessages
+    );
+    const filteredTotalTokens = filteredStats.reduce((sum, current) => sum + current.tokens, 0) + presetTokens;
+    if (filteredTotalTokens <= threshold) {
+      await this._rewriteChatHistory(filteredMessages);
+      import_koishi_plugin_chatluna.logger.info(
+        "[InfiniteContext] Filtered tool-related messages reduced tokens from %d to %d",
+        totalTokens,
+        filteredTotalTokens
+      );
+      return;
+    }
     const compressionResult = await this._compressMessages(
       wrapper,
-      stats,
+      filteredStats,
       maxTokenLimit
     );
     if (!compressionResult) {
+      if (filteredMessages.length !== messages.length) {
+        await this._rewriteChatHistory(filteredMessages);
+        import_koishi_plugin_chatluna.logger.info(
+          "[InfiniteContext] Filtered tool-related messages (compression skipped) reduced tokens from %d to %d",
+          totalTokens,
+          filteredTotalTokens
+        );
+      }
       return;
     }
     const { messages: rewrittenMessages, tokenCount } = compressionResult;
@@ -246,12 +274,9 @@ var InfiniteContextManager = class {
         0,
         contentStats.length - preserveCount
       );
-      compressible = contentStats.filter((stat, index) => {
-        if (this._isCompressedMessage(stat.message)) {
-          return false;
-        }
-        return index < thresholdIndex;
-      });
+      compressible = contentStats.filter(
+        (stat, index) => index < thresholdIndex
+      );
       if (compressible.length > 0) {
         break;
       }
@@ -272,10 +297,16 @@ var InfiniteContextManager = class {
       return null;
     }
     const compressor = this._ensureInfiniteContextChain(wrapper);
-    const compressedSegments = [];
+    const chunkSummaries = [];
+    const previousSummaries = compressible.filter((stat) => this._isCompressedMessage(stat.message)).map(
+      (stat) => (0, import_string2.getMessageContent)(stat.message.content)?.trim() ?? ""
+    ).filter((text) => text.length > 0);
     for (let index = 0; index < chunkStats.length; index++) {
       const chunk = chunkStats[index];
-      const chunkMessages = chunk.map((item) => item.message);
+      const chunkMessages = chunk.map((item) => item.message).filter((message) => !this._isCompressedMessage(message));
+      if (chunkMessages.length === 0) {
+        continue;
+      }
       const chunkText = this._formatChunkForCompression(chunkMessages);
       const compressedText = await compressor.compressChunk({
         chunk: chunkText,
@@ -284,40 +315,28 @@ var InfiniteContextManager = class {
       if (!compressedText) {
         continue;
       }
-      compressedSegments.push({
+      chunkSummaries.push({
         content: compressedText,
         chunkIndex: index,
         chunkSize: chunkMessages.length
       });
     }
-    if (compressedSegments.length === 0) {
+    if (chunkSummaries.length === 0 && previousSummaries.length === 0) {
+      return null;
+    }
+    const finalSummary = await this._buildFinalSummary(
+      compressor,
+      previousSummaries,
+      chunkSummaries
+    );
+    if (!finalSummary) {
       return null;
     }
-    const aggregatedSummary = compressedSegments.map(
-      (segment, idx) => `### Segment ${idx + 1}
-${segment.content.trim()}`
-    ).join("\n\n");
-    const summaryMeta = {
-      source: "infinite-context",
-      segments: compressedSegments.length,
-      segmentDetail: compressedSegments.map((segment, idx) => ({
-        segment: idx + 1,
-        chunkIndex: segment.chunkIndex,
-        chunkSize: segment.chunkSize
-      }))
-    };
     const compressedMessages = [
       new import_messages.HumanMessage({
-        content: `<infinite_context segments="${compressedSegments.length}">
-${aggregatedSummary}
-</infinite_context>`,
+        content: finalSummary.content,
         name: "infinite_context",
-        additional_kwargs: summaryMeta
-      }),
-      new import_messages.AIMessage({
-        content: "Summary acknowledged. I will prioritise available tools when future turns involve these archived topics, and if no tools are available I will note the memory gap and ask the user to confirm the details.",
-        name: "infinite_context_ack",
-        additional_kwargs: summaryMeta
+        additional_kwargs: finalSummary.meta
       })
     ];
     const mergedMessages = [
@@ -345,6 +364,32 @@ ${aggregatedSummary}
     const content = (0, import_string2.getMessageContent)(message?.content ?? "");
     return typeof content === "string" && /<\/?infinite_context/iu.test(content);
   }
+  _isToolRelatedMessage(message) {
+    if (message.getType() === "tool") {
+      return true;
+    }
+    const anyMessage = message;
+    if (Array.isArray(anyMessage.tool_calls) && anyMessage.tool_calls.length > 0) {
+      return true;
+    }
+    const additionalToolCalls = anyMessage.additional_kwargs?.["tool_calls"];
+    return Array.isArray(additionalToolCalls) && additionalToolCalls.length > 0;
+  }
+  async _rewriteChatHistory(messages) {
+    const additionalArgs = {
+      ...await this.options.chatHistory.getAdditionalArgs()
+    };
+    await this.options.chatHistory.clear();
+    for (const message of messages) {
+      await this.options.chatHistory.addMessage(message);
+    }
+    if (Object.keys(additionalArgs).length > 0) {
+      await this.options.chatHistory.overrideAdditionalArgs(
+        additionalArgs
+      );
+    }
+    await this.options.chatHistory.loadConversation();
+  }
   _splitChunksForCompression(stats, maxTokenLimit) {
     const chunkTokenTarget = Math.max(Math.floor(maxTokenLimit * 0.15), 300);
     const chunks = [];
@@ -384,33 +429,8 @@ ${aggregatedSummary}
       const role = message.getType().toUpperCase();
       const nameSuffix = message.name ? ` (${message.name})` : "";
       const content = (0, import_string2.getMessageContent)(message.content).trim();
-      const extras = [];
-      const aiMessage = message;
-      if (Array.isArray(aiMessage.tool_calls) && aiMessage.tool_calls.length > 0) {
-        const toolCalls = aiMessage.tool_calls.map((call) => {
-          let args = "";
-          try {
-            args = JSON.stringify(call.args);
-          } catch (error) {
-            args = "[unserializable]";
-          }
-          return `${call.name} => ${args}`;
-        }).join("; ");
-        extras.push(`tool calls: ${toolCalls}`);
-      }
-      if (message.getType() === "tool") {
-        const toolMessage = message;
-        if (toolMessage.tool_call_id) {
-          extras.push(`tool call id: ${toolMessage.tool_call_id}`);
-        }
-        if (toolMessage.name) {
-          extras.push(`tool name: ${toolMessage.name}`);
-        }
-      }
-      const extraText = extras.length > 0 ? `
-Meta: ${extras.join(", ")}` : "";
       return `[${role}${nameSuffix}]
-${content || "(empty)"}${extraText}`;
+${content || "(empty)"}`;
     }).join("\n---\n");
   }
   _ensureInfiniteContextChain(wrapper) {
@@ -421,6 +441,64 @@ ${content || "(empty)"}${extraText}`;
     }
     return this._chain;
   }
+  async _buildFinalSummary(compressor, previousSummaries, chunkSummaries) {
+    const cleanedPrevious = previousSummaries.filter(
+      (text) => text.trim().length > 0
+    );
+    const cleanedChunks = chunkSummaries.filter(
+      (summary) => summary.content.trim().length > 0
+    );
+    if (cleanedPrevious.length === 0 && cleanedChunks.length === 0) {
+      return null;
+    }
+    if (cleanedPrevious.length === 0 && cleanedChunks.length === 1) {
+      return {
+        content: cleanedChunks[0].content.trim(),
+        meta: {
+          source: "infinite-context",
+          mergedSegments: 1,
+          previousSummaries: 0,
+          chunkDetail: [
+            {
+              chunkIndex: cleanedChunks[0].chunkIndex,
+              chunkSize: cleanedChunks[0].chunkSize
+            }
+          ]
+        }
+      };
+    }
+    const virtualTranscript = [];
+    if (cleanedPrevious.length > 0) {
+      virtualTranscript.push(
+        `Existing summary snapshot:
+${cleanedPrevious.join("\n\n")}`
+      );
+    }
+    cleanedChunks.forEach((chunk, index) => {
+      virtualTranscript.push(
+        `Recent segment ${index + 1} (${chunk.chunkSize} turns):
+${chunk.content}`
+      );
+    });
+    const mergedInput = virtualTranscript.join("\n\n---\n\n");
+    const refined = await compressor.compressChunk({
+      chunk: mergedInput,
+      conversationId: this.options.conversationId
+    });
+    const content = refined?.trim() || mergedInput;
+    return {
+      content,
+      meta: {
+        source: "infinite-context",
+        mergedSegments: cleanedChunks.length,
+        previousSummaries: cleanedPrevious.length,
+        chunkDetail: cleanedChunks.map((chunk) => ({
+          chunkIndex: chunk.chunkIndex,
+          chunkSize: chunk.chunkSize
+        }))
+      }
+    };
+  }
 };
 
 // src/llm-core/chat/app.ts
@@ -445,7 +523,7 @@ var ChatInterface = class {
   _historyMemory;
   _infiniteContextManager;
   _chatCount = 0;
-  async handleChatError(arg, wrapper, error) {
+  async handleChatError(arg, wrapper, error, throwError = true) {
     await this.ctx.parallel(
       "chatluna/after-chat-error",
       error,
@@ -455,6 +533,9 @@ var ChatInterface = class {
       this,
       wrapper
     );
+    if (!throwError) {
+      return;
+    }
     if (error instanceof import_error2.ChatLunaError && error.errorCode === import_error2.ChatLunaErrorCode.API_UNSAFE_CONTENT) {
       throw error;
     }
@@ -561,15 +642,19 @@ var ChatInterface = class {
       }
       await this.chatHistory.addMessage(saveMessage);
     }
-    this.ctx.parallel(
-      "chatluna/after-chat",
-      arg.conversationId,
-      arg.message,
-      displayResponse,
-      { ...arg.variables, chatCount: this._chatCount },
-      this,
-      arg.session
-    );
+    try {
+      await this.ctx.parallel(
+        "chatluna/after-chat",
+        arg.conversationId,
+        arg.message,
+        displayResponse,
+        { ...arg.variables, chatCount: this._chatCount },
+        this,
+        arg.session
+      );
+    } catch (error) {
+      await this.handleChatError(arg, wrapper, error, false);
+    }
     return { message: displayResponse };
   }
   async handlePostProcessing(arg, message) {

package/lib/llm-core/chat/app.mjs

@@ -20,15 +20,12 @@ import { ChatLunaChatModel } from "koishi-plugin-chatluna/llm-core/platform/mode
 import {
   ModelCapabilities
 } from "koishi-plugin-chatluna/llm-core/platform/types";
-import { AIMessage as AIMessage2, ToolMessage as ToolMessage2 } from "@langchain/core/messages";
+import { AIMessage, ToolMessage } from "@langchain/core/messages";
 import { getMessageContent as getMessageContent3 } from "koishi-plugin-chatluna/utils/string";
 import { computed } from "@vue/reactivity";
 
 // src/llm-core/chat/infinite_context.ts
-import {
-  AIMessage,
-  HumanMessage
-} from "@langchain/core/messages";
+import { HumanMessage } from "@langchain/core/messages";
 import { getMessageContent as getMessageContent2 } from "koishi-plugin-chatluna/utils/string";
 import { logger } from "koishi-plugin-chatluna";
 
@@ -185,12 +182,40 @@ var InfiniteContextManager = class {
     logger.info(
       `[InfiniteContext] Start compression with total tokens: ${totalTokens}, threshold: ${threshold}`
     );
+    const filteredMessages = messages.filter(
+      (message) => !this._isToolRelatedMessage(message)
+    );
+    if (filteredMessages.length === 0) {
+      return;
+    }
+    const filteredStats = await this._calculateMessageTokenStats(
+      model,
+      filteredMessages
+    );
+    const filteredTotalTokens = filteredStats.reduce((sum, current) => sum + current.tokens, 0) + presetTokens;
+    if (filteredTotalTokens <= threshold) {
+      await this._rewriteChatHistory(filteredMessages);
+      logger.info(
+        "[InfiniteContext] Filtered tool-related messages reduced tokens from %d to %d",
+        totalTokens,
+        filteredTotalTokens
+      );
+      return;
+    }
     const compressionResult = await this._compressMessages(
       wrapper,
-      stats,
+      filteredStats,
       maxTokenLimit
     );
     if (!compressionResult) {
+      if (filteredMessages.length !== messages.length) {
+        await this._rewriteChatHistory(filteredMessages);
+        logger.info(
+          "[InfiniteContext] Filtered tool-related messages (compression skipped) reduced tokens from %d to %d",
+          totalTokens,
+          filteredTotalTokens
+        );
+      }
       return;
     }
     const { messages: rewrittenMessages, tokenCount } = compressionResult;
@@ -243,12 +268,9 @@ var InfiniteContextManager = class {
         0,
         contentStats.length - preserveCount
       );
-      compressible = contentStats.filter((stat, index) => {
-        if (this._isCompressedMessage(stat.message)) {
-          return false;
-        }
-        return index < thresholdIndex;
-      });
+      compressible = contentStats.filter(
+        (stat, index) => index < thresholdIndex
+      );
       if (compressible.length > 0) {
         break;
       }
@@ -269,10 +291,16 @@ var InfiniteContextManager = class {
       return null;
     }
     const compressor = this._ensureInfiniteContextChain(wrapper);
-    const compressedSegments = [];
+    const chunkSummaries = [];
+    const previousSummaries = compressible.filter((stat) => this._isCompressedMessage(stat.message)).map(
+      (stat) => getMessageContent2(stat.message.content)?.trim() ?? ""
+    ).filter((text) => text.length > 0);
     for (let index = 0; index < chunkStats.length; index++) {
       const chunk = chunkStats[index];
-      const chunkMessages = chunk.map((item) => item.message);
+      const chunkMessages = chunk.map((item) => item.message).filter((message) => !this._isCompressedMessage(message));
+      if (chunkMessages.length === 0) {
+        continue;
+      }
       const chunkText = this._formatChunkForCompression(chunkMessages);
       const compressedText = await compressor.compressChunk({
         chunk: chunkText,
@@ -281,40 +309,28 @@ var InfiniteContextManager = class {
       if (!compressedText) {
         continue;
       }
-      compressedSegments.push({
+      chunkSummaries.push({
         content: compressedText,
         chunkIndex: index,
         chunkSize: chunkMessages.length
       });
     }
-    if (compressedSegments.length === 0) {
+    if (chunkSummaries.length === 0 && previousSummaries.length === 0) {
+      return null;
+    }
+    const finalSummary = await this._buildFinalSummary(
+      compressor,
+      previousSummaries,
+      chunkSummaries
+    );
+    if (!finalSummary) {
       return null;
     }
-    const aggregatedSummary = compressedSegments.map(
-      (segment, idx) => `### Segment ${idx + 1}
-${segment.content.trim()}`
-    ).join("\n\n");
-    const summaryMeta = {
-      source: "infinite-context",
-      segments: compressedSegments.length,
-      segmentDetail: compressedSegments.map((segment, idx) => ({
-        segment: idx + 1,
-        chunkIndex: segment.chunkIndex,
-        chunkSize: segment.chunkSize
-      }))
-    };
     const compressedMessages = [
       new HumanMessage({
-        content: `<infinite_context segments="${compressedSegments.length}">
-${aggregatedSummary}
-</infinite_context>`,
+        content: finalSummary.content,
         name: "infinite_context",
-        additional_kwargs: summaryMeta
-      }),
-      new AIMessage({
-        content: "Summary acknowledged. I will prioritise available tools when future turns involve these archived topics, and if no tools are available I will note the memory gap and ask the user to confirm the details.",
-        name: "infinite_context_ack",
-        additional_kwargs: summaryMeta
+        additional_kwargs: finalSummary.meta
       })
     ];
     const mergedMessages = [
@@ -342,6 +358,32 @@ ${aggregatedSummary}
     const content = getMessageContent2(message?.content ?? "");
     return typeof content === "string" && /<\/?infinite_context/iu.test(content);
   }
+  _isToolRelatedMessage(message) {
+    if (message.getType() === "tool") {
+      return true;
+    }
+    const anyMessage = message;
+    if (Array.isArray(anyMessage.tool_calls) && anyMessage.tool_calls.length > 0) {
+      return true;
+    }
+    const additionalToolCalls = anyMessage.additional_kwargs?.["tool_calls"];
+    return Array.isArray(additionalToolCalls) && additionalToolCalls.length > 0;
+  }
+  async _rewriteChatHistory(messages) {
+    const additionalArgs = {
+      ...await this.options.chatHistory.getAdditionalArgs()
+    };
+    await this.options.chatHistory.clear();
+    for (const message of messages) {
+      await this.options.chatHistory.addMessage(message);
+    }
+    if (Object.keys(additionalArgs).length > 0) {
+      await this.options.chatHistory.overrideAdditionalArgs(
+        additionalArgs
+      );
+    }
+    await this.options.chatHistory.loadConversation();
+  }
   _splitChunksForCompression(stats, maxTokenLimit) {
     const chunkTokenTarget = Math.max(Math.floor(maxTokenLimit * 0.15), 300);
     const chunks = [];
@@ -381,33 +423,8 @@ ${aggregatedSummary}
       const role = message.getType().toUpperCase();
       const nameSuffix = message.name ? ` (${message.name})` : "";
       const content = getMessageContent2(message.content).trim();
-      const extras = [];
-      const aiMessage = message;
-      if (Array.isArray(aiMessage.tool_calls) && aiMessage.tool_calls.length > 0) {
-        const toolCalls = aiMessage.tool_calls.map((call) => {
-          let args = "";
-          try {
-            args = JSON.stringify(call.args);
-          } catch (error) {
-            args = "[unserializable]";
-          }
-          return `${call.name} => ${args}`;
-        }).join("; ");
-        extras.push(`tool calls: ${toolCalls}`);
-      }
-      if (message.getType() === "tool") {
-        const toolMessage = message;
-        if (toolMessage.tool_call_id) {
-          extras.push(`tool call id: ${toolMessage.tool_call_id}`);
-        }
-        if (toolMessage.name) {
-          extras.push(`tool name: ${toolMessage.name}`);
-        }
-      }
-      const extraText = extras.length > 0 ? `
-Meta: ${extras.join(", ")}` : "";
       return `[${role}${nameSuffix}]
-${content || "(empty)"}${extraText}`;
+${content || "(empty)"}`;
     }).join("\n---\n");
   }
   _ensureInfiniteContextChain(wrapper) {
@@ -418,6 +435,64 @@ ${content || "(empty)"}${extraText}`;
     }
     return this._chain;
   }
+  async _buildFinalSummary(compressor, previousSummaries, chunkSummaries) {
+    const cleanedPrevious = previousSummaries.filter(
+      (text) => text.trim().length > 0
+    );
+    const cleanedChunks = chunkSummaries.filter(
+      (summary) => summary.content.trim().length > 0
+    );
+    if (cleanedPrevious.length === 0 && cleanedChunks.length === 0) {
+      return null;
+    }
+    if (cleanedPrevious.length === 0 && cleanedChunks.length === 1) {
+      return {
+        content: cleanedChunks[0].content.trim(),
+        meta: {
+          source: "infinite-context",
+          mergedSegments: 1,
+          previousSummaries: 0,
+          chunkDetail: [
+            {
+              chunkIndex: cleanedChunks[0].chunkIndex,
+              chunkSize: cleanedChunks[0].chunkSize
+            }
+          ]
+        }
+      };
+    }
+    const virtualTranscript = [];
+    if (cleanedPrevious.length > 0) {
+      virtualTranscript.push(
+        `Existing summary snapshot:
+${cleanedPrevious.join("\n\n")}`
+      );
+    }
+    cleanedChunks.forEach((chunk, index) => {
+      virtualTranscript.push(
+        `Recent segment ${index + 1} (${chunk.chunkSize} turns):
+${chunk.content}`
+      );
+    });
+    const mergedInput = virtualTranscript.join("\n\n---\n\n");
+    const refined = await compressor.compressChunk({
+      chunk: mergedInput,
+      conversationId: this.options.conversationId
+    });
+    const content = refined?.trim() || mergedInput;
+    return {
+      content,
+      meta: {
+        source: "infinite-context",
+        mergedSegments: cleanedChunks.length,
+        previousSummaries: cleanedPrevious.length,
+        chunkDetail: cleanedChunks.map((chunk) => ({
+          chunkIndex: chunk.chunkIndex,
+          chunkSize: chunk.chunkSize
+        }))
+      }
+    };
+  }
 };
 
 // src/llm-core/chat/app.ts
@@ -442,7 +517,7 @@ var ChatInterface = class {
   _historyMemory;
   _infiniteContextManager;
   _chatCount = 0;
-  async handleChatError(arg, wrapper, error) {
+  async handleChatError(arg, wrapper, error, throwError = true) {
     await this.ctx.parallel(
       "chatluna/after-chat-error",
       error,
@@ -452,6 +527,9 @@ var ChatInterface = class {
       this,
       wrapper
     );
+    if (!throwError) {
+      return;
+    }
     if (error instanceof ChatLunaError2 && error.errorCode === ChatLunaErrorCode2.API_UNSAFE_CONTENT) {
       throw error;
     }
@@ -510,7 +588,7 @@ var ChatInterface = class {
       maxToken: this.preset?.value?.config?.maxOutputToken
     });
     const responseMessage = response.message;
-    const displayResponse = new AIMessage2({
+    const displayResponse = new AIMessage({
       content: responseMessage.content
     });
     displayResponse.additional_kwargs = responseMessage.additional_kwargs;
@@ -536,7 +614,7 @@ var ChatInterface = class {
         const intermediateSteps = response["parallelIntermediateSteps"];
         for (const parallelSteps of intermediateSteps) {
           await this.chatHistory.addMessage(
-            new AIMessage2({
+            new AIMessage({
               content: "",
               tool_calls: parallelSteps.map((step) => ({
                 id: step.action.toolCallId,
@@ -547,7 +625,7 @@ var ChatInterface = class {
           );
           for (const step of parallelSteps) {
             await this.chatHistory.addMessage(
-              new ToolMessage2({
+              new ToolMessage({
                 content: step.observation,
                 tool_call_id: step.action.toolCallId,
                 name: step.action.tool
@@ -558,15 +636,19 @@ var ChatInterface = class {
       }
       await this.chatHistory.addMessage(saveMessage);
     }
-    this.ctx.parallel(
-      "chatluna/after-chat",
-      arg.conversationId,
-      arg.message,
-      displayResponse,
-      { ...arg.variables, chatCount: this._chatCount },
-      this,
-      arg.session
-    );
+    try {
+      await this.ctx.parallel(
+        "chatluna/after-chat",
+        arg.conversationId,
+        arg.message,
+        displayResponse,
+        { ...arg.variables, chatCount: this._chatCount },
+        this,
+        arg.session
+      );
+    } catch (error) {
+      await this.handleChatError(arg, wrapper, error, false);
+    }
     return { message: displayResponse };
   }
   async handlePostProcessing(arg, message) {

package/lib/llm-core/chat/infinite_context.d.ts

@@ -14,9 +14,12 @@ export declare class InfiniteContextManager {
     compressIfNeeded(wrapper: ChatLunaLLMChainWrapper): Promise<void>;
     private _compressMessages;
     private _isCompressedMessage;
+    private _isToolRelatedMessage;
+    private _rewriteChatHistory;
     private _splitChunksForCompression;
     private _calculateMessageTokenStats;
     private _countMessagesTokens;
     private _formatChunkForCompression;
     private _ensureInfiniteContextChain;
+    private _buildFinalSummary;
 }

package/lib/llm-core/platform/model.cjs

@@ -234,6 +234,10 @@ var ChatLunaChatModel = class extends import_chat_models.BaseChatModel {
     if (maxTokenLimit != null && maxTokenLimit >= this.getModelMaxContextSize()) {
       maxTokenLimit = this.getModelMaxContextSize();
     }
+    let id = options?.id ?? this._options.id;
+    if (!id) {
+      id = options?.variables_hide?.["built"]?.["conversationId"];
+    }
     return {
       model: modelName,
       temperature: options?.temperature ?? this._options.temperature,
@@ -244,12 +248,12 @@ var ChatLunaChatModel = class extends import_chat_models.BaseChatModel {
       logitBias: options?.logitBias ?? this._options.logitBias,
       maxTokens: options?.maxTokens ?? this._options.maxTokens,
       maxTokenLimit,
-      variables: options?.["variables_hide"] ?? {},
+      variables: options?.["variables_hide"] ?? options?.["variables"] ?? {},
       overrideRequestParams: options?.overrideRequestParams ?? this._options.overrideRequestParams ?? {},
       stop: options?.stop ?? this._options.stop,
       stream: options?.stream ?? this._options.stream,
       tools: options?.tools ?? this._options.tools,
-      id: options?.id ?? this._options.id,
+      id,
       signal: options?.signal ?? this._options.signal,
       timeout: options?.timeout ?? this._options.timeout
     };

package/lib/llm-core/platform/model.mjs

@@ -216,6 +216,10 @@ var ChatLunaChatModel = class extends BaseChatModel {
     if (maxTokenLimit != null && maxTokenLimit >= this.getModelMaxContextSize()) {
       maxTokenLimit = this.getModelMaxContextSize();
     }
+    let id = options?.id ?? this._options.id;
+    if (!id) {
+      id = options?.variables_hide?.["built"]?.["conversationId"];
+    }
     return {
       model: modelName,
       temperature: options?.temperature ?? this._options.temperature,
@@ -226,12 +230,12 @@ var ChatLunaChatModel = class extends BaseChatModel {
       logitBias: options?.logitBias ?? this._options.logitBias,
       maxTokens: options?.maxTokens ?? this._options.maxTokens,
       maxTokenLimit,
-      variables: options?.["variables_hide"] ?? {},
+      variables: options?.["variables_hide"] ?? options?.["variables"] ?? {},
       overrideRequestParams: options?.overrideRequestParams ?? this._options.overrideRequestParams ?? {},
       stop: options?.stop ?? this._options.stop,
       stream: options?.stream ?? this._options.stream,
       tools: options?.tools ?? this._options.tools,
-      id: options?.id ?? this._options.id,
+      id,
       signal: options?.signal ?? this._options.signal,
       timeout: options?.timeout ?? this._options.timeout
     };

package/lib/utils/string.cjs

@@ -330,6 +330,7 @@ function getSystemPromptVariables(session, config, room) {
     group_id: session.guildId ?? session.event?.guild?.id,
     group_name: session.event?.guild?.name || session.guildId,
     user_id: session.author?.user?.id ?? session.event?.user?.id ?? session.userId ?? "0",
+    platform: session.platform,
     user: getNotEmptyString(
       session.author?.nick,
       session.author?.name,
@@ -338,6 +339,7 @@ function getSystemPromptVariables(session, config, room) {
     ),
     built: {
       preset: room.preset,
+      platform: session.platform,
       conversationId: room.conversationId
     },
     noop: "",

package/lib/utils/string.d.ts

@@ -45,9 +45,11 @@ export declare function getSystemPromptVariables(session: Session, config: Confi
     group_id: string;
     group_name: string;
     user_id: string;
+    platform: string;
     user: string;
     built: {
         preset: string;
+        platform: string;
         conversationId: string;
     };
     noop: string;

package/lib/utils/string.mjs

@@ -277,6 +277,7 @@ function getSystemPromptVariables(session, config, room) {
     group_id: session.guildId ?? session.event?.guild?.id,
     group_name: session.event?.guild?.name || session.guildId,
     user_id: session.author?.user?.id ?? session.event?.user?.id ?? session.userId ?? "0",
+    platform: session.platform,
     user: getNotEmptyString(
       session.author?.nick,
       session.author?.name,
@@ -285,6 +286,7 @@ function getSystemPromptVariables(session, config, room) {
     ),
     built: {
       preset: room.preset,
+      platform: session.platform,
       conversationId: room.conversationId
     },
     noop: "",

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-chatluna",
   "description": "chatluna for koishi",
-  "version": "1.3.4",
+  "version": "1.3.6",
   "main": "lib/index.cjs",
   "module": "lib/index.mjs",
   "typings": "lib/index.d.ts",