@mastra/cloudflare 0.1.1-alpha.3 → 0.1.1-alpha.5

package/dist/_tsup-dts-rollup.d.cts

@@ -13,10 +13,13 @@ import type { TABLE_NAMES } from '@mastra/core/storage';
 import type { TABLE_THREADS } from '@mastra/core/storage';
 import type { TABLE_TRACES } from '@mastra/core/storage';
 import type { TABLE_WORKFLOW_SNAPSHOT } from '@mastra/core/storage';
+import type { WorkflowRun } from '@mastra/core/storage';
 import type { WorkflowRuns } from '@mastra/core/storage';
 import type { WorkflowRunState } from '@mastra/core/workflows';
 import type { WorkflowRunState as WorkflowRunState_2 } from '@mastra/core';
 
+export declare const checkWorkflowSnapshot: (snapshot: WorkflowRunState_2 | string, stepId: string, status: string) => void;
+
 /**
  * Configuration for Cloudflare KV using REST API
  */
@@ -46,10 +49,7 @@ declare class CloudflareStore extends MastraStorage {
     private getNamespaceValue;
     private putNamespaceValue;
     private deleteNamespaceValue;
-    listNamespaceKeys(tableName: TABLE_NAMES, options?: {
-        limit?: number;
-        prefix?: string;
-    }): Promise<KVNamespaceListKey_2<unknown, string>[]>;
+    listNamespaceKeys(tableName: TABLE_NAMES, options?: ListOptions): Promise<KVNamespaceListKey_2<unknown, string>[]>;
     private createNamespaceById;
     private getNamespaceIdByName;
     private createNamespace;
@@ -103,9 +103,9 @@ declare class CloudflareStore extends MastraStorage {
     clearTable({ tableName }: {
         tableName: TABLE_NAMES;
     }): Promise<void>;
-    insert<T extends TABLE_NAMES>({ tableName, record }: {
+    insert<T extends TABLE_NAMES>({ tableName, record, }: {
         tableName: T;
-        record: RecordTypes[T];
+        record: Record<string, any>;
     }): Promise<void>;
     load<R>({ tableName, keys }: {
         tableName: TABLE_NAMES;
@@ -162,14 +162,22 @@ declare class CloudflareStore extends MastraStorage {
     }): Promise<any[]>;
     private parseJSON;
     getEvalsByAgentName(_agentName: string, _type?: 'test' | 'live'): Promise<EvalRow[]>;
-    getWorkflowRuns(_args?: {
+    private parseWorkflowRun;
+    private buildWorkflowSnapshotPrefix;
+    getWorkflowRuns({ namespace, workflowName, limit, offset, resourceId, fromDate, toDate, }?: {
         namespace?: string;
         workflowName?: string;
-        fromDate?: Date;
-        toDate?: Date;
         limit?: number;
         offset?: number;
+        resourceId?: string;
+        fromDate?: Date;
+        toDate?: Date;
     }): Promise<WorkflowRuns>;
+    getWorkflowRunById({ namespace, runId, workflowName, }: {
+        namespace: string;
+        runId: string;
+        workflowName: string;
+    }): Promise<WorkflowRun | null>;
     close(): Promise<void>;
 }
 export { CloudflareStore }
@@ -222,7 +230,11 @@ export declare const createSampleTrace: (name: string, scope?: string, attribute
     createdAt: string;
 };
 
-export declare const createSampleWorkflowSnapshot: (threadId: string) => WorkflowRunState_2;
+export declare const createSampleWorkflowSnapshot: (threadId: string, status: string, createdAt?: Date) => {
+    snapshot: WorkflowRunState_2;
+    runId: string;
+    stepId: string;
+};
 
 /**
  * Helper to determine if a config is using Workers bindings
@@ -243,6 +255,11 @@ export declare interface KVOperation {
     metadata?: any;
 }
 
+export declare type ListOptions = {
+    limit?: number;
+    prefix?: string;
+};
+
 export declare type RecordTypes = {
     [TABLE_THREADS]: StorageThreadType;
     [TABLE_MESSAGES]: MessageType;

package/dist/_tsup-dts-rollup.d.ts

@@ -13,10 +13,13 @@ import type { TABLE_NAMES } from '@mastra/core/storage';
 import type { TABLE_THREADS } from '@mastra/core/storage';
 import type { TABLE_TRACES } from '@mastra/core/storage';
 import type { TABLE_WORKFLOW_SNAPSHOT } from '@mastra/core/storage';
+import type { WorkflowRun } from '@mastra/core/storage';
 import type { WorkflowRuns } from '@mastra/core/storage';
 import type { WorkflowRunState } from '@mastra/core/workflows';
 import type { WorkflowRunState as WorkflowRunState_2 } from '@mastra/core';
 
+export declare const checkWorkflowSnapshot: (snapshot: WorkflowRunState_2 | string, stepId: string, status: string) => void;
+
 /**
  * Configuration for Cloudflare KV using REST API
  */
@@ -46,10 +49,7 @@ declare class CloudflareStore extends MastraStorage {
     private getNamespaceValue;
     private putNamespaceValue;
     private deleteNamespaceValue;
-    listNamespaceKeys(tableName: TABLE_NAMES, options?: {
-        limit?: number;
-        prefix?: string;
-    }): Promise<KVNamespaceListKey_2<unknown, string>[]>;
+    listNamespaceKeys(tableName: TABLE_NAMES, options?: ListOptions): Promise<KVNamespaceListKey_2<unknown, string>[]>;
     private createNamespaceById;
     private getNamespaceIdByName;
     private createNamespace;
@@ -103,9 +103,9 @@ declare class CloudflareStore extends MastraStorage {
     clearTable({ tableName }: {
         tableName: TABLE_NAMES;
     }): Promise<void>;
-    insert<T extends TABLE_NAMES>({ tableName, record }: {
+    insert<T extends TABLE_NAMES>({ tableName, record, }: {
         tableName: T;
-        record: RecordTypes[T];
+        record: Record<string, any>;
     }): Promise<void>;
     load<R>({ tableName, keys }: {
         tableName: TABLE_NAMES;
@@ -162,14 +162,22 @@ declare class CloudflareStore extends MastraStorage {
     }): Promise<any[]>;
     private parseJSON;
     getEvalsByAgentName(_agentName: string, _type?: 'test' | 'live'): Promise<EvalRow[]>;
-    getWorkflowRuns(_args?: {
+    private parseWorkflowRun;
+    private buildWorkflowSnapshotPrefix;
+    getWorkflowRuns({ namespace, workflowName, limit, offset, resourceId, fromDate, toDate, }?: {
         namespace?: string;
         workflowName?: string;
-        fromDate?: Date;
-        toDate?: Date;
         limit?: number;
         offset?: number;
+        resourceId?: string;
+        fromDate?: Date;
+        toDate?: Date;
     }): Promise<WorkflowRuns>;
+    getWorkflowRunById({ namespace, runId, workflowName, }: {
+        namespace: string;
+        runId: string;
+        workflowName: string;
+    }): Promise<WorkflowRun | null>;
     close(): Promise<void>;
 }
 export { CloudflareStore }
@@ -222,7 +230,11 @@ export declare const createSampleTrace: (name: string, scope?: string, attribute
     createdAt: string;
 };
 
-export declare const createSampleWorkflowSnapshot: (threadId: string) => WorkflowRunState_2;
+export declare const createSampleWorkflowSnapshot: (threadId: string, status: string, createdAt?: Date) => {
+    snapshot: WorkflowRunState_2;
+    runId: string;
+    stepId: string;
+};
 
 /**
  * Helper to determine if a config is using Workers bindings
@@ -243,6 +255,11 @@ export declare interface KVOperation {
     metadata?: any;
 }
 
+export declare type ListOptions = {
+    limit?: number;
+    prefix?: string;
+};
+
 export declare type RecordTypes = {
     [TABLE_THREADS]: StorageThreadType;
     [TABLE_MESSAGES]: MessageType;

package/dist/index.cjs

@@ -63,7 +63,8 @@ var CloudflareStore = class extends storage.MastraStorage {
         this.logger.info("Using Cloudflare KV REST API");
       }
     } catch (error) {
-      this.logger.error("Failed to initialize CloudflareStore:", { error });
+      const message = error instanceof Error ? error.message : String(error);
+      this.logger.error("Failed to initialize CloudflareStore:", { message });
       throw error;
     }
   }
@@ -105,7 +106,8 @@ var CloudflareStore = class extends storage.MastraStorage {
       if (error.message && error.message.includes("key not found")) {
         return null;
       }
-      this.logger.error(`Failed to get value for ${tableName} ${key}:`, { error });
+      const message = error instanceof Error ? error.message : String(error);
+      this.logger.error(`Failed to get value for ${tableName} ${key}:`, { message });
       throw error;
     }
   }
@@ -130,7 +132,8 @@ var CloudflareStore = class extends storage.MastraStorage {
         });
       }
     } catch (error) {
-      this.logger.error(`Failed to put value for ${tableName} ${key}:`, { error });
+      const message = error instanceof Error ? error.message : String(error);
+      this.logger.error(`Failed to put value for ${tableName} ${key}:`, { message });
       throw error;
     }
   }
@@ -253,7 +256,8 @@ var CloudflareStore = class extends storage.MastraStorage {
       }
       return data;
     } catch (error) {
-      this.logger.error("Failed to parse text:", { error, text });
+      const message = error instanceof Error ? error.message : String(error);
+      this.logger.error("Failed to parse text:", { message, text });
       return null;
     }
   }
@@ -287,9 +291,9 @@ var CloudflareStore = class extends storage.MastraStorage {
       throw new Error(`Failed to delete KV value: ${error.message}`);
     }
   }
-  async listKV(tableName) {
+  async listKV(tableName, options) {
     try {
-      return await this.listNamespaceKeys(tableName);
+      return await this.listNamespaceKeys(tableName, options);
     } catch (error) {
       this.logger.error(`Failed to list KV for ${tableName}:`, error);
       throw new Error(`Failed to list KV: ${error.message}`);
@@ -362,7 +366,8 @@ var CloudflareStore = class extends storage.MastraStorage {
           if (!data) return null;
           return typeof data === "string" ? JSON.parse(data) : data;
         } catch (error) {
-          this.logger.error(`Error retrieving message ${id}:`, { error });
+          const message = error instanceof Error ? error.message : String(error);
+          this.logger.error(`Error retrieving message ${id}:`, { message });
           return null;
         }
       })
@@ -395,7 +400,8 @@ var CloudflareStore = class extends storage.MastraStorage {
           value: JSON.stringify(updatedOrder)
         });
       } catch (error) {
-        this.logger.error(`Error updating sorted order for key ${orderKey}:`, { error });
+        const message = error instanceof Error ? error.message : String(error);
+        this.logger.error(`Error updating sorted order for key ${orderKey}:`, { message });
         throw error;
       } finally {
         if (this.updateQueue.get(orderKey) === nextPromise) {
@@ -437,7 +443,11 @@ var CloudflareStore = class extends storage.MastraStorage {
         if (!record.namespace || !record.workflow_name || !record.run_id) {
           throw new Error("Namespace, workflow name, and run ID are required");
         }
-        return `${prefix}${tableName}:${record.namespace}:${record.workflow_name}:${record.run_id}`;
+        let key = `${prefix}${tableName}:${record.namespace}:${record.workflow_name}:${record.run_id}`;
+        if (record.resourceId) {
+          key = `${key}:${record.resourceId}`;
+        }
+        return key;
       case storage.TABLE_TRACES:
         if (!record.id) throw new Error("Trace ID is required");
         return `${prefix}${tableName}:${record.id}`;
@@ -454,7 +464,8 @@ var CloudflareStore = class extends storage.MastraStorage {
       const schemaKey = this.getSchemaKey(tableName);
       return await this.getKV(tableName, schemaKey);
     } catch (error) {
-      this.logger.error(`Failed to get schema for ${tableName}:`, { error });
+      const message = error instanceof Error ? error.message : String(error);
+      this.logger.error(`Failed to get schema for ${tableName}:`, { message });
       return null;
     }
   }
@@ -499,7 +510,8 @@ var CloudflareStore = class extends storage.MastraStorage {
         }
       }
     } catch (error) {
-      this.logger.error(`Error validating record against schema:`, { error, record, schema });
+      const message = error instanceof Error ? error.message : String(error);
+      this.logger.error(`Error validating record against schema:`, { message, record, schema });
       throw error;
     }
   }
@@ -526,7 +538,7 @@ var CloudflareStore = class extends storage.MastraStorage {
           }
           break;
         case storage.TABLE_WORKFLOW_SNAPSHOT:
-          if (!("namespace" in recordTyped) || !("workflowName" in recordTyped) || !("runId" in recordTyped)) {
+          if (!("namespace" in recordTyped) || !("workflow_name" in recordTyped) || !("run_id" in recordTyped)) {
             throw new Error("Workflow record missing required fields");
           }
           break;
@@ -539,7 +551,8 @@ var CloudflareStore = class extends storage.MastraStorage {
           throw new Error(`Unknown table type: ${tableName}`);
       }
     } catch (error) {
-      this.logger.error(`Failed to validate record for ${tableName}:`, { error, record });
+      const message = error instanceof Error ? error.message : String(error);
+      this.logger.error(`Failed to validate record for ${tableName}:`, { message, record });
       throw error;
     }
   }
@@ -579,7 +592,10 @@ var CloudflareStore = class extends storage.MastraStorage {
       await Promise.all(keys.map((keyObj) => this.deleteKV(tableName, keyObj.name)));
     }
   }
-  async insert({ tableName, record }) {
+  async insert({
+    tableName,
+    record
+  }) {
     try {
       const key = this.getKey(tableName, record);
       const processedRecord = {
@@ -591,7 +607,8 @@ var CloudflareStore = class extends storage.MastraStorage {
       await this.validateRecord(processedRecord, tableName);
       await this.putKV({ tableName, key, value: processedRecord });
     } catch (error) {
-      this.logger.error(`Failed to insert record for ${tableName}:`, { error });
+      const message = error instanceof Error ? error.message : String(error);
+      this.logger.error(`Failed to insert record for ${tableName}:`, { message });
       throw error;
     }
   }
@@ -608,7 +625,9 @@ var CloudflareStore = class extends storage.MastraStorage {
       };
       return processed;
     } catch (error) {
-      this.logger.error(`Failed to load data for ${tableName}:`, { error });
+      this.logger.error(`Failed to load data for ${tableName}:`, {
+        error: error instanceof Error ? error.message : String(error)
+      });
       return null;
     }
   }
@@ -623,7 +642,9 @@ var CloudflareStore = class extends storage.MastraStorage {
         metadata: this.ensureMetadata(thread.metadata)
       };
     } catch (error) {
-      this.logger.error(`Error processing thread ${threadId}:`, { error });
+      this.logger.error(`Error processing thread ${threadId}:`, {
+        error: error instanceof Error ? error.message : String(error)
+      });
       return null;
     }
   }
@@ -644,14 +665,16 @@ var CloudflareStore = class extends storage.MastraStorage {
               metadata: this.ensureMetadata(thread.metadata)
             };
           } catch (error) {
-            this.logger.error(`Error processing thread from key ${keyObj.name}:`, { error });
+            const message = error instanceof Error ? error.message : String(error);
+            this.logger.error(`Error processing thread from key ${keyObj.name}:`, { message });
             return null;
           }
         })
       );
       return threads.filter((thread) => thread !== null);
     } catch (error) {
-      this.logger.error(`Error getting threads for resourceId ${resourceId}:`, { error });
+      const message = error instanceof Error ? error.message : String(error);
+      this.logger.error(`Error getting threads for resourceId ${resourceId}:`, { message });
       return [];
     }
   }
@@ -660,7 +683,8 @@ var CloudflareStore = class extends storage.MastraStorage {
       await this.insert({ tableName: storage.TABLE_THREADS, record: thread });
       return thread;
     } catch (error) {
-      this.logger.error("Error saving thread:", { error });
+      const message = error instanceof Error ? error.message : String(error);
+      this.logger.error("Error saving thread:", { message });
       throw error;
     }
   }
@@ -686,7 +710,8 @@ var CloudflareStore = class extends storage.MastraStorage {
       await this.insert({ tableName: storage.TABLE_THREADS, record: updatedThread });
       return updatedThread;
     } catch (error) {
-      this.logger.error(`Error updating thread ${id}:`, { error });
+      const message = error instanceof Error ? error.message : String(error);
+      this.logger.error(`Error updating thread ${id}:`, { message });
       throw error;
     }
   }
@@ -707,7 +732,8 @@ var CloudflareStore = class extends storage.MastraStorage {
         this.deleteKV(storage.TABLE_THREADS, this.getKey(storage.TABLE_THREADS, { id: threadId }))
       ]);
     } catch (error) {
-      this.logger.error(`Error deleting thread ${threadId}:`, { error });
+      const message = error instanceof Error ? error.message : String(error);
+      this.logger.error(`Error deleting thread ${threadId}:`, { message });
       throw error;
     }
   }
@@ -715,7 +741,8 @@ var CloudflareStore = class extends storage.MastraStorage {
     try {
       return this.getKey(storage.TABLE_MESSAGES, { threadId, id: messageId });
     } catch (error) {
-      this.logger.error(`Error getting message key for thread ${threadId} and message ${messageId}:`, { error });
+      const message = error instanceof Error ? error.message : String(error);
+      this.logger.error(`Error getting message key for thread ${threadId} and message ${messageId}:`, { message });
       throw error;
     }
   }
@@ -723,7 +750,8 @@ var CloudflareStore = class extends storage.MastraStorage {
     try {
       return this.getKey(storage.TABLE_MESSAGES, { threadId, id: "messages" });
     } catch (error) {
-      this.logger.error(`Error getting thread messages key for thread ${threadId}:`, { error });
+      const message = error instanceof Error ? error.message : String(error);
+      this.logger.error(`Error getting thread messages key for thread ${threadId}:`, { message });
       throw error;
     }
   }
@@ -874,10 +902,20 @@ var CloudflareStore = class extends storage.MastraStorage {
       const { namespace, workflowName, runId, snapshot } = params;
       const normalizedState = this.normalizeWorkflowState(snapshot);
       this.validateWorkflowState(normalizedState);
-      const key = this.getKey(storage.TABLE_WORKFLOW_SNAPSHOT, { namespace, workflow_name: workflowName, run_id: runId });
-      await this.putKV({ tableName: storage.TABLE_WORKFLOW_SNAPSHOT, key, value: normalizedState });
+      await this.insert({
+        tableName: storage.TABLE_WORKFLOW_SNAPSHOT,
+        record: {
+          namespace,
+          workflow_name: workflowName,
+          run_id: runId,
+          snapshot: normalizedState,
+          createdAt: /* @__PURE__ */ new Date(),
+          updatedAt: /* @__PURE__ */ new Date()
+        }
+      });
     } catch (error) {
-      this.logger.error("Error persisting workflow snapshot:", { error });
+      const message = error instanceof Error ? error.message : String(error);
+      this.logger.error("Error persisting workflow snapshot:", { message });
       throw error;
     }
   }
@@ -888,11 +926,13 @@ var CloudflareStore = class extends storage.MastraStorage {
       const key = this.getKey(storage.TABLE_WORKFLOW_SNAPSHOT, { namespace, workflow_name: workflowName, run_id: runId });
       const data = await this.getKV(storage.TABLE_WORKFLOW_SNAPSHOT, key);
       if (!data) return null;
-      const state = this.normalizeWorkflowState(data);
+      const state = this.normalizeWorkflowState(data.snapshot || data);
       this.validateWorkflowState(state);
       return state;
     } catch (error) {
-      this.logger.error("Error loading workflow snapshot:", { error });
+      this.logger.error("Error loading workflow snapshot:", {
+        error: error instanceof Error ? error.message : String(error)
+      });
       return null;
     }
   }
@@ -912,7 +952,8 @@ var CloudflareStore = class extends storage.MastraStorage {
         })
       );
     } catch (error) {
-      this.logger.error("Error in batch insert:", { error });
+      const message = error instanceof Error ? error.message : String(error);
+      this.logger.error("Error in batch insert:", { message });
       throw error;
     }
   }
@@ -985,7 +1026,8 @@ var CloudflareStore = class extends storage.MastraStorage {
         createdAt: record.createdAt
       }));
     } catch (error) {
-      this.logger.error("Failed to get traces:", { message: error instanceof Error ? error.message : String(error) });
+      const message = error instanceof Error ? error.message : String(error);
+      this.logger.error("Failed to get traces:", { message });
       return [];
     }
   }
