getcompanion-ai/co-mono
1
0
Fork 0
mirror of https://github.com/getcompanion-ai/co-mono.git synced 2026-04-16 18:03:50 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix auto-compaction TUI integration and cut point logic

Browse source 
- Trigger auto-compaction after agent_end instead of during message_end
- Show CompactionComponent after auto-compaction (same as manual /compact)
- Fix cut point to include bash executions before kept user message
- Stop backward scan at compaction, assistant, user, or toolResult boundaries
This commit is contained in:
Mario Zechner 2025-12-09 02:45:24 +01:00
parent 75c2eea151
commit 4227fd5996
3 changed files with 46 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

35
packages/coding-agent/src/core/agent-session.ts
View file

@ -135,6 +135,9 @@ export class AgentSession {
}
}
// Track last assistant message for auto-compaction check
private _lastAssistantMessage: AssistantMessage | null = null;
/** Internal handler for agent events - shared by subscribe and reconnect */
private _handleAgentEvent = async (event: AgentEvent): Promise<void> => {
// Notify all listeners
@ -149,11 +152,18 @@ export class AgentSession {
this.sessionManager.startSession(this.agent.state);
}
// Check auto-compaction after assistant messages
// Track assistant message for auto-compaction (checked on agent_end)
if (event.message.role === "assistant") {
await this._runAutoCompaction();
this._lastAssistantMessage = event.message as AssistantMessage;
}
}
// Check auto-compaction after agent completes (after agent_end clears UI)
if (event.type === "agent_end" && this._lastAssistantMessage) {
const msg = this._lastAssistantMessage;
this._lastAssistantMessage = null;
this._runAutoCompaction(msg).catch(() => {});
}
};
/**
@ -584,26 +594,14 @@ export class AgentSession {
* Internal: Run auto-compaction with events.
* Called after assistant messages complete.
*/
private async _runAutoCompaction(): Promise<void> {
private async _runAutoCompaction(assistantMessage: AssistantMessage): Promise<void> {
const settings = this.settingsManager.getCompactionSettings();
if (!settings.enabled) return;
// Get last non-aborted assistant message
const messages = this.messages;
let lastAssistant: AssistantMessage | null = null;
for (let i = messages.length - 1; i >= 0; i--) {
const msg = messages[i];
if (msg.role === "assistant") {
const assistantMsg = msg as AssistantMessage;
if (assistantMsg.stopReason !== "aborted") {
lastAssistant = assistantMsg;
break;
}
}
}
if (!lastAssistant) return;
// Skip if message was aborted
if (assistantMessage.stopReason === "aborted") return;
const contextTokens = calculateContextTokens(lastAssistant.usage);
const contextTokens = calculateContextTokens(assistantMessage.usage);
const contextWindow = this.model?.contextWindow ?? 0;
if (!shouldCompact(contextTokens, contextWindow, settings)) return;
@ -624,6 +622,7 @@ export class AgentSession {
return;
}
// Load entries (sync file read) then yield to let UI render
const entries = this.sessionManager.loadEntries();
const compactionEntry = await compact(
entries,