inferoa 0.1.0

package/dist/src/tools/process-tools.js

@@ -0,0 +1,199 @@
+import { spawn } from "node:child_process";
+import { resolveInside } from "../util/fs.js";
+import { fail, ok, truncateText } from "../util/limit.js";
+import { randomId } from "../util/hash.js";
+const liveProcesses = new Map();
+function key(sessionId, processId) {
+    return `${sessionId}:${processId}`;
+}
+export async function runCommand(args, context) {
+    const command = String(args.command);
+    const cwd = resolveInside(context.workspace.root, String(args.cwd ?? "."));
+    const env = {
+        ...process.env,
+        ...(typeof args.env === "object" && args.env ? stringEnv(args.env) : {}),
+    };
+    if (args.background) {
+        const processId = randomId("p");
+        const child = spawn(command, {
+            cwd,
+            env,
+            shell: true,
+            detached: process.platform !== "win32",
+        });
+        context.store.upsertProcess({
+            session_id: context.session_id,
+            process_id: processId,
+            pid: child.pid,
+            command,
+            cwd,
+            status: "running",
+        });
+        context.store.appendEvent({
+            session_id: context.session_id,
+            run_id: context.run_id,
+            type: "process.started",
+            data: { process_id: processId, pid: child.pid, command, cwd },
+        });
+        liveProcesses.set(key(context.session_id, processId), { child, session_id: context.session_id, process_id: processId });
+        child.stdout.on("data", (chunk) => {
+            safeStoreWrite(() => context.store.appendProcessOutput(context.session_id, processId, "stdout", String(chunk)));
+        });
+        child.stderr.on("data", (chunk) => {
+            safeStoreWrite(() => context.store.appendProcessOutput(context.session_id, processId, "stderr", String(chunk)));
+        });
+        child.on("close", (code) => {
+            safeStoreWrite(() => {
+                context.store.upsertProcess({
+                    session_id: context.session_id,
+                    process_id: processId,
+                    pid: child.pid,
+                    command,
+                    cwd,
+                    status: "stopped",
+                    exit_code: code,
+                });
+                context.store.appendEvent({
+                    session_id: context.session_id,
+                    run_id: context.run_id,
+                    type: "process.stopped",
+                    data: { process_id: processId, code },
+                });
+            });
+            liveProcesses.delete(key(context.session_id, processId));
+        });
+        return ok(`Started background process ${processId}`, { process_id: processId, pid: child.pid ?? null, command, cwd });
+    }
+    const timeoutMs = typeof args.timeout_ms === "number" ? Math.max(100, Math.min(args.timeout_ms, 600_000)) : 120_000;
+    return await new Promise((resolve) => {
+        const child = spawn(command, {
+            cwd,
+            env,
+            shell: true,
+        });
+        let stdout = "";
+        let stderr = "";
+        let timedOut = false;
+        const timeout = setTimeout(() => {
+            timedOut = true;
+            child.kill("SIGTERM");
+            setTimeout(() => {
+                if (!child.killed) {
+                    child.kill("SIGKILL");
+                }
+            }, 2000).unref();
+        }, timeoutMs);
+        child.stdout.on("data", (chunk) => {
+            stdout += String(chunk);
+        });
+        child.stderr.on("data", (chunk) => {
+            stderr += String(chunk);
+        });
+        child.on("close", (code) => {
+            clearTimeout(timeout);
+            const combined = [stdout, stderr].filter(Boolean).join("\n");
+            const truncated = truncateText(combined);
+            const resource = truncated.truncated || combined.length > 24_000
+                ? context.store.putResource(context.session_id, "command.output", combined, { command, cwd, code, timed_out: timedOut }).uri
+                : undefined;
+            context.store.appendEvent({
+                session_id: context.session_id,
+                run_id: context.run_id,
+                type: "tool.shell.completed",
+                data: { command, cwd, code, timed_out: timedOut, resource_uri: resource },
+            });
+            resolve({
+                ok: code === 0 && !timedOut,
+                summary: `Command exited ${code}${timedOut ? " after timeout" : ""}`,
+                data: {
+                    command,
+                    cwd,
+                    code,
+                    timed_out: timedOut,
+                    output: truncated.text,
+                },
+                resource_uri: resource,
+                error: code === 0 && !timedOut ? undefined : { code: timedOut ? "command_timeout" : "command_failed", message: stderr || stdout },
+            });
+        });
+        child.on("error", (error) => {
+            clearTimeout(timeout);
+            resolve(fail("command_spawn_failed", error.message));
+        });
+    });
+}
+export async function readProcess(args, context) {
+    const processId = String(args.process_id);
+    const sinceSeq = typeof args.since_seq === "number" ? args.since_seq : 0;
+    const maxBytes = typeof args.max_bytes === "number" ? Math.max(1, Math.min(args.max_bytes, 100_000)) : 24_000;
+    const output = context.store.readProcessOutput(context.session_id, processId, sinceSeq, maxBytes);
+    return ok(`Read process ${processId} through seq ${output.seq}`, {
+        process_id: processId,
+        since_seq: sinceSeq,
+        next_seq: output.seq,
+        output: output.text,
+        live: liveProcesses.has(key(context.session_id, processId)),
+    });
+}
+export async function writeProcess(args, context) {
+    const processId = String(args.process_id);
+    const live = liveProcesses.get(key(context.session_id, processId));
+    if (!live) {
+        return fail("process_not_live", `Process is not live in this runtime: ${processId}`);
+    }
+    live.child.stdin.write(String(args.input));
+    if (args.close_stdin) {
+        live.child.stdin.end();
+    }
+    context.store.appendEvent({
+        session_id: context.session_id,
+        run_id: context.run_id,
+        type: "process.stdin",
+        data: { process_id: processId, bytes: Buffer.byteLength(String(args.input)), close_stdin: Boolean(args.close_stdin) },
+    });
+    return ok(`Wrote stdin to ${processId}`, { process_id: processId });
+}
+export async function stopProcess(args, context) {
+    const processId = String(args.process_id);
+    const live = liveProcesses.get(key(context.session_id, processId));
+    if (!live) {
+        return fail("process_not_live", `Process is not live in this runtime: ${processId}`);
+    }
+    const signal = String(args.signal ?? "SIGTERM");
+    killProcessTree(live.child, signal);
+    context.store.appendEvent({
+        session_id: context.session_id,
+        run_id: context.run_id,
+        type: "process.stop_requested",
+        data: { process_id: processId, signal },
+    });
+    return ok(`Sent ${signal} to ${processId}`, { process_id: processId, signal });
+}
+function stringEnv(env) {
+    return Object.fromEntries(Object.entries(env).map(([name, value]) => [name, String(value)]));
+}
+function killProcessTree(child, signal) {
+    const pid = child.pid;
+    if (pid && process.platform !== "win32") {
+        try {
+            process.kill(-pid, signal);
+            return;
+        }
+        catch (error) {
+            if (error.code !== "ESRCH") {
+                child.kill(signal);
+            }
+            return;
+        }
+    }
+    child.kill(signal);
+}
+function safeStoreWrite(write) {
+    try {
+        write();
+    }
+    catch {
+        // Process output can arrive after a test or session has closed its store.
+    }
+}
+//# sourceMappingURL=process-tools.js.map

