@markusylisiurunen/tau 0.2.35 → 0.2.37

package/dist/core/modes/rpc_server.js

@@ -0,0 +1,275 @@
+import { createInterface } from "node:readline";
+import { wrapCoreEvent } from "../events/types.js";
+import { createRpcErrorResponse, createRpcEventMessage, createRpcReadyMessage, createRpcSuccessResponse, parseRpcRequestLine, RPC_ERROR_CODES, RPC_METHODS, RPC_PROTOCOL_VERSION, serializeRpcMessage, } from "./rpc_protocol.js";
+export class RpcServer {
+    runtime;
+    send;
+    unsubscribeSubagent;
+    activeSubmit;
+    initialized = false;
+    rpcShutdown = false;
+    closed = false;
+    constructor(options) {
+        this.runtime = options.runtime;
+        this.send = options.send;
+        this.unsubscribeSubagent = this.runtime.session.onSubagentEvent((event) => {
+            if (this.rpcShutdown || this.closed) {
+                return;
+            }
+            this.sendMessage(createRpcEventMessage(wrapCoreEvent(event)));
+        });
+        if (options.emitReadyOnStart ?? true) {
+            this.emitReady();
+        }
+    }
+    async handleLine(line) {
+        if (this.closed) {
+            return;
+        }
+        const parsed = parseRpcRequestLine(line);
+        if (!parsed.ok) {
+            this.sendParseFailure(parsed);
+            return;
+        }
+        const request = parsed.request;
+        if (this.rpcShutdown && request.method !== "initialize") {
+            this.sendMessage(createRpcErrorResponse(request.id, RPC_ERROR_CODES.invalidRequest, "rpc server is shut down"));
+            return;
+        }
+        try {
+            switch (request.method) {
+                case "initialize":
+                    this.handleInitialize(request);
+                    return;
+                case "session.submit":
+                    await this.handleSubmit(request);
+                    return;
+                case "session.interrupt":
+                    this.handleInterrupt(request);
+                    return;
+                case "session.snapshot":
+                    this.handleSnapshot(request);
+                    return;
+                case "session.reset":
+                    await this.handleReset(request);
+                    return;
+                case "session.shutdown":
+                    await this.handleShutdown(request);
+                    return;
+            }
+        }
+        catch (error) {
+            this.sendMessage(createRpcErrorResponse(request.id, RPC_ERROR_CODES.internalError, "rpc request failed", {
+                cause: error instanceof Error ? error.message : String(error),
+            }));
+        }
+    }
+    async close(options = {}) {
+        if (this.closed) {
+            return;
+        }
+        this.closed = true;
+        this.unsubscribeSubagentListener();
+        if (options.interruptActiveSubmit && (this.activeSubmit || this.runtime.isTurnRunning)) {
+            this.runtime.interruptTurn();
+        }
+    }
+    async shutdown(options = {}) {
+        await this.close(options);
+    }
+    emitReady() {
+        this.sendMessage(createRpcReadyMessage({
+            sessionId: this.runtime.session.sessionId,
+        }));
+    }
+    sendParseFailure(parsed) {
+        this.sendMessage(createRpcErrorResponse(parsed.id, parsed.error.code, parsed.error.message, parsed.error.data));
+    }
+    handleInitialize(request) {
+        const result = {
+            protocolVersion: RPC_PROTOCOL_VERSION,
+            sessionId: this.runtime.session.sessionId,
+            methods: [...RPC_METHODS],
+            alreadyInitialized: this.initialized,
+        };
+        this.initialized = true;
+        this.sendMessage(createRpcSuccessResponse(request.id, result));
+    }
+    async handleSubmit(request) {
+        if (this.activeSubmit || this.runtime.isTurnRunning) {
+            this.sendMessage(createRpcErrorResponse(request.id, RPC_ERROR_CODES.busy, "a session turn is already running"));
+            return;
+        }
+        const addOptions = request.params.historyEntryId
+            ? { historyEntryId: request.params.historyEntryId }
+            : undefined;
+        const userHistoryEntryId = this.runtime.session.addUserText(request.params.text, addOptions);
+        const submitPromise = this.executeSubmit(request.id, userHistoryEntryId);
+        this.activeSubmit = submitPromise;
+        try {
+            await submitPromise;
+        }
+        finally {
+            if (this.activeSubmit === submitPromise) {
+                this.activeSubmit = undefined;
+            }
+        }
+    }
+    async executeSubmit(requestId, userHistoryEntryId) {
+        try {
+            const turnResult = await this.runtime.runTurn((event) => {
+                this.sendMessage(createRpcEventMessage(wrapCoreEvent(event), { requestId }));
+            });
+            const result = {
+                userHistoryEntryId,
+                turn: {
+                    aborted: turnResult.aborted,
+                },
+            };
+            this.sendMessage(createRpcSuccessResponse(requestId, result));
+        }
+        catch (error) {
+            this.sendMessage(createRpcErrorResponse(requestId, RPC_ERROR_CODES.internalError, "failed to run session turn", { cause: error instanceof Error ? error.message : String(error) }));
+        }
+    }
+    handleInterrupt(request) {
+        const interrupted = this.runtime.interruptTurn();
+        const result = {
+            interrupted,
+            isTurnRunning: this.runtime.isTurnRunning,
+        };
+        this.sendMessage(createRpcSuccessResponse(request.id, result));
+    }
+    handleSnapshot(request) {
+        const result = {
+            sessionId: this.runtime.session.sessionId,
+            isTurnRunning: this.runtime.isTurnRunning,
+            historyLength: this.runtime.session.history.length,
+            history: [...this.runtime.session.history],
+            historyEntries: this.runtime.session.historyEntries.map((entry) => ({
+                id: entry.id,
+                message: entry.message,
+            })),
+        };
+        this.sendMessage(createRpcSuccessResponse(request.id, result));
+    }
+    async handleReset(request) {
+        if (this.activeSubmit || this.runtime.isTurnRunning) {
+            this.runtime.interruptTurn();
+            if (this.activeSubmit) {
+                try {
+                    await this.activeSubmit;
+                }
+                catch {
+                    // ignore submit failure while resetting
+                }
+            }
+        }
+        const previousSessionId = this.runtime.session.sessionId;
+        this.runtime.session.reset();
+        const result = {
+            previousSessionId,
+            sessionId: this.runtime.session.sessionId,
+        };
+        this.sendMessage(createRpcSuccessResponse(request.id, result));
+    }
+    async handleShutdown(request) {
+        if (this.activeSubmit || this.runtime.isTurnRunning) {
+            this.runtime.interruptTurn();
+            if (this.activeSubmit) {
+                try {
+                    await this.activeSubmit;
+                }
+                catch {
+                    // ignore submit failure while shutting down
+                }
+            }
+        }
+        this.rpcShutdown = true;
+        this.unsubscribeSubagentListener();
+        const result = {
+            shutdown: true,
+        };
+        this.sendMessage(createRpcSuccessResponse(request.id, result));
+    }
+    unsubscribeSubagentListener() {
+        if (!this.unsubscribeSubagent) {
+            return;
+        }
+        this.unsubscribeSubagent();
+        this.unsubscribeSubagent = undefined;
+    }
+    sendMessage(message) {
+        this.send(serializeRpcMessage(message));
+    }
+}
+export async function runRpcServer(options) {
+    const server = new RpcServer({
+        runtime: options.runtime,
+        send: (line) => {
+            options.output.write(`${line}\n`);
+        },
+        emitReadyOnStart: true,
+    });
+    const lineReader = createInterface({
+        input: options.input,
+        crlfDelay: Number.POSITIVE_INFINITY,
+    });
+    const inFlightHandlers = new Set();
+    let firstError;
+    let stopRequested = false;
+    const trackInFlight = (promise) => {
+        inFlightHandlers.add(promise);
+        void promise
+            .catch((error) => {
+            if (firstError === undefined) {
+                firstError = error;
+            }
+        })
+            .finally(() => {
+            inFlightHandlers.delete(promise);
+        });
+    };
+    const requestStop = () => {
+        if (stopRequested) {
+            return;
+        }
+        stopRequested = true;
+        lineReader.close();
+        if (typeof options.input.pause === "function") {
+            options.input.pause();
+        }
+    };
+    const onLine = (line) => {
+        const handler = server.handleLine(line);
+        trackInFlight(handler);
+    };
+    const onAbort = () => {
+        requestStop();
+    };
+    lineReader.on("line", onLine);
+    const closePromise = new Promise((resolve) => {
+        lineReader.once("close", resolve);
+    });
+    if (options.signal) {
+        if (options.signal.aborted) {
+            requestStop();
+        }
+        options.signal.addEventListener("abort", onAbort);
+    }
+    try {
+        await closePromise;
+    }
+    finally {
+        lineReader.off("line", onLine);
+        options.signal?.removeEventListener("abort", onAbort);
+        await server.shutdown({ interruptActiveSubmit: true });
+        if (inFlightHandlers.size > 0) {
+            await Promise.allSettled([...inFlightHandlers]);
+        }
+    }
+    if (firstError !== undefined) {
+        throw firstError;
+    }
+}
+//# sourceMappingURL=rpc_server.js.map

