lightrace 0.1.7 → 0.1.9

package/dist/integrations/_base.js

@@ -0,0 +1,189 @@
+/**
+ * Shared tracing infrastructure for framework integrations.
+ *
+ * Provides `TracingMixin` with `registerRun` / `endRun` to manage OTel spans
+ * with `lightrace.*` attributes, and `normalizeUsage` for multi-provider
+ * token usage extraction.
+ */
+import { trace as otelTrace } from "@opentelemetry/api";
+import { generateId, jsonSerializable } from "../utils.js";
+import * as attrs from "../otel-exporter.js";
+import { Lightrace } from "../client.js";
+/**
+ * Normalize token usage from any provider format to a canonical form.
+ *
+ * Supports OpenAI (`prompt_tokens`/`completion_tokens`), Anthropic
+ * (`input_tokens`/`output_tokens`), and camelCase variants.
+ */
+export function normalizeUsage(raw) {
+    if (!raw || typeof raw !== "object")
+        return null;
+    const result = {};
+    const prompt = raw.promptTokens ?? raw.prompt_tokens ?? raw.input_tokens ?? undefined;
+    const completion = raw.completionTokens ?? raw.completion_tokens ?? raw.output_tokens ?? undefined;
+    const total = raw.totalTokens ?? raw.total_tokens ?? undefined;
+    if (prompt !== undefined)
+        result.promptTokens = Number(prompt);
+    if (completion !== undefined)
+        result.completionTokens = Number(completion);
+    if (total !== undefined)
+        result.totalTokens = Number(total);
+    else if (result.promptTokens !== undefined && result.completionTokens !== undefined)
+        result.totalTokens = result.promptTokens + result.completionTokens;
+    return Object.keys(result).length > 0 ? result : null;
+}
+/**
+ * Base class providing shared OTel tracing infrastructure for framework
+ * integrations. Manages span lifecycle, run-state tracking, and
+ * `lightrace.*` attribute setting.
+ */
+export class TracingMixin {
+    runs = new Map();
+    runParents = new Map();
+    completionStartTimes = new Map();
+    rootRunId = null;
+    _traceId = null;
+    userId;
+    sessionId;
+    traceName;
+    rootMetadata;
+    otelExporter = null;
+    rootSpan = null;
+    /** True when rootSpan was borrowed from an existing trace() wrapper (don't end it). */
+    borrowedRootSpan = false;
+    /** The trace ID from the most recently completed root run. */
+    lastTraceId = null;
+    constructor(opts) {
+        this.userId = opts?.userId;
+        this.sessionId = opts?.sessionId;
+        this.traceName = opts?.traceName;
+        this.rootMetadata = opts?.metadata;
+        const client = opts?.client ?? Lightrace.getInstance();
+        this.otelExporter = client?.getOtelExporter() ?? null;
+    }
+    /** Get the current active trace ID (null if no run is active). */
+    get traceId() {
+        return this._traceId;
+    }
+    // ── Core span management ──────────────────────────────────────────
+    registerRun(runId, parentRunId, info) {
+        const tracer = this.otelExporter?.tracer;
+        const isRoot = !this.rootRunId;
+        if (isRoot) {
+            this.rootRunId = runId;
+            const activeSpan = otelTrace.getActiveSpan();
+            if (activeSpan) {
+                // Reuse the existing trace span — don't create a duplicate root.
+                this.rootSpan = activeSpan;
+                this.borrowedRootSpan = true;
+                this._traceId = activeSpan.spanContext().traceId;
+            }
+            else if (tracer) {
+                this._traceId = generateId();
+                this.rootSpan = tracer.startSpan(this.traceName ?? info.name, {
+                    startTime: info.startTime,
+                });
+                this.rootSpan.setAttribute(attrs.AS_ROOT, "true");
+                this.rootSpan.setAttribute(attrs.TRACE_NAME, this.traceName ?? info.name);
+                this.rootSpan.setAttribute(attrs.TRACE_INPUT, attrs.safeJson(jsonSerializable(info.input)));
+                if (this.userId)
+                    this.rootSpan.setAttribute(attrs.TRACE_USER_ID, this.userId);
+                if (this.sessionId)
+                    this.rootSpan.setAttribute(attrs.TRACE_SESSION_ID, this.sessionId);
+                if (this.rootMetadata) {
+                    this.rootSpan.setAttribute(attrs.TRACE_METADATA, attrs.safeJson(this.rootMetadata));
+                }
+            }
+            else {
+                this._traceId = generateId();
+            }
+        }
+        const observationId = generateId();
+        let span;
+        if (tracer) {
+            span = tracer.startSpan(info.name, { startTime: info.startTime });
+        }
+        else {
+            span = {
+                setAttribute: () => span,
+                setAttributes: () => span,
+                addEvent: () => span,
+                setStatus: () => span,
+                end: () => { },
+                isRecording: () => false,
+                recordException: () => { },
+                spanContext: () => ({ traceId: "", spanId: "", traceFlags: 0 }),
+                updateName: () => span,
+            };
+        }
+        const runInfo = { ...info, observationId, parentRunId, span };
+        this.runs.set(runId, runInfo);
+        this.runParents.set(observationId, parentRunId ? this.runs.get(parentRunId)?.observationId : undefined);
+        return runInfo;
+    }
+    endObservationSpan(run, output, level, statusMessage, extra) {
+        const obsTypeMap = {
+            span: "SPAN",
+            generation: "GENERATION",
+            event: "EVENT",
+            tool: "TOOL",
+            chain: "CHAIN",
+        };
+        const span = run.span;
+        span.setAttribute(attrs.OBSERVATION_TYPE, obsTypeMap[run.type] ?? run.type);
+        span.setAttribute(attrs.OBSERVATION_INPUT, attrs.safeJson(jsonSerializable(run.input)));
+        span.setAttribute(attrs.OBSERVATION_OUTPUT, attrs.safeJson(jsonSerializable(output)));
+        span.setAttribute(attrs.OBSERVATION_LEVEL, level);
+        if (run.metadata) {
+            span.setAttribute(attrs.OBSERVATION_METADATA, attrs.safeJson(run.metadata));
+        }
+        if (run.model) {
+            span.setAttribute(attrs.OBSERVATION_MODEL, run.model);
+        }
+        if (statusMessage) {
+            span.setAttribute(attrs.OBSERVATION_STATUS_MESSAGE, statusMessage);
+        }
+        if (run.modelParameters && Object.keys(run.modelParameters).length > 0) {
+            span.setAttribute(attrs.OBSERVATION_MODEL_PARAMETERS, attrs.safeJson(run.modelParameters));
+        }
+        if (extra) {
+            const usageDetails = {};
+            if (extra.promptTokens !== undefined)
+                usageDetails.promptTokens = extra.promptTokens;
+            if (extra.completionTokens !== undefined)
+                usageDetails.completionTokens = extra.completionTokens;
+            if (extra.totalTokens !== undefined)
+                usageDetails.totalTokens = extra.totalTokens;
+            if (Object.keys(usageDetails).length > 0) {
+                span.setAttribute(attrs.OBSERVATION_USAGE_DETAILS, attrs.safeJson(usageDetails));
+            }
+        }
+        span.end();
+    }
+    endRun(runId, output, level = "DEFAULT", statusMessage = null, extra) {
+        const run = this.runs.get(runId);
+        if (!run)
+            return;
+        this.endObservationSpan(run, output, level, statusMessage, extra);
+        if (runId === this.rootRunId) {
+            if (this.rootSpan) {
+                this.rootSpan.setAttribute(attrs.TRACE_OUTPUT, attrs.safeJson(jsonSerializable(output)));
+                if (!this.borrowedRootSpan) {
+                    this.rootSpan.end();
+                }
+                this.rootSpan = null;
+            }
+            this.lastTraceId = this._traceId;
+            this.rootRunId = null;
+            this._traceId = null;
+            this.borrowedRootSpan = false;
+            this.runs.clear();
+            this.runParents.clear();
+            this.completionStartTimes.clear();
+        }
+        else {
+            this.runs.delete(runId);
+        }
+    }
+}
+//# sourceMappingURL=_base.js.map

