From a66ad2e2af3d4ffd0a7ab125215e7ed3c3325452 Mon Sep 17 00:00:00 2001 From: Juliette Love Date: Sun, 20 Apr 2025 20:20:40 +0100 Subject: Simple debug mode for CLI side (#66) * Adds debug mode config flag. * Wire through debug lines * Add debug mode logging example * Run format * Run format again --- packages/cli/src/ui/hooks/useGeminiStream.ts | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) (limited to 'packages/cli/src/ui/hooks') diff --git a/packages/cli/src/ui/hooks/useGeminiStream.ts b/packages/cli/src/ui/hooks/useGeminiStream.ts index 56203179..1d839998 100644 --- a/packages/cli/src/ui/hooks/useGeminiStream.ts +++ b/packages/cli/src/ui/hooks/useGeminiStream.ts @@ -49,6 +49,7 @@ export const useGeminiStream = ( const [streamingState, setStreamingState] = useState( StreamingState.Idle, ); + const [debugMessage, setDebugMessage] = useState(''); const [initError, setInitError] = useState(null); const abortControllerRef = useRef(null); const chatSessionRef = useRef(null); @@ -104,6 +105,10 @@ export const useGeminiStream = ( if (streamingState === StreamingState.Responding) return; if (typeof query === 'string' && query.trim().length === 0) return; + if (typeof query === 'string') { + setDebugMessage(`User query: ${query}`); + } + const userMessageTimestamp = Date.now(); const client = geminiClientRef.current; if (!client) { @@ -403,7 +408,7 @@ export const useGeminiStream = ( ], ); - return { streamingState, submitQuery, initError }; + return { streamingState, submitQuery, initError, debugMessage }; }; // Define ServerTool interface here if not importing from server (circular dep issue?) -- cgit v1.2.3