package/dist/core/modes/rpc_server.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"rpc_server.js","sourceRoot":"","sources":["../../../src/core/modes/rpc_server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AAEhD,OAAO,EAAkB,aAAa,EAAE,MAAM,oBAAoB,CAAC;AAGnE,OAAO,EACL,sBAAsB,EACtB,qBAAqB,EACrB,qBAAqB,EACrB,wBAAwB,EACxB,mBAAmB,EACnB,eAAe,EACf,WAAW,EACX,oBAAoB,EAKpB,mBAAmB,GACpB,MAAM,mBAAmB,CAAC;AA0B3B,MAAM,OAAO,SAAS;IACH,OAAO,CAAmB;IAC1B,IAAI,CAAyB;IACtC,mBAAmB,CAAc;IACjC,YAAY,CAAiB;IAC7B,WAAW,GAAG,KAAK,CAAC;IACpB,WAAW,GAAG,KAAK,CAAC;IACpB,MAAM,GAAG,KAAK,CAAC;IAEvB,YAAY,OAAyB;QACnC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;QAC/B,IAAI,CAAC,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;QAEzB,IAAI,CAAC,mBAAmB,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,KAAK,EAAE,EAAE;YACxE,IAAI,IAAI,CAAC,WAAW,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;gBACpC,OAAO;YACT,CAAC;YACD,IAAI,CAAC,WAAW,CAAC,qBAAqB,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,IAAI,OAAO,CAAC,gBAAgB,IAAI,IAAI,EAAE,CAAC;YACrC,IAAI,CAAC,SAAS,EAAE,CAAC;QACnB,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,IAAY;QAC3B,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,OAAO;QACT,CAAC;QAED,MAAM,MAAM,GAAG,mBAAmB,CAAC,IAAI,CAAC,CAAC;QACzC,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,CAAC;YACf,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC;YAC9B,OAAO;QACT,CAAC;QAED,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC;QAE/B,IAAI,IAAI,CAAC,WAAW,IAAI,OAAO,CAAC,MAAM,KAAK,YAAY,EAAE,CAAC;YACxD,IAAI,CAAC,WAAW,CACd,sBAAsB,CACpB,OAAO,CAAC,EAAE,EACV,eAAe,CAAC,cAAc,EAC9B,yBAAyB,CAC1B,CACF,CAAC;YACF,OAAO;QACT,CAAC;QAED,IAAI,CAAC;YACH,QAAQ,OAAO,CAAC,MAAM,EAAE,CAAC;gBACvB,KAAK,YAAY;oBACf,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;oBAC/B,OAAO;gBACT,KAAK,gBAAgB;oBACnB,MAAM,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;oBACjC,OAAO;gBACT,KAAK,mBAAmB;oBACtB,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;oBAC9B,OAAO;gBACT,KAAK,kBAAkB;oBACrB,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;oBAC7B,OAAO;gBACT,KAAK,eAAe;oBAClB,MAAM,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;oBAChC,OAAO;gBACT,KAAK,kBAAkB;oBACrB,MAAM,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;oBACnC,OAAO;YACX,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,WAAW,CACd,sBAAsB,CAAC,OAAO,CAAC,EAAE,EAAE,eAAe,CAAC,aAAa,EAAE,oBAAoB,EAAE;gBACtF,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;aAC9D,CAAC,CACH,CAAC;QACJ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,UAA+C,EAAE;QAC3D,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,OAAO;QACT,CAAC;QAED,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QACnB,IAAI,CAAC,2BAA2B,EAAE,CAAC;QAEnC,IAAI,OAAO,CAAC,qBAAqB,IAAI,CAAC,IAAI,CAAC,YAAY,IAAI,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,EAAE,CAAC;YACvF,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,CAAC;QAC/B,CAAC;IACH,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,UAA+C,EAAE;QAC9D,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IAC5B,CAAC;IAEO,SAAS;QACf,IAAI,CAAC,WAAW,CACd,qBAAqB,CAAC;YACpB,SAAS,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS;SAC1C,CAAC,CACH,CAAC;IACJ,CAAC;IAEO,gBAAgB,CAAC,MAAuB;QAC9C,IAAI,CAAC,WAAW,CACd,sBAAsB,CAAC,MAAM,CAAC,EAAE,EAAE,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAC9F,CAAC;IACJ,CAAC;IAEO,gBAAgB,CAAC,OAA6D;QACpF,MAAM,MAAM,GAAoC;YAC9C,eAAe,EAAE,oBAAoB;YACrC,SAAS,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS;YACzC,OAAO,EAAE,CAAC,GAAG,WAAW,CAAC;YACzB,kBAAkB,EAAE,IAAI,CAAC,WAAW;SACrC,CAAC;QAEF,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;QACxB,IAAI,CAAC,WAAW,CAAC,wBAAwB,CAAC,OAAO,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC;IACjE,CAAC;IAEO,KAAK,CAAC,YAAY,CACxB,OAAiE;QAEjE,IAAI,IAAI,CAAC,YAAY,IAAI,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,CAAC;YACpD,IAAI,CAAC,WAAW,CACd,sBAAsB,CACpB,OAAO,CAAC,EAAE,EACV,eAAe,CAAC,IAAI,EACpB,mCAAmC,CACpC,CACF,CAAC;YACF,OAAO;QACT,CAAC;QAED,MAAM,UAAU,GAAG,OAAO,CAAC,MAAM,CAAC,cAAc;YAC9C,CAAC,CAAC,EAAE,cAAc,EAAE,OAAO,CAAC,MAAM,CAAC,cAAc,EAAE;YACnD,CAAC,CAAC,SAAS,CAAC;QACd,MAAM,kBAAkB,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,WAAW,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QAE7F,MAAM,aAAa,GAAG,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,EAAE,EAAE,kBAAkB,CAAC,CAAC;QACzE,IAAI,CAAC,YAAY,GAAG,aAAa,CAAC;QAElC,IAAI,CAAC;YACH,MAAM,aAAa,CAAC;QACtB,CAAC;gBAAS,CAAC;YACT,IAAI,IAAI,CAAC,YAAY,KAAK,aAAa,EAAE,CAAC;gBACxC,IAAI,CAAC,YAAY,GAAG,SAAS,CAAC;YAChC,CAAC;QACH,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,aAAa,CAAC,SAAuB,EAAE,kBAA0B;QAC7E,IAAI,CAAC;YACH,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE;gBACtD,IAAI,CAAC,WAAW,CAAC,qBAAqB,CAAC,aAAa,CAAC,KAAK,CAAC,EAAE,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC;YAC/E,CAAC,CAAC,CAAC;YAEH,MAAM,MAAM,GAAwC;gBAClD,kBAAkB;gBAClB,IAAI,EAAE;oBACJ,OAAO,EAAE,UAAU,CAAC,OAAO;iBAC5B;aACF,CAAC;YAEF,IAAI,CAAC,WAAW,CAAC,wBAAwB,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC,CAAC;QAChE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,WAAW,CACd,sBAAsB,CACpB,SAAS,EACT,eAAe,CAAC,aAAa,EAC7B,4BAA4B,EAC5B,EAAE,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAClE,CACF,CAAC;QACJ,CAAC;IACH,CAAC;IAEO,eAAe,CACrB,OAAoE;QAEpE,MAAM,WAAW,GAAG,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,CAAC;QAEjD,MAAM,MAAM,GAA2C;YACrD,WAAW;YACX,aAAa,EAAE,IAAI,CAAC,OAAO,CAAC,aAAa;SAC1C,CAAC;QAEF,IAAI,CAAC,WAAW,CAAC,wBAAwB,CAAC,OAAO,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC;IACjE,CAAC;IAEO,cAAc,CACpB,OAAmE;QAEnE,MAAM,MAAM,GAA0C;YACpD,SAAS,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS;YACzC,aAAa,EAAE,IAAI,CAAC,OAAO,CAAC,aAAa;YACzC,aAAa,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM;YAClD,OAAO,EAAE,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC;YAC1C,cAAc,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;gBAClE,EAAE,EAAE,KAAK,CAAC,EAAE;gBACZ,OAAO,EAAE,KAAK,CAAC,OAAO;aACvB,CAAC,CAAC;SACJ,CAAC;QAEF,IAAI,CAAC,WAAW,CAAC,wBAAwB,CAAC,OAAO,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC;IACjE,CAAC;IAEO,KAAK,CAAC,WAAW,CACvB,OAAgE;QAEhE,IAAI,IAAI,CAAC,YAAY,IAAI,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,CAAC;YACpD,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,CAAC;YAC7B,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;gBACtB,IAAI,CAAC;oBACH,MAAM,IAAI,CAAC,YAAY,CAAC;gBAC1B,CAAC;gBAAC,MAAM,CAAC;oBACP,wCAAwC;gBAC1C,CAAC;YACH,CAAC;QACH,CAAC;QAED,MAAM,iBAAiB,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC;QACzD,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QAE7B,MAAM,MAAM,GAAuC;YACjD,iBAAiB;YACjB,SAAS,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS;SAC1C,CAAC;QAEF,IAAI,CAAC,WAAW,CAAC,wBAAwB,CAAC,OAAO,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC;IACjE,CAAC;IAEO,KAAK,CAAC,cAAc,CAC1B,OAAmE;QAEnE,IAAI,IAAI,CAAC,YAAY,IAAI,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,CAAC;YACpD,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,CAAC;YAC7B,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;gBACtB,IAAI,CAAC;oBACH,MAAM,IAAI,CAAC,YAAY,CAAC;gBAC1B,CAAC;gBAAC,MAAM,CAAC;oBACP,4CAA4C;gBAC9C,CAAC;YACH,CAAC;QACH,CAAC;QAED,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;QACxB,IAAI,CAAC,2BAA2B,EAAE,CAAC;QAEnC,MAAM,MAAM,GAA0C;YACpD,QAAQ,EAAE,IAAI;SACf,CAAC;QAEF,IAAI,CAAC,WAAW,CAAC,wBAAwB,CAAC,OAAO,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC;IACjE,CAAC;IAEO,2BAA2B;QACjC,IAAI,CAAC,IAAI,CAAC,mBAAmB,EAAE,CAAC;YAC9B,OAAO;QACT,CAAC;QAED,IAAI,CAAC,mBAAmB,EAAE,CAAC;QAC3B,IAAI,CAAC,mBAAmB,GAAG,SAAS,CAAC;IACvC,CAAC;IAEO,WAAW,CAAC,OAAkD;QACpE,IAAI,CAAC,IAAI,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC,CAAC;IAC1C,CAAC;CACF;AAED,MAAM,CAAC,KAAK,UAAU,YAAY,CAAC,OAA4B;IAC7D,MAAM,MAAM,GAAG,IAAI,SAAS,CAAC;QAC3B,OAAO,EAAE,OAAO,CAAC,OAAO;QACxB,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE;YACb,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,IAAI,IAAI,CAAC,CAAC;QACpC,CAAC;QACD,gBAAgB,EAAE,IAAI;KACvB,CAAC,CAAC;IAEH,MAAM,UAAU,GAAG,eAAe,CAAC;QACjC,KAAK,EAAE,OAAO,CAAC,KAAK;QACpB,SAAS,EAAE,MAAM,CAAC,iBAAiB;KACpC,CAAC,CAAC;IAEH,MAAM,gBAAgB,GAAG,IAAI,GAAG,EAAiB,CAAC;IAClD,IAAI,UAAmB,CAAC;IACxB,IAAI,aAAa,GAAG,KAAK,CAAC;IAE1B,MAAM,aAAa,GAAG,CAAC,OAAsB,EAAE,EAAE;QAC/C,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAC9B,KAAK,OAAO;aACT,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;YACf,IAAI,UAAU,KAAK,SAAS,EAAE,CAAC;gBAC7B,UAAU,GAAG,KAAK,CAAC;YACrB,CAAC;QACH,CAAC,CAAC;aACD,OAAO,CAAC,GAAG,EAAE;YACZ,gBAAgB,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QACnC,CAAC,CAAC,CAAC;IACP,CAAC,CAAC;IAEF,MAAM,WAAW,GAAG,GAAG,EAAE;QACvB,IAAI,aAAa,EAAE,CAAC;YAClB,OAAO;QACT,CAAC;QAED,aAAa,GAAG,IAAI,CAAC;QACrB,UAAU,CAAC,KAAK,EAAE,CAAC;QAEnB,IAAI,OAAO,OAAO,CAAC,KAAK,CAAC,KAAK,KAAK,UAAU,EAAE,CAAC;YAC9C,OAAO,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;QACxB,CAAC;IACH,CAAC,CAAC;IAEF,MAAM,MAAM,GAAG,CAAC,IAAY,EAAE,EAAE;QAC9B,MAAM,OAAO,GAAG,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;QACxC,aAAa,CAAC,OAAO,CAAC,CAAC;IACzB,CAAC,CAAC;IAEF,MAAM,OAAO,GAAG,GAAG,EAAE;QACnB,WAAW,EAAE,CAAC;IAChB,CAAC,CAAC;IAEF,UAAU,CAAC,EAAE,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAE9B,MAAM,YAAY,GAAG,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE;QACjD,UAAU,CAAC,IAAI,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;QACnB,IAAI,OAAO,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YAC3B,WAAW,EAAE,CAAC;QAChB,CAAC;QACD,OAAO,CAAC,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IACpD,CAAC;IAED,IAAI,CAAC;QACH,MAAM,YAAY,CAAC;IACrB,CAAC;YAAS,CAAC;QACT,UAAU,CAAC,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAC/B,OAAO,CAAC,MAAM,EAAE,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QAEtD,MAAM,MAAM,CAAC,QAAQ,CAAC,EAAE,qBAAqB,EAAE,IAAI,EAAE,CAAC,CAAC;QAEvD,IAAI,gBAAgB,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;YAC9B,MAAM,OAAO,CAAC,UAAU,CAAC,CAAC,GAAG,gBAAgB,CAAC,CAAC,CAAC;QAClD,CAAC;IACH,CAAC;IAED,IAAI,UAAU,KAAK,SAAS,EAAE,CAAC;QAC7B,MAAM,UAAU,CAAC;IACnB,CAAC;AACH,CAAC"}

