Fix the way we decide if a DeepInfra model has reasoning capabilities
This commit is contained in:
parent
905e98604d
commit
beb45cba22
@ -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,
|
||||
};
|
||||
|
||||
@ -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 {
|
||||
|
||||
@ -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,
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user