elasticdash-sdk 0.2.0

package/dist/runner.js

@@ -0,0 +1,138 @@
+import { clearRegistry, getRegistry } from './core/registry.js';
+import { startTraceSession, setCurrentTrace } from './trace-adapter/context.js';
+import { startLLMProxy, fetchCapturedTrace } from './proxy/llm-capture.js';
+import { pathToFileURL } from 'node:url';
+import { randomUUID } from 'node:crypto';
+import path from 'node:path';
+export async function runFiles(files, options = {}) {
+    // Optional local LLM capture proxy (opt-in via env). Default behavior stays unchanged when disabled.
+    const proxyOptIn = process.env.ELASTICDASH_LLM_PROXY === '1' || Boolean(process.env.ELASTICDASH_LLM_PROXY_URL);
+    const proxyPort = Number.parseInt(process.env.ELASTICDASH_LLM_PROXY_PORT || '8787', 10);
+    let proxyUrl = process.env.ELASTICDASH_LLM_PROXY_URL;
+    const proxyHandle = proxyOptIn && !proxyUrl ? await startLLMProxy({ port: proxyPort }) : null;
+    if (proxyHandle) {
+        proxyUrl = proxyHandle.url;
+    }
+    const fileResults = [];
+    for (const file of files) {
+        const result = await runFile(file, options, { proxyOptIn, proxyUrl });
+        fileResults.push(result);
+    }
+    if (proxyHandle) {
+        await proxyHandle.stop();
+    }
+    return fileResults;
+}
+async function runFile(file, options, proxyCtx) {
+    const { hooks = {} } = options;
+    // 1. Clear the global registry before loading the file
+    clearRegistry();
+    // 2. Dynamically import the test file (triggers aiTest() registrations)
+    const resolvedPath = file.startsWith('file://')
+        ? file
+        : pathToFileURL(path.resolve(file)).href;
+    if (resolvedPath.endsWith('.ts') && typeof globalThis.Deno === 'undefined') {
+        await import('tsx/esm');
+        await import('tsx/cjs');
+    }
+    await import(resolvedPath);
+    const registry = getRegistry();
+    const results = [];
+    // Shared unhandled error trap for this file's test run
+    let currentTestName = null;
+    let pendingUnhandled;
+    const onUnhandled = (reason) => {
+        if (!pendingUnhandled)
+            pendingUnhandled = reason instanceof Error ? reason : new Error(String(reason));
+    };
+    process.on('unhandledRejection', onUnhandled);
+    process.on('uncaughtException', onUnhandled);
+    // 3. Run beforeAll hooks
+    for (const hook of registry.beforeAllHooks) {
+        await hook();
+    }
+    // 4. Execute each test sequentially
+    for (const entry of registry.tests) {
+        const { context, finalise } = startTraceSession();
+        const traceId = proxyCtx.proxyOptIn ? randomUUID() : null;
+        setCurrentTrace(context.trace);
+        if (traceId) {
+            process.env.ELASTICDASH_TRACE_ID = traceId;
+        }
+        if (hooks.onTestStart) {
+            await hooks.onTestStart(entry.name);
+        }
+        const startTime = Date.now();
+        let passed = false;
+        let error;
+        // Reset per-test unhandled capture and mark current test name
+        pendingUnhandled = undefined;
+        currentTestName = entry.name;
+        setCurrentTrace(context.trace);
+        try {
+            for (const hook of registry.beforeEachHooks) {
+                await hook();
+            }
+            await entry.fn(context);
+            passed = true;
+        }
+        catch (err) {
+            error = err instanceof Error ? err : new Error(String(err));
+        }
+        finally {
+            try {
+                for (const hook of registry.afterEachHooks) {
+                    await hook();
+                }
+            }
+            catch (afterErr) {
+                if (!error) {
+                    error = afterErr instanceof Error ? afterErr : new Error(String(afterErr));
+                    passed = false;
+                }
+            }
+            setCurrentTrace(undefined);
+            if (!error && pendingUnhandled) {
+                error = pendingUnhandled;
+                passed = false;
+            }
+            currentTestName = null;
+        }
+        const durationMs = Date.now() - startTime;
+        if (hooks.onTestFinish) {
+            await hooks.onTestFinish(entry.name, passed, durationMs, error);
+        }
+        if (hooks.onTraceComplete) {
+            await hooks.onTraceComplete(entry.name, context.trace);
+        }
+        // If proxy mode is enabled, pull captured LLM steps and fold into the trace
+        if (traceId && proxyCtx.proxyUrl) {
+            try {
+                const captured = await fetchCapturedTrace(proxyCtx.proxyUrl, traceId);
+                for (const step of captured) {
+                    context.trace.recordLLMStep(step);
+                }
+            }
+            catch (proxyErr) {
+                // Non-fatal: keep test result as-is
+                // eslint-disable-next-line no-console
+                console.warn('[elasticdash] Failed to fetch proxy-captured steps:', proxyErr);
+            }
+        }
+        finalise();
+        setCurrentTrace(undefined);
+        if (traceId) {
+            delete process.env.ELASTICDASH_TRACE_ID;
+        }
+        results.push({ name: entry.name, passed, durationMs, error });
+    }
+    // 5. Run afterAll hooks
+    for (const hook of registry.afterAllHooks) {
+        await hook();
+    }
+    // Cleanup shared handlers
+    process.off('unhandledRejection', onUnhandled);
+    process.off('uncaughtException', onUnhandled);
+    return { file, results };
+}
+//# sourceMappingURL=runner.js.map