package/dist/integrations/_base.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"_base.js","sourceRoot":"","sources":["../../src/integrations/_base.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AACH,OAAO,EAAE,KAAK,IAAI,SAAS,EAAa,MAAM,oBAAoB,CAAC;AACnE,OAAO,EAAE,UAAU,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAE3D,OAAO,KAAK,KAAK,MAAM,qBAAqB,CAAC;AAC7C,OAAO,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AA6BzC;;;;;GAKG;AACH,MAAM,UAAU,cAAc,CAAC,GAA4B;IACzD,IAAI,CAAC,GAAG,IAAI,OAAO,GAAG,KAAK,QAAQ;QAAE,OAAO,IAAI,CAAC;IAEjD,MAAM,MAAM,GAA2B,EAAE,CAAC;IAE1C,MAAM,MAAM,GAAG,GAAG,CAAC,YAAY,IAAI,GAAG,CAAC,aAAa,IAAI,GAAG,CAAC,YAAY,IAAI,SAAS,CAAC;IACtF,MAAM,UAAU,GACd,GAAG,CAAC,gBAAgB,IAAI,GAAG,CAAC,iBAAiB,IAAI,GAAG,CAAC,aAAa,IAAI,SAAS,CAAC;IAClF,MAAM,KAAK,GAAG,GAAG,CAAC,WAAW,IAAI,GAAG,CAAC,YAAY,IAAI,SAAS,CAAC;IAE/D,IAAI,MAAM,KAAK,SAAS;QAAE,MAAM,CAAC,YAAY,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC;IAC/D,IAAI,UAAU,KAAK,SAAS;QAAE,MAAM,CAAC,gBAAgB,GAAG,MAAM,CAAC,UAAU,CAAC,CAAC;IAC3E,IAAI,KAAK,KAAK,SAAS;QAAE,MAAM,CAAC,WAAW,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;SACvD,IAAI,MAAM,CAAC,YAAY,KAAK,SAAS,IAAI,MAAM,CAAC,gBAAgB,KAAK,SAAS;QACjF,MAAM,CAAC,WAAW,GAAG,MAAM,CAAC,YAAY,GAAG,MAAM,CAAC,gBAAgB,CAAC;IAErE,OAAO,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC;AACxD,CAAC;AAED;;;;GAIG;AACH,MAAM,OAAO,YAAY;IACb,IAAI,GAAG,IAAI,GAAG,EAAsB,CAAC;IACrC,UAAU,GAAG,IAAI,GAAG,EAA8B,CAAC;IACnD,oBAAoB,GAAG,IAAI,GAAG,EAAgB,CAAC;IAC/C,SAAS,GAAkB,IAAI,CAAC;IAChC,QAAQ,GAAkB,IAAI,CAAC;IAE/B,MAAM,CAAU;IAChB,SAAS,CAAU;IACnB,SAAS,CAAU;IACnB,YAAY,CAA2B;IAEvC,YAAY,GAAiC,IAAI,CAAC;IAElD,QAAQ,GAAgB,IAAI,CAAC;IACvC,uFAAuF;IAC/E,gBAAgB,GAAG,KAAK,CAAC;IAEjC,8DAA8D;IAC9D,WAAW,GAAkB,IAAI,CAAC;IAElC,YAAY,IAA0B;QACpC,IAAI,CAAC,MAAM,GAAG,IAAI,EAAE,MAAM,CAAC;QAC3B,IAAI,CAAC,SAAS,GAAG,IAAI,EAAE,SAAS,CAAC;QACjC,IAAI,CAAC,SAAS,GAAG,IAAI,EAAE,SAAS,CAAC;QACjC,IAAI,CAAC,YAAY,GAAG,IAAI,EAAE,QAAQ,CAAC;QAEnC,MAAM,MAAM,GAAG,IAAI,EAAE,MAAM,IAAI,SAAS,CAAC,WAAW,EAAE,CAAC;QACvD,IAAI,CAAC,YAAY,GAAG,MAAM,EAAE,eAAe,EAAE,IAAI,IAAI,CAAC;IACxD,CAAC;IAED,kEAAkE;IAClE,IAAI,OAAO;QACT,OAAO,IAAI,CAAC,QAAQ,CAAC;IACvB,CAAC;IAED,qEAAqE;IAE3D,WAAW,CACnB,KAAa,EACb,WAA+B,EAC/B,IAAgE;QAEhE,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,EAAE,MAAM,CAAC;QACzC,MAAM,MAAM,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC;QAE/B,IAAI,MAAM,EAAE,CAAC;YACX,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;YAEvB,MAAM,UAAU,GAAG,SAAS,CAAC,aAAa,EAAE,CAAC;YAE7C,IAAI,UAAU,EAAE,CAAC;gBACf,iEAAiE;gBACjE,IAAI,CAAC,QAAQ,GAAG,UAAU,CAAC;gBAC3B,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC;gBAC7B,IAAI,CAAC,QAAQ,GAAG,UAAU,CAAC,WAAW,EAAE,CAAC,OAAO,CAAC;YACnD,CAAC;iBAAM,IAAI,MAAM,EAAE,CAAC;gBAClB,IAAI,CAAC,QAAQ,GAAG,UAAU,EAAE,CAAC;gBAC7B,IAAI,CAAC,QAAQ,GAAG,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,IAAI,EAAE;oBAC5D,SAAS,EAAE,IAAI,CAAC,SAAS;iBAC1B,CAAC,CAAC;gBACH,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,KAAK,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;gBAClD,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,KAAK,CAAC,UAAU,EAAE,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,IAAI,CAAC,CAAC;gBAC1E,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,KAAK,CAAC,WAAW,EAAE,KAAK,CAAC,QAAQ,CAAC,gBAAgB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;gBAC5F,IAAI,IAAI,CAAC,MAAM;oBAAE,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,KAAK,CAAC,aAAa,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;gBAC9E,IAAI,IAAI,CAAC,SAAS;oBAAE,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,KAAK,CAAC,gBAAgB,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;gBACvF,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;oBACtB,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,KAAK,CAAC,cAAc,EAAE,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC;gBACtF,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,QAAQ,GAAG,UAAU,EAAE,CAAC;YAC/B,CAAC;QACH,CAAC;QAED,MAAM,aAAa,GAAG,UAAU,EAAE,CAAC;QAEnC,IAAI,IAAU,CAAC;QACf,IAAI,MAAM,EAAE,CAAC;YACX,IAAI,GAAG,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC;QACpE,CAAC;aAAM,CAAC;YACN,IAAI,GAAG;gBACL,YAAY,EAAE,GAAG,EAAE,CAAC,IAAI;gBACxB,aAAa,EAAE,GAAG,EAAE,CAAC,IAAI;gBACzB,QAAQ,EAAE,GAAG,EAAE,CAAC,IAAI;gBACpB,SAAS,EAAE,GAAG,EAAE,CAAC,IAAI;gBACrB,GAAG,EAAE,GAAG,EAAE,GAAE,CAAC;gBACb,WAAW,EAAE,GAAG,EAAE,CAAC,KAAK;gBACxB,eAAe,EAAE,GAAG,EAAE,GAAE,CAAC;gBACzB,WAAW,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC;gBAC/D,UAAU,EAAE,GAAG,EAAE,CAAC,IAAI;aACJ,CAAC;QACvB,CAAC;QAED,MAAM,OAAO,GAAe,EAAE,GAAG,IAAI,EAAE,aAAa,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC;QAC1E,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QAC9B,IAAI,CAAC,UAAU,CAAC,GAAG,CACjB,aAAa,EACb,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,aAAa,CAAC,CAAC,CAAC,SAAS,CACpE,CAAC;QAEF,OAAO,OAAO,CAAC;IACjB,CAAC;IAES,kBAAkB,CAC1B,GAAe,EACf,MAAe,EACf,KAAa,EACb,aAA4B,EAC5B,KAA+B;QAE/B,MAAM,UAAU,GAA2B;YACzC,IAAI,EAAE,MAAM;YACZ,UAAU,EAAE,YAAY;YACxB,KAAK,EAAE,OAAO;YACd,IAAI,EAAE,MAAM;YACZ,KAAK,EAAE,OAAO;SACf,CAAC;QAEF,MAAM,IAAI,GAAG,GAAG,CAAC,IAAI,CAAC;QAEtB,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,gBAAgB,EAAE,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,GAAG,CAAC,IAAI,CAAC,CAAC;QAC5E,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,iBAAiB,EAAE,KAAK,CAAC,QAAQ,CAAC,gBAAgB,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACxF,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,kBAAkB,EAAE,KAAK,CAAC,QAAQ,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;QACtF,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,iBAAiB,EAAE,KAAK,CAAC,CAAC;QAElD,IAAI,GAAG,CAAC,QAAQ,EAAE,CAAC;YACjB,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,oBAAoB,EAAE,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC;QAC9E,CAAC;QACD,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;YACd,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,iBAAiB,EAAE,GAAG,CAAC,KAAK,CAAC,CAAC;QACxD,CAAC;QACD,IAAI,aAAa,EAAE,CAAC;YAClB,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,0BAA0B,EAAE,aAAa,CAAC,CAAC;QACrE,CAAC;QACD,IAAI,GAAG,CAAC,eAAe,IAAI,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACvE,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,4BAA4B,EAAE,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC,CAAC;QAC7F,CAAC;QACD,IAAI,KAAK,EAAE,CAAC;YACV,MAAM,YAAY,GAA4B,EAAE,CAAC;YACjD,IAAI,KAAK,CAAC,YAAY,KAAK,SAAS;gBAAE,YAAY,CAAC,YAAY,GAAG,KAAK,CAAC,YAAY,CAAC;YACrF,IAAI,KAAK,CAAC,gBAAgB,KAAK,SAAS;gBACtC,YAAY,CAAC,gBAAgB,GAAG,KAAK,CAAC,gBAAgB,CAAC;YACzD,IAAI,KAAK,CAAC,WAAW,KAAK,SAAS;gBAAE,YAAY,CAAC,WAAW,GAAG,KAAK,CAAC,WAAW,CAAC;YAClF,IAAI,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACzC,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,yBAAyB,EAAE,KAAK,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,CAAC;YACnF,CAAC;QACH,CAAC;QAED,IAAI,CAAC,GAAG,EAAE,CAAC;IACb,CAAC;IAED,MAAM,CACJ,KAAa,EACb,MAAe,EACf,KAAK,GAAG,SAAS,EACjB,gBAA+B,IAAI,EACnC,KAA+B;QAE/B,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACjC,IAAI,CAAC,GAAG;YAAE,OAAO;QAEjB,IAAI,CAAC,kBAAkB,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,aAAa,EAAE,KAAK,CAAC,CAAC;QAElE,IAAI,KAAK,KAAK,IAAI,CAAC,SAAS,EAAE,CAAC;YAC7B,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;gBAClB,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,CAAC,QAAQ,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;gBACzF,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,CAAC;oBAC3B,IAAI,CAAC,QAAQ,CAAC,GAAG,EAAE,CAAC;gBACtB,CAAC;gBACD,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;YACvB,CAAC;YAED,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,QAAQ,CAAC;YACjC,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;YACtB,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;YACrB,IAAI,CAAC,gBAAgB,GAAG,KAAK,CAAC;YAC9B,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;YAClB,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;YACxB,IAAI,CAAC,oBAAoB,CAAC,KAAK,EAAE,CAAC;QACpC,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC1B,CAAC;IACH,CAAC;CACF"}

