@mastra/memory 1.21.3-alpha.0 → 1.21.3-alpha.2

package/dist/{chunk-UHQBCO7I.js → chunk-X7DOZRJK.js}

@@ -1107,7 +1107,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) {
@@ -1700,6 +1700,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) {
@@ -1881,10 +1882,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) {
@@ -4268,7 +4273,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
     };
   }
   /**
@@ -4382,7 +4388,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
     };
   }
 };
@@ -4762,6 +4769,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);
@@ -4839,6 +4847,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(
@@ -4892,7 +4901,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
     };
   }
   /**
@@ -4909,8 +4919,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({
@@ -5029,7 +5042,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.
@@ -5301,6 +5314,7 @@ ${unreflectedContent}` : freshRecord.bufferedReflection;
       resourceId: record.resourceId ?? void 0
     } : void 0;
     let reflectionUsage;
+    let reflectionProviderMetadata;
     let reflectionError;
     try {
       const compressionStartLevel = await this.getCompressionStartLevel(requestContext);
@@ -5316,6 +5330,7 @@ ${unreflectedContent}` : freshRecord.bufferedReflection;
         observabilityContext
       );
       reflectionUsage = reflectResult.usage;
+      reflectionProviderMetadata = reflectResult.providerMetadata;
       const reflectionTokenCount = this.tokenCounter.countObservations(reflectResult.observations);
       await this.storage.createReflectionGeneration({
         currentRecord: record,
@@ -5369,7 +5384,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");
     }
   }
@@ -9424,6 +9443,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);
@@ -9454,11 +9474,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);
@@ -10013,5 +10038,5 @@ function getObservationsAsOf(activeObservations, asOf) {
 }
 
 export { ModelByInputTokens, OBSERVER_SYSTEM_PROMPT, ObservationalMemory, ObservationalMemoryProcessor, TokenCounter, buildObserverPrompt, buildObserverSystemPrompt, combineObservationGroupRanges, deriveObservationGroupProvenance, extractCurrentTask, formatMessagesForObserver, formatToolResultForObserver, getObservationsAsOf, hasCurrentTaskSection, injectAnchorIds, optimizeObservationsForContext, parseAnchorId, parseObservationGroups, parseObserverOutput, reconcileObservationGroupsFromReflection, renderObservationGroupsForReflection, resolveToolResultValue, stripEphemeralAnchorIds, stripObservationGroups, truncateStringByTokens, wrapInObservationGroup };
-//# sourceMappingURL=chunk-UHQBCO7I.js.map
-//# sourceMappingURL=chunk-UHQBCO7I.js.map
+//# sourceMappingURL=chunk-X7DOZRJK.js.map
+//# sourceMappingURL=chunk-X7DOZRJK.js.map