summaryrefslogtreecommitdiff
path: root/packages/cli/src/ui/components
diff options
context:
space:
mode:
Diffstat (limited to 'packages/cli/src/ui/components')
-rw-r--r--packages/cli/src/ui/components/LoadingIndicator.test.tsx52
-rw-r--r--packages/cli/src/ui/components/LoadingIndicator.tsx42
2 files changed, 77 insertions, 17 deletions
diff --git a/packages/cli/src/ui/components/LoadingIndicator.test.tsx b/packages/cli/src/ui/components/LoadingIndicator.test.tsx
index 3d31818b..ba161062 100644
--- a/packages/cli/src/ui/components/LoadingIndicator.test.tsx
+++ b/packages/cli/src/ui/components/LoadingIndicator.test.tsx
@@ -159,4 +159,56 @@ describe('<LoadingIndicator />', () => {
);
expect(lastFrame()).toBe('');
});
+
+ it('should display fallback phrase if thought is empty', () => {
+ const props = {
+ thought: null,
+ currentLoadingPhrase: 'Loading...',
+ elapsedTime: 5,
+ };
+ const { lastFrame } = renderWithContext(
+ <LoadingIndicator {...props} />,
+ StreamingState.Responding,
+ );
+ const output = lastFrame();
+ expect(output).toContain('Loading...');
+ });
+
+ it('should display the subject of a thought', () => {
+ const props = {
+ thought: {
+ subject: 'Thinking about something...',
+ description: 'and other stuff.',
+ },
+ elapsedTime: 5,
+ };
+ const { lastFrame } = renderWithContext(
+ <LoadingIndicator {...props} />,
+ StreamingState.Responding,
+ );
+ const output = lastFrame();
+ expect(output).toBeDefined();
+ if (output) {
+ expect(output).toContain('Thinking about something...');
+ expect(output).not.toContain('and other stuff.');
+ }
+ });
+
+ it('should prioritize thought.subject over currentLoadingPhrase', () => {
+ const props = {
+ thought: {
+ subject: 'This should be displayed',
+ description: 'A description',
+ },
+ currentLoadingPhrase: 'This should not be displayed',
+ elapsedTime: 5,
+ };
+ const { lastFrame } = renderWithContext(
+ <LoadingIndicator {...props} />,
+ StreamingState.Responding,
+ );
+ const output = lastFrame();
+ expect(output).toContain('This should be displayed');
+ expect(output).not.toContain('This should not be displayed');
+ });
});
diff --git a/packages/cli/src/ui/components/LoadingIndicator.tsx b/packages/cli/src/ui/components/LoadingIndicator.tsx
index 61b74b89..855894e6 100644
--- a/packages/cli/src/ui/components/LoadingIndicator.tsx
+++ b/packages/cli/src/ui/components/LoadingIndicator.tsx
@@ -4,6 +4,7 @@
* SPDX-License-Identifier: Apache-2.0
*/
+import { ThoughtSummary } from '@gemini-cli/core';
import React from 'react';
import { Box, Text } from 'ink';
import { Colors } from '../colors.js';
@@ -15,12 +16,14 @@ interface LoadingIndicatorProps {
currentLoadingPhrase?: string;
elapsedTime: number;
rightContent?: React.ReactNode;
+ thought?: ThoughtSummary | null;
}
export const LoadingIndicator: React.FC<LoadingIndicatorProps> = ({
currentLoadingPhrase,
elapsedTime,
rightContent,
+ thought,
}) => {
const streamingState = useStreamingContext();
@@ -28,25 +31,30 @@ export const LoadingIndicator: React.FC<LoadingIndicatorProps> = ({
return null;
}
+ const primaryText = thought?.subject || currentLoadingPhrase;
+
return (
- <Box marginTop={1} paddingLeft={0}>
- <Box marginRight={1}>
- <GeminiRespondingSpinner
- nonRespondingDisplay={
- streamingState === StreamingState.WaitingForConfirmation ? '⠏' : ''
- }
- />
+ <Box marginTop={1} paddingLeft={0} flexDirection="column">
+ {/* Main loading line */}
+ <Box>
+ <Box marginRight={1}>
+ <GeminiRespondingSpinner
+ nonRespondingDisplay={
+ streamingState === StreamingState.WaitingForConfirmation
+ ? '⠏'
+ : ''
+ }
+ />
+ </Box>
+ {primaryText && <Text color={Colors.AccentPurple}>{primaryText}</Text>}
+ <Text color={Colors.Gray}>
+ {streamingState === StreamingState.WaitingForConfirmation
+ ? ''
+ : ` (esc to cancel, ${elapsedTime}s)`}
+ </Text>
+ <Box flexGrow={1}>{/* Spacer */}</Box>
+ {rightContent && <Box>{rightContent}</Box>}
</Box>
- {currentLoadingPhrase && (
- <Text color={Colors.AccentPurple}>{currentLoadingPhrase}</Text>
- )}
- <Text color={Colors.Gray}>
- {streamingState === StreamingState.WaitingForConfirmation
- ? ''
- : ` (esc to cancel, ${elapsedTime}s)`}
- </Text>
- <Box flexGrow={1}>{/* Spacer */}</Box>
- {rightContent && <Box>{rightContent}</Box>}
</Box>
);
};