diff --git a/src/hooks/preemptive-compaction/index.ts b/src/hooks/preemptive-compaction/index.ts index 657f2c8..ee7a257 100644 --- a/src/hooks/preemptive-compaction/index.ts +++ b/src/hooks/preemptive-compaction/index.ts @@ -184,9 +184,15 @@ export function createPreemptiveCompactionHook( setTimeout(async () => { try { + const messageDir = getMessageDir(sessionID) + const storedMessage = messageDir ? findNearestMessageWithFields(messageDir) : null + await ctx.client.session.promptAsync({ path: { id: sessionID }, - body: { parts: [{ type: "text", text: "Continue" }] }, + body: { + agent: storedMessage?.agent, + parts: [{ type: "text", text: "Continue" }], + }, query: { directory: ctx.directory }, }) } catch {}