Mirror_Repos/Kilo-Org_kilocode
2
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

feat: add Claude extended thinking variants for Kilo Gateway (#86)

Browse Source 
Enable reasoning effort level cycling (Ctrl+T) for Claude models through
Kilo Gateway by:

- Adding @kilocode/kilo-gateway to sdkKey() to use 'openrouter' key for
  provider options
- Adding dedicated variants() case for Claude/Anthropic models with full
  effort levels (none, minimal, low, medium, high, xhigh)
- Supporting grok-3-mini reasoning variants through Kilo Gateway
- Fixing smallOptions() to use correct reasoning.effort format

OpenRouter maps effort levels to budget_tokens percentages:
- xhigh: 95%, high: 80%, medium: 50%, low: 20%, minimal: 10%, none: 0%
This commit is contained in:
Marius
2026-02-03 09:42:20 +01:00
committed by GitHub
parent b468a59726
commit 38c6b26db9
2 changed files with 114 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/opencode/src/provider/transform.ts
+17 -4
View File
@@ -34,6 +34,7 @@ export namespace ProviderTransform {
case "@ai-sdk/gateway":
return "gateway"
case "@openrouter/ai-sdk-provider":
case "@kilocode/kilo-gateway": // kilocode_change
return "openrouter"
}
return undefined
@@ -330,7 +331,8 @@ export namespace ProviderTransform {
// see: https://docs.x.ai/docs/guides/reasoning#control-how-hard-the-model-thinks
if (id.includes("grok") && id.includes("grok-3-mini")) {
if (model.api.npm === "@openrouter/ai-sdk-provider") {
if (model.api.npm === "@openrouter/ai-sdk-provider" || model.api.npm === "@kilocode/kilo-gateway") {
// kilocode_change
return {
low: { reasoning: { effort: "low" } },
high: { reasoning: { effort: "high" } },
@@ -344,8 +346,18 @@ export namespace ProviderTransform {
if (id.includes("grok")) return {}
switch (model.api.npm) {
// kilocode_change start
case "@kilocode/kilo-gateway":
// Claude/Anthropic models support reasoning via effort levels
// OpenRouter maps these to budget_tokens percentages (xhigh=95%, high=80%, medium=50%, low=20%, minimal=10%, none=0%)
if (model.id.includes("claude") || model.id.includes("anthropic")) {
return Object.fromEntries(OPENAI_EFFORTS.map((effort) => [effort, { reasoning: { effort } }]))
}
if (!model.id.includes("gpt") && !model.id.includes("gemini-3")) return {}
return Object.fromEntries(OPENAI_EFFORTS.map((effort) => [effort, { reasoning: { effort } }]))
// kilocode_change end
case "@openrouter/ai-sdk-provider":
case "@kilocode/kilo-gateway": // kilocode_change
if (!model.id.includes("gpt") && !model.id.includes("gemini-3")) return {}
return Object.fromEntries(OPENAI_EFFORTS.map((effort) => [effort, { reasoning: { effort } }]))
@@ -632,11 +644,12 @@ export namespace ProviderTransform {
}
return { thinkingConfig: { thinkingBudget: 0 } }
}
if (model.providerID === "openrouter") {
if (model.providerID === "openrouter" || model.api.npm === "@kilocode/kilo-gateway") {
// kilocode_change
if (model.api.id.includes("google")) {
return { reasoning: { enabled: false } }
}
return { reasoningEffort: "minimal" }
return { reasoning: { effort: "minimal" } } // kilocode_change - use reasoning.effort for OpenRouter API
}
return {}
}
packages/opencode/test/provider/transform.test.ts
+97
View File
@@ -1206,6 +1206,103 @@ describe("ProviderTransform.variants", () => {
})
})
// kilocode_change start
describe("@kilocode/kilo-gateway", () => {
test("claude models return OPENAI_EFFORTS with reasoning", () => {
const model = createMockModel({
id: "kilo/anthropic/claude-sonnet-4",
providerID: "kilo",
api: {
id: "anthropic/claude-sonnet-4",
url: "https://gateway.kilo.ai",
npm: "@kilocode/kilo-gateway",
},
})
const result = ProviderTransform.variants(model)
expect(Object.keys(result)).toEqual(["none", "minimal", "low", "medium", "high", "xhigh"])
expect(result.none).toEqual({ reasoning: { effort: "none" } })
expect(result.low).toEqual({ reasoning: { effort: "low" } })
expect(result.medium).toEqual({ reasoning: { effort: "medium" } })
expect(result.high).toEqual({ reasoning: { effort: "high" } })
expect(result.xhigh).toEqual({ reasoning: { effort: "xhigh" } })
})
test("anthropic models return OPENAI_EFFORTS with reasoning", () => {
const model = createMockModel({
id: "kilo/anthropic/claude-opus-4",
providerID: "kilo",
api: {
id: "anthropic/claude-opus-4",
url: "https://gateway.kilo.ai",
npm: "@kilocode/kilo-gateway",
},
})
const result = ProviderTransform.variants(model)
expect(Object.keys(result)).toEqual(["none", "minimal", "low", "medium", "high", "xhigh"])
expect(result.low).toEqual({ reasoning: { effort: "low" } })
})
test("gpt models return OPENAI_EFFORTS with reasoning", () => {
const model = createMockModel({
id: "kilo/openai/gpt-5",
providerID: "kilo",
api: {
id: "openai/gpt-5",
url: "https://gateway.kilo.ai",
npm: "@kilocode/kilo-gateway",
},
})
const result = ProviderTransform.variants(model)
expect(Object.keys(result)).toEqual(["none", "minimal", "low", "medium", "high", "xhigh"])
expect(result.low).toEqual({ reasoning: { effort: "low" } })
})
test("gemini-3 models return OPENAI_EFFORTS with reasoning", () => {
const model = createMockModel({
id: "kilo/google/gemini-3-pro",
providerID: "kilo",
api: {
id: "google/gemini-3-pro",
url: "https://gateway.kilo.ai",
npm: "@kilocode/kilo-gateway",
},
})
const result = ProviderTransform.variants(model)
expect(Object.keys(result)).toEqual(["none", "minimal", "low", "medium", "high", "xhigh"])
})
test("non-qualifying models return empty object", () => {
const model = createMockModel({
id: "kilo/meta/llama-4",
providerID: "kilo",
api: {
id: "meta/llama-4",
url: "https://gateway.kilo.ai",
npm: "@kilocode/kilo-gateway",
},
})
const result = ProviderTransform.variants(model)
expect(result).toEqual({})
})
test("grok-3-mini returns low and high with reasoning", () => {
const model = createMockModel({
id: "kilo/x-ai/grok-3-mini",
providerID: "kilo",
api: {
id: "x-ai/grok-3-mini",
url: "https://gateway.kilo.ai",
npm: "@kilocode/kilo-gateway",
},
})
const result = ProviderTransform.variants(model)
expect(Object.keys(result)).toEqual(["low", "high"])
expect(result.low).toEqual({ reasoning: { effort: "low" } })
expect(result.high).toEqual({ reasoning: { effort: "high" } })
})
})
// kilocode_change end
describe("@ai-sdk/gateway", () => {
test("returns OPENAI_EFFORTS with reasoningEffort", () => {
const model = createMockModel({