@uselemma/tracing 3.0.3 → 4.0.1

package/dist/vercel-ai.js

@@ -0,0 +1,392 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.vercelAI = vercelAI;
+const client_1 = require("./client");
+function addMs(startedAt, durationMs) {
+    return typeof durationMs === "number"
+        ? new Date(startedAt.getTime() + durationMs)
+        : new Date();
+}
+function subtractMs(endedAt, durationMs) {
+    return typeof durationMs === "number"
+        ? new Date(endedAt.getTime() - durationMs)
+        : endedAt;
+}
+function v7StepKey(callId, stepNumber) {
+    return `${callId}:${stepNumber}`;
+}
+function isV7StepStart(event) {
+    return "callId" in event && "provider" in event && "modelId" in event;
+}
+function getTrace(trace) {
+    if (trace)
+        return trace;
+    try {
+        return (0, client_1.active)();
+    }
+    catch {
+        console.warn("@uselemma/tracing: vercelAI() telemetry received an event outside lemma.trace(); event ignored");
+        return undefined;
+    }
+}
+function stringifyContent(content) {
+    const text = content
+        .map((part) => {
+        if (!part || typeof part !== "object")
+            return "";
+        if (part.type !== "text")
+            return "";
+        return String(part.text ?? "");
+    })
+        .join("");
+    if (text)
+        return text;
+    return JSON.stringify(content);
+}
+function toolOutput(event, recordOutputs) {
+    if (!recordOutputs) {
+        return event.toolOutput.type === "tool-error"
+            ? { status: "ERROR" }
+            : {};
+    }
+    if (event.toolOutput.type === "tool-result") {
+        return { output: event.toolOutput.output };
+    }
+    return {
+        error: event.toolOutput.error,
+        status: "ERROR",
+    };
+}
+function v6ToolOutput(event, recordOutputs) {
+    if (!recordOutputs) {
+        return event.success ? {} : { status: "ERROR" };
+    }
+    if (event.success) {
+        return { output: event.output };
+    }
+    return {
+        error: event.error,
+        status: "ERROR",
+    };
+}
+function v6Input(event) {
+    if ("messages" in event && event.messages)
+        return event.messages;
+    return "prompt" in event ? event.prompt : undefined;
+}
+function v6InputMessages(event) {
+    const input = v6Input(event);
+    return Array.isArray(input) ? input : undefined;
+}
+function v6Output(event) {
+    if (typeof event.text === "string")
+        return event.text;
+    if (event.content)
+        return stringifyContent(event.content);
+    return undefined;
+}
+function vercelAI(options = {}) {
+    const modelCalls = new Map();
+    const v7Steps = new Map();
+    const v6Steps = new Map();
+    const v6Starts = [];
+    const generationSpanIdsByCallId = new Map();
+    const generationSpanIdsByToolCallId = new Map();
+    const toolExecutions = new Map();
+    let recordedV6Step = false;
+    let endedExplicitTrace = false;
+    function endOutput(event) {
+        if (typeof event.text === "string")
+            return event.text;
+        if (event.content)
+            return stringifyContent(event.content);
+        return undefined;
+    }
+    async function endExplicitTrace(event) {
+        if (endedExplicitTrace)
+            return;
+        const trace = options.trace;
+        if (typeof trace?.end !== "function")
+            return;
+        endedExplicitTrace = true;
+        const output = endOutput(event);
+        if (options.recordOutputs === false || output === undefined) {
+            await trace.end();
+            return;
+        }
+        await trace.end({ output });
+    }
+    function recordV6Generation(event, stored) {
+        const trace = getTrace(options.trace);
+        if (!trace)
+            return;
+        const startedAt = stored?.startedAt ?? new Date();
+        const endedAt = new Date();
+        const id = crypto.randomUUID();
+        const name = typeof options.generationName === "function"
+            ? options.generationName(event)
+            : (options.generationName ?? "vercel-ai-generation");
+        const usage = "totalUsage" in event ? event.totalUsage : event.usage;
+        const output = v6Output(event);
+        trace.recordGeneration({
+            id,
+            name,
+            input: options.recordInputs === false
+                ? undefined
+                : stored?.event && v6Input(stored.event),
+            output: options.recordOutputs === false ? undefined : output,
+            metadata: options.metadata,
+            model: event.model.modelId,
+            usage: {
+                inputTokens: usage?.inputTokens,
+                outputTokens: usage?.outputTokens,
+            },
+            startedAt,
+            endedAt,
+            llmProvider: event.model.provider,
+            llmInputMessages: options.recordInputs === false || !stored?.event
+                ? undefined
+                : v6InputMessages(stored.event),
+            llmOutputMessages: options.recordOutputs === false || output === undefined
+                ? undefined
+                : [{ role: "assistant", content: output }],
+            llmTools: stored?.event.tools,
+        });
+        generationSpanIdsByCallId.set("v6-latest", id);
+    }
+    function startV7Generation(event) {
+        const trace = getTrace(options.trace);
+        if (!trace)
+            return;
+        const name = typeof options.generationName === "function"
+            ? options.generationName({
+                callId: event.callId,
+                provider: event.provider,
+                modelId: event.modelId,
+                usage: {},
+                content: [],
+                performance: {},
+            })
+            : (options.generationName ?? "vercel-ai-generation");
+        const startedAt = new Date();
+        const handle = trace.startGeneration({
+            name,
+            input: options.recordInputs === false ? undefined : event.messages,
+            metadata: options.metadata,
+            model: event.modelId,
+            startedAt,
+            llmProvider: event.provider,
+            llmInputMessages: options.recordInputs === false ? undefined : event.messages,
+            llmTools: event.tools,
+        });
+        const stored = { event, startedAt, handle };
+        v7Steps.set(v7StepKey(event.callId, event.stepNumber), stored);
+        generationSpanIdsByCallId.set(event.callId, handle.id);
+    }
+    function endV7Generation(event) {
+        const stored = v7Steps.get(v7StepKey(event.callId, event.stepNumber));
+        v7Steps.delete(v7StepKey(event.callId, event.stepNumber));
+        if (!stored)
+            return;
+        const durationMs = event.performance?.stepTimeMs ?? event.performance?.responseTimeMs;
+        const output = typeof event.text === "string"
+            ? event.text
+            : event.content
+                ? stringifyContent(event.content)
+                : undefined;
+        for (const toolCall of event.toolCalls ?? []) {
+            if (toolCall.toolCallId) {
+                generationSpanIdsByToolCallId.set(toolCall.toolCallId, stored.handle.id);
+            }
+        }
+        stored.handle.end({
+            output: options.recordOutputs === false ? undefined : output,
+            usage: {
+                inputTokens: event.usage?.inputTokens,
+                outputTokens: event.usage?.outputTokens,
+            },
+            model: event.model.modelId,
+            durationMs,
+            endedAt: addMs(stored.startedAt, durationMs),
+            llmProvider: event.model.provider,
+            llmOutputMessages: options.recordOutputs === false || output === undefined
+                ? undefined
+                : [{ role: "assistant", content: output }],
+        });
+    }
+    return {
+        onLanguageModelCallStart(event) {
+            const trace = getTrace(options.trace);
+            if (!trace)
+                return;
+            if (generationSpanIdsByCallId.has(event.callId)) {
+                modelCalls.set(event.callId, { event, startedAt: new Date() });
+                return;
+            }
+            const startedAt = new Date();
+            const name = typeof options.generationName === "function"
+                ? options.generationName({
+                    callId: event.callId,
+                    provider: event.provider,
+                    modelId: event.modelId,
+                    usage: {},
+                    content: [],
+                    performance: {},
+                })
+                : (options.generationName ?? "vercel-ai-generation");
+            const handle = trace.startGeneration({
+                name,
+                input: options.recordInputs === false ? undefined : event.messages,
+                metadata: options.metadata,
+                model: event.modelId,
+                startedAt,
+                llmProvider: event.provider,
+                llmInputMessages: options.recordInputs === false ? undefined : event.messages,
+                llmTools: event.tools,
+            });
+            modelCalls.set(event.callId, { event, startedAt, handle });
+            generationSpanIdsByCallId.set(event.callId, handle.id);
+        },
+        onLanguageModelCallEnd(event) {
+            const stored = modelCalls.get(event.callId);
+            modelCalls.delete(event.callId);
+            if (!stored?.handle)
+                return;
+            stored.handle.end({
+                output: options.recordOutputs === false
+                    ? undefined
+                    : stringifyContent(event.content),
+                usage: {
+                    inputTokens: event.usage.inputTokens,
+                    outputTokens: event.usage.outputTokens,
+                },
+                model: event.modelId,
+                durationMs: event.performance.responseTimeMs,
+                endedAt: addMs(stored.startedAt, event.performance.responseTimeMs),
+                llmProvider: event.provider,
+                llmOutputMessages: options.recordOutputs === false
+                    ? undefined
+                    : [{ role: "assistant", content: stringifyContent(event.content) }],
+            });
+        },
+        onToolExecutionStart(event) {
+            const trace = getTrace(options.trace);
+            if (!trace)
+                return;
+            const parentId = event.toolCall.toolCallId &&
+                generationSpanIdsByToolCallId.get(event.toolCall.toolCallId);
+            const fallbackParentId = event.callId
+                ? generationSpanIdsByCallId.get(event.callId)
+                : undefined;
+            const name = typeof options.toolName === "function"
+                ? options.toolName({
+                    ...event,
+                    toolExecutionMs: undefined,
+                    toolOutput: { type: "tool-result" },
+                })
+                : (options.toolName ?? event.toolCall.toolName);
+            const startedAt = new Date();
+            const handle = trace.startTool({
+                name,
+                parentId: parentId ?? fallbackParentId,
+                input: options.recordInputs === false ? undefined : event.toolCall.input,
+                metadata: options.metadata,
+                startedAt,
+            });
+            if (event.toolCall.toolCallId) {
+                toolExecutions.set(event.toolCall.toolCallId, { handle, startedAt });
+            }
+        },
+        onToolExecutionEnd(event) {
+            const trace = getTrace(options.trace);
+            if (!trace)
+                return;
+            const name = typeof options.toolName === "function"
+                ? options.toolName(event)
+                : (options.toolName ?? event.toolCall.toolName);
+            const storedTool = event.toolCall.toolCallId
+                ? toolExecutions.get(event.toolCall.toolCallId)
+                : undefined;
+            if (event.toolCall.toolCallId) {
+                toolExecutions.delete(event.toolCall.toolCallId);
+            }
+            if (storedTool) {
+                storedTool.handle.end({
+                    durationMs: event.toolExecutionMs,
+                    endedAt: addMs(storedTool.startedAt, event.toolExecutionMs),
+                    ...toolOutput(event, options.recordOutputs !== false),
+                });
+                return;
+            }
+            const endedAt = new Date();
+            const startedAt = subtractMs(endedAt, event.toolExecutionMs);
+            const parentId = event.toolCall.toolCallId &&
+                generationSpanIdsByToolCallId.get(event.toolCall.toolCallId);
+            const fallbackParentId = event.callId
+                ? generationSpanIdsByCallId.get(event.callId)
+                : undefined;
+            trace.recordTool({
+                name,
+                parentId: parentId ?? fallbackParentId,
+                input: options.recordInputs === false ? undefined : event.toolCall.input,
+                metadata: options.metadata,
+                durationMs: event.toolExecutionMs,
+                startedAt,
+                endedAt,
+                ...toolOutput(event, options.recordOutputs !== false),
+            });
+        },
+        onStart(event) {
+            recordedV6Step = false;
+            v6Starts.push({ event, startedAt: new Date() });
+        },
+        onStepStart(event) {
+            if (isV7StepStart(event)) {
+                startV7Generation(event);
+                return;
+            }
+            v6Steps.set(event.stepNumber, { event, startedAt: new Date() });
+        },
+        onStepEnd(event) {
+            endV7Generation(event);
+        },
+        onStepFinish(event) {
+            recordedV6Step = true;
+            const stored = v6Steps.get(event.stepNumber);
+            v6Steps.delete(event.stepNumber);
+            recordV6Generation(event, stored);
+        },
+        async onFinish(event) {
+            if (recordedV6Step) {
+                v6Starts.shift();
+                recordedV6Step = false;
+                await endExplicitTrace(event);
+                return;
+            }
+            recordV6Generation(event, v6Starts.shift());
+            await endExplicitTrace(event);
+        },
+        async onEnd(event) {
+            await endExplicitTrace(event);
+        },
+        onToolCallFinish(event) {
+            const trace = getTrace(options.trace);
+            if (!trace)
+                return;
+            const name = typeof options.toolName === "function"
+                ? options.toolName(event)
+                : (options.toolName ?? event.toolCall.toolName);
+            const parentId = event.toolCall.toolCallId &&
+                generationSpanIdsByToolCallId.get(event.toolCall.toolCallId);
+            trace.recordTool({
+                name,
+                parentId: parentId ?? generationSpanIdsByCallId.get("v6-latest"),
+                input: options.recordInputs === false ? undefined : event.toolCall.input,
+                metadata: options.metadata,
+                durationMs: event.durationMs,
+                ...v6ToolOutput(event, options.recordOutputs !== false),
+            });
+        },
+    };
+}
+//# sourceMappingURL=vercel-ai.js.map

