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

package/dist/index.cjs

@@ -1,6 +1,6 @@
 'use strict';
 
-var chunkMECAZ6GY_cjs = require('./chunk-MECAZ6GY.cjs');
+var chunk6P3LEDHY_cjs = require('./chunk-6P3LEDHY.cjs');
 var chunkSFRHJGSM_cjs = require('./chunk-SFRHJGSM.cjs');
 var cache = require('@mastra/core/cache');
 var storage = require('@mastra/core/storage');
@@ -227,7 +227,8 @@ var ConvexAdminClient = class {
     }
     return {
       result: storageResponse.result,
-      hasMore: storageResponse.hasMore
+      hasMore: storageResponse.hasMore,
+      continuationCursor: storageResponse.continuationCursor
     };
   }
   async callStorage(request) {
@@ -280,6 +281,7 @@ var ConvexAdminClient = class {
     return result;
   }
 };
+var LOAD_MANY_REQUEST_BATCH_SIZE = 10;
 function resolveConvexConfig(config) {
   if ("client" in config) {
     return config.client;
@@ -355,6 +357,38 @@ var ConvexDB = class extends base.MastraBase {
       record: this.normalizePatch(record)
     });
   }
+  async updateThread({
+    id,
+    title,
+    metadata,
+    updatedAt
+  }) {
+    return this.client.callStorage({
+      op: "updateThread",
+      tableName: storage.TABLE_THREADS,
+      id,
+      title,
+      metadata,
+      updatedAt: updatedAt.toISOString()
+    });
+  }
+  async updateResource({
+    resourceId,
+    workingMemory,
+    metadata,
+    createdAt,
+    updatedAt
+  }) {
+    return this.client.callStorage({
+      op: "updateResource",
+      tableName: storage.TABLE_RESOURCES,
+      resourceId,
+      ...workingMemory !== void 0 ? { workingMemory } : {},
+      ...metadata !== void 0 ? { metadata } : {},
+      createdAt: createdAt.toISOString(),
+      updatedAt: updatedAt.toISOString()
+    });
+  }
   async load({ tableName, keys }) {
     const result = await this.client.callStorage({
       op: "load",
@@ -363,6 +397,21 @@ var ConvexDB = class extends base.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",
@@ -841,6 +890,23 @@ var ChannelsConvex = class extends storage.ChannelsStorage {
     await this.#db.deleteMany(storage.TABLE_CHANNEL_CONFIG, [platform]);
   }
 };
+function parseStoredThread(row) {
+  return {
+    ...row,
+    metadata: typeof row.metadata === "string" ? storage.safelyParseJSON(row.metadata) : row.metadata,
+    createdAt: new Date(row.createdAt),
+    updatedAt: new Date(row.updatedAt)
+  };
+}
+function parseStoredResource(record) {
+  const metadata = typeof record.metadata === "string" ? storage.safelyParseJSON(record.metadata) : record.metadata;
+  return {
+    ...record,
+    metadata: metadata ?? {},
+    createdAt: new Date(record.createdAt),
+    updatedAt: new Date(record.updatedAt)
+  };
+}
 var MemoryConvex = class extends storage.MemoryStorage {
   #db;
   constructor(config) {
@@ -864,12 +930,7 @@ var MemoryConvex = class extends storage.MemoryStorage {
       keys: { id: threadId }
     });
     if (!row || resourceId !== void 0 && row.resourceId !== resourceId) return null;
-    return {
-      ...row,
-      metadata: typeof row.metadata === "string" ? JSON.parse(row.metadata) : row.metadata,
-      createdAt: new Date(row.createdAt),
-      updatedAt: new Date(row.updatedAt)
-    };
+    return parseStoredThread(row);
   }
   async saveThread({ thread }) {
     await this.#db.insert({
@@ -886,8 +947,13 @@ var MemoryConvex = class extends storage.MemoryStorage {
     title,
     metadata
   }) {
-    const existing = await this.getThreadById({ threadId: id });
-    if (!existing) {
+    const updated = await this.#db.updateThread({
+      id,
+      title,
+      metadata,
+      updatedAt: /* @__PURE__ */ new Date()
+    });
+    if (!updated) {
       throw new error.MastraError({
         id: storage.createStorageErrorId("CONVEX", "UPDATE_THREAD", "THREAD_NOT_FOUND"),
         domain: error.ErrorDomain.STORAGE,
@@ -895,17 +961,7 @@ var MemoryConvex = class extends storage.MemoryStorage {
         text: `Thread ${id} not found`
       });
     }
-    const updated = {
-      ...existing,
-      title,
-      metadata: {
-        ...existing.metadata,
-        ...metadata
-      },
-      updatedAt: /* @__PURE__ */ new Date()
-    };
-    await this.saveThread({ thread: updated });
-    return updated;
+    return parseStoredThread(updated);
   }
   async deleteThread({ threadId }) {
     const messages = await this.#db.queryTable(storage.TABLE_MESSAGES, [
@@ -933,6 +989,19 @@ var MemoryConvex = class extends storage.MemoryStorage {
       );
     }
     const perPage = storage.normalizePerPage(perPageInput, 100);
+    try {
+      this.validateMetadataKeys(filter?.metadata);
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: storage.createStorageErrorId("CONVEX", "LIST_THREADS", "INVALID_METADATA_KEY"),
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.USER,
+          details: { metadataKeys: filter?.metadata ? Object.keys(filter.metadata).join(", ") : "" }
+        },
+        error$1 instanceof Error ? error$1 : new Error("Invalid metadata key")
+      );
+    }
     const { field, direction } = this.parseOrderBy(orderBy);
     const { offset, perPage: perPageForResponse } = storage.calculatePagination(page, perPageInput, perPage);
     const queryFilters = [];
