fix(amazonq): should emit UTDE telemetry before applying the Edits content

Author: atonaamz

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

@@ -354,23 +354,6 @@ export async function displaySvgDecoration(
             // Handle accept
             getLogger().info('Edit suggestion accepted')
 
-            // Replace content
-            try {
-                vsCodeState.isCodeWhispererEditing = true
-                await replaceEditorContent(editor, newCode)
-            } finally {
-                vsCodeState.isCodeWhispererEditing = false
-            }
-
-            // Move cursor to end of the actual changed content
-            const endPosition = getEndOfEditPosition(originalCode, newCode)
-            editor.selection = new vscode.Selection(endPosition, endPosition)
-
-            // Move cursor to end of the actual changed content
-            editor.selection = new vscode.Selection(endPosition, endPosition)
-
-            await decorationManager.clearDecorations(editor)
-            documentChangeListener.dispose()
             const params: LogInlineCompletionSessionResultsParams = {
                 sessionId: session.sessionId,
                 completionSessionResult: {
@@ -386,19 +369,21 @@ export async function displaySvgDecoration(
             }
             languageClient.sendNotification('aws/logInlineCompletionSessionResults', params)
             session.triggerOnAcceptance = true
-            // VS Code triggers suggestion on every keystroke, temporarily disable trigger on acceptance
-            // if (inlineCompletionProvider && session.editsStreakPartialResultToken) {
-            //     await inlineCompletionProvider.provideInlineCompletionItems(
-            //         editor.document,
-            //         endPosition,
-            //         {
-            //             triggerKind: vscode.InlineCompletionTriggerKind.Automatic,
-            //             selectedCompletionInfo: undefined,
-            //         },
-            //         new vscode.CancellationTokenSource().token,
-            //         { emitTelemetry: false, showUi: false, editsStreakToken: session.editsStreakPartialResultToken }
-            //     )
-            // }
+
+            // Replace content
+            try {
+                vsCodeState.isCodeWhispererEditing = true
+                await replaceEditorContent(editor, newCode)
+            } finally {
+                vsCodeState.isCodeWhispererEditing = false
+            }
+
+            // Move cursor to end of the actual changed content
+            const endPosition = getEndOfEditPosition(originalCode, newCode)
+            editor.selection = new vscode.Selection(endPosition, endPosition)
+
+            await decorationManager.clearDecorations(editor)
+            documentChangeListener.dispose()
         },
         async () => {
             // Handle reject