package/dist/vercel-ai.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"vercel-ai.js","sourceRoot":"","sources":["../src/vercel-ai.ts"],"names":[],"mappings":";;AAgUA,4BA0WC;AA1qBD,qCAAsE;AA8NtE,SAAS,KAAK,CAAC,SAAe,EAAE,UAA8B;IAC5D,OAAO,OAAO,UAAU,KAAK,QAAQ;QACnC,CAAC,CAAC,IAAI,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,GAAG,UAAU,CAAC;QAC5C,CAAC,CAAC,IAAI,IAAI,EAAE,CAAC;AACjB,CAAC;AAED,SAAS,UAAU,CAAC,OAAa,EAAE,UAA8B;IAC/D,OAAO,OAAO,UAAU,KAAK,QAAQ;QACnC,CAAC,CAAC,IAAI,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,GAAG,UAAU,CAAC;QAC1C,CAAC,CAAC,OAAO,CAAC;AACd,CAAC;AAED,SAAS,SAAS,CAAC,MAAc,EAAE,UAAkB;IACnD,OAAO,GAAG,MAAM,IAAI,UAAU,EAAE,CAAC;AACnC,CAAC;AAED,SAAS,aAAa,CACpB,KAAwD;IAExD,OAAO,QAAQ,IAAI,KAAK,IAAI,UAAU,IAAI,KAAK,IAAI,SAAS,IAAI,KAAK,CAAC;AACxE,CAAC;AAED,SAAS,QAAQ,CAAC,KAA+B;IAC/C,IAAI,KAAK;QAAE,OAAO,KAAK,CAAC;IACxB,IAAI,CAAC;QACH,OAAO,IAAA,eAAM,GAAE,CAAC;IAClB,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,CAAC,IAAI,CACV,gGAAgG,CACjG,CAAC;QACF,OAAO,SAAS,CAAC;IACnB,CAAC;AACH,CAAC;AAED,SAAS,gBAAgB,CAAC,OAA+B;IACvD,MAAM,IAAI,GAAG,OAAO;SACjB,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;QACZ,IAAI,CAAC,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ;YAAE,OAAO,EAAE,CAAC;QACjD,IAAK,IAA2B,CAAC,IAAI,KAAK,MAAM;YAAE,OAAO,EAAE,CAAC;QAC5D,OAAO,MAAM,CAAE,IAA2B,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC;IACzD,CAAC,CAAC;SACD,IAAI,CAAC,EAAE,CAAC,CAAC;IACZ,IAAI,IAAI;QAAE,OAAO,IAAI,CAAC;IACtB,OAAO,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;AACjC,CAAC;AAED,SAAS,UAAU,CACjB,KAAoC,EACpC,aAAsB;IAEtB,IAAI,CAAC,aAAa,EAAE,CAAC;QACnB,OAAO,KAAK,CAAC,UAAU,CAAC,IAAI,KAAK,YAAY;YAC3C,CAAC,CAAC,EAAE,MAAM,EAAE,OAAgB,EAAE;YAC9B,CAAC,CAAC,EAAE,CAAC;IACT,CAAC;IACD,IAAI,KAAK,CAAC,UAAU,CAAC,IAAI,KAAK,aAAa,EAAE,CAAC;QAC5C,OAAO,EAAE,MAAM,EAAE,KAAK,CAAC,UAAU,CAAC,MAAM,EAAE,CAAC;IAC7C,CAAC;IACD,OAAO;QACL,KAAK,EAAE,KAAK,CAAC,UAAU,CAAC,KAAK;QAC7B,MAAM,EAAE,OAAgB;KACzB,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CACnB,KAAoC,EACpC,aAAsB;IAEtB,IAAI,CAAC,aAAa,EAAE,CAAC;QACnB,OAAO,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,OAAgB,EAAE,CAAC;IAC3D,CAAC;IACD,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;QAClB,OAAO,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE,CAAC;IAClC,CAAC;IACD,OAAO;QACL,KAAK,EAAE,KAAK,CAAC,KAAK;QAClB,MAAM,EAAE,OAAgB;KACzB,CAAC;AACJ,CAAC;AAED,SAAS,OAAO,CAAC,KAAsD;IACrE,IAAI,UAAU,IAAI,KAAK,IAAI,KAAK,CAAC,QAAQ;QAAE,OAAO,KAAK,CAAC,QAAQ,CAAC;IACjE,OAAO,QAAQ,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC;AACtD,CAAC;AAED,SAAS,eAAe,CACtB,KAAsD;IAEtD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC;IAC7B,OAAO,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC;AAClD,CAAC;AAED,SAAS,QAAQ,CAAC,KAAwD;IACxE,IAAI,OAAO,KAAK,CAAC,IAAI,KAAK,QAAQ;QAAE,OAAO,KAAK,CAAC,IAAI,CAAC;IACtD,IAAI,KAAK,CAAC,OAAO;QAAE,OAAO,gBAAgB,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IAC1D,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAgB,QAAQ,CACtB,UAAsC,EAAE;IAExC,MAAM,UAAU,GAAG,IAAI,GAAG,EAA2B,CAAC;IACtD,MAAM,OAAO,GAAG,IAAI,GAAG,EAAwB,CAAC;IAChD,MAAM,OAAO,GAAG,IAAI,GAAG,EAAwB,CAAC;IAChD,MAAM,QAAQ,GAAmB,EAAE,CAAC;IACpC,MAAM,yBAAyB,GAAG,IAAI,GAAG,EAAkB,CAAC;IAC5D,MAAM,6BAA6B,GAAG,IAAI,GAAG,EAAkB,CAAC;IAChE,MAAM,cAAc,GAAG,IAAI,GAAG,EAA+B,CAAC;IAC9D,IAAI,cAAc,GAAG,KAAK,CAAC;IAC3B,IAAI,kBAAkB,GAAG,KAAK,CAAC;IAE/B,SAAS,SAAS,CAAC,KAA+C;QAChE,IAAI,OAAO,KAAK,CAAC,IAAI,KAAK,QAAQ;YAAE,OAAO,KAAK,CAAC,IAAI,CAAC;QACtD,IAAI,KAAK,CAAC,OAAO;YAAE,OAAO,gBAAgB,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAC1D,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,KAAK,UAAU,gBAAgB,CAC7B,KAA+C;QAE/C,IAAI,kBAAkB;YAAE,OAAO;QAC/B,MAAM,KAAK,GAAG,OAAO,CAAC,KAIT,CAAC;QACd,IAAI,OAAO,KAAK,EAAE,GAAG,KAAK,UAAU;YAAE,OAAO;QAE7C,kBAAkB,GAAG,IAAI,CAAC;QAC1B,MAAM,MAAM,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC;QAChC,IAAI,OAAO,CAAC,aAAa,KAAK,KAAK,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;YAC5D,MAAM,KAAK,CAAC,GAAG,EAAE,CAAC;YAClB,OAAO;QACT,CAAC;QACD,MAAM,KAAK,CAAC,GAAG,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC;IAC9B,CAAC;IAED,SAAS,kBAAkB,CACzB,KAAwD,EACxD,MAAgC;QAEhC,MAAM,KAAK,GAAG,QAAQ,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACtC,IAAI,CAAC,KAAK;YAAE,OAAO;QAEnB,MAAM,SAAS,GAAG,MAAM,EAAE,SAAS,IAAI,IAAI,IAAI,EAAE,CAAC;QAClD,MAAM,OAAO,GAAG,IAAI,IAAI,EAAE,CAAC;QAC3B,MAAM,EAAE,GAAG,MAAM,CAAC,UAAU,EAAE,CAAC;QAC/B,MAAM,IAAI,GACR,OAAO,OAAO,CAAC,cAAc,KAAK,UAAU;YAC1C,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,KAAK,CAAC;YAC/B,CAAC,CAAC,CAAC,OAAO,CAAC,cAAc,IAAI,sBAAsB,CAAC,CAAC;QACzD,MAAM,KAAK,GAAG,YAAY,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC;QACrE,MAAM,MAAM,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;QAE/B,KAAK,CAAC,gBAAgB,CAAC;YACrB,EAAE;YACF,IAAI;YACJ,KAAK,EACH,OAAO,CAAC,YAAY,KAAK,KAAK;gBAC5B,CAAC,CAAC,SAAS;gBACX,CAAC,CAAC,MAAM,EAAE,KAAK,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC;YAC5C,MAAM,EAAE,OAAO,CAAC,aAAa,KAAK,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM;YAC5D,QAAQ,EAAE,OAAO,CAAC,QAAQ;YAC1B,KAAK,EAAE,KAAK,CAAC,KAAK,CAAC,OAAO;YAC1B,KAAK,EAAE;gBACL,WAAW,EAAE,KAAK,EAAE,WAAW;gBAC/B,YAAY,EAAE,KAAK,EAAE,YAAY;aAClC;YACD,SAAS;YACT,OAAO;YACP,WAAW,EAAE,KAAK,CAAC,KAAK,CAAC,QAAQ;YACjC,gBAAgB,EACd,OAAO,CAAC,YAAY,KAAK,KAAK,IAAI,CAAC,MAAM,EAAE,KAAK;gBAC9C,CAAC,CAAC,SAAS;gBACX,CAAC,CAAC,eAAe,CAAC,MAAM,CAAC,KAAK,CAAC;YACnC,iBAAiB,EACf,OAAO,CAAC,aAAa,KAAK,KAAK,IAAI,MAAM,KAAK,SAAS;gBACrD,CAAC,CAAC,SAAS;gBACX,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;YAC9C,QAAQ,EAAE,MAAM,EAAE,KAAK,CAAC,KAAK;SAC9B,CAAC,CAAC;QACH,yBAAyB,CAAC,GAAG,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;IACjD,CAAC;IAED,SAAS,iBAAiB,CAAC,KAA6B;QACtD,MAAM,KAAK,GAAG,QAAQ,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACtC,IAAI,CAAC,KAAK;YAAE,OAAO;QAEnB,MAAM,IAAI,GACR,OAAO,OAAO,CAAC,cAAc,KAAK,UAAU;YAC1C,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC;gBACrB,MAAM,EAAE,KAAK,CAAC,MAAM;gBACpB,QAAQ,EAAE,KAAK,CAAC,QAAQ;gBACxB,OAAO,EAAE,KAAK,CAAC,OAAO;gBACtB,KAAK,EAAE,EAAE;gBACT,OAAO,EAAE,EAAE;gBACX,WAAW,EAAE,EAAE;aAChB,CAAC;YACJ,CAAC,CAAC,CAAC,OAAO,CAAC,cAAc,IAAI,sBAAsB,CAAC,CAAC;QACzD,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC;QAC7B,MAAM,MAAM,GAAG,KAAK,CAAC,eAAe,CAAC;YACnC,IAAI;YACJ,KAAK,EAAE,OAAO,CAAC,YAAY,KAAK,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,QAAQ;YAClE,QAAQ,EAAE,OAAO,CAAC,QAAQ;YAC1B,KAAK,EAAE,KAAK,CAAC,OAAO;YACpB,SAAS;YACT,WAAW,EAAE,KAAK,CAAC,QAAQ;YAC3B,gBAAgB,EACd,OAAO,CAAC,YAAY,KAAK,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,QAAQ;YAC7D,QAAQ,EAAE,KAAK,CAAC,KAAK;SACtB,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,EAAE,KAAK,EAAE,SAAS,EAAE,MAAM,EAAE,CAAC;QAC5C,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,KAAK,CAAC,MAAM,EAAE,KAAK,CAAC,UAAU,CAAC,EAAE,MAAM,CAAC,CAAC;QAC/D,yBAAyB,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,EAAE,MAAM,CAAC,EAAE,CAAC,CAAC;IACzD,CAAC;IAED,SAAS,eAAe,CAAC,KAA2B;QAClD,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,KAAK,CAAC,MAAM,EAAE,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC;QACtE,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,MAAM,EAAE,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC;QAC1D,IAAI,CAAC,MAAM;YAAE,OAAO;QAEpB,MAAM,UAAU,GACd,KAAK,CAAC,WAAW,EAAE,UAAU,IAAI,KAAK,CAAC,WAAW,EAAE,cAAc,CAAC;QACrE,MAAM,MAAM,GACV,OAAO,KAAK,CAAC,IAAI,KAAK,QAAQ;YAC5B,CAAC,CAAC,KAAK,CAAC,IAAI;YACZ,CAAC,CAAC,KAAK,CAAC,OAAO;gBACb,CAAC,CAAC,gBAAgB,CAAC,KAAK,CAAC,OAAO,CAAC;gBACjC,CAAC,CAAC,SAAS,CAAC;QAElB,KAAK,MAAM,QAAQ,IAAI,KAAK,CAAC,SAAS,IAAI,EAAE,EAAE,CAAC;YAC7C,IAAI,QAAQ,CAAC,UAAU,EAAE,CAAC;gBACxB,6BAA6B,CAAC,GAAG,CAC/B,QAAQ,CAAC,UAAU,EACnB,MAAM,CAAC,MAAM,CAAC,EAAE,CACjB,CAAC;YACJ,CAAC;QACH,CAAC;QAED,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC;YAChB,MAAM,EAAE,OAAO,CAAC,aAAa,KAAK,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM;YAC5D,KAAK,EAAE;gBACL,WAAW,EAAE,KAAK,CAAC,KAAK,EAAE,WAAW;gBACrC,YAAY,EAAE,KAAK,CAAC,KAAK,EAAE,YAAY;aACxC;YACD,KAAK,EAAE,KAAK,CAAC,KAAK,CAAC,OAAO;YAC1B,UAAU;YACV,OAAO,EAAE,KAAK,CAAC,MAAM,CAAC,SAAS,EAAE,UAAU,CAAC;YAC5C,WAAW,EAAE,KAAK,CAAC,KAAK,CAAC,QAAQ;YACjC,iBAAiB,EACf,OAAO,CAAC,aAAa,KAAK,KAAK,IAAI,MAAM,KAAK,SAAS;gBACrD,CAAC,CAAC,SAAS;gBACX,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;SAC/C,CAAC,CAAC;IACL,CAAC;IAED,OAAO;QACL,wBAAwB,CAAC,KAAK;YAC5B,MAAM,KAAK,GAAG,QAAQ,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YACtC,IAAI,CAAC,KAAK;gBAAE,OAAO;YACnB,IAAI,yBAAyB,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC;gBAChD,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,IAAI,EAAE,EAAE,CAAC,CAAC;gBAC/D,OAAO;YACT,CAAC;YAED,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC;YAC7B,MAAM,IAAI,GACR,OAAO,OAAO,CAAC,cAAc,KAAK,UAAU;gBAC1C,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC;oBACrB,MAAM,EAAE,KAAK,CAAC,MAAM;oBACpB,QAAQ,EAAE,KAAK,CAAC,QAAQ;oBACxB,OAAO,EAAE,KAAK,CAAC,OAAO;oBACtB,KAAK,EAAE,EAAE;oBACT,OAAO,EAAE,EAAE;oBACX,WAAW,EAAE,EAAE;iBAChB,CAAC;gBACJ,CAAC,CAAC,CAAC,OAAO,CAAC,cAAc,IAAI,sBAAsB,CAAC,CAAC;YACzD,MAAM,MAAM,GAAG,KAAK,CAAC,eAAe,CAAC;gBACnC,IAAI;gBACJ,KAAK,EAAE,OAAO,CAAC,YAAY,KAAK,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,QAAQ;gBAClE,QAAQ,EAAE,OAAO,CAAC,QAAQ;gBAC1B,KAAK,EAAE,KAAK,CAAC,OAAO;gBACpB,SAAS;gBACT,WAAW,EAAE,KAAK,CAAC,QAAQ;gBAC3B,gBAAgB,EACd,OAAO,CAAC,YAAY,KAAK,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,QAAQ;gBAC7D,QAAQ,EAAE,KAAK,CAAC,KAAK;aACtB,CAAC,CAAC;YAEH,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,MAAM,EAAE,CAAC,CAAC;YAC3D,yBAAyB,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,EAAE,MAAM,CAAC,EAAE,CAAC,CAAC;QACzD,CAAC;QAED,sBAAsB,CAAC,KAAK;YAC1B,MAAM,MAAM,GAAG,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAC5C,UAAU,CAAC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAChC,IAAI,CAAC,MAAM,EAAE,MAAM;gBAAE,OAAO;YAE5B,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC;gBAChB,MAAM,EACJ,OAAO,CAAC,aAAa,KAAK,KAAK;oBAC7B,CAAC,CAAC,SAAS;oBACX,CAAC,CAAC,gBAAgB,CAAC,KAAK,CAAC,OAAO,CAAC;gBACrC,KAAK,EAAE;oBACL,WAAW,EAAE,KAAK,CAAC,KAAK,CAAC,WAAW;oBACpC,YAAY,EAAE,KAAK,CAAC,KAAK,CAAC,YAAY;iBACvC;gBACD,KAAK,EAAE,KAAK,CAAC,OAAO;gBACpB,UAAU,EAAE,KAAK,CAAC,WAAW,CAAC,cAAc;gBAC5C,OAAO,EAAE,KAAK,CAAC,MAAM,CAAC,SAAS,EAAE,KAAK,CAAC,WAAW,CAAC,cAAc,CAAC;gBAClE,WAAW,EAAE,KAAK,CAAC,QAAQ;gBAC3B,iBAAiB,EACf,OAAO,CAAC,aAAa,KAAK,KAAK;oBAC7B,CAAC,CAAC,SAAS;oBACX,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,gBAAgB,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC;aACxE,CAAC,CAAC;QACL,CAAC;QAED,oBAAoB,CAAC,KAAK;YACxB,MAAM,KAAK,GAAG,QAAQ,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YACtC,IAAI,CAAC,KAAK;gBAAE,OAAO;YAEnB,MAAM,QAAQ,GACZ,KAAK,CAAC,QAAQ,CAAC,UAAU;gBACzB,6BAA6B,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;YAC/D,MAAM,gBAAgB,GAAG,KAAK,CAAC,MAAM;gBACnC,CAAC,CAAC,yBAAyB,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC;gBAC7C,CAAC,CAAC,SAAS,CAAC;YACd,MAAM,IAAI,GACR,OAAO,OAAO,CAAC,QAAQ,KAAK,UAAU;gBACpC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC;oBACf,GAAG,KAAK;oBACR,eAAe,EAAE,SAAS;oBAC1B,UAAU,EAAE,EAAE,IAAI,EAAE,aAAa,EAAE;iBACpC,CAAC;gBACJ,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,IAAI,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;YACpD,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC;YAC7B,MAAM,MAAM,GAAG,KAAK,CAAC,SAAS,CAAC;gBAC7B,IAAI;gBACJ,QAAQ,EAAE,QAAQ,IAAI,gBAAgB;gBACtC,KAAK,EACH,OAAO,CAAC,YAAY,KAAK,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,QAAQ,CAAC,KAAK;gBACnE,QAAQ,EAAE,OAAO,CAAC,QAAQ;gBAC1B,SAAS;aACV,CAAC,CAAC;YAEH,IAAI,KAAK,CAAC,QAAQ,CAAC,UAAU,EAAE,CAAC;gBAC9B,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,UAAU,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,CAAC;YACvE,CAAC;QACH,CAAC;QAED,kBAAkB,CAAC,KAAK;YACtB,MAAM,KAAK,GAAG,QAAQ,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YACtC,IAAI,CAAC,KAAK;gBAAE,OAAO;YAEnB,MAAM,IAAI,GACR,OAAO,OAAO,CAAC,QAAQ,KAAK,UAAU;gBACpC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC;gBACzB,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,IAAI,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;YACpD,MAAM,UAAU,GAAG,KAAK,CAAC,QAAQ,CAAC,UAAU;gBAC1C,CAAC,CAAC,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,UAAU,CAAC;gBAC/C,CAAC,CAAC,SAAS,CAAC;YACd,IAAI,KAAK,CAAC,QAAQ,CAAC,UAAU,EAAE,CAAC;gBAC9B,cAAc,CAAC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;YACnD,CAAC;YAED,IAAI,UAAU,EAAE,CAAC;gBACf,UAAU,CAAC,MAAM,CAAC,GAAG,CAAC;oBACpB,UAAU,EAAE,KAAK,CAAC,eAAe;oBACjC,OAAO,EAAE,KAAK,CAAC,UAAU,CAAC,SAAS,EAAE,KAAK,CAAC,eAAe,CAAC;oBAC3D,GAAG,UAAU,CAAC,KAAK,EAAE,OAAO,CAAC,aAAa,KAAK,KAAK,CAAC;iBACtD,CAAC,CAAC;gBACH,OAAO;YACT,CAAC;YAED,MAAM,OAAO,GAAG,IAAI,IAAI,EAAE,CAAC;YAC3B,MAAM,SAAS,GAAG,UAAU,CAAC,OAAO,EAAE,KAAK,CAAC,eAAe,CAAC,CAAC;YAC7D,MAAM,QAAQ,GACZ,KAAK,CAAC,QAAQ,CAAC,UAAU;gBACzB,6BAA6B,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;YAC/D,MAAM,gBAAgB,GAAG,KAAK,CAAC,MAAM;gBACnC,CAAC,CAAC,yBAAyB,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC;gBAC7C,CAAC,CAAC,SAAS,CAAC;YAEd,KAAK,CAAC,UAAU,CAAC;gBACf,IAAI;gBACJ,QAAQ,EAAE,QAAQ,IAAI,gBAAgB;gBACtC,KAAK,EACH,OAAO,CAAC,YAAY,KAAK,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,QAAQ,CAAC,KAAK;gBACnE,QAAQ,EAAE,OAAO,CAAC,QAAQ;gBAC1B,UAAU,EAAE,KAAK,CAAC,eAAe;gBACjC,SAAS;gBACT,OAAO;gBACP,GAAG,UAAU,CAAC,KAAK,EAAE,OAAO,CAAC,aAAa,KAAK,KAAK,CAAC;aACtD,CAAC,CAAC;QACL,CAAC;QAED,OAAO,CAAC,KAAK;YACX,cAAc,GAAG,KAAK,CAAC;YACvB,QAAQ,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,IAAI,EAAE,EAAE,CAAC,CAAC;QAClD,CAAC;QAED,WAAW,CAAC,KAAK;YACf,IAAI,aAAa,CAAC,KAAK,CAAC,EAAE,CAAC;gBACzB,iBAAiB,CAAC,KAAK,CAAC,CAAC;gBACzB,OAAO;YACT,CAAC;YACD,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,UAAU,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,IAAI,EAAE,EAAE,CAAC,CAAC;QAClE,CAAC;QAED,SAAS,CAAC,KAAK;YACb,eAAe,CAAC,KAAK,CAAC,CAAC;QACzB,CAAC;QAED,YAAY,CAAC,KAAK;YAChB,cAAc,GAAG,IAAI,CAAC;YACtB,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;YAC7C,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;YACjC,kBAAkB,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QACpC,CAAC;QAED,KAAK,CAAC,QAAQ,CAAC,KAAK;YAClB,IAAI,cAAc,EAAE,CAAC;gBACnB,QAAQ,CAAC,KAAK,EAAE,CAAC;gBACjB,cAAc,GAAG,KAAK,CAAC;gBACvB,MAAM,gBAAgB,CAAC,KAAK,CAAC,CAAC;gBAC9B,OAAO;YACT,CAAC;YACD,kBAAkB,CAAC,KAAK,EAAE,QAAQ,CAAC,KAAK,EAAE,CAAC,CAAC;YAC5C,MAAM,gBAAgB,CAAC,KAAK,CAAC,CAAC;QAChC,CAAC;QAED,KAAK,CAAC,KAAK,CAAC,KAAK;YACf,MAAM,gBAAgB,CAAC,KAAK,CAAC,CAAC;QAChC,CAAC;QAED,gBAAgB,CAAC,KAAK;YACpB,MAAM,KAAK,GAAG,QAAQ,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YACtC,IAAI,CAAC,KAAK;gBAAE,OAAO;YAEnB,MAAM,IAAI,GACR,OAAO,OAAO,CAAC,QAAQ,KAAK,UAAU;gBACpC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC;gBACzB,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,IAAI,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;YACpD,MAAM,QAAQ,GACZ,KAAK,CAAC,QAAQ,CAAC,UAAU;gBACzB,6BAA6B,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;YAE/D,KAAK,CAAC,UAAU,CAAC;gBACf,IAAI;gBACJ,QAAQ,EAAE,QAAQ,IAAI,yBAAyB,CAAC,GAAG,CAAC,WAAW,CAAC;gBAChE,KAAK,EACH,OAAO,CAAC,YAAY,KAAK,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,QAAQ,CAAC,KAAK;gBACnE,QAAQ,EAAE,OAAO,CAAC,QAAQ;gBAC1B,UAAU,EAAE,KAAK,CAAC,UAAU;gBAC5B,GAAG,YAAY,CAAC,KAAK,EAAE,OAAO,CAAC,aAAa,KAAK,KAAK,CAAC;aACxD,CAAC,CAAC;QACL,CAAC;KACF,CAAC;AACJ,CAAC"}

