fix(autoedit): fix suffix matching logic

sourcegraph · Nov 21, 2024 · 325e5cc · 325e5cc
1 parent 3c353d9
commit 325e5cc
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/vscode/src/autoedits/autoedits-provider.ts b/vscode/src/autoedits/autoedits-provider.ts
@@ -212,10 +212,7 @@ export class AutoeditsProvider implements vscode.InlineCompletionItemProvider, v
         const isPrefixMatch = prediction.startsWith(codeToReplace.codeToRewritePrefix)
         const isSuffixMatch =
             // The current line suffix should not require any char removals to render the completion.
-            completionMatchesSuffix(
-                { insertText: codeToReplace.codeToRewriteSuffix },
-                docContext.currentLineSuffix
-            ) &&
+            completionMatchesSuffix({ insertText: prediction }, docContext.currentLineSuffix) &&
             // The new lines suggested after the current line must be equal to the prediction.
             prediction.endsWith(codeToRewriteAfterCurrentLine)