getcompanion-ai/co-mono
1
0
Fork 0
mirror of https://github.com/getcompanion-ai/co-mono.git synced 2026-04-15 19:05:11 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Cleanup: unify HookMessage naming and simplify SessionContext

Browse source 
- Rename HookAppMessage to HookMessage, isHookAppMessage to isHookMessage
- Remove entries array from SessionContext (use isHookMessage type guard instead)
- HookMessage.content now accepts string directly (not just array)
- Fix streamMessage type in AgentState (AppMessage, not Message)
- Rename CustomMessageComponent to HookMessageComponent
- Fix test hook to use pi.sendMessage
This commit is contained in:
Mario Zechner 2025-12-27 20:52:12 +01:00
parent a2515cf43f
commit 204d27581b
13 changed files with 62 additions and 106 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -131,8 +131,6 @@ export interface SessionTreeNode {
export interface SessionContext {
messages: AppMessage[];
/** Entries in the current path (root to leaf). Use to identify custom_message entries for rendering. */
entries: SessionEntry[];
thinkingLevel: string;
model: { provider: string; modelId: string } | null;
}
@ -292,7 +290,7 @@ export function buildSessionContext(
}
if (!leaf) {
return { messages: [], entries: [], thinkingLevel: "off", model: null };
return { messages: [], thinkingLevel: "off", model: null };
}
// Walk from leaf to root, collecting path
@ -326,12 +324,10 @@ export function buildSessionContext(
// 2. Emit kept messages (from firstKeptEntryId up to compaction)
// 3. Emit messages after compaction
const messages: AppMessage[] = [];
const contextEntries: SessionEntry[] = [];
if (compaction) {
// Emit summary first
messages.push(createSummaryMessage(compaction.summary, compaction.timestamp));
contextEntries.push(compaction);
// Find compaction index in path
const compactionIdx = path.findIndex((e) => e.type === "compaction" && e.id === compaction.id);
@ -346,13 +342,10 @@ export function buildSessionContext(
if (foundFirstKept) {
if (entry.type === "message") {
messages.push(entry.message);
contextEntries.push(entry);
} else if (entry.type === "custom_message") {
messages.push(createCustomMessage(entry));
contextEntries.push(entry);
} else if (entry.type === "branch_summary") {
messages.push(createSummaryMessage(entry.summary, entry.timestamp));
contextEntries.push(entry);
}
}
}
@ -362,13 +355,10 @@ export function buildSessionContext(
const entry = path[i];
if (entry.type === "message") {
messages.push(entry.message);
contextEntries.push(entry);
} else if (entry.type === "custom_message") {
messages.push(createCustomMessage(entry));
contextEntries.push(entry);
} else if (entry.type === "branch_summary") {
messages.push(createSummaryMessage(entry.summary, entry.timestamp));
contextEntries.push(entry);
}
}
} else {
@ -376,18 +366,15 @@ export function buildSessionContext(
for (const entry of path) {
if (entry.type === "message") {
messages.push(entry.message);
contextEntries.push(entry);
} else if (entry.type === "custom_message") {
messages.push(createCustomMessage(entry));
contextEntries.push(entry);
} else if (entry.type === "branch_summary") {
messages.push(createSummaryMessage(entry.summary, entry.timestamp));
contextEntries.push(entry);
}
}
}
return { messages, entries: contextEntries, thinkingLevel, model };
return { messages, thinkingLevel, model };
}
/**