@kolisachint/hoocode-agent 0.4.21 → 0.4.24

package/dist/core/agent-session.js

@@ -18,6 +18,7 @@ import { clampThinkingLevel, cleanupSessionResources, getSupportedThinkingLevels
 import { theme } from "../modes/interactive/theme/theme.js";
 import { stripFrontmatter } from "../utils/frontmatter.js";
 import { sleep } from "../utils/sleep.js";
+import { loadAgentRegistry } from "./agent-registry.js";
 import { formatNoApiKeyFoundMessage, formatNoModelSelectedMessage } from "./auth-guidance.js";
 import { executeBashWithOperations } from "./bash-executor.js";
 import { calculateContextTokens, collectEntriesForBranchSummary, compact, estimateContextTokens, generateBranchSummary, prepareCompaction, shouldCompact, } from "./compaction/index.js";
@@ -27,8 +28,10 @@ import { createToolHtmlRenderer } from "./export-html/tool-renderer.js";
 import { ExtensionRunner, wrapRegisteredTools, } from "./extensions/index.js";
 import { emitSessionShutdownEvent } from "./extensions/runner.js";
 import { expandPromptTemplate } from "./prompt-templates.js";
+import { clearProviderExhaustion, isProviderQuotaError, markProviderExhausted } from "./provider-health.js";
 import { CURRENT_SESSION_VERSION, getLatestCompactionEntry } from "./session-manager.js";
 import { createSyntheticSourceInfo } from "./source-info.js";
+import { updateSubagentSkillPaths } from "./subagent-pool-instance.js";
 import { buildSystemPrompt } from "./system-prompt.js";
 import { createLocalBashOperations } from "./tools/bash.js";
 import { createAllToolDefinitions } from "./tools/index.js";
@@ -118,6 +121,7 @@ export class AgentSession {
         this.settingsManager = config.settingsManager;
         this._scopedModels = config.scopedModels ?? [];
         this._resourceLoader = config.resourceLoader;
+        updateSubagentSkillPaths(this._resourceLoader.getSkillPaths());
         this._customTools = config.customTools ?? [];
         this._cwd = config.cwd;
         this._modelRegistry = config.modelRegistry;
@@ -314,6 +318,11 @@ export class AgentSession {
                 const assistantMsg = event.message;
                 if (assistantMsg.stopReason !== "error") {
                     this._overflowRecoveryAttempted = false;
+                    // A successful response clears any prior provider-exhaustion flag so
+                    // subagent dispatch is unblocked as soon as the provider recovers.
+                    const provider = this.model?.provider;
+                    if (provider)
+                        clearProviderExhaustion(provider);
                 }
                 // Reset retry counter immediately on successful assistant response
                 // This prevents accumulation across multiple LLM calls within a turn
@@ -336,6 +345,14 @@ export class AgentSession {
                 const didRetry = await this._handleRetryableError(msg);
                 if (didRetry)
                     return; // Retry was initiated, don't proceed to compaction
+                // Retries are exhausted/disabled and a quota or rate-limit error
+                // persists: flag the provider so subagent dispatch can skip pointless
+                // spawns (subagents inherit this provider). Self-expires via TTL and is
+                // cleared on the next successful response.
+                const provider = this.model?.provider;
+                if (provider && isProviderQuotaError(msg.errorMessage)) {
+                    markProviderExhausted(provider, msg.errorMessage ?? "provider error");
+                }
             }
             this._resolveRetry();
             await this._checkCompaction(msg);
@@ -663,9 +680,13 @@ export class AgentSession {
         const appendSystemPrompt = loaderAppendSystemPrompt.length > 0 ? loaderAppendSystemPrompt.join("\n\n") : undefined;
         const loadedSkills = this._resourceLoader.getSkills().skills;
         const loadedContextFiles = this._resourceLoader.getAgentsFiles().agentsFiles;
+        // Include agents in the system prompt only when the Task tool is active.
+        const hasTaskTool = validToolNames.includes("Task");
+        const loadedAgents = hasTaskTool ? loadAgentRegistry({ cwd: this._cwd }).list() : [];
         this._baseSystemPromptOptions = {
             cwd: this._cwd,
             skills: loadedSkills,
+            agents: loadedAgents,
             contextFiles: loadedContextFiles,
             customPrompt: loaderSystemPrompt,
             appendSystemPrompt,
@@ -1588,6 +1609,7 @@ export class AgentSession {
             themePaths: this.buildExtensionResourcePaths(themePaths),
         };
         this._resourceLoader.extendResources(extensionPaths);
+        updateSubagentSkillPaths(this._resourceLoader.getSkillPaths());
         this._baseSystemPrompt = this._rebuildSystemPrompt(this.getActiveToolNames());
         this.agent.state.systemPrompt = this._baseSystemPrompt;
     }
@@ -1850,6 +1872,7 @@ export class AgentSession {
         await this.settingsManager.reload();
         resetApiProviders();
         await this._resourceLoader.reload();
+        updateSubagentSkillPaths(this._resourceLoader.getSkillPaths());
         this._buildRuntime({
             activeToolNames: this.getActiveToolNames(),
             flagValues: previousFlagValues,