langwatch 1.11.0 → 1.12.0

package/dist/agent/index.mjs

@@ -44,7 +44,7 @@ import * as os from "os";
 import { randomUUID } from "crypto";
 
 // package.json
-var version = "1.11.0";
+var version = "1.12.0";
 
 // src/internal/constants.ts
 var LANGWATCH_SDK_VERSION = version;
@@ -1169,6 +1169,8 @@ var AgentClient = class {
     try {
       const result = await context.with(parent, () => runtime.run(call));
       if (entry.cancelled) return;
+      this.releaseCall({ callId: frame.callId, entry });
+      await this.flushSpans();
       this.send({
         type: "result",
         protocol: PROTOCOL_VERSION,
@@ -1181,11 +1183,34 @@ var AgentClient = class {
       const code = error instanceof AgentParameterError ? error.code : "agent_call_failed";
       const message = describeError(error);
       this.logger.warn(`agent "${runtime.name}" call ${frame.callId} failed: ${message}`);
+      this.releaseCall({ callId: frame.callId, entry });
+      await this.flushSpans();
       this.sendError({ callId: frame.callId, code, message });
     } finally {
       this.releaseCall({ callId: frame.callId, entry });
     }
   }
+  /**
+   * Exports the spans of the call now instead of at the exporter's next
+   * schedule. The judge reads the agent's spans right after the last turn,
+   * and a batch exporter would otherwise hold them for seconds, which is what
+   * made the judge report the spans missing.
+   *
+   * The call awaits this before it sends its result or its error: the frame is
+   * what tells the platform the turn is over, so a frame that goes out first
+   * lets the judge read the call while its spans are still in the exporter.
+   */
+  async flushSpans() {
+    const provider = trace.getTracerProvider();
+    const delegate = typeof provider.getDelegate === "function" ? provider.getDelegate() : provider;
+    const flush = delegate?.forceFlush;
+    if (typeof flush !== "function") return;
+    try {
+      await flush.call(delegate);
+    } catch (error) {
+      this.logger.debug(`span flush after a call failed: ${describeError(error)}`);
+    }
+  }
   /** Frees the slot the call holds and drops its deadline timer. */
   releaseCall({ callId, entry }) {
     if (entry.timer) {