get model when create new conversation

fix send second message

Author: JYC0413

components/Chat/Chat.tsx

@@ -129,7 +129,7 @@ export const Chat = memo(({stopConversationRef}: Props) => {
                         if (!queryDoneRef.current) {
                             await delay(textListRef.current.length > 10 ? 300 : 100)
                         } else {
-                            await delay(20)
+                            // await delay(20)
                         }
                     }
                 }
@@ -139,10 +139,6 @@ export const Chat = memo(({stopConversationRef}: Props) => {
 
         const whileShowText = async () => {
             while (textListRef.current.length !== 0) {
-                if (textListRef.current.length > 0) {
-                    console.log(textListRef.current)
-                    console.log("show")
-                }
                 await showText();
             }
         }
@@ -219,6 +215,7 @@ export const Chat = memo(({stopConversationRef}: Props) => {
                             messagesToSend = [message, ...messagesToSend];
                         }
                         if (isStream) {
+                            queryDoneRef.current = false;
                             let response: ReadableStream<Uint8Array> | null = await ChatStream(model, promptToSend, temperatureToUse, api, key, messagesToSend);
                             if (response) {
                                 let notFinishData = "";
@@ -227,7 +224,6 @@ export const Chat = memo(({stopConversationRef}: Props) => {
                                 while (!queryDoneRef.current) {
                                     const {value, done} = await reader.read();
                                     if (done) {
-                                        console.log("done-221")
                                         queryDoneRef.current = true;
                                     }
                                     let chunkValue = decoder.decode(value);
@@ -240,7 +236,6 @@ export const Chat = memo(({stopConversationRef}: Props) => {
                                                 part = part.substring(6).trim();
                                             }
                                             if (part === "[DONE]") {
-                                                console.log("done-234")
                                                 queryDoneRef.current = true;
                                             } else {
                                                 if (!part.startsWith('{')) {

pages/api/home/home.tsx

@@ -18,7 +18,7 @@ import {getSettings} from '@/utils/app/settings';
 import {Conversation} from '@/types/chat';
 import {KeyValuePair} from '@/types/data';
 import {FolderInterface, FolderType} from '@/types/folder';
-import {fallbackModelID, OpenAIModelID} from '@/types/openai';
+import {fallbackModelID, OpenAIModel, OpenAIModelID} from '@/types/openai';
 import {Prompt} from '@/types/prompt';
 
 import {Chat} from '@/components/Chat/Chat';
@@ -181,14 +181,24 @@ const Home = ({
 
     // CONVERSATION OPERATIONS  --------------------------------------------
 
-    const handleNewConversation = () => {
+    function checkObjectExistence(objectList: OpenAIModel[], object: OpenAIModel): boolean {
+        for (let obj of objectList) {
+            if (obj.id === object.id) {
+                return true;
+            }
+        }
+        return false;
+    }
+
+    const handleNewConversation = async () => {
+        await getData()
         const lastConversation = conversations[conversations.length - 1];
         if (models && models.length > 0) {
             const newConversation: Conversation = {
                 id: uuidv4(),
                 name: t('New Conversation'),
                 messages: [],
-                model: lastConversation?.model || models[0],
+                model: (lastConversation?.model && checkObjectExistence(models, lastConversation?.model) && lastConversation?.model) || models[0],
                 prompt: promptsList.find(prompt =>
                     prompt.id?.toLowerCase() === models[0].name?.toLowerCase()
                 )?.content || "",