package/dist/src/tools/process-tools.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"process-tools.js","sourceRoot":"","sources":["../../../src/tools/process-tools.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAuC,MAAM,oBAAoB,CAAC;AAEhF,OAAO,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AAC9C,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE,YAAY,EAAE,MAAM,kBAAkB,CAAC;AAC1D,OAAO,EAAE,QAAQ,EAAE,MAAM,iBAAiB,CAAC;AAS3C,MAAM,aAAa,GAAG,IAAI,GAAG,EAAuB,CAAC;AAErD,SAAS,GAAG,CAAC,SAAiB,EAAE,SAAiB;IAC/C,OAAO,GAAG,SAAS,IAAI,SAAS,EAAE,CAAC;AACrC,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,UAAU,CAAC,IAAgB,EAAE,OAA6B;IAC9E,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACrC,MAAM,GAAG,GAAG,aAAa,CAAC,OAAO,CAAC,SAAS,CAAC,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC,GAAG,IAAI,GAAG,CAAC,CAAC,CAAC;IAC3E,MAAM,GAAG,GAAG;QACV,GAAG,OAAO,CAAC,GAAG;QACd,GAAG,CAAC,OAAO,IAAI,CAAC,GAAG,KAAK,QAAQ,IAAI,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,GAA8B,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;KACpG,CAAC;IACF,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;QACpB,MAAM,SAAS,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC;QAChC,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,EAAE;YAC3B,GAAG;YACH,GAAG;YACH,KAAK,EAAE,IAAI;YACX,QAAQ,EAAE,OAAO,CAAC,QAAQ,KAAK,OAAO;SACvC,CAAC,CAAC;QACH,OAAO,CAAC,KAAK,CAAC,aAAa,CAAC;YAC1B,UAAU,EAAE,OAAO,CAAC,UAAU;YAC9B,UAAU,EAAE,SAAS;YACrB,GAAG,EAAE,KAAK,CAAC,GAAG;YACd,OAAO;YACP,GAAG;YACH,MAAM,EAAE,SAAS;SAClB,CAAC,CAAC;QACH,OAAO,CAAC,KAAK,CAAC,WAAW,CAAC;YACxB,UAAU,EAAE,OAAO,CAAC,UAAU;YAC9B,MAAM,EAAE,OAAO,CAAC,MAAM;YACtB,IAAI,EAAE,iBAAiB;YACvB,IAAI,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,GAAG,EAAE,KAAK,CAAC,GAAG,EAAE,OAAO,EAAE,GAAG,EAAE;SAC9D,CAAC,CAAC;QACH,aAAa,CAAC,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,UAAU,EAAE,SAAS,CAAC,EAAE,EAAE,KAAK,EAAE,UAAU,EAAE,OAAO,CAAC,UAAU,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC,CAAC;QACxH,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAK,EAAE,EAAE;YAChC,cAAc,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,mBAAmB,CAAC,OAAO,CAAC,UAAU,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAClH,CAAC,CAAC,CAAC;QACH,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAK,EAAE,EAAE;YAChC,cAAc,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,mBAAmB,CAAC,OAAO,CAAC,UAAU,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAClH,CAAC,CAAC,CAAC;QACH,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,IAAI,EAAE,EAAE;YACzB,cAAc,CAAC,GAAG,EAAE;gBAClB,OAAO,CAAC,KAAK,CAAC,aAAa,CAAC;oBAC1B,UAAU,EAAE,OAAO,CAAC,UAAU;oBAC9B,UAAU,EAAE,SAAS;oBACrB,GAAG,EAAE,KAAK,CAAC,GAAG;oBACd,OAAO;oBACP,GAAG;oBACH,MAAM,EAAE,SAAS;oBACjB,SAAS,EAAE,IAAI;iBAChB,CAAC,CAAC;gBACH,OAAO,CAAC,KAAK,CAAC,WAAW,CAAC;oBACxB,UAAU,EAAE,OAAO,CAAC,UAAU;oBAC9B,MAAM,EAAE,OAAO,CAAC,MAAM;oBACtB,IAAI,EAAE,iBAAiB;oBACvB,IAAI,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,IAAI,EAAE;iBACtC,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;YACH,aAAa,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,UAAU,EAAE,SAAS,CAAC,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;QACH,OAAO,EAAE,CAAC,8BAA8B,SAAS,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,GAAG,EAAE,KAAK,CAAC,GAAG,IAAI,IAAI,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC,CAAC;IACxH,CAAC;IAED,MAAM,SAAS,GAAG,OAAO,IAAI,CAAC,UAAU,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;IACpH,OAAO,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QACnC,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,EAAE;YAC3B,GAAG;YACH,GAAG;YACH,KAAK,EAAE,IAAI;SACZ,CAAC,CAAC;QACH,IAAI,MAAM,GAAG,EAAE,CAAC;QAChB,IAAI,MAAM,GAAG,EAAE,CAAC;QAChB,IAAI,QAAQ,GAAG,KAAK,CAAC;QACrB,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,EAAE;YAC9B,QAAQ,GAAG,IAAI,CAAC;YAChB,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YACtB,UAAU,CAAC,GAAG,EAAE;gBACd,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC;oBAClB,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBACxB,CAAC;YACH,CAAC,EAAE,IAAI,CAAC,CAAC,KAAK,EAAE,CAAC;QACnB,CAAC,EAAE,SAAS,CAAC,CAAC;QACd,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAK,EAAE,EAAE;YAChC,MAAM,IAAI,MAAM,CAAC,KAAK,CAAC,CAAC;QAC1B,CAAC,CAAC,CAAC;QACH,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAK,EAAE,EAAE;YAChC,MAAM,IAAI,MAAM,CAAC,KAAK,CAAC,CAAC;QAC1B,CAAC,CAAC,CAAC;QACH,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,IAAI,EAAE,EAAE;YACzB,YAAY,CAAC,OAAO,CAAC,CAAC;YACtB,MAAM,QAAQ,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC7D,MAAM,SAAS,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;YACzC,MAAM,QAAQ,GACZ,SAAS,CAAC,SAAS,IAAI,QAAQ,CAAC,MAAM,GAAG,MAAM;gBAC7C,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,WAAW,CAAC,OAAO,CAAC,UAAU,EAAE,gBAAgB,EAAE,QAAQ,EAAE,EAAE,OAAO,EAAE,GAAG,EAAE,IAAI,EAAE,SAAS,EAAE,QAAQ,EAAE,CAAC,CAAC,GAAG;gBAC5H,CAAC,CAAC,SAAS,CAAC;YAChB,OAAO,CAAC,KAAK,CAAC,WAAW,CAAC;gBACxB,UAAU,EAAE,OAAO,CAAC,UAAU;gBAC9B,MAAM,EAAE,OAAO,CAAC,MAAM;gBACtB,IAAI,EAAE,sBAAsB;gBAC5B,IAAI,EAAE,EAAE,OAAO,EAAE,GAAG,EAAE,IAAI,EAAE,SAAS,EAAE,QAAQ,EAAE,YAAY,EAAE,QAAQ,EAAE;aAC1E,CAAC,CAAC;YACH,OAAO,CAAC;gBACN,EAAE,EAAE,IAAI,KAAK,CAAC,IAAI,CAAC,QAAQ;gBAC3B,OAAO,EAAE,kBAAkB,IAAI,GAAG,QAAQ,CAAC,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,EAAE,EAAE;gBACpE,IAAI,EAAE;oBACJ,OAAO;oBACP,GAAG;oBACH,IAAI;oBACJ,SAAS,EAAE,QAAQ;oBACnB,MAAM,EAAE,SAAS,CAAC,IAAI;iBACvB;gBACD,YAAY,EAAE,QAAQ;gBACtB,KAAK,EAAE,IAAI,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,QAAQ,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC,gBAAgB,EAAE,OAAO,EAAE,MAAM,IAAI,MAAM,EAAE;aAClI,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QACH,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;YAC1B,YAAY,CAAC,OAAO,CAAC,CAAC;YACtB,OAAO,CAAC,IAAI,CAAC,sBAAsB,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,WAAW,CAAC,IAAgB,EAAE,OAA6B;IAC/E,MAAM,SAAS,GAAG,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAC1C,MAAM,QAAQ,GAAG,OAAO,IAAI,CAAC,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;IACzE,MAAM,QAAQ,GAAG,OAAO,IAAI,CAAC,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;IAC9G,MAAM,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC,iBAAiB,CAAC,OAAO,CAAC,UAAU,EAAE,SAAS,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC;IAClG,OAAO,EAAE,CAAC,gBAAgB,SAAS,gBAAgB,MAAM,CAAC,GAAG,EAAE,EAAE;QAC/D,UAAU,EAAE,SAAS;QACrB,SAAS,EAAE,QAAQ;QACnB,QAAQ,EAAE,MAAM,CAAC,GAAG;QACpB,MAAM,EAAE,MAAM,CAAC,IAAI;QACnB,IAAI,EAAE,aAAa,CAAC,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,UAAU,EAAE,SAAS,CAAC,CAAC;KAC5D,CAAC,CAAC;AACL,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,YAAY,CAAC,IAAgB,EAAE,OAA6B;IAChF,MAAM,SAAS,GAAG,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAC1C,MAAM,IAAI,GAAG,aAAa,CAAC,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,UAAU,EAAE,SAAS,CAAC,CAAC,CAAC;IACnE,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,OAAO,IAAI,CAAC,kBAAkB,EAAE,wCAAwC,SAAS,EAAE,CAAC,CAAC;IACvF,CAAC;IACD,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;IAC3C,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;QACrB,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC;IACzB,CAAC;IACD,OAAO,CAAC,KAAK,CAAC,WAAW,CAAC;QACxB,UAAU,EAAE,OAAO,CAAC,UAAU;QAC9B,MAAM,EAAE,OAAO,CAAC,MAAM;QACtB,IAAI,EAAE,eAAe;QACrB,IAAI,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,KAAK,EAAE,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,WAAW,EAAE,OAAO,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE;KACtH,CAAC,CAAC;IACH,OAAO,EAAE,CAAC,kBAAkB,SAAS,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC,CAAC;AACtE,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,WAAW,CAAC,IAAgB,EAAE,OAA6B;IAC/E,MAAM,SAAS,GAAG,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAC1C,MAAM,IAAI,GAAG,aAAa,CAAC,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,UAAU,EAAE,SAAS,CAAC,CAAC,CAAC;IACnE,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,OAAO,IAAI,CAAC,kBAAkB,EAAE,wCAAwC,SAAS,EAAE,CAAC,CAAC;IACvF,CAAC;IACD,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,MAAM,IAAI,SAAS,CAAmB,CAAC;IAClE,eAAe,CAAC,IAAI,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;IACpC,OAAO,CAAC,KAAK,CAAC,WAAW,CAAC;QACxB,UAAU,EAAE,OAAO,CAAC,UAAU;QAC9B,MAAM,EAAE,OAAO,CAAC,MAAM;QACtB,IAAI,EAAE,wBAAwB;QAC9B,IAAI,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,EAAE;KACxC,CAAC,CAAC;IACH,OAAO,EAAE,CAAC,QAAQ,MAAM,OAAO,SAAS,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,EAAE,CAAC,CAAC;AACjF,CAAC;AAED,SAAS,SAAS,CAAC,GAA4B;IAC7C,OAAO,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;AAC/F,CAAC;AAED,SAAS,eAAe,CAAC,KAAqC,EAAE,MAAsB;IACpF,MAAM,GAAG,GAAG,KAAK,CAAC,GAAG,CAAC;IACtB,IAAI,GAAG,IAAI,OAAO,CAAC,QAAQ,KAAK,OAAO,EAAE,CAAC;QACxC,IAAI,CAAC;YACH,OAAO,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;YAC3B,OAAO;QACT,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAK,KAA+B,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;gBACtD,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACrB,CAAC;YACD,OAAO;QACT,CAAC;IACH,CAAC;IACD,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;AACrB,CAAC;AAED,SAAS,cAAc,CAAC,KAAiB;IACvC,IAAI,CAAC;QACH,KAAK,EAAE,CAAC;IACV,CAAC;IAAC,MAAM,CAAC;QACP,0EAA0E;IAC5E,CAAC;AACH,CAAC"}

