File size: 2,366 Bytes
88c4c60 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 | // Some thinking-mode providers (DeepSeek, Kimi, MiniMax, ...) require reasoning_content
// to be echoed back on assistant messages. Clients in OpenAI format don't send it,
// so we inject a non-empty placeholder to satisfy upstream validation.
const PLACEHOLDER = " ";
// Provider-level rules: keyed by executor.provider
const PROVIDER_RULES = {
deepseek: { scope: "all" },
minimax: { scope: "all" },
"minimax-cn": { scope: "all" }
};
// Model-level rules: matched by predicate against model id
const MODEL_RULES = [
{ match: m => /^kimi-/i.test(m || ""), scope: "toolCalls" },
{ match: m => /deepseek/i.test(m || ""), 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;
if (typeof rc === "string" && rc.length > 0) return false;
if (scope === "toolCalls") return Array.isArray(message.tool_calls) && message.tool_calls.length > 0;
return true;
}
function applyRule(body, rule) {
if (!rule || !body?.messages) return body;
const messages = body.messages.map(m =>
shouldInject(m, rule.scope) ? { ...m, reasoning_content: PLACEHOLDER } : m
);
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;
const nextBody = applyDeepSeekV4ProAlias({ provider, model, body });
return applyRule(nextBody, rule);
}
|