@selesai/code 0.2.0 → 0.3.0

package/dist/core/agent-session.js

@@ -19,6 +19,7 @@ import { getThemeByName, theme } from "../modes/interactive/theme/theme.js";
 import { stripFrontmatter } from "../utils/frontmatter.js";
 import { resolvePath } from "../utils/paths.js";
 import { sleep } from "../utils/sleep.js";
+import { normalizeAssistantThinkingTags } from "../utils/thinking-tags.js";
 import { formatNoApiKeyFoundMessage, formatNoModelSelectedMessage } from "./auth-guidance.js";
 import { executeBashWithOperations } from "./bash-executor.js";
 import { calculateContextTokens, collectEntriesForBranchSummary, compact, estimateContextTokens, estimateTokens, generateBranchSummary, prepareCompaction, shouldCompact, } from "./compaction/index.js";
@@ -355,6 +356,12 @@ export class AgentSession {
         }
         Object.assign(targetRecord, replacement);
     }
+    _normalizeAssistantMessageInPlace(message) {
+        if (message.role !== "assistant")
+            return;
+        const normalized = normalizeAssistantThinkingTags(message);
+        this._replaceMessageInPlace(message, normalized);
+    }
     /** Emit extension events based on agent events */
     async _emitExtensionEvent(event) {
         if (event.type === "agent_start") {
@@ -398,6 +405,7 @@ export class AgentSession {
             await this._extensionRunner.emit(extensionEvent);
         }
         else if (event.type === "message_end") {
+            this._normalizeAssistantMessageInPlace(event.message);
             const extensionEvent = {
                 type: "message_end",
                 message: event.message,