From b0d6484d748fe8d9c4f68e278102f04f354650c9 Mon Sep 17 00:00:00 2001
From: heimoshuiyu <heimoshuiyu@gmail.com>
Date: Fri, 25 Aug 2023 10:21:45 +0800
Subject: [PATCH] fix: stream response model name and cost

---
 src/chatbox.tsx | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/chatbox.tsx b/src/chatbox.tsx
index 2c9a602..fac8bcb 100644
--- a/src/chatbox.tsx
+++ b/src/chatbox.tsx
@@ -32,7 +32,6 @@ export default function ChatBOX(props: {
 
   const messagesEndRef = createRef();
   useEffect(() => {
-    console.log("ref", messagesEndRef);
     messagesEndRef.current.scrollIntoView({ behavior: "smooth" });
   }, [showRetry, showGenerating, generatingMessage]);
 
@@ -57,6 +56,7 @@ export default function ChatBOX(props: {
     const allChunkMessage: string[] = [];
     setShowGenerating(true);
     for await (const i of client.processStreamResponse(response)) {
+      chatStore.responseModelName = i.model;
       responseTokenCount += 1;
       allChunkMessage.push(i.choices[0].delta.content ?? "");
       setGeneratingMessage(allChunkMessage.join(""));
@@ -79,6 +79,7 @@ export default function ChatBOX(props: {
       cost += sum * (models[chatStore.responseModelName]?.price?.prompt ?? 0);
     }
 
+    console.log("cost", cost);
     chatStore.cost += cost;
     addTotalCost(cost);