raindrop-ai 0.3.0 → 0.3.1

package/dist/{chunk-5Y3LWBKV.mjs → chunk-4TM4IWHB.mjs}

@@ -4,7 +4,7 @@ if (!globalThis.RAINDROP_ASYNC_LOCAL_STORAGE) {
   globalThis.RAINDROP_ASYNC_LOCAL_STORAGE = 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"]));
@@ -552,6 +552,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/";
@@ -682,6 +696,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",
@@ -690,6 +705,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",
@@ -697,6 +713,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"]
 };
@@ -1040,7 +1097,7 @@ installTracingSuppressionHook();
 // 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",
@@ -1352,7 +1409,7 @@ function traceloopConfiguredWithoutClaim() {
 }
 
 // src/tracing/tracer-core.ts
-import { context as context5, SpanStatusCode as SpanStatusCode5, trace as trace5 } from "@opentelemetry/api";
+import { SpanStatusCode as SpanStatusCode5, context as context5, trace as trace5 } from "@opentelemetry/api";
 import { OTLPTraceExporter as OTLPHttpTraceExporter } from "@opentelemetry/exporter-trace-otlp-http";
 import { AnthropicInstrumentation } from "@traceloop/instrumentation-anthropic";
 import { BedrockInstrumentation } from "@traceloop/instrumentation-bedrock";
@@ -1636,7 +1693,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;
@@ -1680,6 +1737,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 };
@@ -1831,16 +1889,69 @@ var DirectTraceShipper = class {
   }
 };
 
+// src/tracing/exit-flush.ts
+import * as traceloop from "@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
 import {
+  SpanStatusCode as SpanStatusCode3,
+  TraceFlags,
   context as context3,
   isValidSpanId as isValidSpanId2,
   isValidTraceId as isValidTraceId2,
-  SpanStatusCode as SpanStatusCode3,
-  TraceFlags,
   trace as trace3
 } from "@opentelemetry/api";
-import * as traceloop from "@traceloop/node-server-sdk";
+import * as traceloop2 from "@traceloop/node-server-sdk";
 
 // src/tracing/subagents.ts
 import { SpanStatusCode as SpanStatusCode2, context as context2, trace as trace2 } from "@opentelemetry/api";
