From e7f8c453d3f9eed4b3e402df26310a6b349fea39 Mon Sep 17 00:00:00 2001
From: Luo Tim <shiwen@merico.dev>
Date: Tue, 8 Oct 2024 23:05:39 +0800
Subject: [PATCH] Add missing comment prefixes

---
 .../ai/devchat/plugin/completion/agent/ContextBuilder.kt      | 4 ++--
 1 file changed, 2 insertions(+), 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
index e07beb9..ce4736b 100644
--- 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>"
     }
 }
\ No newline at end of file