@@ -940,15 +1009,10 @@ var MemoryConvex = class extends storage.MemoryStorage {
       queryFilters.push({ field: "resourceId", value: filter.resourceId });
     }
     const rows = await this.#db.queryTable(storage.TABLE_THREADS, queryFilters);
-    let threads = rows.map((row) => ({
-      ...row,
-      metadata: typeof row.metadata === "string" ? JSON.parse(row.metadata) : row.metadata,
-      createdAt: new Date(row.createdAt),
-      updatedAt: new Date(row.updatedAt)
-    }));
+    let threads = rows.map((row) => parseStoredThread(row));
     if (filter?.metadata && Object.keys(filter.metadata).length > 0) {
       threads = threads.filter((thread) => {
-        if (!thread.metadata) return false;
+        if (!thread.metadata || typeof thread.metadata !== "object" || Array.isArray(thread.metadata)) return false;
         return Object.entries(filter.metadata).every(([key, value]) => thread.metadata[key] === value);
       });
     }
@@ -1053,8 +1117,8 @@ var MemoryConvex = class extends storage.MemoryStorage {
     if (messageIds.length === 0) {
       return { messages: [] };
     }
-    const rows = await this.#db.queryTable(storage.TABLE_MESSAGES, void 0);
-    const filtered = rows.filter((row) => messageIds.includes(row.id)).map((row) => this.parseStoredMessage(row));
+    const rows = await this.#db.loadMany(storage.TABLE_MESSAGES, messageIds);
+    const filtered = rows.map((row) => this.parseStoredMessage(row));
     const list = new agent.MessageList().add(filtered, "memory");
     return { messages: list.get.all.db() };
   }
@@ -1085,19 +1149,11 @@ var MemoryConvex = class extends storage.MemoryStorage {
     const threadIds = [...new Set(messages.map((m) => m.threadId).filter(Boolean))];
     const now = /* @__PURE__ */ new Date();
     for (const threadId of threadIds) {
-      const thread = await this.getThreadById({ threadId });
-      if (thread) {
-        await this.#db.insert({
-          tableName: storage.TABLE_THREADS,
-          record: {
-            ...thread,
-            id: thread.id,
-            updatedAt: now.toISOString(),
-            createdAt: thread.createdAt instanceof Date ? thread.createdAt.toISOString() : thread.createdAt,
-            metadata: thread.metadata ?? {}
-          }
-        });
-      }
+      await this.#db.patch({
+        tableName: storage.TABLE_THREADS,
+        id: threadId,
+        record: { updatedAt: now.toISOString() }
+      });
     }
     const list = new agent.MessageList().add(messages, "memory");
     return { messages: list.get.all.db() };
