90 lines
3.5 KiB
TypeScript
90 lines
3.5 KiB
TypeScript
|
|
import React, { useState, useEffect } from 'react';
|
||
|
|
import { Box, Text } from 'ink';
|
||
|
|
import type { HistoryItem } from './types.js';
|
||
|
|
import { useGeminiStream } from './hooks/useGeminiStream.js';
|
||
|
|
import { useLoadingIndicator } from './hooks/useLoadingIndicator.js';
|
||
|
|
import Header from './components/Header.js';
|
||
|
|
import Tips from './components/Tips.js';
|
||
|
|
import HistoryDisplay from './components/HistoryDisplay.js';
|
||
|
|
import LoadingIndicator from './components/LoadingIndicator.js';
|
||
|
|
import InputPrompt from './components/InputPrompt.js';
|
||
|
|
import Footer from './components/Footer.js';
|
||
|
|
import { StreamingState } from '../core/StreamingState.js';
|
||
|
|
import { PartListUnion } from '@google/genai';
|
||
|
|
|
||
|
|
interface AppProps {
|
||
|
|
directory: string;
|
||
|
|
}
|
||
|
|
|
||
|
|
const App = ({ directory }: AppProps) => {
|
||
|
|
const [query, setQuery] = useState('');
|
||
|
|
const [history, setHistory] = useState<HistoryItem[]>([]);
|
||
|
|
const { streamingState, submitQuery, initError } = useGeminiStream(setHistory);
|
||
|
|
const { elapsedTime, currentLoadingPhrase } = useLoadingIndicator(streamingState);
|
||
|
|
|
||
|
|
const handleInputSubmit = (value: PartListUnion) => {
|
||
|
|
submitQuery(value).then(() => {
|
||
|
|
setQuery('');
|
||
|
|
}).catch(() => {
|
||
|
|
setQuery('');
|
||
|
|
});
|
||
|
|
};
|
||
|
|
|
||
|
|
useEffect(() => {
|
||
|
|
if (initError && !history.some(item => item.type === 'error' && item.text?.includes(initError))) {
|
||
|
|
setHistory(prev => [
|
||
|
|
...prev,
|
||
|
|
{ id: Date.now(), type: 'error', text: `Initialization Error: ${initError}. Please check API key and configuration.` } as HistoryItem
|
||
|
|
]);
|
||
|
|
}
|
||
|
|
}, [initError, history]);
|
||
|
|
|
||
|
|
const isWaitingForToolConfirmation = history.some(item =>
|
||
|
|
item.type === 'tool_group' && item.tools.some(tool => tool.confirmationDetails !== undefined)
|
||
|
|
);
|
||
|
|
const isInputActive = streamingState === StreamingState.Idle && !initError;
|
||
|
|
|
||
|
|
|
||
|
|
return (
|
||
|
|
<Box flexDirection="column" padding={1} marginBottom={1} width="100%">
|
||
|
|
<Header cwd={directory} />
|
||
|
|
|
||
|
|
<Tips />
|
||
|
|
|
||
|
|
{initError && streamingState !== StreamingState.Responding && !isWaitingForToolConfirmation && (
|
||
|
|
<Box borderStyle="round" borderColor="red" paddingX={1} marginBottom={1}>
|
||
|
|
{history.find(item => item.type === 'error' && item.text?.includes(initError))?.text ? (
|
||
|
|
<Text color="red">{history.find(item => item.type === 'error' && item.text?.includes(initError))?.text}</Text>
|
||
|
|
) : (
|
||
|
|
<>
|
||
|
|
<Text color="red">Initialization Error: {initError}</Text>
|
||
|
|
<Text color="red"> Please check API key and configuration.</Text>
|
||
|
|
</>
|
||
|
|
)}
|
||
|
|
</Box>
|
||
|
|
)}
|
||
|
|
|
||
|
|
<Box flexDirection="column">
|
||
|
|
<HistoryDisplay history={history} onSubmit={handleInputSubmit} />
|
||
|
|
<LoadingIndicator
|
||
|
|
isLoading={streamingState === StreamingState.Responding}
|
||
|
|
currentLoadingPhrase={currentLoadingPhrase}
|
||
|
|
elapsedTime={elapsedTime}
|
||
|
|
/>
|
||
|
|
</Box>
|
||
|
|
|
||
|
|
{!isWaitingForToolConfirmation && isInputActive && (
|
||
|
|
<InputPrompt
|
||
|
|
query={query}
|
||
|
|
setQuery={setQuery}
|
||
|
|
onSubmit={handleInputSubmit}
|
||
|
|
isActive={isInputActive}
|
||
|
|
/>
|
||
|
|
)}
|
||
|
|
|
||
|
|
<Footer queryLength={query.length} />
|
||
|
|
</Box>
|
||
|
|
);
|
||
|
|
};
|
||
|
|
|
||
|
|
export default App;
|