@mastra/server 1.48.0-alpha.3 → 1.48.0-alpha.5

package/dist/{chunk-PVZX6FFO.cjs → chunk-3SSXARFU.cjs}

@@ -1534,7 +1534,7 @@ function getCharacterCount(text) {
   return Array.from(text).length;
 }
 
-// ../memory/dist/chunk-UHQBCO7I.js
+// ../memory/dist/chunk-X7DOZRJK.js
 var import_image_size = chunkO7I5CWRX_cjs.__toESM(require_dist(), 1);
 var MINUTE = 6e4;
 var HOUR = 60 * MINUTE;
@@ -2609,7 +2609,7 @@ var ObservationStrategy = class _ObservationStrategy {
           observabilityContext: this.opts.observabilityContext
         });
       }
-      return { observed: true, usage: output.usage };
+      return { observed: true, usage: output.usage, providerMetadata: output.providerMetadata };
     } catch (error) {
       await this.emitFailedMarkers(cycleId, error);
       if (!this.rethrowOnFailure) {
@@ -3202,6 +3202,7 @@ var ResourceScopedObservationStrategy = class extends ObservationStrategy {
   messagesByThread = /* @__PURE__ */ new Map();
   multiThreadResults = /* @__PURE__ */ new Map();
   totalBatchUsage = { inputTokens: 0, outputTokens: 0, totalTokens: 0 };
+  lastBatchProviderMetadata;
   observationResults = [];
   priorMetadataByThread = /* @__PURE__ */ new Map();
   constructor(deps, opts) {
@@ -3383,10 +3384,14 @@ var ResourceScopedObservationStrategy = class extends ObservationStrategy {
         this.totalBatchUsage.outputTokens += batchResult.usage.outputTokens ?? 0;
         this.totalBatchUsage.totalTokens += batchResult.usage.totalTokens ?? 0;
       }
+      if (batchResult.providerMetadata) {
+        this.lastBatchProviderMetadata = batchResult.providerMetadata;
+      }
     }
     return {
       observations: "",
-      usage: this.totalBatchUsage.totalTokens > 0 ? this.totalBatchUsage : void 0
+      usage: this.totalBatchUsage.totalTokens > 0 ? this.totalBatchUsage : void 0,
+      providerMetadata: this.lastBatchProviderMetadata
     };
   }
   async process(_output, existingObservations) {
@@ -5754,7 +5759,8 @@ var ObserverRunner = class {
       currentTask: parsed.currentTask,
       suggestedContinuation: parsed.suggestedContinuation,
       threadTitle: parsed.threadTitle,
-      usage: usage ? { inputTokens: usage.inputTokens, outputTokens: usage.outputTokens, totalTokens: usage.totalTokens } : void 0
+      usage: usage ? { inputTokens: usage.inputTokens, outputTokens: usage.outputTokens, totalTokens: usage.totalTokens } : void 0,
+      providerMetadata: result.providerMetadata
     };
   }
   /**
@@ -5868,7 +5874,8 @@ var ObserverRunner = class {
     const usage = result.totalUsage ?? result.usage;
     return {
       results,
-      usage: usage ? { inputTokens: usage.inputTokens, outputTokens: usage.outputTokens, totalTokens: usage.totalTokens } : void 0
+      usage: usage ? { inputTokens: usage.inputTokens, outputTokens: usage.outputTokens, totalTokens: usage.totalTokens } : void 0,
+      providerMetadata: result.providerMetadata
     };
   }
 };
@@ -6244,6 +6251,7 @@ var ReflectorRunner = class {
     const internalRequestContext = withOmInternalThreadId(requestContext, agent.id);
     const targetThreshold = observationTokensThreshold ?? getMaxThreshold(this.reflectionConfig.observationTokens);
     let totalUsage = { inputTokens: 0, outputTokens: 0, totalTokens: 0 };
+    let finalProviderMetadata;
     const startLevel = compressionStartLevel ?? 0;
     let currentLevel = startLevel;
     const maxLevel = Math.min(MAX_COMPRESSION_LEVEL, startLevel + 3);
@@ -6321,6 +6329,7 @@ var ReflectorRunner = class {
         totalUsage.outputTokens += usage.outputTokens ?? 0;
         totalUsage.totalTokens += usage.totalTokens ?? 0;
       }
+      finalProviderMetadata = result.providerMetadata ?? finalProviderMetadata;
       parsed = parseReflectorOutput(result.text, observations);
       if (parsed.degenerate) {
         omDebug(
@@ -6374,7 +6383,8 @@ var ReflectorRunner = class {
     return {
       observations: parsed.observations,
       suggestedContinuation: parsed.suggestedContinuation,
-      usage: totalUsage.totalTokens > 0 ? totalUsage : void 0
+      usage: totalUsage.totalTokens > 0 ? totalUsage : void 0,
+      providerMetadata: finalProviderMetadata
     };
   }
   /**
@@ -6391,8 +6401,11 @@ var ReflectorRunner = class {
       omError("[OM] Failed to set buffering reflection flag", err);
     });
     reflectionHooks?.onReflectionStart?.();
-    const asyncOp = this.doAsyncBufferedReflection(record, bufferKey, writer, requestContext, observabilityContext).then((usage) => {
-      reflectionHooks?.onReflectionEnd?.({ usage });
+    const asyncOp = this.doAsyncBufferedReflection(record, bufferKey, writer, requestContext, observabilityContext).then((outcome) => {
+      reflectionHooks?.onReflectionEnd?.({
+        usage: outcome?.usage,
+        ...outcome?.providerMetadata ? { providerMetadata: outcome.providerMetadata } : {}
+      });
     }).catch(async (error) => {
       if (writer) {
         const failedMarker = createBufferingFailedMarker({
@@ -6511,7 +6524,7 @@ var ReflectorRunner = class {
       });
       await this.persistMarkerToStorage(endMarker, currentRecord.threadId ?? "", currentRecord.resourceId ?? void 0);
     }
-    return reflectResult.usage;
+    return { usage: reflectResult.usage, providerMetadata: reflectResult.providerMetadata };
   }
   /**
    * Try to activate buffered reflection when threshold is reached.
@@ -6783,6 +6796,7 @@ ${unreflectedContent}` : freshRecord.bufferedReflection;
       resourceId: record.resourceId ?? void 0
     } : void 0;
     let reflectionUsage;
+    let reflectionProviderMetadata;
     let reflectionError;
     try {
       const compressionStartLevel = await this.getCompressionStartLevel(requestContext);
@@ -6798,6 +6812,7 @@ ${unreflectedContent}` : freshRecord.bufferedReflection;
         observabilityContext
       );
       reflectionUsage = reflectResult.usage;
+      reflectionProviderMetadata = reflectResult.providerMetadata;
       const reflectionTokenCount = this.tokenCounter.countObservations(reflectResult.observations);
       await this.storage.createReflectionGeneration({
         currentRecord: record,
@@ -6851,7 +6866,11 @@ ${unreflectedContent}` : freshRecord.bufferedReflection;
       omError("[OM] Reflection failed", error);
     } finally {
       await this.storage.setReflectingFlag(record.id, false);
-      reflectionHooks?.onReflectionEnd?.({ usage: reflectionUsage, error: reflectionError });
+      reflectionHooks?.onReflectionEnd?.({
+        usage: reflectionUsage,
+        error: reflectionError,
+        ...reflectionProviderMetadata ? { providerMetadata: reflectionProviderMetadata } : {}
+      });
       unregisterOp(record.id, "reflecting");
     }
   }
@@ -10904,6 +10923,7 @@ ${grouped}` : grouped;
     const reflectionHooks = hooks ? { onReflectionStart: hooks.onReflectionStart, onReflectionEnd: hooks.onReflectionEnd } : void 0;
     let observed = false;
     let observationUsage;
+    let observationProviderMetadata;
     let generationBefore = -1;
     await this.withLock(lockKey, async () => {
       const freshRecord = await this.getOrCreateRecord(threadId, resourceId);
@@ -10934,11 +10954,16 @@ ${grouped}` : grouped;
         }).run();
         observed = result.observed;
         observationUsage = result.usage;
+        observationProviderMetadata = result.providerMetadata;
       } catch (error) {
         observationError = error instanceof Error ? error : new Error(String(error));
         throw error;
       } finally {
-        hooks?.onObservationEnd?.({ usage: observationUsage, error: observationError });
+        hooks?.onObservationEnd?.({
+          usage: observationUsage,
+          error: observationError,
+          ...observationProviderMetadata ? { providerMetadata: observationProviderMetadata } : {}
+        });
       }
     });
     const record = await this.getOrCreateRecord(threadId, resourceId);
@@ -11514,5 +11539,5 @@ exports.stripEphemeralAnchorIds = stripEphemeralAnchorIds;
 exports.stripObservationGroups = stripObservationGroups;
 exports.truncateStringByTokens = truncateStringByTokens;
 exports.wrapInObservationGroup = wrapInObservationGroup;
-//# sourceMappingURL=chunk-PVZX6FFO.cjs.map
-//# sourceMappingURL=chunk-PVZX6FFO.cjs.map
+//# sourceMappingURL=chunk-3SSXARFU.cjs.map
+//# sourceMappingURL=chunk-3SSXARFU.cjs.map