package/dist/runner.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"runner.js","sourceRoot":"","sources":["../src/runner.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAA;AAC/D,OAAO,EAAE,iBAAiB,EAAE,eAAe,EAAE,MAAM,4BAA4B,CAAA;AAC/E,OAAO,EAAE,aAAa,EAAE,kBAAkB,EAAE,MAAM,wBAAwB,CAAA;AAE1E,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAA;AACxC,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAA;AACxC,OAAO,IAAI,MAAM,WAAW,CAAA;AAkB5B,MAAM,CAAC,KAAK,UAAU,QAAQ,CAAC,KAAe,EAAE,UAAyB,EAAE;IACzE,qGAAqG;IACrG,MAAM,UAAU,GAAG,OAAO,CAAC,GAAG,CAAC,qBAAqB,KAAK,GAAG,IAAI,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,yBAAyB,CAAC,CAAA;IAC9G,MAAM,SAAS,GAAG,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,0BAA0B,IAAI,MAAM,EAAE,EAAE,CAAC,CAAA;IACvF,IAAI,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,yBAAyB,CAAA;IACpD,MAAM,WAAW,GAAG,UAAU,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,aAAa,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAA;IAC7F,IAAI,WAAW,EAAE,CAAC;QAChB,QAAQ,GAAG,WAAW,CAAC,GAAG,CAAA;IAC5B,CAAC;IAED,MAAM,WAAW,GAAiB,EAAE,CAAA;IAEpC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,EAAE,OAAO,EAAE,EAAE,UAAU,EAAE,QAAQ,EAAE,CAAC,CAAA;QACrE,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;IAC1B,CAAC;IAED,IAAI,WAAW,EAAE,CAAC;QAChB,MAAM,WAAW,CAAC,IAAI,EAAE,CAAA;IAC1B,CAAC;IAED,OAAO,WAAW,CAAA;AACpB,CAAC;AAED,KAAK,UAAU,OAAO,CAAC,IAAY,EAAE,OAAsB,EAAE,QAAsE;IACjI,MAAM,EAAE,KAAK,GAAG,EAAE,EAAE,GAAG,OAAO,CAAA;IAE9B,uDAAuD;IACvD,aAAa,EAAE,CAAA;IAEf,wEAAwE;IACxE,MAAM,YAAY,GAAG,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC;QAC7C,CAAC,CAAC,IAAI;QACN,CAAC,CAAC,aAAa,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAA;IAE1C,IAAI,YAAY,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,OAAQ,UAAkB,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;QACpF,MAAM,MAAM,CAAC,SAAS,CAAC,CAAA;QACvB,MAAM,MAAM,CAAC,SAAS,CAAC,CAAA;IACzB,CAAC;IAED,MAAM,MAAM,CAAC,YAAY,CAAC,CAAA;IAE1B,MAAM,QAAQ,GAAG,WAAW,EAAE,CAAA;IAC9B,MAAM,OAAO,GAAiB,EAAE,CAAA;IAEhC,uDAAuD;IACvD,IAAI,eAAe,GAAkB,IAAI,CAAA;IACzC,IAAI,gBAAmC,CAAA;IACvC,MAAM,WAAW,GAAG,CAAC,MAAe,EAAE,EAAE;QACtC,IAAI,CAAC,gBAAgB;YAAE,gBAAgB,GAAG,MAAM,YAAY,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAA;IACxG,CAAC,CAAA;IACD,OAAO,CAAC,EAAE,CAAC,oBAAoB,EAAE,WAAW,CAAC,CAAA;IAC7C,OAAO,CAAC,EAAE,CAAC,mBAAmB,EAAE,WAAW,CAAC,CAAA;IAE5C,yBAAyB;IACzB,KAAK,MAAM,IAAI,IAAI,QAAQ,CAAC,cAAc,EAAE,CAAC;QAC3C,MAAM,IAAI,EAAE,CAAA;IACd,CAAC;IAED,oCAAoC;IACpC,KAAK,MAAM,KAAK,IAAI,QAAQ,CAAC,KAAK,EAAE,CAAC;QACnC,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,GAAG,iBAAiB,EAAE,CAAA;QACjD,MAAM,OAAO,GAAG,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,IAAI,CAAA;QACzD,eAAe,CAAC,OAAO,CAAC,KAAK,CAAC,CAAA;QAC9B,IAAI,OAAO,EAAE,CAAC;YACZ,OAAO,CAAC,GAAG,CAAC,oBAAoB,GAAG,OAAO,CAAA;QAC5C,CAAC;QAED,IAAI,KAAK,CAAC,WAAW,EAAE,CAAC;YACtB,MAAM,KAAK,CAAC,WAAW,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;QACrC,CAAC;QAED,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAA;QAC5B,IAAI,MAAM,GAAG,KAAK,CAAA;QAClB,IAAI,KAAwB,CAAA;QAE5B,8DAA8D;QAC9D,gBAAgB,GAAG,SAAS,CAAA;QAC5B,eAAe,GAAG,KAAK,CAAC,IAAI,CAAA;QAE5B,eAAe,CAAC,OAAO,CAAC,KAAK,CAAC,CAAA;QAC9B,IAAI,CAAC;YACH,KAAK,MAAM,IAAI,IAAI,QAAQ,CAAC,eAAe,EAAE,CAAC;gBAC5C,MAAM,IAAI,EAAE,CAAA;YACd,CAAC;YAED,MAAM,KAAK,CAAC,EAAE,CAAC,OAAO,CAAC,CAAA;YACvB,MAAM,GAAG,IAAI,CAAA;QACf,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,KAAK,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAA;QAC7D,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC;gBACH,KAAK,MAAM,IAAI,IAAI,QAAQ,CAAC,cAAc,EAAE,CAAC;oBAC3C,MAAM,IAAI,EAAE,CAAA;gBACd,CAAC;YACH,CAAC;YAAC,OAAO,QAAQ,EAAE,CAAC;gBAClB,IAAI,CAAC,KAAK,EAAE,CAAC;oBACX,KAAK,GAAG,QAAQ,YAAY,KAAK,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAA;oBAC1E,MAAM,GAAG,KAAK,CAAA;gBAChB,CAAC;YACH,CAAC;YAED,eAAe,CAAC,SAAS,CAAC,CAAA;YAC1B,IAAI,CAAC,KAAK,IAAI,gBAAgB,EAAE,CAAC;gBAC/B,KAAK,GAAG,gBAAgB,CAAA;gBACxB,MAAM,GAAG,KAAK,CAAA;YAChB,CAAC;YACD,eAAe,GAAG,IAAI,CAAA;QACxB,CAAC;QAED,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAA;QAEzC,IAAI,KAAK,CAAC,YAAY,EAAE,CAAC;YACvB,MAAM,KAAK,CAAC,YAAY,CAAC,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,UAAU,EAAE,KAAK,CAAC,CAAA;QACjE,CAAC;QAED,IAAI,KAAK,CAAC,eAAe,EAAE,CAAC;YAC1B,MAAM,KAAK,CAAC,eAAe,CAAC,KAAK,CAAC,IAAI,EAAE,OAAO,CAAC,KAAK,CAAC,CAAA;QACxD,CAAC;QAED,4EAA4E;QAC5E,IAAI,OAAO,IAAI,QAAQ,CAAC,QAAQ,EAAE,CAAC;YACjC,IAAI,CAAC;gBACH,MAAM,QAAQ,GAAG,MAAM,kBAAkB,CAAC,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAA;gBACrE,KAAK,MAAM,IAAI,IAAI,QAAQ,EAAE,CAAC;oBAC5B,OAAO,CAAC,KAAK,CAAC,aAAa,CAAC,IAAI,CAAC,CAAA;gBACnC,CAAC;YACH,CAAC;YAAC,OAAO,QAAQ,EAAE,CAAC;gBAClB,oCAAoC;gBACpC,sCAAsC;gBACtC,OAAO,CAAC,IAAI,CAAC,qDAAqD,EAAE,QAAQ,CAAC,CAAA;YAC/E,CAAC;QACH,CAAC;QAED,QAAQ,EAAE,CAAA;QACV,eAAe,CAAC,SAAS,CAAC,CAAA;QAC1B,IAAI,OAAO,EAAE,CAAC;YACZ,OAAO,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAA;QACzC,CAAC;QAED,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC,CAAA;IAC/D,CAAC;IAED,wBAAwB;IACxB,KAAK,MAAM,IAAI,IAAI,QAAQ,CAAC,aAAa,EAAE,CAAC;QAC1C,MAAM,IAAI,EAAE,CAAA;IACd,CAAC;IAED,0BAA0B;IAC1B,OAAO,CAAC,GAAG,CAAC,oBAAoB,EAAE,WAAW,CAAC,CAAA;IAC9C,OAAO,CAAC,GAAG,CAAC,mBAAmB,EAAE,WAAW,CAAC,CAAA;IAE7C,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,CAAA;AAC1B,CAAC"}