@@ -1002,8 +1044,115 @@ var CloudflareStore = class extends storage.MastraStorage {
   getEvalsByAgentName(_agentName, _type) {
     throw new Error("Method not implemented.");
   }
-  getWorkflowRuns(_args) {
-    throw new Error("Method not implemented.");
+  parseWorkflowRun(row) {
+    let parsedSnapshot = row.snapshot;
+    if (typeof parsedSnapshot === "string") {
+      try {
+        parsedSnapshot = JSON.parse(row.snapshot);
+      } catch (e) {
+        console.warn(`Failed to parse snapshot for workflow ${row.workflow_name}: ${e}`);
+      }
+    }
+    return {
+      workflowName: row.workflow_name,
+      runId: row.run_id,
+      snapshot: parsedSnapshot,
+      createdAt: this.ensureDate(row.createdAt),
+      updatedAt: this.ensureDate(row.updatedAt),
+      resourceId: row.resourceId
+    };
+  }
+  buildWorkflowSnapshotPrefix({
+    namespace,
+    workflowName,
+    runId,
+    resourceId
+  }) {
+    const prefix = this.namespacePrefix ? `${this.namespacePrefix}:` : "";
+    let key = `${prefix}${storage.TABLE_WORKFLOW_SNAPSHOT}`;
+    if (namespace) key += `:${namespace}`;
+    if (workflowName) key += `:${workflowName}`;
+    if (runId) key += `:${runId}`;
+    if (resourceId) key += `:${resourceId}`;
+    if (!resourceId && (runId || workflowName || namespace)) key += ":";
+    return key;
+  }
+  async getWorkflowRuns({
+    namespace,
+    workflowName,
+    limit = 20,
+    offset = 0,
+    resourceId,
+    fromDate,
+    toDate
+  } = {}) {
+    try {
+      const prefix = this.buildWorkflowSnapshotPrefix({ namespace, workflowName });
+      const keyObjs = await this.listKV(storage.TABLE_WORKFLOW_SNAPSHOT, { prefix });
+      const runs = [];
+      for (const { name: key } of keyObjs) {
+        const parts = key.split(":");
+        const idx = parts.indexOf(storage.TABLE_WORKFLOW_SNAPSHOT);
+        if (idx === -1 || parts.length < idx + 4) continue;
+        const ns = parts[idx + 1];
+        const wfName = parts[idx + 2];
+        const keyResourceId = parts.length > idx + 4 ? parts[idx + 4] : void 0;
+        if (namespace && ns !== namespace || workflowName && wfName !== workflowName) continue;
+        if (resourceId && keyResourceId && keyResourceId !== resourceId) continue;
+        const data = await this.getKV(storage.TABLE_WORKFLOW_SNAPSHOT, key);
+        if (!data) continue;
+        try {
+          if (resourceId && data.resourceId && data.resourceId !== resourceId) continue;
+          const createdAt = this.ensureDate(data.createdAt);
+          if (fromDate && createdAt && createdAt < fromDate) continue;
+          if (toDate && createdAt && createdAt > toDate) continue;
+          const state = this.normalizeWorkflowState(data.snapshot || data);
+          this.validateWorkflowState(state);
+          const run = this.parseWorkflowRun({ ...data, snapshot: state });
+          runs.push(run);
+        } catch (err) {
+          this.logger.error("Failed to parse workflow snapshot:", { key, error: err });
+        }
+      }
+      runs.sort((a, b) => {
+        const aDate = a.createdAt ? new Date(a.createdAt).getTime() : 0;
+        const bDate = b.createdAt ? new Date(b.createdAt).getTime() : 0;
+        return bDate - aDate;
+      });
+      const pagedRuns = runs.slice(offset, offset + limit);
+      return {
+        runs: pagedRuns,
+        total: runs.length
+      };
+    } catch (error) {
+      const message = error instanceof Error ? error.message : String(error);
+      this.logger.error("Error in getWorkflowRuns:", { message });
+      return { runs: [], total: 0 };
+    }
+  }
+  async getWorkflowRunById({
+    namespace,
+    runId,
+    workflowName
+  }) {
+    try {
+      if (!runId || !workflowName || !namespace) {
+        throw new Error("runId, workflowName, and namespace are required");
+      }
+      const prefix = this.buildWorkflowSnapshotPrefix({ namespace, workflowName, runId });
+      const keyObjs = await this.listKV(storage.TABLE_WORKFLOW_SNAPSHOT, { prefix });
+      if (!keyObjs.length) return null;
+      const key = keyObjs[0]?.name;
+      const data = await this.getKV(storage.TABLE_WORKFLOW_SNAPSHOT, key);
+      if (!data) return null;
+      const state = this.normalizeWorkflowState(data.snapshot || data);
+      this.validateWorkflowState(state);
+      return this.parseWorkflowRun({ ...data, snapshot: state });
+    } catch (error) {
+      const message = error instanceof Error ? error.message : String(error);
+      this.logger.error("Error in getWorkflowRunById:", { message });
+      return null;
+    }
   }
   async close() {
   }

package/dist/index.js

@@ -57,7 +57,8 @@ var CloudflareStore = class extends MastraStorage {
         this.logger.info("Using Cloudflare KV REST API");
       }
     } catch (error) {
-      this.logger.error("Failed to initialize CloudflareStore:", { error });
+      const message = error instanceof Error ? error.message : String(error);
+      this.logger.error("Failed to initialize CloudflareStore:", { message });
       throw error;
     }
   }
