diff --git a/src/autoComplete.ts b/src/autoComplete.ts index 7c77e64..e750865 100644 --- a/src/autoComplete.ts +++ b/src/autoComplete.ts @@ -143,6 +143,9 @@ const tokenProvider = async ( export const getAutoCompleteProvider = (extension: ExtensionState) => { const provider: vscode.InlineCompletionItemProvider = { async provideInlineCompletionItems(document, position, context, token) { + if (!extension.configuration.inlineCompletion.enable) { + return []; + } try { const completions = await tokenProvider(extension, document, position, context, token); diff --git a/src/config.ts b/src/config.ts index 6ddf6f6..3863adb 100644 --- a/src/config.ts +++ b/src/config.ts @@ -18,10 +18,10 @@ const KEY_INLINE_COMPLETION_TRIE_PRUNE_TIMEOUT = 'inlineCompletion.triePruneTime const DEFAULT_OLLAMA_HOST = 'http://localhost:11434'; const DEFAULT_INLINE_COMPLETION_ENABLE = true; const DEFAULT_INLINE_COMPLETION_MODEL = 'deepseek-coder:6.7b'; -const DEFAULT_INLINE_COMPLETION_PREFIX_START = ''; -const DEFAULT_INLINE_COMPLETION_PREFIX_ENDS = ['', '', '', ''].join(); -const DEFAULT_INLINE_COMPLETION_SUFFIX_START = ''; -const DEFAULT_INLINE_COMPLETION_SUFFIX_END = ''; +const DEFAULT_INLINE_COMPLETION_PREFIX_START = '<|prefix_start|>'; +const DEFAULT_INLINE_COMPLETION_PREFIX_ENDS = ['<|prefix_start|>', '<|prefix_end|>'].join(); +const DEFAULT_INLINE_COMPLETION_SUFFIX_START = '<|suffix_start|>'; +const DEFAULT_INLINE_COMPLETION_SUFFIX_END = '<|suffix_end|>'; const DEFAULT_INLINE_COMPLETION_MAX_TOKENS = 50; const DEFAULT_INLINE_COMPLETION_GENERATION_TIMEOUT = 200; const DEFAULT_INLINE_COMPLETION_TRIE_PRUNE_TIMEOUT = 10000;