@di-code/coding-agent 0.1.0

package/dist/core/session.js

@@ -0,0 +1,262 @@
+import { randomUUID } from "node:crypto";
+import { Agent, estimateContextTokens, generateCompactionSummary, prepareCompaction, resolveContextBudget, shouldCompact, } from "@di-code/agent";
+import { buildSessionContext } from "./context-builder.js";
+import { createBashTool } from "./tools/bash.js";
+import { createEditTool } from "./tools/edit.js";
+import { createReadTool } from "./tools/read.js";
+import { createWriteTool } from "./tools/write.js";
+export class AgentSession {
+    agent;
+    allowedRootValue;
+    sessionManager;
+    provider;
+    model;
+    sessionIdValue;
+    now;
+    compactionEnabledValue;
+    contextBudget;
+    keepRecentTokens;
+    usageTotals;
+    persistenceError;
+    promptActive = false;
+    sessionListeners = new Set();
+    constructor(options) {
+        this.allowedRootValue = options.allowedRoot;
+        this.sessionManager = options.sessionManager;
+        this.provider = options.provider;
+        this.model = options.model;
+        this.sessionIdValue = options.sessionManager?.header.id ?? randomUUID();
+        this.now = options.now ?? Date.now;
+        this.contextBudget = resolveContextBudget(options.model);
+        if (options.compaction?.enabled !== undefined && typeof options.compaction.enabled !== "boolean") {
+            throw new TypeError("compaction.enabled must be a boolean");
+        }
+        this.compactionEnabledValue = options.sessionManager !== undefined && options.compaction?.enabled !== false;
+        const defaultKeepRecentTokens = Math.max(1, Math.min(20_000, Math.floor(this.contextBudget.triggerTokens / 2)));
+        this.keepRecentTokens = options.compaction?.keepRecentTokens ?? defaultKeepRecentTokens;
+        if (!Number.isInteger(this.keepRecentTokens) || this.keepRecentTokens <= 0) {
+            throw new RangeError("compaction.keepRecentTokens must be a positive integer");
+        }
+        this.usageTotals = {
+            requestCount: 0,
+            inputTokens: 0,
+            outputTokens: 0,
+            cacheReadTokens: 0,
+            cacheWriteTokens: 0,
+            totalTokens: 0,
+            cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 },
+        };
+        for (const message of options.sessionManager?.messages ?? []) {
+            if (message.role === "assistant")
+                this.addUsage(message.usage);
+        }
+        const initialContext = options.sessionManager ? buildSessionContext(options.sessionManager.entries) : undefined;
+        this.agent = new Agent({
+            provider: options.provider,
+            model: options.model,
+            sessionId: this.sessionIdValue,
+            tools: [
+                createReadTool(options.allowedRoot),
+                createWriteTool(options.allowedRoot),
+                createEditTool(options.allowedRoot),
+                createBashTool(options.allowedRoot),
+            ],
+            now: this.now,
+            initialMessages: options.sessionManager?.messages,
+            initialContextMessages: initialContext?.messages,
+        });
+        this.agent.subscribe(async (event) => {
+            if (event.type !== "message_end" || this.sessionManager === undefined || this.persistenceError !== undefined) {
+                return;
+            }
+            try {
+                await this.sessionManager.appendMessage(event.message);
+            }
+            catch (cause) {
+                this.persistenceError = cause;
+                throw cause;
+            }
+        });
+        this.agent.subscribe(async (event) => {
+            if (event.type !== "message_end" || event.message.role !== "assistant")
+                return;
+            this.addUsage(event.message.usage);
+            await this.emitSession({ type: "usage_update", usage: this.usage });
+        });
+        this.agent.subscribe(async (event) => {
+            await this.emitSession(event);
+        });
+    }
+    get transcript() {
+        return this.agent.transcript;
+    }
+    get isStreaming() {
+        return this.promptActive;
+    }
+    get sessionId() {
+        return this.sessionIdValue;
+    }
+    get modelId() {
+        return this.model.id;
+    }
+    get allowedRoot() {
+        return this.allowedRootValue;
+    }
+    get availableModels() {
+        return structuredClone(this.provider.models);
+    }
+    setModel(modelId) {
+        if (this.promptActive)
+            throw new Error("Cannot change model while AgentSession is processing a prompt.");
+        const next = this.provider.models.find((model) => model.id === modelId);
+        if (!next)
+            throw new Error(`Unknown model "${modelId}" for provider "${this.provider.id}".`);
+        this.model = structuredClone(next);
+        this.agent.setModel(this.model);
+        this.contextBudget = resolveContextBudget(this.model);
+        this.keepRecentTokens = Math.min(this.keepRecentTokens, Math.max(1, Math.min(20_000, Math.floor(this.contextBudget.triggerTokens / 2))));
+        return structuredClone(this.model);
+    }
+    get compactionEnabled() {
+        return this.compactionEnabledValue;
+    }
+    setCompactionEnabled(enabled) {
+        if (this.promptActive)
+            throw new Error("Cannot change compaction while AgentSession is processing a prompt.");
+        this.compactionEnabledValue = enabled && this.sessionManager !== undefined;
+        return this.compactionEnabledValue;
+    }
+    get sessionFile() {
+        return this.sessionManager?.filePath;
+    }
+    get sessionDiagnostics() {
+        return this.sessionManager?.diagnostics ?? [];
+    }
+    get usage() {
+        return {
+            ...this.usageTotals,
+            cost: { ...this.usageTotals.cost },
+            estimatedContextTokens: estimateContextTokens(this.agent.contextMessages),
+            contextWindow: this.contextBudget.contextWindow,
+            reserveTokens: this.contextBudget.reserveTokens,
+            triggerTokens: this.contextBudget.triggerTokens,
+        };
+    }
+    async prompt(text, signal) {
+        if (this.persistenceError !== undefined) {
+            throw this.persistenceError;
+        }
+        if (this.promptActive) {
+            throw new Error("AgentSession is already processing a prompt.");
+        }
+        this.promptActive = true;
+        try {
+            await this.compactIfNeeded(text, signal);
+            return await this.agent.prompt(text, signal);
+        }
+        finally {
+            this.promptActive = false;
+        }
+    }
+    async compact(signal) {
+        if (this.promptActive)
+            throw new Error("AgentSession is already processing a prompt.");
+        if (!this.sessionManager)
+            throw new Error("Cannot compact without a persisted session.");
+        this.promptActive = true;
+        try {
+            await this.compactNow(signal, "manual");
+        }
+        finally {
+            this.promptActive = false;
+        }
+    }
+    subscribe(listener) {
+        return this.agent.subscribe(listener);
+    }
+    subscribeSession(listener) {
+        this.sessionListeners.add(listener);
+        return () => this.sessionListeners.delete(listener);
+    }
+    async emitSession(event) {
+        for (const listener of this.sessionListeners)
+            await listener(structuredClone(event));
+    }
+    addUsage(usage) {
+        this.usageTotals = {
+            requestCount: this.usageTotals.requestCount + 1,
+            inputTokens: this.usageTotals.inputTokens + usage.input,
+            outputTokens: this.usageTotals.outputTokens + usage.output,
+            cacheReadTokens: this.usageTotals.cacheReadTokens + usage.cacheRead,
+            cacheWriteTokens: this.usageTotals.cacheWriteTokens + usage.cacheWrite,
+            totalTokens: this.usageTotals.totalTokens + usage.totalTokens,
+            cost: {
+                input: this.usageTotals.cost.input + usage.cost.input,
+                output: this.usageTotals.cost.output + usage.cost.output,
+                cacheRead: this.usageTotals.cost.cacheRead + usage.cost.cacheRead,
+                cacheWrite: this.usageTotals.cost.cacheWrite + usage.cost.cacheWrite,
+                total: this.usageTotals.cost.total + usage.cost.total,
+            },
+        };
+    }
+    async compactIfNeeded(text, signal) {
+        if (!this.compactionEnabledValue || !this.sessionManager)
+            return;
+        const context = buildSessionContext(this.sessionManager.entries);
+        const pendingUser = {
+            role: "user",
+            content: [{ type: "text", text }],
+            timestamp: 0,
+        };
+        const estimatedTokens = estimateContextTokens([...context.messages, pendingUser]);
+        if (!shouldCompact(estimatedTokens, this.contextBudget))
+            return;
+        await this.compactNow(signal, "threshold");
+    }
+    async compactNow(signal, reason) {
+        if (!this.sessionManager)
+            throw new Error("Cannot compact without a persisted session.");
+        const context = buildSessionContext(this.sessionManager.entries);
+        await this.emitSession({ type: "compaction_start", reason });
+        const preparation = prepareCompaction(context.messages, this.keepRecentTokens);
+        const firstKeptEntryId = preparation ? context.sourceEntryIds[preparation.firstKeptMessageIndex] : undefined;
+        if (!preparation || typeof firstKeptEntryId !== "string") {
+            const errorMessage = reason === "manual"
+                ? "No valid compaction cut point was found for the current session."
+                : "Context limit reached but no valid compaction cut point was found.";
+            await this.emitSession({ type: "compaction_end", reason, success: false, errorMessage });
+            throw new Error(errorMessage);
+        }
+        let summary;
+        try {
+            summary = await generateCompactionSummary(preparation, this.provider, this.model, {
+                reserveTokens: this.contextBudget.reserveTokens,
+                signal,
+                now: this.now,
+                onUsage: (usage) => this.addUsage(usage),
+            });
+            await this.emitSession({ type: "usage_update", usage: this.usage });
+        }
+        catch (cause) {
+            const errorMessage = cause instanceof Error ? cause.message : String(cause);
+            await this.emitSession({ type: "compaction_end", reason, success: false, errorMessage });
+            throw cause;
+        }
+        try {
+            await this.sessionManager.appendSummary({
+                summary,
+                firstKeptEntryId,
+                tokensBefore: preparation.tokensBefore,
+            });
+        }
+        catch (cause) {
+            this.persistenceError = cause;
+            const errorMessage = cause instanceof Error ? cause.message : String(cause);
+            await this.emitSession({ type: "compaction_end", reason, success: false, errorMessage });
+            throw cause;
+        }
+        this.agent.replaceContext(buildSessionContext(this.sessionManager.entries).messages);
+        await this.emitSession({ type: "compaction_end", reason, success: true });
+    }
+}
+//# sourceMappingURL=session.js.map

