lmin/zmo-cli
1
0
Fork 0
Code Issues Pull Requests Actions 6 Packages Projects Releases Wiki Activity

cleanup: make smallOptions rely on variants (#27390)

Browse Source
This commit is contained in:
Aiden Cline
2026-05-13 18:30:36 -05:00
committed by GitHub
parent 55e0af1405
commit df3895d74f
2 changed files with 75 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

84
packages/opencode/test/provider/transform.test.ts
View File

@@ -3602,8 +3602,8 @@ describe("ProviderTransform.variants", () => {
})
describe("ProviderTransform.smallOptions - gpt-5 chat/search", () => {
const createModel = (apiId: string) =>
({
const createModel = (apiId: string) => {
const model = {
id: `openai/${apiId}`,
providerID: "openai",
api: {
@@ -3611,13 +3611,43 @@ describe("ProviderTransform.smallOptions - gpt-5 chat/search", () => {
url: "https://api.openai.com",
npm: "@ai-sdk/openai",
},
}) as any
capabilities: { reasoning: true },
limit: { output: 64_000 },
release_date: "2026-01-01",
} as any
model.variants = ProviderTransform.variants(model)
return model
}
for (const testCase of [
{ id: "gpt-5-chat-latest", options: { store: false } },
{ id: "gpt-5.1-chat-latest", options: { store: false, reasoningEffort: "medium" } },
{ id: "gpt-5.2-chat-latest", options: { store: false, reasoningEffort: "medium" } },
{ id: "gpt-5-search-api", options: { store: false } },
{
id: "gpt-5.1-chat-latest",
options: {
store: false,
reasoningEffort: "medium",
reasoningSummary: "auto",
include: ["reasoning.encrypted_content"],
},
},
{
id: "gpt-5.2-chat-latest",
options: {
store: false,
reasoningEffort: "medium",
reasoningSummary: "auto",
include: ["reasoning.encrypted_content"],
},
},
{
id: "gpt-5-search-api",
options: {
store: false,
reasoningEffort: "none",
reasoningSummary: "auto",
include: ["reasoning.encrypted_content"],
},
},
]) {
test(`${testCase.id} returns only supported small options`, () => {
expect(ProviderTransform.smallOptions(createModel(testCase.id))).toEqual(testCase.options)
@@ -3626,8 +3656,8 @@ describe("ProviderTransform.smallOptions - gpt-5 chat/search", () => {
})
describe("ProviderTransform.smallOptions - google thinking controls", () => {
const createGoogleModel = (apiId: string) =>
({
const createGoogleModel = (apiId: string) => {
const model = {
id: `google/${apiId}`,
providerID: "google",
api: {
@@ -3635,20 +3665,44 @@ describe("ProviderTransform.smallOptions - google thinking controls", () => {
url: "https://generativelanguage.googleapis.com",
npm: "@ai-sdk/google",
},
}) as any
capabilities: { reasoning: true },
limit: { output: 64_000 },
} as any
model.variants = ProviderTransform.variants(model)
return model
}
for (const testCase of [
{ id: "gemini-3-pro-preview", options: { thinkingConfig: { thinkingLevel: "low" } } },
{ id: "gemini-3-flash-preview", options: { thinkingConfig: { thinkingLevel: "minimal" } } },
{ id: "gemini-3.1-flash-image-preview", options: { thinkingConfig: { thinkingLevel: "minimal" } } },
{ id: "gemini-3-pro-image-preview", options: { thinkingConfig: { thinkingLevel: "high" } } },
{ id: "gemini-2.5-pro", options: { thinkingConfig: { thinkingBudget: 128 } } },
{ id: "gemini-2.5-flash", options: { thinkingConfig: { thinkingBudget: 0 } } },
{ id: "gemini-3-pro-preview", options: { thinkingConfig: { includeThoughts: true, thinkingLevel: "low" } } },
{ id: "gemini-3-flash-preview", options: { thinkingConfig: { includeThoughts: true, thinkingLevel: "minimal" } } },
{
id: "gemini-3.1-flash-image-preview",
options: { thinkingConfig: { includeThoughts: true, thinkingLevel: "minimal" } },
},
{ id: "gemini-3-pro-image-preview", options: { thinkingConfig: { includeThoughts: true, thinkingLevel: "high" } } },
{ id: "gemini-2.5-pro", options: { thinkingConfig: { includeThoughts: true, thinkingBudget: 16000 } } },
{ id: "gemini-2.5-flash", options: { thinkingConfig: { includeThoughts: true, thinkingBudget: 16000 } } },
]) {
test(`${testCase.id} returns supported small thinking options`, () => {
expect(ProviderTransform.smallOptions(createGoogleModel(testCase.id))).toEqual(testCase.options)
})
}
test("uses the first configured variant when available", () => {
expect(
ProviderTransform.smallOptions({
...createGoogleModel("gemini-2.5-pro"),
variants: {
high: { thinkingConfig: { includeThoughts: true, thinkingBudget: 16000 } },
max: { thinkingConfig: { includeThoughts: true, thinkingBudget: 32768 } },
},
}),
).toEqual({ thinkingConfig: { includeThoughts: true, thinkingBudget: 16000 } })
})
test("does not synthesize thinking options when variants are empty", () => {
expect(ProviderTransform.smallOptions({ ...createGoogleModel("gemini-2.5-pro"), variants: {} })).toEqual({})
})
})
describe("ProviderTransform.providerOptions - ai-gateway-provider", () => {