@anthropic-field/core 0.1.0

package/dist/src/scheduler/state-manager.js

@@ -0,0 +1,356 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { ROOT_SCHEDULER_ID } from './types.js';
+import { MessageBusType, } from '../confirmation-bus/types.js';
+/**
+ * Manages the state of tool calls.
+ * Publishes state changes to the MessageBus via TOOL_CALLS_UPDATE events.
+ */
+export class SchedulerStateManager {
+    messageBus;
+    schedulerId;
+    onTerminalCall;
+    activeCalls = new Map();
+    queue = [];
+    _completedBatch = [];
+    constructor(messageBus, schedulerId = ROOT_SCHEDULER_ID, onTerminalCall) {
+        this.messageBus = messageBus;
+        this.schedulerId = schedulerId;
+        this.onTerminalCall = onTerminalCall;
+    }
+    addToolCalls(calls) {
+        this.enqueue(calls);
+    }
+    getToolCall(callId) {
+        return (this.activeCalls.get(callId) ||
+            this.queue.find((c) => c.request.callId === callId) ||
+            this._completedBatch.find((c) => c.request.callId === callId));
+    }
+    enqueue(calls) {
+        this.queue.push(...calls);
+        this.emitUpdate();
+    }
+    dequeue() {
+        const next = this.queue.shift();
+        if (next) {
+            this.activeCalls.set(next.request.callId, next);
+            this.emitUpdate();
+        }
+        return next;
+    }
+    get isActive() {
+        return this.activeCalls.size > 0;
+    }
+    get activeCallCount() {
+        return this.activeCalls.size;
+    }
+    get queueLength() {
+        return this.queue.length;
+    }
+    get firstActiveCall() {
+        return this.activeCalls.values().next().value;
+    }
+    updateStatus(callId, status, auxiliaryData) {
+        const call = this.activeCalls.get(callId);
+        if (!call)
+            return;
+        const updatedCall = this.transitionCall(call, status, auxiliaryData);
+        this.activeCalls.set(callId, updatedCall);
+        this.emitUpdate();
+    }
+    finalizeCall(callId) {
+        const call = this.activeCalls.get(callId);
+        if (!call)
+            return;
+        if (this.isTerminalCall(call)) {
+            this._completedBatch.push(call);
+            this.activeCalls.delete(callId);
+            this.onTerminalCall?.(call);
+            this.emitUpdate();
+        }
+    }
+    updateArgs(callId, newArgs, newInvocation) {
+        const call = this.activeCalls.get(callId);
+        if (!call || call.status === 'error')
+            return;
+        this.activeCalls.set(callId, this.patchCall(call, {
+            request: { ...call.request, args: newArgs },
+            invocation: newInvocation,
+        }));
+        this.emitUpdate();
+    }
+    setOutcome(callId, outcome) {
+        const call = this.activeCalls.get(callId);
+        if (!call)
+            return;
+        this.activeCalls.set(callId, this.patchCall(call, { outcome }));
+        this.emitUpdate();
+    }
+    cancelAllQueued(reason) {
+        if (this.queue.length === 0) {
+            return;
+        }
+        while (this.queue.length > 0) {
+            const queuedCall = this.queue.shift();
+            if (queuedCall.status === 'error') {
+                this._completedBatch.push(queuedCall);
+                this.onTerminalCall?.(queuedCall);
+                continue;
+            }
+            const cancelledCall = this.toCancelled(queuedCall, reason);
+            this._completedBatch.push(cancelledCall);
+            this.onTerminalCall?.(cancelledCall);
+        }
+        this.emitUpdate();
+    }
+    getSnapshot() {
+        return [
+            ...this._completedBatch,
+            ...Array.from(this.activeCalls.values()),
+            ...this.queue,
+        ];
+    }
+    clearBatch() {
+        if (this._completedBatch.length === 0)
+            return;
+        this._completedBatch = [];
+        this.emitUpdate();
+    }
+    get completedBatch() {
+        return [...this._completedBatch];
+    }
+    emitUpdate() {
+        const snapshot = this.getSnapshot();
+        // Fire and forget - The message bus handles the publish and error handling.
+        void this.messageBus.publish({
+            type: MessageBusType.TOOL_CALLS_UPDATE,
+            toolCalls: snapshot,
+            schedulerId: this.schedulerId,
+        });
+    }
+    isTerminalCall(call) {
+        const { status } = call;
+        return status === 'success' || status === 'error' || status === 'cancelled';
+    }
+    transitionCall(call, newStatus, auxiliaryData) {
+        switch (newStatus) {
+            case 'success': {
+                if (!this.isToolCallResponseInfo(auxiliaryData)) {
+                    throw new Error(`Invalid data for 'success' transition (callId: ${call.request.callId})`);
+                }
+                return this.toSuccess(call, auxiliaryData);
+            }
+            case 'error': {
+                if (!this.isToolCallResponseInfo(auxiliaryData)) {
+                    throw new Error(`Invalid data for 'error' transition (callId: ${call.request.callId})`);
+                }
+                return this.toError(call, auxiliaryData);
+            }
+            case 'awaiting_approval': {
+                if (!auxiliaryData) {
+                    throw new Error(`Missing data for 'awaiting_approval' transition (callId: ${call.request.callId})`);
+                }
+                return this.toAwaitingApproval(call, auxiliaryData);
+            }
+            case 'scheduled':
+                return this.toScheduled(call);
+            case 'cancelled': {
+                if (typeof auxiliaryData !== 'string') {
+                    throw new Error(`Invalid reason (string) for 'cancelled' transition (callId: ${call.request.callId})`);
+                }
+                return this.toCancelled(call, auxiliaryData);
+            }
+            case 'validating':
+                return this.toValidating(call);
+            case 'executing': {
+                if (auxiliaryData !== undefined &&
+                    !this.isExecutingToolCallPatch(auxiliaryData)) {
+                    throw new Error(`Invalid patch for 'executing' transition (callId: ${call.request.callId})`);
+                }
+                return this.toExecuting(call, auxiliaryData);
+            }
+            default: {
+                const exhaustiveCheck = newStatus;
+                return exhaustiveCheck;
+            }
+        }
+    }
+    isToolCallResponseInfo(data) {
+        return (typeof data === 'object' &&
+            data !== null &&
+            'callId' in data &&
+            'responseParts' in data);
+    }
+    isExecutingToolCallPatch(data) {
+        // A partial can be an empty object, but it must be a non-null object.
+        return typeof data === 'object' && data !== null;
+    }
+    // --- Transition Helpers ---
+    /**
+     * Ensures the tool call has an associated tool and invocation before
+     * transitioning to states that require them.
+     */
+    validateHasToolAndInvocation(call, targetStatus) {
+        if (!('tool' in call && call.tool && 'invocation' in call && call.invocation)) {
+            throw new Error(`Invalid state transition: cannot transition to ${targetStatus} without tool/invocation (callId: ${call.request.callId})`);
+        }
+    }
+    toSuccess(call, response) {
+        this.validateHasToolAndInvocation(call, 'success');
+        const startTime = 'startTime' in call ? call.startTime : undefined;
+        return {
+            request: call.request,
+            tool: call.tool,
+            invocation: call.invocation,
+            status: 'success',
+            response,
+            durationMs: startTime ? Date.now() - startTime : undefined,
+            outcome: call.outcome,
+            schedulerId: call.schedulerId,
+        };
+    }
+    toError(call, response) {
+        const startTime = 'startTime' in call ? call.startTime : undefined;
+        return {
+            request: call.request,
+            status: 'error',
+            tool: 'tool' in call ? call.tool : undefined,
+            response,
+            durationMs: startTime ? Date.now() - startTime : undefined,
+            outcome: call.outcome,
+            schedulerId: call.schedulerId,
+        };
+    }
+    toAwaitingApproval(call, data) {
+        this.validateHasToolAndInvocation(call, 'awaiting_approval');
+        let confirmationDetails;
+        let correlationId;
+        if (this.isEventDrivenApprovalData(data)) {
+            correlationId = data.correlationId;
+            confirmationDetails = data.confirmationDetails;
+        }
+        else {
+            // TODO: Remove legacy callback shape once event-driven migration is complete
+            confirmationDetails = data;
+        }
+        return {
+            request: call.request,
+            tool: call.tool,
+            status: 'awaiting_approval',
+            correlationId,
+            confirmationDetails,
+            startTime: 'startTime' in call ? call.startTime : undefined,
+            outcome: call.outcome,
+            invocation: call.invocation,
+            schedulerId: call.schedulerId,
+        };
+    }
+    isEventDrivenApprovalData(data) {
+        return (typeof data === 'object' &&
+            data !== null &&
+            'correlationId' in data &&
+            'confirmationDetails' in data);
+    }
+    toScheduled(call) {
+        this.validateHasToolAndInvocation(call, 'scheduled');
+        return {
+            request: call.request,
+            tool: call.tool,
+            status: 'scheduled',
+            startTime: 'startTime' in call ? call.startTime : undefined,
+            outcome: call.outcome,
+            invocation: call.invocation,
+            schedulerId: call.schedulerId,
+        };
+    }
+    toCancelled(call, reason) {
+        this.validateHasToolAndInvocation(call, 'cancelled');
+        const startTime = 'startTime' in call ? call.startTime : undefined;
+        // TODO: Refactor this tool-specific logic into the confirmation details payload.
+        // See: https://github.com/google-gemini/gemini-cli/issues/16716
+        let resultDisplay = undefined;
+        if (this.isWaitingToolCall(call)) {
+            const details = call.confirmationDetails;
+            if (details.type === 'edit' &&
+                'fileDiff' in details &&
+                'fileName' in details &&
+                'filePath' in details &&
+                'originalContent' in details &&
+                'newContent' in details) {
+                resultDisplay = {
+                    fileDiff: details.fileDiff,
+                    fileName: details.fileName,
+                    filePath: details.filePath,
+                    originalContent: details.originalContent,
+                    newContent: details.newContent,
+                };
+            }
+        }
+        const errorMessage = `[Operation Cancelled] Reason: ${reason}`;
+        return {
+            request: call.request,
+            tool: call.tool,
+            invocation: call.invocation,
+            status: 'cancelled',
+            response: {
+                callId: call.request.callId,
+                responseParts: [
+                    {
+                        functionResponse: {
+                            id: call.request.callId,
+                            name: call.request.name,
+                            response: { error: errorMessage },
+                        },
+                    },
+                ],
+                resultDisplay,
+                error: undefined,
+                errorType: undefined,
+                contentLength: errorMessage.length,
+            },
+            durationMs: startTime ? Date.now() - startTime : undefined,
+            outcome: call.outcome,
+            schedulerId: call.schedulerId,
+        };
+    }
+    isWaitingToolCall(call) {
+        return call.status === 'awaiting_approval';
+    }
+    patchCall(call, patch) {
+        return { ...call, ...patch };
+    }
+    toValidating(call) {
+        this.validateHasToolAndInvocation(call, 'validating');
+        return {
+            request: call.request,
+            tool: call.tool,
+            status: 'validating',
+            startTime: 'startTime' in call ? call.startTime : undefined,
+            outcome: call.outcome,
+            invocation: call.invocation,
+            schedulerId: call.schedulerId,
+        };
+    }
+    toExecuting(call, data) {
+        this.validateHasToolAndInvocation(call, 'executing');
+        const execData = data;
+        const liveOutput = execData?.liveOutput ??
+            ('liveOutput' in call ? call.liveOutput : undefined);
+        const pid = execData?.pid ?? ('pid' in call ? call.pid : undefined);
+        return {
+            request: call.request,
+            tool: call.tool,
+            status: 'executing',
+            startTime: 'startTime' in call ? call.startTime : undefined,
+            outcome: call.outcome,
+            invocation: call.invocation,
+            liveOutput,
+            pid,
+            schedulerId: call.schedulerId,
+        };
+    }
+}
+//# sourceMappingURL=state-manager.js.map

