@visibe.ai/node 0.1.26 → 0.1.27

package/dist/cjs/integrations/langchain.js

@@ -3,6 +3,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.LangChainCallback = exports.LANGGRAPH_INTERNAL_NODES = exports.activeLangChainStorage = void 0;
 exports.patchRunnableSequence = patchRunnableSequence;
 exports.patchAgentExecutor = patchAgentExecutor;
+exports.serializeInput = serializeInput;
 const node_async_hooks_1 = require("node:async_hooks");
 const node_crypto_1 = require("node:crypto");
 const utils_1 = require("../utils");
@@ -79,6 +80,8 @@ class LangChainCallback {
         this.totalOutputTokens = 0;
         this.totalCost = 0;
         this.llmCallCount = 0;
+        // First model seen across all LLM calls in this trace — set on first handleLLMEnd.
+        this.firstModel = undefined;
         this.visibe = options.visibe;
         this.traceId = options.traceId;
         this.agentName = options.agentName;
@@ -144,6 +147,8 @@ class LangChainCallback {
         this.totalOutputTokens += outputTokens;
         this.totalCost += cost;
         this.llmCallCount++;
+        if (!this.firstModel)
+            this.firstModel = model;
         this._onLLMSpan?.(inputTokens, outputTokens, cost);
     }
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
@@ -250,6 +255,7 @@ function patchRunnableSequence(lcModule, visibe) {
             name: 'langchain',
             framework: 'langchain',
             started_at: startedAt,
+            prompt: serializeInput(input),
             ...(visibe.sessionId ? { session_id: visibe.sessionId } : {}),
         });
         const cb = new LangChainCallback({ visibe, traceId, agentName: 'langchain' });
@@ -274,6 +280,7 @@ function patchRunnableSequence(lcModule, visibe) {
                 total_tokens: cb.totalInputTokens + cb.totalOutputTokens,
                 total_input_tokens: cb.totalInputTokens,
                 total_output_tokens: cb.totalOutputTokens,
+                ...(cb.firstModel ? { model: cb.firstModel } : {}),
             });
         }
         return result;
@@ -292,6 +299,7 @@ function patchRunnableSequence(lcModule, visibe) {
             name: 'langchain',
             framework: 'langchain',
             started_at: startedAt,
+            prompt: serializeInput(input),
             ...(visibe.sessionId ? { session_id: visibe.sessionId } : {}),
         });
         const cb = new LangChainCallback({ visibe, traceId, agentName: 'langchain' });
@@ -315,6 +323,7 @@ function patchRunnableSequence(lcModule, visibe) {
                 total_tokens: cb.totalInputTokens + cb.totalOutputTokens,
                 total_input_tokens: cb.totalInputTokens,
                 total_output_tokens: cb.totalOutputTokens,
+                ...(cb.firstModel ? { model: cb.firstModel } : {}),
             });
         }
     };
@@ -352,6 +361,7 @@ function patchAgentExecutor(agentsModule, visibe) {
             name: agentName,
             framework: 'langchain',
             started_at: startedAt,
+            prompt: serializeInput(input),
             ...(visibe.sessionId ? { session_id: visibe.sessionId } : {}),
         });
         const cb = new LangChainCallback({ visibe, traceId, agentName });
@@ -376,6 +386,7 @@ function patchAgentExecutor(agentsModule, visibe) {
                 total_tokens: cb.totalInputTokens + cb.totalOutputTokens,
                 total_input_tokens: cb.totalInputTokens,
                 total_output_tokens: cb.totalOutputTokens,
+                ...(cb.firstModel ? { model: cb.firstModel } : {}),
             });
         }
         return result;
@@ -395,6 +406,7 @@ function patchAgentExecutor(agentsModule, visibe) {
             name: agentName,
             framework: 'langchain',
             started_at: startedAt,
+            prompt: serializeInput(input),
             ...(visibe.sessionId ? { session_id: visibe.sessionId } : {}),
         });
         const cb = new LangChainCallback({ visibe, traceId, agentName });
@@ -418,6 +430,7 @@ function patchAgentExecutor(agentsModule, visibe) {
                 total_tokens: cb.totalInputTokens + cb.totalOutputTokens,
                 total_input_tokens: cb.totalInputTokens,
                 total_output_tokens: cb.totalOutputTokens,
+                ...(cb.firstModel ? { model: cb.firstModel } : {}),
             });
         }
     };
@@ -436,3 +449,37 @@ function _mergeCallbacks(config, cb) {
     const existing = Array.isArray(config.callbacks) ? config.callbacks : [];
     return { ...config, callbacks: [...existing, cb] };
 }
