fix: when user input a prompt, it wont have stop message before the new prompt

Author: Randall-Jiang

chat-client/src/client/mynahUi.ts

@@ -94,7 +94,7 @@ export const handlePromptInputChange = (mynahUi: MynahUI, tabId: string, options
     }
 }
 
-export const handleChatPrompt = (
+export const handleChatPrompt = async (
     mynahUi: MynahUI,
     tabId: string,
     prompt: ChatPrompt,
@@ -104,7 +104,24 @@ export const handleChatPrompt = (
     agenticMode?: boolean
 ) => {
     let userPrompt = prompt.escapedPrompt
-    messager.onStopChatResponse(tabId)
+
+    const tabStore = mynahUi.getTabData(tabId)?.getStore()
+    const isLoading = tabStore?.loadingChat === true
+    if (isLoading) {
+        messager.onStopChatResponse(tabId)
+        mynahUi.addChatItem(tabId, {
+            type: ChatItemType.DIRECTIVE,
+            messageId: `stopped-${Date.now()}`,
+        })
+        mynahUi.updateStore(tabId, {
+            loadingChat: false,
+            cancelButtonWhenLoading: true,
+            promptInputDisabledState: false,
+        })
+
+        await new Promise(resolve => setTimeout(resolve, 50))
+    }
+
     if (prompt.command) {
         // Temporary solution to handle clear quick actions on the client side
         if (prompt.command === '/clear') {