runtimeuse 0.2.0

package/dist/session.test.js

@@ -0,0 +1,339 @@
+import { describe, it, expect, vi, beforeEach } from "vitest";
+import { EventEmitter } from "events";
+import { WebSocket } from "ws";
+const mockArtifactManager = {
+    setLogger: vi.fn(),
+    handleUploadResponse: vi.fn().mockResolvedValue(undefined),
+    waitForPendingRequests: vi.fn().mockResolvedValue(undefined),
+    stopWatching: vi.fn().mockResolvedValue(undefined),
+};
+vi.mock("./artifact-manager.js", () => ({
+    ArtifactManager: vi.fn().mockImplementation(function () {
+        return mockArtifactManager;
+    }),
+}));
+const mockDownloadHandler = {
+    download: vi.fn().mockResolvedValue(undefined),
+};
+vi.mock("./download-handler.js", () => ({
+    default: vi.fn().mockImplementation(function () {
+        return mockDownloadHandler;
+    }),
+}));
+const mockCommandExecute = vi.fn();
+vi.mock("./command-handler.js", () => ({
+    default: vi.fn().mockImplementation(function () {
+        return { execute: mockCommandExecute };
+    }),
+}));
+vi.mock("fs", async () => {
+    const actual = await vi.importActual("fs");
+    return {
+        ...actual,
+        default: {
+            ...actual,
+            createWriteStream: vi.fn(() => ({ write: vi.fn(), end: vi.fn(), close: vi.fn() })),
+        },
+    };
+});
+import { WebSocketSession } from "./session.js";
+import { UploadTracker } from "./upload-tracker.js";
+import CommandHandler from "./command-handler.js";
+const mockHandlerRun = vi.fn();
+const mockHandler = {
+    run: mockHandlerRun,
+};
+class MockWebSocket extends EventEmitter {
+    readyState = WebSocket.OPEN;
+    send = vi.fn();
+    close = vi.fn(() => {
+        if (this.readyState !== WebSocket.CLOSED) {
+            this.readyState = WebSocket.CLOSED;
+            this.emit("close");
+        }
+    });
+}
+function createSession() {
+    const ws = new MockWebSocket();
+    const uploadTracker = new UploadTracker();
+    const config = {
+        handler: mockHandler,
+        uploadTracker,
+    };
+    const session = new WebSocketSession(ws, config);
+    return { session, ws, config, uploadTracker };
+}
+function sendMessage(ws, message) {
+    ws.emit("message", Buffer.from(JSON.stringify(message)));
+}
+const INVOCATION_MSG = {
+    message_type: "invocation_message",
+    source_id: "test-source-id",
+    system_prompt: "You are a tester",
+    user_prompt: "Test the login flow",
+    secrets_to_redact: ["secret123"],
+    output_format_json_schema_str: JSON.stringify({
+        type: "json_schema",
+        schema: { type: "object" },
+    }),
+    preferred_model: "test-model",
+};
+describe("WebSocketSession", () => {
+    beforeEach(() => {
+        vi.clearAllMocks();
+        mockHandlerRun.mockResolvedValue({
+            structuredOutput: { success: true },
+            metadata: { duration_ms: 1000 },
+        });
+        mockArtifactManager.handleUploadResponse.mockResolvedValue(undefined);
+        mockArtifactManager.waitForPendingRequests.mockResolvedValue(undefined);
+        mockArtifactManager.stopWatching.mockResolvedValue(undefined);
+        mockDownloadHandler.download.mockResolvedValue(undefined);
+        mockCommandExecute.mockResolvedValue({ exitCode: 0 });
+    });
+    describe("lifecycle", () => {
+        it("resolves when invocation finishes", async () => {
+            const { session, ws } = createSession();
+            const done = session.run();
+            sendMessage(ws, INVOCATION_MSG);
+            await done;
+        });
+    });
+    describe("message routing", () => {
+        it("rejects non-invocation messages before invocation", async () => {
+            const { session, ws } = createSession();
+            session.run();
+            sendMessage(ws, { message_type: "cancel_message" });
+            await tick();
+            expectSentError(ws, "non-invocation message before invocation");
+        });
+        it("rejects duplicate invocation messages", async () => {
+            let resolveAgent;
+            mockHandlerRun.mockImplementation(() => new Promise((r) => {
+                resolveAgent = () => r({ structuredOutput: { success: true } });
+            }));
+            const { session, ws } = createSession();
+            session.run();
+            sendMessage(ws, INVOCATION_MSG);
+            await tick();
+            sendMessage(ws, INVOCATION_MSG);
+            await tick();
+            expectSentError(ws, "multiple invocation messages");
+            resolveAgent();
+        });
+        it("delegates artifact upload responses to ArtifactManager", async () => {
+            let resolveAgent;
+            mockHandlerRun.mockImplementation(() => new Promise((r) => {
+                resolveAgent = () => r({ structuredOutput: { success: true } });
+            }));
+            const { session, ws } = createSession();
+            const done = session.run();
+            sendMessage(ws, INVOCATION_MSG);
+            await tick();
+            const uploadResponse = {
+                message_type: "artifact_upload_response_message",
+                filename: "shot.png",
+                filepath: "/artifacts/shot.png",
+                presigned_url: "https://s3.example.com/upload",
+            };
+            sendMessage(ws, uploadResponse);
+            await tick();
+            expect(mockArtifactManager.handleUploadResponse).toHaveBeenCalledWith(uploadResponse);
+            resolveAgent();
+            await done;
+        });
+        it("aborts and closes on cancel message", async () => {
+            let resolveAgent;
+            mockHandlerRun.mockImplementation(() => new Promise((r) => {
+                resolveAgent = () => r({ structuredOutput: {} });
+            }));
+            const { session, ws } = createSession();
+            const done = session.run();
+            sendMessage(ws, INVOCATION_MSG);
+            await tick();
+            sendMessage(ws, { message_type: "cancel_message" });
+            await tick();
+            expect(ws.close).toHaveBeenCalled();
+            resolveAgent();
+        });
+    });
+    describe("invocation", () => {
+        it("calls handler.run with correct arguments", async () => {
+            const { session, ws } = createSession();
+            const done = session.run();
+            sendMessage(ws, INVOCATION_MSG);
+            await done;
+            expect(mockHandlerRun).toHaveBeenCalledWith(expect.objectContaining({
+                systemPrompt: "You are a tester",
+                userPrompt: "Test the login flow",
+                outputFormat: { type: "json_schema", schema: { type: "object" } },
+                model: "test-model",
+                secrets: ["secret123"],
+            }), expect.objectContaining({
+                sendAssistantMessage: expect.any(Function),
+                sendErrorMessage: expect.any(Function),
+            }));
+        });
+        it("sends result_message from handler result", async () => {
+            mockHandlerRun.mockResolvedValue({
+                structuredOutput: { success: true, steps: ["step1"] },
+                metadata: { duration_ms: 1000 },
+            });
+            const { session, ws } = createSession();
+            const done = session.run();
+            sendMessage(ws, INVOCATION_MSG);
+            await done;
+            const sent = parseSentMessages(ws);
+            const result = sent.find((m) => m.message_type === "result_message");
+            expect(result).toBeDefined();
+            expect(result.structured_output.success).toBe(true);
+            expect(result.metadata).toMatchObject({ duration_ms: 1000 });
+        });
+        it("sends error message when agent throws", async () => {
+            mockHandlerRun.mockRejectedValueOnce(new Error("agent crashed"));
+            const { session, ws } = createSession();
+            const done = session.run();
+            sendMessage(ws, INVOCATION_MSG);
+            await done;
+            expectSentError(ws, "agent crashed");
+        });
+    });
+    describe("finalization", () => {
+        it("stops the artifact watcher", async () => {
+            const { session, ws } = createSession();
+            const done = session.run();
+            sendMessage(ws, INVOCATION_MSG);
+            await done;
+            expect(mockArtifactManager.stopWatching).toHaveBeenCalled();
+        });
+        it("waits for pending artifact requests", async () => {
+            const { session, ws } = createSession();
+            const done = session.run();
+            sendMessage(ws, INVOCATION_MSG);
+            await done;
+            expect(mockArtifactManager.waitForPendingRequests).toHaveBeenCalledWith(60_000);
+        });
+        it("closes the websocket after finalization", async () => {
+            const { session, ws } = createSession();
+            const done = session.run();
+            sendMessage(ws, INVOCATION_MSG);
+            await done;
+            expect(ws.close).toHaveBeenCalled();
+        });
+    });
+    describe("runtime environment downloadables", () => {
+        it("downloads all runtime environment downloadables before running agent", async () => {
+            const { session, ws } = createSession();
+            const done = session.run();
+            sendMessage(ws, {
+                ...INVOCATION_MSG,
+                runtime_environment_downloadables: [
+                    {
+                        download_url: "https://example.com/test.zip",
+                        working_dir: "/tmp/test",
+                    },
+                    {
+                        download_url: "https://example.com/data.tar.gz",
+                        working_dir: "/tmp/data",
+                    },
+                ],
+            });
+            await done;
+            expect(mockDownloadHandler.download).toHaveBeenCalledTimes(2);
+            expect(mockDownloadHandler.download).toHaveBeenCalledWith("https://example.com/test.zip", "/tmp/test");
+            expect(mockDownloadHandler.download).toHaveBeenCalledWith("https://example.com/data.tar.gz", "/tmp/data");
+        });
+        it("does not download when no downloadables are provided", async () => {
+            const { session, ws } = createSession();
+            const done = session.run();
+            sendMessage(ws, INVOCATION_MSG);
+            await done;
+            expect(mockDownloadHandler.download).not.toHaveBeenCalled();
+        });
+    });
+    describe("pre-agent invocation commands", () => {
+        it("executes pre-agent commands and sends success result", async () => {
+            mockCommandExecute.mockResolvedValueOnce({ exitCode: 0 });
+            const { session, ws } = createSession();
+            const done = session.run();
+            sendMessage(ws, {
+                ...INVOCATION_MSG,
+                pre_agent_invocation_commands: [
+                    { command: "npm test", cwd: "/app", env: { CI: "true" } },
+                ],
+            });
+            await done;
+            expect(CommandHandler).toHaveBeenCalledWith({ command: "npm test", cwd: "/app", env: { CI: "true" } }, expect.any(Object), expect.any(Object), expect.any(Object), expect.any(AbortController), expect.any(Function), expect.any(Function));
+            const sent = parseSentMessages(ws);
+            const result = sent.find((m) => m.message_type === "result_message");
+            expect(result).toBeDefined();
+            expect(result.structured_output.success).toBe(true);
+        });
+        it("does not call handler.run when pre-agent command succeeds", async () => {
+            mockCommandExecute.mockResolvedValueOnce({ exitCode: 0 });
+            const { session, ws } = createSession();
+            const done = session.run();
+            sendMessage(ws, {
+                ...INVOCATION_MSG,
+                pre_agent_invocation_commands: [{ command: "npm test", cwd: "/app" }],
+            });
+            await done;
+            expect(mockHandlerRun).not.toHaveBeenCalled();
+        });
+        it("sends failure result when pre-agent command exits with non-zero code", async () => {
+            mockCommandExecute.mockResolvedValueOnce({
+                exitCode: 1,
+                error: new Error("test failed"),
+            });
+            const { session, ws } = createSession();
+            const done = session.run();
+            sendMessage(ws, {
+                ...INVOCATION_MSG,
+                pre_agent_invocation_commands: [{ command: "npm test", cwd: "/app" }],
+            });
+            await done;
+            const sent = parseSentMessages(ws);
+            const result = sent.find((m) => m.message_type === "result_message");
+            expect(result).toBeDefined();
+            expect(result.structured_output.success).toBe(false);
+        });
+        it("sends error and result messages when command execution throws", async () => {
+            mockCommandExecute.mockRejectedValueOnce(new Error("spawn failed"));
+            const { session, ws } = createSession();
+            const done = session.run();
+            sendMessage(ws, {
+                ...INVOCATION_MSG,
+                pre_agent_invocation_commands: [{ command: "bad-cmd" }],
+            });
+            await done;
+            const sent = parseSentMessages(ws);
+            const errors = sent.filter((m) => m.message_type === "error_message");
+            expect(errors.length).toBeGreaterThan(0);
+            const result = sent.find((m) => m.message_type === "result_message");
+            expect(result).toBeDefined();
+            expect(result.structured_output.success).toBe(false);
+        });
+        it("runs agent normally when no pre-agent commands are provided", async () => {
+            const { session, ws } = createSession();
+            const done = session.run();
+            sendMessage(ws, INVOCATION_MSG);
+            await done;
+            expect(CommandHandler).not.toHaveBeenCalled();
+            expect(mockHandlerRun).toHaveBeenCalled();
+        });
+    });
+});
+async function tick() {
+    await new Promise((r) => setTimeout(r, 10));
+}
+function parseSentMessages(ws) {
+    return ws.send.mock.calls.map((c) => JSON.parse(c[0]));
+}
+function expectSentError(ws, substring) {
+    const sent = parseSentMessages(ws);
+    const errors = sent.filter((m) => m.message_type === "error_message");
+    expect(errors.length).toBeGreaterThan(0);
+    const match = errors.some((m) => m.error.includes(substring));
+    expect(match).toBe(true);
+}
+//# sourceMappingURL=session.test.js.map

