Merge pull request #347 from drivecore/feature/sub-agent-modes-344

bhouston · web-flow · commit 5794367511ad · 2025-03-21T21:49:21.000-04:00
feat: implement sub-agent workflow modes (disabled, sync, async)
diff --git a/mycoder.config.js b/mycoder.config.js
@@ -20,6 +20,9 @@ export default {
     // executablePath: null, // e.g., '/path/to/chrome'
   },
 
+  // Sub-agent workflow mode: 'disabled' (default), 'sync' (experimental), or 'async' (experimental)
+  subAgentMode: 'disabled',
+
   // Model settings
   //provider: 'anthropic',
   //model: 'claude-3-7-sonnet-20250219',
diff --git a/packages/agent/src/tools/getTools.ts b/packages/agent/src/tools/getTools.ts
@@ -3,6 +3,7 @@ import { Tool } from '../core/types.js';
 
 // Import tools
 import { agentDoneTool } from './agent/agentDone.js';
+import { agentExecuteTool } from './agent/agentExecute.js';
 import { agentMessageTool } from './agent/agentMessage.js';
 import { agentStartTool } from './agent/agentStart.js';
 import { listAgentsTool } from './agent/listAgents.js';
@@ -21,38 +22,52 @@ import { textEditorTool } from './textEditor/textEditor.js';
 
 // Import these separately to avoid circular dependencies
 
