diff options
Diffstat (limited to 'packages/cli/src/ui')
| -rw-r--r-- | packages/cli/src/ui/components/InputPrompt.tsx | 6 | ||||
| -rw-r--r-- | packages/cli/src/ui/hooks/useGeminiStream.ts | 3 |
2 files changed, 6 insertions, 3 deletions
diff --git a/packages/cli/src/ui/components/InputPrompt.tsx b/packages/cli/src/ui/components/InputPrompt.tsx index f79aeaa3..1102e75d 100644 --- a/packages/cli/src/ui/components/InputPrompt.tsx +++ b/packages/cli/src/ui/components/InputPrompt.tsx @@ -1,7 +1,9 @@ import React from 'react'; import { Box, Text } from 'ink'; import TextInput from 'ink-text-input'; -import { getModel } from '../../config/globalConfig.js'; +import { globalConfig } from '../../config/config.js'; + + interface InputPromptProps { query: string; @@ -15,7 +17,7 @@ const InputPrompt: React.FC<InputPromptProps> = ({ setQuery, onSubmit, }) => { - const model = getModel(); + const model = globalConfig.getModel(); return ( <Box marginTop={1} borderStyle="round" borderColor={'white'} paddingX={1}> diff --git a/packages/cli/src/ui/hooks/useGeminiStream.ts b/packages/cli/src/ui/hooks/useGeminiStream.ts index 63f110b5..0b62a40b 100644 --- a/packages/cli/src/ui/hooks/useGeminiStream.ts +++ b/packages/cli/src/ui/hooks/useGeminiStream.ts @@ -4,6 +4,7 @@ import { GeminiClient } from '../../core/gemini-client.js'; import { type Chat, type PartListUnion } from '@google/genai'; import { HistoryItem } from '../types.js'; import { processGeminiStream , StreamingState } from '../../core/gemini-stream.js'; +import { globalConfig } from '../../config/config.js'; const addHistoryItem = ( setHistory: React.Dispatch<React.SetStateAction<HistoryItem[]>>, @@ -34,7 +35,7 @@ export const useGeminiStream = ( setInitError(null); if (!geminiClientRef.current) { try { - geminiClientRef.current = new GeminiClient(); + geminiClientRef.current = new GeminiClient(globalConfig); } catch (error: any) { setInitError( `Failed to initialize client: ${error.message || 'Unknown error'}`, |