@vybestack/llxprt-code-core 0.1.12

package/dist/src/config/config.test.js

@@ -0,0 +1,291 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach } from 'vitest';
+import { Config } from './config.js';
+import * as path from 'path';
+import { setLlxprtMdFilename as mockSetGeminiMdFilename } from '../tools/memoryTool.js';
+import { DEFAULT_TELEMETRY_TARGET, DEFAULT_OTLP_ENDPOINT, } from '../telemetry/index.js';
+import { AuthType, createContentGeneratorConfig, } from '../core/contentGenerator.js';
+import { GeminiClient } from '../core/client.js';
+import { GitService } from '../services/gitService.js';
+import { loadServerHierarchicalMemory } from '../utils/memoryDiscovery.js';
+// Mock dependencies that might be called during Config construction or createServerConfig
+vi.mock('../tools/tool-registry', () => {
+    const ToolRegistryMock = vi.fn();
+    ToolRegistryMock.prototype.registerTool = vi.fn();
+    ToolRegistryMock.prototype.discoverTools = vi.fn();
+    ToolRegistryMock.prototype.getAllTools = vi.fn(() => []); // Mock methods if needed
+    ToolRegistryMock.prototype.getTool = vi.fn();
+    ToolRegistryMock.prototype.getFunctionDeclarations = vi.fn(() => []);
+    return { ToolRegistry: ToolRegistryMock };
+});
+vi.mock('../utils/memoryDiscovery.js', () => ({
+    loadServerHierarchicalMemory: vi.fn(),
+}));
+// Mock individual tools if their constructors are complex or have side effects
+vi.mock('../tools/ls');
+vi.mock('../tools/read-file');
+vi.mock('../tools/grep');
+vi.mock('../tools/glob');
+vi.mock('../tools/edit');
+vi.mock('../tools/shell');
+vi.mock('../tools/write-file');
+vi.mock('../tools/web-fetch');
+vi.mock('../tools/read-many-files');
+vi.mock('../tools/memoryTool', () => ({
+    MemoryTool: vi.fn(),
+    setLlxprtMdFilename: vi.fn(),
+    getCurrentLlxprtMdFilename: vi.fn(() => 'LLXPRT.md'), // Mock the original filename
+    DEFAULT_CONTEXT_FILENAME: 'LLXPRT.md',
+    LLXPRT_CONFIG_DIR: '.llxprt',
+}));
+vi.mock('../core/contentGenerator.js', async (importOriginal) => {
+    const actual = await importOriginal();
+    return {
+        ...actual,
+        createContentGeneratorConfig: vi.fn(),
+    };
+});
+vi.mock('../core/client.js', () => ({
+    GeminiClient: vi.fn().mockImplementation(() => ({
+        initialize: vi.fn().mockResolvedValue(undefined),
+    })),
+}));
+vi.mock('../telemetry/index.js', async (importOriginal) => {
+    const actual = await importOriginal();
+    return {
+        ...actual,
+        initializeTelemetry: vi.fn(),
+    };
+});
+vi.mock('../services/gitService.js', () => {
+    const GitServiceMock = vi.fn();
+    GitServiceMock.prototype.initialize = vi.fn();
+    return { GitService: GitServiceMock };
+});
+describe('Server Config (config.ts)', () => {
+    const MODEL = 'gemini-pro';
+    const SANDBOX = {
+        command: 'docker',
+        image: 'gemini-cli-sandbox',
+    };
+    const TARGET_DIR = '/path/to/target';
+    const DEBUG_MODE = false;
+    const QUESTION = 'test question';
+    const FULL_CONTEXT = false;
+    const USER_MEMORY = 'Test User Memory';
+    const TELEMETRY_SETTINGS = { enabled: false };
+    const EMBEDDING_MODEL = 'gemini-embedding';
+    const SESSION_ID = 'test-session-id';
+    const baseParams = {
+        cwd: '/tmp',
+        embeddingModel: EMBEDDING_MODEL,
+        sandbox: SANDBOX,
+        targetDir: TARGET_DIR,
+        debugMode: DEBUG_MODE,
+        question: QUESTION,
+        fullContext: FULL_CONTEXT,
+        userMemory: USER_MEMORY,
+        telemetry: TELEMETRY_SETTINGS,
+        sessionId: SESSION_ID,
+        model: MODEL,
+    };
+    beforeEach(() => {
+        // Reset mocks if necessary
+        vi.clearAllMocks();
+    });
+    describe('initialize', () => {
+        it('should throw an error if checkpointing is enabled and GitService fails', async () => {
+            const gitError = new Error('Git is not installed');
+            GitService.prototype.initialize.mockRejectedValue(gitError);
+            const config = new Config({
+                ...baseParams,
+                checkpointing: true,
+            });
+            await expect(config.initialize()).rejects.toThrow(gitError);
+        });
+        it('should not throw an error if checkpointing is disabled and GitService fails', async () => {
+            const gitError = new Error('Git is not installed');
+            GitService.prototype.initialize.mockRejectedValue(gitError);
+            const config = new Config({
+                ...baseParams,
+                checkpointing: false,
+            });
+            await expect(config.initialize()).resolves.toBeUndefined();
+        });
+    });
+    describe('refreshAuth', () => {
+        it('should refresh auth and update config', async () => {
+            const config = new Config(baseParams);
+            // Initialize config to create GeminiClient instance
+            await config.initialize();
+            const authType = AuthType.USE_GEMINI;
+            const newModel = 'gemini-flash';
+            const mockContentConfig = {
+                model: newModel,
+                apiKey: 'test-key',
+            };
+            createContentGeneratorConfig.mockResolvedValue(mockContentConfig);
+            await config.refreshAuth(authType);
+            expect(createContentGeneratorConfig).toHaveBeenCalledWith(MODEL, // Should be called with the original model 'gemini-pro'
+            authType);
+            // Verify that contentGeneratorConfig is updated with the new model
+            expect(config.getContentGeneratorConfig()).toEqual(mockContentConfig);
+            expect(config.getContentGeneratorConfig()?.model).toBe(newModel);
+            expect(config.getModel()).toBe(newModel); // getModel() should return the updated model
+            expect(GeminiClient).toHaveBeenCalledWith(config);
+        });
+    });
+    it('Config constructor should store userMemory correctly', () => {
+        const config = new Config(baseParams);
+        expect(config.getUserMemory()).toBe(USER_MEMORY);
+        // Verify other getters if needed
+        expect(config.getTargetDir()).toBe(path.resolve(TARGET_DIR)); // Check resolved path
+    });
+    it('Config constructor should default userMemory to empty string if not provided', () => {
+        const paramsWithoutMemory = { ...baseParams };
+        delete paramsWithoutMemory.userMemory;
+        const config = new Config(paramsWithoutMemory);
+        expect(config.getUserMemory()).toBe('');
+    });
+    it('Config constructor should call setLlxprtMdFilename with contextFileName if provided', () => {
+        const contextFileName = 'CUSTOM_AGENTS.md';
+        const paramsWithContextFile = {
+            ...baseParams,
+            contextFileName,
+        };
+        new Config(paramsWithContextFile);
+        expect(mockSetGeminiMdFilename).toHaveBeenCalledWith(contextFileName);
+    });
+    it('Config constructor should not call setLlxprtMdFilename if contextFileName is not provided', () => {
+        new Config(baseParams); // baseParams does not have contextFileName
+        expect(mockSetGeminiMdFilename).not.toHaveBeenCalled();
+    });
+    it('should set default file filtering settings when not provided', () => {
+        const config = new Config(baseParams);
+        expect(config.getFileFilteringRespectGitIgnore()).toBe(true);
+    });
+    it('should set custom file filtering settings when provided', () => {
+        const paramsWithFileFiltering = {
+            ...baseParams,
+            fileFiltering: {
+                respectGitIgnore: false,
+            },
+        };
+        const config = new Config(paramsWithFileFiltering);
+        expect(config.getFileFilteringRespectGitIgnore()).toBe(false);
+    });
+    it('Config constructor should set telemetry to true when provided as true', () => {
+        const paramsWithTelemetry = {
+            ...baseParams,
+            telemetry: { enabled: true },
+        };
+        const config = new Config(paramsWithTelemetry);
+        expect(config.getTelemetryEnabled()).toBe(true);
+    });
+    it('Config constructor should set telemetry to false when provided as false', () => {
+        const paramsWithTelemetry = {
+            ...baseParams,
+            telemetry: { enabled: false },
+        };
+        const config = new Config(paramsWithTelemetry);
+        expect(config.getTelemetryEnabled()).toBe(false);
+    });
+    it('Config constructor should default telemetry to default value if not provided', () => {
+        const paramsWithoutTelemetry = { ...baseParams };
+        delete paramsWithoutTelemetry.telemetry;
+        const config = new Config(paramsWithoutTelemetry);
+        expect(config.getTelemetryEnabled()).toBe(TELEMETRY_SETTINGS.enabled);
+    });
+    it('should have a getFileService method that returns FileDiscoveryService', () => {
+        const config = new Config(baseParams);
+        const fileService = config.getFileService();
+        expect(fileService).toBeDefined();
+    });
+    describe('Telemetry Settings', () => {
+        it('should return default telemetry target if not provided', () => {
+            const params = {
+                ...baseParams,
+                telemetry: { enabled: true },
+            };
+            const config = new Config(params);
+            expect(config.getTelemetryTarget()).toBe(DEFAULT_TELEMETRY_TARGET);
+        });
+        it('should return provided OTLP endpoint', () => {
+            const endpoint = 'http://custom.otel.collector:4317';
+            const params = {
+                ...baseParams,
+                telemetry: { enabled: true, otlpEndpoint: endpoint },
+            };
+            const config = new Config(params);
+            expect(config.getTelemetryOtlpEndpoint()).toBe(endpoint);
+        });
+        it('should return default OTLP endpoint if not provided', () => {
+            const params = {
+                ...baseParams,
+                telemetry: { enabled: true },
+            };
+            const config = new Config(params);
+            expect(config.getTelemetryOtlpEndpoint()).toBe(DEFAULT_OTLP_ENDPOINT);
+        });
+        it('should return provided logPrompts setting', () => {
+            const params = {
+                ...baseParams,
+                telemetry: { enabled: true, logPrompts: false },
+            };
+            const config = new Config(params);
+            expect(config.getTelemetryLogPromptsEnabled()).toBe(false);
+        });
+        it('should return default logPrompts setting (true) if not provided', () => {
+            const params = {
+                ...baseParams,
+                telemetry: { enabled: true },
+            };
+            const config = new Config(params);
+            expect(config.getTelemetryLogPromptsEnabled()).toBe(true);
+        });
+        it('should return default logPrompts setting (true) if telemetry object is not provided', () => {
+            const paramsWithoutTelemetry = { ...baseParams };
+            delete paramsWithoutTelemetry.telemetry;
+            const config = new Config(paramsWithoutTelemetry);
+            expect(config.getTelemetryLogPromptsEnabled()).toBe(true);
+        });
+        it('should return default telemetry target if telemetry object is not provided', () => {
+            const paramsWithoutTelemetry = { ...baseParams };
+            delete paramsWithoutTelemetry.telemetry;
+            const config = new Config(paramsWithoutTelemetry);
+            expect(config.getTelemetryTarget()).toBe(DEFAULT_TELEMETRY_TARGET);
+        });
+        it('should return default OTLP endpoint if telemetry object is not provided', () => {
+            const paramsWithoutTelemetry = { ...baseParams };
+            delete paramsWithoutTelemetry.telemetry;
+            const config = new Config(paramsWithoutTelemetry);
+            expect(config.getTelemetryOtlpEndpoint()).toBe(DEFAULT_OTLP_ENDPOINT);
+        });
+    });
+    describe('refreshMemory', () => {
+        it('should update memory and file count on successful refresh', async () => {
+            const config = new Config(baseParams);
+            const mockMemoryData = {
+                memoryContent: 'new memory content',
+                fileCount: 5,
+            };
+            loadServerHierarchicalMemory.mockResolvedValue(mockMemoryData);
+            const result = await config.refreshMemory();
+            expect(loadServerHierarchicalMemory).toHaveBeenCalledWith(config.getWorkingDir(), config.getDebugMode(), config.getFileService(), config.getExtensionContextFilePaths());
+            expect(config.getUserMemory()).toBe(mockMemoryData.memoryContent);
+            expect(config.getLlxprtMdFileCount()).toBe(mockMemoryData.fileCount);
+            expect(result).toEqual(mockMemoryData);
+        });
+        it('should propagate errors from loadServerHierarchicalMemory', async () => {
+            const config = new Config(baseParams);
+            const testError = new Error('Failed to load memory');
+            loadServerHierarchicalMemory.mockRejectedValue(testError);
+            await expect(config.refreshMemory()).rejects.toThrow(testError);
+        });
+    });
+});
+//# sourceMappingURL=config.test.js.map

