lightrace 0.1.0

package/dist/tool-client.js

@@ -0,0 +1,186 @@
+/**
+ * WebSocket client for remote tool invocation.
+ */
+import WebSocket from "ws";
+import { _getToolRegistry } from "./trace.js";
+import { generateId } from "./utils.js";
+import { sign, verify, NonceTracker } from "./security.js";
+import { jsonSerializable } from "./utils.js";
+export class ToolClient {
+    host;
+    publicKey;
+    secretKey;
+    sdkInstanceId;
+    heartbeatInterval;
+    maxReconnectDelay;
+    sessionToken = null;
+    nonceTracker = new NonceTracker(60);
+    running = false;
+    ws = null;
+    reconnectTimer = null;
+    heartbeatTimer = null;
+    constructor(options) {
+        this.host = options.host.replace(/\/$/, "");
+        this.publicKey = options.publicKey;
+        this.secretKey = options.secretKey;
+        this.sdkInstanceId = options.sdkInstanceId ?? generateId();
+        this.heartbeatInterval = (options.heartbeatInterval ?? 30) * 1000;
+        this.maxReconnectDelay = (options.maxReconnectDelay ?? 30) * 1000;
+    }
+    get wsUrl() {
+        return (this.host.replace("http://", "ws://").replace("https://", "wss://") + "/api/public/tools/ws");
+    }
+    start() {
+        const registry = _getToolRegistry();
+        if (registry.size === 0)
+            return;
+        this.running = true;
+        this.connect(1000);
+    }
+    stop() {
+        this.running = false;
+        if (this.reconnectTimer)
+            clearTimeout(this.reconnectTimer);
+        if (this.heartbeatTimer)
+            clearInterval(this.heartbeatTimer);
+        if (this.ws) {
+            this.ws.close();
+            this.ws = null;
+        }
+    }
+    connect(delay) {
+        if (!this.running)
+            return;
+        const auth = Buffer.from(`${this.publicKey}:${this.secretKey}`).toString("base64");
+        const ws = new WebSocket(this.wsUrl, {
+            headers: { Authorization: `Basic ${auth}` },
+        });
+        ws.on("open", () => {
+            this.ws = ws;
+        });
+        ws.on("message", (data) => {
+            try {
+                const msg = JSON.parse(data.toString());
+                this.handleMessage(ws, msg);
+            }
+            catch {
+                // ignore malformed messages
+            }
+        });
+        ws.on("close", () => {
+            this.cleanup();
+            if (this.running) {
+                const nextDelay = Math.min(delay * 2, this.maxReconnectDelay);
+                this.reconnectTimer = setTimeout(() => this.connect(nextDelay), delay);
+            }
+        });
+        ws.on("error", () => {
+            // close event will handle reconnection
+        });
+    }
+    cleanup() {
+        if (this.heartbeatTimer) {
+            clearInterval(this.heartbeatTimer);
+            this.heartbeatTimer = null;
+        }
+        this.sessionToken = null;
+        this.ws = null;
+    }
+    handleMessage(ws, msg) {
+        switch (msg.type) {
+            case "connected":
+                this.sessionToken = msg.sessionToken;
+                this.registerTools(ws);
+                this.startHeartbeat(ws);
+                break;
+            case "registered":
+                // Tools registered successfully
+                break;
+            case "invoke":
+                this.handleInvoke(ws, msg);
+                break;
+            case "heartbeat_ack":
+                break;
+            case "error":
+                console.error(`[lightrace] Server error: ${msg.message}`);
+                break;
+        }
+    }
+    registerTools(ws) {
+        const registry = _getToolRegistry();
+        const tools = Array.from(registry.entries()).map(([name, entry]) => ({
+            name,
+            inputSchema: entry.inputSchema,
+        }));
+        ws.send(JSON.stringify({
+            type: "register",
+            sdkInstanceId: this.sdkInstanceId,
+            tools,
+        }));
+    }
+    startHeartbeat(ws) {
+        this.heartbeatTimer = setInterval(() => {
+            if (ws.readyState === WebSocket.OPEN) {
+                ws.send(JSON.stringify({ type: "heartbeat" }));
+            }
+        }, this.heartbeatInterval);
+    }
+    async handleInvoke(ws, msg) {
+        const { nonce, tool, input, signature: sig } = msg;
+        // Verify HMAC
+        if (!this.sessionToken || !verify(this.sessionToken, nonce, tool, input, sig)) {
+            ws.send(JSON.stringify({ type: "error", message: "Invalid signature" }));
+            return;
+        }
+        // Check nonce freshness
+        if (!this.nonceTracker.checkAndMark(nonce)) {
+            ws.send(JSON.stringify({ type: "error", message: "Replayed nonce" }));
+            return;
+        }
+        const registry = _getToolRegistry();
+        const entry = registry.get(tool);
+        if (!entry) {
+            ws.send(JSON.stringify({
+                type: "result",
+                nonce,
+                output: null,
+                error: `Tool not found: ${tool}`,
+                durationMs: 0,
+                signature: sign(this.sessionToken, nonce, tool, null),
+            }));
+            return;
+        }
+        const start = performance.now();
+        try {
+            let output;
+            if (input && typeof input === "object" && !Array.isArray(input)) {
+                output = await entry.fn(input);
+            }
+            else {
+                output = input !== null && input !== undefined ? await entry.fn(input) : await entry.fn();
+            }
+            const durationMs = Math.round(performance.now() - start);
+            output = jsonSerializable(output);
+            ws.send(JSON.stringify({
+                type: "result",
+                nonce,
+                output,
+                durationMs,
+                signature: sign(this.sessionToken, nonce, tool, output),
+            }));
+        }
+        catch (err) {
+            const durationMs = Math.round(performance.now() - start);
+            const error = err instanceof Error ? err.message : String(err);
+            ws.send(JSON.stringify({
+                type: "result",
+                nonce,
+                output: null,
+                error,
+                durationMs,
+                signature: sign(this.sessionToken, nonce, tool, null),
+            }));
+        }
+    }
+}
+//# sourceMappingURL=tool-client.js.map

