diff --git a/.changeset/slimy-pans-obey.md b/.changeset/slimy-pans-obey.md new file mode 100644 index 0000000..01ea6c7 --- /dev/null +++ b/.changeset/slimy-pans-obey.md @@ -0,0 +1,5 @@ +--- +"roo-cline": patch +--- + +Add toggle to enable/disable the MCP-related sections of the system prompt (thanks @daniel-lxs!) diff --git a/README.md b/README.md index 5d1cd7b..892a5de 100644 --- a/README.md +++ b/README.md @@ -18,7 +18,8 @@ A fork of Cline, an autonomous coding agent, with some additional experimental f - Support for Meta 3, 3.1, and 3.2 models via AWS Bedrock - Support for listing models from OpenAI-compatible providers - Per-tool MCP auto-approval -- Enable/disable MCP servers +- Enable/disable individual MCP servers +- Enable/disable the MCP feature overall - Configurable delay after auto-writes to allow diagnostics to detect potential problems - Control the number of terminal output lines to pass to the model when executing commands - Runs alongside the original Cline diff --git a/src/core/Cline.ts b/src/core/Cline.ts index 7e06c22..6a73a5f 100644 --- a/src/core/Cline.ts +++ b/src/core/Cline.ts @@ -50,6 +50,7 @@ import { ClineProvider, GlobalFileNames } from "./webview/ClineProvider" import { detectCodeOmission } from "../integrations/editor/detect-omission" import { BrowserSession } from "../services/browser/BrowserSession" import { OpenRouterHandler } from "../api/providers/openrouter" +import { McpHub } from "../services/mcp/McpHub" const cwd = vscode.workspace.workspaceFolders?.map((folder) => folder.uri.fsPath).at(0) ?? path.join(os.homedir(), "Desktop") // may or may not exist but fs checking existence would immediately ask for permission which would be bad UX, need to come up with a better solution @@ -778,14 +779,19 @@ export class Cline { } async *attemptApiRequest(previousApiReqIndex: number): ApiStream { - // Wait for MCP servers to be connected before generating system prompt - await pWaitFor(() => this.providerRef.deref()?.mcpHub?.isConnecting !== true, { timeout: 10_000 }).catch(() => { - console.error("MCP servers failed to connect in time") - }) + let mcpHub: McpHub | undefined - const mcpHub = this.providerRef.deref()?.mcpHub - if (!mcpHub) { - throw new Error("MCP hub not available") + const { mcpEnabled } = await this.providerRef.deref()?.getState() ?? {} + + if (mcpEnabled ?? true) { + mcpHub = this.providerRef.deref()?.mcpHub + if (!mcpHub) { + throw new Error("MCP hub not available") + } + // Wait for MCP servers to be connected before generating system prompt + await pWaitFor(() => mcpHub!.isConnecting !== true, { timeout: 10_000 }).catch(() => { + console.error("MCP servers failed to connect in time") + }) } const { browserViewportSize, preferredLanguage } = await this.providerRef.deref()?.getState() ?? {} diff --git a/src/core/prompts/system.ts b/src/core/prompts/system.ts index b1a835a..88bfc26 100644 --- a/src/core/prompts/system.ts +++ b/src/core/prompts/system.ts @@ -9,7 +9,7 @@ import { McpHub } from "../../services/mcp/McpHub" export const SYSTEM_PROMPT = async ( cwd: string, supportsComputerUse: boolean, - mcpHub: McpHub, + mcpHub?: McpHub, diffStrategy?: DiffStrategy, browserViewportSize?: string ) => `You are Cline, a highly skilled software engineer with extensive knowledge in many programming languages, frameworks, design patterns, and best practices. @@ -146,6 +146,7 @@ Usage: : "" } +${mcpHub ? ` ## use_mcp_tool Description: Request to use a tool provided by a connected MCP server. Each MCP server can provide multiple tools with different capabilities. Tools have defined input schemas that specify required and optional parameters. Parameters: @@ -173,7 +174,7 @@ Usage: server name here resource URI here - +` : ''} ## ask_followup_question Description: Ask the user a question to gather additional information needed to complete the task. This tool should be used when you encounter ambiguities, need clarification, or require more details to proceed effectively. It allows for interactive problem-solving by enabling direct communication with the user. Use this tool judiciously to maintain a balance between gathering necessary information and avoiding excessive back-and-forth. @@ -229,6 +230,7 @@ Your final result description here 14 +${mcpHub ? ` ## Example 3: Requesting to use an MCP tool @@ -247,7 +249,7 @@ Your final result description here weather-server weather://san-francisco/current - +` : ''} # Tool Use Guidelines @@ -272,6 +274,7 @@ By waiting for and carefully considering the user's response after each tool use ==== +${mcpHub ? ` MCP SERVERS The Model Context Protocol (MCP) enables communication between the system and locally running MCP servers that provide additional tools and resources to extend your capabilities. @@ -675,7 +678,7 @@ However some MCP servers may be running from installed packages rather than a lo The user may not always request the use or creation of MCP servers. Instead, they might provide tasks that can be completed with existing tools. While using the MCP SDK to extend your capabilities can be useful, it's important to understand that this is just one specialized type of task you can accomplish. You should only implement MCP servers when the user explicitly requests it (e.g., "add a tool that..."). -Remember: The MCP documentation and example provided above are to help you understand and work with existing MCP servers or create new ones when requested by the user. You already have access to tools and capabilities that can be used to accomplish a wide range of tasks. +Remember: The MCP documentation and example provided above are to help you understand and work with existing MCP servers or create new ones when requested by the user. You already have access to tools and capabilities that can be used to accomplish a wide range of tasks.` : ''} ==== @@ -693,7 +696,9 @@ CAPABILITIES ? "\n- You can use the browser_action tool to interact with websites (including html files and locally running development servers) through a Puppeteer-controlled browser when you feel it is necessary in accomplishing the user's task. This tool is particularly useful for web development tasks as it allows you to launch a browser, navigate to pages, interact with elements through clicks and keyboard input, and capture the results through screenshots and console logs. This tool may be useful at key stages of web development tasks-such as after implementing new features, making substantial changes, when troubleshooting issues, or to verify the result of your work. You can analyze the provided screenshots to ensure correct rendering or identify errors, and review console logs for runtime issues.\n - For example, if asked to add a component to a react website, you might create the necessary files, use execute_command to run the site locally, then use browser_action to launch the browser, navigate to the local server, and verify the component renders & functions correctly before closing the browser." : "" } +${mcpHub ? ` - You have access to MCP servers that may provide additional tools and resources. Each server may provide different capabilities that you can use to accomplish tasks more effectively. +` : ''} ==== diff --git a/src/core/webview/ClineProvider.ts b/src/core/webview/ClineProvider.ts index f2fae33..45f9d06 100644 --- a/src/core/webview/ClineProvider.ts +++ b/src/core/webview/ClineProvider.ts @@ -78,7 +78,7 @@ type GlobalStateKey = | "preferredLanguage" // Language setting for Cline's communication | "writeDelayMs" | "terminalOutputLineLimit" - + | "mcpEnabled" export const GlobalFileNames = { apiConversationHistory: "api_conversation_history.json", uiMessages: "ui_messages.json", @@ -606,6 +606,11 @@ export class ClineProvider implements vscode.WebviewViewProvider { } break } + case "mcpEnabled": + const mcpEnabled = message.bool ?? true + await this.updateGlobalState("mcpEnabled", mcpEnabled) + await this.postStateToWebview() + break case "playSound": if (message.audioType) { const soundPath = path.join(this.context.extensionPath, "audio", `${message.audioType}.wav`) @@ -1056,6 +1061,7 @@ export class ClineProvider implements vscode.WebviewViewProvider { writeDelayMs, terminalOutputLineLimit, fuzzyMatchThreshold, + mcpEnabled, } = await this.getState() const allowedCommands = vscode.workspace @@ -1087,6 +1093,7 @@ export class ClineProvider implements vscode.WebviewViewProvider { writeDelayMs: writeDelayMs ?? 1000, terminalOutputLineLimit: terminalOutputLineLimit ?? 500, fuzzyMatchThreshold: fuzzyMatchThreshold ?? 1.0, + mcpEnabled: mcpEnabled ?? true, } } @@ -1188,6 +1195,7 @@ export class ClineProvider implements vscode.WebviewViewProvider { writeDelayMs, screenshotQuality, terminalOutputLineLimit, + mcpEnabled, ] = await Promise.all([ this.getGlobalState("apiProvider") as Promise, this.getGlobalState("apiModelId") as Promise, @@ -1234,6 +1242,7 @@ export class ClineProvider implements vscode.WebviewViewProvider { this.getGlobalState("writeDelayMs") as Promise, this.getGlobalState("screenshotQuality") as Promise, this.getGlobalState("terminalOutputLineLimit") as Promise, + this.getGlobalState("mcpEnabled") as Promise, ]) let apiProvider: ApiProvider @@ -1324,6 +1333,7 @@ export class ClineProvider implements vscode.WebviewViewProvider { // Return mapped language or default to English return langMap[vscodeLang.split('-')[0]] ?? 'English'; })(), + mcpEnabled: mcpEnabled ?? true, } } diff --git a/src/core/webview/__tests__/ClineProvider.test.ts b/src/core/webview/__tests__/ClineProvider.test.ts index af75631..2add65e 100644 --- a/src/core/webview/__tests__/ClineProvider.test.ts +++ b/src/core/webview/__tests__/ClineProvider.test.ts @@ -255,6 +255,7 @@ describe('ClineProvider', () => { writeDelayMs: 1000, browserViewportSize: "900x600", fuzzyMatchThreshold: 1.0, + mcpEnabled: true, } const message: ExtensionMessage = { diff --git a/src/shared/ExtensionMessage.ts b/src/shared/ExtensionMessage.ts index 3acc740..c00fa6c 100644 --- a/src/shared/ExtensionMessage.ts +++ b/src/shared/ExtensionMessage.ts @@ -62,6 +62,7 @@ export interface ExtensionState { preferredLanguage: string writeDelayMs: number terminalOutputLineLimit?: number + mcpEnabled: boolean } export interface ClineMessage { diff --git a/src/shared/WebviewMessage.ts b/src/shared/WebviewMessage.ts index bfa659a..ee602ed 100644 --- a/src/shared/WebviewMessage.ts +++ b/src/shared/WebviewMessage.ts @@ -49,6 +49,7 @@ export interface WebviewMessage { | "draggedImages" | "deleteMessage" | "terminalOutputLineLimit" + | "mcpEnabled" text?: string disabled?: boolean askResponse?: ClineAskResponse diff --git a/webview-ui/src/components/mcp/McpEnabledToggle.tsx b/webview-ui/src/components/mcp/McpEnabledToggle.tsx new file mode 100644 index 0000000..41c94c7 --- /dev/null +++ b/webview-ui/src/components/mcp/McpEnabledToggle.tsx @@ -0,0 +1,34 @@ +import { VSCodeCheckbox } from "@vscode/webview-ui-toolkit/react" +import { FormEvent } from "react" +import { useExtensionState } from "../../context/ExtensionStateContext" +import { vscode } from "../../utils/vscode" + +const McpEnabledToggle = () => { + const { mcpEnabled, setMcpEnabled } = useExtensionState() + + const handleChange = (e: Event | FormEvent) => { + const target = ('target' in e ? e.target : null) as HTMLInputElement | null + if (!target) return + setMcpEnabled(target.checked) + vscode.postMessage({ type: "mcpEnabled", bool: target.checked }) + } + + return ( +
+ + Enable MCP Servers + +

