@visibe.ai/node 0.1.17 → 0.1.18

package/dist/cjs/integrations/langchain.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.LangChainCallback = exports.activeLangChainStorage = void 0;
+exports.LangChainCallback = exports.LANGGRAPH_INTERNAL_NODES = exports.activeLangChainStorage = void 0;
 exports.patchRunnableSequence = patchRunnableSequence;
 const node_async_hooks_1 = require("node:async_hooks");
 const node_crypto_1 = require("node:crypto");
@@ -11,19 +11,19 @@ exports.activeLangChainStorage = new node_async_hooks_1.AsyncLocalStorage();
 // ---------------------------------------------------------------------------
 // LangChain token extraction
 // Different providers nest token usage in different locations.
-// Check in the order specified by the spec.
 // ---------------------------------------------------------------------------
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
 function extractTokenUsage(output) {
     const usage = output?.llmOutput?.tokenUsage
         ?? output?.llmOutput?.usage
         ?? output?.generations?.[0]?.[0]?.generationInfo?.usage;
-    // Use ?? not || so token counts of 0 are preserved correctly.
     return {
         inputTokens: usage?.promptTokens ?? usage?.input_tokens ?? 0,
         outputTokens: usage?.completionTokens ?? usage?.output_tokens ?? 0,
     };
 }
+// Internal LangGraph system node names — never emit agent_start spans for these.
+exports.LANGGRAPH_INTERNAL_NODES = new Set(['__start__', '__end__', 'LangGraph']);
 // ---------------------------------------------------------------------------
 // LangChainCallback
 // ---------------------------------------------------------------------------
