@agent-inspect/langchain 1.0.3 → 1.1.0

package/dist/index.mjs

@@ -1,5 +1,5 @@
 import { BaseCallbackHandler } from '@langchain/core/callbacks/base';
-import { Redactor } from 'agent-inspect';
+import { Redactor, hasActiveContext, getTraceDirFromContext, resolveTraceDir, getCurrentRunId, createRunId, resolveTraceSafetyOptions, createStepId, initializeTraceFile, prepareTraceEventForDisk, writeTraceEvent } from 'agent-inspect';
 
 // packages/langchain/src/agent-inspect-callback.ts
 
@@ -145,6 +145,217 @@ function toPlainMetadata(value) {
     return {};
   }
 }
+function kindToStepType(kind) {
+  switch (kind) {
+    case "LLM":
+      return "llm";
+    case "TOOL":
+      return "tool";
+    case "DECISION":
+      return "decision";
+    default:
+      return "logic";
+  }
+}
+function toStepMetadata(attrs) {
+  const out = {
+    adapter: "langchain",
+    confidence: "explicit"
+  };
+  for (const [k, v] of Object.entries(attrs)) {
+    out[k] = v;
+  }
+  return out;
+}
+var LangChainTracePersistence = class {
+  #traceDir;
+  #runId;
+  #runName;
+  #standalone;
+  #silent;
+  #safety;
+  #runStarted = false;
+  #runCompleted = false;
+  #runStartTime;
+  #rootLcRunId;
+  #lcToStepId = /* @__PURE__ */ new Map();
+  constructor(options = {}) {
+    const inContext = hasActiveContext();
+    this.#standalone = !inContext;
+    this.#silent = options.silent ?? false;
+    this.#traceDir = inContext ? getTraceDirFromContext() ?? resolveTraceDir({ dir: options.traceDir }) : resolveTraceDir({ dir: options.traceDir });
+    this.#runId = (inContext ? getCurrentRunId() : void 0) ?? options.runId ?? createRunId();
+    this.#runName = options.runName ?? "langchain-agent";
+    this.#safety = resolveTraceSafetyOptions({
+      redact: options.redact ? { rules: options.redact } : true,
+      maxPreviewLength: options.maxPreviewChars
+    });
+  }
+  get runId() {
+    return this.#runId;
+  }
+  get traceDir() {
+    return this.#traceDir;
+  }
+  reset() {
+    this.#runStarted = false;
+    this.#runCompleted = false;
+    this.#runStartTime = void 0;
+    this.#rootLcRunId = void 0;
+    this.#lcToStepId.clear();
+  }
+  noteRoot(lcRunId, parentRunId) {
+    if (!parentRunId && !this.#rootLcRunId) {
+      this.#rootLcRunId = lcRunId;
+    }
+  }
+  resolveParentId(lcParentRunId) {
+    if (!lcParentRunId) return void 0;
+    return this.#lcToStepId.get(lcParentRunId);
+  }
+  async onStepStart(params) {
+    try {
+      this.noteRoot(params.lcRunId, params.lcParentRunId);
+      if (this.#standalone && !this.#runStarted) {
+        await this.#ensureRunStarted(params.startTime, params.attributes);
+      }
+      const stepId = createStepId();
+      this.#lcToStepId.set(params.lcRunId, stepId);
+      const parentId = this.resolveParentId(params.lcParentRunId);
+      const event = {
+        schemaVersion: "0.1",
+        event: "step_started",
+        timestamp: params.startTime,
+        runId: this.#runId,
+        stepId,
+        ...parentId ? { parentId } : {},
+        name: params.name,
+        type: kindToStepType(params.kind),
+        startTime: params.startTime,
+        metadata: toStepMetadata(params.attributes)
+      };
+      await this.#write(event);
+    } catch (err) {
+      this.#warn(err);
+    }
+  }
+  async onStepEnd(params) {
+    try {
+      const stepId = this.#lcToStepId.get(params.lcRunId);
+      if (!stepId) return;
+      const durationMs = typeof params.durationMs === "number" && Number.isFinite(params.durationMs) ? Math.max(0, Math.floor(params.durationMs)) : Math.max(0, params.endTime - (this.#runStartTime ?? params.endTime));
+      const event = {
+        schemaVersion: "0.1",
+        event: "step_completed",
+        timestamp: params.endTime,
+        runId: this.#runId,
+        stepId,
+        status: params.status,
+        endTime: params.endTime,
+        durationMs,
+        ...params.status === "error" && params.errorMessage ? { error: { message: params.errorMessage } } : {}
+      };
+      await this.#write(event);
+      if (this.#standalone && !this.#runCompleted && this.#rootLcRunId === params.lcRunId && !params.lcParentRunId) {
+        await this.#ensureRunCompleted(
+          params.endTime,
+          params.status,
+          params.errorMessage
+        );
+      }
+    } catch (err) {
+      this.#warn(err);
+    }
+  }
+  /** Point-in-time adapter events (e.g. agent action) — writes start + completed pair. */
+  async onInstantStep(params) {
+    try {
+      this.noteRoot(params.lcRunId, params.lcParentRunId);
+      if (this.#standalone && !this.#runStarted) {
+        await this.#ensureRunStarted(params.timestamp, params.attributes);
+      }
+      const stepId = createStepId();
+      this.#lcToStepId.set(params.lcRunId, stepId);
+      const parentId = this.resolveParentId(params.lcParentRunId);
+      const started = {
+        schemaVersion: "0.1",
+        event: "step_started",
+        timestamp: params.timestamp,
+        runId: this.#runId,
+        stepId,
+        ...parentId ? { parentId } : {},
+        name: params.name,
+        type: kindToStepType(params.kind),
+        startTime: params.timestamp,
+        metadata: toStepMetadata(params.attributes)
+      };
+      await this.#write(started);
+      const completed = {
+        schemaVersion: "0.1",
+        event: "step_completed",
+        timestamp: params.timestamp,
+        runId: this.#runId,
+        stepId,
+        status: params.status,
+        endTime: params.timestamp,
+        durationMs: 0,
+        ...params.status === "error" && params.errorMessage ? { error: { message: params.errorMessage } } : {}
+      };
+      await this.#write(completed);
+    } catch (err) {
+      this.#warn(err);
+    }
+  }
+  async #ensureRunStarted(startTime, attrs) {
+    if (this.#runStarted) return;
+    this.#runStarted = true;
+    this.#runStartTime = startTime;
+    await initializeTraceFile(this.#runId, this.#traceDir);
+    const metadata = {
+      adapter: "langchain",
+      confidence: "explicit"
+    };
+    if (attrs.langchainRunId) metadata.langchainRunId = attrs.langchainRunId;
+    if (attrs.adapterRunName) metadata.adapterRunName = attrs.adapterRunName;
+    const event = {
+      schemaVersion: "0.1",
+      event: "run_started",
+      timestamp: startTime,
+      runId: this.#runId,
+      name: this.#runName,
+      startTime,
+      metadata
+    };
+    await this.#write(event);
+  }
+  async #ensureRunCompleted(endTime, stepStatus, errorMessage) {
+    if (this.#runCompleted || !this.#runStarted) return;
+    this.#runCompleted = true;
+    const startTime = this.#runStartTime ?? endTime;
+    const durationMs = Math.max(0, endTime - startTime);
+    const runStatus = stepStatus === "error" ? "error" : "success";
+    const event = {
+      schemaVersion: "0.1",
+      event: "run_completed",
+      timestamp: endTime,
+      runId: this.#runId,
+      status: runStatus,
+      endTime,
+      durationMs,
+      ...runStatus === "error" && errorMessage ? { error: { message: errorMessage } } : {}
+    };
+    await this.#write(event);
+  }
+  async #write(event) {
+    const safe = prepareTraceEventForDisk(event, this.#safety);
+    await writeTraceEvent(safe, this.#traceDir);
+  }
+  #warn(err) {
+    if (!this.#silent) {
+      console.error("[agent-inspect:langchain]", err);
+    }
+  }
+};
 
 // packages/langchain/src/agent-inspect-callback.ts
 function serializedLabel(s) {
@@ -162,6 +373,7 @@ var AgentInspectCallback = class extends BaseCallbackHandler {
   name = "agent-inspect";
   #opts;
   #redactor;
+  #persistence;
   #events = [];
   #starts = /* @__PURE__ */ new Map();
   #rootRunId;
@@ -171,9 +383,21 @@ var AgentInspectCallback = class extends BaseCallbackHandler {
       capture: options.capture ?? "metadata-only",
       silent: options.silent ?? false,
       maxPreviewChars: options.maxPreviewChars ?? 200,
+      persist: options.persist ?? false,
+      runName: options.runName ?? "langchain-agent",
       ...options
     };
     this.#redactor = new Redactor({ rules: this.#opts.redact });
+    if (this.#opts.persist) {
+      this.#persistence = new LangChainTracePersistence({
+        runName: this.#opts.runName,
+        traceDir: this.#opts.traceDir,
+        runId: this.#opts.runId,
+        redact: this.#opts.redact,
+        silent: this.#opts.silent,
+        maxPreviewChars: this.#opts.maxPreviewChars
+      });
+    }
   }
   getEvents() {
     return this.#events.map((e) => ({
@@ -186,6 +410,7 @@ var AgentInspectCallback = class extends BaseCallbackHandler {
     this.#events = [];
     this.#starts.clear();
     this.#rootRunId = void 0;
+    this.#persistence?.reset();
   }
   #ensureRoot(lcRunId, parentRunId) {
     if (parentRunId) return;
@@ -239,6 +464,38 @@ var AgentInspectCallback = class extends BaseCallbackHandler {
       }
     }
   }
