@@ -54,7 +54,7 @@ import { Disposable, LanguageClient, Position, TextDocumentIdentifier } from 'vs
5454import * as jose from 'jose'
5555import { AmazonQChatViewProvider } from './webviewProvider'
5656import { AuthUtil } from 'aws-core-vscode/codewhisperer'
57- import { AmazonQPromptSettings , messages , openUrl } from 'aws-core-vscode/shared'
57+ import { amazonQDiffScheme , AmazonQPromptSettings , messages , openUrl } from 'aws-core-vscode/shared'
5858import { DefaultAmazonQAppInitContext , messageDispatcher , EditorContentController } from 'aws-core-vscode/amazonq'
5959
6060export function registerLanguageServerEventListener ( languageClient : LanguageClient , provider : AmazonQChatViewProvider ) {
@@ -427,13 +427,17 @@ export function registerMessageListeners(
427427 new vscode . Position ( 0 , 0 ) ,
428428 new vscode . Position ( doc . lineCount - 1 , doc . lineAt ( doc . lineCount - 1 ) . text . length )
429429 )
430- await edc . viewDiff ( {
431- context : {
432- activeFileContext : { filePath : params . originalFileUri } ,
433- focusAreaContext : { selectionInsideExtendedCodeBlock : entireDocumentSelection } ,
430+ await edc . viewDiff (
431+ {
432+ context : {
433+ activeFileContext : { filePath : params . originalFileUri } ,
434+ focusAreaContext : { selectionInsideExtendedCodeBlock : entireDocumentSelection } ,
435+ } ,
436+ code : params . fileContent ,
434437 } ,
435- code : params . fileContent ,
436- } )
438+ amazonQDiffScheme ,
439+ true
440+ )
437441 } )
438442}
439443
0 commit comments