package/dist/socket-connector.d.ts

@@ -0,0 +1,22 @@
+import { Socket } from 'socket.io-client';
+export interface SocketConnectorOptions {
+    serverUrl: string;
+    apiKey?: string;
+    sessionId: string;
+}
+/**
+ * Establish a persistent socket.io connection to the backend.
+ *
+ * The SDK registers its sessionId and available tools/workflows
+ * on connect. The backend can then push portal tasks and trigger signals
+ * to this SDK instance for execution.
+ *
+ * Auto-reconnects on disconnect with exponential backoff (1s–30s).
+ * Safe to call multiple times — subsequent calls return the existing socket.
+ */
+export declare function connectToBackend(options: SocketConnectorOptions): Socket;
+/**
+ * Disconnect from the backend and clean up the socket.
+ */
+export declare function disconnectFromBackend(): Promise<void>;
+//# sourceMappingURL=socket-connector.d.ts.map

package/dist/socket-connector.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"socket-connector.d.ts","sourceRoot":"","sources":["../src/socket-connector.ts"],"names":[],"mappings":"AAAA,OAAO,EAAM,MAAM,EAAE,MAAM,kBAAkB,CAAA;AAQ7C,MAAM,WAAW,sBAAsB;IACrC,SAAS,EAAE,MAAM,CAAA;IACjB,MAAM,CAAC,EAAE,MAAM,CAAA;IACf,SAAS,EAAE,MAAM,CAAA;CAClB;AAID;;;;;;;;;GASG;AACH,wBAAgB,gBAAgB,CAAC,OAAO,EAAE,sBAAsB,GAAG,MAAM,CAgFxE;AAED;;GAEG;AACH,wBAAsB,qBAAqB,IAAI,OAAO,CAAC,IAAI,CAAC,CAK3D"}

package/dist/socket-connector.js

@@ -0,0 +1,104 @@
+import { io } from 'socket.io-client';
+import { executePortalTask } from './portal-executor.js';
+import { executeTrigger } from './trigger-executor.js';
+import { scanTools, scanWorkflows } from './execution/tool-runner.js';
+import { debugLog } from './utils/debug.js';
+let socket = null;
+/**
+ * Establish a persistent socket.io connection to the backend.
+ *
+ * The SDK registers its sessionId and available tools/workflows
+ * on connect. The backend can then push portal tasks and trigger signals
+ * to this SDK instance for execution.
+ *
+ * Auto-reconnects on disconnect with exponential backoff (1s–30s).
+ * Safe to call multiple times — subsequent calls return the existing socket.
+ */
+export function connectToBackend(options) {
+    if (socket?.connected)
+        return socket;
+    // If a socket exists but is disconnected/reconnecting, reuse it
+    if (socket)
+        return socket;
+    const { serverUrl, apiKey, sessionId } = options;
+    const cwd = process.cwd();
+    socket = io(serverUrl, {
+        auth: {
+            ...(apiKey ? { apiKey } : {}),
+            sessionId,
+        },
+        transports: ['websocket', 'polling'],
+        reconnection: true,
+        reconnectionDelay: 1000,
+        reconnectionDelayMax: 30_000,
+    });
+    socket.on('connect', () => {
+        debugLog(`[elasticdash] Socket connected: ${socket.id}`);
+        const tools = scanTools(cwd);
+        const workflows = scanWorkflows(cwd);
+        socket.emit('register', {
+            sessionId,
+            tools: tools.map(t => t.name),
+            workflows: workflows.map(w => w.name),
+        });
+    });
+    socket.on('auth:ok', (data) => {
+        debugLog(`[elasticdash] Authenticated for project ${data.projectId}`);
+        socket.emit('join', `observability:project:${data.projectId}`);
+    });
+    socket.on('portal:task', async (task, ack) => {
+        debugLog(`[elasticdash] Socket received portal task: ${task.taskId} type=${task.type} name=${task.name}`);
+        try {
+            const tools = scanTools(cwd);
+            const result = await executePortalTask(task, cwd, tools);
+            if (typeof ack === 'function') {
+                ack(result);
+            }
+            else {
+                socket.emit('portal:result', result);
+            }
+        }
+        catch (e) {
+            const errorResult = {
+                taskId: task.taskId,
+                ok: false,
+                output: null,
+                error: e instanceof Error ? e.message : String(e),
+                durationMs: 0,
+                metadata: task.metadata,
+            };
+            if (typeof ack === 'function') {
+                ack(errorResult);
+            }
+            else {
+                socket.emit('portal:result', errorResult);
+            }
+        }
+    });
+    socket.on('trigger', async (trigger) => {
+        debugLog(`[elasticdash] Socket received trigger: ${trigger.triggerId} steps=${trigger.steps.length} runs=${trigger.runCount}`);
+        try {
+            await executeTrigger(serverUrl, apiKey, trigger);
+        }
+        catch (e) {
+            debugLog(`[elasticdash] Socket trigger execution failed: ${e instanceof Error ? e.message : String(e)}`);
+        }
+    });
+    socket.on('disconnect', (reason) => {
+        debugLog(`[elasticdash] Socket disconnected: ${reason}`);
+    });
+    socket.on('connect_error', (err) => {
+        debugLog(`[elasticdash] Socket connection error: ${err.message}`);
+    });
+    return socket;
+}
+/**
+ * Disconnect from the backend and clean up the socket.
+ */
+export async function disconnectFromBackend() {
+    if (socket) {
+        socket.disconnect();
+        socket = null;
+    }
+}
+//# sourceMappingURL=socket-connector.js.map

