@amodalai/runtime 0.1.25 → 0.2.0

package/dist/src/agent/states/dispatching.js

@@ -0,0 +1,241 @@
+/**
+ * @license
+ * Copyright 2026 Amodal Labs, Inc.
+ * SPDX-License-Identifier: MIT
+ */
+/**
+ * DISPATCHING state handler.
+ *
+ * Spawns a child agent with a subset of tools and a specific prompt.
+ * The child runs its own runAgent() loop; each child event is wrapped
+ * as an SSESubagentEvent and yielded as an effect so the client can
+ * display real-time sub-agent activity.
+ *
+ * After the child completes, the handler builds a ToolResult from the
+ * child's text response and transitions back through the normal
+ * EXECUTING post-result flow (queue, compaction, thinking).
+ */
+import { SSEEventType } from '../../types.js';
+import { createToolRegistry } from '../../tools/registry.js';
+import { DEFAULT_LOOP_CONFIG } from '../loop-types.js';
+import { runAgent } from '../loop.js';
+import { nextAfterToolResult } from './executing.js';
+import { DEFAULT_CHILD_MAX_TURNS, DISPATCH_TOOL_NAME } from '../../tools/dispatch-tool.js';
+// ---------------------------------------------------------------------------
+// Constants
+// ---------------------------------------------------------------------------
+const DEFAULT_CHILD_MAX_CONTEXT_TOKENS = 100_000;
+const DEFAULT_CHILD_TIMEOUT_MS = 60_000;
+// ---------------------------------------------------------------------------
+// Handler
+// ---------------------------------------------------------------------------
+/**
+ * Handle the DISPATCHING state — run a child agent loop.
+ *
+ * Creates a child AgentContext with:
+ * - Same provider, logger, signal, tenant, user, permission checker
+ * - Tool registry subset (only the requested tools, never dispatch_task)
+ * - Same buildToolContext factory (child shares parent's connections/stores)
+ * - Reduced maxTurns (default 10) and maxContextTokens
+ * - Fresh messages (dispatch prompt as user message)
+ * - Fresh usage counters (merged back to parent after completion)
+ */
+export async function handleDispatching(state, ctx) {
+    const { task, toolCallId } = state;
+    const effects = [];
+    const startedAt = Date.now();
+    ctx.logger.info('dispatch_start', {
+        session: ctx.sessionId,
+        agent: task.agentName,
+        tools: task.toolSubset,
+        maxTurns: task.maxTurns ?? DEFAULT_CHILD_MAX_TURNS,
+        parentToolCallId: toolCallId,
+    });
+    // Build child tool registry from the parent's subset
+    const childRegistry = createToolRegistry();
+    const subsetTools = ctx.toolRegistry.subset(task.toolSubset);
+    for (const [name, def] of Object.entries(subsetTools)) {
+        childRegistry.register(name, def);
+    }
+    // Child-specific timeout — prevents a slow child from starving the parent.
+    // Combined with parent's signal so parent abort also stops child.
+    const childTimeoutMs = task.timeoutMs ?? DEFAULT_CHILD_TIMEOUT_MS;
+    const childSignal = AbortSignal.any([ctx.signal, AbortSignal.timeout(childTimeoutMs)]);
+    // Minimal system prompt for the child — the full parent prompt (skills,
+    // knowledge, connection docs) is unnecessary for a focused sub-task and
+    // wastes tokens. The child gets a task-scoped prompt with available tools.
+    const childToolNames = childRegistry.names();
+    const childSystemPrompt = buildChildSystemPrompt(task.agentName, task.prompt, childToolNames);
+    // Build child context
+    const childCtx = {
+        provider: ctx.provider,
+        toolRegistry: childRegistry,
+        permissionChecker: ctx.permissionChecker,
+        logger: ctx.logger,
+        signal: childSignal,
+        sessionId: ctx.sessionId,
+        tenantId: ctx.tenantId,
+        user: ctx.user,
+        systemPrompt: childSystemPrompt,
+        messages: [],
+        usage: { inputTokens: 0, outputTokens: 0, totalTokens: 0 },
+        turnCount: 0,
+        maxTurns: task.maxTurns ?? DEFAULT_CHILD_MAX_TURNS,
+        maxContextTokens: task.maxContextTokens ?? DEFAULT_CHILD_MAX_CONTEXT_TOKENS,
+        config: { ...DEFAULT_LOOP_CONFIG },
+        compactionFailures: 0,
+        preExecutionCache: new Map(),
+        waitForConfirmation: ctx.waitForConfirmation,
+        buildToolContext: ctx.buildToolContext,
+    };
+    // Run child agent loop and translate events to SubagentEvents
+    let childResponse = '';
+    let childFailed = false;
+    try {
+        for await (const childEvent of runAgent({ messages: [{ role: 'user', content: task.prompt }], context: childCtx })) {
+            if (ctx.signal.aborted)
+                break;
+            // eslint-disable-next-line @typescript-eslint/switch-exhaustiveness-check -- intentionally skip child-internal events (Init, Widget, etc.)
+            switch (childEvent.type) {
+                case SSEEventType.TextDelta:
+                    childResponse += childEvent.content;
+                    effects.push({
+                        type: SSEEventType.SubagentEvent,
+                        parent_tool_id: toolCallId,
+                        agent_name: task.agentName,
+                        event_type: 'thought',
+                        text: childEvent.content,
+                        timestamp: childEvent.timestamp,
+                    });
+                    break;
+                case SSEEventType.ToolCallStart:
+                    effects.push({
+                        type: SSEEventType.SubagentEvent,
+                        parent_tool_id: toolCallId,
+                        agent_name: task.agentName,
+                        event_type: 'tool_call_start',
+                        tool_name: childEvent.tool_name,
+                        tool_args: childEvent.parameters,
+                        timestamp: childEvent.timestamp,
+                    });
+                    break;
+                case SSEEventType.ToolCallResult:
+                    effects.push({
+                        type: SSEEventType.SubagentEvent,
+                        parent_tool_id: toolCallId,
+                        agent_name: task.agentName,
+                        event_type: 'tool_call_end',
+                        tool_name: undefined,
+                        result: childEvent.status === 'error' ? childEvent.error : undefined,
+                        timestamp: childEvent.timestamp,
+                    });
+                    break;
+                case SSEEventType.Error:
+                    effects.push({
+                        type: SSEEventType.SubagentEvent,
+                        parent_tool_id: toolCallId,
+                        agent_name: task.agentName,
+                        event_type: 'error',
+                        error: childEvent.message,
+                        timestamp: childEvent.timestamp,
+                    });
+                    break;
+                case SSEEventType.Done:
+                    effects.push({
+                        type: SSEEventType.SubagentEvent,
+                        parent_tool_id: toolCallId,
+                        agent_name: task.agentName,
+                        event_type: 'complete',
+                        timestamp: childEvent.timestamp,
+                    });
+                    break;
+                default:
+                    // Skip Init and other events — they're child-internal
+                    break;
+            }
+        }
+    }
+    catch (err) {
+        const errorMessage = err instanceof Error ? err.message : String(err);
+        ctx.logger.error('dispatch_child_error', {
+            session: ctx.sessionId,
+            agent: task.agentName,
+            error: errorMessage,
+            parentToolCallId: toolCallId,
+        });
+        effects.push({
+            type: SSEEventType.SubagentEvent,
+            parent_tool_id: toolCallId,
+            agent_name: task.agentName,
+            event_type: 'error',
+            error: errorMessage,
+            timestamp: new Date().toISOString(),
+        });
+        childResponse = `Sub-agent "${task.agentName}" failed: ${errorMessage}`;
+        childFailed = true;
+    }
+    // Merge child usage into parent
+    ctx.usage.inputTokens += childCtx.usage.inputTokens;
+    ctx.usage.outputTokens += childCtx.usage.outputTokens;
+    ctx.usage.totalTokens += childCtx.usage.totalTokens;
+    if (childCtx.usage.cachedInputTokens) {
+        ctx.usage.cachedInputTokens = (ctx.usage.cachedInputTokens ?? 0) + childCtx.usage.cachedInputTokens;
+    }
+    if (childCtx.usage.cacheCreationInputTokens) {
+        ctx.usage.cacheCreationInputTokens = (ctx.usage.cacheCreationInputTokens ?? 0) + childCtx.usage.cacheCreationInputTokens;
+    }
+    const duration = Date.now() - startedAt;
+    ctx.logger.info('dispatch_complete', {
+        session: ctx.sessionId,
+        agent: task.agentName,
+        childTurns: childCtx.turnCount,
+        childUsage: childCtx.usage,
+        responseLength: childResponse.length,
+        parentToolCallId: toolCallId,
+        duration,
+    });
+    // Build tool result from child response
+    const result = {
+        callId: toolCallId,
+        toolName: DISPATCH_TOOL_NAME,
+        status: childFailed ? 'error' : 'success',
+        content: childResponse || '(no response from sub-agent)',
+    };
+    // Emit ToolCallResult for the dispatch_task call
+    effects.push({
+        type: SSEEventType.ToolCallResult,
+        tool_id: toolCallId,
+        status: result.status,
+        duration_ms: duration,
+        ...(result.status === 'error' ? { error: result.content } : {}),
+        timestamp: new Date().toISOString(),
+    });
+    // Use the shared post-result transition logic from EXECUTING.
+    // This handles: append tool result message, continue queue, compaction check.
+    const executingState = {
+        type: 'executing',
+        current: { toolCallId, toolName: DISPATCH_TOOL_NAME, args: {} },
+        queue: state.queue,
+        results: state.results,
+    };
+    return nextAfterToolResult(executingState, result, effects, ctx);
+}
+// ---------------------------------------------------------------------------
+// Child system prompt
+// ---------------------------------------------------------------------------
+/**
+ * Build a minimal system prompt for the child agent.
+ *
+ * The full parent prompt (30K+ chars of skills, knowledge, connection docs)
+ * is wasteful for a focused sub-task. The child gets a concise, task-scoped
+ * prompt listing only its available tools.
+ */
+function buildChildSystemPrompt(agentName, taskPrompt, toolNames) {
+    const toolList = toolNames.length > 0
+        ? `\n\nAvailable tools: ${toolNames.join(', ')}`
+        : '';
+    return `You are "${agentName}", a sub-agent executing a delegated task. Complete the task and return a concise summary of your findings or actions.
+
+Do not ask clarifying questions — work with what you have. If you cannot complete the task, explain what went wrong.${toolList}`;
+}
+//# sourceMappingURL=dispatching.js.map

