Feat/storage flavour durable objects

packages/core/src/storage/base.ts

@@ -164,6 +164,8 @@ export abstract class MastraStorage extends MastraBase {
         id: { type: 'text', nullable: false, primaryKey: true },
         thread_id: { type: 'text', nullable: false },
         content: { type: 'text', nullable: false },
+        role: { type: 'text', nullable: false },
+        type: { type: 'text', nullable: false },
         createdAt: { type: 'timestamp', nullable: false },
       },
     });

packages/core/src/storage/libsql.test.ts

@@ -30,21 +30,29 @@ describe('MastraStorageLibSql', () => {
       type: 'text',
       threadId,
       content: [{ type: 'text', text: 'Hello' }],
-      created_at: new Date().toISOString(),
+      createdAt: new Date().toISOString(),
     }) as any;
 
   beforeAll(async () => {
     await storage.init();
   });
 
-  afterAll(async () => {
+  beforeEach(async () => {
     // Clear tables before each test
     await storage.clearTable({ tableName: 'workflow_snapshot' });
     await storage.clearTable({ tableName: 'evals' });
     await storage.clearTable({ tableName: 'messages' });
     await storage.clearTable({ tableName: 'threads' });
   });
 
+  afterAll(async () => {
+    // Clear tables after tests
+    await storage.clearTable({ tableName: 'workflow_snapshot' });
+    await storage.clearTable({ tableName: 'evals' });
+    await storage.clearTable({ tableName: 'messages' });
+    await storage.clearTable({ tableName: 'threads' });
+  });
+
   describe('Thread Operations', () => {
     it('should create and retrieve a thread', async () => {
       const thread = createSampleThread();

packages/core/src/storage/libsql.ts

@@ -237,9 +237,9 @@ export class MastraStorageLibSql extends MastraStorage {
 
       for (const message of messages) {
         await tx.execute({
-          sql: `INSERT INTO ${MastraStorage.TABLE_MESSAGES} (id, thread_id, content, createdAt) 
-                              VALUES (?, ?, ?, ?)`,
-          args: [message.id, threadId, JSON.stringify(message), message.createdAt || new Date().toISOString()],
+          sql: `INSERT INTO ${MastraStorage.TABLE_MESSAGES} (id, thread_id, content, role, type, createdAt) 
+                              VALUES (?, ?, ?, ?, ?, ?)`,
+          args: [message.id, threadId, JSON.stringify(message), message.role, message.type, message.createdAt || new Date().toISOString()],
         });
       }
 

packages/core/src/storage/types.ts

@@ -1,9 +1,13 @@
 import { WorkflowRunState } from '../workflows';
 
 export interface StorageColumn {
-  type: 'text' | 'timestamp';
+  type: 'text' | 'timestamp' | 'uuid' | 'jsonb' | 'integer';
   primaryKey?: boolean;
   nullable?: boolean;
+  references?: {
+    table: string;
+    column: string;
+  };
 }
 
 export interface WorkflowRow {

storage/durable-object/package.json

@@ -0,0 +1,38 @@
+{
+  "name": "@mastra/storage-durable-object",
+  "version": "0.0.0-alpha.0",
+  "description": "Cloudflare Durable Objects storage provider for Mastra",
+  "type": "module",
+  "main": "dist/index.js",
+  "module": "dist/storage-durable-object.esm.js",
+  "types": "dist/index.d.ts",
+  "exports": {
+    ".": {
+      "import": {
+        "types": "./dist/index.d.ts",
+        "default": "./dist/storage-durable-object.esm.js"
+      },
+      "require": {
+        "types": "./dist/index.d.ts",
+        "default": "./dist/index.js"
+      }
+    },
+    "./package.json": "./package.json"
+  },
+  "scripts": {
+    "build": "dts build",
+    "build:dev": "dts watch",
+    "test": "vitest run"
+  },
+  "dependencies": {
+    "@mastra/core": "workspace:*"
+  },
+  "devDependencies": {
+    "@cloudflare/workers-types": "^4.20250124.3",
+    "@tsconfig/recommended": "^1.0.7",
+    "@types/node": "^22.9.0",
+    "dts-cli": "^2.0.5",
+    "miniflare": "^3.20240117.0",
+    "vitest": "^2.1.8"
+  }
+}