@stigmer/react 0.0.36 → 0.0.40

package/execution/__tests__/useExecutionStream.test.js

@@ -0,0 +1,190 @@
+import { jsx as _jsx } from "react/jsx-runtime";
+import { describe, it, expect, vi, beforeEach } from "vitest";
+import { renderHook, act, waitFor } from "@testing-library/react";
+import { create } from "@bufbuild/protobuf";
+import { AgentExecutionSchema, AgentExecutionStatusSchema, } from "@stigmer/protos/ai/stigmer/agentic/agentexecution/v1/api_pb";
+import { ExecutionPhase } from "@stigmer/protos/ai/stigmer/agentic/agentexecution/v1/enum_pb";
+import { StigmerContext } from "../../context";
+import { useExecutionStream } from "../useExecutionStream";
+/**
+ * Creates a controllable async generator. Call `push(value)` to yield
+ * the next value, `finish()` to end the stream, or `fail(err)` to
+ * throw an error from the stream.
+ */
+function createControllableStream() {
+    let resolve = null;
+    let reject = null;
+    const generator = {
+        next() {
+            return new Promise((res, rej) => {
+                resolve = res;
+                reject = rej;
+            });
+        },
+        return(value) {
+            return Promise.resolve({ done: true, value: value });
+        },
+        throw(err) {
+            return Promise.reject(err);
+        },
+        [Symbol.asyncIterator]() {
+            return generator;
+        },
+    };
+    return {
+        generator,
+        push(value) {
+            resolve?.({ done: false, value });
+        },
+        finish() {
+            resolve?.({ done: true, value: undefined });
+        },
+        fail(err) {
+            reject?.(err);
+        },
+    };
+}
+function makeSnapshot(phase) {
+    const exec = create(AgentExecutionSchema);
+    const status = create(AgentExecutionStatusSchema);
+    status.phase = phase;
+    exec.status = status;
+    return exec;
+}
+function createMockStigmer(subscribeFn) {
+    return {
+        agentExecution: { subscribe: subscribeFn },
+    };
+}
+function createWrapper(client) {
+    return function Wrapper({ children }) {
+        return (_jsx(StigmerContext.Provider, { value: client, children: children }));
+    };
+}
+describe("useExecutionStream", () => {
+    let stream;
+    let subscribeFn;
+    let mockStigmer;
+    beforeEach(() => {
+        stream = createControllableStream();
+        subscribeFn = vi.fn().mockReturnValue(stream.generator);
+        mockStigmer = createMockStigmer(subscribeFn);
+    });
+    it("returns idle state when executionId is null", () => {
+        const { result } = renderHook(() => useExecutionStream(null), {
+            wrapper: createWrapper(mockStigmer),
+        });
+        expect(result.current.execution).toBeNull();
+        expect(result.current.isConnecting).toBe(false);
+        expect(result.current.isStreaming).toBe(false);
+        expect(result.current.error).toBeNull();
+        expect(subscribeFn).not.toHaveBeenCalled();
+    });
+    it("starts in isConnecting state when given an ID", () => {
+        const { result } = renderHook(() => useExecutionStream("exec-1"), {
+            wrapper: createWrapper(mockStigmer),
+        });
+        expect(result.current.isConnecting).toBe(true);
+        expect(result.current.isStreaming).toBe(false);
+        expect(result.current.execution).toBeNull();
+    });
+    it("transitions to isStreaming after first non-terminal snapshot", async () => {
+        const { result } = renderHook(() => useExecutionStream("exec-1"), {
+            wrapper: createWrapper(mockStigmer),
+        });
+        act(() => {
+            stream.push(makeSnapshot(ExecutionPhase.EXECUTION_IN_PROGRESS));
+        });
+        await waitFor(() => {
+            expect(result.current.isStreaming).toBe(true);
+            expect(result.current.isConnecting).toBe(false);
+            expect(result.current.execution).not.toBeNull();
+        });
+    });
+    it("updates execution with each streamed snapshot", async () => {
+        const { result } = renderHook(() => useExecutionStream("exec-1"), {
+            wrapper: createWrapper(mockStigmer),
+        });
+        const snap1 = makeSnapshot(ExecutionPhase.EXECUTION_PENDING);
+        act(() => stream.push(snap1));
+        await waitFor(() => {
+            expect(result.current.execution).toBe(snap1);
+        });
+        const snap2 = makeSnapshot(ExecutionPhase.EXECUTION_IN_PROGRESS);
+        act(() => stream.push(snap2));
+        await waitFor(() => {
+            expect(result.current.execution).toBe(snap2);
+        });
+    });
+    it("sets isStreaming to false when a terminal phase is received", async () => {
+        const { result } = renderHook(() => useExecutionStream("exec-1"), {
+            wrapper: createWrapper(mockStigmer),
+        });
+        act(() => {
+            stream.push(makeSnapshot(ExecutionPhase.EXECUTION_COMPLETED));
+        });
+        await waitFor(() => {
+            expect(result.current.isStreaming).toBe(false);
+            expect(result.current.execution).not.toBeNull();
+            expect(result.current.phase).toBe(ExecutionPhase.EXECUTION_COMPLETED);
+        });
+    });
+    it("sets error when the stream throws", async () => {
+        const { result } = renderHook(() => useExecutionStream("exec-1"), {
+            wrapper: createWrapper(mockStigmer),
+        });
+        act(() => {
+            stream.fail(new Error("connection lost"));
+        });
+        await waitFor(() => {
+            expect(result.current.error?.message).toBe("connection lost");
+            expect(result.current.isConnecting).toBe(false);
+            expect(result.current.isStreaming).toBe(false);
+        });
+    });
+    it("reconnect clears error and re-subscribes", async () => {
+        const { result } = renderHook(() => useExecutionStream("exec-1"), {
+            wrapper: createWrapper(mockStigmer),
+        });
+        act(() => stream.fail(new Error("timeout")));
+        await waitFor(() => {
+            expect(result.current.error?.message).toBe("timeout");
+        });
+        const newStream = createControllableStream();
+        subscribeFn.mockReturnValue(newStream.generator);
+        act(() => result.current.reconnect());
+        await waitFor(() => {
+            expect(result.current.error).toBeNull();
+            expect(result.current.isConnecting).toBe(true);
+        });
+        expect(subscribeFn).toHaveBeenCalledTimes(2);
+    });
+    it("aborts previous subscription when executionId changes", async () => {
+        const { result, rerender } = renderHook(({ id }) => useExecutionStream(id), {
+            wrapper: createWrapper(mockStigmer),
+            initialProps: { id: "exec-1" },
+        });
+        act(() => {
+            stream.push(makeSnapshot(ExecutionPhase.EXECUTION_IN_PROGRESS));
+        });
+        await waitFor(() => {
+            expect(result.current.isStreaming).toBe(true);
+        });
+        const newStream = createControllableStream();
+        subscribeFn.mockReturnValue(newStream.generator);
+        rerender({ id: "exec-2" });
+        expect(result.current.isConnecting).toBe(true);
+        expect(result.current.execution).toBeNull();
+        expect(subscribeFn).toHaveBeenCalledTimes(2);
+    });
+    it("cleans up subscription on unmount", () => {
+        const { unmount } = renderHook(() => useExecutionStream("exec-1"), {
+            wrapper: createWrapper(mockStigmer),
+        });
+        const signal = subscribeFn.mock.calls[0][1];
+        expect(signal.aborted).toBe(false);
+        unmount();
+        expect(signal.aborted).toBe(true);
+    });
+});
+//# sourceMappingURL=useExecutionStream.test.js.map

