feat(chat-client): open use input prompt for agentic chat and new prompt should st… (aws#1081)

feat: open use input prompt for agentic chat and new prompt should stop current response

Author: ashishrp-aws

chat-client/src/client/mynahUi.test.ts

@@ -98,7 +98,7 @@ describe('MynahUI', () => {
             assert.notCalled(onQuickActionSpy)
             assert.calledWith(onChatPromptSpy, { prompt, tabId, context: undefined })
             assert.calledWith(addChatItemSpy, tabId, { type: ChatItemType.PROMPT, body: prompt.escapedPrompt })
-            assert.calledWith(updateStoreSpy, tabId, { loadingChat: true, promptInputDisabledState: true })
+            assert.calledWith(updateStoreSpy, tabId, { loadingChat: true, promptInputDisabledState: false })
             assert.calledWith(addChatItemSpy, tabId, { type: ChatItemType.ANSWER_STREAM })
         })
 
@@ -127,7 +127,7 @@ describe('MynahUI', () => {
                 tabId,
             })
             assert.calledOnce(updateStoreSpy)
-            assert.calledWith(updateStoreSpy, tabId, { loadingChat: true, promptInputDisabledState: true })
+            assert.calledWith(updateStoreSpy, tabId, { loadingChat: true, promptInputDisabledState: false })
         })
     })
 
@@ -282,7 +282,7 @@ describe('MynahUI', () => {
 
             sinon.assert.calledOnceWithMatch(updateStoreSpy, tabId, {
                 loadingChat: true,
-                promptInputDisabledState: true,
+                promptInputDisabledState: false,
             })
         })
     })

chat-client/src/client/mynahUi.ts

@@ -90,6 +90,7 @@ export const handleChatPrompt = (
     _eventId?: string
 ) => {
     let userPrompt = prompt.escapedPrompt
+    messager.onStopChatResponse(tabId)
     if (prompt.command) {
         // Temporary solution to handle clear quick actions on the client side
         if (prompt.command === '/clear') {
@@ -124,7 +125,7 @@ export const handleChatPrompt = (
     // Set UI to loading state
     mynahUi.updateStore(tabId, {
         loadingChat: true,
-        promptInputDisabledState: true,
+        promptInputDisabledState: false,
     })
 
     // Create initial empty response