package/dist/integrations/anthropic.d.ts

@@ -0,0 +1,15 @@
+import { TracingMixin, type TracingMixinOptions } from "./_base.js";
+export declare class LightraceAnthropicInstrumentor extends TracingMixin {
+    private patchedTargets;
+    constructor(opts?: TracingMixinOptions);
+    /**
+     * Patch an Anthropic client instance so that `messages.create` and
+     * `messages.stream` calls are automatically traced.
+     */
+    instrument(client: unknown): void;
+    /** Restore the original methods. */
+    uninstrument(client: unknown): void;
+    /** Start a trace for an Anthropic messages call. */
+    private startAnthropicTrace;
+}
+//# sourceMappingURL=anthropic.d.ts.map

package/dist/integrations/anthropic.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"anthropic.d.ts","sourceRoot":"","sources":["../../src/integrations/anthropic.ts"],"names":[],"mappings":"AAkBA,OAAO,EAAE,YAAY,EAAE,KAAK,mBAAmB,EAAE,MAAM,YAAY,CAAC;AAKpE,qBAAa,8BAA+B,SAAQ,YAAY;IAC9D,OAAO,CAAC,cAAc,CAGlB;gBAEQ,IAAI,CAAC,EAAE,mBAAmB;IAItC;;;OAGG;IACH,UAAU,CAAC,MAAM,EAAE,OAAO,GAAG,IAAI;IAmEjC,oCAAoC;IACpC,YAAY,CAAC,MAAM,EAAE,OAAO,GAAG,IAAI;IAYnC,oDAAoD;IACpD,OAAO,CAAC,mBAAmB;CA8B5B"}