package/dist/socket-connector.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"socket-connector.js","sourceRoot":"","sources":["../src/socket-connector.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,EAAE,EAAU,MAAM,kBAAkB,CAAA;AAG7C,OAAO,EAAE,iBAAiB,EAAE,MAAM,sBAAsB,CAAA;AACxD,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAA;AACtD,OAAO,EAAE,SAAS,EAAE,aAAa,EAAE,MAAM,4BAA4B,CAAA;AACrE,OAAO,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAA;AAQ3C,IAAI,MAAM,GAAkB,IAAI,CAAA;AAEhC;;;;;;;;;GASG;AACH,MAAM,UAAU,gBAAgB,CAAC,OAA+B;IAC9D,IAAI,MAAM,EAAE,SAAS;QAAE,OAAO,MAAM,CAAA;IACpC,gEAAgE;IAChE,IAAI,MAAM;QAAE,OAAO,MAAM,CAAA;IAEzB,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,OAAO,CAAA;IAChD,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,EAAE,CAAA;IAEzB,MAAM,GAAG,EAAE,CAAC,SAAS,EAAE;QACrB,IAAI,EAAE;YACJ,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC7B,SAAS;SACV;QACD,UAAU,EAAE,CAAC,WAAW,EAAE,SAAS,CAAC;QACpC,YAAY,EAAE,IAAI;QAClB,iBAAiB,EAAE,IAAI;QACvB,oBAAoB,EAAE,MAAM;KAC7B,CAAC,CAAA;IAEF,MAAM,CAAC,EAAE,CAAC,SAAS,EAAE,GAAG,EAAE;QACxB,QAAQ,CAAC,mCAAmC,MAAO,CAAC,EAAE,EAAE,CAAC,CAAA;QACzD,MAAM,KAAK,GAAG,SAAS,CAAC,GAAG,CAAC,CAAA;QAC5B,MAAM,SAAS,GAAG,aAAa,CAAC,GAAG,CAAC,CAAA;QACpC,MAAO,CAAC,IAAI,CAAC,UAAU,EAAE;YACvB,SAAS;YACT,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;YAC7B,SAAS,EAAE,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;SACtC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,MAAM,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,IAA2B,EAAE,EAAE;QACnD,QAAQ,CAAC,2CAA2C,IAAI,CAAC,SAAS,EAAE,CAAC,CAAA;QACrE,MAAO,CAAC,IAAI,CAAC,MAAM,EAAE,yBAAyB,IAAI,CAAC,SAAS,EAAE,CAAC,CAAA;IACjE,CAAC,CAAC,CAAA;IAEF,MAAM,CAAC,EAAE,CAAC,aAAa,EAAE,KAAK,EAAE,IAAgB,EAAE,GAAwC,EAAE,EAAE;QAC5F,QAAQ,CAAC,8CAA8C,IAAI,CAAC,MAAM,SAAS,IAAI,CAAC,IAAI,SAAS,IAAI,CAAC,IAAI,EAAE,CAAC,CAAA;QACzG,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,SAAS,CAAC,GAAG,CAAC,CAAA;YAC5B,MAAM,MAAM,GAAG,MAAM,iBAAiB,CAAC,IAAI,EAAE,GAAG,EAAE,KAAK,CAAC,CAAA;YACxD,IAAI,OAAO,GAAG,KAAK,UAAU,EAAE,CAAC;gBAC9B,GAAG,CAAC,MAAM,CAAC,CAAA;YACb,CAAC;iBAAM,CAAC;gBACN,MAAO,CAAC,IAAI,CAAC,eAAe,EAAE,MAAM,CAAC,CAAA;YACvC,CAAC;QACH,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,MAAM,WAAW,GAAqB;gBACpC,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,EAAE,EAAE,KAAK;gBACT,MAAM,EAAE,IAAI;gBACZ,KAAK,EAAE,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC;gBACjD,UAAU,EAAE,CAAC;gBACb,QAAQ,EAAE,IAAI,CAAC,QAAQ;aACxB,CAAA;YACD,IAAI,OAAO,GAAG,KAAK,UAAU,EAAE,CAAC;gBAC9B,GAAG,CAAC,WAAW,CAAC,CAAA;YAClB,CAAC;iBAAM,CAAC;gBACN,MAAO,CAAC,IAAI,CAAC,eAAe,EAAE,WAAW,CAAC,CAAA;YAC5C,CAAC;QACH,CAAC;IACH,CAAC,CAAC,CAAA;IAEF,MAAM,CAAC,EAAE,CAAC,SAAS,EAAE,KAAK,EAAE,OAAsB,EAAE,EAAE;QACpD,QAAQ,CAAC,0CAA0C,OAAO,CAAC,SAAS,UAAU,OAAO,CAAC,KAAK,CAAC,MAAM,SAAS,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAA;QAC9H,IAAI,CAAC;YACH,MAAM,cAAc,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QAClD,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,QAAQ,CAAC,kDAAkD,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAA;QAC1G,CAAC;IACH,CAAC,CAAC,CAAA;IAEF,MAAM,CAAC,EAAE,CAAC,YAAY,EAAE,CAAC,MAAM,EAAE,EAAE;QACjC,QAAQ,CAAC,sCAAsC,MAAM,EAAE,CAAC,CAAA;IAC1D,CAAC,CAAC,CAAA;IAEF,MAAM,CAAC,EAAE,CAAC,eAAe,EAAE,CAAC,GAAG,EAAE,EAAE;QACjC,QAAQ,CAAC,0CAA0C,GAAG,CAAC,OAAO,EAAE,CAAC,CAAA;IACnE,CAAC,CAAC,CAAA;IAEF,OAAO,MAAM,CAAA;AACf,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,qBAAqB;IACzC,IAAI,MAAM,EAAE,CAAC;QACX,MAAM,CAAC,UAAU,EAAE,CAAA;QACnB,MAAM,GAAG,IAAI,CAAA;IACf,CAAC;AACH,CAAC"}

package/dist/telemetry-batcher.d.ts

@@ -0,0 +1,56 @@
+import type { WorkflowEvent } from './capture/event.js';
+export interface TriggerStep {
+    eventId: number;
+    eventType: 'ai' | 'tool' | 'http' | 'db';
+    eventName: string;
+    originalEventDbId: number;
+    input: unknown;
+    model?: string;
+    provider?: string;
+}
+export interface FrozenEvent {
+    id: number;
+    type: string;
+    name: string;
+    input: unknown;
+    output: unknown;
+    timestamp: number;
+    durationMs: number | null;
+    streamed?: boolean;
+    streamRaw?: string | null;
+}
+export interface TriggerSignal {
+    triggerId: number;
+    runCount: number;
+    steps: TriggerStep[];
+    /** HTTP/DB events from the same trace to freeze (mock) during step reruns */
+    frozenEvents?: FrozenEvent[];
+}
+export interface TelemetryBatcherOptions {
+    serverUrl: string;
+    apiKey?: string;
+    sessionId: string;
+    metadata?: Record<string, unknown>;
+    batchIntervalMs?: number;
+    maxBatchSize?: number;
+    redactKeys?: string[];
+    onTrigger?: (trigger: TriggerSignal) => Promise<void>;
+}
+export declare class TelemetryBatcher {
+    private buffer;
+    private timer;
+    private readonly serverUrl;
+    private readonly apiKey;
+    private readonly sessionId;
+    private readonly metadata;
+    private readonly maxBatchSize;
+    private readonly redactKeys;
+    private readonly onTrigger;
+    private shuttingDown;
+    constructor(opts: TelemetryBatcherOptions);
+    enqueue(event: WorkflowEvent): void;
+    flush(): Promise<void>;
+    private send;
+    shutdown(): Promise<void>;
+}
+//# sourceMappingURL=telemetry-batcher.d.ts.map