package/dist/src/scheduler/state-manager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"state-manager.js","sourceRoot":"","sources":["../../../src/scheduler/state-manager.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAeH,OAAO,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AAS/C,OAAO,EACL,cAAc,GAEf,MAAM,8BAA8B,CAAC;AAOtC;;;GAGG;AACH,MAAM,OAAO,qBAAqB;IAMb;IACA;IACA;IAPF,WAAW,GAAG,IAAI,GAAG,EAAoB,CAAC;IAC1C,KAAK,GAAe,EAAE,CAAC;IAChC,eAAe,GAAwB,EAAE,CAAC;IAElD,YACmB,UAAsB,EACtB,cAAsB,iBAAiB,EACvC,cAAoC;QAFpC,eAAU,GAAV,UAAU,CAAY;QACtB,gBAAW,GAAX,WAAW,CAA4B;QACvC,mBAAc,GAAd,cAAc,CAAsB;IACpD,CAAC;IAEJ,YAAY,CAAC,KAAiB;QAC5B,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IACtB,CAAC;IAED,WAAW,CAAC,MAAc;QACxB,OAAO,CACL,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,MAAM,CAAC;YAC5B,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,KAAK,MAAM,CAAC;YACnD,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,KAAK,MAAM,CAAC,CAC9D,CAAC;IACJ,CAAC;IAED,OAAO,CAAC,KAAiB;QACvB,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,CAAC;QAC1B,IAAI,CAAC,UAAU,EAAE,CAAC;IACpB,CAAC;IAED,OAAO;QACL,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;QAChC,IAAI,IAAI,EAAE,CAAC;YACT,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;YAChD,IAAI,CAAC,UAAU,EAAE,CAAC;QACpB,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,IAAI,QAAQ;QACV,OAAO,IAAI,CAAC,WAAW,CAAC,IAAI,GAAG,CAAC,CAAC;IACnC,CAAC;IAED,IAAI,eAAe;QACjB,OAAO,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC;IAC/B,CAAC;IAED,IAAI,WAAW;QACb,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC;IAC3B,CAAC;IAED,IAAI,eAAe;QACjB,OAAO,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC;IAChD,CAAC;IAgCD,YAAY,CAAC,MAAc,EAAE,MAAc,EAAE,aAAuB;QAClE,MAAM,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAC1C,IAAI,CAAC,IAAI;YAAE,OAAO;QAElB,MAAM,WAAW,GAAG,IAAI,CAAC,cAAc,CAAC,IAAI,EAAE,MAAM,EAAE,aAAa,CAAC,CAAC;QACrE,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;QAE1C,IAAI,CAAC,UAAU,EAAE,CAAC;IACpB,CAAC;IAED,YAAY,CAAC,MAAc;QACzB,MAAM,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAC1C,IAAI,CAAC,IAAI;YAAE,OAAO;QAElB,IAAI,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,EAAE,CAAC;YAC9B,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAChC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YAEhC,IAAI,CAAC,cAAc,EAAE,CAAC,IAAI,CAAC,CAAC;YAC5B,IAAI,CAAC,UAAU,EAAE,CAAC;QACpB,CAAC;IACH,CAAC;IAED,UAAU,CACR,MAAc,EACd,OAAgC,EAChC,aAAgC;QAEhC,MAAM,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAC1C,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,MAAM,KAAK,OAAO;YAAE,OAAO;QAE7C,IAAI,CAAC,WAAW,CAAC,GAAG,CAClB,MAAM,EACN,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE;YACnB,OAAO,EAAE,EAAE,GAAG,IAAI,CAAC,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE;YAC3C,UAAU,EAAE,aAAa;SAC1B,CAAC,CACH,CAAC;QACF,IAAI,CAAC,UAAU,EAAE,CAAC;IACpB,CAAC;IAED,UAAU,CAAC,MAAc,EAAE,OAAgC;QACzD,MAAM,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAC1C,IAAI,CAAC,IAAI;YAAE,OAAO;QAElB,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC;QAChE,IAAI,CAAC,UAAU,EAAE,CAAC;IACpB,CAAC;IAED,eAAe,CAAC,MAAc;QAC5B,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC5B,OAAO;QACT,CAAC;QAED,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC7B,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,EAAG,CAAC;YACvC,IAAI,UAAU,CAAC,MAAM,KAAK,OAAO,EAAE,CAAC;gBAClC,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;gBACtC,IAAI,CAAC,cAAc,EAAE,CAAC,UAAU,CAAC,CAAC;gBAClC,SAAS;YACX,CAAC;YACD,MAAM,aAAa,GAAG,IAAI,CAAC,WAAW,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;YAC3D,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YACzC,IAAI,CAAC,cAAc,EAAE,CAAC,aAAa,CAAC,CAAC;QACvC,CAAC;QACD,IAAI,CAAC,UAAU,EAAE,CAAC;IACpB,CAAC;IAED,WAAW;QACT,OAAO;YACL,GAAG,IAAI,CAAC,eAAe;YACvB,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC;YACxC,GAAG,IAAI,CAAC,KAAK;SACd,CAAC;IACJ,CAAC;IAED,UAAU;QACR,IAAI,IAAI,CAAC,eAAe,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QAC9C,IAAI,CAAC,eAAe,GAAG,EAAE,CAAC;QAC1B,IAAI,CAAC,UAAU,EAAE,CAAC;IACpB,CAAC;IAED,IAAI,cAAc;QAChB,OAAO,CAAC,GAAG,IAAI,CAAC,eAAe,CAAC,CAAC;IACnC,CAAC;IAEO,UAAU;QAChB,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;QAEpC,4EAA4E;QAC5E,KAAK,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC;YAC3B,IAAI,EAAE,cAAc,CAAC,iBAAiB;YACtC,SAAS,EAAE,QAAQ;YACnB,WAAW,EAAE,IAAI,CAAC,WAAW;SAC9B,CAAC,CAAC;IACL,CAAC;IAEO,cAAc,CAAC,IAAc;QACnC,MAAM,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC;QACxB,OAAO,MAAM,KAAK,SAAS,IAAI,MAAM,KAAK,OAAO,IAAI,MAAM,KAAK,WAAW,CAAC;IAC9E,CAAC;IAEO,cAAc,CACpB,IAAc,EACd,SAAiB,EACjB,aAAuB;QAEvB,QAAQ,SAAS,EAAE,CAAC;YAClB,KAAK,SAAS,CAAC,CAAC,CAAC;gBACf,IAAI,CAAC,IAAI,CAAC,sBAAsB,CAAC,aAAa,CAAC,EAAE,CAAC;oBAChD,MAAM,IAAI,KAAK,CACb,kDAAkD,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CACzE,CAAC;gBACJ,CAAC;gBACD,OAAO,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,aAAa,CAAC,CAAC;YAC7C,CAAC;YACD,KAAK,OAAO,CAAC,CAAC,CAAC;gBACb,IAAI,CAAC,IAAI,CAAC,sBAAsB,CAAC,aAAa,CAAC,EAAE,CAAC;oBAChD,MAAM,IAAI,KAAK,CACb,gDAAgD,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CACvE,CAAC;gBACJ,CAAC;gBACD,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,aAAa,CAAC,CAAC;YAC3C,CAAC;YACD,KAAK,mBAAmB,CAAC,CAAC,CAAC;gBACzB,IAAI,CAAC,aAAa,EAAE,CAAC;oBACnB,MAAM,IAAI,KAAK,CACb,4DAA4D,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CACnF,CAAC;gBACJ,CAAC;gBACD,OAAO,IAAI,CAAC,kBAAkB,CAAC,IAAI,EAAE,aAAa,CAAC,CAAC;YACtD,CAAC;YACD,KAAK,WAAW;gBACd,OAAO,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;YAChC,KAAK,WAAW,CAAC,CAAC,CAAC;gBACjB,IAAI,OAAO,aAAa,KAAK,QAAQ,EAAE,CAAC;oBACtC,MAAM,IAAI,KAAK,CACb,+DAA+D,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CACtF,CAAC;gBACJ,CAAC;gBACD,OAAO,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,aAAa,CAAC,CAAC;YAC/C,CAAC;YACD,KAAK,YAAY;gBACf,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;YACjC,KAAK,WAAW,CAAC,CAAC,CAAC;gBACjB,IACE,aAAa,KAAK,SAAS;oBAC3B,CAAC,IAAI,CAAC,wBAAwB,CAAC,aAAa,CAAC,EAC7C,CAAC;oBACD,MAAM,IAAI,KAAK,CACb,qDAAqD,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAC5E,CAAC;gBACJ,CAAC;gBACD,OAAO,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,aAAa,CAAC,CAAC;YAC/C,CAAC;YACD,OAAO,CAAC,CAAC,CAAC;gBACR,MAAM,eAAe,GAAU,SAAS,CAAC;gBACzC,OAAO,eAAe,CAAC;YACzB,CAAC;QACH,CAAC;IACH,CAAC;IAEO,sBAAsB,CAAC,IAAa;QAC1C,OAAO,CACL,OAAO,IAAI,KAAK,QAAQ;YACxB,IAAI,KAAK,IAAI;YACb,QAAQ,IAAI,IAAI;YAChB,eAAe,IAAI,IAAI,CACxB,CAAC;IACJ,CAAC;IAEO,wBAAwB,CAC9B,IAAa;QAEb,sEAAsE;QACtE,OAAO,OAAO,IAAI,KAAK,QAAQ,IAAI,IAAI,KAAK,IAAI,CAAC;IACnD,CAAC;IAED,6BAA6B;IAE7B;;;OAGG;IACK,4BAA4B,CAClC,IAAc,EACd,YAAoB;QAKpB,IACE,CAAC,CAAC,MAAM,IAAI,IAAI,IAAI,IAAI,CAAC,IAAI,IAAI,YAAY,IAAI,IAAI,IAAI,IAAI,CAAC,UAAU,CAAC,EACzE,CAAC;YACD,MAAM,IAAI,KAAK,CACb,kDAAkD,YAAY,qCAAqC,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAC1H,CAAC;QACJ,CAAC;IACH,CAAC;IAEO,SAAS,CACf,IAAc,EACd,QAA8B;QAE9B,IAAI,CAAC,4BAA4B,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;QACnD,MAAM,SAAS,GAAG,WAAW,IAAI,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC;QACnE,OAAO;YACL,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,UAAU,EAAE,IAAI,CAAC,UAAU;YAC3B,MAAM,EAAE,SAAS;YACjB,QAAQ;YACR,UAAU,EAAE,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,CAAC,CAAC,SAAS;YAC1D,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,WAAW,EAAE,IAAI,CAAC,WAAW;SAC9B,CAAC;IACJ,CAAC;IAEO,OAAO,CACb,IAAc,EACd,QAA8B;QAE9B,MAAM,SAAS,GAAG,WAAW,IAAI,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC;QACnE,OAAO;YACL,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,MAAM,EAAE,OAAO;YACf,IAAI,EAAE,MAAM,IAAI,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS;YAC5C,QAAQ;YACR,UAAU,EAAE,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,CAAC,CAAC,SAAS;YAC1D,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,WAAW,EAAE,IAAI,CAAC,WAAW;SAC9B,CAAC;IACJ,CAAC;IAEO,kBAAkB,CAAC,IAAc,EAAE,IAAa;QACtD,IAAI,CAAC,4BAA4B,CAAC,IAAI,EAAE,mBAAmB,CAAC,CAAC;QAE7D,IAAI,mBAE+B,CAAC;QACpC,IAAI,aAAiC,CAAC;QAEtC,IAAI,IAAI,CAAC,yBAAyB,CAAC,IAAI,CAAC,EAAE,CAAC;YACzC,aAAa,GAAG,IAAI,CAAC,aAAa,CAAC;YACnC,mBAAmB,GAAG,IAAI,CAAC,mBAAmB,CAAC;QACjD,CAAC;aAAM,CAAC;YACN,6EAA6E;YAC7E,mBAAmB,GAAG,IAAmC,CAAC;QAC5D,CAAC;QAED,OAAO;YACL,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,MAAM,EAAE,mBAAmB;YAC3B,aAAa;YACb,mBAAmB;YACnB,SAAS,EAAE,WAAW,IAAI,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;YAC3D,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,UAAU,EAAE,IAAI,CAAC,UAAU;YAC3B,WAAW,EAAE,IAAI,CAAC,WAAW;SAC9B,CAAC;IACJ,CAAC;IAEO,yBAAyB,CAAC,IAAa;QAI7C,OAAO,CACL,OAAO,IAAI,KAAK,QAAQ;YACxB,IAAI,KAAK,IAAI;YACb,eAAe,IAAI,IAAI;YACvB,qBAAqB,IAAI,IAAI,CAC9B,CAAC;IACJ,CAAC;IAEO,WAAW,CAAC,IAAc;QAChC,IAAI,CAAC,4BAA4B,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;QACrD,OAAO;YACL,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,MAAM,EAAE,WAAW;YACnB,SAAS,EAAE,WAAW,IAAI,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;YAC3D,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,UAAU,EAAE,IAAI,CAAC,UAAU;YAC3B,WAAW,EAAE,IAAI,CAAC,WAAW;SAC9B,CAAC;IACJ,CAAC;IAEO,WAAW,CAAC,IAAc,EAAE,MAAc;QAChD,IAAI,CAAC,4BAA4B,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;QACrD,MAAM,SAAS,GAAG,WAAW,IAAI,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC;QAEnE,iFAAiF;QACjF,gEAAgE;QAChE,IAAI,aAAa,GAAkC,SAAS,CAAC;QAC7D,IAAI,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,EAAE,CAAC;YACjC,MAAM,OAAO,GAAG,IAAI,CAAC,mBAAmB,CAAC;YACzC,IACE,OAAO,CAAC,IAAI,KAAK,MAAM;gBACvB,UAAU,IAAI,OAAO;gBACrB,UAAU,IAAI,OAAO;gBACrB,UAAU,IAAI,OAAO;gBACrB,iBAAiB,IAAI,OAAO;gBAC5B,YAAY,IAAI,OAAO,EACvB,CAAC;gBACD,aAAa,GAAG;oBACd,QAAQ,EAAE,OAAO,CAAC,QAAQ;oBAC1B,QAAQ,EAAE,OAAO,CAAC,QAAQ;oBAC1B,QAAQ,EAAE,OAAO,CAAC,QAAQ;oBAC1B,eAAe,EAAE,OAAO,CAAC,eAAe;oBACxC,UAAU,EAAE,OAAO,CAAC,UAAU;iBAC/B,CAAC;YACJ,CAAC;QACH,CAAC;QAED,MAAM,YAAY,GAAG,iCAAiC,MAAM,EAAE,CAAC;QAC/D,OAAO;YACL,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,UAAU,EAAE,IAAI,CAAC,UAAU;YAC3B,MAAM,EAAE,WAAW;YACnB,QAAQ,EAAE;gBACR,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM;gBAC3B,aAAa,EAAE;oBACb;wBACE,gBAAgB,EAAE;4BAChB,EAAE,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM;4BACvB,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,IAAI;4BACvB,QAAQ,EAAE,EAAE,KAAK,EAAE,YAAY,EAAE;yBAClC;qBACF;iBACF;gBACD,aAAa;gBACb,KAAK,EAAE,SAAS;gBAChB,SAAS,EAAE,SAAS;gBACpB,aAAa,EAAE,YAAY,CAAC,MAAM;aACnC;YACD,UAAU,EAAE,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,CAAC,CAAC,SAAS;YAC1D,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,WAAW,EAAE,IAAI,CAAC,WAAW;SAC9B,CAAC;IACJ,CAAC;IAEO,iBAAiB,CAAC,IAAc;QACtC,OAAO,IAAI,CAAC,MAAM,KAAK,mBAAmB,CAAC;IAC7C,CAAC;IAEO,SAAS,CAAqB,IAAO,EAAE,KAAiB;QAC9D,OAAO,EAAE,GAAG,IAAI,EAAE,GAAG,KAAK,EAAE,CAAC;IAC/B,CAAC;IAEO,YAAY,CAAC,IAAc;QACjC,IAAI,CAAC,4BAA4B,CAAC,IAAI,EAAE,YAAY,CAAC,CAAC;QACtD,OAAO;YACL,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,MAAM,EAAE,YAAY;YACpB,SAAS,EAAE,WAAW,IAAI,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;YAC3D,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,UAAU,EAAE,IAAI,CAAC,UAAU;YAC3B,WAAW,EAAE,IAAI,CAAC,WAAW;SAC9B,CAAC;IACJ,CAAC;IAEO,WAAW,CAAC,IAAc,EAAE,IAAc;QAChD,IAAI,CAAC,4BAA4B,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;QACrD,MAAM,QAAQ,GAAG,IAA8C,CAAC;QAChE,MAAM,UAAU,GACd,QAAQ,EAAE,UAAU;YACpB,CAAC,YAAY,IAAI,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;QACvD,MAAM,GAAG,GAAG,QAAQ,EAAE,GAAG,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;QAEpE,OAAO;YACL,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,MAAM,EAAE,WAAW;YACnB,SAAS,EAAE,WAAW,IAAI,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;YAC3D,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,UAAU,EAAE,IAAI,CAAC,UAAU;YAC3B,UAAU;YACV,GAAG;YACH,WAAW,EAAE,IAAI,CAAC,WAAW;SAC9B,CAAC;IACJ,CAAC;CACF"}