package/examples/callback-tracing.ts

@@ -0,0 +1,55 @@
+import { Lemma, active } from "@uselemma/tracing";
+
+const lemma = new Lemma({
+  apiKey: process.env.LEMMA_API_KEY,
+  projectId: process.env.LEMMA_PROJECT_ID,
+});
+
+async function searchDocs(query: string) {
+  return [
+    { id: "doc-1", text: "Orders ship in two business days.", score: 0.91 },
+  ];
+}
+
+async function callModel(message: string, docs: unknown[]) {
+  return {
+    text: `Answer for ${message}`,
+    messages: [{ role: "user", content: message }],
+    usage: { inputTokens: 42, outputTokens: 12 },
+    docs,
+  };
+}
+
+export async function runSupportAgent(userMessage: string) {
+  return lemma.trace(
+    {
+      name: "support-agent",
+      input: userMessage,
+      threadId: "thread-123",
+      userId: "user-456",
+    },
+    async (trace) => {
+      const docs = await searchDocs(userMessage);
+
+      trace.recordTool({
+        name: "search_docs",
+        input: { query: userMessage },
+        output: docs,
+        toolParameters: { query: "string" },
+      });
+
+      const response = await callModel(userMessage, docs);
+      active().recordGeneration({
+        name: "draft-reply",
+        input: response.messages,
+        output: response.text,
+        model: "gpt-4o",
+        usage: response.usage,
+        llmInputMessages: response.messages,
+        llmInvocationParameters: { temperature: 0.2 },
+      });
+
+      return response.text;
+    },
+  );
+}

