diff --git a/packages/cli/src/ui/components/HistoryItemDisplay.test.tsx b/packages/cli/src/ui/components/HistoryItemDisplay.test.tsx index 81a84a3b..180d66c6 100644 --- a/packages/cli/src/ui/components/HistoryItemDisplay.test.tsx +++ b/packages/cli/src/ui/components/HistoryItemDisplay.test.tsx @@ -10,6 +10,7 @@ import { HistoryItemDisplay } from './HistoryItemDisplay.js'; import type { HistoryItem } from '../types.js'; import { MessageType } from '../types.js'; import { SessionStatsProvider } from '../contexts/SessionContext.js'; +import type { Config } from '@google/gemini-cli-core'; // Mock child components vi.mock('./messages/ToolGroupMessage.js', () => ({ @@ -17,11 +18,13 @@ vi.mock('./messages/ToolGroupMessage.js', () => ({ })); describe('', () => { + const mockConfig = {} as unknown as Config; const baseItem = { id: 1, timestamp: 12345, isPending: false, terminalWidth: 80, + config: mockConfig, }; it('renders UserMessage for "user" type', () => { diff --git a/packages/cli/src/ui/components/HistoryItemDisplay.tsx b/packages/cli/src/ui/components/HistoryItemDisplay.tsx index 7e6ea8cd..7881f9be 100644 --- a/packages/cli/src/ui/components/HistoryItemDisplay.tsx +++ b/packages/cli/src/ui/components/HistoryItemDisplay.tsx @@ -29,7 +29,7 @@ interface HistoryItemDisplayProps { availableTerminalHeight?: number; terminalWidth: number; isPending: boolean; - config?: Config; + config: Config; isFocused?: boolean; commands?: readonly SlashCommand[]; } diff --git a/packages/cli/src/ui/components/messages/ToolConfirmationMessage.test.tsx b/packages/cli/src/ui/components/messages/ToolConfirmationMessage.test.tsx index fceb86b6..32fac4c6 100644 --- a/packages/cli/src/ui/components/messages/ToolConfirmationMessage.test.tsx +++ b/packages/cli/src/ui/components/messages/ToolConfirmationMessage.test.tsx @@ -13,6 +13,11 @@ import type { import { renderWithProviders } from '../../../test-utils/render.js'; describe('ToolConfirmationMessage', () => { + const mockConfig = { + isTrustedFolder: () => true, + getIdeMode: () => false, + } as unknown as Config; + it('should not display urls if prompt and url are the same', () => { const confirmationDetails: ToolCallConfirmationDetails = { type: 'info', @@ -25,6 +30,7 @@ describe('ToolConfirmationMessage', () => { const { lastFrame } = renderWithProviders( , @@ -48,6 +54,7 @@ describe('ToolConfirmationMessage', () => { const { lastFrame } = renderWithProviders( , diff --git a/packages/cli/src/ui/components/messages/ToolConfirmationMessage.tsx b/packages/cli/src/ui/components/messages/ToolConfirmationMessage.tsx index bec6666a..acddae11 100644 --- a/packages/cli/src/ui/components/messages/ToolConfirmationMessage.tsx +++ b/packages/cli/src/ui/components/messages/ToolConfirmationMessage.tsx @@ -23,7 +23,7 @@ import { useKeypress } from '../../hooks/useKeypress.js'; export interface ToolConfirmationMessageProps { confirmationDetails: ToolCallConfirmationDetails; - config?: Config; + config: Config; isFocused?: boolean; availableTerminalHeight?: number; terminalWidth: number; @@ -43,8 +43,8 @@ export const ToolConfirmationMessage: React.FC< const handleConfirm = async (outcome: ToolConfirmationOutcome) => { if (confirmationDetails.type === 'edit') { - const ideClient = config?.getIdeClient(); - if (config?.getIdeMode()) { + const ideClient = config.getIdeClient(); + if (config.getIdeMode()) { const cliOutcome = outcome === ToolConfirmationOutcome.Cancel ? 'rejected' : 'accepted'; await ideClient?.resolveDiffFromCli( @@ -56,7 +56,7 @@ export const ToolConfirmationMessage: React.FC< onConfirm(outcome); }; - const isTrustedFolder = config?.isTrustedFolder() !== false; + const isTrustedFolder = config.isTrustedFolder() !== false; useKeypress( (key) => { @@ -137,7 +137,7 @@ export const ToolConfirmationMessage: React.FC< value: ToolConfirmationOutcome.ProceedAlways, }); } - if (config?.getIdeMode()) { + if (config.getIdeMode()) { options.push({ label: 'No (esc)', value: ToolConfirmationOutcome.Cancel, diff --git a/packages/cli/src/ui/components/messages/ToolGroupMessage.tsx b/packages/cli/src/ui/components/messages/ToolGroupMessage.tsx index a200f98d..3558f926 100644 --- a/packages/cli/src/ui/components/messages/ToolGroupMessage.tsx +++ b/packages/cli/src/ui/components/messages/ToolGroupMessage.tsx @@ -20,7 +20,7 @@ interface ToolGroupMessageProps { toolCalls: IndividualToolCallDisplay[]; availableTerminalHeight?: number; terminalWidth: number; - config?: Config; + config: Config; isFocused?: boolean; }