+  async #persistStepStart(lcRunId, lcParentRunId, name, kind, attrs, startTime) {
+    await this.#persistence?.onStepStart({
+      lcRunId,
+      lcParentRunId,
+      name,
+      kind,
+      startTime,
+      attributes: attrs
+    });
+  }
+  async #persistStepEnd(lcRunId, lcParentRunId, status, endTime, durationMs, errorMessage) {
+    await this.#persistence?.onStepEnd({
+      lcRunId,
+      lcParentRunId,
+      endTime,
+      durationMs,
+      status,
+      errorMessage
+    });
+  }
+  async #persistInstant(lcRunId, lcParentRunId, name, kind, attrs, status, errorMessage) {
+    await this.#persistence?.onInstantStep({
+      lcRunId,
+      lcParentRunId,
+      name,
+      kind,
+      timestamp: Date.now(),
+      attributes: attrs,
+      status,
+      errorMessage
+    });
+  }
   async handleChainStart(chain, inputs, runId, runType, tags, metadata, runName, parentRunId, _extra) {
     this.#ensureRoot(runId, parentRunId);
     this.#rememberStart(runId, "CHAIN");
@@ -250,18 +507,20 @@ var AgentInspectCallback = class extends BaseCallbackHandler {
     };
     this.#mergeMetadata(attrs, metadata);
     this.#applyPreview(attrs, previews);
+    const ts = Date.now();
     this.#pushEvent({
       eventId: `${runId}:CHAIN:start`,
       runId: this.#traceRunId(runId),
       parentId: parentRunId,
       name: `chain:${runName ?? label}`,
       kind: "CHAIN",
-      timestamp: Date.now(),
+      timestamp: ts,
       status: "running",
       attributes: attrs,
       confidence: "explicit",
       source: { type: "adapter" }
     });
