titan-agent 5.4.0 → 5.4.2

package/dist/agent/agentLoop.js

@@ -26,6 +26,7 @@ async function chatWithTimeout(fn, label, timeoutMs = 3e5) {
 import { buildSmartContext } from "./contextManager.js";
 import { isKilled } from "../safety/killSwitch.js";
 import { estimateTokens } from "../utils/tokens.js";
+import { DEFAULT_MAX_TOKENS } from "../utils/constants.js";
 import { getCachedResponse, setCachedResponse } from "./responseCache.js";
 import { shouldReflect, reflect, resetProgress, recordProgress, setProgressSession } from "./reflection.js";
 import { recordToolResult, classifyTaskType, recordToolPreference, getErrorResolution, recordErrorResolution } from "../memory/learning.js";
@@ -442,7 +443,7 @@ async function runAgentLoop(ctx) {
   let budgetWarningSent = false;
   let pendingToolCalls = [];
   let pendingAssistantContent = "";
-  const tokenBudget = ctx.voiceFastPath ? 2e3 : ctx.config.agent.tokenBudget || 12e3;
+  const tokenBudget = ctx.voiceFastPath ? 2e3 : ctx.config.agent.tokenBudget || DEFAULT_MAX_TOKENS;
   globalSteerQueues.set(ctx.sessionId, ctx.steerQueue ?? []);
   const traceCtx = newTraceContext();
   initOtel();
@@ -663,6 +664,23 @@ Try a COMPLETELY DIFFERENT strategy. Do NOT repeat the same tools or approach.`
               ctx.streamCallbacks.onToolCall?.(chunk.toolCall.function.name, JSON.parse(chunk.toolCall.function.arguments || "{}"));
             } else if (chunk.type === "error") {
               logger.error(COMPONENT, `Stream error: ${chunk.error}`);
+            } else if (chunk.type === "retry") {
+              logger.warn(COMPONENT, `Stream retrying ${chunk.provider}/${chunk.model} (attempt ${chunk.attempt}/${chunk.maxRetries}, reason=${chunk.reason}, delay=${chunk.delayMs}ms)`);
+              ctx.streamCallbacks.onRetry?.({
+                attempt: chunk.attempt,
+                maxRetries: chunk.maxRetries,
+                reason: chunk.reason,
+                provider: chunk.provider,
+                model: chunk.model,
+                delayMs: chunk.delayMs
+              });
+            } else if (chunk.type === "failover") {
+              logger.warn(COMPONENT, `Stream failover from ${chunk.originalProvider}/${chunk.originalModel}${chunk.error ? ` (${chunk.error})` : ""}`);
+              ctx.streamCallbacks.onFailover?.({
+                originalProvider: chunk.originalProvider,
+                originalModel: chunk.originalModel,
+                error: chunk.error
+              });
             }
           }
           const estCompletionTokens = estimateTokens(streamContent + JSON.stringify(streamToolCalls));
@@ -929,7 +947,11 @@ Execute the next ACTION.` });
               phase = "done";
               break;
             }
