add handleSteps to .agents

Author: charleslien

.agents/file-explorer.ts

@@ -35,6 +35,27 @@ const config: AgentConfig = {
     'You are a file explorer agent that spawns multiple file picker agents in parallel to comprehensively explore the codebase.',
   instructionsPrompt: '',
   stepPrompt: '',
+  handleSteps: function* ({ prompt, params }) {
+    const filePickerPrompts = params.prompts.map(
+        (focusPrompt) =>
+          `Based on the overall goal "${prompt}", find files related to this specific area: ${focusPrompt}`,
+      ),
+      { toolResult: spawnResult } = yield {
+        toolName: 'spawn_agents',
+        args: {
+          agents: filePickerPrompts.map((promptText) => ({
+            agent_type: 'file_picker',
+            prompt: promptText,
+          })),
+        },
+      }
+    yield {
+      toolName: 'set_output',
+      args: {
+        results: spawnResult?.result,
+      },
+    }
+  },
 }
 
 export default config

.agents/file-picker.ts

@@ -21,6 +21,13 @@ const config: AgentConfig = {
     'Provide a short analysis of the locations in the codebase that could be helpful. Focus on the files that are most relevant to the user prompt.\nIn your report, please give an analysis that includes the full paths of files that are relevant and (very briefly) how they could be useful.',
   stepPrompt:
     'Do not use the find_files tool or any tools again. Just give your response.',
+  handleSteps: function* ({ agentState, prompt, params }) {
+    yield {
+      toolName: 'find_files',
+      args: { prompt },
+    }
+    yield 'STEP_ALL'
+  },
 }
 
 export default config

.agents/researcher.ts

@@ -22,6 +22,13 @@ const config: AgentConfig = {
   instructionsPrompt: '',
   stepPrompt:
     "Don't forget to end your response with the end_turn tool: <end_turn></end_turn>",
+  handleSteps: function* ({ agentState, prompt, params }) {
+    yield {
+      toolName: 'web_search',
+      args: { prompt },
+    }
+    yield 'STEP_ALL'
+  },
 }
 
 export default config