@dreb/coding-agent 2.57.0 → 2.58.0

package/dist/core/agent-session.js

@@ -37,11 +37,11 @@ import { expandPromptTemplate } from "./prompt-templates.js";
 import { scrubSecrets } from "./secret-scrubber.js";
 import { isSensitivePath } from "./sensitive-paths.js";
 import { CURRENT_SESSION_VERSION, getLatestCompactionEntry } from "./session-manager.js";
-import { DEFAULT_BG_PARENT_TURN_LIMIT } from "./settings-manager.js";
+import { DEFAULT_BG_PARENT_TURN_LIMIT, DEFAULT_MAX_CONCURRENT_SUBAGENTS, } from "./settings-manager.js";
 import { createSyntheticSourceInfo } from "./source-info.js";
 import { buildSystemPrompt } from "./system-prompt.js";
 import { resolveThinkingDisplay } from "./thinking.js";
-import { createAllToolDefinitions, discoverAgentTypes, getRunningBackgroundAgents, } from "./tools/index.js";
+import { createAllToolDefinitions, createSubagentConcurrencyGate, discoverAgentTypes, getRunningBackgroundAgents, } from "./tools/index.js";
 import { expandSkillContent } from "./tools/skill.js";
 import { createToolDefinitionFromAgentTool, wrapToolDefinition } from "./tools/tool-definition-wrapper.js";
 // ============================================================================
@@ -152,6 +152,14 @@ export class AgentSession {
     // Base system prompt (without extension appends) - used to apply fresh appends each turn
     _baseSystemPrompt = "";
     _uiType;
+    _maxConcurrentSubagents;
+    _subagentsDisabledBySetting;
+    /**
+     * Session-owned concurrency gate. Created once and reused across `_buildRuntime` rebuilds so
+     * the running-child count stays accurate through `/reload` while children launched before the
+     * reload are still in flight.
+     */
+    _subagentConcurrencyGate;
     performanceTracker;
     _ownsPerformanceTracker;
     // Git repo state captured once at session start
@@ -184,6 +192,15 @@ export class AgentSession {
         this._initialActiveToolNames = config.initialActiveToolNames;
         this._baseToolsOverride = config.baseToolsOverride;
         this._uiType = config.uiType;
+        const configuredMaxConcurrentSubagents = config.maxConcurrentSubagents ?? this.settingsManager.getMaxConcurrentSubagents();
+        if (!Number.isSafeInteger(configuredMaxConcurrentSubagents) || configuredMaxConcurrentSubagents < 0) {
+            throw new Error("maxConcurrentSubagents must be a non-negative whole number");
+        }
+        const isChildAgentSession = this.sessionManager.getHeader()?.agentType !== undefined;
+        this._subagentsDisabledBySetting = !isChildAgentSession && configuredMaxConcurrentSubagents === 0;
+        this._maxConcurrentSubagents =
+            configuredMaxConcurrentSubagents > 0 ? configuredMaxConcurrentSubagents : DEFAULT_MAX_CONCURRENT_SUBAGENTS;
+        this._subagentConcurrencyGate = createSubagentConcurrencyGate(this._maxConcurrentSubagents);
         // Capture git repo state once at session start (before building runtime/system prompt)
         this._gitRepoState = getGitRepoState(this._cwd) ?? undefined;
         this._dispatchArbiter = new DispatchArbiter({
@@ -1105,6 +1122,8 @@ export class AgentSession {
         const tools = [];
         const validToolNames = [];
         for (const name of toolNames) {
+            if (this._subagentsDisabledBySetting && name === "subagent")
+                continue;
             const tool = this._toolRegistry.get(name);
             if (tool) {
                 tools.push(tool);
@@ -1231,6 +1250,7 @@ export class AgentSession {
             uiType: this._uiType,
             gitRepoState: this._gitRepoState,
             currentModel: this.model ? { provider: this.model.provider, id: this.model.id } : undefined,
+            subagentsDisabled: this._subagentsDisabledBySetting,
         });
     }
     // =========================================================================
@@ -2562,7 +2582,10 @@ export class AgentSession {
                 }
             }
         }
-        this.setActiveToolsByName([...new Set(nextActiveToolNames)]);
+        const uniqueActiveToolNames = [...new Set(nextActiveToolNames)];
+        this.setActiveToolsByName(this._subagentsDisabledBySetting
+            ? uniqueActiveToolNames.filter((name) => name !== "subagent")
+            : uniqueActiveToolNames);
     }
     _buildRuntime(options) {
         const autoResizeImages = this.settingsManager.getImageAutoResize();
@@ -2617,6 +2640,8 @@ export class AgentSession {
                     modelRegistry: this._modelRegistry,
                     getAgentModelsForAgent: (name) => this.settingsManager?.getAgentModelsForAgent(name),
                     defaultThinkingLevel: () => this.settingsManager.getDefaultThinkingLevel() ?? DEFAULT_THINKING_LEVEL,
+                    maxConcurrentSubagents: this._maxConcurrentSubagents,
+                    concurrencyGate: this._subagentConcurrencyGate,
                     arbitrate: (request, signal) => this._dispatchArbiter.arbitrate(request, signal),
                     onArbitration: (event) => {
                         this.sessionManager.appendCustomEntry("subagent_arbitration", event);