@code-yeongyu/senpi 2026.7.24 → 2026.7.25-2

package/node_modules/@earendil-works/pi-ai/dist/api/openai-completions.js

@@ -15,6 +15,87 @@ import { buildCopilotDynamicHeaders, hasCopilotVisionInput } from "./github-copi
 import { clampOpenAIPromptCacheKey } from "./openai-prompt-cache.js";
 import { applyExtraBody, buildBaseOptions, clampMaxForOpenAI, OPENAI_COMPLETIONS_RESERVED_BODY_KEYS, } from "./simple-options.js";
 import { transformMessages } from "./transform-messages.js";
+const KIMI_K3_THINKING_LEVEL_MAP = {
+    off: null,
+    minimal: null,
+    low: "low",
+    medium: null,
+    high: "high",
+    xhigh: null,
+    max: "max",
+};
+const DEEPSEEK_THINKING_LEVEL_MAP = {
+    minimal: "high",
+    low: "high",
+    medium: "high",
+    high: "high",
+    xhigh: "max",
+    max: "max",
+};
+const OPENROUTER_DEEPSEEK_THINKING_LEVEL_MAP = {
+    minimal: "high",
+    low: "high",
+    medium: "high",
+    high: "high",
+    xhigh: "high",
+    max: "high",
+};
+const MIMO_THINKING_LEVEL_MAP = {
+    minimal: "low",
+    low: "low",
+    medium: "medium",
+    high: "high",
+    xhigh: "high",
+    max: null,
+};
+const OLLAMA_THINKING_LEVEL_MAP = {
+    off: "none",
+    minimal: null,
+    low: "low",
+    medium: "medium",
+    high: "high",
+    xhigh: null,
+    max: "max",
+};
+function getThinkingLevelMap(model, compat) {
+    if (model.thinkingLevelMap !== undefined) {
+        return model.thinkingLevelMap;
+    }
+    const id = model.id.toLowerCase();
+    const isKimiK3 = id === "k3" || id.startsWith("k3-") || /(?:^|[/:-])kimi-k3(?:$|[/.:_-])/.test(id);
+    const isDeepSeek = id.includes("deepseek");
+    const isMiMo = /\bmimo\b/.test(id);
+    const isGlm52 = /(?:^|[/:-])glm-5\.2(?:$|[/.:_-])/.test(id);
+    if (model.provider === "ollama") {
+        return OLLAMA_THINKING_LEVEL_MAP;
+    }
+    if (isKimiK3) {
+        return KIMI_K3_THINKING_LEVEL_MAP;
+    }
+    if (compat.thinkingFormat === "openrouter" && isDeepSeek) {
+        return OPENROUTER_DEEPSEEK_THINKING_LEVEL_MAP;
+    }
+    if ((compat.thinkingFormat === "openai" || compat.thinkingFormat === "openrouter") && isMiMo) {
+        return MIMO_THINKING_LEVEL_MAP;
+    }
+    if (isDeepSeek) {
+        return DEEPSEEK_THINKING_LEVEL_MAP;
+    }
+    if (isGlm52) {
+        if (compat.thinkingFormat === "zai") {
+            return DEEPSEEK_THINKING_LEVEL_MAP;
+        }
+        if (compat.thinkingFormat === "openrouter") {
+            return { xhigh: "xhigh" };
+        }
+        return { max: "max" };
+    }
+    return undefined;
+}
+function resolveReasoningEffort(thinkingLevelMap, effort) {
+    const mapped = thinkingLevelMap?.[effort];
+    return mapped === undefined ? effort : typeof mapped === "string" ? mapped : undefined;
+}
 /**
  * Check if conversation messages contain tool calls or tool results.
  * This is needed because Anthropic (via proxy) requires the tools param
@@ -497,12 +578,15 @@ export const stream = (model, context, options) => {
 export const streamSimple = (model, context, options) => {
     getClientApiKey(model.provider, options?.apiKey, options?.headers);
     const base = buildBaseOptions(model, context, options, options?.apiKey);
-    const clampedReasoning = options?.reasoning ? clampThinkingLevel(model, options.reasoning) : undefined;
+    const compat = getCompat(model);
+    const thinkingLevelMap = getThinkingLevelMap(model, compat);
+    const thinkingModel = thinkingLevelMap === model.thinkingLevelMap ? model : { ...model, thinkingLevelMap };
+    const clampedReasoning = options?.reasoning ? clampThinkingLevel(thinkingModel, options.reasoning) : undefined;
     const reasoningEffort = clampedReasoning === "off"
         ? undefined
-        : clampedReasoning === "max" && model.thinkingLevelMap?.max !== undefined
+        : clampedReasoning === "max" && thinkingLevelMap?.max !== undefined
             ? "max"
-            : clampMaxForOpenAI(clampedReasoning, supportsXhigh(model));
+            : clampMaxForOpenAI(clampedReasoning, supportsXhigh(thinkingModel));
     const toolChoice = options?.toolChoice;
     return stream(model, context, {
         ...base,
@@ -549,6 +633,7 @@ function buildParams(model, context, options, compat = getCompat(model), cacheRe
         grammarToolInputProperties,
     });
     const cacheControl = getCompatCacheControl(compat, cacheRetention);
+    const thinkingLevelMap = getThinkingLevelMap(model, compat);
     const params = {
         model: model.id,
         messages,
@@ -598,9 +683,8 @@ function buildParams(model, context, options, compat = getCompat(model), cacheRe
         const zaiParams = params;
         zaiParams.thinking = options?.reasoningEffort ? { type: "enabled", clear_thinking: false } : { type: "disabled" };
         if (options?.reasoningEffort && compat.supportsReasoningEffort) {
-            const mappedEffort = model.thinkingLevelMap?.[options.reasoningEffort];
-            const effort = mappedEffort === undefined ? options.reasoningEffort : mappedEffort;
-            if (typeof effort === "string") {
+            const effort = resolveReasoningEffort(thinkingLevelMap, options.reasoningEffort);
+            if (effort !== undefined) {
                 zaiParams.reasoning_effort = effort;
             }
         }
@@ -624,10 +708,13 @@ function buildParams(model, context, options, compat = getCompat(model), cacheRe
         if (options?.reasoningEffort) {
             params.thinking = { type: "enabled" };
             if (compat.supportsReasoningEffort) {
-                params.reasoning_effort = model.thinkingLevelMap?.[options.reasoningEffort] ?? options.reasoningEffort;
+                const effort = resolveReasoningEffort(thinkingLevelMap, options.reasoningEffort);
+                if (effort !== undefined) {
+                    params.reasoning_effort = effort;
+                }
             }
         }
-        else if (compat.supportsDisabledThinking !== false && model.thinkingLevelMap?.off !== null) {
+        else if (compat.supportsDisabledThinking !== false && thinkingLevelMap?.off !== null) {
             params.thinking = { type: "disabled" };
         }
     }
@@ -635,16 +722,17 @@ function buildParams(model, context, options, compat = getCompat(model), cacheRe
         // OpenRouter normalizes reasoning across providers via a nested reasoning object.
         const openRouterParams = params;
         if (options?.reasoningEffort) {
-            openRouterParams.reasoning = {
-                effort: model.thinkingLevelMap?.[options.reasoningEffort] ?? options.reasoningEffort,
-            };
+            const effort = resolveReasoningEffort(thinkingLevelMap, options.reasoningEffort);
+            if (effort !== undefined) {
+                openRouterParams.reasoning = { effort };
+            }
         }
-        else if (model.thinkingLevelMap?.off !== null) {
-            openRouterParams.reasoning = { effort: model.thinkingLevelMap?.off ?? "none" };
+        else if (thinkingLevelMap?.off !== null) {
+            openRouterParams.reasoning = { effort: thinkingLevelMap?.off ?? "none" };
         }
     }
     else if (compat.thinkingFormat === "ant-ling" && model.reasoning && options?.reasoningEffort) {
-        const effort = model.thinkingLevelMap?.[options.reasoningEffort];
+        const effort = thinkingLevelMap?.[options.reasoningEffort];
         if (typeof effort === "string") {
             params.reasoning = { effort };
         }
@@ -653,23 +741,32 @@ function buildParams(model, context, options, compat = getCompat(model), cacheRe
         const togetherParams = params;
         togetherParams.reasoning = { enabled: !!options?.reasoningEffort };
         if (options?.reasoningEffort && compat.supportsReasoningEffort) {
-            togetherParams.reasoning_effort = model.thinkingLevelMap?.[options.reasoningEffort] ?? options.reasoningEffort;
+            const effort = resolveReasoningEffort(thinkingLevelMap, options.reasoningEffort);
+            if (effort !== undefined) {
+                togetherParams.reasoning_effort = effort;
+            }
         }
     }
     else if (compat.thinkingFormat === "string-thinking" && model.reasoning) {
         if (options?.reasoningEffort) {
-            params.thinking = model.thinkingLevelMap?.[options.reasoningEffort] ?? options.reasoningEffort;
+            const effort = resolveReasoningEffort(thinkingLevelMap, options.reasoningEffort);
+            if (effort !== undefined) {
+                params.thinking = effort;
+            }
         }
-        else if (model.thinkingLevelMap?.off !== null) {
-            params.thinking = model.thinkingLevelMap?.off ?? "none";
+        else if (thinkingLevelMap?.off !== null) {
+            params.thinking = thinkingLevelMap?.off ?? "none";
         }
     }
     else if (options?.reasoningEffort && model.reasoning && compat.supportsReasoningEffort) {
         // OpenAI-style reasoning_effort
-        params.reasoning_effort = model.thinkingLevelMap?.[options.reasoningEffort] ?? options.reasoningEffort;
+        const effort = resolveReasoningEffort(thinkingLevelMap, options.reasoningEffort);
+        if (effort !== undefined) {
+            params.reasoning_effort = effort;
+        }
     }
     else if (!options?.reasoningEffort && model.reasoning && compat.supportsReasoningEffort) {
-        const offValue = model.thinkingLevelMap?.off;
+        const offValue = thinkingLevelMap?.off;
         if (typeof offValue === "string") {
             params.reasoning_effort = offValue;
         }
@@ -696,14 +793,14 @@ function buildParams(model, context, options, compat = getCompat(model), cacheRe
 function buildChatTemplateKwargs(model, options, compat) {
     const kwargs = {};
     for (const [key, value] of Object.entries(compat.chatTemplateKwargs)) {
-        const resolved = resolveChatTemplateKwargValue(model, options, value);
+        const resolved = resolveChatTemplateKwargValue(model, options, compat, value);
         if (resolved !== undefined) {
             kwargs[key] = resolved;
         }
     }
     return Object.keys(kwargs).length > 0 ? kwargs : undefined;
 }
-function resolveChatTemplateKwargValue(model, options, value) {
+function resolveChatTemplateKwargValue(model, options, compat, value) {
     if (typeof value !== "object" || value === null) {
         return value;
     }
@@ -714,7 +811,8 @@ function resolveChatTemplateKwargValue(model, options, value) {
     if (value.$var === "thinking.enabled") {
         return !!reasoningEffort;
     }
-    const mappedValue = reasoningEffort ? model.thinkingLevelMap?.[reasoningEffort] : model.thinkingLevelMap?.off;
+    const thinkingLevelMap = getThinkingLevelMap(model, compat);
+    const mappedValue = reasoningEffort ? thinkingLevelMap?.[reasoningEffort] : thinkingLevelMap?.off;
     return mappedValue === undefined ? reasoningEffort : typeof mappedValue === "string" ? mappedValue : undefined;
 }
 function getCompatCacheControl(compat, cacheRetention) {