@didim365/agent-cli-core 0.1.0

package/dist/src/core/contentGenerator.multiProvider.test.js

@@ -0,0 +1,314 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+/**
+ * M3.0.5 regression tests for multi-provider selection in createContentGenerator.
+ *
+ * Tests the 7 scenarios from the phase3 todolist covering all combinations
+ * of ENABLE_MULTI_PROVIDER, LLM_PROVIDER, and authType.
+ *
+ * @see docs/ai_adapter/todolist/phase3_provider_extension_todolist.md §3.0.5.5
+ */
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import { createContentGenerator, AuthType, isProviderIndependentGenerator, } from './contentGenerator.js';
+import { setMultiProviderOverride, clearMultiProviderOverride, } from '../providers/gemini/featureFlag.js';
+import { ProviderRegistry } from '../providers/registry.js';
+import { GoogleGenAI } from '@google/genai';
+import { LoggingContentGenerator } from './loggingContentGenerator.js';
+// ============================================================================
+// Module mocks
+// ============================================================================
+vi.mock('@google/genai');
+vi.mock('@anthropic-ai/sdk', () => ({
+    default: vi.fn().mockImplementation(() => ({
+        messages: {
+            create: vi.fn(),
+        },
+    })),
+}));
+vi.mock('openai', () => ({
+    default: vi.fn().mockImplementation(() => ({
+        chat: {
+            completions: {
+                create: vi.fn(),
+            },
+        },
+    })),
+}));
+vi.mock('../code_assist/codeAssist.js');
+vi.mock('./apiKeyCredentialStorage.js', () => ({
+    loadApiKey: vi.fn(),
+}));
+vi.mock('./fakeContentGenerator.js');
+// ============================================================================
+// Helpers
+// ============================================================================
+function createMockConfig(overrides = {}) {
+    return {
+        getModel: vi.fn().mockReturnValue('gemini-pro'),
+        setModel: vi.fn(),
+        getProxy: vi.fn().mockReturnValue(undefined),
+        getUsageStatisticsEnabled: vi.fn().mockReturnValue(false),
+        getPreviewFeatures: vi.fn().mockReturnValue(false),
+        getContentGeneratorConfig: vi.fn().mockReturnValue(undefined),
+        setLatestApiRequest: vi.fn(),
+        ...overrides,
+    };
+}
+/** Create a mock BaseAdapter for non-Gemini providers. */
+function createMockAdapter(providerName) {
+    return {
+        providerName,
+        capabilities: {
+            supportsStreaming: true,
+            supportsToolCalls: true,
+            supportsImageInput: false,
+            supportsImageGeneration: false,
+            supportsEmbedding: false,
+            supportsTokenCount: true,
+            supportsSystemMessage: true,
+            supportsThought: false,
+            maxContextLength: 200_000,
+            maxOutputTokens: 4_096,
+        },
+        generateContent: vi.fn().mockResolvedValue({
+            id: 'resp-1',
+            content: [{ type: 'text', text: 'mock response' }],
+            model: 'mock-model',
+            stopReason: 'end_turn',
+            usage: { promptTokens: 10, completionTokens: 5, totalTokens: 15 },
+        }),
+        generateContentStream: vi.fn(),
+        countTokens: vi.fn().mockResolvedValue({ totalTokens: 100 }),
+        embedContent: vi.fn(),
+    };
+}
+// ============================================================================
+// Tests
+// ============================================================================
+describe('Multi-provider selection in createContentGenerator', () => {
+    let registry;
+    const mockGoogleGenAI = {
+        models: {
+            generateContent: vi.fn(),
+            generateContentStream: vi.fn(),
+            countTokens: vi.fn(),
+        },
+    };
+    beforeEach(() => {
+        vi.clearAllMocks();
+        clearMultiProviderOverride();
+        registry = ProviderRegistry.getInstance();
+        registry.clear();
+        // Setup GoogleGenAI mock (for Gemini paths)
+        vi.mocked(GoogleGenAI).mockImplementation(() => mockGoogleGenAI);
+    });
+    afterEach(() => {
+        vi.unstubAllEnvs();
+        clearMultiProviderOverride();
+        registry.clear();
+    });
+    // ========================================================================
+    // Scenario 1: flag=false, no LLM_PROVIDER, USE_GEMINI → Gemini (legacy)
+    // ========================================================================
+    it('Scenario 1: flag=false, USE_GEMINI → Gemini via legacy path', async () => {
+        setMultiProviderOverride(false);
+        const generator = await createContentGenerator({ apiKey: 'test-key', authType: AuthType.USE_GEMINI }, createMockConfig());
+        expect(GoogleGenAI).toHaveBeenCalled();
+        expect(generator).toBeInstanceOf(LoggingContentGenerator);
+    });
+    // ========================================================================
+    // Scenario 2: flag=false, LLM_PROVIDER=claude, USE_GEMINI → Gemini
+    //             (flag off → legacy path, LLM_PROVIDER ignored)
+    // ========================================================================
+    it('Scenario 2: flag=false, LLM_PROVIDER=claude → Gemini (flag off ignores LLM_PROVIDER)', async () => {
+        setMultiProviderOverride(false);
+        vi.stubEnv('LLM_PROVIDER', 'claude');
+        vi.stubEnv('ANTHROPIC_API_KEY', 'test-claude-key');
+        const generator = await createContentGenerator({ apiKey: 'test-key', authType: AuthType.USE_GEMINI }, createMockConfig());
+        // Still uses GoogleGenAI (legacy path)
+        expect(GoogleGenAI).toHaveBeenCalled();
+        expect(generator).toBeInstanceOf(LoggingContentGenerator);
+    });
+    // ========================================================================
+    // Scenario 3: flag=true, no LLM_PROVIDER, USE_GEMINI → Gemini
+    //             (selectProvider returns Gemini → falls through to legacy)
+    // ========================================================================
+    it('Scenario 3: flag=true, USE_GEMINI → Gemini (falls through to legacy path)', async () => {
+        setMultiProviderOverride(true);
+        vi.stubEnv('GEMINI_API_KEY', 'test-key');
+        const generator = await createContentGenerator({ apiKey: 'test-key', authType: AuthType.USE_GEMINI }, createMockConfig());
+        expect(GoogleGenAI).toHaveBeenCalled();
+        expect(generator).toBeInstanceOf(LoggingContentGenerator);
+    });
+    // ========================================================================
+    // Scenario 4: flag=true, LLM_PROVIDER=claude → Claude
+    //             (non-Gemini → ProviderFactory path)
+    // ========================================================================
+    it('Scenario 4: flag=true, LLM_PROVIDER=claude → Claude via ProviderFactory', async () => {
+        setMultiProviderOverride(true);
+        vi.stubEnv('LLM_PROVIDER', 'claude');
+        vi.stubEnv('ANTHROPIC_API_KEY', 'test-claude-key');
+        // Register mock Claude adapter
+        const mockClaudeAdapter = createMockAdapter('claude');
+        registry.register('claude', () => mockClaudeAdapter);
+        const generator = await createContentGenerator({}, createMockConfig());
+        // Should NOT use GoogleGenAI
+        expect(GoogleGenAI).not.toHaveBeenCalled();
+        // Should return LoggingContentGenerator wrapping the adapter
+        expect(generator).toBeInstanceOf(LoggingContentGenerator);
+        // The wrapped generator should have llm* methods
+        expect(isProviderIndependentGenerator(generator)).toBe(true);
+    });
+    // ========================================================================
+    // Scenario 5: flag=true, LLM_PROVIDER=openai, USE_VERTEX_AI → OpenAI
+    //             (LLM_PROVIDER takes priority over authType)
+    // ========================================================================
+    it('Scenario 5: flag=true, LLM_PROVIDER=openai → OpenAI (LLM_PROVIDER priority over authType)', async () => {
+        setMultiProviderOverride(true);
+        vi.stubEnv('LLM_PROVIDER', 'openai');
+        vi.stubEnv('OPENAI_API_KEY', 'test-openai-key');
+        // Register mock OpenAI adapter
+        const mockOpenAIAdapter = createMockAdapter('openai');
+        registry.register('openai', () => mockOpenAIAdapter);
+        const generator = await createContentGenerator({ authType: AuthType.USE_VERTEX_AI }, createMockConfig());
+        // Should NOT use GoogleGenAI (LLM_PROVIDER=openai overrides authType)
+        expect(GoogleGenAI).not.toHaveBeenCalled();
+        expect(generator).toBeInstanceOf(LoggingContentGenerator);
+        expect(isProviderIndependentGenerator(generator)).toBe(true);
+    });
+    // ========================================================================
+    // Scenario 5b: flag=true, LLM_PROVIDER=openai → auto-bootstrap
+    //              (Review fix: OpenAI provider is auto-bootstrapped in contentGenerator)
+    // ========================================================================
+    it('Scenario 5b: flag=true, LLM_PROVIDER=openai → auto-bootstrap without manual registration', async () => {
+        setMultiProviderOverride(true);
+        vi.stubEnv('LLM_PROVIDER', 'openai');
+        vi.stubEnv('OPENAI_API_KEY', 'test-openai-key');
+        // Do NOT manually register OpenAI — rely on bootstrapOpenAiProvider() in contentGenerator
+        const generator = await createContentGenerator({}, createMockConfig());
+        expect(GoogleGenAI).not.toHaveBeenCalled();
+        expect(generator).toBeInstanceOf(LoggingContentGenerator);
+        expect(isProviderIndependentGenerator(generator)).toBe(true);
+    });
+    // ========================================================================
+    // Scenario 6: flag=true, no LLM_PROVIDER, USE_VERTEX_AI → Gemini
+    //             (authType implies Gemini → falls through to legacy)
+    // ========================================================================
+    it('Scenario 6: flag=true, USE_VERTEX_AI → Gemini (authType implies Gemini)', async () => {
+        setMultiProviderOverride(true);
+        vi.stubEnv('GEMINI_API_KEY', 'test-key');
+        const generator = await createContentGenerator({ apiKey: 'test-key', authType: AuthType.USE_VERTEX_AI, vertexai: true }, createMockConfig());
+        expect(GoogleGenAI).toHaveBeenCalled();
+        expect(generator).toBeInstanceOf(LoggingContentGenerator);
+    });
+    // ========================================================================
+    // Scenario 7: flag=true, no LLM_PROVIDER, no authType → Gemini (default)
+    //             (selectProvider returns Gemini as default fallback)
+    // ========================================================================
+    it('Scenario 7: flag=true, no LLM_PROVIDER, no authType → selectProvider defaults to Gemini', async () => {
+        setMultiProviderOverride(true);
+        // No LLM_PROVIDER, no authType
+        // selectProvider({}) → returns { type: ProviderType.Gemini }
+        // Falls through to existing code → throws because no authType is set
+        // The selection correctly returns Gemini (default), but without authType
+        // the existing Gemini code path throws. This is expected behavior —
+        // selectProvider correctly identifies Gemini as the default.
+        await expect(createContentGenerator({}, createMockConfig())).rejects.toThrow('Unsupported authType');
+    });
+    // ========================================================================
+    // Scenario 8: flag=true, LLM_PROVIDER=gemini, GEMINI_API_KEY, no authType
+    //             → authType backfilled to USE_GEMINI, Gemini via legacy path
+    //             (Review fix: selectProvider returns Gemini + apiKey → backfill)
+    // ========================================================================
+    it('Scenario 8: flag=true, LLM_PROVIDER=gemini, API key, no authType → backfill USE_GEMINI', async () => {
+        setMultiProviderOverride(true);
+        vi.stubEnv('LLM_PROVIDER', 'gemini');
+        vi.stubEnv('GEMINI_API_KEY', 'test-gemini-key');
+        const generator = await createContentGenerator({}, createMockConfig());
+        // Should use GoogleGenAI (Gemini path, not throw Unsupported authType)
+        expect(GoogleGenAI).toHaveBeenCalled();
+        expect(generator).toBeInstanceOf(LoggingContentGenerator);
+    });
+    // ========================================================================
+    // Scenario 9: flag=true, no LLM_PROVIDER, GEMINI_API_KEY, no authType
+    //             → selectProvider detects env key → authType backfilled
+    // ========================================================================
+    it('Scenario 9: flag=true, GEMINI_API_KEY, no authType → backfill USE_GEMINI from env key', async () => {
+        setMultiProviderOverride(true);
+        vi.stubEnv('GEMINI_API_KEY', 'test-gemini-key');
+        // No LLM_PROVIDER, no authType
+        const generator = await createContentGenerator({}, createMockConfig());
+        expect(GoogleGenAI).toHaveBeenCalled();
+        expect(generator).toBeInstanceOf(LoggingContentGenerator);
+    });
+    // ========================================================================
+    // Scenario 10: flag=true, LLM_PROVIDER=claude, no manual registration
+    //              → auto-bootstrap via bootstrapClaudeProvider() in contentGenerator
+    //              (Review fix: non-Gemini providers must be bootstrapped at runtime)
+    // ========================================================================
+    it('Scenario 10: flag=true, LLM_PROVIDER=claude → auto-bootstrap without manual registration', async () => {
+        setMultiProviderOverride(true);
+        vi.stubEnv('LLM_PROVIDER', 'claude');
+        vi.stubEnv('ANTHROPIC_API_KEY', 'test-claude-key');
+        // Do NOT manually register Claude — rely on bootstrapClaudeProvider() in contentGenerator
+        const generator = await createContentGenerator({}, createMockConfig());
+        // Should NOT use GoogleGenAI
+        expect(GoogleGenAI).not.toHaveBeenCalled();
+        // Should return LoggingContentGenerator wrapping the Claude adapter
+        expect(generator).toBeInstanceOf(LoggingContentGenerator);
+        // The wrapped generator should have llm* methods
+        expect(isProviderIndependentGenerator(generator)).toBe(true);
+    });
+    // ========================================================================
+    // Additional: wrapAdapterAsGenerator provides llm* methods
+    // ========================================================================
+    describe('wrapAdapterAsGenerator', () => {
+        it('should provide llmGenerateContent that delegates to adapter', async () => {
+            setMultiProviderOverride(true);
+            vi.stubEnv('LLM_PROVIDER', 'claude');
+            vi.stubEnv('ANTHROPIC_API_KEY', 'test-key');
+            const mockAdapter = createMockAdapter('claude');
+            registry.register('claude', () => mockAdapter);
+            const generator = await createContentGenerator({}, createMockConfig());
+            // Call llmGenerateContent through the LoggingContentGenerator
+            const response = await generator.llmGenerateContent({
+                model: 'claude-3-sonnet',
+                messages: [
+                    { role: 'user', content: [{ type: 'text', text: 'Hello' }] },
+                ],
+            }, 'prompt-1');
+            expect(response.content).toBeDefined();
+            expect(mockAdapter.generateContent).toHaveBeenCalled();
+        });
+        it('should throw on legacy generateContent for non-Gemini provider', async () => {
+            setMultiProviderOverride(true);
+            vi.stubEnv('LLM_PROVIDER', 'claude');
+            vi.stubEnv('ANTHROPIC_API_KEY', 'test-key');
+            const mockAdapter = createMockAdapter('claude');
+            registry.register('claude', () => mockAdapter);
+            const generator = await createContentGenerator({}, createMockConfig());
+            // Legacy generateContent should throw for non-Gemini
+            await expect(generator.generateContent({ model: 'claude-3', contents: 'test' }, 'prompt-1')).rejects.toThrow('does not support legacy Gemini API');
+        });
+    });
+    // ========================================================================
+    // Bootstrap idempotency
+    // ========================================================================
+    describe('bootstrap idempotency', () => {
+        it('should bootstrap Gemini provider on multi-provider path without errors', async () => {
+            setMultiProviderOverride(true);
+            vi.stubEnv('LLM_PROVIDER', 'claude');
+            vi.stubEnv('ANTHROPIC_API_KEY', 'test-key');
+            const mockAdapter = createMockAdapter('claude');
+            registry.register('claude', () => mockAdapter);
+            // First call — bootstraps Gemini
+            await createContentGenerator({}, createMockConfig());
+            // Second call — should not throw (idempotent bootstrap)
+            await createContentGenerator({}, createMockConfig());
+        });
+    });
+});
+//# sourceMappingURL=contentGenerator.multiProvider.test.js.map

