@uselemma/tracing 7.9.0 → 7.11.0

package/dist/coding-agent.js

@@ -0,0 +1,186 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.startCodingAgentTurn = startCodingAgentTurn;
+exports.recordCodingAgentToolStart = recordCodingAgentToolStart;
+exports.recordCodingAgentToolResult = recordCodingAgentToolResult;
+exports.completeCodingAgentTurn = completeCodingAgentTurn;
+exports.codingAgentTurnTrace = codingAgentTurnTrace;
+const node_crypto_1 = require("node:crypto");
+const client_1 = require("./client");
+const error_message_1 = require("./error-message");
+function requireOpen(turn) {
+    if (turn.status === "completed") {
+        throw new Error(`Coding agent turn ${turn.sessionId}/${turn.turnId} already completed`);
+    }
+    return turn;
+}
+function harnessAttributes(turn) {
+    return {
+        "lemma.harness.id": turn.harness,
+        "lemma.harness.session_id": turn.sessionId,
+        "lemma.harness.turn_id": turn.turnId,
+        "lemma.sdk.integration": "coding-agent",
+    };
+}
+function deterministicUuid(value) {
+    const hash = (0, node_crypto_1.createHash)("sha256").update(value).digest("hex");
+    const variant = (8 + (Number.parseInt(hash[16], 16) % 4)).toString(16);
+    return `${hash.slice(0, 8)}-${hash.slice(8, 12)}-5${hash.slice(13, 16)}-${variant}${hash.slice(17, 20)}-${hash.slice(20, 32)}`;
+}
+function turnIdentity(options) {
+    return `${options.harness}\0${options.sessionId}\0${options.turnId}`;
+}
+function startCodingAgentTurn(options) {
+    return {
+        version: 1,
+        status: "open",
+        harness: options.harness,
+        sessionId: options.sessionId,
+        turnId: options.turnId,
+        traceId: options.traceId ??
+            deterministicUuid(`lemma-coding-agent-trace\0${turnIdentity(options)}`),
+        generationId: options.generationId ??
+            deterministicUuid(`lemma-coding-agent-generation\0${turnIdentity(options)}`),
+        prompt: options.prompt,
+        startedAt: options.startedAt,
+        model: options.model,
+        provider: options.provider,
+        metadata: options.metadata,
+        tools: [],
+    };
+}
+function recordCodingAgentToolStart(turn, event) {
+    const open = requireOpen(turn);
+    const existingIndex = open.tools.findIndex((tool) => tool.toolUseId === event.toolUseId);
+    if (existingIndex >= 0) {
+        const existing = open.tools[existingIndex];
+        if (!existing.startTimeMissing)
+            return open;
+        const tools = [...open.tools];
+        tools[existingIndex] = {
+            ...existing,
+            toolName: event.toolName,
+            input: event.input === undefined ? existing.input : event.input,
+            startedAt: event.startedAt,
+            startTimeMissing: undefined,
+        };
+        return { ...open, tools };
+    }
+    return {
+        ...open,
+        tools: [
+            ...open.tools,
+            {
+                toolUseId: event.toolUseId,
+                toolName: event.toolName,
+                input: event.input,
+                startedAt: event.startedAt,
+            },
+        ],
+    };
+}
+function recordCodingAgentToolResult(turn, event) {
+    const open = requireOpen(turn);
+    const existingIndex = open.tools.findIndex((tool) => tool.toolUseId === event.toolUseId);
+    const completed = {
+        toolUseId: event.toolUseId,
+        toolName: event.toolName,
+        input: event.input === undefined && existingIndex >= 0
+            ? open.tools[existingIndex].input
+            : event.input,
+        output: event.output,
+        error: (0, error_message_1.failureMessage)(event.error) ?? undefined,
+        startedAt: existingIndex >= 0 ? open.tools[existingIndex].startedAt : undefined,
+        endedAt: event.endedAt,
+        startTimeMissing: existingIndex >= 0 ? open.tools[existingIndex].startTimeMissing : true,
+    };
+    if (existingIndex < 0) {
+        return { ...open, tools: [...open.tools, completed] };
+    }
+    const tools = [...open.tools];
+    tools[existingIndex] = completed;
+    return { ...open, tools };
+}
+function completeCodingAgentTurn(turn, event) {
+    if (turn.status === "completed")
+        return turn;
+    if ((event.generationStartedAt === undefined) !==
+        (event.generationEndedAt === undefined)) {
+        throw new Error("Coding agent generation timing requires both startedAt and endedAt");
+    }
+    const tools = turn.tools.map((tool) => {
+        const error = (0, error_message_1.failureMessage)(tool.error) ?? undefined;
+        return tool.endedAt
+            ? { ...tool, error }
+            : { ...tool, error, resultMissing: true };
+    });
+    return {
+        ...turn,
+        tools,
+        status: "completed",
+        response: event.response,
+        endedAt: event.endedAt,
+        model: event.model ?? turn.model,
+        provider: event.provider ?? turn.provider,
+        generationStartedAt: event.generationStartedAt,
+        generationEndedAt: event.generationEndedAt,
+    };
+}
+function codingAgentTurnTrace(turn) {
+    const attributes = harnessAttributes(turn);
+    const context = new client_1.TraceContext({
+        id: turn.traceId,
+        name: `${turn.harness} coding agent`,
+        input: turn.prompt,
+        output: turn.response,
+        threadId: turn.sessionId,
+        startedAt: turn.startedAt,
+        metadata: {
+            ...(turn.metadata ?? {}),
+            ...attributes,
+        },
+    });
+    for (const tool of turn.tools) {
+        const missingStart = tool.startTimeMissing === true || tool.startedAt === undefined;
+        const missingResult = tool.resultMissing === true || tool.endedAt === undefined;
+        const error = tool.error;
+        context.recordTool({
+            id: tool.toolUseId,
+            name: tool.toolName,
+            toolName: tool.toolName,
+            input: tool.input,
+            output: tool.output,
+            error,
+            status: error == null ? (missingResult ? undefined : "OK") : "ERROR",
+            startedAt: tool.startedAt ?? null,
+            endedAt: tool.endedAt ?? null,
+            attributes,
+            metadata: {
+                tool_use_id: tool.toolUseId,
+                ...(missingStart ? { start_time_missing: true } : {}),
+                ...(missingResult ? { result_missing: true } : {}),
+            },
+        });
+    }
+    const generationTimingMissing = turn.generationStartedAt === undefined;
+    context.recordGeneration({
+        id: turn.generationId,
+        name: `${turn.harness} response`,
+        input: turn.prompt,
+        output: turn.response,
+        model: turn.model,
+        llmProvider: turn.provider,
+        llmInputMessages: [{ role: "user", content: turn.prompt }],
+        llmOutputMessages: [{ role: "assistant", content: turn.response }],
+        startedAt: turn.generationStartedAt ?? null,
+        endedAt: turn.generationEndedAt ?? null,
+        attributes,
+        metadata: generationTimingMissing ? { timing_missing: true } : undefined,
+    });
+    return {
+        context,
+        startedAt: turn.startedAt,
+        endedAt: turn.endedAt,
+    };
+}
+//# sourceMappingURL=coding-agent.js.map