package/dist/tool-client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-client.js","sourceRoot":"","sources":["../src/tool-client.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,OAAO,SAAS,MAAM,IAAI,CAAC;AAE3B,OAAO,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAC9C,OAAO,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AACxC,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAC3D,OAAO,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAE9C,MAAM,OAAO,UAAU;IACb,IAAI,CAAS;IACb,SAAS,CAAS;IAClB,SAAS,CAAS;IAClB,aAAa,CAAS;IACtB,iBAAiB,CAAS;IAC1B,iBAAiB,CAAS;IAE1B,YAAY,GAAkB,IAAI,CAAC;IACnC,YAAY,GAAG,IAAI,YAAY,CAAC,EAAE,CAAC,CAAC;IACpC,OAAO,GAAG,KAAK,CAAC;IAChB,EAAE,GAAqB,IAAI,CAAC;IAC5B,cAAc,GAAyC,IAAI,CAAC;IAC5D,cAAc,GAA0C,IAAI,CAAC;IAErE,YAAY,OAOX;QACC,IAAI,CAAC,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QAC5C,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;QACnC,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;QACnC,IAAI,CAAC,aAAa,GAAG,OAAO,CAAC,aAAa,IAAI,UAAU,EAAE,CAAC;QAC3D,IAAI,CAAC,iBAAiB,GAAG,CAAC,OAAO,CAAC,iBAAiB,IAAI,EAAE,CAAC,GAAG,IAAI,CAAC;QAClE,IAAI,CAAC,iBAAiB,GAAG,CAAC,OAAO,CAAC,iBAAiB,IAAI,EAAE,CAAC,GAAG,IAAI,CAAC;IACpE,CAAC;IAED,IAAY,KAAK;QACf,OAAO,CACL,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC,OAAO,CAAC,UAAU,EAAE,QAAQ,CAAC,GAAG,sBAAsB,CAC7F,CAAC;IACJ,CAAC;IAED,KAAK;QACH,MAAM,QAAQ,GAAG,gBAAgB,EAAE,CAAC;QACpC,IAAI,QAAQ,CAAC,IAAI,KAAK,CAAC;YAAE,OAAO;QAEhC,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;QACpB,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IACrB,CAAC;IAED,IAAI;QACF,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC;QACrB,IAAI,IAAI,CAAC,cAAc;YAAE,YAAY,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAC3D,IAAI,IAAI,CAAC,cAAc;YAAE,aAAa,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAC5D,IAAI,IAAI,CAAC,EAAE,EAAE,CAAC;YACZ,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC;YAChB,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC;QACjB,CAAC;IACH,CAAC;IAEO,OAAO,CAAC,KAAa;QAC3B,IAAI,CAAC,IAAI,CAAC,OAAO;YAAE,OAAO;QAE1B,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QAEnF,MAAM,EAAE,GAAG,IAAI,SAAS,CAAC,IAAI,CAAC,KAAK,EAAE;YACnC,OAAO,EAAE,EAAE,aAAa,EAAE,SAAS,IAAI,EAAE,EAAE;SAC5C,CAAC,CAAC;QAEH,EAAE,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE;YACjB,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QACf,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,IAAI,EAAE,EAAE;YACxB,IAAI,CAAC;gBACH,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAkB,CAAC;gBACzD,IAAI,CAAC,aAAa,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;YAC9B,CAAC;YAAC,MAAM,CAAC;gBACP,4BAA4B;YAC9B,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;YAClB,IAAI,CAAC,OAAO,EAAE,CAAC;YACf,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;gBACjB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,GAAG,CAAC,EAAE,IAAI,CAAC,iBAAiB,CAAC,CAAC;gBAC9D,IAAI,CAAC,cAAc,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,KAAK,CAAC,CAAC;YACzE,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;YAClB,uCAAuC;QACzC,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,OAAO;QACb,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,aAAa,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YACnC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAC7B,CAAC;QACD,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;QACzB,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC;IACjB,CAAC;IAEO,aAAa,CAAC,EAAa,EAAE,GAAkB;QACrD,QAAQ,GAAG,CAAC,IAAI,EAAE,CAAC;YACjB,KAAK,WAAW;gBACd,IAAI,CAAC,YAAY,GAAG,GAAG,CAAC,YAAY,CAAC;gBACrC,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC;gBACvB,IAAI,CAAC,cAAc,CAAC,EAAE,CAAC,CAAC;gBACxB,MAAM;YAER,KAAK,YAAY;gBACf,gCAAgC;gBAChC,MAAM;YAER,KAAK,QAAQ;gBACX,IAAI,CAAC,YAAY,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;gBAC3B,MAAM;YAER,KAAK,eAAe;gBAClB,MAAM;YAER,KAAK,OAAO;gBACV,OAAO,CAAC,KAAK,CAAC,6BAA6B,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;gBAC1D,MAAM;QACV,CAAC;IACH,CAAC;IAEO,aAAa,CAAC,EAAa;QACjC,MAAM,QAAQ,GAAG,gBAAgB,EAAE,CAAC;QACpC,MAAM,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC;YACnE,IAAI;YACJ,WAAW,EAAE,KAAK,CAAC,WAAW;SAC/B,CAAC,CAAC,CAAC;QAEJ,EAAE,CAAC,IAAI,CACL,IAAI,CAAC,SAAS,CAAC;YACb,IAAI,EAAE,UAAU;YAChB,aAAa,EAAE,IAAI,CAAC,aAAa;YACjC,KAAK;SACN,CAAC,CACH,CAAC;IACJ,CAAC;IAEO,cAAc,CAAC,EAAa;QAClC,IAAI,CAAC,cAAc,GAAG,WAAW,CAAC,GAAG,EAAE;YACrC,IAAI,EAAE,CAAC,UAAU,KAAK,SAAS,CAAC,IAAI,EAAE,CAAC;gBACrC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC;YACjD,CAAC;QACH,CAAC,EAAE,IAAI,CAAC,iBAAiB,CAAC,CAAC;IAC7B,CAAC;IAEO,KAAK,CAAC,YAAY,CACxB,EAAa,EACb,GAAuE;QAEvE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,SAAS,EAAE,GAAG,EAAE,GAAG,GAAG,CAAC;QAEnD,cAAc;QACd,IAAI,CAAC,IAAI,CAAC,YAAY,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,YAAY,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,CAAC,EAAE,CAAC;YAC9E,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,mBAAmB,EAAE,CAAC,CAAC,CAAC;YACzE,OAAO;QACT,CAAC;QAED,wBAAwB;QACxB,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,YAAY,CAAC,KAAK,CAAC,EAAE,CAAC;YAC3C,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,gBAAgB,EAAE,CAAC,CAAC,CAAC;YACtE,OAAO;QACT,CAAC;QAED,MAAM,QAAQ,GAAG,gBAAgB,EAAE,CAAC;QACpC,MAAM,KAAK,GAAG,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACjC,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,EAAE,CAAC,IAAI,CACL,IAAI,CAAC,SAAS,CAAC;gBACb,IAAI,EAAE,QAAQ;gBACd,KAAK;gBACL,MAAM,EAAE,IAAI;gBACZ,KAAK,EAAE,mBAAmB,IAAI,EAAE;gBAChC,UAAU,EAAE,CAAC;gBACb,SAAS,EAAE,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,CAAC;aACtD,CAAC,CACH,CAAC;YACF,OAAO;QACT,CAAC;QAED,MAAM,KAAK,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;QAChC,IAAI,CAAC;YACH,IAAI,MAAe,CAAC;YACpB,IAAI,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;gBAChE,MAAM,GAAG,MAAM,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC;YACjC,CAAC;iBAAM,CAAC;gBACN,MAAM,GAAG,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,KAAK,CAAC,EAAE,EAAE,CAAC;YAC5F,CAAC;YACD,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC,CAAC;YACzD,MAAM,GAAG,gBAAgB,CAAC,MAAM,CAAC,CAAC;YAElC,EAAE,CAAC,IAAI,CACL,IAAI,CAAC,SAAS,CAAC;gBACb,IAAI,EAAE,QAAQ;gBACd,KAAK;gBACL,MAAM;gBACN,UAAU;gBACV,SAAS,EAAE,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,CAAC;aACxD,CAAC,CACH,CAAC;QACJ,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC,CAAC;YACzD,MAAM,KAAK,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YAE/D,EAAE,CAAC,IAAI,CACL,IAAI,CAAC,SAAS,CAAC;gBACb,IAAI,EAAE,QAAQ;gBACd,KAAK;gBACL,MAAM,EAAE,IAAI;gBACZ,KAAK;gBACL,UAAU;gBACV,SAAS,EAAE,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,CAAC;aACtD,CAAC,CACH,CAAC;QACJ,CAAC;IACH,CAAC;CACF"}