package/dist/core/session.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"session.js","sourceRoot":"","sources":["../../src/core/session.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EACN,KAAK,EAGL,qBAAqB,EACrB,yBAAyB,EACzB,iBAAiB,EACjB,oBAAoB,EACpB,aAAa,GACb,MAAM,gBAAgB,CAAC;AAExB,OAAO,EAAE,mBAAmB,EAAE,MAAM,sBAAsB,CAAC;AAG3D,OAAO,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AACjD,OAAO,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AACjD,OAAO,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AACjD,OAAO,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAsCnD,MAAM,OAAO,YAAY;IACP,KAAK,CAAQ;IACb,gBAAgB,CAAS;IACzB,cAAc,CAAkB;IAChC,QAAQ,CAAW;IAC5B,KAAK,CAAQ;IACJ,cAAc,CAAS;IACvB,GAAG,CAAe;IAC3B,sBAAsB,CAAU;IAChC,aAAa,CAAgB;IAC7B,gBAAgB,CAAS;IACzB,WAAW,CAGjB;IACM,gBAAgB,CAAW;IAC3B,YAAY,GAAG,KAAK,CAAC;IACZ,gBAAgB,GAAG,IAAI,GAAG,EAAwB,CAAC;IAEpE,YAAY,OAA4B;QACvC,IAAI,CAAC,gBAAgB,GAAG,OAAO,CAAC,WAAW,CAAC;QAC5C,IAAI,CAAC,cAAc,GAAG,OAAO,CAAC,cAAc,CAAC;QAC7C,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC;QACjC,IAAI,CAAC,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;QAC3B,IAAI,CAAC,cAAc,GAAG,OAAO,CAAC,cAAc,EAAE,MAAM,CAAC,EAAE,IAAI,UAAU,EAAE,CAAC;QACxE,IAAI,CAAC,GAAG,GAAG,OAAO,CAAC,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC;QACnC,IAAI,CAAC,aAAa,GAAG,oBAAoB,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACzD,IAAI,OAAO,CAAC,UAAU,EAAE,OAAO,KAAK,SAAS,IAAI,OAAO,OAAO,CAAC,UAAU,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;YAClG,MAAM,IAAI,SAAS,CAAC,sCAAsC,CAAC,CAAC;QAC7D,CAAC;QACD,IAAI,CAAC,sBAAsB,GAAG,OAAO,CAAC,cAAc,KAAK,SAAS,IAAI,OAAO,CAAC,UAAU,EAAE,OAAO,KAAK,KAAK,CAAC;QAC5G,MAAM,uBAAuB,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,aAAa,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QAChH,IAAI,CAAC,gBAAgB,GAAG,OAAO,CAAC,UAAU,EAAE,gBAAgB,IAAI,uBAAuB,CAAC;QACxF,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,gBAAgB,CAAC,IAAI,IAAI,CAAC,gBAAgB,IAAI,CAAC,EAAE,CAAC;YAC5E,MAAM,IAAI,UAAU,CAAC,wDAAwD,CAAC,CAAC;QAChF,CAAC;QACD,IAAI,CAAC,WAAW,GAAG;YAClB,YAAY,EAAE,CAAC;YACf,WAAW,EAAE,CAAC;YACd,YAAY,EAAE,CAAC;YACf,eAAe,EAAE,CAAC;YAClB,gBAAgB,EAAE,CAAC;YACnB,WAAW,EAAE,CAAC;YACd,IAAI,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE;SACpE,CAAC;QACF,KAAK,MAAM,OAAO,IAAI,OAAO,CAAC,cAAc,EAAE,QAAQ,IAAI,EAAE,EAAE,CAAC;YAC9D,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW;gBAAE,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAChE,CAAC;QACD,MAAM,cAAc,GAAG,OAAO,CAAC,cAAc,CAAC,CAAC,CAAC,mBAAmB,CAAC,OAAO,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAChH,IAAI,CAAC,KAAK,GAAG,IAAI,KAAK,CAAC;YACtB,QAAQ,EAAE,OAAO,CAAC,QAAQ;YAC1B,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,SAAS,EAAE,IAAI,CAAC,cAAc;YAC9B,KAAK,EAAE;gBACN,cAAc,CAAC,OAAO,CAAC,WAAW,CAAC;gBACnC,eAAe,CAAC,OAAO,CAAC,WAAW,CAAC;gBACpC,cAAc,CAAC,OAAO,CAAC,WAAW,CAAC;gBACnC,cAAc,CAAC,OAAO,CAAC,WAAW,CAAC;aACnC;YACD,GAAG,EAAE,IAAI,CAAC,GAAG;YACb,eAAe,EAAE,OAAO,CAAC,cAAc,EAAE,QAAQ;YACjD,sBAAsB,EAAE,cAAc,EAAE,QAAQ;SAChD,CAAC,CAAC;QACH,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE;YACpC,IAAI,KAAK,CAAC,IAAI,KAAK,aAAa,IAAI,IAAI,CAAC,cAAc,KAAK,SAAS,IAAI,IAAI,CAAC,gBAAgB,KAAK,SAAS,EAAE,CAAC;gBAC9G,OAAO;YACR,CAAC;YACD,IAAI,CAAC;gBACJ,MAAM,IAAI,CAAC,cAAc,CAAC,aAAa,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YACxD,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBAChB,IAAI,CAAC,gBAAgB,GAAG,KAAK,CAAC;gBAC9B,MAAM,KAAK,CAAC;YACb,CAAC;QACF,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE;YACpC,IAAI,KAAK,CAAC,IAAI,KAAK,aAAa,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,KAAK,WAAW;gBAAE,OAAO;YAC/E,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YACnC,MAAM,IAAI,CAAC,WAAW,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC;QACrE,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE;YACpC,MAAM,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC;IACJ,CAAC;IAED,IAAI,UAAU;QACb,OAAO,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC;IAC9B,CAAC;IAED,IAAI,WAAW;QACd,OAAO,IAAI,CAAC,YAAY,CAAC;IAC1B,CAAC;IAED,IAAI,SAAS;QACZ,OAAO,IAAI,CAAC,cAAc,CAAC;IAC5B,CAAC;IAED,IAAI,OAAO;QACV,OAAO,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;IACtB,CAAC;IAED,IAAI,WAAW;QACd,OAAO,IAAI,CAAC,gBAAgB,CAAC;IAC9B,CAAC;IAED,IAAI,eAAe;QAClB,OAAO,eAAe,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;IAC9C,CAAC;IAED,QAAQ,CAAC,OAAe;QACvB,IAAI,IAAI,CAAC,YAAY;YAAE,MAAM,IAAI,KAAK,CAAC,gEAAgE,CAAC,CAAC;QACzG,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,KAAK,OAAO,CAAC,CAAC;QACxE,IAAI,CAAC,IAAI;YAAE,MAAM,IAAI,KAAK,CAAC,kBAAkB,OAAO,mBAAmB,IAAI,CAAC,QAAQ,CAAC,EAAE,IAAI,CAAC,CAAC;QAC7F,IAAI,CAAC,KAAK,GAAG,eAAe,CAAC,IAAI,CAAC,CAAC;QACnC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAChC,IAAI,CAAC,aAAa,GAAG,oBAAoB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACtD,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC,GAAG,CAC/B,IAAI,CAAC,gBAAgB,EACrB,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,aAAa,GAAG,CAAC,CAAC,CAAC,CAAC,CAC/E,CAAC;QACF,OAAO,eAAe,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACpC,CAAC;IAED,IAAI,iBAAiB;QACpB,OAAO,IAAI,CAAC,sBAAsB,CAAC;IACpC,CAAC;IAED,oBAAoB,CAAC,OAAgB;QACpC,IAAI,IAAI,CAAC,YAAY;YAAE,MAAM,IAAI,KAAK,CAAC,qEAAqE,CAAC,CAAC;QAC9G,IAAI,CAAC,sBAAsB,GAAG,OAAO,IAAI,IAAI,CAAC,cAAc,KAAK,SAAS,CAAC;QAC3E,OAAO,IAAI,CAAC,sBAAsB,CAAC;IACpC,CAAC;IAED,IAAI,WAAW;QACd,OAAO,IAAI,CAAC,cAAc,EAAE,QAAQ,CAAC;IACtC,CAAC;IAED,IAAI,kBAAkB;QACrB,OAAO,IAAI,CAAC,cAAc,EAAE,WAAW,IAAI,EAAE,CAAC;IAC/C,CAAC;IAED,IAAI,KAAK;QACR,OAAO;YACN,GAAG,IAAI,CAAC,WAAW;YACnB,IAAI,EAAE,EAAE,GAAG,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE;YAClC,sBAAsB,EAAE,qBAAqB,CAAC,IAAI,CAAC,KAAK,CAAC,eAAe,CAAC;YACzE,aAAa,EAAE,IAAI,CAAC,aAAa,CAAC,aAAa;YAC/C,aAAa,EAAE,IAAI,CAAC,aAAa,CAAC,aAAa;YAC/C,aAAa,EAAE,IAAI,CAAC,aAAa,CAAC,aAAa;SAC/C,CAAC;IACH,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,IAAY,EAAE,MAAoB;QAC9C,IAAI,IAAI,CAAC,gBAAgB,KAAK,SAAS,EAAE,CAAC;YACzC,MAAM,IAAI,CAAC,gBAAgB,CAAC;QAC7B,CAAC;QACD,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACvB,MAAM,IAAI,KAAK,CAAC,8CAA8C,CAAC,CAAC;QACjE,CAAC;QAED,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;QACzB,IAAI,CAAC;YACJ,MAAM,IAAI,CAAC,eAAe,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;YACzC,OAAO,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QAC9C,CAAC;gBAAS,CAAC;YACV,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC;QAC3B,CAAC;IACF,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAAoB;QACjC,IAAI,IAAI,CAAC,YAAY;YAAE,MAAM,IAAI,KAAK,CAAC,8CAA8C,CAAC,CAAC;QACvF,IAAI,CAAC,IAAI,CAAC,cAAc;YAAE,MAAM,IAAI,KAAK,CAAC,6CAA6C,CAAC,CAAC;QACzF,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;QACzB,IAAI,CAAC;YACJ,MAAM,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;QACzC,CAAC;gBAAS,CAAC;YACV,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC;QAC3B,CAAC;IACF,CAAC;IAED,SAAS,CAAC,QAAuB;QAChC,OAAO,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;IACvC,CAAC;IAED,gBAAgB,CAAC,QAA8B;QAC9C,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACpC,OAAO,GAAG,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IACrD,CAAC;IAEO,KAAK,CAAC,WAAW,CAAC,KAAwB;QACjD,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,gBAAgB;YAAE,MAAM,QAAQ,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC,CAAC;IACtF,CAAC;IAEO,QAAQ,CAAC,KAAY;QAC5B,IAAI,CAAC,WAAW,GAAG;YAClB,YAAY,EAAE,IAAI,CAAC,WAAW,CAAC,YAAY,GAAG,CAAC;YAC/C,WAAW,EAAE,IAAI,CAAC,WAAW,CAAC,WAAW,GAAG,KAAK,CAAC,KAAK;YACvD,YAAY,EAAE,IAAI,CAAC,WAAW,CAAC,YAAY,GAAG,KAAK,CAAC,MAAM;YAC1D,eAAe,EAAE,IAAI,CAAC,WAAW,CAAC,eAAe,GAAG,KAAK,CAAC,SAAS;YACnE,gBAAgB,EAAE,IAAI,CAAC,WAAW,CAAC,gBAAgB,GAAG,KAAK,CAAC,UAAU;YACtE,WAAW,EAAE,IAAI,CAAC,WAAW,CAAC,WAAW,GAAG,KAAK,CAAC,WAAW;YAC7D,IAAI,EAAE;gBACL,KAAK,EAAE,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,KAAK;gBACrD,MAAM,EAAE,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC,MAAM;gBACxD,SAAS,EAAE,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC,IAAI,CAAC,SAAS;gBACjE,UAAU,EAAE,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC,IAAI,CAAC,UAAU;gBACpE,KAAK,EAAE,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,KAAK;aACrD;SACD,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,eAAe,CAAC,IAAY,EAAE,MAAoB;QAC/D,IAAI,CAAC,IAAI,CAAC,sBAAsB,IAAI,CAAC,IAAI,CAAC,cAAc;YAAE,OAAO;QAEjE,MAAM,OAAO,GAAG,mBAAmB,CAAC,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QACjE,MAAM,WAAW,GAAY;YAC5B,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;YACjC,SAAS,EAAE,CAAC;SACZ,CAAC;QACF,MAAM,eAAe,GAAG,qBAAqB,CAAC,CAAC,GAAG,OAAO,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC,CAAC;QAClF,IAAI,CAAC,aAAa,CAAC,eAAe,EAAE,IAAI,CAAC,aAAa,CAAC;YAAE,OAAO;QAEhE,MAAM,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;IAC5C,CAAC;IAEO,KAAK,CAAC,UAAU,CAAC,MAA+B,EAAE,MAA8B;QACvF,IAAI,CAAC,IAAI,CAAC,cAAc;YAAE,MAAM,IAAI,KAAK,CAAC,6CAA6C,CAAC,CAAC;QACzF,MAAM,OAAO,GAAG,mBAAmB,CAAC,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QACjE,MAAM,IAAI,CAAC,WAAW,CAAC,EAAE,IAAI,EAAE,kBAAkB,EAAE,MAAM,EAAE,CAAC,CAAC;QAC7D,MAAM,WAAW,GAAG,iBAAiB,CAAC,OAAO,CAAC,QAAQ,EAAE,IAAI,CAAC,gBAAgB,CAAC,CAAC;QAC/E,MAAM,gBAAgB,GAAG,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,WAAW,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAC7G,IAAI,CAAC,WAAW,IAAI,OAAO,gBAAgB,KAAK,QAAQ,EAAE,CAAC;YAC1D,MAAM,YAAY,GACjB,MAAM,KAAK,QAAQ;gBAClB,CAAC,CAAC,kEAAkE;gBACpE,CAAC,CAAC,oEAAoE,CAAC;YACzE,MAAM,IAAI,CAAC,WAAW,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC,CAAC;YACzF,MAAM,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC;QAC/B,CAAC;QAED,IAAI,OAAe,CAAC;QACpB,IAAI,CAAC;YACJ,OAAO,GAAG,MAAM,yBAAyB,CAAC,WAAW,EAAE,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,KAAK,EAAE;gBACjF,aAAa,EAAE,IAAI,CAAC,aAAa,CAAC,aAAa;gBAC/C,MAAM;gBACN,GAAG,EAAE,IAAI,CAAC,GAAG;gBACb,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;aACxC,CAAC,CAAC;YACH,MAAM,IAAI,CAAC,WAAW,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC;QACrE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC5E,MAAM,IAAI,CAAC,WAAW,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC,CAAC;YACzF,MAAM,KAAK,CAAC;QACb,CAAC;QAED,IAAI,CAAC;YACJ,MAAM,IAAI,CAAC,cAAc,CAAC,aAAa,CAAC;gBACvC,OAAO;gBACP,gBAAgB;gBAChB,YAAY,EAAE,WAAW,CAAC,YAAY;aACtC,CAAC,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,IAAI,CAAC,gBAAgB,GAAG,KAAK,CAAC;YAC9B,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC5E,MAAM,IAAI,CAAC,WAAW,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC,CAAC;YACzF,MAAM,KAAK,CAAC;QACb,CAAC;QAED,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,mBAAmB,CAAC,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,QAAQ,CAAC,CAAC;QACrF,MAAM,IAAI,CAAC,WAAW,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;IAC3E,CAAC;CACD","sourcesContent":["import { randomUUID } from \"node:crypto\";\nimport {\n\tAgent,\n\ttype AgentListener,\n\ttype ContextBudget,\n\testimateContextTokens,\n\tgenerateCompactionSummary,\n\tprepareCompaction,\n\tresolveContextBudget,\n\tshouldCompact,\n} from \"@di-code/agent\";\nimport type { AssistantMessage, Message, Model, Provider, Usage } from \"@di-code/ai\";\nimport { buildSessionContext } from \"./context-builder.ts\";\nimport type { SessionManager } from \"./session/session-manager.ts\";\nimport type { SessionDiagnostic } from \"./session/types.ts\";\nimport { createBashTool } from \"./tools/bash.ts\";\nimport { createEditTool } from \"./tools/edit.ts\";\nimport { createReadTool } from \"./tools/read.ts\";\nimport { createWriteTool } from \"./tools/write.ts\";\n\nexport interface AgentSessionCompactionOptions {\n\treadonly enabled?: boolean;\n\treadonly keepRecentTokens?: number;\n}\n\nexport interface SessionUsage {\n\treadonly requestCount: number;\n\treadonly inputTokens: number;\n\treadonly outputTokens: number;\n\treadonly cacheReadTokens: number;\n\treadonly cacheWriteTokens: number;\n\treadonly totalTokens: number;\n\treadonly cost: Usage[\"cost\"];\n\treadonly estimatedContextTokens: number;\n\treadonly contextWindow: number;\n\treadonly reserveTokens: number;\n\treadonly triggerTokens: number;\n}\n\nexport interface AgentSessionOptions {\n\treadonly allowedRoot: string;\n\treadonly provider: Provider;\n\treadonly model: Model;\n\treadonly now?: () => number;\n\treadonly sessionManager?: SessionManager;\n\treadonly compaction?: AgentSessionCompactionOptions;\n}\n\nexport type AgentSessionEvent =\n\t| import(\"@di-code/agent\").AgentEvent\n\t| { type: \"compaction_start\"; reason: \"threshold\" | \"manual\" }\n\t| { type: \"compaction_end\"; reason: \"threshold\" | \"manual\"; success: boolean; errorMessage?: string }\n\t| { type: \"usage_update\"; usage: SessionUsage };\n\nexport type AgentSessionListener = (event: AgentSessionEvent) => void | Promise<void>;\n\nexport class AgentSession {\n\tprivate readonly agent: Agent;\n\tprivate readonly allowedRootValue: string;\n\tprivate readonly sessionManager?: SessionManager;\n\tprivate readonly provider: Provider;\n\tprivate model: Model;\n\tprivate readonly sessionIdValue: string;\n\tprivate readonly now: () => number;\n\tprivate compactionEnabledValue: boolean;\n\tprivate contextBudget: ContextBudget;\n\tprivate keepRecentTokens: number;\n\tprivate usageTotals: Omit<\n\t\tSessionUsage,\n\t\t\"estimatedContextTokens\" | \"contextWindow\" | \"reserveTokens\" | \"triggerTokens\"\n\t>;\n\tprivate persistenceError?: unknown;\n\tprivate promptActive = false;\n\tprivate readonly sessionListeners = new Set<AgentSessionListener>();\n\n\tconstructor(options: AgentSessionOptions) {\n\t\tthis.allowedRootValue = options.allowedRoot;\n\t\tthis.sessionManager = options.sessionManager;\n\t\tthis.provider = options.provider;\n\t\tthis.model = options.model;\n\t\tthis.sessionIdValue = options.sessionManager?.header.id ?? randomUUID();\n\t\tthis.now = options.now ?? Date.now;\n\t\tthis.contextBudget = resolveContextBudget(options.model);\n\t\tif (options.compaction?.enabled !== undefined && typeof options.compaction.enabled !== \"boolean\") {\n\t\t\tthrow new TypeError(\"compaction.enabled must be a boolean\");\n\t\t}\n\t\tthis.compactionEnabledValue = options.sessionManager !== undefined && options.compaction?.enabled !== false;\n\t\tconst defaultKeepRecentTokens = Math.max(1, Math.min(20_000, Math.floor(this.contextBudget.triggerTokens / 2)));\n\t\tthis.keepRecentTokens = options.compaction?.keepRecentTokens ?? defaultKeepRecentTokens;\n\t\tif (!Number.isInteger(this.keepRecentTokens) || this.keepRecentTokens <= 0) {\n\t\t\tthrow new RangeError(\"compaction.keepRecentTokens must be a positive integer\");\n\t\t}\n\t\tthis.usageTotals = {\n\t\t\trequestCount: 0,\n\t\t\tinputTokens: 0,\n\t\t\toutputTokens: 0,\n\t\t\tcacheReadTokens: 0,\n\t\t\tcacheWriteTokens: 0,\n\t\t\ttotalTokens: 0,\n\t\t\tcost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 },\n\t\t};\n\t\tfor (const message of options.sessionManager?.messages ?? []) {\n\t\t\tif (message.role === \"assistant\") this.addUsage(message.usage);\n\t\t}\n\t\tconst initialContext = options.sessionManager ? buildSessionContext(options.sessionManager.entries) : undefined;\n\t\tthis.agent = new Agent({\n\t\t\tprovider: options.provider,\n\t\t\tmodel: options.model,\n\t\t\tsessionId: this.sessionIdValue,\n\t\t\ttools: [\n\t\t\t\tcreateReadTool(options.allowedRoot),\n\t\t\t\tcreateWriteTool(options.allowedRoot),\n\t\t\t\tcreateEditTool(options.allowedRoot),\n\t\t\t\tcreateBashTool(options.allowedRoot),\n\t\t\t],\n\t\t\tnow: this.now,\n\t\t\tinitialMessages: options.sessionManager?.messages,\n\t\t\tinitialContextMessages: initialContext?.messages,\n\t\t});\n\t\tthis.agent.subscribe(async (event) => {\n\t\t\tif (event.type !== \"message_end\" || this.sessionManager === undefined || this.persistenceError !== undefined) {\n\t\t\t\treturn;\n\t\t\t}\n\t\t\ttry {\n\t\t\t\tawait this.sessionManager.appendMessage(event.message);\n\t\t\t} catch (cause) {\n\t\t\t\tthis.persistenceError = cause;\n\t\t\t\tthrow cause;\n\t\t\t}\n\t\t});\n\t\tthis.agent.subscribe(async (event) => {\n\t\t\tif (event.type !== \"message_end\" || event.message.role !== \"assistant\") return;\n\t\t\tthis.addUsage(event.message.usage);\n\t\t\tawait this.emitSession({ type: \"usage_update\", usage: this.usage });\n\t\t});\n\t\tthis.agent.subscribe(async (event) => {\n\t\t\tawait this.emitSession(event);\n\t\t});\n\t}\n\n\tget transcript(): readonly Message[] {\n\t\treturn this.agent.transcript;\n\t}\n\n\tget isStreaming(): boolean {\n\t\treturn this.promptActive;\n\t}\n\n\tget sessionId(): string {\n\t\treturn this.sessionIdValue;\n\t}\n\n\tget modelId(): string {\n\t\treturn this.model.id;\n\t}\n\n\tget allowedRoot(): string {\n\t\treturn this.allowedRootValue;\n\t}\n\n\tget availableModels(): readonly Model[] {\n\t\treturn structuredClone(this.provider.models);\n\t}\n\n\tsetModel(modelId: string): Model {\n\t\tif (this.promptActive) throw new Error(\"Cannot change model while AgentSession is processing a prompt.\");\n\t\tconst next = this.provider.models.find((model) => model.id === modelId);\n\t\tif (!next) throw new Error(`Unknown model \"${modelId}\" for provider \"${this.provider.id}\".`);\n\t\tthis.model = structuredClone(next);\n\t\tthis.agent.setModel(this.model);\n\t\tthis.contextBudget = resolveContextBudget(this.model);\n\t\tthis.keepRecentTokens = Math.min(\n\t\t\tthis.keepRecentTokens,\n\t\t\tMath.max(1, Math.min(20_000, Math.floor(this.contextBudget.triggerTokens / 2))),\n\t\t);\n\t\treturn structuredClone(this.model);\n\t}\n\n\tget compactionEnabled(): boolean {\n\t\treturn this.compactionEnabledValue;\n\t}\n\n\tsetCompactionEnabled(enabled: boolean): boolean {\n\t\tif (this.promptActive) throw new Error(\"Cannot change compaction while AgentSession is processing a prompt.\");\n\t\tthis.compactionEnabledValue = enabled && this.sessionManager !== undefined;\n\t\treturn this.compactionEnabledValue;\n\t}\n\n\tget sessionFile(): string | undefined {\n\t\treturn this.sessionManager?.filePath;\n\t}\n\n\tget sessionDiagnostics(): readonly SessionDiagnostic[] {\n\t\treturn this.sessionManager?.diagnostics ?? [];\n\t}\n\n\tget usage(): SessionUsage {\n\t\treturn {\n\t\t\t...this.usageTotals,\n\t\t\tcost: { ...this.usageTotals.cost },\n\t\t\testimatedContextTokens: estimateContextTokens(this.agent.contextMessages),\n\t\t\tcontextWindow: this.contextBudget.contextWindow,\n\t\t\treserveTokens: this.contextBudget.reserveTokens,\n\t\t\ttriggerTokens: this.contextBudget.triggerTokens,\n\t\t};\n\t}\n\n\tasync prompt(text: string, signal?: AbortSignal): Promise<AssistantMessage> {\n\t\tif (this.persistenceError !== undefined) {\n\t\t\tthrow this.persistenceError;\n\t\t}\n\t\tif (this.promptActive) {\n\t\t\tthrow new Error(\"AgentSession is already processing a prompt.\");\n\t\t}\n\n\t\tthis.promptActive = true;\n\t\ttry {\n\t\t\tawait this.compactIfNeeded(text, signal);\n\t\t\treturn await this.agent.prompt(text, signal);\n\t\t} finally {\n\t\t\tthis.promptActive = false;\n\t\t}\n\t}\n\n\tasync compact(signal?: AbortSignal): Promise<void> {\n\t\tif (this.promptActive) throw new Error(\"AgentSession is already processing a prompt.\");\n\t\tif (!this.sessionManager) throw new Error(\"Cannot compact without a persisted session.\");\n\t\tthis.promptActive = true;\n\t\ttry {\n\t\t\tawait this.compactNow(signal, \"manual\");\n\t\t} finally {\n\t\t\tthis.promptActive = false;\n\t\t}\n\t}\n\n\tsubscribe(listener: AgentListener): () => void {\n\t\treturn this.agent.subscribe(listener);\n\t}\n\n\tsubscribeSession(listener: AgentSessionListener): () => void {\n\t\tthis.sessionListeners.add(listener);\n\t\treturn () => this.sessionListeners.delete(listener);\n\t}\n\n\tprivate async emitSession(event: AgentSessionEvent): Promise<void> {\n\t\tfor (const listener of this.sessionListeners) await listener(structuredClone(event));\n\t}\n\n\tprivate addUsage(usage: Usage): void {\n\t\tthis.usageTotals = {\n\t\t\trequestCount: this.usageTotals.requestCount + 1,\n\t\t\tinputTokens: this.usageTotals.inputTokens + usage.input,\n\t\t\toutputTokens: this.usageTotals.outputTokens + usage.output,\n\t\t\tcacheReadTokens: this.usageTotals.cacheReadTokens + usage.cacheRead,\n\t\t\tcacheWriteTokens: this.usageTotals.cacheWriteTokens + usage.cacheWrite,\n\t\t\ttotalTokens: this.usageTotals.totalTokens + usage.totalTokens,\n\t\t\tcost: {\n\t\t\t\tinput: this.usageTotals.cost.input + usage.cost.input,\n\t\t\t\toutput: this.usageTotals.cost.output + usage.cost.output,\n\t\t\t\tcacheRead: this.usageTotals.cost.cacheRead + usage.cost.cacheRead,\n\t\t\t\tcacheWrite: this.usageTotals.cost.cacheWrite + usage.cost.cacheWrite,\n\t\t\t\ttotal: this.usageTotals.cost.total + usage.cost.total,\n\t\t\t},\n\t\t};\n\t}\n\n\tprivate async compactIfNeeded(text: string, signal?: AbortSignal): Promise<void> {\n\t\tif (!this.compactionEnabledValue || !this.sessionManager) return;\n\n\t\tconst context = buildSessionContext(this.sessionManager.entries);\n\t\tconst pendingUser: Message = {\n\t\t\trole: \"user\",\n\t\t\tcontent: [{ type: \"text\", text }],\n\t\t\ttimestamp: 0,\n\t\t};\n\t\tconst estimatedTokens = estimateContextTokens([...context.messages, pendingUser]);\n\t\tif (!shouldCompact(estimatedTokens, this.contextBudget)) return;\n\n\t\tawait this.compactNow(signal, \"threshold\");\n\t}\n\n\tprivate async compactNow(signal: AbortSignal | undefined, reason: \"threshold\" | \"manual\"): Promise<void> {\n\t\tif (!this.sessionManager) throw new Error(\"Cannot compact without a persisted session.\");\n\t\tconst context = buildSessionContext(this.sessionManager.entries);\n\t\tawait this.emitSession({ type: \"compaction_start\", reason });\n\t\tconst preparation = prepareCompaction(context.messages, this.keepRecentTokens);\n\t\tconst firstKeptEntryId = preparation ? context.sourceEntryIds[preparation.firstKeptMessageIndex] : undefined;\n\t\tif (!preparation || typeof firstKeptEntryId !== \"string\") {\n\t\t\tconst errorMessage =\n\t\t\t\treason === \"manual\"\n\t\t\t\t\t? \"No valid compaction cut point was found for the current session.\"\n\t\t\t\t\t: \"Context limit reached but no valid compaction cut point was found.\";\n\t\t\tawait this.emitSession({ type: \"compaction_end\", reason, success: false, errorMessage });\n\t\t\tthrow new Error(errorMessage);\n\t\t}\n\n\t\tlet summary: string;\n\t\ttry {\n\t\t\tsummary = await generateCompactionSummary(preparation, this.provider, this.model, {\n\t\t\t\treserveTokens: this.contextBudget.reserveTokens,\n\t\t\t\tsignal,\n\t\t\t\tnow: this.now,\n\t\t\t\tonUsage: (usage) => this.addUsage(usage),\n\t\t\t});\n\t\t\tawait this.emitSession({ type: \"usage_update\", usage: this.usage });\n\t\t} catch (cause) {\n\t\t\tconst errorMessage = cause instanceof Error ? cause.message : String(cause);\n\t\t\tawait this.emitSession({ type: \"compaction_end\", reason, success: false, errorMessage });\n\t\t\tthrow cause;\n\t\t}\n\n\t\ttry {\n\t\t\tawait this.sessionManager.appendSummary({\n\t\t\t\tsummary,\n\t\t\t\tfirstKeptEntryId,\n\t\t\t\ttokensBefore: preparation.tokensBefore,\n\t\t\t});\n\t\t} catch (cause) {\n\t\t\tthis.persistenceError = cause;\n\t\t\tconst errorMessage = cause instanceof Error ? cause.message : String(cause);\n\t\t\tawait this.emitSession({ type: \"compaction_end\", reason, success: false, errorMessage });\n\t\t\tthrow cause;\n\t\t}\n\n\t\tthis.agent.replaceContext(buildSessionContext(this.sessionManager.entries).messages);\n\t\tawait this.emitSession({ type: \"compaction_end\", reason, success: true });\n\t}\n}\n"]}