package/dist/coding-agent.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"coding-agent.js","sourceRoot":"","sources":["../src/coding-agent.ts"],"names":[],"mappings":";;AA+HA,oDAwBC;AAED,gEAiCC;AAED,kEA6BC;AAED,0DA8BC;AAED,oDA+DC;AA1TD,6CAAyC;AAEzC,qCAAwC;AACxC,mDAAiD;AAgGjD,SAAS,WAAW,CAAC,IAAqB;IACxC,IAAI,IAAI,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;QAChC,MAAM,IAAI,KAAK,CACb,qBAAqB,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,MAAM,oBAAoB,CACvE,CAAC;IACJ,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,iBAAiB,CAAC,IAAqB;IAC9C,OAAO;QACL,kBAAkB,EAAE,IAAI,CAAC,OAAO;QAChC,0BAA0B,EAAE,IAAI,CAAC,SAAS;QAC1C,uBAAuB,EAAE,IAAI,CAAC,MAAM;QACpC,uBAAuB,EAAE,cAAc;KACxC,CAAC;AACJ,CAAC;AAED,SAAS,iBAAiB,CAAC,KAAa;IACtC,MAAM,IAAI,GAAG,IAAA,wBAAU,EAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAC9D,MAAM,OAAO,GAAG,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;IACvE,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,IAAI,CAAC,KAAK,CAAC,EAAE,EAAE,EAAE,CAAC,IAAI,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,EAAE,EAAE,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC;AACjI,CAAC;AAED,SAAS,YAAY,CAAC,OAAoC;IACxD,OAAO,GAAG,OAAO,CAAC,OAAO,KAAK,OAAO,CAAC,SAAS,KAAK,OAAO,CAAC,MAAM,EAAE,CAAC;AACvE,CAAC;AAED,SAAgB,oBAAoB,CAClC,OAAoC;IAEpC,OAAO;QACL,OAAO,EAAE,CAAC;QACV,MAAM,EAAE,MAAM;QACd,OAAO,EAAE,OAAO,CAAC,OAAO;QACxB,SAAS,EAAE,OAAO,CAAC,SAAS;QAC5B,MAAM,EAAE,OAAO,CAAC,MAAM;QACtB,OAAO,EACL,OAAO,CAAC,OAAO;YACf,iBAAiB,CAAC,6BAA6B,YAAY,CAAC,OAAO,CAAC,EAAE,CAAC;QACzE,YAAY,EACV,OAAO,CAAC,YAAY;YACpB,iBAAiB,CACf,kCAAkC,YAAY,CAAC,OAAO,CAAC,EAAE,CAC1D;QACH,MAAM,EAAE,OAAO,CAAC,MAAM;QACtB,SAAS,EAAE,OAAO,CAAC,SAAS;QAC5B,KAAK,EAAE,OAAO,CAAC,KAAK;QACpB,QAAQ,EAAE,OAAO,CAAC,QAAQ;QAC1B,QAAQ,EAAE,OAAO,CAAC,QAAQ;QAC1B,KAAK,EAAE,EAAE;KACV,CAAC;AACJ,CAAC;AAED,SAAgB,0BAA0B,CACxC,IAAqB,EACrB,KAA2B;IAE3B,MAAM,IAAI,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC;IAC/B,MAAM,aAAa,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CACxC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,KAAK,KAAK,CAAC,SAAS,CAC7C,CAAC;IACF,IAAI,aAAa,IAAI,CAAC,EAAE,CAAC;QACvB,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;QAC3C,IAAI,CAAC,QAAQ,CAAC,gBAAgB;YAAE,OAAO,IAAI,CAAC;QAC5C,MAAM,KAAK,GAAG,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC;QAC9B,KAAK,CAAC,aAAa,CAAC,GAAG;YACrB,GAAG,QAAQ;YACX,QAAQ,EAAE,KAAK,CAAC,QAAQ;YACxB,KAAK,EAAE,KAAK,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK;YAC/D,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,gBAAgB,EAAE,SAAS;SAC5B,CAAC;QACF,OAAO,EAAE,GAAG,IAAI,EAAE,KAAK,EAAE,CAAC;IAC5B,CAAC;IACD,OAAO;QACL,GAAG,IAAI;QACP,KAAK,EAAE;YACL,GAAG,IAAI,CAAC,KAAK;YACb;gBACE,SAAS,EAAE,KAAK,CAAC,SAAS;gBAC1B,QAAQ,EAAE,KAAK,CAAC,QAAQ;gBACxB,KAAK,EAAE,KAAK,CAAC,KAAK;gBAClB,SAAS,EAAE,KAAK,CAAC,SAAS;aAC3B;SACF;KACF,CAAC;AACJ,CAAC;AAED,SAAgB,2BAA2B,CACzC,IAAqB,EACrB,KAA4B;IAE5B,MAAM,IAAI,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC;IAC/B,MAAM,aAAa,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CACxC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,KAAK,KAAK,CAAC,SAAS,CAC7C,CAAC;IACF,MAAM,SAAS,GAAwB;QACrC,SAAS,EAAE,KAAK,CAAC,SAAS;QAC1B,QAAQ,EAAE,KAAK,CAAC,QAAQ;QACxB,KAAK,EACH,KAAK,CAAC,KAAK,KAAK,SAAS,IAAI,aAAa,IAAI,CAAC;YAC7C,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC,KAAK;YACjC,CAAC,CAAC,KAAK,CAAC,KAAK;QACjB,MAAM,EAAE,KAAK,CAAC,MAAM;QACpB,KAAK,EAAE,IAAA,8BAAc,EAAC,KAAK,CAAC,KAAK,CAAC,IAAI,SAAS;QAC/C,SAAS,EACP,aAAa,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;QACtE,OAAO,EAAE,KAAK,CAAC,OAAO;QACtB,gBAAgB,EACd,aAAa,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI;KACzE,CAAC;IACF,IAAI,aAAa,GAAG,CAAC,EAAE,CAAC;QACtB,OAAO,EAAE,GAAG,IAAI,EAAE,KAAK,EAAE,CAAC,GAAG,IAAI,CAAC,KAAK,EAAE,SAAS,CAAC,EAAE,CAAC;IACxD,CAAC;IACD,MAAM,KAAK,GAAG,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC;IAC9B,KAAK,CAAC,aAAa,CAAC,GAAG,SAAS,CAAC;IACjC,OAAO,EAAE,GAAG,IAAI,EAAE,KAAK,EAAE,CAAC;AAC5B,CAAC;AAED,SAAgB,uBAAuB,CACrC,IAAqB,EACrB,KAAqC;IAErC,IAAI,IAAI,CAAC,MAAM,KAAK,WAAW;QAAE,OAAO,IAAI,CAAC;IAC7C,IACE,CAAC,KAAK,CAAC,mBAAmB,KAAK,SAAS,CAAC;QACzC,CAAC,KAAK,CAAC,iBAAiB,KAAK,SAAS,CAAC,EACvC,CAAC;QACD,MAAM,IAAI,KAAK,CACb,oEAAoE,CACrE,CAAC;IACJ,CAAC;IACD,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;QACpC,MAAM,KAAK,GAAG,IAAA,8BAAc,EAAC,IAAI,CAAC,KAAK,CAAC,IAAI,SAAS,CAAC;QACtD,OAAO,IAAI,CAAC,OAAO;YACjB,CAAC,CAAC,EAAE,GAAG,IAAI,EAAE,KAAK,EAAE;YACpB,CAAC,CAAC,EAAE,GAAG,IAAI,EAAE,KAAK,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC;IAC9C,CAAC,CAAC,CAAC;IACH,OAAO;QACL,GAAG,IAAI;QACP,KAAK;QACL,MAAM,EAAE,WAAW;QACnB,QAAQ,EAAE,KAAK,CAAC,QAAQ;QACxB,OAAO,EAAE,KAAK,CAAC,OAAO;QACtB,KAAK,EAAE,KAAK,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK;QAChC,QAAQ,EAAE,KAAK,CAAC,QAAQ,IAAI,IAAI,CAAC,QAAQ;QACzC,mBAAmB,EAAE,KAAK,CAAC,mBAAmB;QAC9C,iBAAiB,EAAE,KAAK,CAAC,iBAAiB;KAC3C,CAAC;AACJ,CAAC;AAED,SAAgB,oBAAoB,CAClC,IAA8B;IAE9B,MAAM,UAAU,GAAG,iBAAiB,CAAC,IAAI,CAAC,CAAC;IAC3C,MAAM,OAAO,GAAG,IAAI,qBAAY,CAAC;QAC/B,EAAE,EAAE,IAAI,CAAC,OAAO;QAChB,IAAI,EAAE,GAAG,IAAI,CAAC,OAAO,eAAe;QACpC,KAAK,EAAE,IAAI,CAAC,MAAM;QAClB,MAAM,EAAE,IAAI,CAAC,QAAQ;QACrB,QAAQ,EAAE,IAAI,CAAC,SAAS;QACxB,SAAS,EAAE,IAAI,CAAC,SAAS;QACzB,QAAQ,EAAE;YACR,GAAG,CAAC,IAAI,CAAC,QAAQ,IAAI,EAAE,CAAC;YACxB,GAAG,UAAU;SACd;KACF,CAAC,CAAC;IAEH,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;QAC9B,MAAM,YAAY,GAChB,IAAI,CAAC,gBAAgB,KAAK,IAAI,IAAI,IAAI,CAAC,SAAS,KAAK,SAAS,CAAC;QACjE,MAAM,aAAa,GACjB,IAAI,CAAC,aAAa,KAAK,IAAI,IAAI,IAAI,CAAC,OAAO,KAAK,SAAS,CAAC;QAC5D,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC;QACzB,OAAO,CAAC,UAAU,CAAC;YACjB,EAAE,EAAE,IAAI,CAAC,SAAS;YAClB,IAAI,EAAE,IAAI,CAAC,QAAQ;YACnB,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,KAAK;YACL,MAAM,EAAE,KAAK,IAAI,IAAI,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,OAAO;YACpE,SAAS,EAAE,IAAI,CAAC,SAAS,IAAI,IAAI;YACjC,OAAO,EAAE,IAAI,CAAC,OAAO,IAAI,IAAI;YAC7B,UAAU;YACV,QAAQ,EAAE;gBACR,WAAW,EAAE,IAAI,CAAC,SAAS;gBAC3B,GAAG,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,kBAAkB,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBACrD,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,cAAc,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aACnD;SACF,CAAC,CAAC;IACL,CAAC;IAED,MAAM,uBAAuB,GAAG,IAAI,CAAC,mBAAmB,KAAK,SAAS,CAAC;IACvE,OAAO,CAAC,gBAAgB,CAAC;QACvB,EAAE,EAAE,IAAI,CAAC,YAAY;QACrB,IAAI,EAAE,GAAG,IAAI,CAAC,OAAO,WAAW;QAChC,KAAK,EAAE,IAAI,CAAC,MAAM;QAClB,MAAM,EAAE,IAAI,CAAC,QAAQ;QACrB,KAAK,EAAE,IAAI,CAAC,KAAK;QACjB,WAAW,EAAE,IAAI,CAAC,QAAQ;QAC1B,gBAAgB,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC;QAC1D,iBAAiB,EAAE,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,IAAI,CAAC,QAAQ,EAAE,CAAC;QAClE,SAAS,EAAE,IAAI,CAAC,mBAAmB,IAAI,IAAI;QAC3C,OAAO,EAAE,IAAI,CAAC,iBAAiB,IAAI,IAAI;QACvC,UAAU;QACV,QAAQ,EAAE,uBAAuB,CAAC,CAAC,CAAC,EAAE,cAAc,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,SAAS;KACzE,CAAC,CAAC;IAEH,OAAO;QACL,OAAO;QACP,SAAS,EAAE,IAAI,CAAC,SAAS;QACzB,OAAO,EAAE,IAAI,CAAC,OAAO;KACtB,CAAC;AACJ,CAAC"}