package/dist/trace.d.ts

@@ -0,0 +1,25 @@
+import type { TraceOptions, ToolRegistryEntry } from "./types.js";
+import type { BatchExporter } from "./exporter.js";
+/** Context for trace propagation. */
+interface TraceContext {
+    traceId: string;
+    observationId: string | null;
+}
+export declare function _setExporter(exporter: BatchExporter | null): void;
+export declare function _setClientDefaults(defaults: {
+    userId?: string;
+    sessionId?: string;
+}): void;
+/** Get the current trace context (used by imperative API). */
+export declare function _getTraceContext(): TraceContext | undefined;
+export declare function _getToolRegistry(): Map<string, ToolRegistryEntry>;
+/**
+ * Unified trace wrapper.
+ *
+ * Signatures:
+ *   trace(name, fn)              — root trace (no options)
+ *   trace(name, options, fn)     — with options
+ */
+export declare function trace<T extends (...args: any[]) => any>(name: string, fnOrOptions: T | TraceOptions, maybeFn?: T): T;
+export {};
+//# sourceMappingURL=trace.d.ts.map

package/dist/trace.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"trace.d.ts","sourceRoot":"","sources":["../src/trace.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAc,YAAY,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AAG9E,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AAEnD,qCAAqC;AACrC,UAAU,YAAY;IACpB,OAAO,EAAE,MAAM,CAAC;IAChB,aAAa,EAAE,MAAM,GAAG,IAAI,CAAC;CAC9B;AAaD,wBAAgB,YAAY,CAAC,QAAQ,EAAE,aAAa,GAAG,IAAI,GAAG,IAAI,CAEjE;AAED,wBAAgB,kBAAkB,CAAC,QAAQ,EAAE;IAAE,MAAM,CAAC,EAAE,MAAM,CAAC;IAAC,SAAS,CAAC,EAAE,MAAM,CAAA;CAAE,GAAG,IAAI,CAE1F;AAED,8DAA8D;AAC9D,wBAAgB,gBAAgB,IAAI,YAAY,GAAG,SAAS,CAE3D;AAED,wBAAgB,gBAAgB,IAAI,GAAG,CAAC,MAAM,EAAE,iBAAiB,CAAC,CAEjE;AAED;;;;;;GAMG;AACH,wBAAgB,KAAK,CAAC,CAAC,SAAS,CAAC,GAAG,IAAI,EAAE,GAAG,EAAE,KAAK,GAAG,EACrD,IAAI,EAAE,MAAM,EACZ,WAAW,EAAE,CAAC,GAAG,YAAY,EAC7B,OAAO,CAAC,EAAE,CAAC,GACV,CAAC,CA4IH"}