package/dist/core/runtime/chat_runtime.js

@@ -0,0 +1,124 @@
+import { CoreSession } from "../session/core_session.js";
+import { ConversationTurnRuntime, } from "./conversation_turn_runtime.js";
+import { composeSessionPrompts } from "./session_prompt_composer.js";
+export class ChatRuntime {
+    sessionInstance;
+    turnRuntime;
+    currentPersona;
+    riskLevel;
+    promptContext;
+    environment;
+    latestPromptComposition;
+    static create(options) {
+        const promptComposition = composeSessionPrompts({
+            persona: options.persona,
+            riskLevel: options.riskLevel,
+            cwd: options.promptContext.cwd,
+            datetime: new Date(options.environment.now()).toISOString(),
+            platform: options.environment.platform(),
+            nodeVersion: options.environment.nodeVersion(),
+            skillsBlock: options.promptContext.skillsBlock,
+            projectContextBlock: options.promptContext.projectContextBlock,
+            sandboxEnabled: options.promptContext.sandboxEnabled,
+            sandboxEnvironmentInfo: options.promptContext.sandboxEnvironmentInfo,
+        });
+        const session = new CoreSession({
+            persona: options.persona,
+            systemPrompt: promptComposition.baseSystemPrompt,
+            subagentPrompts: promptComposition.subagentPrompts,
+            riskLevel: options.riskLevel,
+            toolRegistry: options.toolRegistry,
+            config: options.config,
+            deps: options.deps,
+        });
+        return new ChatRuntime({
+            session,
+            persona: options.persona,
+            riskLevel: options.riskLevel,
+            promptContext: options.promptContext,
+            environment: options.environment,
+            initialPromptComposition: promptComposition,
+        });
+    }
+    constructor(options) {
+        this.sessionInstance = options.session;
+        this.turnRuntime = new ConversationTurnRuntime(this.sessionInstance);
+        this.currentPersona = options.persona;
+        this.riskLevel = options.riskLevel;
+        this.promptContext = { ...options.promptContext };
+        this.environment = options.environment;
+        this.latestPromptComposition = options.initialPromptComposition ?? this.composePromptSet();
+        if (!options.initialPromptComposition) {
+            this.sessionInstance.setPersona(this.currentPersona, this.latestPromptComposition.baseSystemPrompt, this.latestPromptComposition.subagentPrompts);
+        }
+    }
+    get session() {
+        return this.sessionInstance;
+    }
+    get isTurnRunning() {
+        return this.turnRuntime.isRunning;
+    }
+    get promptComposition() {
+        return this.latestPromptComposition;
+    }
+    runTurn(onEvent) {
+        return this.turnRuntime.run(onEvent);
+    }
+    interruptTurn() {
+        return this.turnRuntime.interrupt();
+    }
+    setConfig(config) {
+        this.sessionInstance.setConfig(config);
+    }
+    setRiskLevel(level) {
+        const previous = this.riskLevel;
+        this.riskLevel = level;
+        this.sessionInstance.setRiskLevel(level);
+        if (previous !== level) {
+            this.rebuildSubagentPrompts();
+        }
+    }
+    setPersona(persona, options) {
+        this.currentPersona = persona;
+        this.rebuildSystemPrompts(options);
+    }
+    updatePromptContext(context) {
+        this.promptContext = {
+            ...this.promptContext,
+            ...context,
+        };
+    }
+    rebuildSystemPrompts(options) {
+        if (options?.skillsBlock !== undefined) {
+            this.promptContext.skillsBlock = options.skillsBlock;
+        }
+        this.latestPromptComposition = this.composePromptSet(options?.skillsBlock);
+        this.sessionInstance.setPersona(this.currentPersona, this.latestPromptComposition.baseSystemPrompt, this.latestPromptComposition.subagentPrompts);
+    }
+    rebuildSubagentPrompts(options) {
+        if (options?.skillsBlock !== undefined) {
+            this.promptContext.skillsBlock = options.skillsBlock;
+        }
+        const nextPromptComposition = this.composePromptSet(options?.skillsBlock);
+        this.latestPromptComposition = {
+            ...this.latestPromptComposition,
+            subagentPrompts: nextPromptComposition.subagentPrompts,
+        };
+        this.sessionInstance.setPersona(this.currentPersona, this.latestPromptComposition.baseSystemPrompt, this.latestPromptComposition.subagentPrompts);
+    }
+    composePromptSet(skillsBlock) {
+        return composeSessionPrompts({
+            persona: this.currentPersona,
+            riskLevel: this.riskLevel,
+            cwd: this.promptContext.cwd,
+            datetime: new Date(this.environment.now()).toISOString(),
+            platform: this.environment.platform(),
+            nodeVersion: this.environment.nodeVersion(),
+            skillsBlock: skillsBlock ?? this.promptContext.skillsBlock,
+            projectContextBlock: this.promptContext.projectContextBlock,
+            sandboxEnabled: this.promptContext.sandboxEnabled,
+            sandboxEnvironmentInfo: this.promptContext.sandboxEnvironmentInfo,
+        });
+    }
+}
+//# sourceMappingURL=chat_runtime.js.map