package/dist/index.d.ts

@@ -1,4 +1,6 @@
-export { Lemma, NoopSpanHandle, SpanHandle, TraceContext, TraceHandle, type DetachedGenerationOptions, type DetachedSpanOptions, type DetachedToolOptions, type GenerationOptions, type JsonValue, type LemmaClientOptions, type SpanOptions, type ToolOptions, type TraceEndOptions, type TraceOptions, type DebugSmokeTestResult, } from "./client";
+export { Lemma, NoopSpanHandle, SpanHandle, TraceContext, TraceHandle, type DetachedGenerationOptions, type DetachedSpanOptions, type DetachedToolOptions, type GenerationOptions, type JsonValue, type LemmaClientOptions, type SdkIntegration, type SpanOptions, type OperationUsageResult, type TokenUsage, type ToolOptions, type TraceEndOptions, type TraceOptions, type DebugSmokeTestResult, type WireTokenUsage, attachResultUsage, normalizeTokenUsage, toWireTokenUsage, } from "./client";
+export { codingAgentTurnTrace, completeCodingAgentTurn, recordCodingAgentToolResult, recordCodingAgentToolStart, startCodingAgentTurn, type CodingAgentHarnessId, type CodingAgentToolCall, type CodingAgentToolResult, type CodingAgentToolStart, type CodingAgentTurn, type CodingAgentTurnTrace, type CompleteCodingAgentTurnOptions, type CompletedCodingAgentTurn, type OpenCodingAgentTurn, type StartCodingAgentTurnOptions, } from "./coding-agent";
+export { normalizeRelease, RELEASE_MAX_LENGTH } from "./release";
 export { disableDebugMode, enableDebugMode, isDebugModeEnabled, isDebugVerifyEnabled, lemmaDebug, } from "./debug-mode";
 export { LangChainCallbackHandler, LemmaLangChainCallbackHandler, langChain, langGraph, type LangChainIntegrationOptions, type LemmaLangChainIntegrationOptions, type LemmaLangGraphIntegrationOptions, } from "./langchain";
 export { LemmaMastraExporter, mastra, type LemmaMastraIntegrationOptions, type MastraErrorInfo, type MastraExportedSpan, type MastraIntegrationOptions, type MastraSpanType, type MastraTracingEvent, } from "./mastra";

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,KAAK,EACL,cAAc,EACd,UAAU,EACV,YAAY,EACZ,WAAW,EACX,KAAK,yBAAyB,EAC9B,KAAK,mBAAmB,EACxB,KAAK,mBAAmB,EACxB,KAAK,iBAAiB,EACtB,KAAK,SAAS,EACd,KAAK,kBAAkB,EACvB,KAAK,WAAW,EAChB,KAAK,WAAW,EAChB,KAAK,eAAe,EACpB,KAAK,YAAY,EACjB,KAAK,oBAAoB,GAC1B,MAAM,UAAU,CAAC;AAClB,OAAO,EACL,gBAAgB,EAChB,eAAe,EACf,kBAAkB,EAClB,oBAAoB,EACpB,UAAU,GACX,MAAM,cAAc,CAAC;AACtB,OAAO,EACL,wBAAwB,EACxB,6BAA6B,EAC7B,SAAS,EACT,SAAS,EACT,KAAK,2BAA2B,EAChC,KAAK,gCAAgC,EACrC,KAAK,gCAAgC,GACtC,MAAM,aAAa,CAAC;AACrB,OAAO,EACL,mBAAmB,EACnB,MAAM,EACN,KAAK,6BAA6B,EAClC,KAAK,eAAe,EACpB,KAAK,kBAAkB,EACvB,KAAK,wBAAwB,EAC7B,KAAK,cAAc,EACnB,KAAK,kBAAkB,GACxB,MAAM,UAAU,CAAC;AAClB,OAAO,EACL,YAAY,EACZ,KAAK,8BAA8B,EACnC,KAAK,gBAAgB,EACrB,KAAK,oBAAoB,EACzB,KAAK,iBAAiB,EACtB,KAAK,4BAA4B,GAClC,MAAM,iBAAiB,CAAC;AACzB,OAAO,EACL,QAAQ,EACR,KAAK,+BAA+B,EACpC,KAAK,0BAA0B,EAC/B,KAAK,4BAA4B,GAClC,MAAM,aAAa,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,KAAK,EACL,cAAc,EACd,UAAU,EACV,YAAY,EACZ,WAAW,EACX,KAAK,yBAAyB,EAC9B,KAAK,mBAAmB,EACxB,KAAK,mBAAmB,EACxB,KAAK,iBAAiB,EACtB,KAAK,SAAS,EACd,KAAK,kBAAkB,EACvB,KAAK,cAAc,EACnB,KAAK,WAAW,EAChB,KAAK,oBAAoB,EACzB,KAAK,UAAU,EACf,KAAK,WAAW,EAChB,KAAK,eAAe,EACpB,KAAK,YAAY,EACjB,KAAK,oBAAoB,EACzB,KAAK,cAAc,EACnB,iBAAiB,EACjB,mBAAmB,EACnB,gBAAgB,GACjB,MAAM,UAAU,CAAC;AAClB,OAAO,EACL,oBAAoB,EACpB,uBAAuB,EACvB,2BAA2B,EAC3B,0BAA0B,EAC1B,oBAAoB,EACpB,KAAK,oBAAoB,EACzB,KAAK,mBAAmB,EACxB,KAAK,qBAAqB,EAC1B,KAAK,oBAAoB,EACzB,KAAK,eAAe,EACpB,KAAK,oBAAoB,EACzB,KAAK,8BAA8B,EACnC,KAAK,wBAAwB,EAC7B,KAAK,mBAAmB,EACxB,KAAK,2BAA2B,GACjC,MAAM,gBAAgB,CAAC;AACxB,OAAO,EAAE,gBAAgB,EAAE,kBAAkB,EAAE,MAAM,WAAW,CAAC;AACjE,OAAO,EACL,gBAAgB,EAChB,eAAe,EACf,kBAAkB,EAClB,oBAAoB,EACpB,UAAU,GACX,MAAM,cAAc,CAAC;AACtB,OAAO,EACL,wBAAwB,EACxB,6BAA6B,EAC7B,SAAS,EACT,SAAS,EACT,KAAK,2BAA2B,EAChC,KAAK,gCAAgC,EACrC,KAAK,gCAAgC,GACtC,MAAM,aAAa,CAAC;AACrB,OAAO,EACL,mBAAmB,EACnB,MAAM,EACN,KAAK,6BAA6B,EAClC,KAAK,eAAe,EACpB,KAAK,kBAAkB,EACvB,KAAK,wBAAwB,EAC7B,KAAK,cAAc,EACnB,KAAK,kBAAkB,GACxB,MAAM,UAAU,CAAC;AAClB,OAAO,EACL,YAAY,EACZ,KAAK,8BAA8B,EACnC,KAAK,gBAAgB,EACrB,KAAK,oBAAoB,EACzB,KAAK,iBAAiB,EACtB,KAAK,4BAA4B,GAClC,MAAM,iBAAiB,CAAC;AACzB,OAAO,EACL,QAAQ,EACR,KAAK,+BAA+B,EACpC,KAAK,0BAA0B,EAC/B,KAAK,4BAA4B,GAClC,MAAM,aAAa,CAAC"}

