@mastra/observability 1.20.0 → 1.20.1-alpha.1

package/dist/index.js

@@ -1287,7 +1287,8 @@ const BEDROCK_GEOGRAPHY_PREFIXES = /* @__PURE__ */ new Set([
 	"eu",
 	"apac",
 	"jp",
-	"au"
+	"au",
+	"in"
 ]);
 const AI_SDK_VERCEL_GATEWAY_PROVIDER_ID = "gateway";
 const AZURE_OPENAI_GATEWAY_ID = "azure-openai";
@@ -6396,6 +6397,8 @@ var TrackingExporter = class extends BaseExporter {
 	#pendingCleanups = /* @__PURE__ */ new Map();
 	/** Waiting-queue replays started by setImmediate that haven't finished yet */
 	#activeReplays = /* @__PURE__ */ new Set();
+	/** Hard-cap enforcements started by getTraceData() that haven't finished yet */
+	#activeHardCapEnforcements = /* @__PURE__ */ new Set();
 	/** Subclass configuration with resolved values */
 	config;
 	/** Maximum attempts to process a queued event before dropping */
@@ -6481,6 +6484,15 @@ var TrackingExporter = class extends BaseExporter {
 		}
 	}
 	/**
+	* Wait for fire-and-forget hard-cap enforcements, so the aborts they issue are
+	* handed to the vendor before flush() / shutdown() send or abort anything else.
+	* Only the ones running right now are joined; enforcements started by spans that
+	* arrive later belong to a later flush, so this never spins under traffic.
+	*/
+	async #drainHardCapEnforcements() {
+		await Promise.all([...this.#activeHardCapEnforcements]);
+	}
+	/**
 	* Process all events waiting for root span.
 	* Takes the queue before processing, so a concurrent call can't process the same events.
 	* @returns The number of events that left the queue (processed, moved or dropped)
@@ -7014,9 +7026,11 @@ var TrackingExporter = class extends BaseExporter {
 				traceId,
 				method
 			});
-			this.#enforceHardCap().catch((error) => {
+			const enforcement = this.#enforceHardCap().catch((error) => {
 				this.logger.error(`${this.name}: Error enforcing hard cap`, { error });
 			});
+			this.#activeHardCapEnforcements.add(enforcement);
+			enforcement.finally(() => this.#activeHardCapEnforcements.delete(enforcement));
 		}
 		return this.#traceMap.get(traceId);
 	}
@@ -7049,6 +7063,7 @@ var TrackingExporter = class extends BaseExporter {
 		if (this.isDisabled) return;
 		this.logger.debug(`${this.name}: Flushing`);
 		if (!this.#shutdownStarted) await this.#drainWaitingQueues();
+		await this.#drainHardCapEnforcements();
 		await this._flush();
 	}
 	/**
@@ -7069,6 +7084,7 @@ var TrackingExporter = class extends BaseExporter {
 		if (this.isDisabled) return;
 		this.#shutdownStarted = true;
 		await this.#drainWaitingQueues();
+		await this.#drainHardCapEnforcements();
 		await this._preShutdown();
 		for (const [traceId, timeout] of this.#pendingCleanups) {
 			clearTimeout(timeout);