chore: supplemental context log (#2185)

Author: Will-ShaoHua

server/aws-lsp-codewhisperer/src/shared/codeWhispererService.ts

@@ -444,6 +444,8 @@ export class CodeWhispererServiceToken extends CodeWhispererServiceBase {
     "endpoint": ${this.codeWhispererEndpoint},
     "predictionType": ${request.predictionTypes?.toString() ?? 'Not specified (COMPLETIONS)'},
     "filename": ${request.fileContext.filename},
+    "leftContextLength": ${request.fileContext.leftFileContent.length},
+    rightContextLength: ${request.fileContext.rightFileContent.length},
     "language": ${request.fileContext.programmingLanguage.languageName},
     "supplementalContextCount": ${request.supplementalContexts?.length ?? 0},
     "request.nextToken": ${request.nextToken},

server/aws-lsp-codewhisperer/src/shared/supplementalContextUtil/crossFileContextUtil.ts

@@ -71,12 +71,13 @@ export async function fetchSupplementalContextForSrc(
     const supplementalContextConfig = getSupplementalContextConfig(document.languageId)
 
     if (supplementalContextConfig === undefined) {
-        return supplementalContextConfig
+        return undefined
     }
-    //TODO: add logic for other strategies once available
+
     if (supplementalContextConfig === 'codemap') {
         return await codemapContext(document, position, workspace, cancellationToken, openTabFiles)
     }
+
     return { supplementalContextItems: [], strategy: 'Empty' }
 }
 
@@ -264,10 +265,7 @@ function getInputChunk(document: TextDocument, cursorPosition: Position, chunkSi
  * @returns specifically returning undefined if the langueage is not supported,
  * otherwise true/false depending on if the language is fully supported or not belonging to the user group
  */
-function getSupplementalContextConfig(
-    languageId: TextDocument['languageId'],
-    _userGroup?: any
-): SupplementalContextStrategy | undefined {
+function getSupplementalContextConfig(languageId: TextDocument['languageId']): SupplementalContextStrategy | undefined {
     return isCrossFileSupported(languageId) ? 'codemap' : undefined
 }
 

server/aws-lsp-codewhisperer/src/shared/supplementalContextUtil/supplementalContextUtil.ts

@@ -91,7 +91,24 @@ export async function fetchSupplementalContext(
                 strategy: supplementalContextValue.strategy,
             }
 
-            return truncateSupplementalContext(resBeforeTruncation)
+            const r = truncateSupplementalContext(resBeforeTruncation)
+
+            let logstr = `@@supplemental context@@
+\tisUtg: ${r.isUtg},
+\tisProcessTimeout: ${r.isProcessTimeout},
+\tcontents.length: ${r.contentsLength},
+\tlatency: ${r.latency},
+\tstrategy: ${r.strategy},
+`
+            r.supplementalContextItems.forEach((item, index) => {
+                logstr += `\tChunk [${index}th]:\n`
+                logstr += `\t\tPath: ${item.filePath}\n`
+                logstr += `\t\tLength: ${item.content.length}\n`
+                logstr += `\t\tScore: ${item.score}\n`
+            })
+            logging.info(logstr)
+
+            return r
         } else {
             return undefined
         }