feat(cli-tools): configure Grok Build subagent models
Add separate model selectors for Grok Build main, general-purpose, explore, and plan agents. Each override gets an independent 9Router custom-model slot and context_window derived from 9Router model capabilities. Preserve and restore pre-existing config on reset.
This commit is contained in:
rixzkiye
247
src/lib/grokBuildConfig.js
Normal file
247
src/lib/grokBuildConfig.js
Normal file
@@ -0,0 +1,247 @@
|
||||
export const GROK_MAIN_MODEL_SLOT = "9router";
|
||||
export const GROK_BUILTIN_DEFAULT = "grok-build";
|
||||
export const GROK_SUBAGENT_TYPES = ["general-purpose", "explore", "plan"];
|
||||
|
||||
const UNSET_SENTINEL = "__9router_unset__";
|
||||
const MODELS_SECTION = "models";
|
||||
const SUBAGENT_MODELS_SECTION = "subagents.models";
|
||||
|
||||
const escapeRegExp = (value) => value.replace(/[.*+?^${}()|[\]\\]/g, "\\$&");
|
||||
const tomlString = (value) => JSON.stringify(String(value));
|
||||
|
||||
const sectionRegExp = (section) =>
|
||||
new RegExp(
|
||||
`^\\[${escapeRegExp(section)}\\][ \\t]*\\r?\\n((?:(?!\\[)[^\\r\\n]*\\r?\\n?)*)`,
|
||||
"m",
|
||||
);
|
||||
|
||||
const modelSlot = (type) => `${GROK_MAIN_MODEL_SLOT}-${type}`;
|
||||
|
||||
const previousDefaultRegExp = /^# 9router-prev-default = "([^"]*)"[ \t]*\r?\n?/m;
|
||||
const previousSubagentRegExp = (type) =>
|
||||
new RegExp(
|
||||
`^# 9router-prev-subagent-${escapeRegExp(type)} = "([^"]*)"[ \\t]*\\r?\\n?`,
|
||||
"m",
|
||||
);
|
||||
|
||||
function getSectionField(toml, section, key) {
|
||||
const match = toml.match(sectionRegExp(section));
|
||||
if (!match) return null;
|
||||
const field = match[1].match(
|
||||
new RegExp(`^[ \\t]*${escapeRegExp(key)}[ \\t]*=[ \\t]*"([^"]*)"`, "m"),
|
||||
);
|
||||
return field ? field[1] : null;
|
||||
}
|
||||
|
||||
function getSectionNumber(toml, section, key) {
|
||||
const match = toml.match(sectionRegExp(section));
|
||||
if (!match) return null;
|
||||
const field = match[1].match(
|
||||
new RegExp(`^[ \\t]*${escapeRegExp(key)}[ \\t]*=[ \\t]*([0-9]+(?:\\.[0-9]+)?)`, "m"),
|
||||
);
|
||||
if (!field) return null;
|
||||
const value = Number(field[1]);
|
||||
return Number.isFinite(value) ? value : null;
|
||||
}
|
||||
|
||||
function setSectionField(toml, section, key, value) {
|
||||
const match = toml.match(sectionRegExp(section));
|
||||
const line = `${key} = ${tomlString(value)}`;
|
||||
if (!match) {
|
||||
const prefix = toml.length > 0 && !toml.endsWith("\n") ? `${toml}\n` : toml;
|
||||
return `${prefix}\n[${section}]\n${line}\n`;
|
||||
}
|
||||
|
||||
const body = match[1] || "";
|
||||
const fieldRegExp = new RegExp(
|
||||
`^[ \\t]*${escapeRegExp(key)}[ \\t]*=[ \\t]*"[^"]*"`,
|
||||
"m",
|
||||
);
|
||||
const nextBody = fieldRegExp.test(body)
|
||||
? body.replace(fieldRegExp, line)
|
||||
: `${line}\n${body}`;
|
||||
return toml.replace(match[0], `[${section}]\n${nextBody}`);
|
||||
}
|
||||
|
||||
function deleteSectionField(toml, section, key) {
|
||||
const match = toml.match(sectionRegExp(section));
|
||||
if (!match) return toml;
|
||||
const fieldRegExp = new RegExp(
|
||||
`^[ \\t]*${escapeRegExp(key)}[ \\t]*=[^\\r\\n]*\\r?\\n?`,
|
||||
"m",
|
||||
);
|
||||
const nextBody = (match[1] || "").replace(fieldRegExp, "");
|
||||
if (!nextBody.trim()) return toml.replace(match[0], "").replace(/\n{3,}/g, "\n\n");
|
||||
return toml.replace(match[0], `[${section}]\n${nextBody}`);
|
||||
}
|
||||
|
||||
function parseModelSection(toml, slot) {
|
||||
const match = toml.match(sectionRegExp(`model.${slot}`));
|
||||
if (!match) return null;
|
||||
const body = match[1] || "";
|
||||
const contextWindow = getSectionNumber(toml, `model.${slot}`, "context_window");
|
||||
return {
|
||||
model: getSectionField(toml, `model.${slot}`, "model"),
|
||||
base_url: getSectionField(toml, `model.${slot}`, "base_url"),
|
||||
name: getSectionField(toml, `model.${slot}`, "name"),
|
||||
api_key: getSectionField(toml, `model.${slot}`, "api_key"),
|
||||
api_backend: getSectionField(toml, `model.${slot}`, "api_backend"),
|
||||
context_window: Number.isFinite(contextWindow) && contextWindow > 0 ? contextWindow : null,
|
||||
raw: body,
|
||||
};
|
||||
}
|
||||
|
||||
function buildModelSection({ slot, model, baseUrl, apiKey, contextWindow, name }) {
|
||||
const lines = [
|
||||
`[model.${slot}]`,
|
||||
`model = ${tomlString(model)}`,
|
||||
`base_url = ${tomlString(baseUrl)}`,
|
||||
`name = ${tomlString(name)}`,
|
||||
`description = ${tomlString("Routed via 9Router gateway")}`,
|
||||
`api_backend = "chat_completions"`,
|
||||
];
|
||||
if (apiKey) lines.push(`api_key = ${tomlString(apiKey)}`);
|
||||
if (Number.isFinite(contextWindow) && contextWindow > 0) {
|
||||
lines.push(`context_window = ${Math.floor(contextWindow)}`);
|
||||
}
|
||||
return `${lines.join("\n")}\n`;
|
||||
}
|
||||
|
||||
function upsertModelSection(toml, config) {
|
||||
const regexp = sectionRegExp(`model.${config.slot}`);
|
||||
const section = buildModelSection(config);
|
||||
if (regexp.test(toml)) return toml.replace(regexp, section);
|
||||
const prefix = toml.length > 0 && !toml.endsWith("\n") ? `${toml}\n` : toml;
|
||||
return `${prefix}\n${section}`;
|
||||
}
|
||||
|
||||
function removeModelSection(toml, slot) {
|
||||
return toml.replace(sectionRegExp(`model.${slot}`), "").replace(/\n{3,}/g, "\n\n");
|
||||
}
|
||||
|
||||
function insertMarker(toml, marker) {
|
||||
const mainSection = sectionRegExp(`model.${GROK_MAIN_MODEL_SLOT}`);
|
||||
if (mainSection.test(toml)) {
|
||||
return toml.replace(mainSection, (section) => `${marker}${section}`);
|
||||
}
|
||||
const prefix = toml.length > 0 && !toml.endsWith("\n") ? `${toml}\n` : toml;
|
||||
return `${prefix}${marker}`;
|
||||
}
|
||||
|
||||
function rememberPreviousDefault(toml) {
|
||||
if (previousDefaultRegExp.test(toml)) return toml;
|
||||
const current = getSectionField(toml, MODELS_SECTION, "default");
|
||||
if (!current || current === GROK_MAIN_MODEL_SLOT) return toml;
|
||||
return insertMarker(toml, `# 9router-prev-default = ${tomlString(current)}\n`);
|
||||
}
|
||||
|
||||
function restorePreviousDefault(toml) {
|
||||
const previous = toml.match(previousDefaultRegExp)?.[1] || GROK_BUILTIN_DEFAULT;
|
||||
let next = toml.replace(previousDefaultRegExp, "");
|
||||
if (getSectionField(next, MODELS_SECTION, "default") === GROK_MAIN_MODEL_SLOT) {
|
||||
next = setSectionField(next, MODELS_SECTION, "default", previous);
|
||||
}
|
||||
return next;
|
||||
}
|
||||
|
||||
function rememberPreviousSubagent(toml, type) {
|
||||
const regexp = previousSubagentRegExp(type);
|
||||
if (regexp.test(toml)) return toml;
|
||||
const current = getSectionField(toml, SUBAGENT_MODELS_SECTION, type);
|
||||
const previous = current == null ? UNSET_SENTINEL : current;
|
||||
return insertMarker(
|
||||
toml,
|
||||
`# 9router-prev-subagent-${type} = ${tomlString(previous)}\n`,
|
||||
);
|
||||
}
|
||||
|
||||
function restorePreviousSubagent(toml, type) {
|
||||
const regexp = previousSubagentRegExp(type);
|
||||
const previous = toml.match(regexp)?.[1] || UNSET_SENTINEL;
|
||||
let next = toml.replace(regexp, "");
|
||||
if (getSectionField(next, SUBAGENT_MODELS_SECTION, type) !== modelSlot(type)) {
|
||||
return next;
|
||||
}
|
||||
if (previous === UNSET_SENTINEL) {
|
||||
return deleteSectionField(next, SUBAGENT_MODELS_SECTION, type);
|
||||
}
|
||||
return setSectionField(next, SUBAGENT_MODELS_SECTION, type, previous);
|
||||
}
|
||||
|
||||
export function parseGrokBuildConfig(toml) {
|
||||
const subagentModels = {};
|
||||
const subagentMappings = {};
|
||||
for (const type of GROK_SUBAGENT_TYPES) {
|
||||
const mapping = getSectionField(toml, SUBAGENT_MODELS_SECTION, type);
|
||||
subagentMappings[type] = mapping;
|
||||
subagentModels[type] = mapping === modelSlot(type)
|
||||
? parseModelSection(toml, mapping)
|
||||
: null;
|
||||
}
|
||||
|
||||
return {
|
||||
model: parseModelSection(toml, GROK_MAIN_MODEL_SLOT),
|
||||
default: getSectionField(toml, MODELS_SECTION, "default"),
|
||||
subagentModels,
|
||||
subagentMappings,
|
||||
};
|
||||
}
|
||||
|
||||
/**
|
||||
* Apply main model and optional per-type subagent overrides while preserving all unrelated TOML.
|
||||
* `subagentModels === undefined` leaves existing subagent config untouched for API compatibility.
|
||||
*/
|
||||
export function applyGrokBuildConfig(
|
||||
toml,
|
||||
{ baseUrl, apiKey, model, contextWindow, subagentModels },
|
||||
) {
|
||||
let next = rememberPreviousDefault(toml);
|
||||
next = upsertModelSection(next, {
|
||||
slot: GROK_MAIN_MODEL_SLOT,
|
||||
model,
|
||||
baseUrl,
|
||||
apiKey,
|
||||
contextWindow,
|
||||
name: "9Router",
|
||||
});
|
||||
next = setSectionField(next, MODELS_SECTION, "default", GROK_MAIN_MODEL_SLOT);
|
||||
|
||||
if (subagentModels && typeof subagentModels === "object") {
|
||||
for (const type of GROK_SUBAGENT_TYPES) {
|
||||
const selected = subagentModels[type];
|
||||
const slot = modelSlot(type);
|
||||
if (selected?.model) {
|
||||
next = rememberPreviousSubagent(next, type);
|
||||
next = upsertModelSection(next, {
|
||||
slot,
|
||||
model: selected.model,
|
||||
baseUrl,
|
||||
apiKey,
|
||||
contextWindow: selected.contextWindow,
|
||||
name: `9Router ${type}`,
|
||||
});
|
||||
next = setSectionField(next, SUBAGENT_MODELS_SECTION, type, slot);
|
||||
} else {
|
||||
next = restorePreviousSubagent(next, type);
|
||||
next = removeModelSection(next, slot);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return next;
|
||||
}
|
||||
|
||||
export function resetGrokBuildConfig(toml) {
|
||||
let next = toml;
|
||||
for (const type of GROK_SUBAGENT_TYPES) {
|
||||
next = restorePreviousSubagent(next, type);
|
||||
next = removeModelSection(next, modelSlot(type));
|
||||
}
|
||||
next = removeModelSection(next, GROK_MAIN_MODEL_SLOT);
|
||||
next = restorePreviousDefault(next);
|
||||
return next.replace(/\n{3,}/g, "\n\n");
|
||||
}
|
||||
|
||||
export function getGrokSubagentSlot(type) {
|
||||
return GROK_SUBAGENT_TYPES.includes(type) ? modelSlot(type) : null;
|
||||
}
|
||||
Reference in New Issue
Block a user