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

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

@ -176,17 +176,20 @@ 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 (tokens < MAX_TOKENS) { if (message.status === "DONE") {
tokens += countTextTokens(message.content); if (tokens < MAX_TOKENS) {
usageMessageList.unshift(message); tokens += countTextTokens(message.content);
formatedMessageList.unshift({ usageMessageList.unshift(message);
role: message.creatorRole, formatedMessageList.unshift({
content: message.content, role: message.creatorRole,
}); content: message.content,
});
}
} }
} }
usageMessageList.unshift({ usageMessageList.unshift({