@mastra/convex 0.0.1 → 0.1.0-beta.1

package/dist/index.js

@@ -1,5 +1,5 @@
 export { TABLE_MESSAGES, TABLE_RESOURCES, TABLE_SCORERS, TABLE_THREADS, TABLE_WORKFLOW_SNAPSHOT, mastraDocumentsTable, mastraMessagesTable, mastraResourcesTable, mastraScoresTable, mastraStorage, mastraThreadsTable, mastraVectorIndexesTable, mastraVectorsTable, mastraWorkflowSnapshotsTable } from './chunk-NZCHEPNU.js';
-import { MastraStorage, StoreOperations, TABLE_WORKFLOW_SNAPSHOT, MemoryStorage, TABLE_THREADS, TABLE_MESSAGES, normalizePerPage, calculatePagination, safelyParseJSON, TABLE_RESOURCES, WorkflowsStorage, ScoresStorage, TABLE_SCORERS } from '@mastra/core/storage';
+import { MastraStorage, StoreOperations, TABLE_WORKFLOW_SNAPSHOT, MemoryStorage, TABLE_THREADS, createStorageErrorId, TABLE_MESSAGES, normalizePerPage, calculatePagination, safelyParseJSON, TABLE_RESOURCES, WorkflowsStorage, ScoresStorage, TABLE_SCORERS } from '@mastra/core/storage';
 import { MessageList } from '@mastra/core/agent';
 import { MastraError, ErrorCategory, ErrorDomain } from '@mastra/core/error';
 import crypto from 'crypto';
@@ -104,7 +104,7 @@ var MemoryConvex = class extends MemoryStorage {
     const existing = await this.getThreadById({ threadId: id });
     if (!existing) {
       throw new MastraError({
-        id: "CONVEX_STORAGE_THREAD_NOT_FOUND",
+        id: createStorageErrorId("CONVEX", "UPDATE_THREAD", "THREAD_NOT_FOUND"),
         domain: ErrorDomain.STORAGE,
         category: ErrorCategory.USER,
         text: `Thread ${id} not found`
@@ -163,23 +163,28 @@ var MemoryConvex = class extends MemoryStorage {
   }
   async listMessages(args) {
     const { threadId, resourceId, include, filter, perPage: perPageInput, page = 0, orderBy } = args;
-    if (!threadId.trim()) {
+    const threadIds = Array.isArray(threadId) ? threadId : [threadId];
+    if (threadIds.length === 0 || threadIds.some((id) => !id.trim())) {
       throw new MastraError(
         {
-          id: "CONVEX_STORAGE_LIST_MESSAGES_INVALID_THREAD_ID",
+          id: createStorageErrorId("CONVEX", "LIST_MESSAGES", "INVALID_THREAD_ID"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER,
-          details: { threadId }
+          details: { threadId: Array.isArray(threadId) ? threadId.join(",") : threadId }
         },
-        new Error("threadId must be a non-empty string")
+        new Error("threadId must be a non-empty string or array of non-empty strings")
       );
     }
     const perPage = normalizePerPage(perPageInput, 40);
     const { offset, perPage: perPageForResponse } = calculatePagination(page, perPageInput, perPage);
     const { field, direction } = this.parseOrderBy(orderBy, "ASC");
-    let rows = await this.operations.queryTable(TABLE_MESSAGES, [
-      { field: "thread_id", value: threadId }
-    ]);
+    let rows = [];
+    for (const tid of threadIds) {
+      const threadRows = await this.operations.queryTable(TABLE_MESSAGES, [
+        { field: "thread_id", value: tid }
+      ]);
+      rows.push(...threadRows);
+    }
     if (resourceId) {
       rows = rows.filter((row) => row.resourceId === resourceId);
     }
@@ -206,21 +211,41 @@ var MemoryConvex = class extends MemoryStorage {
     const messageIds = new Set(messages.map((msg) => msg.id));
     if (include && include.length > 0) {
       const threadMessagesCache = /* @__PURE__ */ new Map();
-      threadMessagesCache.set(threadId, rows);
+      for (const tid of threadIds) {
+        const tidRows = rows.filter((r) => r.thread_id === tid);
+        threadMessagesCache.set(tid, tidRows);
+      }
       for (const includeItem of include) {
-        const targetThreadId = includeItem.threadId || threadId;
-        if (!threadMessagesCache.has(targetThreadId)) {
-          const otherThreadRows = await this.operations.queryTable(TABLE_MESSAGES, [
-            { field: "thread_id", value: targetThreadId }
+        let targetThreadId;
+        let target;
+        for (const [tid, cachedRows] of threadMessagesCache) {
+          target = cachedRows.find((row) => row.id === includeItem.id);
+          if (target) {
+            targetThreadId = tid;
+            break;
+          }
+        }
+        if (!target) {
+          const messageRows = await this.operations.queryTable(TABLE_MESSAGES, [
+            { field: "id", value: includeItem.id }
           ]);
-          threadMessagesCache.set(targetThreadId, otherThreadRows);
+          if (messageRows.length > 0) {
+            target = messageRows[0];
+            targetThreadId = target.thread_id;
+            if (targetThreadId && !threadMessagesCache.has(targetThreadId)) {
+              const otherThreadRows = await this.operations.queryTable(TABLE_MESSAGES, [
+                { field: "thread_id", value: targetThreadId }
+              ]);
+              threadMessagesCache.set(targetThreadId, otherThreadRows);
+            }
+          }
         }
-        const targetThreadRows = threadMessagesCache.get(targetThreadId) || [];
-        const target = targetThreadRows.find((row) => row.id === includeItem.id);
-        if (target && !messageIds.has(target.id)) {
+        if (!target || !targetThreadId) continue;
+        if (!messageIds.has(target.id)) {
           messages.push(this.parseStoredMessage(target));
           messageIds.add(target.id);
         }
+        const targetThreadRows = threadMessagesCache.get(targetThreadId) || [];
         await this.addContextMessages({
           includeItem,
           allMessages: targetThreadRows,
@@ -525,7 +550,7 @@ var ScoresConvex = class extends ScoresStorage {
     if (pagination.page < 0) {
       throw new MastraError(
         {
-          id: "CONVEX_STORAGE_INVALID_PAGINATION",
+          id: createStorageErrorId("CONVEX", "LIST_SCORES", "INVALID_PAGINATION"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER
         },
@@ -804,7 +829,7 @@ var ConvexStore = class extends MastraStorage {
   workflows;
   scores;
   constructor(config) {
-    super({ id: config.id, name: config.name ?? "ConvexStore" });
+    super({ id: config.id, name: config.name ?? "ConvexStore", disableInit: config.disableInit });
     const client = new ConvexAdminClient(config);
     this.operations = new StoreOperationsConvex(client);
     this.memory = new MemoryConvex(this.operations);