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

Use serializeConversation in handoff hook

Browse source 
Properly handles all message types including tool calls, thinking,
and tool results instead of manual text extraction.
This commit is contained in:
Mario Zechner 2026-01-02 00:45:03 +01:00
parent ace0063a00
commit 91c52de8be
1 changed files with 9 additions and 26 deletions
Download patch file Download diff file Expand all files Collapse all files

35
packages/coding-agent/examples/hooks/handoff.ts
View file

@ -13,8 +13,8 @@
*/ */
import { complete, type Message } from "@mariozechner/pi-ai"; import { complete, type Message } from "@mariozechner/pi-ai";
import type { HookAPI } from "@mariozechner/pi-coding-agent"; import type { HookAPI, SessionEntry } from "@mariozechner/pi-coding-agent";
import { BorderedLoader } from "@mariozechner/pi-coding-agent"; import { BorderedLoader, convertToLlm, serializeConversation } from "@mariozechner/pi-coding-agent";
const SYSTEM_PROMPT = `You are a context transfer assistant. Given a conversation history and the user's goal for a new thread, generate a focused prompt that: const SYSTEM_PROMPT = `You are a context transfer assistant. Given a conversation history and the user's goal for a new thread, generate a focused prompt that:
@ -60,35 +60,18 @@ export default function (pi: HookAPI) {
// Gather conversation context from current branch // Gather conversation context from current branch
const branch = ctx.sessionManager.getBranch(); const branch = ctx.sessionManager.getBranch();
const conversationParts: string[] = []; const messages = branch
.filter((entry): entry is SessionEntry & { type: "message" } => entry.type === "message")
.map((entry) => entry.message);
for (const entry of branch) { if (messages.length === 0) {
if (entry.type === "message") {
const msg = entry.message;
if ("role" in msg && (msg.role === "user" || msg.role === "assistant")) {
let text: string;
if (typeof msg.content === "string") {
text = msg.content;
} else {
text = msg.content
.filter((c): c is { type: "text"; text: string } => c.type === "text")
.map((c) => c.text)
.join("\n");
}
if (text) {
const role = msg.role === "user" ? "User" : "Assistant";
conversationParts.push(`${role}: ${text}`);
}
}
}
}
if (conversationParts.length === 0) {
ctx.ui.notify("No conversation to hand off", "error"); ctx.ui.notify("No conversation to hand off", "error");
return; return;
} }
const conversationText = conversationParts.join("\n\n"); // Convert to LLM format and serialize
const llmMessages = convertToLlm(messages);
const conversationText = serializeConversation(llmMessages);
const currentSessionFile = ctx.sessionManager.getSessionFile(); const currentSessionFile = ctx.sessionManager.getSessionFile();
// Generate the handoff prompt with loader UI // Generate the handoff prompt with loader UI