@@ -2264,6 +2375,7 @@ var DirectToolSpan = class {
 };
 var LiveInteraction = class {
   constructor(context6, traceId, analytics, directShipper, routing) {
+    this.usageEmitted = false;
     this.context = context6;
     this.analytics = analytics;
     this.tracer = trace3.getTracer("traceloop.tracer");
@@ -2348,10 +2460,7 @@ var LiveInteraction = class {
         spanId: base64ToHex2(spanIdB64),
         traceFlags: TraceFlags.SAMPLED
       };
-      const wrappedContext = trace3.setSpan(
-        context3.active(),
-        trace3.wrapSpanContext(spanContext)
-      );
+      const wrappedContext = trace3.setSpan(context3.active(), trace3.wrapSpanContext(spanContext));
       const invoke = () => this.routing ? withRoutingContext(this.routing, () => fn.apply(thisArg, args)) : fn.apply(thisArg, args);
       try {
         const result = await context3.with(wrappedContext, invoke);
@@ -2402,7 +2511,7 @@ var LiveInteraction = class {
       }
       return fn.apply(thisArg, fnArgs);
     };
-    return traceloop.withTask(
+    return traceloop2.withTask(
       {
         name: taskName,
         associationProperties: serializeAssociationProperties(properties),
@@ -2443,10 +2552,7 @@ var LiveInteraction = class {
         spanId: base64ToHex2(spanIdB64),
         traceFlags: TraceFlags.SAMPLED
       };
-      const wrappedContext = trace3.setSpan(
-        context3.active(),
-        trace3.wrapSpanContext(spanContext)
-      );
+      const wrappedContext = trace3.setSpan(context3.active(), trace3.wrapSpanContext(spanContext));
       const invoke = () => this.routing ? withRoutingContext(this.routing, () => fn.apply(thisArg, args)) : fn.apply(thisArg, args);
       let result;
       let error;
@@ -2556,7 +2662,7 @@ var LiveInteraction = class {
         throw error;
       }
     };
-    return traceloop.withTool(
+    return traceloop2.withTool(
       {
         name: toolName,
         associationProperties: serializeAssociationProperties(properties),
@@ -2859,7 +2965,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);
@@ -2867,6 +2973,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 {
@@ -2874,21 +2989,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
@@ -3005,7 +3119,7 @@ var LiveInteraction = class {
 
 // src/tracing/liveTracer.ts
 import { context as context4, isValidSpanId as isValidSpanId3, SpanStatusCode as SpanStatusCode4, trace as trace4 } from "@opentelemetry/api";
-import * as traceloop2 from "@traceloop/node-server-sdk";
+import * as traceloop3 from "@traceloop/node-server-sdk";
 var LiveTracer = class {
   constructor(globalProperties, directShipper, routing) {
     this.tracer = trace4.getTracer("traceloop.tracer");
@@ -3049,7 +3163,7 @@ var LiveTracer = class {
       }
       return fn.apply(thisArg, fnArgs);
     };
-    return traceloop2.withTask(
+    return traceloop3.withTask(
       {
         name: taskName,
         associationProperties: properties,
@@ -3161,59 +3275,6 @@ var LiveTracer = class {
   }
 };
 
-// src/tracing/exit-flush.ts
-import * as traceloop3 from "@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) {
@@ -3931,6 +3992,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 = trace5.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.
@@ -4042,6 +4156,7 @@ var tracing = (analytics, apiUrl, writeKey, options, isDebug = false) => {
     begin(traceContext) {
       return beginInteraction(traceContext);
     },
+    recordModelUsage,
     resumeSubagent(carrierOrHeaders, options2 = {}) {
       return openSubagentRun(carrierOrHeaders, options2);
     },

package/dist/{index-DBdUrUXc.d.mts → index-DuXeVnSq.d.mts}

@@ -330,11 +330,42 @@ interface IdentifyEvent {
  * @property input - An optional input property for the event, required if output is not provided.
  * @property output - An optional output property for the event, required if input is not provided.
  */
+declare const AiTrackUsageSchema: z.ZodObject<{
+    provider: z.ZodString;
+    inputTokens: z.ZodNumber;
+    outputTokens: z.ZodNumber;
+    cacheReadTokens: z.ZodOptional<z.ZodNumber>;
+    cacheWriteTokens: z.ZodOptional<z.ZodNumber>;
+    reasoningTokens: z.ZodOptional<z.ZodNumber>;
+    providerReportedCost: z.ZodOptional<z.ZodNumber>;
+}, "strip", z.ZodTypeAny, {
+    provider: string;
+    inputTokens: number;
+    outputTokens: number;
+    cacheReadTokens?: number | undefined;
+    cacheWriteTokens?: number | undefined;
+    reasoningTokens?: number | undefined;
+    providerReportedCost?: number | undefined;
+}, {
+    provider: string;
+    inputTokens: number;
+    outputTokens: number;
+    cacheReadTokens?: number | undefined;
+    cacheWriteTokens?: number | undefined;
+    reasoningTokens?: number | undefined;
+    providerReportedCost?: number | undefined;
+}>;
+type AiTrackUsage = z.infer<typeof AiTrackUsageSchema>;
 type AiTrackEvent = Omit<TrackEvent, "type"> & {
     model?: string;
     convoId?: string;
     attachments?: Attachment[];
     featureFlags?: RaindropFeatureFlags;
+    /**
+     * Raw values copied from the provider's usage object. Raindrop derives
+     * provider-specific billing pools on the server.
+     */
+    usage?: AiTrackUsage;
 } & ({
     input: string;
     output?: string;
@@ -1454,4 +1485,4 @@ declare class Raindrop {
     private closeWithinDeadline;
 }
 
-export { type AiTrackEvent as A, type BeginInteractionOptions as B, type CarrierHeaders as C, type TraceContext as D, type TrackToolParams as E, type FinishInteractionOptions as F, resolveDisableBatching as G, type Interaction as I, type LocalDebuggerLiveEventInput as L, MAX_INGEST_SIZE_BYTES as M, type PartialAiTrackEvent as P, Raindrop as R, type SubagentResumeOptions as S, type TraceCarrier as T, type WorkflowParams as W, type RaindropSpanProcessor as a, type SubagentRun as b, type Tracer as c, type Attachment as d, type AttachmentType as e, type IdentifyEvent as f, type RaindropFeatureFlagValue as g, type RaindropFeatureFlags as h, type RaindropProperties as i, type RaindropPropertyLeaf as j, type RaindropPropertyValue as k, type SelfDiagnoseOptions as l, type SelfDiagnosticsExecuteContext as m, type SelfDiagnosticsExecuteResult as n, type SelfDiagnosticsSignalDefinition as o, type SelfDiagnosticsSignalDefinitions as p, type SelfDiagnosticsTool as q, type SelfDiagnosticsToolInputSchema as r, type SelfDiagnosticsToolOptions as s, type SignalEvent as t, type SpanParams as u, type SubagentDispatch as v, type SubagentDispatchOptions as w, type SubagentFinishOptions as x, type ToolParams as y, type ToolSpan as z };
+export { type AiTrackUsage as A, type BeginInteractionOptions as B, type CarrierHeaders as C, type ToolSpan as D, type TraceContext as E, type FinishInteractionOptions as F, type TrackToolParams as G, resolveDisableBatching as H, type Interaction as I, type LocalDebuggerLiveEventInput as L, MAX_INGEST_SIZE_BYTES as M, type PartialAiTrackEvent as P, Raindrop as R, type SubagentResumeOptions as S, type TraceCarrier as T, type WorkflowParams as W, type RaindropSpanProcessor as a, type SubagentRun as b, type Tracer as c, type AiTrackEvent as d, type Attachment as e, type AttachmentType as f, type IdentifyEvent as g, type RaindropFeatureFlagValue as h, type RaindropFeatureFlags as i, type RaindropProperties as j, type RaindropPropertyLeaf as k, type RaindropPropertyValue as l, type SelfDiagnoseOptions as m, type SelfDiagnosticsExecuteContext as n, type SelfDiagnosticsExecuteResult as o, type SelfDiagnosticsSignalDefinition as p, type SelfDiagnosticsSignalDefinitions as q, type SelfDiagnosticsTool as r, type SelfDiagnosticsToolInputSchema as s, type SelfDiagnosticsToolOptions as t, type SignalEvent as u, type SpanParams as v, type SubagentDispatch as w, type SubagentDispatchOptions as x, type SubagentFinishOptions as y, type ToolParams as z };

package/dist/{index-DBdUrUXc.d.ts → index-DuXeVnSq.d.ts}

@@ -330,11 +330,42 @@ interface IdentifyEvent {
  * @property input - An optional input property for the event, required if output is not provided.
  * @property output - An optional output property for the event, required if input is not provided.
  */
+declare const AiTrackUsageSchema: z.ZodObject<{
+    provider: z.ZodString;
+    inputTokens: z.ZodNumber;
+    outputTokens: z.ZodNumber;
+    cacheReadTokens: z.ZodOptional<z.ZodNumber>;
+    cacheWriteTokens: z.ZodOptional<z.ZodNumber>;
+    reasoningTokens: z.ZodOptional<z.ZodNumber>;
+    providerReportedCost: z.ZodOptional<z.ZodNumber>;
+}, "strip", z.ZodTypeAny, {
+    provider: string;
+    inputTokens: number;
+    outputTokens: number;
+    cacheReadTokens?: number | undefined;
+    cacheWriteTokens?: number | undefined;
+    reasoningTokens?: number | undefined;
+    providerReportedCost?: number | undefined;
+}, {
+    provider: string;
+    inputTokens: number;
+    outputTokens: number;
+    cacheReadTokens?: number | undefined;
+    cacheWriteTokens?: number | undefined;
+    reasoningTokens?: number | undefined;
+    providerReportedCost?: number | undefined;
+}>;
+type AiTrackUsage = z.infer<typeof AiTrackUsageSchema>;
 type AiTrackEvent = Omit<TrackEvent, "type"> & {
     model?: string;
     convoId?: string;
     attachments?: Attachment[];
     featureFlags?: RaindropFeatureFlags;
+    /**
+     * Raw values copied from the provider's usage object. Raindrop derives
+     * provider-specific billing pools on the server.
+     */
+    usage?: AiTrackUsage;
 } & ({
     input: string;
     output?: string;
@@ -1454,4 +1485,4 @@ declare class Raindrop {
     private closeWithinDeadline;
 }
 
-export { type AiTrackEvent as A, type BeginInteractionOptions as B, type CarrierHeaders as C, type TraceContext as D, type TrackToolParams as E, type FinishInteractionOptions as F, resolveDisableBatching as G, type Interaction as I, type LocalDebuggerLiveEventInput as L, MAX_INGEST_SIZE_BYTES as M, type PartialAiTrackEvent as P, Raindrop as R, type SubagentResumeOptions as S, type TraceCarrier as T, type WorkflowParams as W, type RaindropSpanProcessor as a, type SubagentRun as b, type Tracer as c, type Attachment as d, type AttachmentType as e, type IdentifyEvent as f, type RaindropFeatureFlagValue as g, type RaindropFeatureFlags as h, type RaindropProperties as i, type RaindropPropertyLeaf as j, type RaindropPropertyValue as k, type SelfDiagnoseOptions as l, type SelfDiagnosticsExecuteContext as m, type SelfDiagnosticsExecuteResult as n, type SelfDiagnosticsSignalDefinition as o, type SelfDiagnosticsSignalDefinitions as p, type SelfDiagnosticsTool as q, type SelfDiagnosticsToolInputSchema as r, type SelfDiagnosticsToolOptions as s, type SignalEvent as t, type SpanParams as u, type SubagentDispatch as v, type SubagentDispatchOptions as w, type SubagentFinishOptions as x, type ToolParams as y, type ToolSpan as z };
+export { type AiTrackUsage as A, type BeginInteractionOptions as B, type CarrierHeaders as C, type ToolSpan as D, type TraceContext as E, type FinishInteractionOptions as F, type TrackToolParams as G, resolveDisableBatching as H, type Interaction as I, type LocalDebuggerLiveEventInput as L, MAX_INGEST_SIZE_BYTES as M, type PartialAiTrackEvent as P, Raindrop as R, type SubagentResumeOptions as S, type TraceCarrier as T, type WorkflowParams as W, type RaindropSpanProcessor as a, type SubagentRun as b, type Tracer as c, type AiTrackEvent as d, type Attachment as e, type AttachmentType as f, type IdentifyEvent as g, type RaindropFeatureFlagValue as h, type RaindropFeatureFlags as i, type RaindropProperties as j, type RaindropPropertyLeaf as k, type RaindropPropertyValue as l, type SelfDiagnoseOptions as m, type SelfDiagnosticsExecuteContext as n, type SelfDiagnosticsExecuteResult as o, type SelfDiagnosticsSignalDefinition as p, type SelfDiagnosticsSignalDefinitions as q, type SelfDiagnosticsTool as r, type SelfDiagnosticsToolInputSchema as s, type SelfDiagnosticsToolOptions as t, type SignalEvent as u, type SpanParams as v, type SubagentDispatch as w, type SubagentDispatchOptions as x, type SubagentFinishOptions as y, type ToolParams as z };

package/dist/index.d.mts

@@ -1,5 +1,5 @@
 import '@opentelemetry/instrumentation';
-export { A as AiTrackEvent, d as Attachment, e as AttachmentType, B as BeginInteractionOptions, F as FinishInteractionOptions, f as IdentifyEvent, I as Interaction, L as LocalDebuggerLiveEventInput, M as MAX_INGEST_SIZE_BYTES, P as PartialAiTrackEvent, R as Raindrop, g as RaindropFeatureFlagValue, h as RaindropFeatureFlags, i as RaindropProperties, j as RaindropPropertyLeaf, k as RaindropPropertyValue, a as RaindropSpanProcessor, l as SelfDiagnoseOptions, m as SelfDiagnosticsExecuteContext, n as SelfDiagnosticsExecuteResult, o as SelfDiagnosticsSignalDefinition, p as SelfDiagnosticsSignalDefinitions, q as SelfDiagnosticsTool, r as SelfDiagnosticsToolInputSchema, s as SelfDiagnosticsToolOptions, t as SignalEvent, u as SpanParams, v as SubagentDispatch, w as SubagentDispatchOptions, x as SubagentFinishOptions, S as SubagentResumeOptions, b as SubagentRun, y as ToolParams, z as ToolSpan, D as TraceContext, c as Tracer, E as TrackToolParams, W as WorkflowParams, R as default, G as resolveDisableBatching } from './index-DBdUrUXc.mjs';
+export { d as AiTrackEvent, A as AiTrackUsage, e as Attachment, f as AttachmentType, B as BeginInteractionOptions, F as FinishInteractionOptions, g as IdentifyEvent, I as Interaction, L as LocalDebuggerLiveEventInput, M as MAX_INGEST_SIZE_BYTES, P as PartialAiTrackEvent, R as Raindrop, h as RaindropFeatureFlagValue, i as RaindropFeatureFlags, j as RaindropProperties, k as RaindropPropertyLeaf, l as RaindropPropertyValue, a as RaindropSpanProcessor, m as SelfDiagnoseOptions, n as SelfDiagnosticsExecuteContext, o as SelfDiagnosticsExecuteResult, p as SelfDiagnosticsSignalDefinition, q as SelfDiagnosticsSignalDefinitions, r as SelfDiagnosticsTool, s as SelfDiagnosticsToolInputSchema, t as SelfDiagnosticsToolOptions, u as SignalEvent, v as SpanParams, w as SubagentDispatch, x as SubagentDispatchOptions, y as SubagentFinishOptions, S as SubagentResumeOptions, b as SubagentRun, z as ToolParams, D as ToolSpan, E as TraceContext, c as Tracer, G as TrackToolParams, W as WorkflowParams, R as default, H as resolveDisableBatching } from './index-DuXeVnSq.mjs';
 import '@traceloop/node-server-sdk';
 import '@opentelemetry/api';
 import 'zod';

package/dist/index.d.ts

@@ -1,5 +1,5 @@
 import '@opentelemetry/instrumentation';
-export { A as AiTrackEvent, d as Attachment, e as AttachmentType, B as BeginInteractionOptions, F as FinishInteractionOptions, f as IdentifyEvent, I as Interaction, L as LocalDebuggerLiveEventInput, M as MAX_INGEST_SIZE_BYTES, P as PartialAiTrackEvent, R as Raindrop, g as RaindropFeatureFlagValue, h as RaindropFeatureFlags, i as RaindropProperties, j as RaindropPropertyLeaf, k as RaindropPropertyValue, a as RaindropSpanProcessor, l as SelfDiagnoseOptions, m as SelfDiagnosticsExecuteContext, n as SelfDiagnosticsExecuteResult, o as SelfDiagnosticsSignalDefinition, p as SelfDiagnosticsSignalDefinitions, q as SelfDiagnosticsTool, r as SelfDiagnosticsToolInputSchema, s as SelfDiagnosticsToolOptions, t as SignalEvent, u as SpanParams, v as SubagentDispatch, w as SubagentDispatchOptions, x as SubagentFinishOptions, S as SubagentResumeOptions, b as SubagentRun, y as ToolParams, z as ToolSpan, D as TraceContext, c as Tracer, E as TrackToolParams, W as WorkflowParams, R as default, G as resolveDisableBatching } from './index-DBdUrUXc.js';
+export { d as AiTrackEvent, A as AiTrackUsage, e as Attachment, f as AttachmentType, B as BeginInteractionOptions, F as FinishInteractionOptions, g as IdentifyEvent, I as Interaction, L as LocalDebuggerLiveEventInput, M as MAX_INGEST_SIZE_BYTES, P as PartialAiTrackEvent, R as Raindrop, h as RaindropFeatureFlagValue, i as RaindropFeatureFlags, j as RaindropProperties, k as RaindropPropertyLeaf, l as RaindropPropertyValue, a as RaindropSpanProcessor, m as SelfDiagnoseOptions, n as SelfDiagnosticsExecuteContext, o as SelfDiagnosticsExecuteResult, p as SelfDiagnosticsSignalDefinition, q as SelfDiagnosticsSignalDefinitions, r as SelfDiagnosticsTool, s as SelfDiagnosticsToolInputSchema, t as SelfDiagnosticsToolOptions, u as SignalEvent, v as SpanParams, w as SubagentDispatch, x as SubagentDispatchOptions, y as SubagentFinishOptions, S as SubagentResumeOptions, b as SubagentRun, z as ToolParams, D as ToolSpan, E as TraceContext, c as Tracer, G as TrackToolParams, W as WorkflowParams, R as default, H as resolveDisableBatching } from './index-DuXeVnSq.js';
 import '@traceloop/node-server-sdk';
 import '@opentelemetry/api';
 import 'zod';