@google/gemini-cli 0.26.0-preview.4 → 0.27.0-nightly.20260126.cb772a5b7

package/dist/src/ui/hooks/useToolExecutionScheduler.js

@@ -0,0 +1,113 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { MessageBusType, ToolConfirmationOutcome, Scheduler, } from '@google/gemini-cli-core';
+import { useCallback, useState, useMemo, useEffect, useRef } from 'react';
+/**
+ * Modern tool scheduler hook using the event-driven Core Scheduler.
+ *
+ * This hook acts as an Adapter between the new MessageBus-driven Core
+ * and the legacy callback-based UI components.
+ */
+export function useToolExecutionScheduler(onComplete, config, getPreferredEditor) {
+    // State stores Core objects, not Display objects
+    const [toolCalls, setToolCalls] = useState([]);
+    const [lastToolOutputTime, setLastToolOutputTime] = useState(0);
+    const messageBus = useMemo(() => config.getMessageBus(), [config]);
+    const onCompleteRef = useRef(onComplete);
+    useEffect(() => {
+        onCompleteRef.current = onComplete;
+    }, [onComplete]);
+    const getPreferredEditorRef = useRef(getPreferredEditor);
+    useEffect(() => {
+        getPreferredEditorRef.current = getPreferredEditor;
+    }, [getPreferredEditor]);
+    const scheduler = useMemo(() => new Scheduler({
+        config,
+        messageBus,
+        getPreferredEditor: () => getPreferredEditorRef.current(),
+    }), [config, messageBus]);
+    const internalAdaptToolCalls = useCallback((coreCalls, prevTracked) => adaptToolCalls(coreCalls, prevTracked, messageBus), [messageBus]);
+    useEffect(() => {
+        const handler = (event) => {
+            setToolCalls((prev) => {
+                const adapted = internalAdaptToolCalls(event.toolCalls, prev);
+                // Update output timer for UI spinners
+                if (event.toolCalls.some((tc) => tc.status === 'executing')) {
+                    setLastToolOutputTime(Date.now());
+                }
+                return adapted;
+            });
+        };
+        messageBus.subscribe(MessageBusType.TOOL_CALLS_UPDATE, handler);
+        return () => {
+            messageBus.unsubscribe(MessageBusType.TOOL_CALLS_UPDATE, handler);
+        };
+    }, [messageBus, internalAdaptToolCalls]);
+    const schedule = useCallback(async (request, signal) => {
+        // Clear state for new run
+        setToolCalls([]);
+        // 1. Await Core Scheduler directly
+        const results = await scheduler.schedule(request, signal);
+        // 2. Trigger legacy reinjection logic (useGeminiStream loop)
+        await onCompleteRef.current(results);
+        return results;
+    }, [scheduler]);
+    const cancelAll = useCallback((_signal) => {
+        scheduler.cancelAll();
+    }, [scheduler]);
+    const markToolsAsSubmitted = useCallback((callIdsToMark) => {
+        setToolCalls((prevCalls) => prevCalls.map((tc) => callIdsToMark.includes(tc.request.callId)
+            ? { ...tc, responseSubmittedToGemini: true }
+            : tc));
+    }, []);
+    return [
+        toolCalls,
+        schedule,
+        markToolsAsSubmitted,
+        setToolCalls,
+        cancelAll,
+        lastToolOutputTime,
+    ];
+}
+/**
+ * ADAPTER: Merges UI metadata (submitted flag) and injects legacy callbacks.
+ */
+function adaptToolCalls(coreCalls, prevTracked, messageBus) {
+    const prevMap = new Map(prevTracked.map((t) => [t.request.callId, t]));
+    return coreCalls.map((coreCall) => {
+        const prev = prevMap.get(coreCall.request.callId);
+        const responseSubmittedToGemini = prev?.responseSubmittedToGemini ?? false;
+        // Inject onConfirm adapter for tools awaiting approval.
+        // The Core provides data-only (serializable) confirmationDetails. We must
+        // inject the legacy callback function that proxies responses back to the
+        // MessageBus.
+        if (coreCall.status === 'awaiting_approval' && coreCall.correlationId) {
+            const correlationId = coreCall.correlationId;
+            return {
+                ...coreCall,
+                confirmationDetails: {
+                    ...coreCall.confirmationDetails,
+                    onConfirm: async (outcome, payload) => {
+                        await messageBus.publish({
+                            type: MessageBusType.TOOL_CONFIRMATION_RESPONSE,
+                            correlationId,
+                            confirmed: outcome !== ToolConfirmationOutcome.Cancel,
+                            requiresUserConfirmation: false,
+                            outcome,
+                            payload,
+                        });
+                    },
+                },
+                responseSubmittedToGemini,
+            };
+        }
+        return {
+            ...coreCall,
+            responseSubmittedToGemini,
+        };
+    });
+}
+//# sourceMappingURL=useToolExecutionScheduler.js.map