+    await this.#persistStepStart(runId, parentRunId, `chain:${runName ?? label}`, "CHAIN", attrs, ts);
   }
   async handleChainEnd(outputs, runId, parentRunId, tags, _kwargs) {
     this.#ensureRoot(runId, parentRunId);
@@ -273,19 +532,21 @@ var AgentInspectCallback = class extends BaseCallbackHandler {
       ...this.#baseAttrs(runId, parentRunId, tags, void 0)
     };
     this.#applyPreview(attrs, previews);
+    const ts = Date.now();
     this.#pushEvent({
       eventId: `${runId}:CHAIN:end`,
       runId: this.#traceRunId(runId),
       parentId: parentRunId,
       name: "chain:end",
       kind: "CHAIN",
-      timestamp: Date.now(),
+      timestamp: ts,
       status: "ok",
       durationMs,
       attributes: attrs,
       confidence: "explicit",
       source: { type: "adapter" }
     });
+    await this.#persistStepEnd(runId, parentRunId, "success", ts, durationMs);
   }
   async handleChainError(err, runId, parentRunId, tags, _kwargs) {
     this.#ensureRoot(runId, parentRunId);
@@ -297,19 +558,21 @@ var AgentInspectCallback = class extends BaseCallbackHandler {
       errorName,
       errorMessage
     };
+    const ts = Date.now();
     this.#pushEvent({
       eventId: `${runId}:CHAIN:error`,
       runId: this.#traceRunId(runId),
       parentId: parentRunId,
       name: "chain:error",
       kind: "CHAIN",
-      timestamp: Date.now(),
+      timestamp: ts,
       status: "error",
       durationMs,
       attributes: attrs,
       confidence: "explicit",
       source: { type: "adapter" }
     });