package/dist/telemetry-batcher.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"telemetry-batcher.d.ts","sourceRoot":"","sources":["../src/telemetry-batcher.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAA;AAMvD,MAAM,WAAW,WAAW;IAC1B,OAAO,EAAE,MAAM,CAAA;IACf,SAAS,EAAE,IAAI,GAAG,MAAM,GAAG,MAAM,GAAG,IAAI,CAAA;IACxC,SAAS,EAAE,MAAM,CAAA;IACjB,iBAAiB,EAAE,MAAM,CAAA;IACzB,KAAK,EAAE,OAAO,CAAA;IACd,KAAK,CAAC,EAAE,MAAM,CAAA;IACd,QAAQ,CAAC,EAAE,MAAM,CAAA;CAClB;AAED,MAAM,WAAW,WAAW;IAC1B,EAAE,EAAE,MAAM,CAAA;IACV,IAAI,EAAE,MAAM,CAAA;IACZ,IAAI,EAAE,MAAM,CAAA;IACZ,KAAK,EAAE,OAAO,CAAA;IACd,MAAM,EAAE,OAAO,CAAA;IACf,SAAS,EAAE,MAAM,CAAA;IACjB,UAAU,EAAE,MAAM,GAAG,IAAI,CAAA;IACzB,QAAQ,CAAC,EAAE,OAAO,CAAA;IAClB,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI,CAAA;CAC1B;AAED,MAAM,WAAW,aAAa;IAC5B,SAAS,EAAE,MAAM,CAAA;IACjB,QAAQ,EAAE,MAAM,CAAA;IAChB,KAAK,EAAE,WAAW,EAAE,CAAA;IACpB,6EAA6E;IAC7E,YAAY,CAAC,EAAE,WAAW,EAAE,CAAA;CAC7B;AAED,MAAM,WAAW,uBAAuB;IACtC,SAAS,EAAE,MAAM,CAAA;IACjB,MAAM,CAAC,EAAE,MAAM,CAAA;IACf,SAAS,EAAE,MAAM,CAAA;IACjB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;IAClC,eAAe,CAAC,EAAE,MAAM,CAAA;IACxB,YAAY,CAAC,EAAE,MAAM,CAAA;IACrB,UAAU,CAAC,EAAE,MAAM,EAAE,CAAA;IACrB,SAAS,CAAC,EAAE,CAAC,OAAO,EAAE,aAAa,KAAK,OAAO,CAAC,IAAI,CAAC,CAAA;CACtD;AAED,qBAAa,gBAAgB;IAC3B,OAAO,CAAC,MAAM,CAAsB;IACpC,OAAO,CAAC,KAAK,CAA8C;IAC3D,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAQ;IAClC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAoB;IAC3C,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAQ;IAClC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAqC;IAC9D,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAQ;IACrC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAU;IACrC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAyD;IACnF,OAAO,CAAC,YAAY,CAAQ;gBAEhB,IAAI,EAAE,uBAAuB;IAiBzC,OAAO,CAAC,KAAK,EAAE,aAAa,GAAG,IAAI;IAW7B,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;YAMd,IAAI;IAuEZ,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;CAShC"}

package/dist/telemetry-batcher.js

@@ -0,0 +1,143 @@
+import { Readable } from 'node:stream';
+import { randomUUID } from 'node:crypto';
+import { getOriginalFetch } from './interceptors/http.js';
+import { debugLog } from './utils/debug.js';
+import { notifyLicenseError } from './utils/license-error.js';
+import { redactPayload } from './utils/redact.js';
+export class TelemetryBatcher {
+    buffer = [];
+    timer = null;
+    serverUrl;
+    apiKey;
+    sessionId;
+    metadata;
+    maxBatchSize;
+    redactKeys;
+    onTrigger;
+    shuttingDown = false;
+    constructor(opts) {
+        this.serverUrl = opts.serverUrl.replace(/\/$/, '');
+        this.apiKey = opts.apiKey;
+        this.sessionId = opts.sessionId;
+        this.metadata = opts.metadata;
+        this.maxBatchSize = opts.maxBatchSize ?? 50;
+        this.redactKeys = opts.redactKeys ?? [];
+        this.onTrigger = opts.onTrigger;
+        const intervalMs = opts.batchIntervalMs ?? 2000;
+        this.timer = setInterval(() => { this.flush().catch(() => { }); }, intervalMs);
+        // Allow the process to exit even if the timer is still scheduled
+        if (this.timer && typeof this.timer === 'object' && 'unref' in this.timer) {
+            this.timer.unref();
+        }
+    }
+    enqueue(event) {
+        if (this.shuttingDown)
+            return;
+        const redacted = this.redactKeys.length > 0
+            ? { ...event, input: redactPayload(event.input, this.redactKeys), output: redactPayload(event.output, this.redactKeys) }
+            : event;
+        this.buffer.push(redacted);
+        if (this.buffer.length >= this.maxBatchSize) {
+            this.flush().catch(() => { });
+        }
+    }
+    async flush() {
+        if (this.buffer.length === 0)
+            return;
+        const batch = this.buffer.splice(0, this.buffer.length);
+        await this.send(batch, 0);
+    }
+    async send(batch, attempt) {
+        if (!this.serverUrl) {
+            debugLog(`[elasticdash] Dropping ${batch.length} events: serverUrl is empty`);
+            return;
+        }
+        const maxRetries = 3;
+        const url = `${this.serverUrl}/api/observability/events`;
+        const headers = {
+            'Content-Type': 'application/json',
+            'X-Correlation-ID': randomUUID(),
+        };
+        if (this.apiKey)
+            headers['Authorization'] = `Bearer ${this.apiKey}`;
+        try {
+            // Stream the JSON body to avoid buffering large payloads (e.g. events
+            // with streamRaw can exceed 100 MB). Node.js fetch (undici) supports
+            // Readable as body and uses chunked Transfer-Encoding automatically.
+            const body = Readable.from(jsonStreamEvents(this.sessionId, batch, this.metadata));
+            const res = await getOriginalFetch()(url, {
+                method: 'POST',
+                headers,
+                body,
+                duplex: 'half',
+            });
+            if (res.status === 402) {
+                notifyLicenseError(res.status, 'telemetry');
+                debugLog(`[elasticdash] Dropping ${batch.length} events: no available license (402)`);
+                return;
+            }
+            if (res.status === 429 || res.status >= 500) {
+                if (attempt < maxRetries) {
+                    const delayMs = Math.pow(2, attempt) * 1000;
+                    debugLog(`[elasticdash] Telemetry flush failed (${res.status}), retrying in ${delayMs}ms (attempt ${attempt + 1}/${maxRetries})`);
+                    await new Promise((r) => setTimeout(r, delayMs));
+                    return this.send(batch, attempt + 1);
+                }
+                debugLog(`[elasticdash] Dropping ${batch.length} events after ${maxRetries} retries (last status: ${res.status})`);
+                return;
+            }
+            debugLog(`[elasticdash] Flushed ${batch.length} events (status ${res.status})`);
+            // Parse response for trigger signal
+            if (res.ok && this.onTrigger) {
+                try {
+                    const body = await res.json();
+                    if (body.trigger && typeof body.trigger.triggerId === 'number' && Array.isArray(body.trigger.steps)) {
+                        debugLog(`[elasticdash] Trigger received: id=${body.trigger.triggerId} steps=${body.trigger.steps.length} runCount=${body.trigger.runCount}`);
+                        // Fire-and-forget — don't block the flush pipeline
+                        this.onTrigger(body.trigger).catch((err) => {
+                            debugLog(`[elasticdash] Trigger execution failed: ${err instanceof Error ? err.message : String(err)}`);
+                        });
+                    }
+                }
+                catch {
+                    // Response parsing failed — not critical, ignore
+                }
+            }
+        }
+        catch (err) {
+            if (attempt < maxRetries) {
+                const delayMs = Math.pow(2, attempt) * 1000;
+                debugLog(`[elasticdash] Telemetry flush error, retrying in ${delayMs}ms (attempt ${attempt + 1}/${maxRetries})`);
+                await new Promise((r) => setTimeout(r, delayMs));
+                return this.send(batch, attempt + 1);
+            }
+            debugLog(`[elasticdash] Dropping ${batch.length} events after ${maxRetries} retries: ${err instanceof Error ? err.message : String(err)}`);
+        }
+    }
+    async shutdown() {
+        if (this.shuttingDown)
+            return;
+        this.shuttingDown = true;
+        if (this.timer) {
+            clearInterval(this.timer);
+            this.timer = null;
+        }
+        await this.flush();
+    }
+}
+/**
+ * Yields JSON fragments for the events payload without building a single
+ * giant string. Each event is serialized individually so the process never
+ * holds the full serialized body in memory at once.
+ */
+function* jsonStreamEvents(sessionId, events, metadata) {
+    const metaPart = metadata ? `,"metadata":${JSON.stringify(metadata)}` : '';
+    yield `{"sessionId":${JSON.stringify(sessionId)}${metaPart},"events":[`;
+    for (let i = 0; i < events.length; i++) {
+        if (i > 0)
+            yield ',';
+        yield JSON.stringify(events[i]);
+    }
+    yield ']}';
+}
+//# sourceMappingURL=telemetry-batcher.js.map

