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;