+// Serialize a graph/chain input to a prompt string for the trace header.
+// Handles strings, common object shapes, and falls back to JSON.
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+function serializeInput(input, limit = 2000) {
+    if (!input)
+        return undefined;
+    let s;
+    if (typeof input === 'string') {
+        s = input;
+    }
+    else if (typeof input === 'object') {
+        // Try common single-field patterns first.
+        for (const key of ['task', 'prompt', 'input', 'query', 'question', 'text', 'content']) {
+            if (typeof input[key] === 'string') {
+                s = input[key];
+                break;
+            }
+        }
+        if (!s) {
+            // HumanMessage / messages array — grab the last message content.
+            if (Array.isArray(input.messages) && input.messages.length > 0) {
+                const last = input.messages[input.messages.length - 1];
+                s = typeof last?.content === 'string' ? last.content : JSON.stringify(last?.content ?? '');
+            }
+            else {
+                s = JSON.stringify(input);
+            }
+        }
+    }
+    else {
+        s = String(input);
+    }
+    return s.length > limit ? s.slice(0, limit) + '…' : s;
+}

package/dist/cjs/integrations/langgraph.js

@@ -68,6 +68,7 @@ function patchCompiledStateGraph(lgModule, visibe) {
             name: graphName,
             framework: 'langgraph',
             started_at: startedAt,
+            prompt: (0, langchain_1.serializeInput)(input),
             ...(visibe.sessionId ? { session_id: visibe.sessionId } : {}),
         });
         // Collect node names from the graph's node registry.
@@ -98,6 +99,7 @@ function patchCompiledStateGraph(lgModule, visibe) {
                 total_tokens: cb.totalInputTokens + cb.totalOutputTokens,
                 total_input_tokens: cb.totalInputTokens,
                 total_output_tokens: cb.totalOutputTokens,
+                ...(cb.firstModel ? { model: cb.firstModel } : {}),
             });
         }
         return result;
@@ -119,6 +121,7 @@ function patchCompiledStateGraph(lgModule, visibe) {
             name: graphName,
             framework: 'langgraph',
             started_at: startedAt,
+            prompt: (0, langchain_1.serializeInput)(input),
             ...(visibe.sessionId ? { session_id: visibe.sessionId } : {}),
         });
         const nodeNames = _extractNodeNames(this);
@@ -150,6 +153,7 @@ function patchCompiledStateGraph(lgModule, visibe) {
                 total_tokens: cb.totalInputTokens + cb.totalOutputTokens,
                 total_input_tokens: cb.totalInputTokens,
                 total_output_tokens: cb.totalOutputTokens,
+                ...(cb.firstModel ? { model: cb.firstModel } : {}),
             });
         }
     };

package/dist/esm/integrations/langchain.js

@@ -74,6 +74,8 @@ export class LangChainCallback {
         this.totalOutputTokens = 0;
         this.totalCost = 0;
         this.llmCallCount = 0;
+        // First model seen across all LLM calls in this trace — set on first handleLLMEnd.
+        this.firstModel = undefined;
         this.visibe = options.visibe;
         this.traceId = options.traceId;
         this.agentName = options.agentName;
@@ -139,6 +141,8 @@ export class LangChainCallback {
         this.totalOutputTokens += outputTokens;
         this.totalCost += cost;
         this.llmCallCount++;
+        if (!this.firstModel)
+            this.firstModel = model;
         this._onLLMSpan?.(inputTokens, outputTokens, cost);
     }
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
@@ -244,6 +248,7 @@ export function patchRunnableSequence(lcModule, visibe) {
             name: 'langchain',
             framework: 'langchain',
             started_at: startedAt,
+            prompt: serializeInput(input),
             ...(visibe.sessionId ? { session_id: visibe.sessionId } : {}),
         });
         const cb = new LangChainCallback({ visibe, traceId, agentName: 'langchain' });
@@ -268,6 +273,7 @@ export function patchRunnableSequence(lcModule, visibe) {
                 total_tokens: cb.totalInputTokens + cb.totalOutputTokens,
                 total_input_tokens: cb.totalInputTokens,
                 total_output_tokens: cb.totalOutputTokens,
+                ...(cb.firstModel ? { model: cb.firstModel } : {}),
             });
         }
         return result;
@@ -286,6 +292,7 @@ export function patchRunnableSequence(lcModule, visibe) {
             name: 'langchain',
             framework: 'langchain',
             started_at: startedAt,
+            prompt: serializeInput(input),
             ...(visibe.sessionId ? { session_id: visibe.sessionId } : {}),
         });
         const cb = new LangChainCallback({ visibe, traceId, agentName: 'langchain' });
@@ -309,6 +316,7 @@ export function patchRunnableSequence(lcModule, visibe) {
                 total_tokens: cb.totalInputTokens + cb.totalOutputTokens,
                 total_input_tokens: cb.totalInputTokens,
                 total_output_tokens: cb.totalOutputTokens,
+                ...(cb.firstModel ? { model: cb.firstModel } : {}),
             });
         }
     };
@@ -346,6 +354,7 @@ export function patchAgentExecutor(agentsModule, visibe) {
             name: agentName,
             framework: 'langchain',
             started_at: startedAt,
+            prompt: serializeInput(input),
             ...(visibe.sessionId ? { session_id: visibe.sessionId } : {}),
         });
         const cb = new LangChainCallback({ visibe, traceId, agentName });