package/dist/telemetry-batcher.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"telemetry-batcher.js","sourceRoot":"","sources":["../src/telemetry-batcher.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAA;AACtC,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAA;AAExC,OAAO,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAA;AACzD,OAAO,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAA;AAC3C,OAAO,EAAE,kBAAkB,EAAE,MAAM,0BAA0B,CAAA;AAC7D,OAAO,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAA;AA2CjD,MAAM,OAAO,gBAAgB;IACnB,MAAM,GAAoB,EAAE,CAAA;IAC5B,KAAK,GAA0C,IAAI,CAAA;IAC1C,SAAS,CAAQ;IACjB,MAAM,CAAoB;IAC1B,SAAS,CAAQ;IACjB,QAAQ,CAAqC;IAC7C,YAAY,CAAQ;IACpB,UAAU,CAAU;IACpB,SAAS,CAAyD;IAC3E,YAAY,GAAG,KAAK,CAAA;IAE5B,YAAY,IAA6B;QACvC,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAA;QAClD,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAA;QACzB,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,SAAS,CAAA;QAC/B,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAA;QAC7B,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,YAAY,IAAI,EAAE,CAAA;QAC3C,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,UAAU,IAAI,EAAE,CAAA;QACvC,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,SAAS,CAAA;QAE/B,MAAM,UAAU,GAAG,IAAI,CAAC,eAAe,IAAI,IAAI,CAAA;QAC/C,IAAI,CAAC,KAAK,GAAG,WAAW,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAA,CAAC,CAAC,EAAE,UAAU,CAAC,CAAA;QAC5E,iEAAiE;QACjE,IAAI,IAAI,CAAC,KAAK,IAAI,OAAO,IAAI,CAAC,KAAK,KAAK,QAAQ,IAAI,OAAO,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YAC1E,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAA;QACpB,CAAC;IACH,CAAC;IAED,OAAO,CAAC,KAAoB;QAC1B,IAAI,IAAI,CAAC,YAAY;YAAE,OAAM;QAC7B,MAAM,QAAQ,GAAG,IAAI,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC;YACzC,CAAC,CAAC,EAAE,GAAG,KAAK,EAAE,KAAK,EAAE,aAAa,CAAC,KAAK,CAAC,KAAK,EAAE,IAAI,CAAC,UAAU,CAAC,EAAE,MAAM,EAAE,aAAa,CAAC,KAAK,CAAC,MAAM,EAAE,IAAI,CAAC,UAAU,CAAC,EAAE;YACxH,CAAC,CAAC,KAAK,CAAA;QACT,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,QAAyB,CAAC,CAAA;QAC3C,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YAC5C,IAAI,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAA;QAC9B,CAAC;IACH,CAAC;IAED,KAAK,CAAC,KAAK;QACT,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC;YAAE,OAAM;QACpC,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAA;QACvD,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,CAAA;IAC3B,CAAC;IAEO,KAAK,CAAC,IAAI,CAAC,KAAsB,EAAE,OAAe;QACxD,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,QAAQ,CAAC,0BAA0B,KAAK,CAAC,MAAM,6BAA6B,CAAC,CAAA;YAC7E,OAAM;QACR,CAAC;QACD,MAAM,UAAU,GAAG,CAAC,CAAA;QACpB,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,SAAS,2BAA2B,CAAA;QACxD,MAAM,OAAO,GAA2B;YACtC,cAAc,EAAE,kBAAkB;YAClC,kBAAkB,EAAE,UAAU,EAAE;SACjC,CAAA;QACD,IAAI,IAAI,CAAC,MAAM;YAAE,OAAO,CAAC,eAAe,CAAC,GAAG,UAAU,IAAI,CAAC,MAAM,EAAE,CAAA;QAEnE,IAAI,CAAC;YACH,sEAAsE;YACtE,qEAAqE;YACrE,qEAAqE;YACrE,MAAM,IAAI,GAAG,QAAQ,CAAC,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAA;YAElF,MAAM,GAAG,GAAG,MAAM,gBAAgB,EAAE,CAAC,GAAG,EAAE;gBACxC,MAAM,EAAE,MAAM;gBACd,OAAO;gBACP,IAAI;gBACJ,MAAM,EAAE,MAAM;aACf,CAAC,CAAA;YAEF,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBACvB,kBAAkB,CAAC,GAAG,CAAC,MAAM,EAAE,WAAW,CAAC,CAAA;gBAC3C,QAAQ,CAAC,0BAA0B,KAAK,CAAC,MAAM,qCAAqC,CAAC,CAAA;gBACrF,OAAM;YACR,CAAC;YAED,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,IAAI,GAAG,CAAC,MAAM,IAAI,GAAG,EAAE,CAAC;gBAC5C,IAAI,OAAO,GAAG,UAAU,EAAE,CAAC;oBACzB,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC,GAAG,IAAI,CAAA;oBAC3C,QAAQ,CAAC,yCAAyC,GAAG,CAAC,MAAM,kBAAkB,OAAO,eAAe,OAAO,GAAG,CAAC,IAAI,UAAU,GAAG,CAAC,CAAA;oBACjI,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC,CAAA;oBAChD,OAAO,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,OAAO,GAAG,CAAC,CAAC,CAAA;gBACtC,CAAC;gBACD,QAAQ,CAAC,0BAA0B,KAAK,CAAC,MAAM,iBAAiB,UAAU,0BAA0B,GAAG,CAAC,MAAM,GAAG,CAAC,CAAA;gBAClH,OAAM;YACR,CAAC;YAED,QAAQ,CAAC,yBAAyB,KAAK,CAAC,MAAM,mBAAmB,GAAG,CAAC,MAAM,GAAG,CAAC,CAAA;YAE/E,oCAAoC;YACpC,IAAI,GAAG,CAAC,EAAE,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;gBAC7B,IAAI,CAAC;oBACH,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,IAAI,EAAiC,CAAA;oBAC5D,IAAI,IAAI,CAAC,OAAO,IAAI,OAAO,IAAI,CAAC,OAAO,CAAC,SAAS,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;wBACpG,QAAQ,CAAC,sCAAsC,IAAI,CAAC,OAAO,CAAC,SAAS,UAAU,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,MAAM,aAAa,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAA;wBAC7I,mDAAmD;wBACnD,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;4BACzC,QAAQ,CAAC,2CAA2C,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAA;wBACzG,CAAC,CAAC,CAAA;oBACJ,CAAC;gBACH,CAAC;gBAAC,MAAM,CAAC;oBACP,iDAAiD;gBACnD,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,OAAO,GAAG,UAAU,EAAE,CAAC;gBACzB,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC,GAAG,IAAI,CAAA;gBAC3C,QAAQ,CAAC,oDAAoD,OAAO,eAAe,OAAO,GAAG,CAAC,IAAI,UAAU,GAAG,CAAC,CAAA;gBAChH,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC,CAAA;gBAChD,OAAO,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,OAAO,GAAG,CAAC,CAAC,CAAA;YACtC,CAAC;YACD,QAAQ,CAAC,0BAA0B,KAAK,CAAC,MAAM,iBAAiB,UAAU,aAAa,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAA;QAC5I,CAAC;IACH,CAAC;IAED,KAAK,CAAC,QAAQ;QACZ,IAAI,IAAI,CAAC,YAAY;YAAE,OAAM;QAC7B,IAAI,CAAC,YAAY,GAAG,IAAI,CAAA;QACxB,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACf,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;YACzB,IAAI,CAAC,KAAK,GAAG,IAAI,CAAA;QACnB,CAAC;QACD,MAAM,IAAI,CAAC,KAAK,EAAE,CAAA;IACpB,CAAC;CACF;AAED;;;;GAIG;AACH,QAAQ,CAAC,CAAC,gBAAgB,CAAC,SAAiB,EAAE,MAAuB,EAAE,QAAkC;IACvG,MAAM,QAAQ,GAAG,QAAQ,CAAC,CAAC,CAAC,eAAe,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAA;IAC1E,MAAM,gBAAgB,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,GAAG,QAAQ,aAAa,CAAA;IACvE,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACvC,IAAI,CAAC,GAAG,CAAC;YAAE,MAAM,GAAG,CAAA;QACpB,MAAM,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAA;IACjC,CAAC;IACD,MAAM,IAAI,CAAA;AACZ,CAAC"}

