general agent: add context pruner, tweak prompts

Author: jahooma

agents/general-agent/general-agent.ts

@@ -19,8 +19,8 @@ export const createGeneralAgent = (options: {
     displayName: isGpt5 ? 'GPT-5 Agent' : 'Opus Agent',
     spawnerPrompt:
       isGpt5 ?
-        'A general-purpose, deep-thinking (and slow) agent that can be used to solve a wide range of problems. Use this to help you solve a specific problem that requires extended reasoning.'
-        : 'A general-purpose capable agent that can be used to solve a wide range of problems. Use this to help you solve any problem.',
+        'A general-purpose, deep-thinking (and slow) agent that can be used to solve a wide range of problems. Use this to help you solve a specific problem that requires extended reasoning. This agent has no context on the conversation history so you must provide all the relevant context (via the prompt or filePaths) for this agent to work well.'
+        : 'A general-purpose capable agent that can be used to solve a wide range of problems. Use this to help you solve any problem. This agent has no context on the conversation history so you must provide all the relevant context (via the prompt or filePaths) for this agent to work well.',
     inputSchema: {
       prompt: {
         type: 'string',
@@ -50,6 +50,7 @@ export const createGeneralAgent = (options: {
       'directory-lister',
       'glob-matcher',
       'commander',
+      'context-pruner',
     ],
     toolNames: [
       'spawn_agents',
@@ -59,7 +60,7 @@ export const createGeneralAgent = (options: {
       'write_file',
     ],
 
-    instructionsPrompt: `Use the spawn_agents tool to spawn agents to help you complete the user request. file-picker is really good at finding relevant files in the codebase and so you should spawn it if at all relevant. You should spawn multiple agents in parallel when possible to speed up the process. (e.g. spawn 3 file-pickers + 1 code-searcher + 1 researcher-web in one spawn_agents call or 3 commanders in one spawn_agents call). Read multiple files at once to speed up the process and get more context.`,
+    instructionsPrompt: `Use the spawn_agents tool to spawn agents to help you complete the user request. If you need to find more information in the codebase, file-picker is really good at finding relevant files. You should spawn multiple agents in parallel when possible to speed up the process. (e.g. spawn 3 file-pickers + 1 code-searcher + 1 researcher-web in one spawn_agents call or 3 commanders in one spawn_agents call).`,
 
     handleSteps: function* ({ params }) {
       const filePaths = params?.filePaths as string[] | undefined
@@ -71,8 +72,20 @@ export const createGeneralAgent = (options: {
         }
       }
 
-      // Allow multiple steps for extended reasoning
-      yield 'STEP_ALL'
+      while (true) {
+        // Run context-pruner before each step
+        yield {
+          toolName: 'spawn_agent_inline',
+          input: {
+            agent_type: 'context-pruner',
+            params: params ?? {},
+          },
+          includeToolCall: false,
+        } as any
+
+        const { stepsComplete } = yield 'STEP'
+        if (stepsComplete) break
+      }
     },
   }
 }