package/dist/session.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"session.test.js","sourceRoot":"","sources":["../src/session.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AACtC,OAAO,EAAE,SAAS,EAAE,MAAM,IAAI,CAAC;AAE/B,MAAM,mBAAmB,GAAG;IAC1B,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE;IAClB,oBAAoB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;IAC1D,sBAAsB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;IAC5D,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;CACnD,CAAC;AAEF,EAAE,CAAC,IAAI,CAAC,uBAAuB,EAAE,GAAG,EAAE,CAAC,CAAC;IACtC,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC;QAC1C,OAAO,mBAAmB,CAAC;IAC7B,CAAC,CAAC;CACH,CAAC,CAAC,CAAC;AAEJ,MAAM,mBAAmB,GAAG;IAC1B,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;CAC/C,CAAC;AAEF,EAAE,CAAC,IAAI,CAAC,uBAAuB,EAAE,GAAG,EAAE,CAAC,CAAC;IACtC,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC;QAClC,OAAO,mBAAmB,CAAC;IAC7B,CAAC,CAAC;CACH,CAAC,CAAC,CAAC;AAEJ,MAAM,kBAAkB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;AAEnC,EAAE,CAAC,IAAI,CAAC,sBAAsB,EAAE,GAAG,EAAE,CAAC,CAAC;IACrC,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC;QAClC,OAAO,EAAE,OAAO,EAAE,kBAAkB,EAAE,CAAC;IACzC,CAAC,CAAC;CACH,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,IAAI,EAAE;IACvB,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC,YAAY,CAAsB,IAAI,CAAC,CAAC;IAChE,OAAO;QACL,GAAG,MAAM;QACT,OAAO,EAAE;YACP,GAAG,MAAM;YACT,iBAAiB,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;SACnF;KACF,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,OAAO,EAAE,gBAAgB,EAAsB,MAAM,cAAc,CAAC;AACpE,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AAEpD,OAAO,cAAc,MAAM,sBAAsB,CAAC;AAElD,MAAM,cAAc,GAAG,EAAE,CAAC,EAAE,EAAgF,CAAC;AAE7G,MAAM,WAAW,GAAiB;IAChC,GAAG,EAAE,cAAc;CACpB,CAAC;AAEF,MAAM,aAAc,SAAQ,YAAY;IACtC,UAAU,GAAW,SAAS,CAAC,IAAI,CAAC;IACpC,IAAI,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IACf,KAAK,GAAG,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE;QACjB,IAAI,IAAI,CAAC,UAAU,KAAK,SAAS,CAAC,MAAM,EAAE,CAAC;YACzC,IAAI,CAAC,UAAU,GAAG,SAAS,CAAC,MAAM,CAAC;YACnC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACrB,CAAC;IACH,CAAC,CAAC,CAAC;CACJ;AAED,SAAS,aAAa;IACpB,MAAM,EAAE,GAAG,IAAI,aAAa,EAAE,CAAC;IAC/B,MAAM,aAAa,GAAG,IAAI,aAAa,EAAE,CAAC;IAC1C,MAAM,MAAM,GAAkB;QAC5B,OAAO,EAAE,WAAW;QACpB,aAAa;KACd,CAAC;IACF,MAAM,OAAO,GAAG,IAAI,gBAAgB,CAAC,EAAS,EAAE,MAAM,CAAC,CAAC;IACxD,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,MAAM,EAAE,aAAa,EAAE,CAAC;AAChD,CAAC;AAED,SAAS,WAAW,CAAC,EAAiB,EAAE,OAAgB;IACtD,EAAE,CAAC,IAAI,CAAC,SAAS,EAAE,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;AAC3D,CAAC;AAED,MAAM,cAAc,GAAG;IACrB,YAAY,EAAE,oBAA6B;IAC3C,SAAS,EAAE,gBAAgB;IAC3B,aAAa,EAAE,kBAAkB;IACjC,WAAW,EAAE,qBAAqB;IAClC,iBAAiB,EAAE,CAAC,WAAW,CAAC;IAChC,6BAA6B,EAAE,IAAI,CAAC,SAAS,CAAC;QAC5C,IAAI,EAAE,aAAa;QACnB,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;KAC3B,CAAC;IACF,eAAe,EAAE,YAAY;CAC9B,CAAC;AAEF,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QAEnB,cAAc,CAAC,iBAAiB,CAAC;YAC/B,gBAAgB,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE;YACnC,QAAQ,EAAE,EAAE,WAAW,EAAE,IAAI,EAAE;SAChC,CAAC,CAAC;QAEH,mBAAmB,CAAC,oBAAoB,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QACtE,mBAAmB,CAAC,sBAAsB,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QACxE,mBAAmB,CAAC,YAAY,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAC9D,mBAAmB,CAAC,QAAQ,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAC1D,kBAAkB,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;QACzB,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;YACjD,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,GAAG,aAAa,EAAE,CAAC;YACxC,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;YAC3B,WAAW,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC;YAChC,MAAM,IAAI,CAAC;QACb,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;YACjE,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,GAAG,aAAa,EAAE,CAAC;YACxC,OAAO,CAAC,GAAG,EAAE,CAAC;YAEd,WAAW,CAAC,EAAE,EAAE,EAAE,YAAY,EAAE,gBAAgB,EAAE,CAAC,CAAC;YACpD,MAAM,IAAI,EAAE,CAAC;YAEb,eAAe,CAAC,EAAE,EAAE,0CAA0C,CAAC,CAAC;QAClE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACrD,IAAI,YAAyB,CAAC;YAC9B,cAAc,CAAC,kBAAkB,CAC/B,GAAG,EAAE,CACH,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE;gBAChB,YAAY,GAAG,GAAG,EAAE,CAClB,CAAC,CAAC,EAAE,gBAAgB,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC;YAC/C,CAAC,CAAC,CACL,CAAC;YAEF,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,GAAG,aAAa,EAAE,CAAC;YACxC,OAAO,CAAC,GAAG,EAAE,CAAC;YAEd,WAAW,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC;YAChC,MAAM,IAAI,EAAE,CAAC;YACb,WAAW,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC;YAChC,MAAM,IAAI,EAAE,CAAC;YAEb,eAAe,CAAC,EAAE,EAAE,8BAA8B,CAAC,CAAC;YAEpD,YAAY,EAAE,CAAC;QACjB,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;YACtE,IAAI,YAAyB,CAAC;YAC9B,cAAc,CAAC,kBAAkB,CAC/B,GAAG,EAAE,CACH,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE;gBAChB,YAAY,GAAG,GAAG,EAAE,CAClB,CAAC,CAAC,EAAE,gBAAgB,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC;YAC/C,CAAC,CAAC,CACL,CAAC;YAEF,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,GAAG,aAAa,EAAE,CAAC;YACxC,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;YAE3B,WAAW,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC;YAChC,MAAM,IAAI,EAAE,CAAC;YAEb,MAAM,cAAc,GAAG;gBACrB,YAAY,EAAE,kCAAkC;gBAChD,QAAQ,EAAE,UAAU;gBACpB,QAAQ,EAAE,qBAAqB;gBAC/B,aAAa,EAAE,+BAA+B;aAC/C,CAAC;YACF,WAAW,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC;YAChC,MAAM,IAAI,EAAE,CAAC;YAEb,MAAM,CAAC,mBAAmB,CAAC,oBAAoB,CAAC,CAAC,oBAAoB,CACnE,cAAc,CACf,CAAC;YAEF,YAAY,EAAE,CAAC;YACf,MAAM,IAAI,CAAC;QACb,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;YACnD,IAAI,YAAyB,CAAC;YAC9B,cAAc,CAAC,kBAAkB,CAC/B,GAAG,EAAE,CACH,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE;gBAChB,YAAY,GAAG,GAAG,EAAE,CAAC,CAAC,CAAC,EAAE,gBAAgB,EAAE,EAAE,EAAE,CAAC,CAAC;YACnD,CAAC,CAAC,CACL,CAAC;YAEF,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,GAAG,aAAa,EAAE,CAAC;YACxC,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;YAE3B,WAAW,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC;YAChC,MAAM,IAAI,EAAE,CAAC;YAEb,WAAW,CAAC,EAAE,EAAE,EAAE,YAAY,EAAE,gBAAgB,EAAE,CAAC,CAAC;YACpD,MAAM,IAAI,EAAE,CAAC;YAEb,MAAM,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,gBAAgB,EAAE,CAAC;YAEpC,YAAY,EAAE,CAAC;QACjB,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACxD,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,GAAG,aAAa,EAAE,CAAC;YACxC,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;YAC3B,WAAW,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC;YAChC,MAAM,IAAI,CAAC;YAEX,MAAM,CAAC,cAAc,CAAC,CAAC,oBAAoB,CACzC,MAAM,CAAC,gBAAgB,CAAC;gBACtB,YAAY,EAAE,kBAAkB;gBAChC,UAAU,EAAE,qBAAqB;gBACjC,YAAY,EAAE,EAAE,IAAI,EAAE,aAAa,EAAE,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;gBACjE,KAAK,EAAE,YAAY;gBACnB,OAAO,EAAE,CAAC,WAAW,CAAC;aACvB,CAAC,EACF,MAAM,CAAC,gBAAgB,CAAC;gBACtB,oBAAoB,EAAE,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC;gBAC1C,gBAAgB,EAAE,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC;aACvC,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACxD,cAAc,CAAC,iBAAiB,CAAC;gBAC/B,gBAAgB,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,OAAO,CAAC,EAAE;gBACrD,QAAQ,EAAE,EAAE,WAAW,EAAE,IAAI,EAAE;aAChC,CAAC,CAAC;YAEH,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,GAAG,aAAa,EAAE,CAAC;YACxC,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;YAC3B,WAAW,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC;YAChC,MAAM,IAAI,CAAC;YAEX,MAAM,IAAI,GAAG,iBAAiB,CAAC,EAAE,CAAC,CAAC;YACnC,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,KAAK,gBAAgB,CAAC,CAAC;YACrE,MAAM,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC;YAC7B,MAAM,CAAC,MAAO,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACrD,MAAM,CAAC,MAAO,CAAC,QAAQ,CAAC,CAAC,aAAa,CAAC,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACrD,cAAc,CAAC,qBAAqB,CAClC,IAAI,KAAK,CAAC,eAAe,CAAC,CAC3B,CAAC;YAEF,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,GAAG,aAAa,EAAE,CAAC;YACxC,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;YAC3B,WAAW,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC;YAChC,MAAM,IAAI,CAAC;YAEX,eAAe,CAAC,EAAE,EAAE,eAAe,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,EAAE,CAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;YAC1C,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,GAAG,aAAa,EAAE,CAAC;YACxC,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;YAC3B,WAAW,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC;YAChC,MAAM,IAAI,CAAC;YAEX,MAAM,CAAC,mBAAmB,CAAC,YAAY,CAAC,CAAC,gBAAgB,EAAE,CAAC;QAC9D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;YACnD,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,GAAG,aAAa,EAAE,CAAC;YACxC,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;YAC3B,WAAW,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC;YAChC,MAAM,IAAI,CAAC;YAEX,MAAM,CAAC,mBAAmB,CAAC,sBAAsB,CAAC,CAAC,oBAAoB,CACrE,MAAM,CACP,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,GAAG,aAAa,EAAE,CAAC;YACxC,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;YAC3B,WAAW,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC;YAChC,MAAM,IAAI,CAAC;YAEX,MAAM,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACtC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,mCAAmC,EAAE,GAAG,EAAE;QACjD,EAAE,CAAC,sEAAsE,EAAE,KAAK,IAAI,EAAE;YACpF,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,GAAG,aAAa,EAAE,CAAC;YACxC,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;YAC3B,WAAW,CAAC,EAAE,EAAE;gBACd,GAAG,cAAc;gBACjB,iCAAiC,EAAE;oBACjC;wBACE,YAAY,EAAE,8BAA8B;wBAC5C,WAAW,EAAE,WAAW;qBACzB;oBACD;wBACE,YAAY,EAAE,iCAAiC;wBAC/C,WAAW,EAAE,WAAW;qBACzB;iBACF;aACF,CAAC,CAAC;YACH,MAAM,IAAI,CAAC;YAEX,MAAM,CAAC,mBAAmB,CAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC9D,MAAM,CAAC,mBAAmB,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CACvD,8BAA8B,EAC9B,WAAW,CACZ,CAAC;YACF,MAAM,CAAC,mBAAmB,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CACvD,iCAAiC,EACjC,WAAW,CACZ,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;YACpE,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,GAAG,aAAa,EAAE,CAAC;YACxC,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;YAC3B,WAAW,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC;YAChC,MAAM,IAAI,CAAC;YAEX,MAAM,CAAC,mBAAmB,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC9D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,+BAA+B,EAAE,GAAG,EAAE;QAC7C,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;YACpE,kBAAkB,CAAC,qBAAqB,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAC;YAE1D,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,GAAG,aAAa,EAAE,CAAC;YACxC,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;YAC3B,WAAW,CAAC,EAAE,EAAE;gBACd,GAAG,cAAc;gBACjB,6BAA6B,EAAE;oBAC7B,EAAE,OAAO,EAAE,UAAU,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE;iBAC1D;aACF,CAAC,CAAC;YACH,MAAM,IAAI,CAAC;YAEX,MAAM,CAAC,cAAc,CAAC,CAAC,oBAAoB,CACzC,EAAE,OAAO,EAAE,UAAU,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EACzD,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,EAClB,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,EAClB,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,EAClB,MAAM,CAAC,GAAG,CAAC,eAAe,CAAC,EAC3B,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,EACpB,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CACrB,CAAC;YAEF,MAAM,IAAI,GAAG,iBAAiB,CAAC,EAAE,CAAC,CAAC;YACnC,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,KAAK,gBAAgB,CAAC,CAAC;YACrE,MAAM,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC;YAC7B,MAAM,CAAC,MAAO,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;YACzE,kBAAkB,CAAC,qBAAqB,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAC;YAE1D,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,GAAG,aAAa,EAAE,CAAC;YACxC,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;YAC3B,WAAW,CAAC,EAAE,EAAE;gBACd,GAAG,cAAc;gBACjB,6BAA6B,EAAE,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,GAAG,EAAE,MAAM,EAAE,CAAC;aACtE,CAAC,CAAC;YACH,MAAM,IAAI,CAAC;YAEX,MAAM,CAAC,cAAc,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sEAAsE,EAAE,KAAK,IAAI,EAAE;YACpF,kBAAkB,CAAC,qBAAqB,CAAC;gBACvC,QAAQ,EAAE,CAAC;gBACX,KAAK,EAAE,IAAI,KAAK,CAAC,aAAa,CAAC;aAChC,CAAC,CAAC;YAEH,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,GAAG,aAAa,EAAE,CAAC;YACxC,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;YAC3B,WAAW,CAAC,EAAE,EAAE;gBACd,GAAG,cAAc;gBACjB,6BAA6B,EAAE,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,GAAG,EAAE,MAAM,EAAE,CAAC;aACtE,CAAC,CAAC;YACH,MAAM,IAAI,CAAC;YAEX,MAAM,IAAI,GAAG,iBAAiB,CAAC,EAAE,CAAC,CAAC;YACnC,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,KAAK,gBAAgB,CAAC,CAAC;YACrE,MAAM,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC;YAC7B,MAAM,CAAC,MAAO,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACxD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;YAC7E,kBAAkB,CAAC,qBAAqB,CAAC,IAAI,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC;YAEpE,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,GAAG,aAAa,EAAE,CAAC;YACxC,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;YAC3B,WAAW,CAAC,EAAE,EAAE;gBACd,GAAG,cAAc;gBACjB,6BAA6B,EAAE,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC;aACxD,CAAC,CAAC;YACH,MAAM,IAAI,CAAC;YAEX,MAAM,IAAI,GAAG,iBAAiB,CAAC,EAAE,CAAC,CAAC;YACnC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,KAAK,eAAe,CAAC,CAAC;YACtE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAEzC,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,KAAK,gBAAgB,CAAC,CAAC;YACrE,MAAM,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC;YAC7B,MAAM,CAAC,MAAO,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACxD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;YAC3E,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,GAAG,aAAa,EAAE,CAAC;YACxC,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;YAC3B,WAAW,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC;YAChC,MAAM,IAAI,CAAC;YAEX,MAAM,CAAC,cAAc,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;YAC9C,MAAM,CAAC,cAAc,CAAC,CAAC,gBAAgB,EAAE,CAAC;QAC5C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,KAAK,UAAU,IAAI;IACjB,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;AAC9C,CAAC;AAED,SAAS,iBAAiB,CAAC,EAAiB;IAC1C,OAAO,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAY,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAW,CAAC,CAAC,CAAC;AAC9E,CAAC;AAED,SAAS,eAAe,CAAC,EAAiB,EAAE,SAAiB;IAC3D,MAAM,IAAI,GAAG,iBAAiB,CAAC,EAAE,CAAC,CAAC;IACnC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,KAAK,eAAe,CAAC,CAAC;IACtE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;IACzC,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC;IAC9D,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC3B,CAAC"}