+    await this.#persistStepEnd(runId, parentRunId, "error", ts, durationMs, errorMessage);
   }
   async handleLLMStart(llm, prompts, runId, parentRunId, _extraParams, tags, metadata, runName) {
     this.#ensureRoot(runId, parentRunId);
@@ -325,18 +588,21 @@ var AgentInspectCallback = class extends BaseCallbackHandler {
     if (model && this.#opts.capture !== "none") attrs.model = model;
     this.#mergeMetadata(attrs, metadata);
     this.#applyPreview(attrs, previews);
+    const ts = Date.now();
+    const stepName = `llm:${model ?? "llm"}`;
     this.#pushEvent({
       eventId: `${runId}:LLM:start`,
       runId: this.#traceRunId(runId),
       parentId: parentRunId,
-      name: `llm:${model ?? "llm"}`,
+      name: stepName,
       kind: "LLM",
-      timestamp: Date.now(),
+      timestamp: ts,
       status: "running",
       attributes: attrs,
       confidence: "explicit",
       source: { type: "adapter" }
     });
+    await this.#persistStepStart(runId, parentRunId, stepName, "LLM", attrs, ts);
   }
   async handleChatModelStart(llm, messages, runId, parentRunId, _extraParams, tags, metadata, runName) {
     this.#ensureRoot(runId, parentRunId);
@@ -350,18 +616,21 @@ var AgentInspectCallback = class extends BaseCallbackHandler {
     if (model && this.#opts.capture !== "none") attrs.model = model;
     this.#mergeMetadata(attrs, metadata);
     this.#applyPreview(attrs, previews);
+    const ts = Date.now();
+    const stepName = `llm:${model ?? "llm"}`;
     this.#pushEvent({
       eventId: `${runId}:CHAT:start`,
       runId: this.#traceRunId(runId),
       parentId: parentRunId,
-      name: `llm:${model ?? "llm"}`,
+      name: stepName,
       kind: "LLM",
-      timestamp: Date.now(),
+      timestamp: ts,
       status: "running",
       attributes: attrs,
       confidence: "explicit",
       source: { type: "adapter" }
     });
+    await this.#persistStepStart(runId, parentRunId, stepName, "LLM", attrs, ts);
   }
   async handleLLMEnd(output, runId, parentRunId, tags, _extraParams) {
     this.#ensureRoot(runId, parentRunId);
@@ -377,19 +646,21 @@ var AgentInspectCallback = class extends BaseCallbackHandler {
     if (model && this.#opts.capture !== "none") attrs.model = model;
     if (tokens && this.#opts.capture !== "none") attrs.tokens = tokens;
     this.#applyPreview(attrs, previews);
+    const ts = Date.now();
     this.#pushEvent({
       eventId: `${runId}:LLM:end`,
       runId: this.#traceRunId(runId),
       parentId: parentRunId,
       name: `llm:${model ?? "llm"}`,
       kind: "LLM",
-      timestamp: Date.now(),
+      timestamp: ts,
       status: "ok",
       durationMs,
       attributes: attrs,
       confidence: "explicit",
       source: { type: "adapter" }
     });
+    await this.#persistStepEnd(runId, parentRunId, "success", ts, durationMs);
   }
   async handleLLMError(err, runId, parentRunId, tags, _extraParams) {
     this.#ensureRoot(runId, parentRunId);
@@ -401,19 +672,21 @@ var AgentInspectCallback = class extends BaseCallbackHandler {
       errorName,
       errorMessage
     };
+    const ts = Date.now();
     this.#pushEvent({
       eventId: `${runId}:LLM:error`,
       runId: this.#traceRunId(runId),
       parentId: parentRunId,
       name: "llm:error",
       kind: "LLM",
-      timestamp: Date.now(),
+      timestamp: ts,
       status: "error",
       durationMs,
       attributes: attrs,
       confidence: "explicit",
       source: { type: "adapter" }
     });