package/dist/src/ui/hooks/useToolExecutionScheduler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"useToolExecutionScheduler.js","sourceRoot":"","sources":["../../../../src/ui/hooks/useToolExecutionScheduler.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAOL,cAAc,EACd,uBAAuB,EACvB,SAAS,GAGV,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,WAAW,EAAE,QAAQ,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,OAAO,CAAC;AAmB1E;;;;;GAKG;AACH,MAAM,UAAU,yBAAyB,CACvC,UAAyD,EACzD,MAAc,EACd,kBAAgD;IAShD,iDAAiD;IACjD,MAAM,CAAC,SAAS,EAAE,YAAY,CAAC,GAAG,QAAQ,CAAoB,EAAE,CAAC,CAAC;IAClE,MAAM,CAAC,kBAAkB,EAAE,qBAAqB,CAAC,GAAG,QAAQ,CAAS,CAAC,CAAC,CAAC;IAExE,MAAM,UAAU,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,aAAa,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC;IAEnE,MAAM,aAAa,GAAG,MAAM,CAAC,UAAU,CAAC,CAAC;IACzC,SAAS,CAAC,GAAG,EAAE;QACb,aAAa,CAAC,OAAO,GAAG,UAAU,CAAC;IACrC,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC,CAAC;IAEjB,MAAM,qBAAqB,GAAG,MAAM,CAAC,kBAAkB,CAAC,CAAC;IACzD,SAAS,CAAC,GAAG,EAAE;QACb,qBAAqB,CAAC,OAAO,GAAG,kBAAkB,CAAC;IACrD,CAAC,EAAE,CAAC,kBAAkB,CAAC,CAAC,CAAC;IAEzB,MAAM,SAAS,GAAG,OAAO,CACvB,GAAG,EAAE,CACH,IAAI,SAAS,CAAC;QACZ,MAAM;QACN,UAAU;QACV,kBAAkB,EAAE,GAAG,EAAE,CAAC,qBAAqB,CAAC,OAAO,EAAE;KAC1D,CAAC,EACJ,CAAC,MAAM,EAAE,UAAU,CAAC,CACrB,CAAC;IAEF,MAAM,sBAAsB,GAAG,WAAW,CACxC,CAAC,SAAqB,EAAE,WAA8B,EAAE,EAAE,CACxD,cAAc,CAAC,SAAS,EAAE,WAAW,EAAE,UAAU,CAAC,EACpD,CAAC,UAAU,CAAC,CACb,CAAC;IAEF,SAAS,CAAC,GAAG,EAAE;QACb,MAAM,OAAO,GAAG,CAAC,KAA6B,EAAE,EAAE;YAChD,YAAY,CAAC,CAAC,IAAI,EAAE,EAAE;gBACpB,MAAM,OAAO,GAAG,sBAAsB,CAAC,KAAK,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;gBAE9D,sCAAsC;gBACtC,IAAI,KAAK,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,MAAM,KAAK,WAAW,CAAC,EAAE,CAAC;oBAC5D,qBAAqB,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;gBACpC,CAAC;gBAED,OAAO,OAAO,CAAC;YACjB,CAAC,CAAC,CAAC;QACL,CAAC,CAAC;QAEF,UAAU,CAAC,SAAS,CAAC,cAAc,CAAC,iBAAiB,EAAE,OAAO,CAAC,CAAC;QAChE,OAAO,GAAG,EAAE;YACV,UAAU,CAAC,WAAW,CAAC,cAAc,CAAC,iBAAiB,EAAE,OAAO,CAAC,CAAC;QACpE,CAAC,CAAC;IACJ,CAAC,EAAE,CAAC,UAAU,EAAE,sBAAsB,CAAC,CAAC,CAAC;IAEzC,MAAM,QAAQ,GAAe,WAAW,CACtC,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE;QACxB,0BAA0B;QAC1B,YAAY,CAAC,EAAE,CAAC,CAAC;QAEjB,mCAAmC;QACnC,MAAM,OAAO,GAAG,MAAM,SAAS,CAAC,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAE1D,6DAA6D;QAC7D,MAAM,aAAa,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAErC,OAAO,OAAO,CAAC;IACjB,CAAC,EACD,CAAC,SAAS,CAAC,CACZ,CAAC;IAEF,MAAM,SAAS,GAAgB,WAAW,CACxC,CAAC,OAAO,EAAE,EAAE;QACV,SAAS,CAAC,SAAS,EAAE,CAAC;IACxB,CAAC,EACD,CAAC,SAAS,CAAC,CACZ,CAAC;IAEF,MAAM,oBAAoB,GAA2B,WAAW,CAC9D,CAAC,aAAuB,EAAE,EAAE;QAC1B,YAAY,CAAC,CAAC,SAAS,EAAE,EAAE,CACzB,SAAS,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CACnB,aAAa,CAAC,QAAQ,CAAC,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC;YACvC,CAAC,CAAC,EAAE,GAAG,EAAE,EAAE,yBAAyB,EAAE,IAAI,EAAE;YAC5C,CAAC,CAAC,EAAE,CACP,CACF,CAAC;IACJ,CAAC,EACD,EAAE,CACH,CAAC;IAEF,OAAO;QACL,SAAS;QACT,QAAQ;QACR,oBAAoB;QACpB,YAAY;QACZ,SAAS;QACT,kBAAkB;KACnB,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,SAAS,cAAc,CACrB,SAAqB,EACrB,WAA8B,EAC9B,UAAsB;IAEtB,MAAM,OAAO,GAAG,IAAI,GAAG,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IAEvE,OAAO,SAAS,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAmB,EAAE;QACjD,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAClD,MAAM,yBAAyB,GAAG,IAAI,EAAE,yBAAyB,IAAI,KAAK,CAAC;QAE3E,wDAAwD;QACxD,0EAA0E;QAC1E,yEAAyE;QACzE,cAAc;QACd,IAAI,QAAQ,CAAC,MAAM,KAAK,mBAAmB,IAAI,QAAQ,CAAC,aAAa,EAAE,CAAC;YACtE,MAAM,aAAa,GAAG,QAAQ,CAAC,aAAa,CAAC;YAC7C,OAAO;gBACL,GAAG,QAAQ;gBACX,mBAAmB,EAAE;oBACnB,GAAG,QAAQ,CAAC,mBAAmB;oBAC/B,SAAS,EAAE,KAAK,EACd,OAAgC,EAChC,OAAiC,EACjC,EAAE;wBACF,MAAM,UAAU,CAAC,OAAO,CAAC;4BACvB,IAAI,EAAE,cAAc,CAAC,0BAA0B;4BAC/C,aAAa;4BACb,SAAS,EAAE,OAAO,KAAK,uBAAuB,CAAC,MAAM;4BACrD,wBAAwB,EAAE,KAAK;4BAC/B,OAAO;4BACP,OAAO;yBACR,CAAC,CAAC;oBACL,CAAC;iBACF;gBACD,yBAAyB;aAC1B,CAAC;QACJ,CAAC;QAED,OAAO;YACL,GAAG,QAAQ;YACX,yBAAyB;SAC1B,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC"}

