@uselemma/tracing 0.15.0 → 1.1.0

package/README.md

@@ -1,6 +1,6 @@
 # @uselemma/tracing
 
-Utilities for OpenTelemetry-based tracing and prompt management.
+OpenTelemetry-based tracing for AI agents. Capture inputs, outputs, timing, token usage, and errors — then view everything in [Lemma](https://uselemma.ai).
 
 ## Installation
 
@@ -8,118 +8,51 @@ Utilities for OpenTelemetry-based tracing and prompt management.
 npm install @uselemma/tracing
 ```
 
-## Components
+## Quick Start
 
-### MemorySpanExporter
+### 1. Register the tracer provider
 
-A custom OpenTelemetry span exporter that stores spans in memory for programmatic access. Useful for testing, debugging, or capturing trace data for custom processing.
-
-#### Usage
+Call `registerOTel` once when your application starts. It reads `LEMMA_API_KEY` and `LEMMA_PROJECT_ID` from environment variables by default.
 
 ```typescript
-import { NodeSDK } from "@opentelemetry/sdk-node";
-import { SimpleSpanProcessor } from "@opentelemetry/sdk-trace-base";
-import { MemorySpanExporter } from "@uselemma/tracing";
-
-// Create and configure the exporter
-const memoryExporter = new MemorySpanExporter();
-
-const sdk = new NodeSDK({
-  spanProcessors: [new SimpleSpanProcessor(memoryExporter)],
-});
-
-sdk.start();
-
-// Later, retrieve spans
-const allSpans = memoryExporter.getSpans();
-const spansAsDicts = memoryExporter.getSpansAsDicts();
-const traceSpans = memoryExporter.getSpansByTraceId("your-trace-id");
+import { registerOTel } from "@uselemma/tracing";
 
-// Clear memory when needed
-memoryExporter.clear();
+registerOTel();
 ```
 
-#### Methods
+### 2. Wrap your agent
 
-- **`getSpans(): ReadableSpan[]`** - Get all stored spans as OpenTelemetry ReadableSpan objects
-- **`getSpansAsDicts(): SpanDict[]`** - Get all stored spans as formatted dictionaries
-- **`getSpansByTraceId(traceId: string): SpanDict[]`** - Get all spans for a specific trace ID
-- **`clear(): void`** - Clear all stored spans from memory
-- **`export(spans: ReadableSpan[]): Promise<{ code: ExportResultCode }>`** - Export spans (called automatically by OpenTelemetry)
-- **`shutdown(): Promise<void>`** - Shutdown the exporter
-- **`forceFlush(): Promise<void>`** - Force flush pending spans
-
-### CandidatePromptManager
-
-Manages prompt template overrides using AsyncLocalStorage for context-local state. Useful for A/B testing or evaluating different prompt variations.
-
-#### Usage
+`wrapAgent` creates an OpenTelemetry span around your agent function and provides helpers for recording results.
 
 ```typescript
-import { CandidatePromptManager } from "@uselemma/tracing";
-
-const promptManager = new CandidatePromptManager();
-
-// Run code with prompt overrides
-await promptManager.run(
-  {
-    greeting: "Hello {{ name }}, welcome!",
-    farewell: "Goodbye {{ name }}!",
+import { wrapAgent } from "@uselemma/tracing";
+
+const wrappedFn = wrapAgent(
+  "my-agent",
+  { initialState: { userMessage } },
+  async ({ recordGenerationResults }) => {
+    const result = await doWork(userMessage);
+    recordGenerationResults({ response: result });
+    return result;
   },
-  async () => {
-    // Within this context, candidate prompts will be used
-    const [template, wasOverridden] = promptManager.getEffectiveTemplate(
-      "greeting",
-      "Hi {{ name }}" // default template
-    );
-
-    console.log(template); // "Hello {{ name }}, welcome!"
-    console.log(wasOverridden); // true
-  }
 );
-```
-
-#### Methods
-
-- **`run<T>(candidatePrompts: Record<string, string> | null, callback: () => Promise<T> | T): Promise<T>`**  
-  Run a callback with candidate prompts set in the async context
-- **`getEffectiveTemplate(promptName: string, defaultTemplate: string): [string, boolean]`**  
-  Get the effective template, applying candidate override if present. Returns `[template, wasOverridden]`
-- **`annotateSpan(span: { setAttribute: (key: string, value: unknown) => void }): void`**  
-  Annotate an OpenTelemetry span with candidate prompt metadata
-
-## Example: Dual Processor Setup
-
-Use `MemorySpanExporter` alongside other exporters to both send traces to your backend and capture them locally:
 
-```typescript
-import { NodeSDK } from "@opentelemetry/sdk-node";
-import { SimpleSpanProcessor } from "@opentelemetry/sdk-trace-base";
-import { MemorySpanExporter } from "@uselemma/tracing";
-import { OtherSpanProcessor } from "your-backend";
-
-export const memoryExporter = new MemorySpanExporter();
+const { result, runId } = await wrappedFn();
+```
 
-const sdk = new NodeSDK({
-  spanProcessors: [
-    new OtherSpanProcessor(), // Send to your backend
-    new SimpleSpanProcessor(memoryExporter), // Store in memory for local access
-  ],
-});
+## Environment Variables
 
-sdk.start();
+| Variable           | Description           |
+| ------------------ | --------------------- |
+| `LEMMA_API_KEY`    | Your Lemma API key    |
+| `LEMMA_PROJECT_ID` | Your Lemma project ID |
 
-// In your application code
-import { memoryExporter } from "./instrumentation";
+Both are required unless passed explicitly to `registerOTel()`.
 
-function myTracedFunction() {
-  // ... your code ...
+## Documentation
 
-  // Access spans programmatically
-  const allSpans = memoryExporter.getSpansAsDicts();
-  const myTrace = memoryExporter.getSpansByTraceId(currentTraceId);
-}
-```
+- [Tracing Overview](https://docs.uselemma.ai/tracing/overview) — concepts, API reference, and usage patterns
+- [Vercel AI SDK Integration](https://docs.uselemma.ai/tracing/integrations/vercel-ai-sdk) — framework setup, streaming, and examples
 
 ## License
 

package/dist/index.d.ts

@@ -1,3 +1,3 @@
-export { CandidatePromptManager } from "./candidate-prompt-manager";
+export { registerOTel, type RegisterOTelOptions } from "./register";
 export { wrapAgent, type TraceContext } from "./trace-wrapper";
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,sBAAsB,EAAE,MAAM,4BAA4B,CAAC;AACpE,OAAO,EAAE,SAAS,EAAE,KAAK,YAAY,EAAE,MAAM,iBAAiB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,KAAK,mBAAmB,EAAE,MAAM,YAAY,CAAC;AACpE,OAAO,EAAE,SAAS,EAAE,KAAK,YAAY,EAAE,MAAM,iBAAiB,CAAC"}

package/dist/index.js

@@ -1,8 +1,8 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.wrapAgent = exports.CandidatePromptManager = void 0;
-var candidate_prompt_manager_1 = require("./candidate-prompt-manager");
-Object.defineProperty(exports, "CandidatePromptManager", { enumerable: true, get: function () { return candidate_prompt_manager_1.CandidatePromptManager; } });
+exports.wrapAgent = exports.registerOTel = void 0;
+var register_1 = require("./register");
+Object.defineProperty(exports, "registerOTel", { enumerable: true, get: function () { return register_1.registerOTel; } });
 var trace_wrapper_1 = require("./trace-wrapper");
 Object.defineProperty(exports, "wrapAgent", { enumerable: true, get: function () { return trace_wrapper_1.wrapAgent; } });
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;AAAA,uEAAoE;AAA3D,kIAAA,sBAAsB,OAAA;AAC/B,iDAA+D;AAAtD,0GAAA,SAAS,OAAA"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;AAAA,uCAAoE;AAA3D,wGAAA,YAAY,OAAA;AACrB,iDAA+D;AAAtD,0GAAA,SAAS,OAAA"}

package/dist/register.d.ts

@@ -0,0 +1,33 @@
+import { NodeTracerProvider } from "@opentelemetry/sdk-trace-node";
+export type RegisterOTelOptions = {
+    /** Lemma API key. Defaults to LEMMA_API_KEY environment variable. */
+    apiKey?: string;
+    /** Lemma project ID. Defaults to LEMMA_PROJECT_ID environment variable. */
+    projectId?: string;
+    /** Base URL for the Lemma API. Defaults to https://api.uselemma.ai */
+    baseUrl?: string;
+};
+/**
+ * Registers an OpenTelemetry tracer provider configured to send traces to Lemma.
+ *
+ * This is a convenience wrapper that sets up `NodeTracerProvider` with a
+ * `BatchSpanProcessor` and `OTLPTraceExporter` pointing at the Lemma ingest endpoint.
+ *
+ * @example
+ * // instrumentation.ts (Next.js)
+ * export async function register() {
+ *   if (process.env.NEXT_RUNTIME === 'nodejs') {
+ *     const { registerOTel } = await import('@uselemma/tracing');
+ *     registerOTel();
+ *   }
+ * }
+ *
+ * @example
+ * // With explicit options
+ * registerOTel({
+ *   apiKey: 'lma_...',
+ *   projectId: 'proj_...',
+ * });
+ */
+export declare function registerOTel(options?: RegisterOTelOptions): NodeTracerProvider;
+//# sourceMappingURL=register.d.ts.map

package/dist/register.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"register.d.ts","sourceRoot":"","sources":["../src/register.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAE,MAAM,+BAA+B,CAAC;AAInE,MAAM,MAAM,mBAAmB,GAAG;IAChC,qEAAqE;IACrE,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,2EAA2E;IAC3E,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,sEAAsE;IACtE,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB,CAAC;AAEF;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,wBAAgB,YAAY,CAAC,OAAO,GAAE,mBAAwB,sBAkC7D"}

package/dist/register.js

@@ -0,0 +1,56 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.registerOTel = registerOTel;
+const sdk_trace_node_1 = require("@opentelemetry/sdk-trace-node");
+const sdk_trace_base_1 = require("@opentelemetry/sdk-trace-base");
+const exporter_trace_otlp_proto_1 = require("@opentelemetry/exporter-trace-otlp-proto");
+/**
+ * Registers an OpenTelemetry tracer provider configured to send traces to Lemma.
+ *
+ * This is a convenience wrapper that sets up `NodeTracerProvider` with a
+ * `BatchSpanProcessor` and `OTLPTraceExporter` pointing at the Lemma ingest endpoint.
+ *
+ * @example
+ * // instrumentation.ts (Next.js)
+ * export async function register() {
+ *   if (process.env.NEXT_RUNTIME === 'nodejs') {
+ *     const { registerOTel } = await import('@uselemma/tracing');
+ *     registerOTel();
+ *   }
+ * }
+ *
+ * @example
+ * // With explicit options
+ * registerOTel({
+ *   apiKey: 'lma_...',
+ *   projectId: 'proj_...',
+ * });
+ */
+function registerOTel(options = {}) {
+    // Resolve options, falling back to environment variables / defaults
+    const apiKey = options.apiKey ?? process.env.LEMMA_API_KEY;
+    const projectId = options.projectId ?? process.env.LEMMA_PROJECT_ID;
+    const baseUrl = options.baseUrl ?? "https://api.uselemma.ai";
+    // Fail fast if required credentials are missing
+    if (!apiKey || !projectId) {
+        throw new Error("@uselemma/tracing: Missing API key and/or project ID. Set the LEMMA_API_KEY and LEMMA_PROJECT_ID environment variables or pass them to registerOTel().");
+    }
+    // Create a Node tracer provider with a batch processor that exports
+    // spans to the Lemma OTLP ingest endpoint over HTTP/protobuf
+    const tracerProvider = new sdk_trace_node_1.NodeTracerProvider({
+        spanProcessors: [
+            new sdk_trace_base_1.BatchSpanProcessor(new exporter_trace_otlp_proto_1.OTLPTraceExporter({
+                url: `${baseUrl}/otel/v1/traces`,
+                headers: {
+                    Authorization: `Bearer ${apiKey}`,
+                    "X-Lemma-Project-ID": projectId,
+                },
+            })),
+        ],
+    });
+    // Register this provider as the global tracer provider so all
+    // subsequent `trace.getTracer()` calls use it
+    tracerProvider.register();
+    return tracerProvider;
+}
+//# sourceMappingURL=register.js.map

package/dist/register.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"register.js","sourceRoot":"","sources":["../src/register.ts"],"names":[],"mappings":";;AAmCA,oCAkCC;AArED,kEAAmE;AACnE,kEAAmE;AACnE,wFAA6E;AAW7E;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,SAAgB,YAAY,CAAC,UAA+B,EAAE;IAC5D,oEAAoE;IACpE,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC;IAC3D,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC;IACpE,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,IAAI,yBAAyB,CAAC;IAE7D,gDAAgD;IAChD,IAAI,CAAC,MAAM,IAAI,CAAC,SAAS,EAAE,CAAC;QAC1B,MAAM,IAAI,KAAK,CACb,wJAAwJ,CACzJ,CAAC;IACJ,CAAC;IAED,oEAAoE;IACpE,6DAA6D;IAC7D,MAAM,cAAc,GAAG,IAAI,mCAAkB,CAAC;QAC5C,cAAc,EAAE;YACd,IAAI,mCAAkB,CACpB,IAAI,6CAAiB,CAAC;gBACpB,GAAG,EAAE,GAAG,OAAO,iBAAiB;gBAChC,OAAO,EAAE;oBACP,aAAa,EAAE,UAAU,MAAM,EAAE;oBACjC,oBAAoB,EAAE,SAAS;iBAChC;aACF,CAAC,CACH;SACF;KACF,CAAC,CAAC;IAEH,8DAA8D;IAC9D,8CAA8C;IAC9C,cAAc,CAAC,QAAQ,EAAE,CAAC;IAE1B,OAAO,cAAc,CAAC;AACxB,CAAC"}

package/dist/trace-wrapper.d.ts

@@ -1,16 +1,57 @@
 import { Span } from "@opentelemetry/api";
 export type TraceContext = {
+    /** The active OpenTelemetry span for this agent run. */
     span: Span;
+    /** Unique identifier for this agent run. */
     runId: string;
+    /** Signal that the agent completed successfully. Records the result and ends the span. */
     onComplete: (result: unknown) => void;
+    /** Signal that the agent encountered an error. Records the exception and ends the span. */
     onError: (error: unknown) => void;
+    /** Attach arbitrary generation results (e.g. model outputs) to the span. */
     recordGenerationResults: (results: Record<string, string>) => void;
 };
-export declare function wrapAgent<A extends unknown[]>(agentName: string, options: {
+/**
+ * Wraps an agent function with OpenTelemetry tracing, automatically creating
+ * a span for the agent run and providing a `TraceContext` to the wrapped function.
+ *
+ * The returned function creates a new span on every invocation, attaches agent
+ * metadata (run ID, input, experiment flag), and handles error recording.
+ * The `input` passed to the returned function is recorded as the agent's initial
+ * state on the span.
+ *
+ * @example
+ * const myAgent = wrapAgent(
+ *   'my-agent',
+ *   async (ctx, input: { topic: string }) => {
+ *     // use ctx.span, ctx.runId, ctx.onComplete, etc.
+ *     const result = await doWork(input.topic);
+ *     ctx.onComplete(result);
+ *   },
+ * );
+ * await myAgent({ topic: 'math' });
+ *
+ * @example
+ * // Keep the span open after the function exits
+ * const longRunning = wrapAgent(
+ *   'streaming-agent',
+ *   async (ctx, input) => {
+ *     // caller is responsible for calling ctx.onComplete / ctx.onError
+ *   },
+ *   { endOnExit: false },
+ * );
+ *
+ * @param agentName - Human-readable name used as the span name.
+ * @param fn - The agent function to wrap. Receives a {@link TraceContext} as its first argument and the call-time input as its second.
+ * @param options - Configuration for the agent trace.
+ * @param options.isExperiment - Mark this run as an experiment in Lemma.
+ * @param options.endOnExit - Whether to auto-end the span when the function returns. Defaults to `true`.
+ * @returns An async function that accepts an `input`, executes `fn` inside a traced context, and returns `{ result, runId, span }`.
+ */
+export declare function wrapAgent<Input = unknown>(agentName: string, fn: (traceContext: TraceContext, input: Input) => any, options?: {
     isExperiment?: boolean;
-    initialState?: any;
     endOnExit?: boolean;
-}, fn: (traceContext: TraceContext, ...args: A) => any): (this: any, ...args: A) => Promise<{
+}): (this: any, input: Input) => Promise<{
     result: any;
     runId: string;
     span: Span;

package/dist/trace-wrapper.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"trace-wrapper.d.ts","sourceRoot":"","sources":["../src/trace-wrapper.ts"],"names":[],"mappings":"AAAA,OAAO,EAAkB,IAAI,EAAE,MAAM,oBAAoB,CAAC;AAG1D,MAAM,MAAM,YAAY,GAAG;IACzB,IAAI,EAAE,IAAI,CAAC;IACX,KAAK,EAAE,MAAM,CAAC;IACd,UAAU,EAAE,CAAC,MAAM,EAAE,OAAO,KAAK,IAAI,CAAC;IACtC,OAAO,EAAE,CAAC,KAAK,EAAE,OAAO,KAAK,IAAI,CAAC;IAClC,uBAAuB,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,KAAK,IAAI,CAAC;CACpE,CAAC;AAEF,wBAAgB,SAAS,CAAC,CAAC,SAAS,OAAO,EAAE,EAAE,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE;IAAE,YAAY,CAAC,EAAE,OAAO,CAAC;IAAC,YAAY,CAAC,EAAE,GAAG,CAAC;IAAC,SAAS,CAAC,EAAE,OAAO,CAAA;CAAE,EAAE,EAAE,EAAE,CAAC,YAAY,EAAE,YAAY,EAAE,GAAG,IAAI,EAAE,CAAC,KAAK,GAAG,UACnJ,GAAG,WAAW,CAAC;;;;GAoD9D"}
+{"version":3,"file":"trace-wrapper.d.ts","sourceRoot":"","sources":["../src/trace-wrapper.ts"],"names":[],"mappings":"AAAA,OAAO,EAAkB,IAAI,EAAE,MAAM,oBAAoB,CAAC;AAG1D,MAAM,MAAM,YAAY,GAAG;IACzB,wDAAwD;IACxD,IAAI,EAAE,IAAI,CAAC;IACX,4CAA4C;IAC5C,KAAK,EAAE,MAAM,CAAC;IACd,0FAA0F;IAC1F,UAAU,EAAE,CAAC,MAAM,EAAE,OAAO,KAAK,IAAI,CAAC;IACtC,2FAA2F;IAC3F,OAAO,EAAE,CAAC,KAAK,EAAE,OAAO,KAAK,IAAI,CAAC;IAClC,4EAA4E;IAC5E,uBAAuB,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,KAAK,IAAI,CAAC;CACpE,CAAC;AAEF;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAoCG;AACH,wBAAgB,SAAS,CAAC,KAAK,GAAG,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,YAAY,EAAE,YAAY,EAAE,KAAK,EAAE,KAAK,KAAK,GAAG,EAAE,OAAO,CAAC,EAAE;IAAE,YAAY,CAAC,EAAE,OAAO,CAAC;IAAC,SAAS,CAAC,EAAE,OAAO,CAAA;CAAE,UAC9H,GAAG,SAAS,KAAK;;;;GA8DhE"}

package/dist/trace-wrapper.js

@@ -3,33 +3,78 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.wrapAgent = wrapAgent;
 const api_1 = require("@opentelemetry/api");
 const uuid_1 = require("uuid");
-function wrapAgent(agentName, options, fn) {
-    const wrappedFunction = async function (...args) {
+/**
+ * Wraps an agent function with OpenTelemetry tracing, automatically creating
+ * a span for the agent run and providing a `TraceContext` to the wrapped function.
+ *
+ * The returned function creates a new span on every invocation, attaches agent
+ * metadata (run ID, input, experiment flag), and handles error recording.
+ * The `input` passed to the returned function is recorded as the agent's initial
+ * state on the span.
+ *
+ * @example
+ * const myAgent = wrapAgent(
+ *   'my-agent',
+ *   async (ctx, input: { topic: string }) => {
+ *     // use ctx.span, ctx.runId, ctx.onComplete, etc.
+ *     const result = await doWork(input.topic);
+ *     ctx.onComplete(result);
+ *   },
+ * );
+ * await myAgent({ topic: 'math' });
+ *
+ * @example
+ * // Keep the span open after the function exits
+ * const longRunning = wrapAgent(
+ *   'streaming-agent',
+ *   async (ctx, input) => {
+ *     // caller is responsible for calling ctx.onComplete / ctx.onError
+ *   },
+ *   { endOnExit: false },
+ * );
+ *
+ * @param agentName - Human-readable name used as the span name.
+ * @param fn - The agent function to wrap. Receives a {@link TraceContext} as its first argument and the call-time input as its second.
+ * @param options - Configuration for the agent trace.
+ * @param options.isExperiment - Mark this run as an experiment in Lemma.
+ * @param options.endOnExit - Whether to auto-end the span when the function returns. Defaults to `true`.
+ * @returns An async function that accepts an `input`, executes `fn` inside a traced context, and returns `{ result, runId, span }`.
+ */
+function wrapAgent(agentName, fn, options) {
+    const wrappedFunction = async function (input) {
+        // Obtain the Lemma tracer from the global OTel provider
         const tracer = api_1.trace.getTracer("lemma");
+        // Generate a unique run ID and open a new span for this agent invocation
         const runId = (0, uuid_1.v4)();
         const span = tracer.startSpan(agentName, {
             attributes: {
                 "lemma.agent.run_id": runId,
-                "lemma.agent.input": JSON.stringify(options.initialState),
-                "lemma.agent.is_experiment": options.isExperiment,
+                "lemma.agent.input": JSON.stringify(input),
+                "lemma.agent.is_experiment": options?.isExperiment,
             },
         });
+        // Propagate the span as the active context so child spans are nested correctly
         const ctx = api_1.trace.setSpan(api_1.context.active(), span);
         try {
             return await api_1.context.with(ctx, async () => {
+                // Build the TraceContext callbacks that the agent function can use
+                // to manually signal completion, errors, or record generation results
                 const onComplete = (result) => {
                     span.setAttribute("lemma.agent.output", JSON.stringify(result));
                     span.end();
                 };
                 const onError = (error) => {
+                    // Normalise non-Error values so OTel always receives an Error instance
                     span.recordException(error instanceof Error ? error : new Error(String(error)));
-                    span.setStatus({ code: 2 });
+                    span.setStatus({ code: 2 }); // SpanStatusCode.ERROR
                     span.end();
                 };
                 const recordGenerationResults = (results) => {
                     span.setAttribute("lemma.agent.generation_results", JSON.stringify(results));
                 };
-                const result = await fn.call(this, { span, runId, onComplete, onError, recordGenerationResults }, ...args);
+                // Invoke the wrapped agent function with the trace context and call-time input
+                const result = await fn.call(this, { span, runId, onComplete, onError, recordGenerationResults }, input);
+                // Auto-end the span unless the caller opted out (e.g. for streaming)
                 if (options?.endOnExit !== false) {
                     span.end();
                 }
@@ -37,8 +82,9 @@ function wrapAgent(agentName, options, fn) {
             });
         }
         catch (err) {
+            // Record the exception on the span and mark it as errored
             span.recordException(err);
-            span.setStatus({ code: 2 });
+            span.setStatus({ code: 2 }); // SpanStatusCode.ERROR
             if (options?.endOnExit !== false) {
                 span.end();
             }

package/dist/trace-wrapper.js.map

@@ -1 +1 @@
-{"version":3,"file":"trace-wrapper.js","sourceRoot":"","sources":["../src/trace-wrapper.ts"],"names":[],"mappings":";;AAWA,8BAqDC;AAhED,4CAA0D;AAC1D,+BAAoC;AAUpC,SAAgB,SAAS,CAAsB,SAAiB,EAAE,OAA4E,EAAE,EAAmD;IACjM,MAAM,eAAe,GAAG,KAAK,WAAsB,GAAG,IAAO;QAC3D,MAAM,MAAM,GAAG,WAAK,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QAExC,MAAM,KAAK,GAAG,IAAA,SAAM,GAAE,CAAC;QACvB,MAAM,IAAI,GAAG,MAAM,CAAC,SAAS,CAAC,SAAS,EAAE;YACvC,UAAU,EAAE;gBACV,oBAAoB,EAAE,KAAK;gBAC3B,mBAAmB,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,YAAY,CAAC;gBACzD,2BAA2B,EAAE,OAAO,CAAC,YAAY;aAClD;SACF,CAAC,CAAC;QAEH,MAAM,GAAG,GAAG,WAAK,CAAC,OAAO,CAAC,aAAO,CAAC,MAAM,EAAE,EAAE,IAAI,CAAC,CAAC;QAElD,IAAI,CAAC;YACH,OAAO,MAAM,aAAO,CAAC,IAAI,CAAC,GAAG,EAAE,KAAK,IAAI,EAAE;gBACxC,MAAM,UAAU,GAAG,CAAC,MAAe,EAAE,EAAE;oBACrC,IAAI,CAAC,YAAY,CAAC,oBAAoB,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC;oBAChE,IAAI,CAAC,GAAG,EAAE,CAAC;gBACb,CAAC,CAAC;gBAEF,MAAM,OAAO,GAAG,CAAC,KAAc,EAAE,EAAE;oBACjC,IAAI,CAAC,eAAe,CAAC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;oBAChF,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;oBAC5B,IAAI,CAAC,GAAG,EAAE,CAAC;gBACb,CAAC,CAAC;gBAEF,MAAM,uBAAuB,GAAG,CAAC,OAA+B,EAAE,EAAE;oBAClE,IAAI,CAAC,YAAY,CAAC,gCAAgC,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC;gBAC/E,CAAC,CAAC;gBAEF,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,UAAU,EAAE,OAAO,EAAE,uBAAuB,EAAE,EAAE,GAAG,IAAI,CAAC,CAAC;gBAE3G,IAAI,OAAO,EAAE,SAAS,KAAK,KAAK,EAAE,CAAC;oBACjC,IAAI,CAAC,GAAG,EAAE,CAAC;gBACb,CAAC;gBAED,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;YACjC,CAAC,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,CAAC,eAAe,CAAC,GAAY,CAAC,CAAC;YACnC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;YAE5B,IAAI,OAAO,EAAE,SAAS,KAAK,KAAK,EAAE,CAAC;gBACjC,IAAI,CAAC,GAAG,EAAE,CAAC;YACb,CAAC;YAED,MAAM,GAAG,CAAC;QACZ,CAAC;IACH,CAAC,CAAC;IAEF,OAAO,eAAe,CAAC;AACzB,CAAC"}
+{"version":3,"file":"trace-wrapper.js","sourceRoot":"","sources":["../src/trace-wrapper.ts"],"names":[],"mappings":";;AAqDA,8BA+DC;AApHD,4CAA0D;AAC1D,+BAAoC;AAepC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAoCG;AACH,SAAgB,SAAS,CAAkB,SAAiB,EAAE,EAAqD,EAAE,OAAyD;IAC5K,MAAM,eAAe,GAAG,KAAK,WAAsB,KAAY;QAC7D,wDAAwD;QACxD,MAAM,MAAM,GAAG,WAAK,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QAExC,yEAAyE;QACzE,MAAM,KAAK,GAAG,IAAA,SAAM,GAAE,CAAC;QACvB,MAAM,IAAI,GAAG,MAAM,CAAC,SAAS,CAAC,SAAS,EAAE;YACvC,UAAU,EAAE;gBACV,oBAAoB,EAAE,KAAK;gBAC3B,mBAAmB,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC;gBAC1C,2BAA2B,EAAE,OAAO,EAAE,YAAY;aACnD;SACF,CAAC,CAAC;QAEH,+EAA+E;QAC/E,MAAM,GAAG,GAAG,WAAK,CAAC,OAAO,CAAC,aAAO,CAAC,MAAM,EAAE,EAAE,IAAI,CAAC,CAAC;QAElD,IAAI,CAAC;YACH,OAAO,MAAM,aAAO,CAAC,IAAI,CAAC,GAAG,EAAE,KAAK,IAAI,EAAE;gBACxC,mEAAmE;gBACnE,sEAAsE;gBAEtE,MAAM,UAAU,GAAG,CAAC,MAAe,EAAE,EAAE;oBACrC,IAAI,CAAC,YAAY,CAAC,oBAAoB,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC;oBAChE,IAAI,CAAC,GAAG,EAAE,CAAC;gBACb,CAAC,CAAC;gBAEF,MAAM,OAAO,GAAG,CAAC,KAAc,EAAE,EAAE;oBACjC,uEAAuE;oBACvE,IAAI,CAAC,eAAe,CAAC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;oBAChF,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,uBAAuB;oBACpD,IAAI,CAAC,GAAG,EAAE,CAAC;gBACb,CAAC,CAAC;gBAEF,MAAM,uBAAuB,GAAG,CAAC,OAA+B,EAAE,EAAE;oBAClE,IAAI,CAAC,YAAY,CAAC,gCAAgC,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC;gBAC/E,CAAC,CAAC;gBAEF,+EAA+E;gBAC/E,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,UAAU,EAAE,OAAO,EAAE,uBAAuB,EAAE,EAAE,KAAK,CAAC,CAAC;gBAEzG,qEAAqE;gBACrE,IAAI,OAAO,EAAE,SAAS,KAAK,KAAK,EAAE,CAAC;oBACjC,IAAI,CAAC,GAAG,EAAE,CAAC;gBACb,CAAC;gBAED,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;YACjC,CAAC,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,0DAA0D;YAC1D,IAAI,CAAC,eAAe,CAAC,GAAY,CAAC,CAAC;YACnC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,uBAAuB;YAEpD,IAAI,OAAO,EAAE,SAAS,KAAK,KAAK,EAAE,CAAC;gBACjC,IAAI,CAAC,GAAG,EAAE,CAAC;YACb,CAAC;YAED,MAAM,GAAG,CAAC;QACZ,CAAC;IACH,CAAC,CAAC;IAEF,OAAO,eAAe,CAAC;AACzB,CAAC"}

package/package.json

@@ -1,16 +1,16 @@
 {
   "name": "@uselemma/tracing",
-  "version": "0.15.0",
+  "version": "1.1.0",
   "description": "OpenTelemetry-based tracing module for Lemma",
   "license": "MIT",
   "author": "Lemma",
   "repository": {
     "type": "git",
-    "url": "https://github.com/uselemma/tracing"
+    "url": "https://github.com/uselemma/sdk"
   },
-  "homepage": "https://github.com/uselemma/tracing#readme",
+  "homepage": "https://github.com/uselemma/sdk#readme",
   "bugs": {
-    "url": "https://github.com/uselemma/tracing/issues"
+    "url": "https://github.com/uselemma/sdk/issues"
   },
   "keywords": [
     "opentelemetry",
@@ -23,7 +23,10 @@
   "main": "./dist/index.js",
   "types": "./dist/index.d.ts",
   "exports": {
-    ".": "./dist/index.js"
+    ".": {
+      "types": "./dist/index.d.ts",
+      "default": "./dist/index.js"
+    }
   },
   "files": [
     "dist",
@@ -32,19 +35,14 @@
   ],
   "dependencies": {
     "@opentelemetry/api": "^1.9.0",
-    "@opentelemetry/context-async-hooks": "^2.2.0",
-    "@opentelemetry/core": "^2.2.0",
     "@opentelemetry/exporter-trace-otlp-proto": "^0.211.0",
-    "@opentelemetry/resources": "^2.2.0",
     "@opentelemetry/sdk-trace-base": "^2.2.0",
     "@opentelemetry/sdk-trace-node": "^2.5.0",
-    "nunjucks": "^3.2.4",
     "uuid": "^13.0.0"
   },
   "devDependencies": {
     "@trivago/prettier-plugin-sort-imports": "^5.2.2",
     "@types/node": "^22.18.8",
-    "@types/nunjucks": "^3.2.6",
     "prettier": "^3.6.2",
     "prettier-plugin-tailwindcss": "^0.6.14",
     "typescript": "^5.9.3"

package/dist/candidate-prompt-manager.d.ts

@@ -1,32 +0,0 @@
-/**
- * Manages candidate prompt overrides using AsyncLocalStorage for context-local state.
- */
-export declare class CandidatePromptManager {
-    private readonly _overrides;
-    constructor();
-    /**
-     * Runs a callback with candidate prompts set in the async context.
-     *
-     * @param candidatePrompts - Optional dictionary of prompt name -> candidate template
-     * @param callback - Function to run with the context
-     * @returns Result of the callback
-     */
-    run<T>(candidatePrompts: Record<string, string> | null | undefined, callback: () => Promise<T> | T): Promise<T>;
-    /**
-     * Get the effective template, applying candidate override if present.
-     *
-     * @param promptName - Name of the prompt
-     * @param defaultTemplate - Default template to use if no override
-     * @returns Tuple of [effectiveTemplate, overrideApplied]
-     */
-    getEffectiveTemplate(promptName: string, defaultTemplate: string): [string, boolean];
-    /**
-     * Annotate span with candidate prompt metadata.
-     *
-     * @param span - OpenTelemetry span to annotate
-     */
-    annotateSpan(span: {
-        setAttribute: (key: string, value: unknown) => void;
-    }): void;
-}
-//# sourceMappingURL=candidate-prompt-manager.d.ts.map

package/dist/candidate-prompt-manager.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"candidate-prompt-manager.d.ts","sourceRoot":"","sources":["../src/candidate-prompt-manager.ts"],"names":[],"mappings":"AAEA;;GAEG;AACH,qBAAa,sBAAsB;IACjC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAmD;;IAM9E;;;;;;OAMG;IACG,GAAG,CAAC,CAAC,EACT,gBAAgB,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,IAAI,GAAG,SAAS,EAC3D,QAAQ,EAAE,MAAM,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,GAC7B,OAAO,CAAC,CAAC,CAAC;IAIb;;;;;;OAMG;IACH,oBAAoB,CAClB,UAAU,EAAE,MAAM,EAClB,eAAe,EAAE,MAAM,GACtB,CAAC,MAAM,EAAE,OAAO,CAAC;IAQpB;;;;OAIG;IACH,YAAY,CAAC,IAAI,EAAE;QACjB,YAAY,EAAE,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,KAAK,IAAI,CAAC;KACrD,GAAG,IAAI;CAaT"}

package/dist/candidate-prompt-manager.js

@@ -1,55 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.CandidatePromptManager = void 0;
-const node_async_hooks_1 = require("node:async_hooks");
-/**
- * Manages candidate prompt overrides using AsyncLocalStorage for context-local state.
- */
-class CandidatePromptManager {
-    _overrides;
-    constructor() {
-        this._overrides = new node_async_hooks_1.AsyncLocalStorage();
-    }
-    /**
-     * Runs a callback with candidate prompts set in the async context.
-     *
-     * @param candidatePrompts - Optional dictionary of prompt name -> candidate template
-     * @param callback - Function to run with the context
-     * @returns Result of the callback
-     */
-    async run(candidatePrompts, callback) {
-        return this._overrides.run(candidatePrompts ?? null, callback);
-    }
-    /**
-     * Get the effective template, applying candidate override if present.
-     *
-     * @param promptName - Name of the prompt
-     * @param defaultTemplate - Default template to use if no override
-     * @returns Tuple of [effectiveTemplate, overrideApplied]
-     */
-    getEffectiveTemplate(promptName, defaultTemplate) {
-        const overrides = this._overrides.getStore();
-        if (overrides && promptName in overrides) {
-            return [overrides[promptName], true];
-        }
-        return [defaultTemplate, false];
-    }
-    /**
-     * Annotate span with candidate prompt metadata.
-     *
-     * @param span - OpenTelemetry span to annotate
-     */
-    annotateSpan(span) {
-        const overrides = this._overrides.getStore();
-        if (overrides !== null && overrides !== undefined) {
-            try {
-                span.setAttribute("candidate_prompts.count", Object.keys(overrides).length);
-            }
-            catch {
-                // Best-effort; avoid breaking tracing on attribute errors
-            }
-        }
-    }
-}
-exports.CandidatePromptManager = CandidatePromptManager;
-//# sourceMappingURL=candidate-prompt-manager.js.map

package/dist/candidate-prompt-manager.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"candidate-prompt-manager.js","sourceRoot":"","sources":["../src/candidate-prompt-manager.ts"],"names":[],"mappings":";;;AAAA,uDAAqD;AAErD;;GAEG;AACH,MAAa,sBAAsB;IAChB,UAAU,CAAmD;IAE9E;QACE,IAAI,CAAC,UAAU,GAAG,IAAI,oCAAiB,EAAiC,CAAC;IAC3E,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,GAAG,CACP,gBAA2D,EAC3D,QAA8B;QAE9B,OAAO,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,gBAAgB,IAAI,IAAI,EAAE,QAAQ,CAAC,CAAC;IACjE,CAAC;IAED;;;;;;OAMG;IACH,oBAAoB,CAClB,UAAkB,EAClB,eAAuB;QAEvB,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC,QAAQ,EAAE,CAAC;QAC7C,IAAI,SAAS,IAAI,UAAU,IAAI,SAAS,EAAE,CAAC;YACzC,OAAO,CAAC,SAAS,CAAC,UAAU,CAAC,EAAE,IAAI,CAAC,CAAC;QACvC,CAAC;QACD,OAAO,CAAC,eAAe,EAAE,KAAK,CAAC,CAAC;IAClC,CAAC;IAED;;;;OAIG;IACH,YAAY,CAAC,IAEZ;QACC,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC,QAAQ,EAAE,CAAC;QAC7C,IAAI,SAAS,KAAK,IAAI,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;YAClD,IAAI,CAAC;gBACH,IAAI,CAAC,YAAY,CACf,yBAAyB,EACzB,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,MAAM,CAC9B,CAAC;YACJ,CAAC;YAAC,MAAM,CAAC;gBACP,0DAA0D;YAC5D,CAAC;QACH,CAAC;IACH,CAAC;CACF;AA3DD,wDA2DC"}

package/dist/memory-exporter.d.ts

@@ -1,43 +0,0 @@
-import { ExportResultCode } from "@opentelemetry/core";
-import type { ReadableSpan, SpanExporter } from "@opentelemetry/sdk-trace-base";
-export interface SpanDict {
-    timestamp: string;
-    trace_id: string;
-    span_id: string;
-    parent_span_id: string | null;
-    name: string;
-    kind: string;
-    start_time_ns: string;
-    end_time_ns: string | null;
-    duration_ms: number | null;
-    attributes: Record<string, unknown>;
-    status: {
-        status_code: string;
-        description?: string;
-    } | null;
-    events: Array<{
-        name: string;
-        timestamp_ns: string;
-        attributes: Record<string, unknown>;
-    }>;
-    resource: Record<string, unknown>;
-}
-/**
- * Span exporter that stores spans in memory for retrieval.
- */
-export declare class MemorySpanExporter implements SpanExporter {
-    private readonly _spans;
-    export(spans: ReadableSpan[]): Promise<{
-        code: ExportResultCode;
-    }>;
-    getSpans(): ReadableSpan[];
-    getSpansAsDicts(): SpanDict[];
-    getSpansByTraceId(traceId: string): SpanDict[];
-    private _formatTraceId;
-    private _formatSpanId;
-    private _spanToDict;
-    clear(): void;
-    shutdown(): Promise<void>;
-    forceFlush(): Promise<void>;
-}
-//# sourceMappingURL=memory-exporter.d.ts.map

package/dist/memory-exporter.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"memory-exporter.d.ts","sourceRoot":"","sources":["../src/memory-exporter.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AACvD,OAAO,KAAK,EAAE,YAAY,EAAE,YAAY,EAAE,MAAM,+BAA+B,CAAC;AAEhF,MAAM,WAAW,QAAQ;IACvB,SAAS,EAAE,MAAM,CAAC;IAClB,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,MAAM,CAAC;IAChB,cAAc,EAAE,MAAM,GAAG,IAAI,CAAC;IAC9B,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,aAAa,EAAE,MAAM,CAAC;IACtB,WAAW,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,WAAW,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACpC,MAAM,EAAE;QACN,WAAW,EAAE,MAAM,CAAC;QACpB,WAAW,CAAC,EAAE,MAAM,CAAC;KACtB,GAAG,IAAI,CAAC;IACT,MAAM,EAAE,KAAK,CAAC;QACZ,IAAI,EAAE,MAAM,CAAC;QACb,YAAY,EAAE,MAAM,CAAC;QACrB,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;KACrC,CAAC,CAAC;IACH,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACnC;AAED;;GAEG;AACH,qBAAa,kBAAmB,YAAW,YAAY;IACrD,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAsB;IAE7C,MAAM,CAAC,KAAK,EAAE,YAAY,EAAE,GAAG,OAAO,CAAC;QAAE,IAAI,EAAE,gBAAgB,CAAA;KAAE,CAAC;IAKlE,QAAQ,IAAI,YAAY,EAAE;IAI1B,eAAe,IAAI,QAAQ,EAAE;IAI7B,iBAAiB,CAAC,OAAO,EAAE,MAAM,GAAG,QAAQ,EAAE;IAiB9C,OAAO,CAAC,cAAc;IAKtB,OAAO,CAAC,aAAa;IAKrB,OAAO,CAAC,WAAW;IAkDnB,KAAK,IAAI,IAAI;IAIb,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;IAIzB,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;CAG5B"}

package/dist/memory-exporter.js

@@ -1,95 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.MemorySpanExporter = void 0;
-const core_1 = require("@opentelemetry/core");
-/**
- * Span exporter that stores spans in memory for retrieval.
- */
-class MemorySpanExporter {
-    _spans = [];
-    export(spans) {
-        this._spans.push(...spans);
-        return Promise.resolve({ code: core_1.ExportResultCode.SUCCESS });
-    }
-    getSpans() {
-        return [...this._spans];
-    }
-    getSpansAsDicts() {
-        return this._spans.map((span) => this._spanToDict(span));
-    }
-    getSpansByTraceId(traceId) {
-        const formattedTraceId = this._formatTraceId(traceId);
-        console.log(this._spans.map((span) => {
-            const spanDict = this._spanToDict(span);
-            return {
-                traceId: spanDict.trace_id,
-                spanId: spanDict.span_id,
-                parentSpanId: spanDict.parent_span_id,
-            };
-        }));
-        return this._spans
-            .map((span) => this._spanToDict(span))
-            .filter((span) => span.trace_id === formattedTraceId);
-    }
-    _formatTraceId(traceId) {
-        // Ensure trace ID is 32-character hex string
-        return traceId.padStart(32, "0").slice(0, 32);
-    }
-    _formatSpanId(spanId) {
-        // Ensure span ID is 16-character hex string
-        return spanId.padStart(16, "0").slice(0, 16);
-    }
-    _spanToDict(span) {
-        const spanCtx = span.spanContext();
-        const traceId = this._formatTraceId(spanCtx.traceId);
-        const spanId = this._formatSpanId(spanCtx.spanId);
-        const parentSpanId = span.parentSpanContext
-            ? this._formatSpanId(span.parentSpanContext.spanId)
-            : null;
-        return {
-            timestamp: new Date().toISOString(),
-            trace_id: traceId,
-            span_id: spanId,
-            parent_span_id: parentSpanId,
-            name: span.name,
-            kind: span.kind.toString(),
-            start_time_ns: (BigInt(span.startTime[0]) * 1000000000n +
-                BigInt(span.startTime[1])).toString(),
-            end_time_ns: span.endTime
-                ? (BigInt(span.endTime[0]) * 1000000000n +
-                    BigInt(span.endTime[1])).toString()
-                : null,
-            duration_ms: span.endTime
-                ? (span.endTime[0] - span.startTime[0]) * 1000 +
-                    (span.endTime[1] - span.startTime[1]) / 1_000_000
-                : null,
-            attributes: span.attributes ? { ...span.attributes } : {},
-            status: span.status
-                ? {
-                    status_code: span.status.code.toString(),
-                    description: span.status.message,
-                }
-                : null,
-            events: (span.events || []).map((event) => ({
-                name: event.name,
-                timestamp_ns: (BigInt(event.time[0]) * 1000000000n +
-                    BigInt(event.time[1])).toString(),
-                attributes: event.attributes ? { ...event.attributes } : {},
-            })),
-            resource: span.resource?.attributes
-                ? { ...span.resource.attributes }
-                : {},
-        };
-    }
-    clear() {
-        this._spans.length = 0;
-    }
-    shutdown() {
-        return Promise.resolve();
-    }
-    forceFlush() {
-        return Promise.resolve();
-    }
-}
-exports.MemorySpanExporter = MemorySpanExporter;
-//# sourceMappingURL=memory-exporter.js.map

package/dist/memory-exporter.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"memory-exporter.js","sourceRoot":"","sources":["../src/memory-exporter.ts"],"names":[],"mappings":";;;AAAA,8CAAuD;AA0BvD;;GAEG;AACH,MAAa,kBAAkB;IACZ,MAAM,GAAmB,EAAE,CAAC;IAE7C,MAAM,CAAC,KAAqB;QAC1B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,CAAC;QAC3B,OAAO,OAAO,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,uBAAgB,CAAC,OAAO,EAAE,CAAC,CAAC;IAC7D,CAAC;IAED,QAAQ;QACN,OAAO,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC;IAC1B,CAAC;IAED,eAAe;QACb,OAAO,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC;IAC3D,CAAC;IAED,iBAAiB,CAAC,OAAe;QAC/B,MAAM,gBAAgB,GAAG,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QAEtD,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;YACnC,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;YACxC,OAAO;gBACL,OAAO,EAAE,QAAQ,CAAC,QAAQ;gBAC1B,MAAM,EAAE,QAAQ,CAAC,OAAO;gBACxB,YAAY,EAAE,QAAQ,CAAC,cAAc;aACtC,CAAC;QACJ,CAAC,CAAC,CAAC,CAAA;QAEH,OAAO,IAAI,CAAC,MAAM;aACf,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;aACrC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,KAAK,gBAAgB,CAAC,CAAC;IAC1D,CAAC;IAEO,cAAc,CAAC,OAAe;QACpC,6CAA6C;QAC7C,OAAO,OAAO,CAAC,QAAQ,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;IAChD,CAAC;IAEO,aAAa,CAAC,MAAc;QAClC,4CAA4C;QAC5C,OAAO,MAAM,CAAC,QAAQ,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;IAC/C,CAAC;IAEO,WAAW,CAAC,IAAkB;QACpC,MAAM,OAAO,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;QACnC,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QACrD,MAAM,MAAM,GAAG,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAClD,MAAM,YAAY,GAAG,IAAI,CAAC,iBAAiB;YACzC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC;YACnD,CAAC,CAAC,IAAI,CAAC;QAET,OAAO;YACL,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,QAAQ,EAAE,OAAO;YACjB,OAAO,EAAE,MAAM;YACf,cAAc,EAAE,YAAY;YAC5B,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE;YAC1B,aAAa,EAAE,CACb,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,GAAG,WAAc;gBAC1C,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAC1B,CAAC,QAAQ,EAAE;YACZ,WAAW,EAAE,IAAI,CAAC,OAAO;gBACvB,CAAC,CAAC,CACE,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,GAAG,WAAc;oBACxC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CACxB,CAAC,QAAQ,EAAE;gBACd,CAAC,CAAC,IAAI;YACR,WAAW,EAAE,IAAI,CAAC,OAAO;gBACvB,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,GAAG,IAAI;oBAC5C,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,GAAG,SAAS;gBACnD,CAAC,CAAC,IAAI;YACR,UAAU,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE;YACzD,MAAM,EAAE,IAAI,CAAC,MAAM;gBACjB,CAAC,CAAC;oBACE,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,EAAE;oBACxC,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO;iBACjC;gBACH,CAAC,CAAC,IAAI;YACR,MAAM,EAAE,CAAC,IAAI,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;gBAC1C,IAAI,EAAE,KAAK,CAAC,IAAI;gBAChB,YAAY,EAAE,CACZ,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,WAAc;oBACtC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CACtB,CAAC,QAAQ,EAAE;gBACZ,UAAU,EAAE,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,GAAG,KAAK,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE;aAC5D,CAAC,CAAC;YACH,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,UAAU;gBACjC,CAAC,CAAC,EAAE,GAAG,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE;gBACjC,CAAC,CAAC,EAAE;SACP,CAAC;IACJ,CAAC;IAED,KAAK;QACH,IAAI,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC;IACzB,CAAC;IAED,QAAQ;QACN,OAAO,OAAO,CAAC,OAAO,EAAE,CAAC;IAC3B,CAAC;IAED,UAAU;QACR,OAAO,OAAO,CAAC,OAAO,EAAE,CAAC;IAC3B,CAAC;CACF;AAxGD,gDAwGC"}