fix(chat): Remove enforceMaxSize from fsRead

Author: jguoamz

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

@@ -93,7 +93,7 @@ import {
 } from '../../constants'
 import { ChatSession } from '../../clients/chat/v0/chat'
 import { amazonQTabSuffix } from '../../../shared/constants'
-import { maxToolOutputCharacterLength, OutputKind } from '../../tools/toolShared'
+import { maxToolResponseSize, OutputKind } from '../../tools/toolShared'
 import { ToolUtils, Tool, ToolType } from '../../tools/toolUtils'
 import { ChatStream } from '../../tools/chatStream'
 import { ChatHistoryStorage } from '../../storages/chatHistoryStorage'
@@ -723,10 +723,8 @@ 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}`
-                                )
+                            if (output.output.content.length > maxToolResponseSize) {
+                                throw Error(`Tool output exceeds maximum character limit of ${maxToolResponseSize}`)
                             }
 
                             toolResults.push({

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

@@ -5,7 +5,7 @@
 import * as vscode from 'vscode'
 import { getLogger } from '../../shared/logger/logger'
 import fs from '../../shared/fs/fs'
-import { InvokeOutput, maxToolResponseSize, OutputKind, sanitizePath } from './toolShared'
+import { InvokeOutput, OutputKind, sanitizePath } from './toolShared'
 import { Writable } from 'stream'
 import path from 'path'
 
@@ -89,7 +89,7 @@ export class FsRead {
     private handleFileRange(fullText: string): InvokeOutput {
         if (!this.readRange || this.readRange.length === 0) {
             this.logger.info('No range provided. returning entire file.')
-            return this.createOutput(this.enforceMaxSize(fullText))
+            return this.createOutput(fullText)
         }
 
         const lines = fullText.split('\n')
@@ -101,7 +101,7 @@ export class FsRead {
 
         this.logger.info(`Reading file: ${this.fsPath}, lines ${start + 1}-${end + 1}`)
         const slice = lines.slice(start, end + 1).join('\n')
-        return this.createOutput(this.enforceMaxSize(slice))
+        return this.createOutput(slice)
     }
 
     private parseLineRange(lineCount: number, range: number[]): [number, number] {
@@ -121,17 +121,6 @@ export class FsRead {
         return [finalStart, finalEnd]
     }
 
-    private enforceMaxSize(content: string): string {
-        const byteCount = Buffer.byteLength(content, 'utf8')
-        if (byteCount > maxToolResponseSize) {
-            throw new Error(
-                `This tool only supports reading ${maxToolResponseSize} bytes at a time.
-                You tried to read ${byteCount} bytes. Try executing with fewer lines specified.`
-            )
-        }
-        return content
-    }
-
     private createOutput(content: string): InvokeOutput {
         return {
             output: {

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

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

packages/core/src/test/codewhispererChat/tools/fsRead.test.ts

@@ -54,8 +54,8 @@ describe('FsRead Tool', () => {
         )
     })
 
-    it('throws error if content exceeds 30KB', async function () {
-        const bigContent = 'x'.repeat(35_000)
+    it('throws error if content exceeds 800KB', async function () {
+        const bigContent = 'x'.repeat(800_001)
         const bigFilePath = await testFolder.write('bigFile.txt', bigContent)
 
         const fsRead = new FsRead({ path: bigFilePath })