WIP: Experiment: Create temporary planning documents with new create_plan tool

.changeset/ephemeral-planning-experiment.md

@@ -0,0 +1,5 @@
+---
+"kilo-code": minor
+---
+
+Add `ephemeralPlanning` experiment to gate the planning tool

.changeset/planning-doc-tool.md

@@ -0,0 +1,5 @@
+---
+"kilo-code": minor
+---
+
+Add `create_plan` tool for creating ephemeral planning documents

packages/types/src/experiment.ts

@@ -6,7 +6,7 @@ import type { Keys, Equals, AssertEqual } from "./type-fu.js"
  * ExperimentId
  */
 
-const kilocodeExperimentIds = ["morphFastApply", "speechToText"] as const // kilocode_change
+const kilocodeExperimentIds = ["morphFastApply", "speechToText", "ephemeralPlanning"] as const // kilocode_change
 export const experimentIds = [
 	"powerSteering",
 	"multiFileApplyDiff",
@@ -27,6 +27,7 @@ export type ExperimentId = z.infer<typeof experimentIdsSchema>
 export const experimentsSchema = z.object({
 	morphFastApply: z.boolean().optional(), // kilocode_change
 	speechToText: z.boolean().optional(), // kilocode_change
+	ephemeralPlanning: z.boolean().optional(), // kilocode_change
 	powerSteering: z.boolean().optional(),
 	multiFileApplyDiff: z.boolean().optional(),
 	preventFocusDisruption: z.boolean().optional(),

packages/types/src/tool.ts

@@ -39,6 +39,7 @@ export const toolNames = [
 	"report_bug",
 	"condense",
 	"delete_file",
+	"create_plan", // kilocode_change
 	// kilocode_change end
 	"update_todo_list",
 	"run_slash_command",

src/__tests__/extension.spec.ts

@@ -28,6 +28,10 @@ vi.mock("vscode", () => ({
 	},
 	workspace: {
 		registerTextDocumentContentProvider: vi.fn(),
+		// kilocode_change start
+		registerFileSystemProvider: vi.fn().mockReturnValue({
+			dispose: vi.fn(),
+		}),
 		getConfiguration: vi.fn().mockReturnValue({
 			get: vi.fn().mockReturnValue([]),
 		}),
@@ -50,6 +54,13 @@ vi.mock("vscode", () => ({
 		onDidCloseTextDocument: vi.fn().mockReturnValue({
 			dispose: vi.fn(),
 		}),
+		// kilocode_change start
+		fs: {
+			readFile: vi.fn(),
+			writeFile: vi.fn(),
+			stat: vi.fn(),
+		},
+		// kilocode_change end
 	},
 	languages: {
 		registerCodeActionsProvider: vi.fn(),
@@ -63,6 +74,7 @@ vi.mock("vscode", () => ({
 	env: {
 		language: "en",
 		appName: "Visual Studio Code",
+		version: "1.0.0", // kilocode_change
 	},
 	ExtensionMode: {
 		Production: 1,

src/core/assistant-message/presentAssistantMessage.ts

@@ -31,6 +31,7 @@ import { switchModeTool } from "../tools/SwitchModeTool"
 import { attemptCompletionTool, AttemptCompletionCallbacks } from "../tools/AttemptCompletionTool"
 import { newTaskTool } from "../tools/NewTaskTool"
 
+import { createPlanTool } from "../tools/CreatePlanTool" // kilocode_change
 import { updateTodoListTool } from "../tools/UpdateTodoListTool"
 import { runSlashCommandTool } from "../tools/RunSlashCommandTool"
 import { generateImageTool } from "../tools/GenerateImageTool"
@@ -466,6 +467,8 @@ export async function presentAssistantMessage(cline: Task) {
 						return `[${block.name}]`
 					case "condense":
 						return `[${block.name}]`
+					case "create_plan":
+						return `[${block.name} for '${block.params.title}']` // kilocode_change
 					// kilocode_change end
 					case "run_slash_command":
 						return `[${block.name} for '${block.params.command}'${block.params.args ? ` with args: ${block.params.args}` : ""}]`
@@ -1099,6 +1102,15 @@ export async function presentAssistantMessage(cline: Task) {
 				case "condense":
 					await condenseTool(cline, block, askApproval, handleError, pushToolResult, removeClosingTag)
 					break
+				case "create_plan":
+					await createPlanTool.handle(cline, block as ToolUse<"create_plan">, {
+						askApproval,
+						handleError,
+						pushToolResult,
+						removeClosingTag,
+						toolProtocol,
+					}) // kilocode_change
+					break
 				// kilocode_change end
 
 				case "run_slash_command":

src/core/kilocode/wrapper.ts

@@ -3,7 +3,7 @@ import { JETBRAIN_PRODUCTS, KiloCodeWrapperProperties } from "../../shared/kiloc
 
 export const getKiloCodeWrapperProperties = (): KiloCodeWrapperProperties => {
 	const appName = vscode.env.appName
-	const kiloCodeWrapped = appName.includes("wrapper")
+	const kiloCodeWrapped = appName?.includes("wrapper") ?? false // kilocode_change
 	let kiloCodeWrapper = null
 	let kiloCodeWrapperTitle = null
 	let kiloCodeWrapperCode = null
@@ -37,8 +37,8 @@ export const getEditorNameHeader = () => {
 	return (
 		props.kiloCodeWrapped
 			? [props.kiloCodeWrapperTitle, props.kiloCodeWrapperVersion]
-			: [vscode.env.appName, vscode.version]
-	)
+			: [vscode.env.appName || "VS Code", vscode.version]
+	) // kilocode_change
 		.filter(Boolean)
 		.join(" ")
 }