@mastra/memory 1.18.0 → 1.18.1

package/dist/{chunk-NRYX4QWV.cjs → chunk-LCRYFBV3.cjs}

@@ -1020,6 +1020,8 @@ var ObservationStrategy = class _ObservationStrategy {
     this.scope = deps.scope;
     this.retrieval = deps.retrieval;
   }
+  deps;
+  opts;
   storage;
   messageHistory;
   tokenCounter;
@@ -2048,6 +2050,8 @@ var ObservationStep = class {
     this.turn = turn;
     this.stepNumber = stepNumber;
   }
+  turn;
+  stepNumber;
   _prepared = false;
   _context;
   /** Whether this step has been prepared. */
@@ -5536,6 +5540,7 @@ var IMAGE_FILE_EXTENSIONS = /* @__PURE__ */ new Set([
   "avif"
 ]);
 var TOKEN_ESTIMATE_CACHE_VERSION = 6;
+var CLIENT_TOKEN_ESTIMATE_SOURCE = "client";
 var DEFAULT_IMAGE_ESTIMATOR = {
   baseTokens: 85,
   tileTokens: 170,
@@ -5643,6 +5648,16 @@ function setPartCacheEntry(part, key, entry) {
   const mastraMetadata = ensurePartMastraMetadata(part);
   mastraMetadata.tokenEstimate = mergeCacheEntry(mastraMetadata.tokenEstimate, key, entry);
 }
+function getClientPartTokenEstimate(part) {
+  const cache = getPartMastraMetadata(part)?.tokenEstimate;
+  if (!cache || typeof cache !== "object") return void 0;
+  const matches = (entry) => isTokenEstimateEntry(entry) && entry.source === CLIENT_TOKEN_ESTIMATE_SOURCE && Number.isFinite(entry.tokens) && entry.tokens >= 0;
+  if (matches(cache)) return cache;
+  for (const value of Object.values(cache)) {
+    if (matches(value)) return value;
+  }
+  return void 0;
+}
 function getMessageCacheEntry(message, key) {
   const contentLevelEntry = getCacheEntry(getContentMastraMetadata(message.content)?.tokenEstimate, key);
   if (contentLevelEntry) return contentLevelEntry;
@@ -6447,6 +6462,12 @@ var TokenCounter = class _TokenCounter {
     return this.estimateImageAssetTokens(part, part.data, "file");
   }
   countAttachmentPartSync(part) {
+    if (part.type === "image" || part.type === "file") {
+      const clientEstimate = getClientPartTokenEstimate(part);
+      if (clientEstimate) {
+        return clientEstimate.tokens;
+      }
+    }
     if (part.type === "image") {
       const estimate = this.estimateImageTokens(part);
       return this.readOrPersistFixedPartEstimate(part, "image", estimate.cachePayload, estimate.tokens);
@@ -6511,6 +6532,12 @@ var TokenCounter = class _TokenCounter {
     return void 0;
   }
   async countAttachmentPartAsync(part) {
+    if (part.type === "image" || part.type === "file") {
+      const clientEstimate = getClientPartTokenEstimate(part);
+      if (clientEstimate) {
+        return clientEstimate.tokens;
+      }
+    }
     const isImageAttachment = part.type === "image" || part.type === "file" && isImageLikeFilePart(part);
     const remotePayload = this.buildRemoteAttachmentCachePayload(part);
     if (remotePayload) {
@@ -9665,5 +9692,5 @@ exports.stripEphemeralAnchorIds = stripEphemeralAnchorIds;
 exports.stripObservationGroups = stripObservationGroups;
 exports.truncateStringByTokens = truncateStringByTokens;
 exports.wrapInObservationGroup = wrapInObservationGroup;
-//# sourceMappingURL=chunk-NRYX4QWV.cjs.map
-//# sourceMappingURL=chunk-NRYX4QWV.cjs.map
+//# sourceMappingURL=chunk-LCRYFBV3.cjs.map
+//# sourceMappingURL=chunk-LCRYFBV3.cjs.map