package/dist/src/ui/hooks/useToolExecutionScheduler.test.d.ts

@@ -0,0 +1,6 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export {};

package/dist/src/ui/hooks/useToolExecutionScheduler.test.js

@@ -0,0 +1,296 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach } from 'vitest';
+import { act } from 'react';
+import { renderHook } from '../../test-utils/render.js';
+import { useToolExecutionScheduler } from './useToolExecutionScheduler.js';
+import { MessageBusType, ToolConfirmationOutcome, Scheduler, } from '@google/gemini-cli-core';
+import { createMockMessageBus } from '@google/gemini-cli-core/src/test-utils/mock-message-bus.js';
+// Mock Core Scheduler
+vi.mock('@google/gemini-cli-core', async (importOriginal) => {
+    const actual = await importOriginal();
+    return {
+        ...actual,
+        Scheduler: vi.fn().mockImplementation(() => ({
+            schedule: vi.fn().mockResolvedValue([]),
+            cancelAll: vi.fn(),
+        })),
+    };
+});
+const createMockTool = (overrides = {}) => ({
+    name: 'test_tool',
+    displayName: 'Test Tool',
+    description: 'A test tool',
+    kind: 'function',
+    parameterSchema: {},
+    isOutputMarkdown: false,
+    build: vi.fn(),
+    ...overrides,
+});
+const createMockInvocation = (overrides = {}) => ({
+    getDescription: () => 'Executing test tool',
+    shouldConfirmExecute: vi.fn(),
+    execute: vi.fn(),
+    params: {},
+    toolLocations: [],
+    ...overrides,
+});
+describe('useToolExecutionScheduler', () => {
+    let mockConfig;
+    let mockMessageBus;
+    beforeEach(() => {
+        vi.clearAllMocks();
+        mockMessageBus = createMockMessageBus();
+        mockConfig = {
+            getMessageBus: () => mockMessageBus,
+        };
+    });
+    it('initializes with empty tool calls', () => {
+        const { result } = renderHook(() => useToolExecutionScheduler(vi.fn().mockResolvedValue(undefined), mockConfig, () => undefined));
+        const [toolCalls] = result.current;
+        expect(toolCalls).toEqual([]);
+    });
+    it('updates tool calls when MessageBus emits TOOL_CALLS_UPDATE', () => {
+        const { result } = renderHook(() => useToolExecutionScheduler(vi.fn().mockResolvedValue(undefined), mockConfig, () => undefined));
+        const mockToolCall = {
+            status: 'executing',
+            request: {
+                callId: 'call-1',
+                name: 'test_tool',
+                args: {},
+                isClientInitiated: false,
+                prompt_id: 'p1',
+            },
+            tool: createMockTool(),
+            invocation: createMockInvocation(),
+            liveOutput: 'Loading...',
+        };
+        act(() => {
+            void mockMessageBus.publish({
+                type: MessageBusType.TOOL_CALLS_UPDATE,
+                toolCalls: [mockToolCall],
+            });
+        });
+        const [toolCalls] = result.current;
+        expect(toolCalls).toHaveLength(1);
+        // Expect Core Object structure, not Display Object
+        expect(toolCalls[0]).toMatchObject({
+            request: { callId: 'call-1', name: 'test_tool' },
+            status: 'executing', // Core status
+            liveOutput: 'Loading...',
+            responseSubmittedToGemini: false,
+        });
+    });
+    it('injects onConfirm callback for awaiting_approval tools (Adapter Pattern)', async () => {
+        const { result } = renderHook(() => useToolExecutionScheduler(vi.fn().mockResolvedValue(undefined), mockConfig, () => undefined));
+        const mockToolCall = {
+            status: 'awaiting_approval',
+            request: {
+                callId: 'call-1',
+                name: 'test_tool',
+                args: {},
+                isClientInitiated: false,
+                prompt_id: 'p1',
+            },
+            tool: createMockTool(),
+            invocation: createMockInvocation({
+                getDescription: () => 'Confirming test tool',
+            }),
+            confirmationDetails: { type: 'info', title: 'Confirm', prompt: 'Sure?' },
+            correlationId: 'corr-123',
+        };
+        act(() => {
+            void mockMessageBus.publish({
+                type: MessageBusType.TOOL_CALLS_UPDATE,
+                toolCalls: [mockToolCall],
+            });
+        });
+        const [toolCalls] = result.current;
+        const call = toolCalls[0];
+        if (call.status !== 'awaiting_approval') {
+            throw new Error('Expected status to be awaiting_approval');
+        }
+        const confirmationDetails = call.confirmationDetails;
+        expect(confirmationDetails).toBeDefined();
+        expect(typeof confirmationDetails.onConfirm).toBe('function');
+        // Test that onConfirm publishes to MessageBus
+        const publishSpy = vi.spyOn(mockMessageBus, 'publish');
+        await confirmationDetails.onConfirm(ToolConfirmationOutcome.ProceedOnce);
+        expect(publishSpy).toHaveBeenCalledWith({
+            type: MessageBusType.TOOL_CONFIRMATION_RESPONSE,
+            correlationId: 'corr-123',
+            confirmed: true,
+            requiresUserConfirmation: false,
+            outcome: ToolConfirmationOutcome.ProceedOnce,
+            payload: undefined,
+        });
+    });
+    it('injects onConfirm with payload (Inline Edit support)', async () => {
+        const { result } = renderHook(() => useToolExecutionScheduler(vi.fn().mockResolvedValue(undefined), mockConfig, () => undefined));
+        const mockToolCall = {
+            status: 'awaiting_approval',
+            request: {
+                callId: 'call-1',
+                name: 'test_tool',
+                args: {},
+                isClientInitiated: false,
+                prompt_id: 'p1',
+            },
+            tool: createMockTool(),
+            invocation: createMockInvocation(),
+            confirmationDetails: { type: 'edit', title: 'Edit', filePath: 'test.ts' },
+            correlationId: 'corr-edit',
+        };
+        act(() => {
+            void mockMessageBus.publish({
+                type: MessageBusType.TOOL_CALLS_UPDATE,
+                toolCalls: [mockToolCall],
+            });
+        });
+        const [toolCalls] = result.current;
+        const call = toolCalls[0];
+        if (call.status !== 'awaiting_approval') {
+            throw new Error('Expected awaiting_approval');
+        }
+        const confirmationDetails = call.confirmationDetails;
+        const publishSpy = vi.spyOn(mockMessageBus, 'publish');
+        const mockPayload = { newContent: 'updated code' };
+        await confirmationDetails.onConfirm(ToolConfirmationOutcome.ProceedOnce, mockPayload);
+        expect(publishSpy).toHaveBeenCalledWith({
+            type: MessageBusType.TOOL_CONFIRMATION_RESPONSE,
+            correlationId: 'corr-edit',
+            confirmed: true,
+            requiresUserConfirmation: false,
+            outcome: ToolConfirmationOutcome.ProceedOnce,
+            payload: mockPayload,
+        });
+    });
+    it('preserves responseSubmittedToGemini flag across updates', () => {
+        const { result } = renderHook(() => useToolExecutionScheduler(vi.fn().mockResolvedValue(undefined), mockConfig, () => undefined));
+        const mockToolCall = {
+            status: 'success',
+            request: {
+                callId: 'call-1',
+                name: 'test',
+                args: {},
+                isClientInitiated: false,
+                prompt_id: 'p1',
+            },
+            tool: createMockTool(),
+            invocation: createMockInvocation(),
+            response: {
+                callId: 'call-1',
+                resultDisplay: 'OK',
+                responseParts: [],
+                error: undefined,
+                errorType: undefined,
+            },
+        };
+        // 1. Initial success
+        act(() => {
+            void mockMessageBus.publish({
+                type: MessageBusType.TOOL_CALLS_UPDATE,
+                toolCalls: [mockToolCall],
+            });
+        });
+        // 2. Mark as submitted
+        act(() => {
+            const [, , markAsSubmitted] = result.current;
+            markAsSubmitted(['call-1']);
+        });
+        expect(result.current[0][0].responseSubmittedToGemini).toBe(true);
+        // 3. Receive another update (should preserve the true flag)
+        act(() => {
+            void mockMessageBus.publish({
+                type: MessageBusType.TOOL_CALLS_UPDATE,
+                toolCalls: [mockToolCall],
+            });
+        });
+        expect(result.current[0][0].responseSubmittedToGemini).toBe(true);
+    });
+    it('updates lastToolOutputTime when tools are executing', () => {
+        vi.useFakeTimers();
+        const { result } = renderHook(() => useToolExecutionScheduler(vi.fn().mockResolvedValue(undefined), mockConfig, () => undefined));
+        const startTime = Date.now();
+        vi.advanceTimersByTime(1000);
+        act(() => {
+            void mockMessageBus.publish({
+                type: MessageBusType.TOOL_CALLS_UPDATE,
+                toolCalls: [
+                    {
+                        status: 'executing',
+                        request: {
+                            callId: 'call-1',
+                            name: 'test',
+                            args: {},
+                            isClientInitiated: false,
+                            prompt_id: 'p1',
+                        },
+                        tool: createMockTool(),
+                        invocation: createMockInvocation(),
+                    },
+                ],
+            });
+        });
+        const [, , , , , lastOutputTime] = result.current;
+        expect(lastOutputTime).toBeGreaterThan(startTime);
+        vi.useRealTimers();
+    });
+    it('delegates cancelAll to the Core Scheduler', () => {
+        const { result } = renderHook(() => useToolExecutionScheduler(vi.fn().mockResolvedValue(undefined), mockConfig, () => undefined));
+        const [, , , , cancelAll] = result.current;
+        const signal = new AbortController().signal;
+        // We need to find the mock instance of Scheduler
+        // Since we used vi.mock at top level, we can get it from vi.mocked(Scheduler)
+        const schedulerInstance = vi.mocked(Scheduler).mock.results[0].value;
+        cancelAll(signal);
+        expect(schedulerInstance.cancelAll).toHaveBeenCalled();
+    });
+    it('resolves the schedule promise when scheduler resolves', async () => {
+        const onComplete = vi.fn().mockResolvedValue(undefined);
+        const completedToolCall = {
+            status: 'success',
+            request: {
+                callId: 'call-1',
+                name: 'test',
+                args: {},
+                isClientInitiated: false,
+                prompt_id: 'p1',
+            },
+            tool: createMockTool(),
+            invocation: createMockInvocation(),
+            response: {
+                callId: 'call-1',
+                responseParts: [],
+                resultDisplay: 'Success',
+                error: undefined,
+                errorType: undefined,
+            },
+        };
+        // Mock the specific return value for this test
+        const { Scheduler } = await import('@google/gemini-cli-core');
+        vi.mocked(Scheduler).mockImplementation(() => ({
+            schedule: vi.fn().mockResolvedValue([completedToolCall]),
+            cancelAll: vi.fn(),
+        }));
+        const { result } = renderHook(() => useToolExecutionScheduler(onComplete, mockConfig, () => undefined));
+        const [, schedule] = result.current;
+        const signal = new AbortController().signal;
+        let completedResult = [];
+        await act(async () => {
+            completedResult = await schedule({
+                callId: 'call-1',
+                name: 'test',
+                args: {},
+                isClientInitiated: false,
+                prompt_id: 'p1',
+            }, signal);
+        });
+        expect(completedResult).toEqual([completedToolCall]);
+        expect(onComplete).toHaveBeenCalledWith([completedToolCall]);
+    });
+});
+//# sourceMappingURL=useToolExecutionScheduler.test.js.map

