fix(codewhisperer): fix extractContextForCodeWhisperer tests

Author: Jiatong Li

packages/amazonq/test/unit/codewhisperer/util/editorContext.test.ts

@@ -56,6 +56,7 @@ describe('editorContext', function () {
             const editor = createMockTextEditor('import math\ndef two_sum(nums, target):\n', 'test.py', 'python', 1, 17)
             const actual = EditorContext.extractContextForCodeWhisperer(editor)
             const expected: codewhispererClient.FileContext = {
+                fileUri: 'file:///test.py',
                 filename: 'test.py',
                 programmingLanguage: {
                     languageName: 'python',
@@ -76,6 +77,7 @@ describe('editorContext', function () {
             )
             const actual = EditorContext.extractContextForCodeWhisperer(editor)
             const expected: codewhispererClient.FileContext = {
+                fileUri: 'file:///test.py',
                 filename: 'test.py',
                 programmingLanguage: {
                     languageName: 'python',
@@ -112,6 +114,7 @@ describe('editorContext', function () {
 
             const actual = EditorContext.extractContextForCodeWhisperer(editor)
             const expected: codewhispererClient.FileContext = {
+                fileUri: 'vscode-notebook-cell:Untitled-1.ipynb',
                 filename: 'Untitled-1.py',
                 programmingLanguage: {
                     languageName: 'python',

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

@@ -167,7 +167,7 @@ export function extractContextForCodeWhisperer(editor: vscode.TextEditor): codew
     }
 
     return {
-        fileUri: editor.document.uri.toString().substring(0, CodeWhispererConstants.filenameCharsLimit),
+        fileUri: editor.document.uri.toString().split('?')[0].substring(0, CodeWhispererConstants.filenameCharsLimit),
         filename: getFileRelativePath(editor),
         programmingLanguage: {
             languageName: languageName,