@mastra/cloudflare-d1 0.0.0-taofeeqInngest-20250603090617 → 0.0.0-tool-call-parts-20250630193309

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

@@ -1,43 +1,18 @@
 import type { D1Database as D1Database_2 } from '@cloudflare/workers-types';
 import type { EvalRow } from '@mastra/core/storage';
+import type { MastraMessageContentV2 } from '@mastra/core/agent';
 import type { MastraMessageV1 } from '@mastra/core/memory';
 import type { MastraMessageV2 } from '@mastra/core/memory';
-import type { MastraMessageV2 as MastraMessageV2_2 } from '@mastra/core';
 import { MastraStorage } from '@mastra/core/storage';
+import type { PaginationInfo } from '@mastra/core/storage';
 import type { StorageColumn } from '@mastra/core/storage';
 import type { StorageGetMessagesArg } from '@mastra/core/storage';
 import type { StorageThreadType } from '@mastra/core/memory';
 import type { TABLE_NAMES } 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';
-
-export declare const checkWorkflowSnapshot: (snapshot: WorkflowRunState_2 | string, stepId: string, status: string) => void;
-
-export declare const createSampleMessage: (threadId: string, parts?: MastraMessageV2_2["content"]["parts"]) => MastraMessageV2_2;
-
-export declare const createSampleThread: () => {
-    id: string;
-    resourceId: string;
-    title: string;
-    createdAt: Date;
-    updatedAt: Date;
-    metadata: {
-        key: string;
-    };
-};
-
-export declare const createSampleThreadWithParams: (threadId: string, resourceId: string, createdAt: Date, updatedAt: Date) => {
-    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;
@@ -56,12 +31,6 @@ export declare const createSampleTrace: (name: string, scope?: string, attribute
     createdAt: string;
 };
 
-export declare const createSampleWorkflowSnapshot: (threadId: string, status: string, createdAt?: Date) => {
-    snapshot: WorkflowRunState_2;
-    runId: string;
-    stepId: string;
-};
-
 export declare function createSqlBuilder(): SqlBuilder;
 
 /**
@@ -101,15 +70,25 @@ declare class D1Store extends MastraStorage {
      * @returns Query results as an array or a single object if first=true
      */
     private executeQuery;
-    private getSqlType;
-    private ensureDate;
-    private serializeDate;
+    private getTableColumns;
     private serializeValue;
     private deserializeValue;
+    protected getSqlType(type: StorageColumn['type']): string;
     createTable({ tableName, schema, }: {
         tableName: TABLE_NAMES;
         schema: Record<string, StorageColumn>;
     }): Promise<void>;
+    /**
+     * Alters table schema to add columns if they don't exist
+     * @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({ tableName, schema, ifNotExists, }: {
+        tableName: TABLE_NAMES;
+        schema: Record<string, StorageColumn>;
+        ifNotExists: string[];
+    }): Promise<void>;
     clearTable({ tableName }: {
         tableName: TABLE_NAMES;
     }): Promise<void>;
@@ -125,9 +104,19 @@ declare class D1Store extends MastraStorage {
     getThreadById({ threadId }: {
         threadId: string;
     }): Promise<StorageThreadType | null>;
+    /**
+     * @deprecated use getThreadsByResourceIdPaginated instead
+     */
     getThreadsByResourceId({ resourceId }: {
         resourceId: string;
     }): Promise<StorageThreadType[]>;
+    getThreadsByResourceIdPaginated(args: {
+        resourceId: string;
+        page: number;
+        perPage: number;
+    }): Promise<PaginationInfo & {
+        threads: StorageThreadType[];
+    }>;
     saveThread({ thread }: {
         thread: StorageThreadType;
     }): Promise<StorageThreadType>;
@@ -147,12 +136,21 @@ declare class D1Store extends MastraStorage {
         messages: MastraMessageV2[];
         format: 'v2';
     }): Promise<MastraMessageV2[]>;
+    private _getIncludedMessages;
+    /**
+     * @deprecated use getMessagesPaginated instead
+     */
     getMessages(args: StorageGetMessagesArg & {
         format?: 'v1';
     }): Promise<MastraMessageV1[]>;
     getMessages(args: StorageGetMessagesArg & {
         format: 'v2';
     }): Promise<MastraMessageV2[]>;
