Merge pull request #165 from drivecore/feature/disable-user-prompt

Add enableUserPrompt command line option

Author: bhouston

.changeset/enable-user-prompt-option.md

@@ -0,0 +1,6 @@
+---
+'mycoder': patch
+'mycoder-agent': patch
+---
+
+Add `--enableUserPrompt` command line option that defaults to true but can be set to false to disable the userPrompt tool for fully automated sessions.

README.md

@@ -34,6 +34,9 @@ mycoder "Implement a React component that displays a list of items"
 # Run with a prompt from a file
 mycoder -f prompt.txt
 
+# Disable user prompts for fully automated sessions
+mycoder --enableUserPrompt false "Generate a basic Express.js server"
+
 # Enable GitHub mode
 mycoder config set githubMode true
 ```

packages/agent/src/core/types.ts

@@ -19,6 +19,7 @@ export type ToolContext = {
   githubMode: boolean;
   customPrompt?: string;
   tokenCache?: boolean;
+  enableUserPrompt?: boolean;
 };
 
 export type Tool<TParams = Record<string, any>, TReturn = any> = {

packages/agent/src/tools/getTools.ts

@@ -1,5 +1,6 @@
 import { Tool } from '../core/types.js';
 
+// Import tools
 import { browseMessageTool } from './browser/browseMessage.js';
 import { browseStartTool } from './browser/browseStart.js';
 import { subAgentTool } from './interaction/subAgent.js';
@@ -12,18 +13,33 @@ import { shellMessageTool } from './system/shellMessage.js';
 import { shellStartTool } from './system/shellStart.js';
 import { sleepTool } from './system/sleep.js';
 
-export function getTools(): Tool[] {
-  return [
-    textEditorTool,
-    subAgentTool,
-    userPromptTool,
-    sequenceCompleteTool,
-    fetchTool,
-    shellStartTool,
-    shellMessageTool,
-    browseStartTool,
-    browseMessageTool,
-    respawnTool,
-    sleepTool,
-  ] as Tool[];
+// Import these separately to avoid circular dependencies
+
+interface GetToolsOptions {
+  enableUserPrompt?: boolean;
+}
+
+export function getTools(options?: GetToolsOptions): Tool[] {
+  const enableUserPrompt = options?.enableUserPrompt !== false; // Default to true if not specified
+
+  // Force cast to Tool type to avoid TypeScript issues
+  const tools: Tool[] = [
+    textEditorTool as unknown as Tool,
+    subAgentTool as unknown as Tool,
+    sequenceCompleteTool as unknown as Tool,
+    fetchTool as unknown as Tool,
+    shellStartTool as unknown as Tool,
+    shellMessageTool as unknown as Tool,
+    browseStartTool as unknown as Tool,
+    browseMessageTool as unknown as Tool,
+    respawnTool as unknown as Tool,
+    sleepTool as unknown as Tool,
+  ];
+
+  // Only include userPrompt tool if enabled
+  if (enableUserPrompt) {
+    tools.push(userPromptTool as unknown as Tool);
+  }
+
+  return tools;
 }

packages/agent/src/tools/interaction/subAgent.ts

@@ -90,7 +90,7 @@ export const subAgentTool: Tool<Parameters, ReturnType> = {
       .filter(Boolean)
       .join('\n');
 
-    const tools = getTools().filter((tool) => tool.name !== 'userPrompt');
+    const tools = getTools({ enableUserPrompt: false });
 
     // Update config if timeout is specified
     const config = {

packages/cli/src/commands/$default.ts

@@ -170,7 +170,10 @@ export const command: CommandModule<SharedOptions, DefaultArgs> = {
         'Once the task is complete ask the user, via the userPrompt tool if the results are acceptable or if changes are needed or if there are additional follow on tasks.',
       ].join('\n');
 
-      const tools = getTools();
+      const tools = getTools({
+        enableUserPrompt:
+          argv.enableUserPrompt !== undefined ? argv.enableUserPrompt : true,
+      });
 
       // Error handling
       process.on('SIGINT', () => {
@@ -210,6 +213,8 @@ export const command: CommandModule<SharedOptions, DefaultArgs> = {
         customPrompt: config.customPrompt,
         tokenCache:
           argv.tokenCache !== undefined ? argv.tokenCache : config.tokenCache,
+        enableUserPrompt:
+          argv.enableUserPrompt !== undefined ? argv.enableUserPrompt : true,
       });
 
       const output =

packages/cli/src/options.ts

@@ -16,6 +16,7 @@ export type SharedOptions = {
   readonly temperature?: number;
   readonly profile?: boolean;
   readonly tokenCache?: boolean;
+  readonly enableUserPrompt?: boolean;
 };
 
 export const sharedOptions = {
@@ -102,4 +103,10 @@ export const sharedOptions = {
     type: 'boolean',
     description: 'Enable token caching for LLM API calls',
   } as const,
+  enableUserPrompt: {
+    type: 'boolean',
+    description:
+      'Enable or disable the userPrompt tool (disable for fully automated sessions)',
+    default: true,
+  } as const,
 };