elasticdash-test 0.1.18-alpha-6 → 0.1.18-alpha-8

package/dist/index.cjs

@@ -875,15 +875,15 @@ function normaliseMockResult(value) {
   return value;
 }
 function resolveMock(toolName) {
-  const g4 = globalThis;
-  const mocks = g4["__ELASTICDASH_TOOL_MOCKS__"];
+  const g5 = globalThis;
+  const mocks = g5["__ELASTICDASH_TOOL_MOCKS__"];
   if (!mocks) return { mocked: false };
   const entry = mocks[toolName];
   if (!entry || entry.mode === "live") return { mocked: false };
-  if (!g4["__ELASTICDASH_TOOL_CALL_COUNTERS__"]) {
-    g4["__ELASTICDASH_TOOL_CALL_COUNTERS__"] = {};
+  if (!g5["__ELASTICDASH_TOOL_CALL_COUNTERS__"]) {
+    g5["__ELASTICDASH_TOOL_CALL_COUNTERS__"] = {};
   }
-  const counters = g4["__ELASTICDASH_TOOL_CALL_COUNTERS__"];
+  const counters = g5["__ELASTICDASH_TOOL_CALL_COUNTERS__"];
   counters[toolName] = (counters[toolName] ?? 0) + 1;
   const callNumber = counters[toolName];
   if (entry.mode === "mock-all") {
@@ -946,8 +946,8 @@ function replaceSystemPrompt(input, newSystemPrompt) {
   return input;
 }
 function resolvePromptMock(input) {
-  const g4 = globalThis;
-  const mocks = g4["__ELASTICDASH_PROMPT_MOCKS__"];
+  const g5 = globalThis;
+  const mocks = g5["__ELASTICDASH_PROMPT_MOCKS__"];
   if (!mocks || Object.keys(mocks).length === 0) return void 0;
   const systemPrompt = extractSystemPrompt(input);
   if (systemPrompt === void 0) return void 0;
@@ -958,10 +958,10 @@ function resolvePromptMock(input) {
     return replaceSystemPrompt(input, entry);
   }
   if (entry.mode === "live") return void 0;
-  if (!g4["__ELASTICDASH_PROMPT_CALL_COUNTERS__"]) {
-    g4["__ELASTICDASH_PROMPT_CALL_COUNTERS__"] = {};
+  if (!g5["__ELASTICDASH_PROMPT_CALL_COUNTERS__"]) {
+    g5["__ELASTICDASH_PROMPT_CALL_COUNTERS__"] = {};
   }
-  const counters = g4["__ELASTICDASH_PROMPT_CALL_COUNTERS__"];
+  const counters = g5["__ELASTICDASH_PROMPT_CALL_COUNTERS__"];
   counters[match.key] = (counters[match.key] ?? 0) + 1;
   const callNumber = counters[match.key];
   if (entry.mode === "replace-all") {
@@ -1024,15 +1024,15 @@ function replaceUserPrompt(input, originalText, newText) {
   return input;
 }
 function resolveUserPromptMock(input) {
-  const g4 = globalThis;
-  const mocks = g4["__ELASTICDASH_USER_PROMPT_MOCKS__"];
+  const g5 = globalThis;
+  const mocks = g5["__ELASTICDASH_USER_PROMPT_MOCKS__"];
   if (!mocks || Object.keys(mocks).length === 0) return void 0;
   const userPrompts = extractUserPrompts(input);
   if (userPrompts.length === 0) return void 0;
-  if (!g4["__ELASTICDASH_USER_PROMPT_CALL_COUNTERS__"]) {
-    g4["__ELASTICDASH_USER_PROMPT_CALL_COUNTERS__"] = {};
+  if (!g5["__ELASTICDASH_USER_PROMPT_CALL_COUNTERS__"]) {
+    g5["__ELASTICDASH_USER_PROMPT_CALL_COUNTERS__"] = {};
   }
-  const counters = g4["__ELASTICDASH_USER_PROMPT_CALL_COUNTERS__"];
+  const counters = g5["__ELASTICDASH_USER_PROMPT_CALL_COUNTERS__"];
   const uniqueTexts = Array.from(new Set(userPrompts));
   let modified = false;
   let result = input;
@@ -1108,15 +1108,15 @@ function applySystemPromptMocks(input, promptMocks, callCounters) {
   return input;
 }
 function resolveAIMock(modelName) {
-  const g4 = globalThis;
-  const mocks = g4["__ELASTICDASH_AI_MOCKS__"];
+  const g5 = globalThis;
+  const mocks = g5["__ELASTICDASH_AI_MOCKS__"];
   if (!mocks) return { mocked: false };
   const entry = mocks[modelName];
   if (!entry || entry.mode === "live") return { mocked: false };
-  if (!g4["__ELASTICDASH_AI_CALL_COUNTERS__"]) {
-    g4["__ELASTICDASH_AI_CALL_COUNTERS__"] = {};
+  if (!g5["__ELASTICDASH_AI_CALL_COUNTERS__"]) {
+    g5["__ELASTICDASH_AI_CALL_COUNTERS__"] = {};
   }
-  const counters = g4["__ELASTICDASH_AI_CALL_COUNTERS__"];
+  const counters = g5["__ELASTICDASH_AI_CALL_COUNTERS__"];
   counters[modelName] = (counters[modelName] ?? 0) + 1;
   const callNumber = counters[modelName];
   if (entry.mode === "mock-all") {
@@ -1475,13 +1475,354 @@ var init_http = __esm({
   }
 });
 
+// src/interceptors/workflow-ai.ts
+function isAIWrapperActive() {
+  return g3[AI_WRAPPER_KEY] > 0;
+}
+function enterAIWrapper() {
+  g3[AI_WRAPPER_KEY] = (g3[AI_WRAPPER_KEY] ?? 0) + 1;
+}
+function leaveAIWrapper() {
+  g3[AI_WRAPPER_KEY] = Math.max(0, (g3[AI_WRAPPER_KEY] ?? 0) - 1);
+}
+function extractUsage(output) {
+  if (!output || typeof output !== "object") return void 0;
+  const o = output;
+  if (o.usage && typeof o.usage === "object") {
+    const u = o.usage;
+    if (u.input_tokens != null || u.output_tokens != null) {
+      return {
+        inputTokens: u.input_tokens,
+        outputTokens: u.output_tokens,
+        totalTokens: (u.input_tokens ?? 0) + (u.output_tokens ?? 0)
+      };
+    }
+    if (u.prompt_tokens != null || u.completion_tokens != null) {
+      return {
+        inputTokens: u.prompt_tokens,
+        outputTokens: u.completion_tokens,
+        totalTokens: u.total_tokens
+      };
+    }
+  }
+  if (o.usageMetadata && typeof o.usageMetadata === "object") {
+    const u = o.usageMetadata;
+    return {
+      inputTokens: u.promptTokenCount,
+      outputTokens: u.candidatesTokenCount,
+      totalTokens: u.totalTokenCount
+    };
+  }
+  return void 0;
+}
+function isReadableStream(v) {
+  return typeof v === "object" && v !== null && typeof v.getReader === "function" && typeof v.tee === "function";
+}
+function isAsyncIterable(v) {
+  return typeof v === "object" && v !== null && Symbol.asyncIterator in v;
+}
+async function bufferReadableStream(stream) {
+  const decoder = new TextDecoder();
+  const reader = stream.getReader();
+  let raw = "";
+  try {
+    for (; ; ) {
+      const { done, value } = await reader.read();
+      if (done) break;
+      raw += decoder.decode(value, { stream: true });
+    }
+  } finally {
+    reader.releaseLock();
+  }
+  return raw;
+}
+function reconstructStream2(raw) {
+  const encoder = new TextEncoder();
+  return new ReadableStream({
+    start(ctrl) {
+      ctrl.enqueue(encoder.encode(raw));
+      ctrl.close();
+    }
+  });
+}
+function wrapAsyncIterable(source, onComplete) {
+  return {
+    [Symbol.asyncIterator]() {
+      const iter = source[Symbol.asyncIterator]();
+      const collected = [];
+      return {
+        async next() {
+          const result = await iter.next();
+          if (!result.done) {
+            collected.push(result.value);
+          } else {
+            onComplete(collected);
+          }
+          return result;
+        },
+        async return(value) {
+          onComplete(collected);
+          return iter.return ? iter.return(value) : { done: true, value: void 0 };
+        }
+      };
+    }
+  };
+}
+function wrapAI(modelName, callFn) {
+  return async (...args) => {
+    await tryAutoInitHttpContext();
+    const ctx = getCaptureContext();
+    const httpCtx = getHttpRunContext();
+    const obsCtx = getObservabilityContext();
+    if (!ctx && !httpCtx && !obsCtx) return callFn(...args);
+    enterAIWrapper();
+    try {
+      const start = rawDateNow();
+      if (!ctx && !httpCtx && obsCtx) {
+        const id2 = obsCtx.nextId();
+        const input2 = args.length === 1 ? args[0] : args;
+        try {
+          const output = await callFn(...args);
+          if (isReadableStream(output)) {
+            const [streamForCaller, streamForRecorder] = output.tee();
+            bufferReadableStream(streamForRecorder).then((rawText) => {
+              const durationMs2 = rawDateNow() - start;
+              pushTelemetryEvent({ id: id2, type: "ai", name: modelName, input: input2, output: null, streamed: true, streamRaw: rawText, timestamp: start, durationMs: durationMs2 });
+            }).catch(() => {
+              const durationMs2 = rawDateNow() - start;
+              pushTelemetryEvent({ id: id2, type: "ai", name: modelName, input: input2, output: null, streamed: true, streamRaw: "", timestamp: start, durationMs: durationMs2 });
+            });
+            return streamForCaller;
+          }
+          if (isAsyncIterable(output)) {
+            return wrapAsyncIterable(output, (chunks) => {
+              const durationMs2 = rawDateNow() - start;
+              const rawText = chunks.map((c) => typeof c === "string" ? c : JSON.stringify(c)).join("");
+              pushTelemetryEvent({ id: id2, type: "ai", name: modelName, input: input2, output: null, streamed: true, streamRaw: rawText, timestamp: start, durationMs: durationMs2 });
+            });
+          }
+          const durationMs = rawDateNow() - start;
+          const usage = extractUsage(output);
+          const event = {
+            id: id2,
+            type: "ai",
+            name: modelName,
+            input: input2,
+            output,
+            timestamp: start,
+            durationMs,
+            ...usage ? { usage } : {}
+          };
+          pushTelemetryEvent(event);
+          return output;
+        } catch (e) {
+          const durationMs = rawDateNow() - start;
+          pushTelemetryEvent({
+            id: id2,
+            type: "ai",
+            name: modelName,
+            input: input2,
+            output: { error: String(e) },
+            timestamp: start,
+            durationMs
+          });
+          throw e;
+        }
+      }
+      if (ctx) {
+        const { recorder, replay } = ctx;
+        const id2 = recorder.nextId();
+        if (replay.shouldReplay(id2)) {
+          const historical = replay.getRecordedEvent(id2);
+          if (historical?.streamed === true) {
+            const raw = typeof historical.streamRaw === "string" ? historical.streamRaw : "";
+            return reconstructStream2(raw);
+          }
+          return replay.getRecordedResult(id2);
+        }
+        const aiMock2 = resolveAIMock(modelName);
+        if (aiMock2.mocked) {
+          const input3 = args.length === 1 ? args[0] : args;
+          const event = {
+            id: id2,
+            type: "ai",
+            name: modelName,
+            input: input3,
+            output: aiMock2.result,
+            timestamp: start,
+            durationMs: 0
+          };
+          recorder.record(event);
+          if (httpCtx) pushTelemetryEvent(event);
+          return aiMock2.result;
+        }
+        const rawInput = args.length === 1 ? args[0] : args;
+        const promptModifiedInput = resolvePromptMock(rawInput);
+        const userPromptModifiedInput = resolveUserPromptMock(promptModifiedInput !== void 0 ? promptModifiedInput : rawInput);
+        const modifiedInput = userPromptModifiedInput !== void 0 ? userPromptModifiedInput : promptModifiedInput;
+        const effectiveArgs = modifiedInput !== void 0 ? [modifiedInput] : args;
+        const input2 = modifiedInput !== void 0 ? modifiedInput : rawInput;
+        try {
+          const output = await callFn(...effectiveArgs);
+          if (isReadableStream(output)) {
+            const [streamForCaller, streamForRecorder] = output.tee();
+            bufferReadableStream(streamForRecorder).then((rawText) => {
+              const durationMs2 = rawDateNow() - start;
+              const event2 = { id: id2, type: "ai", name: modelName, input: input2, output: null, streamed: true, streamRaw: rawText, timestamp: start, durationMs: durationMs2 };
+              recorder.record(event2);
+              if (httpCtx) pushTelemetryEvent(event2);
+            }).catch(() => {
+              const durationMs2 = rawDateNow() - start;
+              const event2 = { id: id2, type: "ai", name: modelName, input: input2, output: null, streamed: true, streamRaw: "", timestamp: start, durationMs: durationMs2 };
+              recorder.record(event2);
+              if (httpCtx) pushTelemetryEvent(event2);
+            });
+            return streamForCaller;
+          }
+          if (isAsyncIterable(output)) {
+            return wrapAsyncIterable(output, (chunks) => {
+              const durationMs2 = rawDateNow() - start;
+              const rawText = chunks.map((c) => typeof c === "string" ? c : JSON.stringify(c)).join("");
+              const event2 = { id: id2, type: "ai", name: modelName, input: input2, output: null, streamed: true, streamRaw: rawText, timestamp: start, durationMs: durationMs2 };
+              recorder.record(event2);
+              if (httpCtx) pushTelemetryEvent(event2);
+            });
+          }
+          const durationMs = rawDateNow() - start;
+          const usage = extractUsage(output);
+          const event = {
+            id: id2,
+            type: "ai",
+            name: modelName,
+            input: input2,
+            output,
+            timestamp: start,
+            durationMs,
+            ...usage ? { usage } : {}
+          };
+          recorder.record(event);
+          if (httpCtx) pushTelemetryEvent(event);
+          return output;
+        } catch (e) {
+          const durationMs = rawDateNow() - start;
+          const event = {
+            id: id2,
+            type: "ai",
+            name: modelName,
+            input: input2,
+            output: { error: String(e) },
+            timestamp: start,
+            durationMs
+          };
+          recorder.record(event);
+          if (httpCtx) pushTelemetryEvent(event);
+          throw e;
+        }
+      }
+      const id = httpCtx.nextId();
+      const frozen = getHttpFrozenEvent(id);
+      if (frozen) {
+        pushTelemetryEvent(frozen);
+        if (frozen.streamed === true) {
+          const raw = typeof frozen.streamRaw === "string" ? frozen.streamRaw : "";
+          return reconstructStream2(raw);
+        }
+        return frozen.output;
+      }
+      const aiMock = getHttpAIMock(modelName);
+      if (aiMock.mocked) {
+        const input2 = args.length === 1 ? args[0] : args;
+        const event = {
+          id,
+          type: "ai",
+          name: modelName,
+          input: input2,
+          output: aiMock.result,
+          timestamp: start,
+          durationMs: 0
+        };
+        pushTelemetryEvent(event);
+        return aiMock.result;
+      }
+      const rawHttpInput = args.length === 1 ? args[0] : args;
+      const httpSystemModified = getHttpPromptMock(rawHttpInput);
+      const httpUserModified = getHttpUserPromptMock(httpSystemModified !== void 0 ? httpSystemModified : rawHttpInput);
+      const httpModifiedInput = httpUserModified !== void 0 ? httpUserModified : httpSystemModified;
+      const httpEffectiveArgs = httpModifiedInput !== void 0 ? [httpModifiedInput] : args;
+      const input = httpModifiedInput !== void 0 ? httpModifiedInput : rawHttpInput;
+      try {
+        const output = await callFn(...httpEffectiveArgs);
+        if (isReadableStream(output)) {
+          const [streamForCaller, streamForRecorder] = output.tee();
+          bufferReadableStream(streamForRecorder).then((rawText) => {
+            const durationMs2 = rawDateNow() - start;
+            pushTelemetryEvent({ id, type: "ai", name: modelName, input, output: null, streamed: true, streamRaw: rawText, timestamp: start, durationMs: durationMs2 });
+          }).catch(() => {
+            const durationMs2 = rawDateNow() - start;
+            pushTelemetryEvent({ id, type: "ai", name: modelName, input, output: null, streamed: true, streamRaw: "", timestamp: start, durationMs: durationMs2 });
+          });
+          return streamForCaller;
+        }
+        if (isAsyncIterable(output)) {
+          return wrapAsyncIterable(output, (chunks) => {
+            const durationMs2 = rawDateNow() - start;
+            const rawText = chunks.map((c) => typeof c === "string" ? c : JSON.stringify(c)).join("");
+            pushTelemetryEvent({ id, type: "ai", name: modelName, input, output: null, streamed: true, streamRaw: rawText, timestamp: start, durationMs: durationMs2 });
+          });
+        }
+        const durationMs = rawDateNow() - start;
+        const usage = extractUsage(output);
+        const event = {
+          id,
+          type: "ai",
+          name: modelName,
+          input,
+          output,
+          timestamp: start,
+          durationMs,
+          ...usage ? { usage } : {}
+        };
+        pushTelemetryEvent(event);
+        return output;
+      } catch (e) {
+        const durationMs = rawDateNow() - start;
+        pushTelemetryEvent({
+          id,
+          type: "ai",
+          name: modelName,
+          input,
+          output: { error: String(e) },
+          timestamp: start,
+          durationMs
+        });
+        throw e;
+      }
+    } finally {
+      leaveAIWrapper();
+    }
+  };
+}
+var AI_WRAPPER_KEY, g3;
+var init_workflow_ai = __esm({
+  "src/interceptors/workflow-ai.ts"() {
+    "use strict";
+    init_recorder();
+    init_side_effects();
+    init_telemetry_push();
+    init_mock_resolver();
+    AI_WRAPPER_KEY = "__elasticdash_ai_wrapper_depth__";
+    g3 = globalThis;
+    if (g3[AI_WRAPPER_KEY] == null) g3[AI_WRAPPER_KEY] = 0;
+  }
+});
+
 // src/interceptors/ai-interceptor.ts
 var ai_interceptor_exports = {};
 __export(ai_interceptor_exports, {
   installAIInterceptor: () => installAIInterceptor,
   uninstallAIInterceptor: () => uninstallAIInterceptor
 });
-function extractUsage(provider, body) {
+function extractUsage2(provider, body) {
   if (provider === "openai" || provider === "grok" || provider === "kimi") {
     const u = body.usage;
     if (!u) return void 0;
@@ -1799,6 +2140,9 @@ function installAIInterceptor() {
   globalThis.fetch = async function patchedFetch(input, init) {
     const url = typeof input === "string" ? input : input instanceof URL ? input.href : input.url;
     const provider = detectProvider2(url);
+    if (provider && isAIWrapperActive()) {
+      return originalFetch2(input, init);
+    }
     const traceAtCall = getCurrentTrace();
     const obsCtx = getObservabilityContext();
     const httpCtx = getHttpRunContext();
@@ -1842,7 +2186,7 @@ function installAIInterceptor() {
         const cloned = response2.clone();
         const responseBody = await cloned.json();
         const completion = extractCompletion2(provider, responseBody);
-        const usage = extractUsage(provider, responseBody);
+        const usage = extractUsage2(provider, responseBody);
         const durationMs = rawDateNow() - start;
         const event = {
           id,
@@ -1898,7 +2242,7 @@ function installAIInterceptor() {
         const cloned = response2.clone();
         const responseBody = await cloned.json();
         const completion = extractCompletion2(provider, responseBody);
-        const usage = extractUsage(provider, responseBody);
+        const usage = extractUsage2(provider, responseBody);
         const durationMs = rawDateNow() - start;
         pushTelemetryEvent({ id, type: "ai", name: model, input: eventInput, output: { completion }, timestamp: start, durationMs, ...usage ? { usage } : {} });
       } catch {
@@ -1970,7 +2314,7 @@ function installAIInterceptor() {
           const cloned = response2.clone();
           const responseBody = await cloned.json();
           const completion = extractCompletion2(provider, responseBody);
-          const usage = extractUsage(provider, responseBody);
+          const usage = extractUsage2(provider, responseBody);
           const assistantMessage = extractAssistantMessage(provider, responseBody);
           traceAtCall.recordLLMStep({ model, provider, prompt, completion, workflowEventId: id, durationMs });
           recorder.record({ id, type: "ai", name: model, input: { url, provider, model, prompt, messages }, output: assistantMessage ?? responseBody, timestamp: start, durationMs, usage });
@@ -2018,6 +2362,7 @@ var init_ai_interceptor = __esm({
     init_recorder();
     init_side_effects();
     init_telemetry_push();
+    init_workflow_ai();
     AI_PATTERNS2 = {
       openai: /https?:\/\/api\.openai\.com\/v1\/((chat\/)?completions|embeddings)/,
       anthropic: /https?:\/\/api\.anthropic\.com\/v1\/messages/,
@@ -2905,6 +3250,8 @@ async function executeTrigger(serverUrl, apiKey, trigger) {
     return;
   }
   handledTriggers.add(trigger.triggerId);
+  const obsCtx = getObservabilityContext();
+  if (obsCtx) obsCtx.isRerun = true;
   const cwd = process.cwd();
   const tools = scanTools(cwd);
   const frozenEvents = trigger.frozenEvents || [];
@@ -2984,6 +3331,7 @@ async function executeTrigger(serverUrl, apiKey, trigger) {
       debugLog(`[elasticdash] Trigger ${trigger.triggerId} step ${stepIndex + 1}/${totalSteps} POST to event ${step.originalEventDbId} failed: ${err instanceof Error ? err.message : String(err)}`);
     }
   }
+  if (obsCtx) obsCtx.isRerun = false;
 }
 var handledTriggers;
 var init_trigger_executor = __esm({
@@ -2991,6 +3339,7 @@ var init_trigger_executor = __esm({
     "use strict";
     init_portal_executor();
     init_http();
+    init_telemetry_push();
     init_tool_runner();
     init_debug();
     handledTriggers = /* @__PURE__ */ new Set();
@@ -3290,14 +3639,14 @@ var init_observability = __esm({
 // src/interceptors/telemetry-push.ts
 function setObservabilityContext(ctx) {
   obsAls.enterWith(ctx);
-  g3[GLOBAL_OBS_KEY] = ctx;
+  g4[GLOBAL_OBS_KEY] = ctx;
 }
 function getObservabilityContext() {
-  return obsAls.getStore() ?? g3[GLOBAL_OBS_KEY];
+  return obsAls.getStore() ?? g4[GLOBAL_OBS_KEY];
 }
 function clearObservabilityContext() {
   obsAls.enterWith(void 0);
-  g3[GLOBAL_OBS_KEY] = void 0;
+  g4[GLOBAL_OBS_KEY] = void 0;
 }
 function buildContext(runId, dashboardUrl, frozenEvents, promptMocksRecord = {}, toolMockConfig, aiMockConfig, userPromptMocks) {
   let counter = 0;
@@ -3319,15 +3668,15 @@ function buildContext(runId, dashboardUrl, frozenEvents, promptMocksRecord = {},
   };
 }
 function setGlobalHttpContext(ctx) {
-  g3[GLOBAL_CTX_KEY] = ctx;
+  g4[GLOBAL_CTX_KEY] = ctx;
 }
 function clearGlobalHttpContext() {
-  g3[GLOBAL_CTX_KEY] = void 0;
+  g4[GLOBAL_CTX_KEY] = void 0;
 }
 function setHttpRunContext(runId, dashboardUrl) {
   const ctx = buildContext(runId, dashboardUrl, [], {});
   httpRunAls.enterWith(ctx);
-  g3[GLOBAL_CTX_KEY] = ctx;
+  g4[GLOBAL_CTX_KEY] = ctx;
   ensureInterceptorsInstalled().catch(() => {
   });
 }
@@ -3357,11 +3706,11 @@ async function initHttpRunContext(runId, dashboardUrl) {
   }
   const ctx = buildContext(runId, dashboardUrl, frozenEvents, promptMocks, toolMockConfig, aiMockConfig, userPromptMocks);
   httpRunAls.enterWith(ctx);
-  g3[GLOBAL_CTX_KEY] = ctx;
+  g4[GLOBAL_CTX_KEY] = ctx;
   await ensureInterceptorsInstalled();
 }
 function getHttpRunContext() {
-  return httpRunAls.getStore() ?? g3[GLOBAL_CTX_KEY];
+  return httpRunAls.getStore() ?? g4[GLOBAL_CTX_KEY];
 }
 function getHttpFrozenEvent(id) {
   return getHttpRunContext()?.frozenEvents.get(id);
@@ -3469,6 +3818,9 @@ function pushTelemetryEvent(event, explicitCtx) {
     if (!event.traceId && obsCtx.traceId) {
       event.traceId = obsCtx.traceId;
     }
+    if (obsCtx.isRerun) {
+      event.isRerun = true;
+    }
     obsCtx.batcher.enqueue(event);
     return;
   }
@@ -3490,8 +3842,8 @@ function pushTelemetryEvent(event, explicitCtx) {
   });
 }
 async function ensureInterceptorsInstalled() {
-  if (g3[INTERCEPTORS_KEY]) return;
-  g3[INTERCEPTORS_KEY] = true;
+  if (g4[INTERCEPTORS_KEY]) return;
+  g4[INTERCEPTORS_KEY] = true;
   try {
     const [httpMod, aiMod, dbMod] = await Promise.all([
       Promise.resolve().then(() => (init_http(), http_exports)),
@@ -3507,11 +3859,11 @@ async function ensureInterceptorsInstalled() {
 async function runInHttpContext(runId, dashboardUrl, callback) {
   await ensureInterceptorsInstalled();
   const ctx = buildContext(runId, dashboardUrl, [], {});
-  g3[GLOBAL_CTX_KEY] = ctx;
+  g4[GLOBAL_CTX_KEY] = ctx;
   try {
     return await httpRunAls.run(ctx, callback);
   } finally {
-    g3[GLOBAL_CTX_KEY] = void 0;
+    g4[GLOBAL_CTX_KEY] = void 0;
   }
 }
 async function runWithInitializedHttpContext(runId, dashboardUrl, callback) {
@@ -3548,11 +3900,11 @@ async function runWithInitializedHttpContext(runId, dashboardUrl, callback) {
   }
   await ensureInterceptorsInstalled();
   const ctx = buildContext(runId, dashboardUrl, frozenEvents, promptMocks, toolMockConfig, aiMockConfig, userPromptMocks);
-  g3[GLOBAL_CTX_KEY] = ctx;
+  g4[GLOBAL_CTX_KEY] = ctx;
   try {
     return await httpRunAls.run(ctx, callback);
   } finally {
-    g3[GLOBAL_CTX_KEY] = void 0;
+    g4[GLOBAL_CTX_KEY] = void 0;
   }
 }
 async function tryAutoInitHttpContext() {
@@ -3561,8 +3913,8 @@ async function tryAutoInitHttpContext() {
   const apiUrl = (typeof process !== "undefined" && process.env?.ELASTICDASH_API_URL) ?? "";
   if (apiUrl) {
     const obsInitKey = "__elasticdash_obs_auto_init__";
-    if (!g3[obsInitKey]) {
-      g3[obsInitKey] = (async () => {
+    if (!g4[obsInitKey]) {
+      g4[obsInitKey] = (async () => {
         try {
           const { initObservability: initObservability2 } = await Promise.resolve().then(() => (init_observability(), observability_exports));
           initObservability2({
@@ -3575,13 +3927,13 @@ async function tryAutoInitHttpContext() {
         }
       })();
     }
-    await g3[obsInitKey];
+    await g4[obsInitKey];
     return;
   }
   const serverUrl = (typeof process !== "undefined" && process.env?.ELASTICDASH_SERVER) ?? "";
   if (!serverUrl) return;
-  if (!g3[AUTO_INIT_KEY]) {
-    g3[AUTO_INIT_KEY] = (async () => {
+  if (!g4[AUTO_INIT_KEY]) {
+    g4[AUTO_INIT_KEY] = (async () => {
       try {
         const runId = (typeof process !== "undefined" && process.env?.ELASTICDASH_RUN_ID) ?? "";
         if (runId) {
@@ -3593,9 +3945,9 @@ async function tryAutoInitHttpContext() {
       }
     })();
   }
-  await g3[AUTO_INIT_KEY];
+  await g4[AUTO_INIT_KEY];
 }
-var import_node_async_hooks3, import_node_crypto4, g3, HTTP_RUN_ALS_KEY, GLOBAL_CTX_KEY, OBS_ALS_KEY, GLOBAL_OBS_KEY, httpRunAls, obsAls, INTERCEPTORS_KEY, AUTO_INIT_KEY;
+var import_node_async_hooks3, import_node_crypto4, g4, HTTP_RUN_ALS_KEY, GLOBAL_CTX_KEY, OBS_ALS_KEY, GLOBAL_OBS_KEY, httpRunAls, obsAls, INTERCEPTORS_KEY, AUTO_INIT_KEY;
 var init_telemetry_push = __esm({
   "src/interceptors/telemetry-push.ts"() {
     "use strict";
@@ -3603,15 +3955,15 @@ var init_telemetry_push = __esm({
     import_node_crypto4 = require("node:crypto");
     init_mock_resolver();
     init_debug();
-    g3 = globalThis;
+    g4 = globalThis;
     HTTP_RUN_ALS_KEY = "__elasticdash_http_run_als__";
     GLOBAL_CTX_KEY = "__elasticdash_global_http_ctx__";
     OBS_ALS_KEY = "__elasticdash_obs_als__";
     GLOBAL_OBS_KEY = "__elasticdash_global_obs_ctx__";
-    httpRunAls = g3[HTTP_RUN_ALS_KEY] ?? new import_node_async_hooks3.AsyncLocalStorage();
-    if (!g3[HTTP_RUN_ALS_KEY]) g3[HTTP_RUN_ALS_KEY] = httpRunAls;
-    obsAls = g3[OBS_ALS_KEY] ?? new import_node_async_hooks3.AsyncLocalStorage();
-    if (!g3[OBS_ALS_KEY]) g3[OBS_ALS_KEY] = obsAls;
+    httpRunAls = g4[HTTP_RUN_ALS_KEY] ?? new import_node_async_hooks3.AsyncLocalStorage();
+    if (!g4[HTTP_RUN_ALS_KEY]) g4[HTTP_RUN_ALS_KEY] = httpRunAls;
+    obsAls = g4[OBS_ALS_KEY] ?? new import_node_async_hooks3.AsyncLocalStorage();
+    if (!g4[OBS_ALS_KEY]) g4[OBS_ALS_KEY] = obsAls;
     INTERCEPTORS_KEY = "__elasticdash_interceptors_installed__";
     AUTO_INIT_KEY = "__elasticdash_auto_init_promise__";
   }
@@ -4308,13 +4660,13 @@ function toTraceArgs2(input) {
   if (input === void 0) return void 0;
   return { value: input };
 }
-function isReadableStream(v) {
+function isReadableStream2(v) {
   return typeof v === "object" && v !== null && typeof v.getReader === "function" && typeof v.tee === "function";
 }
-function isAsyncIterable(v) {
+function isAsyncIterable2(v) {
   return typeof v === "object" && v !== null && Symbol.asyncIterator in v;
 }
-async function bufferReadableStream(stream) {
+async function bufferReadableStream2(stream) {
   const decoder = new TextDecoder();
   const reader = stream.getReader();
   let raw = "";
@@ -4329,7 +4681,7 @@ async function bufferReadableStream(stream) {
   }
   return raw;
 }
-function reconstructStream2(raw) {
+function reconstructStream3(raw) {
   const encoder = new TextEncoder();
   return new ReadableStream({
     start(ctrl) {
@@ -4338,7 +4690,7 @@ function reconstructStream2(raw) {
     }
   });
 }
-function wrapAsyncIterable(source, onComplete) {
+function wrapAsyncIterable2(source, onComplete) {
   return {
     [Symbol.asyncIterator]() {
       const iter = source[Symbol.asyncIterator]();
@@ -4378,9 +4730,9 @@ function wrapTool(name, fn) {
       try {
         const output = await fn(...args);
         const durationMs = rawDateNow() - start2;
-        if (isReadableStream(output)) {
+        if (isReadableStream2(output)) {
           const [streamForCaller, streamForRecorder] = output.tee();
-          bufferReadableStream(streamForRecorder).then((rawText) => {
+          bufferReadableStream2(streamForRecorder).then((rawText) => {
             const durationMs2 = rawDateNow() - start2;
             pushTelemetryEvent({ id: id2, type: "tool", name, input: input2, output: null, streamed: true, streamRaw: rawText, timestamp: start2, durationMs: durationMs2 });
           }).catch(() => {
@@ -4389,8 +4741,8 @@ function wrapTool(name, fn) {
           });
           return streamForCaller;
         }
-        if (isAsyncIterable(output)) {
-          return wrapAsyncIterable(output, (chunks) => {
+        if (isAsyncIterable2(output)) {
+          return wrapAsyncIterable2(output, (chunks) => {
             const durationMs2 = rawDateNow() - start2;
             const rawText = chunks.map((c) => typeof c === "string" ? c : JSON.stringify(c)).join("");
             pushTelemetryEvent({ id: id2, type: "tool", name, input: input2, output: null, streamed: true, streamRaw: rawText, timestamp: start2, durationMs: durationMs2 });
@@ -4412,7 +4764,7 @@ function wrapTool(name, fn) {
         pushTelemetryEvent(frozen);
         if (frozen.streamed === true) {
           const raw = typeof frozen.streamRaw === "string" ? frozen.streamRaw : "";
-          return reconstructStream2(raw);
+          return reconstructStream3(raw);
         }
         return frozen.output;
       }
@@ -4425,9 +4777,9 @@ function wrapTool(name, fn) {
       const start2 = rawDateNow();
       try {
         const output = await fn(...args);
-        if (isReadableStream(output)) {
+        if (isReadableStream2(output)) {
           const [streamForCaller, streamForRecorder] = output.tee();
-          bufferReadableStream(streamForRecorder).then((rawText) => {
+          bufferReadableStream2(streamForRecorder).then((rawText) => {
             const durationMs2 = rawDateNow() - start2;
             pushTelemetryEvent({ id: id2, type: "tool", name, input: input2, output: null, streamed: true, streamRaw: rawText, timestamp: start2, durationMs: durationMs2 });
           }).catch(() => {
@@ -4436,8 +4788,8 @@ function wrapTool(name, fn) {
           });
           return streamForCaller;
         }
-        if (isAsyncIterable(output)) {
-          return wrapAsyncIterable(output, (chunks) => {
+        if (isAsyncIterable2(output)) {
+          return wrapAsyncIterable2(output, (chunks) => {
             const durationMs2 = rawDateNow() - start2;
             const rawText = chunks.map((c) => typeof c === "string" ? c : JSON.stringify(c)).join("");
             pushTelemetryEvent({ id: id2, type: "tool", name, input: input2, output: null, streamed: true, streamRaw: rawText, timestamp: start2, durationMs: durationMs2 });
@@ -4461,7 +4813,7 @@ function wrapTool(name, fn) {
       if (historical) recorder.record(historical);
       if (historical?.streamed === true) {
         const raw = typeof historical.streamRaw === "string" ? historical.streamRaw : "";
-        const stream = reconstructStream2(raw);
+        const stream = reconstructStream3(raw);
         if (trace && typeof trace.recordToolCall === "function") {
           trace.recordToolCall({ name, args: toTraceArgs2(input), result: stream, workflowEventId: id });
         }
@@ -4473,15 +4825,15 @@ function wrapTool(name, fn) {
       }
       return replayed;
     }
-    const g4 = globalThis;
-    const prev = g4[TOOL_WRAPPER_ACTIVE_KEY];
-    g4[TOOL_WRAPPER_ACTIVE_KEY] = true;
+    const g5 = globalThis;
+    const prev = g5[TOOL_WRAPPER_ACTIVE_KEY];
+    g5[TOOL_WRAPPER_ACTIVE_KEY] = true;
     const start = rawDateNow();
     try {
       const output = await fn(...args);
-      if (isReadableStream(output)) {
+      if (isReadableStream2(output)) {
         const [streamForCaller, streamForRecorder] = output.tee();
-        bufferReadableStream(streamForRecorder).then((rawText) => {
+        bufferReadableStream2(streamForRecorder).then((rawText) => {
           const durationMs2 = rawDateNow() - start;
           const event = { id, type: "tool", name, input, output: null, streamed: true, streamRaw: rawText, timestamp: start, durationMs: durationMs2 };
           recorder.record(event);
@@ -4498,8 +4850,8 @@ function wrapTool(name, fn) {
         }
         return result;
       }
-      if (isAsyncIterable(output)) {
-        const wrapped = wrapAsyncIterable(output, (chunks) => {
+      if (isAsyncIterable2(output)) {
+        const wrapped = wrapAsyncIterable2(output, (chunks) => {
           const durationMs2 = rawDateNow() - start;
           const rawText = chunks.map((c) => typeof c === "string" ? c : JSON.stringify(c)).join("");
           const event = { id, type: "tool", name, input, output: null, streamed: true, streamRaw: rawText, timestamp: start, durationMs: durationMs2 };
@@ -4529,331 +4881,14 @@ function wrapTool(name, fn) {
       }
       throw e;
     } finally {
-      if (prev === void 0) delete g4[TOOL_WRAPPER_ACTIVE_KEY];
-      else g4[TOOL_WRAPPER_ACTIVE_KEY] = prev;
-    }
-  };
-}
-
-// src/interceptors/workflow-ai.ts
-init_recorder();
-init_side_effects();
-init_telemetry_push();
-init_mock_resolver();
-function extractUsage2(output) {
-  if (!output || typeof output !== "object") return void 0;
-  const o = output;
-  if (o.usage && typeof o.usage === "object") {
-    const u = o.usage;
-    if (u.input_tokens != null || u.output_tokens != null) {
-      return {
-        inputTokens: u.input_tokens,
-        outputTokens: u.output_tokens,
-        totalTokens: (u.input_tokens ?? 0) + (u.output_tokens ?? 0)
-      };
-    }
-    if (u.prompt_tokens != null || u.completion_tokens != null) {
-      return {
-        inputTokens: u.prompt_tokens,
-        outputTokens: u.completion_tokens,
-        totalTokens: u.total_tokens
-      };
-    }
-  }
-  if (o.usageMetadata && typeof o.usageMetadata === "object") {
-    const u = o.usageMetadata;
-    return {
-      inputTokens: u.promptTokenCount,
-      outputTokens: u.candidatesTokenCount,
-      totalTokens: u.totalTokenCount
-    };
-  }
-  return void 0;
-}
-function isReadableStream2(v) {
-  return typeof v === "object" && v !== null && typeof v.getReader === "function" && typeof v.tee === "function";
-}
-function isAsyncIterable2(v) {
-  return typeof v === "object" && v !== null && Symbol.asyncIterator in v;
-}
-async function bufferReadableStream2(stream) {
-  const decoder = new TextDecoder();
-  const reader = stream.getReader();
-  let raw = "";
-  try {
-    for (; ; ) {
-      const { done, value } = await reader.read();
-      if (done) break;
-      raw += decoder.decode(value, { stream: true });
-    }
-  } finally {
-    reader.releaseLock();
-  }
-  return raw;
-}
-function reconstructStream3(raw) {
-  const encoder = new TextEncoder();
-  return new ReadableStream({
-    start(ctrl) {
-      ctrl.enqueue(encoder.encode(raw));
-      ctrl.close();
-    }
-  });
-}
-function wrapAsyncIterable2(source, onComplete) {
-  return {
-    [Symbol.asyncIterator]() {
-      const iter = source[Symbol.asyncIterator]();
-      const collected = [];
-      return {
-        async next() {
-          const result = await iter.next();
-          if (!result.done) {
-            collected.push(result.value);
-          } else {
-            onComplete(collected);
-          }
-          return result;
-        },
-        async return(value) {
-          onComplete(collected);
-          return iter.return ? iter.return(value) : { done: true, value: void 0 };
-        }
-      };
-    }
-  };
-}
-function wrapAI(modelName, callFn) {
-  return async (...args) => {
-    await tryAutoInitHttpContext();
-    const ctx = getCaptureContext();
-    const httpCtx = getHttpRunContext();
-    const obsCtx = getObservabilityContext();
-    if (!ctx && !httpCtx && !obsCtx) return callFn(...args);
-    const start = rawDateNow();
-    if (!ctx && !httpCtx && obsCtx) {
-      const id2 = obsCtx.nextId();
-      const input2 = args.length === 1 ? args[0] : args;
-      try {
-        const output = await callFn(...args);
-        if (isReadableStream2(output)) {
-          const [streamForCaller, streamForRecorder] = output.tee();
-          bufferReadableStream2(streamForRecorder).then((rawText) => {
-            const durationMs2 = rawDateNow() - start;
-            pushTelemetryEvent({ id: id2, type: "ai", name: modelName, input: input2, output: null, streamed: true, streamRaw: rawText, timestamp: start, durationMs: durationMs2 });
-          }).catch(() => {
-            const durationMs2 = rawDateNow() - start;
-            pushTelemetryEvent({ id: id2, type: "ai", name: modelName, input: input2, output: null, streamed: true, streamRaw: "", timestamp: start, durationMs: durationMs2 });
-          });
-          return streamForCaller;
-        }
-        if (isAsyncIterable2(output)) {
-          return wrapAsyncIterable2(output, (chunks) => {
-            const durationMs2 = rawDateNow() - start;
-            const rawText = chunks.map((c) => typeof c === "string" ? c : JSON.stringify(c)).join("");
-            pushTelemetryEvent({ id: id2, type: "ai", name: modelName, input: input2, output: null, streamed: true, streamRaw: rawText, timestamp: start, durationMs: durationMs2 });
-          });
-        }
-        const durationMs = rawDateNow() - start;
-        const usage = extractUsage2(output);
-        const event = {
-          id: id2,
-          type: "ai",
-          name: modelName,
-          input: input2,
-          output,
-          timestamp: start,
-          durationMs,
-          ...usage ? { usage } : {}
-        };
-        pushTelemetryEvent(event);
-        return output;
-      } catch (e) {
-        const durationMs = rawDateNow() - start;
-        pushTelemetryEvent({
-          id: id2,
-          type: "ai",
-          name: modelName,
-          input: input2,
-          output: { error: String(e) },
-          timestamp: start,
-          durationMs
-        });
-        throw e;
-      }
-    }
-    if (ctx) {
-      const { recorder, replay } = ctx;
-      const id2 = recorder.nextId();
-      if (replay.shouldReplay(id2)) {
-        const historical = replay.getRecordedEvent(id2);
-        if (historical?.streamed === true) {
-          const raw = typeof historical.streamRaw === "string" ? historical.streamRaw : "";
-          return reconstructStream3(raw);
-        }
-        return replay.getRecordedResult(id2);
-      }
-      const aiMock2 = resolveAIMock(modelName);
-      if (aiMock2.mocked) {
-        const input3 = args.length === 1 ? args[0] : args;
-        const event = {
-          id: id2,
-          type: "ai",
-          name: modelName,
-          input: input3,
-          output: aiMock2.result,
-          timestamp: start,
-          durationMs: 0
-        };
-        recorder.record(event);
-        if (httpCtx) pushTelemetryEvent(event);
-        return aiMock2.result;
-      }
-      const rawInput = args.length === 1 ? args[0] : args;
-      const promptModifiedInput = resolvePromptMock(rawInput);
-      const userPromptModifiedInput = resolveUserPromptMock(promptModifiedInput !== void 0 ? promptModifiedInput : rawInput);
-      const modifiedInput = userPromptModifiedInput !== void 0 ? userPromptModifiedInput : promptModifiedInput;
-      const effectiveArgs = modifiedInput !== void 0 ? [modifiedInput] : args;
-      const input2 = modifiedInput !== void 0 ? modifiedInput : rawInput;
-      try {
-        const output = await callFn(...effectiveArgs);
-        if (isReadableStream2(output)) {
-          const [streamForCaller, streamForRecorder] = output.tee();
-          bufferReadableStream2(streamForRecorder).then((rawText) => {
-            const durationMs2 = rawDateNow() - start;
-            const event2 = { id: id2, type: "ai", name: modelName, input: input2, output: null, streamed: true, streamRaw: rawText, timestamp: start, durationMs: durationMs2 };
-            recorder.record(event2);
-            if (httpCtx) pushTelemetryEvent(event2);
-          }).catch(() => {
-            const durationMs2 = rawDateNow() - start;
-            const event2 = { id: id2, type: "ai", name: modelName, input: input2, output: null, streamed: true, streamRaw: "", timestamp: start, durationMs: durationMs2 };
-            recorder.record(event2);
-            if (httpCtx) pushTelemetryEvent(event2);
-          });
-          return streamForCaller;
-        }
-        if (isAsyncIterable2(output)) {
-          return wrapAsyncIterable2(output, (chunks) => {
-            const durationMs2 = rawDateNow() - start;
-            const rawText = chunks.map((c) => typeof c === "string" ? c : JSON.stringify(c)).join("");
-            const event2 = { id: id2, type: "ai", name: modelName, input: input2, output: null, streamed: true, streamRaw: rawText, timestamp: start, durationMs: durationMs2 };
-            recorder.record(event2);
-            if (httpCtx) pushTelemetryEvent(event2);
-          });
-        }
-        const durationMs = rawDateNow() - start;
-        const usage = extractUsage2(output);
-        const event = {
-          id: id2,
-          type: "ai",
-          name: modelName,
-          input: input2,
-          output,
-          timestamp: start,
-          durationMs,
-          ...usage ? { usage } : {}
-        };
-        recorder.record(event);
-        if (httpCtx) pushTelemetryEvent(event);
-        return output;
-      } catch (e) {
-        const durationMs = rawDateNow() - start;
-        const event = {
-          id: id2,
-          type: "ai",
-          name: modelName,
-          input: input2,
-          output: { error: String(e) },
-          timestamp: start,
-          durationMs
-        };
-        recorder.record(event);
-        if (httpCtx) pushTelemetryEvent(event);
-        throw e;
-      }
-    }
-    const id = httpCtx.nextId();
-    const frozen = getHttpFrozenEvent(id);
-    if (frozen) {
-      pushTelemetryEvent(frozen);
-      if (frozen.streamed === true) {
-        const raw = typeof frozen.streamRaw === "string" ? frozen.streamRaw : "";
-        return reconstructStream3(raw);
-      }
-      return frozen.output;
-    }
-    const aiMock = getHttpAIMock(modelName);
-    if (aiMock.mocked) {
-      const input2 = args.length === 1 ? args[0] : args;
-      const event = {
-        id,
-        type: "ai",
-        name: modelName,
-        input: input2,
-        output: aiMock.result,
-        timestamp: start,
-        durationMs: 0
-      };
-      pushTelemetryEvent(event);
-      return aiMock.result;
-    }
-    const rawHttpInput = args.length === 1 ? args[0] : args;
-    const httpSystemModified = getHttpPromptMock(rawHttpInput);
-    const httpUserModified = getHttpUserPromptMock(httpSystemModified !== void 0 ? httpSystemModified : rawHttpInput);
-    const httpModifiedInput = httpUserModified !== void 0 ? httpUserModified : httpSystemModified;
-    const httpEffectiveArgs = httpModifiedInput !== void 0 ? [httpModifiedInput] : args;
-    const input = httpModifiedInput !== void 0 ? httpModifiedInput : rawHttpInput;
-    try {
-      const output = await callFn(...httpEffectiveArgs);
-      if (isReadableStream2(output)) {
-        const [streamForCaller, streamForRecorder] = output.tee();
-        bufferReadableStream2(streamForRecorder).then((rawText) => {
-          const durationMs2 = rawDateNow() - start;
-          pushTelemetryEvent({ id, type: "ai", name: modelName, input, output: null, streamed: true, streamRaw: rawText, timestamp: start, durationMs: durationMs2 });
-        }).catch(() => {
-          const durationMs2 = rawDateNow() - start;
-          pushTelemetryEvent({ id, type: "ai", name: modelName, input, output: null, streamed: true, streamRaw: "", timestamp: start, durationMs: durationMs2 });
-        });
-        return streamForCaller;
-      }
-      if (isAsyncIterable2(output)) {
-        return wrapAsyncIterable2(output, (chunks) => {
-          const durationMs2 = rawDateNow() - start;
-          const rawText = chunks.map((c) => typeof c === "string" ? c : JSON.stringify(c)).join("");
-          pushTelemetryEvent({ id, type: "ai", name: modelName, input, output: null, streamed: true, streamRaw: rawText, timestamp: start, durationMs: durationMs2 });
-        });
-      }
-      const durationMs = rawDateNow() - start;
-      const usage = extractUsage2(output);
-      const event = {
-        id,
-        type: "ai",
-        name: modelName,
-        input,
-        output,
-        timestamp: start,
-        durationMs,
-        ...usage ? { usage } : {}
-      };
-      pushTelemetryEvent(event);
-      return output;
-    } catch (e) {
-      const durationMs = rawDateNow() - start;
-      pushTelemetryEvent({
-        id,
-        type: "ai",
-        name: modelName,
-        input,
-        output: { error: String(e) },
-        timestamp: start,
-        durationMs
-      });
-      throw e;
+      if (prev === void 0) delete g5[TOOL_WRAPPER_ACTIVE_KEY];
+      else g5[TOOL_WRAPPER_ACTIVE_KEY] = prev;
     }
   };
 }
 
 // src/index.ts
+init_workflow_ai();
 init_telemetry_push();
 init_mock_resolver();