@mastra/dsql 1.3.0-alpha.0 → 1.3.0-alpha.1

package/dist/index.js

@@ -2172,7 +2172,14 @@ var MemoryDSQL = class MemoryDSQL extends MemoryStorage {
 			}, error);
 		});
 	}
-	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 || include.length === 0) return null;
 		const unionQueries = [];
 		const params = [];
@@ -2183,17 +2190,18 @@ var MemoryDSQL = class MemoryDSQL extends MemoryStorage {
 		});
 		for (const inc of include) {
 			const { id, withPreviousMessages = 0, withNextMessages = 0 } = inc;
+			const resourceCondition = resourceId ? ` AND "resourceId" = $${paramIdx + 3}` : "";
 			unionQueries.push(`
             SELECT * FROM (
               WITH target_thread AS (
-                SELECT thread_id FROM ${tableName} WHERE id = $${paramIdx}
+                SELECT thread_id FROM ${tableName} WHERE id = $${paramIdx}${resourceCondition}
               ),
               ordered_messages AS (
                 SELECT
                   *,
                   ROW_NUMBER() OVER (ORDER BY "createdAt" ASC) as row_num
                 FROM ${tableName}
-                WHERE thread_id = (SELECT thread_id FROM target_thread)
+                WHERE thread_id = (SELECT thread_id FROM target_thread)${resourceCondition}
               )
               SELECT
                 m.id,
@@ -2219,6 +2227,10 @@ var MemoryDSQL = class MemoryDSQL extends MemoryStorage {
             `);
 			params.push(id, withPreviousMessages, withNextMessages);
 			paramIdx += 3;
+			if (resourceId) {
+				params.push(resourceId);
+				paramIdx += 1;
+			}
 		}
 		const finalQuery = unionQueries.join(" UNION ALL ") + " ORDER BY \"createdAt\" ASC";
 		const includedRows = await this.#db.client.manyOrNone(finalQuery, params);
@@ -2344,7 +2356,10 @@ var MemoryDSQL = class MemoryDSQL extends MemoryStorage {
 			};
 			const messageIds = new Set(messages.map((m) => m.id));
 			if (include && include.length > 0) {
-				const includeMessages = await this._getIncludedMessages({ include });
+				const includeMessages = await this._getIncludedMessages({
+					include,
+					resourceId
+				});
 				if (includeMessages) {
 					for (const includeMsg of includeMessages) if (!messageIds.has(includeMsg.id)) {
 						messages.push(includeMsg);