32 lines
934 B
TypeScript
32 lines
934 B
TypeScript
import { CONFIG_PATH_CLAWDBOT } from "../../config/config.js";
|
|
import type { RuntimeEnv } from "../../runtime.js";
|
|
import {
|
|
buildAllowlistSet,
|
|
modelKey,
|
|
resolveModelTarget,
|
|
updateConfig,
|
|
} from "./shared.js";
|
|
|
|
export async function modelsSetCommand(modelRaw: string, runtime: RuntimeEnv) {
|
|
const updated = await updateConfig((cfg) => {
|
|
const resolved = resolveModelTarget({ raw: modelRaw, cfg });
|
|
const allowlist = buildAllowlistSet(cfg);
|
|
if (allowlist.size > 0) {
|
|
const key = modelKey(resolved.provider, resolved.model);
|
|
if (!allowlist.has(key)) {
|
|
throw new Error(`Model ${key} is not in agent.allowedModels.`);
|
|
}
|
|
}
|
|
return {
|
|
...cfg,
|
|
agent: {
|
|
...cfg.agent,
|
|
model: `${resolved.provider}/${resolved.model}`,
|
|
},
|
|
};
|
|
});
|
|
|
|
runtime.log(`Updated ${CONFIG_PATH_CLAWDBOT}`);
|
|
runtime.log(`Default model: ${updated.agent?.model ?? modelRaw}`);
|
|
}
|