feat: set horizon-alpha model max tokens to 32k for OpenRouter (#6470)

Co-authored-by: Roo Code <roomote@roocode.com>
This commit is contained in:
roomote[bot] 2025-07-30 23:27:18 -04:00 committed by GitHub
parent 01f5320b4d
commit c47e857048
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 54 additions and 1 deletions

View file

@ -11,7 +11,7 @@ import {
OPEN_ROUTER_REQUIRED_REASONING_BUDGET_MODELS,
} from "@roo-code/types"
import { getOpenRouterModelEndpoints, getOpenRouterModels } from "../openrouter"
import { getOpenRouterModelEndpoints, getOpenRouterModels, parseOpenRouterModel } from "../openrouter"
nockBack.fixtures = path.join(__dirname, "fixtures")
nockBack.setMode("lockdown")
@ -251,4 +251,52 @@ describe("OpenRouter API", () => {
nockDone()
})
})
describe("parseOpenRouterModel", () => {
it("sets horizon-alpha model to 32k max tokens", () => {
const mockModel = {
name: "Horizon Alpha",
description: "Test model",
context_length: 128000,
max_completion_tokens: 128000,
pricing: {
prompt: "0.000003",
completion: "0.000015",
},
}
const result = parseOpenRouterModel({
id: "openrouter/horizon-alpha",
model: mockModel,
modality: "text",
maxTokens: 128000,
})
expect(result.maxTokens).toBe(32768)
expect(result.contextWindow).toBe(128000)
})
it("does not override max tokens for other models", () => {
const mockModel = {
name: "Other Model",
description: "Test model",
context_length: 128000,
max_completion_tokens: 64000,
pricing: {
prompt: "0.000003",
completion: "0.000015",
},
}
const result = parseOpenRouterModel({
id: "openrouter/other-model",
model: mockModel,
modality: "text",
maxTokens: 64000,
})
expect(result.maxTokens).toBe(64000)
expect(result.contextWindow).toBe(128000)
})
})
})

View file

@ -232,5 +232,10 @@ export const parseOpenRouterModel = ({
modelInfo.maxTokens = anthropicModels["claude-3-7-sonnet-20250219:thinking"].maxTokens
}
// Set horizon-alpha model to 32k max tokens
if (id === "openrouter/horizon-alpha") {
modelInfo.maxTokens = 32768
}
return modelInfo
}