@agent-inspect/langchain 1.6.0 → 1.8.0

package/dist/index.mjs

@@ -109,6 +109,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 {};
@@ -145,6 +210,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() {
@@ -284,6 +375,11 @@ var LangChainTracePersistence = class {
       const stepId = 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",
@@ -294,7 +390,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) {
@@ -308,6 +404,11 @@ var LangChainTracePersistence = class {
         stepId = 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",
@@ -321,7 +422,7 @@ var LangChainTracePersistence = class {
             params.completionAttributes.kind ?? "LLM"
           ),
           startTime,
-          metadata: toStepMetadata(params.completionAttributes)
+          metadata
         };
         await this.#write(started);
       }
@@ -360,6 +461,11 @@ var LangChainTracePersistence = class {
       const stepId = 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",
@@ -370,7 +476,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 = {
@@ -441,6 +547,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];
@@ -574,6 +683,15 @@ var AgentInspectCallback = class extends 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);
   }
@@ -592,6 +710,8 @@ var AgentInspectCallback = class extends 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;
@@ -655,6 +775,7 @@ var AgentInspectCallback = class extends BaseCallbackHandler {
     };
     this.#mergeMetadata(attrs, metadata);
     this.#applyPreview(attrs, previews);
+    this.#rememberStartMetadata(runId, attrs);
     const ts = Date.now();
     this.#pushEvent({
       eventId: `${runId}:CHAIN:start`,
@@ -673,12 +794,13 @@ var AgentInspectCallback = class extends 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({
@@ -699,13 +821,14 @@ var AgentInspectCallback = class extends 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`,
@@ -737,6 +860,7 @@ var AgentInspectCallback = class extends 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({
@@ -766,6 +890,7 @@ var AgentInspectCallback = class extends 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({
@@ -800,7 +925,6 @@ var AgentInspectCallback = class extends 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 = {};
@@ -810,6 +934,8 @@ var AgentInspectCallback = class extends 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);
@@ -838,13 +964,14 @@ var AgentInspectCallback = class extends 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();
@@ -881,6 +1008,7 @@ var AgentInspectCallback = class extends BaseCallbackHandler {
     };
     this.#mergeMetadata(attrs, metadata);
     this.#applyPreview(attrs, previews);
+    this.#rememberStartMetadata(runId, attrs);
     const ts = Date.now();
     const stepName = `tool:${toolName}`;
     this.#pushEvent({
@@ -900,12 +1028,13 @@ var AgentInspectCallback = class extends 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({
@@ -926,13 +1055,14 @@ var AgentInspectCallback = class extends 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`,
@@ -958,6 +1088,7 @@ var AgentInspectCallback = class extends BaseCallbackHandler {
       retriever: rname
     };
     this.#mergeMetadata(attrs, metadata);
+    this.#rememberStartMetadata(runId, attrs);
     const ts = Date.now();
     const stepName = `retriever:${rname}`;
     this.#pushEvent({
@@ -977,7 +1108,6 @@ var AgentInspectCallback = class extends 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);
@@ -986,6 +1116,8 @@ var AgentInspectCallback = class extends 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({
@@ -1006,13 +1138,14 @@ var AgentInspectCallback = class extends 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`,