-            if (ctx.isAutonomous && round < ctx.effectiveMaxRounds - 2) {
+            const isMultiStepPipeline = ctx.completionStrategy === "terminal-tool" || ctx.minRounds !== void 0;
+            const nudgeMinRounds = ctx.minRounds ?? (ctx.completionStrategy === "single-round" ? 1 : 2);
+            const nudgeMinRoundsMet = round >= nudgeMinRounds;
+            const nudgeEnabled = ctx.isAutonomous || isMultiStepPipeline && !nudgeMinRoundsMet;
+            if (nudgeEnabled && round < ctx.effectiveMaxRounds - 2) {
               const futureIntentOpener = /^(let me\s+\w+|I['']?ll\s+(?:start|begin|check|look|read|run|edit|write|create|try|go|investigate|verify|test|install|build|fix|update|change|set)|I\s+(?:will|need to|should|can|am going to|plan to)\s+\w+|first,?\s+I|now\s+I|to\s+(?:fix|resolve|complete|edit|write|create|update|change|run))\b/i.test(response.content.trim());
               const describesWork = /\b(?:I['']?ll|I (?:will|need to|should|plan to|am going to)|let me)\b[^.]{0,80}\b(?:fix|edit|change|update|create|write|modify|run|install|build|start|restart|read|open|debug|set up|check|look at|examine|investigate|verify|confirm|test)\b/i.test(response.content);
               if (futureIntentOpener && describesWork && noToolsRetryCount < 3) {
@@ -1056,6 +1078,17 @@ ${gateText}` : gateText;
             tr.diff
           );
         }
+        const pendingApproval = toolResults.find((r) => r.approvalPending);
+        if (pendingApproval) {
+          logger.info(COMPONENT, `[ApprovalPause] Tool "${pendingApproval.name}" awaiting approval (req ${pendingApproval.approvalRequestId}) \u2014 exiting loop`);
+          result.content = pendingApproval.content;
+          ctx.messages.push({
+            role: "assistant",
+            content: `I need your approval before I can use ${pendingApproval.name}. ${pendingApproval.content}`
+          });
+          phase = "done";
+          break;
+        }
         try {
           const { getSubdirTracker } = await import("./subdirHints.js");
           const tracker = getSubdirTracker(ctx.sessionId);
@@ -1342,17 +1375,32 @@ Examples: use read_file instead of cat/head/tail/grep, and edit_file instead of
             }
           }
         } else {
-          const discoveryTools = ["gallery_search", "tool_search"];
-          const lastWasDiscovery = pendingToolCalls.length === 1 && discoveryTools.includes(pendingToolCalls[0].function.name);
-          const resultHintsFollowUp = lastWasDiscovery && toolResults.some(
-            (r) => /call gallery_get/i.test(r.content) || /hint:.*?(call|use|try|fetch|get|search)/i.test(r.content)
-          );
-          if (resultHintsFollowUp && round < ctx.effectiveMaxRounds - 1) {
-            logger.info(COMPONENT, `[DiscoveryChain] ${pendingToolCalls[0].function.name} hinted follow-up \u2014 allowing another think round`);
-            phase = "think";
+          const hasPipelineConfig = ctx.completionStrategy !== void 0 || ctx.minRounds !== void 0;
+          const isMultiStep = hasPipelineConfig && (ctx.completionStrategy === "terminal-tool" || ctx.minRounds !== void 0);
+          if (isMultiStep) {
+            const defaultTerminals = ["write_file", "append_file", "fb_post", "fb_reply", "content_publish"];
+            const terminalTools = new Set(ctx.pipelineTerminalTools || defaultTerminals);
+            const terminalMissing = !toolResults.some((r) => terminalTools.has(r.name));
+            if (terminalMissing && round < ctx.effectiveMaxRounds - 1) {
+              logger.info(COMPONENT, `[NonAutonomous] Multi-step task needs terminal tool \u2014 staying in THINK`);
+              ctx.messages.push({ role: "user", content: "Continue with the task. Call the required tool to finish." });
+              phase = "think";
+            } else {
+              phase = "respond";
+            }
           } else {
-            phase = "respond";
-            logger.info(COMPONENT, `[ThinkAct] Tools executed \u2014 entering respond phase (tools will be stripped)`);
+            const discoveryTools = ["gallery_search", "tool_search"];
+            const lastWasDiscovery = pendingToolCalls.length === 1 && discoveryTools.includes(pendingToolCalls[0].function.name);
+            const resultHintsFollowUp = lastWasDiscovery && toolResults.some(
+              (r) => /call gallery_get/i.test(r.content) || /hint:.*?(call|use|try|fetch|get|search)/i.test(r.content)
+            );
+            if (resultHintsFollowUp && round < ctx.effectiveMaxRounds - 1) {
+              logger.info(COMPONENT, `[DiscoveryChain] ${pendingToolCalls[0].function.name} hinted follow-up \u2014 allowing another think round`);
+              phase = "think";
+            } else {
+              phase = "respond";
+              logger.info(COMPONENT, `[ThinkAct] Tools executed \u2014 entering respond phase (tools will be stripped)`);
+            }
           }
         }
         break;
@@ -1423,6 +1471,23 @@ Examples: use read_file instead of cat/head/tail/grep, and edit_file instead of
               ctx.streamCallbacks.onToken(chunk.content);
             } else if (chunk.type === "error") {
               logger.error(COMPONENT, `Stream error: ${chunk.error}`);
+            } else if (chunk.type === "retry") {
+              logger.warn(COMPONENT, `Respond-phase stream retrying ${chunk.provider}/${chunk.model} (attempt ${chunk.attempt}/${chunk.maxRetries}, reason=${chunk.reason})`);
+              ctx.streamCallbacks.onRetry?.({
+                attempt: chunk.attempt,
+                maxRetries: chunk.maxRetries,
+                reason: chunk.reason,
+                provider: chunk.provider,
+                model: chunk.model,
+                delayMs: chunk.delayMs
+              });
+            } else if (chunk.type === "failover") {
+              logger.warn(COMPONENT, `Respond-phase stream failover from ${chunk.originalProvider}/${chunk.originalModel}`);
+              ctx.streamCallbacks.onFailover?.({
+                originalProvider: chunk.originalProvider,
+                originalModel: chunk.originalModel,
+                error: chunk.error
+              });
             }
           }
           response = {