Fix race condition by awaiting scheduleToolCalls (#16759)

Co-authored-by: Tommaso Sciortino <sciortino@gmail.com>
This commit is contained in:
christine betts
2026-01-16 11:55:15 -05:00
committed by GitHub
parent a15978593a
commit cfdc4cfca8
3 changed files with 86 additions and 52 deletions

View File

@@ -460,7 +460,7 @@ export const useGeminiStream = (
isClientInitiated: true,
prompt_id,
};
scheduleToolCalls([toolCallRequest], abortSignal);
await scheduleToolCalls([toolCallRequest], abortSignal);
return { queryToSend: null, shouldProceed: false };
}
case 'submit_prompt': {
@@ -923,7 +923,7 @@ export const useGeminiStream = (
}
}
if (toolCallRequests.length > 0) {
scheduleToolCalls(toolCallRequests, signal);
await scheduleToolCalls(toolCallRequests, signal);
}
return StreamProcessingStatus.Completed;
},