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

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
@@ -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 } },
})
})