package/dist/core/runtime/chat_runtime.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chat_runtime.js","sourceRoot":"","sources":["../../../src/core/runtime/chat_runtime.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AAGzD,OAAO,EAEL,uBAAuB,GACxB,MAAM,gCAAgC,CAAC;AAExC,OAAO,EAAE,qBAAqB,EAAiC,MAAM,8BAA8B,CAAC;AAmCpG,MAAM,OAAO,WAAW;IACL,eAAe,CAAc;IAC7B,WAAW,CAA0B;IAC9C,cAAc,CAAU;IACxB,SAAS,CAAY;IACrB,aAAa,CAA2B;IAC/B,WAAW,CAAyB;IAC7C,uBAAuB,CAA2B;IAE1D,MAAM,CAAC,MAAM,CAAC,OAAiC;QAC7C,MAAM,iBAAiB,GAAG,qBAAqB,CAAC;YAC9C,OAAO,EAAE,OAAO,CAAC,OAAO;YACxB,SAAS,EAAE,OAAO,CAAC,SAAS;YAC5B,GAAG,EAAE,OAAO,CAAC,aAAa,CAAC,GAAG;YAC9B,QAAQ,EAAE,IAAI,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,GAAG,EAAE,CAAC,CAAC,WAAW,EAAE;YAC3D,QAAQ,EAAE,OAAO,CAAC,WAAW,CAAC,QAAQ,EAAE;YACxC,WAAW,EAAE,OAAO,CAAC,WAAW,CAAC,WAAW,EAAE;YAC9C,WAAW,EAAE,OAAO,CAAC,aAAa,CAAC,WAAW;YAC9C,mBAAmB,EAAE,OAAO,CAAC,aAAa,CAAC,mBAAmB;YAC9D,cAAc,EAAE,OAAO,CAAC,aAAa,CAAC,cAAc;YACpD,sBAAsB,EAAE,OAAO,CAAC,aAAa,CAAC,sBAAsB;SACrE,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,IAAI,WAAW,CAAC;YAC9B,OAAO,EAAE,OAAO,CAAC,OAAO;YACxB,YAAY,EAAE,iBAAiB,CAAC,gBAAgB;YAChD,eAAe,EAAE,iBAAiB,CAAC,eAAe;YAClD,SAAS,EAAE,OAAO,CAAC,SAAS;YAC5B,YAAY,EAAE,OAAO,CAAC,YAAY;YAClC,MAAM,EAAE,OAAO,CAAC,MAAM;YACtB,IAAI,EAAE,OAAO,CAAC,IAAI;SACnB,CAAC,CAAC;QAEH,OAAO,IAAI,WAAW,CAAC;YACrB,OAAO;YACP,OAAO,EAAE,OAAO,CAAC,OAAO;YACxB,SAAS,EAAE,OAAO,CAAC,SAAS;YAC5B,aAAa,EAAE,OAAO,CAAC,aAAa;YACpC,WAAW,EAAE,OAAO,CAAC,WAAW;YAChC,wBAAwB,EAAE,iBAAiB;SAC5C,CAAC,CAAC;IACL,CAAC;IAED,YAAY,OAA2B;QACrC,IAAI,CAAC,eAAe,GAAG,OAAO,CAAC,OAAO,CAAC;QACvC,IAAI,CAAC,WAAW,GAAG,IAAI,uBAAuB,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QACrE,IAAI,CAAC,cAAc,GAAG,OAAO,CAAC,OAAO,CAAC;QACtC,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;QACnC,IAAI,CAAC,aAAa,GAAG,EAAE,GAAG,OAAO,CAAC,aAAa,EAAE,CAAC;QAClD,IAAI,CAAC,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC;QAEvC,IAAI,CAAC,uBAAuB,GAAG,OAAO,CAAC,wBAAwB,IAAI,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAE3F,IAAI,CAAC,OAAO,CAAC,wBAAwB,EAAE,CAAC;YACtC,IAAI,CAAC,eAAe,CAAC,UAAU,CAC7B,IAAI,CAAC,cAAc,EACnB,IAAI,CAAC,uBAAuB,CAAC,gBAAgB,EAC7C,IAAI,CAAC,uBAAuB,CAAC,eAAe,CAC7C,CAAC;QACJ,CAAC;IACH,CAAC;IAED,IAAI,OAAO;QACT,OAAO,IAAI,CAAC,eAAe,CAAC;IAC9B,CAAC;IAED,IAAI,aAAa;QACf,OAAO,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC;IACpC,CAAC;IAED,IAAI,iBAAiB;QACnB,OAAO,IAAI,CAAC,uBAAuB,CAAC;IACtC,CAAC;IAED,OAAO,CAAC,OAAmD;QACzD,OAAO,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IACvC,CAAC;IAED,aAAa;QACX,OAAO,IAAI,CAAC,WAAW,CAAC,SAAS,EAAE,CAAC;IACtC,CAAC;IAED,SAAS,CAAC,MAAc;QACtB,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IACzC,CAAC;IAED,YAAY,CAAC,KAAgB;QAC3B,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC;QAChC,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;QACvB,IAAI,CAAC,eAAe,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;QAEzC,IAAI,QAAQ,KAAK,KAAK,EAAE,CAAC;YACvB,IAAI,CAAC,sBAAsB,EAAE,CAAC;QAChC,CAAC;IACH,CAAC;IAED,UAAU,CAAC,OAAgB,EAAE,OAAkC;QAC7D,IAAI,CAAC,cAAc,GAAG,OAAO,CAAC;QAC9B,IAAI,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC;IACrC,CAAC;IAED,mBAAmB,CAAC,OAA0C;QAC5D,IAAI,CAAC,aAAa,GAAG;YACnB,GAAG,IAAI,CAAC,aAAa;YACrB,GAAG,OAAO;SACX,CAAC;IACJ,CAAC;IAED,oBAAoB,CAAC,OAAkC;QACrD,IAAI,OAAO,EAAE,WAAW,KAAK,SAAS,EAAE,CAAC;YACvC,IAAI,CAAC,aAAa,CAAC,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC;QACvD,CAAC;QAED,IAAI,CAAC,uBAAuB,GAAG,IAAI,CAAC,gBAAgB,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;QAC3E,IAAI,CAAC,eAAe,CAAC,UAAU,CAC7B,IAAI,CAAC,cAAc,EACnB,IAAI,CAAC,uBAAuB,CAAC,gBAAgB,EAC7C,IAAI,CAAC,uBAAuB,CAAC,eAAe,CAC7C,CAAC;IACJ,CAAC;IAED,sBAAsB,CAAC,OAAkC;QACvD,IAAI,OAAO,EAAE,WAAW,KAAK,SAAS,EAAE,CAAC;YACvC,IAAI,CAAC,aAAa,CAAC,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC;QACvD,CAAC;QAED,MAAM,qBAAqB,GAAG,IAAI,CAAC,gBAAgB,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;QAC1E,IAAI,CAAC,uBAAuB,GAAG;YAC7B,GAAG,IAAI,CAAC,uBAAuB;YAC/B,eAAe,EAAE,qBAAqB,CAAC,eAAe;SACvD,CAAC;QAEF,IAAI,CAAC,eAAe,CAAC,UAAU,CAC7B,IAAI,CAAC,cAAc,EACnB,IAAI,CAAC,uBAAuB,CAAC,gBAAgB,EAC7C,IAAI,CAAC,uBAAuB,CAAC,eAAe,CAC7C,CAAC;IACJ,CAAC;IAEO,gBAAgB,CAAC,WAAoB;QAC3C,OAAO,qBAAqB,CAAC;YAC3B,OAAO,EAAE,IAAI,CAAC,cAAc;YAC5B,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,GAAG,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG;YAC3B,QAAQ,EAAE,IAAI,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,GAAG,EAAE,CAAC,CAAC,WAAW,EAAE;YACxD,QAAQ,EAAE,IAAI,CAAC,WAAW,CAAC,QAAQ,EAAE;YACrC,WAAW,EAAE,IAAI,CAAC,WAAW,CAAC,WAAW,EAAE;YAC3C,WAAW,EAAE,WAAW,IAAI,IAAI,CAAC,aAAa,CAAC,WAAW;YAC1D,mBAAmB,EAAE,IAAI,CAAC,aAAa,CAAC,mBAAmB;YAC3D,cAAc,EAAE,IAAI,CAAC,aAAa,CAAC,cAAc;YACjD,sBAAsB,EAAE,IAAI,CAAC,aAAa,CAAC,sBAAsB;SAClE,CAAC,CAAC;IACL,CAAC;CACF"}

