@uselemma/tracing 3.0.2 → 4.0.0

package/examples/vercel-ai-v6.ts

@@ -0,0 +1,27 @@
+import { generateText } from "ai";
+import { Lemma, vercelAI } from "@uselemma/tracing";
+
+const lemma = new Lemma({
+  apiKey: process.env.LEMMA_API_KEY,
+  projectId: process.env.LEMMA_PROJECT_ID,
+});
+
+export async function runVercelAIV6(
+  model: Parameters<typeof generateText>[0]["model"],
+  userMessage: string,
+) {
+  const trace = lemma.trace({
+    name: "support-agent",
+    input: userMessage,
+  });
+
+  const result = await generateText({
+    model,
+    prompt: userMessage,
+    experimental_telemetry: {
+      integrations: [vercelAI({ trace })],
+    },
+  });
+
+  return result.text;
+}

package/examples/vercel-ai-v7.ts

@@ -0,0 +1,27 @@
+import { generateText } from "ai";
+import { Lemma, vercelAI } from "@uselemma/tracing";
+
+const lemma = new Lemma({
+  apiKey: process.env.LEMMA_API_KEY,
+  projectId: process.env.LEMMA_PROJECT_ID,
+});
+
+export async function runVercelAIV7(
+  model: Parameters<typeof generateText>[0]["model"],
+  userMessage: string,
+) {
+  const trace = lemma.trace({
+    name: "support-agent",
+    input: userMessage,
+  });
+
+  const result = await generateText({
+    model,
+    prompt: userMessage,
+    telemetry: {
+      integrations: [vercelAI({ trace })],
+    },
+  });
+
+  return result.text;
+}

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@uselemma/tracing",
-  "version": "3.0.2",
-  "description": "OpenTelemetry-based tracing module for Lemma",
+  "version": "4.0.0",
+  "description": "HTTP tracing SDK for Lemma",
   "license": "MIT",
   "author": "Lemma",
   "repository": {
@@ -13,12 +13,11 @@
     "url": "https://github.com/uselemma/sdk/issues"
   },
   "keywords": [
-    "opentelemetry",
     "tracing",
     "observability",
     "monitoring",
     "llm",
-    "instrumentation"
+    "sdk"
   ],
   "main": "./dist/index.js",
   "types": "./dist/index.d.ts",
@@ -30,19 +29,12 @@
   },
   "files": [
     "dist",
+    "examples",
     "README.md",
     "LICENSE"
   ],
