@mastra/convex 0.0.0-netlify-no-bundle-20251127120354 → 0.0.0-new-button-export-20251219133013

package/dist/index.js

@@ -1,8 +1,10 @@
-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';
+export { mastraStorage } from './chunk-KSAPIIEJ.js';
+export { TABLE_MESSAGES, TABLE_RESOURCES, TABLE_SCORERS, TABLE_THREADS, TABLE_WORKFLOW_SNAPSHOT, mastraDocumentsTable, mastraMessagesTable, mastraResourcesTable, mastraScoresTable, mastraThreadsTable, mastraVectorIndexesTable, mastraVectorsTable, mastraWorkflowSnapshotsTable } from './chunk-PKUUSREO.js';
+import { MastraStorage, MemoryStorage, TABLE_THREADS, TABLE_MESSAGES, TABLE_RESOURCES, createStorageErrorId, normalizePerPage, calculatePagination, safelyParseJSON, WorkflowsStorage, TABLE_WORKFLOW_SNAPSHOT, 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';
+import { MastraBase } from '@mastra/core/base';
 import { MastraVector } from '@mastra/core/vector';
 
 // src/storage/client.ts
@@ -68,13 +70,114 @@ var ConvexAdminClient = class {
     return result;
   }
 };
+function resolveConvexConfig(config) {
+  if ("client" in config) {
+    return config.client;
+  }
+  return new ConvexAdminClient(config);
+}
+var ConvexDB = class extends MastraBase {
+  constructor(client) {
+    super({ name: "convex-db" });
+    this.client = client;
+  }
+  async hasColumn(_table, _column) {
+    return true;
+  }
+  async clearTable({ tableName }) {
+    let hasMore = true;
+    while (hasMore) {
+      const response = await this.client.callStorageRaw({
+        op: "clearTable",
+        tableName
+      });
+      hasMore = response.hasMore ?? false;
+    }
+  }
+  async dropTable({ tableName }) {
+    let hasMore = true;
+    while (hasMore) {
+      const response = await this.client.callStorageRaw({
+        op: "dropTable",
+        tableName
+      });
+      hasMore = response.hasMore ?? false;
+    }
+  }
+  async insert({ tableName, record }) {
+    await this.client.callStorage({
+      op: "insert",
+      tableName,
+      record: this.normalizeRecord(tableName, record)
+    });
+  }
+  async batchInsert({ tableName, records }) {
+    if (records.length === 0) return;
+    await this.client.callStorage({
+      op: "batchInsert",
+      tableName,
+      records: records.map((record) => this.normalizeRecord(tableName, record))
+    });
+  }
+  async load({ tableName, keys }) {
+    const result = await this.client.callStorage({
+      op: "load",
+      tableName,
+      keys
+    });
+    return result;
+  }
+  async queryTable(tableName, filters) {
+    return this.client.callStorage({
+      op: "queryTable",
+      tableName,
+      filters
+    });
+  }
+  async deleteMany(tableName, ids) {
+    if (ids.length === 0) return;
+    await this.client.callStorage({
+      op: "deleteMany",
+      tableName,
+      ids
+    });
+  }
+  normalizeRecord(tableName, record) {
+    const normalized = { ...record };
+    if (tableName === TABLE_WORKFLOW_SNAPSHOT && !normalized.id) {
+      const runId = normalized.run_id || normalized.runId;
+      const workflowName = normalized.workflow_name || normalized.workflowName;
+      normalized.id = workflowName ? `${workflowName}-${runId}` : runId;
+    }
+    if (!normalized.id) {
+      normalized.id = crypto.randomUUID();
+    }
+    for (const [key, value] of Object.entries(normalized)) {
+      if (value instanceof Date) {
+        normalized[key] = value.toISOString();
+      }
+    }
+    return normalized;
+  }
+};
+
+// src/storage/domains/memory/index.ts
 var MemoryConvex = class extends MemoryStorage {
-  constructor(operations) {
+  #db;
+  constructor(config) {
     super();
-    this.operations = operations;
+    const client = resolveConvexConfig(config);
+    this.#db = new ConvexDB(client);
+  }
+  async init() {
+  }
+  async dangerouslyClearAll() {
+    await this.#db.clearTable({ tableName: TABLE_THREADS });
+    await this.#db.clearTable({ tableName: TABLE_MESSAGES });
+    await this.#db.clearTable({ tableName: TABLE_RESOURCES });
   }
   async getThreadById({ threadId }) {
-    const row = await this.operations.load({
+    const row = await this.#db.load({
       tableName: TABLE_THREADS,
       keys: { id: threadId }
     });
@@ -87,7 +190,7 @@ var MemoryConvex = class extends MemoryStorage {
     };
   }
   async saveThread({ thread }) {
-    await this.operations.insert({
+    await this.#db.insert({
       tableName: TABLE_THREADS,
       record: {
         ...thread,
@@ -104,7 +207,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`
@@ -123,21 +226,21 @@ var MemoryConvex = class extends MemoryStorage {
     return updated;
   }
   async deleteThread({ threadId }) {
-    const messages = await this.operations.queryTable(TABLE_MESSAGES, [
+    const messages = await this.#db.queryTable(TABLE_MESSAGES, [
       { field: "thread_id", value: threadId }
     ]);
-    await this.operations.deleteMany(
+    await this.#db.deleteMany(
       TABLE_MESSAGES,
       messages.map((msg) => msg.id)
     );
-    await this.operations.deleteMany(TABLE_THREADS, [threadId]);
+    await this.#db.deleteMany(TABLE_THREADS, [threadId]);
   }
   async listThreadsByResourceId(args) {
     const { resourceId, page = 0, perPage: perPageInput, orderBy } = args;
     const perPage = normalizePerPage(perPageInput, 100);
     const { field, direction } = this.parseOrderBy(orderBy);
     const { offset, perPage: perPageForResponse } = calculatePagination(page, perPageInput, perPage);
-    const rows = await this.operations.queryTable(TABLE_THREADS, [{ field: "resourceId", value: resourceId }]);
+    const rows = await this.#db.queryTable(TABLE_THREADS, [{ field: "resourceId", value: resourceId }]);
     const threads = rows.map((row) => ({
       ...row,
       metadata: typeof row.metadata === "string" ? JSON.parse(row.metadata) : row.metadata,
@@ -163,23 +266,26 @@ 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.#db.queryTable(TABLE_MESSAGES, [{ field: "thread_id", value: tid }]);
+      rows.push(...threadRows);
+    }
     if (resourceId) {
       rows = rows.filter((row) => row.resourceId === resourceId);
     }
@@ -206,21 +312,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.#db.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.#db.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,
@@ -251,7 +377,7 @@ var MemoryConvex = class extends MemoryStorage {
     if (messageIds.length === 0) {
       return { messages: [] };
     }
-    const rows = await this.operations.queryTable(TABLE_MESSAGES, void 0);
+    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 list = new MessageList().add(filtered, "memory");
     return { messages: list.get.all.db() };
@@ -276,7 +402,7 @@ var MemoryConvex = class extends MemoryStorage {
         resourceId: message.resourceId
       };
     });
-    await this.operations.batchInsert({
+    await this.#db.batchInsert({
       tableName: TABLE_MESSAGES,
       records: normalized
     });
@@ -285,7 +411,7 @@ var MemoryConvex = class extends MemoryStorage {
     for (const threadId of threadIds) {
       const thread = await this.getThreadById({ threadId });
       if (thread) {
-        await this.operations.insert({
+        await this.#db.insert({
           tableName: TABLE_THREADS,
           record: {
             ...thread,
@@ -304,7 +430,7 @@ var MemoryConvex = class extends MemoryStorage {
     messages
   }) {
     if (messages.length === 0) return [];
-    const existing = await this.operations.queryTable(TABLE_MESSAGES, void 0);
+    const existing = await this.#db.queryTable(TABLE_MESSAGES, void 0);
     const updated = [];
     const affectedThreadIds = /* @__PURE__ */ new Set();
     for (const update of messages) {
@@ -333,7 +459,7 @@ var MemoryConvex = class extends MemoryStorage {
         };
         current.content = JSON.stringify(mergedContent);
       }
-      await this.operations.insert({
+      await this.#db.insert({
         tableName: TABLE_MESSAGES,
         record: current
       });
@@ -343,7 +469,7 @@ var MemoryConvex = class extends MemoryStorage {
     for (const threadId of affectedThreadIds) {
       const thread = await this.getThreadById({ threadId });
       if (thread) {
-        await this.operations.insert({
+        await this.#db.insert({
           tableName: TABLE_THREADS,
           record: {
             ...thread,
@@ -358,7 +484,7 @@ var MemoryConvex = class extends MemoryStorage {
     return updated;
   }
   async deleteMessages(messageIds) {
-    await this.operations.deleteMany(TABLE_MESSAGES, messageIds);
+    await this.#db.deleteMany(TABLE_MESSAGES, messageIds);
   }
   async saveResource({ resource }) {
     const record = {
@@ -369,14 +495,14 @@ var MemoryConvex = class extends MemoryStorage {
     if (resource.metadata !== void 0) {
       record.metadata = resource.metadata;
     }
-    await this.operations.insert({
+    await this.#db.insert({
       tableName: TABLE_RESOURCES,
       record
     });
     return resource;
   }
   async getResourceById({ resourceId }) {
-    const record = await this.operations.load({
+    const record = await this.#db.load({
       tableName: TABLE_RESOURCES,
       keys: { id: resourceId }
     });
@@ -462,12 +588,19 @@ var MemoryConvex = class extends MemoryStorage {
   }
 };
 var ScoresConvex = class extends ScoresStorage {
-  constructor(operations) {
+  #db;
+  constructor(config) {
     super();
-    this.operations = operations;
+    const client = resolveConvexConfig(config);
+    this.#db = new ConvexDB(client);
+  }
+  async init() {
+  }
+  async dangerouslyClearAll() {
+    await this.#db.clearTable({ tableName: TABLE_SCORERS });
   }
   async getScoreById({ id }) {
-    const row = await this.operations.load({
+    const row = await this.#db.load({
       tableName: TABLE_SCORERS,
       keys: { id }
     });
@@ -481,7 +614,7 @@ var ScoresConvex = class extends ScoresStorage {
       createdAt: now.toISOString(),
       updatedAt: now.toISOString()
     };
-    await this.operations.insert({
+    await this.#db.insert({
       tableName: TABLE_SCORERS,
       record
     });
@@ -525,14 +658,14 @@ 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
         },
         new Error("page must be >= 0")
       );
     }
-    const rows = await this.operations.queryTable(TABLE_SCORERS, void 0);
+    const rows = await this.#db.queryTable(TABLE_SCORERS, void 0);
     const filtered = rows.filter((row) => filters.scorerId ? row.scorerId === filters.scorerId : true).filter((row) => filters.entityId ? row.entityId === filters.entityId : true).filter((row) => filters.entityType ? row.entityType === filters.entityType : true).filter((row) => filters.runId ? row.runId === filters.runId : true).filter((row) => filters.source ? row.source === filters.source : true).sort((a, b) => new Date(b.createdAt).getTime() - new Date(a.createdAt).getTime());
     const { perPage, page } = pagination;
     const perPageValue = perPage === false ? filtered.length : perPage;
@@ -558,9 +691,16 @@ var ScoresConvex = class extends ScoresStorage {
   }
 };
 var WorkflowsConvex = class extends WorkflowsStorage {
-  constructor(operations) {
+  #db;
+  constructor(config) {
     super();
-    this.operations = operations;
+    const client = resolveConvexConfig(config);
+    this.#db = new ConvexDB(client);
+  }
+  async init() {
+  }
+  async dangerouslyClearAll() {
+    await this.#db.clearTable({ tableName: TABLE_WORKFLOW_SNAPSHOT });
   }
   async updateWorkflowResults({
     workflowName,
@@ -607,11 +747,11 @@ var WorkflowsConvex = class extends WorkflowsStorage {
     snapshot
   }) {
     const now = /* @__PURE__ */ new Date();
-    const existing = await this.operations.load({
+    const existing = await this.#db.load({
       tableName: TABLE_WORKFLOW_SNAPSHOT,
       keys: { workflow_name: workflowName, run_id: runId }
     });
-    await this.operations.insert({
+    await this.#db.insert({
       tableName: TABLE_WORKFLOW_SNAPSHOT,
       record: {
         workflow_name: workflowName,
@@ -627,7 +767,7 @@ var WorkflowsConvex = class extends WorkflowsStorage {
     workflowName,
     runId
   }) {
-    const row = await this.operations.load({
+    const row = await this.#db.load({
       tableName: TABLE_WORKFLOW_SNAPSHOT,
       keys: { workflow_name: workflowName, run_id: runId }
     });
@@ -636,7 +776,7 @@ var WorkflowsConvex = class extends WorkflowsStorage {
   }
   async listWorkflowRuns(args = {}) {
     const { workflowName, fromDate, toDate, perPage, page, resourceId, status } = args;
-    let rows = await this.operations.queryTable(TABLE_WORKFLOW_SNAPSHOT, void 0);
+    let rows = await this.#db.queryTable(TABLE_WORKFLOW_SNAPSHOT, void 0);
     if (workflowName) rows = rows.filter((run) => run.workflow_name === workflowName);
     if (resourceId) rows = rows.filter((run) => run.resourceId === resourceId);
     if (fromDate) rows = rows.filter((run) => new Date(run.createdAt).getTime() >= fromDate.getTime());
@@ -668,7 +808,7 @@ var WorkflowsConvex = class extends WorkflowsStorage {
     runId,
     workflowName
   }) {
-    const runs = await this.operations.queryTable(TABLE_WORKFLOW_SNAPSHOT, void 0);
+    const runs = await this.#db.queryTable(TABLE_WORKFLOW_SNAPSHOT, void 0);
     const match = runs.find((run) => run.run_id === runId && (!workflowName || run.workflow_name === workflowName));
     if (!match) return null;
     return {
@@ -680,8 +820,11 @@ var WorkflowsConvex = class extends WorkflowsStorage {
       resourceId: match.resourceId
     };
   }
+  async deleteWorkflowRunById({ runId, workflowName }) {
+    await this.#db.deleteMany(TABLE_WORKFLOW_SNAPSHOT, [`${workflowName}-${runId}`]);
+  }
   async getRun(workflowName, runId) {
-    const runs = await this.operations.queryTable(TABLE_WORKFLOW_SNAPSHOT, [
+    const runs = await this.#db.queryTable(TABLE_WORKFLOW_SNAPSHOT, [
       { field: "workflow_name", value: workflowName }
     ]);
     return runs.find((run) => run.run_id === runId) ?? null;
@@ -708,110 +851,20 @@ var WorkflowsConvex = class extends WorkflowsStorage {
     return JSON.parse(JSON.stringify(run.snapshot));
   }
 };
-var StoreOperationsConvex = class extends StoreOperations {
-  constructor(client) {
-    super();
-    this.client = client;
-  }
-  async hasColumn(_table, _column) {
-    return true;
-  }
-  async createTable(_args) {
-  }
-  async clearTable({ tableName }) {
-    let hasMore = true;
-    while (hasMore) {
-      const response = await this.client.callStorageRaw({
-        op: "clearTable",
-        tableName
-      });
-      hasMore = response.hasMore ?? false;
-    }
-  }
-  async dropTable({ tableName }) {
-    let hasMore = true;
-    while (hasMore) {
-      const response = await this.client.callStorageRaw({
-        op: "dropTable",
-        tableName
-      });
-      hasMore = response.hasMore ?? false;
-    }
-  }
-  async alterTable(_args) {
-  }
-  async insert({ tableName, record }) {
-    await this.client.callStorage({
-      op: "insert",
-      tableName,
-      record: this.normalizeRecord(tableName, record)
-    });
-  }
-  async batchInsert({ tableName, records }) {
-    if (records.length === 0) return;
-    await this.client.callStorage({
-      op: "batchInsert",
-      tableName,
-      records: records.map((record) => this.normalizeRecord(tableName, record))
-    });
-  }
-  async load({ tableName, keys }) {
-    const result = await this.client.callStorage({
-      op: "load",
-      tableName,
-      keys
-    });
-    return result;
-  }
-  async queryTable(tableName, filters) {
-    return this.client.callStorage({
-      op: "queryTable",
-      tableName,
-      filters
-    });
-  }
-  async deleteMany(tableName, ids) {
-    if (ids.length === 0) return;
-    await this.client.callStorage({
-      op: "deleteMany",
-      tableName,
-      ids
-    });
-  }
-  normalizeRecord(tableName, record) {
-    const normalized = { ...record };
-    if (tableName === TABLE_WORKFLOW_SNAPSHOT && !normalized.id) {
-      const runId = normalized.run_id || normalized.runId;
-      const workflowName = normalized.workflow_name || normalized.workflowName;
-      normalized.id = workflowName ? `${workflowName}-${runId}` : runId;
-    }
-    if (!normalized.id) {
-      normalized.id = crypto.randomUUID();
-    }
-    for (const [key, value] of Object.entries(normalized)) {
-      if (value instanceof Date) {
-        normalized[key] = value.toISOString();
-      }
-    }
-    return normalized;
-  }
-};
 
 // src/storage/index.ts
 var ConvexStore = class extends MastraStorage {
-  operations;
   memory;
   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);
-    this.workflows = new WorkflowsConvex(this.operations);
-    this.scores = new ScoresConvex(this.operations);
+    const domainConfig = { client };
+    this.memory = new MemoryConvex(domainConfig);
+    this.workflows = new WorkflowsConvex(domainConfig);
+    this.scores = new ScoresConvex(domainConfig);
     this.stores = {
-      operations: this.operations,
       memory: this.memory,
       workflows: this.workflows,
       scores: this.scores
@@ -828,25 +881,6 @@ var ConvexStore = class extends MastraStorage {
       listScoresBySpan: false
     };
   }
-  async createTable(_args) {
-  }
-  async clearTable({ tableName }) {
-    await this.operations.clearTable({ tableName });
-  }
-  async dropTable({ tableName }) {
-    await this.operations.dropTable({ tableName });
-  }
-  async alterTable(_args) {
-  }
-  async insert({ tableName, record }) {
-    await this.operations.insert({ tableName, record });
-  }
-  async batchInsert({ tableName, records }) {
-    await this.operations.batchInsert({ tableName, records });
-  }
-  async load({ tableName, keys }) {
-    return this.operations.load({ tableName, keys });
-  }
   async getThreadById({ threadId }) {
     return this.memory.getThreadById({ threadId });
   }
@@ -925,6 +959,9 @@ var ConvexStore = class extends MastraStorage {
   }) {
     return this.workflows.getWorkflowRunById({ runId, workflowName });
   }
+  async deleteWorkflowRunById({ runId, workflowName }) {
+    return this.workflows.deleteWorkflowRunById({ runId, workflowName });
+  }
   async getScoreById({ id }) {
     return this.scores.getScoreById({ id });
   }