Skip to content

Commit 5648c21

Browse files
remove "deepseek" from TemplateType
1 parent 8f67a00 commit 5648c21

File tree

2 files changed

+4
-10
lines changed

2 files changed

+4
-10
lines changed

core/index.d.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -903,7 +903,7 @@ export type TemplateType =
903903
| "chatml"
904904
| "none"
905905
| "openchat"
906-
| "deepseek"
906+
// | "deepseek" ---- remove this because add tool call support for deepseek models
907907
| "xwin-coder"
908908
| "neural-chat"
909909
| "codellama-70b"

core/llm/autodetect.ts

Lines changed: 3 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -186,7 +186,8 @@ function autodetectTemplateType(model: string): TemplateType | undefined {
186186
lower.includes("gemini") ||
187187
lower.includes("grok") ||
188188
lower.includes("moonshot") ||
189-
lower.includes("mercury")
189+
lower.includes("mercury") ||
190+
lower.includes("deepseek")
190191
) {
191192
return undefined;
192193
}
@@ -247,10 +248,6 @@ function autodetectTemplateType(model: string): TemplateType | undefined {
247248
return "llama2";
248249
}
249250

250-
if (lower.includes("deepseek")) {
251-
return "deepseek";
252-
}
253-
254251
if (lower.includes("ninja") || lower.includes("openchat")) {
255252
return "openchat";
256253
}
@@ -293,7 +290,6 @@ function autodetectTemplateFunction(
293290
anthropic: anthropicTemplateMessages,
294291
chatml: chatmlTemplateMessages,
295292
// deepseek: deepseekTemplateMessages,
296-
deepseek: null,
297293
openchat: openchatTemplateMessages,
298294
"xwin-coder": xWinCoderTemplateMessages,
299295
"neural-chat": neuralChatTemplateMessages,
@@ -316,7 +312,7 @@ const USES_OS_MODELS_EDIT_PROMPT: TemplateType[] = [
316312
"alpaca",
317313
"chatml",
318314
// "codellama-70b", Doesn't respond well to this prompt
319-
"deepseek",
315+
// "deepseek",
320316
"gemma",
321317
"llama2",
322318
"llava",
@@ -356,8 +352,6 @@ function autodetectPromptTemplates(
356352
}
357353
} else if (templateType === "alpaca") {
358354
editTemplate = alpacaEditPrompt;
359-
} else if (templateType === "deepseek") {
360-
editTemplate = deepseekEditPrompt;
361355
} else if (templateType === "openchat") {
362356
editTemplate = openchatEditPrompt;
363357
} else if (templateType === "xwin-coder") {

0 commit comments

Comments
 (0)