Ling0925
/
buster
mirror of https://github.com/buster-so/buster.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update useChatStreamMessage.ts

This commit is contained in:
Nate Kelley 2025-03-05 16:22:45 -07:00
parent 5b38037419
commit d7430f1594
No known key found for this signature in database
GPG Key ID: FD90372AB8D98B4F
1 changed files with 12 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
web/src/context/Chats/NewChatProvider/useChatStreamMessage.ts
View File

@ -219,15 +219,18 @@ export const useChatStreamMessage = () => {
reasoning.message_chunk !== null || reasoning.message_chunk !== undefined; reasoning.message_chunk !== null || reasoning.message_chunk !== undefined;
initializeOrUpdateMessage(chat_id, message_id, (draft) => { initializeOrUpdateMessage(chat_id, message_id, (draft) => {
if (!draft[chat_id]?.messages?.[message_id]?.reasoning_messages?.[reasoningMessageId]) const reasoningMessage =
return; draft[chat_id]?.messages?.[message_id]?.reasoning_messages?.[reasoningMessageId];
const messageText = draft[chat_id].messages[message_id].reasoning_messages[ if (!reasoningMessage) return;
reasoningMessageId const messageText = reasoningMessage as BusterChatMessageReasoning_text;
] as BusterChatMessageReasoning_text;
Object.assign(messageText, existingReasoningMessageText); Object.assign(messageText, {
messageText.message = isStreaming ...existingReasoningMessageText,
? (existingReasoningMessageText?.message || '') + (reasoning.message_chunk || '') ...reasoning,
: reasoning.message; message: isStreaming
? (existingReasoningMessageText?.message || '') + (reasoning.message_chunk || '')
: reasoning.message
});
}); });
break; break;