@@ -99,7 +100,8 @@ var CloudflareStore = class extends MastraStorage {
       if (error.message && error.message.includes("key not found")) {
         return null;
       }
-      this.logger.error(`Failed to get value for ${tableName} ${key}:`, { error });
+      const message = error instanceof Error ? error.message : String(error);
+      this.logger.error(`Failed to get value for ${tableName} ${key}:`, { message });
       throw error;
     }
   }
@@ -124,7 +126,8 @@ var CloudflareStore = class extends MastraStorage {
         });
       }
     } catch (error) {
-      this.logger.error(`Failed to put value for ${tableName} ${key}:`, { error });
+      const message = error instanceof Error ? error.message : String(error);
+      this.logger.error(`Failed to put value for ${tableName} ${key}:`, { message });
       throw error;
     }
   }
@@ -247,7 +250,8 @@ var CloudflareStore = class extends MastraStorage {
       }
       return data;
     } catch (error) {
-      this.logger.error("Failed to parse text:", { error, text });
+      const message = error instanceof Error ? error.message : String(error);
+      this.logger.error("Failed to parse text:", { message, text });
       return null;
     }
   }
@@ -281,9 +285,9 @@ var CloudflareStore = class extends MastraStorage {
       throw new Error(`Failed to delete KV value: ${error.message}`);
     }
   }