package/dist/core/tools/bash.d.ts

@@ -0,0 +1,39 @@
+import type { AgentTool } from "@di-code/agent";
+import { type Static, Type } from "@di-code/ai";
+export declare const DEFAULT_BASH_TIMEOUT_MS = 30000;
+export declare const MAX_BASH_TIMEOUT_MS = 300000;
+export declare const DEFAULT_BASH_MAX_OUTPUT_BYTES: number;
+export declare const bashParameters: Type.TObject<{
+    command: Type.TString;
+    timeoutMs: Type.TOptional<Type.TInteger>;
+}>;
+export type BashParameters = Static<typeof bashParameters>;
+export interface BashRunOptions {
+    readonly signal?: AbortSignal;
+    readonly timeoutMs: number;
+    readonly onStdout: (chunk: Buffer) => void;
+    readonly onStderr: (chunk: Buffer) => void;
+}
+export interface BashExecution {
+    readonly exitCode: number | null;
+    readonly timedOut: boolean;
+    readonly aborted: boolean;
+}
+export interface BashOperations {
+    run(command: string, cwd: string, options: BashRunOptions): Promise<BashExecution>;
+}
+export interface BashToolOptions {
+    readonly operations?: BashOperations;
+    readonly maxOutputBytes?: number;
+}
+export type BashTool = AgentTool<typeof bashParameters>;
+export interface ShellRuntime {
+    readonly executable: string;
+    readonly detached: boolean;
+    readonly description: string;
+    args(command: string): readonly string[];
+}
+export declare function resolveShellRuntime(platform?: NodeJS.Platform): ShellRuntime;
+export declare function createLocalBashOperations(): BashOperations;
+export declare function createBashTool(allowedRoot: string, options?: BashToolOptions): BashTool;
+//# sourceMappingURL=bash.d.ts.map

