@anthropic-field/core 0.1.0

package/dist/src/scheduler/scheduler.d.ts

@@ -0,0 +1,61 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { Config } from '../config/config.js';
+import type { MessageBus } from '../confirmation-bus/message-bus.js';
+import { type ToolCallRequestInfo, type CompletedToolCall } from './types.js';
+import type { EditorType } from '../utils/editor.js';
+export interface SchedulerOptions {
+    config: Config;
+    messageBus: MessageBus;
+    getPreferredEditor: () => EditorType | undefined;
+    schedulerId: string;
+    parentCallId?: string;
+}
+/**
+ * Event-Driven Orchestrator for Tool Execution.
+ * Coordinates execution via state updates and event listening.
+ */
+export declare class Scheduler {
+    private static subscribedMessageBuses;
+    private readonly state;
+    private readonly executor;
+    private readonly modifier;
+    private readonly config;
+    private readonly messageBus;
+    private readonly getPreferredEditor;
+    private readonly schedulerId;
+    private readonly parentCallId?;
+    private isProcessing;
+    private isCancelling;
+    private readonly requestQueue;
+    constructor(options: SchedulerOptions);
+    private setupMessageBusListener;
+    /**
+     * Schedules a batch of tool calls.
+     * @returns A promise that resolves with the results of the completed batch.
+     */
+    schedule(request: ToolCallRequestInfo | ToolCallRequestInfo[], signal: AbortSignal): Promise<CompletedToolCall[]>;
+    private _enqueueRequest;
+    cancelAll(): void;
+    get completedCalls(): CompletedToolCall[];
+    private isTerminal;
+    private _startBatch;
+    private _createToolNotFoundErroredToolCall;
+    private _validateAndCreateToolCall;
+    private _processQueue;
+    /**
+     * Processes the next item in the queue.
+     * @returns true if the loop should continue, false if it should terminate.
+     */
+    private _processNextItem;
+    private _processValidatingCall;
+    private _processToolCall;
+    /**
+     * Executes the tool and records the result.
+     */
+    private _execute;
+    private _processNextInRequestQueue;
+}

package/dist/src/scheduler/scheduler.js