package/dist/src/scheduler/tool-executor.d.ts

@@ -0,0 +1,22 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { Config, AnsiOutput } from '../index.js';
+import type { CompletedToolCall, ToolCall } from './types.js';
+export interface ToolExecutionContext {
+    call: ToolCall;
+    signal: AbortSignal;
+    outputUpdateHandler?: (callId: string, output: string | AnsiOutput) => void;
+    onUpdateToolCall: (updatedCall: ToolCall) => void;
+}
+export declare class ToolExecutor {
+    private readonly config;
+    constructor(config: Config);
+    execute(context: ToolExecutionContext): Promise<CompletedToolCall>;
+    private createCancelledResult;
+    private createSuccessResult;
+    private createErrorResult;
+    private createErrorResponse;
+}

package/dist/src/scheduler/tool-executor.js

@@ -0,0 +1,201 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { ToolErrorType, ToolOutputTruncatedEvent, logToolOutputTruncated, runInDevTraceSpan, } from '../index.js';
+import { SHELL_TOOL_NAME } from '../tools/tool-names.js';
+import { ShellToolInvocation } from '../tools/shell.js';
+import { executeToolWithHooks } from '../core/coreToolHookTriggers.js';
+import { saveTruncatedToolOutput, formatTruncatedToolOutput, } from '../utils/fileUtils.js';
+import { convertToFunctionResponse } from '../utils/generateContentResponseUtilities.js';
+export class ToolExecutor {
+    config;
+    constructor(config) {
+        this.config = config;
+    }
+    async execute(context) {
+        const { call, signal, outputUpdateHandler, onUpdateToolCall } = context;
+        const { request } = call;
+        const toolName = request.name;
+        const callId = request.callId;
+        if (!('tool' in call) || !call.tool || !('invocation' in call)) {
+            throw new Error(`Cannot execute tool call ${callId}: Tool or Invocation missing.`);
+        }
+        const { tool, invocation } = call;
+        // Setup live output handling
+        const liveOutputCallback = tool.canUpdateOutput && outputUpdateHandler
+            ? (outputChunk) => {
+                outputUpdateHandler(callId, outputChunk);
+            }
+            : undefined;
+        const shellExecutionConfig = this.config.getShellExecutionConfig();
+        return runInDevTraceSpan({
+            name: tool.name,
+            attributes: { type: 'tool-call' },
+        }, async ({ metadata: spanMetadata }) => {
+            spanMetadata.input = { request };
+            try {
+                let promise;
+                if (invocation instanceof ShellToolInvocation) {
+                    const setPidCallback = (pid) => {
+                        const executingCall = {
+                            ...call,
+                            status: 'executing',
+                            tool,
+                            invocation,
+                            pid,
+                            startTime: 'startTime' in call ? call.startTime : undefined,
+                        };
+                        onUpdateToolCall(executingCall);
+                    };
+                    promise = executeToolWithHooks(invocation, toolName, signal, tool, liveOutputCallback, shellExecutionConfig, setPidCallback, this.config);
+                }
+                else {
+                    promise = executeToolWithHooks(invocation, toolName, signal, tool, liveOutputCallback, shellExecutionConfig, undefined, this.config);
+                }
+                const toolResult = await promise;
+                spanMetadata.output = toolResult;
+                if (signal.aborted) {
+                    return this.createCancelledResult(call, 'User cancelled tool execution.');
+                }
+                else if (toolResult.error === undefined) {
+                    return await this.createSuccessResult(call, toolResult);
+                }
+                else {
+                    const displayText = typeof toolResult.returnDisplay === 'string'
+                        ? toolResult.returnDisplay
+                        : undefined;
+                    return this.createErrorResult(call, new Error(toolResult.error.message), toolResult.error.type, displayText);
+                }
+            }
+            catch (executionError) {
+                spanMetadata.error = executionError;
+                if (signal.aborted) {
+                    return this.createCancelledResult(call, 'User cancelled tool execution.');
+                }
+                const error = executionError instanceof Error
+                    ? executionError
+                    : new Error(String(executionError));
+                return this.createErrorResult(call, error, ToolErrorType.UNHANDLED_EXCEPTION);
+            }
+        });
+    }
+    createCancelledResult(call, reason) {
+        const errorMessage = `[Operation Cancelled] ${reason}`;
+        const startTime = 'startTime' in call ? call.startTime : undefined;
+        if (!('tool' in call) || !('invocation' in call)) {
+            // This should effectively never happen in execution phase, but we handle
+            // it safely
+            throw new Error('Cancelled tool call missing tool/invocation references');
+        }
+        return {
+            status: 'cancelled',
+            request: call.request,
+            response: {
+                callId: call.request.callId,
+                responseParts: [
+                    {
+                        functionResponse: {
+                            id: call.request.callId,
+                            name: call.request.name,
+                            response: { error: errorMessage },
+                        },
+                    },
+                ],
+                resultDisplay: undefined,
+                error: undefined,
+                errorType: undefined,
+                contentLength: errorMessage.length,
+            },
+            tool: call.tool,
+            invocation: call.invocation,
+            durationMs: startTime ? Date.now() - startTime : undefined,
+            outcome: call.outcome,
+        };
+    }
+    async createSuccessResult(call, toolResult) {
+        let content = toolResult.llmContent;
+        let outputFile;
+        const toolName = call.request.name;
+        const callId = call.request.callId;
+        if (typeof content === 'string' &&
+            toolName === SHELL_TOOL_NAME &&
+            this.config.getEnableToolOutputTruncation() &&
+            this.config.getTruncateToolOutputThreshold() > 0 &&
+            this.config.getTruncateToolOutputLines() > 0) {
+            const originalContentLength = content.length;
+            const threshold = this.config.getTruncateToolOutputThreshold();
+            const lines = this.config.getTruncateToolOutputLines();
+            if (content.length > threshold) {
+                const { outputFile: savedPath } = await saveTruncatedToolOutput(content, toolName, callId, this.config.storage.getProjectTempDir());
+                outputFile = savedPath;
+                content = formatTruncatedToolOutput(content, outputFile, lines);
+                logToolOutputTruncated(this.config, new ToolOutputTruncatedEvent(call.request.prompt_id, {
+                    toolName,
+                    originalContentLength,
+                    truncatedContentLength: content.length,
+                    threshold,
+                    lines,
+                }));
+            }
+        }
+        const response = convertToFunctionResponse(toolName, callId, content, this.config.getActiveModel());
+        const successResponse = {
+            callId,
+            responseParts: response,
+            resultDisplay: toolResult.returnDisplay,
+            error: undefined,
+            errorType: undefined,
+            outputFile,
+            contentLength: typeof content === 'string' ? content.length : undefined,
+            data: toolResult.data,
+        };
+        const startTime = 'startTime' in call ? call.startTime : undefined;
+        // Ensure we have tool and invocation
+        if (!('tool' in call) || !('invocation' in call)) {
+            throw new Error('Successful tool call missing tool or invocation');
+        }
+        return {
+            status: 'success',
+            request: call.request,
+            tool: call.tool,
+            response: successResponse,
+            invocation: call.invocation,
+            durationMs: startTime ? Date.now() - startTime : undefined,
+            outcome: call.outcome,
+        };
+    }
+    createErrorResult(call, error, errorType, returnDisplay) {
+        const response = this.createErrorResponse(call.request, error, errorType, returnDisplay);
+        const startTime = 'startTime' in call ? call.startTime : undefined;
+        return {
+            status: 'error',
+            request: call.request,
+            response,
+            tool: call.tool,
+            durationMs: startTime ? Date.now() - startTime : undefined,
+            outcome: call.outcome,
+        };
+    }
+    createErrorResponse(request, error, errorType, returnDisplay) {
+        const displayText = returnDisplay ?? error.message;
+        return {
+            callId: request.callId,
+            error,
+            responseParts: [
+                {
+                    functionResponse: {
+                        id: request.callId,
+                        name: request.name,
+                        response: { error: error.message },
+                    },
+                },
+            ],
+            resultDisplay: displayText,
+            errorType,
+            contentLength: displayText.length,
+        };
+    }
+}
+//# sourceMappingURL=tool-executor.js.map

