@trigger.dev/core 3.3.16 → 4.0.0-v4-beta.0

package/dist/esm/v3/serverOnly/shutdownManager.js

@@ -0,0 +1,88 @@
+import { isTest } from "std-env";
+import { SimpleStructuredLogger } from "../utils/structuredLogger.js";
+import { singleton } from "./singleton.js";
+export class ShutdownManager {
+    disableForTesting;
+    isShuttingDown = false;
+    signalNumbers = {
+        SIGINT: 2,
+        SIGTERM: 15,
+    };
+    logger = new SimpleStructuredLogger("shutdownManager");
+    handlers = new Map();
+    constructor(disableForTesting = true) {
+        this.disableForTesting = disableForTesting;
+        if (disableForTesting)
+            return;
+        process.on("SIGTERM", () => this.shutdown("SIGTERM"));
+        process.on("SIGINT", () => this.shutdown("SIGINT"));
+    }
+    register(name, handler, signals = ["SIGTERM", "SIGINT"]) {
+        if (!this.isEnabled())
+            return;
+        if (this.handlers.has(name)) {
+            throw new Error(`Shutdown handler "${name}" already registered`);
+        }
+        this.handlers.set(name, { handler, signals });
+    }
+    unregister(name) {
+        if (!this.isEnabled())
+            return;
+        if (!this.handlers.has(name)) {
+            throw new Error(`Shutdown handler "${name}" not registered`);
+        }
+        this.handlers.delete(name);
+    }
+    async shutdown(signal) {
+        if (!this.isEnabled())
+            return;
+        if (this.isShuttingDown)
+            return;
+        this.isShuttingDown = true;
+        this.logger.info(`Received ${signal}. Starting graceful shutdown...`);
+        // Get handlers that are registered for this signal
+        const handlersToRun = Array.from(this.handlers.entries()).filter(([_, { signals }]) => signals.includes(signal));
+        try {
+            const results = await Promise.allSettled(handlersToRun.map(async ([name, { handler }]) => {
+                try {
+                    this.logger.info(`Running shutdown handler: ${name}`);
+                    await handler(signal);
+                    this.logger.info(`Shutdown handler completed: ${name}`);
+                }
+                catch (error) {
+                    this.logger.error(`Shutdown handler failed: ${name}`, { error });
+                    throw error;
+                }
+            }));
+            // Log any failures
+            results.forEach((result, index) => {
+                if (result.status === "rejected") {
+                    const handlerEntry = handlersToRun[index];
+                    if (handlerEntry) {
+                        const [name] = handlerEntry;
+                        this.logger.error(`Shutdown handler "${name}" failed:`, { reason: result.reason });
+                    }
+                }
+            });
+        }
+        catch (error) {
+            this.logger.error("Error during shutdown:", { error });
+        }
+        finally {
+            // Exit with the correct signal number
+            process.exit(128 + this.signalNumbers[signal]);
+        }
+    }
+    isEnabled() {
+        if (!this.disableForTesting) {
+            return true;
+        }
+        return !isTest;
+    }
+    // Only for testing
+    _getHandlersForTesting() {
+        return new Map(this.handlers);
+    }
+}
+export const shutdownManager = singleton("shutdownManager", () => new ShutdownManager());
+//# sourceMappingURL=shutdownManager.js.map

package/dist/esm/v3/serverOnly/shutdownManager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"shutdownManager.js","sourceRoot":"","sources":["../../../../src/v3/serverOnly/shutdownManager.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AACjC,OAAO,EAAE,sBAAsB,EAAE,MAAM,8BAA8B,CAAC;AACtE,OAAO,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAM3C,MAAM,OAAO,eAAe;IAWN;IAVZ,cAAc,GAAG,KAAK,CAAC;IACvB,aAAa,GAAmC;QACtD,MAAM,EAAE,CAAC;QACT,OAAO,EAAE,EAAE;KACZ,CAAC;IAEM,MAAM,GAAG,IAAI,sBAAsB,CAAC,iBAAiB,CAAC,CAAC;IACvD,QAAQ,GACd,IAAI,GAAG,EAAE,CAAC;IAEZ,YAAoB,oBAAoB,IAAI;QAAxB,sBAAiB,GAAjB,iBAAiB,CAAO;QAC1C,IAAI,iBAAiB;YAAE,OAAO;QAE9B,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC;QACtD,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC;IACtD,CAAC;IAED,QAAQ,CACN,IAAY,EACZ,OAAwB,EACxB,UAA4B,CAAC,SAAS,EAAE,QAAQ,CAAC;QAEjD,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE;YAAE,OAAO;QAE9B,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;YAC5B,MAAM,IAAI,KAAK,CAAC,qBAAqB,IAAI,sBAAsB,CAAC,CAAC;QACnE,CAAC;QACD,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;IAChD,CAAC;IAED,UAAU,CAAC,IAAY;QACrB,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE;YAAE,OAAO;QAE9B,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,qBAAqB,IAAI,kBAAkB,CAAC,CAAC;QAC/D,CAAC;QAED,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IAC7B,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,MAAsB;QACnC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE;YAAE,OAAO;QAE9B,IAAI,IAAI,CAAC,cAAc;YAAE,OAAO;QAChC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAE3B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,YAAY,MAAM,iCAAiC,CAAC,CAAC;QAEtE,mDAAmD;QACnD,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,OAAO,EAAE,CAAC,EAAE,EAAE,CACpF,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CACzB,CAAC;QAEF,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,UAAU,CACtC,aAAa,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC,IAAI,EAAE,EAAE,OAAO,EAAE,CAAC,EAAE,EAAE;gBAC9C,IAAI,CAAC;oBACH,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,6BAA6B,IAAI,EAAE,CAAC,CAAC;oBACtD,MAAM,OAAO,CAAC,MAAM,CAAC,CAAC;oBACtB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,+BAA+B,IAAI,EAAE,CAAC,CAAC;gBAC1D,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,4BAA4B,IAAI,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;oBACjE,MAAM,KAAK,CAAC;gBACd,CAAC;YACH,CAAC,CAAC,CACH,CAAC;YAEF,mBAAmB;YACnB,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,KAAK,EAAE,EAAE;gBAChC,IAAI,MAAM,CAAC,MAAM,KAAK,UAAU,EAAE,CAAC;oBACjC,MAAM,YAAY,GAAG,aAAa,CAAC,KAAK,CAAC,CAAC;oBAC1C,IAAI,YAAY,EAAE,CAAC;wBACjB,MAAM,CAAC,IAAI,CAAC,GAAG,YAAY,CAAC;wBAC5B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,qBAAqB,IAAI,WAAW,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;oBACrF,CAAC;gBACH,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,wBAAwB,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QACzD,CAAC;gBAAS,CAAC;YACT,sCAAsC;YACtC,OAAO,CAAC,IAAI,CAAC,GAAG,GAAG,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,CAAC;QACjD,CAAC;IACH,CAAC;IAEO,SAAS;QACf,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC5B,OAAO,IAAI,CAAC;QACd,CAAC;QAED,OAAO,CAAC,MAAM,CAAC;IACjB,CAAC;IAED,mBAAmB;IACZ,sBAAsB;QAI3B,OAAO,IAAI,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAChC,CAAC;CACF;AAED,MAAM,CAAC,MAAM,eAAe,GAAG,SAAS,CAAC,iBAAiB,EAAE,GAAG,EAAE,CAAC,IAAI,eAAe,EAAE,CAAC,CAAC"}

package/dist/esm/v3/serverOnly/shutdownManager.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/esm/v3/serverOnly/shutdownManager.test.js

