@aexol/spectral 0.9.230 → 0.9.234

package/dist/sdk/agent-core/agent-loop.js

@@ -2,7 +2,10 @@
  * Agent loop that works with AgentMessage throughout.
  * Transforms to Message[] only at the LLM call boundary.
  */
-import { EventStream, formatPaymentRequiredMessage, formatRateLimitRetryMessage, getRateLimitRetryDelayMs, isPaymentRequiredError, MAX_RATE_LIMIT_RETRIES, streamSimple, isRateLimitError, isTransientApiError, validateToolArguments, } from "../ai/index.js";
+import { EventStream, formatPaymentRequiredMessage, formatRateLimitRetryMessage, getRateLimitRetryDelayMs, isPaymentRequiredError, MAX_RATE_LIMIT_RETRIES, streamSimple, isRateLimitError, isTransientApiError, } from "../ai/index.js";
+import { executeToolCalls } from "./tool-execution.js";
+// Re-exported for backwards compatibility (moved to tool-execution.ts).
+export { hasMissingRequiredArguments } from "./tool-execution.js";
 /** Maximum consecutive tool call errors before terminating the loop. */
 const MAX_CONSECUTIVE_TOOL_ERRORS = 10;
 /** Maximum retries for transient provider API errors. */
@@ -264,13 +267,21 @@ async function runLoop(initialContext, newMessages, initialConfig, signal, emit,
                     newMessages.push(result);
                 }
                 // Fix 1: Track consecutive tool call errors; terminate after threshold.
