@mastra/mssql 1.7.0-alpha.0 → 1.7.0-alpha.2

package/dist/index.js

@@ -2392,7 +2392,7 @@ var MemoryMSSQL = class MemoryMSSQL extends MemoryStorage {
 			text: `Thread ${id} not found`,
 			details: {
 				threadId: id,
-				title
+				title: title ?? null
 			}
 		});
 		const mergedMetadata = {
@@ -2411,7 +2411,7 @@ var MemoryMSSQL = class MemoryMSSQL extends MemoryStorage {
         WHERE id = @id`;
 			const req = this.pool.request();
 			req.input("id", id);
-			req.input("title", title);
+			req.input("title", title ?? existingThread.title);
 			req.input("metadata", JSON.stringify(mergedMetadata));
 			req.input("updatedAt", /* @__PURE__ */ new Date());
 			const result = await req.query(sql);
@@ -2427,7 +2427,7 @@ var MemoryMSSQL = class MemoryMSSQL extends MemoryStorage {
 				text: `Thread ${id} not found after update`,
 				details: {
 					threadId: id,
-					title
+					title: title ?? null
 				}
 			});
 			return {
@@ -2443,7 +2443,7 @@ var MemoryMSSQL = class MemoryMSSQL extends MemoryStorage {
 				category: ErrorCategory.THIRD_PARTY,
 				details: {
 					threadId: id,
-					title
+					title: title ?? null
 				}
 			}, error);
 		}
@@ -2487,7 +2487,14 @@ var MemoryMSSQL = class MemoryMSSQL extends MemoryStorage {
 			return diff !== 0 ? diff : a.id.localeCompare(b.id);
 		});
 	}
-	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 paramValues = [];
@@ -2497,6 +2504,7 @@ var MemoryMSSQL = class MemoryMSSQL extends MemoryStorage {
 			indexName: TABLE_MESSAGES,
 			schemaName: getSchemaName(this.schema)
 		});
+		const resourceCondition = resourceId ? ` AND [resourceId] = @presource` : "";
 		for (const inc of include) {
 			const { id, withPreviousMessages = 0, withNextMessages = 0 } = inc;
 			const pId = `@p${paramIdx}`;
@@ -2515,7 +2523,7 @@ var MemoryMSSQL = class MemoryMSSQL extends MemoryStorage {
           FROM (
             SELECT *, ROW_NUMBER() OVER (ORDER BY [createdAt] ASC) as row_num
             FROM ${tableName}
-            WHERE [thread_id] = (SELECT thread_id FROM ${tableName} WHERE id = ${pId})
+            WHERE [thread_id] = (SELECT thread_id FROM ${tableName} WHERE id = ${pId}${resourceCondition})${resourceCondition}
           ) AS m
           WHERE m.id = ${pId}
           OR EXISTS (
@@ -2523,7 +2531,7 @@ var MemoryMSSQL = class MemoryMSSQL extends MemoryStorage {
             FROM (
               SELECT *, ROW_NUMBER() OVER (ORDER BY [createdAt] ASC) as row_num
               FROM ${tableName}
-              WHERE [thread_id] = (SELECT thread_id FROM ${tableName} WHERE id = ${pId})
+              WHERE [thread_id] = (SELECT thread_id FROM ${tableName} WHERE id = ${pId}${resourceCondition})${resourceCondition}
             ) AS target
             WHERE target.id = ${pId}
             AND (
@@ -2547,6 +2555,7 @@ var MemoryMSSQL = class MemoryMSSQL extends MemoryStorage {
     `;
 		const req = this.pool.request();
 		for (let i = 0; i < paramValues.length; ++i) req.input(paramNames[i], paramValues[i]);
+		if (resourceId) req.input("presource", resourceId);
 		const includedRows = (await req.query(finalQuery)).recordset || [];
 		const seen = /* @__PURE__ */ new Set();
 		return includedRows.filter((row) => {
@@ -2646,7 +2655,10 @@ var MemoryMSSQL = class MemoryMSSQL 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 messages = this._parseAndFormatMessages(includeMessages ?? [], "v2");
 				return {
 					messages: this._sortMessages(messages, field, direction),
@@ -2683,7 +2695,10 @@ var MemoryMSSQL = class MemoryMSSQL extends MemoryStorage {
 			};
 			if (include?.length) {
 				const messageIds = new Set(messages.map((m) => m.id));
-				(await this._getIncludedMessages({ include }))?.forEach((msg) => {
+				(await this._getIncludedMessages({
+					include,
+					resourceId
+				}))?.forEach((msg) => {
 					if (!messageIds.has(msg.id)) {
 						messages.push(msg);
 						messageIds.add(msg.id);