@@ -34,23 +34,21 @@ class LangChainCallback {
     constructor(options) {
         // Required by @langchain/core v1+ for proper callback registration.
         // Without `name`, ensureHandler() wraps via fromMethods() which drops prototype methods.
-        // Without `awaitHandlers`, callbacks run in a background queue (p-queue) and fire
-        // after model.invoke() returns — causing spans to be missed on flush/completeTrace.
+        // Without `awaitHandlers`, callbacks run in a background queue and fire after
+        // model.invoke() returns — causing spans to be missed on flush/completeTrace.
         this.name = 'visibe-langchain-callback';
         this.awaitHandlers = true;
         this.raiseError = false;
         // Maps LangChain runId → our spanId so we can set parent_span_id.
         this.runIdToSpanId = new Map();
-        // Tracks start times so we can compute durationMs.
-        this.pendingLLMCalls = new Map(); // runId → startMs
+        // Pending LLM calls: runId → { startMs, model, inputText }
+        this.pendingLLMCalls = new Map();
         this.pendingToolCalls = new Map();
         this.stepCounter = 0;
-        // Agents we have already emitted agent_start spans for.
-        this.seenAgents = new Set();
-        // Token / call accumulators — updated by handleLLMEnd, read by patchCompiledStateGraph
-        // and patchRunnableSequence to populate completeTrace totals.
+        // Token / call / cost accumulators — read by patchCompiledStateGraph / patchRunnableSequence.
         this.totalInputTokens = 0;
         this.totalOutputTokens = 0;
+        this.totalCost = 0;
         this.llmCallCount = 0;
         this.visibe = options.visibe;
         this.traceId = options.traceId;
@@ -59,17 +57,40 @@ class LangChainCallback {
     // ---------------------------------------------------------------------------
     // LLM events
     // ---------------------------------------------------------------------------
+    // Called for text-completion models.
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    async handleLLMStart(_llm, _messages, runId) {
-        this.pendingLLMCalls.set(runId, Date.now());
+    async handleLLMStart(llm, prompts, runId) {
+        const model = llm?.kwargs?.model ?? llm?.kwargs?.model_name ?? undefined;
+        const inputText = Array.isArray(prompts) ? prompts.join('\n') : '';
+        this.pendingLLMCalls.set(runId, { startMs: Date.now(), model, inputText });
+    }
+    // Called for chat models (ChatOpenAI, ChatAnthropic, etc.).
+    // messages is BaseMessage[][] — one array per parallel completion request.
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    async handleChatModelStart(llm, messages, runId) {
+        const model = llm?.kwargs?.model ?? llm?.kwargs?.model_name ?? undefined;
+        let inputText = '';
+        try {
+            const msgs = messages?.[0] ?? [];
+            inputText = msgs.map((m) => {
+                const role = m?.getType?.() ?? m?._getType?.() ?? m?.role ?? 'user';
+                const content = typeof m?.content === 'string'
+                    ? m.content
+                    : JSON.stringify(m?.content ?? '');
+                return `${role}: ${content}`;
+            }).join('\n');
+        }
+        catch { /* ignore serialisation errors */ }
+        this.pendingLLMCalls.set(runId, { startMs: Date.now(), model, inputText });
     }
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
     async handleLLMEnd(output, runId, parentRunId) {
-        const startMs = this.pendingLLMCalls.get(runId) ?? Date.now();
+        const pending = this.pendingLLMCalls.get(runId) ?? { startMs: Date.now() };
         this.pendingLLMCalls.delete(runId);
         const { inputTokens, outputTokens } = extractTokenUsage(output);
         const gen = output?.generations?.[0]?.[0];
-        const model = gen?.generationInfo?.model ?? this.agentName;
+        // Prefer model saved at LLM-start, fall back to generationInfo, then agentName.
+        const model = pending.model ?? gen?.generationInfo?.model ?? this.agentName;
         const cost = (0, utils_1.calculateCost)(model, inputTokens, outputTokens);
         // eslint-disable-next-line @typescript-eslint/no-explicit-any
         const rawText = gen?.text ?? gen?.message?.content ?? '';
@@ -85,16 +106,15 @@ class LangChainCallback {
             status: 'success',
             inputTokens,
             outputTokens,
-            inputText: '', // LangChain doesn't surface the raw prompt here
+            inputText: pending.inputText ?? '',
             outputText,
-            durationMs: Date.now() - startMs,
+            durationMs: Date.now() - pending.startMs,
         });
         this.visibe.batcher.add(this.traceId, span);
-        // Update local accumulators (used by patchCompiledStateGraph / patchRunnableSequence).
         this.totalInputTokens += inputTokens;
         this.totalOutputTokens += outputTokens;
+        this.totalCost += cost;
         this.llmCallCount++;
-        // Notify track() accumulator if running inside a group tracker.
         this._onLLMSpan?.(inputTokens, outputTokens, cost);
     }
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
@@ -143,16 +163,15 @@ class LangChainCallback {
     }
     // ---------------------------------------------------------------------------
     // Chain events
+    //
+    // In LangGraph v1.2+, the node key is passed as the 8th `name` parameter, NOT
+    // via chain.name. We always store runId → spanId so nested LLM/tool calls get
+    // a valid parent_span_id regardless of whether the chain is a user-defined node.
     // ---------------------------------------------------------------------------
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    async handleChainStart(chain, _inputs, runId, parentRunId) {
-        // Emit an agent_start span the first time we see a named chain.
-        const chainName = chain?.id?.at(-1) ?? '';
-        if (chainName && !this.seenAgents.has(chainName)) {
-            this.seenAgents.add(chainName);
-            const spanId = this.nextSpanId();
-            this.runIdToSpanId.set(runId, spanId);
-            void parentRunId; // suppress unused warning
+    async handleChainStart(_chain, _inputs, runId, _parentRunId, _tags, _metadata, _runType, _name) {
+        if (!this.runIdToSpanId.has(runId)) {
+            this.runIdToSpanId.set(runId, this.nextSpanId());
         }
     }
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
@@ -179,7 +198,6 @@ function patchRunnableSequence(lcModule, visibe) {
     const originalStream = RunnableSequence.prototype.stream;
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
     RunnableSequence.prototype.invoke = async function (input, config) {
-        // If already inside a LangChain trace, pass through.
         if (exports.activeLangChainStorage.getStore() !== undefined) {
             return originalInvoke.call(this, input, config);
         }
@@ -194,6 +212,7 @@ function patchRunnableSequence(lcModule, visibe) {
             ...(visibe.sessionId ? { session_id: visibe.sessionId } : {}),
         });
         const cb = new LangChainCallback({ visibe, traceId, agentName: 'langchain' });
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
         let result;
         let status = 'completed';
         try {
@@ -210,6 +229,7 @@ function patchRunnableSequence(lcModule, visibe) {
                 ended_at: new Date().toISOString(),
                 duration_ms: Date.now() - startMs,
                 llm_call_count: cb.llmCallCount,
+                total_cost: cb.totalCost,
                 total_tokens: cb.totalInputTokens + cb.totalOutputTokens,
                 total_input_tokens: cb.totalInputTokens,
                 total_output_tokens: cb.totalOutputTokens,
@@ -236,9 +256,6 @@ function patchRunnableSequence(lcModule, visibe) {
         const cb = new LangChainCallback({ visibe, traceId, agentName: 'langchain' });
         let status = 'completed';
         try {
-            // RunnableSequence.stream() is an async function (not async generator) returning
-            // a Promise<AsyncIterable>. activeLangChainStorage.run returns that Promise,
-            // so we must await before yield*.
             const gen = await exports.activeLangChainStorage.run(cb, () => originalStream.call(this, input, _mergeCallbacks(config, cb)));
             yield* gen;
         }
@@ -253,6 +270,7 @@ function patchRunnableSequence(lcModule, visibe) {
                 ended_at: new Date().toISOString(),
                 duration_ms: Date.now() - startMs,
                 llm_call_count: cb.llmCallCount,
+                total_cost: cb.totalCost,
                 total_tokens: cb.totalInputTokens + cb.totalOutputTokens,
                 total_input_tokens: cb.totalInputTokens,
                 total_output_tokens: cb.totalOutputTokens,
@@ -267,7 +285,6 @@ function patchRunnableSequence(lcModule, visibe) {
 // ---------------------------------------------------------------------------
 // Private helpers
 // ---------------------------------------------------------------------------
-// Merge our callback into an existing LangChain config object.
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
 function _mergeCallbacks(config, cb) {
     if (!config)

package/dist/cjs/integrations/langgraph.js

@@ -13,24 +13,26 @@ class LangGraphCallback extends langchain_1.LangChainCallback {
         super(options);
         this.nodeNames = new Set(options.nodeNames ?? []);
     }
-    // Override handleChainStart to emit agent_start for known graph nodes.
+    // Override handleChainStart to emit agent_start spans for LangGraph nodes.
+    //
+    // In LangGraph v1.2+, the node key is the 8th `name` parameter (not chain.name).
+    // Internal system nodes (__start__, __end__, LangGraph) are filtered out.
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    async handleChainStart(chain, inputs, runId, parentRunId) {
-        // LangGraph may surface the node key in chain.name (run name) or chain.id[-1] (class name).
-        const chainName = chain?.name ?? chain?.id?.at(-1) ?? '';
-        if (chainName && this.nodeNames.has(chainName)) {
-            const spanId = this.nextSpanId();
+    async handleChainStart(chain, inputs, runId, parentRunId, tags, metadata, runType, name) {
+        // Always track the runId so child LLM calls can resolve parent_span_id.
+        await super.handleChainStart(chain, inputs, runId, parentRunId, tags, metadata, runType, name);
+        // The node key is passed as the 8th `name` parameter in LangGraph v1.2+.
+        // Fall back to chain.name for older versions.
+        const nodeName = name ?? chain?.name ?? '';
+        if (nodeName && !langchain_1.LANGGRAPH_INTERNAL_NODES.has(nodeName)) {
+            // Use the spanId already assigned by super for this runId.
+            const spanId = this.runIdToSpanId.get(runId) ?? this.nextSpanId();
             this.runIdToSpanId.set(runId, spanId);
-            // Emit an agent_start span for this node.
-            // type MUST be exactly "agent_start" — the backend validates this string.
             this.visibe.batcher.add(this.traceId, this.visibe.buildAgentStartSpan({
                 spanId,
-                agentName: chainName,
+                agentName: nodeName,
             }));
-            // Don't call super — we've already set the runId mapping.
-            return;
         }
-        await super.handleChainStart(chain, inputs, runId, parentRunId);
     }
 }
 exports.LangGraphCallback = LangGraphCallback;
@@ -87,6 +89,7 @@ function patchCompiledStateGraph(lgModule, visibe) {
                 ended_at: new Date().toISOString(),
                 duration_ms: Date.now() - startMs,
                 llm_call_count: cb.llmCallCount,
+                total_cost: cb.totalCost,
                 total_tokens: cb.totalInputTokens + cb.totalOutputTokens,
                 total_input_tokens: cb.totalInputTokens,
                 total_output_tokens: cb.totalOutputTokens,
@@ -138,6 +141,7 @@ function patchCompiledStateGraph(lgModule, visibe) {
                 ended_at: new Date().toISOString(),
                 duration_ms: Date.now() - startMs,
                 llm_call_count: cb.llmCallCount,
+                total_cost: cb.totalCost,
                 total_tokens: cb.totalInputTokens + cb.totalOutputTokens,
                 total_input_tokens: cb.totalInputTokens,
                 total_output_tokens: cb.totalOutputTokens,

package/dist/esm/integrations/langchain.js

@@ -7,19 +7,19 @@ export const activeLangChainStorage = new AsyncLocalStorage();
 // ---------------------------------------------------------------------------
 // LangChain token extraction
 // Different providers nest token usage in different locations.
-// Check in the order specified by the spec.
 // ---------------------------------------------------------------------------
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
 function extractTokenUsage(output) {
     const usage = output?.llmOutput?.tokenUsage
         ?? output?.llmOutput?.usage
         ?? output?.generations?.[0]?.[0]?.generationInfo?.usage;
-    // Use ?? not || so token counts of 0 are preserved correctly.
     return {
         inputTokens: usage?.promptTokens ?? usage?.input_tokens ?? 0,
         outputTokens: usage?.completionTokens ?? usage?.output_tokens ?? 0,
     };
 }
+// Internal LangGraph system node names — never emit agent_start spans for these.
+export const LANGGRAPH_INTERNAL_NODES = new Set(['__start__', '__end__', 'LangGraph']);
 // ---------------------------------------------------------------------------
 // LangChainCallback
 // ---------------------------------------------------------------------------
@@ -30,23 +30,21 @@ export class LangChainCallback {
     constructor(options) {
         // Required by @langchain/core v1+ for proper callback registration.
         // Without `name`, ensureHandler() wraps via fromMethods() which drops prototype methods.
-        // Without `awaitHandlers`, callbacks run in a background queue (p-queue) and fire
-        // after model.invoke() returns — causing spans to be missed on flush/completeTrace.
+        // Without `awaitHandlers`, callbacks run in a background queue and fire after
+        // model.invoke() returns — causing spans to be missed on flush/completeTrace.
         this.name = 'visibe-langchain-callback';
         this.awaitHandlers = true;
         this.raiseError = false;
         // Maps LangChain runId → our spanId so we can set parent_span_id.
         this.runIdToSpanId = new Map();
-        // Tracks start times so we can compute durationMs.
-        this.pendingLLMCalls = new Map(); // runId → startMs
+        // Pending LLM calls: runId → { startMs, model, inputText }
+        this.pendingLLMCalls = new Map();
         this.pendingToolCalls = new Map();
         this.stepCounter = 0;
-        // Agents we have already emitted agent_start spans for.
-        this.seenAgents = new Set();
-        // Token / call accumulators — updated by handleLLMEnd, read by patchCompiledStateGraph
-        // and patchRunnableSequence to populate completeTrace totals.
+        // Token / call / cost accumulators — read by patchCompiledStateGraph / patchRunnableSequence.
         this.totalInputTokens = 0;
         this.totalOutputTokens = 0;
+        this.totalCost = 0;
         this.llmCallCount = 0;
         this.visibe = options.visibe;
         this.traceId = options.traceId;
@@ -55,17 +53,40 @@ export class LangChainCallback {
     // ---------------------------------------------------------------------------
     // LLM events
     // ---------------------------------------------------------------------------
+    // Called for text-completion models.
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    async handleLLMStart(_llm, _messages, runId) {
-        this.pendingLLMCalls.set(runId, Date.now());
+    async handleLLMStart(llm, prompts, runId) {
+        const model = llm?.kwargs?.model ?? llm?.kwargs?.model_name ?? undefined;
+        const inputText = Array.isArray(prompts) ? prompts.join('\n') : '';
+        this.pendingLLMCalls.set(runId, { startMs: Date.now(), model, inputText });
+    }
+    // Called for chat models (ChatOpenAI, ChatAnthropic, etc.).
+    // messages is BaseMessage[][] — one array per parallel completion request.
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    async handleChatModelStart(llm, messages, runId) {
+        const model = llm?.kwargs?.model ?? llm?.kwargs?.model_name ?? undefined;
+        let inputText = '';
+        try {
+            const msgs = messages?.[0] ?? [];
+            inputText = msgs.map((m) => {
+                const role = m?.getType?.() ?? m?._getType?.() ?? m?.role ?? 'user';
+                const content = typeof m?.content === 'string'
+                    ? m.content
+                    : JSON.stringify(m?.content ?? '');
+                return `${role}: ${content}`;
+            }).join('\n');
+        }
+        catch { /* ignore serialisation errors */ }
+        this.pendingLLMCalls.set(runId, { startMs: Date.now(), model, inputText });
     }
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
     async handleLLMEnd(output, runId, parentRunId) {
-        const startMs = this.pendingLLMCalls.get(runId) ?? Date.now();
+        const pending = this.pendingLLMCalls.get(runId) ?? { startMs: Date.now() };
         this.pendingLLMCalls.delete(runId);
         const { inputTokens, outputTokens } = extractTokenUsage(output);
         const gen = output?.generations?.[0]?.[0];
-        const model = gen?.generationInfo?.model ?? this.agentName;
+        // Prefer model saved at LLM-start, fall back to generationInfo, then agentName.
+        const model = pending.model ?? gen?.generationInfo?.model ?? this.agentName;
         const cost = calculateCost(model, inputTokens, outputTokens);
         // eslint-disable-next-line @typescript-eslint/no-explicit-any
         const rawText = gen?.text ?? gen?.message?.content ?? '';
@@ -81,16 +102,15 @@ export class LangChainCallback {
             status: 'success',
             inputTokens,
             outputTokens,
-            inputText: '', // LangChain doesn't surface the raw prompt here
+            inputText: pending.inputText ?? '',
             outputText,
-            durationMs: Date.now() - startMs,
+            durationMs: Date.now() - pending.startMs,
         });
         this.visibe.batcher.add(this.traceId, span);
-        // Update local accumulators (used by patchCompiledStateGraph / patchRunnableSequence).
         this.totalInputTokens += inputTokens;
         this.totalOutputTokens += outputTokens;
+        this.totalCost += cost;
         this.llmCallCount++;
-        // Notify track() accumulator if running inside a group tracker.
         this._onLLMSpan?.(inputTokens, outputTokens, cost);
     }
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
@@ -139,16 +159,15 @@ export class LangChainCallback {
     }
     // ---------------------------------------------------------------------------
     // Chain events
+    //
+    // In LangGraph v1.2+, the node key is passed as the 8th `name` parameter, NOT
+    // via chain.name. We always store runId → spanId so nested LLM/tool calls get
+    // a valid parent_span_id regardless of whether the chain is a user-defined node.
     // ---------------------------------------------------------------------------
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    async handleChainStart(chain, _inputs, runId, parentRunId) {
-        // Emit an agent_start span the first time we see a named chain.
-        const chainName = chain?.id?.at(-1) ?? '';
-        if (chainName && !this.seenAgents.has(chainName)) {
-            this.seenAgents.add(chainName);
-            const spanId = this.nextSpanId();
-            this.runIdToSpanId.set(runId, spanId);
-            void parentRunId; // suppress unused warning
+    async handleChainStart(_chain, _inputs, runId, _parentRunId, _tags, _metadata, _runType, _name) {
+        if (!this.runIdToSpanId.has(runId)) {
+            this.runIdToSpanId.set(runId, this.nextSpanId());
         }
     }
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
@@ -174,7 +193,6 @@ export function patchRunnableSequence(lcModule, visibe) {
     const originalStream = RunnableSequence.prototype.stream;
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
     RunnableSequence.prototype.invoke = async function (input, config) {
-        // If already inside a LangChain trace, pass through.
         if (activeLangChainStorage.getStore() !== undefined) {
             return originalInvoke.call(this, input, config);
         }
@@ -189,6 +207,7 @@ export function patchRunnableSequence(lcModule, visibe) {
             ...(visibe.sessionId ? { session_id: visibe.sessionId } : {}),
         });
         const cb = new LangChainCallback({ visibe, traceId, agentName: 'langchain' });
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
         let result;
         let status = 'completed';
         try {
@@ -205,6 +224,7 @@ export function patchRunnableSequence(lcModule, visibe) {
                 ended_at: new Date().toISOString(),
                 duration_ms: Date.now() - startMs,
                 llm_call_count: cb.llmCallCount,
+                total_cost: cb.totalCost,
                 total_tokens: cb.totalInputTokens + cb.totalOutputTokens,
                 total_input_tokens: cb.totalInputTokens,
                 total_output_tokens: cb.totalOutputTokens,
@@ -231,9 +251,6 @@ export function patchRunnableSequence(lcModule, visibe) {
         const cb = new LangChainCallback({ visibe, traceId, agentName: 'langchain' });
         let status = 'completed';
         try {
-            // RunnableSequence.stream() is an async function (not async generator) returning
-            // a Promise<AsyncIterable>. activeLangChainStorage.run returns that Promise,
-            // so we must await before yield*.
             const gen = await activeLangChainStorage.run(cb, () => originalStream.call(this, input, _mergeCallbacks(config, cb)));
             yield* gen;
         }
@@ -248,6 +265,7 @@ export function patchRunnableSequence(lcModule, visibe) {
                 ended_at: new Date().toISOString(),
                 duration_ms: Date.now() - startMs,
                 llm_call_count: cb.llmCallCount,
+                total_cost: cb.totalCost,
                 total_tokens: cb.totalInputTokens + cb.totalOutputTokens,
                 total_input_tokens: cb.totalInputTokens,
                 total_output_tokens: cb.totalOutputTokens,
@@ -262,7 +280,6 @@ export function patchRunnableSequence(lcModule, visibe) {
 // ---------------------------------------------------------------------------
 // Private helpers
 // ---------------------------------------------------------------------------
-// Merge our callback into an existing LangChain config object.
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
 function _mergeCallbacks(config, cb) {
     if (!config)

package/dist/esm/integrations/langgraph.js

@@ -1,5 +1,5 @@
 import { randomUUID } from 'node:crypto';
-import { LangChainCallback, activeLangChainStorage } from './langchain.js';
+import { LangChainCallback, activeLangChainStorage, LANGGRAPH_INTERNAL_NODES } from './langchain.js';
 // ---------------------------------------------------------------------------
 // LangGraphCallback
 // Extends LangChainCallback and adds node-level agent_start spans.
@@ -9,24 +9,26 @@ export class LangGraphCallback extends LangChainCallback {
         super(options);
         this.nodeNames = new Set(options.nodeNames ?? []);
     }
-    // Override handleChainStart to emit agent_start for known graph nodes.
+    // Override handleChainStart to emit agent_start spans for LangGraph nodes.
+    //
+    // In LangGraph v1.2+, the node key is the 8th `name` parameter (not chain.name).
+    // Internal system nodes (__start__, __end__, LangGraph) are filtered out.
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    async handleChainStart(chain, inputs, runId, parentRunId) {
-        // LangGraph may surface the node key in chain.name (run name) or chain.id[-1] (class name).
-        const chainName = chain?.name ?? chain?.id?.at(-1) ?? '';
-        if (chainName && this.nodeNames.has(chainName)) {
-            const spanId = this.nextSpanId();
+    async handleChainStart(chain, inputs, runId, parentRunId, tags, metadata, runType, name) {
+        // Always track the runId so child LLM calls can resolve parent_span_id.
+        await super.handleChainStart(chain, inputs, runId, parentRunId, tags, metadata, runType, name);
+        // The node key is passed as the 8th `name` parameter in LangGraph v1.2+.
+        // Fall back to chain.name for older versions.
+        const nodeName = name ?? chain?.name ?? '';
+        if (nodeName && !LANGGRAPH_INTERNAL_NODES.has(nodeName)) {
+            // Use the spanId already assigned by super for this runId.
+            const spanId = this.runIdToSpanId.get(runId) ?? this.nextSpanId();
             this.runIdToSpanId.set(runId, spanId);
-            // Emit an agent_start span for this node.
-            // type MUST be exactly "agent_start" — the backend validates this string.
             this.visibe.batcher.add(this.traceId, this.visibe.buildAgentStartSpan({
                 spanId,
-                agentName: chainName,
+                agentName: nodeName,
             }));
-            // Don't call super — we've already set the runId mapping.
-            return;
         }
-        await super.handleChainStart(chain, inputs, runId, parentRunId);
     }
 }
 // ---------------------------------------------------------------------------
@@ -82,6 +84,7 @@ export function patchCompiledStateGraph(lgModule, visibe) {
                 ended_at: new Date().toISOString(),
                 duration_ms: Date.now() - startMs,
                 llm_call_count: cb.llmCallCount,
+                total_cost: cb.totalCost,
                 total_tokens: cb.totalInputTokens + cb.totalOutputTokens,
                 total_input_tokens: cb.totalInputTokens,
                 total_output_tokens: cb.totalOutputTokens,
@@ -133,6 +136,7 @@ export function patchCompiledStateGraph(lgModule, visibe) {
                 ended_at: new Date().toISOString(),
                 duration_ms: Date.now() - startMs,
                 llm_call_count: cb.llmCallCount,
+                total_cost: cb.totalCost,
                 total_tokens: cb.totalInputTokens + cb.totalOutputTokens,
                 total_input_tokens: cb.totalInputTokens,
                 total_output_tokens: cb.totalOutputTokens,

package/dist/types/integrations/langchain.d.ts

@@ -1,6 +1,7 @@
 import { AsyncLocalStorage } from 'node:async_hooks';
 import type { Visibe } from '../client';
 export declare const activeLangChainStorage: AsyncLocalStorage<LangChainCallback>;
+export declare const LANGGRAPH_INTERNAL_NODES: Set<string>;
 export declare class LangChainCallback {
     readonly name = "visibe-langchain-callback";
     readonly awaitHandlers = true;
@@ -9,7 +10,11 @@ export declare class LangChainCallback {
     protected readonly traceId: string;
     protected readonly agentName: string;
     protected runIdToSpanId: Map<string, string>;
-    protected pendingLLMCalls: Map<string, number>;
+    protected pendingLLMCalls: Map<string, {
+        startMs: number;
+        model?: string;
+        inputText?: string;
+    }>;
     protected pendingToolCalls: Map<string, {
         startMs: number;
         inputText: string;
@@ -17,22 +22,23 @@ export declare class LangChainCallback {
     }>;
     protected stepCounter: number;
     protected nextSpanId(): string;
-    protected seenAgents: Set<string>;
     totalInputTokens: number;
     totalOutputTokens: number;
+    totalCost: number;
     llmCallCount: number;
     constructor(options: {
         visibe: Visibe;
         traceId: string;
         agentName: string;
     });
-    handleLLMStart(_llm: any, _messages: any[], runId: string): Promise<void>;
+    handleLLMStart(llm: any, prompts: any[], runId: string): Promise<void>;
+    handleChatModelStart(llm: any, messages: any[][], runId: string): Promise<void>;
     handleLLMEnd(output: any, runId: string, parentRunId?: string): Promise<void>;
     handleLLMError(err: Error, runId: string): Promise<void>;
     handleToolStart(tool: any, input: string, runId: string): Promise<void>;
     handleToolEnd(output: string, runId: string, parentRunId?: string): Promise<void>;
     handleToolError(err: Error, runId: string): Promise<void>;
-    handleChainStart(chain: any, _inputs: any, runId: string, parentRunId?: string): Promise<void>;
+    handleChainStart(_chain: any, _inputs: any, runId: string, _parentRunId?: string, _tags?: string[], _metadata?: any, _runType?: string, _name?: string): Promise<void>;
     handleChainEnd(_outputs: any, _runId: string): Promise<void>;
     handleChainError(err: Error, _runId: string): Promise<void>;
     _onLLMSpan?: (inputTokens: number, outputTokens: number, cost: number) => void;

package/dist/types/integrations/langgraph.d.ts

@@ -8,6 +8,6 @@ export declare class LangGraphCallback extends LangChainCallback {
         agentName: string;
         nodeNames?: string[];
     });
-    handleChainStart(chain: any, inputs: any, runId: string, parentRunId?: string): Promise<void>;
+    handleChainStart(chain: any, inputs: any, runId: string, parentRunId?: string, tags?: string[], metadata?: any, runType?: string, name?: string): Promise<void>;
 }
 export declare function patchCompiledStateGraph(lgModule: any, visibe: Visibe): () => void;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@visibe.ai/node",
-  "version": "0.1.17",
+  "version": "0.1.18",
   "description": "AI Agent Observability — Track OpenAI, LangChain, LangGraph, Bedrock, Vercel AI, Anthropic",
   "main": "dist/cjs/index.js",
   "module": "dist/esm/index.js",