+ When enabled, Cline will be able to interact with MCP servers for advanced functionality. If you're not using MCP, you can disable this to reduce Cline's token usage. +

+
+ ) +} + +export default McpEnabledToggle \ No newline at end of file diff --git a/webview-ui/src/components/mcp/McpView.tsx b/webview-ui/src/components/mcp/McpView.tsx index 318cbab..ea2ea2b 100644 --- a/webview-ui/src/components/mcp/McpView.tsx +++ b/webview-ui/src/components/mcp/McpView.tsx @@ -11,13 +11,14 @@ import { useExtensionState } from "../../context/ExtensionStateContext" import { McpServer } from "../../../../src/shared/mcp" import McpToolRow from "./McpToolRow" import McpResourceRow from "./McpResourceRow" +import McpEnabledToggle from "./McpEnabledToggle" type McpViewProps = { onDone: () => void } const McpView = ({ onDone }: McpViewProps) => { - const { mcpServers: servers, alwaysAllowMcp } = useExtensionState() + const { mcpServers: servers, alwaysAllowMcp, mcpEnabled } = useExtensionState() // const [servers, setServers] = useState([ // // Add some mock servers for testing // { @@ -106,7 +107,7 @@ const McpView = ({ onDone }: McpViewProps) => { style={{ color: "var(--vscode-foreground)", fontSize: "13px", - marginBottom: "20px", + marginBottom: "10px", marginTop: "5px", }}> The{" "} @@ -122,27 +123,33 @@ const McpView = ({ onDone }: McpViewProps) => { npm docs"). - {/* Server List */} - {servers.length > 0 && ( -
- {servers.map((server) => ( - - ))} -
- )} + - {/* Edit Settings Button */} -
- { - vscode.postMessage({ type: "openMcpSettings" }) - }}> - - Edit MCP Settings - -
+ {mcpEnabled && ( + <> + {/* Server List */} + {servers.length > 0 && ( +
+ {servers.map((server) => ( + + ))} +
+ )} + + {/* Edit Settings Button */} +
+ { + vscode.postMessage({ type: "openMcpSettings" }) + }}> + + Edit MCP Settings + +
+ + )} {/* Bottom padding */}
diff --git a/webview-ui/src/components/settings/SettingsView.tsx b/webview-ui/src/components/settings/SettingsView.tsx index 7ac1563..645a260 100644 --- a/webview-ui/src/components/settings/SettingsView.tsx +++ b/webview-ui/src/components/settings/SettingsView.tsx @@ -4,6 +4,7 @@ import { useExtensionState } from "../../context/ExtensionStateContext" import { validateApiConfiguration, validateModelId } from "../../utils/validate" import { vscode } from "../../utils/vscode" import ApiOptions from "./ApiOptions" +import McpEnabledToggle from "../mcp/McpEnabledToggle" const IS_DEV = false // FIXME: use flags when packaging @@ -48,6 +49,7 @@ const SettingsView = ({ onDone }: SettingsViewProps) => { setScreenshotQuality, terminalOutputLineLimit, setTerminalOutputLineLimit, + mcpEnabled, } = useExtensionState() const [apiErrorMessage, setApiErrorMessage] = useState(undefined) const [modelIdErrorMessage, setModelIdErrorMessage] = useState(undefined) @@ -79,6 +81,7 @@ const SettingsView = ({ onDone }: SettingsViewProps) => { vscode.postMessage({ type: "writeDelayMs", value: writeDelayMs }) vscode.postMessage({ type: "screenshotQuality", value: screenshotQuality ?? 75 }) vscode.postMessage({ type: "terminalOutputLineLimit", value: terminalOutputLineLimit ?? 500 }) + vscode.postMessage({ type: "mcpEnabled", bool: mcpEnabled }) onDone() } } @@ -135,12 +138,13 @@ const SettingsView = ({ onDone }: SettingsViewProps) => { paddingRight: 17, }}> -