@@ -1106,11 +1162,15 @@ var MemoryConvex = class extends storage.MemoryStorage {
     messages
   }) {
     if (messages.length === 0) return [];
-    const existing = await this.#db.queryTable(storage.TABLE_MESSAGES, void 0);
+    const existingRows = await this.#db.loadMany(
+      storage.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) {
@@ -1143,19 +1203,11 @@ var MemoryConvex = class extends storage.MemoryStorage {
     }
     const now = /* @__PURE__ */ new Date();
     for (const threadId of affectedThreadIds) {
-      const thread = await this.getThreadById({ threadId });
-      if (thread) {
-        await this.#db.insert({
-          tableName: storage.TABLE_THREADS,
-          record: {
-            ...thread,
-            id: thread.id,
-            updatedAt: now.toISOString(),
-            createdAt: thread.createdAt instanceof Date ? thread.createdAt.toISOString() : thread.createdAt,
-            metadata: thread.metadata ?? {}
-          }
-        });
-      }
+      await this.#db.patch({
+        tableName: storage.TABLE_THREADS,
+        id: threadId,
+        record: { updatedAt: now.toISOString() }
+      });
     }
     return updated;
   }
@@ -1183,41 +1235,22 @@ var MemoryConvex = class extends storage.MemoryStorage {
       keys: { id: resourceId }
     });
     if (!record) return null;