package/execution/__tests__/useExecutionStream.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"useExecutionStream.test.js","sourceRoot":"","sources":["../../../src/execution/__tests__/useExecutionStream.test.tsx"],"names":[],"mappings":";AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,UAAU,EAAE,GAAG,EAAE,OAAO,EAAE,MAAM,wBAAwB,CAAC;AAElE,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAC5C,OAAO,EACL,oBAAoB,EACpB,0BAA0B,GAE3B,MAAM,6DAA6D,CAAC;AACrE,OAAO,EAAE,cAAc,EAAE,MAAM,8DAA8D,CAAC;AAE9F,OAAO,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAC/C,OAAO,EAAE,kBAAkB,EAAE,MAAM,uBAAuB,CAAC;AAE3D;;;;GAIG;AACH,SAAS,wBAAwB;IAC/B,IAAI,OAAO,GAA4C,IAAI,CAAC;IAC5D,IAAI,MAAM,GAAoC,IAAI,CAAC;IAEnD,MAAM,SAAS,GAAsB;QACnC,IAAI;YACF,OAAO,IAAI,OAAO,CAAoB,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE;gBACjD,OAAO,GAAG,GAAG,CAAC;gBACd,MAAM,GAAG,GAAG,CAAC;YACf,CAAC,CAAC,CAAC;QACL,CAAC;QACD,MAAM,CAAC,KAAe;YACpB,OAAO,OAAO,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,IAAa,EAAE,KAAK,EAAE,KAAU,EAAE,CAAC,CAAC;QACrE,CAAC;QACD,KAAK,CAAC,GAAa;YACjB,OAAO,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAC7B,CAAC;QACD,CAAC,MAAM,CAAC,aAAa,CAAC;YACpB,OAAO,SAAS,CAAC;QACnB,CAAC;KACF,CAAC;IAEF,OAAO;QACL,SAAS;QACT,IAAI,CAAC,KAAQ;YACX,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;QACpC,CAAC;QACD,MAAM;YACJ,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,SAAyB,EAAE,CAAC,CAAC;QAC9D,CAAC;QACD,IAAI,CAAC,GAAU;YACb,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC;QAChB,CAAC;KACF,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CAAC,KAAqB;IACzC,MAAM,IAAI,GAAG,MAAM,CAAC,oBAAoB,CAAC,CAAC;IAC1C,MAAM,MAAM,GAAG,MAAM,CAAC,0BAA0B,CAAC,CAAC;IAClD,MAAM,CAAC,KAAK,GAAG,KAAK,CAAC;IACrB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACrB,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,iBAAiB,CAAC,WAAmD;IAC5E,OAAO;QACL,cAAc,EAAE,EAAE,SAAS,EAAE,WAAW,EAAE;KACrB,CAAC;AAC1B,CAAC;AAED,SAAS,aAAa,CAAC,MAAe;IACpC,OAAO,SAAS,OAAO,CAAC,EAAE,QAAQ,EAA2B;QAC3D,OAAO,CACL,KAAC,cAAc,CAAC,QAAQ,IAAC,KAAK,EAAE,MAAM,YAAG,QAAQ,GAA2B,CAC7E,CAAC;IACJ,CAAC,CAAC;AACJ,CAAC;AAED,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;IAClC,IAAI,MAAmE,CAAC;IACxE,IAAI,WAAqC,CAAC;IAC1C,IAAI,WAAoB,CAAC;IAEzB,UAAU,CAAC,GAAG,EAAE;QACd,MAAM,GAAG,wBAAwB,EAAkB,CAAC;QACpD,WAAW,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QACxD,WAAW,GAAG,iBAAiB,CAAC,WAAW,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;QACrD,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,kBAAkB,CAAC,IAAI,CAAC,EAAE;YAC5D,OAAO,EAAE,aAAa,CAAC,WAAW,CAAC;SACpC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,QAAQ,EAAE,CAAC;QAC5C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAChD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC/C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,QAAQ,EAAE,CAAC;QACxC,MAAM,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;QACvD,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,kBAAkB,CAAC,QAAQ,CAAC,EAAE;YAChE,OAAO,EAAE,aAAa,CAAC,WAAW,CAAC;SACpC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC/C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC/C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,QAAQ,EAAE,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;QAC5E,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,kBAAkB,CAAC,QAAQ,CAAC,EAAE;YAChE,OAAO,EAAE,aAAa,CAAC,WAAW,CAAC;SACpC,CAAC,CAAC;QAEH,GAAG,CAAC,GAAG,EAAE;YACP,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,cAAc,CAAC,qBAAqB,CAAC,CAAC,CAAC;QAClE,CAAC,CAAC,CAAC;QAEH,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC9C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAChD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;QAClD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;QAC7D,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,kBAAkB,CAAC,QAAQ,CAAC,EAAE;YAChE,OAAO,EAAE,aAAa,CAAC,WAAW,CAAC;SACpC,CAAC,CAAC;QAEH,MAAM,KAAK,GAAG,YAAY,CAAC,cAAc,CAAC,iBAAiB,CAAC,CAAC;QAC7D,GAAG,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;QAE9B,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,MAAM,KAAK,GAAG,YAAY,CAAC,cAAc,CAAC,qBAAqB,CAAC,CAAC;QACjE,GAAG,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;QAE9B,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;QAC3E,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,kBAAkB,CAAC,QAAQ,CAAC,EAAE;YAChE,OAAO,EAAE,aAAa,CAAC,WAAW,CAAC;SACpC,CAAC,CAAC;QAEH,GAAG,CAAC,GAAG,EAAE;YACP,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,cAAc,CAAC,mBAAmB,CAAC,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC/C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;YAChD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,mBAAmB,CAAC,CAAC;QACxE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;QACjD,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,kBAAkB,CAAC,QAAQ,CAAC,EAAE;YAChE,OAAO,EAAE,aAAa,CAAC,WAAW,CAAC;SACpC,CAAC,CAAC;QAEH,GAAG,CAAC,GAAG,EAAE;YACP,MAAM,CAAC,IAAI,CAAC,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;YAC9D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAChD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;QACxD,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,kBAAkB,CAAC,QAAQ,CAAC,EAAE;YAChE,OAAO,EAAE,aAAa,CAAC,WAAW,CAAC;SACpC,CAAC,CAAC;QAEH,GAAG,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;QAE7C,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACxD,CAAC,CAAC,CAAC;QAEH,MAAM,SAAS,GAAG,wBAAwB,EAAkB,CAAC;QAC7D,WAAW,CAAC,eAAe,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QAEjD,GAAG,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC;QAEtC,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,QAAQ,EAAE,CAAC;YACxC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,WAAW,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;QACrE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,GAAG,UAAU,CACrC,CAAC,EAAE,EAAE,EAAyB,EAAE,EAAE,CAAC,kBAAkB,CAAC,EAAE,CAAC,EACzD;YACE,OAAO,EAAE,aAAa,CAAC,WAAW,CAAC;YACnC,YAAY,EAAE,EAAE,EAAE,EAAE,QAAQ,EAAE;SAC/B,CACF,CAAC;QAEF,GAAG,CAAC,GAAG,EAAE;YACP,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,cAAc,CAAC,qBAAqB,CAAC,CAAC,CAAC;QAClE,CAAC,CAAC,CAAC;QAEH,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,MAAM,SAAS,GAAG,wBAAwB,EAAkB,CAAC;QAC7D,WAAW,CAAC,eAAe,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QAEjD,QAAQ,CAAC,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAC;QAE3B,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC/C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,QAAQ,EAAE,CAAC;QAC5C,MAAM,CAAC,WAAW,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;QAC3C,MAAM,EAAE,OAAO,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,kBAAkB,CAAC,QAAQ,CAAC,EAAE;YACjE,OAAO,EAAE,aAAa,CAAC,WAAW,CAAC;SACpC,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAgB,CAAC;QAC3D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAEnC,OAAO,EAAE,CAAC;QAEV,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/execution/__tests__/useExecutionUsage.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=useExecutionUsage.test.d.ts.map

