mirror of
https://github.com/pacnpal/Roo-Code.git
synced 2025-12-20 04:11:10 -05:00
Refactor modes
This commit is contained in:
Matt Rubens
@@ -1,8 +1,125 @@
|
||||
export const codeMode = 'code' as const;
|
||||
export const architectMode = 'architect' as const;
|
||||
export const askMode = 'ask' as const;
|
||||
// Tool options for specific tools
|
||||
export type ToolOptions = {
|
||||
string: readonly string[];
|
||||
}
|
||||
|
||||
export type Mode = typeof codeMode | typeof architectMode | typeof askMode;
|
||||
// Tool configuration tuple type
|
||||
export type ToolConfig = readonly [string] | readonly [string, ToolOptions];
|
||||
|
||||
// Mode types
|
||||
export type Mode = string;
|
||||
|
||||
// Mode configuration type
|
||||
export type ModeConfig = {
|
||||
slug: string;
|
||||
name: string;
|
||||
roleDefinition: string;
|
||||
tools: readonly ToolConfig[];
|
||||
}
|
||||
|
||||
// Separate enhance prompt type and definition
|
||||
export type EnhanceConfig = {
|
||||
prompt: string;
|
||||
}
|
||||
|
||||
export const enhance: EnhanceConfig = {
|
||||
prompt: "Generate an enhanced version of this prompt (reply with only the enhanced prompt - no conversation, explanations, lead-in, bullet points, placeholders, or surrounding quotes):"
|
||||
} as const;
|
||||
|
||||
// Main modes configuration as an ordered array
|
||||
export const modes: readonly ModeConfig[] = [
|
||||
{
|
||||
slug: 'code',
|
||||
name: 'Code',
|
||||
roleDefinition: "You are Cline, a highly skilled software engineer with extensive knowledge in many programming languages, frameworks, design patterns, and best practices.",
|
||||
tools: [
|
||||
['execute_command'],
|
||||
['read_file'],
|
||||
['write_to_file'],
|
||||
['apply_diff'],
|
||||
['search_files'],
|
||||
['list_files'],
|
||||
['list_code_definition_names'],
|
||||
['browser_action'],
|
||||
['use_mcp_tool'],
|
||||
['access_mcp_resource'],
|
||||
['ask_followup_question'],
|
||||
['attempt_completion'],
|
||||
] as const
|
||||
},
|
||||
{
|
||||
slug: 'architect',
|
||||
name: 'Architect',
|
||||
roleDefinition: "You are Cline, a software architecture expert specializing in analyzing codebases, identifying patterns, and providing high-level technical guidance. You excel at understanding complex systems, evaluating architectural decisions, and suggesting improvements while maintaining a read-only approach to the codebase. Make sure to help the user come up with a solid implementation plan for their project and don't rush to switch to implementing code.",
|
||||
tools: [
|
||||
['read_file'],
|
||||
['search_files'],
|
||||
['list_files'],
|
||||
['list_code_definition_names'],
|
||||
['browser_action'],
|
||||
['use_mcp_tool'],
|
||||
['access_mcp_resource'],
|
||||
['ask_followup_question'],
|
||||
['attempt_completion'],
|
||||
] as const
|
||||
},
|
||||
{
|
||||
slug: 'ask',
|
||||
name: 'Ask',
|
||||
roleDefinition: "You are Cline, a knowledgeable technical assistant focused on answering questions and providing information about software development, technology, and related topics. You can analyze code, explain concepts, and access external resources while maintaining a read-only approach to the codebase. Make sure to answer the user's questions and don't rush to switch to implementing code.",
|
||||
tools: [
|
||||
['read_file'],
|
||||
['search_files'],
|
||||
['list_files'],
|
||||
['list_code_definition_names'],
|
||||
['browser_action'],
|
||||
['use_mcp_tool'],
|
||||
['access_mcp_resource'],
|
||||
['ask_followup_question'],
|
||||
['attempt_completion'],
|
||||
] as const
|
||||
},
|
||||
] as const;
|
||||
|
||||
// Export the default mode slug
|
||||
export const defaultModeSlug = modes[0].slug;
|
||||
|
||||
// Helper functions
|
||||
export function getModeBySlug(slug: string): ModeConfig | undefined {
|
||||
return modes.find(mode => mode.slug === slug);
|
||||
}
|
||||
|
||||
export function getModeConfig(slug: string): ModeConfig {
|
||||
const mode = getModeBySlug(slug);
|
||||
if (!mode) {
|
||||
throw new Error(`No mode found for slug: ${slug}`);
|
||||
}
|
||||
return mode;
|
||||
}
|
||||
|
||||
// Derive tool names from the modes configuration
|
||||
export type ToolName = typeof modes[number]['tools'][number][0];
|
||||
export type TestToolName = ToolName | 'unknown_tool';
|
||||
|
||||
export function isToolAllowedForMode(tool: TestToolName, modeSlug: string): boolean {
|
||||
if (tool === 'unknown_tool') {
|
||||
return false;
|
||||
}
|
||||
const mode = getModeBySlug(modeSlug);
|
||||
if (!mode) {
|
||||
return false;
|
||||
}
|
||||
return mode.tools.some(([toolName]) => toolName === tool);
|
||||
}
|
||||
|
||||
export function getToolOptions(tool: ToolName, modeSlug: string): ToolOptions | undefined {
|
||||
const mode = getModeBySlug(modeSlug);
|
||||
if (!mode) {
|
||||
return undefined;
|
||||
}
|
||||
const toolConfig = mode.tools.find(([toolName]) => toolName === tool);
|
||||
return toolConfig?.[1];
|
||||
}
|
||||
|
||||
export type PromptComponent = {
|
||||
roleDefinition?: string;
|
||||
@@ -10,21 +127,26 @@ export type PromptComponent = {
|
||||
}
|
||||
|
||||
export type CustomPrompts = {
|
||||
ask?: PromptComponent;
|
||||
code?: PromptComponent;
|
||||
architect?: PromptComponent;
|
||||
enhance?: string;
|
||||
[key: string]: PromptComponent | string | undefined;
|
||||
}
|
||||
|
||||
export const defaultPrompts = {
|
||||
[askMode]: {
|
||||
roleDefinition: "You are Cline, a knowledgeable technical assistant focused on answering questions and providing information about software development, technology, and related topics. You can analyze code, explain concepts, and access external resources while maintaining a read-only approach to the codebase. Make sure to answer the user's questions and don't rush to switch to implementing code.",
|
||||
},
|
||||
[codeMode]: {
|
||||
roleDefinition: "You are Cline, a highly skilled software engineer with extensive knowledge in many programming languages, frameworks, design patterns, and best practices.",
|
||||
},
|
||||
[architectMode]: {
|
||||
roleDefinition: "You are Cline, a software architecture expert specializing in analyzing codebases, identifying patterns, and providing high-level technical guidance. You excel at understanding complex systems, evaluating architectural decisions, and suggesting improvements while maintaining a read-only approach to the codebase. Make sure to help the user come up with a solid implementation plan for their project and don't rush to switch to implementing code.",
|
||||
},
|
||||
enhance: "Generate an enhanced version of this prompt (reply with only the enhanced prompt - no conversation, explanations, lead-in, bullet points, placeholders, or surrounding quotes):"
|
||||
} as const;
|
||||
// Create the defaultPrompts object with the correct type
|
||||
export const defaultPrompts: CustomPrompts = {
|
||||
...Object.fromEntries(modes.map(mode => [
|
||||
mode.slug,
|
||||
{ roleDefinition: mode.roleDefinition }
|
||||
])),
|
||||
enhance: enhance.prompt
|
||||
} as const;
|
||||
|
||||
// Helper function to safely get role definition
|
||||
export function getRoleDefinition(modeSlug: string): string {
|
||||
const prompt = defaultPrompts[modeSlug];
|
||||
if (!prompt || typeof prompt === 'string') {
|
||||
throw new Error(`Invalid mode slug: ${modeSlug}`);
|
||||
}
|
||||
if (!prompt.roleDefinition) {
|
||||
throw new Error(`No role definition found for mode: ${modeSlug}`);
|
||||
}
|
||||
return prompt.roleDefinition;
|
||||
}
|
||||
Reference in New Issue
Block a user