Merge pull request #250 from drivecore/feature/issue-249-multiple-line-custom-prompts

feat: support multiple line custom prompts in mycoder.config.js

Author: bhouston

README.md

@@ -72,7 +72,14 @@ export default {
   temperature: 0.7,
 
   // Custom settings
+  // customPrompt can be a string or an array of strings for multiple lines
   customPrompt: '',
+  // Example of multiple line custom prompts:
+  // customPrompt: [
+  //   'Custom instruction line 1',
+  //   'Custom instruction line 2',
+  //   'Custom instruction line 3',
+  // ],
   profile: false,
   tokenCache: true,
 

mycoder.config.js

@@ -20,7 +20,14 @@ export default {
   temperature: 0.7,
 
   // Custom settings
+  // customPrompt can be a string or an array of strings for multiple lines
   customPrompt: '',
+  // Example of multiple line custom prompts:
+  // customPrompt: [
+  //   'Custom instruction line 1',
+  //   'Custom instruction line 2',
+  //   'Custom instruction line 3',
+  // ],
   profile: false,
   tokenCache: true,
 };

packages/agent/src/core/toolAgent/config.ts

@@ -199,6 +199,13 @@ export function getDefaultSystemPrompt(toolContext: ToolContext): string {
     'When you run into issues or unexpected results, take a step back and read the project documentation and configuration files and look at other source files in the project for examples of what works.',
     '',
     'Use sub-agents for parallel tasks, providing them with specific context they need rather than having them rediscover it.',
-    toolContext.customPrompt ? `\n\n${toolContext.customPrompt}` : '',
+    (() => {
+      if (!toolContext.customPrompt) return '';
+      if (typeof toolContext.customPrompt === 'string') {
+        return `\n\n${toolContext.customPrompt}`;
+      }
+      // It's an array of strings
+      return `\n\n${toolContext.customPrompt.join('\n')}`;
+    })(),
   ].join('\n');
 }

packages/agent/src/core/types.ts

@@ -19,7 +19,7 @@ export type ToolContext = {
   pageFilter: pageFilter;
   tokenTracker: TokenTracker;
   githubMode: boolean;
-  customPrompt?: string;
+  customPrompt?: string | string[];
   tokenCache?: boolean;
   userPrompt?: boolean;
   agentId?: string; // Unique identifier for the agent, used for background tool tracking

packages/cli/src/settings/config.ts

@@ -13,7 +13,7 @@ export type Config = {
   model: string;
   maxTokens: number;
   temperature: number;
-  customPrompt: string;
+  customPrompt: string | string[];
   profile: boolean;
   tokenCache: boolean;
   userPrompt: boolean;