package/examples/record-by-id.ts

@@ -0,0 +1,47 @@
+import { Lemma } from "@uselemma/tracing";
+
+const lemma = new Lemma({
+  apiKey: process.env.LEMMA_API_KEY,
+  projectId: process.env.LEMMA_PROJECT_ID,
+});
+
+async function searchDocs(query: string) {
+  return [
+    { id: "doc-1", text: "Orders ship in two business days.", score: 0.91 },
+  ];
+}
+
+export async function runDetachedHelpers(userMessage: string) {
+  const trace = lemma.trace({
+    name: "support-agent",
+    input: userMessage,
+  });
+
+  const retrieve = lemma.startSpan({
+    traceId: trace.id,
+    name: "retrieve-context",
+    input: { query: userMessage },
+  });
+
+  const docs = await searchDocs(userMessage);
+
+  lemma.recordTool({
+    traceId: trace.id,
+    parentSpanId: retrieve.id,
+    name: "search_docs",
+    input: { query: userMessage },
+    output: docs,
+    toolParameters: { query: "string" },
+  });
+
+  retrieve.end({
+    output: { count: docs.length },
+    retrievalDocuments: docs.map((doc) => ({
+      id: doc.id,
+      content: doc.text,
+      score: doc.score,
+    })),
+  });
+
+  await trace.end({ output: `Found ${docs.length} document(s).` });
+}

