Kevin van Dijk
93 lines
3.0 KiB
TypeScript
93 lines
3.0 KiB
TypeScript
import z from "zod"
|
|
import path from "path"
|
|
import { Tool } from "./tool"
|
|
import { Question } from "../question"
|
|
import { Session } from "../session"
|
|
import { MessageV2 } from "../session/message-v2"
|
|
import { Identifier } from "../id/id"
|
|
import { Provider } from "../provider/provider"
|
|
import { Instance } from "../project/instance"
|
|
import EXIT_DESCRIPTION from "./plan-exit.txt"
|
|
|
|
async function getLastModel(sessionID: string) {
|
|
for await (const item of MessageV2.stream(sessionID)) {
|
|
if (item.info.role === "user" && item.info.model) return item.info.model
|
|
}
|
|
return Provider.defaultModel()
|
|
}
|
|
|
|
// kilocode_change start - simplified plan_exit: readiness signal only, no user prompt
|
|
export const PlanExitTool = Tool.define("plan_exit", {
|
|
description: EXIT_DESCRIPTION,
|
|
parameters: z.object({}),
|
|
async execute(_params, ctx) {
|
|
const session = await Session.get(ctx.sessionID)
|
|
const plan = path.relative(Instance.worktree, Session.plan(session))
|
|
return {
|
|
title: "Planning complete",
|
|
output: `Plan is ready at ${plan}. Ending planning turn.`,
|
|
metadata: { plan },
|
|
}
|
|
},
|
|
})
|
|
// kilocode_change end
|
|
|
|
/*
|
|
export const PlanEnterTool = Tool.define("plan_enter", {
|
|
description: ENTER_DESCRIPTION,
|
|
parameters: z.object({}),
|
|
async execute(_params, ctx) {
|
|
const session = await Session.get(ctx.sessionID)
|
|
const plan = path.relative(Instance.worktree, Session.plan(session))
|
|
|
|
const answers = await Question.ask({
|
|
sessionID: ctx.sessionID,
|
|
questions: [
|
|
{
|
|
question: `Would you like to switch to the plan agent and create a plan saved to ${plan}?`,
|
|
header: "Plan Mode",
|
|
custom: false,
|
|
options: [
|
|
{ label: "Yes", description: "Switch to plan agent for research and planning" },
|
|
{ label: "No", description: "Stay with code agent to continue making changes" }, // kilocode_change - renamed from "build" to "code"
|
|
],
|
|
},
|
|
],
|
|
tool: ctx.callID ? { messageID: ctx.messageID, callID: ctx.callID } : undefined,
|
|
})
|
|
|
|
const answer = answers[0]?.[0]
|
|
|
|
if (answer === "No") throw new Question.RejectedError()
|
|
|
|
const model = await getLastModel(ctx.sessionID)
|
|
|
|
const userMsg: MessageV2.User = {
|
|
id: Identifier.ascending("message"),
|
|
sessionID: ctx.sessionID,
|
|
role: "user",
|
|
time: {
|
|
created: Date.now(),
|
|
},
|
|
agent: "plan",
|
|
model,
|
|
}
|
|
await Session.updateMessage(userMsg)
|
|
await Session.updatePart({
|
|
id: Identifier.ascending("part"),
|
|
messageID: userMsg.id,
|
|
sessionID: ctx.sessionID,
|
|
type: "text",
|
|
text: "User has requested to enter plan mode. Switch to plan mode and begin planning.",
|
|
synthetic: true,
|
|
} satisfies MessageV2.TextPart)
|
|
|
|
return {
|
|
title: "Switching to plan agent",
|
|
output: `User confirmed to switch to plan mode. A new message has been created to switch you to plan mode. The plan file will be at ${plan}. Begin planning.`,
|
|
metadata: {},
|
|
}
|
|
},
|
|
})
|
|
*/
|