Skip to content

feat(auto): add Kilo Auto Balanced model #1031

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
chrarnoldus merged 9 commits into from
Mar 12, 2026
Merged

feat(auto): add Kilo Auto Balanced model #1031

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
9 commits
Select commit Hold shift + click to select a range
f0f9afd
feat(auto): add Kilo Auto Balanced model
lambertjosh Mar 11, 2026
2ea7079
fix(auto): filter models without legacy mapping and update test snapshot
lambertjosh Mar 11, 2026
1a11521
Apply suggestion from @lambertjosh
lambertjosh Mar 11, 2026
18cfc8a
Apply suggestion from @lambertjosh
lambertjosh Mar 11, 2026
3baf980
fix(auto): add tool restrictions to balanced model
lambertjosh Mar 11, 2026
5ee2cbc
fix: restore nvidia nemotron in preferredModels and update snapshot
lambertjosh Mar 11, 2026
4f8149e
merge: resolve conflict with main, keep bumped preferredIndex values
lambertjosh Mar 11, 2026
cf60cff
* Remove reference to disabled free model
chrarnoldus Mar 12, 2026
e33985d
Approvals
chrarnoldus Mar 12, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
79 changes: 66 additions & 13 deletions src/lib/kilo-auto-model.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -51,6 +51,22 @@ export const KILO_AUTO_FREE_MODEL: AutoModel = {
opencode_settings: undefined,
};

export const KILO_AUTO_BALANCED_MODEL: AutoModel = {
id: 'kilo-auto/balanced',
name: 'Kilo Auto Balanced',
description: 'Automatically routes your request for a balanced mix of price and performance.',
context_length: 204800,
max_completion_tokens: 131072,
Comment thread
chrarnoldus marked this conversation as resolved.
prompt_price: '0.0000006',
completion_price: '0.000003',
supports_images: false,
roocode_settings: {
included_tools: ['edit_file'],
excluded_tools: ['apply_diff'],
},
opencode_settings: undefined,
};

export const KILO_AUTO_SMALL_MODEL: AutoModel = {
id: 'kilo-auto/small',
name: 'Kilo Auto Small',
Expand All @@ -67,7 +83,12 @@ export const KILO_AUTO_SMALL_MODEL: AutoModel = {
},
};

export const AUTO_MODELS = [KILO_AUTO_FRONTIER_MODEL, KILO_AUTO_FREE_MODEL, KILO_AUTO_SMALL_MODEL];
export const AUTO_MODELS = [
KILO_AUTO_FRONTIER_MODEL,
KILO_AUTO_BALANCED_MODEL,
Comment thread
chrarnoldus marked this conversation as resolved.
KILO_AUTO_FREE_MODEL,
KILO_AUTO_SMALL_MODEL,
];