package/dist/storage.d.ts

@@ -0,0 +1,3 @@
+import { type Logger } from "./logger.js";
+export declare function uploadFile(filePath: string, presignedUrl: string, contentType: string, logger?: Logger): Promise<boolean>;
+//# sourceMappingURL=storage.d.ts.map

package/dist/storage.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"storage.d.ts","sourceRoot":"","sources":["../src/storage.ts"],"names":[],"mappings":"AACA,OAAO,EAAiB,KAAK,MAAM,EAAE,MAAM,aAAa,CAAC;AAEzD,wBAAsB,UAAU,CAC9B,QAAQ,EAAE,MAAM,EAChB,YAAY,EAAE,MAAM,EACpB,WAAW,EAAE,MAAM,EACnB,MAAM,GAAE,MAAsB,GAC7B,OAAO,CAAC,OAAO,CAAC,CAmBlB"}

package/dist/storage.js

@@ -0,0 +1,21 @@
+import fs from "fs";
+import { defaultLogger } from "./logger.js";
+export async function uploadFile(filePath, presignedUrl, contentType, logger = defaultLogger) {
+    try {
+        const content = fs.readFileSync(filePath);
+        const response = await fetch(presignedUrl, {
+            method: "PUT",
+            headers: {
+                "Content-Type": contentType,
+            },
+            body: content,
+        });
+        logger.log(`File ${filePath} uploaded with status: ${response.status}`);
+        return response.status === 200;
+    }
+    catch (error) {
+        logger.error("Upload error:", error);
+        return false;
+    }
+}
+//# sourceMappingURL=storage.js.map

