@agent-inspect/langchain 1.7.0 → 1.8.0

package/dist/index.cjs

@@ -111,6 +111,71 @@ function safePreview(value, maxChars) {
   }
 }
 var MAX_METADATA_KEYS = 40;
+var LANGGRAPH_ALIASES = [
+  { out: "graphId", aliases: ["graphId", "graph_id", "langgraph_graph_id"] },
+  { out: "graphName", aliases: ["graphName", "graph_name", "langgraph_graph_name"] },
+  { out: "nodeId", aliases: ["nodeId", "node_id", "langgraph_node_id"] },
+  { out: "nodeName", aliases: ["nodeName", "node_name", "langgraph_node"] },
+  { out: "subgraphId", aliases: ["subgraphId", "subgraph_id", "langgraph_subgraph_id"] },
+  { out: "subgraphName", aliases: ["subgraphName", "subgraph_name", "langgraph_subgraph"] },
+  { out: "taskId", aliases: ["taskId", "task_id", "langgraph_task_id"] },
+  { out: "taskName", aliases: ["taskName", "task_name", "langgraph_task"] },
+  { out: "branch", aliases: ["branch", "branchName", "branch_name", "langgraph_branch"] },
+  { out: "branchPath", aliases: ["branchPath", "branch_path", "langgraph_path"] },
+  { out: "checkpointId", aliases: ["checkpointId", "checkpoint_id", "langgraph_checkpoint_id"] },
+  {
+    out: "checkpointNamespace",
+    aliases: ["checkpointNamespace", "checkpoint_ns", "langgraph_checkpoint_ns"]
+  },
+  { out: "threadId", aliases: ["threadId", "thread_id", "langgraph_thread_id"] },
+  { out: "sessionId", aliases: ["sessionId", "session_id", "langgraph_session_id"] },
+  { out: "retryAttempt", aliases: ["retryAttempt", "retry_attempt", "attempt"] },
+  { out: "handoffFrom", aliases: ["handoffFrom", "handoff_from", "langgraph_handoff_from"] },
+  { out: "handoffTo", aliases: ["handoffTo", "handoff_to", "langgraph_handoff_to"] }
+];
+var LANGGRAPH_SUMMARY_ALIASES = [
+  { out: "checkpointSummary", aliases: ["checkpoint", "langgraph_checkpoint"] },
+  { out: "branchSummary", aliases: ["branches", "langgraph_branches", "langgraph_triggers"] },
+  { out: "taskSummary", aliases: ["tasks", "langgraph_tasks"] }
+];
+function boundedValue(value) {
+  if (value === null) return null;
+  if (typeof value === "string") return value.length <= 200 ? value : `${value.slice(0, 200)}\u2026`;
+  if (typeof value === "number" || typeof value === "boolean") return value;
+  if (typeof value === "bigint") return value.toString();
+  if (Array.isArray(value)) {
+    const items = value.slice(0, 10).map((item) => {
+      if (item === null) return null;
+      if (typeof item === "string" || typeof item === "number" || typeof item === "boolean") {
+        return item;
+      }
+      if (typeof item === "bigint") return item.toString();
+      return summarizeValue(item);
+    });
+    return {
+      type: "array",
+      itemCount: value.length,
+      items,
+      truncated: value.length > items.length ? true : void 0
+    };
+  }
+  return summarizeValue(value);
+}
+function summarizeValue(value) {
+  if (Array.isArray(value)) return { type: "array", itemCount: value.length };
+  if (isRecord(value)) return { type: "object", keyCount: Object.keys(value).length };
+  return { type: typeof value };
+}
+function getAliasValue(sources, aliases) {
+  for (const source of sources) {
+    for (const alias of aliases) {
+      if (Object.prototype.hasOwnProperty.call(source, alias)) {
+        return source[alias];
+      }
+    }
+  }
+  return void 0;
+}
 function toPlainMetadata(value) {
   try {
     if (!isRecord(value)) return {};
@@ -147,6 +212,32 @@ function toPlainMetadata(value) {
     return {};
   }
 }
+function extractLangGraphMetadata(value) {
+  try {
+    if (!isRecord(value)) return void 0;
+    const sources = [value];
+    for (const key of ["langgraph", "configurable"]) {
+      const nested = value[key];
+      if (isRecord(nested)) sources.push(nested);
+    }
+    const config = value.config;
+    if (isRecord(config) && isRecord(config.configurable)) {
+      sources.push(config.configurable);
+    }
+    const out = {};
+    for (const { out: outKey, aliases } of LANGGRAPH_ALIASES) {
+      const raw = getAliasValue(sources, aliases);
+      if (raw !== void 0) out[outKey] = boundedValue(raw);
+    }
+    for (const { out: outKey, aliases } of LANGGRAPH_SUMMARY_ALIASES) {
+      const raw = getAliasValue(sources, aliases);
+      if (raw !== void 0) out[outKey] = summarizeValue(raw);
+    }
+    return Object.keys(out).length > 0 ? out : void 0;
+  } catch {
+    return void 0;
+  }
+}
 
 // packages/langchain/src/streaming-metadata.ts
 function createLlmStreamState() {
@@ -286,6 +377,11 @@ var LangChainTracePersistence = class {
       const stepId = agentInspect.createStepId();
       this.#lcToStepId.set(params.lcRunId, stepId);
       const parentId = this.resolveParentId(params.lcParentRunId);
+      const metadata = toStepMetadata(params.attributes);
+      if (params.lcParentRunId && !parentId) {
+        metadata.parentMapping = "unresolved";
+        metadata.unresolvedParentRunId = params.lcParentRunId;
+      }
       const event = {
         schemaVersion: "0.1",
         event: "step_started",
@@ -296,7 +392,7 @@ var LangChainTracePersistence = class {
         name: params.name,
         type: kindToStepType(params.kind),
         startTime: params.startTime,
-        metadata: toStepMetadata(params.attributes)
+        metadata
       };
       await this.#write(event);
     } catch (err) {
@@ -310,6 +406,11 @@ var LangChainTracePersistence = class {
         stepId = agentInspect.createStepId();
         this.#lcToStepId.set(params.lcRunId, stepId);
         const parentId = this.resolveParentId(params.lcParentRunId);
+        const metadata = toStepMetadata(params.completionAttributes);
+        if (params.lcParentRunId && !parentId) {
+          metadata.parentMapping = "unresolved";
+          metadata.unresolvedParentRunId = params.lcParentRunId;
+        }
         const startTime = params.endTime - (params.durationMs ?? 0);
         const started = {
           schemaVersion: "0.1",
@@ -323,7 +424,7 @@ var LangChainTracePersistence = class {
             params.completionAttributes.kind ?? "LLM"
           ),
           startTime,
-          metadata: toStepMetadata(params.completionAttributes)
+          metadata
         };
         await this.#write(started);
       }
@@ -362,6 +463,11 @@ var LangChainTracePersistence = class {
       const stepId = agentInspect.createStepId();
       this.#lcToStepId.set(params.lcRunId, stepId);
       const parentId = this.resolveParentId(params.lcParentRunId);
+      const metadata = toStepMetadata(params.attributes);
+      if (params.lcParentRunId && !parentId) {
+        metadata.parentMapping = "unresolved";
+        metadata.unresolvedParentRunId = params.lcParentRunId;
+      }
       const started = {
         schemaVersion: "0.1",
         event: "step_started",
@@ -372,7 +478,7 @@ var LangChainTracePersistence = class {
         name: params.name,
         type: kindToStepType(params.kind),
         startTime: params.timestamp,
-        metadata: toStepMetadata(params.attributes)
+        metadata
       };
       await this.#write(started);
       const completed = {
@@ -443,6 +549,9 @@ var LangChainTracePersistence = class {
 };
 
 // packages/langchain/src/agent-inspect-callback.ts
+function isRecord2(v) {
+  return typeof v === "object" && v !== null && !Array.isArray(v);
+}
 function serializedLabel(s) {
   if (typeof s.name === "string" && s.name.trim()) return s.name;
   if (Array.isArray(s.id) && s.id.length > 0) return s.id[s.id.length - 1];
@@ -576,6 +685,15 @@ var AgentInspectCallback = class extends base.BaseCallbackHandler {
   #rememberStart(lcRunId, kind) {
     this.#starts.set(lcRunId, { ts: Date.now(), kind });
   }
+  #rememberStartMetadata(lcRunId, attrs) {
+    const start = this.#starts.get(lcRunId);
+    if (!start || !isRecord2(attrs.langGraph)) return;
+    this.#starts.set(lcRunId, { ...start, langGraph: { ...attrs.langGraph } });
+  }
+  #attachStartMetadata(attrs, lcRunId) {
+    const langGraph = this.#starts.get(lcRunId)?.langGraph;
+    if (langGraph && !attrs.langGraph) attrs.langGraph = { ...langGraph };
+  }
   #clearStart(lcRunId) {
     this.#starts.delete(lcRunId);
   }
@@ -594,6 +712,8 @@ var AgentInspectCallback = class extends base.BaseCallbackHandler {
   #mergeMetadata(attrs, metadata) {
     if (this.#opts.capture === "none" || !metadata) return;
     attrs.metadata = this.#redactor.redactRecord(toPlainMetadata(metadata));
+    const langGraph = extractLangGraphMetadata(metadata);
+    if (langGraph) attrs.langGraph = this.#redactor.redactRecord(langGraph);
   }
   #applyPreview(attrs, previews) {
     if (this.#opts.capture !== "preview") return;
@@ -657,6 +777,7 @@ var AgentInspectCallback = class extends base.BaseCallbackHandler {
     };
     this.#mergeMetadata(attrs, metadata);
     this.#applyPreview(attrs, previews);
+    this.#rememberStartMetadata(runId, attrs);
     const ts = Date.now();
     this.#pushEvent({
       eventId: `${runId}:CHAIN:start`,
@@ -675,12 +796,13 @@ var AgentInspectCallback = class extends base.BaseCallbackHandler {
   async handleChainEnd(outputs, runId, parentRunId, tags, _kwargs) {
     this.#ensureRoot(runId, parentRunId);
     const durationMs = this.#durationFor(runId);
-    this.#clearStart(runId);
     const previews = {};
     if (this.#opts.capture === "preview") previews.outputPreview = outputs;
     const attrs = {
       ...this.#baseAttrs(runId, parentRunId, tags, void 0)
     };
+    this.#attachStartMetadata(attrs, runId);
+    this.#clearStart(runId);
     this.#applyPreview(attrs, previews);
     const ts = Date.now();
     this.#pushEvent({
@@ -701,13 +823,14 @@ var AgentInspectCallback = class extends base.BaseCallbackHandler {
   async handleChainError(err, runId, parentRunId, tags, _kwargs) {
     this.#ensureRoot(runId, parentRunId);
     const durationMs = this.#durationFor(runId);
-    this.#clearStart(runId);
     const { errorName, errorMessage } = errorShape(err);
     const attrs = {
       ...this.#baseAttrs(runId, parentRunId, tags, void 0),
       errorName,
       errorMessage
     };
+    this.#attachStartMetadata(attrs, runId);
+    this.#clearStart(runId);
     const ts = Date.now();
     this.#pushEvent({
       eventId: `${runId}:CHAIN:error`,
@@ -739,6 +862,7 @@ var AgentInspectCallback = class extends base.BaseCallbackHandler {
     this.#mergeMetadata(attrs, metadata);
     this.#mergeCorrelation(attrs);
     this.#applyPreview(attrs, previews);
+    this.#rememberStartMetadata(runId, attrs);
     const ts = Date.now();
     const stepName = `llm:${model ?? "llm"}`;
     this.#pushEvent({
@@ -768,6 +892,7 @@ var AgentInspectCallback = class extends base.BaseCallbackHandler {
     this.#mergeMetadata(attrs, metadata);
     this.#mergeCorrelation(attrs);
     this.#applyPreview(attrs, previews);
+    this.#rememberStartMetadata(runId, attrs);
     const ts = Date.now();
     const stepName = `llm:${model ?? "llm"}`;
     this.#pushEvent({
@@ -802,7 +927,6 @@ var AgentInspectCallback = class extends base.BaseCallbackHandler {
   async handleLLMEnd(output, runId, parentRunId, tags, _extraParams) {
     this.#ensureRoot(runId, parentRunId);
     const durationMs = this.#durationFor(runId);
-    this.#clearStart(runId);
     const tokens = extractTokenUsage(output);
     const model = extractModelName(output);
     const previews = {};
@@ -812,6 +936,8 @@ var AgentInspectCallback = class extends base.BaseCallbackHandler {
     };
     if (model && this.#opts.capture !== "none") attrs.model = model;
     if (tokens && this.#opts.capture !== "none") attrs.tokens = tokens;
+    this.#attachStartMetadata(attrs, runId);
+    this.#clearStart(runId);
     this.#attachStreamMetadata(attrs, runId);
     this.#mergeCorrelation(attrs);
     this.#applyPreview(attrs, previews);
@@ -840,13 +966,14 @@ var AgentInspectCallback = class extends base.BaseCallbackHandler {
   async handleLLMError(err, runId, parentRunId, tags, _extraParams) {
     this.#ensureRoot(runId, parentRunId);
     const durationMs = this.#durationFor(runId);
-    this.#clearStart(runId);
     const { errorName, errorMessage } = errorShape(err);
     const attrs = {
       ...this.#baseAttrs(runId, parentRunId, tags, void 0),
       errorName,
       errorMessage
     };
+    this.#attachStartMetadata(attrs, runId);
+    this.#clearStart(runId);
     this.#attachStreamMetadata(attrs, runId);
     this.#mergeCorrelation(attrs);
     const ts = Date.now();
@@ -883,6 +1010,7 @@ var AgentInspectCallback = class extends base.BaseCallbackHandler {
     };
     this.#mergeMetadata(attrs, metadata);
     this.#applyPreview(attrs, previews);
+    this.#rememberStartMetadata(runId, attrs);
     const ts = Date.now();
     const stepName = `tool:${toolName}`;
     this.#pushEvent({
@@ -902,12 +1030,13 @@ var AgentInspectCallback = class extends base.BaseCallbackHandler {
   async handleToolEnd(output, runId, parentRunId, tags) {
     this.#ensureRoot(runId, parentRunId);
     const durationMs = this.#durationFor(runId);
-    this.#clearStart(runId);
     const previews = {};
     if (this.#opts.capture === "preview") previews.outputPreview = output;
     const attrs = {
       ...this.#baseAttrs(runId, parentRunId, tags, void 0)
     };
+    this.#attachStartMetadata(attrs, runId);
+    this.#clearStart(runId);
     this.#applyPreview(attrs, previews);
     const ts = Date.now();
     this.#pushEvent({
@@ -928,13 +1057,14 @@ var AgentInspectCallback = class extends base.BaseCallbackHandler {
   async handleToolError(err, runId, parentRunId, tags) {
     this.#ensureRoot(runId, parentRunId);
     const durationMs = this.#durationFor(runId);
-    this.#clearStart(runId);
     const { errorName, errorMessage } = errorShape(err);
     const attrs = {
       ...this.#baseAttrs(runId, parentRunId, tags, void 0),
       errorName,
       errorMessage
     };
+    this.#attachStartMetadata(attrs, runId);
+    this.#clearStart(runId);
     const ts = Date.now();
     this.#pushEvent({
       eventId: `${runId}:TOOL:error`,
@@ -960,6 +1090,7 @@ var AgentInspectCallback = class extends base.BaseCallbackHandler {
       retriever: rname
     };
     this.#mergeMetadata(attrs, metadata);
+    this.#rememberStartMetadata(runId, attrs);
     const ts = Date.now();
     const stepName = `retriever:${rname}`;
     this.#pushEvent({
@@ -979,7 +1110,6 @@ var AgentInspectCallback = class extends base.BaseCallbackHandler {
   async handleRetrieverEnd(documents, runId, parentRunId, tags) {
     this.#ensureRoot(runId, parentRunId);
     const durationMs = this.#durationFor(runId);
-    this.#clearStart(runId);
     const previews = {};
     if (this.#opts.capture === "preview" && documents.length > 0) {
       previews.documentPreview = documents.slice(0, 3);
@@ -988,6 +1118,8 @@ var AgentInspectCallback = class extends base.BaseCallbackHandler {
       ...this.#baseAttrs(runId, parentRunId, tags, void 0),
       documentCount: documents.length
     };
+    this.#attachStartMetadata(attrs, runId);
+    this.#clearStart(runId);
     this.#applyPreview(attrs, previews);
     const ts = Date.now();
     this.#pushEvent({
@@ -1008,13 +1140,14 @@ var AgentInspectCallback = class extends base.BaseCallbackHandler {
   async handleRetrieverError(err, runId, parentRunId, tags) {
     this.#ensureRoot(runId, parentRunId);
     const durationMs = this.#durationFor(runId);
-    this.#clearStart(runId);
     const { errorName, errorMessage } = errorShape(err);
     const attrs = {
       ...this.#baseAttrs(runId, parentRunId, tags, void 0),
       errorName,
       errorMessage
     };
+    this.#attachStartMetadata(attrs, runId);
+    this.#clearStart(runId);
     const ts = Date.now();
     this.#pushEvent({
       eventId: `${runId}:RETRIEVER:error`,