fix(amazonq): adding truncation logic to fileText for inline suggestions (#2589)

* fix(amazonq): adding truncation logic to filetext for inline suggestions

* Fixing tests

Author: laileni-aws

server/aws-lsp-codewhisperer/src/language-server/inline-completion/codeWhispererServer.test.ts

@@ -916,6 +916,16 @@ describe('CodeWhisperer Server', () => {
                         { content: 'class Foo', filePath: 'Foo.java' },
                         { content: 'class Bar', filePath: 'Bar.java' },
                     ],
+                    editorState: {
+                        document: {
+                            relativeFilePath: 'file:///TargetFile.java',
+                            programmingLanguage: { languageName: 'java' },
+                            text: '',
+                        },
+                        cursorState: {
+                            position: { line: 0, character: 0 },
+                        },
+                    },
                     // workspaceId: undefined,
                 }
                 sinon.assert.calledOnceWithExactly(test_service.generateSuggestions, expectedGenerateSuggestionsRequest)

server/aws-lsp-codewhisperer/src/language-server/inline-completion/contants/constants.ts

@@ -1,6 +1,7 @@
 export const FILE_URI_CHARS_LIMIT = 1024
 export const FILENAME_CHARS_LIMIT = 1024
 export const CONTEXT_CHARACTERS_LIMIT = 10240
+export const EDITOR_STATE_MAX_LENGTH = 40000
 export const EMPTY_RESULT = { sessionId: '', items: [] }
 export const EDIT_DEBOUNCE_INTERVAL_MS = 500
 // ABAP ADT extensions commonly used with Eclipse

server/aws-lsp-codewhisperer/src/language-server/inline-completion/handler/inlineCompletionHandler.ts

@@ -45,7 +45,7 @@ import {
     emitServiceInvocationTelemetry,
     emitUserTriggerDecisionTelemetry,
 } from '../telemetry/telemetry'
-import { EMPTY_RESULT } from '../contants/constants'
+import { EDITOR_STATE_MAX_LENGTH, EMPTY_RESULT } from '../contants/constants'
 import { IdleWorkspaceManager } from '../../workspaceContext/IdleWorkspaceManager'
 import { mergeSuggestionsWithRightContext } from '../utils/mergeRightUtils'
 import { getTextDocument } from '../utils/textDocumentUtils'
@@ -331,9 +331,36 @@ export class InlineCompletionHandler {
 
         if (codeWhispererService instanceof CodeWhispererServiceToken) {
             const tokenRequest = requestContext as GenerateTokenSuggestionsRequest
+
+            // Build editorState with truncation for token-based requests
+            const documentText = textDocument.getText()
+            const cursorOffset = textDocument.offsetAt(params.position)
+            let fileText = documentText
+            if (documentText.length > EDITOR_STATE_MAX_LENGTH) {
+                const halfLength = Math.floor(EDITOR_STATE_MAX_LENGTH / 2)
+                const leftPart = documentText.substring(Math.max(0, cursorOffset - halfLength), cursorOffset)
+                const rightPart = documentText.substring(cursorOffset, cursorOffset + halfLength)
+                fileText = leftPart + rightPart
+            }
+
             generateCompletionReq = {
                 ...tokenRequest,
                 ...(workspaceId ? { workspaceId } : {}),
+                editorState: {
+                    document: {
+                        relativeFilePath: textDocument.uri,
+                        programmingLanguage: {
+                            languageName: fileContext.programmingLanguage.languageName,
+                        },
+                        text: fileText,
+                    },
+                    cursorState: {
+                        position: {
+                            line: params.position.line,
+                            character: params.position.character,
+                        },
+                    },
+                },
             }
         } else {
             const iamRequest = requestContext as GenerateIAMSuggestionsRequest
@@ -344,7 +371,9 @@ export class InlineCompletionHandler {
 
         try {
             const authType = codeWhispererService instanceof CodeWhispererServiceToken ? 'token' : 'iam'
-            this.logging.debug(`[INLINE_COMPLETION] API call - generateSuggestions (new session, ${authType})`)
+            this.logging.debug(
+                `[INLINE_COMPLETION] API call - generateSuggestions (new session, ${authType}), generateCompletion request = ${JSON.stringify(generateCompletionReq)}`
+            )
             const suggestionResponse = await codeWhispererService.generateSuggestions(generateCompletionReq)
             return await this.processSuggestionResponse(suggestionResponse, newSession, true, selectionRange)
         } catch (error) {