package/dist/storage.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"storage.js","sourceRoot":"","sources":["../src/storage.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,EAAE,aAAa,EAAe,MAAM,aAAa,CAAC;AAEzD,MAAM,CAAC,KAAK,UAAU,UAAU,CAC9B,QAAgB,EAChB,YAAoB,EACpB,WAAmB,EACnB,SAAiB,aAAa;IAE9B,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;QAE1C,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,YAAY,EAAE;YACzC,MAAM,EAAE,KAAK;YACb,OAAO,EAAE;gBACP,cAAc,EAAE,WAAW;aAC5B;YACD,IAAI,EAAE,OAAO;SACd,CAAC,CAAC;QAEH,MAAM,CAAC,GAAG,CAAC,QAAQ,QAAQ,0BAA0B,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;QAExE,OAAO,QAAQ,CAAC,MAAM,KAAK,GAAG,CAAC;IACjC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,CAAC,KAAK,CAAC,eAAe,EAAE,KAAK,CAAC,CAAC;QACrC,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC"}

package/dist/types.d.ts

@@ -0,0 +1,62 @@
+interface Command {
+    command: string;
+    cwd?: string;
+    env?: Record<string, string>;
+}
+interface RuntimeEnvironmentDownloadable {
+    download_url: string;
+    working_dir: string;
+}
+interface InvocationMessage {
+    message_type: "invocation_message";
+    source_id: string;
+    system_prompt: string;
+    user_prompt: string;
+    secrets_to_redact: string[];
+    agent_env?: Record<string, string>;
+    output_format_json_schema_str: string;
+    preferred_model?: string;
+    artifacts_dir?: string;
+    pre_agent_invocation_commands?: Command[];
+    post_agent_invocation_commands?: Command[];
+    runtime_environment_downloadables?: RuntimeEnvironmentDownloadable[];
+}
+interface CancelMessage {
+    message_type: "cancel_message";
+}
+interface ResultMessage {
+    message_type: "result_message";
+    metadata?: Record<string, unknown>;
+    structured_output: Record<string, unknown>;
+    [key: string]: unknown;
+}
+interface AssistantMessage {
+    message_type: "assistant_message";
+    text_blocks: string[];
+}
+interface ArtifactUploadRequestMessage {
+    message_type: "artifact_upload_request_message";
+    artifact_type: "screenshot" | "video" | "tool_calls" | "javascript" | "python" | "shellscript" | "other";
+    filename: string;
+    filepath: string;
+}
+interface ArtifactUploadResponseMessage {
+    message_type: "artifact_upload_response_message";
+    filename: string;
+    filepath: string;
+    presigned_url: string;
+    content_type: string;
+}
+interface ErrorMessage {
+    message_type: "error_message";
+    error: string;
+    metadata: Record<string, unknown>;
+}
+interface StreamEndMessage {
+    message_type: "stream_end";
+    content: string;
+}
+type OutgoingMessage = ResultMessage | AssistantMessage | ArtifactUploadRequestMessage | ErrorMessage | StreamEndMessage;
+type IncomingMessage = InvocationMessage | ArtifactUploadResponseMessage | CancelMessage;
+export type { IncomingMessage, OutgoingMessage, InvocationMessage, CancelMessage, ResultMessage, AssistantMessage, ArtifactUploadRequestMessage, ArtifactUploadResponseMessage, ErrorMessage, StreamEndMessage, Command, RuntimeEnvironmentDownloadable, };
+//# sourceMappingURL=types.d.ts.map

