@nodetool-ai/cli 0.7.0-rc.11

package/dist/useExecutionState.js

@@ -0,0 +1,281 @@
+/**
+ * useExecutionState — Tracks agent task/step hierarchy from ProcessingMessages.
+ *
+ * Processes interleaved messages from parallel task execution and maintains
+ * a structured tree state for rendering by ExecutionTree.
+ */
+import { useCallback, useRef, useState } from "react";
+import { TaskUpdateEvent } from "@nodetool-ai/protocol";
+const INITIAL_STATE = {
+    phase: "idle",
+    planningContent: "",
+    tasks: [],
+    selectedIndex: 0,
+};
+// Max chars to keep in step output preview
+const MAX_OUTPUT_LENGTH = 200;
+// ---------------------------------------------------------------------------
+// Hook
+// ---------------------------------------------------------------------------
+export function useExecutionState() {
+    const [state, setState] = useState({ ...INITIAL_STATE });
+    // Track which step is "active" per task for routing chunks
+    const activeStepRef = useRef(new Map());
+    const reset = useCallback(() => {
+        setState({ ...INITIAL_STATE });
+        activeStepRef.current.clear();
+    }, []);
+    const processMessage = useCallback((msg) => {
+        switch (msg.type) {
+            case "planning_update": {
+                const pu = msg;
+                setState((prev) => ({
+                    ...prev,
+                    phase: "planning",
+                    planningContent: pu.content ?? prev.planningContent,
+                }));
+                break;
+            }
+            case "task_update": {
+                const tu = msg;
+                setState((prev) => {
+                    const tasks = [...prev.tasks];
+                    if (tu.event === TaskUpdateEvent.TaskPlanned) {
+                        const existingIdx = tasks.findIndex((t) => t.id === tu.task.id);
+                        const steps = (tu.task.steps ?? []).map((s) => ({
+                            id: s.id ?? "",
+                            name: s.instructions?.slice(0, 60) ?? s.id ?? "",
+                            status: "waiting",
+                            output: "",
+                        }));
+                        const plannedTask = {
+                            id: tu.task.id ?? "",
+                            name: tu.task.title ?? tu.task.name ?? tu.task.id ?? "",
+                            status: "planned",
+                            steps,
+                            expanded: true,
+                        };
+                        if (existingIdx === -1) {
+                            tasks.push(plannedTask);
+                        }
+                        else {
+                            tasks[existingIdx] = {
+                                ...plannedTask,
+                                startedAt: tasks[existingIdx].startedAt,
+                            };
+                        }
+                        return { ...prev, phase: "planning", tasks };
+                    }
+                    if (tu.event === TaskUpdateEvent.TaskRemoved) {
+                        const filtered = tasks.filter((t) => t.id !== tu.task.id);
+                        return { ...prev, tasks: filtered };
+                    }
+                    if (tu.event === TaskUpdateEvent.TaskCreated) {
+                        const existingIdx = tasks.findIndex((t) => t.id === tu.task.id);
+                        if (existingIdx === -1) {
+                            const steps = (tu.task.steps ?? []).map((s) => ({
+                                id: s.id ?? "",
+                                name: s.instructions?.slice(0, 60) ?? s.id ?? "",
+                                status: "waiting",
+                                output: "",
+                            }));
+                            tasks.push({
+                                id: tu.task.id ?? "",
+                                name: tu.task.title ?? tu.task.name ?? tu.task.id ?? "",
+                                status: "running",
+                                startedAt: Date.now(),
+                                steps,
+                                expanded: true,
+                            });
+                        }
+                        else {
+                            tasks[existingIdx] = {
+                                ...tasks[existingIdx],
+                                status: "running",
+                                startedAt: Date.now(),
+                            };
+                        }
+                        return { ...prev, phase: "executing", tasks };
+                    }
+                    if (tu.event === TaskUpdateEvent.StepStarted) {
+                        const taskIdx = tasks.findIndex((t) => t.id === tu.task.id);
+                        if (taskIdx !== -1) {
+                            const task = { ...tasks[taskIdx], steps: [...tasks[taskIdx].steps] };
+                            const stepIdx = task.steps.findIndex((s) => s.id === tu.step?.id);
+                            if (stepIdx !== -1) {
+                                task.steps[stepIdx] = {
+                                    ...task.steps[stepIdx],
+                                    status: "running",
+                                    startedAt: Date.now(),
+                                };
+                            }
+                            else if (tu.step?.id) {
+                                // Step not in initial plan — add dynamically
+                                task.steps.push({
+                                    id: tu.step.id,
+                                    name: tu.step.instructions?.slice(0, 60) ?? tu.step.id,
+                                    status: "running",
+                                    startedAt: Date.now(),
+                                    output: "",
+                                });
+                            }
+                            task.status = "running";
+                            tasks[taskIdx] = task;
+                            activeStepRef.current.set(task.id, tu.step?.id ?? "");
+                        }
+                        return { ...prev, tasks };
+                    }
+                    if (tu.event === TaskUpdateEvent.StepCompleted ||
+                        tu.event === TaskUpdateEvent.StepFailed) {
+                        const taskIdx = tasks.findIndex((t) => t.id === tu.task.id);
+                        if (taskIdx !== -1) {
+                            const task = { ...tasks[taskIdx], steps: [...tasks[taskIdx].steps] };
+                            const stepIdx = task.steps.findIndex((s) => s.id === tu.step?.id);
+                            if (stepIdx !== -1) {
+                                const step = task.steps[stepIdx];
+                                const now = Date.now();
+                                task.steps[stepIdx] = {
+                                    ...step,
+                                    status: tu.event === TaskUpdateEvent.StepCompleted
+                                        ? "completed"
+                                        : "failed",
+                                    duration: step.startedAt
+                                        ? (now - step.startedAt) / 1000
+                                        : undefined,
+                                };
+                            }
+                            tasks[taskIdx] = task;
+                        }
+                        return { ...prev, tasks };
+                    }
+                    if (tu.event === TaskUpdateEvent.TaskCompleted) {
+                        const taskIdx = tasks.findIndex((t) => t.id === tu.task.id);
+                        if (taskIdx !== -1) {
+                            const task = tasks[taskIdx];
+                            const now = Date.now();
+                            tasks[taskIdx] = {
+                                ...task,
+                                status: "completed",
+                                expanded: false,
+                                duration: task.startedAt
+                                    ? (now - task.startedAt) / 1000
+                                    : undefined,
+                            };
+                            activeStepRef.current.delete(task.id);
+                        }
+                        return { ...prev, tasks };
+                    }
+                    return prev;
+                });
+                break;
+            }
+            case "tool_call_update": {
+                const tc = msg;
+                const stepId = tc.node_id ?? tc.step_id;
+                if (!stepId)
+                    break;
+                setState((prev) => {
+                    const tasks = prev.tasks.map((task) => {
+                        const stepIdx = task.steps.findIndex((s) => s.id === stepId);
+                        if (stepIdx === -1)
+                            return task;
+                        const steps = [...task.steps];
+                        const argsStr = Object.entries(tc.args ?? {})
+                            .map(([k, v]) => {
+                            const val = typeof v === "string" ? v : JSON.stringify(v);
+                            return `${k}: ${val.slice(0, 40)}`;
+                        })
+                            .join(", ");
+                        steps[stepIdx] = {
+                            ...steps[stepIdx],
+                            toolName: tc.name,
+                            toolArgs: argsStr,
+                            status: "running",
+                        };
+                        return { ...task, steps };
+                    });
+                    return { ...prev, tasks };
+                });
+                break;
+            }
+            case "chunk": {
+                const ch = msg;
+                const stepId = ch.node_id;
+                if (!stepId)
+                    break;
+                setState((prev) => {
+                    const tasks = prev.tasks.map((task) => {
+                        const stepIdx = task.steps.findIndex((s) => s.id === stepId);
+                        if (stepIdx === -1)
+                            return task;
+                        const steps = [...task.steps];
+                        const existing = steps[stepIdx].output;
+                        const newOutput = existing + (ch.content ?? "");
+                        steps[stepIdx] = {
+                            ...steps[stepIdx],
+                            output: newOutput.length > MAX_OUTPUT_LENGTH
+                                ? newOutput.slice(-MAX_OUTPUT_LENGTH)
+                                : newOutput,
+                            status: "running",
+                        };
+                        return { ...task, steps };
+                    });
+                    return { ...prev, tasks };
+                });
+                break;
+            }
+            case "step_result": {
+                const sr = msg;
+                const stepId = sr.step?.id;
+                if (!stepId)
+                    break;
+                setState((prev) => {
+                    const tasks = prev.tasks.map((task) => {
+                        const stepIdx = task.steps.findIndex((s) => s.id === stepId);
+                        if (stepIdx === -1)
+                            return task;
+                        const steps = [...task.steps];
+                        const result = typeof sr.result === "string"
+                            ? sr.result
+                            : JSON.stringify(sr.result ?? "");
+                        steps[stepIdx] = {
+                            ...steps[stepIdx],
+                            output: result.slice(0, MAX_OUTPUT_LENGTH),
+                            status: sr.error ? "failed" : "completed",
+                            error: sr.error ?? undefined,
+                        };
+                        return { ...task, steps };
+                    });
+                    return { ...prev, tasks };
+                });
+                break;
+            }
+        }
+    }, []);
+    const navigate = useCallback((direction) => {
+        setState((prev) => {
+            const max = prev.tasks.length - 1;
+            if (max < 0)
+                return prev;
+            const next = direction === "up"
+                ? Math.max(0, prev.selectedIndex - 1)
+                : Math.min(max, prev.selectedIndex + 1);
+            return { ...prev, selectedIndex: next };
+        });
+    }, []);
+    const toggleExpand = useCallback(() => {
+        setState((prev) => {
+            const tasks = [...prev.tasks];
+            const task = tasks[prev.selectedIndex];
+            if (!task)
+                return prev;
+            tasks[prev.selectedIndex] = { ...task, expanded: !task.expanded };
+            return { ...prev, tasks };
+        });
+    }, []);
+    const markDone = useCallback(() => {
+        setState((prev) => ({ ...prev, phase: "done" }));
+    }, []);
+    return { state, processMessage, navigate, toggleExpand, reset, markDone };
+}
+//# sourceMappingURL=useExecutionState.js.map