package/dist/integrations/anthropic.js

@@ -0,0 +1,228 @@
+/**
+ * Anthropic SDK integration for Lightrace.
+ *
+ * Wraps `messages.create` and `messages.stream` on Anthropic client instances
+ * to automatically capture traces via OpenTelemetry spans.
+ *
+ * @example
+ * ```ts
+ * import { LightraceAnthropicInstrumentor } from "lightrace/integrations/anthropic";
+ *
+ * const instrumentor = new LightraceAnthropicInstrumentor();
+ * instrumentor.instrument(client);
+ *
+ * const response = await client.messages.create({ ... });
+ * console.log(instrumentor.lastTraceId);
+ * ```
+ */
+import { generateId, jsonSerializable } from "../utils.js";
+import { TracingMixin } from "./_base.js";
+export class LightraceAnthropicInstrumentor extends TracingMixin {
+    patchedTargets = new Map();
+    constructor(opts) {
+        super(opts);
+    }
+    /**
+     * Patch an Anthropic client instance so that `messages.create` and
+     * `messages.stream` calls are automatically traced.
+     */
+    instrument(client) {
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        const c = client;
+        const messages = c?.messages;
+        if (!messages?.create)
+            return;
+        if (this.patchedTargets.has(client))
+            return;
+        const originalCreate = messages.create.bind(messages);
+        const originalStream = messages.stream ? messages.stream.bind(messages) : undefined;
+        this.patchedTargets.set(client, {
+            messages,
+            originalCreate,
+            originalStream,
+        });
+        // Patch messages.create
+        messages.create = async (...args) => {
+            const kwargs = args.length === 1 && typeof args[0] === "object" && args[0] !== null
+                ? args[0]
+                : {};
+            const { runId } = this.startAnthropicTrace(kwargs);
+            const stream = Boolean(kwargs.stream);
+            try {
+                const result = await originalCreate(...args);
+                if (!stream) {
+                    const output = extractOutput(result);
+                    const usage = extractUsage(result);
+                    this.endRun(runId, output, "DEFAULT", null, usage ?? undefined);
+                }
+                else {
+                    // For streaming, end immediately with partial info
+                    this.endRun(runId, { streaming: true });
+                }
+                return result;
+            }
+            catch (err) {
+                this.endRun(runId, null, "ERROR", err instanceof Error ? err.message : String(err));
+                throw err;
+            }
+        };
+        // Patch messages.stream (separate context-manager streaming API)
+        if (originalStream) {
+            messages.stream = (...args) => {
+                const kwargs = args.length === 1 && typeof args[0] === "object" && args[0] !== null
+                    ? args[0]
+                    : {};
+                const { runId } = this.startAnthropicTrace(kwargs);
+                try {
+                    const streamManager = originalStream(...args);
+                    // Wrap the stream manager to capture the final message
+                    return wrapStreamManager(streamManager, this, runId);
+                }
+                catch (err) {
+                    this.endRun(runId, null, "ERROR", err instanceof Error ? err.message : String(err));
+                    throw err;
+                }
+            };
+        }
+    }
+    /** Restore the original methods. */
+    uninstrument(client) {
+        const entry = this.patchedTargets.get(client);
+        if (!entry)
+            return;
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        const messages = entry.messages;
+        messages.create = entry.originalCreate;
+        if (entry.originalStream) {
+            messages.stream = entry.originalStream;
+        }
+        this.patchedTargets.delete(client);
+    }
+    /** Start a trace for an Anthropic messages call. */
+    startAnthropicTrace(kwargs) {
+        const runId = generateId();
+        const model = kwargs.model;
+        const inputData = {
+            messages: jsonSerializable(kwargs.messages ?? []),
+        };
+        if (model)
+            inputData.model = model;
+        if (kwargs.system)
+            inputData.system = jsonSerializable(kwargs.system);
+        if (kwargs.tools)
+            inputData.tools = jsonSerializable(kwargs.tools);
+        if (kwargs.tool_choice)
+            inputData.tool_choice = jsonSerializable(kwargs.tool_choice);
+        const modelParams = {};
+        for (const key of ["temperature", "max_tokens", "top_p", "top_k", "stop_sequences"]) {
+            if (kwargs[key] !== undefined)
+                modelParams[key] = kwargs[key];
+        }
+        this.registerRun(runId, undefined, {
+            type: "generation",
+            name: model ?? "anthropic",
+            startTime: new Date(),
+            input: inputData,
+            model,
+            modelParameters: Object.keys(modelParams).length > 0 ? modelParams : undefined,
+        });
+        return { runId };
+    }
+}
+// -- Helpers ----------------------------------------------------------------
+function extractOutput(response) {
+    if (!response || typeof response !== "object")
+        return response;
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    const r = response;
+    const content = r.content;
+    if (!Array.isArray(content))
+        return jsonSerializable(response);
+    const blocks = content.map((block) => {
+        if (block.type === "text") {
+            return { type: "text", text: block.text ?? "" };
+        }
+        if (block.type === "tool_use") {
+            return {
+                type: "tool_use",
+                id: block.id ?? "",
+                name: block.name ?? "",
+                input: jsonSerializable(block.input ?? {}),
+            };
+        }
+        return jsonSerializable(block);
+    });
+    const result = {
+        role: r.role ?? "assistant",
+        content: blocks,
+    };
+    if (r.stop_reason)
+        result.stop_reason = r.stop_reason;
+    return result;
+}
+function extractUsage(response) {
+    if (!response || typeof response !== "object")
+        return null;
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    const usage = response.usage;
+    if (!usage)
+        return null;
+    const raw = {};
+    if (usage.input_tokens !== undefined)
+        raw.promptTokens = usage.input_tokens;
+    if (usage.output_tokens !== undefined)
+        raw.completionTokens = usage.output_tokens;
+    if (raw.promptTokens !== undefined && raw.completionTokens !== undefined) {
+        raw.totalTokens = raw.promptTokens + raw.completionTokens;
+    }
+    return Object.keys(raw).length > 0 ? raw : null;
+}
+/**
+ * Wrap a stream manager (from `messages.stream()`) to capture the final
+ * message when the stream ends.
+ */
+function wrapStreamManager(manager, instrumentor, runId) {
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    const m = manager;
+    // For async stream managers (most common in Node.js)
+    if (typeof m[Symbol.asyncIterator] === "function" || typeof m.then === "function") {
+        return {
+            ...m,
+            async *[Symbol.asyncIterator]() {
+                let finalMessage = null;
+                try {
+                    for await (const event of m) {
+                        if (event?.type === "message_stop") {
+                            finalMessage = event.message ?? m.currentMessageSnapshot ?? null;
+                        }
+                        yield event;
+                    }
+                }
+                finally {
+                    const msg = finalMessage ?? m.currentMessageSnapshot ?? m.finalMessage ?? null;
+                    if (msg) {
+                        const output = extractOutput(msg);
+                        const usage = extractUsage(msg);
+                        instrumentor.endRun(runId, output, "DEFAULT", null, usage ?? undefined);
+                    }
+                    else {
+                        instrumentor.endRun(runId, null);
+                    }
+                }
+            },
+            // Forward method calls
+            on: m.on?.bind(m),
+            off: m.off?.bind(m),
+            get finalMessage() {
+                return m.finalMessage;
+            },
+            get currentMessageSnapshot() {
+                return m.currentMessageSnapshot;
+            },
+        };
+    }
+    // Fallback: return as-is and end the span
+    instrumentor.endRun(runId, null);
+    return manager;
+}
+//# sourceMappingURL=anthropic.js.map