package/dist/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,UAAU,OAAO;IACf,OAAO,EAAE,MAAM,CAAC;IAChB,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,GAAG,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;CAC9B;AAED,UAAU,8BAA8B;IACtC,YAAY,EAAE,MAAM,CAAC;IACrB,WAAW,EAAE,MAAM,CAAC;CACrB;AAED,UAAU,iBAAiB;IACzB,YAAY,EAAE,oBAAoB,CAAC;IACnC,SAAS,EAAE,MAAM,CAAC;IAClB,aAAa,EAAE,MAAM,CAAC;IACtB,WAAW,EAAE,MAAM,CAAC;IACpB,iBAAiB,EAAE,MAAM,EAAE,CAAC;IAC5B,SAAS,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACnC,6BAA6B,EAAE,MAAM,CAAC;IACtC,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,6BAA6B,CAAC,EAAE,OAAO,EAAE,CAAC;IAC1C,8BAA8B,CAAC,EAAE,OAAO,EAAE,CAAC;IAC3C,iCAAiC,CAAC,EAAE,8BAA8B,EAAE,CAAC;CACtE;AAED,UAAU,aAAa;IACrB,YAAY,EAAE,gBAAgB,CAAC;CAChC;AAED,UAAU,aAAa;IACrB,YAAY,EAAE,gBAAgB,CAAC;IAC/B,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACnC,iBAAiB,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAC3C,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;CACxB;AAED,UAAU,gBAAgB;IACxB,YAAY,EAAE,mBAAmB,CAAC;IAClC,WAAW,EAAE,MAAM,EAAE,CAAC;CACvB;AAED,UAAU,4BAA4B;IACpC,YAAY,EAAE,iCAAiC,CAAC;IAChD,aAAa,EACT,YAAY,GACZ,OAAO,GACP,YAAY,GACZ,YAAY,GACZ,QAAQ,GACR,aAAa,GACb,OAAO,CAAC;IACZ,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;CAClB;AAED,UAAU,6BAA6B;IACrC,YAAY,EAAE,kCAAkC,CAAC;IACjD,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,aAAa,EAAE,MAAM,CAAC;IACtB,YAAY,EAAE,MAAM,CAAC;CACtB;AAED,UAAU,YAAY;IACpB,YAAY,EAAE,eAAe,CAAC;IAC9B,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACnC;AAED,UAAU,gBAAgB;IACxB,YAAY,EAAE,YAAY,CAAC;IAC3B,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,KAAK,eAAe,GAChB,aAAa,GACb,gBAAgB,GAChB,4BAA4B,GAC5B,YAAY,GACZ,gBAAgB,CAAC;AAErB,KAAK,eAAe,GAChB,iBAAiB,GACjB,6BAA6B,GAC7B,aAAa,CAAC;AAElB,YAAY,EACV,eAAe,EACf,eAAe,EACf,iBAAiB,EACjB,aAAa,EACb,aAAa,EACb,gBAAgB,EAChB,4BAA4B,EAC5B,6BAA6B,EAC7B,YAAY,EACZ,gBAAgB,EAChB,OAAO,EACP,8BAA8B,GAC/B,CAAC"}

