neoctl 0.2.0 → 0.2.2

package/dist/repl/transcript-flush.d.ts

@@ -0,0 +1,6 @@
+import { type TranscriptUiLine } from "./transcript-format.js";
+export declare function TranscriptFlusher({ lines, columns, resetEpoch, }: {
+    lines: readonly TranscriptUiLine[];
+    columns: number;
+    resetEpoch: number;
+}): null;

package/dist/repl/transcript-flush.js

@@ -0,0 +1,54 @@
+import { useStdout } from "ink";
+import { useLayoutEffect } from "react";
+import { markdownRenderKey } from "./markdown-renderer.js";
+import { formatTranscriptBlock } from "./transcript-format.js";
+function messageContentWidth(columns) {
+    return Math.max(10, columns - 2);
+}
+function lineNeedsDynamicRender(line, columns) {
+    const w = messageContentWidth(columns);
+    if (line.live || line.pendingReplacement)
+        return true;
+    if (line.previewStyle === "summary" || line.format === "ansi")
+        return false;
+    return line.renderedKey !== markdownRenderKey(line.text, line.kind, w);
+}
+function lineDigest(line) {
+    return `${line.text}\0${line.renderedKey ?? ""}\0${Boolean(line.live)}\0${Boolean(line.pendingReplacement)}\0${line.title ?? ""}\0${line.format ?? ""}\0${line.previewStyle ?? ""}`;
+}
+/** Survives React remount / HMR so we do not re-append the same transcript blocks to scrollback. */
+let transcriptDigestMap = new Map();
+let transcriptDigestLastEpoch = -1;
+export function TranscriptFlusher({ lines, columns, resetEpoch, }) {
+    const { write } = useStdout();
+    useLayoutEffect(() => {
+        if (transcriptDigestLastEpoch !== resetEpoch) {
+            transcriptDigestMap.clear();
+            transcriptDigestLastEpoch = resetEpoch;
+        }
+        const w = messageContentWidth(columns);
+        let tailIndex = lines.findIndex((l) => lineNeedsDynamicRender(l, columns));
+        if (tailIndex === -1)
+            tailIndex = lines.length;
+        const parts = [];
+        for (let i = 0; i < tailIndex; i++) {
+            const line = lines[i];
+            if (!line)
+                continue;
+            const digest = lineDigest(line);
+            const mapKey = `${resetEpoch}:${line.id}`;
+            const prev = transcriptDigestMap.get(mapKey);
+            if (prev === digest)
+                continue;
+            transcriptDigestMap.set(mapKey, digest);
+            const block = formatTranscriptBlock(line, w);
+            if (block.length === 0)
+                continue;
+            parts.push(block.endsWith("\n") ? block : `${block}\n`);
+        }
+        if (parts.length > 0)
+            write(parts.join(""));
+    }, [lines, columns, write, resetEpoch]);
+    return null;
+}
+//# sourceMappingURL=transcript-flush.js.map

package/dist/repl/transcript-flush.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"transcript-flush.js","sourceRoot":"","sources":["../../src/repl/transcript-flush.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,KAAK,CAAC;AAChC,OAAO,EAAE,eAAe,EAAE,MAAM,OAAO,CAAC;AACxC,OAAO,EAAE,iBAAiB,EAAyB,MAAM,wBAAwB,CAAC;AAClF,OAAO,EAAE,qBAAqB,EAAyB,MAAM,wBAAwB,CAAC;AAEtF,SAAS,mBAAmB,CAAC,OAAe;IAC1C,OAAO,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,OAAO,GAAG,CAAC,CAAC,CAAC;AACnC,CAAC;AAED,SAAS,sBAAsB,CAAC,IAAsB,EAAE,OAAe;IACrE,MAAM,CAAC,GAAG,mBAAmB,CAAC,OAAO,CAAC,CAAC;IACvC,IAAI,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,kBAAkB;QAAE,OAAO,IAAI,CAAC;IACtD,IAAI,IAAI,CAAC,YAAY,KAAK,SAAS,IAAI,IAAI,CAAC,MAAM,KAAK,MAAM;QAAE,OAAO,KAAK,CAAC;IAC5E,OAAO,IAAI,CAAC,WAAW,KAAK,iBAAiB,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,IAAwB,EAAE,CAAC,CAAC,CAAC;AAC7F,CAAC;AAED,SAAS,UAAU,CAAC,IAAsB;IACxC,OAAO,GAAG,IAAI,CAAC,IAAI,KAAK,IAAI,CAAC,WAAW,IAAI,EAAE,KAAK,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,OAAO,CAAC,IAAI,CAAC,kBAAkB,CAAC,KAAK,IAAI,CAAC,KAAK,IAAI,EAAE,KAAK,IAAI,CAAC,MAAM,IAAI,EAAE,KAAK,IAAI,CAAC,YAAY,IAAI,EAAE,EAAE,CAAC;AACtL,CAAC;AAED,oGAAoG;AACpG,IAAI,mBAAmB,GAAG,IAAI,GAAG,EAAkB,CAAC;AACpD,IAAI,yBAAyB,GAAG,CAAC,CAAC,CAAC;AAEnC,MAAM,UAAU,iBAAiB,CAAC,EAChC,KAAK,EACL,OAAO,EACP,UAAU,GAKX;IACC,MAAM,EAAE,KAAK,EAAE,GAAG,SAAS,EAAE,CAAC;IAE9B,eAAe,CAAC,GAAG,EAAE;QACnB,IAAI,yBAAyB,KAAK,UAAU,EAAE,CAAC;YAC7C,mBAAmB,CAAC,KAAK,EAAE,CAAC;YAC5B,yBAAyB,GAAG,UAAU,CAAC;QACzC,CAAC;QAED,MAAM,CAAC,GAAG,mBAAmB,CAAC,OAAO,CAAC,CAAC;QACvC,IAAI,SAAS,GAAG,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,sBAAsB,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC,CAAC;QAC3E,IAAI,SAAS,KAAK,CAAC,CAAC;YAAE,SAAS,GAAG,KAAK,CAAC,MAAM,CAAC;QAE/C,MAAM,KAAK,GAAa,EAAE,CAAC;QAC3B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,EAAE,CAAC,EAAE,EAAE,CAAC;YACnC,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;YACtB,IAAI,CAAC,IAAI;gBAAE,SAAS;YACpB,MAAM,MAAM,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC;YAChC,MAAM,MAAM,GAAG,GAAG,UAAU,IAAI,IAAI,CAAC,EAAE,EAAE,CAAC;YAC1C,MAAM,IAAI,GAAG,mBAAmB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YAC7C,IAAI,IAAI,KAAK,MAAM;gBAAE,SAAS;YAC9B,mBAAmB,CAAC,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;YACxC,MAAM,KAAK,GAAG,qBAAqB,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;YAC7C,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;gBAAE,SAAS;YACjC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,KAAK,IAAI,CAAC,CAAC;QAC1D,CAAC;QACD,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC;YAAE,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;IAC9C,CAAC,EAAE,CAAC,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,UAAU,CAAC,CAAC,CAAC;IAExC,OAAO,IAAI,CAAC;AACd,CAAC"}