package/dist/index.js

@@ -1,12 +1,24 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.vercelAI = exports.openAIAgents = exports.mastra = exports.LemmaMastraExporter = exports.langGraph = exports.langChain = exports.LemmaLangChainCallbackHandler = exports.LangChainCallbackHandler = exports.lemmaDebug = exports.isDebugVerifyEnabled = exports.isDebugModeEnabled = exports.enableDebugMode = exports.disableDebugMode = exports.TraceHandle = exports.TraceContext = exports.SpanHandle = exports.NoopSpanHandle = exports.Lemma = void 0;
+exports.vercelAI = exports.openAIAgents = exports.mastra = exports.LemmaMastraExporter = exports.langGraph = exports.langChain = exports.LemmaLangChainCallbackHandler = exports.LangChainCallbackHandler = exports.lemmaDebug = exports.isDebugVerifyEnabled = exports.isDebugModeEnabled = exports.enableDebugMode = exports.disableDebugMode = exports.RELEASE_MAX_LENGTH = exports.normalizeRelease = exports.startCodingAgentTurn = exports.recordCodingAgentToolStart = exports.recordCodingAgentToolResult = exports.completeCodingAgentTurn = exports.codingAgentTurnTrace = exports.toWireTokenUsage = exports.normalizeTokenUsage = exports.attachResultUsage = exports.TraceHandle = exports.TraceContext = exports.SpanHandle = exports.NoopSpanHandle = exports.Lemma = void 0;
 var client_1 = require("./client");
 Object.defineProperty(exports, "Lemma", { enumerable: true, get: function () { return client_1.Lemma; } });
 Object.defineProperty(exports, "NoopSpanHandle", { enumerable: true, get: function () { return client_1.NoopSpanHandle; } });
 Object.defineProperty(exports, "SpanHandle", { enumerable: true, get: function () { return client_1.SpanHandle; } });
 Object.defineProperty(exports, "TraceContext", { enumerable: true, get: function () { return client_1.TraceContext; } });
 Object.defineProperty(exports, "TraceHandle", { enumerable: true, get: function () { return client_1.TraceHandle; } });
