raindrop-ai 0.3.0 → 0.3.1

package/dist/index.js

@@ -5778,7 +5778,7 @@ if (!globalThis.RAINDROP_ASYNC_LOCAL_STORAGE) {
   globalThis.RAINDROP_ASYNC_LOCAL_STORAGE = import_node_async_hooks.AsyncLocalStorage;
 }
 
-// ../core/dist/chunk-YUIXRQWN.js
+// ../core/dist/chunk-3V4MIA75.js
 function normalizeFeatureFlags(input) {
   if (Array.isArray(input)) {
     return Object.fromEntries(input.map((name) => [name, "true"]));
@@ -6326,6 +6326,20 @@ function stringifyBounded(value, limit) {
   }
   return json.length <= max ? json : truncateToLimit(json, max);
 }
+function attrString(key, value) {
+  if (value === void 0) return void 0;
+  return { key, value: { stringValue: value } };
+}
+function attrInt(key, value) {
+  if (value === void 0) return void 0;
+  if (!Number.isFinite(value)) return void 0;
+  return { key, value: { intValue: String(Math.trunc(value)) } };
+}
+function attrDouble(key, value) {
+  if (value === void 0) return void 0;
+  if (!Number.isFinite(value)) return void 0;
+  return { key, value: { doubleValue: value } };
+}
 var LOCAL_DEBUGGER_ENV_VAR = "RAINDROP_LOCAL_DEBUGGER";
 var WORKSHOP_ENV_VAR = "RAINDROP_WORKSHOP";
 var DEFAULT_LOCAL_WORKSHOP_URL = "http://localhost:5899/v1/";
@@ -6456,6 +6470,7 @@ function projectIdHeaders(projectId) {
 var SHUTDOWN_DEADLINE_MS = 1e4;
 var POST_SHUTDOWN_TIMEOUT_MS = 5e3;
 var MODEL_USAGE_ATTRIBUTES = {
+  usageSource: "raindrop.usage.source",
   providerName: "gen_ai.provider.name",
   requestModel: "gen_ai.request.model",
   responseModel: "gen_ai.response.model",
@@ -6464,6 +6479,7 @@ var MODEL_USAGE_ATTRIBUTES = {
   reasoningTokens: "gen_ai.usage.reasoning_tokens",
   cacheReadInputTokens: "gen_ai.usage.cache_read_input_tokens",
   cacheWriteInputTokens: "gen_ai.usage.cache_write_input_tokens",
+  providerReportedCost: "gen_ai.usage.provider_reported_cost",
   nonCachedInputTokens: "raindrop.usage.input_tokens.non_cached",
   nonReasoningOutputTokens: "raindrop.usage.output_tokens.non_reasoning",
   totalOutputTokens: "raindrop.usage.output_tokens.total",
@@ -6471,6 +6487,47 @@ var MODEL_USAGE_ATTRIBUTES = {
   cacheReadTokens: "raindrop.usage.input_tokens.cache_read",
   cacheWriteTokens: "raindrop.usage.input_tokens.cache_write"
 };
+var MODEL_PROVIDER_NAMES = [
+  ["google.vertex.anthropic", "google-vertex"],
+  ["google.vertex_ai", "google-vertex"],
+  ["google.vertex", "google-vertex"],
+  ["gcp.vertex_ai", "google-vertex"],
+  ["gcp.gemini", "google"],
+  ["google_vertexai", "google-vertex"],
+  ["vertex.anthropic", "google-vertex"],
+  ["amazon-bedrock", "amazon-bedrock"],
+  ["aws.bedrock", "amazon-bedrock"],
+  ["bedrock", "amazon-bedrock"],
+  ["anthropic", "anthropic"],
+  ["openai", "openai"],
+  ["google", "google"],
+  ["az.ai.openai", "azure"],
+  ["azure", "azure"],
+  ["openrouter", "openrouter"]
+];
+function canonicalModelProvider(provider) {
+  var _a;
+  const normalizedProvider = provider.trim().toLowerCase();
+  if (normalizedProvider === "gateway") return "vercel";
+  const match = MODEL_PROVIDER_NAMES.find(
+    ([family]) => normalizedProvider === family || normalizedProvider.startsWith(`${family}.`)
+  );
+  return (_a = match == null ? void 0 : match[1]) != null ? _a : normalizedProvider;
+}
+function tokenCount(value) {
+  return typeof value === "number" && Number.isSafeInteger(value) && value >= 0 ? value : void 0;
+}
+function buildRawModelUsageAttributes(usage) {
+  const providerReportedCost = typeof usage.providerReportedCost === "number" && Number.isFinite(usage.providerReportedCost) && usage.providerReportedCost >= 0 ? usage.providerReportedCost : void 0;
+  return [
+    attrInt(MODEL_USAGE_ATTRIBUTES.inputTokens, tokenCount(usage.inputTokens)),
+    attrInt(MODEL_USAGE_ATTRIBUTES.outputTokens, tokenCount(usage.outputTokens)),
+    attrInt(MODEL_USAGE_ATTRIBUTES.reasoningTokens, tokenCount(usage.reasoningTokens)),
+    attrInt(MODEL_USAGE_ATTRIBUTES.cacheReadInputTokens, tokenCount(usage.cacheReadInputTokens)),
+    attrInt(MODEL_USAGE_ATTRIBUTES.cacheWriteInputTokens, tokenCount(usage.cacheWriteInputTokens)),
+    attrDouble(MODEL_USAGE_ATTRIBUTES.providerReportedCost, providerReportedCost)
+  ].filter((attribute) => attribute !== void 0);
+}
 var STRING_ALIASES = {
   [MODEL_USAGE_ATTRIBUTES.requestModel]: ["ai.model.id", "ai.model"]
 };
@@ -10920,7 +10977,7 @@ var SignalEventSchema = external_exports.object({
 // package.json
 var package_default = {
   name: "raindrop-ai",
-  version: "0.3.0",
+  version: "0.3.1",
   main: "dist/index.js",
   module: "dist/index.mjs",
   types: "dist/index.d.ts",
@@ -11602,7 +11659,7 @@ var DirectTraceShipper = class {
    * so spans still appear as children in the trace tree.
    */
   sendSpan(params) {
-    var _a, _b;
+    var _a, _b, _c;
     if (!this.enabled) return;
     let traceIdB64 = params.traceId;
     let parentSpanIdB64 = params.parentSpanId;
@@ -11646,6 +11703,7 @@ var DirectTraceShipper = class {
     })) {
       attributes.push(attrString2(key, value));
     }
+    attributes.push(...(_c = params.otlpAttributes) != null ? _c : []);
     let status;
     if (params.error) {
       status = { code: 2, message: params.error };
@@ -11797,9 +11855,62 @@ var DirectTraceShipper = class {
   }
 };
 
+// src/tracing/exit-flush.ts
+var traceloop = __toESM(require("@traceloop/node-server-sdk"));
+var STATE_KEY2 = /* @__PURE__ */ Symbol.for("raindrop.tracing.exitFlushState");
+function getState2() {
+  const holder = globalThis;
+  let state = holder[STATE_KEY2];
+  if (!state) {
+    state = {
+      clientFlushers: /* @__PURE__ */ new Set(),
+      pipelineFlushNeeded: false,
+      handlerRegistered: false
+    };
+    holder[STATE_KEY2] = state;
+  }
+  return state;
+}
+function ensureExitHandler(state) {
+  if (state.handlerRegistered) return;
+  if (typeof process === "undefined" || typeof process.once !== "function") return;
+  state.handlerRegistered = true;
+  process.once("beforeExit", () => {
+    void flushAll();
+  });
+}
+function registerClientFlusher(flush) {
+  const state = getState2();
+  state.clientFlushers.add(flush);
+  ensureExitHandler(state);
+  return () => {
+    state.clientFlushers.delete(flush);
+  };
+}
+function markPipelineForExitFlush() {
+  const state = getState2();
+  state.pipelineFlushNeeded = true;
+  ensureExitHandler(state);
+}
+async function flushAll() {
+  const state = getState2();
+  const tasks = [];
+  for (const flush of state.clientFlushers) {
+    tasks.push(Promise.resolve().then(flush).catch(() => {
+    }));
+  }
+  if (state.pipelineFlushNeeded) {
+    tasks.push(
+      Promise.resolve().then(() => traceloop.forceFlush()).catch(() => {
+      })
+    );
+  }
+  await Promise.all(tasks);
+}
+
 // src/tracing/liveInteraction.ts
 var import_api3 = require("@opentelemetry/api");
-var traceloop = __toESM(require("@traceloop/node-server-sdk"));
+var traceloop2 = __toESM(require("@traceloop/node-server-sdk"));
 
 // src/tracing/subagents.ts
 var import_api2 = require("@opentelemetry/api");
@@ -12223,6 +12334,7 @@ var DirectToolSpan = class {
 };
 var LiveInteraction = class {
   constructor(context6, traceId, analytics, directShipper, routing) {
+    this.usageEmitted = false;
     this.context = context6;
     this.analytics = analytics;
     this.tracer = import_api3.trace.getTracer("traceloop.tracer");
@@ -12307,10 +12419,7 @@ var LiveInteraction = class {
         spanId: base64ToHex2(spanIdB64),
         traceFlags: import_api3.TraceFlags.SAMPLED
       };
-      const wrappedContext = import_api3.trace.setSpan(
-        import_api3.context.active(),
-        import_api3.trace.wrapSpanContext(spanContext)
-      );
+      const wrappedContext = import_api3.trace.setSpan(import_api3.context.active(), import_api3.trace.wrapSpanContext(spanContext));
       const invoke = () => this.routing ? withRoutingContext(this.routing, () => fn.apply(thisArg, args)) : fn.apply(thisArg, args);
       try {
         const result = await import_api3.context.with(wrappedContext, invoke);
@@ -12361,7 +12470,7 @@ var LiveInteraction = class {
       }
       return fn.apply(thisArg, fnArgs);
     };
-    return traceloop.withTask(
+    return traceloop2.withTask(
       {
         name: taskName,
         associationProperties: serializeAssociationProperties(properties),
@@ -12402,10 +12511,7 @@ var LiveInteraction = class {
         spanId: base64ToHex2(spanIdB64),
         traceFlags: import_api3.TraceFlags.SAMPLED
       };
-      const wrappedContext = import_api3.trace.setSpan(
-        import_api3.context.active(),
-        import_api3.trace.wrapSpanContext(spanContext)
-      );
+      const wrappedContext = import_api3.trace.setSpan(import_api3.context.active(), import_api3.trace.wrapSpanContext(spanContext));
       const invoke = () => this.routing ? withRoutingContext(this.routing, () => fn.apply(thisArg, args)) : fn.apply(thisArg, args);
       let result;
       let error;
@@ -12515,7 +12621,7 @@ var LiveInteraction = class {
         throw error;
       }
     };
-    return traceloop.withTool(
+    return traceloop2.withTool(
       {
         name: toolName,
         associationProperties: serializeAssociationProperties(properties),
@@ -12818,7 +12924,7 @@ var LiveInteraction = class {
     span.end();
   }
   async finish(resultEvent) {
-    var _a, _b, _c, _d;
+    var _a, _b, _c, _d, _e, _f, _g;
     unbindRoutingContext(this.boundContext);
     this.boundContext = void 0;
     unbindSpanAttributes(this.toolEventsFrame);
@@ -12826,6 +12932,15 @@ var LiveInteraction = class {
     if (!this.traceId) {
       this.resolveLiveTraceId();
     }
+    const eventId = resultEvent.eventId || this.context.eventId;
+    if (!this.usageEmitted) {
+      (_c = this.analytics) == null ? void 0 : _c._recordModelUsage(
+        eventId,
+        (_a = resultEvent.model) != null ? _a : this.context.model,
+        (_b = resultEvent.usage) != null ? _b : this.context.usage
+      );
+      this.usageEmitted = true;
+    }
     const spanFlush = (async () => {
       var _a2;
       try {
@@ -12833,21 +12948,20 @@ var LiveInteraction = class {
       } catch (e) {
       }
     })();
-    if ((_b = (_a = this.analytics) == null ? void 0 : _a._shouldBlockFinishFlush()) != null ? _b : true) {
+    if ((_e = (_d = this.analytics) == null ? void 0 : _d._shouldBlockFinishFlush()) != null ? _e : true) {
       await spanFlush;
     } else {
       void spanFlush.catch(() => {
       });
     }
-    const eventId = resultEvent.eventId || this.context.eventId;
     try {
-      await ((_c = this.analytics) == null ? void 0 : _c._drainInFlightPartials(eventId));
+      await ((_f = this.analytics) == null ? void 0 : _f._drainInFlightPartials(eventId));
     } catch (e) {
     }
     if (this.traceId) {
       this.setProperty("trace_id", this.traceId);
     }
-    await ((_d = this.analytics) == null ? void 0 : _d._trackAiPartial({
+    await ((_g = this.analytics) == null ? void 0 : _g._trackAiPartial({
       ...resultEvent,
       eventId,
       isPending: false
@@ -12964,7 +13078,7 @@ var LiveInteraction = class {
 
 // src/tracing/liveTracer.ts
 var import_api4 = require("@opentelemetry/api");
-var traceloop2 = __toESM(require("@traceloop/node-server-sdk"));
+var traceloop3 = __toESM(require("@traceloop/node-server-sdk"));
 var LiveTracer = class {
   constructor(globalProperties, directShipper, routing) {
     this.tracer = import_api4.trace.getTracer("traceloop.tracer");
@@ -13008,7 +13122,7 @@ var LiveTracer = class {
       }
       return fn.apply(thisArg, fnArgs);
     };
-    return traceloop2.withTask(
+    return traceloop3.withTask(
       {
         name: taskName,
         associationProperties: properties,
@@ -13120,59 +13234,6 @@ var LiveTracer = class {
   }
 };
 
-// src/tracing/exit-flush.ts
-var traceloop3 = __toESM(require("@traceloop/node-server-sdk"));
-var STATE_KEY2 = /* @__PURE__ */ Symbol.for("raindrop.tracing.exitFlushState");
-function getState2() {
-  const holder = globalThis;
-  let state = holder[STATE_KEY2];
-  if (!state) {
-    state = {
-      clientFlushers: /* @__PURE__ */ new Set(),
-      pipelineFlushNeeded: false,
-      handlerRegistered: false
-    };
-    holder[STATE_KEY2] = state;
-  }
-  return state;
-}
-function ensureExitHandler(state) {
-  if (state.handlerRegistered) return;
-  if (typeof process === "undefined" || typeof process.once !== "function") return;
-  state.handlerRegistered = true;
-  process.once("beforeExit", () => {
-    void flushAll();
-  });
-}
-function registerClientFlusher(flush) {
-  const state = getState2();
-  state.clientFlushers.add(flush);
-  ensureExitHandler(state);
-  return () => {
-    state.clientFlushers.delete(flush);
-  };
-}
-function markPipelineForExitFlush() {
-  const state = getState2();
-  state.pipelineFlushNeeded = true;
-  ensureExitHandler(state);
-}
-async function flushAll() {
-  const state = getState2();
-  const tasks = [];
-  for (const flush of state.clientFlushers) {
-    tasks.push(Promise.resolve().then(flush).catch(() => {
-    }));
-  }
-  if (state.pipelineFlushNeeded) {
-    tasks.push(
-      Promise.resolve().then(() => traceloop3.forceFlush()).catch(() => {
-      })
-    );
-  }
-  await Promise.all(tasks);
-}
-
 // src/tracing/tracer-core.ts
 if (localDebuggerEnabled()) {
   if (!process.env.OTEL_EXPORTER_OTLP_PROTOCOL) {
@@ -13890,6 +13951,59 @@ var tracing = (analytics, apiUrl, writeKey, options, isDebug = false) => {
       ...frame ? { frame, frameOwner } : {}
     });
   };
+  const recordModelUsage = (params) => {
+    try {
+      const attributes = [
+        attrString(MODEL_USAGE_ATTRIBUTES.usageSource, "manual"),
+        attrString(
+          MODEL_USAGE_ATTRIBUTES.providerName,
+          canonicalModelProvider(params.usage.provider)
+        ),
+        attrString(MODEL_USAGE_ATTRIBUTES.requestModel, params.model),
+        attrString(MODEL_USAGE_ATTRIBUTES.responseModel, params.model),
+        ...buildRawModelUsageAttributes({
+          inputTokens: params.usage.inputTokens,
+          outputTokens: params.usage.outputTokens,
+          cacheReadInputTokens: params.usage.cacheReadTokens,
+          cacheWriteInputTokens: params.usage.cacheWriteTokens,
+          reasoningTokens: params.usage.reasoningTokens,
+          providerReportedCost: params.usage.providerReportedCost
+        })
+      ].filter((attribute) => attribute !== void 0);
+      if (directShipper) {
+        const timeMs = Date.now();
+        directShipper.sendSpan({
+          name: "ai.model.usage",
+          spanKind: "llm",
+          properties: { event_id: params.eventId },
+          startTimeMs: timeMs,
+          endTimeMs: timeMs,
+          durationMs: 0,
+          otlpAttributes: attributes
+        });
+        return;
+      }
+      const span = import_api5.trace.getTracer("traceloop.tracer").startSpan("ai.model.usage");
+      span.setAttribute("traceloop.span.kind", "llm");
+      span.setAttribute("traceloop.association.properties.event_id", params.eventId);
+      stampSpan(span, projectId, authHint);
+      for (const attribute of attributes) {
+        const value = attribute.value;
+        if (value.stringValue !== void 0) {
+          span.setAttribute(attribute.key, value.stringValue);
+        } else if (value.intValue !== void 0) {
+          const parsed = Number(value.intValue);
+          if (Number.isSafeInteger(parsed)) span.setAttribute(attribute.key, parsed);
+        } else if (value.doubleValue !== void 0) {
+          span.setAttribute(attribute.key, value.doubleValue);
+        } else if (value.boolValue !== void 0) {
+          span.setAttribute(attribute.key, value.boolValue);
+        }
+      }
+      span.end();
+    } catch (e) {
+    }
+  };
   return {
     /**
      * Returns instrumentation instances for your NodeSDK.
@@ -14001,6 +14115,7 @@ var tracing = (analytics, apiUrl, writeKey, options, isDebug = false) => {
     begin(traceContext) {
       return beginInteraction(traceContext);
     },
+    recordModelUsage,
     resumeSubagent(carrierOrHeaders, options2 = {}) {
       return openSubagentRun(carrierOrHeaders, options2);
     },
@@ -14081,6 +14196,15 @@ var PartialClientAiTrack = ClientAiTrack.extend({
     convo_id: import_zod.z.string().optional().nullable()
   }).optional()
 });
+var AiTrackUsageSchema = import_zod.z.object({
+  provider: import_zod.z.string().trim().min(1),
+  inputTokens: import_zod.z.number().int().nonnegative().max(Number.MAX_SAFE_INTEGER),
+  outputTokens: import_zod.z.number().int().nonnegative().max(Number.MAX_SAFE_INTEGER),
+  cacheReadTokens: import_zod.z.number().int().nonnegative().max(Number.MAX_SAFE_INTEGER).optional(),
+  cacheWriteTokens: import_zod.z.number().int().nonnegative().max(Number.MAX_SAFE_INTEGER).optional(),
+  reasoningTokens: import_zod.z.number().int().nonnegative().max(Number.MAX_SAFE_INTEGER).optional(),
+  providerReportedCost: import_zod.z.number().finite().nonnegative().optional()
+});
 var AttachmentTypeSchema = import_zod.z.enum(["code", "text", "image", "iframe"]);
 
 // src/index.ts
@@ -14588,6 +14712,7 @@ var Raindrop = class {
         attachments,
         properties,
         featureFlags,
+        usage,
         timestamp,
         ...rest
       } = e;
@@ -14637,6 +14762,7 @@ var Raindrop = class {
         type: "ai",
         ...toSave
       });
+      this._recordModelUsage(customEventId, model, usage);
       return customEventId;
     };
     if (Array.isArray(event)) {
@@ -15062,6 +15188,7 @@ var Raindrop = class {
       attachments,
       properties,
       featureFlags,
+      usage: _usage,
       ...rest
     } = accumulatedEvent;
     const normalizedFlags = featureFlags ? normalizeFeatureFlags(featureFlags) : void 0;
@@ -15205,6 +15332,29 @@ var Raindrop = class {
     } catch (e) {
     }
   }
+  /**
+   * Emit one model span carrying raw provider usage facts.
+   * @internal
+   */
+  _recordModelUsage(eventId, model, usage) {
+    if (usage === void 0) return;
+    const parsed = AiTrackUsageSchema.safeParse(usage);
+    if (!parsed.success) {
+      console.warn(
+        `[raindrop] Invalid usage provided to .track_ai(): ${this.formatZodError(parsed.error)}`
+      );
+      return;
+    }
+    if (typeof model !== "string" || model.trim() === "") {
+      console.warn("[raindrop] .track_ai() usage requires a non-empty model.");
+      return;
+    }
+    this._tracing.recordModelUsage({
+      eventId,
+      model,
+      usage: parsed.data
+    });
+  }
   /**
    * Wait for already-issued partial-event POSTs to settle. Used to order the
    * terminal partial-event POST after any earlier in-flight POST for the same

package/dist/index.mjs

@@ -21,7 +21,7 @@ import {
   stringifyBounded,
   tracing,
   withRoutingContext
-} from "./chunk-5Y3LWBKV.mjs";
+} from "./chunk-4TM4IWHB.mjs";
 import {
   __commonJS,
   __toESM
@@ -9973,6 +9973,15 @@ var PartialClientAiTrack = ClientAiTrack.extend({
     convo_id: z.string().optional().nullable()
   }).optional()
 });
+var AiTrackUsageSchema = z.object({
+  provider: z.string().trim().min(1),
+  inputTokens: z.number().int().nonnegative().max(Number.MAX_SAFE_INTEGER),
+  outputTokens: z.number().int().nonnegative().max(Number.MAX_SAFE_INTEGER),
+  cacheReadTokens: z.number().int().nonnegative().max(Number.MAX_SAFE_INTEGER).optional(),
+  cacheWriteTokens: z.number().int().nonnegative().max(Number.MAX_SAFE_INTEGER).optional(),
+  reasoningTokens: z.number().int().nonnegative().max(Number.MAX_SAFE_INTEGER).optional(),
+  providerReportedCost: z.number().finite().nonnegative().optional()
+});
 var AttachmentTypeSchema = z.enum(["code", "text", "image", "iframe"]);
 
 // src/index.ts
@@ -10480,6 +10489,7 @@ var Raindrop = class {
         attachments,
         properties,
         featureFlags,
+        usage,
         timestamp,
         ...rest
       } = e;
@@ -10529,6 +10539,7 @@ var Raindrop = class {
         type: "ai",
         ...toSave
       });
+      this._recordModelUsage(customEventId, model, usage);
       return customEventId;
     };
     if (Array.isArray(event)) {
@@ -10954,6 +10965,7 @@ var Raindrop = class {
       attachments,
       properties,
       featureFlags,
+      usage: _usage,
       ...rest
     } = accumulatedEvent;
     const normalizedFlags = featureFlags ? normalizeFeatureFlags(featureFlags) : void 0;
@@ -11097,6 +11109,29 @@ var Raindrop = class {
     } catch (e) {
     }
   }
+  /**
+   * Emit one model span carrying raw provider usage facts.
+   * @internal
+   */
+  _recordModelUsage(eventId, model, usage) {
+    if (usage === void 0) return;
+    const parsed = AiTrackUsageSchema.safeParse(usage);
+    if (!parsed.success) {
+      console.warn(
+        `[raindrop] Invalid usage provided to .track_ai(): ${this.formatZodError(parsed.error)}`
+      );
+      return;
+    }
+    if (typeof model !== "string" || model.trim() === "") {
+      console.warn("[raindrop] .track_ai() usage requires a non-empty model.");
+      return;
+    }
+    this._tracing.recordModelUsage({
+      eventId,
+      model,
+      usage: parsed.data
+    });
+  }
   /**
    * Wait for already-issued partial-event POSTs to settle. Used to order the
    * terminal partial-event POST after any earlier in-flight POST for the same

package/dist/tracing/index.d.mts

@@ -1,24 +1,9 @@
-import { R as Raindrop, a as RaindropSpanProcessor, P as PartialAiTrackEvent, I as Interaction, C as CarrierHeaders, T as TraceCarrier, S as SubagentResumeOptions, b as SubagentRun, c as Tracer } from '../index-DBdUrUXc.mjs';
 import { Instrumentation } from '@opentelemetry/instrumentation';
+import { R as Raindrop, a as RaindropSpanProcessor, P as PartialAiTrackEvent, I as Interaction, A as AiTrackUsage, C as CarrierHeaders, T as TraceCarrier, S as SubagentResumeOptions, b as SubagentRun, c as Tracer } from '../index-DuXeVnSq.mjs';
 import * as traceloop from '@traceloop/node-server-sdk';
 import '@opentelemetry/api';
 import 'zod';
 
-/**
- * Tracing core - the real tracing implementation.
- *
- * Two modes:
- *
- * 1. DEFAULT: Raindrop manages everything
- *    - Creates NodeSDK, registers instrumentations, exports traces
- *    - Customer just uses raindrop.begin(), withSpan(), etc.
- *
- * 2. EXTERNAL OTEL (useExternalOtel: true): Customer has their own NodeSDK
- *    - Raindrop does NOT create a NodeSDK (would conflict!)
- *    - Customer adds raindrop.createSpanProcessor() to their NodeSDK
- *    - Customer calls raindrop.getInstrumentations() to get configured instrumentations
- */
-
 interface TracingOptions extends traceloop.InitializeOptions {
     /**
      * Set to true when you have your own OpenTelemetry NodeSDK.
@@ -103,6 +88,11 @@ declare const tracing: (analytics: Raindrop, apiUrl: string, writeKey: string, o
      */
     createSpanProcessor(processorOptions?: traceloop.SpanProcessorOptions): RaindropSpanProcessor;
     begin(traceContext: PartialAiTrackEvent): Interaction;
+    recordModelUsage: (params: {
+        eventId: string;
+        model: string;
+        usage: AiTrackUsage;
+    }) => void;
     resumeSubagent(carrierOrHeaders: CarrierHeaders | TraceCarrier | null | undefined, options?: SubagentResumeOptions): SubagentRun;
     withSubagentRun<T>(carrierOrHeaders: CarrierHeaders | TraceCarrier | null | undefined, fn: (run: SubagentRun) => Promise<T> | T, options?: SubagentResumeOptions): Promise<T>;
     getActiveInteraction(eventId: string): Interaction;

package/dist/tracing/index.d.ts

@@ -1,24 +1,9 @@
-import { R as Raindrop, a as RaindropSpanProcessor, P as PartialAiTrackEvent, I as Interaction, C as CarrierHeaders, T as TraceCarrier, S as SubagentResumeOptions, b as SubagentRun, c as Tracer } from '../index-DBdUrUXc.js';
 import { Instrumentation } from '@opentelemetry/instrumentation';
+import { R as Raindrop, a as RaindropSpanProcessor, P as PartialAiTrackEvent, I as Interaction, A as AiTrackUsage, C as CarrierHeaders, T as TraceCarrier, S as SubagentResumeOptions, b as SubagentRun, c as Tracer } from '../index-DuXeVnSq.js';
 import * as traceloop from '@traceloop/node-server-sdk';
 import '@opentelemetry/api';
 import 'zod';
 
-/**
- * Tracing core - the real tracing implementation.
- *
- * Two modes:
- *
- * 1. DEFAULT: Raindrop manages everything
- *    - Creates NodeSDK, registers instrumentations, exports traces
- *    - Customer just uses raindrop.begin(), withSpan(), etc.
- *
- * 2. EXTERNAL OTEL (useExternalOtel: true): Customer has their own NodeSDK
- *    - Raindrop does NOT create a NodeSDK (would conflict!)
- *    - Customer adds raindrop.createSpanProcessor() to their NodeSDK
- *    - Customer calls raindrop.getInstrumentations() to get configured instrumentations
- */
-
 interface TracingOptions extends traceloop.InitializeOptions {
     /**
      * Set to true when you have your own OpenTelemetry NodeSDK.
@@ -103,6 +88,11 @@ declare const tracing: (analytics: Raindrop, apiUrl: string, writeKey: string, o
      */
     createSpanProcessor(processorOptions?: traceloop.SpanProcessorOptions): RaindropSpanProcessor;
     begin(traceContext: PartialAiTrackEvent): Interaction;
+    recordModelUsage: (params: {
+        eventId: string;
+        model: string;
+        usage: AiTrackUsage;
+    }) => void;
     resumeSubagent(carrierOrHeaders: CarrierHeaders | TraceCarrier | null | undefined, options?: SubagentResumeOptions): SubagentRun;
     withSubagentRun<T>(carrierOrHeaders: CarrierHeaders | TraceCarrier | null | undefined, fn: (run: SubagentRun) => Promise<T> | T, options?: SubagentResumeOptions): Promise<T>;
     getActiveInteraction(eventId: string): Interaction;