@@ -0,0 +1,137 @@
+import { describe, test, expect, vi, beforeEach } from "vitest";
+import { ShutdownManager } from "./shutdownManager.js";
+describe("ShutdownManager", { concurrent: false }, () => {
+    // Mock process.exit to prevent actual exit
+    const mockExit = vi.spyOn(process, "exit").mockImplementation(() => undefined);
+    beforeEach(() => {
+        vi.clearAllMocks();
+    });
+    test("should successfully register a new handler", () => {
+        const manager = new ShutdownManager(false);
+        const handler = vi.fn();
+        manager.register("test-handler", handler);
+        expect(manager._getHandlersForTesting().has("test-handler")).toBe(true);
+        const registeredHandler = manager._getHandlersForTesting().get("test-handler");
+        expect(registeredHandler?.handler).toBe(handler);
+        expect(registeredHandler?.signals).toEqual(["SIGTERM", "SIGINT"]);
+    });
+    test("should throw error when registering duplicate handler name", () => {
+        const manager = new ShutdownManager(false);
+        const handler = vi.fn();
+        manager.register("duplicate-handler", handler);
+        expect(() => {
+            manager.register("duplicate-handler", handler);
+        }).toThrow('Shutdown handler "duplicate-handler" already registered');
+    });
+    test("should register handler with custom signals", () => {
+        const manager = new ShutdownManager(false);
+        const handler = vi.fn();
+        manager.register("custom-signals", handler, ["SIGTERM"]);
+        const registeredHandler = manager._getHandlersForTesting().get("custom-signals");
+        expect(registeredHandler?.signals).toEqual(["SIGTERM"]);
+    });
+    test("should call registered handlers when shutdown is triggered", async () => {
+        const manager = new ShutdownManager(false);
+        const handler1 = vi.fn();
+        const handler2 = vi.fn();
+        manager.register("handler1", handler1);
+        manager.register("handler2", handler2);
+        await manager.shutdown("SIGTERM");
+        expect(handler1).toHaveBeenCalledWith("SIGTERM");
+        expect(handler2).toHaveBeenCalledWith("SIGTERM");
+        expect(mockExit).toHaveBeenCalledWith(128 + 15); // SIGTERM number
+    });
+    test("should only call handlers registered for specific signal", async () => {
+        const manager = new ShutdownManager(false);
+        const handler1 = vi.fn();
+        const handler2 = vi.fn();
+        manager.register("handler1", handler1, ["SIGTERM"]);
+        manager.register("handler2", handler2, ["SIGINT"]);
+        await manager.shutdown("SIGTERM");
+        expect(handler1).toHaveBeenCalledWith("SIGTERM");
+        expect(handler2).not.toHaveBeenCalled();
+        expect(mockExit).toHaveBeenCalledWith(128 + 15);
+    });
+    test("should handle errors in shutdown handlers gracefully", async () => {
+        const manager = new ShutdownManager(false);
+        const successHandler = vi.fn();
+        const errorHandler = vi.fn().mockRejectedValue(new Error("Handler failed"));
+        manager.register("success-handler", successHandler);
+        manager.register("error-handler", errorHandler);
+        await manager.shutdown("SIGTERM");
+        expect(successHandler).toHaveBeenCalledWith("SIGTERM");
+        expect(errorHandler).toHaveBeenCalledWith("SIGTERM");
+        expect(mockExit).toHaveBeenCalledWith(128 + 15);
+    });
+    test("should only run shutdown sequence once even if called multiple times", async () => {
+        const manager = new ShutdownManager(false);
+        const handler = vi.fn();
+        manager.register("test-handler", handler);
+        await Promise.all([manager.shutdown("SIGTERM"), manager.shutdown("SIGTERM")]);
+        expect(handler).toHaveBeenCalledTimes(1);
+        expect(mockExit).toHaveBeenCalledTimes(1);
+        expect(mockExit).toHaveBeenCalledWith(128 + 15);
+    });
+    test("should exit with correct signal number on SIGINT", async () => {
+        const manager = new ShutdownManager(false);
+        manager.register("test-handler", vi.fn());
+        await manager.shutdown("SIGINT");
+        expect(mockExit).toHaveBeenCalledWith(128 + 2); // SIGINT number
+    });
+    test("should exit with correct signal number on SIGTERM", async () => {
+        const manager = new ShutdownManager(false);
+        manager.register("test-handler", vi.fn());
+        await manager.shutdown("SIGTERM");
+        expect(mockExit).toHaveBeenCalledWith(128 + 15); // SIGTERM number
+    });
+    test("should only exit after all handlers have finished", async () => {
+        const sequence = [];
+        const manager = new ShutdownManager(false);
+        const handler1 = vi.fn().mockImplementation(async () => {
+            sequence.push("handler1 start");
+            await new Promise((resolve) => setTimeout(resolve, 10));
+            sequence.push("handler1 end");
+        });
+        const handler2 = vi.fn().mockImplementation(async () => {
+            sequence.push("handler2 start");
+            await new Promise((resolve) => setTimeout(resolve, 20));
+            sequence.push("handler2 end");
+        });
+        const handler3 = vi.fn().mockImplementation(async () => {
+            sequence.push("handler3 start");
+            await new Promise((resolve) => setTimeout(resolve, 5));
+            sequence.push("handler3 end");
+        });
+        // Store the current mock implementation
+        const currentExit = mockExit.getMockImplementation();
+        // Override with our sequence-tracking implementation
+        mockExit.mockImplementation((code) => {
+            sequence.push("exit");
+            return undefined;
+        });
+        manager.register("handler1", handler1);
+        manager.register("handler2", handler2);
+        manager.register("handler3", handler3);
+        await manager.shutdown("SIGTERM");
+        // Verify the execution order
+        expect(sequence).toEqual([
+            "handler1 start",
+            "handler2 start",
+            "handler3 start",
+            "handler3 end",
+            "handler1 end",
+            "handler2 end",
+            "exit",
+        ]);
+        // Verify the handlers were called with correct arguments
+        expect(handler1).toHaveBeenCalledWith("SIGTERM");
+        expect(handler2).toHaveBeenCalledWith("SIGTERM");
+        expect(handler3).toHaveBeenCalledWith("SIGTERM");
+        expect(mockExit).toHaveBeenCalledWith(128 + 15);
+        // Restore original mock implementation
+        if (currentExit) {
+            mockExit.mockImplementation(currentExit);
+        }
+    });
+});
+//# sourceMappingURL=shutdownManager.test.js.map