+Object.defineProperty(exports, "attachResultUsage", { enumerable: true, get: function () { return client_1.attachResultUsage; } });
+Object.defineProperty(exports, "normalizeTokenUsage", { enumerable: true, get: function () { return client_1.normalizeTokenUsage; } });
+Object.defineProperty(exports, "toWireTokenUsage", { enumerable: true, get: function () { return client_1.toWireTokenUsage; } });
+var coding_agent_1 = require("./coding-agent");
+Object.defineProperty(exports, "codingAgentTurnTrace", { enumerable: true, get: function () { return coding_agent_1.codingAgentTurnTrace; } });
+Object.defineProperty(exports, "completeCodingAgentTurn", { enumerable: true, get: function () { return coding_agent_1.completeCodingAgentTurn; } });
+Object.defineProperty(exports, "recordCodingAgentToolResult", { enumerable: true, get: function () { return coding_agent_1.recordCodingAgentToolResult; } });
+Object.defineProperty(exports, "recordCodingAgentToolStart", { enumerable: true, get: function () { return coding_agent_1.recordCodingAgentToolStart; } });
+Object.defineProperty(exports, "startCodingAgentTurn", { enumerable: true, get: function () { return coding_agent_1.startCodingAgentTurn; } });
+var release_1 = require("./release");
+Object.defineProperty(exports, "normalizeRelease", { enumerable: true, get: function () { return release_1.normalizeRelease; } });
+Object.defineProperty(exports, "RELEASE_MAX_LENGTH", { enumerable: true, get: function () { return release_1.RELEASE_MAX_LENGTH; } });
 var debug_mode_1 = require("./debug-mode");
 Object.defineProperty(exports, "disableDebugMode", { enumerable: true, get: function () { return debug_mode_1.disableDebugMode; } });
 Object.defineProperty(exports, "enableDebugMode", { enumerable: true, get: function () { return debug_mode_1.enableDebugMode; } });

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;AAAA,mCAiBkB;AAhBhB,+FAAA,KAAK,OAAA;AACL,wGAAA,cAAc,OAAA;AACd,oGAAA,UAAU,OAAA;AACV,sGAAA,YAAY,OAAA;AACZ,qGAAA,WAAW,OAAA;AAab,2CAMsB;AALpB,8GAAA,gBAAgB,OAAA;AAChB,6GAAA,eAAe,OAAA;AACf,gHAAA,kBAAkB,OAAA;AAClB,kHAAA,oBAAoB,OAAA;AACpB,wGAAA,UAAU,OAAA;AAEZ,yCAQqB;AAPnB,qHAAA,wBAAwB,OAAA;AACxB,0HAAA,6BAA6B,OAAA;AAC7B,sGAAA,SAAS,OAAA;AACT,sGAAA,SAAS,OAAA;AAKX,mCASkB;AARhB,6GAAA,mBAAmB,OAAA;AACnB,gGAAA,MAAM,OAAA;AAQR,iDAOyB;AANvB,6GAAA,YAAY,OAAA;AAOd,yCAKqB;AAJnB,qGAAA,QAAQ,OAAA"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;AAAA,mCAwBkB;AAvBhB,+FAAA,KAAK,OAAA;AACL,wGAAA,cAAc,OAAA;AACd,oGAAA,UAAU,OAAA;AACV,sGAAA,YAAY,OAAA;AACZ,qGAAA,WAAW,OAAA;AAgBX,2GAAA,iBAAiB,OAAA;AACjB,6GAAA,mBAAmB,OAAA;AACnB,0GAAA,gBAAgB,OAAA;AAElB,+CAgBwB;AAftB,oHAAA,oBAAoB,OAAA;AACpB,uHAAA,uBAAuB,OAAA;AACvB,2HAAA,2BAA2B,OAAA;AAC3B,0HAAA,0BAA0B,OAAA;AAC1B,oHAAA,oBAAoB,OAAA;AAYtB,qCAAiE;AAAxD,2GAAA,gBAAgB,OAAA;AAAE,6GAAA,kBAAkB,OAAA;AAC7C,2CAMsB;AALpB,8GAAA,gBAAgB,OAAA;AAChB,6GAAA,eAAe,OAAA;AACf,gHAAA,kBAAkB,OAAA;AAClB,kHAAA,oBAAoB,OAAA;AACpB,wGAAA,UAAU,OAAA;AAEZ,yCAQqB;AAPnB,qHAAA,wBAAwB,OAAA;AACxB,0HAAA,6BAA6B,OAAA;AAC7B,sGAAA,SAAS,OAAA;AACT,sGAAA,SAAS,OAAA;AAKX,mCASkB;AARhB,6GAAA,mBAAmB,OAAA;AACnB,gGAAA,MAAM,OAAA;AAQR,iDAOyB;AANvB,6GAAA,YAAY,OAAA;AAOd,yCAKqB;AAJnB,qGAAA,QAAQ,OAAA"}

package/dist/langchain.d.ts

