@google/gemini-cli-core 0.21.0-nightly.20251217.db643e916 → 0.21.0-nightly.20251218.739c02bd6

package/dist/src/code_assist/telemetry.test.js

@@ -3,11 +3,13 @@
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-import { describe, it, expect, vi } from 'vitest';
-import { createConversationOffered, formatProtoJsonDuration, } from './telemetry.js';
-import { ActionStatus } from './types.js';
-import { FinishReason, GenerateContentResponse } from '@google/genai';
-function createMockResponse(candidates = [], ok = true) {
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import { createConversationOffered, formatProtoJsonDuration, recordConversationOffered, recordToolCallInteractions, } from './telemetry.js';
+import { ActionStatus, ConversationInteractionInteraction, } from './types.js';
+import { FinishReason, GenerateContentResponse, } from '@google/genai';
+import * as codeAssist from './codeAssist.js';
+import { ToolConfirmationOutcome, } from '../tools/tools.js';
+function createMockResponse(candidates = [], ok = true, functionCalls = undefined) {
     const response = new GenerateContentResponse();
     response.candidates = candidates;
     response.sdkHttpResponse = {
@@ -16,6 +18,17 @@ function createMockResponse(candidates = [], ok = true) {
         },
         json: async () => ({}),
     };
+    // If functionCalls is explicitly provided, mock the getter.
+    // Otherwise, let the default behavior (if any) or undefined prevail.
+    // In the real SDK, functionCalls is a getter derived from candidates.
+    // For testing `createConversationOffered` which guards on functionCalls,
+    // we often need to force it to be present.
+    if (functionCalls !== undefined) {
+        Object.defineProperty(response, 'functionCalls', {
+            get: () => functionCalls,
+            configurable: true,
+        });
+    }
     return response;
 }
 describe('telemetry', () => {
@@ -35,7 +48,7 @@ describe('telemetry', () => {
                     },
                     finishReason: FinishReason.STOP,
                 },
-            ]);
+            ], true, [{ name: 'someTool', args: {} }]);
             const traceId = 'test-trace-id';
             const streamingLatency = { totalLatency: '1s' };
             const result = createConversationOffered(response, traceId, undefined, streamingLatency);
@@ -48,17 +61,34 @@ describe('telemetry', () => {
                 isAgentic: true,
             });
         });
+        it('should return undefined if no function calls', () => {
+            const response = createMockResponse([
+                {
+                    index: 0,
+                    content: {
+                        role: 'model',
+                        parts: [{ text: 'response without function calls' }],
+                    },
+                },
+            ], true, []);
+            const result = createConversationOffered(response, 'trace-id', undefined, {});
+            expect(result).toBeUndefined();
+        });
         it('should set status to CANCELLED if signal is aborted', () => {
-            const response = createMockResponse();
+            const response = createMockResponse([], true, [
+                { name: 'tool', args: {} },
+            ]);
             const signal = new AbortController().signal;
             vi.spyOn(signal, 'aborted', 'get').mockReturnValue(true);
             const result = createConversationOffered(response, 'trace-id', signal, {});
-            expect(result.status).toBe(ActionStatus.ACTION_STATUS_CANCELLED);
+            expect(result?.status).toBe(ActionStatus.ACTION_STATUS_CANCELLED);
         });
         it('should set status to ERROR_UNKNOWN if response has error (non-OK SDK response)', () => {
-            const response = createMockResponse([], false);
+            const response = createMockResponse([], false, [
+                { name: 'tool', args: {} },
+            ]);
             const result = createConversationOffered(response, 'trace-id', undefined, {});
-            expect(result.status).toBe(ActionStatus.ACTION_STATUS_ERROR_UNKNOWN);
+            expect(result?.status).toBe(ActionStatus.ACTION_STATUS_ERROR_UNKNOWN);
         });
         it('should set status to ERROR_UNKNOWN if finishReason is not STOP or MAX_TOKENS', () => {
             const response = createMockResponse([
@@ -66,14 +96,18 @@ describe('telemetry', () => {
                     index: 0,
                     finishReason: FinishReason.SAFETY,
                 },
-            ]);
+            ], true, [{ name: 'tool', args: {} }]);
             const result = createConversationOffered(response, 'trace-id', undefined, {});
-            expect(result.status).toBe(ActionStatus.ACTION_STATUS_ERROR_UNKNOWN);
+            expect(result?.status).toBe(ActionStatus.ACTION_STATUS_ERROR_UNKNOWN);
         });
         it('should set status to EMPTY if candidates is empty', () => {
-            const response = createMockResponse();
+            // We force functionCalls to be present to bypass the guard,
+            // simulating a state where we want to test the candidates check.
+            const response = createMockResponse([], true, [
+                { name: 'tool', args: {} },
+            ]);
             const result = createConversationOffered(response, 'trace-id', undefined, {});
-            expect(result.status).toBe(ActionStatus.ACTION_STATUS_EMPTY);
+            expect(result?.status).toBe(ActionStatus.ACTION_STATUS_EMPTY);
         });
         it('should detect code in response', () => {
             const response = createMockResponse([
@@ -85,9 +119,9 @@ describe('telemetry', () => {
                         ],
                     },
                 },
-            ]);
+            ], true, [{ name: 'tool', args: {} }]);
             const result = createConversationOffered(response, 'id', undefined, {});
-            expect(result.includedCode).toBe(true);
+            expect(result?.includedCode).toBe(true);
         });
         it('should not detect code if no backticks', () => {
             const response = createMockResponse([
@@ -97,9 +131,9 @@ describe('telemetry', () => {
                         parts: [{ text: 'Here is some text.' }],
                     },
                 },
-            ]);
+            ], true, [{ name: 'tool', args: {} }]);
             const result = createConversationOffered(response, 'id', undefined, {});