@@ -0,0 +1,356 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { SchedulerStateManager } from './state-manager.js';
+import { resolveConfirmation } from './confirmation.js';
+import { checkPolicy, updatePolicy } from './policy.js';
+import { ToolExecutor } from './tool-executor.js';
+import { ToolModificationHandler } from './tool-modifier.js';
+import {} from './types.js';
+import { ToolErrorType } from '../tools/tool-error.js';
+import { PolicyDecision } from '../policy/types.js';
+import { ToolConfirmationOutcome, } from '../tools/tools.js';
+import { getToolSuggestion } from '../utils/tool-utils.js';
+import { runInDevTraceSpan } from '../telemetry/trace.js';
+import { logToolCall } from '../telemetry/loggers.js';
+import { ToolCallEvent } from '../telemetry/types.js';
+import { MessageBusType, } from '../confirmation-bus/types.js';
+import { runWithToolCallContext } from '../utils/toolCallContext.js';
+const createErrorResponse = (request, error, errorType) => ({
+    callId: request.callId,
+    error,
+    responseParts: [
+        {
+            functionResponse: {
+                id: request.callId,
+                name: request.name,
+                response: { error: error.message },
+            },
+        },
+    ],
+    resultDisplay: error.message,
+    errorType,
+    contentLength: error.message.length,
+});
+/**
+ * Event-Driven Orchestrator for Tool Execution.
+ * Coordinates execution via state updates and event listening.
+ */
+export class Scheduler {
+    // Tracks which MessageBus instances have the legacy listener attached to prevent duplicates.
+    static subscribedMessageBuses = new WeakSet();
+    state;
+    executor;
+    modifier;
+    config;
+    messageBus;
+    getPreferredEditor;
+    schedulerId;
+    parentCallId;
+    isProcessing = false;
+    isCancelling = false;
+    requestQueue = [];
+    constructor(options) {
+        this.config = options.config;
+        this.messageBus = options.messageBus;
+        this.getPreferredEditor = options.getPreferredEditor;
+        this.schedulerId = options.schedulerId;
+        this.parentCallId = options.parentCallId;
+        this.state = new SchedulerStateManager(this.messageBus, this.schedulerId, (call) => logToolCall(this.config, new ToolCallEvent(call)));
+        this.executor = new ToolExecutor(this.config);
+        this.modifier = new ToolModificationHandler();
+        this.setupMessageBusListener(this.messageBus);
+    }
+    setupMessageBusListener(messageBus) {
+        if (Scheduler.subscribedMessageBuses.has(messageBus)) {
+            return;
+        }
+        // TODO: Optimize policy checks. Currently, tools check policy via
+        // MessageBus even though the Scheduler already checked it.
+        messageBus.subscribe(MessageBusType.TOOL_CONFIRMATION_REQUEST, async (request) => {
+            await messageBus.publish({
+                type: MessageBusType.TOOL_CONFIRMATION_RESPONSE,
+                correlationId: request.correlationId,
+                confirmed: false,
+                requiresUserConfirmation: true,
+            });
+        });
+        Scheduler.subscribedMessageBuses.add(messageBus);
+    }
+    /**
+     * Schedules a batch of tool calls.
+     * @returns A promise that resolves with the results of the completed batch.
+     */
+    async schedule(request, signal) {
+        return runInDevTraceSpan({ name: 'schedule' }, async ({ metadata: spanMetadata }) => {
+            const requests = Array.isArray(request) ? request : [request];
+            spanMetadata.input = requests;
+            if (this.isProcessing || this.state.isActive) {
+                return this._enqueueRequest(requests, signal);
+            }
+            return this._startBatch(requests, signal);
+        });
+    }
+    _enqueueRequest(requests, signal) {
+        return new Promise((resolve, reject) => {
+            const abortHandler = () => {
+                const index = this.requestQueue.findIndex((item) => item.requests === requests);
+                if (index > -1) {
+                    this.requestQueue.splice(index, 1);
+                    reject(new Error('Tool call cancelled while in queue.'));
+                }
+            };
+            if (signal.aborted) {
+                reject(new Error('Operation cancelled'));
+                return;
+            }
+            signal.addEventListener('abort', abortHandler, { once: true });
+            this.requestQueue.push({
+                requests,
+                signal,
+                resolve: (results) => {
+                    signal.removeEventListener('abort', abortHandler);
+                    resolve(results);
+                },
+                reject: (err) => {
+                    signal.removeEventListener('abort', abortHandler);
+                    reject(err);
+                },
+            });
+        });
+    }
+    cancelAll() {
+        if (this.isCancelling)
+            return;
+        this.isCancelling = true;
+        // Clear scheduler request queue
+        while (this.requestQueue.length > 0) {
+            const next = this.requestQueue.shift();
+            next?.reject(new Error('Operation cancelled by user'));
+        }
+        // Cancel active call
+        const activeCall = this.state.firstActiveCall;
+        if (activeCall && !this.isTerminal(activeCall.status)) {
+            this.state.updateStatus(activeCall.request.callId, 'cancelled', 'Operation cancelled by user');
+        }
+        // Clear queue
+        this.state.cancelAllQueued('Operation cancelled by user');
+    }
+    get completedCalls() {
+        return this.state.completedBatch;
+    }
+    isTerminal(status) {
+        return status === 'success' || status === 'error' || status === 'cancelled';
+    }
+    // --- Phase 1: Ingestion & Resolution ---
+    async _startBatch(requests, signal) {
+        this.isProcessing = true;
+        this.isCancelling = false;
+        this.state.clearBatch();
+        try {
+            const toolRegistry = this.config.getToolRegistry();
+            const newCalls = requests.map((request) => {
+                const enrichedRequest = {
+                    ...request,
+                    schedulerId: this.schedulerId,
+                    parentCallId: this.parentCallId,
+                };
+                const tool = toolRegistry.getTool(request.name);
+                if (!tool) {
+                    return this._createToolNotFoundErroredToolCall(enrichedRequest, toolRegistry.getAllToolNames());
+                }
+                return this._validateAndCreateToolCall(enrichedRequest, tool);
+            });
+            this.state.enqueue(newCalls);
+            await this._processQueue(signal);
+            return this.state.completedBatch;
+        }
+        finally {
+            this.isProcessing = false;
+            this.state.clearBatch();
+            this._processNextInRequestQueue();
+        }
+    }
+    _createToolNotFoundErroredToolCall(request, toolNames) {
+        const suggestion = getToolSuggestion(request.name, toolNames);
+        return {
+            status: 'error',
+            request,
+            response: createErrorResponse(request, new Error(`Tool "${request.name}" not found.${suggestion}`), ToolErrorType.TOOL_NOT_REGISTERED),
+            durationMs: 0,
+            schedulerId: this.schedulerId,
+        };
+    }
+    _validateAndCreateToolCall(request, tool) {
+        return runWithToolCallContext({
+            callId: request.callId,
+            schedulerId: this.schedulerId,
+            parentCallId: this.parentCallId,
+        }, () => {
+            try {
+                const invocation = tool.build(request.args);
+                return {
+                    status: 'validating',
+                    request,
+                    tool,
+                    invocation,
+                    startTime: Date.now(),
+                    schedulerId: this.schedulerId,
+                };
+            }
+            catch (e) {
+                return {
+                    status: 'error',
+                    request,
+                    tool,
+                    response: createErrorResponse(request, e instanceof Error ? e : new Error(String(e)), ToolErrorType.INVALID_TOOL_PARAMS),
+                    durationMs: 0,
+                    schedulerId: this.schedulerId,
+                };
+            }
+        });
+    }
+    // --- Phase 2: Processing Loop ---
+    async _processQueue(signal) {
+        while (this.state.queueLength > 0 || this.state.isActive) {
+            const shouldContinue = await this._processNextItem(signal);
+            if (!shouldContinue)
+                break;
+        }
+    }
+    /**
+     * Processes the next item in the queue.
+     * @returns true if the loop should continue, false if it should terminate.
+     */
+    async _processNextItem(signal) {
+        if (signal.aborted || this.isCancelling) {
+            this.state.cancelAllQueued('Operation cancelled');
+            return false;
+        }
+        if (!this.state.isActive) {
+            const next = this.state.dequeue();
+            if (!next)
+                return false;
+            if (next.status === 'error') {
+                this.state.updateStatus(next.request.callId, 'error', next.response);
+                this.state.finalizeCall(next.request.callId);
+                return true;
+            }
+        }
+        const active = this.state.firstActiveCall;
+        if (!active)
+            return false;
+        if (active.status === 'validating') {
+            await this._processValidatingCall(active, signal);
+        }
+        return true;
+    }
+    async _processValidatingCall(active, signal) {
+        try {
+            await this._processToolCall(active, signal);
+        }
+        catch (error) {
+            const err = error instanceof Error ? error : new Error(String(error));
+            // If the signal aborted while we were waiting on something, treat as
+            // cancelled. Otherwise, it's a genuine unhandled system exception.
+            if (signal.aborted || err.name === 'AbortError') {
+                this.state.updateStatus(active.request.callId, 'cancelled', 'Operation cancelled');
+            }
+            else {
+                this.state.updateStatus(active.request.callId, 'error', createErrorResponse(active.request, err, ToolErrorType.UNHANDLED_EXCEPTION));
+            }
+        }
+        this.state.finalizeCall(active.request.callId);
+    }
+    // --- Phase 3: Single Call Orchestration ---
+    async _processToolCall(toolCall, signal) {
+        const callId = toolCall.request.callId;
+        // Policy & Security
+        const { decision, rule } = await checkPolicy(toolCall, this.config);
+        if (decision === PolicyDecision.DENY) {
+            const denyMessage = rule?.denyMessage ? ` ${rule.denyMessage}` : '';
+            this.state.updateStatus(callId, 'error', createErrorResponse(toolCall.request, new Error(`Tool execution denied by policy.${denyMessage}`), ToolErrorType.POLICY_VIOLATION));
+            this.state.finalizeCall(callId);
+            return;
+        }
+        // User Confirmation Loop
+        let outcome = ToolConfirmationOutcome.ProceedOnce;
+        let lastDetails;
+        if (decision === PolicyDecision.ASK_USER) {
+            const result = await resolveConfirmation(toolCall, signal, {
+                config: this.config,
+                messageBus: this.messageBus,
+                state: this.state,
+                modifier: this.modifier,
+                getPreferredEditor: this.getPreferredEditor,
+                schedulerId: this.schedulerId,
+            });
+            outcome = result.outcome;
+            lastDetails = result.lastDetails;
+        }
+        else {
+            this.state.setOutcome(callId, ToolConfirmationOutcome.ProceedOnce);
+        }
+        // Handle Policy Updates
+        await updatePolicy(toolCall.tool, outcome, lastDetails, {
+            config: this.config,
+            messageBus: this.messageBus,
+        });
+        // Handle cancellation (cascades to entire batch)
+        if (outcome === ToolConfirmationOutcome.Cancel) {
+            this.state.updateStatus(callId, 'cancelled', 'User denied execution.');
+            this.state.finalizeCall(callId);
+            this.state.cancelAllQueued('User cancelled operation');
+            return; // Skip execution
+        }
+        // Execution
+        await this._execute(callId, signal);
+    }
+    // --- Sub-phase Handlers ---
+    /**
+     * Executes the tool and records the result.
+     */
+    async _execute(callId, signal) {
+        this.state.updateStatus(callId, 'scheduled');
+        if (signal.aborted)
+            throw new Error('Operation cancelled');
+        this.state.updateStatus(callId, 'executing');
+        const activeCall = this.state.firstActiveCall;
+        const result = await runWithToolCallContext({
+            callId: activeCall.request.callId,
+            schedulerId: this.schedulerId,
+            parentCallId: this.parentCallId,
+        }, () => this.executor.execute({
+            call: activeCall,
+            signal,
+            outputUpdateHandler: (id, out) => this.state.updateStatus(id, 'executing', { liveOutput: out }),
+            onUpdateToolCall: (updated) => {
+                if (updated.status === 'executing' && updated.pid) {
+                    this.state.updateStatus(callId, 'executing', {
+                        pid: updated.pid,
+                    });
+                }
+            },
+        }));
+        if (result.status === 'success') {
+            this.state.updateStatus(callId, 'success', result.response);
+        }
+        else if (result.status === 'cancelled') {
+            this.state.updateStatus(callId, 'cancelled', 'Operation cancelled');
+        }
+        else {
+            this.state.updateStatus(callId, 'error', result.response);
+        }
+    }
+    _processNextInRequestQueue() {
+        if (this.requestQueue.length > 0) {
+            const next = this.requestQueue.shift();
+            this.schedule(next.requests, next.signal)
+                .then(next.resolve)
+                .catch(next.reject);
+        }
+    }
+}
+//# sourceMappingURL=scheduler.js.map