@@ -18,6 +18,7 @@ export type LangChainIntegrationOptions = {
     projectId?: LemmaClientOptions["projectId"];
     baseUrl?: LemmaClientOptions["baseUrl"];
     fetch?: LemmaClientOptions["fetch"];
+    release?: LemmaClientOptions["release"];
     agentName?: string;
     metadata?: Record<string, unknown>;
     /** Key looked up on run metadata / tags for threadId. Default: `threadId`. */
@@ -55,6 +56,11 @@ export declare class LemmaLangChainCallbackHandler {
     private forgetTraceRuns;
     private finalizeTrace;
     private maybeFinalizeOwner;
+    /**
+     * Copy token usage from an `invoke` / chain result onto generation spans
+     * whose callback event omitted it. Call after the operation returns.
+     */
+    recordResult(result: unknown): number;
     private traceName;
     handleChainStart(serialized: Serialized, inputs: unknown, runId: RunId, parentRunId?: RunId, tags?: string[], metadata?: Record<string, unknown>, runType?: string, name?: string): void;
     handleChainEnd(outputs: unknown, runId: RunId): Promise<void>;
@@ -71,7 +77,7 @@ export declare class LemmaLangChainCallbackHandler {
     handleRetrieverStart(serialized: Serialized, query: string, runId: RunId, parentRunId?: RunId, tags?: string[], metadata?: Record<string, unknown>): void;
     handleRetrieverEnd(documents: unknown[], runId: RunId): Promise<void>;
     handleRetrieverError(error: unknown, runId: RunId): Promise<void>;
-    /** Await outstanding terminal deliveries. */
+    /** Send completed traces and await ingest delivery. */
     flush(): Promise<void>;
     /** Finalize open traces and reset integration state. */
     shutdown(): Promise<void>;

package/dist/langchain.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"langchain.d.ts","sourceRoot":"","sources":["../src/langchain.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,KAAK,EACL,KAAK,kBAAkB,EAGxB,MAAM,UAAU,CAAC;AAIlB,KAAK,KAAK,GAAG,MAAM,CAAC;AAEpB,KAAK,UAAU,GAAG;IAChB,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACjC,EAAE,CAAC,EAAE,MAAM,CAAC;IACZ,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;CACxB,CAAC;AAEF,KAAK,SAAS,GAAG;IACf,WAAW,CAAC,EAAE,OAAO,EAAE,EAAE,CAAC;IAC1B,SAAS,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACpC,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;CACxB,CAAC;AAEF,MAAM,MAAM,2BAA2B,GAAG;IACxC,KAAK,CAAC,EAAE,KAAK,CAAC;IACd,MAAM,CAAC,EAAE,kBAAkB,CAAC,QAAQ,CAAC,CAAC;IACtC,SAAS,CAAC,EAAE,kBAAkB,CAAC,WAAW,CAAC,CAAC;IAC5C,OAAO,CAAC,EAAE,kBAAkB,CAAC,SAAS,CAAC,CAAC;IACxC,KAAK,CAAC,EAAE,kBAAkB,CAAC,OAAO,CAAC,CAAC;IACpC,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACnC,8EAA8E;IAC9E,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,6FAA6F;IAC7F,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB,CAAC;AA0jBF,qBAAa,6BAA6B;IAO5B,OAAO,CAAC,QAAQ,CAAC,OAAO;IANpC,IAAI,SAAW;IACf,OAAO,CAAC,KAAK,CAAoB;IACjC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAA+B;IACpD,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAkC;IACzD,OAAO,CAAC,QAAQ,CAAC,OAAO,CAA4B;gBAEvB,OAAO,GAAE,2BAAgC;IAItE,OAAO,CAAC,QAAQ;IAUhB,OAAO,CAAC,eAAe;IAYvB,OAAO,CAAC,aAAa;IAmBrB,OAAO,CAAC,aAAa;IAWrB,OAAO,CAAC,UAAU;IAalB,OAAO,CAAC,aAAa;IAOrB,OAAO,CAAC,cAAc;IAKtB,OAAO,CAAC,aAAa;IAKrB,OAAO,CAAC,YAAY;IAKpB,OAAO,CAAC,gBAAgB;IAyCxB,OAAO,CAAC,WAAW;IAInB,OAAO,CAAC,SAAS;IAKjB;;;;;OAKG;IACH,OAAO,CAAC,iBAAiB;IAgDzB,OAAO,CAAC,eAAe;YAMT,aAAa;IAqC3B,OAAO,CAAC,kBAAkB;IAU1B,OAAO,CAAC,SAAS;IAIjB,gBAAgB,CACd,UAAU,EAAE,UAAU,EACtB,MAAM,EAAE,OAAO,EACf,KAAK,EAAE,KAAK,EACZ,WAAW,CAAC,EAAE,KAAK,EACnB,IAAI,CAAC,EAAE,MAAM,EAAE,EACf,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAClC,OAAO,CAAC,EAAE,MAAM,EAChB,IAAI,CAAC,EAAE,MAAM;IAiET,cAAc,CAAC,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK;IA2B7C,gBAAgB,CAAC,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK;IA6BnD,cAAc,CACZ,UAAU,EAAE,UAAU,EACtB,OAAO,EAAE,MAAM,EAAE,EACjB,KAAK,EAAE,KAAK,EACZ,WAAW,CAAC,EAAE,KAAK,EACnB,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACrC,IAAI,CAAC,EAAE,MAAM,EAAE,EACf,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IA+CpC,oBAAoB,CAClB,UAAU,EAAE,UAAU,EACtB,QAAQ,EAAE,OAAO,EAAE,EAAE,EACrB,KAAK,EAAE,KAAK,EACZ,WAAW,CAAC,EAAE,KAAK,EACnB,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACrC,IAAI,CAAC,EAAE,MAAM,EAAE,EACf,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IAiDpC,OAAO,CAAC,gBAAgB;IAalB,YAAY,CAAC,MAAM,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK;YAoDpC,qBAAqB;IAa7B,cAAc,CAAC,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK;IA2BjD,eAAe,CACb,UAAU,EAAE,UAAU,EACtB,KAAK,EAAE,OAAO,EACd,KAAK,EAAE,KAAK,EACZ,WAAW,CAAC,EAAE,KAAK,EACnB,IAAI,CAAC,EAAE,MAAM,EAAE,EACf,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IA0C9B,aAAa,CAAC,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK;IAkC3C,eAAe,CAAC,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK;IA2BlD,oBAAoB,CAClB,UAAU,EAAE,UAAU,EACtB,KAAK,EAAE,MAAM,EACb,KAAK,EAAE,KAAK,EACZ,WAAW,CAAC,EAAE,KAAK,EACnB,IAAI,CAAC,EAAE,MAAM,EAAE,EACf,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IAwC9B,kBAAkB,CAAC,SAAS,EAAE,OAAO,EAAE,EAAE,KAAK,EAAE,KAAK;IAqBrD,oBAAoB,CAAC,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK;IAuBvD,6CAA6C;IACvC,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAS5B,wDAAwD;IAClD,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;CAKhC;AAED,eAAO,MAAM,wBAAwB,sCAAgC,CAAC;AAEtE,wBAAgB,SAAS,CAAC,OAAO,GAAE,2BAAgC,iCAElE;AAED;;;;GAIG;AACH,wBAAgB,SAAS,CAAC,OAAO,GAAE,2BAAgC,iCAElE;AAED,MAAM,MAAM,gCAAgC,GAAG,2BAA2B,CAAC;AAC3E,MAAM,MAAM,gCAAgC,GAAG,2BAA2B,CAAC"}
+{"version":3,"file":"langchain.d.ts","sourceRoot":"","sources":["../src/langchain.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,KAAK,EACL,KAAK,kBAAkB,EAGxB,MAAM,UAAU,CAAC;AAUlB,KAAK,KAAK,GAAG,MAAM,CAAC;AAEpB,KAAK,UAAU,GAAG;IAChB,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACjC,EAAE,CAAC,EAAE,MAAM,CAAC;IACZ,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;CACxB,CAAC;AAEF,KAAK,SAAS,GAAG;IACf,WAAW,CAAC,EAAE,OAAO,EAAE,EAAE,CAAC;IAC1B,SAAS,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACpC,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;CACxB,CAAC;AAEF,MAAM,MAAM,2BAA2B,GAAG;IACxC,KAAK,CAAC,EAAE,KAAK,CAAC;IACd,MAAM,CAAC,EAAE,kBAAkB,CAAC,QAAQ,CAAC,CAAC;IACtC,SAAS,CAAC,EAAE,kBAAkB,CAAC,WAAW,CAAC,CAAC;IAC5C,OAAO,CAAC,EAAE,kBAAkB,CAAC,SAAS,CAAC,CAAC;IACxC,KAAK,CAAC,EAAE,kBAAkB,CAAC,OAAO,CAAC,CAAC;IACpC,OAAO,CAAC,EAAE,kBAAkB,CAAC,SAAS,CAAC,CAAC;IACxC,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACnC,8EAA8E;IAC9E,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,6FAA6F;IAC7F,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB,CAAC;AAkmBF,qBAAa,6BAA6B;IAO5B,OAAO,CAAC,QAAQ,CAAC,OAAO;IANpC,IAAI,SAAW;IACf,OAAO,CAAC,KAAK,CAAoB;IACjC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAA+B;IACpD,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAkC;IACzD,OAAO,CAAC,QAAQ,CAAC,OAAO,CAA4B;gBAEvB,OAAO,GAAE,2BAAgC;IAItE,OAAO,CAAC,QAAQ;IAWhB,OAAO,CAAC,eAAe;IAYvB,OAAO,CAAC,aAAa;IAmBrB,OAAO,CAAC,aAAa;IAWrB,OAAO,CAAC,UAAU;IAalB,OAAO,CAAC,aAAa;IAOrB,OAAO,CAAC,cAAc;IAKtB,OAAO,CAAC,aAAa;IAKrB,OAAO,CAAC,YAAY;IAKpB,OAAO,CAAC,gBAAgB;IAyCxB,OAAO,CAAC,WAAW;IAInB,OAAO,CAAC,SAAS;IAKjB;;;;;OAKG;IACH,OAAO,CAAC,iBAAiB;IAgDzB,OAAO,CAAC,eAAe;YAMT,aAAa;IAqC3B,OAAO,CAAC,kBAAkB;IAgB1B;;;OAGG;IACH,YAAY,CAAC,MAAM,EAAE,OAAO,GAAG,MAAM;IASrC,OAAO,CAAC,SAAS;IAIjB,gBAAgB,CACd,UAAU,EAAE,UAAU,EACtB,MAAM,EAAE,OAAO,EACf,KAAK,EAAE,KAAK,EACZ,WAAW,CAAC,EAAE,KAAK,EACnB,IAAI,CAAC,EAAE,MAAM,EAAE,EACf,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAClC,OAAO,CAAC,EAAE,MAAM,EAChB,IAAI,CAAC,EAAE,MAAM;IAiET,cAAc,CAAC,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK;IA2B7C,gBAAgB,CAAC,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK;IA6BnD,cAAc,CACZ,UAAU,EAAE,UAAU,EACtB,OAAO,EAAE,MAAM,EAAE,EACjB,KAAK,EAAE,KAAK,EACZ,WAAW,CAAC,EAAE,KAAK,EACnB,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACrC,IAAI,CAAC,EAAE,MAAM,EAAE,EACf,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IA+CpC,oBAAoB,CAClB,UAAU,EAAE,UAAU,EACtB,QAAQ,EAAE,OAAO,EAAE,EAAE,EACrB,KAAK,EAAE,KAAK,EACZ,WAAW,CAAC,EAAE,KAAK,EACnB,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACrC,IAAI,CAAC,EAAE,MAAM,EAAE,EACf,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IAiDpC,OAAO,CAAC,gBAAgB;IAalB,YAAY,CAAC,MAAM,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK;YAqDpC,qBAAqB;IAa7B,cAAc,CAAC,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK;IA2BjD,eAAe,CACb,UAAU,EAAE,UAAU,EACtB,KAAK,EAAE,OAAO,EACd,KAAK,EAAE,KAAK,EACZ,WAAW,CAAC,EAAE,KAAK,EACnB,IAAI,CAAC,EAAE,MAAM,EAAE,EACf,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IA0C9B,aAAa,CAAC,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK;IAkC3C,eAAe,CAAC,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK;IA2BlD,oBAAoB,CAClB,UAAU,EAAE,UAAU,EACtB,KAAK,EAAE,MAAM,EACb,KAAK,EAAE,KAAK,EACZ,WAAW,CAAC,EAAE,KAAK,EACnB,IAAI,CAAC,EAAE,MAAM,EAAE,EACf,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IAwC9B,kBAAkB,CAAC,SAAS,EAAE,OAAO,EAAE,EAAE,KAAK,EAAE,KAAK;IAqBrD,oBAAoB,CAAC,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK;IAuBvD,uDAAuD;IACjD,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAS5B,wDAAwD;IAClD,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;CAKhC;AAED,eAAO,MAAM,wBAAwB,sCAAgC,CAAC;AAEtE,wBAAgB,SAAS,CAAC,OAAO,GAAE,2BAAgC,iCAElE;AAED;;;;GAIG;AACH,wBAAgB,SAAS,CAAC,OAAO,GAAE,2BAAgC,iCAElE;AAED,MAAM,MAAM,gCAAgC,GAAG,2BAA2B,CAAC;AAC3E,MAAM,MAAM,gCAAgC,GAAG,2BAA2B,CAAC"}

package/dist/langchain.js

@@ -5,7 +5,12 @@ exports.langChain = langChain;
 exports.langGraph = langGraph;
 const client_1 = require("./client");
 const error_message_1 = require("./error-message");
+const schedule_1 = require("./schedule");
 const tool_result_1 = require("./tool-result");
+const usage_1 = require("./usage");
+const INTEGRATION_ATTRS = {
+    "lemma.sdk.integration": "langchain",
+};
 const KNOWN_PROVIDERS = [
     "openai",
     "anthropic",
@@ -507,11 +512,53 @@ function durationMs(start, end) {
 }
 function langchainAttributes(runId, parentRunId, runType) {
     return Object.fromEntries(Object.entries({
+        ...INTEGRATION_ATTRS,
         "langchain.run_id": runId,
         "langchain.parent_run_id": parentRunId,
         "langchain.run_type": runType,
     }).filter(([, value]) => value !== undefined && value !== null));
 }
+/**
+ * Extract token usage from LLMResult.llmOutput and/or generation message
+ * usage_metadata. Prefers llmOutput.tokenUsage / token_usage, then the first
+ * message-level usage_metadata that normalizes.
+ */
+function llmTokenUsage(result) {
+    const llmOutput = result.llmOutput;
+    if (llmOutput && typeof llmOutput === "object") {
+        const fromOutput = (0, usage_1.normalizeTokenUsage)(llmOutput);
+        if (fromOutput)
+            return fromOutput;
+    }
+    const generations = result.generations;
+    if (!Array.isArray(generations))
+        return undefined;
+    for (const group of generations) {
+        if (!Array.isArray(group))
+            continue;
+        for (const item of group) {
+            if (!item || typeof item !== "object")
+                continue;
+            const record = item;
+            const message = record.message;
+            if (message && typeof message === "object") {
+                const msg = message;
+                const fromMessage = (0, usage_1.normalizeTokenUsage)(msg.usage_metadata) ??
+                    (0, usage_1.normalizeTokenUsage)(msg.usageMetadata) ??
+                    (0, usage_1.normalizeTokenUsage)(msg.response_metadata) ??
+                    (0, usage_1.normalizeTokenUsage)(msg);
+                if (fromMessage)
+                    return fromMessage;
+            }
+            const fromItem = (0, usage_1.normalizeTokenUsage)(record.usage_metadata) ??
+                (0, usage_1.normalizeTokenUsage)(record.generationInfo) ??
+                (0, usage_1.normalizeTokenUsage)(record);
+            if (fromItem)
+                return fromItem;
+        }
+    }
+    return undefined;
+}
 class LemmaLangChainCallbackHandler {
     options;
     name = "lemma";
@@ -529,6 +576,7 @@ class LemmaLangChainCallbackHandler {
             projectId: this.options.projectId,
             baseUrl: this.options.baseUrl,
             fetch: this.options.fetch,
+            release: this.options.release,
         });
         return this.lemma;
     }
@@ -712,13 +760,32 @@ class LemmaLangChainCallbackHandler {
         if (!run.ownsTrace)
             return;
         const stored = this.traces.get(run.owningTraceId);
-        if (!stored)
+        if (!stored || stored.ended)
             return;
         this.noteBounds(stored, run.startedAt, endedAt);
-        const promise = this.finalizeTrace(run.owningTraceId, stored);
+        // Delay the HTTP send so recordResult(invoke return) can stamp usage
+        // onto generations whose callback event omitted it.
+        const promise = new Promise((resolve, reject) => {
+            (0, schedule_1.scheduleMacrotask)(() => {
+                this.finalizeTrace(run.owningTraceId, stored).then(resolve, reject);
+            });
+        });
         this.trackPending(promise);
         return promise;
     }
+    /**
+     * Copy token usage from an `invoke` / chain result onto generation spans
+     * whose callback event omitted it. Call after the operation returns.
+     */
+    recordResult(result) {
+        let stamped = 0;
+        for (const stored of this.traces.values()) {
+            if (stored.ended)
+                continue;
+            stamped += stored.handle.applyGenerationUsage(result);
+        }
+        return stamped;
+    }
     traceName(serialized, fallback) {
         return this.options.agentName ?? serializedName(serialized, fallback);
     }
@@ -900,6 +967,7 @@ class LemmaLangChainCallbackHandler {
             endedAt,
             durationMs: durationMs(run.startedAt, endedAt),
             llmOutputMessages: softError ? undefined : outputMessages,
+            usage: llmTokenUsage(output),
         });
         const stored = this.storedTrace(run.owningTraceId);
         if (stored) {
@@ -924,14 +992,14 @@ class LemmaLangChainCallbackHandler {
         }
         this.runs.delete(runId);
         if (run.ownsTrace) {
-            await this.maybeFinalizeOwner(run, endedAt);
+            void this.maybeFinalizeOwner(run, endedAt);
             return;
         }
         // Final answer generation under a deferred owned LLM — close that root.
         const deferred = this.deferredOwnerFor(run.owningTraceId);
         if (deferred) {
             this.runs.delete(deferred.rootRunId);
-            await this.maybeFinalizeOwner(deferred, endedAt);
+            void this.maybeFinalizeOwner(deferred, endedAt);
         }
     }
     async finalizeDeferredOwner(owningTraceId, endedAt, rootError) {
@@ -942,7 +1010,7 @@ class LemmaLangChainCallbackHandler {
         if (stored && rootError)
             this.noteRootError(stored, rootError);
         this.runs.delete(deferred.rootRunId);
-        await this.maybeFinalizeOwner(deferred, endedAt);
+        void this.maybeFinalizeOwner(deferred, endedAt);
     }
     async handleLLMError(error, runId) {
         const run = this.runs.get(runId);
@@ -964,10 +1032,10 @@ class LemmaLangChainCallbackHandler {
         }
         this.runs.delete(runId);
         if (run.ownsTrace) {
-            await this.maybeFinalizeOwner(run, endedAt);
+            void this.maybeFinalizeOwner(run, endedAt);
             return;
         }
-        await this.finalizeDeferredOwner(run.owningTraceId, endedAt, message);
+        void this.finalizeDeferredOwner(run.owningTraceId, endedAt, message);
     }
     handleToolStart(serialized, input, runId, parentRunId, tags, metadata) {
         const startedAt = new Date();
@@ -1029,7 +1097,7 @@ class LemmaLangChainCallbackHandler {
             }
         }
         this.runs.delete(runId);
-        await this.maybeFinalizeOwner(run, endedAt);
+        void this.maybeFinalizeOwner(run, endedAt);
     }
     async handleToolError(error, runId) {
         const run = this.runs.get(runId);
@@ -1051,10 +1119,10 @@ class LemmaLangChainCallbackHandler {
         }
         this.runs.delete(runId);
         if (run.ownsTrace) {
-            await this.maybeFinalizeOwner(run, endedAt);
+            void this.maybeFinalizeOwner(run, endedAt);
             return;
         }
-        await this.finalizeDeferredOwner(run.owningTraceId, endedAt, message);
+        void this.finalizeDeferredOwner(run.owningTraceId, endedAt, message);
     }
     handleRetrieverStart(serialized, query, runId, parentRunId, tags, metadata) {
         const startedAt = new Date();
@@ -1099,7 +1167,7 @@ class LemmaLangChainCallbackHandler {
                 this.noteRootOutput(stored, documents);
         }
         this.runs.delete(runId);
-        await this.maybeFinalizeOwner(run, endedAt);
+        void this.maybeFinalizeOwner(run, endedAt);
     }
     async handleRetrieverError(error, runId) {
         const run = this.runs.get(runId);
@@ -1120,9 +1188,9 @@ class LemmaLangChainCallbackHandler {
                 this.noteRootError(stored, message);
         }
         this.runs.delete(runId);
-        await this.maybeFinalizeOwner(run, endedAt);
+        void this.maybeFinalizeOwner(run, endedAt);
     }
-    /** Await outstanding terminal deliveries. */
+    /** Send completed traces and await ingest delivery. */
     async flush() {
         await Promise.all([
             ...Array.from(this.traces.entries(), ([id, stored]) => this.finalizeTrace(id, stored)),