package/dist/useExecutionState.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"useExecutionState.js","sourceRoot":"","sources":["../src/useExecutionState.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,WAAW,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AAStD,OAAO,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AA2CxD,MAAM,aAAa,GAAmB;IACpC,KAAK,EAAE,MAAM;IACb,eAAe,EAAE,EAAE;IACnB,KAAK,EAAE,EAAE;IACT,aAAa,EAAE,CAAC;CACjB,CAAC;AAEF,2CAA2C;AAC3C,MAAM,iBAAiB,GAAG,GAAG,CAAC;AAE9B,8EAA8E;AAC9E,OAAO;AACP,8EAA8E;AAE9E,MAAM,UAAU,iBAAiB;IAC/B,MAAM,CAAC,KAAK,EAAE,QAAQ,CAAC,GAAG,QAAQ,CAAiB,EAAE,GAAG,aAAa,EAAE,CAAC,CAAC;IACzE,2DAA2D;IAC3D,MAAM,aAAa,GAAG,MAAM,CAAsB,IAAI,GAAG,EAAE,CAAC,CAAC;IAE7D,MAAM,KAAK,GAAG,WAAW,CAAC,GAAG,EAAE;QAC7B,QAAQ,CAAC,EAAE,GAAG,aAAa,EAAE,CAAC,CAAC;QAC/B,aAAa,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;IAChC,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,MAAM,cAAc,GAAG,WAAW,CAChC,CAAC,GAAsB,EAAE,EAAE;QACzB,QAAQ,GAAG,CAAC,IAAI,EAAE,CAAC;YACjB,KAAK,iBAAiB,CAAC,CAAC,CAAC;gBACvB,MAAM,EAAE,GAAG,GAAgC,CAAC;gBAC5C,QAAQ,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;oBAClB,GAAG,IAAI;oBACP,KAAK,EAAE,UAAU;oBACjB,eAAe,EAAE,EAAE,CAAC,OAAO,IAAI,IAAI,CAAC,eAAe;iBACpD,CAAC,CAAC,CAAC;gBACJ,MAAM;YACR,CAAC;YAED,KAAK,aAAa,CAAC,CAAC,CAAC;gBACnB,MAAM,EAAE,GAAG,GAA4B,CAAC;gBACxC,QAAQ,CAAC,CAAC,IAAI,EAAE,EAAE;oBAChB,MAAM,KAAK,GAAG,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC;oBAE9B,IAAI,EAAE,CAAC,KAAK,KAAK,eAAe,CAAC,WAAW,EAAE,CAAC;wBAC7C,MAAM,WAAW,GAAG,KAAK,CAAC,SAAS,CACjC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,IAAI,CAAC,EAAE,CAC3B,CAAC;wBACF,MAAM,KAAK,GAAgB,CAAC,EAAE,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;4BAC3D,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE;4BACd,IAAI,EAAE,CAAC,CAAC,YAAY,EAAE,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,EAAE,IAAI,EAAE;4BAChD,MAAM,EAAE,SAAuB;4BAC/B,MAAM,EAAE,EAAE;yBACX,CAAC,CAAC,CAAC;wBACJ,MAAM,WAAW,GAAc;4BAC7B,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE;4BACpB,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE;4BACvD,MAAM,EAAE,SAAS;4BACjB,KAAK;4BACL,QAAQ,EAAE,IAAI;yBACf,CAAC;wBACF,IAAI,WAAW,KAAK,CAAC,CAAC,EAAE,CAAC;4BACvB,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;wBAC1B,CAAC;6BAAM,CAAC;4BACN,KAAK,CAAC,WAAW,CAAC,GAAG;gCACnB,GAAG,WAAW;gCACd,SAAS,EAAE,KAAK,CAAC,WAAW,CAAC,CAAC,SAAS;6BACxC,CAAC;wBACJ,CAAC;wBACD,OAAO,EAAE,GAAG,IAAI,EAAE,KAAK,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC;oBAC/C,CAAC;oBAED,IAAI,EAAE,CAAC,KAAK,KAAK,eAAe,CAAC,WAAW,EAAE,CAAC;wBAC7C,MAAM,QAAQ,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;wBAC1D,OAAO,EAAE,GAAG,IAAI,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC;oBACtC,CAAC;oBAED,IAAI,EAAE,CAAC,KAAK,KAAK,eAAe,CAAC,WAAW,EAAE,CAAC;wBAC7C,MAAM,WAAW,GAAG,KAAK,CAAC,SAAS,CACjC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,IAAI,CAAC,EAAE,CAC3B,CAAC;wBACF,IAAI,WAAW,KAAK,CAAC,CAAC,EAAE,CAAC;4BACvB,MAAM,KAAK,GAAgB,CAAC,EAAE,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;gCAC3D,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE;gCACd,IAAI,EAAE,CAAC,CAAC,YAAY,EAAE,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,EAAE,IAAI,EAAE;gCAChD,MAAM,EAAE,SAAuB;gCAC/B,MAAM,EAAE,EAAE;6BACX,CAAC,CAAC,CAAC;4BACJ,KAAK,CAAC,IAAI,CAAC;gCACT,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE;gCACpB,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE;gCACvD,MAAM,EAAE,SAAS;gCACjB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;gCACrB,KAAK;gCACL,QAAQ,EAAE,IAAI;6BACf,CAAC,CAAC;wBACL,CAAC;6BAAM,CAAC;4BACN,KAAK,CAAC,WAAW,CAAC,GAAG;gCACnB,GAAG,KAAK,CAAC,WAAW,CAAC;gCACrB,MAAM,EAAE,SAAS;gCACjB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;6BACtB,CAAC;wBACJ,CAAC;wBACD,OAAO,EAAE,GAAG,IAAI,EAAE,KAAK,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC;oBAChD,CAAC;oBAED,IAAI,EAAE,CAAC,KAAK,KAAK,eAAe,CAAC,WAAW,EAAE,CAAC;wBAC7C,MAAM,OAAO,GAAG,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;wBAC5D,IAAI,OAAO,KAAK,CAAC,CAAC,EAAE,CAAC;4BACnB,MAAM,IAAI,GAAG,EAAE,GAAG,KAAK,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,CAAC,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC;4BACrE,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAClC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,IAAI,EAAE,EAAE,CAC5B,CAAC;4BACF,IAAI,OAAO,KAAK,CAAC,CAAC,EAAE,CAAC;gCACnB,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG;oCACpB,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC;oCACtB,MAAM,EAAE,SAAS;oCACjB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;iCACtB,CAAC;4BACJ,CAAC;iCAAM,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,EAAE,CAAC;gCACvB,6CAA6C;gCAC7C,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC;oCACd,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE;oCACd,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,EAAE,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,EAAE,CAAC,IAAI,CAAC,EAAE;oCACtD,MAAM,EAAE,SAAS;oCACjB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;oCACrB,MAAM,EAAE,EAAE;iCACX,CAAC,CAAC;4BACL,CAAC;4BACD,IAAI,CAAC,MAAM,GAAG,SAAS,CAAC;4BACxB,KAAK,CAAC,OAAO,CAAC,GAAG,IAAI,CAAC;4BACtB,aAAa,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,CAAC,IAAI,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;wBACxD,CAAC;wBACD,OAAO,EAAE,GAAG,IAAI,EAAE,KAAK,EAAE,CAAC;oBAC5B,CAAC;oBAED,IACE,EAAE,CAAC,KAAK,KAAK,eAAe,CAAC,aAAa;wBAC1C,EAAE,CAAC,KAAK,KAAK,eAAe,CAAC,UAAU,EACvC,CAAC;wBACD,MAAM,OAAO,GAAG,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;wBAC5D,IAAI,OAAO,KAAK,CAAC,CAAC,EAAE,CAAC;4BACnB,MAAM,IAAI,GAAG,EAAE,GAAG,KAAK,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,CAAC,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC;4BACrE,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAClC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,IAAI,EAAE,EAAE,CAC5B,CAAC;4BACF,IAAI,OAAO,KAAK,CAAC,CAAC,EAAE,CAAC;gCACnB,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;gCACjC,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;gCACvB,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG;oCACpB,GAAG,IAAI;oCACP,MAAM,EACJ,EAAE,CAAC,KAAK,KAAK,eAAe,CAAC,aAAa;wCACxC,CAAC,CAAC,WAAW;wCACb,CAAC,CAAC,QAAQ;oCACd,QAAQ,EAAE,IAAI,CAAC,SAAS;wCACtB,CAAC,CAAC,CAAC,GAAG,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,IAAI;wCAC/B,CAAC,CAAC,SAAS;iCACd,CAAC;4BACJ,CAAC;4BACD,KAAK,CAAC,OAAO,CAAC,GAAG,IAAI,CAAC;wBACxB,CAAC;wBACD,OAAO,EAAE,GAAG,IAAI,EAAE,KAAK,EAAE,CAAC;oBAC5B,CAAC;oBAED,IAAI,EAAE,CAAC,KAAK,KAAK,eAAe,CAAC,aAAa,EAAE,CAAC;wBAC/C,MAAM,OAAO,GAAG,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;wBAC5D,IAAI,OAAO,KAAK,CAAC,CAAC,EAAE,CAAC;4BACnB,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC;4BAC5B,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;4BACvB,KAAK,CAAC,OAAO,CAAC,GAAG;gCACf,GAAG,IAAI;gCACP,MAAM,EAAE,WAAW;gCACnB,QAAQ,EAAE,KAAK;gCACf,QAAQ,EAAE,IAAI,CAAC,SAAS;oCACtB,CAAC,CAAC,CAAC,GAAG,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,IAAI;oCAC/B,CAAC,CAAC,SAAS;6BACd,CAAC;4BACF,aAAa,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;wBACxC,CAAC;wBACD,OAAO,EAAE,GAAG,IAAI,EAAE,KAAK,EAAE,CAAC;oBAC5B,CAAC;oBAED,OAAO,IAAI,CAAC;gBACd,CAAC,CAAC,CAAC;gBACH,MAAM;YACR,CAAC;YAED,KAAK,kBAAkB,CAAC,CAAC,CAAC;gBACxB,MAAM,EAAE,GAAG,GAAgC,CAAC;gBAC5C,MAAM,MAAM,GAAG,EAAE,CAAC,OAAO,IAAI,EAAE,CAAC,OAAO,CAAC;gBACxC,IAAI,CAAC,MAAM;oBAAE,MAAM;gBAEnB,QAAQ,CAAC,CAAC,IAAI,EAAE,EAAE;oBAChB,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;wBACpC,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,MAAM,CAAC,CAAC;wBAC7D,IAAI,OAAO,KAAK,CAAC,CAAC;4BAAE,OAAO,IAAI,CAAC;wBAChC,MAAM,KAAK,GAAG,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC;wBAC9B,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,IAAI,IAAI,EAAE,CAAC;6BAC1C,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE;4BACd,MAAM,GAAG,GAAG,OAAO,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;4BAC1D,OAAO,GAAG,CAAC,KAAK,GAAG,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC;wBACrC,CAAC,CAAC;6BACD,IAAI,CAAC,IAAI,CAAC,CAAC;wBACd,KAAK,CAAC,OAAO,CAAC,GAAG;4BACf,GAAG,KAAK,CAAC,OAAO,CAAC;4BACjB,QAAQ,EAAE,EAAE,CAAC,IAAI;4BACjB,QAAQ,EAAE,OAAO;4BACjB,MAAM,EAAE,SAAS;yBAClB,CAAC;wBACF,OAAO,EAAE,GAAG,IAAI,EAAE,KAAK,EAAE,CAAC;oBAC5B,CAAC,CAAC,CAAC;oBACH,OAAO,EAAE,GAAG,IAAI,EAAE,KAAK,EAAE,CAAC;gBAC5B,CAAC,CAAC,CAAC;gBACH,MAAM;YACR,CAAC;YAED,KAAK,OAAO,CAAC,CAAC,CAAC;gBACb,MAAM,EAAE,GAAG,GAAuB,CAAC;gBACnC,MAAM,MAAM,GAAG,EAAE,CAAC,OAAO,CAAC;gBAC1B,IAAI,CAAC,MAAM;oBAAE,MAAM;gBAEnB,QAAQ,CAAC,CAAC,IAAI,EAAE,EAAE;oBAChB,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;wBACpC,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,MAAM,CAAC,CAAC;wBAC7D,IAAI,OAAO,KAAK,CAAC,CAAC;4BAAE,OAAO,IAAI,CAAC;wBAChC,MAAM,KAAK,GAAG,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC;wBAC9B,MAAM,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC;wBACvC,MAAM,SAAS,GAAG,QAAQ,GAAG,CAAC,EAAE,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC;wBAChD,KAAK,CAAC,OAAO,CAAC,GAAG;4BACf,GAAG,KAAK,CAAC,OAAO,CAAC;4BACjB,MAAM,EACJ,SAAS,CAAC,MAAM,GAAG,iBAAiB;gCAClC,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,iBAAiB,CAAC;gCACrC,CAAC,CAAC,SAAS;4BACf,MAAM,EAAE,SAAS;yBAClB,CAAC;wBACF,OAAO,EAAE,GAAG,IAAI,EAAE,KAAK,EAAE,CAAC;oBAC5B,CAAC,CAAC,CAAC;oBACH,OAAO,EAAE,GAAG,IAAI,EAAE,KAAK,EAAE,CAAC;gBAC5B,CAAC,CAAC,CAAC;gBACH,MAAM;YACR,CAAC;YAED,KAAK,aAAa,CAAC,CAAC,CAAC;gBACnB,MAAM,EAAE,GAAG,GAA4B,CAAC;gBACxC,MAAM,MAAM,GAAG,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC;gBAC3B,IAAI,CAAC,MAAM;oBAAE,MAAM;gBAEnB,QAAQ,CAAC,CAAC,IAAI,EAAE,EAAE;oBAChB,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;wBACpC,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,MAAM,CAAC,CAAC;wBAC7D,IAAI,OAAO,KAAK,CAAC,CAAC;4BAAE,OAAO,IAAI,CAAC;wBAChC,MAAM,KAAK,GAAG,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC;wBAC9B,MAAM,MAAM,GACV,OAAO,EAAE,CAAC,MAAM,KAAK,QAAQ;4BAC3B,CAAC,CAAC,EAAE,CAAC,MAAM;4BACX,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC;wBACtC,KAAK,CAAC,OAAO,CAAC,GAAG;4BACf,GAAG,KAAK,CAAC,OAAO,CAAC;4BACjB,MAAM,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,iBAAiB,CAAC;4BAC1C,MAAM,EAAE,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,WAAW;4BACzC,KAAK,EAAE,EAAE,CAAC,KAAK,IAAI,SAAS;yBAC7B,CAAC;wBACF,OAAO,EAAE,GAAG,IAAI,EAAE,KAAK,EAAE,CAAC;oBAC5B,CAAC,CAAC,CAAC;oBACH,OAAO,EAAE,GAAG,IAAI,EAAE,KAAK,EAAE,CAAC;gBAC5B,CAAC,CAAC,CAAC;gBACH,MAAM;YACR,CAAC;QACH,CAAC;IACH,CAAC,EACD,EAAE,CACH,CAAC;IAEF,MAAM,QAAQ,GAAG,WAAW,CAC1B,CAAC,SAAwB,EAAE,EAAE;QAC3B,QAAQ,CAAC,CAAC,IAAI,EAAE,EAAE;YAChB,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC;YAClC,IAAI,GAAG,GAAG,CAAC;gBAAE,OAAO,IAAI,CAAC;YACzB,MAAM,IAAI,GACR,SAAS,KAAK,IAAI;gBAChB,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,aAAa,GAAG,CAAC,CAAC;gBACrC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,CAAC,aAAa,GAAG,CAAC,CAAC,CAAC;YAC5C,OAAO,EAAE,GAAG,IAAI,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC;QAC1C,CAAC,CAAC,CAAC;IACL,CAAC,EACD,EAAE,CACH,CAAC;IAEF,MAAM,YAAY,GAAG,WAAW,CAAC,GAAG,EAAE;QACpC,QAAQ,CAAC,CAAC,IAAI,EAAE,EAAE;YAChB,MAAM,KAAK,GAAG,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC;YAC9B,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YACvC,IAAI,CAAC,IAAI;gBAAE,OAAO,IAAI,CAAC;YACvB,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,GAAG,EAAE,GAAG,IAAI,EAAE,QAAQ,EAAE,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClE,OAAO,EAAE,GAAG,IAAI,EAAE,KAAK,EAAE,CAAC;QAC5B,CAAC,CAAC,CAAC;IACL,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,MAAM,QAAQ,GAAG,WAAW,CAAC,GAAG,EAAE;QAChC,QAAQ,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,IAAI,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;IACnD,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,OAAO,EAAE,KAAK,EAAE,cAAc,EAAE,QAAQ,EAAE,YAAY,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC;AAC5E,CAAC"}

