fix(cli): do not override GOOGLE_CLOUD_PROJECT in Cloud Shell when using Vertex AI (#24455)

Co-authored-by: David Pierce <davidapierce@google.com>

packages/cli/src/ui/auth/AuthDialog.test.tsx

@@ -243,6 +243,32 @@ describe('AuthDialog', () => {
       unmount();
     });
 
+    it('sets auth context with requiresRestart: true for USE_VERTEX_AI in Cloud Shell', async () => {
+      vi.stubEnv('CLOUD_SHELL', 'true');
+      mockedValidateAuthMethod.mockReturnValue(null);
+      const { unmount } = await renderWithProviders(<AuthDialog {...props} />);
+      const { onSelect: handleAuthSelect } =
+        mockedRadioButtonSelect.mock.calls[0][0];
+      await handleAuthSelect(AuthType.USE_VERTEX_AI);
+
+      expect(props.setAuthContext).toHaveBeenCalledWith({
+        requiresRestart: true,
+      });
+      unmount();
+    });
+
+    it('sets auth context with empty object for USE_VERTEX_AI outside Cloud Shell', async () => {
+      vi.stubEnv('CLOUD_SHELL', '');
+      mockedValidateAuthMethod.mockReturnValue(null);
+      const { unmount } = await renderWithProviders(<AuthDialog {...props} />);
+      const { onSelect: handleAuthSelect } =
+        mockedRadioButtonSelect.mock.calls[0][0];
+      await handleAuthSelect(AuthType.USE_VERTEX_AI);
+
+      expect(props.setAuthContext).toHaveBeenCalledWith({});
+      unmount();
+    });
+
     it('sets auth context with empty object for other auth types', async () => {
       mockedValidateAuthMethod.mockReturnValue(null);
       const { unmount } = await renderWithProviders(<AuthDialog {...props} />);

packages/cli/src/ui/auth/AuthDialog.tsx

@@ -119,7 +119,12 @@ export function AuthDialog({
         return;
       }
       if (authType) {
-        if (authType === AuthType.LOGIN_WITH_GOOGLE) {
+        const needsRestart =
+          authType === AuthType.LOGIN_WITH_GOOGLE ||
+          (authType === AuthType.USE_VERTEX_AI &&
+            process.env['CLOUD_SHELL'] === 'true');
+
+        if (needsRestart) {
           setAuthContext({ requiresRestart: true });
         } else {
           setAuthContext({});

packages/cli/src/ui/auth/LoginRestartDialog.test.tsx

@@ -1,12 +1,12 @@
 /**
  * @license
- * Copyright 2025 Google LLC
+ * Copyright 2026 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
 
 import { render } from '../../test-utils/render.js';
 import { describe, it, expect, vi, beforeEach, type Mock } from 'vitest';
-import { LoginWithGoogleRestartDialog } from './LoginWithGoogleRestartDialog.js';
+import { LoginRestartDialog } from './LoginRestartDialog.js';
 import { useKeypress } from '../hooks/useKeypress.js';
 import { runExitCleanup } from '../../utils/cleanup.js';
 import {
@@ -27,7 +27,7 @@ vi.mock('../../utils/cleanup.js', () => ({
 const mockedUseKeypress = useKeypress as Mock;
 const mockedRunExitCleanup = runExitCleanup as Mock;
 
-describe('LoginWithGoogleRestartDialog', () => {
+describe('LoginRestartDialog', () => {
   const onDismiss = vi.fn();
   const exitSpy = vi
     .spyOn(process, 'exit')
@@ -44,11 +44,20 @@ describe('LoginWithGoogleRestartDialog', () => {
     _resetRelaunchStateForTesting();
   });
 
-  it('renders correctly', async () => {
+  it('renders correctly with default message', async () => {
     const { lastFrame, unmount } = await render(
-      <LoginWithGoogleRestartDialog
+      <LoginRestartDialog onDismiss={onDismiss} config={mockConfig} />,
+    );
+    expect(lastFrame()).toMatchSnapshot();
+    unmount();
+  });
+
+  it('renders correctly with custom message', async () => {
+    const { lastFrame, unmount } = await render(
+      <LoginRestartDialog
         onDismiss={onDismiss}
         config={mockConfig}
+        message="Authenticating to Vertex AI in Cloud Shell requires a restart to apply project settings."
       />,
     );
     expect(lastFrame()).toMatchSnapshot();
@@ -57,10 +66,7 @@ describe('LoginWithGoogleRestartDialog', () => {
 
   it('calls onDismiss when escape is pressed', async () => {
     const { unmount } = await render(
-      <LoginWithGoogleRestartDialog
-        onDismiss={onDismiss}
-        config={mockConfig}
-      />,
+      <LoginRestartDialog onDismiss={onDismiss} config={mockConfig} />,
     );
     const keypressHandler = mockedUseKeypress.mock.calls[0][0];
 
@@ -82,10 +88,7 @@ describe('LoginWithGoogleRestartDialog', () => {
       vi.useFakeTimers();
 
       const { unmount } = await render(
-        <LoginWithGoogleRestartDialog
-          onDismiss={onDismiss}
-          config={mockConfig}
-        />,
+        <LoginRestartDialog onDismiss={onDismiss} config={mockConfig} />,
       );
       const keypressHandler = mockedUseKeypress.mock.calls[0][0];
 

packages/cli/src/ui/auth/LoginRestartDialog.tsx

@@ -1,6 +1,6 @@
 /**
  * @license
- * Copyright 2025 Google LLC
+ * Copyright 2026 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
 
@@ -10,15 +10,17 @@ import { theme } from '../semantic-colors.js';
 import { useKeypress } from '../hooks/useKeypress.js';
 import { relaunchApp } from '../../utils/processUtils.js';
 
-interface LoginWithGoogleRestartDialogProps {
+interface LoginRestartDialogProps {
   onDismiss: () => void;
   config: Config;
+  message?: string;
 }
 
-export const LoginWithGoogleRestartDialog = ({
+export const LoginRestartDialog = ({
   onDismiss,
   config,
-}: LoginWithGoogleRestartDialogProps) => {
+  message,
+}: LoginRestartDialogProps) => {
   useKeypress(
     (key) => {
       if (key.name === 'escape') {
@@ -44,14 +46,20 @@ export const LoginWithGoogleRestartDialog = ({
     { isActive: true },
   );
 
-  const message =
+  const displayMessage =
+    message ??
     "You've successfully signed in with Google. Gemini CLI needs to be restarted.";
 
   return (
-    <Box borderStyle="round" borderColor={theme.status.warning} paddingX={1}>
+    <Box
+      borderStyle="round"
+      borderColor={theme.status.warning}
+      paddingX={1}
+      flexDirection="column"
+    >
+      <Text color={theme.status.warning}>{displayMessage}</Text>
       <Text color={theme.status.warning}>
-        {message} Press R to restart, or Esc to choose a different
-        authentication method.
+        Press R to restart, or Esc to choose a different authentication method.
       </Text>
     </Box>
   );

packages/cli/src/ui/auth/__snapshots__/LoginRestartDialog.test.tsx.snap

@@ -0,0 +1,17 @@
+// Vitest Snapshot v1, https://vitest.dev/guide/snapshot.html
+
+exports[`LoginRestartDialog > renders correctly with custom message 1`] = `
+"╭──────────────────────────────────────────────────────────────────────────────────────────────────╮
+│ Authenticating to Vertex AI in Cloud Shell requires a restart to apply project settings.         │
+│ Press R to restart, or Esc to choose a different authentication method.                          │
+╰──────────────────────────────────────────────────────────────────────────────────────────────────╯
+"
+`;
+
+exports[`LoginRestartDialog > renders correctly with default message 1`] = `
+"╭──────────────────────────────────────────────────────────────────────────────────────────────────╮
+│ You've successfully signed in with Google. Gemini CLI needs to be restarted.                     │
+│ Press R to restart, or Esc to choose a different authentication method.                          │
+╰──────────────────────────────────────────────────────────────────────────────────────────────────╯
+"
+`;

packages/cli/src/ui/auth/__snapshots__/LoginWithGoogleRestartDialog.test.tsx.snap

@@ -1,9 +0,0 @@
-// Vitest Snapshot v1, https://vitest.dev/guide/snapshot.html
-
-exports[`LoginWithGoogleRestartDialog > renders correctly 1`] = `
-"╭──────────────────────────────────────────────────────────────────────────────────────────────────╮
-│ You've successfully signed in with Google. Gemini CLI needs to be restarted. Press R to restart, │
-│ or Esc to choose a different authentication method.                                              │
-╰──────────────────────────────────────────────────────────────────────────────────────────────────╯
-"
-`;