fix bug on completion with empty history

2023-07-08 14:16:51 +08:00
parent 1b8eeb0c86
commit 8fb17ba3f8
1 changed files with 4 additions and 2 deletions
--- a/src/chatbox.tsx
+++ b/src/chatbox.tsx
@@ -169,9 +169,11 @@ export default function ChatBOX(props: {
    if (data.usage.prompt_tokens) {
      const userMessageToken = data.usage.prompt_tokens - aboveToken;
      console.log("set user message token");
+      if (chatStore.history.filter((msg) => !msg.hide).length > 0) {
        chatStore.history.filter((msg) => !msg.hide).slice(-1)[0].token =
          userMessageToken;
      }
+    }

    chatStore.history.push({
      role: "assistant",