package/dist/websocket-client.d.ts

@@ -0,0 +1,100 @@
+/**
+ * WebSocket client for connecting the CLI to a NodeTool server.
+ * Handles chat (stateful, with thread history), inference (stateless, for agent mode),
+ * and workflow commands (run_job, cancel_job, get_status).
+ */
+export type ChatEvent = {
+    type: "chunk";
+    content: string;
+} | {
+    type: "tool_call";
+    id: string;
+    name: string;
+    args: Record<string, unknown>;
+} | {
+    type: "tool_result";
+    id: string;
+    name: string;
+    content: string;
+} | {
+    type: "output_update";
+    node_id: string;
+    value: unknown;
+    output_type?: string;
+} | {
+    type: "error";
+    message: string;
+} | {
+    type: "done";
+};
+export type JobEvent = {
+    type: "job_update";
+    status: string;
+    job_id?: string;
+    workflow_id?: string;
+    error?: string;
+    result?: unknown;
+} | {
+    type: "node_update";
+    node_id: string;
+    status: string;
+    error?: string;
+} | {
+    type: "output_update";
+    node_id: string;
+    value: unknown;
+    output_type?: string;
+} | {
+    type: "node_progress";
+    node_id: string;
+    progress: number;
+    total?: number;
+} | {
+    type: "error";
+    message: string;
+} | {
+    type: "done";
+};
+export declare class WebSocketChatClient {
+    private readonly wsUrl;
+    private ws;
+    private contentQueue;
+    private contentWaiters;
+    constructor(wsUrl: string);
+    connect(): Promise<void>;
+    private handleMessage;
+    private isContentEvent;
+    private send;
+    private nextContent;
+    /** Stateful chat — messages are saved to DB, thread history is loaded. */
+    chat(content: string, threadId: string, model: string, provider: string, tools?: unknown[]): AsyncGenerator<ChatEvent>;
+    /** Stateless inference — takes full messages array, streams back, no DB. Used by agent mode. */
+    inference(messages: unknown[], model: string, provider: string, tools?: unknown[]): AsyncGenerator<ChatEvent>;
+    /**
+     * Run a workflow job. Streams job_update, node_update, output_update events.
+     * Terminates when a terminal job_update is received (completed/failed/cancelled).
+     */
+    runJob(opts: {
+        workflowId?: string;
+        graph?: {
+            nodes: Array<Record<string, unknown>>;
+            edges: Array<Record<string, unknown>>;
+        };
+        params?: Record<string, unknown>;
+        jobId?: string;
+    }): AsyncGenerator<JobEvent>;
+    /** Reconnect to a running job's event stream. */
+    reconnectJob(jobId: string, workflowId?: string): AsyncGenerator<JobEvent>;
+    /** Resume a paused/interrupted job. */
+    resumeJob(jobId: string, workflowId?: string): AsyncGenerator<JobEvent>;
+    /** Shared job event consumer used by runJob, reconnectJob, resumeJob. */
+    private consumeJobEvents;
+    /** Cancel a running job. */
+    cancelJob(jobId: string): void;
+    /** Get status of a job or all active jobs. */
+    getStatus(jobId?: string): void;
+    /** Stop in-progress generation. Pass threadId for chat, omit for inference. */
+    stop(threadId?: string): void;
+    disconnect(): void;
+}
+//# sourceMappingURL=websocket-client.d.ts.map