package/dist/repl/transcript-format.d.ts

@@ -0,0 +1,16 @@
+export interface TranscriptUiLine {
+    id: number;
+    kind: "system" | "user" | "assistant" | "thinking" | "tool" | "error" | "meta";
+    text: string;
+    title?: string;
+    bodyTitle?: string;
+    titleStatus?: "success" | "failure";
+    format?: "markdown" | "ansi" | "plain";
+    previewStyle?: "summary";
+    summaryMaxLines?: number;
+    live?: boolean;
+    pendingReplacement?: boolean;
+    renderedKey?: string;
+}
+/** Full message block as ANSI (including role marker), for scrollback via Ink `write`. */
+export declare function formatTranscriptBlock(line: TranscriptUiLine, width: number): string;

package/dist/repl/transcript-format.js

@@ -0,0 +1,199 @@
+import chalk from "chalk";
+import stripAnsi from "strip-ansi";
+import wrapAnsi from "wrap-ansi";
+import { getRenderedMarkdownLines, markdownRenderKey, renderedLinesToAnsi, renderMarkdownPreviewToLines, } from "./markdown-renderer.js";
+const SUMMARY_BLOCK = {
+    maxLines: 6,
+    detailIndent: "    ",
+};
+const THINKING_COLOR = "#a855f7";
+const THINKING_MARKER = "◆";
+const MESSAGE_ROLE_MARKER = "● ";
+function colorForKind(kind) {
+    if (kind === "user")
+        return chalk.cyan;
+    if (kind === "assistant")
+        return chalk.green;
+    if (kind === "thinking")
+        return chalk.hex(THINKING_COLOR);
+    if (kind === "tool")
+        return chalk.hex("#d4b04c");
+    if (kind === "error")
+        return chalk.red;
+    if (kind === "meta")
+        return chalk.gray;
+    return chalk.white;
+}
+function markerColorForKind(kind) {
+    if (kind === "thinking")
+        return chalk.hex(THINKING_COLOR);
+    return colorForKind(kind);
+}
+function messageRoleMarker(kind) {
+    if (kind === "thinking")
+        return `${THINKING_MARKER} `;
+    return MESSAGE_ROLE_MARKER;
+}
+function titleForKind(kind) {
+    if (kind === "thinking")
+        return `${THINKING_MARKER} think`;
+    if (kind === "tool")
+        return "Tool";
+    if (kind === "error")
+        return "Error";
+    if (kind === "meta")
+        return "Meta";
+    if (kind === "system")
+        return "System";
+    if (kind === "user")
+        return "User";
+    return "Assistant";
+}
+function titleStatusMarker(status) {
+    return status === "success" ? "✓" : "✗";
+}
+function titleStatusColor(status) {
+    return status === "success" ? chalk.green : chalk.red;
+}
+function summaryTitle(line) {
+    if (summaryUsesRoleMarker(line))
+        return "";
+    const title = line.title ?? titleForKind(line.kind);
+    if (!line.titleStatus)
+        return title;
+    return `${title} ${titleStatusMarker(line.titleStatus)}`;
+}
+function summaryUsesRoleMarker(line) {
+    return line.previewStyle === "summary" && (line.kind === "system" || line.kind === "meta");
+}
+function titleProvidesToolMarker(line) {
+    return line.kind === "tool" && !!line.title && (line.title.startsWith("◇ ") || line.title.startsWith("◆ "));
+}
+function truncateAnsi(value, maxLength) {
+    let visible = 0;
+    let cut = value.length;
+    let inAnsi = false;
+    for (let i = 0; i < value.length; i++) {
+        const ch = value[i];
+        if (ch === "\x1B") {
+            inAnsi = true;
+            continue;
+        }
+        if (inAnsi) {
+            if (ch === "m")
+                inAnsi = false;
+            continue;
+        }
+        visible++;
+        if (visible > maxLength) {
+            cut = i;
+            break;
+        }
+    }
+    return value.slice(0, cut);
+}
+function renderSummaryLines(line, width) {
+    const content = line.text;
+    const detailWidth = Math.max(10, width - SUMMARY_BLOCK.detailIndent.length);
+    const title = summaryTitle(line);
+    const rawLines = content.replace(/\r\n/g, "\n").split("\n");
+    const wrapped = rawLines.flatMap((rawLine, index) => {
+        const lineWidth = index === 0 && !title ? width : detailWidth;
+        return wrapAnsi(rawLine, Math.max(10, lineWidth), { hard: true, trim: false }).split("\n");
+    });
+    const maxLines = line.summaryMaxLines ?? SUMMARY_BLOCK.maxLines;
+    const preview = [title, ...wrapped].filter((value) => stripAnsi(value).length > 0).slice(0, maxLines);
+    if (wrapped.length + (title ? 1 : 0) > maxLines && preview.length > 0) {
+        preview[preview.length - 1] = truncateAnsi(preview[preview.length - 1], Math.max(1, detailWidth - 1)) + "…";
+    }
+    return preview.length ? preview : [""];
+}
+function formatSummaryTranscript(line, width) {
+    const useRoleMarker = summaryUsesRoleMarker(line);
+    const summaryWidth = useRoleMarker ? width : Math.max(10, width - 2);
+    const lines = renderSummaryLines(line, summaryWidth);
+    const out = [];
+    for (let index = 0; index < lines.length; index++) {
+        const previewLine = lines[index] ?? "";
+        const detail = index > 0;
+        const text = detail ? `${SUMMARY_BLOCK.detailIndent}${previewLine}` : previewLine;
+        if (!detail && line.titleStatus) {
+            const marker = titleStatusMarker(line.titleStatus);
+            const markerSuffix = ` ${marker}`;
+            const titleText = text.endsWith(markerSuffix) ? text.slice(0, -markerSuffix.length) : `${text} `;
+            out.push(`${colorForKind(line.kind).bold(titleText)}${titleStatusColor(line.titleStatus).bold(marker)}`);
+        }
+        else if (line.format === "ansi") {
+            out.push(detail ? chalk.gray.dim(text) : `${colorForKind(line.kind).bold(text)}`);
+        }
+        else {
+            out.push(detail ? chalk.gray.dim(text) : colorForKind(line.kind).bold(text));
+        }
+    }
+    const prefix = useRoleMarker ? markerColorForKind(line.kind)(messageRoleMarker(line.kind)) : "";
+    const body = out.join("\n");
+    return useRoleMarker ? `${prefix}${body}` : body;
+}
+function formatPlainTranscript(line, width) {
+    const useRoleMarker = !titleProvidesToolMarker(line);
+    const lineWidth = useRoleMarker ? width : Math.max(10, width - 2);
+    const parts = [];
+    if (line.title) {
+        const t = line.titleStatus
+            ? `${colorForKind(line.kind).bold(line.title)} ${titleStatusColor(line.titleStatus).bold(titleStatusMarker(line.titleStatus))}`
+            : colorForKind(line.kind).bold(line.title);
+        parts.push(t);
+    }
+    if (line.bodyTitle)
+        parts.push(chalk.bold(line.bodyTitle));
+    const mdKind = line.kind;
+    const rendered = getRenderedMarkdownLines(line.text, mdKind, lineWidth);
+    parts.push(renderedLinesToAnsi(rendered, mdKind));
+    const body = parts.join("\n");
+    return useRoleMarker ? `${markerColorForKind(line.kind)(messageRoleMarker(line.kind))}${body}` : body;
+}
+function formatAnsiTranscript(line, width) {
+    const useRoleMarker = !titleProvidesToolMarker(line);
+    const lineWidth = useRoleMarker ? width : Math.max(10, width - 2);
+    const wrapped = wrapAnsi(line.text, Math.max(10, lineWidth), { hard: true, trim: false }).split("\n");
+    const parts = [];
+    if (line.title) {
+        const t = line.titleStatus
+            ? `${colorForKind(line.kind).bold(line.title)} ${titleStatusColor(line.titleStatus).bold(titleStatusMarker(line.titleStatus))}`
+            : colorForKind(line.kind).bold(line.title);
+        parts.push(t);
+    }
+    if (line.bodyTitle)
+        parts.push(chalk.bold(line.bodyTitle));
+    parts.push(...wrapped);
+    const body = parts.join("\n");
+    return useRoleMarker ? `${markerColorForKind(line.kind)(messageRoleMarker(line.kind))}${body}` : body;
+}
+/** Full message block as ANSI (including role marker), for scrollback via Ink `write`. */
+export function formatTranscriptBlock(line, width) {
+    if (line.previewStyle === "summary")
+        return formatSummaryTranscript(line, width);
+    if (line.format === "ansi")
+        return formatAnsiTranscript(line, width);
+    const key = markdownRenderKey(line.text, line.kind, width);
+    if (line.renderedKey === key) {
+        return formatPlainTranscript(line, width);
+    }
+    const mdKind = line.kind;
+    const useRoleMarker = !titleProvidesToolMarker(line);
+    const lineWidth = useRoleMarker ? width : Math.max(10, width - 2);
+    const parts = [];
+    if (line.title) {
+        const t = line.titleStatus
+            ? `${colorForKind(line.kind).bold(line.title)} ${titleStatusColor(line.titleStatus).bold(titleStatusMarker(line.titleStatus))}`
+            : colorForKind(line.kind).bold(line.title);
+        parts.push(t);
+    }
+    if (line.bodyTitle)
+        parts.push(chalk.bold(line.bodyTitle));
+    const preview = renderMarkdownPreviewToLines(line.text, mdKind, lineWidth);
+    parts.push(renderedLinesToAnsi(preview, mdKind));
+    const body = parts.join("\n");
+    return useRoleMarker ? `${markerColorForKind(line.kind)(messageRoleMarker(line.kind))}${body}` : body;
+}
+//# sourceMappingURL=transcript-format.js.map