package/dist/src/scheduler/scheduler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"scheduler.js","sourceRoot":"","sources":["../../../src/scheduler/scheduler.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAIH,OAAO,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAC3D,OAAO,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AACxD,OAAO,EAAE,WAAW,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AACxD,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EAAE,uBAAuB,EAAE,MAAM,oBAAoB,CAAC;AAC7D,OAAO,EAQN,MAAM,YAAY,CAAC;AACpB,OAAO,EAAE,aAAa,EAAE,MAAM,wBAAwB,CAAC;AACvD,OAAO,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AACpD,OAAO,EACL,uBAAuB,GAExB,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAC3D,OAAO,EAAE,iBAAiB,EAAE,MAAM,uBAAuB,CAAC;AAC1D,OAAO,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AACtD,OAAO,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AAEtD,OAAO,EACL,cAAc,GAGf,MAAM,8BAA8B,CAAC;AACtC,OAAO,EAAE,sBAAsB,EAAE,MAAM,6BAA6B,CAAC;AAiBrE,MAAM,mBAAmB,GAAG,CAC1B,OAA4B,EAC5B,KAAY,EACZ,SAAoC,EACd,EAAE,CAAC,CAAC;IAC1B,MAAM,EAAE,OAAO,CAAC,MAAM;IACtB,KAAK;IACL,aAAa,EAAE;QACb;YACE,gBAAgB,EAAE;gBAChB,EAAE,EAAE,OAAO,CAAC,MAAM;gBAClB,IAAI,EAAE,OAAO,CAAC,IAAI;gBAClB,QAAQ,EAAE,EAAE,KAAK,EAAE,KAAK,CAAC,OAAO,EAAE;aACnC;SACF;KACF;IACD,aAAa,EAAE,KAAK,CAAC,OAAO;IAC5B,SAAS;IACT,aAAa,EAAE,KAAK,CAAC,OAAO,CAAC,MAAM;CACpC,CAAC,CAAC;AAEH;;;GAGG;AACH,MAAM,OAAO,SAAS;IACpB,6FAA6F;IACrF,MAAM,CAAC,sBAAsB,GAAG,IAAI,OAAO,EAAc,CAAC;IAEjD,KAAK,CAAwB;IAC7B,QAAQ,CAAe;IACvB,QAAQ,CAA0B;IAClC,MAAM,CAAS;IACf,UAAU,CAAa;IACvB,kBAAkB,CAA+B;IACjD,WAAW,CAAS;IACpB,YAAY,CAAU;IAE/B,YAAY,GAAG,KAAK,CAAC;IACrB,YAAY,GAAG,KAAK,CAAC;IACZ,YAAY,GAAyB,EAAE,CAAC;IAEzD,YAAY,OAAyB;QACnC,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC7B,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC;QACrC,IAAI,CAAC,kBAAkB,GAAG,OAAO,CAAC,kBAAkB,CAAC;QACrD,IAAI,CAAC,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC;QACvC,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC,YAAY,CAAC;QACzC,IAAI,CAAC,KAAK,GAAG,IAAI,qBAAqB,CACpC,IAAI,CAAC,UAAU,EACf,IAAI,CAAC,WAAW,EAChB,CAAC,IAAI,EAAE,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,aAAa,CAAC,IAAI,CAAC,CAAC,CAC5D,CAAC;QACF,IAAI,CAAC,QAAQ,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC9C,IAAI,CAAC,QAAQ,GAAG,IAAI,uBAAuB,EAAE,CAAC;QAE9C,IAAI,CAAC,uBAAuB,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAChD,CAAC;IAEO,uBAAuB,CAAC,UAAsB;QACpD,IAAI,SAAS,CAAC,sBAAsB,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC;YACrD,OAAO;QACT,CAAC;QAED,kEAAkE;QAClE,2DAA2D;QAC3D,UAAU,CAAC,SAAS,CAClB,cAAc,CAAC,yBAAyB,EACxC,KAAK,EAAE,OAAgC,EAAE,EAAE;YACzC,MAAM,UAAU,CAAC,OAAO,CAAC;gBACvB,IAAI,EAAE,cAAc,CAAC,0BAA0B;gBAC/C,aAAa,EAAE,OAAO,CAAC,aAAa;gBACpC,SAAS,EAAE,KAAK;gBAChB,wBAAwB,EAAE,IAAI;aAC/B,CAAC,CAAC;QACL,CAAC,CACF,CAAC;QAEF,SAAS,CAAC,sBAAsB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;IACnD,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,QAAQ,CACZ,OAAoD,EACpD,MAAmB;QAEnB,OAAO,iBAAiB,CACtB,EAAE,IAAI,EAAE,UAAU,EAAE,EACpB,KAAK,EAAE,EAAE,QAAQ,EAAE,YAAY,EAAE,EAAE,EAAE;YACnC,MAAM,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;YAC9D,YAAY,CAAC,KAAK,GAAG,QAAQ,CAAC;YAE9B,IAAI,IAAI,CAAC,YAAY,IAAI,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC;gBAC7C,OAAO,IAAI,CAAC,eAAe,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;YAChD,CAAC;YAED,OAAO,IAAI,CAAC,WAAW,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QAC5C,CAAC,CACF,CAAC;IACJ,CAAC;IAEO,eAAe,CACrB,QAA+B,EAC/B,MAAmB;QAEnB,OAAO,IAAI,OAAO,CAAsB,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YAC1D,MAAM,YAAY,GAAG,GAAG,EAAE;gBACxB,MAAM,KAAK,GAAG,IAAI,CAAC,YAAY,CAAC,SAAS,CACvC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,KAAK,QAAQ,CACrC,CAAC;gBACF,IAAI,KAAK,GAAG,CAAC,CAAC,EAAE,CAAC;oBACf,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;oBACnC,MAAM,CAAC,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC,CAAC;gBAC3D,CAAC;YACH,CAAC,CAAC;YAEF,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;gBACnB,MAAM,CAAC,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC,CAAC;gBACzC,OAAO;YACT,CAAC;YAED,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,YAAY,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;YAE/D,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC;gBACrB,QAAQ;gBACR,MAAM;gBACN,OAAO,EAAE,CAAC,OAAO,EAAE,EAAE;oBACnB,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC;oBAClD,OAAO,CAAC,OAAO,CAAC,CAAC;gBACnB,CAAC;gBACD,MAAM,EAAE,CAAC,GAAG,EAAE,EAAE;oBACd,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC;oBAClD,MAAM,CAAC,GAAG,CAAC,CAAC;gBACd,CAAC;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAED,SAAS;QACP,IAAI,IAAI,CAAC,YAAY;YAAE,OAAO;QAC9B,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;QAEzB,gCAAgC;QAChC,OAAO,IAAI,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACpC,MAAM,IAAI,GAAG,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC;YACvC,IAAI,EAAE,MAAM,CAAC,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC,CAAC;QACzD,CAAC;QAED,qBAAqB;QACrB,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,eAAe,CAAC;QAC9C,IAAI,UAAU,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;YACtD,IAAI,CAAC,KAAK,CAAC,YAAY,CACrB,UAAU,CAAC,OAAO,CAAC,MAAM,EACzB,WAAW,EACX,6BAA6B,CAC9B,CAAC;QACJ,CAAC;QAED,cAAc;QACd,IAAI,CAAC,KAAK,CAAC,eAAe,CAAC,6BAA6B,CAAC,CAAC;IAC5D,CAAC;IAED,IAAI,cAAc;QAChB,OAAO,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC;IACnC,CAAC;IAEO,UAAU,CAAC,MAAc;QAC/B,OAAO,MAAM,KAAK,SAAS,IAAI,MAAM,KAAK,OAAO,IAAI,MAAM,KAAK,WAAW,CAAC;IAC9E,CAAC;IAED,0CAA0C;IAElC,KAAK,CAAC,WAAW,CACvB,QAA+B,EAC/B,MAAmB;QAEnB,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;QACzB,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC;QAC1B,IAAI,CAAC,KAAK,CAAC,UAAU,EAAE,CAAC;QAExB,IAAI,CAAC;YACH,MAAM,YAAY,GAAG,IAAI,CAAC,MAAM,CAAC,eAAe,EAAE,CAAC;YACnD,MAAM,QAAQ,GAAe,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE;gBACpD,MAAM,eAAe,GAAwB;oBAC3C,GAAG,OAAO;oBACV,WAAW,EAAE,IAAI,CAAC,WAAW;oBAC7B,YAAY,EAAE,IAAI,CAAC,YAAY;iBAChC,CAAC;gBACF,MAAM,IAAI,GAAG,YAAY,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;gBAEhD,IAAI,CAAC,IAAI,EAAE,CAAC;oBACV,OAAO,IAAI,CAAC,kCAAkC,CAC5C,eAAe,EACf,YAAY,CAAC,eAAe,EAAE,CAC/B,CAAC;gBACJ,CAAC;gBAED,OAAO,IAAI,CAAC,0BAA0B,CAAC,eAAe,EAAE,IAAI,CAAC,CAAC;YAChE,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;YAC7B,MAAM,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;YACjC,OAAO,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC;QACnC,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC;YAC1B,IAAI,CAAC,KAAK,CAAC,UAAU,EAAE,CAAC;YACxB,IAAI,CAAC,0BAA0B,EAAE,CAAC;QACpC,CAAC;IACH,CAAC;IAEO,kCAAkC,CACxC,OAA4B,EAC5B,SAAmB;QAEnB,MAAM,UAAU,GAAG,iBAAiB,CAAC,OAAO,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;QAC9D,OAAO;YACL,MAAM,EAAE,OAAO;YACf,OAAO;YACP,QAAQ,EAAE,mBAAmB,CAC3B,OAAO,EACP,IAAI,KAAK,CAAC,SAAS,OAAO,CAAC,IAAI,eAAe,UAAU,EAAE,CAAC,EAC3D,aAAa,CAAC,mBAAmB,CAClC;YACD,UAAU,EAAE,CAAC;YACb,WAAW,EAAE,IAAI,CAAC,WAAW;SAC9B,CAAC;IACJ,CAAC;IAEO,0BAA0B,CAChC,OAA4B,EAC5B,IAAwB;QAExB,OAAO,sBAAsB,CAC3B;YACE,MAAM,EAAE,OAAO,CAAC,MAAM;YACtB,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,YAAY,EAAE,IAAI,CAAC,YAAY;SAChC,EACD,GAAG,EAAE;YACH,IAAI,CAAC;gBACH,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;gBAC5C,OAAO;oBACL,MAAM,EAAE,YAAY;oBACpB,OAAO;oBACP,IAAI;oBACJ,UAAU;oBACV,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;oBACrB,WAAW,EAAE,IAAI,CAAC,WAAW;iBAC9B,CAAC;YACJ,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,OAAO;oBACL,MAAM,EAAE,OAAO;oBACf,OAAO;oBACP,IAAI;oBACJ,QAAQ,EAAE,mBAAmB,CAC3B,OAAO,EACP,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAC7C,aAAa,CAAC,mBAAmB,CAClC;oBACD,UAAU,EAAE,CAAC;oBACb,WAAW,EAAE,IAAI,CAAC,WAAW;iBAC9B,CAAC;YACJ,CAAC;QACH,CAAC,CACF,CAAC;IACJ,CAAC;IAED,mCAAmC;IAE3B,KAAK,CAAC,aAAa,CAAC,MAAmB;QAC7C,OAAO,IAAI,CAAC,KAAK,CAAC,WAAW,GAAG,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC;YACzD,MAAM,cAAc,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC;YAC3D,IAAI,CAAC,cAAc;gBAAE,MAAM;QAC7B,CAAC;IACH,CAAC;IAED;;;OAGG;IACK,KAAK,CAAC,gBAAgB,CAAC,MAAmB;QAChD,IAAI,MAAM,CAAC,OAAO,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACxC,IAAI,CAAC,KAAK,CAAC,eAAe,CAAC,qBAAqB,CAAC,CAAC;YAClD,OAAO,KAAK,CAAC;QACf,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC;YACzB,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;YAClC,IAAI,CAAC,IAAI;gBAAE,OAAO,KAAK,CAAC;YAExB,IAAI,IAAI,CAAC,MAAM,KAAK,OAAO,EAAE,CAAC;gBAC5B,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;gBACrE,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;gBAC7C,OAAO,IAAI,CAAC;YACd,CAAC;QACH,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,eAAe,CAAC;QAC1C,IAAI,CAAC,MAAM;YAAE,OAAO,KAAK,CAAC;QAE1B,IAAI,MAAM,CAAC,MAAM,KAAK,YAAY,EAAE,CAAC;YACnC,MAAM,IAAI,CAAC,sBAAsB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACpD,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,KAAK,CAAC,sBAAsB,CAClC,MAA0B,EAC1B,MAAmB;QAEnB,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,gBAAgB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAC9C,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,GAAG,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;YACtE,qEAAqE;YACrE,mEAAmE;YACnE,IAAI,MAAM,CAAC,OAAO,IAAI,GAAG,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;gBAChD,IAAI,CAAC,KAAK,CAAC,YAAY,CACrB,MAAM,CAAC,OAAO,CAAC,MAAM,EACrB,WAAW,EACX,qBAAqB,CACtB,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,KAAK,CAAC,YAAY,CACrB,MAAM,CAAC,OAAO,CAAC,MAAM,EACrB,OAAO,EACP,mBAAmB,CACjB,MAAM,CAAC,OAAO,EACd,GAAG,EACH,aAAa,CAAC,mBAAmB,CAClC,CACF,CAAC;YACJ,CAAC;QACH,CAAC;QAED,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IACjD,CAAC;IAED,6CAA6C;IAErC,KAAK,CAAC,gBAAgB,CAC5B,QAA4B,EAC5B,MAAmB;QAEnB,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,MAAM,CAAC;QAEvC,oBAAoB;QACpB,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,GAAG,MAAM,WAAW,CAAC,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;QAEpE,IAAI,QAAQ,KAAK,cAAc,CAAC,IAAI,EAAE,CAAC;YACrC,MAAM,WAAW,GAAG,IAAI,EAAE,WAAW,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACpE,IAAI,CAAC,KAAK,CAAC,YAAY,CACrB,MAAM,EACN,OAAO,EACP,mBAAmB,CACjB,QAAQ,CAAC,OAAO,EAChB,IAAI,KAAK,CAAC,mCAAmC,WAAW,EAAE,CAAC,EAC3D,aAAa,CAAC,gBAAgB,CAC/B,CACF,CAAC;YACF,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;YAChC,OAAO;QACT,CAAC;QAED,yBAAyB;QACzB,IAAI,OAAO,GAAG,uBAAuB,CAAC,WAAW,CAAC;QAClD,IAAI,WAAwD,CAAC;QAE7D,IAAI,QAAQ,KAAK,cAAc,CAAC,QAAQ,EAAE,CAAC;YACzC,MAAM,MAAM,GAAG,MAAM,mBAAmB,CAAC,QAAQ,EAAE,MAAM,EAAE;gBACzD,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,UAAU,EAAE,IAAI,CAAC,UAAU;gBAC3B,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,kBAAkB,EAAE,IAAI,CAAC,kBAAkB;gBAC3C,WAAW,EAAE,IAAI,CAAC,WAAW;aAC9B,CAAC,CAAC;YACH,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC;YACzB,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC;QACnC,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,MAAM,EAAE,uBAAuB,CAAC,WAAW,CAAC,CAAC;QACrE,CAAC;QAED,wBAAwB;QACxB,MAAM,YAAY,CAAC,QAAQ,CAAC,IAAI,EAAE,OAAO,EAAE,WAAW,EAAE;YACtD,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,UAAU,EAAE,IAAI,CAAC,UAAU;SAC5B,CAAC,CAAC;QAEH,iDAAiD;QACjD,IAAI,OAAO,KAAK,uBAAuB,CAAC,MAAM,EAAE,CAAC;YAC/C,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,MAAM,EAAE,WAAW,EAAE,wBAAwB,CAAC,CAAC;YACvE,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;YAChC,IAAI,CAAC,KAAK,CAAC,eAAe,CAAC,0BAA0B,CAAC,CAAC;YACvD,OAAO,CAAC,iBAAiB;QAC3B,CAAC;QAED,YAAY;QACZ,MAAM,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACtC,CAAC;IAED,6BAA6B;IAE7B;;OAEG;IACK,KAAK,CAAC,QAAQ,CAAC,MAAc,EAAE,MAAmB;QACxD,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;QAC7C,IAAI,MAAM,CAAC,OAAO;YAAE,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;QAC3D,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;QAE7C,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,eAAoC,CAAC;QAEnE,MAAM,MAAM,GAAG,MAAM,sBAAsB,CACzC;YACE,MAAM,EAAE,UAAU,CAAC,OAAO,CAAC,MAAM;YACjC,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,YAAY,EAAE,IAAI,CAAC,YAAY;SAChC,EACD,GAAG,EAAE,CACH,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;YACpB,IAAI,EAAE,UAAU;YAChB,MAAM;YACN,mBAAmB,EAAE,CAAC,EAAE,EAAE,GAAG,EAAE,EAAE,CAC/B,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,EAAE,EAAE,WAAW,EAAE,EAAE,UAAU,EAAE,GAAG,EAAE,CAAC;YAC/D,gBAAgB,EAAE,CAAC,OAAO,EAAE,EAAE;gBAC5B,IAAI,OAAO,CAAC,MAAM,KAAK,WAAW,IAAI,OAAO,CAAC,GAAG,EAAE,CAAC;oBAClD,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,MAAM,EAAE,WAAW,EAAE;wBAC3C,GAAG,EAAE,OAAO,CAAC,GAAG;qBACjB,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;SACF,CAAC,CACL,CAAC;QAEF,IAAI,MAAM,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;YAChC,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,MAAM,EAAE,SAAS,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC9D,CAAC;aAAM,IAAI,MAAM,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;YACzC,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,MAAM,EAAE,WAAW,EAAE,qBAAqB,CAAC,CAAC;QACtE,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,MAAM,EAAE,OAAO,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC5D,CAAC;IACH,CAAC;IAEO,0BAA0B;QAChC,IAAI,IAAI,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACjC,MAAM,IAAI,GAAG,IAAI,CAAC,YAAY,CAAC,KAAK,EAAG,CAAC;YACxC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC;iBACtC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC;iBAClB,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACxB,CAAC;IACH,CAAC"}

