@google/gemini-cli-core 0.18.0-nightly.20251120.2231497b1 → 0.18.0-preview.0

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

@@ -4,7 +4,7 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
-import { ApiError } from '@google/genai';
+import { ApiError, ThinkingLevel } from '@google/genai';
 import { GeminiChat, InvalidStreamError, StreamEventType, SYNTHETIC_THOUGHT_SIGNATURE, } from './geminiChat.js';
 import { setSimulate429 } from '../utils/testUtils.js';
 import { DEFAULT_GEMINI_FLASH_MODEL, DEFAULT_GEMINI_MODEL, PREVIEW_GEMINI_MODEL, } from '../config/models.js';
@@ -65,7 +65,6 @@ describe('GeminiChat', () => {
     let mockContentGenerator;
     let chat;
     let mockConfig;
-    const config = {};
     beforeEach(() => {
         vi.clearAllMocks();
         vi.mocked(uiTelemetryService.setLastPromptTokenCount).mockClear();
@@ -104,6 +103,17 @@ describe('GeminiChat', () => {
             }),
             getContentGenerator: vi.fn().mockReturnValue(mockContentGenerator),
             getRetryFetchErrors: vi.fn().mockReturnValue(false),
+            modelConfigService: {
+                getResolvedConfig: vi.fn().mockImplementation((modelConfigKey) => ({
+                    model: modelConfigKey.model,
+                    generateContentConfig: {
+                        temperature: 0,
+                        thinkingConfig: {
+                            thinkingBudget: 1000,
+                        },
+                    },
+                })),
+            },
             isPreviewModelBypassMode: vi.fn().mockReturnValue(false),
             setPreviewModelBypassMode: vi.fn(),
             isPreviewModelFallbackMode: vi.fn().mockReturnValue(false),
@@ -113,7 +123,7 @@ describe('GeminiChat', () => {
         // Disable 429 simulation for tests
         setSimulate429(false);
         // Reset history for each test by creating a new instance
-        chat = new GeminiChat(mockConfig, config, []);
+        chat = new GeminiChat(mockConfig);
     });
     afterEach(() => {
         vi.restoreAllMocks();
@@ -125,12 +135,12 @@ describe('GeminiChat', () => {
                 { role: 'user', parts: [{ text: 'Hello' }] },
                 { role: 'model', parts: [{ text: 'Hi there' }] },
             ];
-            const chatWithHistory = new GeminiChat(mockConfig, config, history);
+            const chatWithHistory = new GeminiChat(mockConfig, '', [], history);
             const estimatedTokens = Math.ceil(JSON.stringify(history).length / 4);
             expect(chatWithHistory.getLastPromptTokenCount()).toBe(estimatedTokens);
         });
         it('should initialize lastPromptTokenCount for empty history', () => {
-            const chatEmpty = new GeminiChat(mockConfig, config, []);
+            const chatEmpty = new GeminiChat(mockConfig);
             expect(chatEmpty.getLastPromptTokenCount()).toBe(Math.ceil(JSON.stringify([]).length / 4));
         });
     });
@@ -163,7 +173,7 @@ describe('GeminiChat', () => {
             vi.mocked(mockContentGenerator.generateContentStream).mockResolvedValue(streamWithToolCall);
             // 2. Action & Assert: The stream processing should complete without throwing an error
             // because the presence of a tool call makes the empty final chunk acceptable.
-            const stream = await chat.sendMessageStream('test-model', { message: 'test message' }, 'prompt-id-tool-call-empty-end');
+            const stream = await chat.sendMessageStream({ model: 'test-model' }, 'test message', 'prompt-id-tool-call-empty-end', new AbortController().signal);
             await expect((async () => {
                 for await (const _ of stream) {
                     /* consume stream */
@@ -203,7 +213,7 @@ describe('GeminiChat', () => {
             })();
             vi.mocked(mockContentGenerator.generateContentStream).mockResolvedValue(streamWithNoFinish);
             // 2. Action & Assert: The stream should fail because there's no finish reason.
-            const stream = await chat.sendMessageStream('gemini-2.0-flash', { message: 'test message' }, 'prompt-id-no-finish-empty-end');
+            const stream = await chat.sendMessageStream({ model: 'gemini-2.0-flash' }, 'test message', 'prompt-id-no-finish-empty-end', new AbortController().signal);
             await expect((async () => {
                 for await (const _ of stream) {
                     /* consume stream */
@@ -238,7 +248,7 @@ describe('GeminiChat', () => {
             })();
             vi.mocked(mockContentGenerator.generateContentStream).mockResolvedValue(streamWithInvalidEnd);
             // 2. Action & Assert: The stream should complete without throwing an error.
-            const stream = await chat.sendMessageStream('test-model', { message: 'test message' }, 'prompt-id-valid-then-invalid-end');
+            const stream = await chat.sendMessageStream({ model: 'test-model' }, 'test message', 'prompt-id-valid-then-invalid-end', new AbortController().signal);
             await expect((async () => {
                 for await (const _ of stream) {
                     /* consume stream */
@@ -273,7 +283,7 @@ describe('GeminiChat', () => {
             })();
             vi.mocked(mockContentGenerator.generateContentStream).mockResolvedValue(multiChunkStream);
             // 2. Action: Send a message and consume the stream.
-            const stream = await chat.sendMessageStream('test-model', { message: 'test message' }, 'prompt-id-empty-chunk-consolidation');
+            const stream = await chat.sendMessageStream({ model: 'test-model' }, 'test message', 'prompt-id-empty-chunk-consolidation', new AbortController().signal);
             for await (const _ of stream) {
                 // Consume the stream
             }
@@ -321,7 +331,7 @@ describe('GeminiChat', () => {
             })();
             vi.mocked(mockContentGenerator.generateContentStream).mockResolvedValue(multiChunkStream);
             // 2. Action: Send a message and consume the stream.
-            const stream = await chat.sendMessageStream('test-model', { message: 'test message' }, 'prompt-id-multi-chunk');
+            const stream = await chat.sendMessageStream({ model: 'test-model' }, 'test message', 'prompt-id-multi-chunk', new AbortController().signal);
             for await (const _ of stream) {
                 // Consume the stream to trigger history recording.
             }
@@ -357,7 +367,7 @@ describe('GeminiChat', () => {
             })();
             vi.mocked(mockContentGenerator.generateContentStream).mockResolvedValue(mixedContentStream);
             // 2. Action: Send a message and fully consume the stream to trigger history recording.
-            const stream = await chat.sendMessageStream('test-model', { message: 'test message' }, 'prompt-id-mixed-chunk');
+            const stream = await chat.sendMessageStream({ model: 'test-model' }, 'test message', 'prompt-id-mixed-chunk', new AbortController().signal);
             for await (const _ of stream) {
                 // This loop consumes the stream.
             }
@@ -385,7 +395,7 @@ describe('GeminiChat', () => {
                     ],
                 };
             })());
-            const stream = await chat.sendMessageStream(PREVIEW_GEMINI_MODEL, { message: 'test' }, 'prompt-id-fast-retry');
+            const stream = await chat.sendMessageStream({ model: PREVIEW_GEMINI_MODEL }, 'test', 'prompt-id-fast-retry', new AbortController().signal);
             for await (const _ of stream) {
                 // consume stream
             }
@@ -405,7 +415,7 @@ describe('GeminiChat', () => {
                     ],
                 };
             })());
-            const stream = await chat.sendMessageStream(DEFAULT_GEMINI_FLASH_MODEL, { message: 'test' }, 'prompt-id-normal-retry');
+            const stream = await chat.sendMessageStream({ model: DEFAULT_GEMINI_FLASH_MODEL }, 'test', 'prompt-id-normal-retry', new AbortController().signal);
             for await (const _ of stream) {
                 // consume stream
             }
@@ -437,7 +447,7 @@ describe('GeminiChat', () => {
             }));
             // ACT
             const consumeStream = async () => {
-                const stream = await chat.sendMessageStream(PREVIEW_GEMINI_MODEL, { message: 'test' }, 'prompt-id-bypass');
+                const stream = await chat.sendMessageStream({ model: PREVIEW_GEMINI_MODEL }, 'test', 'prompt-id-bypass', new AbortController().signal);
                 // Consume the stream to trigger execution
                 for await (const _ of stream) {
                     // do nothing
@@ -484,14 +494,12 @@ describe('GeminiChat', () => {
             })();
             vi.mocked(mockContentGenerator.generateContentStream).mockResolvedValue(emptyStreamResponse);
             // 3. Action: Send the function response back to the model and consume the stream.
-            const stream = await chat.sendMessageStream('gemini-2.0-flash', {
-                message: {
-                    functionResponse: {
-                        name: 'find_restaurant',
-                        response: { name: 'Vesuvio' },
-                    },
+            const stream = await chat.sendMessageStream({ model: 'gemini-2.0-flash' }, {
+                functionResponse: {
+                    name: 'find_restaurant',
+                    response: { name: 'Vesuvio' },
                 },
-            }, 'prompt-id-stream-1');
+            }, 'prompt-id-stream-1', new AbortController().signal);
             // 4. Assert: The stream processing should throw an InvalidStreamError.
             await expect((async () => {
                 for await (const _ of stream) {
@@ -522,7 +530,7 @@ describe('GeminiChat', () => {
                 };
             })();
             vi.mocked(mockContentGenerator.generateContentStream).mockResolvedValue(streamWithToolCall);
-            const stream = await chat.sendMessageStream('test-model', { message: 'test' }, 'prompt-id-1');
+            const stream = await chat.sendMessageStream({ model: 'test-model' }, 'test message', 'prompt-id-1', new AbortController().signal);
             // Should not throw an error
             await expect((async () => {
                 for await (const _ of stream) {
@@ -546,7 +554,7 @@ describe('GeminiChat', () => {
                 };
             })();
             vi.mocked(mockContentGenerator.generateContentStream).mockResolvedValue(streamWithoutFinishReason);
-            const stream = await chat.sendMessageStream('gemini-2.0-flash', { message: 'test' }, 'prompt-id-1');
+            const stream = await chat.sendMessageStream({ model: 'gemini-2.0-flash' }, 'test message', 'prompt-id-1', new AbortController().signal);
             await expect((async () => {
                 for await (const _ of stream) {
                     // consume stream
@@ -569,7 +577,7 @@ describe('GeminiChat', () => {
                 };
             })();
             vi.mocked(mockContentGenerator.generateContentStream).mockResolvedValue(streamWithEmptyResponse);
-            const stream = await chat.sendMessageStream('gemini-2.0-flash', { message: 'test' }, 'prompt-id-1');
+            const stream = await chat.sendMessageStream({ model: 'gemini-2.0-flash' }, 'test message', 'prompt-id-1', new AbortController().signal);
             await expect((async () => {
                 for await (const _ of stream) {
                     // consume stream
@@ -592,7 +600,7 @@ describe('GeminiChat', () => {
                 };
             })();
             vi.mocked(mockContentGenerator.generateContentStream).mockResolvedValue(validStream);
-            const stream = await chat.sendMessageStream('test-model', { message: 'test' }, 'prompt-id-1');
+            const stream = await chat.sendMessageStream({ model: 'test-model' }, 'test message', 'prompt-id-1', new AbortController().signal);
             // Should not throw an error
             await expect((async () => {
                 for await (const _ of stream) {
@@ -616,7 +624,7 @@ describe('GeminiChat', () => {
                 };
             })();
             vi.mocked(mockContentGenerator.generateContentStream).mockResolvedValue(streamWithMalformedFunctionCall);
-            const stream = await chat.sendMessageStream('gemini-2.5-pro', { message: 'test' }, 'prompt-id-malformed');
+            const stream = await chat.sendMessageStream({ model: 'gemini-2.5-pro' }, 'test', 'prompt-id-malformed', new AbortController().signal);
             // Should throw an error
             await expect((async () => {
                 for await (const _ of stream) {
@@ -650,7 +658,7 @@ describe('GeminiChat', () => {
                 };
             })());
             // 2. Send a message
-            const stream = await chat.sendMessageStream('gemini-2.5-pro', { message: 'test retry' }, 'prompt-id-retry-malformed');
+            const stream = await chat.sendMessageStream({ model: 'gemini-2.5-pro' }, 'test retry', 'prompt-id-retry-malformed', new AbortController().signal);
             const events = [];
             for await (const event of stream) {
                 events.push(event);
@@ -688,7 +696,7 @@ describe('GeminiChat', () => {
                 };
             })();
             vi.mocked(mockContentGenerator.generateContentStream).mockResolvedValue(response);
-            const stream = await chat.sendMessageStream('test-model', { message: 'hello' }, 'prompt-id-1');
+            const stream = await chat.sendMessageStream({ model: 'test-model' }, 'hello', 'prompt-id-1', new AbortController().signal);
             for await (const _ of stream) {
                 // consume stream
             }
@@ -700,9 +708,43 @@ describe('GeminiChat', () => {
                         parts: [{ text: 'hello' }],
                     },
                 ],
-                config: {},
+                config: {
+                    systemInstruction: '',
+                    tools: [],
+                    temperature: 0,
+                    thinkingConfig: {
+                        thinkingBudget: 1000,
+                    },
+                    abortSignal: expect.any(AbortSignal),
+                },
             }, 'prompt-id-1');
         });
+        it('should use thinkingLevel and remove thinkingBudget for gemini-3 models', async () => {
+            const response = (async function* () {
+                yield {
+                    candidates: [
+                        {
+                            content: { parts: [{ text: 'response' }], role: 'model' },
+                            finishReason: 'STOP',
+                        },
+                    ],
+                };
+            })();
+            vi.mocked(mockContentGenerator.generateContentStream).mockResolvedValue(response);
+            const stream = await chat.sendMessageStream({ model: 'gemini-3-test-only-model-string-for-testing' }, 'hello', 'prompt-id-thinking-level', new AbortController().signal);
+            for await (const _ of stream) {
+                // consume stream
+            }
+            expect(mockContentGenerator.generateContentStream).toHaveBeenCalledWith(expect.objectContaining({
+                model: 'gemini-3-test-only-model-string-for-testing',
+                config: expect.objectContaining({
+                    thinkingConfig: {
+                        thinkingBudget: undefined,
+                        thinkingLevel: ThinkingLevel.HIGH,
+                    },
+                }),
+            }), 'prompt-id-thinking-level');
+        });
     });
     describe('addHistory', () => {
         it('should add a new content item to the history', () => {
@@ -740,7 +782,7 @@ describe('GeminiChat', () => {
                     candidates: [{ content: { parts: [{ text: '' }] } }],
                 };
             })());
-            const stream = await chat.sendMessageStream('gemini-1.5-pro', { message: 'test' }, 'prompt-id-no-retry');
+            const stream = await chat.sendMessageStream({ model: 'gemini-1.5-pro' }, 'test', 'prompt-id-no-retry', new AbortController().signal);
             await expect((async () => {
                 for await (const _ of stream) {
                     // Must loop to trigger the internal logic that throws.
@@ -773,7 +815,7 @@ describe('GeminiChat', () => {
                 };
             })());
             // ACT: Send a message and collect all events from the stream.
-            const stream = await chat.sendMessageStream('gemini-2.0-flash', { message: 'test' }, 'prompt-id-yield-retry');
+            const stream = await chat.sendMessageStream({ model: 'gemini-2.0-flash' }, 'test message', 'prompt-id-yield-retry', new AbortController().signal);
             const events = [];
             for await (const event of stream) {
                 events.push(event);
@@ -805,7 +847,7 @@ describe('GeminiChat', () => {
                     ],
                 };
             })());
-            const stream = await chat.sendMessageStream('gemini-2.0-flash', { message: 'test' }, 'prompt-id-retry-success');
+            const stream = await chat.sendMessageStream({ model: 'gemini-2.0-flash' }, 'test', 'prompt-id-retry-success', new AbortController().signal);
             const chunks = [];
             for await (const chunk of stream) {
                 chunks.push(chunk);
@@ -856,7 +898,7 @@ describe('GeminiChat', () => {
                     ],
                 };
             })());
-            const stream = await chat.sendMessageStream('gemini-2.0-flash', { message: 'test', config: { temperature: 0.5 } }, 'prompt-id-retry-temperature');
+            const stream = await chat.sendMessageStream({ model: 'gemini-2.0-flash' }, 'test message', 'prompt-id-retry-temperature', new AbortController().signal);
             for await (const _ of stream) {
                 // consume stream
             }
@@ -864,7 +906,7 @@ describe('GeminiChat', () => {
             // First call should have original temperature
             expect(mockContentGenerator.generateContentStream).toHaveBeenNthCalledWith(1, expect.objectContaining({
                 config: expect.objectContaining({
-                    temperature: 0.5,
+                    temperature: 0,
                 }),
             }), 'prompt-id-retry-temperature');
             // Second call (retry) should have temperature 1
@@ -887,7 +929,7 @@ describe('GeminiChat', () => {
                     ],
                 };
             })());
-            const stream = await chat.sendMessageStream('gemini-2.0-flash', { message: 'test' }, 'prompt-id-retry-fail');
+            const stream = await chat.sendMessageStream({ model: 'gemini-2.0-flash' }, 'test', 'prompt-id-retry-fail', new AbortController().signal);
             await expect(async () => {
                 for await (const _ of stream) {
                     // Must loop to trigger the internal logic that throws.
@@ -936,7 +978,7 @@ describe('GeminiChat', () => {
             it('should not retry on 400 Bad Request errors', async () => {
                 const error400 = new ApiError({ message: 'Bad Request', status: 400 });
                 vi.mocked(mockContentGenerator.generateContentStream).mockRejectedValue(error400);
-                const stream = await chat.sendMessageStream('gemini-2.0-flash', { message: 'test' }, 'prompt-id-400');
+                const stream = await chat.sendMessageStream({ model: 'gemini-2.0-flash' }, 'test message', 'prompt-id-400', new AbortController().signal);
                 await expect((async () => {
                     for await (const _ of stream) {
                         /* consume stream */
@@ -959,7 +1001,7 @@ describe('GeminiChat', () => {
                         ],
                     };
                 })());
-                const stream = await chat.sendMessageStream('test-model', { message: 'test' }, 'prompt-id-429-retry');
+                const stream = await chat.sendMessageStream({ model: 'test-model' }, 'test message', 'prompt-id-429-retry', new AbortController().signal);
                 const events = [];
                 for await (const event of stream) {
                     events.push(event);
@@ -988,7 +1030,7 @@ describe('GeminiChat', () => {
                         ],
                     };
                 })());
-                const stream = await chat.sendMessageStream('test-model', { message: 'test' }, 'prompt-id-500-retry');
+                const stream = await chat.sendMessageStream({ model: 'test-model' }, 'test message', 'prompt-id-500-retry', new AbortController().signal);
                 const events = [];
                 for await (const event of stream) {
                     events.push(event);
@@ -1024,7 +1066,7 @@ describe('GeminiChat', () => {
                         throw error;
                     }
                 });
-                const stream = await chat.sendMessageStream('test-model', { message: 'test' }, 'prompt-id-fetch-error-retry');
+                const stream = await chat.sendMessageStream({ model: 'test-model' }, 'test message', 'prompt-id-fetch-error-retry', new AbortController().signal);
                 const events = [];
                 for await (const event of stream) {
                     events.push(event);
@@ -1067,7 +1109,7 @@ describe('GeminiChat', () => {
             };
         })());
         // 3. Send a new message
-        const stream = await chat.sendMessageStream('gemini-2.0-flash', { message: 'Second question' }, 'prompt-id-retry-existing');
+        const stream = await chat.sendMessageStream({ model: 'gemini-2.0-flash' }, 'Second question', 'prompt-id-retry-existing', new AbortController().signal);
         for await (const _ of stream) {
             // consume stream
         }
@@ -1119,7 +1161,7 @@ describe('GeminiChat', () => {
             };
         })());
         // 2. Call the method and consume the stream.
-        const stream = await chat.sendMessageStream('gemini-2.0-flash', { message: 'test empty stream' }, 'prompt-id-empty-stream');
+        const stream = await chat.sendMessageStream({ model: 'gemini-2.0-flash' }, 'test empty stream', 'prompt-id-empty-stream', new AbortController().signal);
         const chunks = [];
         for await (const chunk of stream) {
             chunks.push(chunk);
@@ -1180,11 +1222,11 @@ describe('GeminiChat', () => {
             .mockResolvedValueOnce(firstStreamGenerator)
             .mockResolvedValueOnce(secondStreamGenerator);
         // 3. Start the first stream and consume only the first chunk to pause it
-        const firstStream = await chat.sendMessageStream('test-model', { message: 'first' }, 'prompt-1');
+        const firstStream = await chat.sendMessageStream({ model: 'test-model' }, 'first', 'prompt-1', new AbortController().signal);
         const firstStreamIterator = firstStream[Symbol.asyncIterator]();
         await firstStreamIterator.next();
         // 4. While the first stream is paused, start the second call. It will block.
-        const secondStreamPromise = chat.sendMessageStream('test-model', { message: 'second' }, 'prompt-2');
+        const secondStreamPromise = chat.sendMessageStream({ model: 'test-model' }, 'second', 'prompt-2', new AbortController().signal);
         // 5. Assert that only one API call has been made so far.
         expect(mockContentGenerator.generateContentStream).toHaveBeenCalledTimes(1);
         // 6. Unblock and fully consume the first stream to completion.
@@ -1224,7 +1266,7 @@ describe('GeminiChat', () => {
             vi.mocked(mockContentGenerator.generateContentStream).mockImplementation(async () => (async function* () {
                 yield mockResponse;
             })());
-            const stream = await chat.sendMessageStream('test-model', { message: 'test' }, 'prompt-id-res3');
+            const stream = await chat.sendMessageStream({ model: 'test-model' }, 'test message', 'prompt-id-res3', new AbortController().signal);
             for await (const _ of stream) {
                 // consume stream
             }
@@ -1287,7 +1329,7 @@ describe('GeminiChat', () => {
                 isInFallbackModeSpy.mockReturnValue(true);
                 return true; // Signal retry
             });
-            const stream = await chat.sendMessageStream('test-model', { message: 'trigger 429' }, 'prompt-id-fb1');
+            const stream = await chat.sendMessageStream({ model: 'test-model' }, 'trigger 429', 'prompt-id-fb1', new AbortController().signal);
             // Consume stream to trigger logic
             for await (const _ of stream) {
                 // no-op
@@ -1303,7 +1345,7 @@ describe('GeminiChat', () => {
             vi.mocked(mockConfig.getModel).mockReturnValue('gemini-pro');
             vi.mocked(mockContentGenerator.generateContentStream).mockRejectedValue(error429);
             mockHandleFallback.mockResolvedValue(false);
-            const stream = await chat.sendMessageStream('gemini-2.0-flash', { message: 'test stop' }, 'prompt-id-fb2');
+            const stream = await chat.sendMessageStream({ model: 'gemini-2.0-flash' }, 'test stop', 'prompt-id-fb2', new AbortController().signal);
             await expect((async () => {
                 for await (const _ of stream) {
                     /* consume stream */
@@ -1347,7 +1389,7 @@ describe('GeminiChat', () => {
             };
         })());
         // Send a message and consume the stream
-        const stream = await chat.sendMessageStream('gemini-2.0-flash', { message: 'test' }, 'prompt-id-discard-test');
+        const stream = await chat.sendMessageStream({ model: 'gemini-2.0-flash' }, 'test message', 'prompt-id-discard-test', new AbortController().signal);
         const events = [];
         for await (const event of stream) {
             events.push(event);
@@ -1411,7 +1453,7 @@ describe('GeminiChat', () => {
                 };
             })();
             vi.mocked(mockContentGenerator.generateContentStream).mockResolvedValue(stream);
-            await chat.sendMessageStream('test-model', { message: 'test' }, 'prompt-id-preview-model-reset');
+            await chat.sendMessageStream({ model: 'test-model' }, 'test', 'prompt-id-preview-model-reset', new AbortController().signal);
             expect(mockConfig.setPreviewModelBypassMode).toHaveBeenCalledWith(false);
         });
         it('should reset previewModelFallbackMode to false upon successful Preview Model usage', async () => {
@@ -1426,7 +1468,7 @@ describe('GeminiChat', () => {
                 };
             })();
             vi.mocked(mockContentGenerator.generateContentStream).mockResolvedValue(stream);
-            const resultStream = await chat.sendMessageStream(PREVIEW_GEMINI_MODEL, { message: 'test' }, 'prompt-id-preview-model-healing');
+            const resultStream = await chat.sendMessageStream({ model: PREVIEW_GEMINI_MODEL }, 'test', 'prompt-id-preview-model-healing', new AbortController().signal);
             for await (const _ of resultStream) {
                 // consume stream
             }
@@ -1446,7 +1488,7 @@ describe('GeminiChat', () => {
             vi.mocked(mockContentGenerator.generateContentStream).mockResolvedValue(stream);
             // Simulate bypass mode being active (downgrade happened)
             vi.mocked(mockConfig.isPreviewModelBypassMode).mockReturnValue(true);
-            const resultStream = await chat.sendMessageStream(PREVIEW_GEMINI_MODEL, { message: 'test' }, 'prompt-id-bypass-no-healing');
+            const resultStream = await chat.sendMessageStream({ model: PREVIEW_GEMINI_MODEL }, 'test', 'prompt-id-bypass-no-healing', new AbortController().signal);
             for await (const _ of resultStream) {
                 // consume stream
             }
@@ -1455,7 +1497,7 @@ describe('GeminiChat', () => {
     });
     describe('ensureActiveLoopHasThoughtSignatures', () => {
         it('should add thoughtSignature to the first functionCall in each model turn of the active loop', () => {
-            const chat = new GeminiChat(mockConfig, {}, []);
+            const chat = new GeminiChat(mockConfig, '', [], []);
             const history = [
                 { role: 'user', parts: [{ text: 'Old message' }] },
                 {
@@ -1504,7 +1546,7 @@ describe('GeminiChat', () => {
             expect(newContents[5]?.parts?.[1]).not.toHaveProperty('thoughtSignature');
         });
         it('should not modify contents if there is no user text message', () => {
-            const chat = new GeminiChat(mockConfig, {}, []);
+            const chat = new GeminiChat(mockConfig, '', [], []);
             const history = [
                 {
                     role: 'user',
@@ -1520,13 +1562,13 @@ describe('GeminiChat', () => {
             expect(newContents[1]?.parts?.[0]).not.toHaveProperty('thoughtSignature');
         });
         it('should handle an empty history', () => {
-            const chat = new GeminiChat(mockConfig, {}, []);
+            const chat = new GeminiChat(mockConfig, '', []);
             const history = [];
             const newContents = chat.ensureActiveLoopHasThoughtSignatures(history);
             expect(newContents).toEqual([]);
         });
         it('should handle history with only a user message', () => {
-            const chat = new GeminiChat(mockConfig, {}, []);
+            const chat = new GeminiChat(mockConfig, '', []);
             const history = [{ role: 'user', parts: [{ text: 'Hello' }] }];
             const newContents = chat.ensureActiveLoopHasThoughtSignatures(history);
             expect(newContents).toEqual(history);