package/dist/repl/transcript-format.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"transcript-format.js","sourceRoot":"","sources":["../../src/repl/transcript-format.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,SAAS,MAAM,YAAY,CAAC;AACnC,OAAO,QAAQ,MAAM,WAAW,CAAC;AACjC,OAAO,EACL,wBAAwB,EACxB,iBAAiB,EACjB,mBAAmB,EACnB,4BAA4B,GAE7B,MAAM,wBAAwB,CAAC;AAEhC,MAAM,aAAa,GAAG;IACpB,QAAQ,EAAE,CAAC;IACX,YAAY,EAAE,MAAM;CACZ,CAAC;AAEX,MAAM,cAAc,GAAG,SAAS,CAAC;AACjC,MAAM,eAAe,GAAG,GAAG,CAAC;AAC5B,MAAM,mBAAmB,GAAG,IAAI,CAAC;AAiBjC,SAAS,YAAY,CAAC,IAA8B;IAClD,IAAI,IAAI,KAAK,MAAM;QAAE,OAAO,KAAK,CAAC,IAAI,CAAC;IACvC,IAAI,IAAI,KAAK,WAAW;QAAE,OAAO,KAAK,CAAC,KAAK,CAAC;IAC7C,IAAI,IAAI,KAAK,UAAU;QAAE,OAAO,KAAK,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;IAC1D,IAAI,IAAI,KAAK,MAAM;QAAE,OAAO,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IACjD,IAAI,IAAI,KAAK,OAAO;QAAE,OAAO,KAAK,CAAC,GAAG,CAAC;IACvC,IAAI,IAAI,KAAK,MAAM;QAAE,OAAO,KAAK,CAAC,IAAI,CAAC;IACvC,OAAO,KAAK,CAAC,KAAK,CAAC;AACrB,CAAC;AAED,SAAS,kBAAkB,CAAC,IAA8B;IACxD,IAAI,IAAI,KAAK,UAAU;QAAE,OAAO,KAAK,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;IAC1D,OAAO,YAAY,CAAC,IAAI,CAAC,CAAC;AAC5B,CAAC;AAED,SAAS,iBAAiB,CAAC,IAA+B;IACxD,IAAI,IAAI,KAAK,UAAU;QAAE,OAAO,GAAG,eAAe,GAAG,CAAC;IACtD,OAAO,mBAAmB,CAAC;AAC7B,CAAC;AAED,SAAS,YAAY,CAAC,IAA8B;IAClD,IAAI,IAAI,KAAK,UAAU;QAAE,OAAO,GAAG,eAAe,QAAQ,CAAC;IAC3D,IAAI,IAAI,KAAK,MAAM;QAAE,OAAO,MAAM,CAAC;IACnC,IAAI,IAAI,KAAK,OAAO;QAAE,OAAO,OAAO,CAAC;IACrC,IAAI,IAAI,KAAK,MAAM;QAAE,OAAO,MAAM,CAAC;IACnC,IAAI,IAAI,KAAK,QAAQ;QAAE,OAAO,QAAQ,CAAC;IACvC,IAAI,IAAI,KAAK,MAAM;QAAE,OAAO,MAAM,CAAC;IACnC,OAAO,WAAW,CAAC;AACrB,CAAC;AAED,SAAS,iBAAiB,CAAC,MAAoD;IAC7E,OAAO,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC;AAC1C,CAAC;AAED,SAAS,gBAAgB,CAAC,MAAoD;IAC5E,OAAO,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC;AACxD,CAAC;AAED,SAAS,YAAY,CAAC,IAAsB;IAC1C,IAAI,qBAAqB,CAAC,IAAI,CAAC;QAAE,OAAO,EAAE,CAAC;IAC3C,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,IAAI,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACpD,IAAI,CAAC,IAAI,CAAC,WAAW;QAAE,OAAO,KAAK,CAAC;IACpC,OAAO,GAAG,KAAK,IAAI,iBAAiB,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC;AAC3D,CAAC;AAED,SAAS,qBAAqB,CAAC,IAAsB;IACnD,OAAO,IAAI,CAAC,YAAY,KAAK,SAAS,IAAI,CAAC,IAAI,CAAC,IAAI,KAAK,QAAQ,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC;AAC7F,CAAC;AAED,SAAS,uBAAuB,CAAC,IAAsB;IACrD,OAAO,IAAI,CAAC,IAAI,KAAK,MAAM,IAAI,CAAC,CAAC,IAAI,CAAC,KAAK,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC;AAC9G,CAAC;AAED,SAAS,YAAY,CAAC,KAAa,EAAE,SAAiB;IACpD,IAAI,OAAO,GAAG,CAAC,CAAC;IAChB,IAAI,GAAG,GAAG,KAAK,CAAC,MAAM,CAAC;IACvB,IAAI,MAAM,GAAG,KAAK,CAAC;IACnB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACtC,MAAM,EAAE,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QACpB,IAAI,EAAE,KAAK,MAAM,EAAE,CAAC;YAClB,MAAM,GAAG,IAAI,CAAC;YACd,SAAS;QACX,CAAC;QACD,IAAI,MAAM,EAAE,CAAC;YACX,IAAI,EAAE,KAAK,GAAG;gBAAE,MAAM,GAAG,KAAK,CAAC;YAC/B,SAAS;QACX,CAAC;QACD,OAAO,EAAE,CAAC;QACV,IAAI,OAAO,GAAG,SAAS,EAAE,CAAC;YACxB,GAAG,GAAG,CAAC,CAAC;YACR,MAAM;QACR,CAAC;IACH,CAAC;IACD,OAAO,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;AAC7B,CAAC;AAED,SAAS,kBAAkB,CAAC,IAAsB,EAAE,KAAa;IAC/D,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC;IAC1B,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,KAAK,GAAG,aAAa,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;IAC5E,MAAM,KAAK,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC;IACjC,MAAM,QAAQ,GAAG,OAAO,CAAC,OAAO,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAC5D,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE;QAClD,MAAM,SAAS,GAAG,KAAK,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,WAAW,CAAC;QAC9D,OAAO,QAAQ,CAAC,OAAO,EAAE,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,SAAS,CAAC,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAC7F,CAAC,CAAC,CAAC;IACH,MAAM,QAAQ,GAAG,IAAI,CAAC,eAAe,IAAI,aAAa,CAAC,QAAQ,CAAC;IAChE,MAAM,OAAO,GAAG,CAAC,KAAK,EAAE,GAAG,OAAO,CAAC,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC;IACtG,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,QAAQ,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACtE,OAAO,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,GAAG,YAAY,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,WAAW,GAAG,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC;IAC9G,CAAC;IACD,OAAO,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;AACzC,CAAC;AAED,SAAS,uBAAuB,CAAC,IAAsB,EAAE,KAAa;IACpE,MAAM,aAAa,GAAG,qBAAqB,CAAC,IAAI,CAAC,CAAC;IAClD,MAAM,YAAY,GAAG,aAAa,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC;IACrE,MAAM,KAAK,GAAG,kBAAkB,CAAC,IAAI,EAAE,YAAY,CAAC,CAAC;IACrD,MAAM,GAAG,GAAa,EAAE,CAAC;IACzB,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,KAAK,CAAC,MAAM,EAAE,KAAK,EAAE,EAAE,CAAC;QAClD,MAAM,WAAW,GAAG,KAAK,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;QACvC,MAAM,MAAM,GAAG,KAAK,GAAG,CAAC,CAAC;QACzB,MAAM,IAAI,GAAG,MAAM,CAAC,CAAC,CAAC,GAAG,aAAa,CAAC,YAAY,GAAG,WAAW,EAAE,CAAC,CAAC,CAAC,WAAW,CAAC;QAClF,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YAChC,MAAM,MAAM,GAAG,iBAAiB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YACnD,MAAM,YAAY,GAAG,IAAI,MAAM,EAAE,CAAC;YAClC,MAAM,SAAS,GAAG,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,GAAG,IAAI,GAAG,CAAC;YACjG,GAAG,CAAC,IAAI,CACN,GAAG,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,gBAAgB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAC/F,CAAC;QACJ,CAAC;aAAM,IAAI,IAAI,CAAC,MAAM,KAAK,MAAM,EAAE,CAAC;YAClC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACpF,CAAC;aAAM,CAAC;YACN,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QAC/E,CAAC;IACH,CAAC;IACD,MAAM,MAAM,GAAG,aAAa,CAAC,CAAC,CAAC,kBAAkB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,iBAAiB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IAChG,MAAM,IAAI,GAAG,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC5B,OAAO,aAAa,CAAC,CAAC,CAAC,GAAG,MAAM,GAAG,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;AACnD,CAAC;AAED,SAAS,qBAAqB,CAAC,IAAsB,EAAE,KAAa;IAClE,MAAM,aAAa,GAAG,CAAC,uBAAuB,CAAC,IAAI,CAAC,CAAC;IACrD,MAAM,SAAS,GAAG,aAAa,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC;IAClE,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;QACf,MAAM,CAAC,GAAG,IAAI,CAAC,WAAW;YACxB,CAAC,CAAC,GAAG,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,gBAAgB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,EAAE;YAC/H,CAAC,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC7C,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAChB,CAAC;IACD,IAAI,IAAI,CAAC,SAAS;QAAE,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC;IAC3D,MAAM,MAAM,GAAG,IAAI,CAAC,IAAwB,CAAC;IAC7C,MAAM,QAAQ,GAAG,wBAAwB,CAAC,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;IACxE,KAAK,CAAC,IAAI,CAAC,mBAAmB,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC,CAAC;IAClD,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC9B,OAAO,aAAa,CAAC,CAAC,CAAC,GAAG,kBAAkB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,iBAAiB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;AACxG,CAAC;AAED,SAAS,oBAAoB,CAAC,IAAsB,EAAE,KAAa;IACjE,MAAM,aAAa,GAAG,CAAC,uBAAuB,CAAC,IAAI,CAAC,CAAC;IACrD,MAAM,SAAS,GAAG,aAAa,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC;IAClE,MAAM,OAAO,GAAG,QAAQ,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,SAAS,CAAC,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IACtG,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;QACf,MAAM,CAAC,GAAG,IAAI,CAAC,WAAW;YACxB,CAAC,CAAC,GAAG,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,gBAAgB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,EAAE;YAC/H,CAAC,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC7C,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAChB,CAAC;IACD,IAAI,IAAI,CAAC,SAAS;QAAE,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC;IAC3D,KAAK,CAAC,IAAI,CAAC,GAAG,OAAO,CAAC,CAAC;IACvB,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC9B,OAAO,aAAa,CAAC,CAAC,CAAC,GAAG,kBAAkB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,iBAAiB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;AACxG,CAAC;AAED,0FAA0F;AAC1F,MAAM,UAAU,qBAAqB,CAAC,IAAsB,EAAE,KAAa;IACzE,IAAI,IAAI,CAAC,YAAY,KAAK,SAAS;QAAE,OAAO,uBAAuB,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;IACjF,IAAI,IAAI,CAAC,MAAM,KAAK,MAAM;QAAE,OAAO,oBAAoB,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;IACrE,MAAM,GAAG,GAAG,iBAAiB,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,IAAwB,EAAE,KAAK,CAAC,CAAC;IAC/E,IAAI,IAAI,CAAC,WAAW,KAAK,GAAG,EAAE,CAAC;QAC7B,OAAO,qBAAqB,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;IAC5C,CAAC;IACD,MAAM,MAAM,GAAG,IAAI,CAAC,IAAwB,CAAC;IAC7C,MAAM,aAAa,GAAG,CAAC,uBAAuB,CAAC,IAAI,CAAC,CAAC;IACrD,MAAM,SAAS,GAAG,aAAa,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC;IAClE,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;QACf,MAAM,CAAC,GAAG,IAAI,CAAC,WAAW;YACxB,CAAC,CAAC,GAAG,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,gBAAgB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,EAAE;YAC/H,CAAC,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC7C,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAChB,CAAC;IACD,IAAI,IAAI,CAAC,SAAS;QAAE,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC;IAC3D,MAAM,OAAO,GAAG,4BAA4B,CAAC,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;IAC3E,KAAK,CAAC,IAAI,CAAC,mBAAmB,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,CAAC;IACjD,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC9B,OAAO,aAAa,CAAC,CAAC,CAAC,GAAG,kBAAkB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,iBAAiB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;AACxG,CAAC"}

