src/extension.ts

@@ -113,8 +113,13 @@ async function autocompleteCommand(textEditor: vscode.TextEditor, cancellationTo
 
                 // Create a decoration for each line of the response
                 const previewRanges = lines.map((line: string, idx: number) => {
-                    const linePos = new vscode.Position(position.line + idx, 0);
-                    const range = new vscode.Range(linePos, linePos); // Set range at the start of each new line
+                    // Determine the start and end positions for each line
+                    const startPos = new vscode.Position(position.line + idx, 0);
+                    const endPos = new vscode.Position(position.line + idx, line.length);
+                    
+                    // Create a range covering the whole line
+                    const range = new vscode.Range(startPos, endPos); 
+                    
                     return {
                         range,
                         renderOptions: {