-  "dependencies": {
-    "@opentelemetry/api": "^1.9.0",
-    "@opentelemetry/exporter-trace-otlp-proto": "^0.211.0",
-    "@opentelemetry/sdk-trace-base": "^2.2.0",
-    "@opentelemetry/sdk-trace-node": "^2.5.0",
-    "uuid": "^13.0.0"
-  },
+  "dependencies": {},
   "devDependencies": {
-    "@arizeai/openinference-instrumentation-openai": "^4.0.5",
-    "@opentelemetry/instrumentation": "^0.213.0",
     "@trivago/prettier-plugin-sort-imports": "^5.2.2",
     "@types/node": "^22.18.8",
     "@vitest/coverage-v8": "^3.2.4",

package/dist/register.d.ts

@@ -1,61 +0,0 @@
-import { NodeTracerProvider } from "@opentelemetry/sdk-trace-node";
-import { type SpanProcessor } from "@opentelemetry/sdk-trace-base";
-export type LemmaOTelOptions = {
-    /** 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;
-};
-/** @deprecated Use {@link LemmaOTelOptions} */
-export type RegisterOTelOptions = LemmaOTelOptions;
-/** @deprecated Use {@link LemmaOTelOptions} */
-export type CreateLemmaSpanProcessorOptions = LemmaOTelOptions;
-/**
- * Creates a `BatchSpanProcessor` pre-configured to export traces to Lemma.
- *
- * Use this when you need to add Lemma as one of several span processors on an
- * existing `NodeTracerProvider` (e.g. for dual-export to Lemma and another
- * backend simultaneously).
- *
- * @example
- * // Dual export: Lemma + your own OTLP collector
- * import { NodeTracerProvider } from "@opentelemetry/sdk-trace-node";
- * import { BatchSpanProcessor } from "@opentelemetry/sdk-trace-base";
- * import { OTLPTraceExporter } from "@opentelemetry/exporter-trace-otlp-proto";
- * import { createLemmaSpanProcessor } from "@uselemma/tracing";
- *
- * const provider = new NodeTracerProvider({
- *   spanProcessors: [
- *     createLemmaSpanProcessor(),
- *     new BatchSpanProcessor(new OTLPTraceExporter({ url: "http://localhost:4318/v1/traces" })),
- *   ],
- * });
- * provider.register();
- */
-export declare function createLemmaSpanProcessor(options?: LemmaOTelOptions): SpanProcessor;
-/**
- * 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?: LemmaOTelOptions): NodeTracerProvider;
-//# sourceMappingURL=register.d.ts.map

package/dist/register.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"register.d.ts","sourceRoot":"","sources":["../src/register.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAE,MAAM,+BAA+B,CAAC;AACnE,OAAO,EAAE,KAAK,aAAa,EAAE,MAAM,+BAA+B,CAAC;AAInE,MAAM,MAAM,gBAAgB,GAAG;IAC7B,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,+CAA+C;AAC/C,MAAM,MAAM,mBAAmB,GAAG,gBAAgB,CAAC;AACnD,+CAA+C;AAC/C,MAAM,MAAM,+BAA+B,GAAG,gBAAgB,CAAC;AAG/D;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,wBAAgB,wBAAwB,CACtC,OAAO,GAAE,gBAAqB,GAC7B,aAAa,CAoBf;AAED;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,wBAAgB,YAAY,CAAC,OAAO,GAAE,gBAAqB,sBAU1D"}

package/dist/register.js

@@ -1,76 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.createLemmaSpanProcessor = createLemmaSpanProcessor;
-exports.registerOTel = registerOTel;
-const sdk_trace_node_1 = require("@opentelemetry/sdk-trace-node");
-const exporter_trace_otlp_proto_1 = require("@opentelemetry/exporter-trace-otlp-proto");
-const run_batch_span_processor_1 = require("./run-batch-span-processor");
-/**
- * Creates a `BatchSpanProcessor` pre-configured to export traces to Lemma.
- *
- * Use this when you need to add Lemma as one of several span processors on an
- * existing `NodeTracerProvider` (e.g. for dual-export to Lemma and another
- * backend simultaneously).
- *
- * @example
- * // Dual export: Lemma + your own OTLP collector
- * import { NodeTracerProvider } from "@opentelemetry/sdk-trace-node";
- * import { BatchSpanProcessor } from "@opentelemetry/sdk-trace-base";
- * import { OTLPTraceExporter } from "@opentelemetry/exporter-trace-otlp-proto";
- * import { createLemmaSpanProcessor } from "@uselemma/tracing";
- *
- * const provider = new NodeTracerProvider({
- *   spanProcessors: [
- *     createLemmaSpanProcessor(),
- *     new BatchSpanProcessor(new OTLPTraceExporter({ url: "http://localhost:4318/v1/traces" })),
- *   ],
- * });
- * provider.register();
- */
-function createLemmaSpanProcessor(options = {}) {
-    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";
-    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 createLemmaSpanProcessor().");
-    }
-    return new run_batch_span_processor_1.RunBatchSpanProcessor(new exporter_trace_otlp_proto_1.OTLPTraceExporter({
-        url: `${baseUrl}/otel/v1/traces`,
-        headers: {
-            Authorization: `Bearer ${apiKey}`,
-            "X-Lemma-Project-ID": projectId,
-        },
-    }));
-}
-/**
- * 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 = {}) {
-    const tracerProvider = new sdk_trace_node_1.NodeTracerProvider({
-        spanProcessors: [createLemmaSpanProcessor(options)],
-    });
-    // 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

@@ -1 +0,0 @@
-{"version":3,"file":"register.js","sourceRoot":"","sources":["../src/register.ts"],"names":[],"mappings":";;AA0CA,4DAsBC;AAwBD,oCAUC;AAlGD,kEAAmE;AAEnE,wFAA6E;AAC7E,yEAAmE;AAiBnE;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,SAAgB,wBAAwB,CACtC,UAA4B,EAAE;IAE9B,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,IAAI,CAAC,MAAM,IAAI,CAAC,SAAS,EAAE,CAAC;QAC1B,MAAM,IAAI,KAAK,CACb,oKAAoK,CACrK,CAAC;IACJ,CAAC;IAED,OAAO,IAAI,gDAAqB,CAC9B,IAAI,6CAAiB,CAAC;QACpB,GAAG,EAAE,GAAG,OAAO,iBAAiB;QAChC,OAAO,EAAE;YACP,aAAa,EAAE,UAAU,MAAM,EAAE;YACjC,oBAAoB,EAAE,SAAS;SAChC;KACF,CAAC,CACH,CAAC;AACJ,CAAC;AAED;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,SAAgB,YAAY,CAAC,UAA4B,EAAE;IACzD,MAAM,cAAc,GAAG,IAAI,mCAAkB,CAAC;QAC5C,cAAc,EAAE,CAAC,wBAAwB,CAAC,OAAO,CAAC,CAAC;KACpD,CAAC,CAAC;IAEH,8DAA8D;IAC9D,8CAA8C;IAC9C,cAAc,CAAC,QAAQ,EAAE,CAAC;IAE1B,OAAO,cAAc,CAAC;AACxB,CAAC"}

package/dist/run-batch-span-processor.d.ts

@@ -1,25 +0,0 @@
-import type { Context } from "@opentelemetry/api";
-import { type ReadableSpan, type Span, type SpanExporter, type SpanProcessor } from "@opentelemetry/sdk-trace-base";
-export declare class RunBatchSpanProcessor implements SpanProcessor {
-    private isShutdown;
-    private spanIdToRunId;
-    private topLevelSpanIdByRunId;
-    private directChildCountByRunId;
-    private directChildSpanIdToRunId;
-    private batches;
-    private endedRuns;
-    private readonly exporter;
-    constructor(exporter: SpanExporter);
-    onStart(span: Span, _parentContext: Context): void;
-    onEnd(span: ReadableSpan): void;
-    forceFlush(): Promise<void>;
-    shutdown(): Promise<void>;
-    private isTopLevelRun;
-    private getRunIdFromSpan;
-    private getInstrumentationScopeName;
-    private shouldSkipExport;
-    private exportRunBatch;
-    private hasNoOpenDirectChildren;
-    private clearRunMapping;
-}
-//# sourceMappingURL=run-batch-span-processor.d.ts.map

package/dist/run-batch-span-processor.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"run-batch-span-processor.d.ts","sourceRoot":"","sources":["../src/run-batch-span-processor.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EACL,KAAK,YAAY,EACjB,KAAK,IAAI,EACT,KAAK,YAAY,EACjB,KAAK,aAAa,EACnB,MAAM,+BAA+B,CAAC;AAMvC,qBAAa,qBAAsB,YAAW,aAAa;IACzD,OAAO,CAAC,UAAU,CAAS;IAC3B,OAAO,CAAC,aAAa,CAA4B;IACjD,OAAO,CAAC,qBAAqB,CAA4B;IACzD,OAAO,CAAC,uBAAuB,CAA4B;IAC3D,OAAO,CAAC,wBAAwB,CAA4B;IAC5D,OAAO,CAAC,OAAO,CAAoC;IACnD,OAAO,CAAC,SAAS,CAAoB;IACrC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAe;gBAE5B,QAAQ,EAAE,YAAY;IAIlC,OAAO,CAAC,IAAI,EAAE,IAAI,EAAE,cAAc,EAAE,OAAO,GAAG,IAAI;IA6BlD,KAAK,CAAC,IAAI,EAAE,YAAY,GAAG,IAAI;IAoCzB,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAS3B,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;IAS/B,OAAO,CAAC,aAAa;IAIrB,OAAO,CAAC,gBAAgB;IAOxB,OAAO,CAAC,2BAA2B;IAQnC,OAAO,CAAC,gBAAgB;YAIV,cAAc;IAmB5B,OAAO,CAAC,uBAAuB;IAI/B,OAAO,CAAC,eAAe;CAcxB"}

package/dist/run-batch-span-processor.js

@@ -1,139 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.RunBatchSpanProcessor = void 0;
-const crypto_1 = require("crypto");
-const debug_mode_1 = require("./debug-mode");
-class RunBatchSpanProcessor {
-    isShutdown = false;
-    spanIdToRunId = new Map();
-    topLevelSpanIdByRunId = new Map();
-    directChildCountByRunId = new Map();
-    directChildSpanIdToRunId = new Map();
-    batches = new Map();
-    endedRuns = new Set();
-    exporter;
-    constructor(exporter) {
-        this.exporter = exporter;
-    }
-    onStart(span, _parentContext) {
-        const spanId = span.spanContext().spanId;
-        if (this.isTopLevelRun(span)) {
-            const runId = this.getRunIdFromSpan(span) ?? (0, crypto_1.randomUUID)();
-            span.setAttribute("lemma.run_id", runId);
-            this.spanIdToRunId.set(spanId, runId);
-            this.topLevelSpanIdByRunId.set(runId, spanId);
-            this.directChildCountByRunId.set(runId, 0);
-            (0, debug_mode_1.lemmaDebug)("processor", "onStart: top-level run span", { spanId, runId });
-            return;
-        }
-        const parentSpanId = span.parentSpanContext?.spanId;
-        if (!parentSpanId)
-            return;
-        const runId = this.spanIdToRunId.get(parentSpanId);
-        if (!runId)
-            return;
-        if (this.topLevelSpanIdByRunId.get(runId) === parentSpanId) {
-            this.directChildSpanIdToRunId.set(spanId, runId);
-            this.directChildCountByRunId.set(runId, (this.directChildCountByRunId.get(runId) ?? 0) + 1);
-        }
-        this.spanIdToRunId.set(spanId, runId);
-        span.setAttribute("lemma.run_id", runId);
-        (0, debug_mode_1.lemmaDebug)("processor", "onStart: child span", { spanName: span.name, spanId, runId });
-    }
-    onEnd(span) {
-        const spanId = span.spanContext().spanId;
-        const runId = this.spanIdToRunId.get(spanId) ??
-            (span.parentSpanContext?.spanId
-                ? this.spanIdToRunId.get(span.parentSpanContext.spanId)
-                : undefined);
-        if (!runId)
-            return;
-        const isTopLevelRun = this.isTopLevelRun(span);
-        const shouldSkipExport = this.shouldSkipExport(span);
-        const directChildRunId = this.directChildSpanIdToRunId.get(spanId);
-        if (directChildRunId) {
-            this.directChildSpanIdToRunId.delete(spanId);
-            const currentCount = this.directChildCountByRunId.get(directChildRunId) ?? 0;
-            const nextCount = Math.max(0, currentCount - 1);
-            this.directChildCountByRunId.set(directChildRunId, nextCount);
-            (0, debug_mode_1.lemmaDebug)("processor", "onEnd: direct child ended", { spanName: span.name, spanId, runId: directChildRunId, remainingChildren: nextCount });
-        }
-        (0, debug_mode_1.lemmaDebug)("processor", "onEnd: span ended", { spanName: span.name, spanId, runId, isTopLevelRun, skipped: shouldSkipExport });
-        if (!shouldSkipExport) {
-            const batch = this.batches.get(runId);
-            if (batch)
-                batch.push(span);
-            else
-                this.batches.set(runId, [span]);
-        }
-        if (isTopLevelRun) {
-            this.endedRuns.add(runId);
-        }
-        void this.exportRunBatch(runId, false);
-    }
-    async forceFlush() {
-        (0, debug_mode_1.lemmaDebug)("processor", "forceFlush called");
-        const runIds = [...this.batches.keys()];
-        await Promise.all(runIds.map((runId) => this.exportRunBatch(runId, true)));
-        if (this.exporter.forceFlush) {
-            await this.exporter.forceFlush();
-        }
-    }
-    async shutdown() {
-        if (this.isShutdown)
-            return;
-        (0, debug_mode_1.lemmaDebug)("processor", "shutdown called");
-        this.isShutdown = true;
-        await this.forceFlush();
-        this.exporter.shutdown();
-    }
-    isTopLevelRun(span) {
-        return span.name === "ai.agent.run" && !span.parentSpanContext;
-    }
-    getRunIdFromSpan(span) {
-        const attributes = span
-            .attributes;
-        const runId = attributes?.["lemma.run_id"];
-        return typeof runId === "string" && runId.length > 0 ? runId : undefined;
-    }
-    getInstrumentationScopeName(span) {
-        return span.instrumentationScope?.name;
-    }
-    shouldSkipExport(span) {
-        return this.getInstrumentationScopeName(span) === "next.js";
-    }
-    async exportRunBatch(runId, force) {
-        if (!force && (!this.endedRuns.has(runId) || !this.hasNoOpenDirectChildren(runId))) {
-            return;
-        }
-        const batch = this.batches.get(runId);
-        this.batches.delete(runId);
-        this.endedRuns.delete(runId);
-        this.clearRunMapping(runId);
-        if (!batch || batch.length === 0)
-            return;
-        (0, debug_mode_1.lemmaDebug)("processor", "exporting batch", { runId, spanCount: batch.length, force });
-        await new Promise((resolve) => {
-            this.exporter.export(batch, () => resolve());
-        });
-    }
-    hasNoOpenDirectChildren(runId) {
-        return (this.directChildCountByRunId.get(runId) ?? 0) === 0;
-    }
-    clearRunMapping(runId) {
-        for (const [spanId, mappedRunId] of this.spanIdToRunId.entries()) {
-            if (mappedRunId === runId) {
-                this.spanIdToRunId.delete(spanId);
-            }
-        }
-        for (const [spanId, mappedRunId] of this.directChildSpanIdToRunId.entries()) {
-            if (mappedRunId === runId) {
-                this.directChildSpanIdToRunId.delete(spanId);
-            }
-        }
-        this.topLevelSpanIdByRunId.delete(runId);
-        this.directChildCountByRunId.delete(runId);
-    }
-}
-exports.RunBatchSpanProcessor = RunBatchSpanProcessor;
-//# sourceMappingURL=run-batch-span-processor.js.map

package/dist/run-batch-span-processor.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"run-batch-span-processor.js","sourceRoot":"","sources":["../src/run-batch-span-processor.ts"],"names":[],"mappings":";;;AAAA,mCAAoC;AAQpC,6CAA0C;AAK1C,MAAa,qBAAqB;IACxB,UAAU,GAAG,KAAK,CAAC;IACnB,aAAa,GAAG,IAAI,GAAG,EAAiB,CAAC;IACzC,qBAAqB,GAAG,IAAI,GAAG,EAAiB,CAAC;IACjD,uBAAuB,GAAG,IAAI,GAAG,EAAiB,CAAC;IACnD,wBAAwB,GAAG,IAAI,GAAG,EAAiB,CAAC;IACpD,OAAO,GAAG,IAAI,GAAG,EAAyB,CAAC;IAC3C,SAAS,GAAG,IAAI,GAAG,EAAS,CAAC;IACpB,QAAQ,CAAe;IAExC,YAAY,QAAsB;QAChC,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC3B,CAAC;IAED,OAAO,CAAC,IAAU,EAAE,cAAuB;QACzC,MAAM,MAAM,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC,MAAM,CAAC;QAEzC,IAAI,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,EAAE,CAAC;YAC7B,MAAM,KAAK,GAAG,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,IAAI,IAAA,mBAAU,GAAE,CAAC;YAC1D,IAAI,CAAC,YAAY,CAAC,cAAc,EAAE,KAAK,CAAC,CAAC;YACzC,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;YACtC,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;YAC9C,IAAI,CAAC,uBAAuB,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;YAC3C,IAAA,uBAAU,EAAC,WAAW,EAAE,6BAA6B,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;YAC1E,OAAO;QACT,CAAC;QAED,MAAM,YAAY,GAAG,IAAI,CAAC,iBAAiB,EAAE,MAAM,CAAC;QACpD,IAAI,CAAC,YAAY;YAAE,OAAO;QAE1B,MAAM,KAAK,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QACnD,IAAI,CAAC,KAAK;YAAE,OAAO;QAEnB,IAAI,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,YAAY,EAAE,CAAC;YAC3D,IAAI,CAAC,wBAAwB,CAAC,GAAG,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;YACjD,IAAI,CAAC,uBAAuB,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC,IAAI,CAAC,uBAAuB,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;QAC9F,CAAC;QAED,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;QACtC,IAAI,CAAC,YAAY,CAAC,cAAc,EAAE,KAAK,CAAC,CAAC;QACzC,IAAA,uBAAU,EAAC,WAAW,EAAE,qBAAqB,EAAE,EAAE,QAAQ,EAAE,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;IACzF,CAAC;IAED,KAAK,CAAC,IAAkB;QACtB,MAAM,MAAM,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC,MAAM,CAAC;QACzC,MAAM,KAAK,GACT,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,MAAM,CAAC;YAC9B,CAAC,IAAI,CAAC,iBAAiB,EAAE,MAAM;gBAC7B,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC;gBACvD,CAAC,CAAC,SAAS,CAAC,CAAC;QAEjB,IAAI,CAAC,KAAK;YAAE,OAAO;QAEnB,MAAM,aAAa,GAAG,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;QAC/C,MAAM,gBAAgB,GAAG,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;QACrD,MAAM,gBAAgB,GAAG,IAAI,CAAC,wBAAwB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACnE,IAAI,gBAAgB,EAAE,CAAC;YACrB,IAAI,CAAC,wBAAwB,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YAC7C,MAAM,YAAY,GAAG,IAAI,CAAC,uBAAuB,CAAC,GAAG,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;YAC7E,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,YAAY,GAAG,CAAC,CAAC,CAAC;YAChD,IAAI,CAAC,uBAAuB,CAAC,GAAG,CAAC,gBAAgB,EAAE,SAAS,CAAC,CAAC;YAC9D,IAAA,uBAAU,EAAC,WAAW,EAAE,2BAA2B,EAAE,EAAE,QAAQ,EAAE,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,gBAAgB,EAAE,iBAAiB,EAAE,SAAS,EAAE,CAAC,CAAC;QAC/I,CAAC;QAED,IAAA,uBAAU,EAAC,WAAW,EAAE,mBAAmB,EAAE,EAAE,QAAQ,EAAE,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,aAAa,EAAE,OAAO,EAAE,gBAAgB,EAAE,CAAC,CAAC;QAE/H,IAAI,CAAC,gBAAgB,EAAE,CAAC;YACtB,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YACtC,IAAI,KAAK;gBAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;;gBACvB,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC;QACvC,CAAC;QAED,IAAI,aAAa,EAAE,CAAC;YAClB,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAC5B,CAAC;QAED,KAAK,IAAI,CAAC,cAAc,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;IACzC,CAAC;IAED,KAAK,CAAC,UAAU;QACd,IAAA,uBAAU,EAAC,WAAW,EAAE,mBAAmB,CAAC,CAAC;QAC7C,MAAM,MAAM,GAAG,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;QACxC,MAAM,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC;QAC3E,IAAI,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,CAAC;YAC7B,MAAM,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,CAAC;QACnC,CAAC;IACH,CAAC;IAED,KAAK,CAAC,QAAQ;QACZ,IAAI,IAAI,CAAC,UAAU;YAAE,OAAO;QAE5B,IAAA,uBAAU,EAAC,WAAW,EAAE,iBAAiB,CAAC,CAAC;QAC3C,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;QACvB,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QACxB,IAAI,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC;IAC3B,CAAC;IAEO,aAAa,CAAC,IAAyB;QAC7C,OAAO,IAAI,CAAC,IAAI,KAAK,cAAc,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC;IACjE,CAAC;IAEO,gBAAgB,CAAC,IAAU;QACjC,MAAM,UAAU,GAAI,IAA4D;aAC7E,UAAU,CAAC;QACd,MAAM,KAAK,GAAG,UAAU,EAAE,CAAC,cAAc,CAAC,CAAC;QAC3C,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC;IAC3E,CAAC;IAEO,2BAA2B,CAAC,IAAyB;QAC3D,OACE,IAGD,CAAC,oBAAoB,EAAE,IAAI,CAAC;IAC/B,CAAC;IAEO,gBAAgB,CAAC,IAAyB;QAChD,OAAO,IAAI,CAAC,2BAA2B,CAAC,IAAI,CAAC,KAAK,SAAS,CAAC;IAC9D,CAAC;IAEO,KAAK,CAAC,cAAc,CAAC,KAAY,EAAE,KAAc;QACvD,IAAI,CAAC,KAAK,IAAI,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,uBAAuB,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC;YACnF,OAAO;QACT,CAAC;QAED,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAEtC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC3B,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC7B,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;QAE5B,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QAEzC,IAAA,uBAAU,EAAC,WAAW,EAAE,iBAAiB,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,CAAC,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;QACtF,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE;YAClC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,KAAK,EAAE,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,uBAAuB,CAAC,KAAY;QAC1C,OAAO,CAAC,IAAI,CAAC,uBAAuB,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC;IAC9D,CAAC;IAEO,eAAe,CAAC,KAAY;QAClC,KAAK,MAAM,CAAC,MAAM,EAAE,WAAW,CAAC,IAAI,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,EAAE,CAAC;YACjE,IAAI,WAAW,KAAK,KAAK,EAAE,CAAC;gBAC1B,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YACpC,CAAC;QACH,CAAC;QACD,KAAK,MAAM,CAAC,MAAM,EAAE,WAAW,CAAC,IAAI,IAAI,CAAC,wBAAwB,CAAC,OAAO,EAAE,EAAE,CAAC;YAC5E,IAAI,WAAW,KAAK,KAAK,EAAE,CAAC;gBAC1B,IAAI,CAAC,wBAAwB,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YAC/C,CAAC;QACH,CAAC;QACD,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACzC,IAAI,CAAC,uBAAuB,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAC7C,CAAC;CACF;AA7JD,sDA6JC"}

package/dist/span-helpers.d.ts

@@ -1,50 +0,0 @@
-/**
- * Wraps a function with a child span under the current active context.
- *
- * Use this for any internal function you want to appear in the trace — tool
- * implementations, retrieval calls, preprocessing steps, etc.
- *
- * @param name - Span name shown in the Lemma trace view.
- * @param fn   - Function to wrap. Receives the same input as the returned wrapper.
- *
- * @example
- * const formatOutput = trace("format-output", async (raw: string) => {
- *   return raw.trim().toLowerCase();
- * });
- *
- * // Inside an agent() handler:
- * const formatted = await formatOutput(rawText);
- */
-export declare function trace<Input = unknown, Output = unknown>(name: string, fn: (input: Input) => Output | Promise<Output>): (input: Input) => Promise<Output>;
-/**
- * Wraps a tool function with a child span. Sets `span.type = "tool"`.
- *
- * @example
- * const lookupOrder = tool("lookup-order", async (orderId: string) => {
- *   return db.orders.findById(orderId);
- * });
- */
-export declare function tool<Input = unknown, Output = unknown>(name: string, fn: (input: Input) => Output | Promise<Output>): (input: Input) => Promise<Output>;
-/**
- * Wraps an LLM call with a child span. Sets `span.type = "generation"`.
- *
- * Prefer provider instrumentation (OpenInference) for automatic LLM spans
- * with prompt/completion/token attributes. Use this helper for custom or
- * lightly-instrumented models.
- *
- * @example
- * const generate = llm("gpt-4o", async (prompt: string) => {
- *   return openai.chat.completions.create({ ... });
- * });
- */
-export declare function llm<Input = unknown, Output = unknown>(name: string, fn: (input: Input) => Output | Promise<Output>): (input: Input) => Promise<Output>;
-/**
- * Wraps a retrieval function with a child span. Sets `span.type = "retriever"`.
- *
- * @example
- * const search = retrieval("vector-search", async (query: string) => {
- *   return vectorDB.search(query, { topK: 5 });
- * });
- */
-export declare function retrieval<Input = unknown, Output = unknown>(name: string, fn: (input: Input) => Output | Promise<Output>): (input: Input) => Promise<Output>;
-//# sourceMappingURL=span-helpers.d.ts.map

package/dist/span-helpers.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"span-helpers.d.ts","sourceRoot":"","sources":["../src/span-helpers.ts"],"names":[],"mappings":"AAqCA;;;;;;;;;;;;;;;;GAgBG;AACH,wBAAgB,KAAK,CAAC,KAAK,GAAG,OAAO,EAAE,MAAM,GAAG,OAAO,EACrD,IAAI,EAAE,MAAM,EACZ,EAAE,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,GAC7C,CAAC,KAAK,EAAE,KAAK,KAAK,OAAO,CAAC,MAAM,CAAC,CAEnC;AAED;;;;;;;GAOG;AACH,wBAAgB,IAAI,CAAC,KAAK,GAAG,OAAO,EAAE,MAAM,GAAG,OAAO,EACpD,IAAI,EAAE,MAAM,EACZ,EAAE,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,GAC7C,CAAC,KAAK,EAAE,KAAK,KAAK,OAAO,CAAC,MAAM,CAAC,CAEnC;AAED;;;;;;;;;;;GAWG;AACH,wBAAgB,GAAG,CAAC,KAAK,GAAG,OAAO,EAAE,MAAM,GAAG,OAAO,EACnD,IAAI,EAAE,MAAM,EACZ,EAAE,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,GAC7C,CAAC,KAAK,EAAE,KAAK,KAAK,OAAO,CAAC,MAAM,CAAC,CAEnC;AAED;;;;;;;GAOG;AACH,wBAAgB,SAAS,CAAC,KAAK,GAAG,OAAO,EAAE,MAAM,GAAG,OAAO,EACzD,IAAI,EAAE,MAAM,EACZ,EAAE,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,GAC7C,CAAC,KAAK,EAAE,KAAK,KAAK,OAAO,CAAC,MAAM,CAAC,CAEnC"}

