fix(run): inherit main-session tool permissions for continuation prompts

This commit is contained in:
YeonGyu-Kim
2026-02-18 18:02:42 +09:00
parent 80bb262066
commit dba7a1e07c
17 changed files with 297 additions and 28 deletions
+48
View File
@@ -0,0 +1,48 @@
declare const require: (name: string) => any
const { describe, expect, test } = require("bun:test")
import { extractResumeConfig, resumeSession } from "./resume"
import type { MessageData } from "./types"
describe("session-recovery resume", () => {
test("extractResumeConfig carries tools from last user message", () => {
// given
const userMessage: MessageData = {
info: {
agent: "Hephaestus",
model: { providerID: "openai", modelID: "gpt-5.3-codex" },
tools: { question: false, bash: true },
},
}
// when
const config = extractResumeConfig(userMessage, "ses_resume_tools")
// then
expect(config.tools).toEqual({ question: false, bash: true })
})
test("resumeSession sends inherited tools with continuation prompt", async () => {
// given
let promptBody: Record<string, unknown> | undefined
const client = {
session: {
promptAsync: async (input: { body: Record<string, unknown> }) => {
promptBody = input.body
return {}
},
},
}
// when
const ok = await resumeSession(client as never, {
sessionID: "ses_resume_prompt",
agent: "Hephaestus",
model: { providerID: "openai", modelID: "gpt-5.3-codex" },
tools: { question: false, bash: true },
})
// then
expect(ok).toBe(true)
expect(promptBody?.tools).toEqual({ question: false, bash: true })
})
})
+4
View File
@@ -1,5 +1,6 @@
import type { createOpencodeClient } from "@opencode-ai/sdk"
import type { MessageData, ResumeConfig } from "./types"
import { resolveInheritedPromptTools } from "../../shared"
const RECOVERY_RESUME_TEXT = "[session recovered - continuing previous task]"
@@ -19,17 +20,20 @@ export function extractResumeConfig(userMessage: MessageData | undefined, sessio
sessionID,
agent: userMessage?.info?.agent,
model: userMessage?.info?.model,
tools: userMessage?.info?.tools,
}
}
export async function resumeSession(client: Client, config: ResumeConfig): Promise<boolean> {
try {
const inheritedTools = resolveInheritedPromptTools(config.sessionID, config.tools)
await client.session.promptAsync({
path: { id: config.sessionID },
body: {
parts: [{ type: "text", text: RECOVERY_RESUME_TEXT }],
agent: config.agent,
model: config.model,
...(inheritedTools ? { tools: inheritedTools } : {}),
},
})
return true
+1
View File
@@ -95,4 +95,5 @@ export interface ResumeConfig {
providerID: string
modelID: string
}
tools?: Record<string, boolean>
}