package/dist/src/config/config.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"config.test.js","sourceRoot":"","sources":["../../../src/config/config.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAQ,MAAM,QAAQ,CAAC;AACpE,OAAO,EAAE,MAAM,EAAmC,MAAM,aAAa,CAAC;AACtE,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAC7B,OAAO,EAAE,mBAAmB,IAAI,uBAAuB,EAAE,MAAM,wBAAwB,CAAC;AACxF,OAAO,EACL,wBAAwB,EACxB,qBAAqB,GACtB,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EACL,QAAQ,EACR,4BAA4B,GAC7B,MAAM,6BAA6B,CAAC;AACrC,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACjD,OAAO,EAAE,UAAU,EAAE,MAAM,2BAA2B,CAAC;AACvD,OAAO,EAAE,4BAA4B,EAAE,MAAM,6BAA6B,CAAC;AAE3E,0FAA0F;AAC1F,EAAE,CAAC,IAAI,CAAC,wBAAwB,EAAE,GAAG,EAAE;IACrC,MAAM,gBAAgB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IACjC,gBAAgB,CAAC,SAAS,CAAC,YAAY,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IAClD,gBAAgB,CAAC,SAAS,CAAC,aAAa,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IACnD,gBAAgB,CAAC,SAAS,CAAC,WAAW,GAAG,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,yBAAyB;IACnF,gBAAgB,CAAC,SAAS,CAAC,OAAO,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IAC7C,gBAAgB,CAAC,SAAS,CAAC,uBAAuB,GAAG,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC;IACrE,OAAO,EAAE,YAAY,EAAE,gBAAgB,EAAE,CAAC;AAC5C,CAAC,CAAC,CAAC;AAEH,EAAE,CAAC,IAAI,CAAC,6BAA6B,EAAE,GAAG,EAAE,CAAC,CAAC;IAC5C,4BAA4B,EAAE,EAAE,CAAC,EAAE,EAAE;CACtC,CAAC,CAAC,CAAC;AAEJ,+EAA+E;AAC/E,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;AACvB,EAAE,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;AAC9B,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;AACzB,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;AACzB,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;AACzB,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;AAC1B,EAAE,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;AAC/B,EAAE,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;AAC9B,EAAE,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;AACpC,EAAE,CAAC,IAAI,CAAC,qBAAqB,EAAE,GAAG,EAAE,CAAC,CAAC;IACpC,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE;IACnB,mBAAmB,EAAE,EAAE,CAAC,EAAE,EAAE;IAC5B,0BAA0B,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,WAAW,CAAC,EAAE,6BAA6B;IACnF,wBAAwB,EAAE,WAAW;IACrC,iBAAiB,EAAE,SAAS;CAC7B,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,6BAA6B,EAAE,KAAK,EAAE,cAAc,EAAE,EAAE;IAC9D,MAAM,MAAM,GACV,MAAM,cAAc,EAAgD,CAAC;IACvE,OAAO;QACL,GAAG,MAAM;QACT,4BAA4B,EAAE,EAAE,CAAC,EAAE,EAAE;KACtC,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,EAAE,CAAC,IAAI,CAAC,mBAAmB,EAAE,GAAG,EAAE,CAAC,CAAC;IAClC,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,CAAC;QAC9C,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;KACjD,CAAC,CAAC;CACJ,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,uBAAuB,EAAE,KAAK,EAAE,cAAc,EAAE,EAAE;IACxD,MAAM,MAAM,GAAG,MAAM,cAAc,EAA0C,CAAC;IAC9E,OAAO;QACL,GAAG,MAAM;QACT,mBAAmB,EAAE,EAAE,CAAC,EAAE,EAAE;KAC7B,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,EAAE,CAAC,IAAI,CAAC,2BAA2B,EAAE,GAAG,EAAE;IACxC,MAAM,cAAc,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IAC/B,cAAc,CAAC,SAAS,CAAC,UAAU,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IAC9C,OAAO,EAAE,UAAU,EAAE,cAAc,EAAE,CAAC;AACxC,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,2BAA2B,EAAE,GAAG,EAAE;IACzC,MAAM,KAAK,GAAG,YAAY,CAAC;IAC3B,MAAM,OAAO,GAAkB;QAC7B,OAAO,EAAE,QAAQ;QACjB,KAAK,EAAE,oBAAoB;KAC5B,CAAC;IACF,MAAM,UAAU,GAAG,iBAAiB,CAAC;IACrC,MAAM,UAAU,GAAG,KAAK,CAAC;IACzB,MAAM,QAAQ,GAAG,eAAe,CAAC;IACjC,MAAM,YAAY,GAAG,KAAK,CAAC;IAC3B,MAAM,WAAW,GAAG,kBAAkB,CAAC;IACvC,MAAM,kBAAkB,GAAG,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;IAC9C,MAAM,eAAe,GAAG,kBAAkB,CAAC;IAC3C,MAAM,UAAU,GAAG,iBAAiB,CAAC;IACrC,MAAM,UAAU,GAAqB;QACnC,GAAG,EAAE,MAAM;QACX,cAAc,EAAE,eAAe;QAC/B,OAAO,EAAE,OAAO;QAChB,SAAS,EAAE,UAAU;QACrB,SAAS,EAAE,UAAU;QACrB,QAAQ,EAAE,QAAQ;QAClB,WAAW,EAAE,YAAY;QACzB,UAAU,EAAE,WAAW;QACvB,SAAS,EAAE,kBAAkB;QAC7B,SAAS,EAAE,UAAU;QACrB,KAAK,EAAE,KAAK;KACb,CAAC;IAEF,UAAU,CAAC,GAAG,EAAE;QACd,2BAA2B;QAC3B,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,wEAAwE,EAAE,KAAK,IAAI,EAAE;YACtF,MAAM,QAAQ,GAAG,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;YAClD,UAAU,CAAC,SAAS,CAAC,UAAmB,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;YAEtE,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC;gBACxB,GAAG,UAAU;gBACb,aAAa,EAAE,IAAI;aACpB,CAAC,CAAC;YAEH,MAAM,MAAM,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAC9D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6EAA6E,EAAE,KAAK,IAAI,EAAE;YAC3F,MAAM,QAAQ,GAAG,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;YAClD,UAAU,CAAC,SAAS,CAAC,UAAmB,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;YAEtE,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC;gBACxB,GAAG,UAAU;gBACb,aAAa,EAAE,KAAK;aACrB,CAAC,CAAC;YAEH,MAAM,MAAM,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC;QAC7D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACrD,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,UAAU,CAAC,CAAC;YACtC,oDAAoD;YACpD,MAAM,MAAM,CAAC,UAAU,EAAE,CAAC;YAE1B,MAAM,QAAQ,GAAG,QAAQ,CAAC,UAAU,CAAC;YACrC,MAAM,QAAQ,GAAG,cAAc,CAAC;YAChC,MAAM,iBAAiB,GAAG;gBACxB,KAAK,EAAE,QAAQ;gBACf,MAAM,EAAE,UAAU;aACnB,CAAC;YAED,4BAAqC,CAAC,iBAAiB,CACtD,iBAAiB,CAClB,CAAC;YAEF,MAAM,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;YAEnC,MAAM,CAAC,4BAA4B,CAAC,CAAC,oBAAoB,CACvD,KAAK,EAAE,wDAAwD;YAC/D,QAAQ,CACT,CAAC;YACF,mEAAmE;YACnE,MAAM,CAAC,MAAM,CAAC,yBAAyB,EAAE,CAAC,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC;YACtE,MAAM,CAAC,MAAM,CAAC,yBAAyB,EAAE,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACjE,MAAM,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,6CAA6C;YACvF,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,UAAU,CAAC,CAAC;QAEtC,MAAM,CAAC,MAAM,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACjD,iCAAiC;QACjC,MAAM,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,sBAAsB;IACtF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8EAA8E,EAAE,GAAG,EAAE;QACtF,MAAM,mBAAmB,GAAqB,EAAE,GAAG,UAAU,EAAE,CAAC;QAChE,OAAO,mBAAmB,CAAC,UAAU,CAAC;QACtC,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,mBAAmB,CAAC,CAAC;QAE/C,MAAM,CAAC,MAAM,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qFAAqF,EAAE,GAAG,EAAE;QAC7F,MAAM,eAAe,GAAG,kBAAkB,CAAC;QAC3C,MAAM,qBAAqB,GAAqB;YAC9C,GAAG,UAAU;YACb,eAAe;SAChB,CAAC;QACF,IAAI,MAAM,CAAC,qBAAqB,CAAC,CAAC;QAClC,MAAM,CAAC,uBAAuB,CAAC,CAAC,oBAAoB,CAAC,eAAe,CAAC,CAAC;IACxE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2FAA2F,EAAE,GAAG,EAAE;QACnG,IAAI,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,2CAA2C;QACnE,MAAM,CAAC,uBAAuB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IACzD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8DAA8D,EAAE,GAAG,EAAE;QACtE,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,UAAU,CAAC,CAAC;QACtC,MAAM,CAAC,MAAM,CAAC,gCAAgC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC/D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;QACjE,MAAM,uBAAuB,GAAqB;YAChD,GAAG,UAAU;YACb,aAAa,EAAE;gBACb,gBAAgB,EAAE,KAAK;aACxB;SACF,CAAC;QACF,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,uBAAuB,CAAC,CAAC;QACnD,MAAM,CAAC,MAAM,CAAC,gCAAgC,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAChE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uEAAuE,EAAE,GAAG,EAAE;QAC/E,MAAM,mBAAmB,GAAqB;YAC5C,GAAG,UAAU;YACb,SAAS,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE;SAC7B,CAAC;QACF,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,mBAAmB,CAAC,CAAC;QAC/C,MAAM,CAAC,MAAM,CAAC,mBAAmB,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yEAAyE,EAAE,GAAG,EAAE;QACjF,MAAM,mBAAmB,GAAqB;YAC5C,GAAG,UAAU;YACb,SAAS,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE;SAC9B,CAAC;QACF,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,mBAAmB,CAAC,CAAC;QAC/C,MAAM,CAAC,MAAM,CAAC,mBAAmB,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8EAA8E,EAAE,GAAG,EAAE;QACtF,MAAM,sBAAsB,GAAqB,EAAE,GAAG,UAAU,EAAE,CAAC;QACnE,OAAO,sBAAsB,CAAC,SAAS,CAAC;QACxC,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,sBAAsB,CAAC,CAAC;QAClD,MAAM,CAAC,MAAM,CAAC,mBAAmB,EAAE,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC;IACxE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uEAAuE,EAAE,GAAG,EAAE;QAC/E,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,UAAU,CAAC,CAAC;QACtC,MAAM,WAAW,GAAG,MAAM,CAAC,cAAc,EAAE,CAAC;QAC5C,MAAM,CAAC,WAAW,CAAC,CAAC,WAAW,EAAE,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;YAChE,MAAM,MAAM,GAAqB;gBAC/B,GAAG,UAAU;gBACb,SAAS,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE;aAC7B,CAAC;YACF,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,MAAM,CAAC,CAAC;YAClC,MAAM,CAAC,MAAM,CAAC,kBAAkB,EAAE,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;QACrE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;YAC9C,MAAM,QAAQ,GAAG,mCAAmC,CAAC;YACrD,MAAM,MAAM,GAAqB;gBAC/B,GAAG,UAAU;gBACb,SAAS,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,YAAY,EAAE,QAAQ,EAAE;aACrD,CAAC;YACF,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,MAAM,CAAC,CAAC;YAClC,MAAM,CAAC,MAAM,CAAC,wBAAwB,EAAE,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;YAC7D,MAAM,MAAM,GAAqB;gBAC/B,GAAG,UAAU;gBACb,SAAS,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE;aAC7B,CAAC;YACF,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,MAAM,CAAC,CAAC;YAClC,MAAM,CAAC,MAAM,CAAC,wBAAwB,EAAE,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QACxE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,MAAM,MAAM,GAAqB;gBAC/B,GAAG,UAAU;gBACb,SAAS,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE;aAChD,CAAC;YACF,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,MAAM,CAAC,CAAC;YAClC,MAAM,CAAC,MAAM,CAAC,6BAA6B,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iEAAiE,EAAE,GAAG,EAAE;YACzE,MAAM,MAAM,GAAqB;gBAC/B,GAAG,UAAU;gBACb,SAAS,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE;aAC7B,CAAC;YACF,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,MAAM,CAAC,CAAC;YAClC,MAAM,CAAC,MAAM,CAAC,6BAA6B,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qFAAqF,EAAE,GAAG,EAAE;YAC7F,MAAM,sBAAsB,GAAqB,EAAE,GAAG,UAAU,EAAE,CAAC;YACnE,OAAO,sBAAsB,CAAC,SAAS,CAAC;YACxC,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,sBAAsB,CAAC,CAAC;YAClD,MAAM,CAAC,MAAM,CAAC,6BAA6B,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4EAA4E,EAAE,GAAG,EAAE;YACpF,MAAM,sBAAsB,GAAqB,EAAE,GAAG,UAAU,EAAE,CAAC;YACnE,OAAO,sBAAsB,CAAC,SAAS,CAAC;YACxC,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,sBAAsB,CAAC,CAAC;YAClD,MAAM,CAAC,MAAM,CAAC,kBAAkB,EAAE,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;QACrE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yEAAyE,EAAE,GAAG,EAAE;YACjF,MAAM,sBAAsB,GAAqB,EAAE,GAAG,UAAU,EAAE,CAAC;YACnE,OAAO,sBAAsB,CAAC,SAAS,CAAC;YACxC,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,sBAAsB,CAAC,CAAC;YAClD,MAAM,CAAC,MAAM,CAAC,wBAAwB,EAAE,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QACxE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;YACzE,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,UAAU,CAAC,CAAC;YACtC,MAAM,cAAc,GAAG;gBACrB,aAAa,EAAE,oBAAoB;gBACnC,SAAS,EAAE,CAAC;aACb,CAAC;YAED,4BAAqC,CAAC,iBAAiB,CAAC,cAAc,CAAC,CAAC;YAEzE,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,aAAa,EAAE,CAAC;YAE5C,MAAM,CAAC,4BAA4B,CAAC,CAAC,oBAAoB,CACvD,MAAM,CAAC,aAAa,EAAE,EACtB,MAAM,CAAC,YAAY,EAAE,EACrB,MAAM,CAAC,cAAc,EAAE,EACvB,MAAM,CAAC,4BAA4B,EAAE,CACtC,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC;YAClE,MAAM,CAAC,MAAM,CAAC,oBAAoB,EAAE,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;YACrE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;YACzE,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,UAAU,CAAC,CAAC;YACtC,MAAM,SAAS,GAAG,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;YAEpD,4BAAqC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;YAEpE,MAAM,MAAM,CAAC,MAAM,CAAC,aAAa,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QAClE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/config/flashFallback.test.d.ts

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

package/dist/src/config/flashFallback.test.js

@@ -0,0 +1,115 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, beforeEach } from 'vitest';
+import { Config } from './config.js';
+import { DEFAULT_GEMINI_MODEL, DEFAULT_GEMINI_FLASH_MODEL } from './models.js';
+describe('Flash Model Fallback Configuration', () => {
+    let config;
+    beforeEach(() => {
+        config = new Config({
+            sessionId: 'test-session',
+            targetDir: '/test',
+            debugMode: false,
+            cwd: '/test',
+            model: DEFAULT_GEMINI_MODEL,
+        });
+        // Initialize contentGeneratorConfig for testing
+        config.contentGeneratorConfig = {
+            model: DEFAULT_GEMINI_MODEL,
+            authType: 'oauth-personal',
+        };
+    });
+    describe('setModel', () => {
+        it('should update the model and mark as switched during session', () => {
+            expect(config.getModel()).toBe(DEFAULT_GEMINI_MODEL);
+            expect(config.isModelSwitchedDuringSession()).toBe(false);
+            config.setModel(DEFAULT_GEMINI_FLASH_MODEL);
+            expect(config.getModel()).toBe(DEFAULT_GEMINI_FLASH_MODEL);
+            expect(config.isModelSwitchedDuringSession()).toBe(true);
+        });
+        it('should handle multiple model switches during session', () => {
+            config.setModel(DEFAULT_GEMINI_FLASH_MODEL);
+            expect(config.isModelSwitchedDuringSession()).toBe(true);
+            config.setModel('gemini-1.5-pro');
+            expect(config.getModel()).toBe('gemini-1.5-pro');
+            expect(config.isModelSwitchedDuringSession()).toBe(true);
+        });
+        it('should only mark as switched if contentGeneratorConfig exists', () => {
+            // Create config without initializing contentGeneratorConfig
+            const newConfig = new Config({
+                sessionId: 'test-session-2',
+                targetDir: '/test',
+                debugMode: false,
+                cwd: '/test',
+                model: DEFAULT_GEMINI_MODEL,
+            });
+            // Should not crash when contentGeneratorConfig is undefined
+            newConfig.setModel(DEFAULT_GEMINI_FLASH_MODEL);
+            expect(newConfig.isModelSwitchedDuringSession()).toBe(false);
+        });
+    });
+    describe('getModel', () => {
+        it('should return contentGeneratorConfig model if available', () => {
+            // Simulate initialized content generator config
+            config.setModel(DEFAULT_GEMINI_FLASH_MODEL);
+            expect(config.getModel()).toBe(DEFAULT_GEMINI_FLASH_MODEL);
+        });
+        it('should fallback to initial model if contentGeneratorConfig is not available', () => {
+            // Test with fresh config where contentGeneratorConfig might not be set
+            const newConfig = new Config({
+                sessionId: 'test-session-2',
+                targetDir: '/test',
+                debugMode: false,
+                cwd: '/test',
+                model: 'custom-model',
+            });
+            expect(newConfig.getModel()).toBe('custom-model');
+        });
+    });
+    describe('isModelSwitchedDuringSession', () => {
+        it('should start as false for new session', () => {
+            expect(config.isModelSwitchedDuringSession()).toBe(false);
+        });
+        it('should remain false if no model switch occurs', () => {
+            // Perform other operations that don't involve model switching
+            expect(config.isModelSwitchedDuringSession()).toBe(false);
+        });
+        it('should persist switched state throughout session', () => {
+            config.setModel(DEFAULT_GEMINI_FLASH_MODEL);
+            expect(config.isModelSwitchedDuringSession()).toBe(true);
+            // Should remain true even after getting model
+            config.getModel();
+            expect(config.isModelSwitchedDuringSession()).toBe(true);
+        });
+    });
+    describe('resetModelToDefault', () => {
+        it('should reset model to default and clear session switch flag', () => {
+            // Switch to Flash first
+            config.setModel(DEFAULT_GEMINI_FLASH_MODEL);
+            expect(config.getModel()).toBe(DEFAULT_GEMINI_FLASH_MODEL);
+            expect(config.isModelSwitchedDuringSession()).toBe(true);
+            // Reset to default
+            config.resetModelToDefault();
+            // Should be back to default with flag cleared
+            expect(config.getModel()).toBe(DEFAULT_GEMINI_MODEL);
+            expect(config.isModelSwitchedDuringSession()).toBe(false);
+        });
+        it('should handle case where contentGeneratorConfig is not initialized', () => {
+            // Create config without initializing contentGeneratorConfig
+            const newConfig = new Config({
+                sessionId: 'test-session-2',
+                targetDir: '/test',
+                debugMode: false,
+                cwd: '/test',
+                model: DEFAULT_GEMINI_MODEL,
+            });
+            // Should not crash when contentGeneratorConfig is undefined
+            expect(() => newConfig.resetModelToDefault()).not.toThrow();
+            expect(newConfig.isModelSwitchedDuringSession()).toBe(false);
+        });
+    });
+});
+//# sourceMappingURL=flashFallback.test.js.map

