@mastra/pg 0.10.1-alpha.1 → 0.10.1-alpha.2

package/.turbo/turbo-build.log

@@ -1,23 +1,23 @@
 
-> @mastra/pg@0.10.1-alpha.1 build /home/runner/work/mastra/mastra/stores/pg
+> @mastra/pg@0.10.1-alpha.2 build /home/runner/work/mastra/mastra/stores/pg
 > tsup src/index.ts --format esm,cjs --experimental-dts --clean --treeshake=smallest --splitting
 
 CLI Building entry: src/index.ts
 CLI Using tsconfig: tsconfig.json
 CLI tsup v8.4.0
 TSC Build start
-TSC ⚡️ Build success in 11415ms
+TSC ⚡️ Build success in 10689ms
 DTS Build start
 CLI Target: es2022
 Analysis will use the bundled TypeScript version 5.8.3
 Writing package typings: /home/runner/work/mastra/mastra/stores/pg/dist/_tsup-dts-rollup.d.ts
 Analysis will use the bundled TypeScript version 5.8.3
 Writing package typings: /home/runner/work/mastra/mastra/stores/pg/dist/_tsup-dts-rollup.d.cts
-DTS ⚡️ Build success in 11422ms
+DTS ⚡️ Build success in 11615ms
 CLI Cleaning output folder
 ESM Build start
 CJS Build start
-CJS dist/index.cjs 57.87 KB
-CJS ⚡️ Build success in 1565ms
-ESM dist/index.js 57.36 KB
-ESM ⚡️ Build success in 1566ms
+CJS dist/index.cjs 58.05 KB
+CJS ⚡️ Build success in 1169ms
+ESM dist/index.js 57.55 KB
+ESM ⚡️ Build success in 1169ms

package/CHANGELOG.md

@@ -1,5 +1,14 @@
 # @mastra/pg
 
+## 0.10.1-alpha.2
+
+### Patch Changes
+
+- c5bf1ce: Add backwards compat code for new MessageList in storage
+- Updated dependencies [c5bf1ce]
+- Updated dependencies [12b7002]
+  - @mastra/core@0.10.2-alpha.4
+
 ## 0.10.1-alpha.1
 
 ### Patch Changes

package/README.md