package/dist/core/runtime/conversation_turn_runtime.js

@@ -0,0 +1,49 @@
+export class ConversationTurnRuntime {
+    session;
+    abortController;
+    constructor(session) {
+        this.session = session;
+    }
+    get isRunning() {
+        return this.abortController !== undefined;
+    }
+    async run(onEvent) {
+        if (this.abortController) {
+            throw new Error("conversation turn is already running");
+        }
+        const abortController = new AbortController();
+        this.abortController = abortController;
+        try {
+            try {
+                for await (const event of this.session.events(abortController.signal)) {
+                    if (abortController.signal.aborted) {
+                        break;
+                    }
+                    await onEvent(event);
+                }
+            }
+            catch (err) {
+                if (!abortController.signal.aborted) {
+                    throw err;
+                }
+            }
+            return {
+                aborted: abortController.signal.aborted,
+            };
+        }
+        finally {
+            if (this.abortController === abortController) {
+                this.abortController = undefined;
+            }
+        }
+    }
+    interrupt() {
+        const abortController = this.abortController;
+        if (!abortController || abortController.signal.aborted) {
+            return false;
+        }
+        abortController.abort();
+        return true;
+    }
+}
+//# sourceMappingURL=conversation_turn_runtime.js.map

package/dist/core/runtime/conversation_turn_runtime.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"conversation_turn_runtime.js","sourceRoot":"","sources":["../../../src/core/runtime/conversation_turn_runtime.ts"],"names":[],"mappings":"AAOA,MAAM,OAAO,uBAAuB;IACjB,OAAO,CAA8B;IAC9C,eAAe,CAAmB;IAE1C,YAAY,OAAoC;QAC9C,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;IAED,IAAI,SAAS;QACX,OAAO,IAAI,CAAC,eAAe,KAAK,SAAS,CAAC;IAC5C,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,OAAmD;QAC3D,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;YACzB,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;QAC1D,CAAC;QAED,MAAM,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;QAC9C,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;QAEvC,IAAI,CAAC;YACH,IAAI,CAAC;gBACH,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,eAAe,CAAC,MAAM,CAAC,EAAE,CAAC;oBACtE,IAAI,eAAe,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;wBACnC,MAAM;oBACR,CAAC;oBACD,MAAM,OAAO,CAAC,KAAK,CAAC,CAAC;gBACvB,CAAC;YACH,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;oBACpC,MAAM,GAAG,CAAC;gBACZ,CAAC;YACH,CAAC;YAED,OAAO;gBACL,OAAO,EAAE,eAAe,CAAC,MAAM,CAAC,OAAO;aACxC,CAAC;QACJ,CAAC;gBAAS,CAAC;YACT,IAAI,IAAI,CAAC,eAAe,KAAK,eAAe,EAAE,CAAC;gBAC7C,IAAI,CAAC,eAAe,GAAG,SAAS,CAAC;YACnC,CAAC;QACH,CAAC;IACH,CAAC;IAED,SAAS;QACP,MAAM,eAAe,GAAG,IAAI,CAAC,eAAe,CAAC;QAC7C,IAAI,CAAC,eAAe,IAAI,eAAe,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACvD,OAAO,KAAK,CAAC;QACf,CAAC;QAED,eAAe,CAAC,KAAK,EAAE,CAAC;QACxB,OAAO,IAAI,CAAC;IACd,CAAC;CACF"}