package/dist/trace.js

@@ -0,0 +1,163 @@
+/**
+ * Unified trace() wrapper for all observation types.
+ */
+import { AsyncLocalStorage } from "node:async_hooks";
+import { EVENT_TYPE_MAP, OBSERVATION_TYPE_ENUM } from "./types.js";
+import { generateId, jsonSerializable, zodToJsonSchema } from "./utils.js";
+const asyncStorage = new AsyncLocalStorage();
+/** Global exporter reference (set by Client). */
+let _exporter = null;
+/** Client-level defaults (set by Client to avoid circular imports). */
+let _clientDefaults = {};
+/** Global tool registry for invocable tools. */
+const _toolRegistry = new Map();
+export function _setExporter(exporter) {
+    _exporter = exporter;
+}
+export function _setClientDefaults(defaults) {
+    _clientDefaults = defaults;
+}
+/** Get the current trace context (used by imperative API). */
+export function _getTraceContext() {
+    return asyncStorage.getStore();
+}
+export function _getToolRegistry() {
+    return _toolRegistry;
+}
+/**
+ * Unified trace wrapper.
+ *
+ * Signatures:
+ *   trace(name, fn)              — root trace (no options)
+ *   trace(name, options, fn)     — with options
+ */
+export function trace(name, fnOrOptions, maybeFn) {
+    let options;
+    let fn;
+    if (typeof fnOrOptions === "function") {
+        options = {};
+        fn = fnOrOptions;
+    }
+    else {
+        options = fnOrOptions;
+        fn = maybeFn;
+    }
+    const obsName = options.name ?? name;
+    const obsType = options.type ?? null;
+    const model = options.model ?? null;
+    const staticMetadata = options.metadata ?? null;
+    const invoke = options.invoke !== false; // default true
+    const usage = options.usage ?? null;
+    // Resolve userId/sessionId: per-trace option > client default
+    const userId = options.userId ?? _clientDefaults.userId;
+    const sessionId = options.sessionId ?? _clientDefaults.sessionId;
+    // Register tool for remote invocation
+    if (obsType === "tool" && invoke) {
+        const inputSchema = options.inputSchema ? zodToJsonSchema(options.inputSchema) : null;
+        _toolRegistry.set(obsName, {
+            fn: fn,
+            inputSchema,
+        });
+    }
+    const wrapped = ((...args) => {
+        const isRoot = obsType === null;
+        const entityId = generateId();
+        const startTime = new Date();
+        // Get parent context
+        const parentCtx = asyncStorage.getStore();
+        const traceId = isRoot ? entityId : (parentCtx?.traceId ?? generateId());
+        const parentObservationId = isRoot ? null : (parentCtx?.observationId ?? null);
+        // Set context for children
+        const childCtx = {
+            traceId,
+            observationId: isRoot ? null : entityId,
+        };
+        const emitEvent = (output, level, statusMessage) => {
+            const endTime = new Date();
+            const capturedInput = jsonSerializable(args.length === 1 ? args[0] : args);
+            if (isRoot) {
+                const body = {
+                    id: entityId,
+                    name: obsName,
+                    timestamp: startTime.toISOString(),
+                    input: capturedInput,
+                    output: jsonSerializable(output),
+                    metadata: staticMetadata,
+                };
+                if (userId !== undefined)
+                    body.userId = userId;
+                if (sessionId !== undefined)
+                    body.sessionId = sessionId;
+                const event = {
+                    id: generateId(),
+                    type: "trace-create",
+                    timestamp: startTime.toISOString(),
+                    body,
+                };
+                _exporter?.enqueue(event);
+            }
+            else {
+                const createType = EVENT_TYPE_MAP[obsType]?.[0] ?? "span-create";
+                const body = {
+                    id: entityId,
+                    traceId,
+                    type: OBSERVATION_TYPE_ENUM[obsType] ?? obsType,
+                    name: obsName,
+                    startTime: startTime.toISOString(),
+                    endTime: endTime.toISOString(),
+                    input: capturedInput,
+                    output: jsonSerializable(output),
+                    metadata: staticMetadata,
+                    model,
+                    level,
+                    statusMessage,
+                    parentObservationId,
+                };
+                // Add usage fields for generation observations
+                if (usage && obsType === "generation") {
+                    if (usage.promptTokens !== undefined)
+                        body.promptTokens = usage.promptTokens;
+                    if (usage.completionTokens !== undefined)
+                        body.completionTokens = usage.completionTokens;
+                    if (usage.totalTokens !== undefined)
+                        body.totalTokens = usage.totalTokens;
+                }
+                const event = {
+                    id: generateId(),
+                    type: createType,
+                    timestamp: startTime.toISOString(),
+                    body,
+                };
+                _exporter?.enqueue(event);
+            }
+        };
+        const execute = () => {
+            try {
+                const result = fn(...args);
+                // Handle async functions
+                if (result && typeof result === "object" && typeof result.then === "function") {
+                    return result.then((resolved) => {
+                        emitEvent(resolved, "DEFAULT", null);
+                        return resolved;
+                    }, (err) => {
+                        emitEvent(null, "ERROR", err.message ?? String(err));
+                        throw err;
+                    });
+                }
+                // Sync result
+                emitEvent(result, "DEFAULT", null);
+                return result;
+            }
+            catch (err) {
+                const message = err instanceof Error ? err.message : String(err);
+                emitEvent(null, "ERROR", message);
+                throw err;
+            }
+        };
+        return asyncStorage.run(childCtx, execute);
+    });
+    // Preserve function name
+    Object.defineProperty(wrapped, "name", { value: fn.name || name });
+    return wrapped;
+}
+//# sourceMappingURL=trace.js.map