package/dist/types.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=types.js.map

package/dist/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":""}

package/dist/upload-tracker.d.ts

@@ -0,0 +1,10 @@
+import { type Logger } from "./logger.js";
+export declare class UploadTracker {
+    private readonly pending;
+    private logger;
+    setLogger(logger: Logger): void;
+    track(promise: Promise<any>): void;
+    waitForAll(timeoutMs: number): Promise<void>;
+    get size(): number;
+}
+//# sourceMappingURL=upload-tracker.d.ts.map

package/dist/upload-tracker.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"upload-tracker.d.ts","sourceRoot":"","sources":["../src/upload-tracker.ts"],"names":[],"mappings":"AAAA,OAAO,EAAiB,KAAK,MAAM,EAAE,MAAM,aAAa,CAAC;AAEzD,qBAAa,aAAa;IACxB,OAAO,CAAC,QAAQ,CAAC,OAAO,CAA2B;IACnD,OAAO,CAAC,MAAM,CAAyB;IAEvC,SAAS,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI;IAI/B,KAAK,CAAC,OAAO,EAAE,OAAO,CAAC,GAAG,CAAC,GAAG,IAAI;IAO5B,UAAU,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IASlD,IAAI,IAAI,IAAI,MAAM,CAEjB;CACF"}

package/dist/upload-tracker.js