package/dist/core/runtime/session_prompt_composer.js

@@ -0,0 +1,55 @@
+import { formatSubagentsForPrompt, getSubagentBasePrompt } from "../subagents/registry.js";
+import { buildBaseSystemPrompt, buildEnvironmentTag, buildSandboxInfoBlock, } from "../utils/context.js";
+export function composeSessionPrompts(args) {
+    const environmentTag = buildEnvironmentTag({
+        riskLevel: args.riskLevel,
+        cwd: args.cwd,
+        datetime: args.datetime,
+        platform: args.platform,
+        nodeVersion: args.nodeVersion,
+    });
+    const sandboxInfoBlock = args.sandboxEnabled
+        ? buildSandboxInfoBlock(args.sandboxEnvironmentInfo)
+        : undefined;
+    const baseSystemPrompt = buildBaseSystemPrompt({
+        personaSystemPrompt: args.persona.systemPrompt,
+        skillsBlock: args.skillsBlock,
+        projectContextBlock: args.projectContextBlock,
+        sandboxInfoBlock,
+        environmentTag,
+        subagentsBlock: formatSubagentsForPrompt(args.persona),
+    });
+    const subagentPrompts = {};
+    const subagents = args.persona.subagents;
+    if (subagents && Object.keys(subagents).length > 0) {
+        for (const [name, config] of Object.entries(subagents)) {
+            if (!config) {
+                continue;
+            }
+            const personaSystemPrompt = getSubagentBasePrompt(name, config);
+            if (!personaSystemPrompt) {
+                continue;
+            }
+            const subagentEnvironmentTag = buildEnvironmentTag({
+                riskLevel: config.riskLevel ?? args.riskLevel,
+                cwd: args.cwd,
+                datetime: args.datetime,
+                platform: args.platform,
+                nodeVersion: args.nodeVersion,
+            });
+            subagentPrompts[name] = buildBaseSystemPrompt({
+                personaSystemPrompt,
+                skillsBlock: args.skillsBlock,
+                projectContextBlock: args.projectContextBlock,
+                sandboxInfoBlock,
+                environmentTag: subagentEnvironmentTag,
+            });
+        }
+    }
+    return {
+        environmentTag,
+        baseSystemPrompt,
+        subagentPrompts,
+    };
+}
+//# sourceMappingURL=session_prompt_composer.js.map

