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

Update full-compaction example to use resolveApiKey

Browse source
This commit is contained in:
Mario Zechner 2025-12-24 12:30:00 +01:00
parent 43a5447a80
commit e4283294c8
1 changed files with 9 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

11
packages/coding-agent/examples/hooks/full-compaction.ts
View file

@ -21,10 +21,17 @@ export default function (pi: HookAPI) {
pi.on("session", async (event, ctx) => { pi.on("session", async (event, ctx) => {
if (event.reason !== "before_compact") return; if (event.reason !== "before_compact") return;
const { messagesToSummarize, tokensBefore, model, apiKey, cutPoint } = event; const { messagesToSummarize, tokensBefore, model, resolveApiKey, cutPoint } = event;
ctx.ui.notify(`Compacting ${tokensBefore.toLocaleString()} tokens with full summary...`, "info"); ctx.ui.notify(`Compacting ${tokensBefore.toLocaleString()} tokens with full summary...`, "info");
// Resolve API key for the model
const apiKey = await resolveApiKey(model);
if (!apiKey) {
ctx.ui.notify(`No API key for ${model.provider}, using default compaction`, "warning");
return;
}
// Transform app messages to LLM-compatible format // Transform app messages to LLM-compatible format
const transformedMessages = messageTransformer(messagesToSummarize); const transformedMessages = messageTransformer(messagesToSummarize);
@ -55,7 +62,7 @@ Format the summary as structured markdown with clear sections.`,
]; ];
try { try {
// Use the same model and API key that would be used for compaction // Use the same model with resolved API key
const response = await complete(model, { messages: summaryMessages }, { apiKey, maxTokens: 8192 }); const response = await complete(model, { messages: summaryMessages }, { apiKey, maxTokens: 8192 });
const summary = response.content const summary = response.content