package/dist/websocket-client.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"websocket-client.d.ts","sourceRoot":"","sources":["../src/websocket-client.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAIH,MAAM,MAAM,SAAS,GACjB;IAAE,IAAI,EAAE,OAAO,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,GAClC;IACE,IAAI,EAAE,WAAW,CAAC;IAClB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAC/B,GACD;IAAE,IAAI,EAAE,aAAa,CAAC;IAAC,EAAE,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,GAClE;IACE,IAAI,EAAE,eAAe,CAAC;IACtB,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE,OAAO,CAAC;IACf,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB,GACD;IAAE,IAAI,EAAE,OAAO,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,GAClC;IAAE,IAAI,EAAE,MAAM,CAAA;CAAE,CAAC;AAErB,MAAM,MAAM,QAAQ,GAChB;IACE,IAAI,EAAE,YAAY,CAAC;IACnB,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,MAAM,CAAC,EAAE,OAAO,CAAC;CAClB,GACD;IAAE,IAAI,EAAE,aAAa,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,MAAM,CAAC;IAAC,KAAK,CAAC,EAAE,MAAM,CAAA;CAAE,GACxE;IACE,IAAI,EAAE,eAAe,CAAC;IACtB,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE,OAAO,CAAC;IACf,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB,GACD;IAAE,IAAI,EAAE,eAAe,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,QAAQ,EAAE,MAAM,CAAC;IAAC,KAAK,CAAC,EAAE,MAAM,CAAA;CAAE,GAC5E;IAAE,IAAI,EAAE,OAAO,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,GAClC;IAAE,IAAI,EAAE,MAAM,CAAA;CAAE,CAAC;AAErB,qBAAa,mBAAmB;IAOlB,OAAO,CAAC,QAAQ,CAAC,KAAK;IANlC,OAAO,CAAC,EAAE,CAA0B;IACpC,OAAO,CAAC,YAAY,CAAsC;IAC1D,OAAO,CAAC,cAAc,CAEf;gBAEsB,KAAK,EAAE,MAAM;IAEpC,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IAmC9B,OAAO,CAAC,aAAa;IAoCrB,OAAO,CAAC,cAAc;IAetB,OAAO,CAAC,IAAI;IAMZ,OAAO,CAAC,WAAW;IASnB,0EAA0E;IACnE,IAAI,CACT,OAAO,EAAE,MAAM,EACf,QAAQ,EAAE,MAAM,EAChB,KAAK,EAAE,MAAM,EACb,QAAQ,EAAE,MAAM,EAChB,KAAK,CAAC,EAAE,OAAO,EAAE,GAChB,cAAc,CAAC,SAAS,CAAC;IAkF5B,gGAAgG;IACzF,SAAS,CACd,QAAQ,EAAE,OAAO,EAAE,EACnB,KAAK,EAAE,MAAM,EACb,QAAQ,EAAE,MAAM,EAChB,KAAK,CAAC,EAAE,OAAO,EAAE,GAChB,cAAc,CAAC,SAAS,CAAC;IA2C5B;;;OAGG;IACI,MAAM,CAAC,IAAI,EAAE;QAClB,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,KAAK,CAAC,EAAE;YACN,KAAK,EAAE,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC;YACtC,KAAK,EAAE,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC;SACvC,CAAC;QACF,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QACjC,KAAK,CAAC,EAAE,MAAM,CAAC;KAChB,GAAG,cAAc,CAAC,QAAQ,CAAC;IAc5B,iDAAiD;IAC1C,YAAY,CACjB,KAAK,EAAE,MAAM,EACb,UAAU,CAAC,EAAE,MAAM,GAClB,cAAc,CAAC,QAAQ,CAAC;IAS3B,uCAAuC;IAChC,SAAS,CACd,KAAK,EAAE,MAAM,EACb,UAAU,CAAC,EAAE,MAAM,GAClB,cAAc,CAAC,QAAQ,CAAC;IAS3B,yEAAyE;YAC1D,gBAAgB;IAiE/B,4BAA4B;IAC5B,SAAS,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI;IAI9B,8CAA8C;IAC9C,SAAS,CAAC,KAAK,CAAC,EAAE,MAAM,GAAG,IAAI;IAI/B,+EAA+E;IAC/E,IAAI,CAAC,QAAQ,CAAC,EAAE,MAAM,GAAG,IAAI;IAM7B,UAAU,IAAI,IAAI;CAMnB"}