add deepseek-reasoner, and change deepseek's summary model to deepseek-chat

This commit is contained in:
suruiqiang 2025-01-23 18:47:22 +08:00
parent 0d5e66a9ae
commit 2173c82bb5
2 changed files with 6 additions and 1 deletions

View File

@ -393,6 +393,7 @@ You are an AI assistant with access to system tools. Your role is to help users
export const SUMMARIZE_MODEL = "gpt-4o-mini"; export const SUMMARIZE_MODEL = "gpt-4o-mini";
export const GEMINI_SUMMARIZE_MODEL = "gemini-pro"; export const GEMINI_SUMMARIZE_MODEL = "gemini-pro";
export const DEEPSEEK_SUMMARIZE_MODEL = "deepseek-chat";
export const KnowledgeCutOffDate: Record<string, string> = { export const KnowledgeCutOffDate: Record<string, string> = {
default: "2021-09", default: "2021-09",
@ -561,7 +562,7 @@ const iflytekModels = [
"4.0Ultra", "4.0Ultra",
]; ];
const deepseekModels = ["deepseek-chat", "deepseek-coder"]; const deepseekModels = ["deepseek-chat", "deepseek-coder", "deepseek-reasoner"];
const xAIModes = ["grok-beta"]; const xAIModes = ["grok-beta"];

View File

@ -20,6 +20,7 @@ import {
DEFAULT_MODELS, DEFAULT_MODELS,
DEFAULT_SYSTEM_TEMPLATE, DEFAULT_SYSTEM_TEMPLATE,
GEMINI_SUMMARIZE_MODEL, GEMINI_SUMMARIZE_MODEL,
DEEPSEEK_SUMMARIZE_MODEL,
KnowledgeCutOffDate, KnowledgeCutOffDate,
MCP_SYSTEM_TEMPLATE, MCP_SYSTEM_TEMPLATE,
MCP_TOOLS_TEMPLATE, MCP_TOOLS_TEMPLATE,
@ -143,7 +144,10 @@ function getSummarizeModel(
} }
if (currentModel.startsWith("gemini")) { if (currentModel.startsWith("gemini")) {
return [GEMINI_SUMMARIZE_MODEL, ServiceProvider.Google]; return [GEMINI_SUMMARIZE_MODEL, ServiceProvider.Google];
} else if (currentModel.startsWith("deepseek-")) {
return [DEEPSEEK_SUMMARIZE_MODEL, ServiceProvider.DeepSeek];
} }
return [currentModel, providerName]; return [currentModel, providerName];
} }