fix(chat): request fails if character limit exceeded

Author: ctlai95

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

@@ -88,7 +88,7 @@ import {
 } from '../../constants'
 import { ChatSession } from '../../clients/chat/v0/chat'
 import { amazonQTabSuffix } from '../../../shared/constants'
-import { OutputKind } from '../../tools/toolShared'
+import { maxToolOutputCharacterLength, OutputKind } from '../../tools/toolShared'
 import { ToolUtils, Tool, ToolType } from '../../tools/toolUtils'
 import { ChatStream } from '../../tools/chatStream'
 import { ChatHistoryStorage } from '../../storages/chatHistoryStorage'
@@ -667,6 +667,11 @@ export class ChatController {
                             requiresAcceptance: false,
                         })
                         const output = await ToolUtils.invoke(tool, chatStream)
+                        if (output.output.content.length > maxToolOutputCharacterLength) {
+                            throw Error(
+                                `Tool output exceeds maximum character limit of ${maxToolOutputCharacterLength}`
+                            )
+                        }
 
                         toolResults.push({
                             content: [

packages/core/src/codewhispererChat/tools/toolShared.ts

@@ -7,6 +7,7 @@ import path from 'path'
 import fs from '../../shared/fs/fs'
 
 export const maxToolResponseSize = 30720 // 30KB
+export const maxToolOutputCharacterLength = 800_000
 
 export enum OutputKind {
     Text = 'text',