@mastra/dynamodb 1.3.0-alpha.0 → 1.3.0-alpha.2

package/dist/index.js

@@ -1745,7 +1745,10 @@ var MemoryStorageDynamoDB = class extends MemoryStorage {
 				hasMore: false
 			};
 			if (perPage === 0 && include && include.length > 0) {
-				const includeMessages = await this._getIncludedMessages({ include });
+				const includeMessages = await this._getIncludedMessages({
+					include,
+					resourceId
+				});
 				const list = new MessageList().add(includeMessages, "memory");
 				return {
 					messages: this._sortMessages(list.get.all.db(), field, direction),
@@ -1834,7 +1837,10 @@ var MemoryStorageDynamoDB = class extends MemoryStorage {
 			const messageIds = new Set(paginatedMessages.map((m) => m.id));
 			let includeMessages = [];
 			if (include && include.length > 0) {
-				includeMessages = await this._getIncludedMessages({ include });
+				includeMessages = await this._getIncludedMessages({
+					include,
+					resourceId
+				});
 				for (const includeMsg of includeMessages) if (!messageIds.has(includeMsg.id)) {
 					paginatedMessages.push(includeMsg);
 					messageIds.add(includeMsg.id);
@@ -2009,7 +2015,14 @@ var MemoryStorageDynamoDB = class extends MemoryStorage {
 			return direction === "ASC" ? aValue - bValue : bValue - aValue;
 		});
 	}
-	async _getIncludedMessages({ include }) {
+	/**
+	* Fetches the messages named by `include` together with their surrounding context.
+	*
+	* @param include - Message ids to pin, each with an optional before/after window.
+	* @param resourceId - When set, restricts both the pinned messages and their context
+	* to that resource so an id from another resource returns nothing.
+	*/
+	async _getIncludedMessages({ include, resourceId }) {
 		if (!include?.length) return [];
 		const targetResults = await Promise.all(include.map((inc) => this.service.entities.message.get({
 			entity: "message",
@@ -2022,7 +2035,7 @@ var MemoryStorageDynamoDB = class extends MemoryStorage {
 			data: null
 		}))));
 		const targetMap = /* @__PURE__ */ new Map();
-		for (const { id, data } of targetResults) if (data) {
+		for (const { id, data } of targetResults) if (data && (!resourceId || data.resourceId === resourceId)) {
 			const createdAt = typeof data.createdAt === "string" ? data.createdAt : new Date(data.createdAt).toISOString();
 			targetMap.set(id, {
 				threadId: data.threadId,
@@ -2030,7 +2043,7 @@ var MemoryStorageDynamoDB = class extends MemoryStorage {
 			});
 		}
 		if (targetMap.size === 0) return [];
-		const parseQueryMessages = (data) => data.map((item) => this.parseMessageData(item)).filter((msg) => "content" in msg && typeof msg.content === "object");
+		const parseQueryMessages = (data) => data.map((item) => this.parseMessageData(item)).filter((msg) => "content" in msg && typeof msg.content === "object").filter((msg) => !resourceId || msg.resourceId === resourceId);
 		const includeMessages = [];
 		for (const includeItem of include) {
 			const { id, withPreviousMessages = 0, withNextMessages = 0 } = includeItem;