package/dist/server/agent-manager.d.ts

@@ -0,0 +1,56 @@
+import type { QueryEngine } from "../core/query-engine.js";
+import { type CoreRuntime, type CoreServices, type CreateCoreRuntimeOptions } from "../core/runtime.js";
+import type { RpcAgentSnapshot } from "./rpc-types.js";
+import { QueryRun, type QueryRunOptions } from "./query-runtime.js";
+export type ManagedAgentState = "attached" | "idle" | "running" | "detached_idle" | "detached_running" | "disposed";
+export interface AgentManagerOptions {
+    idleAgentTtlMs?: number;
+    reconnectGraceMs?: number;
+    runningDetachedKeepAliveMs?: number;
+    createRuntime?: (options: CreateCoreRuntimeOptions) => Promise<CoreRuntime>;
+    services?: CoreServices;
+    sessionCwd?: string;
+}
+export interface AgentIdentity {
+    userId: string;
+    agentId: string;
+    sessionId?: string;
+}
+export interface ManagedAgent extends AgentIdentity {
+    key: string;
+    runtime: CoreRuntime;
+    engine: QueryEngine;
+    state: ManagedAgentState;
+    attachedConnections: Set<string>;
+    createdAt: Date;
+    updatedAt: Date;
+    lastDetachedAt?: Date;
+    activeQuery?: QueryRun;
+}
+export declare class AgentManager {
+    private readonly agents;
+    private readonly createRuntime;
+    private readonly idleAgentTtlMs;
+    private readonly reconnectGraceMs;
+    private readonly runningDetachedKeepAliveMs;
+    private readonly services?;
+    private readonly sessionCwd?;
+    constructor(options?: AgentManagerOptions);
+    getOrCreate(identity: AgentIdentity, options?: {
+        resume?: boolean;
+    }): Promise<ManagedAgent>;
+    get(identity: AgentIdentity): ManagedAgent | undefined;
+    attach(connectionId: string, identity: AgentIdentity, options?: {
+        create?: boolean;
+        resume?: boolean;
+    }): Promise<ManagedAgent>;
+    detach(connectionId: string, identity: AgentIdentity): ManagedAgent | undefined;
+    detachConnectionFromAll(connectionId: string): void;
+    startQuery(identity: AgentIdentity, input: Pick<QueryRunOptions, "queryId" | "text" | "displayText" | "disconnectPolicy" | "onEvent">): Promise<QueryRun>;
+    cancelQuery(identity: AgentIdentity, queryId: string): boolean;
+    listSnapshots(): RpcAgentSnapshot[];
+    sweep(now?: number): string[];
+    dispose(key: string): boolean;
+    private mustGet;
+}
+export declare function agentKey(identity: AgentIdentity): string;