@@ -0,0 +1,27 @@
+import { defaultLogger } from "./logger.js";
+export class UploadTracker {
+    pending = new Set();
+    logger = defaultLogger;
+    setLogger(logger) {
+        this.logger = logger;
+    }
+    track(promise) {
+        this.pending.add(promise);
+        promise
+            .catch((error) => this.logger.error("Error uploading artifact:", error))
+            .finally(() => this.pending.delete(promise));
+    }
+    async waitForAll(timeoutMs) {
+        if (this.pending.size === 0)
+            return;
+        this.logger.log(`Waiting for ${this.pending.size} uploads...`);
+        await Promise.race([
+            Promise.allSettled(this.pending),
+            new Promise((r) => setTimeout(r, timeoutMs)),
+        ]);
+    }
+    get size() {
+        return this.pending.size;
+    }
+}
+//# sourceMappingURL=upload-tracker.js.map

package/dist/upload-tracker.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"upload-tracker.js","sourceRoot":"","sources":["../src/upload-tracker.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAe,MAAM,aAAa,CAAC;AAEzD,MAAM,OAAO,aAAa;IACP,OAAO,GAAG,IAAI,GAAG,EAAgB,CAAC;IAC3C,MAAM,GAAW,aAAa,CAAC;IAEvC,SAAS,CAAC,MAAc;QACtB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,OAAqB;QACzB,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAC1B,OAAO;aACJ,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,2BAA2B,EAAE,KAAK,CAAC,CAAC;aACvE,OAAO,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC;IACjD,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,SAAiB;QAChC,IAAI,IAAI,CAAC,OAAO,CAAC,IAAI,KAAK,CAAC;YAAE,OAAO;QACpC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,eAAe,IAAI,CAAC,OAAO,CAAC,IAAI,aAAa,CAAC,CAAC;QAC/D,MAAM,OAAO,CAAC,IAAI,CAAC;YACjB,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC;YAChC,IAAI,OAAO,CAAO,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;SACnD,CAAC,CAAC;IACL,CAAC;IAED,IAAI,IAAI;QACN,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC;IAC3B,CAAC;CACF"}

package/dist/upload-tracker.test.d.ts

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

package/dist/upload-tracker.test.d.ts.map

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

package/dist/upload-tracker.test.js