package/execution/__tests__/useExecutionUsage.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"useExecutionUsage.test.d.ts","sourceRoot":"","sources":["../../../src/execution/__tests__/useExecutionUsage.test.tsx"],"names":[],"mappings":""}

package/execution/__tests__/useExecutionUsage.test.js

@@ -0,0 +1,303 @@
+import { describe, it, expect } from "vitest";
+import { renderHook } from "@testing-library/react";
+import { create } from "@bufbuild/protobuf";
+import { AgentExecutionSchema, AgentExecutionStatusSchema, } from "@stigmer/protos/ai/stigmer/agentic/agentexecution/v1/api_pb";
+import { SubAgentExecutionSchema } from "@stigmer/protos/ai/stigmer/agentic/agentexecution/v1/subagent_pb";
+import { UsageMetricsSchema, ModelUsageSchema, LlmCallMetricsSchema, } from "@stigmer/protos/ai/stigmer/agentic/agentexecution/v1/usage_pb";
+import { aggregateUsage, useExecutionUsage } from "../useExecutionUsage";
+// ---------------------------------------------------------------------------
+// Test helpers
+// ---------------------------------------------------------------------------
+function makeUsage(overrides = {}) {
+    return create(UsageMetricsSchema, overrides);
+}
+function makeModelUsage(model, provider, overrides = {}) {
+    return create(ModelUsageSchema, { model, provider, ...overrides });
+}
+function makeLlmCall(sequence, timestamp, overrides = {}) {
+    return create(LlmCallMetricsSchema, { sequence, timestamp, ...overrides });
+}
+function makeExecution(mainUsage, subAgents = []) {
+    const status = create(AgentExecutionStatusSchema, {
+        usage: mainUsage,
+        subAgentExecutions: subAgents.map((s) => create(SubAgentExecutionSchema, { name: s.name, usage: s.usage })),
+    });
+    return create(AgentExecutionSchema, { status });
+}
+// ---------------------------------------------------------------------------
+// aggregateUsage — pure function tests
+// ---------------------------------------------------------------------------
+describe("aggregateUsage", () => {
+    it("returns null when execution is null", () => {
+        expect(aggregateUsage(null)).toBeNull();
+    });
+    it("returns null when status is undefined", () => {
+        const exec = create(AgentExecutionSchema);
+        expect(aggregateUsage(exec)).toBeNull();
+    });
+    it("returns null when status.usage is undefined", () => {
+        const exec = create(AgentExecutionSchema, {
+            status: create(AgentExecutionStatusSchema),
+        });
+        expect(aggregateUsage(exec)).toBeNull();
+    });
+    it("returns main-only usage when no sub-agents exist", () => {
+        const usage = makeUsage({
+            promptTokens: 100,
+            completionTokens: 50,
+            totalTokens: 150,
+            llmCallCount: 3,
+            estimatedCostUsd: 0.005,
+            primaryModel: "claude-sonnet-4",
+            primaryProvider: "anthropic",
+        });
+        const exec = makeExecution(usage);
+        const result = aggregateUsage(exec);
+        expect(result).toBe(usage);
+    });
+    it("returns main-only usage when sub-agents exist but none have usage", () => {
+        const usage = makeUsage({ promptTokens: 100, totalTokens: 100 });
+        const exec = makeExecution(usage, [
+            { name: "researcher" },
+            { name: "coder" },
+        ]);
+        const result = aggregateUsage(exec);
+        expect(result).toBe(usage);
+    });
+    it("sums token counts across main and sub-agents", () => {
+        const main = makeUsage({
+            promptTokens: 100,
+            completionTokens: 50,
+            totalTokens: 150,
+            cacheCreationTokens: 10,
+            cacheReadTokens: 20,
+        });
+        const sub1 = makeUsage({
+            promptTokens: 200,
+            completionTokens: 80,
+            totalTokens: 280,
+            cacheCreationTokens: 5,
+            cacheReadTokens: 30,
+        });
+        const sub2 = makeUsage({
+            promptTokens: 50,
+            completionTokens: 20,
+            totalTokens: 70,
+        });
+        const exec = makeExecution(main, [
+            { name: "sub1", usage: sub1 },
+            { name: "sub2", usage: sub2 },
+        ]);
+        const result = aggregateUsage(exec);
+        expect(result.promptTokens).toBe(350);
+        expect(result.completionTokens).toBe(150);
+        expect(result.totalTokens).toBe(500);
+        expect(result.cacheCreationTokens).toBe(15);
+        expect(result.cacheReadTokens).toBe(50);
+    });
+    it("sums cost across main and sub-agents", () => {
+        const main = makeUsage({ estimatedCostUsd: 0.01 });
+        const sub = makeUsage({ estimatedCostUsd: 0.005 });
+        const exec = makeExecution(main, [{ name: "sub", usage: sub }]);
+        const result = aggregateUsage(exec);
+        expect(result.estimatedCostUsd).toBeCloseTo(0.015);
+    });
+    it("sums durations across main and sub-agents", () => {
+        const main = makeUsage({
+            totalDurationMs: 5000,
+            llmDurationMs: 3000,
+            toolDurationMs: 1500,
+            approvalWaitDurationMs: 500,
+        });
+        const sub = makeUsage({
+            totalDurationMs: 2000,
+            llmDurationMs: 1000,
+            toolDurationMs: 800,
+            approvalWaitDurationMs: 200,
+        });
+        const exec = makeExecution(main, [{ name: "sub", usage: sub }]);
+        const result = aggregateUsage(exec);
+        expect(result.totalDurationMs).toBe(7000);
+        expect(result.llmDurationMs).toBe(4000);
+        expect(result.toolDurationMs).toBe(2300);
+        expect(result.approvalWaitDurationMs).toBe(700);
+    });
+    it("uses main agent primaryModel and primaryProvider", () => {
+        const main = makeUsage({
+            primaryModel: "claude-sonnet-4",
+            primaryProvider: "anthropic",
+        });
+        const sub = makeUsage({
+            primaryModel: "gpt-4o",
+            primaryProvider: "openai",
+        });
+        const exec = makeExecution(main, [{ name: "sub", usage: sub }]);
+        const result = aggregateUsage(exec);
+        expect(result.primaryModel).toBe("claude-sonnet-4");
+        expect(result.primaryProvider).toBe("anthropic");
+    });
+    it("merges modelBreakdown entries with same model+provider key", () => {
+        const main = makeUsage({
+            modelBreakdown: [
+                makeModelUsage("claude-sonnet-4", "anthropic", {
+                    inputTokens: 100,
+                    outputTokens: 50,
+                    callCount: 2,
+                    estimatedCostUsd: 0.005,
+                }),
+            ],
+        });
+        const sub = makeUsage({
+            modelBreakdown: [
+                makeModelUsage("claude-sonnet-4", "anthropic", {
+                    inputTokens: 200,
+                    outputTokens: 80,
+                    callCount: 3,
+                    estimatedCostUsd: 0.008,
+                }),
+            ],
+        });
+        const exec = makeExecution(main, [{ name: "sub", usage: sub }]);
+        const result = aggregateUsage(exec);
+        expect(result.modelBreakdown).toHaveLength(1);
+        expect(result.modelBreakdown[0].model).toBe("claude-sonnet-4");
+        expect(result.modelBreakdown[0].provider).toBe("anthropic");
+        expect(result.modelBreakdown[0].inputTokens).toBe(300);
+        expect(result.modelBreakdown[0].outputTokens).toBe(130);
+        expect(result.modelBreakdown[0].callCount).toBe(5);
+        expect(result.modelBreakdown[0].estimatedCostUsd).toBeCloseTo(0.013);
+    });
+    it("keeps modelBreakdown entries with different models separate", () => {
+        const main = makeUsage({
+            modelBreakdown: [
+                makeModelUsage("claude-sonnet-4", "anthropic", { callCount: 2 }),
+            ],
+        });
+        const sub = makeUsage({
+            modelBreakdown: [
+                makeModelUsage("gpt-4o", "openai", { callCount: 1 }),
+            ],
+        });
+        const exec = makeExecution(main, [{ name: "sub", usage: sub }]);
+        const result = aggregateUsage(exec);
+        expect(result.modelBreakdown).toHaveLength(2);
+        const models = result.modelBreakdown.map((m) => m.model).sort();
+        expect(models).toEqual(["claude-sonnet-4", "gpt-4o"]);
+    });
+    it("concatenates llmCalls from all agents sorted by timestamp", () => {
+        const main = makeUsage({
+            llmCalls: [
+                makeLlmCall(1, "2026-03-19T10:00:00Z"),
+                makeLlmCall(2, "2026-03-19T10:01:00Z"),
+            ],
+        });
+        const sub = makeUsage({
+            llmCalls: [
+                makeLlmCall(1, "2026-03-19T10:00:30Z"),
+            ],
+        });
+        const exec = makeExecution(main, [{ name: "sub", usage: sub }]);
+        const result = aggregateUsage(exec);
+        expect(result.llmCalls).toHaveLength(3);
+        expect(result.llmCalls[0].timestamp).toBe("2026-03-19T10:00:00Z");
+        expect(result.llmCalls[1].timestamp).toBe("2026-03-19T10:00:30Z");
+        expect(result.llmCalls[2].timestamp).toBe("2026-03-19T10:01:00Z");
+    });
+    it("skips sub-agents with undefined usage gracefully", () => {
+        const main = makeUsage({
+            promptTokens: 100,
+            llmCallCount: 2,
+        });
+        const subWithUsage = makeUsage({
+            promptTokens: 50,
+            llmCallCount: 1,
+        });
+        const exec = makeExecution(main, [
+            { name: "has-usage", usage: subWithUsage },
+            { name: "no-usage" },
+        ]);
+        const result = aggregateUsage(exec);
+        expect(result.promptTokens).toBe(150);
+        expect(result.llmCallCount).toBe(3);
+    });
+    it("sums toolResultCharsTruncated as bigint", () => {
+        const main = makeUsage({ toolResultCharsTruncated: 1000n });
+        const sub = makeUsage({ toolResultCharsTruncated: 500n });
+        const exec = makeExecution(main, [{ name: "sub", usage: sub }]);
+        const result = aggregateUsage(exec);
+        expect(result.toolResultCharsTruncated).toBe(1500n);
+    });
+    it("preserves pricing rates from first model entry", () => {
+        const main = makeUsage({
+            modelBreakdown: [
+                makeModelUsage("claude-sonnet-4", "anthropic", {
+                    inputPricePerMillion: 3.0,
+                    outputPricePerMillion: 15.0,
+                    callCount: 1,
+                }),
+            ],
+        });
+        const sub = makeUsage({
+            modelBreakdown: [
+                makeModelUsage("claude-sonnet-4", "anthropic", {
+                    inputPricePerMillion: 3.0,
+                    outputPricePerMillion: 15.0,
+                    callCount: 1,
+                }),
+            ],
+        });
+        const exec = makeExecution(main, [{ name: "sub", usage: sub }]);
+        const result = aggregateUsage(exec);
+        expect(result.modelBreakdown[0].inputPricePerMillion).toBe(3.0);
+        expect(result.modelBreakdown[0].outputPricePerMillion).toBe(15.0);
+    });
+});
+// ---------------------------------------------------------------------------
+// useExecutionUsage — hook tests
+// ---------------------------------------------------------------------------
+describe("useExecutionUsage", () => {
+    it("returns null usage and zero metadata for null execution", () => {
+        const { result } = renderHook(() => useExecutionUsage(null));
+        expect(result.current.usage).toBeNull();
+        expect(result.current.hasSubAgentUsage).toBe(false);
+        expect(result.current.subAgentUsageCount).toBe(0);
+    });
+    it("returns aggregated UsageMetrics for valid execution", () => {
+        const main = makeUsage({
+            promptTokens: 100,
+            estimatedCostUsd: 0.01,
+            primaryModel: "claude-sonnet-4",
+        });
+        const exec = makeExecution(main);
+        const { result } = renderHook(() => useExecutionUsage(exec));
+        expect(result.current.usage).not.toBeNull();
+        expect(result.current.usage.promptTokens).toBe(100);
+        expect(result.current.usage.estimatedCostUsd).toBeCloseTo(0.01);
+        expect(result.current.usage.primaryModel).toBe("claude-sonnet-4");
+        expect(result.current.hasSubAgentUsage).toBe(false);
+        expect(result.current.subAgentUsageCount).toBe(0);
+    });
+    it("returns stable reference when execution has not changed", () => {
+        const exec = makeExecution(makeUsage({ promptTokens: 100 }));
+        const { result, rerender } = renderHook(() => useExecutionUsage(exec));
+        const first = result.current;
+        rerender();
+        const second = result.current;
+        expect(first).toBe(second);
+    });
+    it("sets hasSubAgentUsage and subAgentUsageCount correctly", () => {
+        const main = makeUsage({ promptTokens: 100 });
+        const sub1 = makeUsage({ promptTokens: 50 });
+        const sub2 = makeUsage({ promptTokens: 30 });
+        const exec = makeExecution(main, [
+            { name: "sub1", usage: sub1 },
+            { name: "sub2", usage: sub2 },
+            { name: "sub3-no-usage" },
+        ]);
+        const { result } = renderHook(() => useExecutionUsage(exec));
+        expect(result.current.hasSubAgentUsage).toBe(true);
+        expect(result.current.subAgentUsageCount).toBe(2);
+        expect(result.current.usage.promptTokens).toBe(180);
+    });
+});
+//# sourceMappingURL=useExecutionUsage.test.js.map