-  async listKV(tableName) {
+  async listKV(tableName, options) {
     try {
-      return await this.listNamespaceKeys(tableName);
+      return await this.listNamespaceKeys(tableName, options);
     } catch (error) {
       this.logger.error(`Failed to list KV for ${tableName}:`, error);
       throw new Error(`Failed to list KV: ${error.message}`);
@@ -356,7 +360,8 @@ var CloudflareStore = class extends MastraStorage {
           if (!data) return null;
           return typeof data === "string" ? JSON.parse(data) : data;
         } catch (error) {
-          this.logger.error(`Error retrieving message ${id}:`, { error });
+          const message = error instanceof Error ? error.message : String(error);
+          this.logger.error(`Error retrieving message ${id}:`, { message });
           return null;
         }
       })
@@ -389,7 +394,8 @@ var CloudflareStore = class extends MastraStorage {
           value: JSON.stringify(updatedOrder)
         });
       } catch (error) {
-        this.logger.error(`Error updating sorted order for key ${orderKey}:`, { error });
+        const message = error instanceof Error ? error.message : String(error);
+        this.logger.error(`Error updating sorted order for key ${orderKey}:`, { message });
         throw error;
       } finally {
         if (this.updateQueue.get(orderKey) === nextPromise) {
@@ -431,7 +437,11 @@ var CloudflareStore = class extends MastraStorage {
         if (!record.namespace || !record.workflow_name || !record.run_id) {
           throw new Error("Namespace, workflow name, and run ID are required");
         }
-        return `${prefix}${tableName}:${record.namespace}:${record.workflow_name}:${record.run_id}`;
+        let key = `${prefix}${tableName}:${record.namespace}:${record.workflow_name}:${record.run_id}`;
+        if (record.resourceId) {
+          key = `${key}:${record.resourceId}`;
+        }
+        return key;
       case TABLE_TRACES:
         if (!record.id) throw new Error("Trace ID is required");
         return `${prefix}${tableName}:${record.id}`;
@@ -448,7 +458,8 @@ var CloudflareStore = class extends MastraStorage {
       const schemaKey = this.getSchemaKey(tableName);
       return await this.getKV(tableName, schemaKey);
     } catch (error) {
-      this.logger.error(`Failed to get schema for ${tableName}:`, { error });
+      const message = error instanceof Error ? error.message : String(error);
+      this.logger.error(`Failed to get schema for ${tableName}:`, { message });
       return null;
     }
   }
@@ -493,7 +504,8 @@ var CloudflareStore = class extends MastraStorage {
         }
       }
     } catch (error) {
-      this.logger.error(`Error validating record against schema:`, { error, record, schema });
+      const message = error instanceof Error ? error.message : String(error);
+      this.logger.error(`Error validating record against schema:`, { message, record, schema });
       throw error;
     }
   }