package/dist/src/tools/registry.d.ts

@@ -0,0 +1,20 @@
+import type { ToolCall, ToolDefinition, ToolResult, VllmAgentConfig, WorkspaceIdentity } from "../types.js";
+import { CodeIntelligenceHub } from "../code-intelligence/hub.js";
+import { SessionStore } from "../session/store.js";
+import type { ToolExecutionContext } from "./context.js";
+export declare class ToolRegistry {
+    private readonly config;
+    private readonly workspace;
+    private readonly store;
+    private readonly codeIntelligence;
+    private readonly policy;
+    constructor(config: VllmAgentConfig, workspace: WorkspaceIdentity, store: SessionStore, codeIntelligence?: CodeIntelligenceHub);
+    list(): ToolDefinition[];
+    call(call: ToolCall, context: {
+        session_id: string;
+        run_id?: string;
+        clarify?: ToolExecutionContext["clarify"];
+    }): Promise<ToolResult>;
+    private recordCall;
+    private recordResult;
+}

package/dist/src/tools/registry.js

@@ -0,0 +1,187 @@
+import { CodeIntelligenceHub } from "../code-intelligence/hub.js";
+import { fail, truncateText } from "../util/limit.js";
+import { PermissionPolicy } from "./permissions.js";
+import { configuredToolDefinitions } from "./schemas.js";
+import { applyPatchTool, completeStep, editFile, fileSearch, gitDiff, gitShow, gitStatus, globPaths, listDir, readFile, readResource, sessionNote, todoWrite, writeFile, } from "./workspace-tools.js";
+import { readProcess, runCommand, stopProcess, writeProcess } from "./process-tools.js";
+import { astEdit, astGrep, lspTool } from "./code-intelligence.js";
+import { webFetch, webOpen, webSearch } from "./web-search.js";
+import { skillDisable, skillEnable, skillList, skillRead } from "./skill-tools.js";
+import { goalTool } from "./goal-tools.js";
+import { planTool } from "./plan-tools.js";
+import { clarifyTool } from "./clarify-tool.js";
+import { initExperiment, logExperiment, runExperiment, updateNotes } from "./autoresearch-tools.js";
+import { audioGeneration, audioUnderstanding, imageGeneration, videoGeneration, videoUnderstanding, visionUnderstanding, } from "./omni-tools.js";
+const HANDLERS = {
+    apply_patch: applyPatchTool,
+    ast_edit: astEdit,
+    ast_grep: astGrep,
+    audio_generation: audioGeneration,
+    audio_understanding: audioUnderstanding,
+    clarify: clarifyTool,
+    complete_step: completeStep,
+    edit_file: editFile,
+    file_search: fileSearch,
+    git_diff: gitDiff,
+    git_show: gitShow,
+    git_status: gitStatus,
+    glob: globPaths,
+    goal: goalTool,
+    image_generation: imageGeneration,
+    init_experiment: initExperiment,
+    list_dir: listDir,
+    log_experiment: logExperiment,
+    lsp: lspTool,
+    plan: planTool,
+    read_file: readFile,
+    read_process: readProcess,
+    read_resource: readResource,
+    run_command: runCommand,
+    run_experiment: runExperiment,
+    session_note: sessionNote,
+    skill_disable: skillDisable,
+    skill_enable: skillEnable,
+    skill_list: skillList,
+    skill_read: skillRead,
+    stop_process: stopProcess,
+    todo_write: todoWrite,
+    update_notes: updateNotes,
+    video_generation: videoGeneration,
+    video_understanding: videoUnderstanding,
+    vision_understanding: visionUnderstanding,
+    web_fetch: webFetch,
+    web_open: webOpen,
+    web_search: webSearch,
+    write_file: writeFile,
+    write_process: writeProcess,
+};
+export class ToolRegistry {
+    config;
+    workspace;
+    store;
+    codeIntelligence;
+    policy;
+    constructor(config, workspace, store, codeIntelligence = new CodeIntelligenceHub(config, workspace)) {
+        this.config = config;
+        this.workspace = workspace;
+        this.store = store;
+        this.codeIntelligence = codeIntelligence;
+        this.policy = new PermissionPolicy(config, workspace);
+    }
+    list() {
+        return [...configuredToolDefinitions(this.config), ...this.codeIntelligence.toolDefinitions()].sort((a, b) => a.name.localeCompare(b.name));
+    }
+    async call(call, context) {
+        const definition = this.list().find((tool) => tool.name === call.name);
+        if (!definition) {
+            const result = fail("unknown_tool", `Unknown tool: ${call.name}`);
+            this.recordCall(context, call);
+            this.recordResult(context, call, result);
+            return result;
+        }
+        this.recordCall(context, call);
+        const decision = this.policy.decide(definition, call.arguments);
+        if (decision.status !== "allow") {
+            this.store.appendEvent({
+                session_id: context.session_id,
+                run_id: context.run_id,
+                type: decision.status === "ask" ? "permission.requested" : "permission.denied",
+                data: {
+                    tool_call_id: call.id,
+                    tool_name: call.name,
+                    decision: decision,
+                },
+            });
+            const blocked = fail(decision.status === "ask" ? "permission_required" : "permission_denied", `Tool ${call.name} blocked: ${decision.reason}`);
+            this.store.appendEvent({
+                session_id: context.session_id,
+                run_id: context.run_id,
+                type: "tool.result",
+                data: {
+                    tool_call_id: call.id,
+                    tool_name: call.name,
+                    result: blocked,
+                },
+            });
+            return blocked;
+        }
+        this.store.appendEvent({
+            session_id: context.session_id,
+            run_id: context.run_id,
+            type: "permission.resolved",
+            data: {
+                tool_call_id: call.id,
+                tool_name: call.name,
+                decision: decision,
+            },
+        });
+        const execContext = {
+            config: this.config,
+            workspace: this.workspace,
+            session_id: context.session_id,
+            run_id: context.run_id,
+            store: this.store,
+            clarify: context.clarify,
+        };
+        let result;
+        try {
+            if (this.codeIntelligence.handlesTool(call.name)) {
+                result = await this.codeIntelligence.callTool(call.name, call.arguments);
+            }
+            else {
+                const handler = HANDLERS[call.name];
+                if (!handler) {
+                    result = fail("tool_not_implemented", `Tool schema exists but handler is not implemented: ${call.name}`);
+                }
+                else {
+                    result = await handler(call.arguments, execContext);
+                }
+            }
+        }
+        catch (error) {
+            result = fail("tool_exception", error instanceof Error ? error.message : String(error));
+        }
+        const serialized = JSON.stringify(result);
+        if (serialized.length > 30_000 && !result.resource_uri) {
+            const resource = this.store.putResource(context.session_id, `tool.${call.name}.result`, serialized, {
+                tool_name: call.name,
+                tool_call_id: call.id,
+            });
+            const truncated = truncateText(serialized, 12_000);
+            result = {
+                ok: result.ok,
+                summary: result.summary,
+                data: { truncated_result: truncated.text },
+                resource_uri: resource.uri,
+                error: result.error,
+            };
+        }
+        this.recordResult(context, call, result);
+        return result;
+    }
+    recordCall(context, call) {
+        this.store.appendEvent({
+            session_id: context.session_id,
+            run_id: context.run_id,
+            type: "tool.call",
+            data: {
+                tool_call_id: call.id,
+                tool_name: call.name,
+                arguments: call.arguments,
+            },
+        });
+    }
+    recordResult(context, call, result) {
+        this.store.appendEvent({
+            session_id: context.session_id,
+            run_id: context.run_id,
+            type: "tool.result",
+            data: {
+                tool_call_id: call.id,
+                tool_name: call.name,
+                result: result,
+            },
+        });
+    }
+}
+//# sourceMappingURL=registry.js.map

