fix: flicker of StreamingState to Idle when tool finishes (#1190) (#1216)

Co-authored-by: Asad Memon <asad.lionpk@gmail.com>
This commit is contained in:
N. Taylor Mullen
2025-06-19 18:25:23 -07:00
committed by GitHub
parent 2e5e9d736b
commit 4059a3e8ee
2 changed files with 101 additions and 9 deletions

View File

@@ -155,7 +155,12 @@ export const useGeminiStream = (
(tc) =>
tc.status === 'executing' ||
tc.status === 'scheduled' ||
tc.status === 'validating',
tc.status === 'validating' ||
((tc.status === 'success' ||
tc.status === 'error' ||
tc.status === 'cancelled') &&
!(tc as TrackedCompletedToolCall | TrackedCancelledToolCall)
.responseSubmittedToGemini),
)
) {
return StreamingState.Responding;
@@ -453,8 +458,9 @@ export const useGeminiStream = (
const submitQuery = useCallback(
async (query: PartListUnion, options?: { isContinuation: boolean }) => {
if (
streamingState === StreamingState.Responding ||
streamingState === StreamingState.WaitingForConfirmation
(streamingState === StreamingState.Responding ||
streamingState === StreamingState.WaitingForConfirmation) &&
!options?.isContinuation
)
return;