opensource/sqlchat
1
0
Fork 0
mirror of https://github.com/sqlchat/sqlchat.git synced 2025-09-24 16:46:05 +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);
}
// Sliding window to add messages with DONE status all the way back up to MAX_TOKENS
let usageMessageList: Message[] = [];
let formatedMessageList = [];
for (let i = messageList.length - 1; i >= 0; i--) {
const message = messageList[i];
if (tokens < MAX_TOKENS) {
tokens += countTextTokens(message.content);
usageMessageList.unshift(message);
formatedMessageList.unshift({
role: message.creatorRole,
content: message.content,
});
if (message.status === "DONE") {
if (tokens < MAX_TOKENS) {
tokens += countTextTokens(message.content);
usageMessageList.unshift(message);
formatedMessageList.unshift({
role: message.creatorRole,
content: message.content,
});
}
}
}
usageMessageList.unshift({