package/dist/test-setup.d.ts

@@ -0,0 +1,12 @@
+/**
+ * Import this file in your AI test files to get:
+ *   - aiTest / beforeAll / afterAll / beforeEach / afterEach available as globals (TypeScript types + runtime)
+ *   - Custom matcher types (toHaveLLMStep, toCallTool, toMatchSemanticOutput)
+ *
+ * The CLI registers matchers at startup, so this import is for TypeScript
+ * type awareness only — no double-registration occurs at runtime.
+ */
+import './core/registry.js';
+import './matchers/index.js';
+export {};
+//# sourceMappingURL=test-setup.d.ts.map

package/dist/test-setup.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"test-setup.d.ts","sourceRoot":"","sources":["../src/test-setup.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAGH,OAAO,oBAAoB,CAAA;AAG3B,OAAO,qBAAqB,CAAA;AAE5B,OAAO,EAAE,CAAA"}

package/dist/test-setup.js

@@ -0,0 +1,13 @@
+/**
+ * Import this file in your AI test files to get:
+ *   - aiTest / beforeAll / afterAll / beforeEach / afterEach available as globals (TypeScript types + runtime)
+ *   - Custom matcher types (toHaveLLMStep, toCallTool, toMatchSemanticOutput)
+ *
+ * The CLI registers matchers at startup, so this import is for TypeScript
+ * type awareness only — no double-registration occurs at runtime.
+ */
+// Side-effect: populates globalThis.aiTest + brings declare global into scope
+import './core/registry.js';
+// Side-effect: brings declare module 'expect' augmentation into scope
+import './matchers/index.js';
+//# sourceMappingURL=test-setup.js.map

package/dist/test-setup.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"test-setup.js","sourceRoot":"","sources":["../src/test-setup.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,8EAA8E;AAC9E,OAAO,oBAAoB,CAAA;AAE3B,sEAAsE;AACtE,OAAO,qBAAqB,CAAA"}

package/dist/tool-registry.d.ts

@@ -0,0 +1,31 @@
+/**
+ * Global registry for tools defined via edTool().
+ *
+ * Tools registered here are discoverable for reruns regardless of which
+ * module they live in. The helper also applies wrapTool() so telemetry
+ * fires automatically — symmetric with the Python @ed_tool decorator.
+ */
+export interface RegisteredTool {
+    name: string;
+    fn: (...args: unknown[]) => unknown | Promise<unknown>;
+    wrapped: (...args: unknown[]) => Promise<unknown>;
+    isAsync: boolean;
+    signature: string;
+    sourceFile: string | null;
+    lineNumber: number | null;
+}
+export declare function getRegisteredTools(): RegisteredTool[];
+export declare function getRegisteredTool(name: string): RegisteredTool | undefined;
+export declare function clearToolRegistry(): void;
+/**
+ * Register a function as a rerunnable tool.
+ *
+ *   export const myTool = edTool('my_tool', async (query: string) => { ... })
+ *
+ * The returned function is the telemetry-wrapped version, so calling it from
+ * normal code paths still produces traces. The CLI `run-tool` command and the
+ * ElasticDash MCP `run_tool` tool will resolve the original by name.
+ */
+export declare function edTool<Args extends unknown[], R>(name: string, fn: (...args: Args) => R | Promise<R>): (...args: Args) => Promise<R>;
+export declare const defineTool: typeof edTool;
+//# sourceMappingURL=tool-registry.d.ts.map

