From 3afaa8033bdd9d6af99ff378cafcd84afd76c59e Mon Sep 17 00:00:00 2001
From: Jaana Dogan <jbd@google.com>
Date: Fri, 18 Apr 2025 11:12:18 -0700
Subject: Introduce a config module to manage configuration (#22)

* Introduce a config module to manage configuration

* Remove public modifier
---
 packages/cli/src/ui/components/InputPrompt.tsx | 6 ++++--
 packages/cli/src/ui/hooks/useGeminiStream.ts   | 3 ++-
 2 files changed, 6 insertions(+), 3 deletions(-)

(limited to 'packages/cli/src/ui')

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'}`,
-- 
cgit v1.2.3