package/dist/integrations/anthropic.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"anthropic.js","sourceRoot":"","sources":["../../src/integrations/anthropic.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;GAgBG;AACH,OAAO,EAAE,UAAU,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAC3D,OAAO,EAAE,YAAY,EAA4B,MAAM,YAAY,CAAC;AAKpE,MAAM,OAAO,8BAA+B,SAAQ,YAAY;IACtD,cAAc,GAAG,IAAI,GAAG,EAG7B,CAAC;IAEJ,YAAY,IAA0B;QACpC,KAAK,CAAC,IAAI,CAAC,CAAC;IACd,CAAC;IAED;;;OAGG;IACH,UAAU,CAAC,MAAe;QACxB,8DAA8D;QAC9D,MAAM,CAAC,GAAG,MAAa,CAAC;QACxB,MAAM,QAAQ,GAAG,CAAC,EAAE,QAAQ,CAAC;QAC7B,IAAI,CAAC,QAAQ,EAAE,MAAM;YAAE,OAAO;QAC9B,IAAI,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,MAAM,CAAC;YAAE,OAAO;QAE5C,MAAM,cAAc,GAAG,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACtD,MAAM,cAAc,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QACpF,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,MAAM,EAAE;YAC9B,QAAQ;YACR,cAAc;YACd,cAAc;SACf,CAAC,CAAC;QAEH,wBAAwB;QACxB,QAAQ,CAAC,MAAM,GAAG,KAAK,EAAE,GAAG,IAAe,EAAE,EAAE;YAC7C,MAAM,MAAM,GACV,IAAI,CAAC,MAAM,KAAK,CAAC,IAAI,OAAO,IAAI,CAAC,CAAC,CAAC,KAAK,QAAQ,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,IAAI;gBAClE,CAAC,CAAE,IAAI,CAAC,CAAC,CAA6B;gBACtC,CAAC,CAAC,EAAE,CAAC;YAET,MAAM,EAAE,KAAK,EAAE,GAAG,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAC;YACnD,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YAEtC,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,GAAG,IAAI,CAAC,CAAC;gBAE7C,IAAI,CAAC,MAAM,EAAE,CAAC;oBACZ,MAAM,MAAM,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC;oBACrC,MAAM,KAAK,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC;oBACnC,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,IAAI,SAAS,CAAC,CAAC;gBAClE,CAAC;qBAAM,CAAC;oBACN,mDAAmD;oBACnD,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;gBAC1C,CAAC;gBAED,OAAO,MAAM,CAAC;YAChB,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;gBACpF,MAAM,GAAG,CAAC;YACZ,CAAC;QACH,CAAC,CAAC;QAEF,iEAAiE;QACjE,IAAI,cAAc,EAAE,CAAC;YACnB,QAAQ,CAAC,MAAM,GAAG,CAAC,GAAG,IAAe,EAAE,EAAE;gBACvC,MAAM,MAAM,GACV,IAAI,CAAC,MAAM,KAAK,CAAC,IAAI,OAAO,IAAI,CAAC,CAAC,CAAC,KAAK,QAAQ,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,IAAI;oBAClE,CAAC,CAAE,IAAI,CAAC,CAAC,CAA6B;oBACtC,CAAC,CAAC,EAAE,CAAC;gBAET,MAAM,EAAE,KAAK,EAAE,GAAG,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAC;gBAEnD,IAAI,CAAC;oBACH,MAAM,aAAa,GAAG,cAAc,CAAC,GAAG,IAAI,CAAC,CAAC;oBAE9C,uDAAuD;oBACvD,OAAO,iBAAiB,CAAC,aAAa,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC;gBACvD,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACb,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;oBACpF,MAAM,GAAG,CAAC;gBACZ,CAAC;YACH,CAAC,CAAC;QACJ,CAAC;IACH,CAAC;IAED,oCAAoC;IACpC,YAAY,CAAC,MAAe;QAC1B,MAAM,KAAK,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAC9C,IAAI,CAAC,KAAK;YAAE,OAAO;QACnB,8DAA8D;QAC9D,MAAM,QAAQ,GAAG,KAAK,CAAC,QAAe,CAAC;QACvC,QAAQ,CAAC,MAAM,GAAG,KAAK,CAAC,cAAc,CAAC;QACvC,IAAI,KAAK,CAAC,cAAc,EAAE,CAAC;YACzB,QAAQ,CAAC,MAAM,GAAG,KAAK,CAAC,cAAc,CAAC;QACzC,CAAC;QACD,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IACrC,CAAC;IAED,oDAAoD;IAC5C,mBAAmB,CAAC,MAA+B;QAGzD,MAAM,KAAK,GAAG,UAAU,EAAE,CAAC;QAC3B,MAAM,KAAK,GAAG,MAAM,CAAC,KAA2B,CAAC;QAEjD,MAAM,SAAS,GAA4B;YACzC,QAAQ,EAAE,gBAAgB,CAAC,MAAM,CAAC,QAAQ,IAAI,EAAE,CAAC;SAClD,CAAC;QACF,IAAI,KAAK;YAAE,SAAS,CAAC,KAAK,GAAG,KAAK,CAAC;QACnC,IAAI,MAAM,CAAC,MAAM;YAAE,SAAS,CAAC,MAAM,GAAG,gBAAgB,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QACtE,IAAI,MAAM,CAAC,KAAK;YAAE,SAAS,CAAC,KAAK,GAAG,gBAAgB,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACnE,IAAI,MAAM,CAAC,WAAW;YAAE,SAAS,CAAC,WAAW,GAAG,gBAAgB,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;QAErF,MAAM,WAAW,GAA4B,EAAE,CAAC;QAChD,KAAK,MAAM,GAAG,IAAI,CAAC,aAAa,EAAE,YAAY,EAAE,OAAO,EAAE,OAAO,EAAE,gBAAgB,CAAC,EAAE,CAAC;YACpF,IAAI,MAAM,CAAC,GAAG,CAAC,KAAK,SAAS;gBAAE,WAAW,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC;QAChE,CAAC;QAED,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,SAAS,EAAE;YACjC,IAAI,EAAE,YAAY;YAClB,IAAI,EAAE,KAAK,IAAI,WAAW;YAC1B,SAAS,EAAE,IAAI,IAAI,EAAE;YACrB,KAAK,EAAE,SAAS;YAChB,KAAK;YACL,eAAe,EAAE,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,SAAS;SAC/E,CAAC,CAAC;QAEH,OAAO,EAAE,KAAK,EAAE,CAAC;IACnB,CAAC;CACF;AAED,8EAA8E;AAE9E,SAAS,aAAa,CAAC,QAAiB;IACtC,IAAI,CAAC,QAAQ,IAAI,OAAO,QAAQ,KAAK,QAAQ;QAAE,OAAO,QAAQ,CAAC;IAC/D,8DAA8D;IAC9D,MAAM,CAAC,GAAG,QAAe,CAAC;IAC1B,MAAM,OAAO,GAAG,CAAC,CAAC,OAAO,CAAC;IAC1B,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC;QAAE,OAAO,gBAAgB,CAAC,QAAQ,CAAC,CAAC;IAE/D,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,KAA8B,EAAE,EAAE;QAC5D,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAC1B,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,IAAI,EAAE,EAAE,CAAC;QAClD,CAAC;QACD,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;YAC9B,OAAO;gBACL,IAAI,EAAE,UAAU;gBAChB,EAAE,EAAE,KAAK,CAAC,EAAE,IAAI,EAAE;gBAClB,IAAI,EAAE,KAAK,CAAC,IAAI,IAAI,EAAE;gBACtB,KAAK,EAAE,gBAAgB,CAAC,KAAK,CAAC,KAAK,IAAI,EAAE,CAAC;aAC3C,CAAC;QACJ,CAAC;QACD,OAAO,gBAAgB,CAAC,KAAK,CAAC,CAAC;IACjC,CAAC,CAAC,CAAC;IAEH,MAAM,MAAM,GAA4B;QACtC,IAAI,EAAE,CAAC,CAAC,IAAI,IAAI,WAAW;QAC3B,OAAO,EAAE,MAAM;KAChB,CAAC;IACF,IAAI,CAAC,CAAC,WAAW;QAAE,MAAM,CAAC,WAAW,GAAG,CAAC,CAAC,WAAW,CAAC;IACtD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,YAAY,CAAC,QAAiB;IACrC,IAAI,CAAC,QAAQ,IAAI,OAAO,QAAQ,KAAK,QAAQ;QAAE,OAAO,IAAI,CAAC;IAC3D,8DAA8D;IAC9D,MAAM,KAAK,GAAI,QAAgB,CAAC,KAAK,CAAC;IACtC,IAAI,CAAC,KAAK;QAAE,OAAO,IAAI,CAAC;IAExB,MAAM,GAAG,GAA4B,EAAE,CAAC;IACxC,IAAI,KAAK,CAAC,YAAY,KAAK,SAAS;QAAE,GAAG,CAAC,YAAY,GAAG,KAAK,CAAC,YAAY,CAAC;IAC5E,IAAI,KAAK,CAAC,aAAa,KAAK,SAAS;QAAE,GAAG,CAAC,gBAAgB,GAAG,KAAK,CAAC,aAAa,CAAC;IAClF,IAAI,GAAG,CAAC,YAAY,KAAK,SAAS,IAAI,GAAG,CAAC,gBAAgB,KAAK,SAAS,EAAE,CAAC;QACzE,GAAG,CAAC,WAAW,GAAI,GAAG,CAAC,YAAuB,GAAI,GAAG,CAAC,gBAA2B,CAAC;IACpF,CAAC;IAED,OAAO,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC;AAClD,CAAC;AAED;;;GAGG;AACH,SAAS,iBAAiB,CACxB,OAAgB,EAChB,YAA4C,EAC5C,KAAa;IAEb,8DAA8D;IAC9D,MAAM,CAAC,GAAG,OAAc,CAAC;IAEzB,qDAAqD;IACrD,IAAI,OAAO,CAAC,CAAC,MAAM,CAAC,aAAa,CAAC,KAAK,UAAU,IAAI,OAAO,CAAC,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;QAClF,OAAO;YACL,GAAG,CAAC;YACJ,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,aAAa,CAAC;gBAC3B,IAAI,YAAY,GAAY,IAAI,CAAC;gBACjC,IAAI,CAAC;oBACH,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC,EAAE,CAAC;wBAC5B,IAAI,KAAK,EAAE,IAAI,KAAK,cAAc,EAAE,CAAC;4BACnC,YAAY,GAAG,KAAK,CAAC,OAAO,IAAI,CAAC,CAAC,sBAAsB,IAAI,IAAI,CAAC;wBACnE,CAAC;wBACD,MAAM,KAAK,CAAC;oBACd,CAAC;gBACH,CAAC;wBAAS,CAAC;oBACT,MAAM,GAAG,GAAG,YAAY,IAAI,CAAC,CAAC,sBAAsB,IAAI,CAAC,CAAC,YAAY,IAAI,IAAI,CAAC;oBAC/E,IAAI,GAAG,EAAE,CAAC;wBACR,MAAM,MAAM,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC;wBAClC,MAAM,KAAK,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC;wBAChC,YAAY,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,IAAI,SAAS,CAAC,CAAC;oBAC1E,CAAC;yBAAM,CAAC;wBACN,YAAY,CAAC,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;oBACnC,CAAC;gBACH,CAAC;YACH,CAAC;YACD,uBAAuB;YACvB,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC,CAAC;YACjB,GAAG,EAAE,CAAC,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC,CAAC;YACnB,IAAI,YAAY;gBACd,OAAO,CAAC,CAAC,YAAY,CAAC;YACxB,CAAC;YACD,IAAI,sBAAsB;gBACxB,OAAO,CAAC,CAAC,sBAAsB,CAAC;YAClC,CAAC;SACF,CAAC;IACJ,CAAC;IAED,0CAA0C;IAC1C,YAAY,CAAC,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IACjC,OAAO,OAAO,CAAC;AACjB,CAAC"}