package/examples/trace-handles.ts

@@ -0,0 +1,59 @@
+import { Lemma } from "@uselemma/tracing";
+
+const lemma = new Lemma({
+  apiKey: process.env.LEMMA_API_KEY,
+  projectId: process.env.LEMMA_PROJECT_ID,
+});
+
+async function searchDocs(query: string) {
+  return [
+    { id: "doc-1", text: "Orders ship in two business days.", score: 0.91 },
+  ];
+}
+
+async function callModel(message: string, docs: unknown[]) {
+  return {
+    text: `Answer for ${message}`,
+    usage: { inputTokens: 42, outputTokens: 12 },
+    docs,
+  };
+}
+
+export async function runWithHandles(userMessage: string) {
+  const trace = lemma.trace({
+    name: "support-agent",
+    input: userMessage,
+    threadId: "thread-123",
+  });
+
+  const retrieval = trace.startSpan({
+    name: "retrieve-context",
+    input: { query: userMessage },
+  });
+
+  const docs = await searchDocs(userMessage);
+  retrieval.recordTool({
+    name: "search_docs",
+    input: { query: userMessage },
+    output: docs,
+    toolParameters: { query: "string" },
+  });
+  retrieval.end({
+    output: { count: docs.length },
+    retrievalDocuments: docs.map((doc) => ({
+      id: doc.id,
+      content: doc.text,
+      score: doc.score,
+    })),
+  });
+
+  const response = await callModel(userMessage, docs);
+  trace.recordGeneration({
+    name: "draft-reply",
+    output: response.text,
+    model: "gpt-4o",
+    usage: response.usage,
+  });
+
+  await trace.end({ output: response.text });
+}

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.3",
-  "description": "OpenTelemetry-based tracing module for Lemma",
+  "version": "4.0.1",
+  "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",