-                const turnHasError = toolResults.some((r) => r.isError);
+                // User-stopped results (`stoppedByUser`) are deliberate user actions,
+                // not tool failures — they must not count toward the threshold, or a
+                // run where the user stops every tool call dies as "10 consecutive
+                // tool call errors" (defeating the per-call stop feature). A batch of
+                // only user-stopped results leaves the counter unchanged; normal
+                // results still reset it.
+                const turnHasError = toolResults.some((r) => r.isError && !r.stoppedByUser);
+                const batchAllUserStopped = toolResults.length > 0 &&
+                    toolResults.every((r) => r.isError && r.stoppedByUser === true);
                 if (turnHasError) {
                     consecutiveToolErrors++;
                     if (consecutiveToolErrors >= MAX_CONSECUTIVE_TOOL_ERRORS) {
                         const lastErrorResult = [...toolResults]
                             .reverse()
-                            .find((r) => r.isError);
+                            .find((r) => r.isError && !r.stoppedByUser);
                         const lastErrorMsg = lastErrorResult?.content.find((c) => c.type === "text")?.text ||
                             "Unknown tool error";
                         const finalMessage = {
@@ -300,7 +311,9 @@ async function runLoop(initialContext, newMessages, initialConfig, signal, emit,
                         return;
                     }
                 }
-                else {
+                else if (!batchAllUserStopped) {
+                    // Pure success/normal batch: reset as today. An all-user-stopped
+                    // batch is neither a real error nor a success — counter unchanged.
                     consecutiveToolErrors = 0;
                 }
             }
@@ -432,290 +445,3 @@ async function streamAssistantResponse(context, config, signal, emit, streamFn)
     await emit({ type: "message_end", message: finalMessage });
     return finalMessage;
 }
-/**
- * Execute tool calls from an assistant message.
- */
-async function executeToolCalls(currentContext, assistantMessage, config, signal, emit) {
-    const toolCalls = assistantMessage.content.filter((c) => c.type === "toolCall");
-    const hasSequentialToolCall = toolCalls.some((tc) => currentContext.tools?.find((t) => t.name === tc.name)?.executionMode ===
-        "sequential");
-    if (config.toolExecution === "sequential" || hasSequentialToolCall) {
-        return executeToolCallsSequential(currentContext, assistantMessage, toolCalls, config, signal, emit);
-    }
-    return executeToolCallsParallel(currentContext, assistantMessage, toolCalls, config, signal, emit);
-}
-async function executeToolCallsSequential(currentContext, assistantMessage, toolCalls, config, signal, emit) {
-    const finalizedCalls = [];
-    const messages = [];
-    for (const toolCall of toolCalls) {
-        await emit({
-            type: "tool_execution_start",
-            toolCallId: toolCall.id,
-            toolName: toolCall.name,
-            args: toolCall.arguments,
-        });
-        const preparation = await prepareToolCall(currentContext, assistantMessage, toolCall, config, signal);
-        let finalized;
-        if (preparation.kind === "immediate") {
-            finalized = {
-                toolCall,
-                result: preparation.result,
-                isError: preparation.isError,
-            };
-        }
-        else {
-            const executed = await executePreparedToolCall(preparation, signal, emit);
-            finalized = await finalizeExecutedToolCall(currentContext, assistantMessage, preparation, executed, config, signal);
-        }
-        await emitToolExecutionEnd(finalized, emit);
-        const toolResultMessage = createToolResultMessage(finalized);
-        await emitToolResultMessage(toolResultMessage, emit);
-        finalizedCalls.push(finalized);
-        messages.push(toolResultMessage);
-        if (signal?.aborted) {
-            break;
-        }
-    }
-    return {
-        messages,
-        terminate: shouldTerminateToolBatch(finalizedCalls),
-    };
-}
-async function executeToolCallsParallel(currentContext, assistantMessage, toolCalls, config, signal, emit) {
-    const finalizedCalls = [];
-    for (const toolCall of toolCalls) {
-        await emit({
-            type: "tool_execution_start",
-            toolCallId: toolCall.id,
-            toolName: toolCall.name,
-            args: toolCall.arguments,
-        });
-        const preparation = await prepareToolCall(currentContext, assistantMessage, toolCall, config, signal);
-        if (preparation.kind === "immediate") {
-            const finalized = {
-                toolCall,
-                result: preparation.result,
-                isError: preparation.isError,
-            };
-            await emitToolExecutionEnd(finalized, emit);
-            finalizedCalls.push(finalized);
-            if (signal?.aborted) {
-                break;
-            }
-            continue;
-        }
-        finalizedCalls.push(async () => {
-            const executed = await executePreparedToolCall(preparation, signal, emit);
-            const finalized = await finalizeExecutedToolCall(currentContext, assistantMessage, preparation, executed, config, signal);
-            await emitToolExecutionEnd(finalized, emit);
-            return finalized;
-        });
-        if (signal?.aborted) {
-            break;
-        }
-    }
-    const orderedFinalizedCalls = await Promise.all(finalizedCalls.map((entry) => typeof entry === "function" ? entry() : Promise.resolve(entry)));
-    const messages = [];
-    for (const finalized of orderedFinalizedCalls) {
-        const toolResultMessage = createToolResultMessage(finalized);
-        await emitToolResultMessage(toolResultMessage, emit);
-        messages.push(toolResultMessage);
-    }
-    return {
-        messages,
-        terminate: shouldTerminateToolBatch(orderedFinalizedCalls),
-    };
-}
-function shouldTerminateToolBatch(finalizedCalls) {
-    return (finalizedCalls.length > 0 &&
-        finalizedCalls.every((finalized) => finalized.result.terminate === true));
-}
-function prepareToolCallArguments(tool, toolCall) {
-    if (!tool.prepareArguments) {
-        return toolCall;
-    }
-    const preparedArguments = tool.prepareArguments(toolCall.arguments);
-    if (preparedArguments === toolCall.arguments) {
-        return toolCall;
-    }
-    return {
-        ...toolCall,
-        arguments: preparedArguments,
-    };
-}
-async function prepareToolCall(currentContext, assistantMessage, toolCall, config, signal) {
-    const tool = currentContext.tools?.find((t) => t.name === toolCall.name);
-    if (!tool) {
-        return {
-            kind: "immediate",
-            result: createErrorToolResult(`Tool ${toolCall.name} not found`),
-            isError: true,
-        };
-    }
-    // If the provider truncated the response mid tool-call (stopReason "length"),
-    // the streaming JSON parser may have produced an empty or incomplete arguments
-    // object. Validating it would surface a raw `Validation failed ... You provided:
-    // {}` error to the model. Instead, return a soft error so the agent-session
-    // length-continuation path (which treats truncated tool calls as visible output)
-    // can resubmit and let the model re-emit the full tool call.
-    if (assistantMessage.stopReason === "length" && hasMissingRequiredArguments(tool, toolCall)) {
-        return {
-            kind: "immediate",
-            result: createErrorToolResult(`Tool "${toolCall.name}" was truncated before its arguments were fully emitted. ` +
-                `Retrying: please re-emit the complete tool call with all required parameters.`),
-            isError: true,
-        };
-    }
-    try {
-        const preparedToolCall = prepareToolCallArguments(tool, toolCall);
-        const validatedArgs = validateToolArguments(tool, preparedToolCall);
-        if (config.beforeToolCall) {
-            const beforeResult = await config.beforeToolCall({
-                assistantMessage,
-                toolCall,
-                args: validatedArgs,
-                context: currentContext,
-            }, signal);
-            if (signal?.aborted) {
-                return {
-                    kind: "immediate",
-                    result: createErrorToolResult("Operation aborted"),
-                    isError: true,
-                };
-            }
-            if (beforeResult?.block) {
-                return {
-                    kind: "immediate",
-                    result: createErrorToolResult(beforeResult.reason || "Tool execution was blocked"),
-                    isError: true,
-                };
-            }
-        }
-        if (signal?.aborted) {
-            return {
-                kind: "immediate",
-                result: createErrorToolResult("Operation aborted"),
-                isError: true,
-            };
-        }
-        return {
-            kind: "prepared",
-            toolCall,
-            tool,
-            args: validatedArgs,
-        };
-    }
-    catch (error) {
-        return {
-            kind: "immediate",
-            result: createErrorToolResult(error instanceof Error ? error.message : String(error)),
-            isError: true,
-        };
-    }
-}
-async function executePreparedToolCall(prepared, signal, emit) {
-    const updateEvents = [];
-    try {
-        const result = await prepared.tool.execute(prepared.toolCall.id, prepared.args, signal, (partialResult) => {
-            updateEvents.push(Promise.resolve(emit({
-                type: "tool_execution_update",
-                toolCallId: prepared.toolCall.id,
-                toolName: prepared.toolCall.name,
-                args: prepared.toolCall.arguments,
-                partialResult,
-            })));
-        });
-        await Promise.all(updateEvents);
-        return { result, isError: false };
-    }
-    catch (error) {
-        await Promise.all(updateEvents);
-        return {
-            result: createErrorToolResult(error instanceof Error ? error.message : String(error)),
-            isError: true,
-        };
-    }
-}
-async function finalizeExecutedToolCall(currentContext, assistantMessage, prepared, executed, config, signal) {
-    let result = executed.result;
-    let isError = executed.isError;
-    if (config.afterToolCall) {
-        try {
-            const afterResult = await config.afterToolCall({
-                assistantMessage,
-                toolCall: prepared.toolCall,
-                args: prepared.args,
-                result,
-                isError,
-                context: currentContext,
-            }, signal);
-            if (afterResult) {
-                result = {
-                    content: afterResult.content ?? result.content,
-                    details: afterResult.details ?? result.details,
-                    terminate: afterResult.terminate ?? result.terminate,
-                };
-                isError = afterResult.isError ?? isError;
-            }
-        }
-        catch (error) {
-            result = createErrorToolResult(error instanceof Error ? error.message : String(error));
-            isError = true;
-        }
-    }
-    return {
-        toolCall: prepared.toolCall,
-        result,
-        isError,
-    };
-}
-function createErrorToolResult(message) {
-    return {
-        content: [{ type: "text", text: message }],
-        details: {},
-    };
-}
-/**
- * Detects whether a tool call is missing any required parameter, indicating the
- * provider truncated the response before the tool arguments were fully emitted.
- * Used together with `stopReason === "length"` to short-circuit tool validation
- * when a streamed tool call is incomplete (which would otherwise throw
- * `Validation failed ... You provided: {}`).
- */
-export function hasMissingRequiredArguments(tool, toolCall) {
-    const schema = tool.parameters;
-    const required = schema.required;
-    if (!Array.isArray(required) || required.length === 0)
-        return false;
-    const args = toolCall.arguments;
-    if (!args || typeof args !== "object")
-        return true;
-    return required.some((key) => !(key in args) ||
-        args[key] === undefined ||
-        args[key] === null ||
-        (typeof args[key] === "string" && args[key].trim() === ""));
-}
-async function emitToolExecutionEnd(finalized, emit) {
-    await emit({
-        type: "tool_execution_end",
-        toolCallId: finalized.toolCall.id,
-        toolName: finalized.toolCall.name,
-        result: finalized.result,
-        isError: finalized.isError,
-    });
-}
-function createToolResultMessage(finalized) {
-    return {
-        role: "toolResult",
-        toolCallId: finalized.toolCall.id,
-        toolName: finalized.toolCall.name,
-        content: finalized.result.content,
-        details: finalized.result.details,
-        isError: finalized.isError,
-        timestamp: Date.now(),
-    };
-}
-async function emitToolResultMessage(toolResultMessage, emit) {
-    await emit({ type: "message_start", message: toolResultMessage });
-    await emit({ type: "message_end", message: toolResultMessage });
-}

package/dist/sdk/agent-core/agent.d.ts

@@ -1,4 +1,4 @@
-import { type ImageContent, type Message, type SimpleStreamOptions, type ThinkingBudgets, type Transport } from "../ai/index.js";
+import { type ImageContent, type Message, type SimpleStreamOptions, type Transport } from "../ai/index.js";
 import type { AfterToolCallContext, AfterToolCallResult, AgentEvent, AgentLoopTurnUpdate, AgentMessage, AgentState, BeforeToolCallContext, BeforeToolCallResult, QueueMode, StreamFn, ToolExecutionMode } from "./types.js";
 export type { QueueMode } from "./types.js";
 /** Options for constructing an {@link Agent}. */
@@ -16,7 +16,6 @@ export interface AgentOptions {
     steeringMode?: QueueMode;
     followUpMode?: QueueMode;
     sessionId?: string;
-    thinkingBudgets?: ThinkingBudgets;
     transport?: Transport;
     maxRetryDelayMs?: number;
     toolExecution?: ToolExecutionMode;
@@ -32,6 +31,7 @@ export declare class Agent {
     private readonly listeners;
     private readonly steeringQueue;
     private readonly followUpQueue;
+    private readonly toolCallRegistry;
     convertToLlm: (messages: AgentMessage[]) => Message[] | Promise<Message[]>;
     transformContext?: (messages: AgentMessage[], signal?: AbortSignal) => Promise<AgentMessage[]>;
     streamFn: StreamFn;
@@ -44,8 +44,6 @@ export declare class Agent {
     private activeRun?;
     /** Session identifier forwarded to providers for cache-aware backends. */
     sessionId?: string;
-    /** Optional per-level thinking token budgets forwarded to the stream function. */
-    thinkingBudgets?: ThinkingBudgets;
     /** Preferred transport forwarded to the stream function. */
     transport: Transport;
     /** Optional cap for provider-requested retry delays. */
@@ -92,6 +90,15 @@ export declare class Agent {
     get signal(): AbortSignal | undefined;
     /** Abort the current run, if one is active. */
     abort(): void;
+    /**
+     * Stop a single in-flight tool call (user-initiated per-call stop).
+     *
+     * Aborts only the targeted call's execution — siblings keep running and the
+     * agent loop CONTINUES the turn with a synthesized user-stop tool result
+     * ("Tool call stopped by user (was running too long)"). Returns false for
+     * unknown/finished toolCallIds (no-op).
+     */
+    stopToolCall(toolCallId: string, reason?: string): Promise<boolean>;
     /**
      * Resolve when the current run and all awaited event listeners have finished.
      *

package/dist/sdk/agent-core/agent.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"agent.d.ts","sourceRoot":"","sources":["../../../src/sdk/agent-core/agent.ts"],"names":[],"mappings":"AAAA,OAAO,EACN,KAAK,YAAY,EACjB,KAAK,OAAO,EAEZ,KAAK,mBAAmB,EAGxB,KAAK,eAAe,EACpB,KAAK,SAAS,EACd,MAAM,gBAAgB,CAAC;AAExB,OAAO,KAAK,EACX,oBAAoB,EACpB,mBAAmB,EAEnB,UAAU,EAEV,mBAAmB,EACnB,YAAY,EACZ,UAAU,EAEV,qBAAqB,EACrB,oBAAoB,EACpB,SAAS,EACT,QAAQ,EACR,iBAAiB,EACjB,MAAM,YAAY,CAAC;AAEpB,YAAY,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AAkE5C,iDAAiD;AACjD,MAAM,WAAW,YAAY;IAC5B,YAAY,CAAC,EAAE,OAAO,CAAC,IAAI,CAAC,UAAU,EAAE,kBAAkB,GAAG,aAAa,GAAG,kBAAkB,GAAG,cAAc,CAAC,CAAC,CAAC;IACnH,YAAY,CAAC,EAAE,CAAC,QAAQ,EAAE,YAAY,EAAE,KAAK,OAAO,EAAE,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;IAC5E,gBAAgB,CAAC,EAAE,CAAC,QAAQ,EAAE,YAAY,EAAE,EAAE,MAAM,CAAC,EAAE,WAAW,KAAK,OAAO,CAAC,YAAY,EAAE,CAAC,CAAC;IAC/F,QAAQ,CAAC,EAAE,QAAQ,CAAC;IACpB,SAAS,CAAC,EAAE,CAAC,QAAQ,EAAE,MAAM,KAAK,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC,GAAG,MAAM,GAAG,SAAS,CAAC;IACnF,SAAS,CAAC,EAAE,mBAAmB,CAAC,WAAW,CAAC,CAAC;IAC7C,UAAU,CAAC,EAAE,mBAAmB,CAAC,YAAY,CAAC,CAAC;IAC/C,cAAc,CAAC,EAAE,CAAC,OAAO,EAAE,qBAAqB,EAAE,MAAM,CAAC,EAAE,WAAW,KAAK,OAAO,CAAC,oBAAoB,GAAG,SAAS,CAAC,CAAC;IACrH,aAAa,CAAC,EAAE,CAAC,OAAO,EAAE,oBAAoB,EAAE,MAAM,CAAC,EAAE,WAAW,KAAK,OAAO,CAAC,mBAAmB,GAAG,SAAS,CAAC,CAAC;IAClH,eAAe,CAAC,EAAE,CACjB,MAAM,CAAC,EAAE,WAAW,KAChB,OAAO,CAAC,mBAAmB,GAAG,SAAS,CAAC,GAAG,mBAAmB,GAAG,SAAS,CAAC;IAChF,YAAY,CAAC,EAAE,SAAS,CAAC;IACzB,YAAY,CAAC,EAAE,SAAS,CAAC;IACzB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,eAAe,CAAC,EAAE,eAAe,CAAC;IAClC,SAAS,CAAC,EAAE,SAAS,CAAC;IACtB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,aAAa,CAAC,EAAE,iBAAiB,CAAC;CAClC;AA4CD;;;;;GAKG;AACH,qBAAa,KAAK;IACjB,OAAO,CAAC,MAAM,CAAoB;IAClC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAA+E;IACzG,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAsB;IACpD,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAsB;IAE7C,YAAY,EAAE,CAAC,QAAQ,EAAE,YAAY,EAAE,KAAK,OAAO,EAAE,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;IAC3E,gBAAgB,CAAC,EAAE,CAAC,QAAQ,EAAE,YAAY,EAAE,EAAE,MAAM,CAAC,EAAE,WAAW,KAAK,OAAO,CAAC,YAAY,EAAE,CAAC,CAAC;IAC/F,QAAQ,EAAE,QAAQ,CAAC;IACnB,SAAS,CAAC,EAAE,CAAC,QAAQ,EAAE,MAAM,KAAK,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC,GAAG,MAAM,GAAG,SAAS,CAAC;IACnF,SAAS,CAAC,EAAE,mBAAmB,CAAC,WAAW,CAAC,CAAC;IAC7C,UAAU,CAAC,EAAE,mBAAmB,CAAC,YAAY,CAAC,CAAC;IAC/C,cAAc,CAAC,EAAE,CACvB,OAAO,EAAE,qBAAqB,EAC9B,MAAM,CAAC,EAAE,WAAW,KAChB,OAAO,CAAC,oBAAoB,GAAG,SAAS,CAAC,CAAC;IACxC,aAAa,CAAC,EAAE,CACtB,OAAO,EAAE,oBAAoB,EAC7B,MAAM,CAAC,EAAE,WAAW,KAChB,OAAO,CAAC,mBAAmB,GAAG,SAAS,CAAC,CAAC;IACvC,eAAe,CAAC,EAAE,CACxB,MAAM,CAAC,EAAE,WAAW,KAChB,OAAO,CAAC,mBAAmB,GAAG,SAAS,CAAC,GAAG,mBAAmB,GAAG,SAAS,CAAC;IAChF,OAAO,CAAC,SAAS,CAAC,CAAY;IAC9B,0EAA0E;IACnE,SAAS,CAAC,EAAE,MAAM,CAAC;IAC1B,kFAAkF;IAC3E,eAAe,CAAC,EAAE,eAAe,CAAC;IACzC,4DAA4D;IACrD,SAAS,EAAE,SAAS,CAAC;IAC5B,wDAAwD;IACjD,eAAe,CAAC,EAAE,MAAM,CAAC;IAChC,uFAAuF;IAChF,aAAa,EAAE,iBAAiB,CAAC;gBAE5B,OAAO,GAAE,YAAiB;IAoBtC;;;;;;;;;OASG;IACH,SAAS,CAAC,QAAQ,EAAE,CAAC,KAAK,EAAE,UAAU,EAAE,MAAM,EAAE,WAAW,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,GAAG,MAAM,IAAI;IAKjG;;;;OAIG;IACH,IAAI,KAAK,IAAI,UAAU,CAEtB;IAED,yDAAyD;IACzD,IAAI,YAAY,CAAC,IAAI,EAAE,SAAS,EAE/B;IAED,IAAI,YAAY,IAAI,SAAS,CAE5B;IAED,0DAA0D;IAC1D,IAAI,YAAY,CAAC,IAAI,EAAE,SAAS,EAE/B;IAED,IAAI,YAAY,IAAI,SAAS,CAE5B;IAED,gFAAgF;IAChF,KAAK,CAAC,OAAO,EAAE,YAAY,GAAG,IAAI;IAIlC,wEAAwE;IACxE,QAAQ,CAAC,OAAO,EAAE,YAAY,GAAG,IAAI;IAIrC,2CAA2C;IAC3C,kBAAkB,IAAI,IAAI;IAI1B,4CAA4C;IAC5C,kBAAkB,IAAI,IAAI;IAI1B,yDAAyD;IACzD,cAAc,IAAI,IAAI;IAKtB,sEAAsE;IACtE,iBAAiB,IAAI,OAAO;IAI5B,uDAAuD;IACvD,IAAI,MAAM,IAAI,WAAW,GAAG,SAAS,CAEpC;IAED,+CAA+C;IAC/C,KAAK,IAAI,IAAI;IAIb;;;;OAIG;IACH,WAAW,IAAI,OAAO,CAAC,IAAI,CAAC;IAI5B,kEAAkE;IAClE,KAAK,IAAI,IAAI;IAUb,8EAA8E;IACxE,MAAM,CAAC,OAAO,EAAE,YAAY,GAAG,YAAY,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAC7D,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,YAAY,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAWnE,oGAAoG;IAC9F,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;IA6B/B,OAAO,CAAC,oBAAoB;YAmBd,iBAAiB;YAgBjB,eAAe;IAY7B,OAAO,CAAC,qBAAqB;IAQ7B,OAAO,CAAC,gBAAgB;YA6BV,gBAAgB;YAyBhB,gBAAgB;IAkB9B,OAAO,CAAC,SAAS;IAQjB;;;;;;OAMG;YACW,aAAa;CAgD3B"}
+{"version":3,"file":"agent.d.ts","sourceRoot":"","sources":["../../../src/sdk/agent-core/agent.ts"],"names":[],"mappings":"AAAA,OAAO,EACN,KAAK,YAAY,EACjB,KAAK,OAAO,EAEZ,KAAK,mBAAmB,EAGxB,KAAK,SAAS,EACd,MAAM,gBAAgB,CAAC;AAGxB,OAAO,KAAK,EACX,oBAAoB,EACpB,mBAAmB,EAEnB,UAAU,EAEV,mBAAmB,EACnB,YAAY,EACZ,UAAU,EAEV,qBAAqB,EACrB,oBAAoB,EACpB,SAAS,EACT,QAAQ,EACR,iBAAiB,EACjB,MAAM,YAAY,CAAC;AAEpB,YAAY,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AAkE5C,iDAAiD;AACjD,MAAM,WAAW,YAAY;IAC5B,YAAY,CAAC,EAAE,OAAO,CAAC,IAAI,CAAC,UAAU,EAAE,kBAAkB,GAAG,aAAa,GAAG,kBAAkB,GAAG,cAAc,CAAC,CAAC,CAAC;IACnH,YAAY,CAAC,EAAE,CAAC,QAAQ,EAAE,YAAY,EAAE,KAAK,OAAO,EAAE,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;IAC5E,gBAAgB,CAAC,EAAE,CAAC,QAAQ,EAAE,YAAY,EAAE,EAAE,MAAM,CAAC,EAAE,WAAW,KAAK,OAAO,CAAC,YAAY,EAAE,CAAC,CAAC;IAC/F,QAAQ,CAAC,EAAE,QAAQ,CAAC;IACpB,SAAS,CAAC,EAAE,CAAC,QAAQ,EAAE,MAAM,KAAK,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC,GAAG,MAAM,GAAG,SAAS,CAAC;IACnF,SAAS,CAAC,EAAE,mBAAmB,CAAC,WAAW,CAAC,CAAC;IAC7C,UAAU,CAAC,EAAE,mBAAmB,CAAC,YAAY,CAAC,CAAC;IAC/C,cAAc,CAAC,EAAE,CAAC,OAAO,EAAE,qBAAqB,EAAE,MAAM,CAAC,EAAE,WAAW,KAAK,OAAO,CAAC,oBAAoB,GAAG,SAAS,CAAC,CAAC;IACrH,aAAa,CAAC,EAAE,CAAC,OAAO,EAAE,oBAAoB,EAAE,MAAM,CAAC,EAAE,WAAW,KAAK,OAAO,CAAC,mBAAmB,GAAG,SAAS,CAAC,CAAC;IAClH,eAAe,CAAC,EAAE,CACjB,MAAM,CAAC,EAAE,WAAW,KAChB,OAAO,CAAC,mBAAmB,GAAG,SAAS,CAAC,GAAG,mBAAmB,GAAG,SAAS,CAAC;IAChF,YAAY,CAAC,EAAE,SAAS,CAAC;IACzB,YAAY,CAAC,EAAE,SAAS,CAAC;IACzB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,SAAS,CAAC,EAAE,SAAS,CAAC;IACtB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,aAAa,CAAC,EAAE,iBAAiB,CAAC;CAClC;AA4CD;;;;;GAKG;AACH,qBAAa,KAAK;IACjB,OAAO,CAAC,MAAM,CAAoB;IAClC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAA+E;IACzG,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAsB;IACpD,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAsB;IACpD,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAA0B;IAEpD,YAAY,EAAE,CAAC,QAAQ,EAAE,YAAY,EAAE,KAAK,OAAO,EAAE,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;IAC3E,gBAAgB,CAAC,EAAE,CAAC,QAAQ,EAAE,YAAY,EAAE,EAAE,MAAM,CAAC,EAAE,WAAW,KAAK,OAAO,CAAC,YAAY,EAAE,CAAC,CAAC;IAC/F,QAAQ,EAAE,QAAQ,CAAC;IACnB,SAAS,CAAC,EAAE,CAAC,QAAQ,EAAE,MAAM,KAAK,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC,GAAG,MAAM,GAAG,SAAS,CAAC;IACnF,SAAS,CAAC,EAAE,mBAAmB,CAAC,WAAW,CAAC,CAAC;IAC7C,UAAU,CAAC,EAAE,mBAAmB,CAAC,YAAY,CAAC,CAAC;IAC/C,cAAc,CAAC,EAAE,CACvB,OAAO,EAAE,qBAAqB,EAC9B,MAAM,CAAC,EAAE,WAAW,KAChB,OAAO,CAAC,oBAAoB,GAAG,SAAS,CAAC,CAAC;IACxC,aAAa,CAAC,EAAE,CACtB,OAAO,EAAE,oBAAoB,EAC7B,MAAM,CAAC,EAAE,WAAW,KAChB,OAAO,CAAC,mBAAmB,GAAG,SAAS,CAAC,CAAC;IACvC,eAAe,CAAC,EAAE,CACxB,MAAM,CAAC,EAAE,WAAW,KAChB,OAAO,CAAC,mBAAmB,GAAG,SAAS,CAAC,GAAG,mBAAmB,GAAG,SAAS,CAAC;IAChF,OAAO,CAAC,SAAS,CAAC,CAAY;IAC9B,0EAA0E;IACnE,SAAS,CAAC,EAAE,MAAM,CAAC;IAC1B,4DAA4D;IACrD,SAAS,EAAE,SAAS,CAAC;IAC5B,wDAAwD;IACjD,eAAe,CAAC,EAAE,MAAM,CAAC;IAChC,uFAAuF;IAChF,aAAa,EAAE,iBAAiB,CAAC;gBAE5B,OAAO,GAAE,YAAiB;IAmBtC;;;;;;;;;OASG;IACH,SAAS,CAAC,QAAQ,EAAE,CAAC,KAAK,EAAE,UAAU,EAAE,MAAM,EAAE,WAAW,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,GAAG,MAAM,IAAI;IAKjG;;;;OAIG;IACH,IAAI,KAAK,IAAI,UAAU,CAEtB;IAED,yDAAyD;IACzD,IAAI,YAAY,CAAC,IAAI,EAAE,SAAS,EAE/B;IAED,IAAI,YAAY,IAAI,SAAS,CAE5B;IAED,0DAA0D;IAC1D,IAAI,YAAY,CAAC,IAAI,EAAE,SAAS,EAE/B;IAED,IAAI,YAAY,IAAI,SAAS,CAE5B;IAED,gFAAgF;IAChF,KAAK,CAAC,OAAO,EAAE,YAAY,GAAG,IAAI;IAIlC,wEAAwE;IACxE,QAAQ,CAAC,OAAO,EAAE,YAAY,GAAG,IAAI;IAIrC,2CAA2C;IAC3C,kBAAkB,IAAI,IAAI;IAI1B,4CAA4C;IAC5C,kBAAkB,IAAI,IAAI;IAI1B,yDAAyD;IACzD,cAAc,IAAI,IAAI;IAKtB,sEAAsE;IACtE,iBAAiB,IAAI,OAAO;IAI5B,uDAAuD;IACvD,IAAI,MAAM,IAAI,WAAW,GAAG,SAAS,CAEpC;IAED,+CAA+C;IAC/C,KAAK,IAAI,IAAI;IAIb;;;;;;;OAOG;IACG,YAAY,CAAC,UAAU,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAIzE;;;;OAIG;IACH,WAAW,IAAI,OAAO,CAAC,IAAI,CAAC;IAI5B,kEAAkE;IAClE,KAAK,IAAI,IAAI;IAWb,8EAA8E;IACxE,MAAM,CAAC,OAAO,EAAE,YAAY,GAAG,YAAY,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAC7D,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,YAAY,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAWnE,oGAAoG;IAC9F,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;IA6B/B,OAAO,CAAC,oBAAoB;YAmBd,iBAAiB;YAgBjB,eAAe;IAY7B,OAAO,CAAC,qBAAqB;IAQ7B,OAAO,CAAC,gBAAgB;YA6BV,gBAAgB;YA0BhB,gBAAgB;IAkB9B,OAAO,CAAC,SAAS;IAQjB;;;;;;OAMG;YACW,aAAa;CAgD3B"}

package/dist/sdk/agent-core/agent.js

@@ -1,5 +1,6 @@
 import { streamSimple, } from "../ai/index.js";
 import { runAgentLoop, runAgentLoopContinue } from "./agent-loop.js";
+import { ToolCallRegistry } from "./tool-call-registry.js";
 function defaultConvertToLlm(messages) {
     return messages.filter((message) => message.role === "user" || message.role === "assistant" || message.role === "toolResult");
 }
@@ -88,6 +89,7 @@ export class Agent {
     listeners = new Set();
     steeringQueue;
     followUpQueue;
+    toolCallRegistry = new ToolCallRegistry();
     convertToLlm;
     transformContext;
     streamFn;
@@ -100,8 +102,6 @@ export class Agent {
     activeRun;
     /** Session identifier forwarded to providers for cache-aware backends. */
     sessionId;
-    /** Optional per-level thinking token budgets forwarded to the stream function. */
-    thinkingBudgets;
     /** Preferred transport forwarded to the stream function. */
     transport;
     /** Optional cap for provider-requested retry delays. */
@@ -122,7 +122,6 @@ export class Agent {
         this.steeringQueue = new PendingMessageQueue(options.steeringMode ?? "one-at-a-time");
         this.followUpQueue = new PendingMessageQueue(options.followUpMode ?? "one-at-a-time");
         this.sessionId = options.sessionId;
-        this.thinkingBudgets = options.thinkingBudgets;
         this.transport = options.transport ?? "auto";
         this.maxRetryDelayMs = options.maxRetryDelayMs;
         this.toolExecution = options.toolExecution ?? "parallel";
@@ -196,6 +195,17 @@ export class Agent {
     abort() {
         this.activeRun?.abortController.abort();
     }
+    /**
+     * Stop a single in-flight tool call (user-initiated per-call stop).
+     *
+     * Aborts only the targeted call's execution — siblings keep running and the
+     * agent loop CONTINUES the turn with a synthesized user-stop tool result
+     * ("Tool call stopped by user (was running too long)"). Returns false for
+     * unknown/finished toolCallIds (no-op).
+     */
+    async stopToolCall(toolCallId, reason) {
+        return this.toolCallRegistry.stop(toolCallId, reason);
+    }
     /**
      * Resolve when the current run and all awaited event listeners have finished.
      *
@@ -211,6 +221,7 @@ export class Agent {
         this._state.streamingMessage = undefined;
         this._state.pendingToolCalls = new Set();
         this._state.errorMessage = undefined;
+        this.toolCallRegistry.clear();
         this.clearFollowUpQueue();
         this.clearSteeringQueue();
     }
@@ -284,7 +295,6 @@ export class Agent {
             onPayload: this.onPayload,
             onResponse: this.onResponse,
             transport: this.transport,
-            thinkingBudgets: this.thinkingBudgets,
             maxRetryDelayMs: this.maxRetryDelayMs,
             toolExecution: this.toolExecution,
             beforeToolCall: this.beforeToolCall,
@@ -301,6 +311,7 @@ export class Agent {
                 return this.steeringQueue.drain();
             },
             getFollowUpMessages: async () => this.followUpQueue.drain(),
+            toolCallRegistry: this.toolCallRegistry,
         };
     }
     async runWithLifecycle(executor) {
@@ -316,6 +327,7 @@ export class Agent {
         this._state.isStreaming = true;
         this._state.streamingMessage = undefined;
         this._state.errorMessage = undefined;
+        this.toolCallRegistry.clear();
         try {
             await executor(abortController.signal);
         }

package/dist/sdk/agent-core/tool-call-registry.d.ts

@@ -0,0 +1,75 @@
+/**
+ * Killable registry of in-flight tool executions for one agent run.
+ *
+ * The agent loop hands every `tool.execute()` a run-level abort signal, which
+ * makes whole-turn cancellation the only kill path. This registry adds a
+ * per-call controller chained from the run signal so a single hung/long-running
+ * tool call can be aborted on its own (`stopToolCall`) while its siblings keep
+ * running and the agent loop continues the turn.
+ *
+ * Lifecycle:
+ *  - `register()` is called by the tool executor right before execution and
+ *    returns a handle carrying the per-call abort signal.
+ *  - `settle()` is called by the executor when the call finishes (success or
+ *    error) — it removes the entry and resolves the handle's `settled` promise,
+ *    which is what `stop()` awaits.
+ *  - `stop()` marks a user-initiated stop (so the caught abort error is
+ *    replaced by the synthesized user-stop tool result), aborts only the
+ *    targeted call's controller, and resolves once the call settles.
+ *
+ * Chaining: when the run-level signal aborts (turn cancel), every per-call
+ * controller aborts too, so existing whole-turn cancel behavior is preserved.
+ */
+/** Default reason recorded on the handle when the caller passes none. */
+export declare const DEFAULT_USER_STOP_REASON = "stopped by user";
+/**
+ * Exactly-matching synthesized toolResult content per the `stop_tool_call`
+ * wire contract. The wire tool_result frame carries `stoppedByUser: true`.
+ */
+export declare const USER_STOP_TOOL_RESULT_TEXT = "Tool call stopped by user (was running too long)";
+/** Handle for one in-flight tool call execution. */
+export interface InFlightToolCallHandle {
+    readonly toolCallId: string;
+    readonly toolName: string;
+    /** Per-call abort signal source. `controller.signal` is what tools receive. */
+    readonly controller: AbortController;
+    /** Resolves when the executor settles this call (success or error). */
+    readonly settled: Promise<void>;
+    /** Set true when a user-initiated stop targets this call. */
+    userStopRequested: boolean;
+    /** Reason supplied by the caller of `stop()` (defaults to "stopped by user"). */
+    userStopReason?: string;
+    /** Resolve the `settled` promise. Called by the executor, not by callers. */
+    settle(): void;
+}
+/**
+ * Per-run killable registry of in-flight tool executions.
+ * Keyed by toolCallId; one Agent owns one registry per run.
+ */
+export declare class ToolCallRegistry {
+    private readonly calls;
+    /**
+     * Register an in-flight call and return its handle. The returned
+     * controller's signal aborts when this call is stopped individually OR
+     * when the run-level signal aborts (turn cancel propagates).
+     */
+    register(toolCallId: string, toolName: string, runSignal?: AbortSignal): InFlightToolCallHandle;
+    /** Look up an in-flight call by id. Undefined for unknown/finished ids. */
+    get(toolCallId: string): InFlightToolCallHandle | undefined;
+    /**
+     * Mark a user-initiated stop and abort only this call's controller.
+     * Returns false for unknown/finished ids (no-op). Idempotent: a second
+     * stop for the same id keeps the first marker and reason.
+     */
+    markUserStop(toolCallId: string, reason?: string): boolean;
+    /**
+     * Abort only the targeted call and resolve once it settles.
+     * Returns false for unknown/finished ids (no-op).
+     */
+    stop(toolCallId: string, reason?: string): Promise<boolean>;
+    /** Settle an in-flight call. Called by the executor when execution ends. */
+    settle(toolCallId: string): void;
+    /** Settle and drop every tracked call. Called when a run starts/ends. */
+    clear(): void;
+}
+//# sourceMappingURL=tool-call-registry.d.ts.map

package/dist/sdk/agent-core/tool-call-registry.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-call-registry.d.ts","sourceRoot":"","sources":["../../../src/sdk/agent-core/tool-call-registry.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;GAqBG;AAEH,yEAAyE;AACzE,eAAO,MAAM,wBAAwB,oBAAoB,CAAC;AAE1D;;;GAGG;AACH,eAAO,MAAM,0BAA0B,qDACY,CAAC;AAEpD,oDAAoD;AACpD,MAAM,WAAW,sBAAsB;IACtC,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC;IAC5B,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC;IAC1B,+EAA+E;IAC/E,QAAQ,CAAC,UAAU,EAAE,eAAe,CAAC;IACrC,uEAAuE;IACvE,QAAQ,CAAC,OAAO,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;IAChC,6DAA6D;IAC7D,iBAAiB,EAAE,OAAO,CAAC;IAC3B,iFAAiF;IACjF,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,6EAA6E;IAC7E,MAAM,IAAI,IAAI,CAAC;CACf;AAED;;;GAGG;AACH,qBAAa,gBAAgB;IAC5B,OAAO,CAAC,QAAQ,CAAC,KAAK,CAA6C;IAEnE;;;;OAIG;IACH,QAAQ,CACP,UAAU,EAAE,MAAM,EAClB,QAAQ,EAAE,MAAM,EAChB,SAAS,CAAC,EAAE,WAAW,GACrB,sBAAsB;IA+CzB,2EAA2E;IAC3E,GAAG,CAAC,UAAU,EAAE,MAAM,GAAG,sBAAsB,GAAG,SAAS;IAI3D;;;;OAIG;IACH,YAAY,CAAC,UAAU,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,OAAO;IAe1D;;;OAGG;IACG,IAAI,CAAC,UAAU,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAMjE,4EAA4E;IAC5E,MAAM,CAAC,UAAU,EAAE,MAAM,GAAG,IAAI;IAIhC,yEAAyE;IACzE,KAAK,IAAI,IAAI;CAIb"}