Fix the way we decide if a DeepInfra model has reasoning capabilities

This commit is contained in:
Georgi Atsev 2026-03-19 15:53:51 +02:00
parent 905e98604d
commit beb45cba22
3 changed files with 8 additions and 6 deletions

View File

@ -18,7 +18,7 @@ function makeModelEntry(overrides: Record<string, unknown> = {}) {
output_tokens: 15.0,
cache_read_tokens: 0.3,
},
tags: ["vision", "reasoning_effort", "prompt_cache"],
tags: ["vision", "reasoning_effort", "prompt_cache", "reasoning"],
},
...overrides,
};

View File

@ -29,7 +29,7 @@ interface DeepInfraModelMetadata {
context_length?: number;
max_tokens?: number;
pricing?: DeepInfraModelPricing;
/** e.g. ["vision", "reasoning_effort", "prompt_cache"] */
/** e.g. ["vision", "reasoning_effort", "prompt_cache", "reasoning"] */
tags?: string[];
}
@ -54,7 +54,9 @@ function parseModality(metadata: DeepInfraModelMetadata): Array<"text" | "image"
}
function parseReasoning(metadata: DeepInfraModelMetadata): boolean {
return metadata.tags?.includes("reasoning_effort") ?? false;
return (
(metadata.tags?.includes("reasoning_effort") || metadata.tags?.includes("reasoning")) ?? false
);
}
function toModelDefinition(entry: DeepInfraModelEntry): ModelDefinitionConfig {

View File

@ -29,7 +29,7 @@ export const DEEPINFRA_MODEL_CATALOG: DeepInfraModelCatalogEntry[] = [
{
id: "MiniMaxAI/MiniMax-M2.5",
name: "MiniMax M2.5",
reasoning: false,
reasoning: true,
input: ["text"],
contextWindow: 196608,
maxTokens: 196608,
@ -37,7 +37,7 @@ export const DEEPINFRA_MODEL_CATALOG: DeepInfraModelCatalogEntry[] = [
{
id: "zai-org/GLM-5",
name: "GLM 5",
reasoning: false,
reasoning: true,
input: ["text"],
contextWindow: 202752,
maxTokens: 202752,
@ -45,7 +45,7 @@ export const DEEPINFRA_MODEL_CATALOG: DeepInfraModelCatalogEntry[] = [
{
id: "moonshotai/Kimi-K2.5",
name: "Kimi K2.5",
reasoning: false,
reasoning: true,
input: ["text", "image"],
contextWindow: 262144,
maxTokens: 262144,