export function isKiloAutoModel(model: string) {
return AUTO_MODELS.some(m => m.id === model) || legacyMapping[model] !== undefined;
Expand All @@ -79,15 +100,15 @@ type ResolvedAutoModel = {
verbosity?: Verbosity;
};

const CODE_MODEL: ResolvedAutoModel = {
const FRONTIER_CODE_MODEL: ResolvedAutoModel = {
model: CLAUDE_SONNET_CURRENT_MODEL_ID,
reasoning: { enabled: true },
verbosity: 'low',
};

// Mode → model mappings for kilo-auto/frontier routing.
// Add/remove/modify entries here to change routing behavior.
const MODE_TO_MODEL = new Map<string, ResolvedAutoModel>([
const FRONTIER_MODE_TO_MODEL = new Map<string, ResolvedAutoModel>([
// Opus modes (planning, reasoning, orchestration, debugging)
[
'plan',
Expand Down Expand Up @@ -119,7 +140,31 @@ const MODE_TO_MODEL = new Map<string, ResolvedAutoModel>([
'explore',
{ model: CLAUDE_SONNET_CURRENT_MODEL_ID, reasoning: { enabled: true }, verbosity: 'medium' },
],
['code', CODE_MODEL],
['code', FRONTIER_CODE_MODEL],
]);

const KIMI_K25_MODEL_ID = 'moonshotai/kimi-k2.5';

const MINIMAX_M25_MODEL_ID = minimax_m25_free_model.is_enabled
? minimax_m25_free_model.public_id
Comment thread
chrarnoldus marked this conversation as resolved.
: 'minimax/minimax-m2.5';

const BALANCED_CODE_MODEL: ResolvedAutoModel = {
model: MINIMAX_M25_MODEL_ID,
};

// Mode → model mappings for kilo-auto/balanced routing.
// Uses Kimi K2.5 where Frontier uses Opus, Minimax M2.5 where Frontier uses Sonnet.
const BALANCED_MODE_TO_MODEL = new Map<string, ResolvedAutoModel>([
['plan', { model: KIMI_K25_MODEL_ID, reasoning: { enabled: true } }],
['general', { model: KIMI_K25_MODEL_ID, reasoning: { enabled: true } }],
['architect', { model: KIMI_K25_MODEL_ID, reasoning: { enabled: true } }],
['orchestrator', { model: KIMI_K25_MODEL_ID, reasoning: { enabled: true } }],
['ask', { model: KIMI_K25_MODEL_ID, reasoning: { enabled: true } }],
['debug', { model: KIMI_K25_MODEL_ID, reasoning: { enabled: true } }],
['build', { model: MINIMAX_M25_MODEL_ID }],
['explore', { model: MINIMAX_M25_MODEL_ID }],
['code', BALANCED_CODE_MODEL],
]);

const legacyMapping: Record<string, string | undefined> = {
Expand All @@ -128,14 +173,19 @@ const legacyMapping: Record<string, string | undefined> = {
'kilo/auto-small': KILO_AUTO_SMALL_MODEL.id,
};

export function deprecatedAutoModelsToPreventNewExtensionModelPickerFromGettingStuck() {
const mapping = Object.fromEntries(Object.entries(legacyMapping).map(([a, b]) => [b, a]));
return AUTO_MODELS.map(m => ({
...m,
id: mapping[m.id],
name: 'Deprecated: ' + m.name,
description: `${mapping[m.id]} is deprecated, use ${m.id} instead`,
}));
export function deprecatedAutoModelsToPreventNewExtensionModelPickerFromGettingStuck(): AutoModel[] {
return Object.entries(legacyMapping)
.map(([legacyId, currentId]) => {
const model = AUTO_MODELS.find(m => m.id === currentId);
if (!model) return null;
return {
...model,
id: legacyId,
name: 'Deprecated: ' + model.name,
description: `${legacyId} is deprecated, use ${model.id} instead`,
};
})
.filter(m => m !== null);
}

export function resolveAutoModel(model: string, modeHeader: string | null): ResolvedAutoModel {
Expand All @@ -147,5 +197,8 @@ export function resolveAutoModel(model: string, modeHeader: string | null): Reso
return { model: 'openai/gpt-5-nano' };
}
const mode = modeHeader?.trim().toLowerCase() ?? '';
return MODE_TO_MODEL.get(mode) ?? CODE_MODEL;
if (mappedModel === KILO_AUTO_BALANCED_MODEL.id) {
return BALANCED_MODE_TO_MODEL.get(mode) ?? BALANCED_CODE_MODEL;
}
return FRONTIER_MODE_TO_MODEL.get(mode) ?? FRONTIER_CODE_MODEL;
}
7 changes: 6 additions & 1 deletion src/lib/models.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,11 @@
* Utility functions for working with AI models
*/

import { KILO_AUTO_FREE_MODEL, KILO_AUTO_FRONTIER_MODEL } from '@/lib/kilo-auto-model';
import {
KILO_AUTO_BALANCED_MODEL,
KILO_AUTO_FREE_MODEL,
KILO_AUTO_FRONTIER_MODEL,
} from '@/lib/kilo-auto-model';
import {
CLAUDE_OPUS_CURRENT_MODEL_ID,
CLAUDE_SONNET_CURRENT_MODEL_ID,
Expand All @@ -20,6 +24,7 @@ export const PRIMARY_DEFAULT_MODEL = CLAUDE_SONNET_CURRENT_MODEL_ID;

export const preferredModels = [
KILO_AUTO_FRONTIER_MODEL.id,
KILO_AUTO_BALANCED_MODEL.id,
KILO_AUTO_FREE_MODEL.id,
'nvidia/nemotron-3-super-120b-a12b:free',
minimax_m25_free_model.is_enabled ? minimax_m25_free_model.public_id : 'minimax/minimax-m2.5',
Expand Down
51 changes: 48 additions & 3 deletions src/tests/openrouter-models-sorting.approved.json
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -42,6 +42,51 @@
},
"preferredIndex": 0
},
{
"id": "kilo-auto/balanced",
"name": "Kilo Auto Balanced",
"created": 0,
"description": "Automatically routes your request for a balanced mix of price and performance.",
"architecture": {
"input_modalities": [
"text"
],
"output_modalities": [
"text"
],
"tokenizer": "Other"
},
"top_provider": {
"is_moderated": false,
"context_length": 204800,
"max_completion_tokens": 131072
},
"pricing": {
"prompt": "0.0000006",
"completion": "0.000003",
"request": "0",
"image": "0",
"web_search": "0",
"internal_reasoning": "0"
},
"context_length": 204800,
"supported_parameters": [
"max_tokens",
"temperature",
"tools",
"reasoning",
"include_reasoning"
],
"settings": {
"included_tools": [
"edit_file"
],
"excluded_tools": [
"apply_diff"
]
},
"preferredIndex": 1
},
{
"id": "kilo-auto/free",
"name": "Kilo Auto Free",
Expand Down Expand Up @@ -86,7 +131,7 @@
"edit_file"
]
},
"preferredIndex": 1
"preferredIndex": 2
},
{
"id": "minimax/minimax-m2.5:free",
Expand Down Expand Up @@ -130,7 +175,7 @@
"include_reasoning"
],
"default_parameters": {},
"preferredIndex": 3,
"preferredIndex": 4,
"settings": {
"included_tools": [
"search_and_replace"
Expand Down Expand Up @@ -184,7 +229,7 @@
"include_reasoning"
],
"default_parameters": {},
"preferredIndex": 5,
"preferredIndex": 6,
"versioned_settings": {
"4.146.0": {
"included_tools": [
Expand Down
Loading