Codebuff.json: Added spawnable agents + removed spawnable agents fields to manage base subagents

Author: jahooma

backend/src/main-prompt.ts

@@ -173,9 +173,19 @@ export const mainPrompt = async (
   let updatedSubagents = mainAgentTemplate.spawnableAgents
   if (!agentId) {
     // If --agent is not specified, use the spawnableAgents from the codebuff config or add all local agents
+    const {
+      spawnableAgents,
+      addedSpawnableAgents = [],
+      removedSpawnableAgents = [],
+    } = fileContext.codebuffConfig ?? {}
     updatedSubagents =
-      fileContext.codebuffConfig?.spawnableAgents ??
+      spawnableAgents ??
       uniq([...mainAgentTemplate.spawnableAgents, ...availableAgents])
+
+    updatedSubagents = uniq([
+      ...updatedSubagents,
+      ...addedSpawnableAgents,
+    ]).filter((subagent) => !removedSpawnableAgents.includes(subagent))
   }
   mainAgentTemplate.spawnableAgents = updatedSubagents
   localAgentTemplates[agentType] = mainAgentTemplate

codebuff.json

@@ -1,5 +1,6 @@
 {
   "description": "Template configuration for this project. See https://www.codebuff.com/config for all options.",
+  "addedSpawnableAgents": ["codebuff/git-commiter@0.0.1"],
   "startupProcesses": [
     {
       "name": "drizzle",

common/src/json-config/constants.ts

@@ -66,7 +66,7 @@ export const FileChangeHook = z
   .describe('Defines a single file change hook.')
 
 export const CodebuffConfigSchema = z
-  .object({
+  .looseObject({
     description: z
       .any()
       .optional()
@@ -87,6 +87,14 @@ export const CodebuffConfigSchema = z
         'Maximum number of turns agent will take before being forced to end',
       ),
     baseAgent: z.string().optional().describe('Specify default base agent'),
+    addedSpawnableAgents: z
+      .array(z.string())
+      .optional()
+      .describe('Specify additional agents that the base agent can spawn'),
+    removedSpawnableAgents: z
+      .array(z.string())
+      .optional()
+      .describe('Specify which agents the base agent cannot spawn'),
     spawnableAgents: z
       .array(z.string())
       .optional()

common/src/json-config/default.ts

@@ -9,6 +9,7 @@ export function getDefaultConfig(): CodebuffConfig {
     maxAgentSteps: MAX_AGENT_STEPS_DEFAULT,
     baseAgent: undefined,
     spawnableAgents: undefined,
+    addedSpawnableAgents: undefined,
+    removedSpawnableAgents: undefined,
   }
 }
-