1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
|
// modules/core/model-profiles.js — per-model-class tool budgets.
// class is picked by an ordered rule list; the FIRST match wins, so keep specific ids above size rules.
const MODEL_CLASSES = {
tiny: { ctx: 8192, readPageChars: 4000, toolResultChars: 3000, toolContextBudgetChars: 12000, fetchChars: 3000 },
small: { ctx: 16384, readPageChars: 8000, toolResultChars: 4000, toolContextBudgetChars: 16000, fetchChars: 4000 },
mid: { ctx: 32768, readPageChars: 16000, toolResultChars: 6000, toolContextBudgetChars: 24000, fetchChars: 6000 },
large: { ctx: 65536, readPageChars: 32000, toolResultChars: 12000, toolContextBudgetChars: 48000, fetchChars: 12000 },
};
// Ordered: first match wins. Only ids we know; everything else falls through to the size parse.
const MODEL_IDS = {
'gemma4:31b-cloud': 'large',
'qwen3:0.6b': 'tiny',
'gemma2:2b': 'tiny',
'llama3.1:8b': 'small',
'gemma4:31b': 'mid',
};
// "235b" -> large, "31b" -> mid, "8b" -> small, "0.6b"/"2b" -> tiny
function classFor(modelId) {
const id = String(modelId || '').toLowerCase();
if (!id) return 'mid';
if (MODEL_IDS[id]) return MODEL_IDS[id];
if (id.endsWith('-cloud') || id.includes(':cloud')) return 'large';
const m = id.match(/(\d+(?:\.\d+)?)\s*b\b/);
if (!m) return 'mid';
const p = parseFloat(m[1]);
if (p <= 3) return 'tiny';
if (p <= 9) return 'small';
if (p <= 32) return 'mid';
return 'large';
}
function profileFor(modelId) {
return MODEL_CLASSES[classFor(modelId)] || MODEL_CLASSES.mid;
}
// The bundle concatenates this file FIRST, so this binding exists before any module method runs.
// module-aiassist/ollama-manager resolve budgets through it: a require() there resolved relative
// to whoever called them and always returned null (found by the live ai-config probe 2026-11-02).
const MODEL_PROFILES = { MODEL_CLASSES, classFor, profileFor };
if (typeof module !== 'undefined' && module.exports) module.exports = { MODEL_CLASSES, classFor, profileFor, MODEL_PROFILES };
|
