@mastra/memory 1.17.5 → 1.17.6-alpha.0

package/dist/{chunk-UZDSNIGD.cjs → chunk-ZUUSLGY6.cjs}

@@ -3165,6 +3165,71 @@ function formatObserverAttachmentPlaceholder(part, counter) {
   const label = resolveObserverAttachmentLabel(part);
   return label ? `[${attachmentType} #${attachmentId}: ${label}]` : `[${attachmentType} #${attachmentId}]`;
 }
+function isRecord(value) {
+  return !!value && typeof value === "object";
+}
+function mapToolResultBlockToAttachment(block) {
+  if (!isRecord(block) || typeof block.type !== "string") {
+    return void 0;
+  }
+  const mediaType = typeof block.mediaType === "string" ? block.mediaType : void 0;
+  const filename = typeof block.filename === "string" ? block.filename : void 0;
+  switch (block.type) {
+    case "image-data": {
+      const data = block.data;
+      if (typeof data !== "string") return void 0;
+      const image = mediaType ? `data:${mediaType};base64,${data}` : data;
+      return { type: "image", image, mimeType: mediaType };
+    }
+    case "image-url": {
+      const url = block.url;
+      if (typeof url !== "string") return void 0;
+      return { type: "image", image: url, mimeType: mediaType };
+    }
+    case "media": {
+      const data = block.data;
+      if (typeof data !== "string" || !mediaType) return void 0;
+      const dataUri = `data:${mediaType};base64,${data}`;
+      if (mediaType.toLowerCase().startsWith("image/")) {
+        return { type: "image", image: dataUri, mimeType: mediaType };
+      }
+      return { type: "file", data: dataUri, mimeType: mediaType };
+    }
+    case "file-data": {
+      const data = block.data;
+      if (typeof data !== "string") return void 0;
+      const dataUri = mediaType ? `data:${mediaType};base64,${data}` : data;
+      return { type: "file", data: dataUri, mimeType: mediaType, filename };
+    }
+    case "file-url": {
+      const url = block.url;
+      if (typeof url !== "string") return void 0;
+      return { type: "file", data: url, mimeType: mediaType, filename };
+    }
+    default:
+      return void 0;
+  }
+}
+function extractToolResultAttachments(result, counter) {
+  if (!isRecord(result) || result.type !== "content" || !Array.isArray(result.value)) {
+    return { resultWithoutAttachments: result, attachments: [] };
+  }
+  const record = result;
+  const attachments = [];
+  const newValue = record.value.map((block) => {
+    const attachment = mapToolResultBlockToAttachment(block);
+    if (!attachment) {
+      return block;
+    }
+    attachments.push(toObserverInputAttachmentPart(attachment));
+    const placeholder = formatObserverAttachmentPlaceholder(attachment, counter);
+    return { type: isRecord(block) ? block.type : void 0, placeholder };
+  });
+  if (attachments.length === 0) {
+    return { resultWithoutAttachments: result, attachments };
+  }
+  return { resultWithoutAttachments: { ...record, value: newValue }, attachments };
+}
 function formatObserverPartLine(title, body, time, previousTime) {
   const timeLabel = time && time !== previousTime ? `(${time})` : "";
   if (!title) {
@@ -3254,9 +3319,19 @@ function formatObserverMessage(msg, counter, options) {
             part,
             inv.result
           );
+          const { resultWithoutAttachments, attachments: extractedAttachments } = extractToolResultAttachments(
+            resultForObserver,
+            counter
+          );
+          if (extractedAttachments.length > 0) {
+            attachments.push(...extractedAttachments);
+          }
           pushLine(
             `Tool Result ${inv.toolName}`,
-            maybeTruncate(formatToolResultForObserver(resultForObserver, { maxTokens: maxToolResultTokens }), maxLen),
+            maybeTruncate(
+              formatToolResultForObserver(resultWithoutAttachments, { maxTokens: maxToolResultTokens }),
+              maxLen
+            ),
             partCreatedAt
           );
           return;
@@ -9398,5 +9473,5 @@ exports.stripEphemeralAnchorIds = stripEphemeralAnchorIds;
 exports.stripObservationGroups = stripObservationGroups;
 exports.truncateStringByTokens = truncateStringByTokens;
 exports.wrapInObservationGroup = wrapInObservationGroup;
-//# sourceMappingURL=chunk-UZDSNIGD.cjs.map
-//# sourceMappingURL=chunk-UZDSNIGD.cjs.map
+//# sourceMappingURL=chunk-ZUUSLGY6.cjs.map
+//# sourceMappingURL=chunk-ZUUSLGY6.cjs.map