package/dist/integrations/llamaindex.d.ts

@@ -0,0 +1,14 @@
+import { TracingMixin, type TracingMixinOptions } from "./_base.js";
+export declare class LightraceLlamaIndexHandler extends TracingMixin {
+    private traceRunId;
+    constructor(opts?: TracingMixinOptions);
+    /** Called when a LlamaIndex trace begins. */
+    startTrace(traceId?: string): void;
+    /** Called when a LlamaIndex trace ends. */
+    endTrace(): void;
+    /** Called when a LlamaIndex event starts. */
+    onEventStart(eventType: string, payload?: Record<string, unknown> | null, eventId?: string, parentId?: string): string;
+    /** Called when a LlamaIndex event ends. */
+    onEventEnd(eventType: string, payload?: Record<string, unknown> | null, eventId?: string): void;
+}
+//# sourceMappingURL=llamaindex.d.ts.map

package/dist/integrations/llamaindex.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"llamaindex.d.ts","sourceRoot":"","sources":["../../src/integrations/llamaindex.ts"],"names":[],"mappings":"AAeA,OAAO,EAAE,YAAY,EAAE,KAAK,mBAAmB,EAAE,MAAM,YAAY,CAAC;AAkBpE,qBAAa,0BAA2B,SAAQ,YAAY;IAC1D,OAAO,CAAC,UAAU,CAAuB;gBAE7B,IAAI,CAAC,EAAE,mBAAmB;IAItC,6CAA6C;IAC7C,UAAU,CAAC,OAAO,CAAC,EAAE,MAAM,GAAG,IAAI;IAelC,2CAA2C;IAC3C,QAAQ,IAAI,IAAI;IAWhB,6CAA6C;IAC7C,YAAY,CACV,SAAS,EAAE,MAAM,EACjB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,EACxC,OAAO,CAAC,EAAE,MAAM,EAChB,QAAQ,CAAC,EAAE,MAAM,GAChB,MAAM;IAmCT,2CAA2C;IAC3C,UAAU,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,IAAI;CAmBhG"}