package/dist/esm/v3/serverOnly/shutdownManager.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"shutdownManager.test.js","sourceRoot":"","sources":["../../../../src/v3/serverOnly/shutdownManager.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAChE,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AAEvD,QAAQ,CAAC,iBAAiB,EAAE,EAAE,UAAU,EAAE,KAAK,EAAE,EAAE,GAAG,EAAE;IACtD,2CAA2C;IAC3C,MAAM,QAAQ,GAAG,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,SAAkB,CAAC,CAAC;IAExF,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACtD,MAAM,OAAO,GAAG,IAAI,eAAe,CAAC,KAAK,CAAC,CAAC;QAE3C,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACxB,OAAO,CAAC,QAAQ,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;QAE1C,MAAM,CAAC,OAAO,CAAC,sBAAsB,EAAE,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACxE,MAAM,iBAAiB,GAAG,OAAO,CAAC,sBAAsB,EAAE,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;QAC/E,MAAM,CAAC,iBAAiB,EAAE,OAAO,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACjD,MAAM,CAAC,iBAAiB,EAAE,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC,CAAC;IACpE,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,4DAA4D,EAAE,GAAG,EAAE;QACtE,MAAM,OAAO,GAAG,IAAI,eAAe,CAAC,KAAK,CAAC,CAAC;QAE3C,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACxB,OAAO,CAAC,QAAQ,CAAC,mBAAmB,EAAE,OAAO,CAAC,CAAC;QAE/C,MAAM,CAAC,GAAG,EAAE;YACV,OAAO,CAAC,QAAQ,CAAC,mBAAmB,EAAE,OAAO,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC,OAAO,CAAC,yDAAyD,CAAC,CAAC;IACxE,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,6CAA6C,EAAE,GAAG,EAAE;QACvD,MAAM,OAAO,GAAG,IAAI,eAAe,CAAC,KAAK,CAAC,CAAC;QAE3C,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACxB,OAAO,CAAC,QAAQ,CAAC,gBAAgB,EAAE,OAAO,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC;QAEzD,MAAM,iBAAiB,GAAG,OAAO,CAAC,sBAAsB,EAAE,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;QACjF,MAAM,CAAC,iBAAiB,EAAE,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;IAC1D,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;QAC5E,MAAM,OAAO,GAAG,IAAI,eAAe,CAAC,KAAK,CAAC,CAAC;QAE3C,MAAM,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACzB,MAAM,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAEzB,OAAO,CAAC,QAAQ,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QACvC,OAAO,CAAC,QAAQ,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QAEvC,MAAM,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;QAElC,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,SAAS,CAAC,CAAC;QACjD,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,SAAS,CAAC,CAAC;QACjD,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,GAAG,GAAG,EAAE,CAAC,CAAC,CAAC,iBAAiB;IACpE,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;QAC1E,MAAM,OAAO,GAAG,IAAI,eAAe,CAAC,KAAK,CAAC,CAAC;QAE3C,MAAM,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACzB,MAAM,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAEzB,OAAO,CAAC,QAAQ,CAAC,UAAU,EAAE,QAAQ,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC;QACpD,OAAO,CAAC,QAAQ,CAAC,UAAU,EAAE,QAAQ,EAAE,CAAC,QAAQ,CAAC,CAAC,CAAC;QAEnD,MAAM,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;QAElC,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,SAAS,CAAC,CAAC;QACjD,MAAM,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACxC,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,GAAG,GAAG,EAAE,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;QACtE,MAAM,OAAO,GAAG,IAAI,eAAe,CAAC,KAAK,CAAC,CAAC;QAE3C,MAAM,cAAc,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAC/B,MAAM,YAAY,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,gBAAgB,CAAC,CAAC,CAAC;QAE5E,OAAO,CAAC,QAAQ,CAAC,iBAAiB,EAAE,cAAc,CAAC,CAAC;QACpD,OAAO,CAAC,QAAQ,CAAC,eAAe,EAAE,YAAY,CAAC,CAAC;QAEhD,MAAM,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;QAElC,MAAM,CAAC,cAAc,CAAC,CAAC,oBAAoB,CAAC,SAAS,CAAC,CAAC;QACvD,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,SAAS,CAAC,CAAC;QACrD,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,GAAG,GAAG,EAAE,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,sEAAsE,EAAE,KAAK,IAAI,EAAE;QACtF,MAAM,OAAO,GAAG,IAAI,eAAe,CAAC,KAAK,CAAC,CAAC;QAE3C,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACxB,OAAO,CAAC,QAAQ,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;QAE1C,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;QAE9E,MAAM,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACzC,MAAM,CAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC1C,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,GAAG,GAAG,EAAE,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;QAClE,MAAM,OAAO,GAAG,IAAI,eAAe,CAAC,KAAK,CAAC,CAAC;QAE3C,OAAO,CAAC,QAAQ,CAAC,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;QAE1C,MAAM,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QACjC,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,gBAAgB;IAClE,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;QACnE,MAAM,OAAO,GAAG,IAAI,eAAe,CAAC,KAAK,CAAC,CAAC;QAE3C,OAAO,CAAC,QAAQ,CAAC,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;QAE1C,MAAM,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;QAClC,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,GAAG,GAAG,EAAE,CAAC,CAAC,CAAC,iBAAiB;IACpE,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;QACnE,MAAM,QAAQ,GAAa,EAAE,CAAC;QAC9B,MAAM,OAAO,GAAG,IAAI,eAAe,CAAC,KAAK,CAAC,CAAC;QAE3C,MAAM,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,KAAK,IAAI,EAAE;YACrD,QAAQ,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YAChC,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;YACxD,QAAQ,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAChC,CAAC,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,KAAK,IAAI,EAAE;YACrD,QAAQ,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YAChC,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;YACxD,QAAQ,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAChC,CAAC,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,KAAK,IAAI,EAAE;YACrD,QAAQ,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YAChC,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;YACvD,QAAQ,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAChC,CAAC,CAAC,CAAC;QAEH,wCAAwC;QACxC,MAAM,WAAW,GAAG,QAAQ,CAAC,qBAAqB,EAAE,CAAC;QAErD,qDAAqD;QACrD,QAAQ,CAAC,kBAAkB,CAAC,CAAC,IAA6B,EAAE,EAAE;YAC5D,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACtB,OAAO,SAAkB,CAAC;QAC5B,CAAC,CAAC,CAAC;QAEH,OAAO,CAAC,QAAQ,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QACvC,OAAO,CAAC,QAAQ,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QACvC,OAAO,CAAC,QAAQ,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QAEvC,MAAM,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;QAElC,6BAA6B;QAC7B,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;YACvB,gBAAgB;YAChB,gBAAgB;YAChB,gBAAgB;YAChB,cAAc;YACd,cAAc;YACd,cAAc;YACd,MAAM;SACP,CAAC,CAAC;QAEH,yDAAyD;QACzD,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,SAAS,CAAC,CAAC;QACjD,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,SAAS,CAAC,CAAC;QACjD,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,SAAS,CAAC,CAAC;QACjD,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,GAAG,GAAG,EAAE,CAAC,CAAC;QAEhD,uCAAuC;QACvC,IAAI,WAAW,EAAE,CAAC;YAChB,QAAQ,CAAC,kBAAkB,CAAC,WAAW,CAAC,CAAC;QAC3C,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/esm/v3/serverOnly/singleton.d.ts

@@ -0,0 +1 @@
+export declare function singleton<T>(name: string, getValue: () => T): T;

package/dist/esm/v3/serverOnly/singleton.js

@@ -0,0 +1,7 @@
+export function singleton(name, getValue) {
+    const thusly = globalThis;
+    thusly.__trigger_singletons ??= {};
+    thusly.__trigger_singletons[name] ??= getValue();
+    return thusly.__trigger_singletons[name];
+}
+//# sourceMappingURL=singleton.js.map

package/dist/esm/v3/serverOnly/singleton.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"singleton.js","sourceRoot":"","sources":["../../../../src/v3/serverOnly/singleton.ts"],"names":[],"mappings":"AAAA,MAAM,UAAU,SAAS,CAAI,IAAY,EAAE,QAAiB;IAC1D,MAAM,MAAM,GAAG,UAEd,CAAC;IACF,MAAM,CAAC,oBAAoB,KAAK,EAAE,CAAC;IACnC,MAAM,CAAC,oBAAoB,CAAC,IAAI,CAAC,KAAK,QAAQ,EAAE,CAAC;IACjD,OAAO,MAAM,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC;AAC3C,CAAC"}

package/dist/esm/v3/streams/asyncIterableStream.d.ts

@@ -0,0 +1,4 @@
+export type AsyncIterableStream<T> = AsyncIterable<T> & ReadableStream<T>;
+export declare function createAsyncIterableStream<S, T>(source: ReadableStream<S>, transformer: Transformer<S, T>): AsyncIterableStream<T>;
+export declare function createAsyncIterableReadable<S, T>(source: ReadableStream<S>, transformer: Transformer<S, T>, signal: AbortSignal): AsyncIterableStream<T>;
+export declare function createAsyncIterableStreamFromAsyncIterable<T>(asyncIterable: AsyncIterable<T>, transformer: Transformer<T, T>, signal?: AbortSignal): AsyncIterableStream<T>;

package/dist/esm/v3/streams/asyncIterableStream.js

@@ -0,0 +1,72 @@
+export function createAsyncIterableStream(source, transformer) {
+    const transformedStream = source.pipeThrough(new TransformStream(transformer));
+    transformedStream[Symbol.asyncIterator] = () => {
+        const reader = transformedStream.getReader();
+        return {
+            async next() {
+                const { done, value } = await reader.read();
+                return done ? { done: true, value: undefined } : { done: false, value };
+            },
+        };
+    };
+    return transformedStream;
+}
+export function createAsyncIterableReadable(source, transformer, signal) {
+    return new ReadableStream({
+        async start(controller) {
+            const transformedStream = source.pipeThrough(new TransformStream(transformer));
+            const reader = transformedStream.getReader();
+            signal.addEventListener("abort", () => {
+                queueMicrotask(() => {
+                    reader.cancel();
+                    controller.close();
+                });
+            });
+            while (true) {
+                const { done, value } = await reader.read();
+                if (done) {
+                    controller.close();
+                    break;
+                }
+                controller.enqueue(value);
+            }
+        },
+    });
+}
+export function createAsyncIterableStreamFromAsyncIterable(asyncIterable, transformer, signal) {
+    const stream = new ReadableStream({
+        async start(controller) {
+            try {
+                if (signal) {
+                    signal.addEventListener("abort", () => {
+                        controller.close();
+                    });
+                }
+                const iterator = asyncIterable[Symbol.asyncIterator]();
+                while (true) {
+                    if (signal?.aborted) {
+                        break;
+                    }
+                    const { done, value } = await iterator.next();
+                    if (done) {
+                        controller.close();
+                        break;
+                    }
+                    controller.enqueue(value);
+                }
+            }
+            catch (error) {
+                controller.error(error);
+            }
+        },
+        cancel() {
+            // If the stream is a tinyexec process with a kill method, kill it
+            if ("kill" in asyncIterable) {
+                asyncIterable.kill();
+            }
+        },
+    });
+    const transformedStream = stream.pipeThrough(new TransformStream(transformer));
+    return transformedStream;
+}
+//# sourceMappingURL=asyncIterableStream.js.map

