@tea-agent/loop-agent 0.23.1 → 0.24.1

package/dist/worker/console/chat/pi-console-config.js

@@ -0,0 +1,158 @@
+import { createHash, randomBytes } from "node:crypto";
+import { open, readFile, rename, writeFile } from "node:fs/promises";
+import path from "node:path";
+import { readJsonIfExists, writeSecureJson } from "../app-data.js";
+import { OPERATOR_CHAT_ALLOWED_INSTRUCTION_SKILLS } from "./instruction-skills.js";
+const revision = (raw) => createHash("sha256").update(raw).digest("hex");
+export async function readModelsConfig(agentDir) {
+    let raw = "{}";
+    try {
+        raw = await readFile(path.join(agentDir, "models.json"), "utf8");
+    }
+    catch {
+        /* empty config */
+    }
+    const parsed = JSON.parse(raw);
+    return {
+        revision: revision(raw),
+        providers: Object.entries(parsed.providers ?? {}).map(([name, provider]) => ({
+            name,
+            api: typeof provider.api === "string" ? provider.api : undefined,
+            baseUrl: typeof provider.baseUrl === "string" ? provider.baseUrl : undefined,
+            credential: {
+                configured: typeof provider.apiKey === "string" && provider.apiKey.length > 0,
+            },
+            models: Array.isArray(provider.models)
+                ? provider.models
+                    .map((model) => {
+                    const item = model && typeof model === "object"
+                        ? model
+                        : {};
+                    return {
+                        id: String(item.id ?? ""),
+                        name: typeof item.name === "string" ? item.name : undefined,
+                    };
+                })
+                    .filter((model) => model.id)
+                : [],
+        })),
+    };
+}
+export async function patchModelsConfig(agentDir, input) {
+    if (!["upsert-model", "remove-model", "remove-provider"].includes(input.action))
+        throw Object.assign(new Error("invalid models config action"), {
+            code: "INVALID_INPUT",
+            status: 400,
+        });
+    const file = path.join(agentDir, "models.json");
+    let raw = "{}";
+    try {
+        raw = await readFile(file, "utf8");
+    }
+    catch {
+        /* create below */
+    }
+    if (revision(raw) !== input.expectedRevision)
+        throw Object.assign(new Error("models config changed"), {
+            code: "CONFIG_CONFLICT",
+            status: 409,
+        });
+    if (!/^[A-Za-z0-9._-]+$/.test(input.provider))
+        throw Object.assign(new Error("invalid provider"), {
+            code: "INVALID_INPUT",
+            status: 400,
+        });
+    const parsed = JSON.parse(raw);
+    parsed.providers ??= {};
+    const provider = parsed.providers[input.provider];
+    if (input.action === "remove-provider")
+        delete parsed.providers[input.provider];
+    else {
+        if (!provider)
+            throw Object.assign(new Error("provider not found; configure credentials with Pi CLI"), { code: "NOT_FOUND", status: 404 });
+        const models = Array.isArray(provider.models)
+            ? provider.models.filter((model) => model && typeof model === "object")
+            : [];
+        if (!input.model?.id || input.model.id.length > 160)
+            throw Object.assign(new Error("model id is required"), {
+                code: "INVALID_INPUT",
+                status: 400,
+            });
+        if (input.model.name && input.model.name.length > 160)
+            throw Object.assign(new Error("model name is too long"), {
+                code: "INVALID_INPUT",
+                status: 400,
+            });
+        provider.models =
+            input.action === "remove-model"
+                ? models.filter((model) => model.id !== input.model.id)
+                : [
+                    ...models.filter((model) => model.id !== input.model.id),
+                    {
+                        ...(models.find((model) => model.id === input.model.id) ?? {}),
+                        id: input.model.id,
+                        ...(input.model.name !== undefined
+                            ? { name: input.model.name }
+                            : {}),
+                    },
+                ];
+    }
+    const body = `${JSON.stringify(parsed, null, 2)}\n`;
+    const tmp = path.join(agentDir, `.models.${process.pid}.${randomBytes(4).toString("hex")}.tmp`);
+    await writeFile(tmp, body, { encoding: "utf8", mode: 0o600 });
+    const handle = await open(tmp, "r+");
+    try {
+        await handle.sync();
+    }
+    finally {
+        await handle.close();
+    }
+    await rename(tmp, file);
+    return readModelsConfig(agentDir);
+}
+const preferencesFile = (appData) => path.join(appData.chats, ".instruction-skill-preferences.json");
+export async function readSkillPreferences(appData) {
+    const value = await readJsonIfExists(preferencesFile(appData));
+    return (value?.disabledNames ?? []).filter((name) => OPERATOR_CHAT_ALLOWED_INSTRUCTION_SKILLS.includes(name));
+}
+export async function writeSkillPreferences(appData, disabledNames) {
+    const allowed = new Set(OPERATOR_CHAT_ALLOWED_INSTRUCTION_SKILLS);
+    if (disabledNames.some((name) => !allowed.has(name)))
+        throw Object.assign(new Error("preferences contain a non-allowed skill"), {
+            code: "INVALID_INPUT",
+            status: 400,
+        });
+    const normalized = [...new Set(disabledNames)].sort();
+    await writeSecureJson(preferencesFile(appData), {
+        schemaVersion: 1,
+        disabledNames: normalized,
+    });
+    return normalized;
+}
+export async function readPackageInventory(agentDir, repoRoot) {
+    const read = async (file) => {
+        try {
+            const parsed = JSON.parse(await readFile(file, "utf8"));
+            return (parsed.packages ?? []).filter((item) => typeof item === "string");
+        }
+        catch {
+            return [];
+        }
+    };
+    const global = await read(path.join(agentDir, "settings.json"));
+    const project = await read(path.join(repoRoot, ".pi", "settings.json"));
+    return [
+        ...global.map((spec) => ({
+            spec,
+            scope: "global",
+            enabled: true,
+            resourceCount: 0,
+        })),
+        ...project.map((spec) => ({
+            spec,
+            scope: "project",
+            enabled: true,
+            resourceCount: 0,
+        })),
+    ];
+}