package/dist/tool-registry.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-registry.d.ts","sourceRoot":"","sources":["../src/tool-registry.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAIH,MAAM,WAAW,cAAc;IAC7B,IAAI,EAAE,MAAM,CAAA;IACZ,EAAE,EAAE,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,CAAA;IACtD,OAAO,EAAE,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,OAAO,CAAC,OAAO,CAAC,CAAA;IACjD,OAAO,EAAE,OAAO,CAAA;IAChB,SAAS,EAAE,MAAM,CAAA;IACjB,UAAU,EAAE,MAAM,GAAG,IAAI,CAAA;IACzB,UAAU,EAAE,MAAM,GAAG,IAAI,CAAA;CAC1B;AAID,wBAAgB,kBAAkB,IAAI,cAAc,EAAE,CAErD;AAED,wBAAgB,iBAAiB,CAAC,IAAI,EAAE,MAAM,GAAG,cAAc,GAAG,SAAS,CAE1E;AAED,wBAAgB,iBAAiB,IAAI,IAAI,CAExC;AA6BD;;;;;;;;GAQG;AACH,wBAAgB,MAAM,CAAC,IAAI,SAAS,OAAO,EAAE,EAAE,CAAC,EAC9C,IAAI,EAAE,MAAM,EACZ,EAAE,EAAE,CAAC,GAAG,IAAI,EAAE,IAAI,KAAK,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,GACpC,CAAC,GAAG,IAAI,EAAE,IAAI,KAAK,OAAO,CAAC,CAAC,CAAC,CAkB/B;AAED,eAAO,MAAM,UAAU,eAAS,CAAA"}

package/dist/tool-registry.js

@@ -0,0 +1,73 @@
+/**
+ * Global registry for tools defined via edTool().
+ *
+ * Tools registered here are discoverable for reruns regardless of which
+ * module they live in. The helper also applies wrapTool() so telemetry
+ * fires automatically — symmetric with the Python @ed_tool decorator.
+ */
+import { wrapTool } from './interceptors/tool.js';
+const _registry = new Map();
+export function getRegisteredTools() {
+    return Array.from(_registry.values());
+}
+export function getRegisteredTool(name) {
+    return _registry.get(name);
+}
+export function clearToolRegistry() {
+    _registry.clear();
+}
+function inferCallerLocation() {
+    const stack = new Error().stack;
+    if (!stack)
+        return { file: null, line: null };
+    const lines = stack.split('\n');
+    for (let i = 2; i < lines.length; i++) {
+        const m = lines[i].match(/\((.*?):(\d+):\d+\)|at\s+(.*?):(\d+):\d+/);
+        if (!m)
+            continue;
+        const file = m[1] ?? m[3];
+        const line = parseInt(m[2] ?? m[4], 10);
+        if (file && !file.includes('tool-registry')) {
+            return { file: file.replace(/^file:\/\//, ''), line: isNaN(line) ? null : line };
+        }
+    }
+    return { file: null, line: null };
+}
+function inferSignature(fn) {
+    const src = fn.toString();
+    const m = src.match(/^[^(]*\(([^)]*)\)/);
+    if (!m)
+        return '()';
+    const params = m[1]
+        .split(',')
+        .map(p => p.trim().split(/[\s=:]/)[0])
+        .filter(Boolean);
+    return `(${params.join(', ')})`;
+}
+/**
+ * Register a function as a rerunnable tool.
+ *
+ *   export const myTool = edTool('my_tool', async (query: string) => { ... })
+ *
+ * The returned function is the telemetry-wrapped version, so calling it from
+ * normal code paths still produces traces. The CLI `run-tool` command and the
+ * ElasticDash MCP `run_tool` tool will resolve the original by name.
+ */
+export function edTool(name, fn) {
+    const isAsync = fn.constructor.name === 'AsyncFunction';
+    const signature = inferSignature(fn);
+    const { file, line } = inferCallerLocation();
+    const wrapped = wrapTool(name, fn);
+    _registry.set(name, {
+        name,
+        fn: fn,
+        wrapped: wrapped,
+        isAsync,
+        signature,
+        sourceFile: file,
+        lineNumber: line,
+    });
+    return wrapped;
+}
+export const defineTool = edTool;
+//# sourceMappingURL=tool-registry.js.map

package/dist/tool-registry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-registry.js","sourceRoot":"","sources":["../src/tool-registry.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,EAAE,QAAQ,EAAE,MAAM,wBAAwB,CAAA;AAYjD,MAAM,SAAS,GAAgC,IAAI,GAAG,EAAE,CAAA;AAExD,MAAM,UAAU,kBAAkB;IAChC,OAAO,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC,CAAA;AACvC,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,IAAY;IAC5C,OAAO,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,CAAA;AAC5B,CAAC;AAED,MAAM,UAAU,iBAAiB;IAC/B,SAAS,CAAC,KAAK,EAAE,CAAA;AACnB,CAAC;AAED,SAAS,mBAAmB;IAC1B,MAAM,KAAK,GAAG,IAAI,KAAK,EAAE,CAAC,KAAK,CAAA;IAC/B,IAAI,CAAC,KAAK;QAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,CAAA;IAC7C,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;IAC/B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACtC,MAAM,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,0CAA0C,CAAC,CAAA;QACpE,IAAI,CAAC,CAAC;YAAE,SAAQ;QAChB,MAAM,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAA;QACzB,MAAM,IAAI,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAA;QACvC,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,eAAe,CAAC,EAAE,CAAC;YAC5C,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE,EAAE,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,EAAE,CAAA;QAClF,CAAC;IACH,CAAC;IACD,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,CAAA;AACnC,CAAC;AAED,SAAS,cAAc,CAAC,EAAY;IAClC,MAAM,GAAG,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAA;IACzB,MAAM,CAAC,GAAG,GAAG,CAAC,KAAK,CAAC,mBAAmB,CAAC,CAAA;IACxC,IAAI,CAAC,CAAC;QAAE,OAAO,IAAI,CAAA;IACnB,MAAM,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC;SAChB,KAAK,CAAC,GAAG,CAAC;SACV,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;SACrC,MAAM,CAAC,OAAO,CAAC,CAAA;IAClB,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAA;AACjC,CAAC;AAED;;;;;;;;GAQG;AACH,MAAM,UAAU,MAAM,CACpB,IAAY,EACZ,EAAqC;IAErC,MAAM,OAAO,GAAG,EAAE,CAAC,WAAW,CAAC,IAAI,KAAK,eAAe,CAAA;IACvD,MAAM,SAAS,GAAG,cAAc,CAAC,EAAE,CAAC,CAAA;IACpC,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,mBAAmB,EAAE,CAAA;IAE5C,MAAM,OAAO,GAAG,QAAQ,CAAC,IAAI,EAAE,EAAwC,CAA6C,CAAA;IAEpH,SAAS,CAAC,GAAG,CAAC,IAAI,EAAE;QAClB,IAAI;QACJ,EAAE,EAAE,EAA0B;QAC9B,OAAO,EAAE,OAAoC;QAC7C,OAAO;QACP,SAAS;QACT,UAAU,EAAE,IAAI;QAChB,UAAU,EAAE,IAAI;KACjB,CAAC,CAAA;IAEF,OAAO,OAAO,CAAA;AAChB,CAAC;AAED,MAAM,CAAC,MAAM,UAAU,GAAG,MAAM,CAAA"}