@mastra/server 1.34.1-alpha.0 → 1.35.0-alpha.1

package/dist/{chunk-QXKYA2XA.cjs → chunk-Z35VW4BM.cjs}

@@ -1042,7 +1042,7 @@ function imageSize(input) {
   throw new TypeError(`unsupported file type: ${type}`);
 }
 
-// ../memory/dist/chunk-KGYJHNI6.js
+// ../memory/dist/chunk-LCALB7W6.js
 var OM_DEBUG_LOG = process.env.OM_DEBUG ? path.join(process.cwd(), "om-debug.log") : null;
 function omDebug(msg) {
   if (!OM_DEBUG_LOG) return;
@@ -6523,7 +6523,7 @@ var IMAGE_FILE_EXTENSIONS = /* @__PURE__ */ new Set([
   "heif",
   "avif"
 ]);
-var TOKEN_ESTIMATE_CACHE_VERSION = 6;
+var TOKEN_ESTIMATE_CACHE_VERSION = 7;
 var CLIENT_TOKEN_ESTIMATE_SOURCE = "client";
 var DEFAULT_IMAGE_ESTIMATOR = {
   baseTokens: 85,
@@ -7034,6 +7034,42 @@ function estimateAnthropicImageTokens(dimensions, sourceStats) {
   }
   return 1600;
 }
+function estimateFileTokensFromBytes(provider, mimeType, sizeBytes) {
+  const normalizedMime = (mimeType ?? "").toLowerCase().split(";", 1)[0].trim();
+  const isPdf = normalizedMime === "application/pdf";
+  const isTextish = normalizedMime.startsWith("text/") || ["application/json", "application/xml", "application/x-yaml", "application/yaml"].includes(normalizedMime);
+  if (isPdf) {
+    if (provider === "google") return Math.max(258, Math.ceil(sizeBytes / 20));
+    if (provider === "anthropic") return Math.max(1500, Math.ceil(sizeBytes / 3));
+    return Math.max(500, Math.ceil(sizeBytes / 4));
+  }
+  if (isTextish) return Math.max(1, Math.ceil(sizeBytes / 4));
+  return Math.max(1, Math.ceil(sizeBytes / 4));
+}
+function estimateNonImageFileTokens(modelContext, part) {
+  const sourceStats = resolveImageSourceStats(getObjectValue(part, "data"));
+  if (sourceStats.sizeBytes === void 0) {
+    return void 0;
+  }
+  const provider = resolveProviderId(modelContext);
+  const modelId = modelContext?.modelId ?? null;
+  const mimeType = getAttachmentMimeType(part, "application/octet-stream");
+  const filename = getAttachmentFilename(part) ?? null;
+  const tokens = estimateFileTokensFromBytes(provider, mimeType, sourceStats.sizeBytes);
+  return {
+    tokens,
+    cachePayload: JSON.stringify({
+      kind: "non-image-file",
+      provider: provider ?? "fallback",
+      modelId,
+      estimator: "bytes",
+      source: sourceStats.source,
+      sizeBytes: sourceStats.sizeBytes,
+      mimeType,
+      filename
+    })
+  };
+}
 function estimateGoogleImageTokens(modelContext, part, dimensions) {
   if (isGoogleGemini3Model(modelContext)) {
     const mediaResolution = resolveGoogleMediaResolution(part);
@@ -7461,6 +7497,15 @@ var TokenCounter = class _TokenCounter {
       return this.readOrPersistFixedPartEstimate(part, "image-like-file", estimate.cachePayload, estimate.tokens);
     }
     if (part.type === "file") {
+      const byteEstimate = estimateNonImageFileTokens(this.getModelContext(), part);
+      if (byteEstimate) {
+        return this.readOrPersistFixedPartEstimate(
+          part,
+          "non-image-file",
+          byteEstimate.cachePayload,
+          byteEstimate.tokens
+        );
+      }
       return this.readOrPersistPartEstimate(part, "file-descriptor", serializeNonImageFilePartForTokenCounting(part));
     }
     return void 0;
@@ -10670,5 +10715,5 @@ exports.stripEphemeralAnchorIds = stripEphemeralAnchorIds;
 exports.stripObservationGroups = stripObservationGroups;
 exports.truncateStringByTokens = truncateStringByTokens;
 exports.wrapInObservationGroup = wrapInObservationGroup;
-//# sourceMappingURL=chunk-QXKYA2XA.cjs.map
-//# sourceMappingURL=chunk-QXKYA2XA.cjs.map
+//# sourceMappingURL=chunk-Z35VW4BM.cjs.map
+//# sourceMappingURL=chunk-Z35VW4BM.cjs.map