+/**
+ * Sub-agent workflow modes
+ * - disabled: No sub-agent tools are available
+ * - sync: Parent agent waits for sub-agent completion before continuing
+ * - async: Sub-agents run in the background, parent can check status and provide guidance
+ */
+export type SubAgentMode = 'disabled' | 'sync' | 'async';
+
 interface GetToolsOptions {
   userPrompt?: boolean;
   mcpConfig?: McpConfig;
+  subAgentMode?: SubAgentMode;
 }
 
 export function getTools(options?: GetToolsOptions): Tool[] {
   const userPrompt = options?.userPrompt !== false; // Default to true if not specified
   const mcpConfig = options?.mcpConfig || { servers: [], defaultResources: [] };
+  const subAgentMode = options?.subAgentMode || 'disabled'; // Default to disabled mode
 
   // Force cast to Tool type to avoid TypeScript issues
   const tools: Tool[] = [
     textEditorTool as unknown as Tool,
-
-    //agentExecuteTool as unknown as Tool,
-    agentStartTool as unknown as Tool,
-    agentMessageTool as unknown as Tool,
-    listAgentsTool as unknown as Tool,
-    agentDoneTool as unknown as Tool,
-
     fetchTool as unknown as Tool,
-
     shellStartTool as unknown as Tool,
     shellMessageTool as unknown as Tool,
     listShellsTool as unknown as Tool,
-
     sessionStartTool as unknown as Tool,
     sessionMessageTool as unknown as Tool,
     listSessionsTool as unknown as Tool,
-
     waitTool as unknown as Tool,
   ];
 
+  // Add agent tools based on the configured mode
+  if (subAgentMode === 'sync') {
+    // For sync mode, include only agentExecute and agentDone
+    tools.push(agentExecuteTool as unknown as Tool);
+    tools.push(agentDoneTool as unknown as Tool);
+  } else if (subAgentMode === 'async') {
+    // For async mode, include all async agent tools
+    tools.push(agentStartTool as unknown as Tool);
+    tools.push(agentMessageTool as unknown as Tool);
+    tools.push(listAgentsTool as unknown as Tool);
+    tools.push(agentDoneTool as unknown as Tool);
+  }
+  // For 'disabled' mode, no agent tools are added
+
   // Only include user interaction tools if enabled
   if (userPrompt) {
     tools.push(userPromptTool as unknown as Tool);
diff --git a/packages/cli/src/commands/$default.ts b/packages/cli/src/commands/$default.ts
@@ -158,6 +158,7 @@ export async function executePrompt(
     const tools = getTools({
       userPrompt: config.userPrompt,
       mcpConfig: config.mcp,
+      subAgentMode: config.subAgentMode,
     });
 
     // Error handling
diff --git a/packages/cli/src/commands/tools.ts b/packages/cli/src/commands/tools.ts
@@ -41,7 +41,7 @@ export const command: CommandModule<object, ToolsArgs> = {
   describe: 'List all available tools and their capabilities',
   handler: () => {
     try {
-      const tools = getTools();
+      const tools = getTools({ subAgentMode: 'disabled' });
 
       console.log('Available Tools:\n');
 
diff --git a/packages/cli/src/options.ts b/packages/cli/src/options.ts
@@ -17,6 +17,7 @@ export type SharedOptions = {
   readonly githubMode?: boolean;
   readonly upgradeCheck?: boolean;
   readonly ollamaBaseUrl?: string;
+  readonly subAgentMode?: 'disabled' | 'sync' | 'async';
 };
 
 export const sharedOptions = {
@@ -100,4 +101,9 @@ export const sharedOptions = {
     type: 'string',
     description: 'Base URL for Ollama API (default: http://localhost:11434)',
   } as const,
+  subAgentMode: {
+    type: 'string',
+    description: 'Sub-agent workflow mode (disabled, sync, or async)',
+    choices: ['disabled', 'sync', 'async'],
+  } as const,
 };
diff --git a/packages/cli/src/settings/config.ts b/packages/cli/src/settings/config.ts
@@ -20,6 +20,7 @@ export type Config = {
   upgradeCheck: boolean;
   tokenUsage: boolean;
   interactive: boolean;
+  subAgentMode?: 'disabled' | 'sync' | 'async';
 
   baseUrl?: string;
 
@@ -77,6 +78,7 @@ const defaultConfig: Config = {
   upgradeCheck: true,
   tokenUsage: false,
   interactive: false,
+  subAgentMode: 'disabled',
 
   // MCP configuration
   mcp: {
@@ -103,6 +105,7 @@ export const getConfigFromArgv = (argv: ArgumentsCamelCase<SharedOptions>) => {
     upgradeCheck: argv.upgradeCheck,
     tokenUsage: argv.tokenUsage,
     interactive: argv.interactive,
+    subAgentMode: argv.subAgentMode,
   };
 };
 
diff --git a/packages/docs/docs/usage/configuration.md b/packages/docs/docs/usage/configuration.md
@@ -118,10 +118,11 @@ export default {
 
 ### Behavior Customization
 
-| Option         | Description                    | Possible Values | Default |
-| -------------- | ------------------------------ | --------------- | ------- |
-| `customPrompt` | Custom instructions for the AI | Any string      | `""`    |
-| `githubMode`   | Enable GitHub integration      | `true`, `false` | `false` |
+| Option         | Description                    | Possible Values                   | Default   |
+| -------------- | ------------------------------ | --------------------------------- | --------- |
+| `customPrompt` | Custom instructions for the AI | Any string                                          | `""`        |
+| `githubMode`   | Enable GitHub integration      | `true`, `false`                                 | `false`     |
+| `subAgentMode` | Sub-agent workflow mode        | `'disabled'`, `'sync'` (experimental), `'async'` (experimental) | `'disabled'` |
 
 Example:
 
@@ -209,5 +210,8 @@ export default {
   profile: true,
   tokenUsage: true,
   tokenCache: true,
+
+  // Sub-agent workflow mode
+  subAgentMode: 'disabled', // Options: 'disabled', 'sync' (experimental), 'async' (experimental)
 };
 ```
diff --git a/packages/docs/docs/usage/sub-agent-modes.md b/packages/docs/docs/usage/sub-agent-modes.md
@@ -0,0 +1,119 @@
+---
+sidebar_position: 9
+---
+
+# Sub-Agent Workflow Modes
+
+MyCoder supports different modes for working with sub-agents, giving you flexibility in how tasks are distributed and executed. You can configure the sub-agent workflow mode based on your specific needs and resource constraints.
+
+## Available Modes
+
+MyCoder supports three distinct sub-agent workflow modes:
+
+### 1. Disabled Mode (Default)
+
+In this mode, sub-agent functionality is completely disabled:
+
+- No sub-agent tools are available to the main agent
+- All tasks must be handled by the main agent directly
+- Useful for simpler tasks or when resource constraints are a concern
+- Reduces memory usage and API costs for straightforward tasks
+
+### 2. Synchronous Mode ("sync") - Experimental
+
+In synchronous mode, the parent agent waits for sub-agents to complete before continuing:
+
+- Uses the `agentExecute` tool for synchronous execution
+- Parent agent waits for sub-agent completion before continuing its own workflow
+- Useful for tasks that require sequential execution
+- Simpler to reason about as there's no parallel execution
+- Good for tasks where later steps depend on the results of earlier steps
+
+### 3. Asynchronous Mode ("async") - Experimental
+
+In asynchronous mode, sub-agents run in parallel with the parent agent:
+
+- Uses `agentStart`, `agentMessage`, and `listAgents` tools
+- Sub-agents run in the background while the parent agent continues its work
+- Parent agent can check status and provide guidance to sub-agents
+- Useful for complex tasks that can benefit from parallelization
+- More efficient for tasks that can be executed concurrently
+- Allows the parent agent to coordinate multiple sub-agents
+
+## Configuration
+
+You can set the sub-agent workflow mode in your `mycoder.config.js` file:
+
+```javascript
+// mycoder.config.js
+export default {
+  // Sub-agent workflow mode: 'disabled', 'sync' (experimental), or 'async' (experimental)
+  subAgentMode: 'disabled', // Default value
+
+  // Other configuration options...
+};
+```
+
+You can also specify the mode via the command line:
+
+```bash
+mycoder --subAgentMode disabled "Implement a simple React component"
+```
+
+## Choosing the Right Mode
+
+Consider these factors when choosing a sub-agent workflow mode:
+
+- **Task Complexity**: For complex tasks that can be broken down into independent parts, async mode is often best. For simpler tasks, disabled mode may be sufficient.
+
+- **Resource Constraints**: Disabled mode uses fewer resources. Async mode can use more memory and API tokens but may complete complex tasks faster.
+
+- **Task Dependencies**: If later steps depend heavily on the results of earlier steps, sync mode ensures proper sequencing.
+
+- **Coordination Needs**: If you need to coordinate multiple parallel workflows, async mode gives you more control.
+
+## Example: Using Different Modes
+
+### Disabled Mode
+
+Best for simple, focused tasks:
+
+```javascript
+// mycoder.config.js
+export default {
+  subAgentMode: 'disabled',
+  // Other settings...
+};
+```
+
+### Synchronous Mode
+
+Good for sequential, dependent tasks:
+
+```javascript
+// mycoder.config.js
+export default {
+  subAgentMode: 'sync',
+  // Other settings...
+};
+```
+
+### Asynchronous Mode
+
+Ideal for complex projects with independent components:
+
+```javascript
+// mycoder.config.js
+export default {
+  subAgentMode: 'async', // Experimental
+  // Other settings...
+};
+```
+
+## How It Works Internally
+
+- In **disabled mode**, no agent tools are added to the available tools list.
+- In **sync mode**, only the `agentExecute` and `agentDone` tools are available, ensuring synchronous execution.
+- In **async mode**, the full suite of agent tools (`agentStart`, `agentMessage`, `listAgents`, and `agentDone`) is available, enabling parallel execution.
+
+This implementation allows MyCoder to adapt to different task requirements while maintaining a consistent interface for users.