package/dist/src/ui/hooks/useToolExecutionScheduler.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"useToolExecutionScheduler.test.js","sourceRoot":"","sources":["../../../../src/ui/hooks/useToolExecutionScheduler.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,GAAG,EAAE,MAAM,OAAO,CAAC;AAC5B,OAAO,EAAE,UAAU,EAAE,MAAM,4BAA4B,CAAC;AACxD,OAAO,EAAE,yBAAyB,EAAE,MAAM,gCAAgC,CAAC;AAC3E,OAAO,EACL,cAAc,EACd,uBAAuB,EACvB,SAAS,GAQV,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,oBAAoB,EAAE,MAAM,4DAA4D,CAAC;AAElG,sBAAsB;AACtB,EAAE,CAAC,IAAI,CAAC,yBAAyB,EAAE,KAAK,EAAE,cAAc,EAAE,EAAE;IAC1D,MAAM,MAAM,GACV,MAAM,cAAc,EAA4C,CAAC;IACnE,OAAO;QACL,GAAG,MAAM;QACT,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,CAAC;YAC3C,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;YACvC,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE;SACnB,CAAC,CAAC;KACJ,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,MAAM,cAAc,GAAG,CACrB,YAAyC,EAAE,EACvB,EAAE,CACtB,CAAC;IACC,IAAI,EAAE,WAAW;IACjB,WAAW,EAAE,WAAW;IACxB,WAAW,EAAE,aAAa;IAC1B,IAAI,EAAE,UAAU;IAChB,eAAe,EAAE,EAAE;IACnB,gBAAgB,EAAE,KAAK;IACvB,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;IACd,GAAG,SAAS;CACb,CAAuB,CAAC;AAE3B,MAAM,oBAAoB,GAAG,CAC3B,YAAwC,EAAE,EACvB,EAAE,CACrB,CAAC;IACC,cAAc,EAAE,GAAG,EAAE,CAAC,qBAAqB;IAC3C,oBAAoB,EAAE,EAAE,CAAC,EAAE,EAAE;IAC7B,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;IAChB,MAAM,EAAE,EAAE;IACV,aAAa,EAAE,EAAE;IACjB,GAAG,SAAS;CACb,CAAsB,CAAC;AAE1B,QAAQ,CAAC,2BAA2B,EAAE,GAAG,EAAE;IACzC,IAAI,UAAkB,CAAC;IACvB,IAAI,cAA0B,CAAC;IAE/B,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,cAAc,GAAG,oBAAoB,EAA2B,CAAC;QACjE,UAAU,GAAG;YACX,aAAa,EAAE,GAAG,EAAE,CAAC,cAAc;SACf,CAAC;IACzB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;QAC3C,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,yBAAyB,CACvB,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC,EACpC,UAAU,EACV,GAAG,EAAE,CAAC,SAAS,CAChB,CACF,CAAC;QACF,MAAM,CAAC,SAAS,CAAC,GAAG,MAAM,CAAC,OAAO,CAAC;QACnC,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAChC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;QACpE,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,yBAAyB,CACvB,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC,EACpC,UAAU,EACV,GAAG,EAAE,CAAC,SAAS,CAChB,CACF,CAAC;QAEF,MAAM,YAAY,GAAG;YACnB,MAAM,EAAE,WAAoB;YAC5B,OAAO,EAAE;gBACP,MAAM,EAAE,QAAQ;gBAChB,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,EAAE;gBACR,iBAAiB,EAAE,KAAK;gBACxB,SAAS,EAAE,IAAI;aAChB;YACD,IAAI,EAAE,cAAc,EAAE;YACtB,UAAU,EAAE,oBAAoB,EAAE;YAClC,UAAU,EAAE,YAAY;SACzB,CAAC;QAEF,GAAG,CAAC,GAAG,EAAE;YACP,KAAK,cAAc,CAAC,OAAO,CAAC;gBAC1B,IAAI,EAAE,cAAc,CAAC,iBAAiB;gBACtC,SAAS,EAAE,CAAC,YAAY,CAAC;aACA,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,SAAS,CAAC,GAAG,MAAM,CAAC,OAAO,CAAC;QACnC,MAAM,CAAC,SAAS,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAClC,mDAAmD;QACnD,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC;YACjC,OAAO,EAAE,EAAE,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,WAAW,EAAE;YAChD,MAAM,EAAE,WAAW,EAAE,cAAc;YACnC,UAAU,EAAE,YAAY;YACxB,yBAAyB,EAAE,KAAK;SACjC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0EAA0E,EAAE,KAAK,IAAI,EAAE;QACxF,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,yBAAyB,CACvB,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC,EACpC,UAAU,EACV,GAAG,EAAE,CAAC,SAAS,CAChB,CACF,CAAC;QAEF,MAAM,YAAY,GAAG;YACnB,MAAM,EAAE,mBAA4B;YACpC,OAAO,EAAE;gBACP,MAAM,EAAE,QAAQ;gBAChB,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,EAAE;gBACR,iBAAiB,EAAE,KAAK;gBACxB,SAAS,EAAE,IAAI;aAChB;YACD,IAAI,EAAE,cAAc,EAAE;YACtB,UAAU,EAAE,oBAAoB,CAAC;gBAC/B,cAAc,EAAE,GAAG,EAAE,CAAC,sBAAsB;aAC7C,CAAC;YACF,mBAAmB,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE;YACxE,aAAa,EAAE,UAAU;SAC1B,CAAC;QAEF,GAAG,CAAC,GAAG,EAAE;YACP,KAAK,cAAc,CAAC,OAAO,CAAC;gBAC1B,IAAI,EAAE,cAAc,CAAC,iBAAiB;gBACtC,SAAS,EAAE,CAAC,YAAY,CAAC;aACA,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,SAAS,CAAC,GAAG,MAAM,CAAC,OAAO,CAAC;QACnC,MAAM,IAAI,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC;QAC1B,IAAI,IAAI,CAAC,MAAM,KAAK,mBAAmB,EAAE,CAAC;YACxC,MAAM,IAAI,KAAK,CAAC,yCAAyC,CAAC,CAAC;QAC7D,CAAC;QACD,MAAM,mBAAmB,GACvB,IAAI,CAAC,mBAAkD,CAAC;QAE1D,MAAM,CAAC,mBAAmB,CAAC,CAAC,WAAW,EAAE,CAAC;QAC1C,MAAM,CAAC,OAAO,mBAAmB,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAE9D,8CAA8C;QAC9C,MAAM,UAAU,GAAG,EAAE,CAAC,KAAK,CAAC,cAAc,EAAE,SAAS,CAAC,CAAC;QACvD,MAAM,mBAAmB,CAAC,SAAS,CAAC,uBAAuB,CAAC,WAAW,CAAC,CAAC;QAEzE,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC;YACtC,IAAI,EAAE,cAAc,CAAC,0BAA0B;YAC/C,aAAa,EAAE,UAAU;YACzB,SAAS,EAAE,IAAI;YACf,wBAAwB,EAAE,KAAK;YAC/B,OAAO,EAAE,uBAAuB,CAAC,WAAW;YAC5C,OAAO,EAAE,SAAS;SACnB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;QACpE,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,yBAAyB,CACvB,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC,EACpC,UAAU,EACV,GAAG,EAAE,CAAC,SAAS,CAChB,CACF,CAAC;QAEF,MAAM,YAAY,GAAG;YACnB,MAAM,EAAE,mBAA4B;YACpC,OAAO,EAAE;gBACP,MAAM,EAAE,QAAQ;gBAChB,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,EAAE;gBACR,iBAAiB,EAAE,KAAK;gBACxB,SAAS,EAAE,IAAI;aAChB;YACD,IAAI,EAAE,cAAc,EAAE;YACtB,UAAU,EAAE,oBAAoB,EAAE;YAClC,mBAAmB,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE;YACzE,aAAa,EAAE,WAAW;SAC3B,CAAC;QAEF,GAAG,CAAC,GAAG,EAAE;YACP,KAAK,cAAc,CAAC,OAAO,CAAC;gBAC1B,IAAI,EAAE,cAAc,CAAC,iBAAiB;gBACtC,SAAS,EAAE,CAAC,YAAY,CAAC;aACA,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,SAAS,CAAC,GAAG,MAAM,CAAC,OAAO,CAAC;QACnC,MAAM,IAAI,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC;QAC1B,IAAI,IAAI,CAAC,MAAM,KAAK,mBAAmB,EAAE,CAAC;YACxC,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;QAChD,CAAC;QACD,MAAM,mBAAmB,GACvB,IAAI,CAAC,mBAAkD,CAAC;QAE1D,MAAM,UAAU,GAAG,EAAE,CAAC,KAAK,CAAC,cAAc,EAAE,SAAS,CAAC,CAAC;QACvD,MAAM,WAAW,GAAG,EAAE,UAAU,EAAE,cAAc,EAAE,CAAC;QACnD,MAAM,mBAAmB,CAAC,SAAS,CACjC,uBAAuB,CAAC,WAAW,EACnC,WAAW,CACZ,CAAC;QAEF,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC;YACtC,IAAI,EAAE,cAAc,CAAC,0BAA0B;YAC/C,aAAa,EAAE,WAAW;YAC1B,SAAS,EAAE,IAAI;YACf,wBAAwB,EAAE,KAAK;YAC/B,OAAO,EAAE,uBAAuB,CAAC,WAAW;YAC5C,OAAO,EAAE,WAAW;SACrB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;QACjE,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,yBAAyB,CACvB,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC,EACpC,UAAU,EACV,GAAG,EAAE,CAAC,SAAS,CAChB,CACF,CAAC;QAEF,MAAM,YAAY,GAAG;YACnB,MAAM,EAAE,SAAkB;YAC1B,OAAO,EAAE;gBACP,MAAM,EAAE,QAAQ;gBAChB,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,EAAE;gBACR,iBAAiB,EAAE,KAAK;gBACxB,SAAS,EAAE,IAAI;aAChB;YACD,IAAI,EAAE,cAAc,EAAE;YACtB,UAAU,EAAE,oBAAoB,EAAE;YAClC,QAAQ,EAAE;gBACR,MAAM,EAAE,QAAQ;gBAChB,aAAa,EAAE,IAAI;gBACnB,aAAa,EAAE,EAAE;gBACjB,KAAK,EAAE,SAAS;gBAChB,SAAS,EAAE,SAAS;aACrB;SACF,CAAC;QAEF,qBAAqB;QACrB,GAAG,CAAC,GAAG,EAAE;YACP,KAAK,cAAc,CAAC,OAAO,CAAC;gBAC1B,IAAI,EAAE,cAAc,CAAC,iBAAiB;gBACtC,SAAS,EAAE,CAAC,YAAY,CAAC;aACA,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC;QAEH,uBAAuB;QACvB,GAAG,CAAC,GAAG,EAAE;YACP,MAAM,CAAC,EAAE,AAAD,EAAG,eAAe,CAAC,GAAG,MAAM,CAAC,OAAO,CAAC;YAC7C,eAAe,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC;QAC9B,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,yBAAyB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAElE,4DAA4D;QAC5D,GAAG,CAAC,GAAG,EAAE;YACP,KAAK,cAAc,CAAC,OAAO,CAAC;gBAC1B,IAAI,EAAE,cAAc,CAAC,iBAAiB;gBACtC,SAAS,EAAE,CAAC,YAAY,CAAC;aACA,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,yBAAyB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACpE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,yBAAyB,CACvB,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC,EACpC,UAAU,EACV,GAAG,EAAE,CAAC,SAAS,CAChB,CACF,CAAC;QAEF,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,EAAE,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC;QAE7B,GAAG,CAAC,GAAG,EAAE;YACP,KAAK,cAAc,CAAC,OAAO,CAAC;gBAC1B,IAAI,EAAE,cAAc,CAAC,iBAAiB;gBACtC,SAAS,EAAE;oBACT;wBACE,MAAM,EAAE,WAAoB;wBAC5B,OAAO,EAAE;4BACP,MAAM,EAAE,QAAQ;4BAChB,IAAI,EAAE,MAAM;4BACZ,IAAI,EAAE,EAAE;4BACR,iBAAiB,EAAE,KAAK;4BACxB,SAAS,EAAE,IAAI;yBAChB;wBACD,IAAI,EAAE,cAAc,EAAE;wBACtB,UAAU,EAAE,oBAAoB,EAAE;qBACnC;iBACF;aACwB,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,EAAE,AAAD,EAAG,AAAD,EAAG,AAAD,EAAG,AAAD,EAAG,cAAc,CAAC,GAAG,MAAM,CAAC,OAAO,CAAC;QAClD,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC;QAClD,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACnD,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,yBAAyB,CACvB,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC,EACpC,UAAU,EACV,GAAG,EAAE,CAAC,SAAS,CAChB,CACF,CAAC;QAEF,MAAM,CAAC,EAAE,AAAD,EAAG,AAAD,EAAG,AAAD,EAAG,SAAS,CAAC,GAAG,MAAM,CAAC,OAAO,CAAC;QAC3C,MAAM,MAAM,GAAG,IAAI,eAAe,EAAE,CAAC,MAAM,CAAC;QAE5C,iDAAiD;QACjD,8EAA8E;QAC9E,MAAM,iBAAiB,GAAG,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;QAErE,SAAS,CAAC,MAAM,CAAC,CAAC;QAElB,MAAM,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC,gBAAgB,EAAE,CAAC;IACzD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;QACrE,MAAM,UAAU,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAExD,MAAM,iBAAiB,GAAG;YACxB,MAAM,EAAE,SAAkB;YAC1B,OAAO,EAAE;gBACP,MAAM,EAAE,QAAQ;gBAChB,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,EAAE;gBACR,iBAAiB,EAAE,KAAK;gBACxB,SAAS,EAAE,IAAI;aAChB;YACD,IAAI,EAAE,cAAc,EAAE;YACtB,UAAU,EAAE,oBAAoB,EAAE;YAClC,QAAQ,EAAE;gBACR,MAAM,EAAE,QAAQ;gBAChB,aAAa,EAAE,EAAE;gBACjB,aAAa,EAAE,SAAS;gBACxB,KAAK,EAAE,SAAS;gBAChB,SAAS,EAAE,SAAS;aACrB;SACF,CAAC;QAEF,+CAA+C;QAC/C,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,MAAM,CAAC,yBAAyB,CAAC,CAAC;QAC9D,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,kBAAkB,CACrC,GAAG,EAAE,CACH,CAAC;YACC,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,iBAAiB,CAAC,CAAC;YACxD,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE;SACnB,CAAyB,CAC7B,CAAC;QAEF,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,yBAAyB,CAAC,UAAU,EAAE,UAAU,EAAE,GAAG,EAAE,CAAC,SAAS,CAAC,CACnE,CAAC;QAEF,MAAM,CAAC,EAAE,QAAQ,CAAC,GAAG,MAAM,CAAC,OAAO,CAAC;QACpC,MAAM,MAAM,GAAG,IAAI,eAAe,EAAE,CAAC,MAAM,CAAC;QAE5C,IAAI,eAAe,GAAwB,EAAE,CAAC;QAC9C,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,eAAe,GAAG,MAAM,QAAQ,CAC9B;gBACE,MAAM,EAAE,QAAQ;gBAChB,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,EAAE;gBACR,iBAAiB,EAAE,KAAK;gBACxB,SAAS,EAAE,IAAI;aAChB,EACD,MAAM,CACP,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,eAAe,CAAC,CAAC,OAAO,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC;QACrD,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC;IAC/D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/ui/hooks/useToolScheduler.d.ts

@@ -0,0 +1,29 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { Config, EditorType, CompletedToolCall, ToolCallRequestInfo } from '@google/gemini-cli-core';
+import { type TrackedToolCall as LegacyTrackedToolCall, type TrackedScheduledToolCall, type TrackedValidatingToolCall, type TrackedWaitingToolCall, type TrackedExecutingToolCall, type TrackedCompletedToolCall, type TrackedCancelledToolCall, type MarkToolsAsSubmittedFn, type CancelAllFn } from './useReactToolScheduler.js';
+import { type TrackedToolCall as NewTrackedToolCall } from './useToolExecutionScheduler.js';
+export type { TrackedScheduledToolCall, TrackedValidatingToolCall, TrackedWaitingToolCall, TrackedExecutingToolCall, TrackedCompletedToolCall, TrackedCancelledToolCall, };
+export type TrackedToolCall = LegacyTrackedToolCall | NewTrackedToolCall;
+export type ScheduleFn = (request: ToolCallRequestInfo | ToolCallRequestInfo[], signal: AbortSignal) => Promise<void | CompletedToolCall[]>;
+export type UseToolSchedulerReturn = [
+    TrackedToolCall[],
+    ScheduleFn,
+    MarkToolsAsSubmittedFn,
+    React.Dispatch<React.SetStateAction<TrackedToolCall[]>>,
+    CancelAllFn,
+    number
+];
+/**
+ * Facade hook that switches between the Legacy and Event-Driven schedulers
+ * based on configuration.
+ *
+ * Note: This conditionally calls hooks, which technically violates the standard
+ * Rules of Hooks linting. However, this is safe here because
+ * `config.isEventDrivenSchedulerEnabled()` is static for the lifetime of the
+ * application session (it essentially acts as a compile-time feature flag).
+ */
+export declare function useToolScheduler(onComplete: (tools: CompletedToolCall[]) => Promise<void>, config: Config, getPreferredEditor: () => EditorType | undefined): UseToolSchedulerReturn;

package/dist/src/ui/hooks/useToolScheduler.js

@@ -0,0 +1,30 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { useReactToolScheduler, } from './useReactToolScheduler.js';
+import { useToolExecutionScheduler, } from './useToolExecutionScheduler.js';
+/**
+ * Facade hook that switches between the Legacy and Event-Driven schedulers
+ * based on configuration.
+ *
+ * Note: This conditionally calls hooks, which technically violates the standard
+ * Rules of Hooks linting. However, this is safe here because
+ * `config.isEventDrivenSchedulerEnabled()` is static for the lifetime of the
+ * application session (it essentially acts as a compile-time feature flag).
+ */
+export function useToolScheduler(onComplete, config, getPreferredEditor) {
+    const isEventDriven = config.isEventDrivenSchedulerEnabled();
+    // Note: We return the hooks directly without casting. They return compatible
+    // tuple structures, but use explicit tuple signatures rather than the
+    // UseToolSchedulerReturn named type to avoid circular dependencies back to
+    // this facade.
+    if (isEventDriven) {
+        // eslint-disable-next-line react-hooks/rules-of-hooks
+        return useToolExecutionScheduler(onComplete, config, getPreferredEditor);
+    }
+    // eslint-disable-next-line react-hooks/rules-of-hooks
+    return useReactToolScheduler(onComplete, config, getPreferredEditor);
+}
+//# sourceMappingURL=useToolScheduler.js.map

package/dist/src/ui/hooks/useToolScheduler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"useToolScheduler.js","sourceRoot":"","sources":["../../../../src/ui/hooks/useToolScheduler.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAQH,OAAO,EACL,qBAAqB,GAUtB,MAAM,4BAA4B,CAAC;AACpC,OAAO,EACL,yBAAyB,GAE1B,MAAM,gCAAgC,CAAC;AA+BxC;;;;;;;;GAQG;AACH,MAAM,UAAU,gBAAgB,CAC9B,UAAyD,EACzD,MAAc,EACd,kBAAgD;IAEhD,MAAM,aAAa,GAAG,MAAM,CAAC,6BAA6B,EAAE,CAAC;IAE7D,6EAA6E;IAC7E,sEAAsE;IACtE,2EAA2E;IAC3E,eAAe;IACf,IAAI,aAAa,EAAE,CAAC;QAClB,sDAAsD;QACtD,OAAO,yBAAyB,CAAC,UAAU,EAAE,MAAM,EAAE,kBAAkB,CAAC,CAAC;IAC3E,CAAC;IAED,sDAAsD;IACtD,OAAO,qBAAqB,CAAC,UAAU,EAAE,MAAM,EAAE,kBAAkB,CAAC,CAAC;AACvE,CAAC"}

package/dist/src/ui/hooks/useToolScheduler.test.js

@@ -702,7 +702,7 @@ describe('mapToDisplay', () => {
             name: 'validating',
             status: 'validating',
             extraProps: { tool: baseTool, invocation: baseInvocation },
-            expectedStatus: ToolCallStatus.Executing,
+            expectedStatus: ToolCallStatus.Pending,
             expectedName: baseTool.displayName,
             expectedDescription: baseInvocation.getDescription(),
         },