diff --git a/src/hooks/atlas/boulder-continuation-injector.ts b/src/hooks/atlas/boulder-continuation-injector.ts index 289668b4b..c74b2292a 100644 --- a/src/hooks/atlas/boulder-continuation-injector.ts +++ b/src/hooks/atlas/boulder-continuation-injector.ts @@ -14,6 +14,7 @@ export async function injectBoulderContinuation(input: { remaining: number total: number agent?: string + worktreePath?: string backgroundManager?: BackgroundManager sessionState: SessionState }): Promise { @@ -24,6 +25,7 @@ export async function injectBoulderContinuation(input: { remaining, total, agent, + worktreePath, backgroundManager, sessionState, } = input @@ -37,9 +39,11 @@ export async function injectBoulderContinuation(input: { return } + const worktreeContext = worktreePath ? `\n\n[Worktree: ${worktreePath}]` : "" const prompt = BOULDER_CONTINUATION_PROMPT.replace(/{PLAN_NAME}/g, planName) + - `\n\n[Status: ${total - remaining}/${total} completed, ${remaining} remaining]` + `\n\n[Status: ${total - remaining}/${total} completed, ${remaining} remaining]` + + worktreeContext try { log(`[${HOOK_NAME}] Injecting boulder continuation`, { sessionID, planName, remaining }) diff --git a/src/hooks/atlas/event-handler.ts b/src/hooks/atlas/event-handler.ts index d7e92101c..6346dcbb0 100644 --- a/src/hooks/atlas/event-handler.ts +++ b/src/hooks/atlas/event-handler.ts @@ -130,6 +130,7 @@ export function createAtlasEventHandler(input: { remaining, total: progress.total, agent: boulderState.agent, + worktreePath: boulderState.worktree_path, backgroundManager, sessionState: state, })