-            expect(result.includedCode).toBe(false);
+            expect(result?.includedCode).toBe(false);
         });
     });
     describe('formatProtoJsonDuration', () => {
@@ -108,5 +142,159 @@ describe('telemetry', () => {
             expect(formatProtoJsonDuration(100)).toBe('0.1s');
         });
     });
+    describe('recordConversationOffered', () => {
+        it('should call server.recordConversationOffered if traceId is present', async () => {
+            const serverMock = {
+                recordConversationOffered: vi.fn(),
+            };
+            const response = createMockResponse([], true, [
+                { name: 'tool', args: {} },
+            ]);
+            const streamingLatency = {};
+            await recordConversationOffered(serverMock, 'trace-id', response, streamingLatency, undefined);
+            expect(serverMock.recordConversationOffered).toHaveBeenCalledWith(expect.objectContaining({
+                traceId: 'trace-id',
+            }));
+        });
+        it('should not call server.recordConversationOffered if traceId is undefined', async () => {
+            const serverMock = {
+                recordConversationOffered: vi.fn(),
+            };
+            const response = createMockResponse([], true, [
+                { name: 'tool', args: {} },
+            ]);
+            await recordConversationOffered(serverMock, undefined, response, {}, undefined);
+            expect(serverMock.recordConversationOffered).not.toHaveBeenCalled();
+        });
+    });
+    describe('recordToolCallInteractions', () => {
+        let mockServer;
+        beforeEach(() => {
+            mockServer = {
+                recordConversationInteraction: vi.fn(),
+            };
+            vi.spyOn(codeAssist, 'getCodeAssistServer').mockReturnValue(mockServer);
+        });
+        afterEach(() => {
+            vi.restoreAllMocks();
+        });
+        it('should record ACCEPT_FILE interaction for accepted edit tools', async () => {
+            const toolCalls = [
+                {
+                    request: {
+                        name: 'replace', // in EDIT_TOOL_NAMES
+                        args: {},
+                        callId: 'call-1',
+                        isClientInitiated: false,
+                        prompt_id: 'p1',
+                        traceId: 'trace-1',
+                    },
+                    outcome: ToolConfirmationOutcome.ProceedOnce,
+                    status: 'success',
+                },
+            ];
+            await recordToolCallInteractions({}, toolCalls);
+            expect(mockServer.recordConversationInteraction).toHaveBeenCalledWith({
+                traceId: 'trace-1',
+                status: ActionStatus.ACTION_STATUS_NO_ERROR,
+                interaction: ConversationInteractionInteraction.ACCEPT_FILE,
+                isAgentic: true,
+            });
+        });
+        it('should record UNKNOWN interaction for other accepted tools', async () => {
+            const toolCalls = [
+                {
+                    request: {
+                        name: 'read_file', // NOT in EDIT_TOOL_NAMES
+                        args: {},
+                        callId: 'call-2',
+                        isClientInitiated: false,
+                        prompt_id: 'p2',
+                        traceId: 'trace-2',
+                    },
+                    outcome: ToolConfirmationOutcome.ProceedOnce,
+                    status: 'success',
+                },
+            ];
+            await recordToolCallInteractions({}, toolCalls);
+            expect(mockServer.recordConversationInteraction).toHaveBeenCalledWith({
+                traceId: 'trace-2',
+                status: ActionStatus.ACTION_STATUS_NO_ERROR,
+                interaction: ConversationInteractionInteraction.UNKNOWN,
+                isAgentic: true,
+            });
+        });
+        it('should not record interaction for cancelled status', async () => {
+            const toolCalls = [
+                {
+                    request: {
+                        name: 'tool',
+                        args: {},
+                        callId: 'call-3',
+                        isClientInitiated: false,
+                        prompt_id: 'p3',
+                        traceId: 'trace-3',
+                    },
+                    status: 'cancelled',
+                    response: {},
+                    tool: {},
+                    invocation: {},
+                },
+            ];
+            await recordToolCallInteractions({}, toolCalls);
+            expect(mockServer.recordConversationInteraction).not.toHaveBeenCalled();
+        });
+        it('should not record interaction for error status', async () => {
+            const toolCalls = [
+                {
+                    request: {
+                        name: 'tool',
+                        args: {},
+                        callId: 'call-4',
+                        isClientInitiated: false,
+                        prompt_id: 'p4',
+                        traceId: 'trace-4',
+                    },
+                    status: 'error',
+                    response: {
+                        error: new Error('fail'),
+                    },
+                },
+            ];
+            await recordToolCallInteractions({}, toolCalls);
+            expect(mockServer.recordConversationInteraction).not.toHaveBeenCalled();
+        });
+        it('should not record interaction if tool calls are mixed or not 100% accepted', async () => {
+            // Logic: traceId && acceptedToolCalls / toolCalls.length >= 1
+            const toolCalls = [
+                {
+                    request: {
+                        name: 't1',
+                        args: {},
+                        callId: 'c1',
+                        isClientInitiated: false,
+                        prompt_id: 'p1',
+                        traceId: 't1',
+                    },
+                    outcome: ToolConfirmationOutcome.ProceedOnce,
+                    status: 'success',
+                },
+                {
+                    request: {
+                        name: 't2',
+                        args: {},
+                        callId: 'c2',
+                        isClientInitiated: false,
+                        prompt_id: 'p1',
+                        traceId: 't1',
+                    },
+                    outcome: ToolConfirmationOutcome.Cancel, // Rejected
+                    status: 'success',
+                },
+            ];
+            await recordToolCallInteractions({}, toolCalls);
+            expect(mockServer.recordConversationInteraction).not.toHaveBeenCalled();
+        });
+    });
 });
 //# sourceMappingURL=telemetry.test.js.map

