Skip to content

Commit dc94746

Browse files
committed
refactor the constant into constant.ts
1 parent 9bf6fa7 commit dc94746

File tree

2 files changed

+8
-5
lines changed

2 files changed

+8
-5
lines changed

packages/core/src/codewhisperer/models/constants.ts

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -87,6 +87,9 @@ export const lineBreakWin = '\r\n'
8787
export const supplementalContextTimeoutInMs = 100
8888

8989
export const supplementalContextMaxTotalLength = 20480
90+
91+
export const editorStateMaxLength = 40000
92+
9093
/**
9194
* Ux of recommendations
9295
*/

packages/core/src/codewhisperer/util/editorContext.ts

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@ import { getTabSizeSetting } from '../../shared/utilities/editorUtilities'
1111
import { getLogger } from '../../shared/logger/logger'
1212
import { runtimeLanguageContext } from './runtimeLanguageContext'
1313
import { fetchSupplementalContext } from './supplementalContext/supplementalContextUtil'
14-
import { supplementalContextTimeoutInMs } from '../models/constants'
14+
import { editorStateMaxLength, supplementalContextTimeoutInMs } from '../models/constants'
1515
import { getSelectedCustomization } from './customizationUtil'
1616
import { selectFrom } from '../../shared/utilities/tsUtils'
1717
import { checkLeftContextKeywordsForJson } from './commonUtil'
@@ -220,11 +220,11 @@ export function getEditorState(editor: vscode.TextEditor, fileContext: codewhisp
220220
const cursorOffset = editor.document.offsetAt(cursorPosition)
221221
const documentText = editor.document.getText()
222222

223-
// Check if text needs truncation (longer than 40000 characters)
223+
// Truncate if text needs truncation (longer than 40000 characters)
224224
let fileText = documentText
225-
if (documentText.length > 40000) {
226-
const startOffset = Math.max(0, cursorOffset - 20000)
227-
const endOffset = Math.min(documentText.length, cursorOffset + 20000)
225+
if (documentText.length > editorStateMaxLength) {
226+
const startOffset = Math.max(0, cursorOffset - Math.floor(editorStateMaxLength / 2))
227+
const endOffset = Math.min(documentText.length, cursorOffset + Math.floor(editorStateMaxLength / 2))
228228

229229
fileText = documentText.substring(startOffset, endOffset)
230230
}

0 commit comments

Comments
 (0)