@mastra/convex 1.2.0 → 1.2.1-alpha.1

package/dist/index.cjs

@@ -1,6 +1,6 @@
 'use strict';
 
-var chunkORSDZTO4_cjs = require('./chunk-ORSDZTO4.cjs');
+var chunkZFDTGYKC_cjs = require('./chunk-ZFDTGYKC.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) {
@@ -355,6 +356,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",
@@ -380,6 +413,44 @@ var ConvexDB = class extends base.MastraBase {
       ids
     });
   }
+  async mergeWorkflowStepResult({
+    workflowName,
+    runId,
+    stepId,
+    result,
+    requestContext
+  }) {
+    const context = await this.client.callStorage({
+      op: "mergeWorkflowStepResult",
+      tableName: storage.TABLE_WORKFLOW_SNAPSHOT,
+      workflowName,
+      runId,
+      stepId,
+      result: JSON.stringify(result),
+      requestContext: JSON.stringify(requestContext)
+    });
+    if (!context) {
+      throw new Error(`Convex workflow step merge returned no context for runId ${runId}`);
+    }
+    return JSON.parse(context);
+  }
+  async mergeWorkflowState({
+    workflowName,
+    runId,
+    opts
+  }) {
+    const snapshot = await this.client.callStorage({
+      op: "mergeWorkflowState",
+      tableName: storage.TABLE_WORKFLOW_SNAPSHOT,
+      workflowName,
+      runId,
+      opts: JSON.stringify(opts)
+    });
+    if (!snapshot) {
+      throw new Error(`Convex workflow state merge returned no snapshot for runId ${runId}`);
+    }
+    return JSON.parse(snapshot);
+  }
   async createSchedule(record) {
     if (!record.id) {
       throw new Error(`Schedule is missing an id`);
@@ -803,6 +874,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) {
@@ -826,12 +914,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({
@@ -848,8 +931,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,
@@ -857,17 +945,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, [
@@ -895,6 +973,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 = [];
@@ -902,15 +993,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);
       });
     }
@@ -1047,19 +1133,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() };
@@ -1105,19 +1183,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;
   }
@@ -1145,41 +1215,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) => {
@@ -1640,22 +1691,18 @@ var WorkflowsConvex = class extends storage.WorkflowsStorage {
     this.#db = new ConvexDB(client);
   }
   supportsConcurrentUpdates() {
-    return false;
+    return true;
   }
   async init() {
   }
   async dangerouslyClearAll() {
     await this.#db.clearTable({ tableName: storage.TABLE_WORKFLOW_SNAPSHOT });
   }
-  async updateWorkflowResults(_args) {
-    throw new Error(
-      "updateWorkflowResults is not implemented for Convex storage. Convex does not support atomic read-modify-write operations needed for concurrent workflow updates."
-    );
+  async updateWorkflowResults(args) {
+    return this.#db.mergeWorkflowStepResult(args);
   }
-  async updateWorkflowState(_args) {
-    throw new Error(
-      "updateWorkflowState is not implemented for Convex storage. Convex does not support atomic read-modify-write operations needed for concurrent workflow updates."
-    );
+  async updateWorkflowState(args) {
+    return this.#db.mergeWorkflowState(args);
   }
   async persistWorkflowSnapshot({
     workflowName,
@@ -1816,6 +1863,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) {
@@ -1868,10 +1916,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,
@@ -1908,10 +1953,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,
@@ -1932,10 +1974,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 = {
@@ -2004,10 +2043,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({
@@ -2095,6 +2131,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.
@@ -2372,23 +2429,23 @@ var ConvexNativeVector = class extends vector.MastraVector {
 
 Object.defineProperty(exports, "mastraCache", {
   enumerable: true,
-  get: function () { return chunkORSDZTO4_cjs.mastraCache; }
+  get: function () { return chunkZFDTGYKC_cjs.mastraCache; }
 });
 Object.defineProperty(exports, "mastraNativeVectorAction", {
   enumerable: true,
-  get: function () { return chunkORSDZTO4_cjs.mastraNativeVectorAction; }
+  get: function () { return chunkZFDTGYKC_cjs.mastraNativeVectorAction; }
 });
 Object.defineProperty(exports, "mastraNativeVectorMutation", {
   enumerable: true,
-  get: function () { return chunkORSDZTO4_cjs.mastraNativeVectorMutation; }
+  get: function () { return chunkZFDTGYKC_cjs.mastraNativeVectorMutation; }
 });
 Object.defineProperty(exports, "mastraNativeVectorQuery", {
   enumerable: true,
-  get: function () { return chunkORSDZTO4_cjs.mastraNativeVectorQuery; }
+  get: function () { return chunkZFDTGYKC_cjs.mastraNativeVectorQuery; }
 });
 Object.defineProperty(exports, "mastraStorage", {
   enumerable: true,
-  get: function () { return chunkORSDZTO4_cjs.mastraStorage; }
+  get: function () { return chunkZFDTGYKC_cjs.mastraStorage; }
 });
 Object.defineProperty(exports, "TABLE_BACKGROUND_TASKS", {
   enumerable: true,