+    await this.#persistStepEnd(runId, parentRunId, "error", ts, durationMs, errorMessage);
   }
   async handleToolStart(tool, input, runId, parentRunId, tags, metadata, runName, _toolCallId) {
     this.#ensureRoot(runId, parentRunId);
@@ -427,18 +700,21 @@ var AgentInspectCallback = class extends BaseCallbackHandler {
     };
     this.#mergeMetadata(attrs, metadata);
     this.#applyPreview(attrs, previews);
+    const ts = Date.now();
+    const stepName = `tool:${toolName}`;
     this.#pushEvent({
       eventId: `${runId}:TOOL:start`,
       runId: this.#traceRunId(runId),
       parentId: parentRunId,
-      name: `tool:${toolName}`,
+      name: stepName,
       kind: "TOOL",
-      timestamp: Date.now(),
+      timestamp: ts,
       status: "running",
       attributes: attrs,
       confidence: "explicit",
       source: { type: "adapter" }
     });
+    await this.#persistStepStart(runId, parentRunId, stepName, "TOOL", attrs, ts);
   }
   async handleToolEnd(output, runId, parentRunId, tags) {
     this.#ensureRoot(runId, parentRunId);
@@ -450,19 +726,21 @@ var AgentInspectCallback = class extends BaseCallbackHandler {
       ...this.#baseAttrs(runId, parentRunId, tags, void 0)
     };
     this.#applyPreview(attrs, previews);
+    const ts = Date.now();
     this.#pushEvent({
       eventId: `${runId}:TOOL:end`,
       runId: this.#traceRunId(runId),
       parentId: parentRunId,
       name: "tool:end",
       kind: "TOOL",
-      timestamp: Date.now(),
+      timestamp: ts,
       status: "ok",
       durationMs,
       attributes: attrs,
       confidence: "explicit",
       source: { type: "adapter" }
     });
+    await this.#persistStepEnd(runId, parentRunId, "success", ts, durationMs);
   }
   async handleToolError(err, runId, parentRunId, tags) {
     this.#ensureRoot(runId, parentRunId);
@@ -474,19 +752,21 @@ var AgentInspectCallback = class extends BaseCallbackHandler {
       errorName,
       errorMessage
     };
+    const ts = Date.now();
     this.#pushEvent({
       eventId: `${runId}:TOOL:error`,
       runId: this.#traceRunId(runId),
       parentId: parentRunId,
       name: "tool:error",
       kind: "TOOL",
-      timestamp: Date.now(),
+      timestamp: ts,
       status: "error",
       durationMs,
       attributes: attrs,
       confidence: "explicit",
       source: { type: "adapter" }
     });