@@ -520,7 +532,7 @@ var CloudflareStore = class extends MastraStorage {
           }
           break;
         case TABLE_WORKFLOW_SNAPSHOT:
-          if (!("namespace" in recordTyped) || !("workflowName" in recordTyped) || !("runId" in recordTyped)) {
+          if (!("namespace" in recordTyped) || !("workflow_name" in recordTyped) || !("run_id" in recordTyped)) {
             throw new Error("Workflow record missing required fields");
           }
           break;
@@ -533,7 +545,8 @@ var CloudflareStore = class extends MastraStorage {
           throw new Error(`Unknown table type: ${tableName}`);
       }
     } catch (error) {
-      this.logger.error(`Failed to validate record for ${tableName}:`, { error, record });
+      const message = error instanceof Error ? error.message : String(error);
+      this.logger.error(`Failed to validate record for ${tableName}:`, { message, record });
       throw error;
     }
   }
@@ -573,7 +586,10 @@ var CloudflareStore = class extends MastraStorage {
       await Promise.all(keys.map((keyObj) => this.deleteKV(tableName, keyObj.name)));
     }
   }
-  async insert({ tableName, record }) {
+  async insert({
+    tableName,
+    record
+  }) {
     try {
       const key = this.getKey(tableName, record);
       const processedRecord = {
@@ -585,7 +601,8 @@ var CloudflareStore = class extends MastraStorage {
       await this.validateRecord(processedRecord, tableName);
       await this.putKV({ tableName, key, value: processedRecord });
     } catch (error) {
-      this.logger.error(`Failed to insert record for ${tableName}:`, { error });
+      const message = error instanceof Error ? error.message : String(error);
+      this.logger.error(`Failed to insert record for ${tableName}:`, { message });
       throw error;
     }
   }
@@ -602,7 +619,9 @@ var CloudflareStore = class extends MastraStorage {
       };
       return processed;
     } catch (error) {
-      this.logger.error(`Failed to load data for ${tableName}:`, { error });
+      this.logger.error(`Failed to load data for ${tableName}:`, {
+        error: error instanceof Error ? error.message : String(error)
+      });
       return null;
     }
   }
@@ -617,7 +636,9 @@ var CloudflareStore = class extends MastraStorage {
         metadata: this.ensureMetadata(thread.metadata)
       };
     } catch (error) {
-      this.logger.error(`Error processing thread ${threadId}:`, { error });
+      this.logger.error(`Error processing thread ${threadId}:`, {
+        error: error instanceof Error ? error.message : String(error)
+      });
       return null;
     }
   }