package/dist/core/tools/bash.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"bash.d.ts","sourceRoot":"","sources":["../../../src/core/tools/bash.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAChD,OAAO,EAAE,KAAK,MAAM,EAA0B,IAAI,EAAE,MAAM,aAAa,CAAC;AAExE,eAAO,MAAM,uBAAuB,QAAS,CAAC;AAC9C,eAAO,MAAM,mBAAmB,SAAU,CAAC;AAC3C,eAAO,MAAM,6BAA6B,QAAY,CAAC;AAEvD,eAAO,MAAM,cAAc;;;EAGzB,CAAC;AAEH,MAAM,MAAM,cAAc,GAAG,MAAM,CAAC,OAAO,cAAc,CAAC,CAAC;AAE3D,MAAM,WAAW,cAAc;IAC9B,QAAQ,CAAC,MAAM,CAAC,EAAE,WAAW,CAAC;IAC9B,QAAQ,CAAC,SAAS,EAAE,MAAM,CAAC;IAC3B,QAAQ,CAAC,QAAQ,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC;IAC3C,QAAQ,CAAC,QAAQ,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC;CAC3C;AAED,MAAM,WAAW,aAAa;IAC7B,QAAQ,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;IACjC,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC;IAC3B,QAAQ,CAAC,OAAO,EAAE,OAAO,CAAC;CAC1B;AAED,MAAM,WAAW,cAAc;IAC9B,GAAG,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,OAAO,EAAE,cAAc,GAAG,OAAO,CAAC,aAAa,CAAC,CAAC;CACnF;AAED,MAAM,WAAW,eAAe;IAC/B,QAAQ,CAAC,UAAU,CAAC,EAAE,cAAc,CAAC;IACrC,QAAQ,CAAC,cAAc,CAAC,EAAE,MAAM,CAAC;CACjC;AAED,MAAM,MAAM,QAAQ,GAAG,SAAS,CAAC,OAAO,cAAc,CAAC,CAAC;AAExD,MAAM,WAAW,YAAY;IAC5B,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC;IAC5B,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC;IAC3B,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAC;IAC7B,IAAI,CAAC,OAAO,EAAE,MAAM,GAAG,SAAS,MAAM,EAAE,CAAC;CACzC;AAoED,wBAAgB,mBAAmB,CAAC,QAAQ,GAAE,MAAM,CAAC,QAA2B,GAAG,YAAY,CAkB9F;AAyFD,wBAAgB,yBAAyB,IAAI,cAAc,CAE1D;AAaD,wBAAgB,cAAc,CAAC,WAAW,EAAE,MAAM,EAAE,OAAO,GAAE,eAAoB,GAAG,QAAQ,CAqC3F"}