package/dist/core/runtime/session_prompt_composer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"session_prompt_composer.js","sourceRoot":"","sources":["../../../src/core/runtime/session_prompt_composer.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,wBAAwB,EAAE,qBAAqB,EAAE,MAAM,0BAA0B,CAAC;AAE3F,OAAO,EACL,qBAAqB,EACrB,mBAAmB,EACnB,qBAAqB,GACtB,MAAM,qBAAqB,CAAC;AAqB7B,MAAM,UAAU,qBAAqB,CAAC,IAA+B;IACnE,MAAM,cAAc,GAAG,mBAAmB,CAAC;QACzC,SAAS,EAAE,IAAI,CAAC,SAAS;QACzB,GAAG,EAAE,IAAI,CAAC,GAAG;QACb,QAAQ,EAAE,IAAI,CAAC,QAAQ;QACvB,QAAQ,EAAE,IAAI,CAAC,QAAQ;QACvB,WAAW,EAAE,IAAI,CAAC,WAAW;KAC9B,CAAC,CAAC;IAEH,MAAM,gBAAgB,GAAG,IAAI,CAAC,cAAc;QAC1C,CAAC,CAAC,qBAAqB,CAAC,IAAI,CAAC,sBAAsB,CAAC;QACpD,CAAC,CAAC,SAAS,CAAC;IAEd,MAAM,gBAAgB,GAAG,qBAAqB,CAAC;QAC7C,mBAAmB,EAAE,IAAI,CAAC,OAAO,CAAC,YAAY;QAC9C,WAAW,EAAE,IAAI,CAAC,WAAW;QAC7B,mBAAmB,EAAE,IAAI,CAAC,mBAAmB;QAC7C,gBAAgB;QAChB,cAAc;QACd,cAAc,EAAE,wBAAwB,CAAC,IAAI,CAAC,OAAO,CAAC;KACvD,CAAC,CAAC;IAEH,MAAM,eAAe,GAA2B,EAAE,CAAC;IACnD,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC;IACzC,IAAI,SAAS,IAAI,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACnD,KAAK,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,CAAC;YACvD,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,SAAS;YACX,CAAC;YAED,MAAM,mBAAmB,GAAG,qBAAqB,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;YAChE,IAAI,CAAC,mBAAmB,EAAE,CAAC;gBACzB,SAAS;YACX,CAAC;YAED,MAAM,sBAAsB,GAAG,mBAAmB,CAAC;gBACjD,SAAS,EAAE,MAAM,CAAC,SAAS,IAAI,IAAI,CAAC,SAAS;gBAC7C,GAAG,EAAE,IAAI,CAAC,GAAG;gBACb,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,WAAW,EAAE,IAAI,CAAC,WAAW;aAC9B,CAAC,CAAC;YAEH,eAAe,CAAC,IAAI,CAAC,GAAG,qBAAqB,CAAC;gBAC5C,mBAAmB;gBACnB,WAAW,EAAE,IAAI,CAAC,WAAW;gBAC7B,mBAAmB,EAAE,IAAI,CAAC,mBAAmB;gBAC7C,gBAAgB;gBAChB,cAAc,EAAE,sBAAsB;aACvC,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,OAAO;QACL,cAAc;QACd,gBAAgB;QAChB,eAAe;KAChB,CAAC;AACJ,CAAC"}