package/dist/server/agent-manager.js

@@ -0,0 +1,171 @@
+import { createCoreRuntime, createQueryEngineRuntime, } from "../core/runtime.js";
+import { QueryRun } from "./query-runtime.js";
+export class AgentManager {
+    agents = new Map();
+    createRuntime;
+    idleAgentTtlMs;
+    reconnectGraceMs;
+    runningDetachedKeepAliveMs;
+    services;
+    sessionCwd;
+    constructor(options = {}) {
+        this.services = options.services;
+        this.sessionCwd = options.sessionCwd;
+        this.createRuntime = options.createRuntime ?? ((runtimeOptions) => this.services
+            ? createQueryEngineRuntime(this.services, runtimeOptions)
+            : createCoreRuntime(runtimeOptions));
+        this.idleAgentTtlMs = options.idleAgentTtlMs ?? 10 * 60_000;
+        this.reconnectGraceMs = options.reconnectGraceMs ?? 30_000;
+        this.runningDetachedKeepAliveMs = options.runningDetachedKeepAliveMs ?? 60 * 60_000;
+    }
+    async getOrCreate(identity, options = {}) {
+        const key = agentKey(identity);
+        const existing = this.agents.get(key);
+        if (existing)
+            return existing;
+        const runtime = await this.createRuntime({
+            agentId: identity.agentId,
+            sessionId: identity.sessionId,
+            sessionCwd: this.sessionCwd,
+            resume: options.resume ?? true,
+            queryOrigin: "json-rpc-ws",
+        });
+        const agent = {
+            ...identity,
+            key,
+            runtime,
+            engine: runtime.engine,
+            state: "idle",
+            attachedConnections: new Set(),
+            createdAt: new Date(),
+            updatedAt: new Date(),
+        };
+        this.agents.set(key, agent);
+        return agent;
+    }
+    get(identity) {
+        return this.agents.get(agentKey(identity));
+    }
+    async attach(connectionId, identity, options = {}) {
+        const agent = options.create === false ? this.mustGet(identity) : await this.getOrCreate(identity, { resume: options.resume });
+        agent.attachedConnections.add(connectionId);
+        agent.lastDetachedAt = undefined;
+        agent.updatedAt = new Date();
+        agent.state = agent.activeQuery ? "running" : "attached";
+        return agent;
+    }
+    detach(connectionId, identity) {
+        const agent = this.get(identity);
+        if (!agent)
+            return undefined;
+        agent.attachedConnections.delete(connectionId);
+        agent.updatedAt = new Date();
+        if (agent.attachedConnections.size === 0) {
+            agent.lastDetachedAt = new Date();
+            agent.state = agent.activeQuery ? "detached_running" : "detached_idle";
+        }
+        return agent;
+    }
+    detachConnectionFromAll(connectionId) {
+        for (const agent of this.agents.values()) {
+            if (!agent.attachedConnections.has(connectionId))
+                continue;
+            agent.attachedConnections.delete(connectionId);
+            agent.updatedAt = new Date();
+            if (agent.attachedConnections.size === 0) {
+                agent.lastDetachedAt = new Date();
+                agent.state = agent.activeQuery ? "detached_running" : "detached_idle";
+            }
+        }
+    }
+    async startQuery(identity, input) {
+        const agent = this.mustGet(identity);
+        if (agent.activeQuery && agent.activeQuery.getState() === "running") {
+            throw new Error(`agent already has a running query: ${agent.activeQuery.queryId}`);
+        }
+        const run = new QueryRun(agent.engine, {
+            queryId: input.queryId,
+            text: input.text,
+            displayText: input.displayText,
+            disconnectPolicy: input.disconnectPolicy,
+            onEvent: input.onEvent,
+        });
+        agent.activeQuery = run;
+        agent.updatedAt = new Date();
+        agent.state = agent.attachedConnections.size > 0 ? "running" : "detached_running";
+        run.start()
+            .catch(() => undefined)
+            .finally(() => {
+            if (agent.activeQuery === run) {
+                agent.activeQuery = undefined;
+                agent.updatedAt = new Date();
+                agent.state = agent.attachedConnections.size > 0 ? "idle" : "detached_idle";
+            }
+        });
+        return run;
+    }
+    cancelQuery(identity, queryId) {
+        const agent = this.mustGet(identity);
+        if (!agent.activeQuery || agent.activeQuery.queryId !== queryId)
+            return false;
+        agent.activeQuery.cancel();
+        agent.updatedAt = new Date();
+        return true;
+    }
+    listSnapshots() {
+        return [...this.agents.values()].map((agent) => ({
+            key: agent.key,
+            userId: agent.userId,
+            agentId: agent.agentId,
+            state: agent.state,
+            attachedConnections: agent.attachedConnections.size,
+            activeQueryId: agent.activeQuery?.queryId,
+            snapshot: agent.engine.snapshot(),
+        }));
+    }
+    sweep(now = Date.now()) {
+        const disposed = [];
+        for (const [key, agent] of this.agents) {
+            if (agent.attachedConnections.size > 0)
+                continue;
+            const detachedAt = agent.lastDetachedAt?.getTime() ?? agent.updatedAt.getTime();
+            const age = now - detachedAt;
+            if (agent.activeQuery) {
+                if (agent.activeQuery.getState() === "running") {
+                    if (age < this.runningDetachedKeepAliveMs)
+                        continue;
+                    agent.activeQuery.cancel();
+                }
+                else if (age < this.idleAgentTtlMs) {
+                    continue;
+                }
+            }
+            else {
+                if (age < Math.max(this.idleAgentTtlMs, this.reconnectGraceMs))
+                    continue;
+            }
+            this.dispose(key);
+            disposed.push(key);
+        }
+        return disposed;
+    }
+    dispose(key) {
+        const agent = this.agents.get(key);
+        if (!agent)
+            return false;
+        agent.activeQuery?.cancel();
+        agent.state = "disposed";
+        this.agents.delete(key);
+        return true;
+    }
+    mustGet(identity) {
+        const agent = this.get(identity);
+        if (!agent)
+            throw new Error(`agent not found: ${agentKey(identity)}`);
+        return agent;
+    }
+}
+export function agentKey(identity) {
+    return `${identity.userId}:${identity.sessionId ?? "default"}:${identity.agentId}`;
+}
+//# sourceMappingURL=agent-manager.js.map