package/dist/core/tools/bash.js

@@ -0,0 +1,208 @@
+import { spawn } from "node:child_process";
+import { realpath } from "node:fs/promises";
+import { StringDecoder } from "node:string_decoder";
+import { Type } from "@di-code/ai";
+export const DEFAULT_BASH_TIMEOUT_MS = 30_000;
+export const MAX_BASH_TIMEOUT_MS = 300_000;
+export const DEFAULT_BASH_MAX_OUTPUT_BYTES = 50 * 1024;
+export const bashParameters = Type.Object({
+    command: Type.String({ minLength: 1 }),
+    timeoutMs: Type.Optional(Type.Integer({ minimum: 1, maximum: MAX_BASH_TIMEOUT_MS })),
+});
+function assertTimeout(timeoutMs) {
+    if (!Number.isInteger(timeoutMs) || timeoutMs < 1 || timeoutMs > MAX_BASH_TIMEOUT_MS) {
+        throw new Error(`timeoutMs must be an integer between 1 and ${MAX_BASH_TIMEOUT_MS}`);
+    }
+}
+function assertMaxOutputBytes(maxOutputBytes) {
+    if (!Number.isInteger(maxOutputBytes) || maxOutputBytes < 1) {
+        throw new Error("maxOutputBytes must be a positive integer");
+    }
+}
+function createOutputCapture(maxBytes) {
+    const chunks = [];
+    let bytes = 0;
+    let truncated = false;
+    return {
+        append(chunk) {
+            const remaining = maxBytes - bytes;
+            if (remaining > 0) {
+                const kept = chunk.subarray(0, remaining);
+                chunks.push(Buffer.from(kept));
+                bytes += kept.length;
+            }
+            if (chunk.length > remaining)
+                truncated = true;
+        },
+        snapshot() {
+            const decoder = new StringDecoder("utf8");
+            return { text: decoder.write(Buffer.concat(chunks, bytes)), truncated };
+        },
+    };
+}
+function createPowerShellScript(command) {
+    return [
+        "$OutputEncoding = [Console]::OutputEncoding = [System.Text.UTF8Encoding]::new($false)",
+        "$script:diCodeSucceeded = $false",
+        "$script:diCodeExitCode = 0",
+        "& {",
+        command,
+        "$script:diCodeSucceeded = $?",
+        "$script:diCodeExitCode = $LASTEXITCODE",
+        "} | Out-Default",
+        "if ($script:diCodeSucceeded) { exit 0 }",
+        "if ($script:diCodeExitCode -is [int] -and $script:diCodeExitCode -ne 0) { exit $script:diCodeExitCode }",
+        "exit 1",
+    ].join("\n");
+}
+export function resolveShellRuntime(platform = process.platform) {
+    if (platform === "win32") {
+        return {
+            executable: "powershell.exe",
+            detached: false,
+            description: "Execute a Windows PowerShell command in the allowed root with timeout and bounded output. Use PowerShell syntax such as Get-ChildItem and Get-Content; do not use Bash syntax.",
+            args: (command) => ["-NoLogo", "-NoProfile", "-NonInteractive", "-Command", createPowerShellScript(command)],
+        };
+    }
+    return {
+        executable: "/bin/sh",
+        detached: true,
+        description: "Execute a POSIX shell command using /bin/sh in the allowed root with timeout and bounded output. Use portable POSIX syntax; Bash-only extensions may not be available.",
+        args: (command) => ["-c", command],
+    };
+}
+function waitForProcess(command, cwd, options) {
+    return new Promise((resolve, reject) => {
+        if (options.signal?.aborted) {
+            reject(new Error("Operation aborted"));
+            return;
+        }
+        const runtime = resolveShellRuntime();
+        const child = spawn(runtime.executable, [...runtime.args(command)], {
+            cwd,
+            detached: runtime.detached,
+            stdio: ["ignore", "pipe", "pipe"],
+            windowsHide: true,
+        });
+        let timedOut = false;
+        let aborted = false;
+        let settled = false;
+        let termination;
+        const requestTermination = () => {
+            if (child.pid !== undefined)
+                termination ??= killProcessTree(child.pid, child);
+        };
+        const onAbort = () => {
+            aborted = true;
+            requestTermination();
+        };
+        const timeout = setTimeout(() => {
+            timedOut = true;
+            requestTermination();
+        }, options.timeoutMs);
+        options.signal?.addEventListener("abort", onAbort, { once: true });
+        if (options.signal?.aborted)
+            onAbort();
+        child.stdout.on("data", options.onStdout);
+        child.stderr.on("data", options.onStderr);
+        const cleanup = () => {
+            clearTimeout(timeout);
+            options.signal?.removeEventListener("abort", onAbort);
+        };
+        child.once("error", (error) => {
+            if (settled)
+                return;
+            settled = true;
+            cleanup();
+            reject(error);
+        });
+        child.once("close", (exitCode) => {
+            if (settled)
+                return;
+            settled = true;
+            cleanup();
+            void (termination ?? Promise.resolve()).then(() => resolve({ exitCode, timedOut, aborted }), (error) => reject(error));
+        });
+    });
+}
+function waitForChild(child) {
+    return new Promise((resolve) => {
+        child.once("error", () => resolve());
+        child.once("close", () => resolve());
+    });
+}
+async function killProcessTree(pid, child) {
+    if (process.platform === "win32") {
+        const killer = spawn("taskkill", ["/F", "/T", "/PID", String(pid)], {
+            stdio: "ignore",
+            windowsHide: true,
+        });
+        await waitForChild(killer);
+        return;
+    }
+    try {
+        process.kill(-pid, "SIGKILL");
+    }
+    catch {
+        try {
+            child.kill("SIGKILL");
+        }
+        catch {
+            // The process already exited between the timeout and termination request.
+        }
+    }
+}
+export function createLocalBashOperations() {
+    return { run: waitForProcess };
+}
+function formatResult(stdout, stderr, execution) {
+    const status = {
+        exitCode: execution.exitCode,
+        timedOut: execution.timedOut,
+        aborted: execution.aborted,
+        stdoutTruncated: stdout.truncated,
+        stderrTruncated: stderr.truncated,
+    };
+    return `stdout:\n${stdout.text || "(empty)"}\n\nstderr:\n${stderr.text || "(empty)"}\n\nstatus: ${JSON.stringify(status)}`;
+}
+export function createBashTool(allowedRoot, options = {}) {
+    const operations = options.operations ?? createLocalBashOperations();
+    const maxOutputBytes = options.maxOutputBytes ?? DEFAULT_BASH_MAX_OUTPUT_BYTES;
+    const runtime = resolveShellRuntime();
+    assertMaxOutputBytes(maxOutputBytes);
+    return {
+        name: "bash",
+        description: runtime.description,
+        parameters: bashParameters,
+        async execute(_toolCallId, parameters, signal) {
+            if (signal?.aborted)
+                throw new Error("Operation aborted");
+            if (parameters.command.length === 0)
+                throw new Error("command must not be empty");
+            const timeoutMs = parameters.timeoutMs ?? DEFAULT_BASH_TIMEOUT_MS;
+            assertTimeout(timeoutMs);
+            const cwd = await realpath(allowedRoot);
+            if (signal?.aborted)
+                throw new Error("Operation aborted");
+            const stdout = createOutputCapture(maxOutputBytes);
+            const stderr = createOutputCapture(maxOutputBytes);
+            const executionResult = await operations.run(parameters.command, cwd, {
+                signal,
+                timeoutMs,
+                onStdout: (chunk) => stdout.append(chunk),
+                onStderr: (chunk) => stderr.append(chunk),
+            });
+            const execution = signal?.aborted ? { ...executionResult, aborted: true } : executionResult;
+            const output = formatResult(stdout.snapshot(), stderr.snapshot(), execution);
+            if (execution.aborted)
+                throw new Error(`Command aborted\n\n${output}`);
+            if (execution.timedOut)
+                throw new Error(`Command timed out after ${timeoutMs} ms\n\n${output}`);
+            if (execution.exitCode !== 0) {
+                throw new Error(`Command exited with code ${execution.exitCode ?? "unknown"}\n\n${output}`);
+            }
+            return [{ type: "text", text: output }];
+        },
+    };
+}
+//# sourceMappingURL=bash.js.map

