Elgato_dark/void
1
0
Fork 0
mirror of https://github.com/voideditor/void synced 2026-05-24 09:58:23 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix cancel streaming

Browse source
This commit is contained in:
Andrew Pareles 2025-01-17 16:39:24 -08:00
parent ce7ee62abe
commit dce0378eb2
1 changed files with 8 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
src/vs/workbench/contrib/void/browser/react/src/sidebar-tsx/SidebarChat.tsx
View file

@ -548,7 +548,8 @@ export const SidebarChat = () => {
// stream state
const chatThreadsStreamState = useChatThreadsStreamState(chatThreadsState.currentThreadId)
const isCurrThreadStreaming = !!chatThreadsStreamState?.streamingToken
const streamingToken = chatThreadsStreamState?.streamingToken
const isStreaming = !!streamingToken
const latestError = chatThreadsStreamState?.error
const messageSoFar = chatThreadsStreamState?.messageSoFar
@ -569,7 +570,7 @@ export const SidebarChat = () => {
const onSubmit = async () => {
if (isDisabled) return
if (isCurrThreadStreaming) return
if (isStreaming) return
// send message to LLM
const userMessage = textAreaRef.current?.value ?? ''
@ -582,9 +583,9 @@ export const SidebarChat = () => {
}
const onAbort = () => {
const token = chatThreadsStreamState?.streamingToken
if (!token) return
chatThreadsService.cancelStreaming(token)
// this assumes an instant cancellation doesn't happen, since streamingToken state must have updated by this time
if (!streamingToken) return
chatThreadsService.cancelStreaming(streamingToken)
}
// const [_test_messages, _set_test_messages] = useState<string[]>([])
@ -626,7 +627,7 @@ export const SidebarChat = () => {
)}
{/* message stream */}
<ChatBubble chatMessage={{ role: 'assistant', content: messageSoFar ?? '', displayContent: messageSoFar || null }} isLoading={isCurrThreadStreaming} />
<ChatBubble chatMessage={{ role: 'assistant', content: messageSoFar ?? '', displayContent: messageSoFar || null }} isLoading={isStreaming} />
</ScrollToBottomContainer>
@ -700,7 +701,7 @@ export const SidebarChat = () => {
</div>
{/* submit / stop button */}
{isCurrThreadStreaming ?
{isStreaming ?
// stop button
<ButtonStop
onClick={onAbort}