package/dist/server/agent-manager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent-manager.js","sourceRoot":"","sources":["../../src/server/agent-manager.ts"],"names":[],"mappings":"AACA,OAAO,EACL,iBAAiB,EACjB,wBAAwB,GAIzB,MAAM,oBAAoB,CAAC;AAE5B,OAAO,EAAE,QAAQ,EAAwB,MAAM,oBAAoB,CAAC;AA+BpE,MAAM,OAAO,YAAY;IACN,MAAM,GAAG,IAAI,GAAG,EAAwB,CAAC;IACzC,aAAa,CAA8D;IAC3E,cAAc,CAAS;IACvB,gBAAgB,CAAS;IACzB,0BAA0B,CAAS;IACnC,QAAQ,CAAgB;IACxB,UAAU,CAAU;IAErC,YAAY,UAA+B,EAAE;QAC3C,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC;QACjC,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC;QACrC,IAAI,CAAC,aAAa,GAAG,OAAO,CAAC,aAAa,IAAI,CAAC,CAAC,cAAc,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ;YAC9E,CAAC,CAAC,wBAAwB,CAAC,IAAI,CAAC,QAAQ,EAAE,cAAc,CAAC;YACzD,CAAC,CAAC,iBAAiB,CAAC,cAAc,CAAC,CAAC,CAAC;QACvC,IAAI,CAAC,cAAc,GAAG,OAAO,CAAC,cAAc,IAAI,EAAE,GAAG,MAAM,CAAC;QAC5D,IAAI,CAAC,gBAAgB,GAAG,OAAO,CAAC,gBAAgB,IAAI,MAAM,CAAC;QAC3D,IAAI,CAAC,0BAA0B,GAAG,OAAO,CAAC,0BAA0B,IAAI,EAAE,GAAG,MAAM,CAAC;IACtF,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,QAAuB,EAAE,UAAgC,EAAE;QAC3E,MAAM,GAAG,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAC;QAC/B,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACtC,IAAI,QAAQ;YAAE,OAAO,QAAQ,CAAC;QAC9B,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC;YACvC,OAAO,EAAE,QAAQ,CAAC,OAAO;YACzB,SAAS,EAAE,QAAQ,CAAC,SAAS;YAC7B,UAAU,EAAE,IAAI,CAAC,UAAU;YAC3B,MAAM,EAAE,OAAO,CAAC,MAAM,IAAI,IAAI;YAC9B,WAAW,EAAE,aAAa;SAC3B,CAAC,CAAC;QACH,MAAM,KAAK,GAAiB;YAC1B,GAAG,QAAQ;YACX,GAAG;YACH,OAAO;YACP,MAAM,EAAE,OAAO,CAAC,MAAM;YACtB,KAAK,EAAE,MAAM;YACb,mBAAmB,EAAE,IAAI,GAAG,EAAE;YAC9B,SAAS,EAAE,IAAI,IAAI,EAAE;YACrB,SAAS,EAAE,IAAI,IAAI,EAAE;SACtB,CAAC;QACF,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;QAC5B,OAAO,KAAK,CAAC;IACf,CAAC;IAED,GAAG,CAAC,QAAuB;QACzB,OAAO,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC;IAC7C,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,YAAoB,EAAE,QAAuB,EAAE,UAAkD,EAAE;QAC9G,MAAM,KAAK,GAAG,OAAO,CAAC,MAAM,KAAK,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC,WAAW,CAAC,QAAQ,EAAE,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;QAC/H,KAAK,CAAC,mBAAmB,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QAC5C,KAAK,CAAC,cAAc,GAAG,SAAS,CAAC;QACjC,KAAK,CAAC,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC;QAC7B,KAAK,CAAC,KAAK,GAAG,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,UAAU,CAAC;QACzD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,MAAM,CAAC,YAAoB,EAAE,QAAuB;QAClD,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACjC,IAAI,CAAC,KAAK;YAAE,OAAO,SAAS,CAAC;QAC7B,KAAK,CAAC,mBAAmB,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;QAC/C,KAAK,CAAC,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC;QAC7B,IAAI,KAAK,CAAC,mBAAmB,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;YACzC,KAAK,CAAC,cAAc,GAAG,IAAI,IAAI,EAAE,CAAC;YAClC,KAAK,CAAC,KAAK,GAAG,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,eAAe,CAAC;QACzE,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,uBAAuB,CAAC,YAAoB;QAC1C,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC;YACzC,IAAI,CAAC,KAAK,CAAC,mBAAmB,CAAC,GAAG,CAAC,YAAY,CAAC;gBAAE,SAAS;YAC3D,KAAK,CAAC,mBAAmB,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;YAC/C,KAAK,CAAC,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC;YAC7B,IAAI,KAAK,CAAC,mBAAmB,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;gBACzC,KAAK,CAAC,cAAc,GAAG,IAAI,IAAI,EAAE,CAAC;gBAClC,KAAK,CAAC,KAAK,GAAG,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,eAAe,CAAC;YACzE,CAAC;QACH,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,QAAuB,EAAE,KAAiG;QACzI,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QACrC,IAAI,KAAK,CAAC,WAAW,IAAI,KAAK,CAAC,WAAW,CAAC,QAAQ,EAAE,KAAK,SAAS,EAAE,CAAC;YACpE,MAAM,IAAI,KAAK,CAAC,sCAAsC,KAAK,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC,CAAC;QACrF,CAAC;QACD,MAAM,GAAG,GAAG,IAAI,QAAQ,CAAC,KAAK,CAAC,MAAM,EAAE;YACrC,OAAO,EAAE,KAAK,CAAC,OAAO;YACtB,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,WAAW,EAAE,KAAK,CAAC,WAAW;YAC9B,gBAAgB,EAAE,KAAK,CAAC,gBAAgB;YACxC,OAAO,EAAE,KAAK,CAAC,OAAO;SACvB,CAAC,CAAC;QACH,KAAK,CAAC,WAAW,GAAG,GAAG,CAAC;QACxB,KAAK,CAAC,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC;QAC7B,KAAK,CAAC,KAAK,GAAG,KAAK,CAAC,mBAAmB,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,kBAAkB,CAAC;QAClF,GAAG,CAAC,KAAK,EAAE;aACR,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC;aACtB,OAAO,CAAC,GAAG,EAAE;YACZ,IAAI,KAAK,CAAC,WAAW,KAAK,GAAG,EAAE,CAAC;gBAC9B,KAAK,CAAC,WAAW,GAAG,SAAS,CAAC;gBAC9B,KAAK,CAAC,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC;gBAC7B,KAAK,CAAC,KAAK,GAAG,KAAK,CAAC,mBAAmB,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,eAAe,CAAC;YAC9E,CAAC;QACH,CAAC,CAAC,CAAC;QACL,OAAO,GAAG,CAAC;IACb,CAAC;IAED,WAAW,CAAC,QAAuB,EAAE,OAAe;QAClD,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QACrC,IAAI,CAAC,KAAK,CAAC,WAAW,IAAI,KAAK,CAAC,WAAW,CAAC,OAAO,KAAK,OAAO;YAAE,OAAO,KAAK,CAAC;QAC9E,KAAK,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC;QAC3B,KAAK,CAAC,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC;QAC7B,OAAO,IAAI,CAAC;IACd,CAAC;IAED,aAAa;QACX,OAAO,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;YAC/C,GAAG,EAAE,KAAK,CAAC,GAAG;YACd,MAAM,EAAE,KAAK,CAAC,MAAM;YACpB,OAAO,EAAE,KAAK,CAAC,OAAO;YACtB,KAAK,EAAE,KAAK,CAAC,KAAK;YAClB,mBAAmB,EAAE,KAAK,CAAC,mBAAmB,CAAC,IAAI;YACnD,aAAa,EAAE,KAAK,CAAC,WAAW,EAAE,OAAO;YACzC,QAAQ,EAAE,KAAK,CAAC,MAAM,CAAC,QAAQ,EAAE;SAClC,CAAC,CAAC,CAAC;IACN,CAAC;IAED,KAAK,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE;QACpB,MAAM,QAAQ,GAAa,EAAE,CAAC;QAC9B,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YACvC,IAAI,KAAK,CAAC,mBAAmB,CAAC,IAAI,GAAG,CAAC;gBAAE,SAAS;YACjD,MAAM,UAAU,GAAG,KAAK,CAAC,cAAc,EAAE,OAAO,EAAE,IAAI,KAAK,CAAC,SAAS,CAAC,OAAO,EAAE,CAAC;YAChF,MAAM,GAAG,GAAG,GAAG,GAAG,UAAU,CAAC;YAC7B,IAAI,KAAK,CAAC,WAAW,EAAE,CAAC;gBACtB,IAAI,KAAK,CAAC,WAAW,CAAC,QAAQ,EAAE,KAAK,SAAS,EAAE,CAAC;oBAC/C,IAAI,GAAG,GAAG,IAAI,CAAC,0BAA0B;wBAAE,SAAS;oBACpD,KAAK,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC;gBAC7B,CAAC;qBAAM,IAAI,GAAG,GAAG,IAAI,CAAC,cAAc,EAAE,CAAC;oBACrC,SAAS;gBACX,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,IAAI,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,cAAc,EAAE,IAAI,CAAC,gBAAgB,CAAC;oBAAE,SAAS;YAC3E,CAAC;YACD,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;YAClB,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACrB,CAAC;QACD,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,OAAO,CAAC,GAAW;QACjB,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACnC,IAAI,CAAC,KAAK;YAAE,OAAO,KAAK,CAAC;QACzB,KAAK,CAAC,WAAW,EAAE,MAAM,EAAE,CAAC;QAC5B,KAAK,CAAC,KAAK,GAAG,UAAU,CAAC;QACzB,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QACxB,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,OAAO,CAAC,QAAuB;QACrC,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACjC,IAAI,CAAC,KAAK;YAAE,MAAM,IAAI,KAAK,CAAC,oBAAoB,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;QACtE,OAAO,KAAK,CAAC;IACf,CAAC;CACF;AAED,MAAM,UAAU,QAAQ,CAAC,QAAuB;IAC9C,OAAO,GAAG,QAAQ,CAAC,MAAM,IAAI,QAAQ,CAAC,SAAS,IAAI,SAAS,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;AACrF,CAAC"}