package/dist/src/agent/states/dispatching.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"dispatching.js","sourceRoot":"","sources":["../../../../src/agent/states/dispatching.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH;;;;;;;;;;;GAWG;AAEH,OAAO,EAAC,YAAY,EAAC,MAAM,gBAAgB,CAAC;AAE5C,OAAO,EAAC,kBAAkB,EAAC,MAAM,yBAAyB,CAAC;AAC3D,OAAO,EAAC,mBAAmB,EAAC,MAAM,kBAAkB,CAAC;AAOrD,OAAO,EAAC,QAAQ,EAAC,MAAM,YAAY,CAAC;AACpC,OAAO,EAAC,mBAAmB,EAAC,MAAM,gBAAgB,CAAC;AACnD,OAAO,EAAC,uBAAuB,EAAE,kBAAkB,EAAC,MAAM,8BAA8B,CAAC;AAEzF,8EAA8E;AAC9E,YAAY;AACZ,8EAA8E;AAE9E,MAAM,gCAAgC,GAAG,OAAO,CAAC;AACjD,MAAM,wBAAwB,GAAG,MAAM,CAAC;AAExC,8EAA8E;AAC9E,UAAU;AACV,8EAA8E;AAE9E;;;;;;;;;;GAUG;AACH,MAAM,CAAC,KAAK,UAAU,iBAAiB,CACrC,KAAuB,EACvB,GAAiB;IAEjB,MAAM,EAAC,IAAI,EAAE,UAAU,EAAC,GAAG,KAAK,CAAC;IACjC,MAAM,OAAO,GAAe,EAAE,CAAC;IAC/B,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAE7B,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,gBAAgB,EAAE;QAChC,OAAO,EAAE,GAAG,CAAC,SAAS;QACtB,KAAK,EAAE,IAAI,CAAC,SAAS;QACrB,KAAK,EAAE,IAAI,CAAC,UAAU;QACtB,QAAQ,EAAE,IAAI,CAAC,QAAQ,IAAI,uBAAuB;QAClD,gBAAgB,EAAE,UAAU;KAC7B,CAAC,CAAC;IAEH,qDAAqD;IACrD,MAAM,aAAa,GAAG,kBAAkB,EAAE,CAAC;IAC3C,MAAM,WAAW,GAAG,GAAG,CAAC,YAAY,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAC7D,KAAK,MAAM,CAAC,IAAI,EAAE,GAAG,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,CAAC;QACtD,aAAa,CAAC,QAAQ,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;IACpC,CAAC;IAED,2EAA2E;IAC3E,kEAAkE;IAClE,MAAM,cAAc,GAAG,IAAI,CAAC,SAAS,IAAI,wBAAwB,CAAC;IAClE,MAAM,WAAW,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC;IAEvF,wEAAwE;IACxE,wEAAwE;IACxE,2EAA2E;IAC3E,MAAM,cAAc,GAAG,aAAa,CAAC,KAAK,EAAE,CAAC;IAC7C,MAAM,iBAAiB,GAAG,sBAAsB,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,MAAM,EAAE,cAAc,CAAC,CAAC;IAE9F,sBAAsB;IACtB,MAAM,QAAQ,GAAiB;QAC7B,QAAQ,EAAE,GAAG,CAAC,QAAQ;QACtB,YAAY,EAAE,aAAa;QAC3B,iBAAiB,EAAE,GAAG,CAAC,iBAAiB;QACxC,MAAM,EAAE,GAAG,CAAC,MAAM;QAClB,MAAM,EAAE,WAAW;QACnB,SAAS,EAAE,GAAG,CAAC,SAAS;QACxB,QAAQ,EAAE,GAAG,CAAC,QAAQ;QACtB,IAAI,EAAE,GAAG,CAAC,IAAI;QACd,YAAY,EAAE,iBAAiB;QAC/B,QAAQ,EAAE,EAAE;QACZ,KAAK,EAAE,EAAC,WAAW,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,WAAW,EAAE,CAAC,EAAC;QACxD,SAAS,EAAE,CAAC;QACZ,QAAQ,EAAE,IAAI,CAAC,QAAQ,IAAI,uBAAuB;QAClD,gBAAgB,EAAE,IAAI,CAAC,gBAAgB,IAAI,gCAAgC;QAC3E,MAAM,EAAE,EAAC,GAAG,mBAAmB,EAAC;QAChC,kBAAkB,EAAE,CAAC;QACrB,iBAAiB,EAAE,IAAI,GAAG,EAAE;QAC5B,mBAAmB,EAAE,GAAG,CAAC,mBAAmB;QAC5C,gBAAgB,EAAE,GAAG,CAAC,gBAAgB;KACvC,CAAC;IAEF,8DAA8D;IAC9D,IAAI,aAAa,GAAG,EAAE,CAAC;IACvB,IAAI,WAAW,GAAG,KAAK,CAAC;IAExB,IAAI,CAAC;QACH,IAAI,KAAK,EAAE,MAAM,UAAU,IAAI,QAAQ,CAAC,EAAC,QAAQ,EAAE,CAAC,EAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,CAAC,MAAM,EAAC,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAC,CAAC,EAAE,CAAC;YAC/G,IAAI,GAAG,CAAC,MAAM,CAAC,OAAO;gBAAE,MAAM;YAE9B,2IAA2I;YAC3I,QAAQ,UAAU,CAAC,IAAI,EAAE,CAAC;gBACxB,KAAK,YAAY,CAAC,SAAS;oBACzB,aAAa,IAAI,UAAU,CAAC,OAAO,CAAC;oBACpC,OAAO,CAAC,IAAI,CAAC;wBACX,IAAI,EAAE,YAAY,CAAC,aAAa;wBAChC,cAAc,EAAE,UAAU;wBAC1B,UAAU,EAAE,IAAI,CAAC,SAAS;wBAC1B,UAAU,EAAE,SAAS;wBACrB,IAAI,EAAE,UAAU,CAAC,OAAO;wBACxB,SAAS,EAAE,UAAU,CAAC,SAAS;qBAChC,CAAC,CAAC;oBACH,MAAM;gBAER,KAAK,YAAY,CAAC,aAAa;oBAC7B,OAAO,CAAC,IAAI,CAAC;wBACX,IAAI,EAAE,YAAY,CAAC,aAAa;wBAChC,cAAc,EAAE,UAAU;wBAC1B,UAAU,EAAE,IAAI,CAAC,SAAS;wBAC1B,UAAU,EAAE,iBAAiB;wBAC7B,SAAS,EAAE,UAAU,CAAC,SAAS;wBAC/B,SAAS,EAAE,UAAU,CAAC,UAAU;wBAChC,SAAS,EAAE,UAAU,CAAC,SAAS;qBAChC,CAAC,CAAC;oBACH,MAAM;gBAER,KAAK,YAAY,CAAC,cAAc;oBAC9B,OAAO,CAAC,IAAI,CAAC;wBACX,IAAI,EAAE,YAAY,CAAC,aAAa;wBAChC,cAAc,EAAE,UAAU;wBAC1B,UAAU,EAAE,IAAI,CAAC,SAAS;wBAC1B,UAAU,EAAE,eAAe;wBAC3B,SAAS,EAAE,SAAS;wBACpB,MAAM,EAAE,UAAU,CAAC,MAAM,KAAK,OAAO,CAAC,CAAC,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS;wBACpE,SAAS,EAAE,UAAU,CAAC,SAAS;qBAChC,CAAC,CAAC;oBACH,MAAM;gBAER,KAAK,YAAY,CAAC,KAAK;oBACrB,OAAO,CAAC,IAAI,CAAC;wBACX,IAAI,EAAE,YAAY,CAAC,aAAa;wBAChC,cAAc,EAAE,UAAU;wBAC1B,UAAU,EAAE,IAAI,CAAC,SAAS;wBAC1B,UAAU,EAAE,OAAO;wBACnB,KAAK,EAAE,UAAU,CAAC,OAAO;wBACzB,SAAS,EAAE,UAAU,CAAC,SAAS;qBAChC,CAAC,CAAC;oBACH,MAAM;gBAER,KAAK,YAAY,CAAC,IAAI;oBACpB,OAAO,CAAC,IAAI,CAAC;wBACX,IAAI,EAAE,YAAY,CAAC,aAAa;wBAChC,cAAc,EAAE,UAAU;wBAC1B,UAAU,EAAE,IAAI,CAAC,SAAS;wBAC1B,UAAU,EAAE,UAAU;wBACtB,SAAS,EAAE,UAAU,CAAC,SAAS;qBAChC,CAAC,CAAC;oBACH,MAAM;gBAER;oBACE,sDAAsD;oBACtD,MAAM;YACV,CAAC;QACH,CAAC;IACH,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,MAAM,YAAY,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QACtE,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,sBAAsB,EAAE;YACvC,OAAO,EAAE,GAAG,CAAC,SAAS;YACtB,KAAK,EAAE,IAAI,CAAC,SAAS;YACrB,KAAK,EAAE,YAAY;YACnB,gBAAgB,EAAE,UAAU;SAC7B,CAAC,CAAC;QAEH,OAAO,CAAC,IAAI,CAAC;YACX,IAAI,EAAE,YAAY,CAAC,aAAa;YAChC,cAAc,EAAE,UAAU;YAC1B,UAAU,EAAE,IAAI,CAAC,SAAS;YAC1B,UAAU,EAAE,OAAO;YACnB,KAAK,EAAE,YAAY;YACnB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;SACpC,CAAC,CAAC;QAEH,aAAa,GAAG,cAAc,IAAI,CAAC,SAAS,aAAa,YAAY,EAAE,CAAC;QACxE,WAAW,GAAG,IAAI,CAAC;IACrB,CAAC;IAED,gCAAgC;IAChC,GAAG,CAAC,KAAK,CAAC,WAAW,IAAI,QAAQ,CAAC,KAAK,CAAC,WAAW,CAAC;IACpD,GAAG,CAAC,KAAK,CAAC,YAAY,IAAI,QAAQ,CAAC,KAAK,CAAC,YAAY,CAAC;IACtD,GAAG,CAAC,KAAK,CAAC,WAAW,IAAI,QAAQ,CAAC,KAAK,CAAC,WAAW,CAAC;IACpD,IAAI,QAAQ,CAAC,KAAK,CAAC,iBAAiB,EAAE,CAAC;QACrC,GAAG,CAAC,KAAK,CAAC,iBAAiB,GAAG,CAAC,GAAG,CAAC,KAAK,CAAC,iBAAiB,IAAI,CAAC,CAAC,GAAG,QAAQ,CAAC,KAAK,CAAC,iBAAiB,CAAC;IACtG,CAAC;IACD,IAAI,QAAQ,CAAC,KAAK,CAAC,wBAAwB,EAAE,CAAC;QAC5C,GAAG,CAAC,KAAK,CAAC,wBAAwB,GAAG,CAAC,GAAG,CAAC,KAAK,CAAC,wBAAwB,IAAI,CAAC,CAAC,GAAG,QAAQ,CAAC,KAAK,CAAC,wBAAwB,CAAC;IAC3H,CAAC;IAED,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;IAExC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,mBAAmB,EAAE;QACnC,OAAO,EAAE,GAAG,CAAC,SAAS;QACtB,KAAK,EAAE,IAAI,CAAC,SAAS;QACrB,UAAU,EAAE,QAAQ,CAAC,SAAS;QAC9B,UAAU,EAAE,QAAQ,CAAC,KAAK;QAC1B,cAAc,EAAE,aAAa,CAAC,MAAM;QACpC,gBAAgB,EAAE,UAAU;QAC5B,QAAQ;KACT,CAAC,CAAC;IAEH,wCAAwC;IACxC,MAAM,MAAM,GAAe;QACzB,MAAM,EAAE,UAAU;QAClB,QAAQ,EAAE,kBAAkB;QAC5B,MAAM,EAAE,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS;QACzC,OAAO,EAAE,aAAa,IAAI,8BAA8B;KACzD,CAAC;IAEF,iDAAiD;IACjD,OAAO,CAAC,IAAI,CAAC;QACX,IAAI,EAAE,YAAY,CAAC,cAAc;QACjC,OAAO,EAAE,UAAU;QACnB,MAAM,EAAE,MAAM,CAAC,MAAM;QACrB,WAAW,EAAE,QAAQ;QACrB,GAAG,CAAC,MAAM,CAAC,MAAM,KAAK,OAAO,CAAC,CAAC,CAAC,EAAC,KAAK,EAAE,MAAM,CAAC,OAAO,EAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAC7D,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;KACpC,CAAC,CAAC;IAEH,8DAA8D;IAC9D,8EAA8E;IAC9E,MAAM,cAAc,GAAG;QACrB,IAAI,EAAE,WAAoB;QAC1B,OAAO,EAAE,EAAC,UAAU,EAAE,QAAQ,EAAE,kBAAkB,EAAE,IAAI,EAAE,EAAE,EAAC;QAC7D,KAAK,EAAE,KAAK,CAAC,KAAK;QAClB,OAAO,EAAE,KAAK,CAAC,OAAO;KACvB,CAAC;IAEF,OAAO,mBAAmB,CAAC,cAAc,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,CAAC,CAAC;AACnE,CAAC;AAED,8EAA8E;AAC9E,sBAAsB;AACtB,8EAA8E;AAE9E;;;;;;GAMG;AACH,SAAS,sBAAsB,CAAC,SAAiB,EAAE,UAAkB,EAAE,SAAmB;IACxF,MAAM,QAAQ,GAAG,SAAS,CAAC,MAAM,GAAG,CAAC;QACnC,CAAC,CAAC,wBAAwB,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;QAChD,CAAC,CAAC,EAAE,CAAC;IAEP,OAAO,YAAY,SAAS;;sHAEwF,QAAQ,EAAE,CAAC;AACjI,CAAC"}