package/dist/src/code_assist/telemetry.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"telemetry.test.js","sourceRoot":"","sources":["../../../src/code_assist/telemetry.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAClD,OAAO,EACL,yBAAyB,EACzB,uBAAuB,GACxB,MAAM,gBAAgB,CAAC;AACxB,OAAO,EAAE,YAAY,EAAyB,MAAM,YAAY,CAAC;AACjE,OAAO,EAAE,YAAY,EAAE,uBAAuB,EAAE,MAAM,eAAe,CAAC;AAEtE,SAAS,kBAAkB,CACzB,aAAoD,EAAE,EACtD,EAAE,GAAG,IAAI;IAET,MAAM,QAAQ,GAAG,IAAI,uBAAuB,EAAE,CAAC;IAC/C,QAAQ,CAAC,UAAU,GAAG,UAAU,CAAC;IACjC,QAAQ,CAAC,eAAe,GAAG;QACzB,gBAAgB,EAAE;YAChB,EAAE;SACoB;QACxB,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC;KACvB,CAAC;IACF,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;IACzB,QAAQ,CAAC,2BAA2B,EAAE,GAAG,EAAE;QACzC,EAAE,CAAC,gEAAgE,EAAE,GAAG,EAAE;YACxE,MAAM,QAAQ,GAAG,kBAAkB,CAAC;gBAClC;oBACE,KAAK,EAAE,CAAC;oBACR,OAAO,EAAE;wBACP,IAAI,EAAE,OAAO;wBACb,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,0BAA0B,EAAE,CAAC;qBAC9C;oBACD,gBAAgB,EAAE;wBAChB,SAAS,EAAE;4BACT,EAAE,GAAG,EAAE,qBAAqB,EAAE,UAAU,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE;yBAC5D;qBACF;oBACD,YAAY,EAAE,YAAY,CAAC,IAAI;iBAChC;aACF,CAAC,CAAC;YACH,MAAM,OAAO,GAAG,eAAe,CAAC;YAChC,MAAM,gBAAgB,GAAqB,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC;YAElE,MAAM,MAAM,GAAG,yBAAyB,CACtC,QAAQ,EACR,OAAO,EACP,SAAS,EACT,gBAAgB,CACjB,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,aAAa,EAAE,GAAG;gBAClB,YAAY,EAAE,IAAI;gBAClB,MAAM,EAAE,YAAY,CAAC,sBAAsB;gBAC3C,OAAO;gBACP,gBAAgB;gBAChB,SAAS,EAAE,IAAI;aAChB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;YAC7D,MAAM,QAAQ,GAAG,kBAAkB,EAAE,CAAC;YACtC,MAAM,MAAM,GAAG,IAAI,eAAe,EAAE,CAAC,MAAM,CAAC;YAC5C,EAAE,CAAC,KAAK,CAAC,MAAM,EAAE,SAAS,EAAE,KAAK,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YAEzD,MAAM,MAAM,GAAG,yBAAyB,CACtC,QAAQ,EACR,UAAU,EACV,MAAM,EACN,EAAE,CACH,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,uBAAuB,CAAC,CAAC;QACnE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gFAAgF,EAAE,GAAG,EAAE;YACxF,MAAM,QAAQ,GAAG,kBAAkB,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;YAE/C,MAAM,MAAM,GAAG,yBAAyB,CACtC,QAAQ,EACR,UAAU,EACV,SAAS,EACT,EAAE,CACH,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,2BAA2B,CAAC,CAAC;QACvE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8EAA8E,EAAE,GAAG,EAAE;YACtF,MAAM,QAAQ,GAAG,kBAAkB,CAAC;gBAClC;oBACE,KAAK,EAAE,CAAC;oBACR,YAAY,EAAE,YAAY,CAAC,MAAM;iBAClC;aACF,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,yBAAyB,CACtC,QAAQ,EACR,UAAU,EACV,SAAS,EACT,EAAE,CACH,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,2BAA2B,CAAC,CAAC;QACvE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;YAC3D,MAAM,QAAQ,GAAG,kBAAkB,EAAE,CAAC;YAEtC,MAAM,MAAM,GAAG,yBAAyB,CACtC,QAAQ,EACR,UAAU,EACV,SAAS,EACT,EAAE,CACH,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,mBAAmB,CAAC,CAAC;QAC/D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;YACxC,MAAM,QAAQ,GAAG,kBAAkB,CAAC;gBAClC;oBACE,KAAK,EAAE,CAAC;oBACR,OAAO,EAAE;wBACP,KAAK,EAAE;4BACL,EAAE,IAAI,EAAE,mDAAmD,EAAE;yBAC9D;qBACF;iBACF;aACF,CAAC,CAAC;YACH,MAAM,MAAM,GAAG,yBAAyB,CAAC,QAAQ,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE,CAAC,CAAC;YACxE,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;YAChD,MAAM,QAAQ,GAAG,kBAAkB,CAAC;gBAClC;oBACE,KAAK,EAAE,CAAC;oBACR,OAAO,EAAE;wBACP,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,oBAAoB,EAAE,CAAC;qBACxC;iBACF;aACF,CAAC,CAAC;YACH,MAAM,MAAM,GAAG,yBAAyB,CAAC,QAAQ,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE,CAAC,CAAC;YACxE,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACvC,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,MAAM,CAAC,uBAAuB,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACnD,MAAM,CAAC,uBAAuB,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
+{"version":3,"file":"telemetry.test.js","sourceRoot":"","sources":["../../../src/code_assist/telemetry.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EACL,yBAAyB,EACzB,uBAAuB,EACvB,yBAAyB,EACzB,0BAA0B,GAC3B,MAAM,gBAAgB,CAAC;AACxB,OAAO,EACL,YAAY,EACZ,kCAAkC,GAEnC,MAAM,YAAY,CAAC;AACpB,OAAO,EACL,YAAY,EACZ,uBAAuB,GAExB,MAAM,eAAe,CAAC;AACvB,OAAO,KAAK,UAAU,MAAM,iBAAiB,CAAC;AAG9C,OAAO,EACL,uBAAuB,GAGxB,MAAM,mBAAmB,CAAC;AAI3B,SAAS,kBAAkB,CACzB,aAAoD,EAAE,EACtD,EAAE,GAAG,IAAI,EACT,gBAA4C,SAAS;IAErD,MAAM,QAAQ,GAAG,IAAI,uBAAuB,EAAE,CAAC;IAC/C,QAAQ,CAAC,UAAU,GAAG,UAAU,CAAC;IACjC,QAAQ,CAAC,eAAe,GAAG;QACzB,gBAAgB,EAAE;YAChB,EAAE;SACoB;QACxB,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC;KACvB,CAAC;IAEF,4DAA4D;IAC5D,qEAAqE;IACrE,sEAAsE;IACtE,yEAAyE;IACzE,2CAA2C;IAC3C,IAAI,aAAa,KAAK,SAAS,EAAE,CAAC;QAChC,MAAM,CAAC,cAAc,CAAC,QAAQ,EAAE,eAAe,EAAE;YAC/C,GAAG,EAAE,GAAG,EAAE,CAAC,aAAa;YACxB,YAAY,EAAE,IAAI;SACnB,CAAC,CAAC;IACL,CAAC;IAED,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;IACzB,QAAQ,CAAC,2BAA2B,EAAE,GAAG,EAAE;QACzC,EAAE,CAAC,gEAAgE,EAAE,GAAG,EAAE;YACxE,MAAM,QAAQ,GAAG,kBAAkB,CACjC;gBACE;oBACE,KAAK,EAAE,CAAC;oBACR,OAAO,EAAE;wBACP,IAAI,EAAE,OAAO;wBACb,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,0BAA0B,EAAE,CAAC;qBAC9C;oBACD,gBAAgB,EAAE;wBAChB,SAAS,EAAE;4BACT,EAAE,GAAG,EAAE,qBAAqB,EAAE,UAAU,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE;yBAC5D;qBACF;oBACD,YAAY,EAAE,YAAY,CAAC,IAAI;iBAChC;aACF,EACD,IAAI,EACJ,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CACjC,CAAC;YACF,MAAM,OAAO,GAAG,eAAe,CAAC;YAChC,MAAM,gBAAgB,GAAqB,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC;YAElE,MAAM,MAAM,GAAG,yBAAyB,CACtC,QAAQ,EACR,OAAO,EACP,SAAS,EACT,gBAAgB,CACjB,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,aAAa,EAAE,GAAG;gBAClB,YAAY,EAAE,IAAI;gBAClB,MAAM,EAAE,YAAY,CAAC,sBAAsB;gBAC3C,OAAO;gBACP,gBAAgB;gBAChB,SAAS,EAAE,IAAI;aAChB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,MAAM,QAAQ,GAAG,kBAAkB,CACjC;gBACE;oBACE,KAAK,EAAE,CAAC;oBACR,OAAO,EAAE;wBACP,IAAI,EAAE,OAAO;wBACb,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,iCAAiC,EAAE,CAAC;qBACrD;iBACF;aACF,EACD,IAAI,EACJ,EAAE,CACH,CAAC;YACF,MAAM,MAAM,GAAG,yBAAyB,CACtC,QAAQ,EACR,UAAU,EACV,SAAS,EACT,EAAE,CACH,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,CAAC,aAAa,EAAE,CAAC;QACjC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;YAC7D,MAAM,QAAQ,GAAG,kBAAkB,CAAC,EAAE,EAAE,IAAI,EAAE;gBAC5C,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE;aAC3B,CAAC,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,eAAe,EAAE,CAAC,MAAM,CAAC;YAC5C,EAAE,CAAC,KAAK,CAAC,MAAM,EAAE,SAAS,EAAE,KAAK,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YAEzD,MAAM,MAAM,GAAG,yBAAyB,CACtC,QAAQ,EACR,UAAU,EACV,MAAM,EACN,EAAE,CACH,CAAC;YAEF,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,uBAAuB,CAAC,CAAC;QACpE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gFAAgF,EAAE,GAAG,EAAE;YACxF,MAAM,QAAQ,GAAG,kBAAkB,CAAC,EAAE,EAAE,KAAK,EAAE;gBAC7C,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE;aAC3B,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,yBAAyB,CACtC,QAAQ,EACR,UAAU,EACV,SAAS,EACT,EAAE,CACH,CAAC;YAEF,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,2BAA2B,CAAC,CAAC;QACxE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8EAA8E,EAAE,GAAG,EAAE;YACtF,MAAM,QAAQ,GAAG,kBAAkB,CACjC;gBACE;oBACE,KAAK,EAAE,CAAC;oBACR,YAAY,EAAE,YAAY,CAAC,MAAM;iBAClC;aACF,EACD,IAAI,EACJ,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAC7B,CAAC;YAEF,MAAM,MAAM,GAAG,yBAAyB,CACtC,QAAQ,EACR,UAAU,EACV,SAAS,EACT,EAAE,CACH,CAAC;YAEF,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,2BAA2B,CAAC,CAAC;QACxE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;YAC3D,4DAA4D;YAC5D,iEAAiE;YACjE,MAAM,QAAQ,GAAG,kBAAkB,CAAC,EAAE,EAAE,IAAI,EAAE;gBAC5C,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE;aAC3B,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,yBAAyB,CACtC,QAAQ,EACR,UAAU,EACV,SAAS,EACT,EAAE,CACH,CAAC;YAEF,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,mBAAmB,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;YACxC,MAAM,QAAQ,GAAG,kBAAkB,CACjC;gBACE;oBACE,KAAK,EAAE,CAAC;oBACR,OAAO,EAAE;wBACP,KAAK,EAAE;4BACL,EAAE,IAAI,EAAE,mDAAmD,EAAE;yBAC9D;qBACF;iBACF;aACF,EACD,IAAI,EACJ,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAC7B,CAAC;YACF,MAAM,MAAM,GAAG,yBAAyB,CAAC,QAAQ,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE,CAAC,CAAC;YACxE,MAAM,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;YAChD,MAAM,QAAQ,GAAG,kBAAkB,CACjC;gBACE;oBACE,KAAK,EAAE,CAAC;oBACR,OAAO,EAAE;wBACP,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,oBAAoB,EAAE,CAAC;qBACxC;iBACF;aACF,EACD,IAAI,EACJ,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAC7B,CAAC;YACF,MAAM,MAAM,GAAG,yBAAyB,CAAC,QAAQ,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE,CAAC,CAAC;YACxE,MAAM,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACvC,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,MAAM,CAAC,uBAAuB,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACnD,MAAM,CAAC,uBAAuB,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,2BAA2B,EAAE,GAAG,EAAE;QACzC,EAAE,CAAC,oEAAoE,EAAE,KAAK,IAAI,EAAE;YAClF,MAAM,UAAU,GAAG;gBACjB,yBAAyB,EAAE,EAAE,CAAC,EAAE,EAAE;aACJ,CAAC;YAEjC,MAAM,QAAQ,GAAG,kBAAkB,CAAC,EAAE,EAAE,IAAI,EAAE;gBAC5C,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE;aAC3B,CAAC,CAAC;YACH,MAAM,gBAAgB,GAAG,EAAE,CAAC;YAE5B,MAAM,yBAAyB,CAC7B,UAAU,EACV,UAAU,EACV,QAAQ,EACR,gBAAgB,EAChB,SAAS,CACV,CAAC;YAEF,MAAM,CAAC,UAAU,CAAC,yBAAyB,CAAC,CAAC,oBAAoB,CAC/D,MAAM,CAAC,gBAAgB,CAAC;gBACtB,OAAO,EAAE,UAAU;aACpB,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0EAA0E,EAAE,KAAK,IAAI,EAAE;YACxF,MAAM,UAAU,GAAG;gBACjB,yBAAyB,EAAE,EAAE,CAAC,EAAE,EAAE;aACJ,CAAC;YACjC,MAAM,QAAQ,GAAG,kBAAkB,CAAC,EAAE,EAAE,IAAI,EAAE;gBAC5C,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE;aAC3B,CAAC,CAAC;YAEH,MAAM,yBAAyB,CAC7B,UAAU,EACV,SAAS,EACT,QAAQ,EACR,EAAE,EACF,SAAS,CACV,CAAC;YAEF,MAAM,CAAC,UAAU,CAAC,yBAAyB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACtE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,4BAA4B,EAAE,GAAG,EAAE;QAC1C,IAAI,UAAuE,CAAC;QAE5E,UAAU,CAAC,GAAG,EAAE;YACd,UAAU,GAAG;gBACX,6BAA6B,EAAE,EAAE,CAAC,EAAE,EAAE;aACvC,CAAC;YACF,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,qBAAqB,CAAC,CAAC,eAAe,CACzD,UAAyC,CAC1C,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,SAAS,CAAC,GAAG,EAAE;YACb,EAAE,CAAC,eAAe,EAAE,CAAC;QACvB,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;YAC7E,MAAM,SAAS,GAAwB;gBACrC;oBACE,OAAO,EAAE;wBACP,IAAI,EAAE,SAAS,EAAE,qBAAqB;wBACtC,IAAI,EAAE,EAAE;wBACR,MAAM,EAAE,QAAQ;wBAChB,iBAAiB,EAAE,KAAK;wBACxB,SAAS,EAAE,IAAI;wBACf,OAAO,EAAE,SAAS;qBACnB;oBACD,OAAO,EAAE,uBAAuB,CAAC,WAAW;oBAC5C,MAAM,EAAE,SAAS;iBACc;aAClC,CAAC;YAEF,MAAM,0BAA0B,CAAC,EAAY,EAAE,SAAS,CAAC,CAAC;YAE1D,MAAM,CAAC,UAAU,CAAC,6BAA6B,CAAC,CAAC,oBAAoB,CAAC;gBACpE,OAAO,EAAE,SAAS;gBAClB,MAAM,EAAE,YAAY,CAAC,sBAAsB;gBAC3C,WAAW,EAAE,kCAAkC,CAAC,WAAW;gBAC3D,SAAS,EAAE,IAAI;aAChB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;YAC1E,MAAM,SAAS,GAAwB;gBACrC;oBACE,OAAO,EAAE;wBACP,IAAI,EAAE,WAAW,EAAE,yBAAyB;wBAC5C,IAAI,EAAE,EAAE;wBACR,MAAM,EAAE,QAAQ;wBAChB,iBAAiB,EAAE,KAAK;wBACxB,SAAS,EAAE,IAAI;wBACf,OAAO,EAAE,SAAS;qBACnB;oBACD,OAAO,EAAE,uBAAuB,CAAC,WAAW;oBAC5C,MAAM,EAAE,SAAS;iBACc;aAClC,CAAC;YAEF,MAAM,0BAA0B,CAAC,EAAY,EAAE,SAAS,CAAC,CAAC;YAE1D,MAAM,CAAC,UAAU,CAAC,6BAA6B,CAAC,CAAC,oBAAoB,CAAC;gBACpE,OAAO,EAAE,SAAS;gBAClB,MAAM,EAAE,YAAY,CAAC,sBAAsB;gBAC3C,WAAW,EAAE,kCAAkC,CAAC,OAAO;gBACvD,SAAS,EAAE,IAAI;aAChB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;YAClE,MAAM,SAAS,GAAwB;gBACrC;oBACE,OAAO,EAAE;wBACP,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,EAAE;wBACR,MAAM,EAAE,QAAQ;wBAChB,iBAAiB,EAAE,KAAK;wBACxB,SAAS,EAAE,IAAI;wBACf,OAAO,EAAE,SAAS;qBACnB;oBACD,MAAM,EAAE,WAAW;oBACnB,QAAQ,EAAE,EAAqC;oBAC/C,IAAI,EAAE,EAAmC;oBACzC,UAAU,EAAE,EAAkC;iBAC1B;aACvB,CAAC;YAEF,MAAM,0BAA0B,CAAC,EAAY,EAAE,SAAS,CAAC,CAAC;YAE1D,MAAM,CAAC,UAAU,CAAC,6BAA6B,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC1E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;YAC9D,MAAM,SAAS,GAAwB;gBACrC;oBACE,OAAO,EAAE;wBACP,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,EAAE;wBACR,MAAM,EAAE,QAAQ;wBAChB,iBAAiB,EAAE,KAAK;wBACxB,SAAS,EAAE,IAAI;wBACf,OAAO,EAAE,SAAS;qBACnB;oBACD,MAAM,EAAE,OAAO;oBACf,QAAQ,EAAE;wBACR,KAAK,EAAE,IAAI,KAAK,CAAC,MAAM,CAAC;qBACU;iBAChB;aACvB,CAAC;YAEF,MAAM,0BAA0B,CAAC,EAAY,EAAE,SAAS,CAAC,CAAC;YAE1D,MAAM,CAAC,UAAU,CAAC,6BAA6B,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC1E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4EAA4E,EAAE,KAAK,IAAI,EAAE;YAC1F,8DAA8D;YAC9D,MAAM,SAAS,GAAwB;gBACrC;oBACE,OAAO,EAAE;wBACP,IAAI,EAAE,IAAI;wBACV,IAAI,EAAE,EAAE;wBACR,MAAM,EAAE,IAAI;wBACZ,iBAAiB,EAAE,KAAK;wBACxB,SAAS,EAAE,IAAI;wBACf,OAAO,EAAE,IAAI;qBACd;oBACD,OAAO,EAAE,uBAAuB,CAAC,WAAW;oBAC5C,MAAM,EAAE,SAAS;iBAClB;gBACD;oBACE,OAAO,EAAE;wBACP,IAAI,EAAE,IAAI;wBACV,IAAI,EAAE,EAAE;wBACR,MAAM,EAAE,IAAI;wBACZ,iBAAiB,EAAE,KAAK;wBACxB,SAAS,EAAE,IAAI;wBACf,OAAO,EAAE,IAAI;qBACd;oBACD,OAAO,EAAE,uBAAuB,CAAC,MAAM,EAAE,WAAW;oBACpD,MAAM,EAAE,SAAS;iBAClB;aACgC,CAAC;YAEpC,MAAM,0BAA0B,CAAC,EAAY,EAAE,SAAS,CAAC,CAAC;YAE1D,MAAM,CAAC,UAAU,CAAC,6BAA6B,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC1E,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

@@ -36,6 +36,7 @@ import { PolicyEngine } from '../policy/policy-engine.js';
 import type { PolicyEngineConfig } from '../policy/types.js';
 import { HookSystem } from '../hooks/index.js';
 import type { UserTierId } from '../code_assist/types.js';
+import type { RetrieveUserQuotaResponse } from '../code_assist/types.js';
 import type { Experiments } from '../code_assist/experiments/experiments.js';
 import { AgentRegistry } from '../agents/registry.js';
 import { ApprovalMode } from '../policy/types.js';
@@ -229,7 +230,6 @@ export interface ConfigParameters {
     });
     previewFeatures?: boolean;
     enableAgents?: boolean;
-    enableModelAvailabilityService?: boolean;
     experimentalJitContext?: boolean;
 }
 export declare class Config {
@@ -279,10 +279,10 @@ export declare class Config {
     private readonly bugCommand;
     private model;
     private previewFeatures;
+    private hasAccessToPreviewModel;
     private readonly noBrowser;
     private readonly folderTrust;
     private ideMode;
-    private inFallbackMode;
     private _activeModel;
     private readonly maxSessionTurns;
     private readonly listSessions;
@@ -336,9 +336,6 @@ export declare class Config {
     private experiments;
     private experimentsPromise;
     private hookSystem?;
-    private previewModelFallbackMode;
-    private previewModelBypassMode;
-    private readonly enableModelAvailabilityService;
     private readonly enableAgents;
     private readonly experimentalJitContext;
     private contextManager?;
@@ -365,15 +362,9 @@ export declare class Config {
     setModel(newModel: string): void;
     getActiveModel(): string;
     setActiveModel(model: string): void;
-    resetTurn(): void;
-    isInFallbackMode(): boolean;
-    setFallbackMode(active: boolean): void;
     setFallbackModelHandler(handler: FallbackModelHandler): void;
     getFallbackModelHandler(): FallbackModelHandler | undefined;
-    isPreviewModelFallbackMode(): boolean;
-    setPreviewModelFallbackMode(active: boolean): void;
-    isPreviewModelBypassMode(): boolean;
-    setPreviewModelBypassMode(active: boolean): void;
+    resetTurn(): void;
     getMaxSessionTurns(): number;
     setQuotaErrorOccurred(value: boolean): void;
     getQuotaErrorOccurred(): boolean;
@@ -391,6 +382,9 @@ export declare class Config {
     getQuestion(): string | undefined;
     getPreviewFeatures(): boolean | undefined;
     setPreviewFeatures(previewFeatures: boolean): void;
+    getHasAccessToPreviewModel(): boolean;
+    setHasAccessToPreviewModel(hasAccess: boolean): void;
+    refreshUserQuota(): Promise<RetrieveUserQuotaResponse | undefined>;
     getCoreTools(): string[] | undefined;
     getAllowedTools(): string[] | undefined;
     /**
@@ -472,7 +466,6 @@ export declare class Config {
     getExtensionLoader(): ExtensionLoader;
     getEnabledExtensions(): string[];
     getEnableExtensionReloading(): boolean;
-    isModelAvailabilityServiceEnabled(): boolean;
     isAgentsEnabled(): boolean;
     getNoBrowser(): boolean;
     isBrowserLaunchSuppressed(): boolean;

package/dist/src/config/config.js

@@ -29,7 +29,7 @@ import { GitService } from '../services/gitService.js';
 import { initializeTelemetry, DEFAULT_TELEMETRY_TARGET, DEFAULT_OTLP_ENDPOINT, uiTelemetryService, } from '../telemetry/index.js';
 import { coreEvents } from '../utils/events.js';
 import { tokenLimit } from '../core/tokenLimits.js';
-import { DEFAULT_GEMINI_EMBEDDING_MODEL, DEFAULT_GEMINI_FLASH_MODEL, DEFAULT_THINKING_MODE, } from './models.js';
+import { DEFAULT_GEMINI_EMBEDDING_MODEL, DEFAULT_GEMINI_FLASH_MODEL, DEFAULT_GEMINI_MODEL_AUTO, DEFAULT_THINKING_MODE, isPreviewModel, PREVIEW_GEMINI_MODEL, PREVIEW_GEMINI_MODEL_AUTO, } from './models.js';
 import { shouldAttemptBrowserLaunch } from '../utils/browser.js';
 import { ideContextStore } from '../ide/ideContext.js';
 import { WriteTodosTool } from '../tools/write-todos.js';
@@ -184,10 +184,10 @@ export class Config {
     bugCommand;
     model;
     previewFeatures;
+    hasAccessToPreviewModel = false;
     noBrowser;
     folderTrust;
     ideMode;
-    inFallbackMode = false;
     _activeModel;
     maxSessionTurns;
     listSessions;
@@ -241,9 +241,6 @@ export class Config {
     experiments;
     experimentsPromise;
     hookSystem;
-    previewModelFallbackMode = false;
-    previewModelBypassMode = false;
-    enableModelAvailabilityService;
     enableAgents;
     experimentalJitContext;
     contextManager;
@@ -300,8 +297,6 @@ export class Config {
         this.bugCommand = params.bugCommand;
         this.model = params.model;
         this._activeModel = params.model;
-        this.enableModelAvailabilityService =
-            params.enableModelAvailabilityService ?? false;
         this.enableAgents = params.enableAgents ?? false;
         this.experimentalJitContext = params.experimentalJitContext ?? false;
         this.modelAvailabilityService = new ModelAvailabilityService();
@@ -343,7 +338,10 @@ export class Config {
             params.truncateToolOutputLines ?? DEFAULT_TRUNCATE_TOOL_OUTPUT_LINES;
         this.enableToolOutputTruncation = params.enableToolOutputTruncation ?? true;
         this.useSmartEdit = params.useSmartEdit ?? true;
-        this.useWriteTodos = params.useWriteTodos ?? true;
+        // // TODO(joshualitt): Re-evaluate the todo tool for 3 family.
+        this.useWriteTodos = isPreviewModel(this.model)
+            ? false
+            : (params.useWriteTodos ?? true);
         this.enableHooks = params.enableHooks ?? false;
         this.disabledHooks =
             (params.hooks && 'disabled' in params.hooks
@@ -479,20 +477,24 @@ export class Config {
             // Restore the conversation history to the new client
             this.geminiClient.stripThoughtsFromHistory();
         }
+        // Reset availability status when switching auth (e.g. from limited key to OAuth)
+        this.modelAvailabilityService.reset();
         const newContentGeneratorConfig = await createContentGeneratorConfig(this, authMethod);
         this.contentGenerator = await createContentGenerator(newContentGeneratorConfig, this, this.getSessionId());
         // Only assign to instance properties after successful initialization
         this.contentGeneratorConfig = newContentGeneratorConfig;
         // Initialize BaseLlmClient now that the ContentGenerator is available
         this.baseLlmClient = new BaseLlmClient(this.contentGenerator, this);
-        const previewFeatures = this.getPreviewFeatures();
         const codeAssistServer = getCodeAssistServer(this);
         if (codeAssistServer) {
+            if (codeAssistServer.projectId) {
+                await this.refreshUserQuota();
+            }
             this.experimentsPromise = getExperiments(codeAssistServer)
                 .then((experiments) => {
                 this.setExperiments(experiments);
                 // If preview features have not been set and the user authenticated through Google, we enable preview based on remote config only if it's true
-                if (previewFeatures === undefined) {
+                if (this.getPreviewFeatures() === undefined) {
                     const remotePreviewFeatures = experiments.flags[ExperimentFlags.ENABLE_PREVIEW]?.boolValue;
                     if (remotePreviewFeatures === true) {
                         this.setPreviewFeatures(remotePreviewFeatures);
@@ -507,8 +509,15 @@ export class Config {
             this.experiments = undefined;
             this.experimentsPromise = undefined;
         }
-        // Reset the session flag since we're explicitly changing auth and using default model
-        this.inFallbackMode = false;
+        const authType = this.contentGeneratorConfig.authType;
+        if (authType === AuthType.USE_GEMINI ||
+            authType === AuthType.USE_VERTEX_AI) {
+            this.setHasAccessToPreviewModel(true);
+        }
+        // Update model if user no longer has access to the preview model
+        if (!this.hasAccessToPreviewModel && isPreviewModel(this.model)) {
+            this.setModel(DEFAULT_GEMINI_MODEL_AUTO);
+        }
     }
     async getExperimentsAsync() {
         if (this.experiments) {
@@ -560,13 +569,12 @@ export class Config {
         return this.model;
     }
     setModel(newModel) {
-        if (this.model !== newModel || this.inFallbackMode) {
+        if (this.model !== newModel || this._activeModel !== newModel) {
             this.model = newModel;
             // When the user explicitly sets a model, that becomes the active model.
             this._activeModel = newModel;
             coreEvents.emitModelChanged(newModel);
         }
-        this.setFallbackMode(false);
         this.modelAvailabilityService.reset();
     }
     getActiveModel() {
@@ -575,35 +583,16 @@ export class Config {
     setActiveModel(model) {
         if (this._activeModel !== model) {
             this._activeModel = model;
-            coreEvents.emitModelChanged(model);
         }
     }
-    resetTurn() {
-        this.modelAvailabilityService.resetTurn();
-    }
-    isInFallbackMode() {
-        return this.inFallbackMode;
-    }
-    setFallbackMode(active) {
-        this.inFallbackMode = active;
-    }
     setFallbackModelHandler(handler) {
         this.fallbackModelHandler = handler;
     }
     getFallbackModelHandler() {
         return this.fallbackModelHandler;
     }
-    isPreviewModelFallbackMode() {
-        return this.previewModelFallbackMode;
-    }
-    setPreviewModelFallbackMode(active) {
-        this.previewModelFallbackMode = active;
-    }
-    isPreviewModelBypassMode() {
-        return this.previewModelBypassMode;
-    }
-    setPreviewModelBypassMode(active) {
-        this.previewModelBypassMode = active;
+    resetTurn() {
+        this.modelAvailabilityService.resetTurn();
     }
     getMaxSessionTurns() {
         return this.maxSessionTurns;
@@ -659,7 +648,44 @@ export class Config {
         return this.previewFeatures;
     }
     setPreviewFeatures(previewFeatures) {
+        // No change in state, no action needed
+        if (this.previewFeatures === previewFeatures) {
+            return;
+        }
         this.previewFeatures = previewFeatures;
+        const currentModel = this.getModel();
+        // Case 1: Disabling preview features while on a preview model
+        if (!previewFeatures && isPreviewModel(currentModel)) {
+            this.setModel(DEFAULT_GEMINI_MODEL_AUTO);
+        }
+        // Case 2: Enabling preview features while on the default auto model
+        else if (previewFeatures && currentModel === DEFAULT_GEMINI_MODEL_AUTO) {
+            this.setModel(PREVIEW_GEMINI_MODEL_AUTO);
+        }
+    }
+    getHasAccessToPreviewModel() {
+        return this.hasAccessToPreviewModel;
+    }
+    setHasAccessToPreviewModel(hasAccess) {
+        this.hasAccessToPreviewModel = hasAccess;
+    }
+    async refreshUserQuota() {
+        const codeAssistServer = getCodeAssistServer(this);
+        if (!codeAssistServer || !codeAssistServer.projectId) {
+            return undefined;
+        }
+        try {
+            const quota = await codeAssistServer.retrieveUserQuota({
+                project: codeAssistServer.projectId,
+            });
+            const hasAccess = quota.buckets?.some((b) => b.modelId === PREVIEW_GEMINI_MODEL) ?? false;
+            this.setHasAccessToPreviewModel(hasAccess);
+            return quota;
+        }
+        catch (e) {
+            debugLogger.debug('Failed to retrieve user quota', e);
+            return undefined;
+        }
     }
     getCoreTools() {
         return this.coreTools;
@@ -893,9 +919,6 @@ export class Config {
     getEnableExtensionReloading() {
         return this.enableExtensionReloading;
     }
-    isModelAvailabilityServiceEnabled() {
-        return this.enableModelAvailabilityService;
-    }
     isAgentsEnabled() {
         return this.enableAgents;
     }