@mastra/cloudflare 0.0.0-vnextWorkflows-20250422142014 → 0.0.0-working-memory-per-user-20250620161509

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

@@ -1,18 +1,25 @@
 import type { EvalRow } from '@mastra/core/storage';
 import type { KVNamespace as KVNamespace_2 } from '@cloudflare/workers-types';
 import { KVNamespaceListKey as KVNamespaceListKey_2 } from '@cloudflare/workers-types';
+import type { MastraMessageContentV2 } from '@mastra/core/agent';
+import type { MastraMessageV1 } from '@mastra/core/memory';
+import type { MastraMessageV2 } from '@mastra/core/memory';
 import { MastraStorage } from '@mastra/core/storage';
-import type { MessageType } from '@mastra/core/memory';
-import type { MessageType as MessageType_2 } from '@mastra/core';
+import type { PaginationInfo } from '@mastra/core/storage';
 import type { StorageColumn } from '@mastra/core/storage';
 import type { StorageGetMessagesArg } from '@mastra/core/storage';
+import type { StorageGetTracesArg } from '@mastra/core/storage';
+import type { StorageResourceType } from '@mastra/core/storage';
 import type { StorageThreadType } from '@mastra/core/memory';
 import type { TABLE_EVALS } from '@mastra/core/storage';
 import type { TABLE_MESSAGES } from '@mastra/core/storage';
 import type { TABLE_NAMES } from '@mastra/core/storage';
+import type { TABLE_RESOURCES } 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 { Trace } from '@mastra/core/telemetry';
+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';
@@ -46,15 +53,26 @@ 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;
     private getOrCreateNamespaceId;
     private getNamespaceId;
+    private LEGACY_NAMESPACE_MAP;
+    /**
+     * There were a few legacy mappings for tables such as
+     * - messages -> threads
+     * - workflow_snapshot -> mastra_workflows
+     * - traces -> evals
+     * This has been updated to use dedicated namespaces for each table.
+     * In the case of data for a table existing in the legacy namespace, warn the user to migrate to the new namespace.
+     *
+     * @param tableName The table name to check for legacy data
+     * @param prefix The namespace prefix
+     * @returns The legacy namespace ID if data exists; otherwise, null
+     */
+    private checkLegacyNamespace;
     /**
      * Helper to safely serialize data for KV storage
      */
@@ -93,19 +111,28 @@ declare class CloudflareStore extends MastraStorage {
     private validateColumnValue;
     private validateAgainstSchema;
     private validateRecord;
-    private ensureDate;
-    private serializeDate;
     private ensureMetadata;
     createTable({ tableName, schema, }: {
         tableName: TABLE_NAMES;
         schema: Record<string, StorageColumn>;
     }): Promise<void>;
+    /**
+     * No-op: This backend is schemaless and does not require schema changes.
+     * @param tableName Name of the table
+     * @param schema Schema of the table
+     * @param ifNotExists Array of column names to add if they don't exist
+     */
+    alterTable(_args: {
+        tableName: TABLE_NAMES;
+        schema: Record<string, StorageColumn>;
+        ifNotExists: string[];
+    }): Promise<void>;
     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;
@@ -130,10 +157,20 @@ declare class CloudflareStore extends MastraStorage {
     }): Promise<void>;
     private getMessageKey;
     private getThreadMessagesKey;
-    saveMessages({ messages }: {
-        messages: MessageType[];
-    }): Promise<MessageType[]>;
-    getMessages<T extends MessageType = MessageType>({ threadId, selectBy }: StorageGetMessagesArg): Promise<T[]>;
+    saveMessages(args: {
+        messages: MastraMessageV1[];
+        format?: undefined | 'v1';
+    }): Promise<MastraMessageV1[]>;
+    saveMessages(args: {
+        messages: MastraMessageV2[];
+        format: 'v2';
+    }): Promise<MastraMessageV2[]>;
+    getMessages(args: StorageGetMessagesArg & {
+        format?: 'v1';
+    }): Promise<MastraMessageV1[]>;
+    getMessages(args: StorageGetMessagesArg & {
+        format: 'v2';
+    }): Promise<MastraMessageV2[]>;
     private validateWorkflowParams;
     private validateWorkflowState;
     private normalizeSteps;
@@ -153,24 +190,56 @@ declare class CloudflareStore extends MastraStorage {
         tableName: T;
         records: Partial<RecordTypes[T]>[];
     }): Promise<void>;
