@google/gemini-cli-core 0.0.3-preview.4

package/dist/src/core/subagent.test.js

@@ -0,0 +1,520 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { vi, describe, it, expect, beforeEach, afterEach } from 'vitest';
+import { ContextState, SubAgentScope, SubagentTerminateMode, } from './subagent.js';
+import { Config } from '../config/config.js';
+import { GeminiChat } from './geminiChat.js';
+import { createContentGenerator } from './contentGenerator.js';
+import { getEnvironmentContext } from '../utils/environmentContext.js';
+import { executeToolCall } from './nonInteractiveToolExecutor.js';
+import { DEFAULT_GEMINI_MODEL } from '../config/models.js';
+import { Type } from '@google/genai';
+import { ToolErrorType } from '../tools/tool-error.js';
+vi.mock('./geminiChat.js');
+vi.mock('./contentGenerator.js');
+vi.mock('../utils/environmentContext.js');
+vi.mock('./nonInteractiveToolExecutor.js');
+vi.mock('../ide/ide-client.js');
+async function createMockConfig(toolRegistryMocks = {}) {
+    const configParams = {
+        sessionId: 'test-session',
+        model: DEFAULT_GEMINI_MODEL,
+        targetDir: '.',
+        debugMode: false,
+        cwd: process.cwd(),
+    };
+    const config = new Config(configParams);
+    await config.initialize();
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    await config.refreshAuth('test-auth');
+    // Mock ToolRegistry
+    const mockToolRegistry = {
+        getTool: vi.fn(),
+        getFunctionDeclarationsFiltered: vi.fn().mockReturnValue([]),
+        ...toolRegistryMocks,
+    };
+    vi.spyOn(config, 'getToolRegistry').mockReturnValue(mockToolRegistry);
+    return { config, toolRegistry: mockToolRegistry };
+}
+// Helper to simulate LLM responses (sequence of tool calls over multiple turns)
+const createMockStream = (functionCallsList) => {
+    let index = 0;
+    return vi.fn().mockImplementation(() => {
+        const response = functionCallsList[index] || 'stop';
+        index++;
+        return (async function* () {
+            if (response === 'stop') {
+                // When stopping, the model might return text, but the subagent logic primarily cares about the absence of functionCalls.
+                yield { text: 'Done.' };
+            }
+            else if (response.length > 0) {
+                yield { functionCalls: response };
+            }
+            else {
+                yield { text: 'Done.' }; // Handle empty array also as stop
+            }
+        })();
+    });
+};
+describe('subagent.ts', () => {
+    describe('ContextState', () => {
+        it('should set and get values correctly', () => {
+            const context = new ContextState();
+            context.set('key1', 'value1');
+            context.set('key2', 123);
+            expect(context.get('key1')).toBe('value1');
+            expect(context.get('key2')).toBe(123);
+            expect(context.get_keys()).toEqual(['key1', 'key2']);
+        });
+        it('should return undefined for missing keys', () => {
+            const context = new ContextState();
+            expect(context.get('missing')).toBeUndefined();
+        });
+    });
+    describe('SubAgentScope', () => {
+        let mockSendMessageStream;
+        const defaultModelConfig = {
+            model: 'gemini-1.5-flash-latest',
+            temp: 0.5, // Specific temp to test override
+            top_p: 1,
+        };
+        const defaultRunConfig = {
+            max_time_minutes: 5,
+            max_turns: 10,
+        };
+        beforeEach(async () => {
+            vi.clearAllMocks();
+            vi.mocked(getEnvironmentContext).mockResolvedValue([
+                { text: 'Env Context' },
+            ]);
+            vi.mocked(createContentGenerator).mockResolvedValue({
+                getGenerativeModel: vi.fn(),
+                // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            });
+            mockSendMessageStream = vi.fn();
+            // We mock the implementation of the constructor.
+            vi.mocked(GeminiChat).mockImplementation(() => ({
+                sendMessageStream: mockSendMessageStream,
+            }));
+        });
+        afterEach(() => {
+            vi.restoreAllMocks();
+        });
+        // Helper to safely access generationConfig from mock calls
+        const getGenerationConfigFromMock = (callIndex = 0) => {
+            const callArgs = vi.mocked(GeminiChat).mock.calls[callIndex];
+            const generationConfig = callArgs?.[2];
+            // Ensure it's defined before proceeding
+            expect(generationConfig).toBeDefined();
+            if (!generationConfig)
+                throw new Error('generationConfig is undefined');
+            return generationConfig;
+        };
+        describe('create (Tool Validation)', () => {
+            const promptConfig = { systemPrompt: 'Test prompt' };
+            it('should create a SubAgentScope successfully with minimal config', async () => {
+                const { config } = await createMockConfig();
+                const scope = await SubAgentScope.create('test-agent', config, promptConfig, defaultModelConfig, defaultRunConfig);
+                expect(scope).toBeInstanceOf(SubAgentScope);
+            });
+            it('should throw an error if a tool requires confirmation', async () => {
+                const mockTool = {
+                    name: 'risky_tool',
+                    schema: { parametersJsonSchema: { type: 'object', properties: {} } },
+                    build: vi.fn().mockReturnValue({
+                        shouldConfirmExecute: vi.fn().mockResolvedValue({
+                            type: 'exec',
+                            title: 'Confirm',
+                            command: 'rm -rf /',
+                        }),
+                    }),
+                };
+                const { config } = await createMockConfig({
+                    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+                    getTool: vi.fn().mockReturnValue(mockTool),
+                });
+                const toolConfig = { tools: ['risky_tool'] };
+                const options = { toolConfig };
+                await expect(SubAgentScope.create('test-agent', config, promptConfig, defaultModelConfig, defaultRunConfig, options)).rejects.toThrow('Tool "risky_tool" requires user confirmation and cannot be used in a non-interactive subagent.');
+            });
+            it('should succeed if tools do not require confirmation', async () => {
+                const mockTool = {
+                    name: 'safe_tool',
+                    schema: { parametersJsonSchema: { type: 'object', properties: {} } },
+                    build: vi.fn().mockReturnValue({
+                        shouldConfirmExecute: vi.fn().mockResolvedValue(null),
+                    }),
+                };
+                const { config } = await createMockConfig({
+                    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+                    getTool: vi.fn().mockReturnValue(mockTool),
+                });
+                const toolConfig = { tools: ['safe_tool'] };
+                const options = { toolConfig };
+                const scope = await SubAgentScope.create('test-agent', config, promptConfig, defaultModelConfig, defaultRunConfig, options);
+                expect(scope).toBeInstanceOf(SubAgentScope);
+            });
+            it('should skip interactivity check and warn for tools with required parameters', async () => {
+                const consoleWarnSpy = vi
+                    .spyOn(console, 'warn')
+                    .mockImplementation(() => { });
+                const mockToolWithParams = {
+                    name: 'tool_with_params',
+                    schema: {
+                        parametersJsonSchema: {
+                            type: 'object',
+                            properties: {
+                                path: { type: 'string' },
+                            },
+                            required: ['path'],
+                        },
+                    },
+                    // build should not be called, but we mock it to be safe
+                    build: vi.fn(),
+                };
+                const { config } = await createMockConfig({
+                    getTool: vi.fn().mockReturnValue(mockToolWithParams),
+                    getAllTools: vi.fn().mockReturnValue([mockToolWithParams]),
+                });
+                const toolConfig = { tools: ['tool_with_params'] };
+                const options = { toolConfig };
+                // The creation should succeed without throwing
+                const scope = await SubAgentScope.create('test-agent', config, promptConfig, defaultModelConfig, defaultRunConfig, options);
+                expect(scope).toBeInstanceOf(SubAgentScope);
+                // Check that the warning was logged
+                expect(consoleWarnSpy).toHaveBeenCalledWith('Cannot check tool "tool_with_params" for interactivity because it requires parameters. Assuming it is safe for non-interactive use.');
+                // Ensure build was never called
+                expect(mockToolWithParams.build).not.toHaveBeenCalled();
+                consoleWarnSpy.mockRestore();
+            });
+        });
+        describe('runNonInteractive - Initialization and Prompting', () => {
+            it('should correctly template the system prompt and initialize GeminiChat', async () => {
+                const { config } = await createMockConfig();
+                vi.mocked(GeminiChat).mockClear();
+                const promptConfig = {
+                    systemPrompt: 'Hello ${name}, your task is ${task}.',
+                };
+                const context = new ContextState();
+                context.set('name', 'Agent');
+                context.set('task', 'Testing');
+                // Model stops immediately
+                mockSendMessageStream.mockImplementation(createMockStream(['stop']));
+                const scope = await SubAgentScope.create('test-agent', config, promptConfig, defaultModelConfig, defaultRunConfig);
+                await scope.runNonInteractive(context);
+                // Check if GeminiChat was initialized correctly by the subagent
+                expect(GeminiChat).toHaveBeenCalledTimes(1);
+                const callArgs = vi.mocked(GeminiChat).mock.calls[0];
+                // Check Generation Config
+                const generationConfig = getGenerationConfigFromMock();
+                // Check temperature override
+                expect(generationConfig.temperature).toBe(defaultModelConfig.temp);
+                expect(generationConfig.systemInstruction).toContain('Hello Agent, your task is Testing.');
+                expect(generationConfig.systemInstruction).toContain('Important Rules:');
+                // Check History (should include environment context)
+                const history = callArgs[3];
+                expect(history).toEqual([
+                    { role: 'user', parts: [{ text: 'Env Context' }] },
+                    {
+                        role: 'model',
+                        parts: [{ text: 'Got it. Thanks for the context!' }],
+                    },
+                ]);
+            });
+            it('should include output instructions in the system prompt when outputs are defined', async () => {
+                const { config } = await createMockConfig();
+                vi.mocked(GeminiChat).mockClear();
+                const promptConfig = { systemPrompt: 'Do the task.' };
+                const outputConfig = {
+                    outputs: {
+                        result1: 'The first result',
+                    },
+                };
+                const context = new ContextState();
+                // Model stops immediately
+                mockSendMessageStream.mockImplementation(createMockStream(['stop']));
+                const scope = await SubAgentScope.create('test-agent', config, promptConfig, defaultModelConfig, defaultRunConfig, { outputConfig });
+                await scope.runNonInteractive(context);
+                const generationConfig = getGenerationConfigFromMock();
+                const systemInstruction = generationConfig.systemInstruction;
+                expect(systemInstruction).toContain('Do the task.');
+                expect(systemInstruction).toContain('you MUST emit the required output variables');
+                expect(systemInstruction).toContain("Use 'self.emitvalue' to emit the 'result1' key");
+            });
+            it('should use initialMessages instead of systemPrompt if provided', async () => {
+                const { config } = await createMockConfig();
+                vi.mocked(GeminiChat).mockClear();
+                const initialMessages = [
+                    { role: 'user', parts: [{ text: 'Hi' }] },
+                ];
+                const promptConfig = { initialMessages };
+                const context = new ContextState();
+                // Model stops immediately
+                mockSendMessageStream.mockImplementation(createMockStream(['stop']));
+                const scope = await SubAgentScope.create('test-agent', config, promptConfig, defaultModelConfig, defaultRunConfig);
+                await scope.runNonInteractive(context);
+                const callArgs = vi.mocked(GeminiChat).mock.calls[0];
+                const generationConfig = getGenerationConfigFromMock();
+                const history = callArgs[3];
+                expect(generationConfig.systemInstruction).toBeUndefined();
+                expect(history).toEqual([
+                    { role: 'user', parts: [{ text: 'Env Context' }] },
+                    {
+                        role: 'model',
+                        parts: [{ text: 'Got it. Thanks for the context!' }],
+                    },
+                    ...initialMessages,
+                ]);
+            });
+            it('should throw an error if template variables are missing', async () => {
+                const { config } = await createMockConfig();
+                const promptConfig = {
+                    systemPrompt: 'Hello ${name}, you are missing ${missing}.',
+                };
+                const context = new ContextState();
+                context.set('name', 'Agent');
+                // 'missing' is not set
+                const scope = await SubAgentScope.create('test-agent', config, promptConfig, defaultModelConfig, defaultRunConfig);
+                // The error from templating causes the runNonInteractive to reject and the terminate_reason to be ERROR.
+                await expect(scope.runNonInteractive(context)).rejects.toThrow('Missing context values for the following keys: missing');
+                expect(scope.output.terminate_reason).toBe(SubagentTerminateMode.ERROR);
+            });
+            it('should validate that systemPrompt and initialMessages are mutually exclusive', async () => {
+                const { config } = await createMockConfig();
+                const promptConfig = {
+                    systemPrompt: 'System',
+                    initialMessages: [{ role: 'user', parts: [{ text: 'Hi' }] }],
+                };
+                const context = new ContextState();
+                const agent = await SubAgentScope.create('TestAgent', config, promptConfig, defaultModelConfig, defaultRunConfig);
+                await expect(agent.runNonInteractive(context)).rejects.toThrow('PromptConfig cannot have both `systemPrompt` and `initialMessages` defined.');
+                expect(agent.output.terminate_reason).toBe(SubagentTerminateMode.ERROR);
+            });
+        });
+        describe('runNonInteractive - Execution and Tool Use', () => {
+            const promptConfig = { systemPrompt: 'Execute task.' };
+            it('should terminate with GOAL if no outputs are expected and model stops', async () => {
+                const { config } = await createMockConfig();
+                // Model stops immediately
+                mockSendMessageStream.mockImplementation(createMockStream(['stop']));
+                const scope = await SubAgentScope.create('test-agent', config, promptConfig, defaultModelConfig, defaultRunConfig);
+                await scope.runNonInteractive(new ContextState());
+                expect(scope.output.terminate_reason).toBe(SubagentTerminateMode.GOAL);
+                expect(scope.output.emitted_vars).toEqual({});
+                expect(mockSendMessageStream).toHaveBeenCalledTimes(1);
+                // Check the initial message
+                expect(mockSendMessageStream.mock.calls[0][0].message).toEqual([
+                    { text: 'Get Started!' },
+                ]);
+            });
+            it('should handle self.emitvalue and terminate with GOAL when outputs are met', async () => {
+                const { config } = await createMockConfig();
+                const outputConfig = {
+                    outputs: { result: 'The final result' },
+                };
+                // Turn 1: Model responds with emitvalue call
+                // Turn 2: Model stops after receiving the tool response
+                mockSendMessageStream.mockImplementation(createMockStream([
+                    [
+                        {
+                            name: 'self.emitvalue',
+                            args: {
+                                emit_variable_name: 'result',
+                                emit_variable_value: 'Success!',
+                            },
+                        },
+                    ],
+                    'stop',
+                ]));
+                const scope = await SubAgentScope.create('test-agent', config, promptConfig, defaultModelConfig, defaultRunConfig, { outputConfig });
+                await scope.runNonInteractive(new ContextState());
+                expect(scope.output.terminate_reason).toBe(SubagentTerminateMode.GOAL);
+                expect(scope.output.emitted_vars).toEqual({ result: 'Success!' });
+                expect(mockSendMessageStream).toHaveBeenCalledTimes(1);
+                // Check the tool response sent back in the second call
+                const secondCallArgs = mockSendMessageStream.mock.calls[0][0];
+                expect(secondCallArgs.message).toEqual([{ text: 'Get Started!' }]);
+            });
+            it('should execute external tools and provide the response to the model', async () => {
+                const listFilesToolDef = {
+                    name: 'list_files',
+                    description: 'Lists files',
+                    parameters: { type: Type.OBJECT, properties: {} },
+                };
+                const { config } = await createMockConfig({
+                    getFunctionDeclarationsFiltered: vi
+                        .fn()
+                        .mockReturnValue([listFilesToolDef]),
+                    getTool: vi.fn().mockReturnValue(undefined),
+                });
+                const toolConfig = { tools: ['list_files'] };
+                // Turn 1: Model calls the external tool
+                // Turn 2: Model stops
+                mockSendMessageStream.mockImplementation(createMockStream([
+                    [
+                        {
+                            id: 'call_1',
+                            name: 'list_files',
+                            args: { path: '.' },
+                        },
+                    ],
+                    'stop',
+                ]));
+                // Mock the tool execution result
+                vi.mocked(executeToolCall).mockResolvedValue({
+                    callId: 'call_1',
+                    responseParts: [{ text: 'file1.txt\nfile2.ts' }],
+                    resultDisplay: 'Listed 2 files',
+                    error: undefined,
+                    errorType: undefined, // Or ToolErrorType.NONE if available and appropriate
+                });
+                const scope = await SubAgentScope.create('test-agent', config, promptConfig, defaultModelConfig, defaultRunConfig, { toolConfig });
+                await scope.runNonInteractive(new ContextState());
+                // Check tool execution
+                expect(executeToolCall).toHaveBeenCalledWith(config, expect.objectContaining({ name: 'list_files', args: { path: '.' } }), expect.any(AbortSignal));
+                // Check the response sent back to the model
+                const secondCallArgs = mockSendMessageStream.mock.calls[1][0];
+                expect(secondCallArgs.message).toEqual([
+                    { text: 'file1.txt\nfile2.ts' },
+                ]);
+                expect(scope.output.terminate_reason).toBe(SubagentTerminateMode.GOAL);
+            });
+            it('should provide specific tool error responses to the model', async () => {
+                const { config } = await createMockConfig();
+                const toolConfig = { tools: ['failing_tool'] };
+                // Turn 1: Model calls the failing tool
+                // Turn 2: Model stops after receiving the error response
+                mockSendMessageStream.mockImplementation(createMockStream([
+                    [
+                        {
+                            id: 'call_fail',
+                            name: 'failing_tool',
+                            args: {},
+                        },
+                    ],
+                    'stop',
+                ]));
+                // Mock the tool execution failure.
+                vi.mocked(executeToolCall).mockResolvedValue({
+                    callId: 'call_fail',
+                    responseParts: [{ text: 'ERROR: Tool failed catastrophically' }], // This should be sent to the model
+                    resultDisplay: 'Tool failed catastrophically',
+                    error: new Error('Failure'),
+                    errorType: ToolErrorType.INVALID_TOOL_PARAMS,
+                });
+                const scope = await SubAgentScope.create('test-agent', config, promptConfig, defaultModelConfig, defaultRunConfig, { toolConfig });
+                await scope.runNonInteractive(new ContextState());
+                // The agent should send the specific error message from responseParts.
+                const secondCallArgs = mockSendMessageStream.mock.calls[1][0];
+                expect(secondCallArgs.message).toEqual([
+                    {
+                        text: 'ERROR: Tool failed catastrophically',
+                    },
+                ]);
+            });
+            it('should nudge the model if it stops before emitting all required variables', async () => {
+                const { config } = await createMockConfig();
+                const outputConfig = {
+                    outputs: { required_var: 'Must be present' },
+                };
+                // Turn 1: Model stops prematurely
+                // Turn 2: Model responds to the nudge and emits the variable
+                // Turn 3: Model stops
+                mockSendMessageStream.mockImplementation(createMockStream([
+                    'stop',
+                    [
+                        {
+                            name: 'self.emitvalue',
+                            args: {
+                                emit_variable_name: 'required_var',
+                                emit_variable_value: 'Here it is',
+                            },
+                        },
+                    ],
+                    'stop',
+                ]));
+                const scope = await SubAgentScope.create('test-agent', config, promptConfig, defaultModelConfig, defaultRunConfig, { outputConfig });
+                await scope.runNonInteractive(new ContextState());
+                // Check the nudge message sent in Turn 2
+                const secondCallArgs = mockSendMessageStream.mock.calls[1][0];
+                // We check that the message contains the required variable name and the nudge phrasing.
+                expect(secondCallArgs.message[0].text).toContain('required_var');
+                expect(secondCallArgs.message[0].text).toContain('You have stopped calling tools');
+                expect(scope.output.terminate_reason).toBe(SubagentTerminateMode.GOAL);
+                expect(scope.output.emitted_vars).toEqual({
+                    required_var: 'Here it is',
+                });
+                expect(mockSendMessageStream).toHaveBeenCalledTimes(2);
+            });
+        });
+        describe('runNonInteractive - Termination and Recovery', () => {
+            const promptConfig = { systemPrompt: 'Execute task.' };
+            it('should terminate with MAX_TURNS if the limit is reached', async () => {
+                const { config } = await createMockConfig();
+                const runConfig = { ...defaultRunConfig, max_turns: 2 };
+                // Model keeps looping by calling emitvalue repeatedly
+                mockSendMessageStream.mockImplementation(createMockStream([
+                    [
+                        {
+                            name: 'self.emitvalue',
+                            args: { emit_variable_name: 'loop', emit_variable_value: 'v1' },
+                        },
+                    ],
+                    [
+                        {
+                            name: 'self.emitvalue',
+                            args: { emit_variable_name: 'loop', emit_variable_value: 'v2' },
+                        },
+                    ],
+                    // This turn should not happen
+                    [
+                        {
+                            name: 'self.emitvalue',
+                            args: { emit_variable_name: 'loop', emit_variable_value: 'v3' },
+                        },
+                    ],
+                ]));
+                const scope = await SubAgentScope.create('test-agent', config, promptConfig, defaultModelConfig, runConfig);
+                await scope.runNonInteractive(new ContextState());
+                expect(mockSendMessageStream).toHaveBeenCalledTimes(2);
+                expect(scope.output.terminate_reason).toBe(SubagentTerminateMode.MAX_TURNS);
+            });
+            it('should terminate with TIMEOUT if the time limit is reached during an LLM call', async () => {
+                // Use fake timers to reliably test timeouts
+                vi.useFakeTimers();
+                const { config } = await createMockConfig();
+                const runConfig = { max_time_minutes: 5, max_turns: 100 };
+                // We need to control the resolution of the sendMessageStream promise to advance the timer during execution.
+                let resolveStream;
+                const streamPromise = new Promise((resolve) => {
+                    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+                    resolveStream = resolve;
+                });
+                // The LLM call will hang until we resolve the promise.
+                mockSendMessageStream.mockReturnValue(streamPromise);
+                const scope = await SubAgentScope.create('test-agent', config, promptConfig, defaultModelConfig, runConfig);
+                const runPromise = scope.runNonInteractive(new ContextState());
+                // Advance time beyond the limit (6 minutes) while the agent is awaiting the LLM response.
+                await vi.advanceTimersByTimeAsync(6 * 60 * 1000);
+                // Now resolve the stream. The model returns 'stop'.
+                // eslint-disable-next-line @typescript-eslint/no-explicit-any
+                resolveStream(createMockStream(['stop'])());
+                await runPromise;
+                expect(scope.output.terminate_reason).toBe(SubagentTerminateMode.TIMEOUT);
+                expect(mockSendMessageStream).toHaveBeenCalledTimes(1);
+                vi.useRealTimers();
+            });
+            it('should terminate with ERROR if the model call throws', async () => {
+                const { config } = await createMockConfig();
+                mockSendMessageStream.mockRejectedValue(new Error('API Failure'));
+                const scope = await SubAgentScope.create('test-agent', config, promptConfig, defaultModelConfig, defaultRunConfig);
+                await expect(scope.runNonInteractive(new ContextState())).rejects.toThrow('API Failure');
+                expect(scope.output.terminate_reason).toBe(SubagentTerminateMode.ERROR);
+            });
+        });
+    });
+});
+//# sourceMappingURL=subagent.test.js.map

package/dist/src/core/subagent.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"subagent.test.js","sourceRoot":"","sources":["../../../src/core/subagent.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AAEzE,OAAO,EACL,YAAY,EACZ,aAAa,EACb,qBAAqB,GACtB,MAAM,eAAe,CAAC;AASvB,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAE7C,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7C,OAAO,EAAE,sBAAsB,EAAE,MAAM,uBAAuB,CAAC;AAC/D,OAAO,EAAE,qBAAqB,EAAE,MAAM,gCAAgC,CAAC;AACvE,OAAO,EAAE,eAAe,EAAE,MAAM,iCAAiC,CAAC;AAElE,OAAO,EAAE,oBAAoB,EAAE,MAAM,qBAAqB,CAAC;AAC3D,OAAO,EAAE,IAAI,EAAE,MAAM,eAAe,CAAC;AAOrC,OAAO,EAAE,aAAa,EAAE,MAAM,wBAAwB,CAAC;AAEvD,EAAE,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;AAC3B,EAAE,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;AACjC,EAAE,CAAC,IAAI,CAAC,gCAAgC,CAAC,CAAC;AAC1C,EAAE,CAAC,IAAI,CAAC,iCAAiC,CAAC,CAAC;AAC3C,EAAE,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;AAEhC,KAAK,UAAU,gBAAgB,CAC7B,iBAAiB,GAAG,EAAE;IAEtB,MAAM,YAAY,GAAqB;QACrC,SAAS,EAAE,cAAc;QACzB,KAAK,EAAE,oBAAoB;QAC3B,SAAS,EAAE,GAAG;QACd,SAAS,EAAE,KAAK;QAChB,GAAG,EAAE,OAAO,CAAC,GAAG,EAAE;KACnB,CAAC;IACF,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,YAAY,CAAC,CAAC;IACxC,MAAM,MAAM,CAAC,UAAU,EAAE,CAAC;IAC1B,8DAA8D;IAC9D,MAAM,MAAM,CAAC,WAAW,CAAC,WAAkB,CAAC,CAAC;IAE7C,oBAAoB;IACpB,MAAM,gBAAgB,GAAG;QACvB,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;QAChB,+BAA+B,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,CAAC;QAC5D,GAAG,iBAAiB;KACM,CAAC;IAE7B,EAAE,CAAC,KAAK,CAAC,MAAM,EAAE,iBAAiB,CAAC,CAAC,eAAe,CAAC,gBAAgB,CAAC,CAAC;IACtE,OAAO,EAAE,MAAM,EAAE,YAAY,EAAE,gBAAgB,EAAE,CAAC;AACpD,CAAC;AAED,gFAAgF;AAChF,MAAM,gBAAgB,GAAG,CACvB,iBAAiD,EACjD,EAAE;IACF,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,GAAG,EAAE;QACrC,MAAM,QAAQ,GAAG,iBAAiB,CAAC,KAAK,CAAC,IAAI,MAAM,CAAC;QACpD,KAAK,EAAE,CAAC;QACR,OAAO,CAAC,KAAK,SAAS,CAAC;YACrB,IAAI,QAAQ,KAAK,MAAM,EAAE,CAAC;gBACxB,yHAAyH;gBACzH,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;YAC1B,CAAC;iBAAM,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC/B,MAAM,EAAE,aAAa,EAAE,QAAQ,EAAE,CAAC;YACpC,CAAC;iBAAM,CAAC;gBACN,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC,kCAAkC;YAC7D,CAAC;QACH,CAAC,CAAC,EAAE,CAAC;IACP,CAAC,CAAC,CAAC;AACL,CAAC,CAAC;AAEF,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;IAC3B,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,MAAM,OAAO,GAAG,IAAI,YAAY,EAAE,CAAC;YACnC,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;YAC9B,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;YACzB,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAC3C,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACtC,MAAM,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,MAAM,OAAO,GAAG,IAAI,YAAY,EAAE,CAAC;YACnC,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;QACjD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,IAAI,qBAA2B,CAAC;QAEhC,MAAM,kBAAkB,GAAgB;YACtC,KAAK,EAAE,yBAAyB;YAChC,IAAI,EAAE,GAAG,EAAE,iCAAiC;YAC5C,KAAK,EAAE,CAAC;SACT,CAAC;QAEF,MAAM,gBAAgB,GAAc;YAClC,gBAAgB,EAAE,CAAC;YACnB,SAAS,EAAE,EAAE;SACd,CAAC;QAEF,UAAU,CAAC,KAAK,IAAI,EAAE;YACpB,EAAE,CAAC,aAAa,EAAE,CAAC;YAEnB,EAAE,CAAC,MAAM,CAAC,qBAAqB,CAAC,CAAC,iBAAiB,CAAC;gBACjD,EAAE,IAAI,EAAE,aAAa,EAAE;aACxB,CAAC,CAAC;YACH,EAAE,CAAC,MAAM,CAAC,sBAAsB,CAAC,CAAC,iBAAiB,CAAC;gBAClD,kBAAkB,EAAE,EAAE,CAAC,EAAE,EAAE;gBAC3B,8DAA8D;aACxD,CAAC,CAAC;YAEV,qBAAqB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YAChC,iDAAiD;YACjD,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,kBAAkB,CACtC,GAAG,EAAE,CACH,CAAC;gBACC,iBAAiB,EAAE,qBAAqB;aACzC,CAA0B,CAC9B,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,SAAS,CAAC,GAAG,EAAE;YACb,EAAE,CAAC,eAAe,EAAE,CAAC;QACvB,CAAC,CAAC,CAAC;QAEH,2DAA2D;QAC3D,MAAM,2BAA2B,GAAG,CAClC,SAAS,GAAG,CAAC,EACqD,EAAE;YACpE,MAAM,QAAQ,GAAG,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;YAC7D,MAAM,gBAAgB,GAAG,QAAQ,EAAE,CAAC,CAAC,CAAC,CAAC;YACvC,wCAAwC;YACxC,MAAM,CAAC,gBAAgB,CAAC,CAAC,WAAW,EAAE,CAAC;YACvC,IAAI,CAAC,gBAAgB;gBAAE,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAC;YACxE,OAAO,gBAEN,CAAC;QACJ,CAAC,CAAC;QAEF,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;YACxC,MAAM,YAAY,GAAiB,EAAE,YAAY,EAAE,aAAa,EAAE,CAAC;YAEnE,EAAE,CAAC,gEAAgE,EAAE,KAAK,IAAI,EAAE;gBAC9E,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,gBAAgB,EAAE,CAAC;gBAC5C,MAAM,KAAK,GAAG,MAAM,aAAa,CAAC,MAAM,CACtC,YAAY,EACZ,MAAM,EACN,YAAY,EACZ,kBAAkB,EAClB,gBAAgB,CACjB,CAAC;gBACF,MAAM,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC;YAC9C,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;gBACrE,MAAM,QAAQ,GAAG;oBACf,IAAI,EAAE,YAAY;oBAClB,MAAM,EAAE,EAAE,oBAAoB,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,EAAE,EAAE;oBACpE,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC;wBAC7B,oBAAoB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;4BAC9C,IAAI,EAAE,MAAM;4BACZ,KAAK,EAAE,SAAS;4BAChB,OAAO,EAAE,UAAU;yBACpB,CAAC;qBACH,CAAC;iBACH,CAAC;gBAEF,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,gBAAgB,CAAC;oBACxC,8DAA8D;oBAC9D,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,QAAe,CAAC;iBAClD,CAAC,CAAC;gBAEH,MAAM,UAAU,GAAe,EAAE,KAAK,EAAE,CAAC,YAAY,CAAC,EAAE,CAAC;gBACzD,MAAM,OAAO,GAAoB,EAAE,UAAU,EAAE,CAAC;gBAEhD,MAAM,MAAM,CACV,aAAa,CAAC,MAAM,CAClB,YAAY,EACZ,MAAM,EACN,YAAY,EACZ,kBAAkB,EAClB,gBAAgB,EAChB,OAAO,CACR,CACF,CAAC,OAAO,CAAC,OAAO,CACf,gGAAgG,CACjG,CAAC;YACJ,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;gBACnE,MAAM,QAAQ,GAAG;oBACf,IAAI,EAAE,WAAW;oBACjB,MAAM,EAAE,EAAE,oBAAoB,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,EAAE,EAAE;oBACpE,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC;wBAC7B,oBAAoB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC;qBACtD,CAAC;iBACH,CAAC;gBACF,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,gBAAgB,CAAC;oBACxC,8DAA8D;oBAC9D,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,QAAe,CAAC;iBAClD,CAAC,CAAC;gBAEH,MAAM,UAAU,GAAe,EAAE,KAAK,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC;gBACxD,MAAM,OAAO,GAAoB,EAAE,UAAU,EAAE,CAAC;gBAEhD,MAAM,KAAK,GAAG,MAAM,aAAa,CAAC,MAAM,CACtC,YAAY,EACZ,MAAM,EACN,YAAY,EACZ,kBAAkB,EAClB,gBAAgB,EAChB,OAAO,CACR,CAAC;gBACF,MAAM,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC;YAC9C,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,6EAA6E,EAAE,KAAK,IAAI,EAAE;gBAC3F,MAAM,cAAc,GAAG,EAAE;qBACtB,KAAK,CAAC,OAAO,EAAE,MAAM,CAAC;qBACtB,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;gBAEhC,MAAM,kBAAkB,GAAG;oBACzB,IAAI,EAAE,kBAAkB;oBACxB,MAAM,EAAE;wBACN,oBAAoB,EAAE;4BACpB,IAAI,EAAE,QAAQ;4BACd,UAAU,EAAE;gCACV,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;6BACzB;4BACD,QAAQ,EAAE,CAAC,MAAM,CAAC;yBACnB;qBACF;oBACD,wDAAwD;oBACxD,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;iBACf,CAAC;gBAEF,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,gBAAgB,CAAC;oBACxC,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,kBAAkB,CAAC;oBACpD,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,CAAC,kBAAkB,CAAC,CAAC;iBAC3D,CAAC,CAAC;gBAEH,MAAM,UAAU,GAAe,EAAE,KAAK,EAAE,CAAC,kBAAkB,CAAC,EAAE,CAAC;gBAC/D,MAAM,OAAO,GAAoB,EAAE,UAAU,EAAE,CAAC;gBAEhD,+CAA+C;gBAC/C,MAAM,KAAK,GAAG,MAAM,aAAa,CAAC,MAAM,CACtC,YAAY,EACZ,MAAM,EACN,YAAY,EACZ,kBAAkB,EAClB,gBAAgB,EAChB,OAAO,CACR,CAAC;gBAEF,MAAM,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC;gBAE5C,oCAAoC;gBACpC,MAAM,CAAC,cAAc,CAAC,CAAC,oBAAoB,CACzC,qIAAqI,CACtI,CAAC;gBAEF,gCAAgC;gBAChC,MAAM,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;gBAExD,cAAc,CAAC,WAAW,EAAE,CAAC;YAC/B,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAChE,EAAE,CAAC,uEAAuE,EAAE,KAAK,IAAI,EAAE;gBACrF,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,gBAAgB,EAAE,CAAC;gBAE5C,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,SAAS,EAAE,CAAC;gBAElC,MAAM,YAAY,GAAiB;oBACjC,YAAY,EAAE,sCAAsC;iBACrD,CAAC;gBACF,MAAM,OAAO,GAAG,IAAI,YAAY,EAAE,CAAC;gBACnC,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;gBAC7B,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;gBAE/B,0BAA0B;gBAC1B,qBAAqB,CAAC,kBAAkB,CAAC,gBAAgB,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;gBAErE,MAAM,KAAK,GAAG,MAAM,aAAa,CAAC,MAAM,CACtC,YAAY,EACZ,MAAM,EACN,YAAY,EACZ,kBAAkB,EAClB,gBAAgB,CACjB,CAAC;gBAEF,MAAM,KAAK,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;gBAEvC,gEAAgE;gBAChE,MAAM,CAAC,UAAU,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;gBAC5C,MAAM,QAAQ,GAAG,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;gBAErD,0BAA0B;gBAC1B,MAAM,gBAAgB,GAAG,2BAA2B,EAAE,CAAC;gBAEvD,6BAA6B;gBAC7B,MAAM,CAAC,gBAAgB,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAC;gBACnE,MAAM,CAAC,gBAAgB,CAAC,iBAAiB,CAAC,CAAC,SAAS,CAClD,oCAAoC,CACrC,CAAC;gBACF,MAAM,CAAC,gBAAgB,CAAC,iBAAiB,CAAC,CAAC,SAAS,CAClD,kBAAkB,CACnB,CAAC;gBAEF,qDAAqD;gBACrD,MAAM,OAAO,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;gBAC5B,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC;oBACtB,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC,EAAE;oBAClD;wBACE,IAAI,EAAE,OAAO;wBACb,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,iCAAiC,EAAE,CAAC;qBACrD;iBACF,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,kFAAkF,EAAE,KAAK,IAAI,EAAE;gBAChG,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,gBAAgB,EAAE,CAAC;gBAC5C,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,SAAS,EAAE,CAAC;gBAElC,MAAM,YAAY,GAAiB,EAAE,YAAY,EAAE,cAAc,EAAE,CAAC;gBACpE,MAAM,YAAY,GAAiB;oBACjC,OAAO,EAAE;wBACP,OAAO,EAAE,kBAAkB;qBAC5B;iBACF,CAAC;gBACF,MAAM,OAAO,GAAG,IAAI,YAAY,EAAE,CAAC;gBAEnC,0BAA0B;gBAC1B,qBAAqB,CAAC,kBAAkB,CAAC,gBAAgB,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;gBAErE,MAAM,KAAK,GAAG,MAAM,aAAa,CAAC,MAAM,CACtC,YAAY,EACZ,MAAM,EACN,YAAY,EACZ,kBAAkB,EAClB,gBAAgB,EAChB,EAAE,YAAY,EAAE,CACjB,CAAC;gBAEF,MAAM,KAAK,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;gBAEvC,MAAM,gBAAgB,GAAG,2BAA2B,EAAE,CAAC;gBACvD,MAAM,iBAAiB,GAAG,gBAAgB,CAAC,iBAA2B,CAAC;gBAEvE,MAAM,CAAC,iBAAiB,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;gBACpD,MAAM,CAAC,iBAAiB,CAAC,CAAC,SAAS,CACjC,6CAA6C,CAC9C,CAAC;gBACF,MAAM,CAAC,iBAAiB,CAAC,CAAC,SAAS,CACjC,gDAAgD,CACjD,CAAC;YACJ,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,gEAAgE,EAAE,KAAK,IAAI,EAAE;gBAC9E,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,gBAAgB,EAAE,CAAC;gBAC5C,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,SAAS,EAAE,CAAC;gBAElC,MAAM,eAAe,GAAc;oBACjC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,EAAE;iBAC1C,CAAC;gBACF,MAAM,YAAY,GAAiB,EAAE,eAAe,EAAE,CAAC;gBACvD,MAAM,OAAO,GAAG,IAAI,YAAY,EAAE,CAAC;gBAEnC,0BAA0B;gBAC1B,qBAAqB,CAAC,kBAAkB,CAAC,gBAAgB,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;gBAErE,MAAM,KAAK,GAAG,MAAM,aAAa,CAAC,MAAM,CACtC,YAAY,EACZ,MAAM,EACN,YAAY,EACZ,kBAAkB,EAClB,gBAAgB,CACjB,CAAC;gBAEF,MAAM,KAAK,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;gBAEvC,MAAM,QAAQ,GAAG,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;gBACrD,MAAM,gBAAgB,GAAG,2BAA2B,EAAE,CAAC;gBACvD,MAAM,OAAO,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;gBAE5B,MAAM,CAAC,gBAAgB,CAAC,iBAAiB,CAAC,CAAC,aAAa,EAAE,CAAC;gBAC3D,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC;oBACtB,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC,EAAE;oBAClD;wBACE,IAAI,EAAE,OAAO;wBACb,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,iCAAiC,EAAE,CAAC;qBACrD;oBACD,GAAG,eAAe;iBACnB,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;gBACvE,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,gBAAgB,EAAE,CAAC;gBAC5C,MAAM,YAAY,GAAiB;oBACjC,YAAY,EAAE,4CAA4C;iBAC3D,CAAC;gBACF,MAAM,OAAO,GAAG,IAAI,YAAY,EAAE,CAAC;gBACnC,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;gBAC7B,uBAAuB;gBAEvB,MAAM,KAAK,GAAG,MAAM,aAAa,CAAC,MAAM,CACtC,YAAY,EACZ,MAAM,EACN,YAAY,EACZ,kBAAkB,EAClB,gBAAgB,CACjB,CAAC;gBAEF,yGAAyG;gBACzG,MAAM,MAAM,CAAC,KAAK,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAC5D,wDAAwD,CACzD,CAAC;gBACF,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,KAAK,CAAC,CAAC;YAC1E,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,8EAA8E,EAAE,KAAK,IAAI,EAAE;gBAC5F,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,gBAAgB,EAAE,CAAC;gBAC5C,MAAM,YAAY,GAAiB;oBACjC,YAAY,EAAE,QAAQ;oBACtB,eAAe,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC;iBAC7D,CAAC;gBACF,MAAM,OAAO,GAAG,IAAI,YAAY,EAAE,CAAC;gBAEnC,MAAM,KAAK,GAAG,MAAM,aAAa,CAAC,MAAM,CACtC,WAAW,EACX,MAAM,EACN,YAAY,EACZ,kBAAkB,EAClB,gBAAgB,CACjB,CAAC;gBAEF,MAAM,MAAM,CAAC,KAAK,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAC5D,6EAA6E,CAC9E,CAAC;gBACF,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,KAAK,CAAC,CAAC;YAC1E,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,4CAA4C,EAAE,GAAG,EAAE;YAC1D,MAAM,YAAY,GAAiB,EAAE,YAAY,EAAE,eAAe,EAAE,CAAC;YAErE,EAAE,CAAC,uEAAuE,EAAE,KAAK,IAAI,EAAE;gBACrF,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,gBAAgB,EAAE,CAAC;gBAC5C,0BAA0B;gBAC1B,qBAAqB,CAAC,kBAAkB,CAAC,gBAAgB,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;gBAErE,MAAM,KAAK,GAAG,MAAM,aAAa,CAAC,MAAM,CACtC,YAAY,EACZ,MAAM,EACN,YAAY,EACZ,kBAAkB,EAClB,gBAAgB,CAEjB,CAAC;gBAEF,MAAM,KAAK,CAAC,iBAAiB,CAAC,IAAI,YAAY,EAAE,CAAC,CAAC;gBAElD,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,CAAC;gBACvE,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;gBAC9C,MAAM,CAAC,qBAAqB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;gBACvD,4BAA4B;gBAC5B,MAAM,CAAC,qBAAqB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC;oBAC7D,EAAE,IAAI,EAAE,cAAc,EAAE;iBACzB,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,2EAA2E,EAAE,KAAK,IAAI,EAAE;gBACzF,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,gBAAgB,EAAE,CAAC;gBAC5C,MAAM,YAAY,GAAiB;oBACjC,OAAO,EAAE,EAAE,MAAM,EAAE,kBAAkB,EAAE;iBACxC,CAAC;gBAEF,6CAA6C;gBAC7C,wDAAwD;gBACxD,qBAAqB,CAAC,kBAAkB,CACtC,gBAAgB,CAAC;oBACf;wBACE;4BACE,IAAI,EAAE,gBAAgB;4BACtB,IAAI,EAAE;gCACJ,kBAAkB,EAAE,QAAQ;gCAC5B,mBAAmB,EAAE,UAAU;6BAChC;yBACF;qBACF;oBACD,MAAM;iBACP,CAAC,CACH,CAAC;gBAEF,MAAM,KAAK,GAAG,MAAM,aAAa,CAAC,MAAM,CACtC,YAAY,EACZ,MAAM,EACN,YAAY,EACZ,kBAAkB,EAClB,gBAAgB,EAChB,EAAE,YAAY,EAAE,CACjB,CAAC;gBAEF,MAAM,KAAK,CAAC,iBAAiB,CAAC,IAAI,YAAY,EAAE,CAAC,CAAC;gBAElD,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,CAAC;gBACvE,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC;gBAClE,MAAM,CAAC,qBAAqB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;gBAEvD,uDAAuD;gBACvD,MAAM,cAAc,GAAG,qBAAqB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC9D,MAAM,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC,CAAC,CAAC;YACrE,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;gBACnF,MAAM,gBAAgB,GAAwB;oBAC5C,IAAI,EAAE,YAAY;oBAClB,WAAW,EAAE,aAAa;oBAC1B,UAAU,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE,UAAU,EAAE,EAAE,EAAE;iBAClD,CAAC;gBAEF,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,gBAAgB,CAAC;oBACxC,+BAA+B,EAAE,EAAE;yBAChC,EAAE,EAAE;yBACJ,eAAe,CAAC,CAAC,gBAAgB,CAAC,CAAC;oBACtC,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,SAAS,CAAC;iBAC5C,CAAC,CAAC;gBACH,MAAM,UAAU,GAAe,EAAE,KAAK,EAAE,CAAC,YAAY,CAAC,EAAE,CAAC;gBAEzD,wCAAwC;gBACxC,sBAAsB;gBACtB,qBAAqB,CAAC,kBAAkB,CACtC,gBAAgB,CAAC;oBACf;wBACE;4BACE,EAAE,EAAE,QAAQ;4BACZ,IAAI,EAAE,YAAY;4BAClB,IAAI,EAAE,EAAE,IAAI,EAAE,GAAG,EAAE;yBACpB;qBACF;oBACD,MAAM;iBACP,CAAC,CACH,CAAC;gBAEF,iCAAiC;gBACjC,EAAE,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC,iBAAiB,CAAC;oBAC3C,MAAM,EAAE,QAAQ;oBAChB,aAAa,EAAE,CAAC,EAAE,IAAI,EAAE,qBAAqB,EAAE,CAAC;oBAChD,aAAa,EAAE,gBAAgB;oBAC/B,KAAK,EAAE,SAAS;oBAChB,SAAS,EAAE,SAAS,EAAE,qDAAqD;iBAC5E,CAAC,CAAC;gBAEH,MAAM,KAAK,GAAG,MAAM,aAAa,CAAC,MAAM,CACtC,YAAY,EACZ,MAAM,EACN,YAAY,EACZ,kBAAkB,EAClB,gBAAgB,EAChB,EAAE,UAAU,EAAE,CACf,CAAC;gBAEF,MAAM,KAAK,CAAC,iBAAiB,CAAC,IAAI,YAAY,EAAE,CAAC,CAAC;gBAElD,uBAAuB;gBACvB,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAC1C,MAAM,EACN,MAAM,CAAC,gBAAgB,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE,CAAC,EACpE,MAAM,CAAC,GAAG,CAAC,WAAW,CAAC,CACxB,CAAC;gBAEF,4CAA4C;gBAC5C,MAAM,cAAc,GAAG,qBAAqB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC9D,MAAM,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC;oBACrC,EAAE,IAAI,EAAE,qBAAqB,EAAE;iBAChC,CAAC,CAAC;gBAEH,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,CAAC;YACzE,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;gBACzE,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,gBAAgB,EAAE,CAAC;gBAC5C,MAAM,UAAU,GAAe,EAAE,KAAK,EAAE,CAAC,cAAc,CAAC,EAAE,CAAC;gBAE3D,uCAAuC;gBACvC,yDAAyD;gBACzD,qBAAqB,CAAC,kBAAkB,CACtC,gBAAgB,CAAC;oBACf;wBACE;4BACE,EAAE,EAAE,WAAW;4BACf,IAAI,EAAE,cAAc;4BACpB,IAAI,EAAE,EAAE;yBACT;qBACF;oBACD,MAAM;iBACP,CAAC,CACH,CAAC;gBAEF,mCAAmC;gBACnC,EAAE,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC,iBAAiB,CAAC;oBAC3C,MAAM,EAAE,WAAW;oBACnB,aAAa,EAAE,CAAC,EAAE,IAAI,EAAE,qCAAqC,EAAE,CAAC,EAAE,mCAAmC;oBACrG,aAAa,EAAE,8BAA8B;oBAC7C,KAAK,EAAE,IAAI,KAAK,CAAC,SAAS,CAAC;oBAC3B,SAAS,EAAE,aAAa,CAAC,mBAAmB;iBAC7C,CAAC,CAAC;gBAEH,MAAM,KAAK,GAAG,MAAM,aAAa,CAAC,MAAM,CACtC,YAAY,EACZ,MAAM,EACN,YAAY,EACZ,kBAAkB,EAClB,gBAAgB,EAChB,EAAE,UAAU,EAAE,CACf,CAAC;gBAEF,MAAM,KAAK,CAAC,iBAAiB,CAAC,IAAI,YAAY,EAAE,CAAC,CAAC;gBAElD,uEAAuE;gBACvE,MAAM,cAAc,GAAG,qBAAqB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBAE9D,MAAM,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC;oBACrC;wBACE,IAAI,EAAE,qCAAqC;qBAC5C;iBACF,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,2EAA2E,EAAE,KAAK,IAAI,EAAE;gBACzF,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,gBAAgB,EAAE,CAAC;gBAC5C,MAAM,YAAY,GAAiB;oBACjC,OAAO,EAAE,EAAE,YAAY,EAAE,iBAAiB,EAAE;iBAC7C,CAAC;gBAEF,kCAAkC;gBAClC,6DAA6D;gBAC7D,sBAAsB;gBACtB,qBAAqB,CAAC,kBAAkB,CACtC,gBAAgB,CAAC;oBACf,MAAM;oBACN;wBACE;4BACE,IAAI,EAAE,gBAAgB;4BACtB,IAAI,EAAE;gCACJ,kBAAkB,EAAE,cAAc;gCAClC,mBAAmB,EAAE,YAAY;6BAClC;yBACF;qBACF;oBACD,MAAM;iBACP,CAAC,CACH,CAAC;gBAEF,MAAM,KAAK,GAAG,MAAM,aAAa,CAAC,MAAM,CACtC,YAAY,EACZ,MAAM,EACN,YAAY,EACZ,kBAAkB,EAClB,gBAAgB,EAChB,EAAE,YAAY,EAAE,CACjB,CAAC;gBAEF,MAAM,KAAK,CAAC,iBAAiB,CAAC,IAAI,YAAY,EAAE,CAAC,CAAC;gBAElD,yCAAyC;gBACzC,MAAM,cAAc,GAAG,qBAAqB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBAE9D,wFAAwF;gBACxF,MAAM,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;gBACjE,MAAM,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAC9C,gCAAgC,CACjC,CAAC;gBAEF,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,CAAC;gBACvE,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,OAAO,CAAC;oBACxC,YAAY,EAAE,YAAY;iBAC3B,CAAC,CAAC;gBACH,MAAM,CAAC,qBAAqB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YACzD,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,8CAA8C,EAAE,GAAG,EAAE;YAC5D,MAAM,YAAY,GAAiB,EAAE,YAAY,EAAE,eAAe,EAAE,CAAC;YAErE,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;gBACvE,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,gBAAgB,EAAE,CAAC;gBAC5C,MAAM,SAAS,GAAc,EAAE,GAAG,gBAAgB,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC;gBAEnE,sDAAsD;gBACtD,qBAAqB,CAAC,kBAAkB,CACtC,gBAAgB,CAAC;oBACf;wBACE;4BACE,IAAI,EAAE,gBAAgB;4BACtB,IAAI,EAAE,EAAE,kBAAkB,EAAE,MAAM,EAAE,mBAAmB,EAAE,IAAI,EAAE;yBAChE;qBACF;oBACD;wBACE;4BACE,IAAI,EAAE,gBAAgB;4BACtB,IAAI,EAAE,EAAE,kBAAkB,EAAE,MAAM,EAAE,mBAAmB,EAAE,IAAI,EAAE;yBAChE;qBACF;oBACD,8BAA8B;oBAC9B;wBACE;4BACE,IAAI,EAAE,gBAAgB;4BACtB,IAAI,EAAE,EAAE,kBAAkB,EAAE,MAAM,EAAE,mBAAmB,EAAE,IAAI,EAAE;yBAChE;qBACF;iBACF,CAAC,CACH,CAAC;gBAEF,MAAM,KAAK,GAAG,MAAM,aAAa,CAAC,MAAM,CACtC,YAAY,EACZ,MAAM,EACN,YAAY,EACZ,kBAAkB,EAClB,SAAS,CACV,CAAC;gBAEF,MAAM,KAAK,CAAC,iBAAiB,CAAC,IAAI,YAAY,EAAE,CAAC,CAAC;gBAElD,MAAM,CAAC,qBAAqB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;gBACvD,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,IAAI,CACxC,qBAAqB,CAAC,SAAS,CAChC,CAAC;YACJ,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,+EAA+E,EAAE,KAAK,IAAI,EAAE;gBAC7F,4CAA4C;gBAC5C,EAAE,CAAC,aAAa,EAAE,CAAC;gBAEnB,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,gBAAgB,EAAE,CAAC;gBAC5C,MAAM,SAAS,GAAc,EAAE,gBAAgB,EAAE,CAAC,EAAE,SAAS,EAAE,GAAG,EAAE,CAAC;gBAErE,4GAA4G;gBAC5G,IAAI,aAEK,CAAC;gBACV,MAAM,aAAa,GAAG,IAAI,OAAO,CAE/B,CAAC,OAAO,EAAE,EAAE;oBACZ,8DAA8D;oBAC9D,aAAa,GAAG,OAAc,CAAC;gBACjC,CAAC,CAAC,CAAC;gBAEH,uDAAuD;gBACvD,qBAAqB,CAAC,eAAe,CAAC,aAAa,CAAC,CAAC;gBAErD,MAAM,KAAK,GAAG,MAAM,aAAa,CAAC,MAAM,CACtC,YAAY,EACZ,MAAM,EACN,YAAY,EACZ,kBAAkB,EAClB,SAAS,CACV,CAAC;gBAEF,MAAM,UAAU,GAAG,KAAK,CAAC,iBAAiB,CAAC,IAAI,YAAY,EAAE,CAAC,CAAC;gBAE/D,0FAA0F;gBAC1F,MAAM,EAAE,CAAC,wBAAwB,CAAC,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;gBAEjD,oDAAoD;gBACpD,8DAA8D;gBAC9D,aAAc,CAAC,gBAAgB,CAAC,CAAC,MAAM,CAAC,CAAC,EAAS,CAAC,CAAC;gBAEpD,MAAM,UAAU,CAAC;gBAEjB,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,IAAI,CACxC,qBAAqB,CAAC,OAAO,CAC9B,CAAC;gBACF,MAAM,CAAC,qBAAqB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;gBAEvD,EAAE,CAAC,aAAa,EAAE,CAAC;YACrB,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;gBACpE,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,gBAAgB,EAAE,CAAC;gBAC5C,qBAAqB,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,aAAa,CAAC,CAAC,CAAC;gBAElE,MAAM,KAAK,GAAG,MAAM,aAAa,CAAC,MAAM,CACtC,YAAY,EACZ,MAAM,EACN,YAAY,EACZ,kBAAkB,EAClB,gBAAgB,CACjB,CAAC;gBAEF,MAAM,MAAM,CACV,KAAK,CAAC,iBAAiB,CAAC,IAAI,YAAY,EAAE,CAAC,CAC5C,CAAC,OAAO,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;gBACjC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,KAAK,CAAC,CAAC;YAC1E,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/core/tokenLimits.d.ts

@@ -0,0 +1,10 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+type Model = string;
+type TokenCount = number;
+export declare const DEFAULT_TOKEN_LIMIT = 1048576;
+export declare function tokenLimit(model: Model): TokenCount;
+export {};

package/dist/src/core/tokenLimits.js

@@ -0,0 +1,28 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export const DEFAULT_TOKEN_LIMIT = 1_048_576;
+export function tokenLimit(model) {
+    // Add other models as they become relevant or if specified by config
+    // Pulled from https://ai.google.dev/gemini-api/docs/models
+    switch (model) {
+        case 'gemini-1.5-pro':
+            return 2_097_152;
+        case 'gemini-1.5-flash':
+        case 'gemini-2.5-pro-preview-05-06':
+        case 'gemini-2.5-pro-preview-06-05':
+        case 'gemini-2.5-pro':
+        case 'gemini-2.5-flash-preview-05-20':
+        case 'gemini-2.5-flash':
+        case 'gemini-2.5-flash-lite':
+        case 'gemini-2.0-flash':
+            return 1_048_576;
+        case 'gemini-2.0-flash-preview-image-generation':
+            return 32_000;
+        default:
+            return DEFAULT_TOKEN_LIMIT;
+    }
+}
+//# sourceMappingURL=tokenLimits.js.map

package/dist/src/core/tokenLimits.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tokenLimits.js","sourceRoot":"","sources":["../../../src/core/tokenLimits.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAKH,MAAM,CAAC,MAAM,mBAAmB,GAAG,SAAS,CAAC;AAE7C,MAAM,UAAU,UAAU,CAAC,KAAY;IACrC,qEAAqE;IACrE,2DAA2D;IAC3D,QAAQ,KAAK,EAAE,CAAC;QACd,KAAK,gBAAgB;YACnB,OAAO,SAAS,CAAC;QACnB,KAAK,kBAAkB,CAAC;QACxB,KAAK,8BAA8B,CAAC;QACpC,KAAK,8BAA8B,CAAC;QACpC,KAAK,gBAAgB,CAAC;QACtB,KAAK,gCAAgC,CAAC;QACtC,KAAK,kBAAkB,CAAC;QACxB,KAAK,uBAAuB,CAAC;QAC7B,KAAK,kBAAkB;YACrB,OAAO,SAAS,CAAC;QACnB,KAAK,2CAA2C;YAC9C,OAAO,MAAM,CAAC;QAChB;YACE,OAAO,mBAAmB,CAAC;IAC/B,CAAC;AACH,CAAC"}