From 869fbc31bd4e41a447699f8e31fb4963b1f51e84 Mon Sep 17 00:00:00 2001 From: Leyla Becker Date: Mon, 4 Aug 2025 17:53:19 -0500 Subject: [PATCH 1/2] updated prefix tokens --- src/autoComplete.ts | 3 +++ 1 file changed, 3 insertions(+) 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); From fbc588368e6c5b6e1963c03e40a51e64b44614ba Mon Sep 17 00:00:00 2001 From: Leyla Becker Date: Mon, 4 Aug 2025 17:53:33 -0500 Subject: [PATCH 2/2] made completion depend on enable option --- src/config.ts | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) 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;