+    getMessagesPaginated({ threadId, selectBy, format, }: StorageGetMessagesArg & {
+        format?: 'v1' | 'v2';
+    }): Promise<PaginationInfo & {
+        messages: MastraMessageV1[] | MastraMessageV2[];
+    }>;
     persistWorkflowSnapshot({ workflowName, runId, snapshot, }: {
         workflowName: string;
         runId: string;
@@ -171,6 +169,15 @@ declare class D1Store extends MastraStorage {
         tableName: TABLE_NAMES;
         records: Record<string, any>[];
     }): Promise<void>;
+    /**
+     * Upsert multiple records in a batch operation
+     * @param tableName The table to insert into
+     * @param records The records to insert
+     */
+    private batchUpsert;
+    /**
+     * @deprecated use getTracesPaginated instead
+     */
     getTraces({ name, scope, page, perPage, attributes, fromDate, toDate, }: {
         name?: string;
         scope?: string;
@@ -179,8 +186,32 @@ declare class D1Store extends MastraStorage {
         attributes?: Record<string, string>;
         fromDate?: Date;
         toDate?: Date;
-    }): Promise<Record<string, any>[]>;
+    }): Promise<Trace[]>;
+    getTracesPaginated(args: {
+        name?: string;
+        scope?: string;
+        attributes?: Record<string, string>;
+        page: number;
+        perPage: number;
+        fromDate?: Date;
+        toDate?: Date;
+    }): Promise<PaginationInfo & {
+        traces: Trace[];
+    }>;
+    /**
+     * @deprecated use getEvals instead
+     */
     getEvalsByAgentName(agentName: string, type?: 'test' | 'live'): Promise<EvalRow[]>;