@@ -638,14 +659,16 @@ var CloudflareStore = class extends MastraStorage {
               metadata: this.ensureMetadata(thread.metadata)
             };
           } catch (error) {
-            this.logger.error(`Error processing thread from key ${keyObj.name}:`, { error });
+            const message = error instanceof Error ? error.message : String(error);
+            this.logger.error(`Error processing thread from key ${keyObj.name}:`, { message });
             return null;
           }
         })
       );
       return threads.filter((thread) => thread !== null);
     } catch (error) {
-      this.logger.error(`Error getting threads for resourceId ${resourceId}:`, { error });
+      const message = error instanceof Error ? error.message : String(error);
+      this.logger.error(`Error getting threads for resourceId ${resourceId}:`, { message });
       return [];
     }
   }
@@ -654,7 +677,8 @@ var CloudflareStore = class extends MastraStorage {
       await this.insert({ tableName: TABLE_THREADS, record: thread });
       return thread;
     } catch (error) {
-      this.logger.error("Error saving thread:", { error });
+      const message = error instanceof Error ? error.message : String(error);
+      this.logger.error("Error saving thread:", { message });
       throw error;
     }
   }
@@ -680,7 +704,8 @@ var CloudflareStore = class extends MastraStorage {
       await this.insert({ tableName: TABLE_THREADS, record: updatedThread });
       return updatedThread;
     } catch (error) {
-      this.logger.error(`Error updating thread ${id}:`, { error });
+      const message = error instanceof Error ? error.message : String(error);
+      this.logger.error(`Error updating thread ${id}:`, { message });
       throw error;
     }
   }
