call LLM to get response of shell command rejection

Author: yueny2020

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

@@ -91,7 +91,7 @@ import {
     workspaceChunkMaxSize,
     defaultContextLengths,
 } from '../../constants'
-import { ChatSession } from '../../clients/chat/v0/chat'
+import { ChatSession, ToolUseWithError } from '../../clients/chat/v0/chat'
 import { amazonQTabSuffix } from '../../../shared/constants'
 import { OutputKind } from '../../tools/toolShared'
 import { ToolUtils, Tool, ToolType } from '../../tools/toolUtils'
@@ -806,6 +806,17 @@ export class ChatController {
             type: 'chat_message',
             context: undefined,
         })
+
+        const toolUseWithError: ToolUseWithError = {
+            toolUse: {
+                name: ToolType.ExecuteBash,
+                input: message.action.text,
+                toolUseId: message.action.formItemValues?.toolUseId,
+            },
+            error: new Error('Tool use was rejected by the user.'),
+        }
+        const session = this.sessionStorage.getSession(message.tabID!)
+        session.setToolUseWithError(toolUseWithError)
         await this.generateStaticTextResponse('reject-shell-command', triggerId)
     }
 
@@ -827,6 +838,7 @@ export class ChatController {
                 break
             case 'reject-shell-command':
                 await this.rejectShellCommand(message)
+                await this.processToolUseMessage(message)
                 break
             default:
                 getLogger().warn(`Unhandled action: ${message.action.id}`)