package/dist/core/tools/bash.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"bash.js","sourceRoot":"","sources":["../../../src/core/tools/bash.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,MAAM,oBAAoB,CAAC;AAC3C,OAAO,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AAC5C,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AAEpD,OAAO,EAAuC,IAAI,EAAE,MAAM,aAAa,CAAC;AAExE,MAAM,CAAC,MAAM,uBAAuB,GAAG,MAAM,CAAC;AAC9C,MAAM,CAAC,MAAM,mBAAmB,GAAG,OAAO,CAAC;AAC3C,MAAM,CAAC,MAAM,6BAA6B,GAAG,EAAE,GAAG,IAAI,CAAC;AAEvD,MAAM,CAAC,MAAM,cAAc,GAAG,IAAI,CAAC,MAAM,CAAC;IACzC,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC;IACtC,SAAS,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,OAAO,EAAE,mBAAmB,EAAE,CAAC,CAAC;CACpF,CAAC,CAAC;AAgDH,SAAS,aAAa,CAAC,SAAiB;IACvC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,IAAI,SAAS,GAAG,CAAC,IAAI,SAAS,GAAG,mBAAmB,EAAE,CAAC;QACtF,MAAM,IAAI,KAAK,CAAC,8CAA8C,mBAAmB,EAAE,CAAC,CAAC;IACtF,CAAC;AACF,CAAC;AAED,SAAS,oBAAoB,CAAC,cAAsB;IACnD,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,cAAc,CAAC,IAAI,cAAc,GAAG,CAAC,EAAE,CAAC;QAC7D,MAAM,IAAI,KAAK,CAAC,2CAA2C,CAAC,CAAC;IAC9D,CAAC;AACF,CAAC;AAED,SAAS,mBAAmB,CAAC,QAAgB;IAI5C,MAAM,MAAM,GAAa,EAAE,CAAC;IAC5B,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,IAAI,SAAS,GAAG,KAAK,CAAC;IAEtB,OAAO;QACN,MAAM,CAAC,KAAK;YACX,MAAM,SAAS,GAAG,QAAQ,GAAG,KAAK,CAAC;YACnC,IAAI,SAAS,GAAG,CAAC,EAAE,CAAC;gBACnB,MAAM,IAAI,GAAG,KAAK,CAAC,QAAQ,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;gBAC1C,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;gBAC/B,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC;YACtB,CAAC;YACD,IAAI,KAAK,CAAC,MAAM,GAAG,SAAS;gBAAE,SAAS,GAAG,IAAI,CAAC;QAChD,CAAC;QACD,QAAQ;YACP,MAAM,OAAO,GAAG,IAAI,aAAa,CAAC,MAAM,CAAC,CAAC;YAC1C,OAAO,EAAE,IAAI,EAAE,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC;QACzE,CAAC;KACD,CAAC;AACH,CAAC;AAED,SAAS,sBAAsB,CAAC,OAAe;IAC9C,OAAO;QACN,uFAAuF;QACvF,kCAAkC;QAClC,4BAA4B;QAC5B,KAAK;QACL,OAAO;QACP,8BAA8B;QAC9B,wCAAwC;QACxC,iBAAiB;QACjB,yCAAyC;QACzC,yGAAyG;QACzG,QAAQ;KACR,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACd,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,WAA4B,OAAO,CAAC,QAAQ;IAC/E,IAAI,QAAQ,KAAK,OAAO,EAAE,CAAC;QAC1B,OAAO;YACN,UAAU,EAAE,gBAAgB;YAC5B,QAAQ,EAAE,KAAK;YACf,WAAW,EACV,gLAAgL;YACjL,IAAI,EAAE,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,SAAS,EAAE,YAAY,EAAE,iBAAiB,EAAE,UAAU,EAAE,sBAAsB,CAAC,OAAO,CAAC,CAAC;SAC5G,CAAC;IACH,CAAC;IAED,OAAO;QACN,UAAU,EAAE,SAAS;QACrB,QAAQ,EAAE,IAAI;QACd,WAAW,EACV,wKAAwK;QACzK,IAAI,EAAE,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,OAAO,CAAC;KAClC,CAAC;AACH,CAAC;AAED,SAAS,cAAc,CAAC,OAAe,EAAE,GAAW,EAAE,OAAuB;IAC5E,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACtC,IAAI,OAAO,CAAC,MAAM,EAAE,OAAO,EAAE,CAAC;YAC7B,MAAM,CAAC,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAC,CAAC;YACvC,OAAO;QACR,CAAC;QAED,MAAM,OAAO,GAAG,mBAAmB,EAAE,CAAC;QACtC,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,EAAE;YACnE,GAAG;YACH,QAAQ,EAAE,OAAO,CAAC,QAAQ;YAC1B,KAAK,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,CAAC;YACjC,WAAW,EAAE,IAAI;SACjB,CAAC,CAAC;QAEH,IAAI,QAAQ,GAAG,KAAK,CAAC;QACrB,IAAI,OAAO,GAAG,KAAK,CAAC;QACpB,IAAI,OAAO,GAAG,KAAK,CAAC;QACpB,IAAI,WAAsC,CAAC;QAE3C,MAAM,kBAAkB,GAAG,GAAS,EAAE;YACrC,IAAI,KAAK,CAAC,GAAG,KAAK,SAAS;gBAAE,WAAW,KAAK,eAAe,CAAC,KAAK,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;QAChF,CAAC,CAAC;QACF,MAAM,OAAO,GAAG,GAAS,EAAE;YAC1B,OAAO,GAAG,IAAI,CAAC;YACf,kBAAkB,EAAE,CAAC;QACtB,CAAC,CAAC;QACF,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,EAAE;YAC/B,QAAQ,GAAG,IAAI,CAAC;YAChB,kBAAkB,EAAE,CAAC;QACtB,CAAC,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC;QAEtB,OAAO,CAAC,MAAM,EAAE,gBAAgB,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QACnE,IAAI,OAAO,CAAC,MAAM,EAAE,OAAO;YAAE,OAAO,EAAE,CAAC;QACvC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC;QAC1C,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC;QAE1C,MAAM,OAAO,GAAG,GAAS,EAAE;YAC1B,YAAY,CAAC,OAAO,CAAC,CAAC;YACtB,OAAO,CAAC,MAAM,EAAE,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QACvD,CAAC,CAAC;QACF,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;YAC7B,IAAI,OAAO;gBAAE,OAAO;YACpB,OAAO,GAAG,IAAI,CAAC;YACf,OAAO,EAAE,CAAC;YACV,MAAM,CAAC,KAAK,CAAC,CAAC;QACf,CAAC,CAAC,CAAC;QACH,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE,EAAE;YAChC,IAAI,OAAO;gBAAE,OAAO;YACpB,OAAO,GAAG,IAAI,CAAC;YACf,OAAO,EAAE,CAAC;YACV,KAAK,CAAC,WAAW,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,IAAI,CAC3C,GAAG,EAAE,CAAC,OAAO,CAAC,EAAE,QAAQ,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,EAC9C,CAAC,KAAc,EAAE,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CACjC,CAAC;QACH,CAAC,CAAC,CAAC;IACJ,CAAC,CAAC,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CAAC,KAA+B;IACpD,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QAC9B,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC;QACrC,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,eAAe,CAAC,GAAW,EAAE,KAA+B;IAC1E,IAAI,OAAO,CAAC,QAAQ,KAAK,OAAO,EAAE,CAAC;QAClC,MAAM,MAAM,GAAG,KAAK,CAAC,UAAU,EAAE,CAAC,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC,EAAE;YACnE,KAAK,EAAE,QAAQ;YACf,WAAW,EAAE,IAAI;SACjB,CAAC,CAAC;QACH,MAAM,YAAY,CAAC,MAAM,CAAC,CAAC;QAC3B,OAAO;IACR,CAAC;IAED,IAAI,CAAC;QACJ,OAAO,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;IAC/B,CAAC;IAAC,MAAM,CAAC;QACR,IAAI,CAAC;YACJ,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACvB,CAAC;QAAC,MAAM,CAAC;YACR,0EAA0E;QAC3E,CAAC;IACF,CAAC;AACF,CAAC;AAED,MAAM,UAAU,yBAAyB;IACxC,OAAO,EAAE,GAAG,EAAE,cAAc,EAAE,CAAC;AAChC,CAAC;AAED,SAAS,YAAY,CAAC,MAAsB,EAAE,MAAsB,EAAE,SAAwB;IAC7F,MAAM,MAAM,GAAe;QAC1B,QAAQ,EAAE,SAAS,CAAC,QAAQ;QAC5B,QAAQ,EAAE,SAAS,CAAC,QAAQ;QAC5B,OAAO,EAAE,SAAS,CAAC,OAAO;QAC1B,eAAe,EAAE,MAAM,CAAC,SAAS;QACjC,eAAe,EAAE,MAAM,CAAC,SAAS;KACjC,CAAC;IACF,OAAO,YAAY,MAAM,CAAC,IAAI,IAAI,SAAS,gBAAgB,MAAM,CAAC,IAAI,IAAI,SAAS,eAAe,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC;AAC5H,CAAC;AAED,MAAM,UAAU,cAAc,CAAC,WAAmB,EAAE,UAA2B,EAAE;IAChF,MAAM,UAAU,GAAG,OAAO,CAAC,UAAU,IAAI,yBAAyB,EAAE,CAAC;IACrE,MAAM,cAAc,GAAG,OAAO,CAAC,cAAc,IAAI,6BAA6B,CAAC;IAC/E,MAAM,OAAO,GAAG,mBAAmB,EAAE,CAAC;IACtC,oBAAoB,CAAC,cAAc,CAAC,CAAC;IAErC,OAAO;QACN,IAAI,EAAE,MAAM;QACZ,WAAW,EAAE,OAAO,CAAC,WAAW;QAChC,UAAU,EAAE,cAAc;QAC1B,KAAK,CAAC,OAAO,CAAC,WAAW,EAAE,UAAU,EAAE,MAAM;YAC5C,IAAI,MAAM,EAAE,OAAO;gBAAE,MAAM,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAC;YAC1D,IAAI,UAAU,CAAC,OAAO,CAAC,MAAM,KAAK,CAAC;gBAAE,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;YAClF,MAAM,SAAS,GAAG,UAAU,CAAC,SAAS,IAAI,uBAAuB,CAAC;YAClE,aAAa,CAAC,SAAS,CAAC,CAAC;YACzB,MAAM,GAAG,GAAG,MAAM,QAAQ,CAAC,WAAW,CAAC,CAAC;YACxC,IAAI,MAAM,EAAE,OAAO;gBAAE,MAAM,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAC;YAE1D,MAAM,MAAM,GAAG,mBAAmB,CAAC,cAAc,CAAC,CAAC;YACnD,MAAM,MAAM,GAAG,mBAAmB,CAAC,cAAc,CAAC,CAAC;YACnD,MAAM,eAAe,GAAG,MAAM,UAAU,CAAC,GAAG,CAAC,UAAU,CAAC,OAAO,EAAE,GAAG,EAAE;gBACrE,MAAM;gBACN,SAAS;gBACT,QAAQ,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC;gBACzC,QAAQ,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC;aACzC,CAAC,CAAC;YACH,MAAM,SAAS,GAAG,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,EAAE,GAAG,eAAe,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,eAAe,CAAC;YAC5F,MAAM,MAAM,GAAG,YAAY,CAAC,MAAM,CAAC,QAAQ,EAAE,EAAE,MAAM,CAAC,QAAQ,EAAE,EAAE,SAAS,CAAC,CAAC;YAE7E,IAAI,SAAS,CAAC,OAAO;gBAAE,MAAM,IAAI,KAAK,CAAC,sBAAsB,MAAM,EAAE,CAAC,CAAC;YACvE,IAAI,SAAS,CAAC,QAAQ;gBAAE,MAAM,IAAI,KAAK,CAAC,2BAA2B,SAAS,UAAU,MAAM,EAAE,CAAC,CAAC;YAChG,IAAI,SAAS,CAAC,QAAQ,KAAK,CAAC,EAAE,CAAC;gBAC9B,MAAM,IAAI,KAAK,CAAC,4BAA4B,SAAS,CAAC,QAAQ,IAAI,SAAS,OAAO,MAAM,EAAE,CAAC,CAAC;YAC7F,CAAC;YACD,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;QACzC,CAAC;KACD,CAAC;AACH,CAAC","sourcesContent":["import { spawn } from \"node:child_process\";\nimport { realpath } from \"node:fs/promises\";\nimport { StringDecoder } from \"node:string_decoder\";\nimport type { AgentTool } from \"@di-code/agent\";\nimport { type Static, type ToolResultContent, Type } from \"@di-code/ai\";\n\nexport const DEFAULT_BASH_TIMEOUT_MS = 30_000;\nexport const MAX_BASH_TIMEOUT_MS = 300_000;\nexport const DEFAULT_BASH_MAX_OUTPUT_BYTES = 50 * 1024;\n\nexport const bashParameters = Type.Object({\n\tcommand: Type.String({ minLength: 1 }),\n\ttimeoutMs: Type.Optional(Type.Integer({ minimum: 1, maximum: MAX_BASH_TIMEOUT_MS })),\n});\n\nexport type BashParameters = Static<typeof bashParameters>;\n\nexport interface BashRunOptions {\n\treadonly signal?: AbortSignal;\n\treadonly timeoutMs: number;\n\treadonly onStdout: (chunk: Buffer) => void;\n\treadonly onStderr: (chunk: Buffer) => void;\n}\n\nexport interface BashExecution {\n\treadonly exitCode: number | null;\n\treadonly timedOut: boolean;\n\treadonly aborted: boolean;\n}\n\nexport interface BashOperations {\n\trun(command: string, cwd: string, options: BashRunOptions): Promise<BashExecution>;\n}\n\nexport interface BashToolOptions {\n\treadonly operations?: BashOperations;\n\treadonly maxOutputBytes?: number;\n}\n\nexport type BashTool = AgentTool<typeof bashParameters>;\n\nexport interface ShellRuntime {\n\treadonly executable: string;\n\treadonly detached: boolean;\n\treadonly description: string;\n\targs(command: string): readonly string[];\n}\n\ninterface OutputSnapshot {\n\treadonly text: string;\n\treadonly truncated: boolean;\n}\n\ninterface BashStatus {\n\treadonly exitCode: number | null;\n\treadonly timedOut: boolean;\n\treadonly aborted: boolean;\n\treadonly stdoutTruncated: boolean;\n\treadonly stderrTruncated: boolean;\n}\n\nfunction assertTimeout(timeoutMs: number): void {\n\tif (!Number.isInteger(timeoutMs) || timeoutMs < 1 || timeoutMs > MAX_BASH_TIMEOUT_MS) {\n\t\tthrow new Error(`timeoutMs must be an integer between 1 and ${MAX_BASH_TIMEOUT_MS}`);\n\t}\n}\n\nfunction assertMaxOutputBytes(maxOutputBytes: number): void {\n\tif (!Number.isInteger(maxOutputBytes) || maxOutputBytes < 1) {\n\t\tthrow new Error(\"maxOutputBytes must be a positive integer\");\n\t}\n}\n\nfunction createOutputCapture(maxBytes: number): {\n\tappend(chunk: Buffer): void;\n\tsnapshot(): OutputSnapshot;\n} {\n\tconst chunks: Buffer[] = [];\n\tlet bytes = 0;\n\tlet truncated = false;\n\n\treturn {\n\t\tappend(chunk) {\n\t\t\tconst remaining = maxBytes - bytes;\n\t\t\tif (remaining > 0) {\n\t\t\t\tconst kept = chunk.subarray(0, remaining);\n\t\t\t\tchunks.push(Buffer.from(kept));\n\t\t\t\tbytes += kept.length;\n\t\t\t}\n\t\t\tif (chunk.length > remaining) truncated = true;\n\t\t},\n\t\tsnapshot() {\n\t\t\tconst decoder = new StringDecoder(\"utf8\");\n\t\t\treturn { text: decoder.write(Buffer.concat(chunks, bytes)), truncated };\n\t\t},\n\t};\n}\n\nfunction createPowerShellScript(command: string): string {\n\treturn [\n\t\t\"$OutputEncoding = [Console]::OutputEncoding = [System.Text.UTF8Encoding]::new($false)\",\n\t\t\"$script:diCodeSucceeded = $false\",\n\t\t\"$script:diCodeExitCode = 0\",\n\t\t\"& {\",\n\t\tcommand,\n\t\t\"$script:diCodeSucceeded = $?\",\n\t\t\"$script:diCodeExitCode = $LASTEXITCODE\",\n\t\t\"} | Out-Default\",\n\t\t\"if ($script:diCodeSucceeded) { exit 0 }\",\n\t\t\"if ($script:diCodeExitCode -is [int] -and $script:diCodeExitCode -ne 0) { exit $script:diCodeExitCode }\",\n\t\t\"exit 1\",\n\t].join(\"\\n\");\n}\n\nexport function resolveShellRuntime(platform: NodeJS.Platform = process.platform): ShellRuntime {\n\tif (platform === \"win32\") {\n\t\treturn {\n\t\t\texecutable: \"powershell.exe\",\n\t\t\tdetached: false,\n\t\t\tdescription:\n\t\t\t\t\"Execute a Windows PowerShell command in the allowed root with timeout and bounded output. Use PowerShell syntax such as Get-ChildItem and Get-Content; do not use Bash syntax.\",\n\t\t\targs: (command) => [\"-NoLogo\", \"-NoProfile\", \"-NonInteractive\", \"-Command\", createPowerShellScript(command)],\n\t\t};\n\t}\n\n\treturn {\n\t\texecutable: \"/bin/sh\",\n\t\tdetached: true,\n\t\tdescription:\n\t\t\t\"Execute a POSIX shell command using /bin/sh in the allowed root with timeout and bounded output. Use portable POSIX syntax; Bash-only extensions may not be available.\",\n\t\targs: (command) => [\"-c\", command],\n\t};\n}\n\nfunction waitForProcess(command: string, cwd: string, options: BashRunOptions): Promise<BashExecution> {\n\treturn new Promise((resolve, reject) => {\n\t\tif (options.signal?.aborted) {\n\t\t\treject(new Error(\"Operation aborted\"));\n\t\t\treturn;\n\t\t}\n\n\t\tconst runtime = resolveShellRuntime();\n\t\tconst child = spawn(runtime.executable, [...runtime.args(command)], {\n\t\t\tcwd,\n\t\t\tdetached: runtime.detached,\n\t\t\tstdio: [\"ignore\", \"pipe\", \"pipe\"],\n\t\t\twindowsHide: true,\n\t\t});\n\n\t\tlet timedOut = false;\n\t\tlet aborted = false;\n\t\tlet settled = false;\n\t\tlet termination: Promise<void> | undefined;\n\n\t\tconst requestTermination = (): void => {\n\t\t\tif (child.pid !== undefined) termination ??= killProcessTree(child.pid, child);\n\t\t};\n\t\tconst onAbort = (): void => {\n\t\t\taborted = true;\n\t\t\trequestTermination();\n\t\t};\n\t\tconst timeout = setTimeout(() => {\n\t\t\ttimedOut = true;\n\t\t\trequestTermination();\n\t\t}, options.timeoutMs);\n\n\t\toptions.signal?.addEventListener(\"abort\", onAbort, { once: true });\n\t\tif (options.signal?.aborted) onAbort();\n\t\tchild.stdout.on(\"data\", options.onStdout);\n\t\tchild.stderr.on(\"data\", options.onStderr);\n\n\t\tconst cleanup = (): void => {\n\t\t\tclearTimeout(timeout);\n\t\t\toptions.signal?.removeEventListener(\"abort\", onAbort);\n\t\t};\n\t\tchild.once(\"error\", (error) => {\n\t\t\tif (settled) return;\n\t\t\tsettled = true;\n\t\t\tcleanup();\n\t\t\treject(error);\n\t\t});\n\t\tchild.once(\"close\", (exitCode) => {\n\t\t\tif (settled) return;\n\t\t\tsettled = true;\n\t\t\tcleanup();\n\t\t\tvoid (termination ?? Promise.resolve()).then(\n\t\t\t\t() => resolve({ exitCode, timedOut, aborted }),\n\t\t\t\t(error: unknown) => reject(error),\n\t\t\t);\n\t\t});\n\t});\n}\n\nfunction waitForChild(child: ReturnType<typeof spawn>): Promise<void> {\n\treturn new Promise((resolve) => {\n\t\tchild.once(\"error\", () => resolve());\n\t\tchild.once(\"close\", () => resolve());\n\t});\n}\n\nasync function killProcessTree(pid: number, child: ReturnType<typeof spawn>): Promise<void> {\n\tif (process.platform === \"win32\") {\n\t\tconst killer = spawn(\"taskkill\", [\"/F\", \"/T\", \"/PID\", String(pid)], {\n\t\t\tstdio: \"ignore\",\n\t\t\twindowsHide: true,\n\t\t});\n\t\tawait waitForChild(killer);\n\t\treturn;\n\t}\n\n\ttry {\n\t\tprocess.kill(-pid, \"SIGKILL\");\n\t} catch {\n\t\ttry {\n\t\t\tchild.kill(\"SIGKILL\");\n\t\t} catch {\n\t\t\t// The process already exited between the timeout and termination request.\n\t\t}\n\t}\n}\n\nexport function createLocalBashOperations(): BashOperations {\n\treturn { run: waitForProcess };\n}\n\nfunction formatResult(stdout: OutputSnapshot, stderr: OutputSnapshot, execution: BashExecution): string {\n\tconst status: BashStatus = {\n\t\texitCode: execution.exitCode,\n\t\ttimedOut: execution.timedOut,\n\t\taborted: execution.aborted,\n\t\tstdoutTruncated: stdout.truncated,\n\t\tstderrTruncated: stderr.truncated,\n\t};\n\treturn `stdout:\\n${stdout.text || \"(empty)\"}\\n\\nstderr:\\n${stderr.text || \"(empty)\"}\\n\\nstatus: ${JSON.stringify(status)}`;\n}\n\nexport function createBashTool(allowedRoot: string, options: BashToolOptions = {}): BashTool {\n\tconst operations = options.operations ?? createLocalBashOperations();\n\tconst maxOutputBytes = options.maxOutputBytes ?? DEFAULT_BASH_MAX_OUTPUT_BYTES;\n\tconst runtime = resolveShellRuntime();\n\tassertMaxOutputBytes(maxOutputBytes);\n\n\treturn {\n\t\tname: \"bash\",\n\t\tdescription: runtime.description,\n\t\tparameters: bashParameters,\n\t\tasync execute(_toolCallId, parameters, signal): Promise<ToolResultContent[]> {\n\t\t\tif (signal?.aborted) throw new Error(\"Operation aborted\");\n\t\t\tif (parameters.command.length === 0) throw new Error(\"command must not be empty\");\n\t\t\tconst timeoutMs = parameters.timeoutMs ?? DEFAULT_BASH_TIMEOUT_MS;\n\t\t\tassertTimeout(timeoutMs);\n\t\t\tconst cwd = await realpath(allowedRoot);\n\t\t\tif (signal?.aborted) throw new Error(\"Operation aborted\");\n\n\t\t\tconst stdout = createOutputCapture(maxOutputBytes);\n\t\t\tconst stderr = createOutputCapture(maxOutputBytes);\n\t\t\tconst executionResult = await operations.run(parameters.command, cwd, {\n\t\t\t\tsignal,\n\t\t\t\ttimeoutMs,\n\t\t\t\tonStdout: (chunk) => stdout.append(chunk),\n\t\t\t\tonStderr: (chunk) => stderr.append(chunk),\n\t\t\t});\n\t\t\tconst execution = signal?.aborted ? { ...executionResult, aborted: true } : executionResult;\n\t\t\tconst output = formatResult(stdout.snapshot(), stderr.snapshot(), execution);\n\n\t\t\tif (execution.aborted) throw new Error(`Command aborted\\n\\n${output}`);\n\t\t\tif (execution.timedOut) throw new Error(`Command timed out after ${timeoutMs} ms\\n\\n${output}`);\n\t\t\tif (execution.exitCode !== 0) {\n\t\t\t\tthrow new Error(`Command exited with code ${execution.exitCode ?? \"unknown\"}\\n\\n${output}`);\n\t\t\t}\n\t\t\treturn [{ type: \"text\", text: output }];\n\t\t},\n\t};\n}\n"]}