@@ -701,7 +726,8 @@ var CloudflareStore = class extends MastraStorage {
         this.deleteKV(TABLE_THREADS, this.getKey(TABLE_THREADS, { id: threadId }))
       ]);
     } catch (error) {
-      this.logger.error(`Error deleting thread ${threadId}:`, { error });
+      const message = error instanceof Error ? error.message : String(error);
+      this.logger.error(`Error deleting thread ${threadId}:`, { message });
       throw error;
     }
   }
@@ -709,7 +735,8 @@ var CloudflareStore = class extends MastraStorage {
     try {
       return this.getKey(TABLE_MESSAGES, { threadId, id: messageId });
     } catch (error) {
-      this.logger.error(`Error getting message key for thread ${threadId} and message ${messageId}:`, { error });
+      const message = error instanceof Error ? error.message : String(error);
+      this.logger.error(`Error getting message key for thread ${threadId} and message ${messageId}:`, { message });
       throw error;
     }
   }
@@ -717,7 +744,8 @@ var CloudflareStore = class extends MastraStorage {
     try {
       return this.getKey(TABLE_MESSAGES, { threadId, id: "messages" });
     } catch (error) {
-      this.logger.error(`Error getting thread messages key for thread ${threadId}:`, { error });
+      const message = error instanceof Error ? error.message : String(error);
+      this.logger.error(`Error getting thread messages key for thread ${threadId}:`, { message });
       throw error;
     }
   }
@@ -868,10 +896,20 @@ var CloudflareStore = class extends MastraStorage {
       const { namespace, workflowName, runId, snapshot } = params;
       const normalizedState = this.normalizeWorkflowState(snapshot);
       this.validateWorkflowState(normalizedState);
-      const key = this.getKey(TABLE_WORKFLOW_SNAPSHOT, { namespace, workflow_name: workflowName, run_id: runId });
-      await this.putKV({ tableName: TABLE_WORKFLOW_SNAPSHOT, key, value: normalizedState });
+      await this.insert({
+        tableName: TABLE_WORKFLOW_SNAPSHOT,
+        record: {
+          namespace,
+          workflow_name: workflowName,
+          run_id: runId,
+          snapshot: normalizedState,
+          createdAt: /* @__PURE__ */ new Date(),
+          updatedAt: /* @__PURE__ */ new Date()
+        }
+      });
     } catch (error) {
-      this.logger.error("Error persisting workflow snapshot:", { error });
+      const message = error instanceof Error ? error.message : String(error);
+      this.logger.error("Error persisting workflow snapshot:", { message });
       throw error;
     }
   }
@@ -882,11 +920,13 @@ var CloudflareStore = class extends MastraStorage {
       const key = this.getKey(TABLE_WORKFLOW_SNAPSHOT, { namespace, workflow_name: workflowName, run_id: runId });
       const data = await this.getKV(TABLE_WORKFLOW_SNAPSHOT, key);
       if (!data) return null;
-      const state = this.normalizeWorkflowState(data);
+      const state = this.normalizeWorkflowState(data.snapshot || data);
       this.validateWorkflowState(state);
       return state;
     } catch (error) {
-      this.logger.error("Error loading workflow snapshot:", { error });
+      this.logger.error("Error loading workflow snapshot:", {
+        error: error instanceof Error ? error.message : String(error)
+      });
       return null;
     }
   }
@@ -906,7 +946,8 @@ var CloudflareStore = class extends MastraStorage {
         })
       );
     } catch (error) {
-      this.logger.error("Error in batch insert:", { error });
+      const message = error instanceof Error ? error.message : String(error);
+      this.logger.error("Error in batch insert:", { message });
       throw error;
     }
   }
@@ -979,7 +1020,8 @@ var CloudflareStore = class extends MastraStorage {
         createdAt: record.createdAt
       }));
     } catch (error) {
-      this.logger.error("Failed to get traces:", { message: error instanceof Error ? error.message : String(error) });
+      const message = error instanceof Error ? error.message : String(error);
+      this.logger.error("Failed to get traces:", { message });
       return [];
     }
   }