package/dist/src/agent/states/executing.d.ts

@@ -0,0 +1,21 @@
+/**
+ * @license
+ * Copyright 2026 Amodal Labs, Inc.
+ * SPDX-License-Identifier: MIT
+ */
+import type { SSEEvent } from '../../types.js';
+import type { ToolDefinition } from '../../tools/types.js';
+import type { ExecutingState, AgentContext, TransitionResult, ToolCall, ToolResult } from '../loop-types.js';
+/**
+ * Handle the EXECUTING state — execute a single tool call.
+ */
+export declare function handleExecuting(state: ExecutingState, ctx: AgentContext): Promise<TransitionResult>;
+/**
+ * Execute a tool call, building the ToolContext from the AgentContext.
+ * Enforces a timeout via AbortSignal to prevent hanging on broken tools.
+ */
+export declare function executeTool(call: ToolCall, toolDef: ToolDefinition, ctx: AgentContext): Promise<unknown>;
+/**
+ * After a tool result: append to messages, then continue to next call or back to THINKING.
+ */
+export declare function nextAfterToolResult(state: ExecutingState, result: ToolResult, effects: SSEEvent[], ctx: AgentContext): TransitionResult;

package/dist/src/agent/states/executing.js

@@ -0,0 +1,308 @@
+/**
+ * @license
+ * Copyright 2026 Amodal Labs, Inc.
+ * SPDX-License-Identifier: MIT
+ */
+import { SSEEventType } from '../../types.js';
+import { estimateTokenCount } from '../token-estimate.js';
+import { DISPATCH_TOOL_NAME } from '../../tools/dispatch-tool.js';
+/**
+ * Handle the EXECUTING state — execute a single tool call.
+ */
+export async function handleExecuting(state, ctx) {
+    const { current, queue } = state;
+    const effects = [];
+    const timestamp = new Date().toISOString();
+    // 1. Look up the tool
+    const toolDef = ctx.toolRegistry.get(current.toolName);
+    if (!toolDef) {
+        const result = {
+            callId: current.toolCallId,
+            toolName: current.toolName,
+            status: 'error',
+            content: `Tool "${current.toolName}" not found. Available tools: ${ctx.toolRegistry.names().join(', ')}`,
+        };
+        ctx.logger.warn('tool_not_found', {
+            tool: current.toolName,
+            session: ctx.sessionId,
+        });
+        return nextAfterToolResult(state, result, effects, ctx);
+    }
+    // 2. Validate args against schema (cheap guard against hallucinated params)
+    //    Only for Zod schemas — FlexibleSchema (MCP/custom tools) uses jsonSchema()
+    //    which is validated by the AI SDK at the provider level.
+    if ('safeParse' in toolDef.parameters) {
+        const validation = toolDef.parameters.safeParse(current.args);
+        if (!validation.success) {
+            const result = {
+                callId: current.toolCallId,
+                toolName: current.toolName,
+                status: 'error',
+                content: `Invalid parameters: ${validation.error.message}`,
+            };
+            ctx.logger.warn('tool_args_invalid', {
+                tool: current.toolName,
+                callId: current.toolCallId,
+                session: ctx.sessionId,
+                errors: validation.error.issues,
+            });
+            return nextAfterToolResult(state, result, effects, ctx);
+        }
+    }
+    // 3. Check permissions
+    //    Currently scoped to connection tools only — they have access.json ACLs
+    //    with endpoint-level allow/deny/confirm rules. Store tools and admin tools
+    //    have their own guards (read-only paths, blocked filenames, schema validation).
+    //    TODO: generalize to any tool via a requiresConfirmation metadata flag on
+    //    ToolDefinition so store/custom/admin tools can opt into the same gate.
+    if (toolDef.metadata?.category === 'connection' && toolDef.metadata.connection) {
+        const method = typeof current.args['method'] === 'string' ? current.args['method'] : 'GET';
+        const endpoint = typeof current.args['endpoint'] === 'string' ? current.args['endpoint'] : '/';
+        const rawIntent = current.args['intent'];
+        const intent = rawIntent === 'write' || rawIntent === 'confirmed_write' ? rawIntent : 'read';
+        const permResult = ctx.permissionChecker.check({
+            connection: toolDef.metadata.connection,
+            endpointPath: `${method} ${endpoint}`,
+            intent,
+            method,
+            params: current.args,
+        });
+        if (!permResult.allowed) {
+            const result = {
+                callId: current.toolCallId,
+                toolName: current.toolName,
+                status: 'error',
+                content: `Permission denied: ${permResult.reason}`,
+            };
+            ctx.logger.warn('tool_permission_denied', {
+                tool: current.toolName,
+                reason: permResult.reason,
+                session: ctx.sessionId,
+            });
+            return nextAfterToolResult(state, result, effects, ctx);
+        }
+        if (permResult.allowed && permResult.requiresConfirmation) {
+            return {
+                next: { type: 'confirming', call: current, remainingQueue: queue },
+                effects: [...effects, {
+                        type: SSEEventType.ConfirmationRequired,
+                        endpoint,
+                        method,
+                        reason: permResult.reason,
+                        escalated: false,
+                        timestamp,
+                    }],
+            };
+        }
+    }
+    // 4. Intercept dispatch_task — transition to DISPATCHING instead of executing
+    if (current.toolName === DISPATCH_TOOL_NAME) {
+        const args = current.args;
+        const rawTools = Array.isArray(args['tools']) ? args['tools'] : [];
+        const toolSubset = rawTools
+            .filter((t) => typeof t === 'string')
+            .filter((t) => t !== DISPATCH_TOOL_NAME);
+        effects.push({
+            type: SSEEventType.ToolCallStart,
+            tool_name: current.toolName,
+            tool_id: current.toolCallId,
+            parameters: sanitizeParams(current.args),
+            timestamp,
+        });
+        return {
+            next: {
+                type: 'dispatching',
+                task: {
+                    agentName: String(args['agent_name'] ?? 'sub-agent'),
+                    toolSubset,
+                    prompt: String(args['prompt'] ?? ''),
+                    maxTurns: typeof args['max_turns'] === 'number' ? args['max_turns'] : undefined,
+                },
+                toolCallId: current.toolCallId,
+                queue,
+                results: state.results,
+            },
+            effects,
+        };
+    }
+    // 5. Execute the tool
+    effects.push({
+        type: SSEEventType.ToolCallStart,
+        tool_name: current.toolName,
+        tool_id: current.toolCallId,
+        parameters: sanitizeParams(current.args),
+        timestamp,
+    });
+    const startedAt = Date.now();
+    let result;
+    try {
+        // Check pre-execution cache first (read-only tools started during streaming)
+        const cached = ctx.preExecutionCache.get(current.toolCallId);
+        const output = cached
+            ? await cached
+            : await executeTool(current, toolDef, ctx);
+        const content = typeof output === 'string' ? output : JSON.stringify(output);
+        // Detect error-as-result pattern: tool returned {error: "..."} without throwing.
+        // This happens when tool internals catch errors and return them as data.
+        // Treat these as failures so the SSE event shows status: 'error'.
+        const isErrorResult = typeof output === 'object' && output !== null
+            && 'error' in output && typeof output['error'] === 'string'
+            && Object.keys(output).length === 1;
+        result = {
+            callId: current.toolCallId,
+            toolName: current.toolName,
+            status: isErrorResult ? 'error' : 'success',
+            content,
+        };
+        if (isErrorResult) {
+            ctx.logger.warn('tool_returned_error', {
+                tool: current.toolName,
+                callId: current.toolCallId,
+                error: output['error'],
+                session: ctx.sessionId,
+                duration: Date.now() - startedAt,
+            });
+        }
+    }
+    catch (err) {
+        // Tool execution failed — don't crash the loop, tell the model what happened
+        result = {
+            callId: current.toolCallId,
+            toolName: current.toolName,
+            status: 'error',
+            content: `Tool execution failed: ${err instanceof Error ? err.message : String(err)}`,
+        };
+        ctx.logger.error('tool_execution_error', {
+            tool: current.toolName,
+            callId: current.toolCallId,
+            error: err instanceof Error ? err.message : String(err),
+            session: ctx.sessionId,
+            tenant: ctx.tenantId,
+            duration: Date.now() - startedAt,
+        });
+    }
+    const duration = Date.now() - startedAt;
+    effects.push({
+        type: SSEEventType.ToolCallResult,
+        tool_id: current.toolCallId,
+        status: result.status,
+        duration_ms: duration,
+        ...(result.status === 'error' ? { error: result.content } : {}),
+        timestamp: new Date().toISOString(),
+    });
+    ctx.logger.info('tool_call', {
+        tool: current.toolName,
+        callId: current.toolCallId,
+        status: result.status,
+        session: ctx.sessionId,
+        tenant: ctx.tenantId,
+        duration,
+    });
+    return nextAfterToolResult(state, result, effects, ctx);
+}
+/**
+ * Execute a tool call, building the ToolContext from the AgentContext.
+ * Enforces a timeout via AbortSignal to prevent hanging on broken tools.
+ */
+export async function executeTool(call, toolDef, ctx) {
+    const toolCtx = ctx.buildToolContext(call.toolCallId);
+    // Combine session abort signal with a per-tool timeout.
+    // Mutate instead of spread — toolCtx.request() reads ctx.signal at call
+    // time, so the combined signal propagates to HTTP requests and other
+    // async operations inside the tool.
+    const timeoutSignal = AbortSignal.timeout(ctx.config.toolTimeoutMs);
+    toolCtx.signal = AbortSignal.any([ctx.signal, timeoutSignal]);
+    return toolDef.execute(call.args, toolCtx);
+}
+/**
+ * After a tool result: append to messages, then continue to next call or back to THINKING.
+ */
+export function nextAfterToolResult(state, result, effects, ctx) {
+    // Smart snipping: if a tool result exceeds maxResultSize, keep the first
+    // and last 2K chars with a [snipped] marker in between. This preserves
+    // the beginning (usually headers/structure) and end (usually the answer)
+    // while cutting the middle bulk.
+    if (result.content.length > ctx.config.maxResultSize) {
+        const originalSize = result.content.length;
+        const keepChars = 2_000;
+        const head = result.content.slice(0, keepChars);
+        const tail = result.content.slice(-keepChars);
+        result = {
+            ...result,
+            content: `${head}\n\n[... snipped ${originalSize - keepChars * 2} chars — full output was ${originalSize} chars ...]\n\n${tail}`,
+        };
+        ctx.logger.info('tool_result_snipped', {
+            callId: result.callId,
+            tool: result.toolName,
+            originalSize,
+            snippedTo: keepChars * 2,
+            session: ctx.sessionId,
+        });
+    }
+    // Append tool result message
+    const resultMessage = buildToolResultMessage(result);
+    ctx.messages = [...ctx.messages, resultMessage];
+    const allResults = [...state.results, result];
+    // More tool calls in the queue?
+    if (state.queue.length > 0) {
+        const [next, ...rest] = state.queue;
+        return {
+            next: {
+                type: 'executing',
+                queue: rest,
+                current: next,
+                results: allResults,
+            },
+            effects,
+        };
+    }
+    // All tool calls done — check if context is heavy enough to compact
+    const estimatedTokens = estimateTokenCount(ctx.messages);
+    if (estimatedTokens > ctx.maxContextTokens * ctx.config.compactThreshold) {
+        ctx.logger.info('context_compaction_triggered', {
+            session: ctx.sessionId,
+            estimatedTokens,
+            maxContextTokens: ctx.maxContextTokens,
+            threshold: ctx.config.compactThreshold,
+        });
+        return {
+            next: { type: 'compacting', messages: ctx.messages, estimatedTokens },
+            effects,
+        };
+    }
+    // Context OK — back to THINKING for the next LLM turn
+    return {
+        next: { type: 'thinking', messages: ctx.messages },
+        effects,
+    };
+}
+/**
+ * Build a tool result message in AI SDK format.
+ */
+function buildToolResultMessage(result) {
+    return {
+        role: 'tool',
+        content: [{
+                type: 'tool-result',
+                toolCallId: result.callId,
+                toolName: result.toolName,
+                output: { type: 'text', value: result.content },
+            }],
+    };
+}
+/**
+ * Strip sensitive fields from tool call parameters for SSE events.
+ */
+function sanitizeParams(params) {
+    const sanitized = {};
+    for (const [key, value] of Object.entries(params)) {
+        if (/token|secret|password|key|auth/i.test(key)) {
+            sanitized[key] = '[REDACTED]';
+        }
+        else {
+            sanitized[key] = value;
+        }
+    }
+    return sanitized;
+}
+//# sourceMappingURL=executing.js.map

