setup commit

Author: ashishrp-aws

packages/core/src/amazonq/webview/ui/followUps/handler.ts

@@ -48,7 +48,7 @@ export class FollowUpInteractionHandler {
             this.mynahUI.updateStore(tabID, {
                 loadingChat: true,
                 cancelButtonWhenLoading: false,
-                promptInputDisabledState: true,
+                promptInputDisabledState: false,
             })
             this.mynahUI.addChatItem(tabID, {
                 type: ChatItemType.PROMPT,

packages/core/src/amazonq/webview/ui/main.ts

@@ -76,7 +76,7 @@ export const createMynahUI = (
                 body: 'This conversation has timed out after 48 hours. It will not be saved. Start a new conversation.',
             })
             mynahUI.updateStore(tabID, {
-                promptInputDisabledState: true,
+                promptInputDisabledState: false,
                 promptInputPlaceholder: 'Session ended.',
             })
         },
@@ -422,7 +422,7 @@ export const createMynahUI = (
                 mynahUI.updateStore(tabID, {
                     loadingChat: true,
                     cancelButtonWhenLoading: false,
-                    promptInputDisabledState: true,
+                    promptInputDisabledState: false,
                 })
 
                 tabsStorage.updateTabStatus(tabID, 'busy')
@@ -655,9 +655,10 @@ export const createMynahUI = (
         onTabChange: connector.onTabChange,
         // TODO: update mynah-ui this type doesn't seem correct https://github.com/aws/mynah-ui/blob/3777a39eb534a91fd6b99d6cf421ce78ee5c7526/src/main.ts#L372
         onStopChatResponse: (tabID: string) => {
+            console.log('stop button clicked')
             mynahUI.updateStore(tabID, {
                 loadingChat: false,
-                promptInputDisabledState: false,
+                promptInputDisabledState: true,
             })
             connector.onStopChatResponse(tabID)
         },

packages/core/src/amazonq/webview/ui/messages/controller.ts

@@ -121,7 +121,7 @@ export class MessageController {
             this.mynahUI.updateStore(newTabID, {
                 loadingChat: true,
                 cancelButtonWhenLoading: false,
-                promptInputDisabledState: true,
+                promptInputDisabledState: false,
             })
 
             // We have race condition here with onTabAdd Ui event. This way we need to update store twice to be sure

packages/core/src/amazonq/webview/ui/messages/handler.ts

@@ -36,8 +36,8 @@ export class TextMessageHandler {
 
         this.mynahUI.updateStore(tabID, {
             loadingChat: true,
-            cancelButtonWhenLoading: false,
-            promptInputDisabledState: true,
+            cancelButtonWhenLoading: true,
+            promptInputDisabledState: false,
         })
 
         this.tabsStorage.updateTabStatus(tabID, 'busy')

packages/core/src/amazonq/webview/ui/tabs/generator.ts

@@ -3,7 +3,7 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 
-import { ChatItemType, MynahUIDataModel, QuickActionCommandGroup } from '@aws/mynah-ui'
+import { ChatItemType, MynahIcons, MynahUIDataModel, QuickActionCommandGroup } from '@aws/mynah-ui'
 import { TabType } from '../storages/tabsStorage'
 import { FollowUpGenerator } from '../followUps/generator'
 import { QuickActionGenerator } from '../quickActions/generator'
@@ -72,6 +72,19 @@ export class TabDataGenerator {
                       },
                   ]
                 : [],
+            promptInputOptions: [
+                {
+                    type: 'radiogroup',
+                    id: 'stop-response',
+                    value: 'stop-response',
+                    options: [
+                        {
+                            value: 'stop-response',
+                            icon: MynahIcons.STOP,
+                        },
+                    ],
+                },
+            ],
         }
         return tabData
     }

packages/core/src/codewhisperer/client/codewhisperer.ts

@@ -31,8 +31,8 @@ export interface CodeWhispererConfig {
 }
 
 export const defaultServiceConfig: CodeWhispererConfig = {
-    region: 'us-east-1',
-    endpoint: 'https://codewhisperer.us-east-1.amazonaws.com/',
+    region: 'us-west-2',
+    endpoint: 'https://rts.alpha-us-west-2.codewhisperer.ai.aws.dev/',
 }
 
 export function getCodewhispererConfig(): CodeWhispererConfig {

packages/core/src/codewhispererChat/controllers/chat/controller.ts

@@ -367,6 +367,7 @@ export class ChatController {
 
     private async processStopResponseMessage(message: StopResponseMessage) {
         const session = this.sessionStorage.getSession(message.tabID)
+        console.log('stop process has been triggered')
         session.tokenSource.cancel()
         this.chatHistoryStorage.getTabHistory(message.tabID).clearRecentHistory()
     }
@@ -656,6 +657,7 @@ export class ChatController {
                     return
                 }
                 session.setToolUse(undefined)
+                console.log('session undefined in controller', toolUse)
 
                 const toolResults: ToolResult[] = []
 

packages/core/src/codewhispererChat/view/messages/messageListener.ts

@@ -79,6 +79,7 @@ export class UIMessageListener {
                 this.processTriggerTabIDReceived(msg)
                 break
             case 'stop-response':
+                console.log('this stop-response command  in lsitenrer has been called')
                 this.stopResponse(msg)
                 break
             case 'chat-item-voted':