package/dist/trace.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"trace.js","sourceRoot":"","sources":["../src/trace.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AAErD,OAAO,EAAE,cAAc,EAAE,qBAAqB,EAAE,MAAM,YAAY,CAAC;AACnE,OAAO,EAAE,UAAU,EAAE,gBAAgB,EAAE,eAAe,EAAE,MAAM,YAAY,CAAC;AAS3E,MAAM,YAAY,GAAG,IAAI,iBAAiB,EAAgB,CAAC;AAE3D,iDAAiD;AACjD,IAAI,SAAS,GAAyB,IAAI,CAAC;AAE3C,uEAAuE;AACvE,IAAI,eAAe,GAA4C,EAAE,CAAC;AAElE,gDAAgD;AAChD,MAAM,aAAa,GAAG,IAAI,GAAG,EAA6B,CAAC;AAE3D,MAAM,UAAU,YAAY,CAAC,QAA8B;IACzD,SAAS,GAAG,QAAQ,CAAC;AACvB,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAC,QAAiD;IAClF,eAAe,GAAG,QAAQ,CAAC;AAC7B,CAAC;AAED,8DAA8D;AAC9D,MAAM,UAAU,gBAAgB;IAC9B,OAAO,YAAY,CAAC,QAAQ,EAAE,CAAC;AACjC,CAAC;AAED,MAAM,UAAU,gBAAgB;IAC9B,OAAO,aAAa,CAAC;AACvB,CAAC;AAED;;;;;;GAMG;AACH,MAAM,UAAU,KAAK,CACnB,IAAY,EACZ,WAA6B,EAC7B,OAAW;IAEX,IAAI,OAAqB,CAAC;IAC1B,IAAI,EAAK,CAAC;IAEV,IAAI,OAAO,WAAW,KAAK,UAAU,EAAE,CAAC;QACtC,OAAO,GAAG,EAAE,CAAC;QACb,EAAE,GAAG,WAAW,CAAC;IACnB,CAAC;SAAM,CAAC;QACN,OAAO,GAAG,WAAW,CAAC;QACtB,EAAE,GAAG,OAAQ,CAAC;IAChB,CAAC;IAED,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,IAAI,IAAI,CAAC;IACrC,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,IAAI,IAAI,CAAC;IACrC,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,IAAI,CAAC;IACpC,MAAM,cAAc,GAAG,OAAO,CAAC,QAAQ,IAAI,IAAI,CAAC;IAChD,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,KAAK,KAAK,CAAC,CAAC,eAAe;IACxD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,IAAI,CAAC;IAEpC,8DAA8D;IAC9D,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,eAAe,CAAC,MAAM,CAAC;IACxD,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,eAAe,CAAC,SAAS,CAAC;IAEjE,sCAAsC;IACtC,IAAI,OAAO,KAAK,MAAM,IAAI,MAAM,EAAE,CAAC;QACjC,MAAM,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,eAAe,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;QACtF,aAAa,CAAC,GAAG,CAAC,OAAO,EAAE;YACzB,EAAE,EAAE,EAAqC;YACzC,WAAW;SACZ,CAAC,CAAC;IACL,CAAC;IAED,MAAM,OAAO,GAAG,CAAC,CAAC,GAAG,IAAe,EAAE,EAAE;QACtC,MAAM,MAAM,GAAG,OAAO,KAAK,IAAI,CAAC;QAChC,MAAM,QAAQ,GAAG,UAAU,EAAE,CAAC;QAC9B,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC;QAE7B,qBAAqB;QACrB,MAAM,SAAS,GAAG,YAAY,CAAC,QAAQ,EAAE,CAAC;QAC1C,MAAM,OAAO,GAAG,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,SAAS,EAAE,OAAO,IAAI,UAAU,EAAE,CAAC,CAAC;QACzE,MAAM,mBAAmB,GAAG,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS,EAAE,aAAa,IAAI,IAAI,CAAC,CAAC;QAE/E,2BAA2B;QAC3B,MAAM,QAAQ,GAAiB;YAC7B,OAAO;YACP,aAAa,EAAE,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ;SACxC,CAAC;QAEF,MAAM,SAAS,GAAG,CAAC,MAAe,EAAE,KAAa,EAAE,aAA4B,EAAE,EAAE;YACjF,MAAM,OAAO,GAAG,IAAI,IAAI,EAAE,CAAC;YAC3B,MAAM,aAAa,GAAG,gBAAgB,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;YAE3E,IAAI,MAAM,EAAE,CAAC;gBACX,MAAM,IAAI,GAA4B;oBACpC,EAAE,EAAE,QAAQ;oBACZ,IAAI,EAAE,OAAO;oBACb,SAAS,EAAE,SAAS,CAAC,WAAW,EAAE;oBAClC,KAAK,EAAE,aAAa;oBACpB,MAAM,EAAE,gBAAgB,CAAC,MAAM,CAAC;oBAChC,QAAQ,EAAE,cAAc;iBACzB,CAAC;gBACF,IAAI,MAAM,KAAK,SAAS;oBAAE,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;gBAC/C,IAAI,SAAS,KAAK,SAAS;oBAAE,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;gBAExD,MAAM,KAAK,GAAe;oBACxB,EAAE,EAAE,UAAU,EAAE;oBAChB,IAAI,EAAE,cAAc;oBACpB,SAAS,EAAE,SAAS,CAAC,WAAW,EAAE;oBAClC,IAAI;iBACL,CAAC;gBACF,SAAS,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC;YAC5B,CAAC;iBAAM,CAAC;gBACN,MAAM,UAAU,GAAG,cAAc,CAAC,OAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,aAAa,CAAC;gBAClE,MAAM,IAAI,GAA4B;oBACpC,EAAE,EAAE,QAAQ;oBACZ,OAAO;oBACP,IAAI,EAAE,qBAAqB,CAAC,OAAQ,CAAC,IAAI,OAAO;oBAChD,IAAI,EAAE,OAAO;oBACb,SAAS,EAAE,SAAS,CAAC,WAAW,EAAE;oBAClC,OAAO,EAAE,OAAO,CAAC,WAAW,EAAE;oBAC9B,KAAK,EAAE,aAAa;oBACpB,MAAM,EAAE,gBAAgB,CAAC,MAAM,CAAC;oBAChC,QAAQ,EAAE,cAAc;oBACxB,KAAK;oBACL,KAAK;oBACL,aAAa;oBACb,mBAAmB;iBACpB,CAAC;gBACF,+CAA+C;gBAC/C,IAAI,KAAK,IAAI,OAAO,KAAK,YAAY,EAAE,CAAC;oBACtC,IAAI,KAAK,CAAC,YAAY,KAAK,SAAS;wBAAE,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC,YAAY,CAAC;oBAC7E,IAAI,KAAK,CAAC,gBAAgB,KAAK,SAAS;wBAAE,IAAI,CAAC,gBAAgB,GAAG,KAAK,CAAC,gBAAgB,CAAC;oBACzF,IAAI,KAAK,CAAC,WAAW,KAAK,SAAS;wBAAE,IAAI,CAAC,WAAW,GAAG,KAAK,CAAC,WAAW,CAAC;gBAC5E,CAAC;gBAED,MAAM,KAAK,GAAe;oBACxB,EAAE,EAAE,UAAU,EAAE;oBAChB,IAAI,EAAE,UAAU;oBAChB,SAAS,EAAE,SAAS,CAAC,WAAW,EAAE;oBAClC,IAAI;iBACL,CAAC;gBACF,SAAS,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC;YAC5B,CAAC;QACH,CAAC,CAAC;QAEF,MAAM,OAAO,GAAG,GAAG,EAAE;YACnB,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,EAAE,CAAC,GAAG,IAAI,CAAC,CAAC;gBAE3B,yBAAyB;gBACzB,IAAI,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,OAAO,MAAM,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;oBAC9E,OAAQ,MAA2B,CAAC,IAAI,CACtC,CAAC,QAAQ,EAAE,EAAE;wBACX,SAAS,CAAC,QAAQ,EAAE,SAAS,EAAE,IAAI,CAAC,CAAC;wBACrC,OAAO,QAAQ,CAAC;oBAClB,CAAC,EACD,CAAC,GAAU,EAAE,EAAE;wBACb,SAAS,CAAC,IAAI,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,IAAI,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;wBACrD,MAAM,GAAG,CAAC;oBACZ,CAAC,CACF,CAAC;gBACJ,CAAC;gBAED,cAAc;gBACd,SAAS,CAAC,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,CAAC;gBACnC,OAAO,MAAM,CAAC;YAChB,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,MAAM,OAAO,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;gBACjE,SAAS,CAAC,IAAI,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;gBAClC,MAAM,GAAG,CAAC;YACZ,CAAC;QACH,CAAC,CAAC;QAEF,OAAO,YAAY,CAAC,GAAG,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IAC7C,CAAC,CAAiB,CAAC;IAEnB,yBAAyB;IACzB,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC,IAAI,IAAI,IAAI,EAAE,CAAC,CAAC;IAEnE,OAAO,OAAO,CAAC;AACjB,CAAC"}