package/dist/esm/v3/streams/asyncIterableStream.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"asyncIterableStream.js","sourceRoot":"","sources":["../../../../src/v3/streams/asyncIterableStream.ts"],"names":[],"mappings":"AAEA,MAAM,UAAU,yBAAyB,CACvC,MAAyB,EACzB,WAA8B;IAE9B,MAAM,iBAAiB,GAAQ,MAAM,CAAC,WAAW,CAAC,IAAI,eAAe,CAAC,WAAW,CAAC,CAAC,CAAC;IAEpF,iBAAiB,CAAC,MAAM,CAAC,aAAa,CAAC,GAAG,GAAG,EAAE;QAC7C,MAAM,MAAM,GAAG,iBAAiB,CAAC,SAAS,EAAE,CAAC;QAC7C,OAAO;YACL,KAAK,CAAC,IAAI;gBACR,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;gBAC5C,OAAO,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;YAC1E,CAAC;SACF,CAAC;IACJ,CAAC,CAAC;IAEF,OAAO,iBAAiB,CAAC;AAC3B,CAAC;AAED,MAAM,UAAU,2BAA2B,CACzC,MAAyB,EACzB,WAA8B,EAC9B,MAAmB;IAEnB,OAAO,IAAI,cAAc,CAAI;QAC3B,KAAK,CAAC,KAAK,CAAC,UAAU;YACpB,MAAM,iBAAiB,GAAG,MAAM,CAAC,WAAW,CAAC,IAAI,eAAe,CAAC,WAAW,CAAC,CAAC,CAAC;YAC/E,MAAM,MAAM,GAAG,iBAAiB,CAAC,SAAS,EAAE,CAAC;YAE7C,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,GAAG,EAAE;gBACpC,cAAc,CAAC,GAAG,EAAE;oBAClB,MAAM,CAAC,MAAM,EAAE,CAAC;oBAChB,UAAU,CAAC,KAAK,EAAE,CAAC;gBACrB,CAAC,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;YAEH,OAAO,IAAI,EAAE,CAAC;gBACZ,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;gBAC5C,IAAI,IAAI,EAAE,CAAC;oBACT,UAAU,CAAC,KAAK,EAAE,CAAC;oBACnB,MAAM;gBACR,CAAC;gBAED,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YAC5B,CAAC;QACH,CAAC;KACF,CAA2B,CAAC;AAC/B,CAAC;AAED,MAAM,UAAU,0CAA0C,CACxD,aAA+B,EAC/B,WAA8B,EAC9B,MAAoB;IAEpB,MAAM,MAAM,GAAG,IAAI,cAAc,CAAI;QACnC,KAAK,CAAC,KAAK,CAAC,UAAU;YACpB,IAAI,CAAC;gBACH,IAAI,MAAM,EAAE,CAAC;oBACX,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,GAAG,EAAE;wBACpC,UAAU,CAAC,KAAK,EAAE,CAAC;oBACrB,CAAC,CAAC,CAAC;gBACL,CAAC;gBAED,MAAM,QAAQ,GAAG,aAAa,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,CAAC;gBAEvD,OAAO,IAAI,EAAE,CAAC;oBACZ,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC;wBACpB,MAAM;oBACR,CAAC;oBAED,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;oBAE9C,IAAI,IAAI,EAAE,CAAC;wBACT,UAAU,CAAC,KAAK,EAAE,CAAC;wBACnB,MAAM;oBACR,CAAC;oBAED,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;gBAC5B,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,UAAU,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YAC1B,CAAC;QACH,CAAC;QACD,MAAM;YACJ,kEAAkE;YAClE,IAAI,MAAM,IAAI,aAAa,EAAE,CAAC;gBAC3B,aAAqB,CAAC,IAAI,EAAE,CAAC;YAChC,CAAC;QACH,CAAC;KACF,CAAC,CAAC;IAEH,MAAM,iBAAiB,GAAG,MAAM,CAAC,WAAW,CAAC,IAAI,eAAe,CAAC,WAAW,CAAC,CAAC,CAAC;IAE/E,OAAO,iBAA2C,CAAC;AACrD,CAAC"}

package/dist/esm/v3/taskContext/index.d.ts

