@mastra/cloudflare-d1 0.0.0-vector-query-tool-provider-options-20250828222356 → 0.0.0-vector-extension-schema-20250922130418

package/dist/index.js

@@ -802,6 +802,7 @@ var MemoryStorageD1 = class extends MemoryStorage {
     }
   }
   async _getIncludedMessages(threadId, selectBy) {
+    if (!threadId.trim()) throw new Error("threadId must be a non-empty string");
     const include = selectBy?.include;
     if (!include) return null;
     const unionQueries = [];
@@ -860,17 +861,19 @@ var MemoryStorageD1 = class extends MemoryStorage {
   }
   async getMessages({
     threadId,
+    resourceId,
     selectBy,
     format
   }) {
-    const fullTableName = this.operations.getTableName(TABLE_MESSAGES);
-    const limit = resolveMessageLimit({
-      last: selectBy?.last,
-      defaultLimit: 40
-    });
-    const include = selectBy?.include || [];
-    const messages = [];
     try {
+      if (!threadId.trim()) throw new Error("threadId must be a non-empty string");
+      const fullTableName = this.operations.getTableName(TABLE_MESSAGES);
+      const limit = resolveMessageLimit({
+        last: selectBy?.last,
+        defaultLimit: 40
+      });
+      const include = selectBy?.include || [];
+      const messages = [];
       if (include.length) {
         const includeResult = await this._getIncludedMessages(threadId, selectBy);
         if (Array.isArray(includeResult)) messages.push(...includeResult);
@@ -910,7 +913,7 @@ var MemoryStorageD1 = class extends MemoryStorage {
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           text: `Failed to retrieve messages for thread ${threadId}: ${error instanceof Error ? error.message : String(error)}`,
-          details: { threadId }
+          details: { threadId, resourceId: resourceId ?? "" }
         },
         error
       );
@@ -962,6 +965,7 @@ var MemoryStorageD1 = class extends MemoryStorage {
   }
   async getMessagesPaginated({
     threadId,
+    resourceId,
     selectBy,
     format
   }) {
@@ -971,6 +975,7 @@ var MemoryStorageD1 = class extends MemoryStorage {
     const fullTableName = this.operations.getTableName(TABLE_MESSAGES);
     const messages = [];
     try {
+      if (!threadId.trim()) throw new Error("threadId must be a non-empty string");
       if (selectBy?.include?.length) {
         const includeResult = await this._getIncludedMessages(threadId, selectBy);
         if (Array.isArray(includeResult)) messages.push(...includeResult);
@@ -1059,7 +1064,7 @@ var MemoryStorageD1 = class extends MemoryStorage {
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           text: `Failed to retrieve messages for thread ${threadId}: ${error instanceof Error ? error.message : String(error)}`,
-          details: { threadId }
+          details: { threadId, resourceId: resourceId ?? "" }
         },
         error
       );
@@ -1874,7 +1879,7 @@ var TracesStorageD1 = class extends TracesStorage {
       const allDataResult = await this.operations.executeQuery(
         createSqlBuilder().select("*").from(fullTableName).where("1=1").build()
       );
-      console.log("allDataResult", allDataResult);
+      console.info("allDataResult", allDataResult);
       const countResult = await this.operations.executeQuery(countQuery.build());
       const total = Number(countResult?.[0]?.count ?? 0);
       dataQuery.orderBy("startTime", "DESC").limit(perPage).offset(page * perPage);
@@ -1945,6 +1950,7 @@ var WorkflowsStorageD1 = class extends WorkflowsStorage {
   async persistWorkflowSnapshot({
     workflowName,
     runId,
+    resourceId,
     snapshot
   }) {
     const fullTableName = this.operations.getTableName(TABLE_WORKFLOW_SNAPSHOT);
@@ -1955,11 +1961,13 @@ var WorkflowsStorageD1 = class extends WorkflowsStorage {
     });
     const persisting = currentSnapshot ? {
       ...currentSnapshot,
+      resourceId,
       snapshot: JSON.stringify(snapshot),
       updatedAt: now
     } : {
       workflow_name: workflowName,
       run_id: runId,
+      resourceId,
       snapshot,
       createdAt: now,
       updatedAt: now
@@ -2329,9 +2337,10 @@ var D1Store = class extends MastraStorage {
   async persistWorkflowSnapshot({
     workflowName,
     runId,
+    resourceId,
     snapshot
   }) {
-    return this.stores.workflows.persistWorkflowSnapshot({ workflowName, runId, snapshot });
+    return this.stores.workflows.persistWorkflowSnapshot({ workflowName, runId, resourceId, snapshot });
   }
   async loadWorkflowSnapshot(params) {
     return this.stores.workflows.loadWorkflowSnapshot(params);