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

Fix branch summarization: use convertToLlm instead of messagesToText

Browse source 
The old messagesToText only extracted text content, losing all tool calls.
Now uses convertToLlm like compaction does, preserving the full conversation
including tool calls, which is essential for generating useful summaries.
This commit is contained in:
Mario Zechner 2025-12-29 22:37:46 +01:00
parent 0fe9f74b4e
commit 2ba69878ed
1 changed files with 20 additions and 60 deletions
Download patch file Download diff file Expand all files Collapse all files

80
packages/coding-agent/src/core/compaction/branch-summarization.ts
View file

@ -8,7 +8,12 @@
import type { AgentMessage } from "@mariozechner/pi-agent-core";
import type { Model } from "@mariozechner/pi-ai";
import { complete } from "@mariozechner/pi-ai";
import { createBranchSummaryMessage, createCompactionSummaryMessage, createHookMessage } from "../messages.js";
import {
convertToLlm,
createBranchSummaryMessage,
createCompactionSummaryMessage,
createHookMessage,
} from "../messages.js";
import type { ReadonlySessionManager, SessionEntry } from "../session-manager.js";
import { estimateTokens } from "./compaction.js";
@ -289,50 +294,6 @@ Use this EXACT format:
Keep each section concise. Preserve exact file paths, function names, and error messages.`;
/**
* Convert messages to text for the summarization prompt.
*/
function messagesToText(messages: AgentMessage[]): string {
const parts: string[] = [];
for (const msg of messages) {
let text = "";
if (msg.role === "user" && typeof msg.content === "string") {
text = msg.content;
} else if (msg.role === "user" && Array.isArray(msg.content)) {
text = msg.content
.filter((c): c is { type: "text"; text: string } => c.type === "text")
.map((c) => c.text)
.join("");
} else if (msg.role === "assistant" && "content" in msg && Array.isArray(msg.content)) {
text = msg.content
.filter((c): c is { type: "text"; text: string } => c.type === "text")
.map((c) => c.text)
.join("");
} else if (msg.role === "branchSummary" && "summary" in msg) {
text = `[Branch summary: ${msg.summary}]`;
} else if (msg.role === "compactionSummary" && "summary" in msg) {
text = `[Session summary: ${msg.summary}]`;
} else if (msg.role === "hookMessage" && "content" in msg) {
if (typeof msg.content === "string") {
text = msg.content;
} else if (Array.isArray(msg.content)) {
text = msg.content
.filter((c): c is { type: "text"; text: string } => c.type === "text")
.map((c) => c.text)
.join("");
}
}
if (text) {
parts.push(`${msg.role}: ${text}`);
}
}
return parts.join("\n\n");
}
/**
* Generate a summary of abandoned branch entries.
*
@ -355,25 +316,24 @@ export async function generateBranchSummary(
return { summary: "No content to summarize" };
}
// Transform to LLM-compatible messages (preserves tool calls, etc.)
const transformedMessages = convertToLlm(messages);
// Build prompt
const conversationText = messagesToText(messages);
const instructions = customInstructions || BRANCH_SUMMARY_PROMPT;
const prompt = `${instructions}\n\nConversation:\n${conversationText}`;
// Append summarization prompt as final user message
const summarizationMessages = [
...transformedMessages,
{
role: "user" as const,
content: [{ type: "text" as const, text: instructions }],
timestamp: Date.now(),
},
];
// Call LLM for summarization
const response = await complete(
model,
{
messages: [
{
role: "user",
content: [{ type: "text", text: prompt }],
timestamp: Date.now(),
},
],
},
{ apiKey, signal, maxTokens: 2048 },
);
const response = await complete(model, { messages: summarizationMessages }, { apiKey, signal, maxTokens: 2048 });
// Check if aborted or errored
if (response.stopReason === "aborted") {