package/dist/types.d.ts

@@ -0,0 +1,85 @@
+/** Observation types supported by lightrace. */
+export type ObservationType = "span" | "generation" | "event" | "tool" | "chain";
+/** Map user-facing type strings to ingestion event types [create, update]. */
+export declare const EVENT_TYPE_MAP: Record<string, [createType: string, updateType: string]>;
+/** Map type strings to observation type enum values. */
+export declare const OBSERVATION_TYPE_ENUM: Record<string, string>;
+/** Token / cost tracking for generation observations. */
+export interface UsageDetails {
+    promptTokens?: number;
+    completionTokens?: number;
+    totalTokens?: number;
+}
+/** Options for the trace() wrapper. */
+export interface TraceOptions {
+    /** Observation type. undefined = root trace. */
+    type?: ObservationType;
+    /** Override observation name (defaults to the name argument). */
+    name?: string;
+    /** For type="tool": register for remote invocation. Default: true. */
+    invoke?: boolean;
+    /** For type="generation": LLM model name. */
+    model?: string;
+    /** Optional Zod schema for tool input (auto-converted to JSON Schema). */
+    inputSchema?: {
+        _def?: unknown;
+    };
+    /** Static metadata attached to every call. */
+    metadata?: Record<string, unknown>;
+    /** User ID for this trace/observation. */
+    userId?: string;
+    /** Session ID for this trace/observation. */
+    sessionId?: string;
+    /** Token usage for generation observations. */
+    usage?: UsageDetails;
+}
+/** A single event to be batched and sent to the ingestion endpoint. */
+export interface TraceEvent {
+    id: string;
+    type: string;
+    timestamp: string;
+    body: Record<string, unknown>;
+}
+/** Tool registration entry in the local registry. */
+export interface ToolRegistryEntry {
+    fn: (...args: unknown[]) => unknown;
+    inputSchema: Record<string, unknown> | null;
+}
+/** WebSocket messages from SDK to backend. */
+export type SdkMessage = {
+    type: "register";
+    sdkInstanceId: string;
+    tools: Array<{
+        name: string;
+        inputSchema: unknown;
+    }>;
+} | {
+    type: "result";
+    nonce: string;
+    output: unknown;
+    error?: string;
+    durationMs: number;
+    signature: string;
+} | {
+    type: "heartbeat";
+};
+/** WebSocket messages from backend to SDK. */
+export type ServerMessage = {
+    type: "connected";
+    sessionToken: string;
+} | {
+    type: "registered";
+    tools: string[];
+} | {
+    type: "invoke";
+    nonce: string;
+    tool: string;
+    input: unknown;
+    signature: string;
+} | {
+    type: "heartbeat_ack";
+} | {
+    type: "error";
+    message: string;
+};
+//# 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,gDAAgD;AAChD,MAAM,MAAM,eAAe,GAAG,MAAM,GAAG,YAAY,GAAG,OAAO,GAAG,MAAM,GAAG,OAAO,CAAC;AAEjF,8EAA8E;AAC9E,eAAO,MAAM,cAAc,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,UAAU,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,CAAC,CAMnF,CAAC;AAEF,wDAAwD;AACxD,eAAO,MAAM,qBAAqB,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAMxD,CAAC;AAEF,yDAAyD;AACzD,MAAM,WAAW,YAAY;IAC3B,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AAED,uCAAuC;AACvC,MAAM,WAAW,YAAY;IAC3B,gDAAgD;IAChD,IAAI,CAAC,EAAE,eAAe,CAAC;IACvB,iEAAiE;IACjE,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,sEAAsE;IACtE,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,6CAA6C;IAC7C,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,0EAA0E;IAC1E,WAAW,CAAC,EAAE;QAAE,IAAI,CAAC,EAAE,OAAO,CAAA;KAAE,CAAC;IACjC,8CAA8C;IAC9C,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACnC,0CAA0C;IAC1C,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,6CAA6C;IAC7C,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,+CAA+C;IAC/C,KAAK,CAAC,EAAE,YAAY,CAAC;CACtB;AAED,uEAAuE;AACvE,MAAM,WAAW,UAAU;IACzB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,CAAC;IAClB,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAC/B;AAED,qDAAqD;AACrD,MAAM,WAAW,iBAAiB;IAChC,EAAE,EAAE,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,OAAO,CAAC;IACpC,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC;CAC7C;AAED,8CAA8C;AAC9C,MAAM,MAAM,UAAU,GAClB;IACE,IAAI,EAAE,UAAU,CAAC;IACjB,aAAa,EAAE,MAAM,CAAC;IACtB,KAAK,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,WAAW,EAAE,OAAO,CAAA;KAAE,CAAC,CAAC;CACtD,GACD;IACE,IAAI,EAAE,QAAQ,CAAC;IACf,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,OAAO,CAAC;IAChB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,EAAE,MAAM,CAAC;CACnB,GACD;IAAE,IAAI,EAAE,WAAW,CAAA;CAAE,CAAC;AAE1B,8CAA8C;AAC9C,MAAM,MAAM,aAAa,GACrB;IAAE,IAAI,EAAE,WAAW,CAAC;IAAC,YAAY,EAAE,MAAM,CAAA;CAAE,GAC3C;IAAE,IAAI,EAAE,YAAY,CAAC;IAAC,KAAK,EAAE,MAAM,EAAE,CAAA;CAAE,GACvC;IACE,IAAI,EAAE,QAAQ,CAAC;IACf,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,OAAO,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;CACnB,GACD;IAAE,IAAI,EAAE,eAAe,CAAA;CAAE,GACzB;IAAE,IAAI,EAAE,OAAO,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,CAAC"}