package/dist/server/connection-manager.d.ts

@@ -0,0 +1,18 @@
+import type { AgentIdentity } from "./agent-manager.js";
+export interface ConnectionRecord {
+    id: string;
+    attachedAgents: Map<string, AgentIdentity>;
+    openedAt: Date;
+    closedAt?: Date;
+}
+export declare class ConnectionManager {
+    private readonly connections;
+    open(id: string): ConnectionRecord;
+    get(id: string): ConnectionRecord | undefined;
+    attachAgent(connectionId: string, identity: AgentIdentity): void;
+    detachAgent(connectionId: string, identity: AgentIdentity): void;
+    listAttachedAgents(connectionId: string): AgentIdentity[];
+    listConnectionIdsForAgent(identity: AgentIdentity): string[];
+    close(connectionId: string): AgentIdentity[];
+    private mustGet;
+}

package/dist/server/connection-manager.js

@@ -0,0 +1,52 @@
+export class ConnectionManager {
+    connections = new Map();
+    open(id) {
+        const record = {
+            id,
+            attachedAgents: new Map(),
+            openedAt: new Date(),
+        };
+        this.connections.set(id, record);
+        return record;
+    }
+    get(id) {
+        return this.connections.get(id);
+    }
+    attachAgent(connectionId, identity) {
+        const record = this.mustGet(connectionId);
+        record.attachedAgents.set(agentLocatorKey(identity), identity);
+    }
+    detachAgent(connectionId, identity) {
+        this.mustGet(connectionId).attachedAgents.delete(agentLocatorKey(identity));
+    }
+    listAttachedAgents(connectionId) {
+        return [...this.mustGet(connectionId).attachedAgents.values()];
+    }
+    listConnectionIdsForAgent(identity) {
+        const key = agentLocatorKey(identity);
+        const results = [];
+        for (const [connectionId, record] of this.connections) {
+            if (record.attachedAgents.has(key))
+                results.push(connectionId);
+        }
+        return results;
+    }
+    close(connectionId) {
+        const record = this.connections.get(connectionId);
+        if (!record)
+            return [];
+        record.closedAt = new Date();
+        this.connections.delete(connectionId);
+        return [...record.attachedAgents.values()];
+    }
+    mustGet(connectionId) {
+        const record = this.connections.get(connectionId);
+        if (!record)
+            throw new Error(`connection not found: ${connectionId}`);
+        return record;
+    }
+}
+function agentLocatorKey(identity) {
+    return `${identity.userId}:${identity.sessionId ?? "default"}:${identity.agentId}`;
+}
+//# sourceMappingURL=connection-manager.js.map

