elasticdash-test 0.1.18-alpha-26 → 0.1.18

package/dist/index.cjs

@@ -1468,6 +1468,7 @@ var init_http = __esm({
     init_side_effects();
     AI_URL_PATTERNS = [
       /https?:\/\/api\.openai\.com\/v1\/((chat\/)?completions|embeddings)/,
+      /https?:\/\/api\.anthropic\.com\/v1\/messages/,
       /https?:\/\/generativelanguage\.googleapis\.com\/.*\/models\/[^/:]+:(generateContent|streamGenerateContent)/,
       /https?:\/\/api\.x\.ai\/v1\/(chat\/)?completions/,
       /https?:\/\/api\.moonshot\.ai\/v1\/(chat\/)?completions/
@@ -1478,12 +1479,39 @@ var init_http = __esm({
 // src/interceptors/ai-interceptor.ts
 var ai_interceptor_exports = {};
 __export(ai_interceptor_exports, {
+  consumeCapturedLLMRequest: () => consumeCapturedLLMRequest,
   installAIInterceptor: () => installAIInterceptor,
   uninstallAIInterceptor: () => uninstallAIInterceptor
 });
 function isAIWrapperActive() {
   return (globalThis[AI_WRAPPER_KEY] ?? 0) > 0;
 }
+function consumeCapturedLLMRequest() {
+  const g5 = globalThis;
+  const req = g5[LLM_REQUEST_KEY];
+  if (req) g5[LLM_REQUEST_KEY] = void 0;
+  return req;
+}
+function extractPromptSnippet(body) {
+  let messages;
+  if (Array.isArray(body.messages)) messages = body.messages;
+  else if (Array.isArray(body.contents)) messages = body.contents;
+  if (!messages || messages.length === 0) return void 0;
+  for (let i = messages.length - 1; i >= 0; i--) {
+    const msg = messages[i];
+    if (!msg) continue;
+    if (msg.role === "user") {
+      let content = msg.content;
+      if (Array.isArray(content)) {
+        content = content.map((b) => b && typeof b === "object" ? String(b.text ?? "") : String(b)).filter(Boolean).join("");
+      }
+      if (typeof content === "string") {
+        return content.slice(0, 100);
+      }
+    }
+  }
+  return void 0;
+}
 function extractUsage(provider, body) {
   if (provider === "openai" || provider === "grok" || provider === "kimi") {
     const u = body.usage;
@@ -1723,6 +1751,30 @@ async function bufferSSEStream(provider, stream) {
   }
   return completion;
 }
+function extractStreamUsage(provider, rawSSE) {
+  const lines = rawSSE.split("\n");
+  for (let i = lines.length - 1; i >= 0; i--) {
+    const line = lines[i];
+    if (!line.startsWith("data: ")) continue;
+    const data = line.slice(6).trim();
+    if (data === "[DONE]") continue;
+    try {
+      const obj = JSON.parse(data);
+      const usage = extractUsage(provider, obj);
+      if (usage) return usage;
+      if (obj.type === "message_delta") {
+        const u = obj.usage;
+        if (u) return extractUsage("anthropic", { usage: u });
+      }
+      if (obj.type === "message_start") {
+        const msg = obj.message;
+        if (msg?.usage) return extractUsage("anthropic", { usage: msg.usage });
+      }
+    } catch {
+    }
+  }
+  return void 0;
+}
 function synthesizeCompletionJSON(provider, completion) {
   if (provider === "gemini") {
     return {
@@ -1803,7 +1855,57 @@ function installAIInterceptor() {
     const url = typeof input === "string" ? input : input instanceof URL ? input.href : input.url;
     const provider = detectProvider2(url);
     if (provider && isAIWrapperActive()) {
-      return originalFetch2(input, init);
+      let capturedReq;
+      let capturedModel = "unknown";
+      let capturedMessages;
+      let capturedSnippet;
+      try {
+        const rawBody = init?.body;
+        if (rawBody && typeof rawBody === "string") {
+          capturedReq = JSON.parse(rawBody);
+          capturedModel = extractModel2(provider, capturedReq, url);
+          capturedMessages = Array.isArray(capturedReq.messages) ? capturedReq.messages : Array.isArray(capturedReq.contents) ? capturedReq.contents : void 0;
+          capturedSnippet = extractPromptSnippet(capturedReq);
+        }
+      } catch {
+      }
+      const response2 = await originalFetch2(input, init);
+      if (capturedReq) {
+        const captured = {
+          url,
+          provider,
+          model: capturedModel,
+          messages: capturedMessages,
+          body: capturedReq,
+          promptSnippet: capturedSnippet
+        };
+        const isStreaming2 = capturedReq.stream === true;
+        try {
+          const cloned = response2.clone();
+          if (!isStreaming2) {
+            const responseBody = await cloned.json();
+            captured.usage = extractUsage(provider, responseBody);
+          } else if (cloned.body) {
+            try {
+              const decoder = new TextDecoder();
+              const reader = cloned.body.getReader();
+              let rawSSE = "";
+              for (; ; ) {
+                const { done, value } = await reader.read();
+                if (done) break;
+                rawSSE += decoder.decode(value, { stream: true });
+              }
+              reader.releaseLock();
+              captured.usage = extractStreamUsage(provider, rawSSE);
+            } catch {
+            }
+          }
+        } catch {
+        }
+        ;
+        globalThis[LLM_REQUEST_KEY] = captured;
+      }
+      return response2;
     }
     const traceAtCall = getCurrentTrace();
     const obsCtx = getObservabilityContext();
@@ -1828,7 +1930,7 @@ function installAIInterceptor() {
     } catch {
     }
     const ctx = getCaptureContext();
-    if (!traceAtCall && !ctx && obsCtx) {
+    if (!traceAtCall && !ctx && obsCtx && !isAIWrapperActive()) {
       const id = obsCtx.nextId();
       const start = rawDateNow();
       const eventInput = { url, provider, model, prompt, messages };
@@ -1867,7 +1969,7 @@ function installAIInterceptor() {
       }
       return response2;
     }
-    if (!ctx && httpCtx) {
+    if (!ctx && httpCtx && !isAIWrapperActive()) {
       const id = httpCtx.nextId();
       const eventInput = { url, provider, model, prompt, messages };
       const frozen = getHttpFrozenEvent(id);
@@ -2016,7 +2118,7 @@ function uninstallAIInterceptor() {
     originalFetch2 = null;
   }
 }
-var AI_WRAPPER_KEY, AI_PATTERNS2, originalFetch2;
+var AI_WRAPPER_KEY, LLM_REQUEST_KEY, AI_PATTERNS2, originalFetch2;
 var init_ai_interceptor = __esm({
   "src/interceptors/ai-interceptor.ts"() {
     "use strict";
@@ -2025,6 +2127,7 @@ var init_ai_interceptor = __esm({
     init_side_effects();
     init_telemetry_push();
     AI_WRAPPER_KEY = "__elasticdash_ai_wrapper_depth__";
+    LLM_REQUEST_KEY = "__elasticdash_last_llm_request__";
     AI_PATTERNS2 = {
       openai: /https?:\/\/api\.openai\.com\/v1\/((chat\/)?completions|embeddings)/,
       anthropic: /https?:\/\/api\.anthropic\.com\/v1\/messages/,
@@ -3057,13 +3160,12 @@ var init_trigger_executor = __esm({
 function connectToBackend(options) {
   if (socket?.connected) return socket;
   if (socket) return socket;
-  const { serverUrl, apiKey, sessionId, projectId } = options;
+  const { serverUrl, apiKey, sessionId } = options;
   const cwd = process.cwd();
   socket = (0, import_socket.io)(serverUrl, {
     auth: {
       ...apiKey ? { apiKey } : {},
-      sessionId,
-      ...projectId ? { projectId } : {}
+      sessionId
     },
     transports: ["websocket", "polling"],
     reconnection: true,
@@ -3076,11 +3178,14 @@ function connectToBackend(options) {
     const workflows = scanWorkflows(cwd);
     socket.emit("register", {
       sessionId,
-      ...projectId ? { projectId } : {},
       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 {
@@ -3157,7 +3262,6 @@ function initObservability(options) {
     throw new Error("[elasticdash] initObservability: serverUrl is required (set ELASTICDASH_API_URL or pass serverUrl option)");
   }
   const apiKey = options?.apiKey ?? process.env.ELASTICDASH_API_KEY;
-  const projectId = options?.projectId ?? process.env.ELASTICDASH_PROJECT_ID;
   const sessionId = options?.sessionId ?? process.env.ELASTICDASH_SESSION_ID ?? (0, import_node_crypto3.randomUUID)();
   const sampleRate = options?.sampleRate ?? 1;
   const redactKeys = options?.redactKeys ?? [];
@@ -3194,7 +3298,7 @@ function initObservability(options) {
   interceptFetch();
   installDBAutoInterceptor().catch(() => {
   });
-  connectToBackend({ serverUrl, apiKey, sessionId, projectId });
+  connectToBackend({ serverUrl, apiKey, sessionId });
   heartbeatTimer = setInterval(() => {
     batcher.enqueue({
       id: ctx.nextId(),
@@ -4618,6 +4722,7 @@ init_recorder();
 init_side_effects();
 init_telemetry_push();
 init_mock_resolver();
+init_ai_interceptor();
 var AI_WRAPPER_KEY2 = "__elasticdash_ai_wrapper_depth__";
 var g4 = globalThis;
 if (g4[AI_WRAPPER_KEY2] == null) g4[AI_WRAPPER_KEY2] = 0;
@@ -4646,6 +4751,13 @@ function extractUsage2(output) {
         totalTokens: u.total_tokens
       };
     }
+    if (u.inputTokens != null || u.outputTokens != null) {
+      return {
+        inputTokens: u.inputTokens,
+        outputTokens: u.outputTokens,
+        totalTokens: (u.inputTokens ?? 0) + (u.outputTokens ?? 0)
+      };
+    }
   }
   if (o.usageMetadata && typeof o.usageMetadata === "object") {
     const u = o.usageMetadata;
@@ -4655,8 +4767,30 @@ function extractUsage2(output) {
       totalTokens: u.totalTokenCount
     };
   }
+  if (typeof o.tokens === "number" || typeof o.outputTokens === "number") {
+    return {
+      inputTokens: typeof o.inputTokens === "number" ? o.inputTokens : void 0,
+      outputTokens: typeof o.outputTokens === "number" ? o.outputTokens : typeof o.tokens === "number" ? o.tokens : void 0,
+      totalTokens: (typeof o.inputTokens === "number" ? o.inputTokens : 0) + (typeof o.outputTokens === "number" ? o.outputTokens : typeof o.tokens === "number" ? o.tokens : 0)
+    };
+  }
   return void 0;
 }
+function enrichFromLLMCapture(input, appUsage, fallbackModel, fallbackProvider) {
+  const captured = consumeCapturedLLMRequest();
+  if (captured) {
+    const enrichedInput = input && typeof input === "object" ? { ...input, llmRequest: captured.body, promptSnippet: captured.promptSnippet } : { originalInput: input, llmRequest: captured.body, promptSnippet: captured.promptSnippet };
+    const usage = appUsage ?? captured.usage;
+    return { input: enrichedInput, usage };
+  }
+  const resolvedModel = fallbackModel || (input && typeof input === "object" ? input.model : void 0);
+  const resolvedProvider = fallbackProvider || (input && typeof input === "object" ? input.provider : void 0);
+  if (resolvedModel || resolvedProvider) {
+    const enrichedInput = input && typeof input === "object" ? { ...input, llmRequest: { model: resolvedModel, provider: resolvedProvider } } : { originalInput: input, llmRequest: { model: resolvedModel, provider: resolvedProvider } };
+    return { input: enrichedInput, usage: appUsage };
+  }
+  return { input, usage: appUsage };
+}
 function isReadableStream2(v) {
   return typeof v === "object" && v !== null && typeof v.getReader === "function" && typeof v.tee === "function";
 }
@@ -4710,9 +4844,7 @@ function wrapAsyncIterable2(source, onComplete) {
     }
   };
 }
-function wrapAI(modelName, callFn, options) {
-  const actualModel = options?.model;
-  const actualProvider = options?.provider;
+function wrapAI(modelName, callFn, _options) {
   return async (...args) => {
     await tryAutoInitHttpContext();
     const ctx = getCaptureContext();
@@ -4724,18 +4856,18 @@ function wrapAI(modelName, callFn, options) {
       const start = rawDateNow();
       if (!ctx && !httpCtx && obsCtx) {
         const id2 = obsCtx.nextId();
-        const rawInput = args.length === 1 ? args[0] : args;
-        const input2 = actualModel || actualProvider ? { ...rawInput && typeof rawInput === "object" ? rawInput : { prompt: rawInput }, ...actualModel ? { model: actualModel } : {}, ...actualProvider ? { provider: actualProvider } : {} } : rawInput;
+        const input2 = args.length === 1 ? args[0] : args;
         try {
           const output = await callFn(...args);
+          const enriched = enrichFromLLMCapture(input2, extractUsage2(output), _options?.model, _options?.provider);
           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 });
+              pushTelemetryEvent({ id: id2, type: "ai", name: modelName, input: enriched.input, output: null, streamed: true, streamRaw: rawText, timestamp: start, durationMs: durationMs2, ...enriched.usage ? { usage: enriched.usage } : {} });
             }).catch(() => {
               const durationMs2 = rawDateNow() - start;
-              pushTelemetryEvent({ id: id2, type: "ai", name: modelName, input: input2, output: null, streamed: true, streamRaw: "", timestamp: start, durationMs: durationMs2 });
+              pushTelemetryEvent({ id: id2, type: "ai", name: modelName, input: enriched.input, output: null, streamed: true, streamRaw: "", timestamp: start, durationMs: durationMs2, ...enriched.usage ? { usage: enriched.usage } : {} });
             });
             return streamForCaller;
           }
@@ -4743,30 +4875,30 @@ function wrapAI(modelName, callFn, options) {
             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 });
+              pushTelemetryEvent({ id: id2, type: "ai", name: modelName, input: enriched.input, output: null, streamed: true, streamRaw: rawText, timestamp: start, durationMs: durationMs2, ...enriched.usage ? { usage: enriched.usage } : {} });
             });
           }
           const durationMs = rawDateNow() - start;
-          const usage = extractUsage2(output);
           const event = {
             id: id2,
             type: "ai",
             name: modelName,
-            input: input2,
+            input: enriched.input,
             output,
             timestamp: start,
             durationMs,
-            ...usage ? { usage } : {}
+            ...enriched.usage ? { usage: enriched.usage } : {}
           };
           pushTelemetryEvent(event);
           return output;
         } catch (e) {
+          const enriched = enrichFromLLMCapture(input2, void 0, _options?.model, _options?.provider);
           const durationMs = rawDateNow() - start;
           pushTelemetryEvent({
             id: id2,
             type: "ai",
             name: modelName,
-            input: input2,
+            input: enriched.input,
             output: { error: String(e) },
             timestamp: start,
             durationMs
@@ -4897,14 +5029,15 @@ function wrapAI(modelName, callFn, options) {
       const input = httpModifiedInput !== void 0 ? httpModifiedInput : rawHttpInput;
       try {
         const output = await callFn(...httpEffectiveArgs);
+        const enriched = enrichFromLLMCapture(input, extractUsage2(output), _options?.model, _options?.provider);
         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 });
+            pushTelemetryEvent({ id, type: "ai", name: modelName, input: enriched.input, output: null, streamed: true, streamRaw: rawText, timestamp: start, durationMs: durationMs2, ...enriched.usage ? { usage: enriched.usage } : {} });
           }).catch(() => {
             const durationMs2 = rawDateNow() - start;
-            pushTelemetryEvent({ id, type: "ai", name: modelName, input, output: null, streamed: true, streamRaw: "", timestamp: start, durationMs: durationMs2 });
+            pushTelemetryEvent({ id, type: "ai", name: modelName, input: enriched.input, output: null, streamed: true, streamRaw: "", timestamp: start, durationMs: durationMs2, ...enriched.usage ? { usage: enriched.usage } : {} });
           });
           return streamForCaller;
         }
@@ -4912,30 +5045,30 @@ function wrapAI(modelName, callFn, options) {
           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 });
+            pushTelemetryEvent({ id, type: "ai", name: modelName, input: enriched.input, output: null, streamed: true, streamRaw: rawText, timestamp: start, durationMs: durationMs2, ...enriched.usage ? { usage: enriched.usage } : {} });
           });
         }
         const durationMs = rawDateNow() - start;
-        const usage = extractUsage2(output);
         const event = {
           id,
           type: "ai",
           name: modelName,
-          input,
+          input: enriched.input,
           output,
           timestamp: start,
           durationMs,
-          ...usage ? { usage } : {}
+          ...enriched.usage ? { usage: enriched.usage } : {}
         };
         pushTelemetryEvent(event);
         return output;
       } catch (e) {
+        const enriched = enrichFromLLMCapture(input, void 0, _options?.model, _options?.provider);
         const durationMs = rawDateNow() - start;
         pushTelemetryEvent({
           id,
           type: "ai",
           name: modelName,
-          input,
+          input: enriched.input,
           output: { error: String(e) },
           timestamp: start,
           durationMs
@@ -5440,6 +5573,9 @@ async function startPortalServer(options) {
 
 // src/index.ts
 init_portal_executor();
+init_telemetry_push();
+tryAutoInitHttpContext().catch(() => {
+});
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   ReplayController,

package/dist/index.js

@@ -44,4 +44,11 @@ export { serializeAgentState, deserializeAgentState, extractTaskOutputs, resolve
 // Portal (remote rerun queue)
 export { startPortalServer } from './portal-server.js';
 export { executePortalTask } from './portal-executor.js';
+// ─── Eager auto-init ────────────────────────────────────────
+// When ELASTICDASH_API_KEY is set, automatically initialise observability mode
+// at import time so the socket connection is established immediately (e.g. for
+// onboarding detection). tryAutoInitHttpContext is idempotent and deduplicates
+// concurrent calls, so this is safe even if interceptors trigger it again later.
+import { tryAutoInitHttpContext } from './interceptors/telemetry-push.js';
+tryAutoInitHttpContext().catch(() => { });
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,0CAA0C;AAE1C,iBAAiB;AACjB,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,UAAU,EAAE,SAAS,EAAE,WAAW,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAA;AACnH,OAAO,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAA;AACtC,OAAO,EAAE,aAAa,EAAE,MAAM,eAAe,CAAA;AAC7C,OAAO,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAA;AACtD,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAA;AAG5C,gBAAgB;AAChB,OAAO,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,eAAe,EAAE,eAAe,EAAE,MAAM,4BAA4B,CAAA;AAGnH,4BAA4B;AAC5B,OAAO,EAAE,aAAa,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,MAAM,uBAAuB,CAAA;AAC3F,OAAO,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAA;AAItD,yBAAyB;AACzB,OAAO,EAAE,QAAQ,EAAE,MAAM,wBAAwB,CAAA;AACjD,OAAO,EAAE,MAAM,EAAE,MAAM,+BAA+B,CAAA;AAEtD,oEAAoE;AACpE,OAAO,EACL,iBAAiB,EACjB,kBAAkB,EAClB,iBAAiB,EACjB,kBAAkB,EAClB,iBAAiB,EACjB,kBAAkB,EAClB,sBAAsB,EACtB,gBAAgB,EAChB,6BAA6B,EAC7B,oBAAoB,EACpB,sBAAsB,GACvB,MAAM,kCAAkC,CAAA;AAGzC,4DAA4D;AAC5D,OAAO,EACL,WAAW,EACX,aAAa,EACb,iBAAiB,EACjB,qBAAqB,EACrB,eAAe,EACf,oBAAoB,EACpB,sBAAsB,GACvB,MAAM,8BAA8B,CAAA;AAUrC,wBAAwB;AACxB,OAAO,EAAE,MAAM,EAAE,YAAY,EAAE,QAAQ,EAAE,mBAAmB,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAA;AAC3G,OAAO,EAAE,wBAAwB,EAAE,0BAA0B,EAAE,MAAM,2BAA2B,CAAA;AAEhG,4BAA4B;AAC5B,OAAO,EAAE,cAAc,EAAE,YAAY,EAAE,kBAAkB,EAAE,MAAM,wBAAwB,CAAA;AAGzF,2BAA2B;AAC3B,OAAO,EAAE,eAAe,EAAE,aAAa,EAAE,gBAAgB,EAAE,cAAc,EAAE,MAAM,gCAAgC,CAAA;AAEjH,sCAAsC;AACtC,OAAO,EAAE,oBAAoB,EAAE,sBAAsB,EAAE,MAAM,kCAAkC,CAAA;AAE/F,gBAAgB;AAChB,OAAO,EAAE,iBAAiB,EAAE,qBAAqB,EAAE,UAAU,EAAE,QAAQ,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAA;AAEjH,OAAO,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAA;AAGzD,OAAO,EAAE,qBAAqB,EAAE,mBAAmB,EAAE,MAAM,sBAAsB,CAAA;AAGjF,qDAAqD;AACrD,OAAO,EAAE,gBAAgB,EAAE,qBAAqB,EAAE,MAAM,uBAAuB,CAAA;AAG/E,YAAY;AACZ,OAAO,EAAE,aAAa,EAAE,kBAAkB,EAAE,MAAM,wBAAwB,CAAA;AAE1E,UAAU;AACV,OAAO,EAAE,cAAc,EAAE,MAAM,cAAc,CAAA;AAC7C,OAAO,EAAE,kBAAkB,EAAE,QAAQ,EAAE,MAAM,qCAAqC,CAAA;AAElF,kBAAkB;AAClB,OAAO,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAA;AAGlD,yBAAyB;AACzB,OAAO,EAAE,mBAAmB,EAAE,qBAAqB,EAAE,kBAAkB,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAA;AAGxH,8BAA8B;AAC9B,OAAO,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAA;AACtD,OAAO,EAAE,iBAAiB,EAAE,MAAM,sBAAsB,CAAA"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,0CAA0C;AAE1C,iBAAiB;AACjB,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,UAAU,EAAE,SAAS,EAAE,WAAW,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAA;AACnH,OAAO,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAA;AACtC,OAAO,EAAE,aAAa,EAAE,MAAM,eAAe,CAAA;AAC7C,OAAO,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAA;AACtD,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAA;AAG5C,gBAAgB;AAChB,OAAO,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,eAAe,EAAE,eAAe,EAAE,MAAM,4BAA4B,CAAA;AAGnH,4BAA4B;AAC5B,OAAO,EAAE,aAAa,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,MAAM,uBAAuB,CAAA;AAC3F,OAAO,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAA;AAItD,yBAAyB;AACzB,OAAO,EAAE,QAAQ,EAAE,MAAM,wBAAwB,CAAA;AACjD,OAAO,EAAE,MAAM,EAAE,MAAM,+BAA+B,CAAA;AAEtD,oEAAoE;AACpE,OAAO,EACL,iBAAiB,EACjB,kBAAkB,EAClB,iBAAiB,EACjB,kBAAkB,EAClB,iBAAiB,EACjB,kBAAkB,EAClB,sBAAsB,EACtB,gBAAgB,EAChB,6BAA6B,EAC7B,oBAAoB,EACpB,sBAAsB,GACvB,MAAM,kCAAkC,CAAA;AAGzC,4DAA4D;AAC5D,OAAO,EACL,WAAW,EACX,aAAa,EACb,iBAAiB,EACjB,qBAAqB,EACrB,eAAe,EACf,oBAAoB,EACpB,sBAAsB,GACvB,MAAM,8BAA8B,CAAA;AAUrC,wBAAwB;AACxB,OAAO,EAAE,MAAM,EAAE,YAAY,EAAE,QAAQ,EAAE,mBAAmB,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAA;AAC3G,OAAO,EAAE,wBAAwB,EAAE,0BAA0B,EAAE,MAAM,2BAA2B,CAAA;AAEhG,4BAA4B;AAC5B,OAAO,EAAE,cAAc,EAAE,YAAY,EAAE,kBAAkB,EAAE,MAAM,wBAAwB,CAAA;AAGzF,2BAA2B;AAC3B,OAAO,EAAE,eAAe,EAAE,aAAa,EAAE,gBAAgB,EAAE,cAAc,EAAE,MAAM,gCAAgC,CAAA;AAEjH,sCAAsC;AACtC,OAAO,EAAE,oBAAoB,EAAE,sBAAsB,EAAE,MAAM,kCAAkC,CAAA;AAE/F,gBAAgB;AAChB,OAAO,EAAE,iBAAiB,EAAE,qBAAqB,EAAE,UAAU,EAAE,QAAQ,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAA;AAEjH,OAAO,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAA;AAGzD,OAAO,EAAE,qBAAqB,EAAE,mBAAmB,EAAE,MAAM,sBAAsB,CAAA;AAGjF,qDAAqD;AACrD,OAAO,EAAE,gBAAgB,EAAE,qBAAqB,EAAE,MAAM,uBAAuB,CAAA;AAG/E,YAAY;AACZ,OAAO,EAAE,aAAa,EAAE,kBAAkB,EAAE,MAAM,wBAAwB,CAAA;AAE1E,UAAU;AACV,OAAO,EAAE,cAAc,EAAE,MAAM,cAAc,CAAA;AAC7C,OAAO,EAAE,kBAAkB,EAAE,QAAQ,EAAE,MAAM,qCAAqC,CAAA;AAElF,kBAAkB;AAClB,OAAO,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAA;AAGlD,yBAAyB;AACzB,OAAO,EAAE,mBAAmB,EAAE,qBAAqB,EAAE,kBAAkB,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAA;AAGxH,8BAA8B;AAC9B,OAAO,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAA;AACtD,OAAO,EAAE,iBAAiB,EAAE,MAAM,sBAAsB,CAAA;AAGxD,+DAA+D;AAC/D,+EAA+E;AAC/E,+EAA+E;AAC/E,+EAA+E;AAC/E,iFAAiF;AACjF,OAAO,EAAE,sBAAsB,EAAE,MAAM,kCAAkC,CAAA;AACzE,sBAAsB,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAA"}

package/dist/interceptors/ai-interceptor.d.ts

@@ -1,3 +1,18 @@
+export interface CapturedLLMRequest {
+    url: string;
+    provider: string;
+    model: string;
+    messages?: unknown[];
+    body: Record<string, unknown>;
+    promptSnippet?: string;
+    usage?: UsageInfo;
+}
+export declare function consumeCapturedLLMRequest(): CapturedLLMRequest | undefined;
+type UsageInfo = {
+    inputTokens?: number;
+    outputTokens?: number;
+    totalTokens?: number;
+};
 /**
  * Install the AI fetch interceptor. Wraps globalThis.fetch to automatically
  * record LLM steps into the active trace for OpenAI, Gemini, and Grok calls.
@@ -7,4 +22,5 @@ export declare function installAIInterceptor(): void;
  * Uninstall the AI fetch interceptor, restoring globalThis.fetch to its original value.
  */
 export declare function uninstallAIInterceptor(): void;
+export {};
 //# sourceMappingURL=ai-interceptor.d.ts.map

package/dist/interceptors/ai-interceptor.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ai-interceptor.d.ts","sourceRoot":"","sources":["../../src/interceptors/ai-interceptor.ts"],"names":[],"mappings":"AA8XA;;;GAGG;AACH,wBAAgB,oBAAoB,IAAI,IAAI,CA2Q3C;AAED;;GAEG;AACH,wBAAgB,sBAAsB,IAAI,IAAI,CAK7C"}
+{"version":3,"file":"ai-interceptor.d.ts","sourceRoot":"","sources":["../../src/interceptors/ai-interceptor.ts"],"names":[],"mappings":"AAqBA,MAAM,WAAW,kBAAkB;IACjC,GAAG,EAAE,MAAM,CAAA;IACX,QAAQ,EAAE,MAAM,CAAA;IAChB,KAAK,EAAE,MAAM,CAAA;IACb,QAAQ,CAAC,EAAE,OAAO,EAAE,CAAA;IACpB,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;IAC7B,aAAa,CAAC,EAAE,MAAM,CAAA;IACtB,KAAK,CAAC,EAAE,SAAS,CAAA;CAClB;AAED,wBAAgB,yBAAyB,IAAI,kBAAkB,GAAG,SAAS,CAK1E;AA6BD,KAAK,SAAS,GAAG;IAAE,WAAW,CAAC,EAAE,MAAM,CAAC;IAAC,YAAY,CAAC,EAAE,MAAM,CAAC;IAAC,WAAW,CAAC,EAAE,MAAM,CAAA;CAAE,CAAA;AA2YtF;;;GAGG;AACH,wBAAgB,oBAAoB,IAAI,IAAI,CAmU3C;AAED;;GAEG;AACH,wBAAgB,sBAAsB,IAAI,IAAI,CAK7C"}

package/dist/interceptors/ai-interceptor.js

@@ -11,6 +11,46 @@ const AI_WRAPPER_KEY = '__elasticdash_ai_wrapper_depth__';
 function isAIWrapperActive() {
     return (globalThis[AI_WRAPPER_KEY] ?? 0) > 0;
 }
+/**
+ * When inside a wrapAI call, the ai-interceptor captures the actual HTTP
+ * request payload and stashes it here so wrapAI can attach it to its event.
+ */
+const LLM_REQUEST_KEY = '__elasticdash_last_llm_request__';
+export function consumeCapturedLLMRequest() {
+    const g = globalThis;
+    const req = g[LLM_REQUEST_KEY];
+    if (req)
+        g[LLM_REQUEST_KEY] = undefined;
+    return req;
+}
+function extractPromptSnippet(body) {
+    let messages;
+    if (Array.isArray(body.messages))
+        messages = body.messages;
+    else if (Array.isArray(body.contents))
+        messages = body.contents;
+    if (!messages || messages.length === 0)
+        return undefined;
+    // Find the last user message
+    for (let i = messages.length - 1; i >= 0; i--) {
+        const msg = messages[i];
+        if (!msg)
+            continue;
+        if (msg.role === 'user') {
+            let content = msg.content;
+            if (Array.isArray(content)) {
+                content = content
+                    .map((b) => (b && typeof b === 'object' ? String(b.text ?? '') : String(b)))
+                    .filter(Boolean)
+                    .join('');
+            }
+            if (typeof content === 'string') {
+                return content.slice(0, 100);
+            }
+        }
+    }
+    return undefined;
+}
 function extractUsage(provider, body) {
     if (provider === 'openai' || provider === 'grok' || provider === 'kimi') {
         const u = body.usage;
@@ -312,6 +352,40 @@ async function bufferSSEStream(provider, stream) {
     }
     return completion;
 }
+/** Extract usage from buffered raw SSE text */
+function extractStreamUsage(provider, rawSSE) {
+    const lines = rawSSE.split('\n');
+    // Walk backwards to find usage in the final events
+    for (let i = lines.length - 1; i >= 0; i--) {
+        const line = lines[i];
+        if (!line.startsWith('data: '))
+            continue;
+        const data = line.slice(6).trim();
+        if (data === '[DONE]')
+            continue;
+        try {
+            const obj = JSON.parse(data);
+            // OpenAI / Grok / Kimi: usage in the final chunk
+            const usage = extractUsage(provider, obj);
+            if (usage)
+                return usage;
+            // Anthropic: usage in message_delta event
+            if (obj.type === 'message_delta') {
+                const u = obj.usage;
+                if (u)
+                    return extractUsage('anthropic', { usage: u });
+            }
+            // Anthropic: usage in message_start event
+            if (obj.type === 'message_start') {
+                const msg = obj.message;
+                if (msg?.usage)
+                    return extractUsage('anthropic', { usage: msg.usage });
+            }
+        }
+        catch { /* skip */ }
+    }
+    return undefined;
+}
 /** Build a minimal non-streaming JSON response body from a completion string (for replay) */
 function synthesizeCompletionJSON(provider, completion) {
     if (provider === 'gemini') {
@@ -382,9 +456,66 @@ export function installAIInterceptor() {
     globalThis.fetch = async function patchedFetch(input, init) {
         const url = typeof input === 'string' ? input : input instanceof URL ? input.href : input.url;
         const provider = detectProvider(url);
-        // Skip recording when inside a wrapAI call to avoid duplicate events
+        // Skip recording when inside a wrapAI call to avoid duplicate events,
+        // but capture the actual HTTP request body and response usage so wrapAI
+        // can attach them to its event.
         if (provider && isAIWrapperActive()) {
-            return originalFetch(input, init);
+            let capturedReq;
+            let capturedModel = 'unknown';
+            let capturedMessages;
+            let capturedSnippet;
+            try {
+                const rawBody = init?.body;
+                if (rawBody && typeof rawBody === 'string') {
+                    capturedReq = JSON.parse(rawBody);
+                    capturedModel = extractModel(provider, capturedReq, url);
+                    capturedMessages = Array.isArray(capturedReq.messages) ? capturedReq.messages : Array.isArray(capturedReq.contents) ? capturedReq.contents : undefined;
+                    capturedSnippet = extractPromptSnippet(capturedReq);
+                }
+            }
+            catch {
+                // Ignore parse errors
+            }
+            const response = await originalFetch(input, init);
+            // Extract usage from the response (clone to avoid consuming the body)
+            if (capturedReq) {
+                const captured = {
+                    url, provider, model: capturedModel, messages: capturedMessages,
+                    body: capturedReq, promptSnippet: capturedSnippet,
+                };
+                const isStreaming = capturedReq.stream === true;
+                try {
+                    const cloned = response.clone();
+                    if (!isStreaming) {
+                        // Non-streaming: parse JSON response for usage
+                        const responseBody = await cloned.json();
+                        captured.usage = extractUsage(provider, responseBody);
+                    }
+                    else if (cloned.body) {
+                        // Streaming: read the raw SSE text to extract usage from final events
+                        try {
+                            const decoder = new TextDecoder();
+                            const reader = cloned.body.getReader();
+                            let rawSSE = '';
+                            for (;;) {
+                                const { done, value } = await reader.read();
+                                if (done)
+                                    break;
+                                rawSSE += decoder.decode(value, { stream: true });
+                            }
+                            reader.releaseLock();
+                            captured.usage = extractStreamUsage(provider, rawSSE);
+                        }
+                        catch { /* stream read failed */ }
+                    }
+                }
+                catch {
+                    // Response body not available — usage won't be captured
+                }
+                ;
+                globalThis[LLM_REQUEST_KEY] = captured;
+            }
+            return response;
         }
         const traceAtCall = getCurrentTrace();
         const obsCtx = getObservabilityContext();
@@ -417,7 +548,8 @@ export function installAIInterceptor() {
         }
         const ctx = getCaptureContext();
         // Observability-only mode: no trace handle, no capture context — record via pushTelemetryEvent
-        if (!traceAtCall && !ctx && obsCtx) {
+        // Skip when inside a wrapAI call to avoid duplicate events (wrapAI records its own)
+        if (!traceAtCall && !ctx && obsCtx && !isAIWrapperActive()) {
             const id = obsCtx.nextId();
             const start = rawDateNow();
             const eventInput = { url, provider, model, prompt, messages };
@@ -453,7 +585,8 @@ export function installAIInterceptor() {
             return response;
         }
         // HTTP mode (no capture context): replay frozen AI events or execute live + push telemetry
-        if (!ctx && httpCtx) {
+        // Skip when inside a wrapAI call to avoid duplicate events (wrapAI records its own)
+        if (!ctx && httpCtx && !isAIWrapperActive()) {
             const id = httpCtx.nextId();
             const eventInput = { url, provider, model, prompt, messages };
             // Replay frozen step