@agent-inspect/langchain 6.7.2 → 6.7.4

package/dist/index.mjs

@@ -349,6 +349,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 +374,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 +395,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 +469,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 +482,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 +521,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 +612,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 };
@@ -1017,7 +1065,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 = {