package/dist/src/core/contentGenerator.multiProvider.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"contentGenerator.multiProvider.test.js","sourceRoot":"","sources":["../../../src/core/contentGenerator.multiProvider.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH;;;;;;;GAOG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EACL,sBAAsB,EACtB,QAAQ,EACR,8BAA8B,GAC/B,MAAM,uBAAuB,CAAC;AAE/B,OAAO,EACL,wBAAwB,EACxB,0BAA0B,GAC3B,MAAM,oCAAoC,CAAC;AAC5C,OAAO,EAAE,gBAAgB,EAAE,MAAM,0BAA0B,CAAC;AAC5D,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAC5C,OAAO,EAAE,uBAAuB,EAAE,MAAM,8BAA8B,CAAC;AAEvE,+EAA+E;AAC/E,eAAe;AACf,+EAA+E;AAE/E,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;AACzB,EAAE,CAAC,IAAI,CAAC,mBAAmB,EAAE,GAAG,EAAE,CAAC,CAAC;IAClC,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,CAAC;QACzC,QAAQ,EAAE;YACR,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;SAChB;KACF,CAAC,CAAC;CACJ,CAAC,CAAC,CAAC;AACJ,EAAE,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,EAAE,CAAC,CAAC;IACvB,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,CAAC;QACzC,IAAI,EAAE;YACJ,WAAW,EAAE;gBACX,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;aAChB;SACF;KACF,CAAC,CAAC;CACJ,CAAC,CAAC,CAAC;AACJ,EAAE,CAAC,IAAI,CAAC,8BAA8B,CAAC,CAAC;AACxC,EAAE,CAAC,IAAI,CAAC,8BAA8B,EAAE,GAAG,EAAE,CAAC,CAAC;IAC7C,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE;CACpB,CAAC,CAAC,CAAC;AACJ,EAAE,CAAC,IAAI,CAAC,2BAA2B,CAAC,CAAC;AAErC,+EAA+E;AAC/E,UAAU;AACV,+EAA+E;AAE/E,SAAS,gBAAgB,CAAC,YAA6B,EAAE;IACvD,OAAO;QACL,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,YAAY,CAAC;QAC/C,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;QACjB,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,SAAS,CAAC;QAC5C,yBAAyB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,KAAK,CAAC;QACzD,kBAAkB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,KAAK,CAAC;QAClD,yBAAyB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,SAAS,CAAC;QAC7D,mBAAmB,EAAE,EAAE,CAAC,EAAE,EAAE;QAC5B,GAAG,SAAS;KACQ,CAAC;AACzB,CAAC;AAED,0DAA0D;AAC1D,SAAS,iBAAiB,CAAC,YAAoB;IAC7C,OAAO;QACL,YAAY;QACZ,YAAY,EAAE;YACZ,iBAAiB,EAAE,IAAI;YACvB,iBAAiB,EAAE,IAAI;YACvB,kBAAkB,EAAE,KAAK;YACzB,uBAAuB,EAAE,KAAK;YAC9B,iBAAiB,EAAE,KAAK;YACxB,kBAAkB,EAAE,IAAI;YACxB,qBAAqB,EAAE,IAAI;YAC3B,eAAe,EAAE,KAAK;YACtB,gBAAgB,EAAE,OAAO;YACzB,eAAe,EAAE,KAAK;SACvB;QACD,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;YACzC,EAAE,EAAE,QAAQ;YACZ,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC;YAClD,KAAK,EAAE,YAAY;YACnB,UAAU,EAAE,UAAU;YACtB,KAAK,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,gBAAgB,EAAE,CAAC,EAAE,WAAW,EAAE,EAAE,EAAE;SAClE,CAAC;QACF,qBAAqB,EAAE,EAAE,CAAC,EAAE,EAAE;QAC9B,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,WAAW,EAAE,GAAG,EAAE,CAAC;QAC5D,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE;KACtB,CAAC;AACJ,CAAC;AAED,+EAA+E;AAC/E,QAAQ;AACR,+EAA+E;AAE/E,QAAQ,CAAC,oDAAoD,EAAE,GAAG,EAAE;IAClE,IAAI,QAA0B,CAAC;IAC/B,MAAM,eAAe,GAAG;QACtB,MAAM,EAAE;YACN,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;YACxB,qBAAqB,EAAE,EAAE,CAAC,EAAE,EAAE;YAC9B,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE;SACrB;KACF,CAAC;IAEF,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,0BAA0B,EAAE,CAAC;QAC7B,QAAQ,GAAG,gBAAgB,CAAC,WAAW,EAAE,CAAC;QAC1C,QAAQ,CAAC,KAAK,EAAE,CAAC;QAEjB,4CAA4C;QAC5C,EAAE,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,eAAwB,CAAC,CAAC;IAC5E,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,0BAA0B,EAAE,CAAC;QAC7B,QAAQ,CAAC,KAAK,EAAE,CAAC;IACnB,CAAC,CAAC,CAAC;IAEH,2EAA2E;IAC3E,wEAAwE;IACxE,2EAA2E;IAC3E,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;QAC3E,wBAAwB,CAAC,KAAK,CAAC,CAAC;QAEhC,MAAM,SAAS,GAAG,MAAM,sBAAsB,CAC5C,EAAE,MAAM,EAAE,UAAU,EAAE,QAAQ,EAAE,QAAQ,CAAC,UAAU,EAAE,EACrD,gBAAgB,EAAE,CACnB,CAAC;QAEF,MAAM,CAAC,WAAW,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACvC,MAAM,CAAC,SAAS,CAAC,CAAC,cAAc,CAAC,uBAAuB,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;IAEH,2EAA2E;IAC3E,mEAAmE;IACnE,6DAA6D;IAC7D,2EAA2E;IAC3E,EAAE,CAAC,sFAAsF,EAAE,KAAK,IAAI,EAAE;QACpG,wBAAwB,CAAC,KAAK,CAAC,CAAC;QAChC,EAAE,CAAC,OAAO,CAAC,cAAc,EAAE,QAAQ,CAAC,CAAC;QACrC,EAAE,CAAC,OAAO,CAAC,mBAAmB,EAAE,iBAAiB,CAAC,CAAC;QAEnD,MAAM,SAAS,GAAG,MAAM,sBAAsB,CAC5C,EAAE,MAAM,EAAE,UAAU,EAAE,QAAQ,EAAE,QAAQ,CAAC,UAAU,EAAE,EACrD,gBAAgB,EAAE,CACnB,CAAC;QAEF,uCAAuC;QACvC,MAAM,CAAC,WAAW,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACvC,MAAM,CAAC,SAAS,CAAC,CAAC,cAAc,CAAC,uBAAuB,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;IAEH,2EAA2E;IAC3E,8DAA8D;IAC9D,wEAAwE;IACxE,2EAA2E;IAC3E,EAAE,CAAC,2EAA2E,EAAE,KAAK,IAAI,EAAE;QACzF,wBAAwB,CAAC,IAAI,CAAC,CAAC;QAC/B,EAAE,CAAC,OAAO,CAAC,gBAAgB,EAAE,UAAU,CAAC,CAAC;QAEzC,MAAM,SAAS,GAAG,MAAM,sBAAsB,CAC5C,EAAE,MAAM,EAAE,UAAU,EAAE,QAAQ,EAAE,QAAQ,CAAC,UAAU,EAAE,EACrD,gBAAgB,EAAE,CACnB,CAAC;QAEF,MAAM,CAAC,WAAW,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACvC,MAAM,CAAC,SAAS,CAAC,CAAC,cAAc,CAAC,uBAAuB,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;IAEH,2EAA2E;IAC3E,sDAAsD;IACtD,kDAAkD;IAClD,2EAA2E;IAC3E,EAAE,CAAC,yEAAyE,EAAE,KAAK,IAAI,EAAE;QACvF,wBAAwB,CAAC,IAAI,CAAC,CAAC;QAC/B,EAAE,CAAC,OAAO,CAAC,cAAc,EAAE,QAAQ,CAAC,CAAC;QACrC,EAAE,CAAC,OAAO,CAAC,mBAAmB,EAAE,iBAAiB,CAAC,CAAC;QAEnD,+BAA+B;QAC/B,MAAM,iBAAiB,GAAG,iBAAiB,CAAC,QAAQ,CAAC,CAAC;QACtD,QAAQ,CAAC,QAAQ,CAAC,QAAQ,EAAE,GAAG,EAAE,CAAC,iBAA0B,CAAC,CAAC;QAE9D,MAAM,SAAS,GAAG,MAAM,sBAAsB,CAAC,EAAE,EAAE,gBAAgB,EAAE,CAAC,CAAC;QAEvE,6BAA6B;QAC7B,MAAM,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC3C,6DAA6D;QAC7D,MAAM,CAAC,SAAS,CAAC,CAAC,cAAc,CAAC,uBAAuB,CAAC,CAAC;QAC1D,iDAAiD;QACjD,MAAM,CAAC,8BAA8B,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC/D,CAAC,CAAC,CAAC;IAEH,2EAA2E;IAC3E,qEAAqE;IACrE,0DAA0D;IAC1D,2EAA2E;IAC3E,EAAE,CAAC,2FAA2F,EAAE,KAAK,IAAI,EAAE;QACzG,wBAAwB,CAAC,IAAI,CAAC,CAAC;QAC/B,EAAE,CAAC,OAAO,CAAC,cAAc,EAAE,QAAQ,CAAC,CAAC;QACrC,EAAE,CAAC,OAAO,CAAC,gBAAgB,EAAE,iBAAiB,CAAC,CAAC;QAEhD,+BAA+B;QAC/B,MAAM,iBAAiB,GAAG,iBAAiB,CAAC,QAAQ,CAAC,CAAC;QACtD,QAAQ,CAAC,QAAQ,CAAC,QAAQ,EAAE,GAAG,EAAE,CAAC,iBAA0B,CAAC,CAAC;QAE9D,MAAM,SAAS,GAAG,MAAM,sBAAsB,CAC5C,EAAE,QAAQ,EAAE,QAAQ,CAAC,aAAa,EAAE,EACpC,gBAAgB,EAAE,CACnB,CAAC;QAEF,sEAAsE;QACtE,MAAM,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC3C,MAAM,CAAC,SAAS,CAAC,CAAC,cAAc,CAAC,uBAAuB,CAAC,CAAC;QAC1D,MAAM,CAAC,8BAA8B,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC/D,CAAC,CAAC,CAAC;IAEH,2EAA2E;IAC3E,+DAA+D;IAC/D,sFAAsF;IACtF,2EAA2E;IAC3E,EAAE,CAAC,0FAA0F,EAAE,KAAK,IAAI,EAAE;QACxG,wBAAwB,CAAC,IAAI,CAAC,CAAC;QAC/B,EAAE,CAAC,OAAO,CAAC,cAAc,EAAE,QAAQ,CAAC,CAAC;QACrC,EAAE,CAAC,OAAO,CAAC,gBAAgB,EAAE,iBAAiB,CAAC,CAAC;QAEhD,0FAA0F;QAC1F,MAAM,SAAS,GAAG,MAAM,sBAAsB,CAAC,EAAE,EAAE,gBAAgB,EAAE,CAAC,CAAC;QAEvE,MAAM,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC3C,MAAM,CAAC,SAAS,CAAC,CAAC,cAAc,CAAC,uBAAuB,CAAC,CAAC;QAC1D,MAAM,CAAC,8BAA8B,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC/D,CAAC,CAAC,CAAC;IAEH,2EAA2E;IAC3E,iEAAiE;IACjE,kEAAkE;IAClE,2EAA2E;IAC3E,EAAE,CAAC,yEAAyE,EAAE,KAAK,IAAI,EAAE;QACvF,wBAAwB,CAAC,IAAI,CAAC,CAAC;QAC/B,EAAE,CAAC,OAAO,CAAC,gBAAgB,EAAE,UAAU,CAAC,CAAC;QAEzC,MAAM,SAAS,GAAG,MAAM,sBAAsB,CAC5C,EAAE,MAAM,EAAE,UAAU,EAAE,QAAQ,EAAE,QAAQ,CAAC,aAAa,EAAE,QAAQ,EAAE,IAAI,EAAE,EACxE,gBAAgB,EAAE,CACnB,CAAC;QAEF,MAAM,CAAC,WAAW,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACvC,MAAM,CAAC,SAAS,CAAC,CAAC,cAAc,CAAC,uBAAuB,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;IAEH,2EAA2E;IAC3E,yEAAyE;IACzE,kEAAkE;IAClE,2EAA2E;IAC3E,EAAE,CAAC,yFAAyF,EAAE,KAAK,IAAI,EAAE;QACvG,wBAAwB,CAAC,IAAI,CAAC,CAAC;QAC/B,+BAA+B;QAC/B,6DAA6D;QAC7D,qEAAqE;QAErE,yEAAyE;QACzE,oEAAoE;QACpE,6DAA6D;QAC7D,MAAM,MAAM,CACV,sBAAsB,CAAC,EAAE,EAAE,gBAAgB,EAAE,CAAC,CAC/C,CAAC,OAAO,CAAC,OAAO,CAAC,sBAAsB,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,2EAA2E;IAC3E,0EAA0E;IAC1E,0EAA0E;IAC1E,8EAA8E;IAC9E,2EAA2E;IAC3E,EAAE,CAAC,wFAAwF,EAAE,KAAK,IAAI,EAAE;QACtG,wBAAwB,CAAC,IAAI,CAAC,CAAC;QAC/B,EAAE,CAAC,OAAO,CAAC,cAAc,EAAE,QAAQ,CAAC,CAAC;QACrC,EAAE,CAAC,OAAO,CAAC,gBAAgB,EAAE,iBAAiB,CAAC,CAAC;QAEhD,MAAM,SAAS,GAAG,MAAM,sBAAsB,CAAC,EAAE,EAAE,gBAAgB,EAAE,CAAC,CAAC;QAEvE,uEAAuE;QACvE,MAAM,CAAC,WAAW,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACvC,MAAM,CAAC,SAAS,CAAC,CAAC,cAAc,CAAC,uBAAuB,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;IAEH,2EAA2E;IAC3E,sEAAsE;IACtE,qEAAqE;IACrE,2EAA2E;IAC3E,EAAE,CAAC,uFAAuF,EAAE,KAAK,IAAI,EAAE;QACrG,wBAAwB,CAAC,IAAI,CAAC,CAAC;QAC/B,EAAE,CAAC,OAAO,CAAC,gBAAgB,EAAE,iBAAiB,CAAC,CAAC;QAChD,+BAA+B;QAE/B,MAAM,SAAS,GAAG,MAAM,sBAAsB,CAAC,EAAE,EAAE,gBAAgB,EAAE,CAAC,CAAC;QAEvE,MAAM,CAAC,WAAW,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACvC,MAAM,CAAC,SAAS,CAAC,CAAC,cAAc,CAAC,uBAAuB,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;IAEH,2EAA2E;IAC3E,sEAAsE;IACtE,kFAAkF;IAClF,kFAAkF;IAClF,2EAA2E;IAC3E,EAAE,CAAC,0FAA0F,EAAE,KAAK,IAAI,EAAE;QACxG,wBAAwB,CAAC,IAAI,CAAC,CAAC;QAC/B,EAAE,CAAC,OAAO,CAAC,cAAc,EAAE,QAAQ,CAAC,CAAC;QACrC,EAAE,CAAC,OAAO,CAAC,mBAAmB,EAAE,iBAAiB,CAAC,CAAC;QAEnD,0FAA0F;QAC1F,MAAM,SAAS,GAAG,MAAM,sBAAsB,CAAC,EAAE,EAAE,gBAAgB,EAAE,CAAC,CAAC;QAEvE,6BAA6B;QAC7B,MAAM,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC3C,oEAAoE;QACpE,MAAM,CAAC,SAAS,CAAC,CAAC,cAAc,CAAC,uBAAuB,CAAC,CAAC;QAC1D,iDAAiD;QACjD,MAAM,CAAC,8BAA8B,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC/D,CAAC,CAAC,CAAC;IAEH,2EAA2E;IAC3E,2DAA2D;IAC3D,2EAA2E;IAC3E,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;QACtC,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;YAC3E,wBAAwB,CAAC,IAAI,CAAC,CAAC;YAC/B,EAAE,CAAC,OAAO,CAAC,cAAc,EAAE,QAAQ,CAAC,CAAC;YACrC,EAAE,CAAC,OAAO,CAAC,mBAAmB,EAAE,UAAU,CAAC,CAAC;YAE5C,MAAM,WAAW,GAAG,iBAAiB,CAAC,QAAQ,CAAC,CAAC;YAChD,QAAQ,CAAC,QAAQ,CAAC,QAAQ,EAAE,GAAG,EAAE,CAAC,WAAoB,CAAC,CAAC;YAExD,MAAM,SAAS,GAAG,MAAM,sBAAsB,CAAC,EAAE,EAAE,gBAAgB,EAAE,CAAC,CAAC;YAEvE,8DAA8D;YAC9D,MAAM,QAAQ,GAAG,MAAM,SAAS,CAAC,kBAAmB,CAClD;gBACE,KAAK,EAAE,iBAAiB;gBACxB,QAAQ,EAAE;oBACR,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE;iBAC7D;aACF,EACD,UAAU,CACX,CAAC;YAEF,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,CAAC;YACvC,MAAM,CAAC,WAAW,CAAC,eAAe,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACzD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gEAAgE,EAAE,KAAK,IAAI,EAAE;YAC9E,wBAAwB,CAAC,IAAI,CAAC,CAAC;YAC/B,EAAE,CAAC,OAAO,CAAC,cAAc,EAAE,QAAQ,CAAC,CAAC;YACrC,EAAE,CAAC,OAAO,CAAC,mBAAmB,EAAE,UAAU,CAAC,CAAC;YAE5C,MAAM,WAAW,GAAG,iBAAiB,CAAC,QAAQ,CAAC,CAAC;YAChD,QAAQ,CAAC,QAAQ,CAAC,QAAQ,EAAE,GAAG,EAAE,CAAC,WAAoB,CAAC,CAAC;YAExD,MAAM,SAAS,GAAG,MAAM,sBAAsB,CAAC,EAAE,EAAE,gBAAgB,EAAE,CAAC,CAAC;YAEvE,qDAAqD;YACrD,MAAM,MAAM,CACV,SAAS,CAAC,eAAe,CACvB,EAAE,KAAK,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAW,EAChD,UAAU,CACX,CACF,CAAC,OAAO,CAAC,OAAO,CAAC,oCAAoC,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,2EAA2E;IAC3E,wBAAwB;IACxB,2EAA2E;IAC3E,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACrC,EAAE,CAAC,wEAAwE,EAAE,KAAK,IAAI,EAAE;YACtF,wBAAwB,CAAC,IAAI,CAAC,CAAC;YAC/B,EAAE,CAAC,OAAO,CAAC,cAAc,EAAE,QAAQ,CAAC,CAAC;YACrC,EAAE,CAAC,OAAO,CAAC,mBAAmB,EAAE,UAAU,CAAC,CAAC;YAE5C,MAAM,WAAW,GAAG,iBAAiB,CAAC,QAAQ,CAAC,CAAC;YAChD,QAAQ,CAAC,QAAQ,CAAC,QAAQ,EAAE,GAAG,EAAE,CAAC,WAAoB,CAAC,CAAC;YAExD,iCAAiC;YACjC,MAAM,sBAAsB,CAAC,EAAE,EAAE,gBAAgB,EAAE,CAAC,CAAC;YACrD,wDAAwD;YACxD,MAAM,sBAAsB,CAAC,EAAE,EAAE,gBAAgB,EAAE,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/core/contentGenerator.test.d.ts

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

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

@@ -0,0 +1,299 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import { createContentGenerator, AuthType, createContentGeneratorConfig, } from './contentGenerator.js';
+import { createCodeAssistContentGenerator } from '../code_assist/codeAssist.js';
+import { GoogleGenAI } from '@google/genai';
+import { LoggingContentGenerator } from './loggingContentGenerator.js';
+import { loadApiKey } from './apiKeyCredentialStorage.js';
+import { FakeContentGenerator } from './fakeContentGenerator.js';
+import { RecordingContentGenerator } from './recordingContentGenerator.js';
+vi.mock('../code_assist/codeAssist.js');
+vi.mock('@google/genai');
+vi.mock('./apiKeyCredentialStorage.js', () => ({
+    loadApiKey: vi.fn(),
+}));
+vi.mock('./fakeContentGenerator.js');
+const mockConfig = {
+    getModel: vi.fn().mockReturnValue('gemini-pro'),
+    getProxy: vi.fn().mockReturnValue(undefined),
+    getUsageStatisticsEnabled: vi.fn().mockReturnValue(true),
+    getPreviewFeatures: vi.fn().mockReturnValue(false),
+};
+describe('createContentGenerator', () => {
+    beforeEach(() => {
+        vi.clearAllMocks();
+    });
+    afterEach(() => {
+        vi.unstubAllEnvs();
+    });
+    it('should create a FakeContentGenerator', async () => {
+        const mockGenerator = {};
+        vi.mocked(FakeContentGenerator.fromFile).mockResolvedValue(mockGenerator);
+        const fakeResponsesFile = 'fake/responses.yaml';
+        const mockConfigWithFake = {
+            fakeResponses: fakeResponsesFile,
+        };
+        const generator = await createContentGenerator({
+            authType: AuthType.USE_GEMINI,
+        }, mockConfigWithFake);
+        expect(FakeContentGenerator.fromFile).toHaveBeenCalledWith(fakeResponsesFile);
+        expect(generator).toEqual(new LoggingContentGenerator(mockGenerator, mockConfigWithFake));
+    });
+    it('should create a RecordingContentGenerator', async () => {
+        const fakeResponsesFile = 'fake/responses.yaml';
+        const recordResponsesFile = 'record/responses.yaml';
+        const mockConfigWithRecordResponses = {
+            fakeResponses: fakeResponsesFile,
+            recordResponses: recordResponsesFile,
+        };
+        const generator = await createContentGenerator({
+            authType: AuthType.USE_GEMINI,
+        }, mockConfigWithRecordResponses);
+        expect(generator).toBeInstanceOf(RecordingContentGenerator);
+    });
+    it('should create a CodeAssistContentGenerator when AuthType is LOGIN_WITH_GOOGLE', async () => {
+        const mockGenerator = {};
+        vi.mocked(createCodeAssistContentGenerator).mockResolvedValue(mockGenerator);
+        const generator = await createContentGenerator({
+            authType: AuthType.LOGIN_WITH_GOOGLE,
+        }, mockConfig);
+        expect(createCodeAssistContentGenerator).toHaveBeenCalled();
+        expect(generator).toEqual(new LoggingContentGenerator(mockGenerator, mockConfig));
+    });
+    it('should create a CodeAssistContentGenerator when AuthType is COMPUTE_ADC', async () => {
+        const mockGenerator = {};
+        vi.mocked(createCodeAssistContentGenerator).mockResolvedValue(mockGenerator);
+        const generator = await createContentGenerator({
+            authType: AuthType.COMPUTE_ADC,
+        }, mockConfig);
+        expect(createCodeAssistContentGenerator).toHaveBeenCalled();
+        expect(generator).toEqual(new LoggingContentGenerator(mockGenerator, mockConfig));
+    });
+    it('should create a GoogleGenAI content generator', async () => {
+        const mockConfig = {
+            getModel: vi.fn().mockReturnValue('gemini-pro'),
+            getProxy: vi.fn().mockReturnValue(undefined),
+            getUsageStatisticsEnabled: () => true,
+            getPreviewFeatures: vi.fn().mockReturnValue(false),
+        };
+        // Set a fixed version for testing
+        vi.stubEnv('CLI_VERSION', '1.2.3');
+        const mockGenerator = {
+            models: {},
+        };
+        vi.mocked(GoogleGenAI).mockImplementation(() => mockGenerator);
+        const generator = await createContentGenerator({
+            apiKey: 'test-api-key',
+            authType: AuthType.USE_GEMINI,
+        }, mockConfig);
+        expect(GoogleGenAI).toHaveBeenCalledWith({
+            apiKey: 'test-api-key',
+            vertexai: undefined,
+            httpOptions: {
+                headers: {
+                    'User-Agent': expect.stringContaining('GeminiCLI/1.2.3/gemini-pro'),
+                    'x-gemini-api-privileged-user-id': expect.any(String),
+                },
+            },
+        });
+        expect(generator).toEqual(new LoggingContentGenerator(mockGenerator.models, mockConfig));
+    });
+    it('should include custom headers from GEMINI_CLI_CUSTOM_HEADERS for Code Assist requests', async () => {
+        const mockGenerator = {};
+        vi.mocked(createCodeAssistContentGenerator).mockResolvedValue(mockGenerator);
+        vi.stubEnv('GEMINI_CLI_CUSTOM_HEADERS', 'X-Test-Header: test-value, Another-Header: another value');
+        await createContentGenerator({
+            authType: AuthType.LOGIN_WITH_GOOGLE,
+        }, mockConfig);
+        expect(createCodeAssistContentGenerator).toHaveBeenCalledWith({
+            headers: expect.objectContaining({
+                'User-Agent': expect.any(String),
+                'X-Test-Header': 'test-value',
+                'Another-Header': 'another value',
+            }),
+        }, AuthType.LOGIN_WITH_GOOGLE, mockConfig, undefined);
+    });
+    it('should include custom headers from GEMINI_CLI_CUSTOM_HEADERS for GoogleGenAI requests without inferring auth mechanism', async () => {
+        const mockConfig = {
+            getModel: vi.fn().mockReturnValue('gemini-pro'),
+            getProxy: vi.fn().mockReturnValue(undefined),
+            getUsageStatisticsEnabled: () => false,
+            getPreviewFeatures: vi.fn().mockReturnValue(false),
+        };
+        const mockGenerator = {
+            models: {},
+        };
+        vi.mocked(GoogleGenAI).mockImplementation(() => mockGenerator);
+        vi.stubEnv('GEMINI_CLI_CUSTOM_HEADERS', 'X-Test-Header: test, Another: value');
+        await createContentGenerator({
+            apiKey: 'test-api-key',
+            authType: AuthType.USE_GEMINI,
+        }, mockConfig);
+        expect(GoogleGenAI).toHaveBeenCalledWith({
+            apiKey: 'test-api-key',
+            vertexai: undefined,
+            httpOptions: {
+                headers: expect.objectContaining({
+                    'User-Agent': expect.any(String),
+                    'X-Test-Header': 'test',
+                    Another: 'value',
+                }),
+            },
+        });
+        expect(GoogleGenAI).toHaveBeenCalledWith(expect.not.objectContaining({
+            httpOptions: {
+                headers: expect.objectContaining({
+                    Authorization: expect.any(String),
+                }),
+            },
+        }));
+    });
+    it('should pass api key as Authorization Header when GEMINI_API_KEY_AUTH_MECHANISM is set to bearer', async () => {
+        const mockConfig = {
+            getModel: vi.fn().mockReturnValue('gemini-pro'),
+            getProxy: vi.fn().mockReturnValue(undefined),
+            getUsageStatisticsEnabled: () => false,
+            getPreviewFeatures: vi.fn().mockReturnValue(false),
+        };
+        const mockGenerator = {
+            models: {},
+        };
+        vi.mocked(GoogleGenAI).mockImplementation(() => mockGenerator);
+        vi.stubEnv('GEMINI_API_KEY_AUTH_MECHANISM', 'bearer');
+        await createContentGenerator({
+            apiKey: 'test-api-key',
+            authType: AuthType.USE_GEMINI,
+        }, mockConfig);
+        expect(GoogleGenAI).toHaveBeenCalledWith({
+            apiKey: 'test-api-key',
+            vertexai: undefined,
+            httpOptions: {
+                headers: expect.objectContaining({
+                    'User-Agent': expect.any(String),
+                    Authorization: 'Bearer test-api-key',
+                }),
+            },
+        });
+    });
+    it('should not pass api key as Authorization Header when GEMINI_API_KEY_AUTH_MECHANISM is not set (default behavior)', async () => {
+        const mockConfig = {
+            getModel: vi.fn().mockReturnValue('gemini-pro'),
+            getProxy: vi.fn().mockReturnValue(undefined),
+            getUsageStatisticsEnabled: () => false,
+            getPreviewFeatures: vi.fn().mockReturnValue(false),
+        };
+        const mockGenerator = {
+            models: {},
+        };
+        vi.mocked(GoogleGenAI).mockImplementation(() => mockGenerator);
+        // GEMINI_API_KEY_AUTH_MECHANISM is not stubbed, so it will be undefined, triggering default 'x-goog-api-key'
+        await createContentGenerator({
+            apiKey: 'test-api-key',
+            authType: AuthType.USE_GEMINI,
+        }, mockConfig);
+        expect(GoogleGenAI).toHaveBeenCalledWith({
+            apiKey: 'test-api-key',
+            vertexai: undefined,
+            httpOptions: {
+                headers: expect.objectContaining({
+                    'User-Agent': expect.any(String),
+                }),
+            },
+        });
+        // Explicitly assert that Authorization header is NOT present
+        expect(GoogleGenAI).toHaveBeenCalledWith(expect.not.objectContaining({
+            httpOptions: {
+                headers: expect.objectContaining({
+                    Authorization: expect.any(String),
+                }),
+            },
+        }));
+    });
+    it('should create a GoogleGenAI content generator with client install id logging disabled', async () => {
+        const mockConfig = {
+            getModel: vi.fn().mockReturnValue('gemini-pro'),
+            getUsageStatisticsEnabled: () => false,
+            getPreviewFeatures: vi.fn().mockReturnValue(false),
+        };
+        const mockGenerator = {
+            models: {},
+        };
+        vi.mocked(GoogleGenAI).mockImplementation(() => mockGenerator);
+        const generator = await createContentGenerator({
+            apiKey: 'test-api-key',
+            authType: AuthType.USE_GEMINI,
+        }, mockConfig);
+        expect(GoogleGenAI).toHaveBeenCalledWith({
+            apiKey: 'test-api-key',
+            vertexai: undefined,
+            httpOptions: {
+                headers: {
+                    'User-Agent': expect.any(String),
+                },
+            },
+        });
+        expect(generator).toEqual(new LoggingContentGenerator(mockGenerator.models, mockConfig));
+    });
+});
+describe('createContentGeneratorConfig', () => {
+    const mockConfig = {
+        getModel: vi.fn().mockReturnValue('gemini-pro'),
+        setModel: vi.fn(),
+        flashFallbackHandler: vi.fn(),
+        getProxy: vi.fn(),
+    };
+    beforeEach(() => {
+        // Reset modules to re-evaluate imports and environment variables
+        vi.resetModules();
+        vi.clearAllMocks();
+    });
+    afterEach(() => {
+        vi.unstubAllEnvs();
+    });
+    it('should configure for Gemini using GEMINI_API_KEY when set', async () => {
+        vi.stubEnv('GEMINI_API_KEY', 'env-gemini-key');
+        const config = await createContentGeneratorConfig(mockConfig, AuthType.USE_GEMINI);
+        expect(config.apiKey).toBe('env-gemini-key');
+        expect(config.vertexai).toBe(false);
+    });
+    it('should not configure for Gemini if GEMINI_API_KEY is empty', async () => {
+        vi.stubEnv('GEMINI_API_KEY', '');
+        const config = await createContentGeneratorConfig(mockConfig, AuthType.USE_GEMINI);
+        expect(config.apiKey).toBeUndefined();
+        expect(config.vertexai).toBeUndefined();
+    });
+    it('should not configure for Gemini if GEMINI_API_KEY is not set and storage is empty', async () => {
+        vi.stubEnv('GEMINI_API_KEY', '');
+        vi.mocked(loadApiKey).mockResolvedValue(null);
+        const config = await createContentGeneratorConfig(mockConfig, AuthType.USE_GEMINI);
+        expect(config.apiKey).toBeUndefined();
+        expect(config.vertexai).toBeUndefined();
+    });
+    it('should configure for Vertex AI using GOOGLE_API_KEY when set', async () => {
+        vi.stubEnv('GOOGLE_API_KEY', 'env-google-key');
+        const config = await createContentGeneratorConfig(mockConfig, AuthType.USE_VERTEX_AI);
+        expect(config.apiKey).toBe('env-google-key');
+        expect(config.vertexai).toBe(true);
+    });
+    it('should configure for Vertex AI using GCP project and location when set', async () => {
+        vi.stubEnv('GOOGLE_API_KEY', undefined);
+        vi.stubEnv('GOOGLE_CLOUD_PROJECT', 'env-gcp-project');
+        vi.stubEnv('GOOGLE_CLOUD_LOCATION', 'env-gcp-location');
+        const config = await createContentGeneratorConfig(mockConfig, AuthType.USE_VERTEX_AI);
+        expect(config.vertexai).toBe(true);
+        expect(config.apiKey).toBeUndefined();
+    });
+    it('should not configure for Vertex AI if required env vars are empty', async () => {
+        vi.stubEnv('GOOGLE_API_KEY', '');
+        vi.stubEnv('GOOGLE_CLOUD_PROJECT', '');
+        vi.stubEnv('GOOGLE_CLOUD_LOCATION', '');
+        const config = await createContentGeneratorConfig(mockConfig, AuthType.USE_VERTEX_AI);
+        expect(config.apiKey).toBeUndefined();
+        expect(config.vertexai).toBeUndefined();
+    });
+});
+//# sourceMappingURL=contentGenerator.test.js.map