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

package/dist/src/core/client.js

@@ -0,0 +1,689 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { getDirectoryContextString, getEnvironmentContext, } from '../utils/environmentContext.js';
+import { CompressionStatus } from './turn.js';
+import { Turn, GeminiEventType } from './turn.js';
+import { getCoreSystemPrompt, getCompressionPrompt } from './prompts.js';
+import { getResponseText } from '../utils/partUtils.js';
+import { checkNextSpeaker } from '../utils/nextSpeakerChecker.js';
+import { reportError } from '../utils/errorReporting.js';
+import { GeminiChat } from './geminiChat.js';
+import { retryWithBackoff } from '../utils/retry.js';
+import { getErrorMessage } from '../utils/errors.js';
+import { isFunctionResponse } from '../utils/messageInspectors.js';
+import { tokenLimit } from './tokenLimits.js';
+import { AuthType, createContentGenerator } from './contentGenerator.js';
+import { ProxyAgent, setGlobalDispatcher } from 'undici';
+import { DEFAULT_GEMINI_FLASH_MODEL } from '../config/models.js';
+import { LoopDetectionService } from '../services/loopDetectionService.js';
+import { ideContext } from '../ide/ideContext.js';
+import { logChatCompression, logNextSpeakerCheck, logMalformedJsonResponse, } from '../telemetry/loggers.js';
+import { makeChatCompressionEvent, MalformedJsonResponseEvent, NextSpeakerCheckEvent, } from '../telemetry/types.js';
+function isThinkingSupported(model) {
+    if (model.startsWith('gemini-2.5'))
+        return true;
+    return false;
+}
+/**
+ * Returns the index of the content after the fraction of the total characters in the history.
+ *
+ * Exported for testing purposes.
+ */
+export function findIndexAfterFraction(history, fraction) {
+    if (fraction <= 0 || fraction >= 1) {
+        throw new Error('Fraction must be between 0 and 1');
+    }
+    const contentLengths = history.map((content) => JSON.stringify(content).length);
+    const totalCharacters = contentLengths.reduce((sum, length) => sum + length, 0);
+    const targetCharacters = totalCharacters * fraction;
+    let charactersSoFar = 0;
+    for (let i = 0; i < contentLengths.length; i++) {
+        charactersSoFar += contentLengths[i];
+        if (charactersSoFar >= targetCharacters) {
+            return i;
+        }
+    }
+    return contentLengths.length;
+}
+const MAX_TURNS = 100;
+/**
+ * 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.
+ */
+const COMPRESSION_TOKEN_THRESHOLD = 0.7;
+/**
+ * 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.
+ */
+const COMPRESSION_PRESERVE_THRESHOLD = 0.3;
+export class GeminiClient {
+    config;
+    chat;
+    contentGenerator;
+    embeddingModel;
+    generateContentConfig = {
+        temperature: 0,
+        topP: 1,
+    };
+    sessionTurnCount = 0;
+    loopDetector;
+    lastPromptId;
+    lastSentIdeContext;
+    forceFullIdeContext = true;
+    /**
+     * At any point in this conversation, was compression triggered without
+     * being forced and did it fail?
+     */
+    hasFailedCompressionAttempt = false;
+    constructor(config) {
+        this.config = config;
+        if (config.getProxy()) {
+            setGlobalDispatcher(new ProxyAgent(config.getProxy()));
+        }
+        this.embeddingModel = config.getEmbeddingModel();
+        this.loopDetector = new LoopDetectionService(config);
+        this.lastPromptId = this.config.getSessionId();
+    }
+    async initialize(contentGeneratorConfig) {
+        this.contentGenerator = await createContentGenerator(contentGeneratorConfig, this.config, this.config.getSessionId());
+        this.chat = await this.startChat();
+    }
+    getContentGenerator() {
+        if (!this.contentGenerator) {
+            throw new Error('Content generator not initialized');
+        }
+        return this.contentGenerator;
+    }
+    getUserTier() {
+        return this.contentGenerator?.userTier;
+    }
+    async addHistory(content) {
+        this.getChat().addHistory(content);
+    }
+    getChat() {
+        if (!this.chat) {
+            throw new Error('Chat not initialized');
+        }
+        return this.chat;
+    }
+    isInitialized() {
+        return this.chat !== undefined && this.contentGenerator !== undefined;
+    }
+    getHistory() {
+        return this.getChat().getHistory();
+    }
+    setHistory(history, { stripThoughts = false } = {}) {
+        const historyToSet = stripThoughts
+            ? history.map((content) => {
+                const newContent = { ...content };
+                if (newContent.parts) {
+                    newContent.parts = newContent.parts.map((part) => {
+                        if (part &&
+                            typeof part === 'object' &&
+                            'thoughtSignature' in part) {
+                            const newPart = { ...part };
+                            delete newPart
+                                .thoughtSignature;
+                            return newPart;
+                        }
+                        return part;
+                    });
+                }
+                return newContent;
+            })
+            : history;
+        this.getChat().setHistory(historyToSet);
+        this.forceFullIdeContext = true;
+    }
+    async setTools() {
+        const toolRegistry = this.config.getToolRegistry();
+        const toolDeclarations = toolRegistry.getFunctionDeclarations();
+        const tools = [{ functionDeclarations: toolDeclarations }];
+        this.getChat().setTools(tools);
+    }
+    async resetChat() {
+        this.chat = await this.startChat();
+    }
+    async addDirectoryContext() {
+        if (!this.chat) {
+            return;
+        }
+        this.getChat().addHistory({
+            role: 'user',
+            parts: [{ text: await getDirectoryContextString(this.config) }],
+        });
+    }
+    async startChat(extraHistory) {
+        this.forceFullIdeContext = true;
+        this.hasFailedCompressionAttempt = false;
+        const envParts = await getEnvironmentContext(this.config);
+        const toolRegistry = this.config.getToolRegistry();
+        const toolDeclarations = toolRegistry.getFunctionDeclarations();
+        const tools = [{ functionDeclarations: toolDeclarations }];
+        const history = [
+            {
+                role: 'user',
+                parts: envParts,
+            },
+            {
+                role: 'model',
+                parts: [{ text: 'Got it. Thanks for the context!' }],
+            },
+            ...(extraHistory ?? []),
+        ];
+        try {
+            const userMemory = this.config.getUserMemory();
+            const systemInstruction = getCoreSystemPrompt(userMemory);
+            const generateContentConfigWithThinking = isThinkingSupported(this.config.getModel())
+                ? {
+                    ...this.generateContentConfig,
+                    thinkingConfig: {
+                        thinkingBudget: -1,
+                        includeThoughts: true,
+                    },
+                }
+                : this.generateContentConfig;
+            return new GeminiChat(this.config, this.getContentGenerator(), {
+                systemInstruction,
+                ...generateContentConfigWithThinking,
+                tools,
+            }, history);
+        }
+        catch (error) {
+            await reportError(error, 'Error initializing Gemini chat session.', history, 'startChat');
+            throw new Error(`Failed to initialize chat: ${getErrorMessage(error)}`);
+        }
+    }
+    getIdeContextParts(forceFullContext) {
+        const currentIdeContext = ideContext.getIdeContext();
+        if (!currentIdeContext) {
+            return { contextParts: [], newIdeContext: undefined };
+        }
+        if (forceFullContext || !this.lastSentIdeContext) {
+            // Send full context as JSON
+            const openFiles = currentIdeContext.workspaceState?.openFiles || [];
+            const activeFile = openFiles.find((f) => f.isActive);
+            const otherOpenFiles = openFiles
+                .filter((f) => !f.isActive)
+                .map((f) => f.path);
+            const contextData = {};
+            if (activeFile) {
+                contextData['activeFile'] = {
+                    path: activeFile.path,
+                    cursor: activeFile.cursor
+                        ? {
+                            line: activeFile.cursor.line,
+                            character: activeFile.cursor.character,
+                        }
+                        : undefined,
+                    selectedText: activeFile.selectedText || undefined,
+                };
+            }
+            if (otherOpenFiles.length > 0) {
+                contextData['otherOpenFiles'] = otherOpenFiles;
+            }
+            if (Object.keys(contextData).length === 0) {
+                return { contextParts: [], newIdeContext: currentIdeContext };
+            }
+            const jsonString = JSON.stringify(contextData, null, 2);
+            const contextParts = [
+                "Here is the user's editor context as a JSON object. This is for your information only.",
+                '```json',
+                jsonString,
+                '```',
+            ];
+            if (this.config.getDebugMode()) {
+                console.log(contextParts.join('\n'));
+            }
+            return {
+                contextParts,
+                newIdeContext: currentIdeContext,
+            };
+        }
+        else {
+            // Calculate and send delta as JSON
+            const delta = {};
+            const changes = {};
+            const lastFiles = new Map((this.lastSentIdeContext.workspaceState?.openFiles || []).map((f) => [f.path, f]));
+            const currentFiles = new Map((currentIdeContext.workspaceState?.openFiles || []).map((f) => [
+                f.path,
+                f,
+            ]));
+            const openedFiles = [];
+            for (const [path] of currentFiles.entries()) {
+                if (!lastFiles.has(path)) {
+                    openedFiles.push(path);
+                }
+            }
+            if (openedFiles.length > 0) {
+                changes['filesOpened'] = openedFiles;
+            }
+            const closedFiles = [];
+            for (const [path] of lastFiles.entries()) {
+                if (!currentFiles.has(path)) {
+                    closedFiles.push(path);
+                }
+            }
+            if (closedFiles.length > 0) {
+                changes['filesClosed'] = closedFiles;
+            }
+            const lastActiveFile = (this.lastSentIdeContext.workspaceState?.openFiles || []).find((f) => f.isActive);
+            const currentActiveFile = (currentIdeContext.workspaceState?.openFiles || []).find((f) => f.isActive);
+            if (currentActiveFile) {
+                if (!lastActiveFile || lastActiveFile.path !== currentActiveFile.path) {
+                    changes['activeFileChanged'] = {
+                        path: currentActiveFile.path,
+                        cursor: currentActiveFile.cursor
+                            ? {
+                                line: currentActiveFile.cursor.line,
+                                character: currentActiveFile.cursor.character,
+                            }
+                            : undefined,
+                        selectedText: currentActiveFile.selectedText || undefined,
+                    };
+                }
+                else {
+                    const lastCursor = lastActiveFile.cursor;
+                    const currentCursor = currentActiveFile.cursor;
+                    if (currentCursor &&
+                        (!lastCursor ||
+                            lastCursor.line !== currentCursor.line ||
+                            lastCursor.character !== currentCursor.character)) {
+                        changes['cursorMoved'] = {
+                            path: currentActiveFile.path,
+                            cursor: {
+                                line: currentCursor.line,
+                                character: currentCursor.character,
+                            },
+                        };
+                    }
+                    const lastSelectedText = lastActiveFile.selectedText || '';
+                    const currentSelectedText = currentActiveFile.selectedText || '';
+                    if (lastSelectedText !== currentSelectedText) {
+                        changes['selectionChanged'] = {
+                            path: currentActiveFile.path,
+                            selectedText: currentSelectedText,
+                        };
+                    }
+                }
+            }
+            else if (lastActiveFile) {
+                changes['activeFileChanged'] = {
+                    path: null,
+                    previousPath: lastActiveFile.path,
+                };
+            }
+            if (Object.keys(changes).length === 0) {
+                return { contextParts: [], newIdeContext: currentIdeContext };
+            }
+            delta['changes'] = changes;
+            const jsonString = JSON.stringify(delta, null, 2);
+            const contextParts = [
+                "Here is a summary of changes in the user's editor context, in JSON format. This is for your information only.",
+                '```json',
+                jsonString,
+                '```',
+            ];
+            if (this.config.getDebugMode()) {
+                console.log(contextParts.join('\n'));
+            }
+            return {
+                contextParts,
+                newIdeContext: currentIdeContext,
+            };
+        }
+    }
+    async *sendMessageStream(request, signal, prompt_id, turns = MAX_TURNS, originalModel) {
+        if (this.lastPromptId !== prompt_id) {
+            this.loopDetector.reset(prompt_id);
+            this.lastPromptId = prompt_id;
+        }
+        this.sessionTurnCount++;
+        if (this.config.getMaxSessionTurns() > 0 &&
+            this.sessionTurnCount > this.config.getMaxSessionTurns()) {
+            yield { type: GeminiEventType.MaxSessionTurns };
+            return new Turn(this.getChat(), prompt_id);
+        }
+        // Ensure turns never exceeds MAX_TURNS to prevent infinite loops
+        const boundedTurns = Math.min(turns, MAX_TURNS);
+        if (!boundedTurns) {
+            return new Turn(this.getChat(), prompt_id);
+        }
+        // Track the original model from the first call to detect model switching
+        const initialModel = originalModel || this.config.getModel();
+        const compressed = await this.tryCompressChat(prompt_id);
+        if (compressed.compressionStatus === CompressionStatus.COMPRESSED) {
+            yield { type: GeminiEventType.ChatCompressed, value: compressed };
+        }
+        // Prevent context updates from being sent while a tool call is
+        // waiting for a response. The Gemini API requires that a functionResponse
+        // part from the user immediately follows a functionCall part from the model
+        // in the conversation history . The IDE context is not discarded; it will
+        // be included in the next regular message sent to the model.
+        const history = this.getHistory();
+        const lastMessage = history.length > 0 ? history[history.length - 1] : undefined;
+        const hasPendingToolCall = !!lastMessage &&
+            lastMessage.role === 'model' &&
+            (lastMessage.parts?.some((p) => 'functionCall' in p) || false);
+        if (this.config.getIdeMode() && !hasPendingToolCall) {
+            const { contextParts, newIdeContext } = this.getIdeContextParts(this.forceFullIdeContext || history.length === 0);
+            if (contextParts.length > 0) {
+                this.getChat().addHistory({
+                    role: 'user',
+                    parts: [{ text: contextParts.join('\n') }],
+                });
+            }
+            this.lastSentIdeContext = newIdeContext;
+            this.forceFullIdeContext = false;
+        }
+        const turn = new Turn(this.getChat(), prompt_id);
+        const loopDetected = await this.loopDetector.turnStarted(signal);
+        if (loopDetected) {
+            yield { type: GeminiEventType.LoopDetected };
+            return turn;
+        }
+        const resultStream = turn.run(request, signal);
+        for await (const event of resultStream) {
+            if (this.loopDetector.addAndCheck(event)) {
+                yield { type: GeminiEventType.LoopDetected };
+                return turn;
+            }
+            yield event;
+            if (event.type === GeminiEventType.Error) {
+                return turn;
+            }
+        }
+        if (!turn.pendingToolCalls.length && signal && !signal.aborted) {
+            // Check if model was switched during the call (likely due to quota error)
+            const currentModel = this.config.getModel();
+            if (currentModel !== initialModel) {
+                // Model was switched (likely due to quota error fallback)
+                // Don't continue with recursive call to prevent unwanted Flash execution
+                return turn;
+            }
+            if (this.config.getSkipNextSpeakerCheck()) {
+                return turn;
+            }
+            const nextSpeakerCheck = await checkNextSpeaker(this.getChat(), this, signal);
+            logNextSpeakerCheck(this.config, new NextSpeakerCheckEvent(prompt_id, turn.finishReason?.toString() || '', nextSpeakerCheck?.next_speaker || ''));
+            if (nextSpeakerCheck?.next_speaker === 'model') {
+                const nextRequest = [{ text: 'Please continue.' }];
+                // This recursive call's events will be yielded out, but the final
+                // turn object will be from the top-level call.
+                yield* this.sendMessageStream(nextRequest, signal, prompt_id, boundedTurns - 1, initialModel);
+            }
+        }
+        return turn;
+    }
+    async generateJson(contents, schema, abortSignal, model, config = {}) {
+        // Use current model from config instead of hardcoded Flash model
+        const modelToUse = model || this.config.getModel() || DEFAULT_GEMINI_FLASH_MODEL;
+        try {
+            const userMemory = this.config.getUserMemory();
+            const systemInstruction = getCoreSystemPrompt(userMemory);
+            const requestConfig = {
+                abortSignal,
+                ...this.generateContentConfig,
+                ...config,
+            };
+            const apiCall = () => this.getContentGenerator().generateContent({
+                model: modelToUse,
+                config: {
+                    ...requestConfig,
+                    systemInstruction,
+                    responseJsonSchema: schema,
+                    responseMimeType: 'application/json',
+                },
+                contents,
+            }, this.lastPromptId);
+            const result = await retryWithBackoff(apiCall, {
+                onPersistent429: async (authType, error) => await this.handleFlashFallback(authType, error),
+                authType: this.config.getContentGeneratorConfig()?.authType,
+            });
+            let text = getResponseText(result);
+            if (!text) {
+                const error = new Error('API returned an empty response for generateJson.');
+                await reportError(error, 'Error in generateJson: API returned an empty response.', contents, 'generateJson-empty-response');
+                throw error;
+            }
+            const prefix = '```json';
+            const suffix = '```';
+            if (text.startsWith(prefix) && text.endsWith(suffix)) {
+                logMalformedJsonResponse(this.config, new MalformedJsonResponseEvent(modelToUse));
+                text = text
+                    .substring(prefix.length, text.length - suffix.length)
+                    .trim();
+            }
+            try {
+                return JSON.parse(text);
+            }
+            catch (parseError) {
+                await reportError(parseError, 'Failed to parse JSON response from generateJson.', {
+                    responseTextFailedToParse: text,
+                    originalRequestContents: contents,
+                }, 'generateJson-parse');
+                throw new Error(`Failed to parse API response as JSON: ${getErrorMessage(parseError)}`);
+            }
+        }
+        catch (error) {
+            if (abortSignal.aborted) {
+                throw error;
+            }
+            // Avoid double reporting for the empty response case handled above
+            if (error instanceof Error &&
+                error.message === 'API returned an empty response for generateJson.') {
+                throw error;
+            }
+            await reportError(error, 'Error generating JSON content via API.', contents, 'generateJson-api');
+            throw new Error(`Failed to generate JSON content: ${getErrorMessage(error)}`);
+        }
+    }
+    async generateContent(contents, generationConfig, abortSignal, model) {
+        const modelToUse = model ?? this.config.getModel();
+        const configToUse = {
+            ...this.generateContentConfig,
+            ...generationConfig,
+        };
+        try {
+            const userMemory = this.config.getUserMemory();
+            const systemInstruction = getCoreSystemPrompt(userMemory);
+            const requestConfig = {
+                abortSignal,
+                ...configToUse,
+                systemInstruction,
+            };
+            const apiCall = () => this.getContentGenerator().generateContent({
+                model: modelToUse,
+                config: requestConfig,
+                contents,
+            }, this.lastPromptId);
+            const result = await retryWithBackoff(apiCall, {
+                onPersistent429: async (authType, error) => await this.handleFlashFallback(authType, error),
+                authType: this.config.getContentGeneratorConfig()?.authType,
+            });
+            return result;
+        }
+        catch (error) {
+            if (abortSignal.aborted) {
+                throw error;
+            }
+            await reportError(error, `Error generating content via API with model ${modelToUse}.`, {
+                requestContents: contents,
+                requestConfig: configToUse,
+            }, 'generateContent-api');
+            throw new Error(`Failed to generate content with model ${modelToUse}: ${getErrorMessage(error)}`);
+        }
+    }
+    async generateEmbedding(texts) {
+        if (!texts || texts.length === 0) {
+            return [];
+        }
+        const embedModelParams = {
+            model: this.embeddingModel,
+            contents: texts,
+        };
+        const embedContentResponse = await this.getContentGenerator().embedContent(embedModelParams);
+        if (!embedContentResponse.embeddings ||
+            embedContentResponse.embeddings.length === 0) {
+            throw new Error('No embeddings found in API response.');
+        }
+        if (embedContentResponse.embeddings.length !== texts.length) {
+            throw new Error(`API returned a mismatched number of embeddings. Expected ${texts.length}, got ${embedContentResponse.embeddings.length}.`);
+        }
+        return embedContentResponse.embeddings.map((embedding, index) => {
+            const values = embedding.values;
+            if (!values || values.length === 0) {
+                throw new Error(`API returned an empty embedding for input text at index ${index}: "${texts[index]}"`);
+            }
+            return values;
+        });
+    }
+    async tryCompressChat(prompt_id, force = false) {
+        const curatedHistory = this.getChat().getHistory(true);
+        // Regardless of `force`, don't do anything if the history is empty.
+        if (curatedHistory.length === 0 ||
+            (this.hasFailedCompressionAttempt && !force)) {
+            return {
+                originalTokenCount: 0,
+                newTokenCount: 0,
+                compressionStatus: CompressionStatus.NOOP,
+            };
+        }
+        const model = this.config.getModel();
+        const { totalTokens: originalTokenCount } = await this.getContentGenerator().countTokens({
+            model,
+            contents: curatedHistory,
+        });
+        if (originalTokenCount === undefined) {
+            console.warn(`Could not determine token count for model ${model}.`);
+            this.hasFailedCompressionAttempt = !force && true;
+            return {
+                originalTokenCount: 0,
+                newTokenCount: 0,
+                compressionStatus: CompressionStatus.COMPRESSION_FAILED_TOKEN_COUNT_ERROR,
+            };
+        }
+        const contextPercentageThreshold = this.config.getChatCompression()?.contextPercentageThreshold;
+        // Don't compress if not forced and we are under the limit.
+        if (!force) {
+            const threshold = contextPercentageThreshold ?? COMPRESSION_TOKEN_THRESHOLD;
+            if (originalTokenCount < threshold * tokenLimit(model)) {
+                return {
+                    originalTokenCount,
+                    newTokenCount: originalTokenCount,
+                    compressionStatus: CompressionStatus.NOOP,
+                };
+            }
+        }
+        let compressBeforeIndex = findIndexAfterFraction(curatedHistory, 1 - COMPRESSION_PRESERVE_THRESHOLD);
+        // Find the first user message after the index. This is the start of the next turn.
+        while (compressBeforeIndex < curatedHistory.length &&
+            (curatedHistory[compressBeforeIndex]?.role === 'model' ||
+                isFunctionResponse(curatedHistory[compressBeforeIndex]))) {
+            compressBeforeIndex++;
+        }
+        const historyToCompress = curatedHistory.slice(0, compressBeforeIndex);
+        const historyToKeep = curatedHistory.slice(compressBeforeIndex);
+        this.getChat().setHistory(historyToCompress);
+        const { text: summary } = await this.getChat().sendMessage({
+            message: {
+                text: 'First, reason in your scratchpad. Then, generate the <state_snapshot>.',
+            },
+            config: {
+                systemInstruction: { text: getCompressionPrompt() },
+                maxOutputTokens: originalTokenCount,
+            },
+        }, prompt_id);
+        const chat = await this.startChat([
+            {
+                role: 'user',
+                parts: [{ text: summary }],
+            },
+            {
+                role: 'model',
+                parts: [{ text: 'Got it. Thanks for the additional context!' }],
+            },
+            ...historyToKeep,
+        ]);
+        this.forceFullIdeContext = true;
+        const { totalTokens: newTokenCount } = await this.getContentGenerator().countTokens({
+            // model might change after calling `sendMessage`, so we get the newest value from config
+            model: this.config.getModel(),
+            contents: chat.getHistory(),
+        });
+        if (newTokenCount === undefined) {
+            console.warn('Could not determine compressed history token count.');
+            this.hasFailedCompressionAttempt = !force && true;
+            return {
+                originalTokenCount,
+                newTokenCount: originalTokenCount,
+                compressionStatus: CompressionStatus.COMPRESSION_FAILED_TOKEN_COUNT_ERROR,
+            };
+        }
+        logChatCompression(this.config, makeChatCompressionEvent({
+            tokens_before: originalTokenCount,
+            tokens_after: newTokenCount,
+        }));
+        if (newTokenCount > originalTokenCount) {
+            this.getChat().setHistory(curatedHistory);
+            this.hasFailedCompressionAttempt = !force && true;
+            return {
+                originalTokenCount,
+                newTokenCount,
+                compressionStatus: CompressionStatus.COMPRESSION_FAILED_INFLATED_TOKEN_COUNT,
+            };
+        }
+        else {
+            this.chat = chat; // Chat compression successful, set new state.
+        }
+        return {
+            originalTokenCount,
+            newTokenCount,
+            compressionStatus: CompressionStatus.COMPRESSED,
+        };
+    }
+    /**
+     * Handles falling back to Flash model when persistent 429 errors occur for OAuth users.
+     * Uses a fallback handler if provided by the config; otherwise, returns null.
+     */
+    async handleFlashFallback(authType, error) {
+        // Only handle fallback for OAuth users
+        if (authType !== AuthType.LOGIN_WITH_GOOGLE) {
+            return null;
+        }
+        const currentModel = this.config.getModel();
+        const fallbackModel = DEFAULT_GEMINI_FLASH_MODEL;
+        // Don't fallback if already using Flash model
+        if (currentModel === fallbackModel) {
+            return null;
+        }
+        // Check if config has a fallback handler (set by CLI package)
+        const fallbackHandler = this.config.flashFallbackHandler;
+        if (typeof fallbackHandler === 'function') {
+            try {
+                const accepted = await fallbackHandler(currentModel, fallbackModel, error);
+                if (accepted !== false && accepted !== null) {
+                    this.config.setModel(fallbackModel);
+                    this.config.setFallbackMode(true);
+                    return fallbackModel;
+                }
+                // Check if the model was switched manually in the handler
+                if (this.config.getModel() === fallbackModel) {
+                    return null; // Model was switched but don't continue with current prompt
+                }
+            }
+            catch (error) {
+                console.warn('Flash fallback handler failed:', error);
+            }
+        }
+        return null;
+    }
+}
+export const TEST_ONLY = {
+    COMPRESSION_PRESERVE_THRESHOLD,
+    COMPRESSION_TOKEN_THRESHOLD,
+};
+//# sourceMappingURL=client.js.map