YeonGyu-Kim
72 lines
2.3 KiB
TypeScript
72 lines
2.3 KiB
TypeScript
import type { PluginInput } from "@opencode-ai/plugin"
|
|
import type { BackgroundManager } from "../../features/background-agent"
|
|
import { log } from "../../shared/logger"
|
|
import { resolveInheritedPromptTools } from "../../shared"
|
|
import { HOOK_NAME } from "./hook-name"
|
|
import { BOULDER_CONTINUATION_PROMPT } from "./system-reminder-templates"
|
|
import { resolveRecentPromptContextForSession } from "./recent-model-resolver"
|
|
import type { SessionState } from "./types"
|
|
|
|
export async function injectBoulderContinuation(input: {
|
|
ctx: PluginInput
|
|
sessionID: string
|
|
planName: string
|
|
remaining: number
|
|
total: number
|
|
agent?: string
|
|
backgroundManager?: BackgroundManager
|
|
sessionState: SessionState
|
|
}): Promise<void> {
|
|
const {
|
|
ctx,
|
|
sessionID,
|
|
planName,
|
|
remaining,
|
|
total,
|
|
agent,
|
|
backgroundManager,
|
|
sessionState,
|
|
} = input
|
|
|
|
const hasRunningBgTasks = backgroundManager
|
|
? backgroundManager.getTasksByParentSession(sessionID).some((t: { status: string }) => t.status === "running")
|
|
: false
|
|
|
|
if (hasRunningBgTasks) {
|
|
log(`[${HOOK_NAME}] Skipped injection: background tasks running`, { sessionID })
|
|
return
|
|
}
|
|
|
|
const prompt =
|
|
BOULDER_CONTINUATION_PROMPT.replace(/{PLAN_NAME}/g, planName) +
|
|
`\n\n[Status: ${total - remaining}/${total} completed, ${remaining} remaining]`
|
|
|
|
try {
|
|
log(`[${HOOK_NAME}] Injecting boulder continuation`, { sessionID, planName, remaining })
|
|
|
|
const promptContext = await resolveRecentPromptContextForSession(ctx, sessionID)
|
|
const inheritedTools = resolveInheritedPromptTools(sessionID, promptContext.tools)
|
|
|
|
await ctx.client.session.promptAsync({
|
|
path: { id: sessionID },
|
|
body: {
|
|
agent: agent ?? "atlas",
|
|
...(promptContext.model !== undefined ? { model: promptContext.model } : {}),
|
|
...(inheritedTools ? { tools: inheritedTools } : {}),
|
|
parts: [{ type: "text", text: prompt }],
|
|
},
|
|
query: { directory: ctx.directory },
|
|
})
|
|
|
|
sessionState.promptFailureCount = 0
|
|
log(`[${HOOK_NAME}] Boulder continuation injected`, { sessionID })
|
|
} catch (err) {
|
|
sessionState.promptFailureCount += 1
|
|
log(`[${HOOK_NAME}] Boulder continuation failed`, {
|
|
sessionID,
|
|
error: String(err),
|
|
promptFailureCount: sessionState.promptFailureCount,
|
|
})
|
|
}
|
|
}
|