furyhawk/opencode
1
0
Fork 0
mirror of https://github.com/anomalyco/opencode.git synced 2026-07-21 02:06:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(provider): tweak reasoning option driven max budget variants (#36626)

Browse Source
This commit is contained in:
Aiden Cline
2026-07-13 09:18:32 -05:00
committed by GitHub
parent b3a012cbdd
commit 99668cfdce
2 changed files with 35 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/opencode/src/provider/transform.ts
+4 -8
View File
@@ -1611,17 +1611,13 @@ function effortVariants(model: Provider.Model, values: readonly unknown[]) {
}
function budgetVariants(model: Provider.Model, min?: number, max?: number) {
const limit = model.limit.output - 1
if (limit <= 0) return {}
const high = Math.min(
max === undefined ? Math.max(min ?? 0, 16_000) : Math.min(Math.max(min ?? 0, 16_000), max),
limit,
)
const maximum = max === undefined ? undefined : Math.min(max, limit)
const maximum = Math.min(max ?? OUTPUT_TOKEN_MAX - 1, model.limit.output - 1, OUTPUT_TOKEN_MAX - 1)
if (maximum <= 0) return {}
const high = Math.min(Math.max(min ?? 0, Math.floor((maximum + 1) / 2)), maximum)
return Object.fromEntries(
[
{ id: "high", budget: high },
...(maximum === undefined || maximum === high ? [] : [{ id: "max", budget: maximum }]),
{ id: "max", budget: maximum },
].flatMap((item) => {
const settings = reasoningBudget(model, item.budget)
return settings ? [[item.id, settings]] : []
packages/opencode/test/provider/transform.test.ts
+31 -5
View File
@@ -3111,9 +3111,9 @@ describe("ProviderTransform.reasoningVariants", () => {
["@ai-sdk/cohere", { thinking: { type: "enabled", tokenBudget: 16_000 } }],
["@ai-sdk/alibaba", { enableThinking: true, thinkingBudget: 16_000 }],
])("converts token budgets for %s", (npm, high) => {
expect(
ProviderTransform.reasoningVariants(model([{ type: "budget_tokens", min: 1_024, max: 16_000 }]), target(npm)),
).toEqual({ high })
const variants = ProviderTransform.reasoningVariants(model([{ type: "budget_tokens", min: 1_024 }]), target(npm))
expect(variants?.high).toEqual(high)
expect(Object.keys(variants ?? {})).toEqual(["high", "max"])
})
test("maps null effort to none", () => {
@@ -3150,6 +3150,7 @@ describe("ProviderTransform.reasoningVariants", () => {
).toEqual({
none: { thinking: { type: "disabled" } },
high: { thinking: { type: "enabled", tokenBudget: 16_000 } },
max: { thinking: { type: "enabled", tokenBudget: 31_999 } },
})
})
@@ -3161,7 +3162,7 @@ describe("ProviderTransform.reasoningVariants", () => {
),
).toEqual({
high: { thinking: { type: "enabled", budgetTokens: 16_000 } },
max: { thinking: { type: "enabled", budgetTokens: 63_999 } },
max: { thinking: { type: "enabled", budgetTokens: 31_999 } },
})
})
@@ -3171,7 +3172,32 @@ describe("ProviderTransform.reasoningVariants", () => {
expect(
ProviderTransform.reasoningVariants(model([{ type: "budget_tokens", min: 1_024, max: 64_000 }]), anthropic),
).toEqual({
high: { thinking: { type: "enabled", budgetTokens: 4_999 } },
high: { thinking: { type: "enabled", budgetTokens: 2_500 } },
max: { thinking: { type: "enabled", budgetTokens: 4_999 } },
})
})
test("derives high and max budgets when models.dev omits max", () => {
expect(
ProviderTransform.reasoningVariants(
model([{ type: "budget_tokens", min: 1_024 }]),
target("@ai-sdk/anthropic", "claude-haiku-4-5"),
),
).toEqual({
high: { thinking: { type: "enabled", budgetTokens: 16_000 } },
max: { thinking: { type: "enabled", budgetTokens: 31_999 } },
})
})
test("preserves explicit inclusive budget maxima", () => {
expect(
ProviderTransform.reasoningVariants(
model([{ type: "budget_tokens", min: 1_024, max: 24_576 }]),
target("@ai-sdk/google", "gemini-2.5-pro"),
),
).toEqual({
high: { thinkingConfig: { includeThoughts: true, thinkingBudget: 12_288 } },
max: { thinkingConfig: { includeThoughts: true, thinkingBudget: 24_576 } },
})
})