From 20c31de21d6d66dc2f0b2903e7d0209872c7ddc9 Mon Sep 17 00:00:00 2001
From: squall <wsxqyws@gmail.com>
Date: Thu, 29 Feb 2024 22:23:29 +0800
Subject: [PATCH] fix: completion prompt template for Deepseek Coder (#387)

* fix: completion prompt template for Deepseek Coder

* Add stop token
---
 .../codegpt/codecompletions/InfillPromptTemplate.java         | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/main/java/ee/carlrobert/codegpt/codecompletions/InfillPromptTemplate.java b/src/main/java/ee/carlrobert/codegpt/codecompletions/InfillPromptTemplate.java
index f31e9ca7..c6cc5e2e 100644
--- a/src/main/java/ee/carlrobert/codegpt/codecompletions/InfillPromptTemplate.java
+++ b/src/main/java/ee/carlrobert/codegpt/codecompletions/InfillPromptTemplate.java
@@ -25,10 +25,10 @@ public enum InfillPromptTemplate {
       return format("<fim_prefix>%s<fim_suffix>%s<fim_middle>", prefix, suffix);
     }
   },
-  DEEPSEEK_CODER("DeepSeek Coder") {
+  DEEPSEEK_CODER("DeepSeek Coder", List.of("<|EOT|>")) {
     @Override
     public String buildPrompt(String prefix, String suffix) {
-      return format("<|fim_begin|>%s<|fim_hole|>%s<|fim_end|>", prefix, suffix);
+      return format("<｜fim▁begin｜>%s<｜fim▁hole｜>%s<｜fim▁end｜>", prefix, suffix);
     }
   };