package/dist/src/tools/registry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"registry.js","sourceRoot":"","sources":["../../../src/tools/registry.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,mBAAmB,EAAE,MAAM,6BAA6B,CAAC;AAElE,OAAO,EAAE,IAAI,EAAE,YAAY,EAAE,MAAM,kBAAkB,CAAC;AACtD,OAAO,EAAE,gBAAgB,EAAE,MAAM,kBAAkB,CAAC;AACpD,OAAO,EAAE,yBAAyB,EAAE,MAAM,cAAc,CAAC;AAEzD,OAAO,EACL,cAAc,EACd,YAAY,EACZ,QAAQ,EACR,UAAU,EACV,OAAO,EACP,OAAO,EACP,SAAS,EACT,SAAS,EACT,OAAO,EACP,QAAQ,EACR,YAAY,EACZ,WAAW,EACX,SAAS,EACT,SAAS,GACV,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,WAAW,EAAE,UAAU,EAAE,WAAW,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AACxF,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,wBAAwB,CAAC;AACnE,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAC/D,OAAO,EAAE,YAAY,EAAE,WAAW,EAAE,SAAS,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AACnF,OAAO,EAAE,QAAQ,EAAE,MAAM,iBAAiB,CAAC;AAC3C,OAAO,EAAE,QAAQ,EAAE,MAAM,iBAAiB,CAAC;AAC3C,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAAE,cAAc,EAAE,aAAa,EAAE,aAAa,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AACpG,OAAO,EACL,eAAe,EACf,kBAAkB,EAClB,eAAe,EACf,eAAe,EACf,kBAAkB,EAClB,mBAAmB,GACpB,MAAM,iBAAiB,CAAC;AAEzB,MAAM,QAAQ,GAAgC;IAC5C,WAAW,EAAE,cAAc;IAC3B,QAAQ,EAAE,OAAO;IACjB,QAAQ,EAAE,OAAO;IACjB,gBAAgB,EAAE,eAAe;IACjC,mBAAmB,EAAE,kBAAkB;IACvC,OAAO,EAAE,WAAW;IACpB,aAAa,EAAE,YAAY;IAC3B,SAAS,EAAE,QAAQ;IACnB,WAAW,EAAE,UAAU;IACvB,QAAQ,EAAE,OAAO;IACjB,QAAQ,EAAE,OAAO;IACjB,UAAU,EAAE,SAAS;IACrB,IAAI,EAAE,SAAS;IACf,IAAI,EAAE,QAAQ;IACd,gBAAgB,EAAE,eAAe;IACjC,eAAe,EAAE,cAAc;IAC/B,QAAQ,EAAE,OAAO;IACjB,cAAc,EAAE,aAAa;IAC7B,GAAG,EAAE,OAAO;IACZ,IAAI,EAAE,QAAQ;IACd,SAAS,EAAE,QAAQ;IACnB,YAAY,EAAE,WAAW;IACzB,aAAa,EAAE,YAAY;IAC3B,WAAW,EAAE,UAAU;IACvB,cAAc,EAAE,aAAa;IAC7B,YAAY,EAAE,WAAW;IACzB,aAAa,EAAE,YAAY;IAC3B,YAAY,EAAE,WAAW;IACzB,UAAU,EAAE,SAAS;IACrB,UAAU,EAAE,SAAS;IACrB,YAAY,EAAE,WAAW;IACzB,UAAU,EAAE,SAAS;IACrB,YAAY,EAAE,WAAW;IACzB,gBAAgB,EAAE,eAAe;IACjC,mBAAmB,EAAE,kBAAkB;IACvC,oBAAoB,EAAE,mBAAmB;IACzC,SAAS,EAAE,QAAQ;IACnB,QAAQ,EAAE,OAAO;IACjB,UAAU,EAAE,SAAS;IACrB,UAAU,EAAE,SAAS;IACrB,aAAa,EAAE,YAAY;CAC5B,CAAC;AAEF,MAAM,OAAO,YAAY;IAIJ;IACA;IACA;IACA;IANF,MAAM,CAAmB;IAE1C,YACmB,MAAuB,EACvB,SAA4B,EAC5B,KAAmB,EACnB,mBAAwC,IAAI,mBAAmB,CAAC,MAAM,EAAE,SAAS,CAAC;QAHlF,WAAM,GAAN,MAAM,CAAiB;QACvB,cAAS,GAAT,SAAS,CAAmB;QAC5B,UAAK,GAAL,KAAK,CAAc;QACnB,qBAAgB,GAAhB,gBAAgB,CAAkE;QAEnG,IAAI,CAAC,MAAM,GAAG,IAAI,gBAAgB,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;IACxD,CAAC;IAED,IAAI;QACF,OAAO,CAAC,GAAG,yBAAyB,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,GAAG,IAAI,CAAC,gBAAgB,CAAC,eAAe,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;IAC9I,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,IAAc,EAAE,OAA2F;QACpH,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,IAAI,CAAC,IAAI,CAAC,CAAC;QACvE,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,MAAM,MAAM,GAAG,IAAI,CAAC,cAAc,EAAE,iBAAiB,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;YAClE,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;YAC/B,IAAI,CAAC,YAAY,CAAC,OAAO,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC;YACzC,OAAO,MAAM,CAAC;QAChB,CAAC;QACD,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QAC/B,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,UAAU,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;QAChE,IAAI,QAAQ,CAAC,MAAM,KAAK,OAAO,EAAE,CAAC;YAChC,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC;gBACrB,UAAU,EAAE,OAAO,CAAC,UAAU;gBAC9B,MAAM,EAAE,OAAO,CAAC,MAAM;gBACtB,IAAI,EAAE,QAAQ,CAAC,MAAM,KAAK,KAAK,CAAC,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,mBAAmB;gBAC9E,IAAI,EAAE;oBACJ,YAAY,EAAE,IAAI,CAAC,EAAE;oBACrB,SAAS,EAAE,IAAI,CAAC,IAAI;oBACpB,QAAQ,EAAE,QAAiC;iBAC5C;aACF,CAAC,CAAC;YACH,MAAM,OAAO,GAAG,IAAI,CAClB,QAAQ,CAAC,MAAM,KAAK,KAAK,CAAC,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,mBAAmB,EACvE,QAAQ,IAAI,CAAC,IAAI,aAAa,QAAQ,CAAC,MAAM,EAAE,CAChD,CAAC;YACF,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC;gBACrB,UAAU,EAAE,OAAO,CAAC,UAAU;gBAC9B,MAAM,EAAE,OAAO,CAAC,MAAM;gBACtB,IAAI,EAAE,aAAa;gBACnB,IAAI,EAAE;oBACJ,YAAY,EAAE,IAAI,CAAC,EAAE;oBACrB,SAAS,EAAE,IAAI,CAAC,IAAI;oBACpB,MAAM,EAAE,OAAgC;iBACzC;aACF,CAAC,CAAC;YACH,OAAO,OAAO,CAAC;QACjB,CAAC;QACD,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC;YACrB,UAAU,EAAE,OAAO,CAAC,UAAU;YAC9B,MAAM,EAAE,OAAO,CAAC,MAAM;YACtB,IAAI,EAAE,qBAAqB;YAC3B,IAAI,EAAE;gBACJ,YAAY,EAAE,IAAI,CAAC,EAAE;gBACrB,SAAS,EAAE,IAAI,CAAC,IAAI;gBACpB,QAAQ,EAAE,QAAiC;aAC5C;SACF,CAAC,CAAC;QACH,MAAM,WAAW,GAAyB;YACxC,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,UAAU,EAAE,OAAO,CAAC,UAAU;YAC9B,MAAM,EAAE,OAAO,CAAC,MAAM;YACtB,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,OAAO,EAAE,OAAO,CAAC,OAAO;SACzB,CAAC;QACF,IAAI,MAAkB,CAAC;QACvB,IAAI,CAAC;YACH,IAAI,IAAI,CAAC,gBAAgB,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;gBACjD,MAAM,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;YAC3E,CAAC;iBAAM,CAAC;gBACN,MAAM,OAAO,GAAG,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACpC,IAAI,CAAC,OAAO,EAAE,CAAC;oBACb,MAAM,GAAG,IAAI,CAAC,sBAAsB,EAAE,sDAAsD,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;gBAC3G,CAAC;qBAAM,CAAC;oBACN,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC;gBACtD,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,GAAG,IAAI,CAAC,gBAAgB,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;QAC1F,CAAC;QACD,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QAC1C,IAAI,UAAU,CAAC,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC;YACvD,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,OAAO,CAAC,UAAU,EAAE,QAAQ,IAAI,CAAC,IAAI,SAAS,EAAE,UAAU,EAAE;gBAClG,SAAS,EAAE,IAAI,CAAC,IAAI;gBACpB,YAAY,EAAE,IAAI,CAAC,EAAE;aACtB,CAAC,CAAC;YACH,MAAM,SAAS,GAAG,YAAY,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;YACnD,MAAM,GAAG;gBACP,EAAE,EAAE,MAAM,CAAC,EAAE;gBACb,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,IAAI,EAAE,EAAE,gBAAgB,EAAE,SAAS,CAAC,IAAI,EAAE;gBAC1C,YAAY,EAAE,QAAQ,CAAC,GAAG;gBAC1B,KAAK,EAAE,MAAM,CAAC,KAAK;aACpB,CAAC;QACJ,CAAC;QACD,IAAI,CAAC,YAAY,CAAC,OAAO,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC;QACzC,OAAO,MAAM,CAAC;IAChB,CAAC;IAEO,UAAU,CAAC,OAAgD,EAAE,IAAc;QACjF,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC;YACrB,UAAU,EAAE,OAAO,CAAC,UAAU;YAC9B,MAAM,EAAE,OAAO,CAAC,MAAM;YACtB,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE;gBACJ,YAAY,EAAE,IAAI,CAAC,EAAE;gBACrB,SAAS,EAAE,IAAI,CAAC,IAAI;gBACpB,SAAS,EAAE,IAAI,CAAC,SAAS;aAC1B;SACF,CAAC,CAAC;IACL,CAAC;IAEO,YAAY,CAAC,OAAgD,EAAE,IAAc,EAAE,MAAkB;QACvG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC;YACrB,UAAU,EAAE,OAAO,CAAC,UAAU;YAC9B,MAAM,EAAE,OAAO,CAAC,MAAM;YACtB,IAAI,EAAE,aAAa;YACnB,IAAI,EAAE;gBACJ,YAAY,EAAE,IAAI,CAAC,EAAE;gBACrB,SAAS,EAAE,IAAI,CAAC,IAAI;gBACpB,MAAM,EAAE,MAA+B;aACxC;SACF,CAAC,CAAC;IACL,CAAC;CACF"}

package/dist/src/tools/schemas.d.ts

@@ -0,0 +1,3 @@
+import type { ToolDefinition, VllmAgentConfig } from "../types.js";
+export declare const CORE_TOOL_DEFINITIONS: ToolDefinition[];
+export declare function configuredToolDefinitions(config: VllmAgentConfig): ToolDefinition[];