Merge pull request #526 from RooVetGit/switch_mode_tool

Add a tool to switch modes
2025-12-20 12:21:13 -05:00 · 2025-01-24 12:50:57 -05:00
parent 9464a1bae9 d1754cace0
commit 7c41a9cbf8
9 changed files with 300 additions and 1 deletions
--- a/src/core/Cline.ts
+++ b/src/core/Cline.ts
@@ -1022,6 +1022,8 @@ export class Cline {
 							return `[${block.name} for '${block.params.question}']`
 						case "attempt_completion":
 							return `[${block.name}]`
+						case "switch_mode":
+							return `[${block.name} to '${block.params.mode_slug}'${block.params.reason ? ` because: ${block.params.reason}` : ""}]`
 					}
 				}

@@ -2014,6 +2016,74 @@ export class Cline {
 							break
 						}
 					}
+					case "switch_mode": {
+						const mode_slug: string | undefined = block.params.mode_slug
+						const reason: string | undefined = block.params.reason
+						try {
+							if (block.partial) {
+								const partialMessage = JSON.stringify({
+									tool: "switchMode",
+									mode: removeClosingTag("mode_slug", mode_slug),
+									reason: removeClosingTag("reason", reason),
+								})
+								await this.ask("tool", partialMessage, block.partial).catch(() => {})
+								break
+							} else {
+								if (!mode_slug) {
+									this.consecutiveMistakeCount++
+									pushToolResult(await this.sayAndCreateMissingParamError("switch_mode", "mode_slug"))
+									break
+								}
+								this.consecutiveMistakeCount = 0
+
+								// Verify the mode exists
+								const targetMode = getModeBySlug(
+									mode_slug,
+									(await this.providerRef.deref()?.getState())?.customModes,
+								)
+								if (!targetMode) {
+									pushToolResult(formatResponse.toolError(`Invalid mode: ${mode_slug}`))
+									break
+								}
+
+								// Check if already in requested mode
+								const currentMode =
+									(await this.providerRef.deref()?.getState())?.mode ?? defaultModeSlug
+								if (currentMode === mode_slug) {
+									pushToolResult(`Already in ${targetMode.name} mode.`)
+									break
+								}
+
+								const completeMessage = JSON.stringify({
+									tool: "switchMode",
+									mode: mode_slug,
+									reason,
+								})
+
+								const didApprove = await askApproval("tool", completeMessage)
+								if (!didApprove) {
+									break
+								}
+
+								// Switch the mode
+								const provider = this.providerRef.deref()
+								if (provider) {
+									await provider.updateGlobalState("mode", mode_slug)
+									await provider.postStateToWebview()
+								}
+								pushToolResult(
+									`Successfully switched from ${getModeBySlug(currentMode)?.name ?? currentMode} mode to ${
+										targetMode.name
+									} mode${reason ? ` because: ${reason}` : ""}.`,
+								)
+								break
+							}
+						} catch (error) {
+							await handleError("switching mode", error)
+							break
+						}
+					}
+
 					case "attempt_completion": {
 						/*
 						this.consecutiveMistakeCount = 0
--- a/src/core/assistant-message/index.ts
+++ b/src/core/assistant-message/index.ts
@@ -21,6 +21,7 @@ export const toolUseNames = [
 	"access_mcp_resource",
 	"ask_followup_question",
 	"attempt_completion",
+	"switch_mode",
 ] as const

 // Converts array of tool call names into a union type ("execute_command" | "read_file" | ...)
@@ -47,6 +48,8 @@ export const toolParamNames = [
 	"diff",
 	"start_line",
 	"end_line",
+	"mode_slug",
+	"reason",
 ] as const

 export type ToolParamName = (typeof toolParamNames)[number]
@@ -114,3 +117,8 @@ export interface AttemptCompletionToolUse extends ToolUse {
 	name: "attempt_completion"
 	params: Partial<Pick<Record<ToolParamName, string>, "result" | "command">>
 }
+
+export interface SwitchModeToolUse extends ToolUse {
+	name: "switch_mode"
+	params: Partial<Pick<Record<ToolParamName, string>, "mode_slug" | "reason">>
+}
--- a/src/core/prompts/tests/snapshots/system.test.ts.snap
+++ b/src/core/prompts/tests/snapshots/system.test.ts.snap
@@ -181,6 +181,23 @@ I've updated the CSS
 <command>open index.html</command>
 </attempt_completion>

+## switch_mode
+Description: Request to switch to a different mode. This tool allows modes to request switching to another mode when needed, such as switching to Code mode to make code changes. The user must approve the mode switch.
+Parameters:
+- mode_slug: (required) The slug of the mode to switch to (e.g., "code", "ask", "architect")
+- reason: (optional) The reason for switching modes
+Usage:
+<switch_mode>
+<mode_slug>Mode slug here</mode_slug>
+<reason>Reason for switching here</reason>
+</switch_mode>
+
+Example: Requesting to switch to code mode
+<switch_mode>
+<mode_slug>code</mode_slug>
+<reason>Need to make code changes</reason>
+</switch_mode>
+
 # Tool Use Guidelines

 1. In <thinking> tags, assess what information you already have and what information you need to proceed with the task.
@@ -469,6 +486,23 @@ I've updated the CSS
 <command>open index.html</command>
 </attempt_completion>

+## switch_mode
+Description: Request to switch to a different mode. This tool allows modes to request switching to another mode when needed, such as switching to Code mode to make code changes. The user must approve the mode switch.
+Parameters:
+- mode_slug: (required) The slug of the mode to switch to (e.g., "code", "ask", "architect")
+- reason: (optional) The reason for switching modes
+Usage:
+<switch_mode>
+<mode_slug>Mode slug here</mode_slug>
+<reason>Reason for switching here</reason>
+</switch_mode>
+
+Example: Requesting to switch to code mode
+<switch_mode>
+<mode_slug>code</mode_slug>
+<reason>Need to make code changes</reason>
+</switch_mode>
+
 # Tool Use Guidelines

 1. In <thinking> tags, assess what information you already have and what information you need to proceed with the task.
@@ -757,6 +791,23 @@ I've updated the CSS
 <command>open index.html</command>
 </attempt_completion>

+## switch_mode
+Description: Request to switch to a different mode. This tool allows modes to request switching to another mode when needed, such as switching to Code mode to make code changes. The user must approve the mode switch.
+Parameters:
+- mode_slug: (required) The slug of the mode to switch to (e.g., "code", "ask", "architect")
+- reason: (optional) The reason for switching modes
+Usage:
+<switch_mode>
+<mode_slug>Mode slug here</mode_slug>
+<reason>Reason for switching here</reason>
+</switch_mode>
+
+Example: Requesting to switch to code mode
+<switch_mode>
+<mode_slug>code</mode_slug>
+<reason>Need to make code changes</reason>
+</switch_mode>
+
 # Tool Use Guidelines

 1. In <thinking> tags, assess what information you already have and what information you need to proceed with the task.
@@ -1091,6 +1142,23 @@ I've updated the CSS
 <command>open index.html</command>
 </attempt_completion>

+## switch_mode
+Description: Request to switch to a different mode. This tool allows modes to request switching to another mode when needed, such as switching to Code mode to make code changes. The user must approve the mode switch.
+Parameters:
+- mode_slug: (required) The slug of the mode to switch to (e.g., "code", "ask", "architect")
+- reason: (optional) The reason for switching modes
+Usage:
+<switch_mode>
+<mode_slug>Mode slug here</mode_slug>
+<reason>Reason for switching here</reason>
+</switch_mode>
+
+Example: Requesting to switch to code mode
+<switch_mode>
+<mode_slug>code</mode_slug>
+<reason>Need to make code changes</reason>
+</switch_mode>
+
 # Tool Use Guidelines

 1. In <thinking> tags, assess what information you already have and what information you need to proceed with the task.
@@ -1431,6 +1499,23 @@ I've updated the CSS
 <command>open index.html</command>
 </attempt_completion>

+## switch_mode
+Description: Request to switch to a different mode. This tool allows modes to request switching to another mode when needed, such as switching to Code mode to make code changes. The user must approve the mode switch.
+Parameters:
+- mode_slug: (required) The slug of the mode to switch to (e.g., "code", "ask", "architect")
+- reason: (optional) The reason for switching modes
+Usage:
+<switch_mode>
+<mode_slug>Mode slug here</mode_slug>
+<reason>Reason for switching here</reason>
+</switch_mode>
+
+Example: Requesting to switch to code mode
+<switch_mode>
+<mode_slug>code</mode_slug>
+<reason>Need to make code changes</reason>
+</switch_mode>
+
 # Tool Use Guidelines

 1. In <thinking> tags, assess what information you already have and what information you need to proceed with the task.
@@ -2129,6 +2214,23 @@ I've updated the CSS
 <command>open index.html</command>
 </attempt_completion>

+## switch_mode
+Description: Request to switch to a different mode. This tool allows modes to request switching to another mode when needed, such as switching to Code mode to make code changes. The user must approve the mode switch.
+Parameters:
+- mode_slug: (required) The slug of the mode to switch to (e.g., "code", "ask", "architect")
+- reason: (optional) The reason for switching modes
+Usage:
+<switch_mode>
+<mode_slug>Mode slug here</mode_slug>
+<reason>Reason for switching here</reason>
+</switch_mode>
+
+Example: Requesting to switch to code mode
+<switch_mode>
+<mode_slug>code</mode_slug>
+<reason>Need to make code changes</reason>
+</switch_mode>
+
 # Tool Use Guidelines

 1. In <thinking> tags, assess what information you already have and what information you need to proceed with the task.
@@ -2480,6 +2582,23 @@ I've updated the CSS
 <command>open index.html</command>
 </attempt_completion>

+## switch_mode
+Description: Request to switch to a different mode. This tool allows modes to request switching to another mode when needed, such as switching to Code mode to make code changes. The user must approve the mode switch.
+Parameters:
+- mode_slug: (required) The slug of the mode to switch to (e.g., "code", "ask", "architect")
+- reason: (optional) The reason for switching modes
+Usage:
+<switch_mode>
+<mode_slug>Mode slug here</mode_slug>
+<reason>Reason for switching here</reason>
+</switch_mode>
+
+Example: Requesting to switch to code mode
+<switch_mode>
+<mode_slug>code</mode_slug>
+<reason>Need to make code changes</reason>
+</switch_mode>
+
 # Tool Use Guidelines

 1. In <thinking> tags, assess what information you already have and what information you need to proceed with the task.
@@ -2768,6 +2887,23 @@ I've updated the CSS
 <command>open index.html</command>
 </attempt_completion>

+## switch_mode
+Description: Request to switch to a different mode. This tool allows modes to request switching to another mode when needed, such as switching to Code mode to make code changes. The user must approve the mode switch.
+Parameters:
+- mode_slug: (required) The slug of the mode to switch to (e.g., "code", "ask", "architect")
+- reason: (optional) The reason for switching modes
+Usage:
+<switch_mode>
+<mode_slug>Mode slug here</mode_slug>
+<reason>Reason for switching here</reason>
+</switch_mode>
+
+Example: Requesting to switch to code mode
+<switch_mode>
+<mode_slug>code</mode_slug>
+<reason>Need to make code changes</reason>
+</switch_mode>
+
 # Tool Use Guidelines

 1. In <thinking> tags, assess what information you already have and what information you need to proceed with the task.
@@ -3099,6 +3235,23 @@ I've updated the CSS
 <command>open index.html</command>
 </attempt_completion>

+## switch_mode
+Description: Request to switch to a different mode. This tool allows modes to request switching to another mode when needed, such as switching to Code mode to make code changes. The user must approve the mode switch.
+Parameters:
+- mode_slug: (required) The slug of the mode to switch to (e.g., "code", "ask", "architect")
+- reason: (optional) The reason for switching modes
+Usage:
+<switch_mode>
+<mode_slug>Mode slug here</mode_slug>
+<reason>Reason for switching here</reason>
+</switch_mode>
+
+Example: Requesting to switch to code mode
+<switch_mode>
+<mode_slug>code</mode_slug>
+<reason>Need to make code changes</reason>
+</switch_mode>
+
 # Tool Use Guidelines

 1. In <thinking> tags, assess what information you already have and what information you need to proceed with the task.
@@ -3373,6 +3526,23 @@ I've updated the CSS
 <command>open index.html</command>
 </attempt_completion>

+## switch_mode
+Description: Request to switch to a different mode. This tool allows modes to request switching to another mode when needed, such as switching to Code mode to make code changes. The user must approve the mode switch.
+Parameters:
+- mode_slug: (required) The slug of the mode to switch to (e.g., "code", "ask", "architect")
+- reason: (optional) The reason for switching modes
+Usage:
+<switch_mode>
+<mode_slug>Mode slug here</mode_slug>
+<reason>Reason for switching here</reason>
+</switch_mode>
+
+Example: Requesting to switch to code mode
+<switch_mode>
+<mode_slug>code</mode_slug>
+<reason>Need to make code changes</reason>
+</switch_mode>
+
 # Tool Use Guidelines

 1. In <thinking> tags, assess what information you already have and what information you need to proceed with the task.
--- a/src/core/prompts/tools/index.ts
+++ b/src/core/prompts/tools/index.ts
@@ -9,6 +9,7 @@ import { getAskFollowupQuestionDescription } from "./ask-followup-question"
 import { getAttemptCompletionDescription } from "./attempt-completion"
 import { getUseMcpToolDescription } from "./use-mcp-tool"
 import { getAccessMcpResourceDescription } from "./access-mcp-resource"
+import { getSwitchModeDescription } from "./switch-mode"
 import { DiffStrategy } from "../../diff/DiffStrategy"
 import { McpHub } from "../../../services/mcp/McpHub"
 import { Mode, ModeConfig, getModeConfig, isToolAllowedForMode, getGroupName } from "../../../shared/modes"
@@ -28,6 +29,7 @@ const toolDescriptionMap: Record<string, (args: ToolArgs) => string | undefined>
 	attempt_completion: () => getAttemptCompletionDescription(),
 	use_mcp_tool: (args) => getUseMcpToolDescription(args),
 	access_mcp_resource: (args) => getAccessMcpResourceDescription(args),
+	switch_mode: () => getSwitchModeDescription(),
 	apply_diff: (args) =>
 		args.diffStrategy ? args.diffStrategy.getToolDescription({ cwd: args.cwd, toolOptions: args.toolOptions }) : "",
 }
@@ -97,4 +99,5 @@ export {
 	getAttemptCompletionDescription,
 	getUseMcpToolDescription,
 	getAccessMcpResourceDescription,
+	getSwitchModeDescription,
 }
--- a/src/core/prompts/tools/switch-mode.ts
+++ b/src/core/prompts/tools/switch-mode.ts
@@ -0,0 +1,18 @@
+export function getSwitchModeDescription(): string {
+	return `## switch_mode
+Description: Request to switch to a different mode. This tool allows modes to request switching to another mode when needed, such as switching to Code mode to make code changes. The user must approve the mode switch.
+Parameters:
+- mode_slug: (required) The slug of the mode to switch to (e.g., "code", "ask", "architect")
+- reason: (optional) The reason for switching modes
+Usage:
+<switch_mode>
+<mode_slug>Mode slug here</mode_slug>
+<reason>Reason for switching here</reason>
+</switch_mode>
+
+Example: Requesting to switch to code mode
+<switch_mode>
+<mode_slug>code</mode_slug>
+<reason>Need to make code changes</reason>
+</switch_mode>`
+}