@mastra/mssql 0.0.0-fix-9244-clickhouse-metadata-20251104223105 → 0.0.0-fix-thread-list-20251105222841

package/dist/index.js

@@ -1,5 +1,5 @@
 import { MastraError, ErrorCategory, ErrorDomain } from '@mastra/core/error';
-import { MastraStorage, StoreOperations, TABLE_WORKFLOW_SNAPSHOT, TABLE_SCHEMAS, TABLE_THREADS, TABLE_MESSAGES, TABLE_TRACES, TABLE_SCORERS, TABLE_AI_SPANS, ScoresStorage, normalizePerPage, calculatePagination, WorkflowsStorage, MemoryStorage, resolveMessageLimit, TABLE_RESOURCES, ObservabilityStorage, safelyParseJSON } from '@mastra/core/storage';
+import { MastraStorage, StoreOperations, TABLE_WORKFLOW_SNAPSHOT, TABLE_SCHEMAS, TABLE_THREADS, TABLE_MESSAGES, TABLE_TRACES, TABLE_SCORERS, TABLE_AI_SPANS, ScoresStorage, normalizePerPage, calculatePagination, WorkflowsStorage, MemoryStorage, TABLE_RESOURCES, ObservabilityStorage, safelyParseJSON } from '@mastra/core/storage';
 import sql2 from 'mssql';
 import { MessageList } from '@mastra/core/agent';
 import { parseSqlIdentifier } from '@mastra/core/utils';
@@ -371,10 +371,9 @@ var MemoryMSSQL = class extends MemoryStorage {
   }
   async _getIncludedMessages({
     threadId,
-    selectBy
+    include
   }) {
     if (!threadId.trim()) throw new Error("threadId must be a non-empty string");
-    const include = selectBy?.include;
     if (!include) return null;
     const unionQueries = [];
     const paramValues = [];
@@ -446,75 +445,6 @@ var MemoryMSSQL = class extends MemoryStorage {
     });
     return dedupedRows;
   }