@@ -20,7 +20,7 @@ npm install @mastra/pg
 ```typescript
 import { PgVector } from '@mastra/pg';
 
-const vectorStore = new PgVector('postgresql://user:pass@localhost:5432/db');
+const vectorStore = new PgVector({ connectionString: 'postgresql://user:pass@localhost:5432/db' });
 
 // Create a new table with vector support
 await vectorStore.createIndex({

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

@@ -6,6 +6,7 @@ import type { DescribeIndexParams } from '@mastra/core/vector';
 import type { EvalRow } from '@mastra/core/storage';
 import type { IndexStats } from '@mastra/core/vector';
 import type { ISSLConfig } from 'pg-promise/typescript/pg-subset';
+import type { MastraMessageV1 } from '@mastra/core/memory';
 import type { MastraMessageV2 } from '@mastra/core/agent';
 import { MastraStorage } from '@mastra/core/storage';
 import { MastraVector } from '@mastra/core/vector';
@@ -320,8 +321,13 @@ declare class PostgresStore extends MastraStorage {
         threadId: string;
     }): Promise<void>;
     getMessages<T = unknown>({ threadId, selectBy }: StorageGetMessagesArg): Promise<T[]>;
-    saveMessages({ messages }: {
+    saveMessages(args: {
+        messages: MastraMessageV1[];
+        format?: undefined | 'v1';
+    }): Promise<MastraMessageV1[]>;
+    saveMessages(args: {
         messages: MastraMessageV2[];
+        format: 'v2';
     }): Promise<MastraMessageV2[]>;
     persistWorkflowSnapshot({ workflowName, runId, snapshot, }: {
         workflowName: string;

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

@@ -6,6 +6,7 @@ import type { DescribeIndexParams } from '@mastra/core/vector';
 import type { EvalRow } from '@mastra/core/storage';
 import type { IndexStats } from '@mastra/core/vector';
 import type { ISSLConfig } from 'pg-promise/typescript/pg-subset';
+import type { MastraMessageV1 } from '@mastra/core/memory';
 import type { MastraMessageV2 } from '@mastra/core/agent';
 import { MastraStorage } from '@mastra/core/storage';
 import { MastraVector } from '@mastra/core/vector';
@@ -320,8 +321,13 @@ declare class PostgresStore extends MastraStorage {
         threadId: string;
     }): Promise<void>;
     getMessages<T = unknown>({ threadId, selectBy }: StorageGetMessagesArg): Promise<T[]>;
-    saveMessages({ messages }: {
+    saveMessages(args: {
+        messages: MastraMessageV1[];
+        format?: undefined | 'v1';
+    }): Promise<MastraMessageV1[]>;
+    saveMessages(args: {
         messages: MastraMessageV2[];
+        format: 'v2';
     }): Promise<MastraMessageV2[]>;
     persistWorkflowSnapshot({ workflowName, runId, snapshot, }: {
         workflowName: string;

package/dist/index.cjs

@@ -6,6 +6,7 @@ var asyncMutex = require('async-mutex');
 var pg = require('pg');
 var xxhash = require('xxhash-wasm');
 var filter = require('@mastra/core/vector/filter');
+var agent = require('@mastra/core/agent');
 var storage = require('@mastra/core/storage');
 var pgPromise = require('pg-promise');
 
@@ -1418,7 +1419,10 @@ var PostgresStore = class extends storage.MastraStorage {
       throw error;
     }
   }
-  async saveMessages({ messages }) {
+  async saveMessages({
+    messages,
+    format
+  }) {
     if (messages.length === 0) return messages;
     try {
       const threadId = messages[0]?.threadId;
@@ -1445,7 +1449,9 @@ var PostgresStore = class extends storage.MastraStorage {
           );
         }
       });
-      return messages;
+      const list = new agent.MessageList().add(messages, "memory");
+      if (format === `v2`) return list.get.all.v2();
+      return list.get.all.v1();
     } catch (error) {
       console.error("Error saving messages:", error);
       throw error;

package/dist/index.js

@@ -4,6 +4,7 @@ import { Mutex } from 'async-mutex';
 import pg from 'pg';
 import xxhash from 'xxhash-wasm';
 import { BaseFilterTranslator } from '@mastra/core/vector/filter';
+import { MessageList } from '@mastra/core/agent';
 import { MastraStorage, TABLE_EVALS, TABLE_TRACES, TABLE_WORKFLOW_SNAPSHOT, TABLE_THREADS, TABLE_MESSAGES } from '@mastra/core/storage';
 import pgPromise from 'pg-promise';
 
@@ -1410,7 +1411,10 @@ var PostgresStore = class extends MastraStorage {
       throw error;
     }
   }
-  async saveMessages({ messages }) {
+  async saveMessages({
+    messages,
+    format
+  }) {
     if (messages.length === 0) return messages;
     try {
       const threadId = messages[0]?.threadId;
@@ -1437,7 +1441,9 @@ var PostgresStore = class extends MastraStorage {
           );
         }
       });
-      return messages;
+      const list = new MessageList().add(messages, "memory");
+      if (format === `v2`) return list.get.all.v2();
+      return list.get.all.v1();
     } catch (error) {
       console.error("Error saving messages:", error);
       throw error;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/pg",
-  "version": "0.10.1-alpha.1",
+  "version": "0.10.1-alpha.2",
   "description": "Postgres provider for Mastra - includes both vector and db storage capabilities",
   "type": "module",
   "main": "dist/index.js",
@@ -33,8 +33,8 @@
     "tsup": "^8.4.0",
     "typescript": "^5.8.2",
     "vitest": "^3.1.2",
-    "@mastra/core": "0.10.2-alpha.2",
-    "@internal/lint": "0.0.7"
+    "@internal/lint": "0.0.7",
+    "@mastra/core": "0.10.2-alpha.4"
   },
   "peerDependencies": {
     "@mastra/core": "^0.10.0-alpha.0"

package/src/storage/index.test.ts

@@ -1,6 +1,6 @@
 import { randomUUID } from 'crypto';
 import type { MetricResult } from '@mastra/core/eval';
-import type { MessageType } from '@mastra/core/memory';
+import type { MastraMessageV1 } from '@mastra/core/memory';
 import { TABLE_WORKFLOW_SNAPSHOT, TABLE_MESSAGES, TABLE_THREADS, TABLE_EVALS } from '@mastra/core/storage';
 import type { WorkflowRunState } from '@mastra/core/workflows';
 import pgPromise from 'pg-promise';
@@ -31,10 +31,16 @@ const createSampleThread = () => ({
   metadata: { key: 'value' },
 });
 
-const createSampleMessage = (threadId: string): MessageType => ({
+let role: 'user' | 'assistant' = 'assistant';
+const getRole = () => {
+  if (role === `user`) role = `assistant`;
+  else role = `user`;
+  return role;
+};
+const createSampleMessage = (threadId: string): MastraMessageV1 => ({
   id: `msg-${randomUUID()}`,
   resourceId: `resource-${randomUUID()}`,
-  role: 'user',
+  role: getRole(),
   type: 'text',
   threadId,
   content: [{ type: 'text', text: 'Hello' }],
@@ -58,6 +64,7 @@ const createSampleWorkflowSnapshot = (status: WorkflowRunState['context'][string
       },
       input: {},
     },
+    serializedStepGraph: [],
     activePaths: [],
     suspendedPaths: {},
     runId,
@@ -259,14 +266,17 @@ describe('PostgresStore', () => {
       await store.saveThread({ thread });
 
       const messages = [
-        { ...createSampleMessage(thread.id), content: [{ type: 'text', text: 'First' }] as MessageType['content'] },
-        { ...createSampleMessage(thread.id), content: [{ type: 'text', text: 'Second' }] as MessageType['content'] },
-        { ...createSampleMessage(thread.id), content: [{ type: 'text', text: 'Third' }] as MessageType['content'] },
-      ];
+        { ...createSampleMessage(thread.id), content: [{ type: 'text', text: 'First' }] },
+        {
+          ...createSampleMessage(thread.id),
+          content: [{ type: 'text', text: 'Second' }],
+        },
+        { ...createSampleMessage(thread.id), content: [{ type: 'text', text: 'Third' }] },
+      ] satisfies MastraMessageV1[];
 
       await store.saveMessages({ messages });
 
-      const retrievedMessages = await store.getMessages<MessageType>({ threadId: thread.id });
+      const retrievedMessages = await store.getMessages<MastraMessageV1>({ threadId: thread.id });
       expect(retrievedMessages).toHaveLength(3);
 
       // Verify order is maintained

package/src/storage/index.ts

@@ -1,6 +1,7 @@
+import { MessageList } from '@mastra/core/agent';
 import type { MastraMessageV2 } from '@mastra/core/agent';
 import type { MetricResult } from '@mastra/core/eval';
-import type { StorageThreadType } from '@mastra/core/memory';
+import type { MastraMessageV1, StorageThreadType } from '@mastra/core/memory';
 import {
   MastraStorage,
   TABLE_MESSAGES,
@@ -677,7 +678,14 @@ export class PostgresStore extends MastraStorage {
     }
   }
 
-  async saveMessages({ messages }: { messages: MastraMessageV2[] }): Promise<MastraMessageV2[]> {
+  async saveMessages(args: { messages: MastraMessageV1[]; format?: undefined | 'v1' }): Promise<MastraMessageV1[]>;
+  async saveMessages(args: { messages: MastraMessageV2[]; format: 'v2' }): Promise<MastraMessageV2[]>;
+  async saveMessages({
+    messages,
+    format,
+  }:
+    | { messages: MastraMessageV1[]; format?: undefined | 'v1' }
+    | { messages: MastraMessageV2[]; format: 'v2' }): Promise<MastraMessageV2[] | MastraMessageV1[]> {
     if (messages.length === 0) return messages;
 
     try {
@@ -709,7 +717,9 @@ export class PostgresStore extends MastraStorage {
         }
       });
 
-      return messages;
+      const list = new MessageList().add(messages, 'memory');
+      if (format === `v2`) return list.get.all.v2();
+      return list.get.all.v1();
     } catch (error) {
       console.error('Error saving messages:', error);
       throw error;