diff options
Diffstat (limited to 'packages/cli/src/ui/hooks/useGeminiStream.ts')
| -rw-r--r-- | packages/cli/src/ui/hooks/useGeminiStream.ts | 7 |
1 files changed, 6 insertions, 1 deletions
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>( StreamingState.Idle, ); + const [debugMessage, setDebugMessage] = useState<string>(''); const [initError, setInitError] = useState<string | null>(null); const abortControllerRef = useRef<AbortController | null>(null); const chatSessionRef = useRef<Chat | null>(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?) |