@@ -996,8 +1038,115 @@ var CloudflareStore = class extends MastraStorage {
   getEvalsByAgentName(_agentName, _type) {
     throw new Error("Method not implemented.");
   }
-  getWorkflowRuns(_args) {
-    throw new Error("Method not implemented.");
+  parseWorkflowRun(row) {
+    let parsedSnapshot = row.snapshot;
+    if (typeof parsedSnapshot === "string") {
+      try {
+        parsedSnapshot = JSON.parse(row.snapshot);
+      } catch (e) {
+        console.warn(`Failed to parse snapshot for workflow ${row.workflow_name}: ${e}`);
+      }
+    }
+    return {
+      workflowName: row.workflow_name,
+      runId: row.run_id,
+      snapshot: parsedSnapshot,
+      createdAt: this.ensureDate(row.createdAt),
+      updatedAt: this.ensureDate(row.updatedAt),
+      resourceId: row.resourceId
+    };
+  }
+  buildWorkflowSnapshotPrefix({
+    namespace,
+    workflowName,
+    runId,
+    resourceId
+  }) {
+    const prefix = this.namespacePrefix ? `${this.namespacePrefix}:` : "";
+    let key = `${prefix}${TABLE_WORKFLOW_SNAPSHOT}`;
+    if (namespace) key += `:${namespace}`;
+    if (workflowName) key += `:${workflowName}`;
+    if (runId) key += `:${runId}`;
+    if (resourceId) key += `:${resourceId}`;
+    if (!resourceId && (runId || workflowName || namespace)) key += ":";
+    return key;
+  }
+  async getWorkflowRuns({
+    namespace,
+    workflowName,
+    limit = 20,
+    offset = 0,
+    resourceId,
+    fromDate,
+    toDate
+  } = {}) {
+    try {
+      const prefix = this.buildWorkflowSnapshotPrefix({ namespace, workflowName });
+      const keyObjs = await this.listKV(TABLE_WORKFLOW_SNAPSHOT, { prefix });
+      const runs = [];
+      for (const { name: key } of keyObjs) {
+        const parts = key.split(":");
+        const idx = parts.indexOf(TABLE_WORKFLOW_SNAPSHOT);
+        if (idx === -1 || parts.length < idx + 4) continue;
+        const ns = parts[idx + 1];
+        const wfName = parts[idx + 2];
+        const keyResourceId = parts.length > idx + 4 ? parts[idx + 4] : void 0;
+        if (namespace && ns !== namespace || workflowName && wfName !== workflowName) continue;
+        if (resourceId && keyResourceId && keyResourceId !== resourceId) continue;
+        const data = await this.getKV(TABLE_WORKFLOW_SNAPSHOT, key);
+        if (!data) continue;
+        try {
+          if (resourceId && data.resourceId && data.resourceId !== resourceId) continue;
+          const createdAt = this.ensureDate(data.createdAt);
+          if (fromDate && createdAt && createdAt < fromDate) continue;
+          if (toDate && createdAt && createdAt > toDate) continue;
+          const state = this.normalizeWorkflowState(data.snapshot || data);
+          this.validateWorkflowState(state);
+          const run = this.parseWorkflowRun({ ...data, snapshot: state });
+          runs.push(run);
+        } catch (err) {
+          this.logger.error("Failed to parse workflow snapshot:", { key, error: err });
+        }
+      }
+      runs.sort((a, b) => {
+        const aDate = a.createdAt ? new Date(a.createdAt).getTime() : 0;
+        const bDate = b.createdAt ? new Date(b.createdAt).getTime() : 0;
+        return bDate - aDate;
+      });
+      const pagedRuns = runs.slice(offset, offset + limit);
+      return {
+        runs: pagedRuns,
+        total: runs.length
+      };
+    } catch (error) {
+      const message = error instanceof Error ? error.message : String(error);
+      this.logger.error("Error in getWorkflowRuns:", { message });
+      return { runs: [], total: 0 };
+    }
+  }
+  async getWorkflowRunById({
+    namespace,
+    runId,
+    workflowName
+  }) {
+    try {
+      if (!runId || !workflowName || !namespace) {
+        throw new Error("runId, workflowName, and namespace are required");
+      }
+      const prefix = this.buildWorkflowSnapshotPrefix({ namespace, workflowName, runId });
+      const keyObjs = await this.listKV(TABLE_WORKFLOW_SNAPSHOT, { prefix });
+      if (!keyObjs.length) return null;
+      const key = keyObjs[0]?.name;
+      const data = await this.getKV(TABLE_WORKFLOW_SNAPSHOT, key);
+      if (!data) return null;
+      const state = this.normalizeWorkflowState(data.snapshot || data);
+      this.validateWorkflowState(state);
+      return this.parseWorkflowRun({ ...data, snapshot: state });
+    } catch (error) {
+      const message = error instanceof Error ? error.message : String(error);
+      this.logger.error("Error in getWorkflowRunById:", { message });
+      return null;
+    }
   }
   async close() {
   }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/cloudflare",
-  "version": "0.1.1-alpha.3",
+  "version": "0.1.1-alpha.5",
   "description": "Cloudflare provider for Mastra - includes db storage capabilities",
   "type": "module",
   "files": [
@@ -24,18 +24,18 @@
   "license": "MIT",
   "dependencies": {
     "cloudflare": "^4.1.0",
-    "@mastra/core": "^0.9.1-alpha.3"
+    "@mastra/core": "^0.9.1-alpha.4"
   },
   "devDependencies": {
     "@cloudflare/workers-types": "^4.20250313.0",
-    "@microsoft/api-extractor": "^7.52.1",
+    "@microsoft/api-extractor": "^7.52.5",
     "@types/node": "^20.17.27",
     "dotenv": "^16.4.7",
     "eslint": "^9.23.0",
     "miniflare": "^4.20250310.0",
     "tsup": "^8.4.0",
     "typescript": "^5.8.2",
-    "vitest": "^3.0.9",
+    "vitest": "^3.1.2",
     "@internal/lint": "0.0.2"
   },
   "scripts": {