@@ -9,6 +9,7 @@ export declare class TaskContextAPI {
     get isInsideTask(): boolean;
     get ctx(): TaskRunContext | undefined;
     get worker(): ServerBackgroundWorker | undefined;
+    get isWarmStart(): boolean | undefined;
     get attributes(): Attributes;
     get workerAttributes(): Attributes;
     get contextAttributes(): Attributes;

package/dist/esm/v3/taskContext/index.js

@@ -1,5 +1,5 @@
-import { getGlobal, registerGlobal, unregisterGlobal } from "../utils/globals.js";
 import { SemanticInternalAttributes } from "../semanticInternalAttributes.js";
+import { getGlobal, registerGlobal, unregisterGlobal } from "../utils/globals.js";
 const API_NAME = "task-context";
 export class TaskContextAPI {
     static _instance;
@@ -19,11 +19,15 @@ export class TaskContextAPI {
     get worker() {
         return this.#getTaskContext()?.worker;
     }
+    get isWarmStart() {
+        return this.#getTaskContext()?.isWarmStart;
+    }
     get attributes() {
         if (this.ctx) {
             return {
                 ...this.contextAttributes,
                 ...this.workerAttributes,
+                [SemanticInternalAttributes.WARM_START]: !!this.isWarmStart,
             };
         }
         return {};

package/dist/esm/v3/taskContext/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/v3/taskContext/index.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,SAAS,EAAE,cAAc,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAElF,OAAO,EAAE,0BAA0B,EAAE,MAAM,kCAAkC,CAAC;AAE9E,MAAM,QAAQ,GAAG,cAAc,CAAC;AAEhC,MAAM,OAAO,cAAc;IACjB,MAAM,CAAC,SAAS,CAAkB;IAE1C,gBAAuB,CAAC;IAEjB,MAAM,CAAC,WAAW;QACvB,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,IAAI,CAAC,SAAS,GAAG,IAAI,cAAc,EAAE,CAAC;QACxC,CAAC;QAED,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAED,IAAI,YAAY;QACd,OAAO,IAAI,CAAC,eAAe,EAAE,KAAK,SAAS,CAAC;IAC9C,CAAC;IAED,IAAI,GAAG;QACL,OAAO,IAAI,CAAC,eAAe,EAAE,EAAE,GAAG,CAAC;IACrC,CAAC;IAED,IAAI,MAAM;QACR,OAAO,IAAI,CAAC,eAAe,EAAE,EAAE,MAAM,CAAC;IACxC,CAAC;IAED,IAAI,UAAU;QACZ,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;YACb,OAAO;gBACL,GAAG,IAAI,CAAC,iBAAiB;gBACzB,GAAG,IAAI,CAAC,gBAAgB;aACzB,CAAC;QACJ,CAAC;QAED,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,IAAI,gBAAgB;QAClB,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,OAAO;gBACL,CAAC,0BAA0B,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE;gBACtD,CAAC,0BAA0B,CAAC,cAAc,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO;aACjE,CAAC;QACJ,CAAC;QAED,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,IAAI,iBAAiB;QACnB,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;YACb,OAAO;gBACL,CAAC,0BAA0B,CAAC,UAAU,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE;gBAC5D,CAAC,0BAA0B,CAAC,cAAc,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM;gBACpE,CAAC,0BAA0B,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE;gBACxD,CAAC,0BAA0B,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ;gBAC9D,CAAC,0BAA0B,CAAC,gBAAgB,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU;gBACvE,CAAC,0BAA0B,CAAC,UAAU,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI;gBAC5D,CAAC,0BAA0B,CAAC,QAAQ,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE;gBACxD,CAAC,0BAA0B,CAAC,cAAc,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE;gBACpE,CAAC,0BAA0B,CAAC,gBAAgB,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI;gBACxE,CAAC,0BAA0B,CAAC,eAAe,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,EAAE;gBACtE,CAAC,0BAA0B,CAAC,UAAU,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE;gBAC5D,CAAC,0BAA0B,CAAC,WAAW,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,GAAG;gBAC9D,CAAC,0BAA0B,CAAC,YAAY,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI;gBAChE,CAAC,0BAA0B,CAAC,MAAM,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE;gBACpD,CAAC,0BAA0B,CAAC,WAAW,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM;gBAC7D,CAAC,0BAA0B,CAAC,iBAAiB,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,IAAI;gBAC1E,CAAC,0BAA0B,CAAC,iBAAiB,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,IAAI;gBAC1E,CAAC,0BAA0B,CAAC,QAAQ,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE;gBACzD,CAAC,0BAA0B,CAAC,eAAe,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,cAAc;gBACzE,CAAC,0BAA0B,CAAC,mBAAmB,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI;gBACxE,CAAC,0BAA0B,CAAC,kBAAkB,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,GAAG;gBACtE,CAAC,0BAA0B,CAAC,qBAAqB,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,MAAM;gBAC5E,CAAC,0BAA0B,CAAC,2BAA2B,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,UAAU;aACvF,CAAC;QACJ,CAAC;QAED,OAAO,EAAE,CAAC;IACZ,CAAC;IAEM,OAAO;QACZ,gBAAgB,CAAC,QAAQ,CAAC,CAAC;IAC7B,CAAC;IAEM,oBAAoB,CAAC,WAAwB;QAClD,OAAO,cAAc,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;IAC/C,CAAC;IAED,eAAe;QACb,OAAO,SAAS,CAAC,QAAQ,CAAC,CAAC;IAC7B,CAAC;CACF"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/v3/taskContext/index.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,0BAA0B,EAAE,MAAM,kCAAkC,CAAC;AAC9E,OAAO,EAAE,SAAS,EAAE,cAAc,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAGlF,MAAM,QAAQ,GAAG,cAAc,CAAC;AAEhC,MAAM,OAAO,cAAc;IACjB,MAAM,CAAC,SAAS,CAAkB;IAE1C,gBAAuB,CAAC;IAEjB,MAAM,CAAC,WAAW;QACvB,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,IAAI,CAAC,SAAS,GAAG,IAAI,cAAc,EAAE,CAAC;QACxC,CAAC;QAED,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAED,IAAI,YAAY;QACd,OAAO,IAAI,CAAC,eAAe,EAAE,KAAK,SAAS,CAAC;IAC9C,CAAC;IAED,IAAI,GAAG;QACL,OAAO,IAAI,CAAC,eAAe,EAAE,EAAE,GAAG,CAAC;IACrC,CAAC;IAED,IAAI,MAAM;QACR,OAAO,IAAI,CAAC,eAAe,EAAE,EAAE,MAAM,CAAC;IACxC,CAAC;IAED,IAAI,WAAW;QACb,OAAO,IAAI,CAAC,eAAe,EAAE,EAAE,WAAW,CAAC;IAC7C,CAAC;IAED,IAAI,UAAU;QACZ,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;YACb,OAAO;gBACL,GAAG,IAAI,CAAC,iBAAiB;gBACzB,GAAG,IAAI,CAAC,gBAAgB;gBACxB,CAAC,0BAA0B,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,WAAW;aAC5D,CAAC;QACJ,CAAC;QAED,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,IAAI,gBAAgB;QAClB,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,OAAO;gBACL,CAAC,0BAA0B,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE;gBACtD,CAAC,0BAA0B,CAAC,cAAc,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO;aACjE,CAAC;QACJ,CAAC;QAED,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,IAAI,iBAAiB;QACnB,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;YACb,OAAO;gBACL,CAAC,0BAA0B,CAAC,UAAU,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE;gBAC5D,CAAC,0BAA0B,CAAC,cAAc,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM;gBACpE,CAAC,0BAA0B,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE;gBACxD,CAAC,0BAA0B,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ;gBAC9D,CAAC,0BAA0B,CAAC,gBAAgB,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU;gBACvE,CAAC,0BAA0B,CAAC,UAAU,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI;gBAC5D,CAAC,0BAA0B,CAAC,QAAQ,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE;gBACxD,CAAC,0BAA0B,CAAC,cAAc,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE;gBACpE,CAAC,0BAA0B,CAAC,gBAAgB,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI;gBACxE,CAAC,0BAA0B,CAAC,eAAe,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,EAAE;gBACtE,CAAC,0BAA0B,CAAC,UAAU,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE;gBAC5D,CAAC,0BAA0B,CAAC,WAAW,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,GAAG;gBAC9D,CAAC,0BAA0B,CAAC,YAAY,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI;gBAChE,CAAC,0BAA0B,CAAC,MAAM,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE;gBACpD,CAAC,0BAA0B,CAAC,WAAW,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM;gBAC7D,CAAC,0BAA0B,CAAC,iBAAiB,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,IAAI;gBAC1E,CAAC,0BAA0B,CAAC,iBAAiB,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,IAAI;gBAC1E,CAAC,0BAA0B,CAAC,QAAQ,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE;gBACzD,CAAC,0BAA0B,CAAC,eAAe,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,cAAc;gBACzE,CAAC,0BAA0B,CAAC,mBAAmB,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI;gBACxE,CAAC,0BAA0B,CAAC,kBAAkB,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,GAAG;gBACtE,CAAC,0BAA0B,CAAC,qBAAqB,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,MAAM;gBAC5E,CAAC,0BAA0B,CAAC,2BAA2B,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,UAAU;aACvF,CAAC;QACJ,CAAC;QAED,OAAO,EAAE,CAAC;IACZ,CAAC;IAEM,OAAO;QACZ,gBAAgB,CAAC,QAAQ,CAAC,CAAC;IAC7B,CAAC;IAEM,oBAAoB,CAAC,WAAwB;QAClD,OAAO,cAAc,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;IAC/C,CAAC;IAED,eAAe;QACb,OAAO,SAAS,CAAC,QAAQ,CAAC,CAAC;IAC7B,CAAC;CACF"}

package/dist/esm/v3/taskContext/otelProcessors.d.ts

@@ -1,9 +1,11 @@
 import { LogRecord, LogRecordProcessor } from "@opentelemetry/sdk-logs";
 import { Span, SpanProcessor } from "@opentelemetry/sdk-trace-base";
 import { Context } from "@opentelemetry/api";
+import { Tracer } from "@opentelemetry/api";
 export declare class TaskContextSpanProcessor implements SpanProcessor {
     private _innerProcessor;
-    constructor(innerProcessor: SpanProcessor);
+    private _tracer;
+    constructor(tracer: Tracer, innerProcessor: SpanProcessor);
     onStart(span: Span, parentContext: Context): void;
     onEnd(span: Span): void;
     shutdown(): Promise<void>;

package/dist/esm/v3/taskContext/otelProcessors.js

@@ -3,7 +3,9 @@ import { flattenAttributes } from "../utils/flattenAttributes.js";
 import { taskContext } from "../task-context-api.js";
 export class TaskContextSpanProcessor {
     _innerProcessor;
-    constructor(innerProcessor) {
+    _tracer;
+    constructor(tracer, innerProcessor) {
+        this._tracer = tracer;
         this._innerProcessor = innerProcessor;
     }
     // Called when a span starts
@@ -14,6 +16,10 @@ export class TaskContextSpanProcessor {
                 [SemanticInternalAttributes.ATTEMPT_NUMBER]: taskContext.ctx.attempt.number,
             }, SemanticInternalAttributes.METADATA));
         }
+        if (!isPartialSpan(span) && !skipPartialSpan(span)) {
+            const partialSpan = createPartialSpan(this._tracer, span, parentContext);
+            partialSpan.end();
+        }
         this._innerProcessor.onStart(span, parentContext);
     }
     // Delegate the rest of the methods to the wrapped processor
@@ -27,6 +33,33 @@ export class TaskContextSpanProcessor {
         return this._innerProcessor.forceFlush();
     }
 }
+function isPartialSpan(span) {
+    return span.attributes[SemanticInternalAttributes.SPAN_PARTIAL] === true;
+}
+function skipPartialSpan(span) {
+    return span.attributes[SemanticInternalAttributes.SKIP_SPAN_PARTIAL] === true;
+}
+function createPartialSpan(tracer, span, parentContext) {
+    const partialSpan = tracer.startSpan(span.name, {
+        attributes: {
+            [SemanticInternalAttributes.SPAN_PARTIAL]: true,
+            [SemanticInternalAttributes.SPAN_ID]: span.spanContext().spanId,
+            ...span.attributes,
+        },
+    }, parentContext);
+    if (taskContext.ctx) {
+        partialSpan.setAttributes(flattenAttributes({
+            [SemanticInternalAttributes.ATTEMPT_ID]: taskContext.ctx.attempt.id,
+            [SemanticInternalAttributes.ATTEMPT_NUMBER]: taskContext.ctx.attempt.number,
+        }, SemanticInternalAttributes.METADATA));
+    }
+    if (span.events) {
+        for (const event of span.events) {
+            partialSpan.addEvent(event.name, event.attributes, event.time);
+        }
+    }
+    return partialSpan;
+}
 export class TaskContextLogProcessor {
     _innerProcessor;
     constructor(innerProcessor) {

package/dist/esm/v3/taskContext/otelProcessors.js.map

@@ -1 +1 @@
-{"version":3,"file":"otelProcessors.js","sourceRoot":"","sources":["../../../../src/v3/taskContext/otelProcessors.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,0BAA0B,EAAE,MAAM,kCAAkC,CAAC;AAE9E,OAAO,EAAE,iBAAiB,EAAE,MAAM,+BAA+B,CAAC;AAClE,OAAO,EAAE,WAAW,EAAE,MAAM,wBAAwB,CAAC;AAErD,MAAM,OAAO,wBAAwB;IAC3B,eAAe,CAAgB;IAEvC,YAAY,cAA6B;QACvC,IAAI,CAAC,eAAe,GAAG,cAAc,CAAC;IACxC,CAAC;IAED,4BAA4B;IAC5B,OAAO,CAAC,IAAU,EAAE,aAAsB;QACxC,IAAI,WAAW,CAAC,GAAG,EAAE,CAAC;YACpB,IAAI,CAAC,aAAa,CAChB,iBAAiB,CACf;gBACE,CAAC,0BAA0B,CAAC,UAAU,CAAC,EAAE,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE;gBACnE,CAAC,0BAA0B,CAAC,cAAc,CAAC,EAAE,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM;aAC5E,EACD,0BAA0B,CAAC,QAAQ,CACpC,CACF,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,IAAI,EAAE,aAAa,CAAC,CAAC;IACpD,CAAC;IAED,4DAA4D;IAE5D,KAAK,CAAC,IAAU;QACd,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IACnC,CAAC;IAED,QAAQ;QACN,OAAO,IAAI,CAAC,eAAe,CAAC,QAAQ,EAAE,CAAC;IACzC,CAAC;IAED,UAAU;QACR,OAAO,IAAI,CAAC,eAAe,CAAC,UAAU,EAAE,CAAC;IAC3C,CAAC;CACF;AAED,MAAM,OAAO,uBAAuB;IAC1B,eAAe,CAAqB;IAE5C,YAAY,cAAkC;QAC5C,IAAI,CAAC,eAAe,GAAG,cAAc,CAAC;IACxC,CAAC;IACD,UAAU;QACR,OAAO,IAAI,CAAC,eAAe,CAAC,UAAU,EAAE,CAAC;IAC3C,CAAC;IACD,MAAM,CAAC,SAAoB,EAAE,OAA6B;QACxD,mDAAmD;QACnD,IAAI,WAAW,CAAC,GAAG,EAAE,CAAC;YACpB,SAAS,CAAC,aAAa,CACrB,iBAAiB,CACf;gBACE,CAAC,0BAA0B,CAAC,UAAU,CAAC,EAAE,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE;gBACnE,CAAC,0BAA0B,CAAC,cAAc,CAAC,EAAE,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM;aAC5E,EACD,0BAA0B,CAAC,QAAQ,CACpC,CACF,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;IAClD,CAAC;IACD,QAAQ;QACN,OAAO,IAAI,CAAC,eAAe,CAAC,QAAQ,EAAE,CAAC;IACzC,CAAC;CACF"}
+{"version":3,"file":"otelProcessors.js","sourceRoot":"","sources":["../../../../src/v3/taskContext/otelProcessors.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,0BAA0B,EAAE,MAAM,kCAAkC,CAAC;AAE9E,OAAO,EAAE,iBAAiB,EAAE,MAAM,+BAA+B,CAAC;AAClE,OAAO,EAAE,WAAW,EAAE,MAAM,wBAAwB,CAAC;AAGrD,MAAM,OAAO,wBAAwB;IAC3B,eAAe,CAAgB;IAC/B,OAAO,CAAS;IAExB,YAAY,MAAc,EAAE,cAA6B;QACvD,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,CAAC,eAAe,GAAG,cAAc,CAAC;IACxC,CAAC;IAED,4BAA4B;IAC5B,OAAO,CAAC,IAAU,EAAE,aAAsB;QACxC,IAAI,WAAW,CAAC,GAAG,EAAE,CAAC;YACpB,IAAI,CAAC,aAAa,CAChB,iBAAiB,CACf;gBACE,CAAC,0BAA0B,CAAC,UAAU,CAAC,EAAE,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE;gBACnE,CAAC,0BAA0B,CAAC,cAAc,CAAC,EAAE,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM;aAC5E,EACD,0BAA0B,CAAC,QAAQ,CACpC,CACF,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,EAAE,CAAC;YACnD,MAAM,WAAW,GAAG,iBAAiB,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,EAAE,aAAa,CAAC,CAAC;YACzE,WAAW,CAAC,GAAG,EAAE,CAAC;QACpB,CAAC;QAED,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,IAAI,EAAE,aAAa,CAAC,CAAC;IACpD,CAAC;IAED,4DAA4D;IAE5D,KAAK,CAAC,IAAU;QACd,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IACnC,CAAC;IAED,QAAQ;QACN,OAAO,IAAI,CAAC,eAAe,CAAC,QAAQ,EAAE,CAAC;IACzC,CAAC;IAED,UAAU;QACR,OAAO,IAAI,CAAC,eAAe,CAAC,UAAU,EAAE,CAAC;IAC3C,CAAC;CACF;AAED,SAAS,aAAa,CAAC,IAAU;IAC/B,OAAO,IAAI,CAAC,UAAU,CAAC,0BAA0B,CAAC,YAAY,CAAC,KAAK,IAAI,CAAC;AAC3E,CAAC;AAED,SAAS,eAAe,CAAC,IAAU;IACjC,OAAO,IAAI,CAAC,UAAU,CAAC,0BAA0B,CAAC,iBAAiB,CAAC,KAAK,IAAI,CAAC;AAChF,CAAC;AAED,SAAS,iBAAiB,CAAC,MAAc,EAAE,IAAU,EAAE,aAAsB;IAC3E,MAAM,WAAW,GAAG,MAAM,CAAC,SAAS,CAClC,IAAI,CAAC,IAAI,EACT;QACE,UAAU,EAAE;YACV,CAAC,0BAA0B,CAAC,YAAY,CAAC,EAAE,IAAI;YAC/C,CAAC,0BAA0B,CAAC,OAAO,CAAC,EAAE,IAAI,CAAC,WAAW,EAAE,CAAC,MAAM;YAC/D,GAAG,IAAI,CAAC,UAAU;SACnB;KACF,EACD,aAAa,CACd,CAAC;IAEF,IAAI,WAAW,CAAC,GAAG,EAAE,CAAC;QACpB,WAAW,CAAC,aAAa,CACvB,iBAAiB,CACf;YACE,CAAC,0BAA0B,CAAC,UAAU,CAAC,EAAE,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE;YACnE,CAAC,0BAA0B,CAAC,cAAc,CAAC,EAAE,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM;SAC5E,EACD,0BAA0B,CAAC,QAAQ,CACpC,CACF,CAAC;IACJ,CAAC;IAED,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;QAChB,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChC,WAAW,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,UAAU,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;QACjE,CAAC;IACH,CAAC;IAED,OAAO,WAAW,CAAC;AACrB,CAAC;AAED,MAAM,OAAO,uBAAuB;IAC1B,eAAe,CAAqB;IAE5C,YAAY,cAAkC;QAC5C,IAAI,CAAC,eAAe,GAAG,cAAc,CAAC;IACxC,CAAC;IACD,UAAU;QACR,OAAO,IAAI,CAAC,eAAe,CAAC,UAAU,EAAE,CAAC;IAC3C,CAAC;IACD,MAAM,CAAC,SAAoB,EAAE,OAA6B;QACxD,mDAAmD;QACnD,IAAI,WAAW,CAAC,GAAG,EAAE,CAAC;YACpB,SAAS,CAAC,aAAa,CACrB,iBAAiB,CACf;gBACE,CAAC,0BAA0B,CAAC,UAAU,CAAC,EAAE,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE;gBACnE,CAAC,0BAA0B,CAAC,cAAc,CAAC,EAAE,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM;aAC5E,EACD,0BAA0B,CAAC,QAAQ,CACpC,CACF,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;IAClD,CAAC;IACD,QAAQ;QACN,OAAO,IAAI,CAAC,eAAe,CAAC,QAAQ,EAAE,CAAC;IACzC,CAAC;CACF"}

package/dist/esm/v3/taskContext/types.d.ts

@@ -2,4 +2,5 @@ import { ServerBackgroundWorker, TaskRunContext } from "../schemas/index.js";
 export type TaskContext = {
     ctx: TaskRunContext;
     worker: ServerBackgroundWorker;
+    isWarmStart?: boolean;
 };

package/dist/esm/v3/tracer.d.ts

@@ -1,4 +1,4 @@
-import { Context, SpanOptions, type Span, type Tracer } from "@opentelemetry/api";
+import { Attributes, Context, SpanOptions, TimeInput, type Span, type Tracer } from "@opentelemetry/api";
 import { Logger } from "@opentelemetry/api-logs";
 export type TriggerTracerConfig = {
     name: string;
@@ -7,6 +7,14 @@ export type TriggerTracerConfig = {
     tracer: Tracer;
     logger: Logger;
 };
+export type TriggerTracerSpanEvent = {
+    name: string;
+    attributes?: Attributes;
+    startTime?: TimeInput;
+};
+export type TriggerTracerSpanOptions = SpanOptions & {
+    events?: TriggerTracerSpanEvent[];
+};
 export declare class TriggerTracer {
     private readonly _config;
     constructor(_config: TriggerTracerConfig);
@@ -15,6 +23,6 @@ export declare class TriggerTracer {
     private _logger;
     private get logger();
     extractContext(traceContext?: Record<string, unknown>): Context;
-    startActiveSpan<T>(name: string, fn: (span: Span) => Promise<T>, options?: SpanOptions, ctx?: Context, signal?: AbortSignal): Promise<T>;
+    startActiveSpan<T>(name: string, fn: (span: Span) => Promise<T>, options?: TriggerTracerSpanOptions, ctx?: Context, signal?: AbortSignal): Promise<T>;
     startSpan(name: string, options?: SpanOptions, ctx?: Context): Span;
 }

package/dist/esm/v3/tracer.js

@@ -1,10 +1,11 @@
 import { SpanStatusCode, context, propagation, trace, } from "@opentelemetry/api";
 import { logs } from "@opentelemetry/api-logs";
-import { SemanticInternalAttributes } from "./semanticInternalAttributes.js";
 import { clock } from "./clock-api.js";
-import { usage } from "./usage-api.js";
-import { taskContext } from "./task-context-api.js";
+import { isCompleteTaskWithOutput } from "./errors.js";
 import { recordSpanException } from "./otel/utils.js";
+import { SemanticInternalAttributes } from "./semanticInternalAttributes.js";
+import { taskContext } from "./task-context-api.js";
+import { usage } from "./usage-api.js";
 export class TriggerTracer {
     _config;
     constructor(_config) {
@@ -35,9 +36,17 @@ export class TriggerTracer {
         const parentContext = ctx ?? context.active();
         const attributes = options?.attributes ?? {};
         let spanEnded = false;
+        const createPartialSpanWithEvents = options?.events && options.events.length > 0;
         return this.tracer.startActiveSpan(name, {
             ...options,
-            attributes,
+            attributes: {
+                ...attributes,
+                ...(createPartialSpanWithEvents
+                    ? {
+                        [SemanticInternalAttributes.SKIP_SPAN_PARTIAL]: true,
+                    }
+                    : {}),
+            },
             startTime: clock.preciseNow(),
         }, parentContext, async (span) => {
             signal?.addEventListener("abort", () => {
@@ -47,23 +56,38 @@ export class TriggerTracer {
                     span.end();
                 }
             });
-            if (taskContext.ctx) {
-                this.tracer
-                    .startSpan(name, {
+            if (taskContext.ctx && createPartialSpanWithEvents) {
+                const partialSpan = this.tracer.startSpan(name, {
                     ...options,
                     attributes: {
                         ...attributes,
                         [SemanticInternalAttributes.SPAN_PARTIAL]: true,
                         [SemanticInternalAttributes.SPAN_ID]: span.spanContext().spanId,
                     },
-                }, parentContext)
-                    .end();
+                }, parentContext);
+                if (options?.events) {
+                    for (const event of options.events) {
+                        partialSpan.addEvent(event.name, event.attributes, event.startTime);
+                    }
+                }
+                partialSpan.end();
+            }
+            if (options?.events) {
+                for (const event of options.events) {
+                    span.addEvent(event.name, event.attributes, event.startTime);
+                }
             }
             const usageMeasurement = usage.start();
             try {
                 return await fn(span);
             }
             catch (e) {
+                if (isCompleteTaskWithOutput(e)) {
+                    if (!spanEnded) {
+                        span.end(clock.preciseNow());
+                    }
+                    throw e;
+                }
                 if (!spanEnded) {
                     if (typeof e === "string" || e instanceof Error) {
                         span.recordException(e);
@@ -93,17 +117,7 @@ export class TriggerTracer {
     startSpan(name, options, ctx) {
         const parentContext = ctx ?? context.active();
         const attributes = options?.attributes ?? {};
-        const span = this.tracer.startSpan(name, options, ctx);
-        this.tracer
-            .startSpan(name, {
-            ...options,
-            attributes: {
-                ...attributes,
-                [SemanticInternalAttributes.SPAN_PARTIAL]: true,
-                [SemanticInternalAttributes.SPAN_ID]: span.spanContext().spanId,
-            },
-        }, parentContext)
-            .end();
+        const span = this.tracer.startSpan(name, options, parentContext);
         return span;
     }
 }

package/dist/esm/v3/tracer.js.map

@@ -1 +1 @@
-{"version":3,"file":"tracer.js","sourceRoot":"","sources":["../../../src/v3/tracer.ts"],"names":[],"mappings":"AAAA,OAAO,EAGL,cAAc,EACd,OAAO,EACP,WAAW,EACX,KAAK,GAGN,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EAAU,IAAI,EAAE,MAAM,yBAAyB,CAAC;AACvD,OAAO,EAAE,0BAA0B,EAAE,MAAM,iCAAiC,CAAC;AAC7E,OAAO,EAAE,KAAK,EAAE,MAAM,gBAAgB,CAAC;AACvC,OAAO,EAAE,KAAK,EAAE,MAAM,gBAAgB,CAAC;AACvC,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AACpD,OAAO,EAAE,mBAAmB,EAAE,MAAM,iBAAiB,CAAC;AAYtD,MAAM,OAAO,aAAa;IACK;IAA7B,YAA6B,OAA4B;QAA5B,YAAO,GAAP,OAAO,CAAqB;IAAG,CAAC;IAErD,OAAO,CAAqB;IACpC,IAAY,MAAM;QAChB,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAClB,IAAI,QAAQ,IAAI,IAAI,CAAC,OAAO;gBAAE,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC;YAEzD,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAC1E,CAAC;QAED,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAEO,OAAO,CAAqB;IACpC,IAAY,MAAM;QAChB,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAClB,IAAI,QAAQ,IAAI,IAAI,CAAC,OAAO;gBAAE,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC;YAEzD,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QACzE,CAAC;QAED,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAED,cAAc,CAAC,YAAsC;QACnD,OAAO,WAAW,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,YAAY,IAAI,EAAE,CAAC,CAAC;IACnE,CAAC;IAED,eAAe,CACb,IAAY,EACZ,EAA8B,EAC9B,OAAqB,EACrB,GAAa,EACb,MAAoB;QAEpB,MAAM,aAAa,GAAG,GAAG,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;QAE9C,MAAM,UAAU,GAAG,OAAO,EAAE,UAAU,IAAI,EAAE,CAAC;QAE7C,IAAI,SAAS,GAAG,KAAK,CAAC;QAEtB,OAAO,IAAI,CAAC,MAAM,CAAC,eAAe,CAChC,IAAI,EACJ;YACE,GAAG,OAAO;YACV,UAAU;YACV,SAAS,EAAE,KAAK,CAAC,UAAU,EAAE;SAC9B,EACD,aAAa,EACb,KAAK,EAAE,IAAI,EAAE,EAAE;YACb,MAAM,EAAE,gBAAgB,CAAC,OAAO,EAAE,GAAG,EAAE;gBACrC,IAAI,CAAC,SAAS,EAAE,CAAC;oBACf,SAAS,GAAG,IAAI,CAAC;oBACjB,mBAAmB,CAAC,IAAI,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;oBACzC,IAAI,CAAC,GAAG,EAAE,CAAC;gBACb,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,IAAI,WAAW,CAAC,GAAG,EAAE,CAAC;gBACpB,IAAI,CAAC,MAAM;qBACR,SAAS,CACR,IAAI,EACJ;oBACE,GAAG,OAAO;oBACV,UAAU,EAAE;wBACV,GAAG,UAAU;wBACb,CAAC,0BAA0B,CAAC,YAAY,CAAC,EAAE,IAAI;wBAC/C,CAAC,0BAA0B,CAAC,OAAO,CAAC,EAAE,IAAI,CAAC,WAAW,EAAE,CAAC,MAAM;qBAChE;iBACF,EACD,aAAa,CACd;qBACA,GAAG,EAAE,CAAC;YACX,CAAC;YAED,MAAM,gBAAgB,GAAG,KAAK,CAAC,KAAK,EAAE,CAAC;YAEvC,IAAI,CAAC;gBACH,OAAO,MAAM,EAAE,CAAC,IAAI,CAAC,CAAC;YACxB,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,IAAI,CAAC,SAAS,EAAE,CAAC;oBACf,IAAI,OAAO,CAAC,KAAK,QAAQ,IAAI,CAAC,YAAY,KAAK,EAAE,CAAC;wBAChD,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;oBAC1B,CAAC;oBAED,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,cAAc,CAAC,KAAK,EAAE,CAAC,CAAC;gBACjD,CAAC;gBAED,MAAM,CAAC,CAAC;YACV,CAAC;oBAAS,CAAC;gBACT,IAAI,CAAC,SAAS,EAAE,CAAC;oBACf,SAAS,GAAG,IAAI,CAAC;oBAEjB,IAAI,WAAW,CAAC,GAAG,EAAE,CAAC;wBACpB,MAAM,WAAW,GAAG,KAAK,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;wBACjD,MAAM,OAAO,GAAG,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC;wBAExC,IAAI,CAAC,aAAa,CAAC;4BACjB,CAAC,0BAA0B,CAAC,iBAAiB,CAAC,EAAE,WAAW,CAAC,OAAO;4BACnE,CAAC,0BAA0B,CAAC,mBAAmB,CAAC,EAAE,OAAO,EAAE,UAAU;gCACnE,CAAC,CAAC,WAAW,CAAC,OAAO,GAAG,OAAO,CAAC,UAAU;gCAC1C,CAAC,CAAC,CAAC;yBACN,CAAC,CAAC;oBACL,CAAC;oBAED,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,UAAU,EAAE,CAAC,CAAC;gBAC/B,CAAC;YACH,CAAC;QACH,CAAC,CACF,CAAC;IACJ,CAAC;IAED,SAAS,CAAC,IAAY,EAAE,OAAqB,EAAE,GAAa;QAC1D,MAAM,aAAa,GAAG,GAAG,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;QAE9C,MAAM,UAAU,GAAG,OAAO,EAAE,UAAU,IAAI,EAAE,CAAC;QAE7C,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,EAAE,OAAO,EAAE,GAAG,CAAC,CAAC;QAEvD,IAAI,CAAC,MAAM;aACR,SAAS,CACR,IAAI,EACJ;YACE,GAAG,OAAO;YACV,UAAU,EAAE;gBACV,GAAG,UAAU;gBACb,CAAC,0BAA0B,CAAC,YAAY,CAAC,EAAE,IAAI;gBAC/C,CAAC,0BAA0B,CAAC,OAAO,CAAC,EAAE,IAAI,CAAC,WAAW,EAAE,CAAC,MAAM;aAChE;SACF,EACD,aAAa,CACd;aACA,GAAG,EAAE,CAAC;QAET,OAAO,IAAI,CAAC;IACd,CAAC;CACF"}
+{"version":3,"file":"tracer.js","sourceRoot":"","sources":["../../../src/v3/tracer.ts"],"names":[],"mappings":"AAAA,OAAO,EAIL,cAAc,EAEd,OAAO,EACP,WAAW,EACX,KAAK,GAGN,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EAAU,IAAI,EAAE,MAAM,yBAAyB,CAAC;AACvD,OAAO,EAAE,KAAK,EAAE,MAAM,gBAAgB,CAAC;AACvC,OAAO,EAAE,wBAAwB,EAAE,MAAM,aAAa,CAAC;AACvD,OAAO,EAAE,mBAAmB,EAAE,MAAM,iBAAiB,CAAC;AACtD,OAAO,EAAE,0BAA0B,EAAE,MAAM,iCAAiC,CAAC;AAC7E,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AACpD,OAAO,EAAE,KAAK,EAAE,MAAM,gBAAgB,CAAC;AAsBvC,MAAM,OAAO,aAAa;IACK;IAA7B,YAA6B,OAA4B;QAA5B,YAAO,GAAP,OAAO,CAAqB;IAAG,CAAC;IAErD,OAAO,CAAqB;IACpC,IAAY,MAAM;QAChB,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAClB,IAAI,QAAQ,IAAI,IAAI,CAAC,OAAO;gBAAE,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC;YAEzD,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAC1E,CAAC;QAED,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAEO,OAAO,CAAqB;IACpC,IAAY,MAAM;QAChB,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAClB,IAAI,QAAQ,IAAI,IAAI,CAAC,OAAO;gBAAE,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC;YAEzD,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QACzE,CAAC;QAED,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAED,cAAc,CAAC,YAAsC;QACnD,OAAO,WAAW,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,YAAY,IAAI,EAAE,CAAC,CAAC;IACnE,CAAC;IAED,eAAe,CACb,IAAY,EACZ,EAA8B,EAC9B,OAAkC,EAClC,GAAa,EACb,MAAoB;QAEpB,MAAM,aAAa,GAAG,GAAG,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;QAE9C,MAAM,UAAU,GAAG,OAAO,EAAE,UAAU,IAAI,EAAE,CAAC;QAE7C,IAAI,SAAS,GAAG,KAAK,CAAC;QAEtB,MAAM,2BAA2B,GAAG,OAAO,EAAE,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC;QAEjF,OAAO,IAAI,CAAC,MAAM,CAAC,eAAe,CAChC,IAAI,EACJ;YACE,GAAG,OAAO;YACV,UAAU,EAAE;gBACV,GAAG,UAAU;gBACb,GAAG,CAAC,2BAA2B;oBAC7B,CAAC,CAAC;wBACE,CAAC,0BAA0B,CAAC,iBAAiB,CAAC,EAAE,IAAI;qBACrD;oBACH,CAAC,CAAC,EAAE,CAAC;aACR;YACD,SAAS,EAAE,KAAK,CAAC,UAAU,EAAE;SAC9B,EACD,aAAa,EACb,KAAK,EAAE,IAAI,EAAE,EAAE;YACb,MAAM,EAAE,gBAAgB,CAAC,OAAO,EAAE,GAAG,EAAE;gBACrC,IAAI,CAAC,SAAS,EAAE,CAAC;oBACf,SAAS,GAAG,IAAI,CAAC;oBACjB,mBAAmB,CAAC,IAAI,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;oBACzC,IAAI,CAAC,GAAG,EAAE,CAAC;gBACb,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,IAAI,WAAW,CAAC,GAAG,IAAI,2BAA2B,EAAE,CAAC;gBACnD,MAAM,WAAW,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,CACvC,IAAI,EACJ;oBACE,GAAG,OAAO;oBACV,UAAU,EAAE;wBACV,GAAG,UAAU;wBACb,CAAC,0BAA0B,CAAC,YAAY,CAAC,EAAE,IAAI;wBAC/C,CAAC,0BAA0B,CAAC,OAAO,CAAC,EAAE,IAAI,CAAC,WAAW,EAAE,CAAC,MAAM;qBAChE;iBACF,EACD,aAAa,CACd,CAAC;gBAEF,IAAI,OAAO,EAAE,MAAM,EAAE,CAAC;oBACpB,KAAK,MAAM,KAAK,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;wBACnC,WAAW,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,UAAU,EAAE,KAAK,CAAC,SAAS,CAAC,CAAC;oBACtE,CAAC;gBACH,CAAC;gBAED,WAAW,CAAC,GAAG,EAAE,CAAC;YACpB,CAAC;YAED,IAAI,OAAO,EAAE,MAAM,EAAE,CAAC;gBACpB,KAAK,MAAM,KAAK,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;oBACnC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,UAAU,EAAE,KAAK,CAAC,SAAS,CAAC,CAAC;gBAC/D,CAAC;YACH,CAAC;YAED,MAAM,gBAAgB,GAAG,KAAK,CAAC,KAAK,EAAE,CAAC;YAEvC,IAAI,CAAC;gBACH,OAAO,MAAM,EAAE,CAAC,IAAI,CAAC,CAAC;YACxB,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,IAAI,wBAAwB,CAAC,CAAC,CAAC,EAAE,CAAC;oBAChC,IAAI,CAAC,SAAS,EAAE,CAAC;wBACf,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,UAAU,EAAE,CAAC,CAAC;oBAC/B,CAAC;oBAED,MAAM,CAAC,CAAC;gBACV,CAAC;gBAED,IAAI,CAAC,SAAS,EAAE,CAAC;oBACf,IAAI,OAAO,CAAC,KAAK,QAAQ,IAAI,CAAC,YAAY,KAAK,EAAE,CAAC;wBAChD,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;oBAC1B,CAAC;oBAED,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,cAAc,CAAC,KAAK,EAAE,CAAC,CAAC;gBACjD,CAAC;gBAED,MAAM,CAAC,CAAC;YACV,CAAC;oBAAS,CAAC;gBACT,IAAI,CAAC,SAAS,EAAE,CAAC;oBACf,SAAS,GAAG,IAAI,CAAC;oBAEjB,IAAI,WAAW,CAAC,GAAG,EAAE,CAAC;wBACpB,MAAM,WAAW,GAAG,KAAK,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;wBACjD,MAAM,OAAO,GAAG,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC;wBAExC,IAAI,CAAC,aAAa,CAAC;4BACjB,CAAC,0BAA0B,CAAC,iBAAiB,CAAC,EAAE,WAAW,CAAC,OAAO;4BACnE,CAAC,0BAA0B,CAAC,mBAAmB,CAAC,EAAE,OAAO,EAAE,UAAU;gCACnE,CAAC,CAAC,WAAW,CAAC,OAAO,GAAG,OAAO,CAAC,UAAU;gCAC1C,CAAC,CAAC,CAAC;yBACN,CAAC,CAAC;oBACL,CAAC;oBAED,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,UAAU,EAAE,CAAC,CAAC;gBAC/B,CAAC;YACH,CAAC;QACH,CAAC,CACF,CAAC;IACJ,CAAC;IAED,SAAS,CAAC,IAAY,EAAE,OAAqB,EAAE,GAAa;QAC1D,MAAM,aAAa,GAAG,GAAG,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;QAE9C,MAAM,UAAU,GAAG,OAAO,EAAE,UAAU,IAAI,EAAE,CAAC;QAE7C,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,EAAE,OAAO,EAAE,aAAa,CAAC,CAAC;QAEjE,OAAO,IAAI,CAAC;IACd,CAAC;CACF"}