@@ -370,6 +379,7 @@ export function patchAgentExecutor(agentsModule, visibe) {
                 total_tokens: cb.totalInputTokens + cb.totalOutputTokens,
                 total_input_tokens: cb.totalInputTokens,
                 total_output_tokens: cb.totalOutputTokens,
+                ...(cb.firstModel ? { model: cb.firstModel } : {}),
             });
         }
         return result;
@@ -389,6 +399,7 @@ export function patchAgentExecutor(agentsModule, visibe) {
             name: agentName,
             framework: 'langchain',
             started_at: startedAt,
+            prompt: serializeInput(input),
             ...(visibe.sessionId ? { session_id: visibe.sessionId } : {}),
         });
         const cb = new LangChainCallback({ visibe, traceId, agentName });
@@ -412,6 +423,7 @@ export function patchAgentExecutor(agentsModule, visibe) {
                 total_tokens: cb.totalInputTokens + cb.totalOutputTokens,
                 total_input_tokens: cb.totalInputTokens,
                 total_output_tokens: cb.totalOutputTokens,
+                ...(cb.firstModel ? { model: cb.firstModel } : {}),
             });
         }
     };
@@ -430,3 +442,37 @@ function _mergeCallbacks(config, cb) {
     const existing = Array.isArray(config.callbacks) ? config.callbacks : [];
     return { ...config, callbacks: [...existing, cb] };
 }
+// Serialize a graph/chain input to a prompt string for the trace header.
+// Handles strings, common object shapes, and falls back to JSON.
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+export function serializeInput(input, limit = 2000) {
+    if (!input)
+        return undefined;
+    let s;
+    if (typeof input === 'string') {
+        s = input;
+    }
+    else if (typeof input === 'object') {
+        // Try common single-field patterns first.
+        for (const key of ['task', 'prompt', 'input', 'query', 'question', 'text', 'content']) {
+            if (typeof input[key] === 'string') {
+                s = input[key];
+                break;
+            }
+        }
+        if (!s) {
+            // HumanMessage / messages array — grab the last message content.
+            if (Array.isArray(input.messages) && input.messages.length > 0) {
+                const last = input.messages[input.messages.length - 1];
+                s = typeof last?.content === 'string' ? last.content : JSON.stringify(last?.content ?? '');
+            }
+            else {
+                s = JSON.stringify(input);
+            }
+        }
+    }
+    else {
+        s = String(input);
+    }
+    return s.length > limit ? s.slice(0, limit) + '…' : s;
+}

package/dist/esm/integrations/langgraph.js

@@ -1,5 +1,5 @@
 import { randomUUID } from 'node:crypto';
-import { LangChainCallback, activeLangChainStorage, LANGGRAPH_INTERNAL_NODES } from './langchain.js';
+import { LangChainCallback, activeLangChainStorage, LANGGRAPH_INTERNAL_NODES, serializeInput } from './langchain.js';
 // ---------------------------------------------------------------------------
 // LangGraphCallback
 // Extends LangChainCallback and adds node-level agent_start spans.
@@ -63,6 +63,7 @@ export function patchCompiledStateGraph(lgModule, visibe) {
             name: graphName,
             framework: 'langgraph',
             started_at: startedAt,
+            prompt: serializeInput(input),
             ...(visibe.sessionId ? { session_id: visibe.sessionId } : {}),
         });
         // Collect node names from the graph's node registry.
@@ -93,6 +94,7 @@ export function patchCompiledStateGraph(lgModule, visibe) {
                 total_tokens: cb.totalInputTokens + cb.totalOutputTokens,
                 total_input_tokens: cb.totalInputTokens,
                 total_output_tokens: cb.totalOutputTokens,
+                ...(cb.firstModel ? { model: cb.firstModel } : {}),
             });
         }
         return result;
@@ -114,6 +116,7 @@ export function patchCompiledStateGraph(lgModule, visibe) {
             name: graphName,
             framework: 'langgraph',
             started_at: startedAt,
+            prompt: serializeInput(input),
             ...(visibe.sessionId ? { session_id: visibe.sessionId } : {}),
         });
         const nodeNames = _extractNodeNames(this);
@@ -145,6 +148,7 @@ export function patchCompiledStateGraph(lgModule, visibe) {
                 total_tokens: cb.totalInputTokens + cb.totalOutputTokens,
                 total_input_tokens: cb.totalInputTokens,
                 total_output_tokens: cb.totalOutputTokens,
+                ...(cb.firstModel ? { model: cb.firstModel } : {}),
             });
         }
     };

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

@@ -30,6 +30,7 @@ export declare class LangChainCallback {
     totalOutputTokens: number;
     totalCost: number;
     llmCallCount: number;
+    firstModel: string | undefined;
     constructor(options: {
         visibe: Visibe;
         traceId: string;
@@ -50,3 +51,4 @@ export declare class LangChainCallback {
 }
 export declare function patchRunnableSequence(lcModule: any, visibe: Visibe): () => void;
 export declare function patchAgentExecutor(agentsModule: any, visibe: Visibe): () => void;
+export declare function serializeInput(input: any, limit?: number): string | undefined;

package/package.json

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