@visibe.ai/node 0.1.19 → 0.1.21

package/dist/cjs/integrations/langchain.js

@@ -31,6 +31,16 @@ class LangChainCallback {
     nextSpanId() {
         return `step_${++this.stepCounter}`;
     }
+    // Walk up the run parent chain to find the nearest agent name.
+    // Falls back to this.agentName if no ancestor has an associated agent.
+    _findAgentName(runId) {
+        if (!runId)
+            return this.agentName;
+        const name = this.runIdToAgentName.get(runId);
+        if (name)
+            return name;
+        return this._findAgentName(this.runIdToParent.get(runId));
+    }
     constructor(options) {
         // Required by @langchain/core v1+ for proper callback registration.
         // Without `name`, ensureHandler() wraps via fromMethods() which drops prototype methods.
@@ -41,10 +51,17 @@ class LangChainCallback {
         this.raiseError = false;
         // Maps LangChain runId → our spanId so we can set parent_span_id.
         this.runIdToSpanId = new Map();
+        // Tracks parent run IDs so we can walk up the chain hierarchy.
+        this.runIdToParent = new Map();
+        // Maps runId → agent name for chains that emitted an agent_start span.
+        // Used by _findAgentName() to resolve the correct agent for llm_call / tool_call spans.
+        this.runIdToAgentName = new Map();
         // Pending LLM calls: runId → { startMs, model, inputText }
         this.pendingLLMCalls = new Map();
         this.pendingToolCalls = new Map();
         this.stepCounter = 0;
+        // Set to false in subclasses that emit their own agent_start spans (e.g. LangGraphCallback).
+        this._emitRootAgentStart = true;
         // Token / call / cost accumulators — read by patchCompiledStateGraph / patchRunnableSequence.
         this.totalInputTokens = 0;
         this.totalOutputTokens = 0;
@@ -101,7 +118,7 @@ class LangChainCallback {
         const span = this.visibe.buildLLMSpan({
             spanId,
             parentSpanId,
-            agentName: this.agentName,
+            agentName: this._findAgentName(parentRunId),
             model,
             status: 'success',
             inputTokens,
@@ -144,7 +161,7 @@ class LangChainCallback {
             spanId,
             parentSpanId,
             toolName: pending?.toolName ?? 'tool',
-            agentName: this.agentName,
+            agentName: this._findAgentName(parentRunId),
             status: 'success',
             durationMs: pending ? Date.now() - pending.startMs : 0,
             inputText: pending?.inputText ?? '',
@@ -173,6 +190,18 @@ class LangChainCallback {
         if (!this.runIdToSpanId.has(runId)) {
             this.runIdToSpanId.set(runId, this.nextSpanId());
         }
+        if (_parentRunId) {
+            this.runIdToParent.set(runId, _parentRunId);
+        }
+        // Emit agent_start for the root chain so LLM spans have a real parent and
+        // the agents breakdown is populated.  Subclasses that manage their own
+        // agent_start spans (LangGraphCallback) set _emitRootAgentStart = false.
+        if (this._emitRootAgentStart && !_parentRunId) {
+            const spanId = this.runIdToSpanId.get(runId);
+            const agentName = _name ?? _chain?.name ?? this.agentName;
+            this.runIdToAgentName.set(runId, agentName);
+            this.visibe.batcher.add(this.traceId, this.visibe.buildAgentStartSpan({ spanId, agentName }));
+        }
     }
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
     async handleChainEnd(_outputs, _runId) { }

package/dist/cjs/integrations/langgraph.js

@@ -12,6 +12,9 @@ class LangGraphCallback extends langchain_1.LangChainCallback {
     constructor(options) {
         super(options);
         this.nodeNames = new Set(options.nodeNames ?? []);
+        // LangGraphCallback emits its own agent_start spans per node — suppress the
+        // base-class root-chain emission to avoid a spurious 'LangGraph' agent span.
+        this._emitRootAgentStart = false;
     }
     // Override handleChainStart to emit agent_start spans for LangGraph nodes.
     //
@@ -28,6 +31,8 @@ class LangGraphCallback extends langchain_1.LangChainCallback {
             // Use the spanId already assigned by super for this runId.
             const spanId = this.runIdToSpanId.get(runId) ?? this.nextSpanId();
             this.runIdToSpanId.set(runId, spanId);
+            // Register agent name so _findAgentName() resolves it for child llm_call / tool_call spans.
+            this.runIdToAgentName.set(runId, nodeName);
             this.visibe.batcher.add(this.traceId, this.visibe.buildAgentStartSpan({
                 spanId,
                 agentName: nodeName,

package/dist/esm/integrations/langchain.js

@@ -27,6 +27,16 @@ export class LangChainCallback {
     nextSpanId() {
         return `step_${++this.stepCounter}`;
     }
+    // Walk up the run parent chain to find the nearest agent name.
+    // Falls back to this.agentName if no ancestor has an associated agent.
+    _findAgentName(runId) {
+        if (!runId)
+            return this.agentName;
+        const name = this.runIdToAgentName.get(runId);
+        if (name)
+            return name;
+        return this._findAgentName(this.runIdToParent.get(runId));
+    }
     constructor(options) {
         // Required by @langchain/core v1+ for proper callback registration.
         // Without `name`, ensureHandler() wraps via fromMethods() which drops prototype methods.
@@ -37,10 +47,17 @@ export class LangChainCallback {
         this.raiseError = false;
         // Maps LangChain runId → our spanId so we can set parent_span_id.
         this.runIdToSpanId = new Map();
+        // Tracks parent run IDs so we can walk up the chain hierarchy.
+        this.runIdToParent = new Map();
+        // Maps runId → agent name for chains that emitted an agent_start span.
+        // Used by _findAgentName() to resolve the correct agent for llm_call / tool_call spans.
+        this.runIdToAgentName = new Map();
         // Pending LLM calls: runId → { startMs, model, inputText }
         this.pendingLLMCalls = new Map();
         this.pendingToolCalls = new Map();
         this.stepCounter = 0;
+        // Set to false in subclasses that emit their own agent_start spans (e.g. LangGraphCallback).
+        this._emitRootAgentStart = true;
         // Token / call / cost accumulators — read by patchCompiledStateGraph / patchRunnableSequence.
         this.totalInputTokens = 0;
         this.totalOutputTokens = 0;
@@ -97,7 +114,7 @@ export class LangChainCallback {
         const span = this.visibe.buildLLMSpan({
             spanId,
             parentSpanId,
-            agentName: this.agentName,
+            agentName: this._findAgentName(parentRunId),
             model,
             status: 'success',
             inputTokens,
@@ -140,7 +157,7 @@ export class LangChainCallback {
             spanId,
             parentSpanId,
             toolName: pending?.toolName ?? 'tool',
-            agentName: this.agentName,
+            agentName: this._findAgentName(parentRunId),
             status: 'success',
             durationMs: pending ? Date.now() - pending.startMs : 0,
             inputText: pending?.inputText ?? '',
@@ -169,6 +186,18 @@ export class LangChainCallback {
         if (!this.runIdToSpanId.has(runId)) {
             this.runIdToSpanId.set(runId, this.nextSpanId());
         }
+        if (_parentRunId) {
+            this.runIdToParent.set(runId, _parentRunId);
+        }
+        // Emit agent_start for the root chain so LLM spans have a real parent and
+        // the agents breakdown is populated.  Subclasses that manage their own
+        // agent_start spans (LangGraphCallback) set _emitRootAgentStart = false.
+        if (this._emitRootAgentStart && !_parentRunId) {
+            const spanId = this.runIdToSpanId.get(runId);
+            const agentName = _name ?? _chain?.name ?? this.agentName;
+            this.runIdToAgentName.set(runId, agentName);
+            this.visibe.batcher.add(this.traceId, this.visibe.buildAgentStartSpan({ spanId, agentName }));
+        }
     }
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
     async handleChainEnd(_outputs, _runId) { }

package/dist/esm/integrations/langgraph.js

@@ -8,6 +8,9 @@ export class LangGraphCallback extends LangChainCallback {
     constructor(options) {
         super(options);
         this.nodeNames = new Set(options.nodeNames ?? []);
+        // LangGraphCallback emits its own agent_start spans per node — suppress the
+        // base-class root-chain emission to avoid a spurious 'LangGraph' agent span.
+        this._emitRootAgentStart = false;
     }
     // Override handleChainStart to emit agent_start spans for LangGraph nodes.
     //
@@ -24,6 +27,8 @@ export class LangGraphCallback extends LangChainCallback {
             // Use the spanId already assigned by super for this runId.
             const spanId = this.runIdToSpanId.get(runId) ?? this.nextSpanId();
             this.runIdToSpanId.set(runId, spanId);
+            // Register agent name so _findAgentName() resolves it for child llm_call / tool_call spans.
+            this.runIdToAgentName.set(runId, nodeName);
             this.visibe.batcher.add(this.traceId, this.visibe.buildAgentStartSpan({
                 spanId,
                 agentName: nodeName,

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

@@ -10,6 +10,8 @@ export declare class LangChainCallback {
     protected readonly traceId: string;
     protected readonly agentName: string;
     protected runIdToSpanId: Map<string, string>;
+    protected runIdToParent: Map<string, string>;
+    protected runIdToAgentName: Map<string, string>;
     protected pendingLLMCalls: Map<string, {
         startMs: number;
         model?: string;
@@ -22,6 +24,8 @@ export declare class LangChainCallback {
     }>;
     protected stepCounter: number;
     protected nextSpanId(): string;
+    protected _emitRootAgentStart: boolean;
+    protected _findAgentName(runId?: string): string;
     totalInputTokens: number;
     totalOutputTokens: number;
     totalCost: number;

package/package.json

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