feat: add DeepSeek V4 Pro effort aliases (#950)

This commit is contained in:
Aqil Aziz
2026-05-14 10:39:34 +07:00
committed by GitHub
parent d1a8dd4987
commit da00a25eef
2 changed files with 41 additions and 1 deletions
+39 -1
View File
@@ -15,6 +15,18 @@ const MODEL_RULES = [
{ match: m => m?.startsWith?.("deepseek-"), scope: "all" }
];
const DEEPSEEK_V4_PRO = "deepseek-v4-pro";
const DEEPSEEK_V4_PRO_ALIASES = {
[`${DEEPSEEK_V4_PRO}-max`]: {
thinkingType: "enabled",
reasoningEffort: "max"
},
[`${DEEPSEEK_V4_PRO}-none`]: {
thinkingType: "disabled",
reasoningEffort: null
}
};
function shouldInject(message, scope) {
if (message?.role !== "assistant") return false;
const rc = message.reasoning_content;
@@ -31,9 +43,35 @@ function applyRule(body, rule) {
return { ...body, messages };
}
function applyDeepSeekV4ProAlias({ provider, model, body }) {
const alias = DEEPSEEK_V4_PRO_ALIASES[model];
if (provider !== "deepseek" || !alias || !body) return body;
const nextBody = {
...body,
model: DEEPSEEK_V4_PRO,
extra_body: {
...(body.extra_body || {}),
thinking: {
...(body.extra_body?.thinking || {}),
type: alias.thinkingType
}
}
};
if (alias.reasoningEffort) {
nextBody.reasoning_effort = alias.reasoningEffort;
} else {
delete nextBody.reasoning_effort;
}
return nextBody;
}
export function injectReasoningContent({ provider, model, body }) {
const providerRule = PROVIDER_RULES[provider];
const modelRule = MODEL_RULES.find(r => r.match(model));
const rule = providerRule || modelRule;
return applyRule(body, rule);
const nextBody = applyDeepSeekV4ProAlias({ provider, model, body });
return applyRule(nextBody, rule);
}