package/dist/types.js

@@ -0,0 +1,17 @@
+/** Map user-facing type strings to ingestion event types [create, update]. */
+export const EVENT_TYPE_MAP = {
+    span: ["span-create", "span-update"],
+    generation: ["generation-create", "generation-update"],
+    event: ["event-create", "event-create"],
+    tool: ["tool-create", "tool-update"],
+    chain: ["chain-create", "chain-update"],
+};
+/** Map type strings to observation type enum values. */
+export const OBSERVATION_TYPE_ENUM = {
+    span: "SPAN",
+    generation: "GENERATION",
+    event: "EVENT",
+    tool: "TOOL",
+    chain: "CHAIN",
+};
+//# sourceMappingURL=types.js.map

package/dist/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAGA,8EAA8E;AAC9E,MAAM,CAAC,MAAM,cAAc,GAA6D;IACtF,IAAI,EAAE,CAAC,aAAa,EAAE,aAAa,CAAC;IACpC,UAAU,EAAE,CAAC,mBAAmB,EAAE,mBAAmB,CAAC;IACtD,KAAK,EAAE,CAAC,cAAc,EAAE,cAAc,CAAC;IACvC,IAAI,EAAE,CAAC,aAAa,EAAE,aAAa,CAAC;IACpC,KAAK,EAAE,CAAC,cAAc,EAAE,cAAc,CAAC;CACxC,CAAC;AAEF,wDAAwD;AACxD,MAAM,CAAC,MAAM,qBAAqB,GAA2B;IAC3D,IAAI,EAAE,MAAM;IACZ,UAAU,EAAE,YAAY;IACxB,KAAK,EAAE,OAAO;IACd,IAAI,EAAE,MAAM;IACZ,KAAK,EAAE,OAAO;CACf,CAAC"}

package/dist/utils.d.ts

@@ -0,0 +1,7 @@
+/** Generate a unique ID. */
+export declare function generateId(): string;
+/** Make a value JSON-serializable. */
+export declare function jsonSerializable(value: unknown): unknown;
+/** Convert a Zod schema to JSON Schema (basic). */
+export declare function zodToJsonSchema(schema: unknown): Record<string, unknown> | null;
+//# 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":"AAEA,4BAA4B;AAC5B,wBAAgB,UAAU,IAAI,MAAM,CAEnC;AAED,sCAAsC;AACtC,wBAAgB,gBAAgB,CAAC,KAAK,EAAE,OAAO,GAAG,OAAO,CAcxD;AAED,mDAAmD;AACnD,wBAAgB,eAAe,CAAC,MAAM,EAAE,OAAO,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,CAwC/E"}