-  /**
-   * @deprecated use listMessages instead
-   */
-  async getMessages(args) {
-    const { threadId, resourceId, selectBy } = args;
-    const selectStatement = `SELECT seq_id, id, content, role, type, [createdAt], thread_id AS threadId, resourceId`;
-    const orderByStatement = `ORDER BY [seq_id] DESC`;
-    const limit = resolveMessageLimit({ last: selectBy?.last, defaultLimit: 40 });
-    try {
-      if (!threadId.trim()) throw new Error("threadId must be a non-empty string");
-      let rows = [];
-      const include = selectBy?.include || [];
-      if (include?.length) {
-        const includeMessages = await this._getIncludedMessages({ threadId, selectBy });
-        if (includeMessages) {
-          rows.push(...includeMessages);
-        }
-      }
-      const excludeIds = rows.map((m) => m.id).filter(Boolean);
-      let query = `${selectStatement} FROM ${getTableName({ indexName: TABLE_MESSAGES, schemaName: getSchemaName(this.schema) })} WHERE [thread_id] = @threadId`;
-      const request = this.pool.request();
-      request.input("threadId", threadId);
-      if (excludeIds.length > 0) {
-        const excludeParams = excludeIds.map((_, idx) => `@id${idx}`);
-        query += ` AND id NOT IN (${excludeParams.join(", ")})`;
-        excludeIds.forEach((id, idx) => {
-          request.input(`id${idx}`, id);
-        });
-      }
-      query += ` ${orderByStatement} OFFSET 0 ROWS FETCH NEXT @limit ROWS ONLY`;
-      request.input("limit", limit);
-      const result = await request.query(query);
-      const remainingRows = result.recordset || [];
-      rows.push(...remainingRows);
-      rows.sort((a, b) => {
-        const timeDiff = a.seq_id - b.seq_id;
-        return timeDiff;
-      });
-      const messagesWithParsedContent = rows.map((row) => {
-        if (typeof row.content === "string") {
-          try {
-            return { ...row, content: JSON.parse(row.content) };
-          } catch {
-            return row;
-          }
-        }
-        return row;
-      });
-      const cleanMessages = messagesWithParsedContent.map(({ seq_id, ...rest }) => rest);
-      const list = new MessageList().add(cleanMessages, "memory");
-      return { messages: list.get.all.db() };
-    } catch (error) {
-      const mastraError = new MastraError(
-        {
-          id: "MASTRA_STORAGE_MSSQL_STORE_GET_MESSAGES_FAILED",
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          details: {
-            threadId,
-            resourceId: resourceId ?? ""
-          }
-        },
-        error
-      );
-      this.logger?.error?.(mastraError.toString());
-      this.logger?.trackException?.(mastraError);
-      return { messages: [] };
-    }
-  }
   async listMessagesById({ messageIds }) {
     if (messageIds.length === 0) return { messages: [] };
     const selectStatement = `SELECT seq_id, id, content, role, type, [createdAt], thread_id AS threadId, resourceId`;
@@ -578,7 +508,7 @@ var MemoryMSSQL = class extends MemoryStorage {
     const perPage = normalizePerPage(perPageInput, 40);
     const { offset, perPage: perPageForResponse } = calculatePagination(page, perPageInput, perPage);
     try {
-      const { field, direction } = this.parseOrderBy(orderBy);
+      const { field, direction } = this.parseOrderBy(orderBy, "ASC");
       const orderByStatement = `ORDER BY [${field}] ${direction}`;
       const selectStatement = `SELECT seq_id, id, content, role, type, [createdAt], thread_id AS threadId, resourceId`;
       const tableName = getTableName({ indexName: TABLE_MESSAGES, schemaName: getSchemaName(this.schema) });
@@ -623,8 +553,7 @@ var MemoryMSSQL = class extends MemoryStorage {
       }
       const messageIds = new Set(messages.map((m) => m.id));
       if (include && include.length > 0) {
-        const selectBy = { include };
-        const includeMessages = await this._getIncludedMessages({ threadId, selectBy });
+        const includeMessages = await this._getIncludedMessages({ threadId, include });
         if (includeMessages) {
           for (const includeMsg of includeMessages) {
             if (!messageIds.has(includeMsg.id)) {
@@ -1037,13 +966,13 @@ var ObservabilityMSSQL = class extends ObservabilityStorage {
     this.operations = operations;
     this.schema = schema;
   }
-  get aiTracingStrategy() {
+  get tracingStrategy() {
     return {
       preferred: "batch-with-updates",
       supported: ["batch-with-updates", "insert-only"]
     };
   }
-  async createAISpan(span) {
+  async createSpan(span) {
     try {
       const startedAt = span.startedAt instanceof Date ? span.startedAt.toISOString() : span.startedAt;
       const endedAt = span.endedAt instanceof Date ? span.endedAt.toISOString() : span.endedAt;
@@ -1114,7 +1043,7 @@ var ObservabilityMSSQL = class extends ObservabilityStorage {
       );
     }
   }
-  async updateAISpan({
+  async updateSpan({
     spanId,
     traceId,
     updates
@@ -1249,7 +1178,7 @@ var ObservabilityMSSQL = class extends ObservabilityStorage {
       );
     }
   }
-  async batchCreateAISpans(args) {
+  async batchCreateSpans(args) {
     if (!args.records || args.records.length === 0) {
       return;
     }
@@ -1276,7 +1205,7 @@ var ObservabilityMSSQL = class extends ObservabilityStorage {
       );
     }
   }
-  async batchUpdateAISpans(args) {
+  async batchUpdateSpans(args) {
     if (!args.records || args.records.length === 0) {
       return;
     }
@@ -2229,7 +2158,7 @@ ${columns}
         table: TABLE_SCORERS,
         columns: ["traceId", "spanId", "seq_id DESC"]
       },
-      // AI Spans indexes for optimal trace querying
+      // Spans indexes for optimal trace querying
       {
         name: `${schemaPrefix}mastra_ai_spans_traceid_startedat_idx`,
         table: TABLE_AI_SPANS,
@@ -3094,7 +3023,7 @@ var MSSQLStore = class extends MastraStorage {
       createTable: true,
       deleteMessages: true,
       listScoresBySpan: true,
-      aiTracing: true,
+      observabilityInstance: true,
       indexManagement: true
     };
   }
@@ -3145,12 +3074,6 @@ var MSSQLStore = class extends MastraStorage {
   async deleteThread({ threadId }) {
     return this.stores.memory.deleteThread({ threadId });
   }
-  /**
-   * @deprecated use listMessages instead
-   */
-  async getMessages(args) {
-    return this.stores.memory.getMessages(args);
-  }
   async listMessagesById({ messageIds }) {
     return this.stores.memory.listMessagesById({ messageIds });
   }
@@ -3246,7 +3169,7 @@ var MSSQLStore = class extends MastraStorage {
     return this.stores.operations.dropIndex(indexName);
   }
   /**
-   * AI Tracing / Observability
+   * Tracing / Observability
    */
   getObservabilityStore() {
     if (!this.stores.observability) {
@@ -3259,15 +3182,15 @@ var MSSQLStore = class extends MastraStorage {
     }
     return this.stores.observability;
   }
-  async createAISpan(span) {
-    return this.getObservabilityStore().createAISpan(span);
+  async createSpan(span) {
+    return this.getObservabilityStore().createSpan(span);
   }
-  async updateAISpan({
+  async updateSpan({
     spanId,
     traceId,
     updates
   }) {
-    return this.getObservabilityStore().updateAISpan({ spanId, traceId, updates });
+    return this.getObservabilityStore().updateSpan({ spanId, traceId, updates });
   }
   async getAITrace(traceId) {
     return this.getObservabilityStore().getAITrace(traceId);
@@ -3275,11 +3198,11 @@ var MSSQLStore = class extends MastraStorage {
   async getAITracesPaginated(args) {
     return this.getObservabilityStore().getAITracesPaginated(args);
   }
-  async batchCreateAISpans(args) {
-    return this.getObservabilityStore().batchCreateAISpans(args);
+  async batchCreateSpans(args) {
+    return this.getObservabilityStore().batchCreateSpans(args);
   }
-  async batchUpdateAISpans(args) {
-    return this.getObservabilityStore().batchUpdateAISpans(args);
+  async batchUpdateSpans(args) {
+    return this.getObservabilityStore().batchUpdateSpans(args);
   }
   async batchDeleteAITraces(args) {
     return this.getObservabilityStore().batchDeleteAITraces(args);