fix(openai.js): bugs of chatWithAI

web-tech-tw · Jun 11, 2024 · bd3b96e · bd3b96e
1 parent db53fe6
commit bd3b96e
Showing 1 changed file with 24 additions and 15 deletions.
diff --git a/src/clients/openai.js b/src/clients/openai.js
@@ -36,30 +36,39 @@ async function chatWithAI(chatId, prompt) {
     if (!chatHistoryMapper.has(chatId)) {
         chatHistoryMapper.set(chatId, []);
     }
-    const chatHistory = chatHistoryMapper.get(chatId);
 
-    const messages = [
-        ...chatHistory,
-        ...prependPrompts,
-        {
-            role: "user",
-            content: prompt,
-        },
-    ];
+    const chatHistory = chatHistoryMapper.get(chatId);
+    const userPromptMessage = {
+        role: "user",
+        content: prompt,
+    };
 
     const response = await client.chat.completions.create({
         model: chatModel,
-        messages,
+        messages: [
+            ...chatHistory,
+            ...prependPrompts,
+            userPromptMessage,
+        ],
     });
 
     const choice = choose(response.choices);
-    const content = choice.message.content;
-    chatHistory.push({
+    const reply = choice.message.content;
+    const assistantReplyMessage = {
         role: "assistant",
-        content,
-    });
+        content: reply,
+    };
+
+    chatHistory.push(
+        userPromptMessage,
+        assistantReplyMessage,
+    );
+    if (chatHistory.length > 30) {
+        chatHistory.shift();
+        chatHistory.shift();
+    }
 
-    return choice.message.content;
+    return reply;
 }
 
 exports.useClient = () => client;