package/dist/core/utils/auto_retry.js

@@ -1,5 +1,5 @@
 import { isContextOverflow } from "@mariozechner/pi-ai";
-const RETRYABLE_ERROR_REGEX = /overloaded|rate.?limit|too many requests|429|500|502|503|504|service.?unavailable|server error|internal error|connection.?error|connection.?refused|other side closed|fetch failed|upstream.?connect|reset before headers|terminated/i;
+const RETRYABLE_ERROR_REGEX = /overloaded|high.?demand|peak.?load|maximum.?usage.?size|provisioned.?throughput|rate.?limit|too many requests|429|500|502|503|504|service.?unavailable|server error|internal error|connection.?error|connection.?refused|other side closed|fetch failed|upstream.?connect|reset before headers|terminated/i;
 function asAssistantMessage(error) {
     if (!error || typeof error !== "object") {
         return undefined;

package/dist/core/utils/auto_retry.js.map

@@ -1 +1 @@
-{"version":3,"file":"auto_retry.js","sourceRoot":"","sources":["../../../src/core/utils/auto_retry.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AAOxD,MAAM,qBAAqB,GACzB,uOAAuO,CAAC;AAE1O,SAAS,kBAAkB,CAAC,KAAc;IACxC,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QACxC,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,OAAO,GAAG,KAAkC,CAAC;IACnD,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,IAAI,OAAO,CAAC,UAAU,KAAK,OAAO,EAAE,CAAC;QACnE,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,IAAI,OAAO,OAAO,CAAC,UAAU,KAAK,QAAQ,EAAE,CAAC;QAC3C,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,OAAO,OAA2B,CAAC;AACrC,CAAC;AAED,MAAM,UAAU,eAAe,CAAC,EAAE,KAAK,EAAE,KAAK,EAAoB;IAChE,MAAM,OAAO,GAAG,kBAAkB,CAAC,KAAK,CAAC,CAAC;IAC1C,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO,KAAK,CAAC;IACf,CAAC;IAED,IAAI,OAAO,CAAC,UAAU,KAAK,OAAO,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE,CAAC;QAC5D,OAAO,KAAK,CAAC;IACf,CAAC;IAED,IAAI,iBAAiB,CAAC,OAAO,EAAE,KAAK,CAAC,aAAa,CAAC,EAAE,CAAC;QACpD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,OAAO,qBAAqB,CAAC,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;AAC1D,CAAC"}
+{"version":3,"file":"auto_retry.js","sourceRoot":"","sources":["../../../src/core/utils/auto_retry.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AAOxD,MAAM,qBAAqB,GACzB,4SAA4S,CAAC;AAE/S,SAAS,kBAAkB,CAAC,KAAc;IACxC,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QACxC,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,OAAO,GAAG,KAAkC,CAAC;IACnD,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,IAAI,OAAO,CAAC,UAAU,KAAK,OAAO,EAAE,CAAC;QACnE,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,IAAI,OAAO,OAAO,CAAC,UAAU,KAAK,QAAQ,EAAE,CAAC;QAC3C,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,OAAO,OAA2B,CAAC;AACrC,CAAC;AAED,MAAM,UAAU,eAAe,CAAC,EAAE,KAAK,EAAE,KAAK,EAAoB;IAChE,MAAM,OAAO,GAAG,kBAAkB,CAAC,KAAK,CAAC,CAAC;IAC1C,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO,KAAK,CAAC;IACf,CAAC;IAED,IAAI,OAAO,CAAC,UAAU,KAAK,OAAO,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE,CAAC;QAC5D,OAAO,KAAK,CAAC;IACf,CAAC;IAED,IAAI,iBAAiB,CAAC,OAAO,EAAE,KAAK,CAAC,aAAa,CAAC,EAAE,CAAC;QACpD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,OAAO,qBAAqB,CAAC,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;AAC1D,CAAC"}

package/dist/core/version.js

@@ -1,2 +1,2 @@
-export const APP_VERSION = "0.2.35";
+export const APP_VERSION = "0.2.37";
 //# sourceMappingURL=version.js.map