package/dist/utils.js

@@ -0,0 +1,66 @@
+import { randomUUID } from "node:crypto";
+/** Generate a unique ID. */
+export function generateId() {
+    return randomUUID();
+}
+/** Make a value JSON-serializable. */
+export function jsonSerializable(value) {
+    if (value === null || value === undefined)
+        return value;
+    if (typeof value === "string" || typeof value === "number" || typeof value === "boolean")
+        return value;
+    if (value instanceof Date)
+        return value.toISOString();
+    if (Array.isArray(value))
+        return value.map(jsonSerializable);
+    if (typeof value === "object") {
+        const result = {};
+        for (const [k, v] of Object.entries(value)) {
+            result[k] = jsonSerializable(v);
+        }
+        return result;
+    }
+    return String(value);
+}
+/** Convert a Zod schema to JSON Schema (basic). */
+export function zodToJsonSchema(schema) {
+    if (!schema || typeof schema !== "object")
+        return null;
+    const s = schema;
+    if (!s._def)
+        return null;
+    // Handle ZodObject
+    if (s._def.typeName === "ZodObject" && typeof s._def.shape === "function") {
+        const shape = s._def.shape();
+        const properties = {};
+        const required = [];
+        for (const [key, field] of Object.entries(shape)) {
+            const f = field;
+            if (!f._def)
+                continue;
+            let typeName = f._def.typeName;
+            // Unwrap ZodOptional
+            if (typeName === "ZodOptional" && f._def.innerType) {
+                const inner = f._def.innerType;
+                typeName = inner._def?.typeName;
+            }
+            else {
+                required.push(key);
+            }
+            const typeMap = {
+                ZodString: "string",
+                ZodNumber: "number",
+                ZodBoolean: "boolean",
+                ZodArray: "array",
+                ZodObject: "object",
+            };
+            properties[key] = { type: typeMap[typeName ?? ""] ?? "string" };
+        }
+        const result = { type: "object", properties };
+        if (required.length)
+            result.required = required;
+        return result;
+    }
+    return null;
+}
+//# sourceMappingURL=utils.js.map

package/dist/utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["../src/utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAEzC,4BAA4B;AAC5B,MAAM,UAAU,UAAU;IACxB,OAAO,UAAU,EAAE,CAAC;AACtB,CAAC;AAED,sCAAsC;AACtC,MAAM,UAAU,gBAAgB,CAAC,KAAc;IAC7C,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,SAAS;QAAE,OAAO,KAAK,CAAC;IACxD,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,OAAO,KAAK,KAAK,SAAS;QACtF,OAAO,KAAK,CAAC;IACf,IAAI,KAAK,YAAY,IAAI;QAAE,OAAO,KAAK,CAAC,WAAW,EAAE,CAAC;IACtD,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;QAAE,OAAO,KAAK,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;IAC7D,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC9B,MAAM,MAAM,GAA4B,EAAE,CAAC;QAC3C,KAAK,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YAC3C,MAAM,CAAC,CAAC,CAAC,GAAG,gBAAgB,CAAC,CAAC,CAAC,CAAC;QAClC,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IACD,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC;AACvB,CAAC;AAED,mDAAmD;AACnD,MAAM,UAAU,eAAe,CAAC,MAAe;IAC7C,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ;QAAE,OAAO,IAAI,CAAC;IACvD,MAAM,CAAC,GAAG,MAAiF,CAAC;IAC5F,IAAI,CAAC,CAAC,CAAC,IAAI;QAAE,OAAO,IAAI,CAAC;IAEzB,mBAAmB;IACnB,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,KAAK,WAAW,IAAI,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,KAAK,UAAU,EAAE,CAAC;QAC1E,MAAM,KAAK,GAAG,CAAC,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;QAC7B,MAAM,UAAU,GAA4B,EAAE,CAAC;QAC/C,MAAM,QAAQ,GAAa,EAAE,CAAC;QAE9B,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YACjD,MAAM,CAAC,GAAG,KAA8D,CAAC;YACzE,IAAI,CAAC,CAAC,CAAC,IAAI;gBAAE,SAAS;YAEtB,IAAI,QAAQ,GAAG,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC;YAC/B,qBAAqB;YACrB,IAAI,QAAQ,KAAK,aAAa,IAAI,CAAC,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;gBACnD,MAAM,KAAK,GAAG,CAAC,CAAC,IAAI,CAAC,SAA6C,CAAC;gBACnE,QAAQ,GAAG,KAAK,CAAC,IAAI,EAAE,QAAQ,CAAC;YAClC,CAAC;iBAAM,CAAC;gBACN,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACrB,CAAC;YAED,MAAM,OAAO,GAA2B;gBACtC,SAAS,EAAE,QAAQ;gBACnB,SAAS,EAAE,QAAQ;gBACnB,UAAU,EAAE,SAAS;gBACrB,QAAQ,EAAE,OAAO;gBACjB,SAAS,EAAE,QAAQ;aACpB,CAAC;YACF,UAAU,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,EAAE,OAAO,CAAC,QAAQ,IAAI,EAAE,CAAC,IAAI,QAAQ,EAAE,CAAC;QAClE,CAAC;QAED,MAAM,MAAM,GAA4B,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,CAAC;QACvE,IAAI,QAAQ,CAAC,MAAM;YAAE,MAAM,CAAC,QAAQ,GAAG,QAAQ,CAAC;QAChD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC"}