@@ -0,0 +1,89 @@
+import { describe, it, expect, vi, beforeEach } from "vitest";
+import { UploadTracker } from "./upload-tracker.js";
+describe("UploadTracker", () => {
+    let tracker;
+    beforeEach(() => {
+        tracker = new UploadTracker();
+    });
+    it("starts with size 0", () => {
+        expect(tracker.size).toBe(0);
+    });
+    it("tracks a pending upload", () => {
+        tracker.track(new Promise(() => { }));
+        expect(tracker.size).toBe(1);
+    });
+    it("tracks multiple pending uploads", () => {
+        tracker.track(new Promise(() => { }));
+        tracker.track(new Promise(() => { }));
+        tracker.track(new Promise(() => { }));
+        expect(tracker.size).toBe(3);
+    });
+    it("removes upload after it resolves", async () => {
+        let resolve;
+        const promise = new Promise((r) => {
+            resolve = r;
+        });
+        tracker.track(promise);
+        expect(tracker.size).toBe(1);
+        resolve();
+        await promise;
+        await flushMicrotasks();
+        expect(tracker.size).toBe(0);
+    });
+    it("removes upload after it rejects", async () => {
+        let reject;
+        const promise = new Promise((_, r) => {
+            reject = r;
+        });
+        vi.spyOn(console, "error").mockImplementation(() => { });
+        tracker.track(promise);
+        expect(tracker.size).toBe(1);
+        reject(new Error("fail"));
+        await flushMicrotasks();
+        expect(tracker.size).toBe(0);
+    });
+    it("logs error when a tracked upload rejects", async () => {
+        const consoleSpy = vi
+            .spyOn(console, "error")
+            .mockImplementation(() => { });
+        let reject;
+        const promise = new Promise((_, r) => {
+            reject = r;
+        });
+        tracker.track(promise);
+        const error = new Error("upload failed");
+        reject(error);
+        await flushMicrotasks();
+        expect(consoleSpy).toHaveBeenCalledWith("Error uploading artifact:", error);
+    });
+    describe("waitForAll", () => {
+        it("resolves immediately when no pending uploads", async () => {
+            await tracker.waitForAll(1000);
+        });
+        it("resolves when all uploads complete", async () => {
+            let resolve1;
+            let resolve2;
+            tracker.track(new Promise((r) => { resolve1 = r; }));
+            tracker.track(new Promise((r) => { resolve2 = r; }));
+            let waited = false;
+            const waitPromise = tracker.waitForAll(5000).then(() => {
+                waited = true;
+            });
+            resolve1();
+            resolve2();
+            await waitPromise;
+            expect(waited).toBe(true);
+        });
+        it("times out if uploads do not complete", async () => {
+            tracker.track(new Promise(() => { }));
+            const start = Date.now();
+            await tracker.waitForAll(50);
+            expect(Date.now() - start).toBeGreaterThanOrEqual(40);
+            expect(tracker.size).toBe(1);
+        });
+    });
+});
+async function flushMicrotasks() {
+    await new Promise((r) => setTimeout(r, 0));
+}
+//# sourceMappingURL=upload-tracker.test.js.map

package/dist/upload-tracker.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"upload-tracker.test.js","sourceRoot":"","sources":["../src/upload-tracker.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AAEpD,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,IAAI,OAAsB,CAAC;IAE3B,UAAU,CAAC,GAAG,EAAE;QACd,OAAO,GAAG,IAAI,aAAa,EAAE,CAAC;IAChC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAC5B,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAC/B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACjC,OAAO,CAAC,KAAK,CAAC,IAAI,OAAO,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC,CAAC;QACrC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAC/B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;QACzC,OAAO,CAAC,KAAK,CAAC,IAAI,OAAO,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC,CAAC;QACrC,OAAO,CAAC,KAAK,CAAC,IAAI,OAAO,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC,CAAC;QACrC,OAAO,CAAC,KAAK,CAAC,IAAI,OAAO,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC,CAAC;QACrC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAC/B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;QAChD,IAAI,OAAoB,CAAC;QACzB,MAAM,OAAO,GAAG,IAAI,OAAO,CAAO,CAAC,CAAC,EAAE,EAAE;YACtC,OAAO,GAAG,CAAC,CAAC;QACd,CAAC,CAAC,CAAC;QACH,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QACvB,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAE7B,OAAO,EAAE,CAAC;QACV,MAAM,OAAO,CAAC;QACd,MAAM,eAAe,EAAE,CAAC;QACxB,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAC/B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;QAC/C,IAAI,MAA2B,CAAC;QAChC,MAAM,OAAO,GAAG,IAAI,OAAO,CAAO,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;YACzC,MAAM,GAAG,CAAC,CAAC;QACb,CAAC,CAAC,CAAC;QACH,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QAExD,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QACvB,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAE7B,MAAM,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;QAC1B,MAAM,eAAe,EAAE,CAAC;QACxB,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAC/B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;QACxD,MAAM,UAAU,GAAG,EAAE;aAClB,KAAK,CAAC,OAAO,EAAE,OAAO,CAAC;aACvB,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QAEhC,IAAI,MAA2B,CAAC;QAChC,MAAM,OAAO,GAAG,IAAI,OAAO,CAAO,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;YACzC,MAAM,GAAG,CAAC,CAAC;QACb,CAAC,CAAC,CAAC;QACH,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAEvB,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,eAAe,CAAC,CAAC;QACzC,MAAM,CAAC,KAAK,CAAC,CAAC;QACd,MAAM,eAAe,EAAE,CAAC;QAExB,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CACrC,2BAA2B,EAC3B,KAAK,CACN,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;YAC5D,MAAM,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;QACjC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,IAAI,QAAqB,CAAC;YAC1B,IAAI,QAAqB,CAAC;YAC1B,OAAO,CAAC,KAAK,CAAC,IAAI,OAAO,CAAO,CAAC,CAAC,EAAE,EAAE,GAAG,QAAQ,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAC3D,OAAO,CAAC,KAAK,CAAC,IAAI,OAAO,CAAO,CAAC,CAAC,EAAE,EAAE,GAAG,QAAQ,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAE3D,IAAI,MAAM,GAAG,KAAK,CAAC;YACnB,MAAM,WAAW,GAAG,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE;gBACrD,MAAM,GAAG,IAAI,CAAC;YAChB,CAAC,CAAC,CAAC;YAEH,QAAQ,EAAE,CAAC;YACX,QAAQ,EAAE,CAAC;YACX,MAAM,WAAW,CAAC;YAClB,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC5B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YACpD,OAAO,CAAC,KAAK,CAAC,IAAI,OAAO,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC,CAAC;YAErC,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YACzB,MAAM,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;YAC7B,MAAM,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC,CAAC,sBAAsB,CAAC,EAAE,CAAC,CAAC;YACtD,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,KAAK,UAAU,eAAe;IAC5B,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;AAC7C,CAAC"}

package/dist/utils.d.ts

@@ -0,0 +1,7 @@
+export declare function sleep(ms: number): Promise<unknown>;
+/**
+ * Recursively redact secret values from an arbitrary data structure.
+ * Any string that contains a secret will have that secret replaced with [REDACTED].
+ */
+export declare function redactSecrets<T>(value: T, secrets: string[]): T;
+//# sourceMappingURL=utils.d.ts.map

package/dist/utils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../src/utils.ts"],"names":[],"mappings":"AAAA,wBAAsB,KAAK,CAAC,EAAE,EAAE,MAAM,oBAErC;AAED;;;GAGG;AACH,wBAAgB,aAAa,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,GAAG,CAAC,CA0B/D"}