@uselemma/tracing 4.2.0 → 4.3.0

package/README.md

@@ -213,6 +213,40 @@ enableDebugMode();
 Use `openAIAgents({ recordInputs: false, recordOutputs: false })` to avoid
 sending prompts, tool inputs, tool outputs, or model output text.
 
+## LangChain and LangGraph
+
+Pass `langChain()` as a LangChain callback handler. The handler creates one
+Lemma trace for the root run, records LLM calls as generations, tools as tool
+spans, retrievers as spans with retrieval documents, and nested chains as child
+spans.
+
+```typescript
+import { ChatOpenAI } from "@langchain/openai";
+import { langChain } from "@uselemma/tracing";
+
+const model = new ChatOpenAI({
+  model: "gpt-4o",
+  callbacks: [langChain({ agentName: "support-agent" })],
+});
+
+const response = await model.invoke(userMessage);
+```
+
+LangGraph uses LangChain callbacks too:
+
+```typescript
+import { langGraph } from "@uselemma/tracing";
+
+const result = await graph.invoke(
+  { input: userMessage },
+  { callbacks: [langGraph({ agentName: "support-graph" })] },
+);
+```
+
+Use `langChain({ recordInputs: false, recordOutputs: false })` or
+`langGraph({ recordInputs: false, recordOutputs: false })` to avoid sending
+prompts, tool inputs, tool outputs, retrieved documents, or generated text.
+
 When a helper only has IDs, use the client-level methods:
 
 ```typescript
@@ -304,7 +338,8 @@ Use `attributes` for raw attributes that do not yet have a native SDK prop.
 ## Examples
 
 See [`examples/`](./examples) for complete callback tracing, trace handle,
-record-by-ID, Vercel AI SDK v6/v7, and OpenAI Agents SDK examples.
+record-by-ID, Vercel AI SDK v6/v7, OpenAI Agents SDK, LangChain, and LangGraph
+examples.
 
 ## License
 

package/dist/index.d.ts

@@ -1,5 +1,6 @@
 export { Lemma, NoopSpanHandle, SpanHandle, TraceContext, TraceHandle, active, type DetachedGenerationOptions, type DetachedSpanOptions, type DetachedToolOptions, type GenerationOptions, type JsonValue, type LemmaClientOptions, type SpanOptions, type ToolOptions, type TraceEndOptions, type TraceOptions, } from "./client";
 export { disableDebugMode, enableDebugMode, isDebugModeEnabled, lemmaDebug, } from "./debug-mode";
+export { LangChainCallbackHandler, LemmaLangChainCallbackHandler, langChain, langGraph, type LangChainIntegrationOptions, type LemmaLangChainIntegrationOptions, type LemmaLangGraphIntegrationOptions, } from "./langchain";
 export { openAIAgents, type OpenAIAgentsIntegrationOptions, type OpenAIAgentsSpan, type OpenAIAgentsSpanData, type OpenAIAgentsTrace, type OpenAIAgentsTracingProcessor, } from "./openai-agents";
 export { vercelAI, type LemmaVercelAIIntegrationOptions, type VercelAIIntegrationOptions, type VercelAITelemetryIntegration, } from "./vercel-ai";
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,KAAK,EACL,cAAc,EACd,UAAU,EACV,YAAY,EACZ,WAAW,EACX,MAAM,EACN,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,GAClB,MAAM,UAAU,CAAC;AAClB,OAAO,EACL,gBAAgB,EAChB,eAAe,EACf,kBAAkB,EAClB,UAAU,GACX,MAAM,cAAc,CAAC;AACtB,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,MAAM,EACN,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,GAClB,MAAM,UAAU,CAAC;AAClB,OAAO,EACL,gBAAgB,EAChB,eAAe,EACf,kBAAkB,EAClB,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,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,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.vercelAI = exports.openAIAgents = exports.lemmaDebug = exports.isDebugModeEnabled = exports.enableDebugMode = exports.disableDebugMode = exports.active = exports.TraceHandle = exports.TraceContext = exports.SpanHandle = exports.NoopSpanHandle = exports.Lemma = void 0;
+exports.vercelAI = exports.openAIAgents = exports.langGraph = exports.langChain = exports.LemmaLangChainCallbackHandler = exports.LangChainCallbackHandler = exports.lemmaDebug = exports.isDebugModeEnabled = exports.enableDebugMode = exports.disableDebugMode = exports.active = 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; } });
@@ -13,6 +13,11 @@ Object.defineProperty(exports, "disableDebugMode", { enumerable: true, get: func
 Object.defineProperty(exports, "enableDebugMode", { enumerable: true, get: function () { return debug_mode_1.enableDebugMode; } });
 Object.defineProperty(exports, "isDebugModeEnabled", { enumerable: true, get: function () { return debug_mode_1.isDebugModeEnabled; } });
 Object.defineProperty(exports, "lemmaDebug", { enumerable: true, get: function () { return debug_mode_1.lemmaDebug; } });
+var langchain_1 = require("./langchain");
+Object.defineProperty(exports, "LangChainCallbackHandler", { enumerable: true, get: function () { return langchain_1.LangChainCallbackHandler; } });
+Object.defineProperty(exports, "LemmaLangChainCallbackHandler", { enumerable: true, get: function () { return langchain_1.LemmaLangChainCallbackHandler; } });
+Object.defineProperty(exports, "langChain", { enumerable: true, get: function () { return langchain_1.langChain; } });
+Object.defineProperty(exports, "langGraph", { enumerable: true, get: function () { return langchain_1.langGraph; } });
 var openai_agents_1 = require("./openai-agents");
 Object.defineProperty(exports, "openAIAgents", { enumerable: true, get: function () { return openai_agents_1.openAIAgents; } });
 var vercel_ai_1 = require("./vercel-ai");

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;AACX,gGAAA,MAAM,OAAA;AAYR,2CAKsB;AAJpB,8GAAA,gBAAgB,OAAA;AAChB,6GAAA,eAAe,OAAA;AACf,gHAAA,kBAAkB,OAAA;AAClB,wGAAA,UAAU,OAAA;AAEZ,iDAOyB;AANvB,6GAAA,YAAY,OAAA;AAOd,yCAKqB;AAJnB,qGAAA,QAAQ,OAAA"}
+{"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;AACX,gGAAA,MAAM,OAAA;AAYR,2CAKsB;AAJpB,8GAAA,gBAAgB,OAAA;AAChB,6GAAA,eAAe,OAAA;AACf,gHAAA,kBAAkB,OAAA;AAClB,wGAAA,UAAU,OAAA;AAEZ,yCAQqB;AAPnB,qHAAA,wBAAwB,OAAA;AACxB,0HAAA,6BAA6B,OAAA;AAC7B,sGAAA,SAAS,OAAA;AACT,sGAAA,SAAS,OAAA;AAKX,iDAOyB;AANvB,6GAAA,YAAY,OAAA;AAOd,yCAKqB;AAJnB,qGAAA,QAAQ,OAAA"}

package/dist/langchain.d.ts

@@ -0,0 +1,55 @@
+import { Lemma, type LemmaClientOptions } from "./client";
+type RunId = string;
+type Serialized = {
+    id?: string[];
+    name?: string;
+    kwargs?: Record<string, unknown>;
+    [key: string]: unknown;
+};
+type LLMResult = {
+    generations?: unknown[][];
+    llmOutput?: Record<string, unknown>;
+    [key: string]: unknown;
+};
+export type LangChainIntegrationOptions = {
+    lemma?: Lemma;
+    apiKey?: LemmaClientOptions["apiKey"];
+    projectId?: LemmaClientOptions["projectId"];
+    baseUrl?: LemmaClientOptions["baseUrl"];
+    fetch?: LemmaClientOptions["fetch"];
+    agentName?: string;
+    metadata?: Record<string, unknown>;
+    recordInputs?: boolean;
+    recordOutputs?: boolean;
+};
+export declare class LemmaLangChainCallbackHandler {
+    private readonly options;
+    name: string;
+    private lemma;
+    private readonly runs;
+    constructor(options?: LangChainIntegrationOptions);
+    private getLemma;
+    private traceName;
+    private startTrace;
+    private parent;
+    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>;
+    handleChainError(error: unknown, runId: RunId): Promise<void>;
+    handleLLMStart(serialized: Serialized, prompts: string[], runId: RunId, parentRunId?: RunId, extraParams?: Record<string, unknown>): void;
+    handleChatModelStart(serialized: Serialized, messages: unknown[][], runId: RunId, parentRunId?: RunId, extraParams?: Record<string, unknown>): void;
+    handleLLMEnd(output: LLMResult, runId: RunId): void;
+    handleLLMError(error: unknown, runId: RunId): void;
+    handleToolStart(serialized: Serialized, input: unknown, runId: RunId, parentRunId?: RunId): void;
+    handleToolEnd(output: unknown, runId: RunId): void;
+    handleToolError(error: unknown, runId: RunId): void;
+    handleRetrieverStart(serialized: Serialized, query: string, runId: RunId, parentRunId?: RunId): void;
+    handleRetrieverEnd(documents: unknown[], runId: RunId): void;
+    handleRetrieverError(error: unknown, runId: RunId): void;
+}
+export declare const LangChainCallbackHandler: typeof LemmaLangChainCallbackHandler;
+export declare function langChain(options?: LangChainIntegrationOptions): LemmaLangChainCallbackHandler;
+export declare function langGraph(options?: LangChainIntegrationOptions): LemmaLangChainCallbackHandler;
+export type LemmaLangChainIntegrationOptions = LangChainIntegrationOptions;
+export type LemmaLangGraphIntegrationOptions = LangChainIntegrationOptions;
+export {};
+//# sourceMappingURL=langchain.d.ts.map

package/dist/langchain.d.ts.map

@@ -0,0 +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;AAElB,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,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,YAAY,CAAC,EAAE,OAAO,CAAC;IACvB,aAAa,CAAC,EAAE,OAAO,CAAC;CACzB,CAAC;AAmFF,qBAAa,6BAA6B;IAK5B,OAAO,CAAC,QAAQ,CAAC,OAAO;IAJpC,IAAI,SAAW;IACf,OAAO,CAAC,KAAK,CAAoB;IACjC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAA+B;gBAEvB,OAAO,GAAE,2BAAgC;IAItE,OAAO,CAAC,QAAQ;IAUhB,OAAO,CAAC,SAAS;IAIjB,OAAO,CAAC,UAAU;IAoBlB,OAAO,CAAC,MAAM;IAKd,gBAAgB,CACd,UAAU,EAAE,UAAU,EACtB,MAAM,EAAE,OAAO,EACf,KAAK,EAAE,KAAK,EACZ,WAAW,CAAC,EAAE,KAAK,EACnB,KAAK,CAAC,EAAE,MAAM,EAAE,EAChB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAClC,QAAQ,CAAC,EAAE,MAAM,EACjB,IAAI,CAAC,EAAE,MAAM;IA0BT,cAAc,CAAC,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK;IAgB7C,gBAAgB,CAAC,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK;IAanD,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;IA2BvC,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;IA0BvC,YAAY,CAAC,MAAM,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK;IAe5C,cAAc,CAAC,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK;IAM3C,eAAe,CACb,UAAU,EAAE,UAAU,EACtB,KAAK,EAAE,OAAO,EACd,KAAK,EAAE,KAAK,EACZ,WAAW,CAAC,EAAE,KAAK;IAsBrB,aAAa,CAAC,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK;IAQ3C,eAAe,CAAC,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK;IAM5C,oBAAoB,CAClB,UAAU,EAAE,UAAU,EACtB,KAAK,EAAE,MAAM,EACb,KAAK,EAAE,KAAK,EACZ,WAAW,CAAC,EAAE,KAAK;IAoBrB,kBAAkB,CAAC,SAAS,EAAE,OAAO,EAAE,EAAE,KAAK,EAAE,KAAK;IAUrD,oBAAoB,CAAC,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK;CAKlD;AAED,eAAO,MAAM,wBAAwB,sCAAgC,CAAC;AAEtE,wBAAgB,SAAS,CAAC,OAAO,GAAE,2BAAgC,iCAElE;AAED,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

@@ -0,0 +1,295 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.LangChainCallbackHandler = exports.LemmaLangChainCallbackHandler = void 0;
+exports.langChain = langChain;
+exports.langGraph = langGraph;
+const client_1 = require("./client");
+function serializedName(serialized, fallback) {
+    if (typeof serialized?.name === "string" && serialized.name) {
+        return serialized.name;
+    }
+    const id = serialized?.id;
+    if (Array.isArray(id) && id.length > 0) {
+        return String(id[id.length - 1]);
+    }
+    return fallback;
+}
+function modelName(serialized) {
+    const kwargs = serialized?.kwargs;
+    const value = kwargs?.model ??
+        kwargs?.modelName ??
+        kwargs?.model_name ??
+        kwargs?.model_id ??
+        serialized?.model ??
+        serialized?.modelName ??
+        serialized?.model_name ??
+        serialized?.model_id;
+    return typeof value === "string" ? value : undefined;
+}
+function firstText(value) {
+    if (typeof value === "string")
+        return value;
+    if (!value || typeof value !== "object")
+        return undefined;
+    const record = value;
+    if (typeof record.text === "string")
+        return record.text;
+    if (typeof record.content === "string")
+        return record.content;
+    const message = record.message;
+    if (message && typeof message === "object") {
+        return firstText(message);
+    }
+    return undefined;
+}
+function llmOutput(result) {
+    const generations = result.generations;
+    if (!Array.isArray(generations))
+        return result;
+    const text = generations.flat().map(firstText).filter(Boolean).join("");
+    return text || generations;
+}
+function usage(result) {
+    const output = result.llmOutput ?? {};
+    const tokenUsage = output.tokenUsage ??
+        output.token_usage ??
+        output.usage ??
+        output.estimatedTokenUsage;
+    if (!tokenUsage || typeof tokenUsage !== "object")
+        return undefined;
+    const usageData = tokenUsage;
+    const inputTokens = usageData.promptTokens ??
+        usageData.prompt_tokens ??
+        usageData.inputTokens ??
+        usageData.input_tokens;
+    const outputTokens = usageData.completionTokens ??
+        usageData.completion_tokens ??
+        usageData.outputTokens ??
+        usageData.output_tokens;
+    return {
+        inputTokens: typeof inputTokens === "number" ? inputTokens : undefined,
+        outputTokens: typeof outputTokens === "number" ? outputTokens : undefined,
+    };
+}
+function errorMessage(error) {
+    return error instanceof Error ? error.message : String(error);
+}
+class LemmaLangChainCallbackHandler {
+    options;
+    name = "lemma";
+    lemma;
+    runs = new Map();
+    constructor(options = {}) {
+        this.options = options;
+        this.lemma = options.lemma;
+    }
+    getLemma() {
+        this.lemma ??= new client_1.Lemma({
+            apiKey: this.options.apiKey,
+            projectId: this.options.projectId,
+            baseUrl: this.options.baseUrl,
+            fetch: this.options.fetch,
+        });
+        return this.lemma;
+    }
+    traceName(serialized, fallback) {
+        return this.options.agentName ?? serializedName(serialized, fallback);
+    }
+    startTrace(runId, serialized, input, fallbackName, metadata) {
+        const trace = this.getLemma().trace({
+            name: this.traceName(serialized, fallbackName),
+            input: this.options.recordInputs === false ? undefined : input,
+            metadata: {
+                ...this.options.metadata,
+                ...(metadata ?? {}),
+                langchainRunId: runId,
+            },
+        });
+        this.runs.set(runId, { trace, type: "chain" });
+        return trace;
+    }
+    parent(runId) {
+        if (!runId)
+            return undefined;
+        return this.runs.get(runId);
+    }
+    handleChainStart(serialized, inputs, runId, parentRunId, _tags, metadata, _runType, name) {
+        const parent = this.parent(parentRunId);
+        if (!parent) {
+            this.startTrace(runId, { ...serialized, name: name ?? serialized?.name }, inputs, "langchain-run", metadata);
+            return;
+        }
+        const handle = (parent.handle ?? parent.trace)?.startSpan({
+            name: name ?? serializedName(serialized, "langchain-chain"),
+            input: this.options.recordInputs === false ? undefined : inputs,
+            metadata: this.options.metadata,
+            attributes: {
+                "langchain.run_id": runId,
+                "langchain.parent_run_id": parentRunId,
+                "langchain.run_type": "chain",
+            },
+        });
+        this.runs.set(runId, { handle, parentRunId, type: "chain" });
+    }
+    async handleChainEnd(outputs, runId) {
+        const run = this.runs.get(runId);
+        if (!run)
+            return;
+        if (run.trace) {
+            await run.trace.end(this.options.recordOutputs === false ? undefined : { output: outputs });
+            this.runs.delete(runId);
+            return;
+        }
+        run.handle?.end({
+            output: this.options.recordOutputs === false ? undefined : outputs,
+        });
+        this.runs.delete(runId);
+    }
+    async handleChainError(error, runId) {
+        const run = this.runs.get(runId);
+        if (!run)
+            return;
+        if (run.trace) {
+            run.trace.fail(error);
+            await run.trace.end();
+            this.runs.delete(runId);
+            return;
+        }
+        run.handle?.end({ status: "ERROR", error: errorMessage(error) });
+        this.runs.delete(runId);
+    }
+    handleLLMStart(serialized, prompts, runId, parentRunId, extraParams) {
+        const parent = this.parent(parentRunId);
+        const trace = parent?.trace ??
+            parent?.handle ??
+            this.startTrace(runId, serialized, prompts, "langchain-llm");
+        const handle = trace.startGeneration({
+            name: serializedName(serialized, "langchain-llm"),
+            input: this.options.recordInputs === false ? undefined : prompts,
+            metadata: this.options.metadata,
+            model: modelName(serialized),
+            llmProvider: "langchain",
+            llmInputMessages: this.options.recordInputs === false
+                ? undefined
+                : prompts.map((content) => ({ role: "user", content })),
+            llmInvocationParameters: extraParams,
+            attributes: {
+                "langchain.run_id": runId,
+                "langchain.parent_run_id": parentRunId,
+                "langchain.run_type": "llm",
+            },
+        });
+        this.runs.set(runId, { handle, parentRunId, type: "llm" });
+    }
+    handleChatModelStart(serialized, messages, runId, parentRunId, extraParams) {
+        const flatMessages = messages.flat();
+        const parent = this.parent(parentRunId);
+        const trace = parent?.trace ??
+            parent?.handle ??
+            this.startTrace(runId, serialized, flatMessages, "langchain-chat-model");
+        const handle = trace.startGeneration({
+            name: serializedName(serialized, "langchain-chat-model"),
+            input: this.options.recordInputs === false ? undefined : flatMessages,
+            metadata: this.options.metadata,
+            model: modelName(serialized),
+            llmProvider: "langchain",
+            llmInputMessages: this.options.recordInputs === false ? undefined : flatMessages,
+            llmInvocationParameters: extraParams,
+            attributes: {
+                "langchain.run_id": runId,
+                "langchain.parent_run_id": parentRunId,
+                "langchain.run_type": "llm",
+            },
+        });
+        this.runs.set(runId, { handle, parentRunId, type: "llm" });
+    }
+    handleLLMEnd(output, runId) {
+        const run = this.runs.get(runId);
+        if (!run?.handle)
+            return;
+        const outputText = llmOutput(output);
+        run.handle.end({
+            output: this.options.recordOutputs === false ? undefined : outputText,
+            usage: usage(output),
+            llmOutputMessages: this.options.recordOutputs === false || outputText === undefined
+                ? undefined
+                : [{ role: "assistant", content: outputText }],
+        });
+        this.runs.delete(runId);
+    }
+    handleLLMError(error, runId) {
+        const run = this.runs.get(runId);
+        run?.handle?.end({ status: "ERROR", error: errorMessage(error) });
+        this.runs.delete(runId);
+    }
+    handleToolStart(serialized, input, runId, parentRunId) {
+        const parent = this.parent(parentRunId);
+        const trace = parent?.trace ??
+            parent?.handle ??
+            this.startTrace(runId, serialized, input, "langchain-tool");
+        const name = serializedName(serialized, "langchain-tool");
+        const handle = trace.startTool({
+            name,
+            toolName: name,
+            input: this.options.recordInputs === false ? undefined : input,
+            metadata: this.options.metadata,
+            attributes: {
+                "langchain.run_id": runId,
+                "langchain.parent_run_id": parentRunId,
+                "langchain.run_type": "tool",
+            },
+        });
+        this.runs.set(runId, { handle, parentRunId, type: "tool" });
+    }
+    handleToolEnd(output, runId) {
+        const run = this.runs.get(runId);
+        run?.handle?.end({
+            output: this.options.recordOutputs === false ? undefined : output,
+        });
+        this.runs.delete(runId);
+    }
+    handleToolError(error, runId) {
+        const run = this.runs.get(runId);
+        run?.handle?.end({ status: "ERROR", error: errorMessage(error) });
+        this.runs.delete(runId);
+    }
+    handleRetrieverStart(serialized, query, runId, parentRunId) {
+        const parent = this.parent(parentRunId);
+        const trace = parent?.trace ??
+            parent?.handle ??
+            this.startTrace(runId, serialized, query, "langchain-retriever");
+        const handle = trace.startSpan({
+            name: serializedName(serialized, "langchain-retriever"),
+            input: this.options.recordInputs === false ? undefined : query,
+            metadata: this.options.metadata,
+            attributes: {
+                "langchain.run_id": runId,
+                "langchain.parent_run_id": parentRunId,
+                "langchain.run_type": "retriever",
+            },
+        });
+        this.runs.set(runId, { handle, parentRunId, type: "retriever" });
+    }
+    handleRetrieverEnd(documents, runId) {
+        const run = this.runs.get(runId);
+        run?.handle?.end({
+            output: this.options.recordOutputs === false ? undefined : documents,
+            retrievalDocuments: this.options.recordOutputs === false ? undefined : documents,
+        });
+        this.runs.delete(runId);
+    }
+    handleRetrieverError(error, runId) {
+        const run = this.runs.get(runId);
+        run?.handle?.end({ status: "ERROR", error: errorMessage(error) });
+        this.runs.delete(runId);
+    }
+}
+exports.LemmaLangChainCallbackHandler = LemmaLangChainCallbackHandler;
+exports.LangChainCallbackHandler = LemmaLangChainCallbackHandler;
+function langChain(options = {}) {
+    return new LemmaLangChainCallbackHandler(options);
+}
+function langGraph(options = {}) {
+    return langChain({ agentName: "langgraph-agent", ...options });
+}
+//# sourceMappingURL=langchain.js.map

package/dist/langchain.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"langchain.js","sourceRoot":"","sources":["../src/langchain.ts"],"names":[],"mappings":";;;AAyYA,8BAEC;AAED,8BAEC;AA/YD,qCAKkB;AAoClB,SAAS,cAAc,CAAC,UAAkC,EAAE,QAAgB;IAC1E,IAAI,OAAO,UAAU,EAAE,IAAI,KAAK,QAAQ,IAAI,UAAU,CAAC,IAAI,EAAE,CAAC;QAC5D,OAAO,UAAU,CAAC,IAAI,CAAC;IACzB,CAAC;IACD,MAAM,EAAE,GAAG,UAAU,EAAE,EAAE,CAAC;IAC1B,IAAI,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACvC,OAAO,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC;IACnC,CAAC;IACD,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,SAAS,SAAS,CAAC,UAAkC;IACnD,MAAM,MAAM,GAAG,UAAU,EAAE,MAAM,CAAC;IAClC,MAAM,KAAK,GACT,MAAM,EAAE,KAAK;QACb,MAAM,EAAE,SAAS;QACjB,MAAM,EAAE,UAAU;QAClB,MAAM,EAAE,QAAQ;QAChB,UAAU,EAAE,KAAK;QACjB,UAAU,EAAE,SAAS;QACrB,UAAU,EAAE,UAAU;QACtB,UAAU,EAAE,QAAQ,CAAC;IACvB,OAAO,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC;AACvD,CAAC;AAED,SAAS,SAAS,CAAC,KAAc;IAC/B,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,KAAK,CAAC;IAC5C,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,SAAS,CAAC;IAC1D,MAAM,MAAM,GAAG,KAAgC,CAAC;IAChD,IAAI,OAAO,MAAM,CAAC,IAAI,KAAK,QAAQ;QAAE,OAAO,MAAM,CAAC,IAAI,CAAC;IACxD,IAAI,OAAO,MAAM,CAAC,OAAO,KAAK,QAAQ;QAAE,OAAO,MAAM,CAAC,OAAO,CAAC;IAC9D,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC;IAC/B,IAAI,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;QAC3C,OAAO,SAAS,CAAC,OAAO,CAAC,CAAC;IAC5B,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,SAAS,CAAC,MAAiB;IAClC,MAAM,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC;IACvC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,WAAW,CAAC;QAAE,OAAO,MAAM,CAAC;IAC/C,MAAM,IAAI,GAAG,WAAW,CAAC,IAAI,EAAE,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACxE,OAAO,IAAI,IAAI,WAAW,CAAC;AAC7B,CAAC;AAED,SAAS,KAAK,CAAC,MAAiB;IAC9B,MAAM,MAAM,GAAG,MAAM,CAAC,SAAS,IAAI,EAAE,CAAC;IACtC,MAAM,UAAU,GACd,MAAM,CAAC,UAAU;QACjB,MAAM,CAAC,WAAW;QAClB,MAAM,CAAC,KAAK;QACZ,MAAM,CAAC,mBAAmB,CAAC;IAC7B,IAAI,CAAC,UAAU,IAAI,OAAO,UAAU,KAAK,QAAQ;QAAE,OAAO,SAAS,CAAC;IACpE,MAAM,SAAS,GAAG,UAAqC,CAAC;IACxD,MAAM,WAAW,GACf,SAAS,CAAC,YAAY;QACtB,SAAS,CAAC,aAAa;QACvB,SAAS,CAAC,WAAW;QACrB,SAAS,CAAC,YAAY,CAAC;IACzB,MAAM,YAAY,GAChB,SAAS,CAAC,gBAAgB;QAC1B,SAAS,CAAC,iBAAiB;QAC3B,SAAS,CAAC,YAAY;QACtB,SAAS,CAAC,aAAa,CAAC;IAC1B,OAAO;QACL,WAAW,EAAE,OAAO,WAAW,KAAK,QAAQ,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,SAAS;QACtE,YAAY,EAAE,OAAO,YAAY,KAAK,QAAQ,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,SAAS;KAC1E,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CAAC,KAAc;IAClC,OAAO,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AAChE,CAAC;AAED,MAAa,6BAA6B;IAKX;IAJ7B,IAAI,GAAG,OAAO,CAAC;IACP,KAAK,CAAoB;IAChB,IAAI,GAAG,IAAI,GAAG,EAAoB,CAAC;IAEpD,YAA6B,UAAuC,EAAE;QAAzC,YAAO,GAAP,OAAO,CAAkC;QACpE,IAAI,CAAC,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;IAC7B,CAAC;IAEO,QAAQ;QACd,IAAI,CAAC,KAAK,KAAK,IAAI,cAAK,CAAC;YACvB,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM;YAC3B,SAAS,EAAE,IAAI,CAAC,OAAO,CAAC,SAAS;YACjC,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO;YAC7B,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,KAAK;SAC1B,CAAC,CAAC;QACH,OAAO,IAAI,CAAC,KAAK,CAAC;IACpB,CAAC;IAEO,SAAS,CAAC,UAAkC,EAAE,QAAgB;QACpE,OAAO,IAAI,CAAC,OAAO,CAAC,SAAS,IAAI,cAAc,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;IACxE,CAAC;IAEO,UAAU,CAChB,KAAY,EACZ,UAAkC,EAClC,KAAc,EACd,YAAoB,EACpB,QAAkC;QAElC,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC,KAAK,CAAC;YAClC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,UAAU,EAAE,YAAY,CAAC;YAC9C,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,YAAY,KAAK,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK;YAC9D,QAAQ,EAAE;gBACR,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ;gBACxB,GAAG,CAAC,QAAQ,IAAI,EAAE,CAAC;gBACnB,cAAc,EAAE,KAAK;aACtB;SACF,CAAC,CAAC;QACH,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;QAC/C,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,MAAM,CAAC,KAAwB;QACrC,IAAI,CAAC,KAAK;YAAE,OAAO,SAAS,CAAC;QAC7B,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IAC9B,CAAC;IAED,gBAAgB,CACd,UAAsB,EACtB,MAAe,EACf,KAAY,EACZ,WAAmB,EACnB,KAAgB,EAChB,QAAkC,EAClC,QAAiB,EACjB,IAAa;QAEb,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;QACxC,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,IAAI,CAAC,UAAU,CACb,KAAK,EACL,EAAE,GAAG,UAAU,EAAE,IAAI,EAAE,IAAI,IAAI,UAAU,EAAE,IAAI,EAAE,EACjD,MAAM,EACN,eAAe,EACf,QAAQ,CACT,CAAC;YACF,OAAO;QACT,CAAC;QACD,MAAM,MAAM,GAAG,CAAC,MAAM,CAAC,MAAM,IAAI,MAAM,CAAC,KAAK,CAAC,EAAE,SAAS,CAAC;YACxD,IAAI,EAAE,IAAI,IAAI,cAAc,CAAC,UAAU,EAAE,iBAAiB,CAAC;YAC3D,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,YAAY,KAAK,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM;YAC/D,QAAQ,EAAE,IAAI,CAAC,OAAO,CAAC,QAAQ;YAC/B,UAAU,EAAE;gBACV,kBAAkB,EAAE,KAAK;gBACzB,yBAAyB,EAAE,WAAW;gBACtC,oBAAoB,EAAE,OAAO;aAC9B;SACF,CAAC,CAAC;QACH,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;IAC/D,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,OAAgB,EAAE,KAAY;QACjD,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACjC,IAAI,CAAC,GAAG;YAAE,OAAO;QACjB,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;YACd,MAAM,GAAG,CAAC,KAAK,CAAC,GAAG,CACjB,IAAI,CAAC,OAAO,CAAC,aAAa,KAAK,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,CACvE,CAAC;YACF,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACxB,OAAO;QACT,CAAC;QACD,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC;YACd,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,aAAa,KAAK,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO;SACnE,CAAC,CAAC;QACH,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAC1B,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,KAAc,EAAE,KAAY;QACjD,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACjC,IAAI,CAAC,GAAG;YAAE,OAAO;QACjB,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;YACd,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACtB,MAAM,GAAG,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC;YACtB,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACxB,OAAO;QACT,CAAC;QACD,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QACjE,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAC1B,CAAC;IAED,cAAc,CACZ,UAAsB,EACtB,OAAiB,EACjB,KAAY,EACZ,WAAmB,EACnB,WAAqC;QAErC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;QACxC,MAAM,KAAK,GACT,MAAM,EAAE,KAAK;YACb,MAAM,EAAE,MAAM;YACd,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,UAAU,EAAE,OAAO,EAAE,eAAe,CAAC,CAAC;QAC/D,MAAM,MAAM,GAAG,KAAK,CAAC,eAAe,CAAC;YACnC,IAAI,EAAE,cAAc,CAAC,UAAU,EAAE,eAAe,CAAC;YACjD,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,YAAY,KAAK,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO;YAChE,QAAQ,EAAE,IAAI,CAAC,OAAO,CAAC,QAAQ;YAC/B,KAAK,EAAE,SAAS,CAAC,UAAU,CAAC;YAC5B,WAAW,EAAE,WAAW;YACxB,gBAAgB,EACd,IAAI,CAAC,OAAO,CAAC,YAAY,KAAK,KAAK;gBACjC,CAAC,CAAC,SAAS;gBACX,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC;YAC3D,uBAAuB,EAAE,WAAW;YACpC,UAAU,EAAE;gBACV,kBAAkB,EAAE,KAAK;gBACzB,yBAAyB,EAAE,WAAW;gBACtC,oBAAoB,EAAE,KAAK;aAC5B;SACF,CAAC,CAAC;QACH,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;IAC7D,CAAC;IAED,oBAAoB,CAClB,UAAsB,EACtB,QAAqB,EACrB,KAAY,EACZ,WAAmB,EACnB,WAAqC;QAErC,MAAM,YAAY,GAAG,QAAQ,CAAC,IAAI,EAAE,CAAC;QACrC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;QACxC,MAAM,KAAK,GACT,MAAM,EAAE,KAAK;YACb,MAAM,EAAE,MAAM;YACd,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,UAAU,EAAE,YAAY,EAAE,sBAAsB,CAAC,CAAC;QAC3E,MAAM,MAAM,GAAG,KAAK,CAAC,eAAe,CAAC;YACnC,IAAI,EAAE,cAAc,CAAC,UAAU,EAAE,sBAAsB,CAAC;YACxD,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,YAAY,KAAK,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,YAAY;YACrE,QAAQ,EAAE,IAAI,CAAC,OAAO,CAAC,QAAQ;YAC/B,KAAK,EAAE,SAAS,CAAC,UAAU,CAAC;YAC5B,WAAW,EAAE,WAAW;YACxB,gBAAgB,EACd,IAAI,CAAC,OAAO,CAAC,YAAY,KAAK,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,YAAY;YAChE,uBAAuB,EAAE,WAAW;YACpC,UAAU,EAAE;gBACV,kBAAkB,EAAE,KAAK;gBACzB,yBAAyB,EAAE,WAAW;gBACtC,oBAAoB,EAAE,KAAK;aAC5B;SACF,CAAC,CAAC;QACH,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;IAC7D,CAAC;IAED,YAAY,CAAC,MAAiB,EAAE,KAAY;QAC1C,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACjC,IAAI,CAAC,GAAG,EAAE,MAAM;YAAE,OAAO;QACzB,MAAM,UAAU,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC;QACrC,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC;YACb,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,aAAa,KAAK,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,UAAU;YACrE,KAAK,EAAE,KAAK,CAAC,MAAM,CAAC;YACpB,iBAAiB,EACf,IAAI,CAAC,OAAO,CAAC,aAAa,KAAK,KAAK,IAAI,UAAU,KAAK,SAAS;gBAC9D,CAAC,CAAC,SAAS;gBACX,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC;SACnD,CAAC,CAAC;QACH,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAC1B,CAAC;IAED,cAAc,CAAC,KAAc,EAAE,KAAY;QACzC,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACjC,GAAG,EAAE,MAAM,EAAE,GAAG,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAClE,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAC1B,CAAC;IAED,eAAe,CACb,UAAsB,EACtB,KAAc,EACd,KAAY,EACZ,WAAmB;QAEnB,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;QACxC,MAAM,KAAK,GACT,MAAM,EAAE,KAAK;YACb,MAAM,EAAE,MAAM;YACd,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,UAAU,EAAE,KAAK,EAAE,gBAAgB,CAAC,CAAC;QAC9D,MAAM,IAAI,GAAG,cAAc,CAAC,UAAU,EAAE,gBAAgB,CAAC,CAAC;QAC1D,MAAM,MAAM,GAAG,KAAK,CAAC,SAAS,CAAC;YAC7B,IAAI;YACJ,QAAQ,EAAE,IAAI;YACd,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,YAAY,KAAK,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK;YAC9D,QAAQ,EAAE,IAAI,CAAC,OAAO,CAAC,QAAQ;YAC/B,UAAU,EAAE;gBACV,kBAAkB,EAAE,KAAK;gBACzB,yBAAyB,EAAE,WAAW;gBACtC,oBAAoB,EAAE,MAAM;aAC7B;SACF,CAAC,CAAC;QACH,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;IAC9D,CAAC;IAED,aAAa,CAAC,MAAe,EAAE,KAAY;QACzC,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACjC,GAAG,EAAE,MAAM,EAAE,GAAG,CAAC;YACf,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,aAAa,KAAK,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM;SAClE,CAAC,CAAC;QACH,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAC1B,CAAC;IAED,eAAe,CAAC,KAAc,EAAE,KAAY;QAC1C,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACjC,GAAG,EAAE,MAAM,EAAE,GAAG,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAClE,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAC1B,CAAC;IAED,oBAAoB,CAClB,UAAsB,EACtB,KAAa,EACb,KAAY,EACZ,WAAmB;QAEnB,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;QACxC,MAAM,KAAK,GACT,MAAM,EAAE,KAAK;YACb,MAAM,EAAE,MAAM;YACd,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,UAAU,EAAE,KAAK,EAAE,qBAAqB,CAAC,CAAC;QACnE,MAAM,MAAM,GAAG,KAAK,CAAC,SAAS,CAAC;YAC7B,IAAI,EAAE,cAAc,CAAC,UAAU,EAAE,qBAAqB,CAAC;YACvD,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,YAAY,KAAK,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK;YAC9D,QAAQ,EAAE,IAAI,CAAC,OAAO,CAAC,QAAQ;YAC/B,UAAU,EAAE;gBACV,kBAAkB,EAAE,KAAK;gBACzB,yBAAyB,EAAE,WAAW;gBACtC,oBAAoB,EAAE,WAAW;aAClC;SACF,CAAC,CAAC;QACH,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC,CAAC;IACnE,CAAC;IAED,kBAAkB,CAAC,SAAoB,EAAE,KAAY;QACnD,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACjC,GAAG,EAAE,MAAM,EAAE,GAAG,CAAC;YACf,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,aAAa,KAAK,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;YACpE,kBAAkB,EAChB,IAAI,CAAC,OAAO,CAAC,aAAa,KAAK,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;SAC/D,CAAC,CAAC;QACH,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAC1B,CAAC;IAED,oBAAoB,CAAC,KAAc,EAAE,KAAY;QAC/C,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACjC,GAAG,EAAE,MAAM,EAAE,GAAG,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAClE,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAC1B,CAAC;CACF;AAlRD,sEAkRC;AAEY,QAAA,wBAAwB,GAAG,6BAA6B,CAAC;AAEtE,SAAgB,SAAS,CAAC,UAAuC,EAAE;IACjE,OAAO,IAAI,6BAA6B,CAAC,OAAO,CAAC,CAAC;AACpD,CAAC;AAED,SAAgB,SAAS,CAAC,UAAuC,EAAE;IACjE,OAAO,SAAS,CAAC,EAAE,SAAS,EAAE,iBAAiB,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC;AACjE,CAAC"}

package/examples/langchain.ts

@@ -0,0 +1,15 @@
+import { ChatOpenAI } from "@langchain/openai";
+import { HumanMessage } from "@langchain/core/messages";
+import { langChain } from "@uselemma/tracing";
+
+const lemmaCallbacks = [langChain({ agentName: "support-agent" })];
+
+export async function callLangChain(userMessage: string) {
+  const model = new ChatOpenAI({
+    model: "gpt-4o",
+    callbacks: lemmaCallbacks,
+  });
+
+  const response = await model.invoke([new HumanMessage(userMessage)]);
+  return response.content;
+}

package/examples/langgraph.ts

@@ -0,0 +1,24 @@
+import { StateGraph, START, END } from "@langchain/langgraph";
+import { langGraph } from "@uselemma/tracing";
+
+type GraphState = {
+  input: string;
+  output?: string;
+};
+
+const graph = new StateGraph<GraphState>()
+  .addNode("answer", async (state) => ({
+    output: `You said: ${state.input}`,
+  }))
+  .addEdge(START, "answer")
+  .addEdge("answer", END)
+  .compile();
+
+export async function callLangGraph(userMessage: string) {
+  const result = await graph.invoke(
+    { input: userMessage },
+    { callbacks: [langGraph({ agentName: "support-graph" })] },
+  );
+
+  return result.output;
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@uselemma/tracing",
-  "version": "4.2.0",
+  "version": "4.3.0",
   "description": "HTTP tracing SDK for Lemma",
   "license": "MIT",
   "author": "Lemma",