+    getEvals(options?: {
+        agentName?: string;
+        type?: 'test' | 'live';
+        page?: number;
+        perPage?: number;
+        fromDate?: Date;
+        toDate?: Date;
+    }): Promise<PaginationInfo & {
+        evals: EvalRow[];
+    }>;
     private parseWorkflowRun;
     private hasColumn;
     getWorkflowRuns({ workflowName, fromDate, toDate, limit, offset, resourceId, }?: {
@@ -200,6 +231,15 @@ declare class D1Store extends MastraStorage {
      * No explicit cleanup needed for D1 in either REST or Workers Binding mode
      */
     close(): Promise<void>;
+    updateMessages(_args: {
+        messages: Partial<Omit<MastraMessageV2, 'createdAt'>> & {
+            id: string;
+            content?: {
+                metadata?: MastraMessageContentV2['metadata'];
+                content?: MastraMessageContentV2['content'];
+            };
+        }[];
+    }): Promise<MastraMessageV2[]>;
 }
 export { D1Store }
 export { D1Store as D1Store_alias_1 }

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

@@ -1,43 +1,18 @@
 import type { D1Database as D1Database_2 } from '@cloudflare/workers-types';
 import type { EvalRow } from '@mastra/core/storage';
+import type { MastraMessageContentV2 } from '@mastra/core/agent';
 import type { MastraMessageV1 } from '@mastra/core/memory';
 import type { MastraMessageV2 } from '@mastra/core/memory';
-import type { MastraMessageV2 as MastraMessageV2_2 } from '@mastra/core';
 import { MastraStorage } from '@mastra/core/storage';
+import type { PaginationInfo } from '@mastra/core/storage';
 import type { StorageColumn } from '@mastra/core/storage';
 import type { StorageGetMessagesArg } from '@mastra/core/storage';
 import type { StorageThreadType } from '@mastra/core/memory';
 import type { TABLE_NAMES } 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';
-
-export declare const checkWorkflowSnapshot: (snapshot: WorkflowRunState_2 | string, stepId: string, status: string) => void;
-
-export declare const createSampleMessage: (threadId: string, parts?: MastraMessageV2_2["content"]["parts"]) => MastraMessageV2_2;
-
-export declare const createSampleThread: () => {
-    id: string;
-    resourceId: string;
-    title: string;
-    createdAt: Date;
-    updatedAt: Date;
-    metadata: {
-        key: string;
-    };
-};
-
-export declare const createSampleThreadWithParams: (threadId: string, resourceId: string, createdAt: Date, updatedAt: Date) => {
-    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;
@@ -56,12 +31,6 @@ export declare const createSampleTrace: (name: string, scope?: string, attribute
     createdAt: string;
 };
 
-export declare const createSampleWorkflowSnapshot: (threadId: string, status: string, createdAt?: Date) => {
-    snapshot: WorkflowRunState_2;
-    runId: string;
-    stepId: string;
-};
-
 export declare function createSqlBuilder(): SqlBuilder;
 
 /**
@@ -101,15 +70,25 @@ declare class D1Store extends MastraStorage {
      * @returns Query results as an array or a single object if first=true
      */
     private executeQuery;
-    private getSqlType;
-    private ensureDate;
-    private serializeDate;
+    private getTableColumns;
     private serializeValue;
     private deserializeValue;
+    protected getSqlType(type: StorageColumn['type']): string;
     createTable({ tableName, schema, }: {
         tableName: TABLE_NAMES;
         schema: Record<string, StorageColumn>;
     }): Promise<void>;
+    /**
+     * Alters table schema to add columns if they don't exist
+     * @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({ tableName, schema, ifNotExists, }: {
+        tableName: TABLE_NAMES;
+        schema: Record<string, StorageColumn>;
+        ifNotExists: string[];
+    }): Promise<void>;
     clearTable({ tableName }: {
         tableName: TABLE_NAMES;
     }): Promise<void>;
@@ -125,9 +104,19 @@ declare class D1Store extends MastraStorage {
     getThreadById({ threadId }: {
         threadId: string;
     }): Promise<StorageThreadType | null>;
+    /**
+     * @deprecated use getThreadsByResourceIdPaginated instead
+     */
     getThreadsByResourceId({ resourceId }: {
         resourceId: string;
     }): Promise<StorageThreadType[]>;
+    getThreadsByResourceIdPaginated(args: {
+        resourceId: string;
+        page: number;
+        perPage: number;
+    }): Promise<PaginationInfo & {
+        threads: StorageThreadType[];
+    }>;
     saveThread({ thread }: {
         thread: StorageThreadType;
     }): Promise<StorageThreadType>;
@@ -147,12 +136,21 @@ declare class D1Store extends MastraStorage {
         messages: MastraMessageV2[];
         format: 'v2';
     }): Promise<MastraMessageV2[]>;
+    private _getIncludedMessages;
+    /**
+     * @deprecated use getMessagesPaginated instead
+     */
     getMessages(args: StorageGetMessagesArg & {
         format?: 'v1';
     }): Promise<MastraMessageV1[]>;
     getMessages(args: StorageGetMessagesArg & {
         format: 'v2';
     }): Promise<MastraMessageV2[]>;
+    getMessagesPaginated({ threadId, selectBy, format, }: StorageGetMessagesArg & {
+        format?: 'v1' | 'v2';
+    }): Promise<PaginationInfo & {
+        messages: MastraMessageV1[] | MastraMessageV2[];
+    }>;
     persistWorkflowSnapshot({ workflowName, runId, snapshot, }: {
         workflowName: string;
         runId: string;
@@ -171,6 +169,15 @@ declare class D1Store extends MastraStorage {
         tableName: TABLE_NAMES;
         records: Record<string, any>[];
     }): Promise<void>;
+    /**
+     * Upsert multiple records in a batch operation
+     * @param tableName The table to insert into
+     * @param records The records to insert
+     */
+    private batchUpsert;
+    /**
+     * @deprecated use getTracesPaginated instead
+     */
     getTraces({ name, scope, page, perPage, attributes, fromDate, toDate, }: {
         name?: string;
         scope?: string;
@@ -179,8 +186,32 @@ declare class D1Store extends MastraStorage {
         attributes?: Record<string, string>;
         fromDate?: Date;
         toDate?: Date;
-    }): Promise<Record<string, any>[]>;
+    }): Promise<Trace[]>;
+    getTracesPaginated(args: {
+        name?: string;
+        scope?: string;
+        attributes?: Record<string, string>;
+        page: number;
+        perPage: number;
+        fromDate?: Date;
+        toDate?: Date;
+    }): Promise<PaginationInfo & {
+        traces: Trace[];
+    }>;
+    /**
+     * @deprecated use getEvals instead
+     */
     getEvalsByAgentName(agentName: string, type?: 'test' | 'live'): Promise<EvalRow[]>;
+    getEvals(options?: {
+        agentName?: string;
+        type?: 'test' | 'live';
+        page?: number;
+        perPage?: number;
+        fromDate?: Date;
+        toDate?: Date;
+    }): Promise<PaginationInfo & {
+        evals: EvalRow[];
+    }>;
     private parseWorkflowRun;
     private hasColumn;
     getWorkflowRuns({ workflowName, fromDate, toDate, limit, offset, resourceId, }?: {
@@ -200,6 +231,15 @@ declare class D1Store extends MastraStorage {
      * No explicit cleanup needed for D1 in either REST or Workers Binding mode
      */
     close(): Promise<void>;
+    updateMessages(_args: {
+        messages: Partial<Omit<MastraMessageV2, 'createdAt'>> & {
+            id: string;
+            content?: {
+                metadata?: MastraMessageContentV2['metadata'];
+                content?: MastraMessageContentV2['content'];
+            };
+        }[];
+    }): Promise<MastraMessageV2[]>;
 }
 export { D1Store }
 export { D1Store as D1Store_alias_1 }