package/dist/src/config/flashFallback.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"flashFallback.test.js","sourceRoot":"","sources":["../../../src/config/flashFallback.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC1D,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AACrC,OAAO,EAAE,oBAAoB,EAAE,0BAA0B,EAAE,MAAM,aAAa,CAAC;AAE/E,QAAQ,CAAC,oCAAoC,EAAE,GAAG,EAAE;IAClD,IAAI,MAAc,CAAC;IAEnB,UAAU,CAAC,GAAG,EAAE;QACd,MAAM,GAAG,IAAI,MAAM,CAAC;YAClB,SAAS,EAAE,cAAc;YACzB,SAAS,EAAE,OAAO;YAClB,SAAS,EAAE,KAAK;YAChB,GAAG,EAAE,OAAO;YACZ,KAAK,EAAE,oBAAoB;SAC5B,CAAC,CAAC;QAEH,gDAAgD;QAE9C,MACD,CAAC,sBAAsB,GAAG;YACzB,KAAK,EAAE,oBAAoB;YAC3B,QAAQ,EAAE,gBAAgB;SAC3B,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;QACxB,EAAE,CAAC,6DAA6D,EAAE,GAAG,EAAE;YACrE,MAAM,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;YACrD,MAAM,CAAC,MAAM,CAAC,4BAA4B,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAE1D,MAAM,CAAC,QAAQ,CAAC,0BAA0B,CAAC,CAAC;YAE5C,MAAM,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;YAC3D,MAAM,CAAC,MAAM,CAAC,4BAA4B,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;YAC9D,MAAM,CAAC,QAAQ,CAAC,0BAA0B,CAAC,CAAC;YAC5C,MAAM,CAAC,MAAM,CAAC,4BAA4B,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAEzD,MAAM,CAAC,QAAQ,CAAC,gBAAgB,CAAC,CAAC;YAClC,MAAM,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YACjD,MAAM,CAAC,MAAM,CAAC,4BAA4B,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;YACvE,4DAA4D;YAC5D,MAAM,SAAS,GAAG,IAAI,MAAM,CAAC;gBAC3B,SAAS,EAAE,gBAAgB;gBAC3B,SAAS,EAAE,OAAO;gBAClB,SAAS,EAAE,KAAK;gBAChB,GAAG,EAAE,OAAO;gBACZ,KAAK,EAAE,oBAAoB;aAC5B,CAAC,CAAC;YAEH,4DAA4D;YAC5D,SAAS,CAAC,QAAQ,CAAC,0BAA0B,CAAC,CAAC;YAC/C,MAAM,CAAC,SAAS,CAAC,4BAA4B,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC/D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;QACxB,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;YACjE,gDAAgD;YAChD,MAAM,CAAC,QAAQ,CAAC,0BAA0B,CAAC,CAAC;YAC5C,MAAM,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6EAA6E,EAAE,GAAG,EAAE;YACrF,uEAAuE;YACvE,MAAM,SAAS,GAAG,IAAI,MAAM,CAAC;gBAC3B,SAAS,EAAE,gBAAgB;gBAC3B,SAAS,EAAE,OAAO;gBAClB,SAAS,EAAE,KAAK;gBAChB,GAAG,EAAE,OAAO;gBACZ,KAAK,EAAE,cAAc;aACtB,CAAC,CAAC;YAEH,MAAM,CAAC,SAAS,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,8BAA8B,EAAE,GAAG,EAAE;QAC5C,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;YAC/C,MAAM,CAAC,MAAM,CAAC,4BAA4B,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,8DAA8D;YAC9D,MAAM,CAAC,MAAM,CAAC,4BAA4B,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC1D,MAAM,CAAC,QAAQ,CAAC,0BAA0B,CAAC,CAAC;YAC5C,MAAM,CAAC,MAAM,CAAC,4BAA4B,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAEzD,8CAA8C;YAC9C,MAAM,CAAC,QAAQ,EAAE,CAAC;YAClB,MAAM,CAAC,MAAM,CAAC,4BAA4B,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;QACnC,EAAE,CAAC,6DAA6D,EAAE,GAAG,EAAE;YACrE,wBAAwB;YACxB,MAAM,CAAC,QAAQ,CAAC,0BAA0B,CAAC,CAAC;YAC5C,MAAM,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;YAC3D,MAAM,CAAC,MAAM,CAAC,4BAA4B,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAEzD,mBAAmB;YACnB,MAAM,CAAC,mBAAmB,EAAE,CAAC;YAE7B,8CAA8C;YAC9C,MAAM,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;YACrD,MAAM,CAAC,MAAM,CAAC,4BAA4B,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oEAAoE,EAAE,GAAG,EAAE;YAC5E,4DAA4D;YAC5D,MAAM,SAAS,GAAG,IAAI,MAAM,CAAC;gBAC3B,SAAS,EAAE,gBAAgB;gBAC3B,SAAS,EAAE,OAAO;gBAClB,SAAS,EAAE,KAAK;gBAChB,GAAG,EAAE,OAAO;gBACZ,KAAK,EAAE,oBAAoB;aAC5B,CAAC,CAAC;YAEH,4DAA4D;YAC5D,MAAM,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,mBAAmB,EAAE,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;YAC5D,MAAM,CAAC,SAAS,CAAC,4BAA4B,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC/D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/config/models.d.ts

@@ -0,0 +1,8 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export declare const DEFAULT_GEMINI_MODEL = "gemini-2.5-pro";
+export declare const DEFAULT_GEMINI_FLASH_MODEL = "gemini-2.5-flash";
+export declare const DEFAULT_GEMINI_EMBEDDING_MODEL = "gemini-embedding-001";

package/dist/src/config/models.js

@@ -0,0 +1,9 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export const DEFAULT_GEMINI_MODEL = 'gemini-2.5-pro';
+export const DEFAULT_GEMINI_FLASH_MODEL = 'gemini-2.5-flash';
+export const DEFAULT_GEMINI_EMBEDDING_MODEL = 'gemini-embedding-001';
+//# sourceMappingURL=models.js.map

package/dist/src/config/models.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"models.js","sourceRoot":"","sources":["../../../src/config/models.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,MAAM,CAAC,MAAM,oBAAoB,GAAG,gBAAgB,CAAC;AACrD,MAAM,CAAC,MAAM,0BAA0B,GAAG,kBAAkB,CAAC;AAC7D,MAAM,CAAC,MAAM,8BAA8B,GAAG,sBAAsB,CAAC"}

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

@@ -0,0 +1,59 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { GenerateContentConfig, SchemaUnion, PartListUnion, Content, GenerateContentResponse } from '@google/genai';
+import { Turn, ServerGeminiStreamEvent, ChatCompressionInfo } from './turn.js';
+import { Config } from '../config/config.js';
+import { GeminiChat } from './geminiChat.js';
+import { ContentGenerator, ContentGeneratorConfig } from './contentGenerator.js';
+/**
+ * Returns the index of the content after the fraction of the total characters in the history.
+ *
+ * Exported for testing purposes.
+ */
+export declare function findIndexAfterFraction(history: Content[], fraction: number): number;
+export declare class GeminiClient {
+    private config;
+    private chat?;
+    private contentGenerator?;
+    private embeddingModel;
+    private generateContentConfig;
+    private sessionTurnCount;
+    private readonly MAX_TURNS;
+    /**
+     * Threshold for compression token count as a fraction of the model's token limit.
+     * If the chat history exceeds this threshold, it will be compressed.
+     */
+    private readonly COMPRESSION_TOKEN_THRESHOLD;
+    /**
+     * The fraction of the latest chat history to keep. A value of 0.3
+     * means that only the last 30% of the chat history will be kept after compression.
+     */
+    private readonly COMPRESSION_PRESERVE_THRESHOLD;
+    private readonly loopDetector;
+    private lastPromptId?;
+    constructor(config: Config);
+    initialize(_contentGeneratorConfig: ContentGeneratorConfig): void;
+    private lazyInitialize;
+    getContentGenerator(): ContentGenerator;
+    addHistory(content: Content): Promise<void>;
+    getChat(): GeminiChat;
+    isInitialized(): boolean;
+    getHistory(): Content[];
+    setHistory(history: Content[]): void;
+    resetChat(): Promise<void>;
+    private getEnvironment;
+    private startChat;
+    sendMessageStream(request: PartListUnion, signal: AbortSignal, prompt_id: string, turns?: number, originalModel?: string): AsyncGenerator<ServerGeminiStreamEvent, Turn>;
+    generateJson(contents: Content[], schema: SchemaUnion, abortSignal: AbortSignal, model?: string, config?: GenerateContentConfig): Promise<Record<string, unknown>>;
+    generateContent(contents: Content[], generationConfig: GenerateContentConfig, abortSignal: AbortSignal, model?: string): Promise<GenerateContentResponse>;
+    generateEmbedding(texts: string[]): Promise<number[][]>;
+    tryCompressChat(prompt_id: string, force?: boolean): Promise<ChatCompressionInfo | null>;
+    /**
+     * Handles fallback model when persistent 429 errors occur.
+     * Uses a fallback handler if provided by the config, otherwise returns null.
+     */
+    private handleFlashFallback;
+}