package/dist/src/scheduler/tool-executor.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-executor.js","sourceRoot":"","sources":["../../../src/scheduler/tool-executor.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AASH,OAAO,EACL,aAAa,EACb,wBAAwB,EACxB,sBAAsB,EACtB,iBAAiB,GAClB,MAAM,aAAa,CAAC;AACrB,OAAO,EAAE,eAAe,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AACxD,OAAO,EAAE,oBAAoB,EAAE,MAAM,iCAAiC,CAAC;AACvE,OAAO,EACL,uBAAuB,EACvB,yBAAyB,GAC1B,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EAAE,yBAAyB,EAAE,MAAM,8CAA8C,CAAC;AAiBzF,MAAM,OAAO,YAAY;IACM;IAA7B,YAA6B,MAAc;QAAd,WAAM,GAAN,MAAM,CAAQ;IAAG,CAAC;IAE/C,KAAK,CAAC,OAAO,CAAC,OAA6B;QACzC,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,mBAAmB,EAAE,gBAAgB,EAAE,GAAG,OAAO,CAAC;QACxE,MAAM,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC;QACzB,MAAM,QAAQ,GAAG,OAAO,CAAC,IAAI,CAAC;QAC9B,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAE9B,IAAI,CAAC,CAAC,MAAM,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,CAAC,YAAY,IAAI,IAAI,CAAC,EAAE,CAAC;YAC/D,MAAM,IAAI,KAAK,CACb,4BAA4B,MAAM,+BAA+B,CAClE,CAAC;QACJ,CAAC;QACD,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,GAAG,IAAI,CAAC;QAElC,6BAA6B;QAC7B,MAAM,kBAAkB,GACtB,IAAI,CAAC,eAAe,IAAI,mBAAmB;YACzC,CAAC,CAAC,CAAC,WAAgC,EAAE,EAAE;gBACnC,mBAAmB,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;YAC3C,CAAC;YACH,CAAC,CAAC,SAAS,CAAC;QAEhB,MAAM,oBAAoB,GAAG,IAAI,CAAC,MAAM,CAAC,uBAAuB,EAAE,CAAC;QAEnE,OAAO,iBAAiB,CACtB;YACE,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,UAAU,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE;SAClC,EACD,KAAK,EAAE,EAAE,QAAQ,EAAE,YAAY,EAAE,EAAE,EAAE;YACnC,YAAY,CAAC,KAAK,GAAG,EAAE,OAAO,EAAE,CAAC;YAEjC,IAAI,CAAC;gBACH,IAAI,OAA4B,CAAC;gBACjC,IAAI,UAAU,YAAY,mBAAmB,EAAE,CAAC;oBAC9C,MAAM,cAAc,GAAG,CAAC,GAAW,EAAE,EAAE;wBACrC,MAAM,aAAa,GAAsB;4BACvC,GAAG,IAAI;4BACP,MAAM,EAAE,WAAW;4BACnB,IAAI;4BACJ,UAAU;4BACV,GAAG;4BACH,SAAS,EAAE,WAAW,IAAI,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;yBAC5D,CAAC;wBACF,gBAAgB,CAAC,aAAa,CAAC,CAAC;oBAClC,CAAC,CAAC;oBACF,OAAO,GAAG,oBAAoB,CAC5B,UAAU,EACV,QAAQ,EACR,MAAM,EACN,IAAI,EACJ,kBAAkB,EAClB,oBAAoB,EACpB,cAAc,EACd,IAAI,CAAC,MAAM,CACZ,CAAC;gBACJ,CAAC;qBAAM,CAAC;oBACN,OAAO,GAAG,oBAAoB,CAC5B,UAAU,EACV,QAAQ,EACR,MAAM,EACN,IAAI,EACJ,kBAAkB,EAClB,oBAAoB,EACpB,SAAS,EACT,IAAI,CAAC,MAAM,CACZ,CAAC;gBACJ,CAAC;gBAED,MAAM,UAAU,GAAe,MAAM,OAAO,CAAC;gBAC7C,YAAY,CAAC,MAAM,GAAG,UAAU,CAAC;gBAEjC,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;oBACnB,OAAO,IAAI,CAAC,qBAAqB,CAC/B,IAAI,EACJ,gCAAgC,CACjC,CAAC;gBACJ,CAAC;qBAAM,IAAI,UAAU,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;oBAC1C,OAAO,MAAM,IAAI,CAAC,mBAAmB,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;gBAC1D,CAAC;qBAAM,CAAC;oBACN,MAAM,WAAW,GACf,OAAO,UAAU,CAAC,aAAa,KAAK,QAAQ;wBAC1C,CAAC,CAAC,UAAU,CAAC,aAAa;wBAC1B,CAAC,CAAC,SAAS,CAAC;oBAChB,OAAO,IAAI,CAAC,iBAAiB,CAC3B,IAAI,EACJ,IAAI,KAAK,CAAC,UAAU,CAAC,KAAK,CAAC,OAAO,CAAC,EACnC,UAAU,CAAC,KAAK,CAAC,IAAI,EACrB,WAAW,CACZ,CAAC;gBACJ,CAAC;YACH,CAAC;YAAC,OAAO,cAAuB,EAAE,CAAC;gBACjC,YAAY,CAAC,KAAK,GAAG,cAAc,CAAC;gBACpC,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;oBACnB,OAAO,IAAI,CAAC,qBAAqB,CAC/B,IAAI,EACJ,gCAAgC,CACjC,CAAC;gBACJ,CAAC;gBACD,MAAM,KAAK,GACT,cAAc,YAAY,KAAK;oBAC7B,CAAC,CAAC,cAAc;oBAChB,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC;gBACxC,OAAO,IAAI,CAAC,iBAAiB,CAC3B,IAAI,EACJ,KAAK,EACL,aAAa,CAAC,mBAAmB,CAClC,CAAC;YACJ,CAAC;QACH,CAAC,CACF,CAAC;IACJ,CAAC;IAEO,qBAAqB,CAC3B,IAAc,EACd,MAAc;QAEd,MAAM,YAAY,GAAG,yBAAyB,MAAM,EAAE,CAAC;QACvD,MAAM,SAAS,GAAG,WAAW,IAAI,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC;QAEnE,IAAI,CAAC,CAAC,MAAM,IAAI,IAAI,CAAC,IAAI,CAAC,CAAC,YAAY,IAAI,IAAI,CAAC,EAAE,CAAC;YACjD,yEAAyE;YACzE,YAAY;YACZ,MAAM,IAAI,KAAK,CAAC,wDAAwD,CAAC,CAAC;QAC5E,CAAC;QAED,OAAO;YACL,MAAM,EAAE,WAAW;YACnB,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,QAAQ,EAAE;gBACR,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM;gBAC3B,aAAa,EAAE;oBACb;wBACE,gBAAgB,EAAE;4BAChB,EAAE,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM;4BACvB,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,IAAI;4BACvB,QAAQ,EAAE,EAAE,KAAK,EAAE,YAAY,EAAE;yBAClC;qBACF;iBACF;gBACD,aAAa,EAAE,SAAS;gBACxB,KAAK,EAAE,SAAS;gBAChB,SAAS,EAAE,SAAS;gBACpB,aAAa,EAAE,YAAY,CAAC,MAAM;aACnC;YACD,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,UAAU,EAAE,IAAI,CAAC,UAAU;YAC3B,UAAU,EAAE,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,CAAC,CAAC,SAAS;YAC1D,OAAO,EAAE,IAAI,CAAC,OAAO;SACtB,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,mBAAmB,CAC/B,IAAc,EACd,UAAsB;QAEtB,IAAI,OAAO,GAAG,UAAU,CAAC,UAAU,CAAC;QACpC,IAAI,UAA8B,CAAC;QACnC,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC;QACnC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC;QAEnC,IACE,OAAO,OAAO,KAAK,QAAQ;YAC3B,QAAQ,KAAK,eAAe;YAC5B,IAAI,CAAC,MAAM,CAAC,6BAA6B,EAAE;YAC3C,IAAI,CAAC,MAAM,CAAC,8BAA8B,EAAE,GAAG,CAAC;YAChD,IAAI,CAAC,MAAM,CAAC,0BAA0B,EAAE,GAAG,CAAC,EAC5C,CAAC;YACD,MAAM,qBAAqB,GAAG,OAAO,CAAC,MAAM,CAAC;YAC7C,MAAM,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,8BAA8B,EAAE,CAAC;YAC/D,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,0BAA0B,EAAE,CAAC;YAEvD,IAAI,OAAO,CAAC,MAAM,GAAG,SAAS,EAAE,CAAC;gBAC/B,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,GAAG,MAAM,uBAAuB,CAC7D,OAAO,EACP,QAAQ,EACR,MAAM,EACN,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,iBAAiB,EAAE,CACxC,CAAC;gBACF,UAAU,GAAG,SAAS,CAAC;gBACvB,OAAO,GAAG,yBAAyB,CAAC,OAAO,EAAE,UAAU,EAAE,KAAK,CAAC,CAAC;gBAEhE,sBAAsB,CACpB,IAAI,CAAC,MAAM,EACX,IAAI,wBAAwB,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE;oBACnD,QAAQ;oBACR,qBAAqB;oBACrB,sBAAsB,EAAE,OAAO,CAAC,MAAM;oBACtC,SAAS;oBACT,KAAK;iBACN,CAAC,CACH,CAAC;YACJ,CAAC;QACH,CAAC;QAED,MAAM,QAAQ,GAAG,yBAAyB,CACxC,QAAQ,EACR,MAAM,EACN,OAAO,EACP,IAAI,CAAC,MAAM,CAAC,cAAc,EAAE,CAC7B,CAAC;QAEF,MAAM,eAAe,GAAyB;YAC5C,MAAM;YACN,aAAa,EAAE,QAAQ;YACvB,aAAa,EAAE,UAAU,CAAC,aAAa;YACvC,KAAK,EAAE,SAAS;YAChB,SAAS,EAAE,SAAS;YACpB,UAAU;YACV,aAAa,EAAE,OAAO,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS;YACvE,IAAI,EAAE,UAAU,CAAC,IAAI;SACtB,CAAC;QAEF,MAAM,SAAS,GAAG,WAAW,IAAI,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC;QACnE,qCAAqC;QACrC,IAAI,CAAC,CAAC,MAAM,IAAI,IAAI,CAAC,IAAI,CAAC,CAAC,YAAY,IAAI,IAAI,CAAC,EAAE,CAAC;YACjD,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC;QACrE,CAAC;QAED,OAAO;YACL,MAAM,EAAE,SAAS;YACjB,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,QAAQ,EAAE,eAAe;YACzB,UAAU,EAAE,IAAI,CAAC,UAAU;YAC3B,UAAU,EAAE,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,CAAC,CAAC,SAAS;YAC1D,OAAO,EAAE,IAAI,CAAC,OAAO;SACtB,CAAC;IACJ,CAAC;IAEO,iBAAiB,CACvB,IAAc,EACd,KAAY,EACZ,SAAyB,EACzB,aAAsB;QAEtB,MAAM,QAAQ,GAAG,IAAI,CAAC,mBAAmB,CACvC,IAAI,CAAC,OAAO,EACZ,KAAK,EACL,SAAS,EACT,aAAa,CACd,CAAC;QACF,MAAM,SAAS,GAAG,WAAW,IAAI,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC;QAEnE,OAAO;YACL,MAAM,EAAE,OAAO;YACf,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,QAAQ;YACR,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,UAAU,EAAE,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,CAAC,CAAC,SAAS;YAC1D,OAAO,EAAE,IAAI,CAAC,OAAO;SACtB,CAAC;IACJ,CAAC;IAEO,mBAAmB,CACzB,OAA4B,EAC5B,KAAY,EACZ,SAAoC,EACpC,aAAsB;QAEtB,MAAM,WAAW,GAAG,aAAa,IAAI,KAAK,CAAC,OAAO,CAAC;QACnD,OAAO;YACL,MAAM,EAAE,OAAO,CAAC,MAAM;YACtB,KAAK;YACL,aAAa,EAAE;gBACb;oBACE,gBAAgB,EAAE;wBAChB,EAAE,EAAE,OAAO,CAAC,MAAM;wBAClB,IAAI,EAAE,OAAO,CAAC,IAAI;wBAClB,QAAQ,EAAE,EAAE,KAAK,EAAE,KAAK,CAAC,OAAO,EAAE;qBACnC;iBACF;aACF;YACD,aAAa,EAAE,WAAW;YAC1B,SAAS;YACT,aAAa,EAAE,WAAW,CAAC,MAAM;SAClC,CAAC;IACJ,CAAC;CACF"}

package/dist/src/scheduler/tool-modifier.d.ts

@@ -0,0 +1,23 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { EditorType } from '../utils/editor.js';
+import type { ToolConfirmationPayload } from '../tools/tools.js';
+import type { WaitingToolCall } from './types.js';
+export interface ModificationResult {
+    updatedParams: Record<string, unknown>;
+    updatedDiff?: string;
+}
+export declare class ToolModificationHandler {
+    /**
+     * Handles the "Modify with Editor" flow where an external editor is launched
+     * to modify the tool's parameters.
+     */
+    handleModifyWithEditor(toolCall: WaitingToolCall, editorType: EditorType, signal: AbortSignal): Promise<ModificationResult | undefined>;
+    /**
+     * Applies user-provided inline content updates (e.g. from the chat UI).
+     */
+    applyInlineModify(toolCall: WaitingToolCall, payload: ToolConfirmationPayload, signal: AbortSignal): Promise<ModificationResult | undefined>;
+}