fix(atlas): gate boulder continuation prompts

This commit is contained in:
YeonGyu-Kim
2026-05-15 11:30:11 +09:00
parent b0a484b403
commit 960baf39bb
2 changed files with 43 additions and 13 deletions
+23 -6
View File
@@ -20,6 +20,7 @@ import { createInternalAgentContinuationTextPart } from "../../shared"
import { getAgentConfigKey } from "../../shared/agent-display-names"
import { log } from "../../shared/logger"
import { shouldPromptAfterSessionIdle } from "../shared/session-idle-settle"
import { promptAsyncAfterSessionIdle } from "../shared/prompt-async-gate"
import { injectBoulderContinuation } from "./boulder-continuation-injector"
import { HOOK_NAME } from "./hook-name"
import { resolveActiveBoulderSession } from "./resolve-active-boulder-session"
@@ -52,6 +53,7 @@ async function injectContinuation(input: {
progress: { total: number; completed: number }
agent?: string
worktreePath?: string
idleSettleMs?: number
}): Promise<void> {
const remaining = input.progress.total - input.progress.completed
if (input.sessionState.isInjectingContinuation) {
@@ -110,6 +112,7 @@ async function injectContinuation(input: {
preferredTaskTitle: preferredTaskSession?.task_title,
backgroundManager: input.options?.backgroundManager,
sessionState: input.sessionState,
idleSettleMs: input.idleSettleMs,
})
if (result === "injected") {
@@ -288,14 +291,27 @@ export async function handleAtlasSessionIdle(input: {
return
}
await ctx.client.session.promptAsync({
path: { id: sessionID },
body: {
agent: atlasAgent,
parts: [createInternalAgentContinuationTextPart(prompt)],
const promptResult = await promptAsyncAfterSessionIdle({
client: ctx.client,
sessionID,
source: HOOK_NAME,
settleMs: options?.idleSettleMs,
input: {
path: { id: sessionID },
body: {
agent: atlasAgent,
parts: [createInternalAgentContinuationTextPart(prompt)],
},
query: { directory: ctx.directory },
},
query: { directory: ctx.directory },
})
if (promptResult.status !== "dispatched") {
log(`[${HOOK_NAME}] Boulder completion nudge skipped by promptAsync gate`, {
sessionID,
status: promptResult.status,
})
return
}
sessionState.boulderCompletionNudgedAt = {
...(sessionState.boulderCompletionNudgedAt ?? {}),
[work.work_id]: Date.now(),
@@ -398,6 +414,7 @@ export async function handleAtlasSessionIdle(input: {
progress,
agent: boulderState.agent,
worktreePath: boulderState.worktree_path,
idleSettleMs: options?.idleSettleMs ?? 0,
})
}