bitfab 0.57.6 → 0.57.7

package/dist/{chunk-UTBA3ASY.js → chunk-P4R3NY2U.js}

@@ -187,7 +187,7 @@ function encodeRequestBody(body) {
 }
 
 // src/version.generated.ts
-var __version__ = "0.57.6";
+var __version__ = "0.57.7";
 var __packageName__ = "bitfab";
 
 // src/constants.ts
@@ -788,7 +788,13 @@ var SIMULATION_PLAN_MAX_HELD = 1e3;
 var DISABLE_SIMULATION_PLAN_ENV = "BITFAB_DISABLE_SIM_PLAN";
 var ROOT_TRACE_FUNCTION_KEY_FIELD = "rootTraceFunctionKey";
 var CONTENT_OFF_KEY = "content_off_by_simulation_plan";
-var CONTENT_KEYS = ["input", "input_meta", "output", "output_meta"];
+var CONTENT_KEYS = [
+  "input",
+  "input_meta",
+  "output",
+  "output_meta",
+  "prompt"
+];
 function planWaitSliceMs(timeoutMs) {
   return Math.min(SIMULATION_PLAN_READ_TIMEOUT_MS, Math.max(timeoutMs, 0) / 2);
 }
@@ -982,13 +988,13 @@ var SimulationPlan = class {
     }
     return this.loaded;
   }
-  send(payload, traceFunctionKey, submit) {
+  send(payload, traceFunctionKey, submit, drop) {
     this.refresh();
     if (this.disabled() || traceFunctionKey === void 0 || spanNameOf(payload) === void 0 || recordedByFramework(payload)) {
       submit(payload);
       return;
     }
-    const entry = { payload, traceFunctionKey, submit };
+    const entry = { payload, traceFunctionKey, submit, drop };
     if (this.holdingForFirstRead()) {
       this.hold(entry);
       return;
@@ -1042,6 +1048,7 @@ var SimulationPlan = class {
     try {
       entry.submit(this.payloadToSend(entry));
     } catch (error) {
+      entry.drop?.();
       warnOnce(
         "sim-plan-held-span-dropped",
         `a span held for the sim plan was dropped when released: ${error instanceof Error ? error.message : String(error)}`
@@ -2312,23 +2319,21 @@ var HttpClient = class {
     }
     const deadline = Date.now() + Math.max(timeoutMs, 0);
     this.closing = (async () => {
-      const released = await this.releaseHeldExternalSpans?.(
-        planWaitSliceMs(timeoutMs)
+      liveHttpClients.delete(this);
+      const settled = await this.settleDeferredWork(
+        Math.max(0, deadline - Date.now())
       );
       this.stopSimulationPlan?.();
+      this.closed = true;
       this.externalSpanTransform = void 0;
       this.externalTraceTransform = void 0;
+      this.frameworkSpanStart = void 0;
       this.releaseHeldExternalSpans = void 0;
       this.stopSimulationPlan = void 0;
-      liveHttpClients.delete(this);
-      const settled = await this.settleDeferredWork(
-        Math.max(0, deadline - Date.now())
-      );
-      this.closed = true;
       const transport = this.traceTransport;
       this.traceTransport = void 0;
       const shutdownOk = await transport?.shutdown(Math.max(0, deadline - Date.now())) ?? true;
-      return released !== false && settled && shutdownOk;
+      return settled && shutdownOk;
     })();
     return this.closing;
   }
@@ -2429,8 +2434,7 @@ var HttpClient = class {
   async getSimulationPlan() {
     return this.get(
       "/api/sdk/sim-plan",
-      SIMULATION_PLAN_READ_TIMEOUT_MS,
-      { Connection: "close" }
+      SIMULATION_PLAN_READ_TIMEOUT_MS
     );
   }
   async getTraceSpan(traceId, lookup) {
@@ -2506,19 +2510,35 @@ var HttpClient = class {
       carrierMeta("internal_trace", body, void 0)
     );
   }
+  reserveFrameworkSpan(traceId, spanId, parentSpanId) {
+    try {
+      this.frameworkSpanStart?.(traceId, spanId, parentSpanId ?? null);
+    } catch (error) {
+      warnOnce(
+        "framework-span-reservation-failed",
+        `a framework span could not be counted toward its trace's span limit: ${describeError(error)}`
+      );
+    }
+  }
   sendExternalSpan(payload) {
     const ref = carrierRef(payload);
     if (ref?.spanId !== void 0) {
       this.traceCompletion.record(ref.traceId, ref.spanId);
     }
+    const drop = () => {
+      if (ref?.spanId !== void 0) {
+        this.traceCompletion.abort(ref.traceId, ref.spanId);
+      }
+    };
     const gate = this.externalSpanTransform;
     if (gate === void 0) {
       this.submitExternalSpan(payload);
       return;
     }
     try {
-      gate(payload, (ready) => this.submitExternalSpan(ready));
+      gate(payload, (ready) => this.submitExternalSpan(ready), drop);
     } catch (error) {
+      drop();
       warnDroppedExternalSpan(error);
     }
   }
@@ -2837,4 +2857,4 @@ export {
   parseRetryAfterMs,
   HttpClient
 };
-//# sourceMappingURL=chunk-UTBA3ASY.js.map
+//# sourceMappingURL=chunk-P4R3NY2U.js.map