@mastra/redis 1.3.1 → 1.4.0-alpha.1

package/dist/index.js

@@ -240,6 +240,7 @@ var StoreMemoryRedis = class extends MemoryStorage {
 				hasMore: perPageInput === false ? false : end < total
 			};
 		} catch (error) {
+			if (error instanceof MastraError && error.category === ErrorCategory.USER) throw error;
 			const mastraError = new MastraError({
 				id: createStorageErrorId("REDIS", "LIST_THREADS", "FAILED"),
 				domain: ErrorDomain.STORAGE,
@@ -253,13 +254,7 @@ var StoreMemoryRedis = class extends MemoryStorage {
 			}, error);
 			this.logger.trackException(mastraError);
 			this.logger.error(mastraError.toString());
-			return {
-				threads: [],
-				total: 0,
-				page,
-				perPage: perPageForResponse,
-				hasMore: false
-			};
+			throw mastraError;
 		}
 	}
 	async saveThread({ thread }) {
@@ -413,16 +408,36 @@ var StoreMemoryRedis = class extends MemoryStorage {
 		if (message.threadId) await this.client.set(getMessageIndexKey(messageId), message.threadId);
 		return message.threadId || null;
 	}
-	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, drops any pinned or context message owned by another
+	* resource so an id from another resource returns nothing.
+	*/
+	async getIncludedMessages(include, resourceId) {
 		if (!include?.length) return [];
 		const messageIds = /* @__PURE__ */ new Set();
 		const messageIdToThreadIds = {};
 		for (const item of include) {
 			const itemThreadId = await this.getThreadIdForMessage(item.id);
 			if (!itemThreadId) continue;
+			const itemThreadMessagesKey = getThreadMessagesKey(itemThreadId);
+			if (resourceId !== void 0) {
+				const threadMessageIds = await this.client.zRange(itemThreadMessagesKey, 0, -1);
+				const threadMessages = (await this.client.mGet(threadMessageIds.map((id) => getMessageKey(itemThreadId, id)))).filter((data) => data !== null).map((data) => JSON.parse(data)).filter((message) => message.resourceId === resourceId);
+				const targetIndex = threadMessages.findIndex((message) => message.id === item.id);
+				if (targetIndex === -1) continue;
+				const start = Math.max(0, targetIndex - (item.withPreviousMessages ?? 0));
+				const end = Math.min(threadMessages.length, targetIndex + (item.withNextMessages ?? 0) + 1);
+				for (const message of threadMessages.slice(start, end)) {
+					messageIds.add(message.id);
+					messageIdToThreadIds[message.id] = itemThreadId;
+				}
+				continue;
+			}
 			messageIds.add(item.id);
 			messageIdToThreadIds[item.id] = itemThreadId;
-			const itemThreadMessagesKey = getThreadMessagesKey(itemThreadId);
 			const rank = await this.client.zRank(itemThreadMessagesKey, item.id);
 			if (rank === null) continue;
 			if (item.withPreviousMessages) {
@@ -439,7 +454,8 @@ var StoreMemoryRedis = class extends MemoryStorage {
 		}
 		if (messageIds.size === 0) return [];
 		const keysToFetch = Array.from(messageIds).map((id) => getMessageKey(messageIdToThreadIds[id], id));
-		return (await this.client.mGet(keysToFetch)).filter((data) => data !== null).map((data) => JSON.parse(data));
+		const includedMessages = (await this.client.mGet(keysToFetch)).filter((data) => data !== null).map((data) => JSON.parse(data));
+		return resourceId ? includedMessages.filter((message) => message.resourceId === resourceId) : includedMessages;
 	}
 	parseStoredMessage(storedMessage) {
 		const defaultMessageContent = {
@@ -543,7 +559,7 @@ var StoreMemoryRedis = class extends MemoryStorage {
 				hasMore: false
 			};
 			let includedMessages = [];
-			if (include && include.length > 0) includedMessages = (await this.getIncludedMessages(include)).map(this.parseStoredMessage);
+			if (include && include.length > 0) includedMessages = (await this.getIncludedMessages(include, resourceId)).map(this.parseStoredMessage);
 			if (perPage === 0 && include && include.length > 0) return {
 				messages: new MessageList().add(includedMessages, "memory").get.all.db().sort((a, b) => {
 					const aValue = getFieldValue(a);
@@ -613,6 +629,7 @@ var StoreMemoryRedis = class extends MemoryStorage {
 				hasMore
 			};
 		} catch (error) {
+			if (error instanceof MastraError && error.category === ErrorCategory.USER) throw error;
 			const mastraError = new MastraError({
 				id: createStorageErrorId("REDIS", "LIST_MESSAGES", "FAILED"),
 				domain: ErrorDomain.STORAGE,
@@ -624,13 +641,7 @@ var StoreMemoryRedis = class extends MemoryStorage {
 			}, error);
 			this.logger.error(mastraError.toString());
 			this.logger.trackException(mastraError);
-			return {
-				messages: [],
-				total: 0,
-				page,
-				perPage: perPageForResponse,
-				hasMore: false
-			};
+			throw mastraError;
 		}
 	}
 	async getResourceById({ resourceId }) {