package/dist/span-helpers.js

@@ -1,95 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.trace = trace;
-exports.tool = tool;
-exports.llm = llm;
-exports.retrieval = retrieval;
-const api_1 = require("@opentelemetry/api");
-/**
- * Core helper: wraps `fn` so every call runs inside a child span named
- * `spanName` under the currently active context. Unlike `wrapAgent`, this
- * does NOT create a new trace root — it adds a child span to whatever span
- * is currently active.
- *
- * The span ends when the function returns (or throws). Return values are
- * passed through unchanged.
- */
-function spanHelper(spanName, fn, spanType) {
-    return async function (input) {
-        const tracer = api_1.trace.getTracer("lemma");
-        return tracer.startActiveSpan(spanName, async (span) => {
-            if (spanType) {
-                span.setAttribute("span.type", spanType);
-            }
-            try {
-                const result = await fn(input);
-                span.end();
-                return result;
-            }
-            catch (err) {
-                span.recordException(err);
-                span.setStatus({ code: 2 }); // SpanStatusCode.ERROR
-                span.end();
-                throw err;
-            }
-        });
-    };
-}
-/**
- * Wraps a function with a child span under the current active context.
- *
- * Use this for any internal function you want to appear in the trace — tool
- * implementations, retrieval calls, preprocessing steps, etc.
- *
- * @param name - Span name shown in the Lemma trace view.
- * @param fn   - Function to wrap. Receives the same input as the returned wrapper.
- *
- * @example
- * const formatOutput = trace("format-output", async (raw: string) => {
- *   return raw.trim().toLowerCase();
- * });
- *
- * // Inside an agent() handler:
- * const formatted = await formatOutput(rawText);
- */
-function trace(name, fn) {
-    return spanHelper(name, fn);
-}
-/**
- * Wraps a tool function with a child span. Sets `span.type = "tool"`.
- *
- * @example
- * const lookupOrder = tool("lookup-order", async (orderId: string) => {
- *   return db.orders.findById(orderId);
- * });
- */
-function tool(name, fn) {
-    return spanHelper(name, fn, "tool");
-}
-/**
- * Wraps an LLM call with a child span. Sets `span.type = "generation"`.
- *
- * Prefer provider instrumentation (OpenInference) for automatic LLM spans
- * with prompt/completion/token attributes. Use this helper for custom or
- * lightly-instrumented models.
- *
- * @example
- * const generate = llm("gpt-4o", async (prompt: string) => {
- *   return openai.chat.completions.create({ ... });
- * });
- */
-function llm(name, fn) {
-    return spanHelper(name, fn, "generation");
-}
-/**
- * Wraps a retrieval function with a child span. Sets `span.type = "retriever"`.
- *
- * @example
- * const search = retrieval("vector-search", async (query: string) => {
- *   return vectorDB.search(query, { topK: 5 });
- * });
- */
-function retrieval(name, fn) {
-    return spanHelper(name, fn, "retriever");
-}
-//# sourceMappingURL=span-helpers.js.map