package/dist/server/connection-manager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"connection-manager.js","sourceRoot":"","sources":["../../src/server/connection-manager.ts"],"names":[],"mappings":"AASA,MAAM,OAAO,iBAAiB;IACX,WAAW,GAAG,IAAI,GAAG,EAA4B,CAAC;IAEnE,IAAI,CAAC,EAAU;QACb,MAAM,MAAM,GAAqB;YAC/B,EAAE;YACF,cAAc,EAAE,IAAI,GAAG,EAAE;YACzB,QAAQ,EAAE,IAAI,IAAI,EAAE;SACrB,CAAC;QACF,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;QACjC,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,GAAG,CAAC,EAAU;QACZ,OAAO,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAClC,CAAC;IAED,WAAW,CAAC,YAAoB,EAAE,QAAuB;QACvD,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;QAC1C,MAAM,CAAC,cAAc,CAAC,GAAG,CAAC,eAAe,CAAC,QAAQ,CAAC,EAAE,QAAQ,CAAC,CAAC;IACjE,CAAC;IAED,WAAW,CAAC,YAAoB,EAAE,QAAuB;QACvD,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,cAAc,CAAC,MAAM,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC,CAAC;IAC9E,CAAC;IAED,kBAAkB,CAAC,YAAoB;QACrC,OAAO,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,cAAc,CAAC,MAAM,EAAE,CAAC,CAAC;IACjE,CAAC;IAED,yBAAyB,CAAC,QAAuB;QAC/C,MAAM,GAAG,GAAG,eAAe,CAAC,QAAQ,CAAC,CAAC;QACtC,MAAM,OAAO,GAAa,EAAE,CAAC;QAC7B,KAAK,MAAM,CAAC,YAAY,EAAE,MAAM,CAAC,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACtD,IAAI,MAAM,CAAC,cAAc,CAAC,GAAG,CAAC,GAAG,CAAC;gBAAE,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QACjE,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,YAAoB;QACxB,MAAM,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QAClD,IAAI,CAAC,MAAM;YAAE,OAAO,EAAE,CAAC;QACvB,MAAM,CAAC,QAAQ,GAAG,IAAI,IAAI,EAAE,CAAC;QAC7B,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;QACtC,OAAO,CAAC,GAAG,MAAM,CAAC,cAAc,CAAC,MAAM,EAAE,CAAC,CAAC;IAC7C,CAAC;IAEO,OAAO,CAAC,YAAoB;QAClC,MAAM,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QAClD,IAAI,CAAC,MAAM;YAAE,MAAM,IAAI,KAAK,CAAC,yBAAyB,YAAY,EAAE,CAAC,CAAC;QACtE,OAAO,MAAM,CAAC;IAChB,CAAC;CACF;AAED,SAAS,eAAe,CAAC,QAAuB;IAC9C,OAAO,GAAG,QAAQ,CAAC,MAAM,IAAI,QAAQ,CAAC,SAAS,IAAI,SAAS,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;AACrF,CAAC"}

package/dist/server/event-mapper.d.ts

@@ -0,0 +1,4 @@
+import type { AgentEvent } from "../types/events.js";
+import type { RpcQueryEventEnvelope, SerializedAgentEvent } from "./rpc-types.js";
+export declare function serializeAgentEvent(event: AgentEvent): SerializedAgentEvent;
+export declare function createQueryEventEnvelope(queryId: string, seq: number, event: AgentEvent): RpcQueryEventEnvelope;

package/dist/server/event-mapper.js

@@ -0,0 +1,31 @@
+export function serializeAgentEvent(event) {
+    if (event.type === "error") {
+        return {
+            ...event,
+            error: serializeError(event.error),
+        };
+    }
+    if (event.type === "retrying") {
+        return {
+            ...event,
+            error: serializeError(event.error),
+        };
+    }
+    return event;
+}
+export function createQueryEventEnvelope(queryId, seq, event) {
+    return {
+        queryId,
+        seq,
+        event: serializeAgentEvent(event),
+        createdAt: new Date().toISOString(),
+    };
+}
+function serializeError(error) {
+    return {
+        name: error.name,
+        message: error.message,
+        stack: error.stack,
+    };
+}
+//# sourceMappingURL=event-mapper.js.map