Provider Settings

+

Settings

Done
+

Provider Settings

{ }}> These instructions are added to the end of the system prompt sent with every request. Custom instructions set in .clinerules and .cursorrules in the working directory are also included.

+ +
diff --git a/webview-ui/src/context/ExtensionStateContext.tsx b/webview-ui/src/context/ExtensionStateContext.tsx index 36ad4db..45e0614 100644 --- a/webview-ui/src/context/ExtensionStateContext.tsx +++ b/webview-ui/src/context/ExtensionStateContext.tsx @@ -41,6 +41,8 @@ export interface ExtensionStateContextType extends ExtensionState { setScreenshotQuality: (value: number) => void terminalOutputLineLimit?: number setTerminalOutputLineLimit: (value: number) => void + mcpEnabled: boolean + setMcpEnabled: (value: boolean) => void } export const ExtensionStateContext = createContext(undefined) @@ -61,6 +63,7 @@ export const ExtensionStateContextProvider: React.FC<{ children: React.ReactNode browserViewportSize: "900x600", screenshotQuality: 75, terminalOutputLineLimit: 500, + mcpEnabled: true, }) const [didHydrateState, setDidHydrateState] = useState(false) const [showWelcome, setShowWelcome] = useState(false) @@ -180,6 +183,7 @@ export const ExtensionStateContextProvider: React.FC<{ children: React.ReactNode setWriteDelayMs: (value) => setState((prevState) => ({ ...prevState, writeDelayMs: value })), setScreenshotQuality: (value) => setState((prevState) => ({ ...prevState, screenshotQuality: value })), setTerminalOutputLineLimit: (value) => setState((prevState) => ({ ...prevState, terminalOutputLineLimit: value })), + setMcpEnabled: (value) => setState((prevState) => ({ ...prevState, mcpEnabled: value })), } return {children}