diff options
| -rw-r--r-- | packages/cli/src/ui/hooks/useGeminiStream.ts | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/packages/cli/src/ui/hooks/useGeminiStream.ts b/packages/cli/src/ui/hooks/useGeminiStream.ts index 6ff41d2f..4fcc503b 100644 --- a/packages/cli/src/ui/hooks/useGeminiStream.ts +++ b/packages/cli/src/ui/hooks/useGeminiStream.ts @@ -483,6 +483,10 @@ export const useGeminiStream = ( error: undefined, }; updateFunctionResponseUI(responseInfo, ToolCallStatus.Success); + if (pendingHistoryItemRef.current) { + addItem(pendingHistoryItemRef.current, Date.now()); + setPendingHistoryItem(null); + } setStreamingState(StreamingState.Idle); await submitQuery(functionResponse); } finally { @@ -529,6 +533,10 @@ export const useGeminiStream = ( // Update UI to show cancellation/error updateFunctionResponseUI(responseInfo, status); + if (pendingHistoryItemRef.current) { + addItem(pendingHistoryItemRef.current, Date.now()); + setPendingHistoryItem(null); + } setStreamingState(StreamingState.Idle); } }; |