@mastra/mssql 1.4.1-alpha.0 → 1.5.0-alpha.0

package/dist/index.js

@@ -1,5 +1,5 @@
 import { MastraError, ErrorCategory, ErrorDomain } from '@mastra/core/error';
-import { AgentsStorage, TABLE_AGENTS, TABLE_AGENT_VERSIONS, AGENTS_SCHEMA, AGENT_VERSIONS_SCHEMA, createStorageErrorId, normalizePerPage, calculatePagination, BackgroundTasksStorage, TABLE_BACKGROUND_TASKS, TABLE_SCHEMAS, MemoryStorage, TABLE_THREADS, TABLE_MESSAGES, TABLE_RESOURCES, ObservabilityStorage, TABLE_SPANS, SPAN_SCHEMA, listTracesArgsSchema, toTraceSpans, ScoresStorage, TABLE_SCORERS, WorkflowsStorage, TABLE_WORKFLOW_SNAPSHOT, MastraCompositeStore, getDefaultValue, transformScoreRow as transformScoreRow$1, TraceStatus } from '@mastra/core/storage';
+import { AgentsStorage, TABLE_AGENTS, TABLE_AGENT_VERSIONS, AGENTS_SCHEMA, AGENT_VERSIONS_SCHEMA, createStorageErrorId, normalizePerPage, calculatePagination, BackgroundTasksStorage, TABLE_BACKGROUND_TASKS, TABLE_SCHEMAS, MemoryStorage, TABLE_THREADS, TABLE_MESSAGES, TABLE_RESOURCES, validateStorageMetadataFilter, ObservabilityStorage, TABLE_SPANS, SPAN_SCHEMA, listTracesArgsSchema, toTraceSpans, ScoresStorage, TABLE_SCORERS, WorkflowsStorage, TABLE_WORKFLOW_SNAPSHOT, MastraCompositeStore, getDefaultValue, transformScoreRow as transformScoreRow$1, TraceStatus } from '@mastra/core/storage';
 import sql from 'mssql';
 import { randomUUID } from 'crypto';
 import { parseSqlIdentifier } from '@mastra/core/utils';
@@ -2286,6 +2286,32 @@ var BackgroundTasksMSSQL = class _BackgroundTasksMSSQL extends BackgroundTasksSt
     return Number(result.recordset[0]?.count ?? 0);
   }
 };
+function bindMssqlMetadataParams(request, params) {
+  for (const [paramName, paramValue] of Object.entries(params)) {
+    request.input(paramName, paramValue);
+  }
+}
+function buildMssqlMessageMetadataFilter(metadataFilter) {
+  if (!metadataFilter) return { clauses: [], params: {} };
+  const clauses = [];
+  const params = {};
+  Object.entries(metadataFilter).forEach(([key, value], index) => {
+    const keyParam = `metadataKey${index}`;
+    params[keyParam] = key;
+    clauses.push("ISJSON(content) = 1");
+    if (value === null) {
+      clauses.push(`EXISTS (SELECT 1 FROM OPENJSON(content, '$.metadata') WHERE [key] = @${keyParam} AND [type] = 0)`);
+      return;
+    }
+    const valueParam = `metadataValue${index}`;
+    params[valueParam] = String(value);
+    const jsonType = typeof value === "string" ? 1 : typeof value === "number" ? 2 : 3;
+    clauses.push(
+      `EXISTS (SELECT 1 FROM OPENJSON(content, '$.metadata') WHERE [key] = @${keyParam} AND [type] = ${jsonType} AND [value] = @${valueParam})`
+    );
+  });
+  return { clauses, params };
+}
 var MemoryMSSQL = class _MemoryMSSQL extends MemoryStorage {
   pool;
   schema;
@@ -2845,6 +2871,7 @@ var MemoryMSSQL = class _MemoryMSSQL extends MemoryStorage {
   }
   async listMessages(args) {
     const { threadId, resourceId, include, filter, perPage: perPageInput, page = 0, orderBy } = args;
+    const metadataFilter = validateStorageMetadataFilter(filter?.metadata);
     const threadIds = Array.isArray(threadId) ? threadId : [threadId];
     if (threadIds.length === 0 || threadIds.some((id) => !id.trim())) {
       throw new MastraError(
@@ -2881,10 +2908,13 @@ var MemoryMSSQL = class _MemoryMSSQL extends MemoryStorage {
         ...resourceId ? { resourceId } : {},
         ...buildDateRangeFilter(filter?.dateRange, "createdAt")
       };
-      const { sql: actualWhereClause = "", params: whereParams } = prepareWhereClause(
+      const { sql: preparedWhereClause = "", params: whereParams } = prepareWhereClause(
         filters);
+      const metadataWhere = buildMssqlMessageMetadataFilter(metadataFilter);
+      const actualWhereClause = metadataWhere.clauses.length ? `${preparedWhereClause || " WHERE 1 = 1"} AND ${metadataWhere.clauses.join(" AND ")}` : preparedWhereClause;
       const bindWhereParams = (req) => {
         Object.entries(whereParams).forEach(([paramName, paramValue]) => req.input(paramName, paramValue));
+        bindMssqlMetadataParams(req, metadataWhere.params);
       };
       if (perPage === 0 && (!include || include.length === 0)) {
         return { messages: [], total: 0, page, perPage: perPageForResponse, hasMore: false };
@@ -2920,6 +2950,7 @@ var MemoryMSSQL = class _MemoryMSSQL extends MemoryStorage {
       };
       const baseRows = perPage === 0 ? [] : await fetchBaseMessages();
       const messages = [...baseRows];
+      const primaryPageCount = messages.length;
       const seqById = /* @__PURE__ */ new Map();
       messages.forEach((msg) => {
         if (typeof msg.seq_id === "number") seqById.set(msg.id, msg.seq_id);
@@ -2960,7 +2991,7 @@ var MemoryMSSQL = class _MemoryMSSQL extends MemoryStorage {
       });
       const threadIdSet = new Set(threadIds);
       const returnedThreadMessageCount = finalMessages.filter((m) => m.threadId && threadIdSet.has(m.threadId)).length;
-      const hasMore = perPageInput !== false && returnedThreadMessageCount < total && offset + perPage < total;
+      const hasMore = metadataFilter ? perPageInput !== false && offset + primaryPageCount < total : perPageInput !== false && returnedThreadMessageCount < total && offset + perPage < total;
       return {
         messages: finalMessages,
         total,