-    return {
-      ...record,
-      metadata: typeof record.metadata === "string" ? storage.safelyParseJSON(record.metadata) : record.metadata,
-      createdAt: new Date(record.createdAt),
-      updatedAt: new Date(record.updatedAt)
-    };
+    return parseStoredResource(record);
   }
   async updateResource({
     resourceId,
     workingMemory,
     metadata
   }) {
-    const existing = await this.getResourceById({ resourceId });
     const now = /* @__PURE__ */ new Date();
-    if (!existing) {
-      const created = {
-        id: resourceId,
-        workingMemory,
-        metadata: metadata ?? {},
-        createdAt: now,
-        updatedAt: now
-      };
-      return this.saveResource({ resource: created });
-    }
-    const updated = {
-      ...existing,
-      workingMemory: workingMemory ?? existing.workingMemory,
-      metadata: {
-        ...existing.metadata,
-        ...metadata
-      },
+    const updated = await this.#db.updateResource({
+      resourceId,
+      workingMemory,
+      metadata,
+      createdAt: now,
       updatedAt: now
-    };
-    await this.saveResource({ resource: updated });
-    return updated;
+    });
+    return parseStoredResource(updated);
   }
   _sortMessages(messages, field, direction) {
     return messages.sort((a, b) => {
@@ -1850,6 +1883,7 @@ var ConvexStore = class extends storage.MastraCompositeStore {
   }
 };
 var INDEX_METADATA_TABLE = "mastra_vector_indexes";
+var VECTOR_QUERY_PAGE_SIZE = 256;
 var ConvexVector = class extends vector.MastraVector {
   client;
   constructor(config) {
@@ -1902,10 +1936,7 @@ var ConvexVector = class extends vector.MastraVector {
     if (!index) {
       throw new Error(`Index ${indexName} not found`);
     }
-    const vectors = await this.callStorage({
-      op: "queryTable",
-      tableName: this.vectorTable(indexName)
-    });
+    const vectors = await this.queryAllVectors(indexName);
     return {
       dimension: index.dimension,
       count: vectors.length,
@@ -1942,10 +1973,7 @@ var ConvexVector = class extends vector.MastraVector {
         details: { indexName }
       });
     }
-    const vectors = await this.callStorage({
-      op: "queryTable",
-      tableName: this.vectorTable(indexName)
-    });
+    const vectors = await this.queryAllVectors(indexName);
     const filtered = filter && !this.isEmptyFilter(filter) ? vectors.filter((record) => this.matchesFilter(record.metadata, filter)) : vectors;
     const scored = filtered.map((record) => ({
       id: record.id,
@@ -1966,10 +1994,7 @@ var ConvexVector = class extends vector.MastraVector {
       if (this.isEmptyFilter(filter)) {
         throw new Error("ConvexVector.updateVector: cannot update with empty filter");
       }
-      const vectors = await this.callStorage({
-        op: "queryTable",
-        tableName: this.vectorTable(params.indexName)
-      });
+      const vectors = await this.queryAllVectors(params.indexName);
       const matching = vectors.filter((record) => this.matchesFilter(record.metadata, filter));
       for (const existing2 of matching) {
         const updated2 = {
@@ -2038,10 +2063,7 @@ var ConvexVector = class extends vector.MastraVector {
     if (this.isEmptyFilter(filter)) {
       throw new Error("ConvexVector.deleteVectors: cannot delete with empty filter");
     }
-    const vectors = await this.callStorage({
-      op: "queryTable",
-      tableName: this.vectorTable(indexName)
-    });
+    const vectors = await this.queryAllVectors(indexName);
     const matchingIds = vectors.filter((record) => this.matchesFilter(record.metadata, filter)).map((record) => record.id);
     if (matchingIds.length > 0) {
       await this.callStorage({
@@ -2129,6 +2151,27 @@ var ConvexVector = class extends vector.MastraVector {
   async callStorage(request) {
     return this.client.callStorage(request);
   }
+  async queryAllVectors(indexName) {
+    const vectors = [];
+    let cursor = null;
+    let hasMore = true;
+    while (hasMore) {
+      const response = await this.client.callStorageRaw({
+        op: "queryTable",
+        tableName: this.vectorTable(indexName),
+        pageSize: VECTOR_QUERY_PAGE_SIZE,
+        cursor
+      });
+      vectors.push(...response.result);
+      const nextCursor = response.continuationCursor ?? null;
+      hasMore = response.hasMore ?? false;
+      if (hasMore && (!nextCursor || nextCursor === cursor)) {
+        throw new Error("ConvexVector: paginated vector query did not return a valid continuation cursor");
+      }
+      cursor = nextCursor;
+    }
+    return vectors;
+  }
   /**
    * Call storage repeatedly until hasMore is false.
    * Use for bulk operations like clearTable that may need multiple batches.
@@ -2406,23 +2449,23 @@ var ConvexNativeVector = class extends vector.MastraVector {
 
 Object.defineProperty(exports, "mastraCache", {
   enumerable: true,
-  get: function () { return chunkMECAZ6GY_cjs.mastraCache; }
+  get: function () { return chunk6P3LEDHY_cjs.mastraCache; }
 });
 Object.defineProperty(exports, "mastraNativeVectorAction", {
   enumerable: true,
-  get: function () { return chunkMECAZ6GY_cjs.mastraNativeVectorAction; }
+  get: function () { return chunk6P3LEDHY_cjs.mastraNativeVectorAction; }
 });
 Object.defineProperty(exports, "mastraNativeVectorMutation", {
   enumerable: true,
-  get: function () { return chunkMECAZ6GY_cjs.mastraNativeVectorMutation; }
+  get: function () { return chunk6P3LEDHY_cjs.mastraNativeVectorMutation; }
 });
 Object.defineProperty(exports, "mastraNativeVectorQuery", {
   enumerable: true,
-  get: function () { return chunkMECAZ6GY_cjs.mastraNativeVectorQuery; }
+  get: function () { return chunk6P3LEDHY_cjs.mastraNativeVectorQuery; }
 });
 Object.defineProperty(exports, "mastraStorage", {
   enumerable: true,
-  get: function () { return chunkMECAZ6GY_cjs.mastraStorage; }
+  get: function () { return chunk6P3LEDHY_cjs.mastraStorage; }
 });
 Object.defineProperty(exports, "TABLE_BACKGROUND_TASKS", {
   enumerable: true,