Add missing comment prefixes

devchat-ai · Oct 8, 2024 · e7f8c45 · e7f8c45
1 parent e225fca
commit e7f8c45
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/main/kotlin/ai/devchat/plugin/completion/agent/ContextBuilder.kt b/src/main/kotlin/ai/devchat/plugin/completion/agent/ContextBuilder.kt
@@ -186,8 +186,8 @@ class ContextBuilder(val file: PsiFile, val offset: Int) {
         ).joinToString("")
         Log.info("Extras completion context:\n$extras")
         return  if (!model.isNullOrEmpty() && model.contains("deepseek"))
-            "<｜fim▁begin｜>$extras<filename>$filepath\n\n$prefix<｜fim▁hole｜>$suffix<｜fim▁end｜>"
+            "<｜fim▁begin｜>$extras$commentPrefix<filename>$filepath\n\n$prefix<｜fim▁hole｜>$suffix<｜fim▁end｜>"
         else
-            "<fim_prefix>$extras<filename>$filepath\n\n$prefix<fim_suffix>$suffix<fim_middle>"
+            "<fim_prefix>$extras$commentPrefix<filename>$filepath\n\n$prefix<fim_suffix>$suffix<fim_middle>"
     }
 }