@agent-inspect/langchain 6.7.3 → 6.7.5

package/dist/index.cjs

@@ -1,9 +1,14 @@
 'use strict';
 
+var path = require('path');
 var base = require('@langchain/core/callbacks/base');
 var advanced = require('agent-inspect/advanced');
 var logs = require('agent-inspect/logs');
 
+function _interopDefault (e) { return e && e.__esModule ? e : { default: e }; }
+
+var path__default = /*#__PURE__*/_interopDefault(path);
+
 // packages/langchain/src/agent-inspect-callback.ts
 
 // packages/langchain/src/metadata.ts
@@ -351,6 +356,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 = advanced.hasActiveContext();
@@ -371,10 +381,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) {
@@ -388,7 +402,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);
       }
@@ -460,13 +476,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);
     }
@@ -474,7 +489,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);
       }
@@ -511,10 +528,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;
@@ -575,6 +619,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 };
@@ -722,7 +775,11 @@ var AgentInspectCallback = class extends base.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__default.default.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;
@@ -1019,7 +1076,7 @@ var AgentInspectCallback = class extends base.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 = {