feat: trigger inline suggestion via acceptance for EDITS

Author: atonaamz

packages/amazonq/src/app/inline/EditRendering/displayImage.ts

@@ -12,6 +12,7 @@ import { LogInlineCompletionSessionResultsParams } from '@aws/language-server-ru
 import { InlineCompletionItemWithReferences } from '@aws/language-server-runtimes/protocol'
 import path from 'path'
 import { imageVerticalOffset } from './svgGenerator'
+import { AmazonQInlineCompletionItemProvider } from '../completion'
 
 export class EditDecorationManager {
     private imageDecorationType: vscode.TextEditorDecorationType
@@ -279,15 +280,16 @@ export async function displaySvgDecoration(
     originalCodeHighlightRanges: Array<{ line: number; start: number; end: number }>,
     session: CodeWhispererSession,
     languageClient: LanguageClient,
-    item: InlineCompletionItemWithReferences
+    item: InlineCompletionItemWithReferences,
+    inlineCompletionProvider?: AmazonQInlineCompletionItemProvider
 ) {
     const originalCode = editor.document.getText()
 
     decorationManager.displayEditSuggestion(
         editor,
         svgImage,
         startLine,
-        () => {
+        async () => {
             // Handle accept
             getLogger().info('Edit suggestion accepted')
 
@@ -315,6 +317,18 @@ export async function displaySvgDecoration(
                 firstCompletionDisplayLatency: session.firstCompletionDisplayLatency,
             }
             languageClient.sendNotification('aws/logInlineCompletionSessionResults', params)
+            if (inlineCompletionProvider) {
+                await inlineCompletionProvider.provideInlineCompletionItems(
+                    editor.document,
+                    endPosition,
+                    {
+                        triggerKind: vscode.InlineCompletionTriggerKind.Automatic,
+                        selectedCompletionInfo: undefined,
+                    },
+                    new vscode.CancellationTokenSource().token,
+                    { emitTelemetry: false, showUi: false }
+                )
+            }
         },
         () => {
             // Handle reject

packages/amazonq/src/app/inline/EditRendering/imageRenderer.ts

@@ -10,12 +10,14 @@ import { getLogger } from 'aws-core-vscode/shared'
 import { LanguageClient } from 'vscode-languageclient'
 import { InlineCompletionItemWithReferences } from '@aws/language-server-runtimes/protocol'
 import { CodeWhispererSession } from '../sessionManager'
+import { AmazonQInlineCompletionItemProvider } from '../completion'
 
 export async function showEdits(
     item: InlineCompletionItemWithReferences,
     editor: vscode.TextEditor | undefined,
     session: CodeWhispererSession,
-    languageClient: LanguageClient
+    languageClient: LanguageClient,
+    inlineCompletionProvider?: AmazonQInlineCompletionItemProvider
 ) {
     if (!editor) {
         return
@@ -37,7 +39,8 @@ export async function showEdits(
                 origionalCodeHighlightRange,
                 session,
                 languageClient,
-                item
+                item,
+                inlineCompletionProvider
             )
         } else {
             getLogger('nextEditPrediction').error('SVG image generation returned an empty result.')

packages/amazonq/src/app/inline/completion.ts

@@ -354,7 +354,7 @@ ${itemLog}
                 if (item.isInlineEdit) {
                     // Check if Next Edit Prediction feature flag is enabled
                     if (Experiments.instance.isExperimentEnabled('amazonqLSPNEP')) {
-                        void showEdits(item, editor, session, this.languageClient).then(() => {
+                        void showEdits(item, editor, session, this.languageClient, this).then(() => {
                             const t3 = performance.now()
                             logstr = logstr + `- duration since trigger to NEP suggestion is displayed: ${t3 - t0}ms`
                             this.logger.info(logstr)