+    await this.#persistStepEnd(runId, parentRunId, "error", ts, durationMs, errorMessage);
   }
   async handleRetrieverStart(retriever, _query, runId, parentRunId, tags, metadata, name) {
     this.#ensureRoot(runId, parentRunId);
@@ -497,18 +777,21 @@ var AgentInspectCallback = class extends BaseCallbackHandler {
       retriever: rname
     };
     this.#mergeMetadata(attrs, metadata);
+    const ts = Date.now();
+    const stepName = `retriever:${rname}`;
     this.#pushEvent({
       eventId: `${runId}:RETRIEVER:start`,
       runId: this.#traceRunId(runId),
       parentId: parentRunId,
-      name: `retriever:${rname}`,
+      name: stepName,
       kind: "RETRIEVER",
-      timestamp: Date.now(),
+      timestamp: ts,
       status: "running",
       attributes: attrs,
       confidence: "explicit",
       source: { type: "adapter" }
     });
+    await this.#persistStepStart(runId, parentRunId, stepName, "RETRIEVER", attrs, ts);
   }
   async handleRetrieverEnd(documents, runId, parentRunId, tags) {
     this.#ensureRoot(runId, parentRunId);
@@ -523,19 +806,21 @@ var AgentInspectCallback = class extends BaseCallbackHandler {
       documentCount: documents.length
     };
     this.#applyPreview(attrs, previews);
+    const ts = Date.now();
     this.#pushEvent({
       eventId: `${runId}:RETRIEVER:end`,
       runId: this.#traceRunId(runId),
       parentId: parentRunId,
       name: "retriever:end",
       kind: "RETRIEVER",
-      timestamp: Date.now(),
+      timestamp: ts,
       status: "ok",
       durationMs,
       attributes: attrs,
       confidence: "explicit",
       source: { type: "adapter" }
     });
+    await this.#persistStepEnd(runId, parentRunId, "success", ts, durationMs);
   }
   async handleRetrieverError(err, runId, parentRunId, tags) {
     this.#ensureRoot(runId, parentRunId);
@@ -547,19 +832,21 @@ var AgentInspectCallback = class extends BaseCallbackHandler {
       errorName,
       errorMessage
     };
+    const ts = Date.now();
     this.#pushEvent({
       eventId: `${runId}:RETRIEVER:error`,
       runId: this.#traceRunId(runId),
       parentId: parentRunId,
       name: "retriever:error",
       kind: "RETRIEVER",
-      timestamp: Date.now(),
+      timestamp: ts,
       status: "error",
       durationMs,
       attributes: attrs,
       confidence: "explicit",
       source: { type: "adapter" }
     });
+    await this.#persistStepEnd(runId, parentRunId, "error", ts, durationMs, errorMessage);
   }
   async handleAgentAction(action, runId, parentRunId, tags) {
     this.#ensureRoot(runId, parentRunId);
@@ -585,6 +872,7 @@ var AgentInspectCallback = class extends BaseCallbackHandler {
       confidence: "explicit",
       source: { type: "adapter" }
     });
+    await this.#persistInstant(runId, parentRunId, "agent:action", "DECISION", attrs, "success");
   }
   async handleAgentEnd(finish, runId, parentRunId, tags) {
     this.#ensureRoot(runId, parentRunId);
@@ -609,6 +897,7 @@ var AgentInspectCallback = class extends BaseCallbackHandler {
       confidence: "explicit",
       source: { type: "adapter" }
     });
+    await this.#persistInstant(runId, parentRunId, "agent:end", "AGENT", attrs, "success");
   }
 };