-    getTraces({ name, scope, page, perPage, attributes, }: {
+    getTraces({ name, scope, page, perPage, attributes, fromDate, toDate, }: {
         name?: string;
         scope?: string;
         page: number;
         perPage: number;
         attributes?: Record<string, string>;
+        fromDate?: Date;
+        toDate?: Date;
     }): 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>;
+    getTracesPaginated(_args: StorageGetTracesArg): Promise<PaginationInfo & {
+        traces: Trace[];
+    }>;
+    getThreadsByResourceIdPaginated(_args: {
+        resourceId: string;
+        page?: number;
+        perPage?: number;
+    }): Promise<PaginationInfo & {
+        threads: StorageThreadType[];
+    }>;
+    getMessagesPaginated(_args: StorageGetMessagesArg): Promise<PaginationInfo & {
+        messages: MastraMessageV1[] | MastraMessageV2[];
+    }>;
     close(): Promise<void>;
+    updateMessages(_args: {
+        messages: Partial<Omit<MastraMessageV2, 'createdAt'>> & {
+            id: string;
+            content?: {
+                metadata?: MastraMessageContentV2['metadata'];
+                content?: MastraMessageContentV2['content'];
+            };
+        }[];
+    }): Promise<MastraMessageV2[]>;
 }
 export { CloudflareStore }
 export { CloudflareStore as CloudflareStore_alias_1 }
@@ -192,19 +261,6 @@ export declare interface CloudflareWorkersConfig {
     keyPrefix?: string;
 }
 
-export declare const createSampleMessage: (threadId: string) => MessageType_2;
-
-export declare const createSampleThread: () => {
-    id: string;
-    resourceId: string;
-    title: string;
-    createdAt: Date;
-    updatedAt: Date;
-    metadata: {
-        key: string;
-    };
-};
-
 export declare const createSampleTrace: (name: string, scope?: string, attributes?: Record<string, string>) => {
     id: string;
     parentSpanId: string;
@@ -222,7 +278,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,12 +303,18 @@ export declare interface KVOperation {
     metadata?: any;
 }
 
+export declare type ListOptions = {
+    limit?: number;
+    prefix?: string;
+};
+
 export declare type RecordTypes = {
     [TABLE_THREADS]: StorageThreadType;
-    [TABLE_MESSAGES]: MessageType;
+    [TABLE_MESSAGES]: MastraMessageV2;
     [TABLE_WORKFLOW_SNAPSHOT]: WorkflowRunState;
     [TABLE_EVALS]: EvalRow;
     [TABLE_TRACES]: any;
+    [TABLE_RESOURCES]: StorageResourceType;
 };
 
 export declare const retryUntil: <T>(fn: () => Promise<T>, condition: (result: T) => boolean, timeout?: number, // REST API needs longer timeout due to higher latency

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

@@ -1,18 +1,25 @@
 import type { EvalRow } from '@mastra/core/storage';
 import type { KVNamespace as KVNamespace_2 } from '@cloudflare/workers-types';
 import { KVNamespaceListKey as KVNamespaceListKey_2 } from '@cloudflare/workers-types';
+import type { MastraMessageContentV2 } from '@mastra/core/agent';
+import type { MastraMessageV1 } from '@mastra/core/memory';
+import type { MastraMessageV2 } from '@mastra/core/memory';
 import { MastraStorage } from '@mastra/core/storage';
-import type { MessageType } from '@mastra/core/memory';
-import type { MessageType as MessageType_2 } from '@mastra/core';
+import type { PaginationInfo } from '@mastra/core/storage';
 import type { StorageColumn } from '@mastra/core/storage';
 import type { StorageGetMessagesArg } from '@mastra/core/storage';
+import type { StorageGetTracesArg } from '@mastra/core/storage';
+import type { StorageResourceType } from '@mastra/core/storage';
 import type { StorageThreadType } from '@mastra/core/memory';
 import type { TABLE_EVALS } from '@mastra/core/storage';
 import type { TABLE_MESSAGES } from '@mastra/core/storage';
 import type { TABLE_NAMES } from '@mastra/core/storage';
+import type { TABLE_RESOURCES } 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 { Trace } from '@mastra/core/telemetry';
+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';
@@ -46,15 +53,26 @@ 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;
     private getOrCreateNamespaceId;
     private getNamespaceId;
+    private LEGACY_NAMESPACE_MAP;
+    /**
+     * There were a few legacy mappings for tables such as
+     * - messages -> threads
+     * - workflow_snapshot -> mastra_workflows
+     * - traces -> evals
+     * This has been updated to use dedicated namespaces for each table.
+     * In the case of data for a table existing in the legacy namespace, warn the user to migrate to the new namespace.
+     *
+     * @param tableName The table name to check for legacy data
+     * @param prefix The namespace prefix
+     * @returns The legacy namespace ID if data exists; otherwise, null
+     */
+    private checkLegacyNamespace;
     /**
      * Helper to safely serialize data for KV storage
      */
@@ -93,19 +111,28 @@ declare class CloudflareStore extends MastraStorage {
     private validateColumnValue;
     private validateAgainstSchema;
     private validateRecord;
-    private ensureDate;
-    private serializeDate;
     private ensureMetadata;
     createTable({ tableName, schema, }: {
         tableName: TABLE_NAMES;
         schema: Record<string, StorageColumn>;
     }): Promise<void>;
+    /**
+     * No-op: This backend is schemaless and does not require schema changes.
+     * @param tableName Name of the table
+     * @param schema Schema of the table
+     * @param ifNotExists Array of column names to add if they don't exist
+     */
+    alterTable(_args: {
+        tableName: TABLE_NAMES;
+        schema: Record<string, StorageColumn>;
+        ifNotExists: string[];
+    }): Promise<void>;
     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;
@@ -130,10 +157,20 @@ declare class CloudflareStore extends MastraStorage {
     }): Promise<void>;
     private getMessageKey;
     private getThreadMessagesKey;
-    saveMessages({ messages }: {
-        messages: MessageType[];
-    }): Promise<MessageType[]>;
-    getMessages<T extends MessageType = MessageType>({ threadId, selectBy }: StorageGetMessagesArg): Promise<T[]>;
+    saveMessages(args: {
+        messages: MastraMessageV1[];
+        format?: undefined | 'v1';
+    }): Promise<MastraMessageV1[]>;
+    saveMessages(args: {
+        messages: MastraMessageV2[];
+        format: 'v2';
+    }): Promise<MastraMessageV2[]>;
+    getMessages(args: StorageGetMessagesArg & {
+        format?: 'v1';
+    }): Promise<MastraMessageV1[]>;
+    getMessages(args: StorageGetMessagesArg & {
+        format: 'v2';
+    }): Promise<MastraMessageV2[]>;
     private validateWorkflowParams;
     private validateWorkflowState;
     private normalizeSteps;
@@ -153,24 +190,56 @@ declare class CloudflareStore extends MastraStorage {
         tableName: T;
         records: Partial<RecordTypes[T]>[];
     }): Promise<void>;
-    getTraces({ name, scope, page, perPage, attributes, }: {
+    getTraces({ name, scope, page, perPage, attributes, fromDate, toDate, }: {
         name?: string;
         scope?: string;
         page: number;
         perPage: number;
         attributes?: Record<string, string>;
+        fromDate?: Date;
+        toDate?: Date;
     }): 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>;
+    getTracesPaginated(_args: StorageGetTracesArg): Promise<PaginationInfo & {
+        traces: Trace[];
+    }>;
+    getThreadsByResourceIdPaginated(_args: {
+        resourceId: string;
+        page?: number;
+        perPage?: number;
+    }): Promise<PaginationInfo & {
+        threads: StorageThreadType[];
+    }>;
+    getMessagesPaginated(_args: StorageGetMessagesArg): Promise<PaginationInfo & {
+        messages: MastraMessageV1[] | MastraMessageV2[];
+    }>;
     close(): Promise<void>;
+    updateMessages(_args: {
+        messages: Partial<Omit<MastraMessageV2, 'createdAt'>> & {
+            id: string;
+            content?: {
+                metadata?: MastraMessageContentV2['metadata'];
+                content?: MastraMessageContentV2['content'];
+            };
+        }[];
+    }): Promise<MastraMessageV2[]>;
 }
 export { CloudflareStore }
 export { CloudflareStore as CloudflareStore_alias_1 }
@@ -192,19 +261,6 @@ export declare interface CloudflareWorkersConfig {
     keyPrefix?: string;
 }
 
-export declare const createSampleMessage: (threadId: string) => MessageType_2;
-
-export declare const createSampleThread: () => {
-    id: string;
-    resourceId: string;
-    title: string;
-    createdAt: Date;
-    updatedAt: Date;
-    metadata: {
-        key: string;
-    };
-};
-
 export declare const createSampleTrace: (name: string, scope?: string, attributes?: Record<string, string>) => {
     id: string;
     parentSpanId: string;
@@ -222,7 +278,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,12 +303,18 @@ export declare interface KVOperation {
     metadata?: any;
 }
 
+export declare type ListOptions = {
+    limit?: number;
+    prefix?: string;
+};
+
 export declare type RecordTypes = {
     [TABLE_THREADS]: StorageThreadType;
-    [TABLE_MESSAGES]: MessageType;
+    [TABLE_MESSAGES]: MastraMessageV2;
     [TABLE_WORKFLOW_SNAPSHOT]: WorkflowRunState;
     [TABLE_EVALS]: EvalRow;
     [TABLE_TRACES]: any;
+    [TABLE_RESOURCES]: StorageResourceType;
 };
 
 export declare const retryUntil: <T>(fn: () => Promise<T>, condition: (result: T) => boolean, timeout?: number, // REST API needs longer timeout due to higher latency