package/dist/core/tools/edit.d.ts

@@ -0,0 +1,18 @@
+import type { AgentTool } from "@di-code/agent";
+import { type Static, Type } from "@di-code/ai";
+export declare const editParameters: Type.TObject<{
+    path: Type.TString;
+    oldText: Type.TString;
+    newText: Type.TString;
+}>;
+export type EditParameters = Static<typeof editParameters>;
+export interface EditOperations {
+    readFile(filePath: string): Promise<Buffer>;
+    writeFile(filePath: string, content: string): Promise<void>;
+}
+export interface EditToolOptions {
+    readonly operations?: EditOperations;
+}
+export type EditTool = AgentTool<typeof editParameters>;
+export declare function createEditTool(allowedRoot: string, options?: EditToolOptions): EditTool;
+//# sourceMappingURL=edit.d.ts.map

package/dist/core/tools/edit.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"edit.d.ts","sourceRoot":"","sources":["../../../src/core/tools/edit.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAChD,OAAO,EAAE,KAAK,MAAM,EAA0B,IAAI,EAAE,MAAM,aAAa,CAAC;AAIxE,eAAO,MAAM,cAAc;;;;EAIzB,CAAC;AAEH,MAAM,MAAM,cAAc,GAAG,MAAM,CAAC,OAAO,cAAc,CAAC,CAAC;AAE3D,MAAM,WAAW,cAAc;IAC9B,QAAQ,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;IAC5C,SAAS,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CAC5D;AAED,MAAM,WAAW,eAAe;IAC/B,QAAQ,CAAC,UAAU,CAAC,EAAE,cAAc,CAAC;CACrC;AAED,MAAM,MAAM,QAAQ,GAAG,SAAS,CAAC,OAAO,cAAc,CAAC,CAAC;AA6DxD,wBAAgB,cAAc,CAAC,WAAW,EAAE,MAAM,EAAE,OAAO,GAAE,eAAoB,GAAG,QAAQ,CAyC3F"}