package/dist/span-helpers.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"span-helpers.js","sourceRoot":"","sources":["../src/span-helpers.ts"],"names":[],"mappings":";;AAsDA,sBAKC;AAUD,oBAKC;AAcD,kBAKC;AAUD,8BAKC;AA5GD,4CAAwD;AAGxD;;;;;;;;GAQG;AACH,SAAS,UAAU,CACjB,QAAgB,EAChB,EAA8C,EAC9C,QAAiB;IAEjB,OAAO,KAAK,WAAW,KAAY;QACjC,MAAM,MAAM,GAAG,WAAS,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QAC5C,OAAO,MAAM,CAAC,eAAe,CAAC,QAAQ,EAAE,KAAK,EAAE,IAAU,EAAE,EAAE;YAC3D,IAAI,QAAQ,EAAE,CAAC;gBACb,IAAI,CAAC,YAAY,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;YAC3C,CAAC;YACD,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC;gBAC/B,IAAI,CAAC,GAAG,EAAE,CAAC;gBACX,OAAO,MAAM,CAAC;YAChB,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,IAAI,CAAC,eAAe,CAAC,GAAY,CAAC,CAAC;gBACnC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,uBAAuB;gBACpD,IAAI,CAAC,GAAG,EAAE,CAAC;gBACX,MAAM,GAAG,CAAC;YACZ,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC,CAAC;AACJ,CAAC;AAED;;;;;;;;;;;;;;;;GAgBG;AACH,SAAgB,KAAK,CACnB,IAAY,EACZ,EAA8C;IAE9C,OAAO,UAAU,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;AAC9B,CAAC;AAED;;;;;;;GAOG;AACH,SAAgB,IAAI,CAClB,IAAY,EACZ,EAA8C;IAE9C,OAAO,UAAU,CAAC,IAAI,EAAE,EAAE,EAAE,MAAM,CAAC,CAAC;AACtC,CAAC;AAED;;;;;;;;;;;GAWG;AACH,SAAgB,GAAG,CACjB,IAAY,EACZ,EAA8C;IAE9C,OAAO,UAAU,CAAC,IAAI,EAAE,EAAE,EAAE,YAAY,CAAC,CAAC;AAC5C,CAAC;AAED;;;;;;;GAOG;AACH,SAAgB,SAAS,CACvB,IAAY,EACZ,EAA8C;IAE9C,OAAO,UAAU,CAAC,IAAI,EAAE,EAAE,EAAE,WAAW,CAAC,CAAC;AAC3C,CAAC"}