package/dist/src/scheduler/state-manager.d.ts

@@ -0,0 +1,73 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { ToolCall, CompletedToolCall, ExecutingToolCall, ToolCallResponseInfo } from './types.js';
+import type { ToolConfirmationOutcome, AnyToolInvocation, ToolCallConfirmationDetails } from '../tools/tools.js';
+import type { MessageBus } from '../confirmation-bus/message-bus.js';
+import { type SerializableConfirmationDetails } from '../confirmation-bus/types.js';
+/**
+ * Handler for terminal tool calls.
+ */
+export type TerminalCallHandler = (call: CompletedToolCall) => void;
+/**
+ * Manages the state of tool calls.
+ * Publishes state changes to the MessageBus via TOOL_CALLS_UPDATE events.
+ */
+export declare class SchedulerStateManager {
+    private readonly messageBus;
+    private readonly schedulerId;
+    private readonly onTerminalCall?;
+    private readonly activeCalls;
+    private readonly queue;
+    private _completedBatch;
+    constructor(messageBus: MessageBus, schedulerId?: string, onTerminalCall?: TerminalCallHandler | undefined);
+    addToolCalls(calls: ToolCall[]): void;
+    getToolCall(callId: string): ToolCall | undefined;
+    enqueue(calls: ToolCall[]): void;
+    dequeue(): ToolCall | undefined;
+    get isActive(): boolean;
+    get activeCallCount(): number;
+    get queueLength(): number;
+    get firstActiveCall(): ToolCall | undefined;
+    /**
+     * Updates the status of a tool call with specific auxiliary data required for certain states.
+     */
+    updateStatus(callId: string, status: 'success', data: ToolCallResponseInfo): void;
+    updateStatus(callId: string, status: 'error', data: ToolCallResponseInfo): void;
+    updateStatus(callId: string, status: 'awaiting_approval', data: ToolCallConfirmationDetails | {
+        correlationId: string;
+        confirmationDetails: SerializableConfirmationDetails;
+    }): void;
+    updateStatus(callId: string, status: 'cancelled', data: string): void;
+    updateStatus(callId: string, status: 'executing', data?: Partial<ExecutingToolCall>): void;
+    updateStatus(callId: string, status: 'scheduled' | 'validating'): void;
+    finalizeCall(callId: string): void;
+    updateArgs(callId: string, newArgs: Record<string, unknown>, newInvocation: AnyToolInvocation): void;
+    setOutcome(callId: string, outcome: ToolConfirmationOutcome): void;
+    cancelAllQueued(reason: string): void;
+    getSnapshot(): ToolCall[];
+    clearBatch(): void;
+    get completedBatch(): CompletedToolCall[];
+    private emitUpdate;
+    private isTerminalCall;
+    private transitionCall;
+    private isToolCallResponseInfo;
+    private isExecutingToolCallPatch;
+    /**
+     * Ensures the tool call has an associated tool and invocation before
+     * transitioning to states that require them.
+     */
+    private validateHasToolAndInvocation;
+    private toSuccess;
+    private toError;
+    private toAwaitingApproval;
+    private isEventDrivenApprovalData;
+    private toScheduled;
+    private toCancelled;
+    private isWaitingToolCall;
+    private patchCall;
+    private toValidating;
+    private toExecuting;
+}