mirror of
https://github.com/diegosouzapw/OmniRoute.git
synced 2026-08-05 14:52:09 +03:00
f8ef562658
* fix(sse): recognize xiaomi-tokenplan mimo as a thinking-mode model (port from 9router#1321) The reasoning_content injector already handles DeepSeek/Kimi/K2/MiniMax thinking-mode upstreams, echoing a placeholder reasoning_content on assistant turns that lack one. Its THINKING_MODEL_PATTERNS list omitted the xiaomi-tokenplan mimo family, so requests through xiaomi-tokenplan/mimo-v2.5-pro still hit upstream's 400 'reasoning_content in the thinking mode must be passed back to the API', making the model unusable in multi-turn conversations (e.g. Codex CLI). Add a /\bmimo\b/i pattern so mimo models get the same treatment. Reported-by: z.wl (@xxue-z) (https://github.com/decolua/9router/issues/1321) * docs(changelog): add fragment for #7098 mimo thinking-model fix
76 lines
2.6 KiB
TypeScript
76 lines
2.6 KiB
TypeScript
/**
|
|
* Thinking-mode upstreams (DeepSeek V4 Flash, Kimi, MiniMax, xiaomi-tokenplan
|
|
* mimo, ...) require
|
|
* `reasoning_content` to be echoed back on every assistant message in the
|
|
* conversation history. Standard OpenAI clients do not preserve that field
|
|
* across turns, so we inject a non-empty placeholder before forwarding.
|
|
*
|
|
* Without the placeholder these upstreams return:
|
|
* 400 Bad Request — reasoning_content must be passed back
|
|
*
|
|
* Ported from decolua/9router#1099 (issue #1543). Pure helper — no I/O, no
|
|
* cross-module deps — kept narrow so it can be reused by other meta-providers
|
|
* that proxy to thinking-mode models.
|
|
*/
|
|
|
|
const PLACEHOLDER = " ";
|
|
|
|
type JsonRecord = Record<string, unknown>;
|
|
|
|
/**
|
|
* Model-id predicates for thinking-mode families that need the echo.
|
|
* Matched case-insensitively against the resolved model id (post upstream
|
|
* routing, e.g. `oc/deepseek-v4-flash-free` for the OpenCode meta-provider).
|
|
*/
|
|
const THINKING_MODEL_PATTERNS: RegExp[] = [
|
|
/deepseek/i,
|
|
/\bkimi\b/i,
|
|
/\bk2\b/i, // moonshot kimi k2 family alias
|
|
/\bminimax\b/i,
|
|
/\bmimo\b/i, // xiaomi-tokenplan mimo family (e.g. xiaomi-tokenplan/mimo-v2.5-pro)
|
|
];
|
|
|
|
export function isThinkingMessageModel(model: string | undefined | null): boolean {
|
|
if (!model || typeof model !== "string") return false;
|
|
return THINKING_MODEL_PATTERNS.some((re) => re.test(model));
|
|
}
|
|
|
|
function hasNonEmptyReasoningContent(message: JsonRecord): boolean {
|
|
return (
|
|
typeof message.reasoning_content === "string" &&
|
|
(message.reasoning_content as string).trim().length > 0
|
|
);
|
|
}
|
|
|
|
function isAssistantMessage(value: unknown): value is JsonRecord {
|
|
return (
|
|
!!value &&
|
|
typeof value === "object" &&
|
|
!Array.isArray(value) &&
|
|
(value as JsonRecord).role === "assistant"
|
|
);
|
|
}
|
|
|
|
/**
|
|
* Inject a placeholder `reasoning_content` on every assistant message in
|
|
* `body.messages` that lacks one. Returns the original object if no mutation
|
|
* was needed, or a shallow-copied body with a new messages array otherwise.
|
|
*
|
|
* No-op when the body shape is unexpected (defensive).
|
|
*/
|
|
export function injectReasoningContentForThinkingModel(body: unknown): unknown {
|
|
if (!body || typeof body !== "object" || Array.isArray(body)) return body;
|
|
const record = body as JsonRecord;
|
|
if (!Array.isArray(record.messages)) return body;
|
|
|
|
let modified = false;
|
|
const messages = record.messages.map((message) => {
|
|
if (!isAssistantMessage(message)) return message;
|
|
if (hasNonEmptyReasoningContent(message)) return message;
|
|
modified = true;
|
|
return { ...message, reasoning_content: PLACEHOLDER };
|
|
});
|
|
|
|
return modified ? { ...record, messages } : body;
|
|
}
|