Update agent builder a bit

Author: jahooma

backend/src/templates/agents/agent-builder.ts

@@ -6,7 +6,6 @@ import {
   openrouterModels,
   AGENT_DEFINITION_FILE,
 } from '@codebuff/common/constants'
-import { AgentTemplateTypes } from '@codebuff/common/types/session-state'
 import z from 'zod/v4'
 
 import type { AgentTemplate } from '../types'
@@ -64,7 +63,10 @@ export const agentBuilder = (
 
       files
         .filter(
-          (file) => file.endsWith('.ts') && file.startsWith('diff-reviewer'),
+          (file) =>
+            file.endsWith('.ts') &&
+            (file.startsWith('diff-reviewer') ||
+              file === 'your-custom-agent.ts'),
         )
         .forEach((filename) => {
           try {
@@ -119,15 +121,7 @@ export const agentBuilder = (
       'set_output',
       'end_turn',
     ] satisfies ToolName[],
-    spawnableAgents: allAvailableAgents
-      ? (allAvailableAgents as any[])
-      : [
-          AgentTemplateTypes.file_picker,
-          AgentTemplateTypes.researcher,
-          AgentTemplateTypes.thinker,
-          AgentTemplateTypes.reviewer,
-          AgentTemplateTypes.agent_builder,
-        ],
+    spawnableAgents: [],
 
     systemPrompt: [
       '# Bob the Agent Builder',
@@ -270,11 +264,17 @@ IMPORTANT: Always end your response with the end_turn tool when you have complet
       // Step 5: Copy example agent files to .agents/ directory
       for (const [filename, content] of Object.entries(exampleAgentContents)) {
         if (content) {
+          // Copy your-custom-agent.ts to top level .agents directory, others to examples
+          const targetPath =
+            filename === 'your-custom-agent.ts'
+              ? `${AGENT_TEMPLATES_DIR}/${filename}`
+              : `${EXAMPLES_DIR}/${filename}`
+
           yield {
             toolName: 'write_file',
             args: {
-              path: `${EXAMPLES_DIR}/${filename}`,
-              instructions: `Copy example agent file ${filename}`,
+              path: targetPath,
+              instructions: `Copy ${filename === 'your-custom-agent.ts' ? 'custom agent template' : 'example agent'} file ${filename}`,
               content: content,
             },
           }

common/src/constants/agents.ts

@@ -24,12 +24,6 @@ export const AGENT_PERSONAS = {
     purpose: 'Base agent that orchestrates the full response.',
   } as const,
 
-  base_agent_builder: {
-    displayName: 'Buffy the Enthusiastic Agent Builder',
-    purpose:
-      'Enhanced base agent that can create custom agents and handle all coding tasks',
-  } as const,
-
   superagent: {
     displayName: 'Superagent',
     purpose:
@@ -77,7 +71,7 @@ export const AGENT_PERSONAS = {
   } as const,
   agent_builder: {
     displayName: 'Bob the Agent Builder',
-    purpose: 'Creates new agent templates for the codebuff mult-agent system',
+    purpose: 'Creates new agent templates for the codebuff multi-agent system',
     hidden: false,
   } as const,
 } as const satisfies Partial<

common/src/util/examples/diff-reviewer-1.ts

@@ -2,9 +2,8 @@ import type { AgentDefinition } from '../types/agent-definition'
 
 const definition: AgentDefinition = {
   id: 'diff-reviewer-1',
-
   displayName: 'Diff Reviewer (Level 1)',
-  model: 'openai/gpt-5',
+  model: 'anthropic/claude-4-sonnet-20250522',
   toolNames: ['read_files', 'run_terminal_command'],
 
   spawnPurposePrompt:

common/src/util/examples/diff-reviewer-2.ts

@@ -6,7 +6,7 @@ import type {
 const definition: AgentDefinition = {
   id: 'diff-reviewer-2',
   displayName: 'Diff Reviewer (Level 2)',
-  model: 'openai/gpt-5',
+  model: 'anthropic/claude-4-sonnet-20250522',
 
   inputSchema: {
     prompt: {

common/src/util/examples/diff-reviewer-3.ts

@@ -5,9 +5,8 @@ import type {
 
 const definition: AgentDefinition = {
   id: 'diff-reviewer-3',
-
   displayName: 'Diff Reviewer (Level 3)',
-  model: 'openai/gpt-5',
+  model: 'anthropic/claude-4-sonnet-20250522',
   inputSchema: {
     prompt: {
       type: 'string',
@@ -18,7 +17,7 @@ const definition: AgentDefinition = {
   outputMode: 'last_message',
 
   toolNames: ['read_files', 'run_terminal_command', 'spawn_agents'],
-  spawnableAgents: ['james/file-explorer@0.1.3'],
+  spawnableAgents: ['codebuff/file-explorer@0.0.1'],
 
   spawnPurposePrompt:
     'Spawn when you need to review code changes in the git diff',
@@ -77,23 +76,11 @@ Use the following guidelines while reviewing the changes:
       args: {
         role: 'assistant',
         content:
-          'Now I will spawn a file explorer to find any missing codebase context.',
-      },
-    }
-
-    yield 'STEP'
-
-    // Step 5: Put words in the AI's mouth to review the changes.
-    yield {
-      toolName: 'add_message',
-      args: {
-        role: 'assistant',
-        content: 'Here is my comprehensive review of the changes.',
+          'Now I will spawn a file explorer to find any missing codebase context, and then review the changes.',
       },
     }
 
-    // Step 6: Let AI review the changes in a final step. (The last message is also the agent's output.)
-    yield 'STEP'
+    yield 'STEP_ALL'
   },
 }
 

common/src/util/your-custom-agent.ts

@@ -0,0 +1,36 @@
+/*
+ *  EDIT ME to create your own agent!
+ *
+ *  Change any field below, and consult the AgentDefinition type for information on all fields and their purpose.
+ *
+ *  Run your agent with:
+ *  > codebuff --agent git-committer
+ *
+ *  Or, run codebuff normally, and use the '@' menu to mention your agent, and codebuff will spawn it for you.
+ * 
+ *  Finally, you can publish your agent with 'codebuff publish your-custom-agent'.
+ */
+
+import type { AgentDefinition } from './types/agent-definition'
+
+const definition: AgentDefinition = {
+  id: 'git-committer',
+  displayName: 'Git Committer',
+
+  model: 'anthropic/claude-4-sonnet-20250522',
+  toolNames: ['read_files', 'run_terminal_command', 'spawn_agents'],
+  spawnableAgents: ['codebuff/file-explorer@0.0.1'],
+
+  spawnPurposePrompt:
+    'Spawn when you need to commit changes to the git repository',
+
+  instructionsPrompt: `Execute the following steps:
+1. Run git diff
+2. Spawn a file explorer to find all relevant files to the change so you have the maximum context
+3. Read any relevant files
+4. Commit the changes to the git repository with a message that describes the changes`,
+
+  // Add more fields here to customize your agent further: system prompt, input/output schema, handleSteps, etc.
+}
+
+export default definition