@agent-inspect/langchain 6.7.3 → 6.7.5

package/dist/index.mjs

@@ -1,3 +1,4 @@
+import path from 'path';
 import { BaseCallbackHandler } from '@langchain/core/callbacks/base';
 import { getCurrentCorrelationMetadata, hasActiveContext, getTraceDirFromContext, resolveTraceDir, getCurrentRunId, createRunId, resolveTraceSafetyOptions, createStepId, initializeTraceFile, prepareTraceEventForDisk, writeTraceEvent } from 'agent-inspect/advanced';
 import { Redactor } from 'agent-inspect/logs';
@@ -349,6 +350,11 @@ var LangChainTracePersistence = class {
   #runCompleted = false;
   #runStartTime;
   #rootLcRunId;
+  /** Live LangChain callback runIds for standalone envelope finalization. */
+  #activeLcRunIds = /* @__PURE__ */ new Set();
+  #sawError = false;
+  #lastErrorMessage;
+  #finalizationToken = 0;
   #lcToStepId = /* @__PURE__ */ new Map();
   constructor(options = {}) {
     const inContext = hasActiveContext();
@@ -369,10 +375,14 @@ var LangChainTracePersistence = class {
     return this.#traceDir;
   }
   reset() {
+    this.#finalizationToken += 1;
     this.#runStarted = false;
     this.#runCompleted = false;
     this.#runStartTime = void 0;
     this.#rootLcRunId = void 0;
+    this.#activeLcRunIds.clear();
+    this.#sawError = false;
+    this.#lastErrorMessage = void 0;
     this.#lcToStepId.clear();
   }
   noteRoot(lcRunId, parentRunId) {
@@ -386,7 +396,9 @@ var LangChainTracePersistence = class {
   }
   async onStepStart(params) {
     try {
+      this.#finalizationToken += 1;
       this.noteRoot(params.lcRunId, params.lcParentRunId);
+      this.#activeLcRunIds.add(params.lcRunId);
       if (this.#standalone && !this.#runStarted) {
         await this.#ensureRunStarted(params.startTime, params.attributes);
       }
@@ -458,13 +470,12 @@ var LangChainTracePersistence = class {
         ...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
-        );
+      if (params.status === "error") {
+        this.#sawError = true;
+        if (params.errorMessage) this.#lastErrorMessage = params.errorMessage;
       }
+      this.#activeLcRunIds.delete(params.lcRunId);
+      await this.#scheduleStandaloneFinalization(params.endTime);
     } catch (err) {
       this.#warn(err);
     }
@@ -472,7 +483,9 @@ var LangChainTracePersistence = class {
   /** Point-in-time adapter events (e.g. agent action) — writes start + completed pair. */
   async onInstantStep(params) {
     try {
+      this.#finalizationToken += 1;
       this.noteRoot(params.lcRunId, params.lcParentRunId);
+      this.#activeLcRunIds.add(params.lcRunId);
       if (this.#standalone && !this.#runStarted) {
         await this.#ensureRunStarted(params.timestamp, params.attributes);
       }
@@ -509,10 +522,37 @@ var LangChainTracePersistence = class {
         ...params.status === "error" && params.errorMessage ? { error: { message: params.errorMessage } } : {}
       };
       await this.#write(completed);
+      if (params.status === "error") {
+        this.#sawError = true;
+        if (params.errorMessage) this.#lastErrorMessage = params.errorMessage;
+      }
+      this.#activeLcRunIds.delete(params.lcRunId);
+      await this.#scheduleStandaloneFinalization(params.timestamp);
     } catch (err) {
       this.#warn(err);
     }
   }
+  /**
+   * When the last active LangChain callback ends, yield one microtask so a
+   * same-turn sibling start can cancel finalization, then write run_completed
+   * before the callback promise settles. Unresolved external parents do not
+   * block the envelope.
+   */
+  async #scheduleStandaloneFinalization(endTime) {
+    if (!this.#standalone || this.#runCompleted || !this.#runStarted) return;
+    if (this.#activeLcRunIds.size > 0) return;
+    const token = ++this.#finalizationToken;
+    await Promise.resolve();
+    if (token !== this.#finalizationToken) return;
+    if (!this.#standalone || this.#runCompleted || !this.#runStarted) return;
+    if (this.#activeLcRunIds.size > 0) return;
+    const status = this.#sawError ? "error" : "success";
+    await this.#ensureRunCompleted(
+      endTime,
+      status,
+      status === "error" ? this.#lastErrorMessage : void 0
+    );
+  }
   async #ensureRunStarted(startTime, attrs) {
     if (this.#runStarted) return;
     this.#runStarted = true;
@@ -573,6 +613,15 @@ function serializedLabel(s) {
   if (Array.isArray(s.id) && s.id.length > 0) return s.id[s.id.length - 1];
   return s.type;
 }
+function resolveToolDisplayName(tool, runName, metadata) {
+  const fromRun = typeof runName === "string" ? runName.trim() : "";
+  if (fromRun) return fromRun;
+  const metaName = metadata?.toolName;
+  if (typeof metaName === "string" && metaName.trim()) return metaName.trim();
+  const metaTool = metadata?.tool;
+  if (typeof metaTool === "string" && metaTool.trim()) return metaTool.trim();
+  return serializedLabel(tool) ?? "tool";
+}
 function errorShape(err) {
   if (err instanceof Error) {
     return { errorName: err.name, errorMessage: err.message };
@@ -720,7 +769,11 @@ var AgentInspectCallback = class extends BaseCallbackHandler {
     if (parentRunId) out.parentRunId = parentRunId;
     if (this.#opts.runName) out.adapterRunName = this.#opts.runName;
     if (runNameArg) out.runName = runNameArg;
-    if (this.#opts.traceDir) out.traceDir = this.#opts.traceDir;
+    if (this.#opts.persist) out.traceStorage = "local";
+    const configuredTraceDir = this.#opts.traceDir?.trim();
+    if (configuredTraceDir && !path.isAbsolute(configuredTraceDir) && !/^[A-Za-z]:[\\/]/.test(configuredTraceDir)) {
+      out.workspaceRelativeTraceDir = configuredTraceDir;
+    }
     const cap = this.#opts.capture;
     if (cap !== "none" && tags?.length) out.tags = [...tags];
     return out;
@@ -1017,7 +1070,7 @@ var AgentInspectCallback = class extends BaseCallbackHandler {
   async handleToolStart(tool, input, runId, parentRunId, tags, metadata, runName, _toolCallId) {
     this.#ensureRoot(runId, parentRunId);
     this.#rememberStart(runId, "TOOL");
-    const toolName = serializedLabel(tool) ?? "tool";
+    const toolName = resolveToolDisplayName(tool, runName, metadata);
     const previews = {};
     if (this.#opts.capture === "preview") previews.inputPreview = input;
     const attrs = {