@mastra/convex 1.2.1-alpha.1 → 1.2.1-alpha.2

package/dist/index.js

@@ -1,4 +1,4 @@
-export { mastraCache, mastraNativeVectorAction, mastraNativeVectorMutation, mastraNativeVectorQuery, mastraStorage } from './chunk-LXPDHMDN.js';
+export { mastraCache, mastraNativeVectorAction, mastraNativeVectorMutation, mastraNativeVectorQuery, mastraStorage } from './chunk-77UWNT5X.js';
 export { TABLE_BACKGROUND_TASKS, TABLE_CHANNEL_CONFIG, TABLE_CHANNEL_INSTALLATIONS, TABLE_MESSAGES, TABLE_RESOURCES, TABLE_SCHEDULES, TABLE_SCHEDULE_TRIGGERS, TABLE_SCORERS, TABLE_THREADS, TABLE_WORKFLOW_SNAPSHOT, defineMastraNativeVectorTable, mastraBackgroundTasksTable, mastraCacheListItemsTable, mastraCacheTable, mastraChannelConfigTable, mastraChannelInstallationsTable, mastraDocumentsTable, mastraMessagesTable, mastraResourcesTable, mastraScheduleTriggersTable, mastraSchedulesTable, mastraScoresTable, mastraThreadsTable, mastraVectorIndexesTable, mastraVectorsTable, mastraWorkflowSnapshotsTable } from './chunk-MC75WADX.js';
 import { MastraServerCache } from '@mastra/core/cache';
 import { MastraCompositeStore, createVectorErrorId, MemoryStorage, TABLE_THREADS, TABLE_MESSAGES, TABLE_RESOURCES, createStorageErrorId, normalizePerPage, calculatePagination, filterByDateRange, safelyParseJSON, WorkflowsStorage, TABLE_WORKFLOW_SNAPSHOT, ScoresStorage, TABLE_SCORERS, ChannelsStorage, TABLE_CHANNEL_INSTALLATIONS, TABLE_CHANNEL_CONFIG, SchedulesStorage, TABLE_SCHEDULE_TRIGGERS, TABLE_SCHEDULES, BackgroundTasksStorage, TABLE_BACKGROUND_TASKS } from '@mastra/core/storage';
@@ -275,6 +275,7 @@ var ConvexAdminClient = class {
     return result;
   }
 };
+var LOAD_MANY_REQUEST_BATCH_SIZE = 10;
 function resolveConvexConfig(config) {
   if ("client" in config) {
     return config.client;
@@ -390,6 +391,21 @@ var ConvexDB = class extends MastraBase {
     });
     return result;
   }
+  async loadMany(tableName, ids) {
+    const uniqueIds = [...new Set(ids)];
+    if (uniqueIds.length === 0) return [];
+    const rows = [];
+    for (let index = 0; index < uniqueIds.length; index += LOAD_MANY_REQUEST_BATCH_SIZE) {
+      rows.push(
+        ...await this.client.callStorage({
+          op: "loadMany",
+          tableName,
+          ids: uniqueIds.slice(index, index + LOAD_MANY_REQUEST_BATCH_SIZE)
+        })
+      );
+    }
+    return rows;
+  }
   async queryTable(tableName, filters, indexHint, limit) {
     return this.client.callStorage({
       op: "queryTable",
@@ -1095,8 +1111,8 @@ var MemoryConvex = class extends MemoryStorage {
     if (messageIds.length === 0) {
       return { messages: [] };
     }
-    const rows = await this.#db.queryTable(TABLE_MESSAGES, void 0);
-    const filtered = rows.filter((row) => messageIds.includes(row.id)).map((row) => this.parseStoredMessage(row));
+    const rows = await this.#db.loadMany(TABLE_MESSAGES, messageIds);
+    const filtered = rows.map((row) => this.parseStoredMessage(row));
     const list = new MessageList().add(filtered, "memory");
     return { messages: list.get.all.db() };
   }
@@ -1140,11 +1156,15 @@ var MemoryConvex = class extends MemoryStorage {
     messages
   }) {
     if (messages.length === 0) return [];
-    const existing = await this.#db.queryTable(TABLE_MESSAGES, void 0);
+    const existingRows = await this.#db.loadMany(
+      TABLE_MESSAGES,
+      messages.map((message) => message.id)
+    );
+    const existing = new Map(existingRows.map((row) => [row.id, row]));
     const updated = [];
     const affectedThreadIds = /* @__PURE__ */ new Set();
     for (const update of messages) {
-      const current = existing.find((row) => row.id === update.id);
+      const current = existing.get(update.id);
       if (!current) continue;
       affectedThreadIds.add(current.thread_id);
       if (update.threadId) {