package/dist/src/agent/states/executing.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"executing.js","sourceRoot":"","sources":["../../../../src/agent/states/executing.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAiBH,OAAO,EAAC,YAAY,EAAC,MAAM,gBAAgB,CAAC;AAU5C,OAAO,EAAC,kBAAkB,EAAC,MAAM,sBAAsB,CAAC;AACxD,OAAO,EAAC,kBAAkB,EAAC,MAAM,8BAA8B,CAAC;AAEhE;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,eAAe,CACnC,KAAqB,EACrB,GAAiB;IAEjB,MAAM,EAAC,OAAO,EAAE,KAAK,EAAC,GAAG,KAAK,CAAC;IAC/B,MAAM,OAAO,GAAe,EAAE,CAAC;IAC/B,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IAE3C,sBAAsB;IACtB,MAAM,OAAO,GAAG,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IACvD,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,MAAM,MAAM,GAAe;YACzB,MAAM,EAAE,OAAO,CAAC,UAAU;YAC1B,QAAQ,EAAE,OAAO,CAAC,QAAQ;YAC1B,MAAM,EAAE,OAAO;YACf,OAAO,EAAE,SAAS,OAAO,CAAC,QAAQ,iCAAiC,GAAG,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;SACzG,CAAC;QACF,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,gBAAgB,EAAE;YAChC,IAAI,EAAE,OAAO,CAAC,QAAQ;YACtB,OAAO,EAAE,GAAG,CAAC,SAAS;SACvB,CAAC,CAAC;QACH,OAAO,mBAAmB,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,CAAC,CAAC;IAC1D,CAAC;IAED,4EAA4E;IAC5E,gFAAgF;IAChF,6DAA6D;IAC7D,IAAI,WAAW,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;QACtC,MAAM,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC,SAAS,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAC9D,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC;YACxB,MAAM,MAAM,GAAe;gBACzB,MAAM,EAAE,OAAO,CAAC,UAAU;gBAC1B,QAAQ,EAAE,OAAO,CAAC,QAAQ;gBAC1B,MAAM,EAAE,OAAO;gBACf,OAAO,EAAE,uBAAuB,UAAU,CAAC,KAAK,CAAC,OAAO,EAAE;aAC3D,CAAC;YACF,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,mBAAmB,EAAE;gBACnC,IAAI,EAAE,OAAO,CAAC,QAAQ;gBACtB,MAAM,EAAE,OAAO,CAAC,UAAU;gBAC1B,OAAO,EAAE,GAAG,CAAC,SAAS;gBACtB,MAAM,EAAE,UAAU,CAAC,KAAK,CAAC,MAAM;aAChC,CAAC,CAAC;YACH,OAAO,mBAAmB,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC;IAED,uBAAuB;IACvB,4EAA4E;IAC5E,+EAA+E;IAC/E,oFAAoF;IACpF,8EAA8E;IAC9E,4EAA4E;IAC5E,IAAI,OAAO,CAAC,QAAQ,EAAE,QAAQ,KAAK,YAAY,IAAI,OAAO,CAAC,QAAQ,CAAC,UAAU,EAAE,CAAC;QAC/E,MAAM,MAAM,GAAG,OAAO,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;QAC3F,MAAM,QAAQ,GAAG,OAAO,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC;QAC/F,MAAM,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACzC,MAAM,MAAM,GACV,SAAS,KAAK,OAAO,IAAI,SAAS,KAAK,iBAAiB,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC;QAEhF,MAAM,UAAU,GAAG,GAAG,CAAC,iBAAiB,CAAC,KAAK,CAAC;YAC7C,UAAU,EAAE,OAAO,CAAC,QAAQ,CAAC,UAAU;YACvC,YAAY,EAAE,GAAG,MAAM,IAAI,QAAQ,EAAE;YACrC,MAAM;YACN,MAAM;YACN,MAAM,EAAE,OAAO,CAAC,IAAI;SACrB,CAAC,CAAC;QAEH,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC;YACxB,MAAM,MAAM,GAAe;gBACzB,MAAM,EAAE,OAAO,CAAC,UAAU;gBAC1B,QAAQ,EAAE,OAAO,CAAC,QAAQ;gBAC1B,MAAM,EAAE,OAAO;gBACf,OAAO,EAAE,sBAAsB,UAAU,CAAC,MAAM,EAAE;aACnD,CAAC;YACF,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,wBAAwB,EAAE;gBACxC,IAAI,EAAE,OAAO,CAAC,QAAQ;gBACtB,MAAM,EAAE,UAAU,CAAC,MAAM;gBACzB,OAAO,EAAE,GAAG,CAAC,SAAS;aACvB,CAAC,CAAC;YACH,OAAO,mBAAmB,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,CAAC,CAAC;QAC1D,CAAC;QAED,IAAI,UAAU,CAAC,OAAO,IAAI,UAAU,CAAC,oBAAoB,EAAE,CAAC;YAC1D,OAAO;gBACL,IAAI,EAAE,EAAC,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,OAAO,EAAE,cAAc,EAAE,KAAK,EAAC;gBAChE,OAAO,EAAE,CAAC,GAAG,OAAO,EAAE;wBACpB,IAAI,EAAE,YAAY,CAAC,oBAAoB;wBACvC,QAAQ;wBACR,MAAM;wBACN,MAAM,EAAE,UAAU,CAAC,MAAM;wBACzB,SAAS,EAAE,KAAK;wBAChB,SAAS;qBACV,CAAC;aACH,CAAC;QACJ,CAAC;IACH,CAAC;IAED,8EAA8E;IAC9E,IAAI,OAAO,CAAC,QAAQ,KAAK,kBAAkB,EAAE,CAAC;QAC5C,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;QAC1B,MAAM,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QACnE,MAAM,UAAU,GAAG,QAAQ;aACxB,MAAM,CAAC,CAAC,CAAC,EAAe,EAAE,CAAC,OAAO,CAAC,KAAK,QAAQ,CAAC;aACjD,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,kBAAkB,CAAC,CAAC;QAE3C,OAAO,CAAC,IAAI,CAAC;YACX,IAAI,EAAE,YAAY,CAAC,aAAa;YAChC,SAAS,EAAE,OAAO,CAAC,QAAQ;YAC3B,OAAO,EAAE,OAAO,CAAC,UAAU;YAC3B,UAAU,EAAE,cAAc,CAAC,OAAO,CAAC,IAAI,CAAC;YACxC,SAAS;SACV,CAAC,CAAC;QAEH,OAAO;YACL,IAAI,EAAE;gBACJ,IAAI,EAAE,aAAa;gBACnB,IAAI,EAAE;oBACJ,SAAS,EAAE,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,IAAI,WAAW,CAAC;oBACpD,UAAU;oBACV,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;oBACpC,QAAQ,EAAE,OAAO,IAAI,CAAC,WAAW,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,SAAS;iBAChF;gBACD,UAAU,EAAE,OAAO,CAAC,UAAU;gBAC9B,KAAK;gBACL,OAAO,EAAE,KAAK,CAAC,OAAO;aACvB;YACD,OAAO;SACR,CAAC;IACJ,CAAC;IAED,sBAAsB;IACtB,OAAO,CAAC,IAAI,CAAC;QACX,IAAI,EAAE,YAAY,CAAC,aAAa;QAChC,SAAS,EAAE,OAAO,CAAC,QAAQ;QAC3B,OAAO,EAAE,OAAO,CAAC,UAAU;QAC3B,UAAU,EAAE,cAAc,CAAC,OAAO,CAAC,IAAI,CAAC;QACxC,SAAS;KACV,CAAC,CAAC;IAEH,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC7B,IAAI,MAAkB,CAAC;IAEvB,IAAI,CAAC;QACH,6EAA6E;QAC7E,MAAM,MAAM,GAAG,GAAG,CAAC,iBAAiB,CAAC,GAAG,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QAC7D,MAAM,MAAM,GAAG,MAAM;YACnB,CAAC,CAAC,MAAM,MAAM;YACd,CAAC,CAAC,MAAM,WAAW,CAAC,OAAO,EAAE,OAAO,EAAE,GAAG,CAAC,CAAC;QAE7C,MAAM,OAAO,GAAG,OAAO,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QAE7E,iFAAiF;QACjF,yEAAyE;QACzE,kEAAkE;QAClE,MAAM,aAAa,GAAG,OAAO,MAAM,KAAK,QAAQ,IAAI,MAAM,KAAK,IAAI;eAC9D,OAAO,IAAI,MAAM,IAAI,OAAQ,MAAkC,CAAC,OAAO,CAAC,KAAK,QAAQ;eACrF,MAAM,CAAC,IAAI,CAAC,MAAiC,CAAC,CAAC,MAAM,KAAK,CAAC,CAAC;QAEjE,MAAM,GAAG;YACP,MAAM,EAAE,OAAO,CAAC,UAAU;YAC1B,QAAQ,EAAE,OAAO,CAAC,QAAQ;YAC1B,MAAM,EAAE,aAAa,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS;YAC3C,OAAO;SACR,CAAC;QAEF,IAAI,aAAa,EAAE,CAAC;YAClB,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,qBAAqB,EAAE;gBACrC,IAAI,EAAE,OAAO,CAAC,QAAQ;gBACtB,MAAM,EAAE,OAAO,CAAC,UAAU;gBAC1B,KAAK,EAAG,MAAkC,CAAC,OAAO,CAAC;gBACnD,OAAO,EAAE,GAAG,CAAC,SAAS;gBACtB,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;aACjC,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,6EAA6E;QAC7E,MAAM,GAAG;YACP,MAAM,EAAE,OAAO,CAAC,UAAU;YAC1B,QAAQ,EAAE,OAAO,CAAC,QAAQ;YAC1B,MAAM,EAAE,OAAO;YACf,OAAO,EAAE,0BAA0B,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE;SACtF,CAAC;QACF,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,sBAAsB,EAAE;YACvC,IAAI,EAAE,OAAO,CAAC,QAAQ;YACtB,MAAM,EAAE,OAAO,CAAC,UAAU;YAC1B,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC;YACvD,OAAO,EAAE,GAAG,CAAC,SAAS;YACtB,MAAM,EAAE,GAAG,CAAC,QAAQ;YACpB,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;SACjC,CAAC,CAAC;IACL,CAAC;IAED,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;IACxC,OAAO,CAAC,IAAI,CAAC;QACX,IAAI,EAAE,YAAY,CAAC,cAAc;QACjC,OAAO,EAAE,OAAO,CAAC,UAAU;QAC3B,MAAM,EAAE,MAAM,CAAC,MAAM;QACrB,WAAW,EAAE,QAAQ;QACrB,GAAG,CAAC,MAAM,CAAC,MAAM,KAAK,OAAO,CAAC,CAAC,CAAC,EAAC,KAAK,EAAE,MAAM,CAAC,OAAO,EAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAC7D,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;KACpC,CAAC,CAAC;IAEH,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,EAAE;QAC3B,IAAI,EAAE,OAAO,CAAC,QAAQ;QACtB,MAAM,EAAE,OAAO,CAAC,UAAU;QAC1B,MAAM,EAAE,MAAM,CAAC,MAAM;QACrB,OAAO,EAAE,GAAG,CAAC,SAAS;QACtB,MAAM,EAAE,GAAG,CAAC,QAAQ;QACpB,QAAQ;KACT,CAAC,CAAC;IAEH,OAAO,mBAAmB,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,CAAC,CAAC;AAC1D,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,WAAW,CAC/B,IAAc,EACd,OAAuB,EACvB,GAAiB;IAEjB,MAAM,OAAO,GAAG,GAAG,CAAC,gBAAgB,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAEtD,wDAAwD;IACxD,wEAAwE;IACxE,qEAAqE;IACrE,oCAAoC;IACpC,MAAM,aAAa,GAAG,WAAW,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;IACpE,OAAO,CAAC,MAAM,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC,CAAC;IAE9D,OAAO,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;AAC7C,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,mBAAmB,CACjC,KAAqB,EACrB,MAAkB,EAClB,OAAmB,EACnB,GAAiB;IAEjB,yEAAyE;IACzE,uEAAuE;IACvE,yEAAyE;IACzE,iCAAiC;IACjC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,GAAG,GAAG,CAAC,MAAM,CAAC,aAAa,EAAE,CAAC;QACrD,MAAM,YAAY,GAAG,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC;QAC3C,MAAM,SAAS,GAAG,KAAK,CAAC;QACxB,MAAM,IAAI,GAAG,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;QAChD,MAAM,IAAI,GAAG,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,CAAC;QAC9C,MAAM,GAAG;YACP,GAAG,MAAM;YACT,OAAO,EAAE,GAAG,IAAI,oBAAoB,YAAY,GAAG,SAAS,GAAG,CAAC,4BAA4B,YAAY,kBAAkB,IAAI,EAAE;SACjI,CAAC;QACF,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,qBAAqB,EAAE;YACrC,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,IAAI,EAAE,MAAM,CAAC,QAAQ;YACrB,YAAY;YACZ,SAAS,EAAE,SAAS,GAAG,CAAC;YACxB,OAAO,EAAE,GAAG,CAAC,SAAS;SACvB,CAAC,CAAC;IACL,CAAC;IAED,6BAA6B;IAC7B,MAAM,aAAa,GAAG,sBAAsB,CAAC,MAAM,CAAC,CAAC;IACrD,GAAG,CAAC,QAAQ,GAAG,CAAC,GAAG,GAAG,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC;IAEhD,MAAM,UAAU,GAAG,CAAC,GAAG,KAAK,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;IAE9C,gCAAgC;IAChC,IAAI,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC3B,MAAM,CAAC,IAAI,EAAE,GAAG,IAAI,CAAC,GAAG,KAAK,CAAC,KAAK,CAAC;QACpC,OAAO;YACL,IAAI,EAAE;gBACJ,IAAI,EAAE,WAAW;gBACjB,KAAK,EAAE,IAAI;gBACX,OAAO,EAAE,IAAI;gBACb,OAAO,EAAE,UAAU;aACpB;YACD,OAAO;SACR,CAAC;IACJ,CAAC;IAED,oEAAoE;IACpE,MAAM,eAAe,GAAG,kBAAkB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IACzD,IAAI,eAAe,GAAG,GAAG,CAAC,gBAAgB,GAAG,GAAG,CAAC,MAAM,CAAC,gBAAgB,EAAE,CAAC;QACzE,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,8BAA8B,EAAE;YAC9C,OAAO,EAAE,GAAG,CAAC,SAAS;YACtB,eAAe;YACf,gBAAgB,EAAE,GAAG,CAAC,gBAAgB;YACtC,SAAS,EAAE,GAAG,CAAC,MAAM,CAAC,gBAAgB;SACvC,CAAC,CAAC;QACH,OAAO;YACL,IAAI,EAAE,EAAC,IAAI,EAAE,YAAY,EAAE,QAAQ,EAAE,GAAG,CAAC,QAAQ,EAAE,eAAe,EAAC;YACnE,OAAO;SACR,CAAC;IACJ,CAAC;IAED,sDAAsD;IACtD,OAAO;QACL,IAAI,EAAE,EAAC,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,GAAG,CAAC,QAAQ,EAAC;QAChD,OAAO;KACR,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,SAAS,sBAAsB,CAAC,MAAkB;IAChD,OAAO;QACL,IAAI,EAAE,MAAM;QACZ,OAAO,EAAE,CAAC;gBACR,IAAI,EAAE,aAAsB;gBAC5B,UAAU,EAAE,MAAM,CAAC,MAAM;gBACzB,QAAQ,EAAE,MAAM,CAAC,QAAQ;gBACzB,MAAM,EAAE,EAAC,IAAI,EAAE,MAAe,EAAE,KAAK,EAAE,MAAM,CAAC,OAAO,EAAC;aACvD,CAAC;KACH,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,SAAS,cAAc,CAAC,MAA+B;IACrD,MAAM,SAAS,GAA4B,EAAE,CAAC;IAC9C,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;QAClD,IAAI,iCAAiC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;YAChD,SAAS,CAAC,GAAG,CAAC,GAAG,YAAY,CAAC;QAChC,CAAC;aAAM,CAAC;YACN,SAAS,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;QACzB,CAAC;IACH,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC"}

package/dist/src/agent/states/streaming.d.ts

@@ -0,0 +1,10 @@
+/**
+ * @license
+ * Copyright 2026 Amodal Labs, Inc.
+ * SPDX-License-Identifier: MIT
+ */
+import type { StreamingState, AgentContext, TransitionResult } from '../loop-types.js';
+/**
+ * Handle the STREAMING state.
+ */
+export declare function handleStreaming(state: StreamingState, ctx: AgentContext): Promise<TransitionResult>;