opensource/sqlchat
1
0
Fork 0
mirror of https://github.com/sqlchat/sqlchat.git synced 2025-09-25 09:03:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: should only add message with DONE status

Browse Source
This commit is contained in:
Tianzhou Chen
2023-05-02 12:14:45 +08:00
parent f3476d8d91
commit 00963f4bf7
1 changed files with 10 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/components/ConversationView/index.tsx
View File

@ -176,10 +176,12 @@ const ConversationView = () => {
prompt = promptGenerator(schema); prompt = promptGenerator(schema);
} }
// Sliding window to add messages with DONE status all the way back up to MAX_TOKENS
let usageMessageList: Message[] = []; let usageMessageList: Message[] = [];
let formatedMessageList = []; let formatedMessageList = [];
for (let i = messageList.length - 1; i >= 0; i--) { for (let i = messageList.length - 1; i >= 0; i--) {
const message = messageList[i]; const message = messageList[i];
if (message.status === "DONE") {
if (tokens < MAX_TOKENS) { if (tokens < MAX_TOKENS) {
tokens += countTextTokens(message.content); tokens += countTextTokens(message.content);
usageMessageList.unshift(message); usageMessageList.unshift(message);
@ -189,6 +191,7 @@ const ConversationView = () => {
}); });
} }
} }
}
usageMessageList.unshift({ usageMessageList.unshift({
id: generateUUID(), id: generateUUID(),
createdAt: first(usageMessageList)?.createdAt || Date.now(), createdAt: first(usageMessageList)?.createdAt || Date.now(),