Fix: missing edit icon for prompts (#4344)

consolidated duplicate handler to fix prompt icon behavior

Co-authored-by: Timothy Carambat <rambat1010@gmail.com>
This commit is contained in:
Colin Perry 2025-09-03 10:44:27 -07:00 committed by GitHub
parent e31465a639
commit 2500c94b83
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -103,8 +103,11 @@ export default function handleChat(
chatId,
metrics,
};
setLoadingResponse(false);
_chatHistory[chatIdx - 1] = { ..._chatHistory[chatIdx - 1], chatId }; // update prompt with chatID
emitAssistantMessageCompleteEvent(chatId);
setLoadingResponse(false);
} else {
updatedHistory = {
...existingHistory,
@ -136,15 +139,6 @@ export default function handleChat(
setChatHistory([..._chatHistory]);
} else if (type === "agentInitWebsocketConnection") {
setWebsocket(chatResult.websocketUUID);
} else if (type === "finalizeResponseStream") {
const chatIdx = _chatHistory.findIndex((chat) => chat.uuid === uuid);
if (chatIdx !== -1) {
_chatHistory[chatIdx - 1] = { ..._chatHistory[chatIdx - 1], chatId }; // update prompt with chatID
_chatHistory[chatIdx] = { ..._chatHistory[chatIdx], chatId }; // update response with chatID
}
setChatHistory([..._chatHistory]);
setLoadingResponse(false);
} else if (type === "stopGeneration") {
const chatIdx = _chatHistory.length - 1;
const existingHistory = { ..._chatHistory[chatIdx] };