package/execution/__tests__/useExecutionUsage.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"useExecutionUsage.test.js","sourceRoot":"","sources":["../../../src/execution/__tests__/useExecutionUsage.test.tsx"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AACpD,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAC5C,OAAO,EACL,oBAAoB,EACpB,0BAA0B,GAE3B,MAAM,6DAA6D,CAAC;AACrE,OAAO,EAAE,uBAAuB,EAAE,MAAM,kEAAkE,CAAC;AAC3G,OAAO,EACL,kBAAkB,EAClB,gBAAgB,EAChB,oBAAoB,GACrB,MAAM,+DAA+D,CAAC;AACvE,OAAO,EAAE,cAAc,EAAE,iBAAiB,EAAE,MAAM,sBAAsB,CAAC;AAEzE,8EAA8E;AAC9E,eAAe;AACf,8EAA8E;AAE9E,SAAS,SAAS,CAChB,YAiBK,EAAE;IAEP,OAAO,MAAM,CAAC,kBAAkB,EAAE,SAAS,CAAC,CAAC;AAC/C,CAAC;AAED,SAAS,cAAc,CACrB,KAAa,EACb,QAAgB,EAChB,YASK,EAAE;IAEP,OAAO,MAAM,CAAC,gBAAgB,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE,GAAG,SAAS,EAAE,CAAC,CAAC;AACrE,CAAC;AAED,SAAS,WAAW,CAClB,QAAgB,EAChB,SAAiB,EACjB,YAOK,EAAE;IAEP,OAAO,MAAM,CAAC,oBAAoB,EAAE,EAAE,QAAQ,EAAE,SAAS,EAAE,GAAG,SAAS,EAAE,CAAC,CAAC;AAC7E,CAAC;AAED,SAAS,aAAa,CACpB,SAAwC,EACxC,YAGK,EAAE;IAEP,MAAM,MAAM,GAAG,MAAM,CAAC,0BAA0B,EAAE;QAChD,KAAK,EAAE,SAAS;QAChB,kBAAkB,EAAE,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CACtC,MAAM,CAAC,uBAAuB,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC,KAAK,EAAE,CAAC,CAClE;KACF,CAAC,CAAC;IACH,OAAO,MAAM,CAAC,oBAAoB,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;AAClD,CAAC;AAED,8EAA8E;AAC9E,uCAAuC;AACvC,8EAA8E;AAE9E,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;QAC7C,MAAM,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;QAC/C,MAAM,IAAI,GAAG,MAAM,CAAC,oBAAoB,CAAC,CAAC;QAC1C,MAAM,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;QACrD,MAAM,IAAI,GAAG,MAAM,CAAC,oBAAoB,EAAE;YACxC,MAAM,EAAE,MAAM,CAAC,0BAA0B,CAAC;SAC3C,CAAC,CAAC;QACH,MAAM,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC1D,MAAM,KAAK,GAAG,SAAS,CAAC;YACtB,YAAY,EAAE,GAAG;YACjB,gBAAgB,EAAE,EAAE;YACpB,WAAW,EAAE,GAAG;YAChB,YAAY,EAAE,CAAC;YACf,gBAAgB,EAAE,KAAK;YACvB,YAAY,EAAE,iBAAiB;YAC/B,eAAe,EAAE,WAAW;SAC7B,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,aAAa,CAAC,KAAK,CAAC,CAAC;QAElC,MAAM,MAAM,GAAG,cAAc,CAAC,IAAI,CAAC,CAAC;QACpC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mEAAmE,EAAE,GAAG,EAAE;QAC3E,MAAM,KAAK,GAAG,SAAS,CAAC,EAAE,YAAY,EAAE,GAAG,EAAE,WAAW,EAAE,GAAG,EAAE,CAAC,CAAC;QACjE,MAAM,IAAI,GAAG,aAAa,CAAC,KAAK,EAAE;YAChC,EAAE,IAAI,EAAE,YAAY,EAAE;YACtB,EAAE,IAAI,EAAE,OAAO,EAAE;SAClB,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,cAAc,CAAC,IAAI,CAAC,CAAC;QACpC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,MAAM,IAAI,GAAG,SAAS,CAAC;YACrB,YAAY,EAAE,GAAG;YACjB,gBAAgB,EAAE,EAAE;YACpB,WAAW,EAAE,GAAG;YAChB,mBAAmB,EAAE,EAAE;YACvB,eAAe,EAAE,EAAE;SACpB,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,SAAS,CAAC;YACrB,YAAY,EAAE,GAAG;YACjB,gBAAgB,EAAE,EAAE;YACpB,WAAW,EAAE,GAAG;YAChB,mBAAmB,EAAE,CAAC;YACtB,eAAe,EAAE,EAAE;SACpB,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,SAAS,CAAC;YACrB,YAAY,EAAE,EAAE;YAChB,gBAAgB,EAAE,EAAE;YACpB,WAAW,EAAE,EAAE;SAChB,CAAC,CAAC;QAEH,MAAM,IAAI,GAAG,aAAa,CAAC,IAAI,EAAE;YAC/B,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE;YAC7B,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE;SAC9B,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,cAAc,CAAC,IAAI,CAAE,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACtC,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC1C,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC5C,MAAM,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;QAC9C,MAAM,IAAI,GAAG,SAAS,CAAC,EAAE,gBAAgB,EAAE,IAAI,EAAE,CAAC,CAAC;QACnD,MAAM,GAAG,GAAG,SAAS,CAAC,EAAE,gBAAgB,EAAE,KAAK,EAAE,CAAC,CAAC;QACnD,MAAM,IAAI,GAAG,aAAa,CAAC,IAAI,EAAE,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC;QAEhE,MAAM,MAAM,GAAG,cAAc,CAAC,IAAI,CAAE,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACnD,MAAM,IAAI,GAAG,SAAS,CAAC;YACrB,eAAe,EAAE,IAAI;YACrB,aAAa,EAAE,IAAI;YACnB,cAAc,EAAE,IAAI;YACpB,sBAAsB,EAAE,GAAG;SAC5B,CAAC,CAAC;QACH,MAAM,GAAG,GAAG,SAAS,CAAC;YACpB,eAAe,EAAE,IAAI;YACrB,aAAa,EAAE,IAAI;YACnB,cAAc,EAAE,GAAG;YACnB,sBAAsB,EAAE,GAAG;SAC5B,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,aAAa,CAAC,IAAI,EAAE,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC;QAEhE,MAAM,MAAM,GAAG,cAAc,CAAC,IAAI,CAAE,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC1C,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACxC,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzC,MAAM,CAAC,MAAM,CAAC,sBAAsB,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC1D,MAAM,IAAI,GAAG,SAAS,CAAC;YACrB,YAAY,EAAE,iBAAiB;YAC/B,eAAe,EAAE,WAAW;SAC7B,CAAC,CAAC;QACH,MAAM,GAAG,GAAG,SAAS,CAAC;YACpB,YAAY,EAAE,QAAQ;YACtB,eAAe,EAAE,QAAQ;SAC1B,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,aAAa,CAAC,IAAI,EAAE,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC;QAEhE,MAAM,MAAM,GAAG,cAAc,CAAC,IAAI,CAAE,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QACpD,MAAM,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;QACpE,MAAM,IAAI,GAAG,SAAS,CAAC;YACrB,cAAc,EAAE;gBACd,cAAc,CAAC,iBAAiB,EAAE,WAAW,EAAE;oBAC7C,WAAW,EAAE,GAAG;oBAChB,YAAY,EAAE,EAAE;oBAChB,SAAS,EAAE,CAAC;oBACZ,gBAAgB,EAAE,KAAK;iBACxB,CAAC;aACH;SACF,CAAC,CAAC;QACH,MAAM,GAAG,GAAG,SAAS,CAAC;YACpB,cAAc,EAAE;gBACd,cAAc,CAAC,iBAAiB,EAAE,WAAW,EAAE;oBAC7C,WAAW,EAAE,GAAG;oBAChB,YAAY,EAAE,EAAE;oBAChB,SAAS,EAAE,CAAC;oBACZ,gBAAgB,EAAE,KAAK;iBACxB,CAAC;aACH;SACF,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,aAAa,CAAC,IAAI,EAAE,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC;QAEhE,MAAM,MAAM,GAAG,cAAc,CAAC,IAAI,CAAE,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC9C,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAC/D,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC5D,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACvD,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACxD,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACnD,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,gBAAgB,CAAC,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;IACvE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6DAA6D,EAAE,GAAG,EAAE;QACrE,MAAM,IAAI,GAAG,SAAS,CAAC;YACrB,cAAc,EAAE;gBACd,cAAc,CAAC,iBAAiB,EAAE,WAAW,EAAE,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC;aACjE;SACF,CAAC,CAAC;QACH,MAAM,GAAG,GAAG,SAAS,CAAC;YACpB,cAAc,EAAE;gBACd,cAAc,CAAC,QAAQ,EAAE,QAAQ,EAAE,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC;aACrD;SACF,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,aAAa,CAAC,IAAI,EAAE,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC;QAEhE,MAAM,MAAM,GAAG,cAAc,CAAC,IAAI,CAAE,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAE9C,MAAM,MAAM,GAAG,MAAM,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC;QAChE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,iBAAiB,EAAE,QAAQ,CAAC,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2DAA2D,EAAE,GAAG,EAAE;QACnE,MAAM,IAAI,GAAG,SAAS,CAAC;YACrB,QAAQ,EAAE;gBACR,WAAW,CAAC,CAAC,EAAE,sBAAsB,CAAC;gBACtC,WAAW,CAAC,CAAC,EAAE,sBAAsB,CAAC;aACvC;SACF,CAAC,CAAC;QACH,MAAM,GAAG,GAAG,SAAS,CAAC;YACpB,QAAQ,EAAE;gBACR,WAAW,CAAC,CAAC,EAAE,sBAAsB,CAAC;aACvC;SACF,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,aAAa,CAAC,IAAI,EAAE,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC;QAEhE,MAAM,MAAM,GAAG,cAAc,CAAC,IAAI,CAAE,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACxC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;QAClE,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;QAClE,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;IACpE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC1D,MAAM,IAAI,GAAG,SAAS,CAAC;YACrB,YAAY,EAAE,GAAG;YACjB,YAAY,EAAE,CAAC;SAChB,CAAC,CAAC;QACH,MAAM,YAAY,GAAG,SAAS,CAAC;YAC7B,YAAY,EAAE,EAAE;YAChB,YAAY,EAAE,CAAC;SAChB,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,aAAa,CAAC,IAAI,EAAE;YAC/B,EAAE,IAAI,EAAE,WAAW,EAAE,KAAK,EAAE,YAAY,EAAE;YAC1C,EAAE,IAAI,EAAE,UAAU,EAAE;SACrB,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,cAAc,CAAC,IAAI,CAAE,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACtC,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;QACjD,MAAM,IAAI,GAAG,SAAS,CAAC,EAAE,wBAAwB,EAAE,KAAK,EAAE,CAAC,CAAC;QAC5D,MAAM,GAAG,GAAG,SAAS,CAAC,EAAE,wBAAwB,EAAE,IAAI,EAAE,CAAC,CAAC;QAC1D,MAAM,IAAI,GAAG,aAAa,CAAC,IAAI,EAAE,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC;QAEhE,MAAM,MAAM,GAAG,cAAc,CAAC,IAAI,CAAE,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,wBAAwB,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;QACxD,MAAM,IAAI,GAAG,SAAS,CAAC;YACrB,cAAc,EAAE;gBACd,cAAc,CAAC,iBAAiB,EAAE,WAAW,EAAE;oBAC7C,oBAAoB,EAAE,GAAG;oBACzB,qBAAqB,EAAE,IAAI;oBAC3B,SAAS,EAAE,CAAC;iBACb,CAAC;aACH;SACF,CAAC,CAAC;QACH,MAAM,GAAG,GAAG,SAAS,CAAC;YACpB,cAAc,EAAE;gBACd,cAAc,CAAC,iBAAiB,EAAE,WAAW,EAAE;oBAC7C,oBAAoB,EAAE,GAAG;oBACzB,qBAAqB,EAAE,IAAI;oBAC3B,SAAS,EAAE,CAAC;iBACb,CAAC;aACH;SACF,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,aAAa,CAAC,IAAI,EAAE,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC;QAEhE,MAAM,MAAM,GAAG,cAAc,CAAC,IAAI,CAAE,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,oBAAoB,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAChE,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,qBAAqB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACpE,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,8EAA8E;AAC9E,iCAAiC;AACjC,8EAA8E;AAE9E,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;IACjC,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;QACjE,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC,CAAC;QAE7D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,QAAQ,EAAE,CAAC;QACxC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACpD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,MAAM,IAAI,GAAG,SAAS,CAAC;YACrB,YAAY,EAAE,GAAG;YACjB,gBAAgB,EAAE,IAAI;YACtB,YAAY,EAAE,iBAAiB;SAChC,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC;QAEjC,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC,CAAC;QAE7D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;QAC5C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,KAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACrD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,KAAM,CAAC,gBAAgB,CAAC,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;QACjE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,KAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QACnE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACpD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;QACjE,MAAM,IAAI,GAAG,aAAa,CAAC,SAAS,CAAC,EAAE,YAAY,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC;QAC7D,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC,CAAC;QAEvE,MAAM,KAAK,GAAG,MAAM,CAAC,OAAO,CAAC;QAC7B,QAAQ,EAAE,CAAC;QACX,MAAM,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC;QAE9B,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;QAChE,MAAM,IAAI,GAAG,SAAS,CAAC,EAAE,YAAY,EAAE,GAAG,EAAE,CAAC,CAAC;QAC9C,MAAM,IAAI,GAAG,SAAS,CAAC,EAAE,YAAY,EAAE,EAAE,EAAE,CAAC,CAAC;QAC7C,MAAM,IAAI,GAAG,SAAS,CAAC,EAAE,YAAY,EAAE,EAAE,EAAE,CAAC,CAAC;QAC7C,MAAM,IAAI,GAAG,aAAa,CAAC,IAAI,EAAE;YAC/B,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE;YAC7B,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE;YAC7B,EAAE,IAAI,EAAE,eAAe,EAAE;SAC1B,CAAC,CAAC;QAEH,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC,CAAC;QAE7D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACnD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,KAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/execution/artifact-utils.d.ts

@@ -0,0 +1,58 @@
+import type { ExecutionArtifact } from "@stigmer/protos/ai/stigmer/agentic/agentexecution/v1/artifact_pb";
+/**
+ * Extracts the lowercase file extension from an artifact's name.
+ *
+ * Returns `null` when the name has no extension or is empty.
+ *
+ * @example
+ * ```ts
+ * getArtifactExtension(artifact); // "yaml"
+ * getArtifactExtension(noExtArtifact); // null
+ * ```
+ */
+export declare function getArtifactExtension(artifact: ExecutionArtifact): string | null;
+/**
+ * Determines whether an artifact likely contains human-readable text.
+ *
+ * Returns `false` for directory artifacts (which are stored as ZIP files)
+ * regardless of name. For file artifacts, checks whether the file
+ * extension is in the {@link TEXT_EXTENSIONS} set.
+ *
+ * Use this to decide whether to call {@link useArtifactContent} for
+ * a given artifact. Binary or unknown-extension artifacts should use
+ * the download URL instead.
+ *
+ * @example
+ * ```ts
+ * const shouldFetch = isTextArtifact(artifact) && Number(artifact.sizeBytes) < MAX_SIZE;
+ * ```
+ */
+export declare function isTextArtifact(artifact: ExecutionArtifact): boolean;
+/**
+ * Checks whether an artifact's pre-signed download URL has expired.
+ *
+ * Returns `true` when:
+ * - `expiresAt` is empty or cannot be parsed as a valid date
+ * - The current time is past the expiration timestamp
+ *
+ * Expired URLs can be refreshed via `stigmer.agentExecution.getArtifactDownloadUrl()`.
+ */
+export declare function isArtifactExpired(artifact: ExecutionArtifact): boolean;
+/**
+ * Formats a byte count as a human-readable file size string.
+ *
+ * Accepts both `bigint` (from protobuf `int64` fields like
+ * `ExecutionArtifact.sizeBytes`) and `number`.
+ *
+ * Uses base-1024 (binary) units with one decimal place.
+ *
+ * @example
+ * ```ts
+ * formatArtifactSize(0n);       // "0 B"
+ * formatArtifactSize(1024);     // "1.0 KB"
+ * formatArtifactSize(1536n);    // "1.5 KB"
+ * formatArtifactSize(2621440n); // "2.5 MB"
+ * ```
+ */
+export declare function formatArtifactSize(sizeBytes: bigint | number): string;
+//# sourceMappingURL=artifact-utils.d.ts.map

package/execution/artifact-utils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"artifact-utils.d.ts","sourceRoot":"","sources":["../../src/execution/artifact-utils.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,kEAAkE,CAAC;AAuC1G;;;;;;;;;;GAUG;AACH,wBAAgB,oBAAoB,CAClC,QAAQ,EAAE,iBAAiB,GAC1B,MAAM,GAAG,IAAI,CAKf;AAED;;;;;;;;;;;;;;;GAeG;AACH,wBAAgB,cAAc,CAAC,QAAQ,EAAE,iBAAiB,GAAG,OAAO,CAOnE;AAED;;;;;;;;GAQG;AACH,wBAAgB,iBAAiB,CAAC,QAAQ,EAAE,iBAAiB,GAAG,OAAO,CAOtE;AAID;;;;;;;;;;;;;;;GAeG;AACH,wBAAgB,kBAAkB,CAAC,SAAS,EAAE,MAAM,GAAG,MAAM,GAAG,MAAM,CAerE"}