Compare commits
41 Commits
Author | SHA1 | Date |
---|---|---|
Falko Victor Habel | d1ccb89544 | |
Falko Victor Habel | c0af28ee1e | |
Falko Victor Habel | 89b179bbe3 | |
Falko Victor Habel | a7292cdcea | |
Falko Victor Habel | 2233f4ec63 | |
Falko Victor Habel | 12f411fbac | |
Falko Victor Habel | a3bf2f93bb | |
Falko Victor Habel | c1d0a53720 | |
Falko Victor Habel | e02296c5a7 | |
Falko Victor Habel | 621db41722 | |
Falko Victor Habel | e234969638 | |
Falko Victor Habel | 439a538c7e | |
Falko Victor Habel | a56d47747d | |
Falko Victor Habel | 96a3971b71 | |
Falko Victor Habel | 424cc124b2 | |
Falko Victor Habel | dcb5a3bbdf | |
Falko Victor Habel | f6ea8494a7 | |
Falko Victor Habel | f122d99ba1 | |
Falko Victor Habel | 804a3113e4 | |
Falko Victor Habel | 4c5eb334df | |
Falko Victor Habel | d8e97c2b4e | |
Falko Victor Habel | b6241855fc | |
Falko Victor Habel | ccd3528325 | |
Falko Victor Habel | 77e328fcee | |
Falko Victor Habel | 6a953b7a12 | |
Falko Victor Habel | ffd22f7cc6 | |
Falko Victor Habel | 0416896254 | |
Falko Victor Habel | 9109c199e6 | |
Falko Victor Habel | 8e72d08d53 | |
Falko Victor Habel | 2a46e061d3 | |
Falko Victor Habel | 27571fcad8 | |
Falko Victor Habel | 66c88a053b | |
Falko Victor Habel | ac7afb4b4e | |
Falko Victor Habel | fb6e9a5d1f | |
Falko Victor Habel | 678c4823b3 | |
Falko Victor Habel | 77e0dbc048 | |
Falko Victor Habel | 8ac3879ee0 | |
Falko Victor Habel | 0450a222e2 | |
Falko Victor Habel | 916ea8ca4a | |
Falko Victor Habel | 58c9af0256 | |
Falko Victor Habel | 91d13d36df |
|
@ -0,0 +1 @@
|
||||||
|
vsce package --baseContentUrl https://gitea.fabelous.app/fabel/Fabelous-Autocoder/src/branch/main --baseImagesUrl https://gitea.fabelous.app/fabel/Fabelous-Autocoder/src/branch/main
|
102
README.md
102
README.md
|
@ -1,76 +1,58 @@
|
||||||
# Fabelous Autocoder
|
# Fabelous Autocoder
|
||||||
|
|
||||||
Fabelous Autocoder is a Visual Studio Code extension that provides an easy-to-use interface for Ollama autocompletion. This extension allows developers to use Ollama's powerful language models to generate code completions as they type. It is highly customizable, allowing users to configure various settings to fit their needs.
|
Fabelous Autocoder is a powerful VS Code extension that provides intelligent code completion using advanced language models. It offers seamless integration with your development workflow, allowing you to generate and preview code suggestions with ease.
|
||||||
|
|
||||||
![Fabelous Autocoder in Action](demo.gif)
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
## Features
|
## Features
|
||||||
|
|
||||||
- Autocompletion using Ollama language models
|
- **Intelligent Code Completion**: Leverages advanced language models to provide context-aware code suggestions.
|
||||||
- Customizable completion keys
|
- **Preview Functionality**: View generated code completions before accepting them.
|
||||||
- Inline preview of generated completions
|
- **Easy Accept/Decline**: Use simple keyboard shortcuts to accept or decline suggestions.
|
||||||
- Configurable maximum tokens predicted
|
- **Customizable**: Configure various parameters like API endpoint, model, and response behavior.
|
||||||
- Configurable prompt window size
|
- **Language Agnostic**: Works with multiple programming languages.
|
||||||
- Configurable response preview delay
|
|
||||||
- Configurable temperature for the model
|
|
||||||
|
|
||||||
## Installation
|
## How It Works
|
||||||
You can also download the extension from the release tab of the following Git repository:
|
|
||||||
|
|
||||||
[Fabelous-Autocoder Git Repository](https://gitea.fabelous.app/fabel/Fabelous-Autocoder.git)
|
1. Trigger the autocompletion by typing a completion key (configurable, default is space).
|
||||||
|
2. The extension sends your current code context to the configured API.
|
||||||
To do so, follow these steps:
|
3. A code completion is generated and displayed as a preview.
|
||||||
|
4. Accept the completion with `Tab` or decline it with `Backspace`.
|
||||||
1. Visit the repository link.
|
|
||||||
2. Click on the "Releases" tab.
|
|
||||||
3. Look for the latest release and click on it.
|
|
||||||
4. Download the extension file compatible with your operating system.
|
|
||||||
5. Install the extension manually in Visual Studio Code.
|
|
||||||
|
|
||||||
After installation, you'll be able to use the Fabelous Autocoder extension in your Visual Studio Code environment.
|
|
||||||
|
|
||||||
## Configuration
|
|
||||||
|
|
||||||
Fabelous Autocoder is highly customizable, allowing users to configure various settings to fit their needs. To access the configuration settings, follow these steps:
|
|
||||||
|
|
||||||
1. Open Visual Studio Code
|
|
||||||
2. Click on the Settings icon on the sidebar (or press `Ctrl+,`)
|
|
||||||
3. Search for "Fabelous Autocoder" in the search bar
|
|
||||||
4. Configure the desired settings
|
|
||||||
|
|
||||||
Here are some of the available configuration options:
|
|
||||||
|
|
||||||
- `fabelous-autocoder.endpoint`: The endpoint of the Ollama REST API
|
|
||||||
- `fabelous-autocoder.authentication`: The authentication token for Ollama
|
|
||||||
- `fabelous-autocoder.model`: The model to use for generating completions
|
|
||||||
- `fabelous-autocoder.max tokens predicted`: The maximum number of tokens generated by the model
|
|
||||||
- `fabelous-autocoder.prompt window size`: The size of the prompt in characters
|
|
||||||
- `fabelous-autocoder.completion keys`: The characters that trigger the autocompletion item provider
|
|
||||||
- `fabelous-autocoder.response preview`: Whether to show a preview of the generated completion inline
|
|
||||||
- `fabelous-autocoder.preview max tokens`: The maximum number of tokens generated for the response preview
|
|
||||||
- `fabelous-autocoder.preview delay`: The time to wait before starting inline preview generation
|
|
||||||
- `fabelous-autocoder.continue inline`: Whether to continue autocompletion after the inline preview
|
|
||||||
- `fabelous-autocoder.temperature`: The temperature of the model
|
|
||||||
- `fabelous-autocoder.keep alive`: The time in minutes before Ollama unloads the model
|
|
||||||
|
|
||||||
Note that changing the `completion keys` setting requires a reload of Visual Studio Code.
|
|
||||||
|
|
||||||
## Usage
|
## Usage
|
||||||
|
|
||||||
To use Fabelous Autocoder, simply start typing in the editor. When the configured completion keys are pressed, the extension will generate a completion using the configured Ollama model. The completion will be displayed inline with a preview of the generated code. If the `continue inline` setting is enabled, the extension will continue generating completions after the inline preview.
|
![Fabelous Autocoder Showcase](demo.gif)
|
||||||
|
|
||||||
To generate a multi-line completion, press `Enter` after the inline preview. This will open a new editor with the generated completion.
|
1. **Trigger Completion**: Type normally and hit the completion key (space by default).
|
||||||
|
2. **Preview**: The suggested completion appears in light gray text.
|
||||||
|
3. **Accept**: Press `Tab` to accept the entire suggestion.
|
||||||
|
4. **Decline**: Press `Backspace` to remove the preview and decline the suggestion.
|
||||||
|
5. **Partial Accept**: You can continue typing to partially accept the suggestion.
|
||||||
|
|
||||||
To customize the behavior of the extension, see the Configuration section above.
|
## Configuration
|
||||||
|
|
||||||
## License
|
Customize Fabelous Autocoder through VS Code settings:
|
||||||
|
|
||||||
Fabelous Autocoder is licensed under the CC BY-ND 4.0 license. See the [LICENSE](https://gitea.fabelous.app/fabel/Fabelous-Autocoder/src/branch/main/LICENSE) file for more information.
|
- `fabelous-autocoder.endpoint`: API endpoint for the language model.
|
||||||
|
- `fabelous-autocoder.model`: Specify the model to use.
|
||||||
|
- `fabelous-autocoder.temperature`: Control the randomness of completions.
|
||||||
|
- `fabelous-autocoder.max tokens predicted`: Set the maximum length of completions.
|
||||||
|
- `fabelous-autocoder.prompt window size`: Adjust the context window size.
|
||||||
|
- `fabelous-autocoder.completion keys`: Set custom completion trigger keys.
|
||||||
|
- `fabelous-autocoder.response preview`: Toggle preview functionality.
|
||||||
|
- `fabelous-autocoder.preview max tokens`: Limit preview length.
|
||||||
|
- `fabelous-autocoder.preview delay`: Add delay before showing preview.
|
||||||
|
- `fabelous-autocoder.continue inline`: Control inline continuation behavior.
|
||||||
|
|
||||||
## Acknowledgments
|
## Installation
|
||||||
|
|
||||||
Fabelous Autocoder was created by [Falko Habel](https://gitea.fabelous.app/fabel). It was inspired by the [Ollama](https://ollama.ai) project.
|
|
||||||
|
1. [Click here to download the latest version](https://gitea.fabelous.app/Fabel/Fabelous-Autocoder/releases/download/latest/fabelous-autocoder-0.2.0.vsix)
|
||||||
|
2. Open Visual Studio Code
|
||||||
|
3. Go to Extensions (Ctrl+Shift+X)
|
||||||
|
4. Click on the three dots in the upper-right corner and select "Install from VSIX..."
|
||||||
|
5. Navigate to the location where you extracted Fabelous Autocoder and select the .vsix file
|
||||||
|
6. Click "Install" to install the extension
|
||||||
|
|
||||||
|
## Requirements
|
||||||
|
|
||||||
|
- VS Code version 1.89.0 or higher
|
||||||
|
- Internet connection for API calls
|
||||||
|
|
BIN
demo.gif
BIN
demo.gif
Binary file not shown.
Before Width: | Height: | Size: 213 KiB After Width: | Height: | Size: 1.3 MiB |
29
package.json
29
package.json
|
@ -1,10 +1,10 @@
|
||||||
{
|
{
|
||||||
"name": "fabelous-autocoder",
|
"name": "fabelous-autocoder",
|
||||||
"version": "0.1.7",
|
"version": "0.2.0",
|
||||||
"displayName": "Fabelous Autocoder",
|
"displayName": "Fabelous Autocoder",
|
||||||
"description": "A simple to use Ollama autocompletion Plugin",
|
"description": "A simple to use Ollama autocompletion Plugin",
|
||||||
"icon": "icon.png",
|
"icon": "icon.png",
|
||||||
"publisher": "fabel",
|
"publisher": "Falko Habel",
|
||||||
"license": "CC BY-ND 4.0",
|
"license": "CC BY-ND 4.0",
|
||||||
"bugs": {
|
"bugs": {
|
||||||
"url": "https://gitea.fabelous.app/fabel/Fabelous-Autocoder/issues"
|
"url": "https://gitea.fabelous.app/fabel/Fabelous-Autocoder/issues"
|
||||||
|
@ -90,7 +90,6 @@
|
||||||
"type": "boolean",
|
"type": "boolean",
|
||||||
"default": true,
|
"default": true,
|
||||||
"description": "Ollama continues autocompletion after what is previewed inline. Disabling disables that feature as some may find it irritating. Multiline completion is still accessible through the shortcut even after disabling."
|
"description": "Ollama continues autocompletion after what is previewed inline. Disabling disables that feature as some may find it irritating. Multiline completion is still accessible through the shortcut even after disabling."
|
||||||
|
|
||||||
},
|
},
|
||||||
"fabelous-autocoder.temperature": {
|
"fabelous-autocoder.temperature": {
|
||||||
"type": "number",
|
"type": "number",
|
||||||
|
@ -104,14 +103,36 @@
|
||||||
},
|
},
|
||||||
"fabelous-autocoder.top p": {
|
"fabelous-autocoder.top p": {
|
||||||
"type": "number",
|
"type": "number",
|
||||||
|
"default": 1,
|
||||||
"description": "Top p sampling for the model."
|
"description": "Top p sampling for the model."
|
||||||
|
},
|
||||||
|
"fabelous-autocoder.enableLineByLineAcceptance": {
|
||||||
|
"type": "boolean",
|
||||||
|
"default": false,
|
||||||
|
"description": "Enable line-by-line acceptance of the generated code."
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
|
"keybindings": [
|
||||||
|
{
|
||||||
|
"command": "fabelous-autocoder.handleTab",
|
||||||
|
"key": "tab",
|
||||||
|
"when": "editorTextFocus && !editorTabMovesFocus"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"command": "fabelous-autocoder.handleBackspace",
|
||||||
|
"key": "backspace",
|
||||||
|
"when": "editorTextFocus"
|
||||||
|
}
|
||||||
|
],
|
||||||
"commands": [
|
"commands": [
|
||||||
{
|
{
|
||||||
"command": "fabelous-autocoder.autocomplete",
|
"command": "fabelous-autocoder.autocomplete",
|
||||||
"title": "Fabelous autocompletion"
|
"title": "Fabelous Autocompletion"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"command": "fabelous-autocoder.handleTab",
|
||||||
|
"title": "Handle Tab"
|
||||||
}
|
}
|
||||||
]
|
]
|
||||||
},
|
},
|
||||||
|
|
392
src/extension.ts
392
src/extension.ts
|
@ -1,146 +1,287 @@
|
||||||
import * as vscode from "vscode";
|
import * as vscode from 'vscode';
|
||||||
import axios from "axios";
|
import axios from 'axios';
|
||||||
|
|
||||||
let VSConfig: vscode.WorkspaceConfiguration;
|
let config: {
|
||||||
let apiEndpoint: string;
|
apiEndpoint: string;
|
||||||
let apiAuthentication: string;
|
apiAuthentication: string;
|
||||||
let apiModel: string;
|
apiModel: string;
|
||||||
let apiTemperature: number;
|
apiTemperature: number;
|
||||||
let numPredict: number;
|
numPredict: number;
|
||||||
let promptWindowSize: number;
|
promptWindowSize: number;
|
||||||
let completionKeys: string;
|
completionKeys: string[];
|
||||||
let responsePreview: boolean | undefined;
|
responsePreview: boolean;
|
||||||
let responsePreviewMaxTokens: number;
|
responsePreviewMaxTokens: number;
|
||||||
let responsePreviewDelay: number;
|
responsePreviewDelay: number;
|
||||||
let continueInline: boolean | undefined;
|
continueInline: boolean;
|
||||||
let keepAlive: number | undefined;
|
keepAlive: number;
|
||||||
let topP: number | undefined;
|
topP: number;
|
||||||
|
};
|
||||||
|
|
||||||
function updateVSConfig() {
|
let previewDecorationType: vscode.TextEditorDecorationType;
|
||||||
VSConfig = vscode.workspace.getConfiguration("fabelous-autocoder");
|
let activeCompletionManager: CompletionManager | null = null;
|
||||||
apiEndpoint = VSConfig.get("endpoint") || "http://localhost:11434/api/generate";
|
|
||||||
apiAuthentication = VSConfig.get("authentication") || "";
|
function updateConfig() {
|
||||||
apiModel = VSConfig.get("model") || "fabelous-coder:latest";
|
const vsConfig = vscode.workspace.getConfiguration('fabelous-autocoder');
|
||||||
numPredict = VSConfig.get("max tokens predicted") || 1000;
|
config = {
|
||||||
promptWindowSize = VSConfig.get("prompt window size") || 2000;
|
apiEndpoint: vsConfig.get('endpoint') || 'http://localhost:11434/api/generate',
|
||||||
completionKeys = VSConfig.get("completion keys") || " ";
|
apiAuthentication: vsConfig.get('authentication') || '',
|
||||||
responsePreview = VSConfig.get("response preview");
|
apiModel: vsConfig.get('model') || 'fabelous-coder:latest',
|
||||||
responsePreviewMaxTokens = VSConfig.get("preview max tokens") || 50;
|
apiTemperature: vsConfig.get('temperature') || 0.7,
|
||||||
responsePreviewDelay = VSConfig.get("preview delay") || 0;
|
numPredict: vsConfig.get('max tokens predicted') || 1000,
|
||||||
continueInline = VSConfig.get("continue inline");
|
promptWindowSize: vsConfig.get('prompt window size') || 2000,
|
||||||
apiTemperature = VSConfig.get("temperature") || 0.7;
|
completionKeys: (vsConfig.get('completion keys') as string || ' ').split(''),
|
||||||
keepAlive = VSConfig.get("keep alive") || 30;
|
responsePreview: vsConfig.get('response preview') || false,
|
||||||
topP = VSConfig.get("top p") || 1;
|
responsePreviewMaxTokens: vsConfig.get('preview max tokens') || 50,
|
||||||
|
responsePreviewDelay: vsConfig.get('preview delay') || 0,
|
||||||
|
continueInline: vsConfig.get('continue inline') || false,
|
||||||
|
keepAlive: vsConfig.get('keep alive') || 30,
|
||||||
|
topP: vsConfig.get('top p') || 1,
|
||||||
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
updateVSConfig();
|
function createPreviewDecorationType() {
|
||||||
vscode.workspace.onDidChangeConfiguration(updateVSConfig);
|
previewDecorationType = vscode.window.createTextEditorDecorationType({
|
||||||
|
after: {
|
||||||
|
color: '#888888',
|
||||||
|
fontStyle: 'italic',
|
||||||
|
},
|
||||||
|
textDecoration: 'none; display: none;',
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
function getContextLines(document: vscode.TextDocument, position: vscode.Position): string {
|
function getContextLines(document: vscode.TextDocument, position: vscode.Position): string {
|
||||||
const lines = [];
|
|
||||||
const startLine = Math.max(0, position.line - 1);
|
const startLine = Math.max(0, position.line - 1);
|
||||||
const endLine = position.line;
|
const endLine = position.line;
|
||||||
|
return document.getText(new vscode.Range(startLine, 0, endLine, position.character));
|
||||||
for (let i = startLine; i <= endLine; i++) {
|
|
||||||
lines.push(document.lineAt(i).text);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return lines.join("\n");
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
function createFIMPrompt(prefix: string, language: string): string {
|
function createFIMPrompt(prefix: string, language: string): string {
|
||||||
return `<fim_prefix>${prefix}<fim_middle><fim_suffix>${language}\n`;
|
return `<fim_prefix>${prefix}<fim_middle><fim_suffix>${language}\n`;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
async function generateCompletion(prompt: string, cancellationToken: vscode.CancellationToken): Promise<string> {
|
||||||
async function autocompleteCommand(textEditor: vscode.TextEditor, cancellationToken?: vscode.CancellationToken) {
|
|
||||||
const document = textEditor.document;
|
|
||||||
const position = textEditor.selection.active;
|
|
||||||
|
|
||||||
const context = getContextLines(document, position);
|
|
||||||
|
|
||||||
const fimPrompt = createFIMPrompt(context, document.languageId);
|
|
||||||
|
|
||||||
vscode.window.withProgress(
|
|
||||||
{
|
|
||||||
location: vscode.ProgressLocation.Notification,
|
|
||||||
title: "Fabelous Autocoder",
|
|
||||||
cancellable: true,
|
|
||||||
},
|
|
||||||
async (progress, progressCancellationToken) => {
|
|
||||||
try {
|
|
||||||
progress.report({ message: "Starting model..." });
|
|
||||||
|
|
||||||
let axiosCancelPost: () => void;
|
|
||||||
const axiosCancelToken = new axios.CancelToken((c) => {
|
const axiosCancelToken = new axios.CancelToken((c) => {
|
||||||
axiosCancelPost = () => {
|
cancellationToken.onCancellationRequested(() => c('Request cancelled'));
|
||||||
c("Autocompletion request terminated by user cancel");
|
|
||||||
};
|
|
||||||
if (cancellationToken) cancellationToken.onCancellationRequested(axiosCancelPost);
|
|
||||||
progressCancellationToken.onCancellationRequested(axiosCancelPost);
|
|
||||||
vscode.workspace.onDidCloseTextDocument(axiosCancelPost);
|
|
||||||
});
|
});
|
||||||
|
|
||||||
const response = await axios.post(apiEndpoint, {
|
const response = await axios.post(config.apiEndpoint, {
|
||||||
model: apiModel,
|
model: config.apiModel,
|
||||||
prompt: fimPrompt,
|
prompt: prompt,
|
||||||
stream: false,
|
stream: false,
|
||||||
raw: true,
|
raw: true,
|
||||||
options: {
|
options: {
|
||||||
num_predict: numPredict,
|
num_predict: config.numPredict,
|
||||||
temperature: apiTemperature,
|
temperature: config.apiTemperature,
|
||||||
stop: ["<fim_suffix>", "```"]
|
stop: ['<fim_suffix>', '```'],
|
||||||
|
keep_alive: config.keepAlive,
|
||||||
|
top_p: config.topP,
|
||||||
}
|
}
|
||||||
}, {
|
}, {
|
||||||
cancelToken: axiosCancelToken,
|
cancelToken: axiosCancelToken,
|
||||||
headers: {
|
headers: {
|
||||||
'Authorization': apiAuthentication
|
'Authorization': config.apiAuthentication
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
|
|
||||||
progress.report({ message: "Generating..." });
|
return response.data.response.replace(/<fim_middle>|<fim_suffix>|<fim_prefix>/g, '').trim();
|
||||||
|
}
|
||||||
|
|
||||||
let completionText = response.data.response;
|
class CompletionManager {
|
||||||
// Remove any FIM tags and leading/trailing whitespace
|
private textEditor: vscode.TextEditor;
|
||||||
completionText = completionText.replace(/<fim_middle>|<fim_suffix>|<fim_prefix>/g, '').trim();
|
private document: vscode.TextDocument;
|
||||||
|
private startPosition: vscode.Position;
|
||||||
|
private completionText: string;
|
||||||
|
private insertedLineCount: number = 0; // Track the number of inserted lines
|
||||||
|
|
||||||
// Remove the context lines
|
constructor(textEditor: vscode.TextEditor, startPosition: vscode.Position, completionText: string) {
|
||||||
const startLine = Math.max(0, position.line - 1);
|
this.textEditor = textEditor;
|
||||||
const endLine = position.line;
|
this.document = textEditor.document;
|
||||||
|
this.startPosition = startPosition;
|
||||||
|
this.completionText = completionText;
|
||||||
|
}
|
||||||
|
|
||||||
|
public async showPreview() {
|
||||||
|
if (!previewDecorationType) {
|
||||||
|
createPreviewDecorationType();
|
||||||
|
}
|
||||||
|
|
||||||
|
const completionLines = this.completionText.split('\n').length;
|
||||||
|
|
||||||
|
// Adjust the start position to line after the original start position
|
||||||
|
const adjustedStartPosition = this.startPosition.translate(0, 0);
|
||||||
|
|
||||||
|
// Step 1: Insert blank lines to make space for the preview
|
||||||
|
const edit = new vscode.WorkspaceEdit();
|
||||||
|
const linePadding = '\n'.repeat(completionLines + 1); // Include extra line break for visual separation
|
||||||
|
edit.insert(this.document.uri, adjustedStartPosition, linePadding);
|
||||||
|
await vscode.workspace.applyEdit(edit);
|
||||||
|
|
||||||
|
this.insertedLineCount = completionLines + 1;
|
||||||
|
|
||||||
|
// Step 2: Apply decorations
|
||||||
|
const previewRanges: vscode.DecorationOptions[] = this.completionText.split('\n').map((line, index) => {
|
||||||
|
const lineNumber = adjustedStartPosition.line + index + 1; // Start preview one line later
|
||||||
|
return {
|
||||||
|
range: new vscode.Range(
|
||||||
|
new vscode.Position(lineNumber, 0),
|
||||||
|
new vscode.Position(lineNumber, 0)
|
||||||
|
),
|
||||||
|
renderOptions: {
|
||||||
|
after: {
|
||||||
|
contentText: line,
|
||||||
|
color: '#888888',
|
||||||
|
fontStyle: 'italic',
|
||||||
|
},
|
||||||
|
},
|
||||||
|
};
|
||||||
|
});
|
||||||
|
|
||||||
|
this.textEditor.setDecorations(previewDecorationType, previewRanges);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
public async acceptCompletion() {
|
||||||
|
const edit = new vscode.WorkspaceEdit();
|
||||||
|
const completionLines = this.completionText.split('\n');
|
||||||
|
const numberOfLines = completionLines.length;
|
||||||
|
|
||||||
|
// Ensure the start position is never negative
|
||||||
|
const safeStartPosition = new vscode.Position(Math.max(0, this.startPosition.line - 1), 0);
|
||||||
|
|
||||||
|
// Prepare the range to replace
|
||||||
const rangeToReplace = new vscode.Range(
|
const rangeToReplace = new vscode.Range(
|
||||||
|
safeStartPosition,
|
||||||
|
this.startPosition.translate(numberOfLines, 0)
|
||||||
|
);
|
||||||
|
|
||||||
|
// Construct the content to insert
|
||||||
|
const contentToInsert = (safeStartPosition.line === 0 ? '' : '\n') + this.completionText + '\n';
|
||||||
|
edit.replace(this.document.uri, rangeToReplace, contentToInsert);
|
||||||
|
|
||||||
|
await vscode.workspace.applyEdit(edit);
|
||||||
|
|
||||||
|
// Clear the preview decorations
|
||||||
|
this.clearPreview();
|
||||||
|
|
||||||
|
// Set activeCompletionManager to null
|
||||||
|
activeCompletionManager = null;
|
||||||
|
|
||||||
|
// Calculate the new cursor position from the inserted content
|
||||||
|
const lastCompletionLine = completionLines[completionLines.length - 1];
|
||||||
|
const newPosition = new vscode.Position(
|
||||||
|
this.startPosition.line + numberOfLines - 1,
|
||||||
|
lastCompletionLine.length
|
||||||
|
);
|
||||||
|
|
||||||
|
// Set the new cursor position
|
||||||
|
this.textEditor.selection = new vscode.Selection(newPosition, newPosition);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
public clearPreview() {
|
||||||
|
this.textEditor.setDecorations(previewDecorationType, []); // Remove all preview decorations
|
||||||
|
}
|
||||||
|
|
||||||
|
public async declineCompletion() {
|
||||||
|
this.clearPreview(); // Clear the preview decorations
|
||||||
|
|
||||||
|
try {
|
||||||
|
const document = this.textEditor.document;
|
||||||
|
const currentPosition = this.textEditor.selection.active;
|
||||||
|
|
||||||
|
// Calculate the range of lines to remove
|
||||||
|
const startLine = this.startPosition.line + 1;
|
||||||
|
const endLine = currentPosition.line;
|
||||||
|
if (endLine > startLine) {
|
||||||
|
const workspaceEdit = new vscode.WorkspaceEdit();
|
||||||
|
|
||||||
|
// Create a range from start of startLine to end of endLine
|
||||||
|
const range = new vscode.Range(
|
||||||
new vscode.Position(startLine, 0),
|
new vscode.Position(startLine, 0),
|
||||||
new vscode.Position(endLine, document.lineAt(endLine).text.length)
|
new vscode.Position(endLine, document.lineAt(endLine).text.length)
|
||||||
);
|
);
|
||||||
|
|
||||||
|
// Delete the range
|
||||||
|
workspaceEdit.delete(document.uri, range);
|
||||||
|
|
||||||
// Apply the edit
|
// Apply the edit
|
||||||
const edit = new vscode.WorkspaceEdit();
|
await vscode.workspace.applyEdit(workspaceEdit);
|
||||||
edit.replace(document.uri, rangeToReplace, completionText);
|
|
||||||
await vscode.workspace.applyEdit(edit);
|
|
||||||
|
|
||||||
// Move the cursor to the end of the inserted text
|
// Move the cursor back to the original position
|
||||||
const newPosition = new vscode.Position(startLine + completionText.split('\n').length - 1, completionText.split('\n').pop()!.length);
|
this.textEditor.selection = new vscode.Selection(this.startPosition, this.startPosition);
|
||||||
textEditor.selection = new vscode.Selection(newPosition, newPosition);
|
|
||||||
|
console.log(`Lines ${startLine + 1} to ${endLine + 1} removed successfully`);
|
||||||
|
activeCompletionManager = null;
|
||||||
|
} else {
|
||||||
|
console.log('No lines to remove');
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
console.error('Error declining completion:', error);
|
||||||
|
vscode.window.showErrorMessage(`Error removing lines: ${error}`);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
async function autocompleteCommand(textEditor: vscode.TextEditor, edit: vscode.TextEditorEdit, ...args: any[]) {
|
||||||
|
const cancellationTokenSource = new vscode.CancellationTokenSource();
|
||||||
|
const cancellationToken = cancellationTokenSource.token;
|
||||||
|
|
||||||
|
try {
|
||||||
|
const document = textEditor.document;
|
||||||
|
const position = textEditor.selection.active;
|
||||||
|
const context = getContextLines(document, position);
|
||||||
|
const fimPrompt = createFIMPrompt(context, document.languageId);
|
||||||
|
|
||||||
|
const completionText = await vscode.window.withProgress({
|
||||||
|
location: vscode.ProgressLocation.Notification,
|
||||||
|
title: 'Fabelous Autocoder',
|
||||||
|
cancellable: true,
|
||||||
|
}, async (progress, progressCancellationToken) => {
|
||||||
|
progress.report({ message: 'Generating...' });
|
||||||
|
return await generateCompletion(fimPrompt, progressCancellationToken);
|
||||||
|
});
|
||||||
|
|
||||||
|
console.log('Completion generated:', completionText);
|
||||||
|
|
||||||
|
|
||||||
|
const completionManager = new CompletionManager(textEditor, position, completionText);
|
||||||
|
await completionManager.showPreview();
|
||||||
|
activeCompletionManager = completionManager;
|
||||||
|
|
||||||
progress.report({ message: "Fabelous completion finished." });
|
|
||||||
|
|
||||||
} catch (err: any) {
|
} catch (err: any) {
|
||||||
vscode.window.showErrorMessage(
|
console.error('Error in autocompleteCommand:', err);
|
||||||
"Fabelous Autocoder encountered an error: " + err.message
|
vscode.window.showErrorMessage(`Fabelous Autocoder encountered an error: ${err.message}`);
|
||||||
);
|
} finally {
|
||||||
console.log(err);
|
cancellationTokenSource.dispose();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
|
async function handleTab() {
|
||||||
|
if (activeCompletionManager) {
|
||||||
|
await activeCompletionManager.acceptCompletion();
|
||||||
|
} else {
|
||||||
|
await vscode.commands.executeCommand('tab');
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async function handleBackspace() {
|
||||||
|
if (activeCompletionManager) {
|
||||||
|
await activeCompletionManager.declineCompletion();
|
||||||
|
} else {
|
||||||
|
await vscode.commands.executeCommand('deleteLeft');
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
async function provideCompletionItems(document: vscode.TextDocument, position: vscode.Position, cancellationToken: vscode.CancellationToken) {
|
async function provideCompletionItems(document: vscode.TextDocument, position: vscode.Position, cancellationToken: vscode.CancellationToken) {
|
||||||
const item = new vscode.CompletionItem("Fabelous autocompletion");
|
const item = new vscode.CompletionItem('Fabelous autocompletion');
|
||||||
item.insertText = new vscode.SnippetString('${1:}');
|
item.insertText = new vscode.SnippetString('${1:}');
|
||||||
|
item.documentation = new vscode.MarkdownString('Press `Enter` to get an autocompletion from Fabelous Autocoder');
|
||||||
|
|
||||||
if (responsePreview) {
|
if (config.responsePreview) {
|
||||||
await new Promise(resolve => setTimeout(resolve, responsePreviewDelay * 1000));
|
await new Promise(resolve => setTimeout(resolve, config.responsePreviewDelay * 1000));
|
||||||
if (cancellationToken.isCancellationRequested) {
|
if (cancellationToken.isCancellationRequested) {
|
||||||
return [item];
|
return [item];
|
||||||
}
|
}
|
||||||
|
@ -149,58 +290,39 @@ async function provideCompletionItems(document: vscode.TextDocument, position: v
|
||||||
const fimPrompt = createFIMPrompt(context, document.languageId);
|
const fimPrompt = createFIMPrompt(context, document.languageId);
|
||||||
|
|
||||||
try {
|
try {
|
||||||
const response_preview = await axios.post(apiEndpoint, {
|
const result = await generateCompletion(fimPrompt, cancellationToken);
|
||||||
model: apiModel,
|
const preview = (result as any).preview;
|
||||||
prompt: fimPrompt,
|
if (preview) {
|
||||||
stream: false,
|
item.detail = preview.split('\n')[0];
|
||||||
raw: true,
|
|
||||||
options: {
|
|
||||||
num_predict: responsePreviewMaxTokens,
|
|
||||||
temperature: apiTemperature,
|
|
||||||
stop: ['<fim_suffix>', '\n', '```'],
|
|
||||||
...(keepAlive && { keep_alive: keepAlive }),
|
|
||||||
...(topP && { top_p: topP }),
|
|
||||||
}
|
}
|
||||||
}, {
|
|
||||||
cancelToken: new axios.CancelToken((c) => {
|
|
||||||
cancellationToken.onCancellationRequested(() => c("Autocompletion request terminated by completion cancel"));
|
|
||||||
})
|
|
||||||
});
|
|
||||||
} catch (error) {
|
} catch (error) {
|
||||||
console.error("Error fetching preview:", error);
|
console.error('Error fetching preview:', error);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
item.documentation = new vscode.MarkdownString('Press `Enter` to get an autocompletion from Fabelous Autocoder');
|
if (config.continueInline || !config.responsePreview) {
|
||||||
if (continueInline || !responsePreview) {
|
|
||||||
item.command = {
|
item.command = {
|
||||||
command: 'fabelous-autocoder.autocomplete',
|
command: 'fabelous-autocoder.autocomplete',
|
||||||
title: 'Fabelous Autocomplete',
|
title: 'Fabelous Autocomplete',
|
||||||
arguments: [cancellationToken]
|
arguments: []
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
return [item];
|
return [item];
|
||||||
}
|
}
|
||||||
|
export function activate(context: vscode.ExtensionContext) {
|
||||||
|
updateConfig();
|
||||||
|
createPreviewDecorationType();
|
||||||
|
|
||||||
function activate(context: vscode.ExtensionContext) {
|
context.subscriptions.push(
|
||||||
const completionProvider = vscode.languages.registerCompletionItemProvider("*", {
|
vscode.workspace.onDidChangeConfiguration(updateConfig),
|
||||||
provideCompletionItems
|
vscode.languages.registerCompletionItemProvider('*', { provideCompletionItems }, ...config.completionKeys),
|
||||||
},
|
vscode.commands.registerTextEditorCommand('fabelous-autocoder.autocomplete', autocompleteCommand),
|
||||||
...completionKeys.split("")
|
vscode.commands.registerCommand('fabelous-autocoder.handleTab', handleTab),
|
||||||
|
vscode.commands.registerCommand('fabelous-autocoder.handleBackspace', handleBackspace) // Add this line
|
||||||
);
|
);
|
||||||
const externalAutocompleteCommand = vscode.commands.registerTextEditorCommand(
|
|
||||||
"fabelous-autocoder.autocomplete",
|
|
||||||
(textEditor, _, cancellationToken?) => {
|
|
||||||
autocompleteCommand(textEditor, cancellationToken);
|
|
||||||
}
|
|
||||||
);
|
|
||||||
context.subscriptions.push(completionProvider);
|
|
||||||
context.subscriptions.push(externalAutocompleteCommand);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
function deactivate() { }
|
|
||||||
|
|
||||||
module.exports = {
|
|
||||||
activate,
|
export function deactivate() {}
|
||||||
deactivate,
|
|
||||||
};
|
|
||||||
|
|
Loading…
Reference in New Issue