@mastra/pg 0.17.4 → 0.17.5-alpha.0

package/CHANGELOG.md

@@ -1,5 +1,16 @@
 # @mastra/pg
 
+## 0.17.5-alpha.0
+
+### Patch Changes
+
+- Use the tz version of the timestamp column in when fetching messages. ([#8944](https://github.com/mastra-ai/mastra/pull/8944))
+
+- Avoid conflicts in pg fn naming when creating new tables on storage.init() ([#8946](https://github.com/mastra-ai/mastra/pull/8946))
+
+- Updated dependencies [[`c67ca32`](https://github.com/mastra-ai/mastra/commit/c67ca32e3c2cf69bfc146580770c720220ca44ac), [`dbc9e12`](https://github.com/mastra-ai/mastra/commit/dbc9e1216ba575ba59ead4afb727a01215f7de4f), [`99e41b9`](https://github.com/mastra-ai/mastra/commit/99e41b94957cdd25137d3ac12e94e8b21aa01b68), [`c28833c`](https://github.com/mastra-ai/mastra/commit/c28833c5b6d8e10eeffd7f7d39129d53b8bca240), [`f053e89`](https://github.com/mastra-ai/mastra/commit/f053e89160dbd0bd3333fc3492f68231b5c7c349), [`9a1a485`](https://github.com/mastra-ai/mastra/commit/9a1a4859b855e37239f652bf14b1ecd1029b8c4e), [`9257233`](https://github.com/mastra-ai/mastra/commit/9257233c4ffce09b2bedc2a9adbd70d7a83fa8e2), [`0f1a4c9`](https://github.com/mastra-ai/mastra/commit/0f1a4c984fb4b104b2f0b63ba18c9fa77f567700), [`2db6160`](https://github.com/mastra-ai/mastra/commit/2db6160e2022ff8827c15d30157e684683b934b5), [`8aeea37`](https://github.com/mastra-ai/mastra/commit/8aeea37efdde347c635a67fed56794943b7f74ec), [`02fe153`](https://github.com/mastra-ai/mastra/commit/02fe15351d6021d214da48ec982a0e9e4150bcee), [`74567b3`](https://github.com/mastra-ai/mastra/commit/74567b3d237ae3915cd0bca3cf55fa0a64e4e4a4), [`15a1733`](https://github.com/mastra-ai/mastra/commit/15a1733074cee8bd37370e1af34cd818e89fa7ac), [`fc2a774`](https://github.com/mastra-ai/mastra/commit/fc2a77468981aaddc3e77f83f0c4ad4a4af140da), [`4e08933`](https://github.com/mastra-ai/mastra/commit/4e08933625464dfde178347af5b6278fcf34188e)]:
+  - @mastra/core@0.21.2-alpha.0
+
 ## 0.17.4
 
 ### Patch Changes

package/dist/index.cjs

@@ -1529,6 +1529,20 @@ var MemoryPG = class extends storage.MemoryStorage {
     this.schema = schema;
     this.operations = operations;
   }
+  /**
+   * Normalizes message row from database by applying createdAtZ fallback
+   */
+  normalizeMessageRow(row) {
+    return {
+      id: row.id,
+      content: row.content,
+      role: row.role,
+      type: row.type,
+      createdAt: row.createdAtZ || row.createdAt,
+      threadId: row.threadId,
+      resourceId: row.resourceId
+    };
+  }
   async getThreadById({ threadId }) {
     try {
       const tableName = getTableName({ indexName: storage.TABLE_THREADS, schemaName: getSchemaName(this.schema) });
@@ -1796,11 +1810,12 @@ var MemoryPG = class extends storage.MemoryStorage {
                 WHERE thread_id = $${paramIdx}
               )
               SELECT
-                m.id, 
-                m.content, 
-                m.role, 
+                m.id,
+                m.content,
+                m.role,
                 m.type,
-                m."createdAt", 
+                m."createdAt",
+                m."createdAtZ",
                 m.thread_id AS "threadId",
                 m."resourceId"
               FROM ordered_messages m
@@ -1834,24 +1849,25 @@ var MemoryPG = class extends storage.MemoryStorage {
     return dedupedRows;
   }
   parseRow(row) {
-    let content = row.content;
+    const normalized = this.normalizeMessageRow(row);
+    let content = normalized.content;
     try {
-      content = JSON.parse(row.content);
+      content = JSON.parse(normalized.content);
     } catch {
     }
     return {
-      id: row.id,
+      id: normalized.id,
       content,
-      role: row.role,
-      createdAt: new Date(row.createdAt),
-      threadId: row.threadId,
-      resourceId: row.resourceId,
-      ...row.type && row.type !== "v2" ? { type: row.type } : {}
+      role: normalized.role,
+      createdAt: new Date(normalized.createdAt),
+      threadId: normalized.threadId,
+      resourceId: normalized.resourceId,
+      ...normalized.type && normalized.type !== "v2" ? { type: normalized.type } : {}
     };
   }
   async getMessages(args) {
     const { threadId, resourceId, format, selectBy } = args;
-    const selectStatement = `SELECT id, content, role, type, "createdAt", thread_id AS "threadId", "resourceId"`;
+    const selectStatement = `SELECT id, content, role, type, "createdAt", "createdAtZ", thread_id AS "threadId", "resourceId"`;
     const orderByStatement = `ORDER BY "createdAt" DESC`;
     const limit = storage.resolveMessageLimit({ last: selectBy?.last, defaultLimit: 40 });
     try {
@@ -1875,7 +1891,8 @@ var MemoryPG = class extends storage.MemoryStorage {
       const queryParams = [threadId, ...excludeIds, limit];
       const remainingRows = await this.client.manyOrNone(query, queryParams);
       rows.push(...remainingRows);
-      const fetchedMessages = (rows || []).map((message) => {
+      const fetchedMessages = (rows || []).map((row) => {
+        const message = this.normalizeMessageRow(row);
         if (typeof message.content === "string") {
           try {
             message.content = JSON.parse(message.content);
@@ -1914,7 +1931,7 @@ var MemoryPG = class extends storage.MemoryStorage {
     format
   }) {
     if (messageIds.length === 0) return [];
-    const selectStatement = `SELECT id, content, role, type, "createdAt", thread_id AS "threadId", "resourceId"`;
+    const selectStatement = `SELECT id, content, role, type, "createdAt", "createdAtZ", thread_id AS "threadId", "resourceId"`;
     try {
       const tableName = getTableName({ indexName: storage.TABLE_MESSAGES, schemaName: getSchemaName(this.schema) });
       const query = `
@@ -1923,7 +1940,10 @@ var MemoryPG = class extends storage.MemoryStorage {
         ORDER BY "createdAt" DESC
       `;
       const resultRows = await this.client.manyOrNone(query, messageIds);
-      const list = new agent.MessageList().add(resultRows.map(this.parseRow), "memory");
+      const list = new agent.MessageList().add(
+        resultRows.map((row) => this.parseRow(row)),
+        "memory"
+      );
       if (format === `v1`) return list.get.all.v1();
       return list.get.all.v2();
     } catch (error$1) {
@@ -1948,7 +1968,7 @@ var MemoryPG = class extends storage.MemoryStorage {
     const { page = 0, perPage: perPageInput, dateRange } = selectBy?.pagination || {};
     const fromDate = dateRange?.start;
     const toDate = dateRange?.end;
-    const selectStatement = `SELECT id, content, role, type, "createdAt", thread_id AS "threadId", "resourceId"`;
+    const selectStatement = `SELECT id, content, role, type, "createdAt", "createdAtZ", thread_id AS "threadId", "resourceId"`;
     const orderByStatement = `ORDER BY "createdAt" DESC`;
     const messages = [];
     try {
@@ -1992,7 +2012,8 @@ var MemoryPG = class extends storage.MemoryStorage {
       const dataQuery = `${selectStatement} FROM ${tableName} ${whereClause} ${excludeIds.length ? `AND id NOT IN (${excludeIdsParam})` : ""}${orderByStatement} LIMIT $${paramIndex++} OFFSET $${paramIndex++}`;
       const rows = await this.client.manyOrNone(dataQuery, [...queryParams, ...excludeIds, perPage, currentOffset]);
       messages.push(...rows || []);
-      const messagesWithParsedContent = messages.map((message) => {
+      const messagesWithParsedContent = messages.map((row) => {
+        const message = this.normalizeMessageRow(row);
         if (typeof message.content === "string") {
           try {
             return { ...message, content: JSON.parse(message.content) };
@@ -2137,7 +2158,7 @@ var MemoryPG = class extends storage.MemoryStorage {
       return [];
     }
     const messageIds = messages.map((m) => m.id);
-    const selectQuery = `SELECT id, content, role, type, "createdAt", thread_id AS "threadId", "resourceId" FROM ${getTableName({ indexName: storage.TABLE_MESSAGES, schemaName: getSchemaName(this.schema) })} WHERE id IN ($1:list)`;
+    const selectQuery = `SELECT id, content, role, type, "createdAt", "createdAtZ", thread_id AS "threadId", "resourceId" FROM ${getTableName({ indexName: storage.TABLE_MESSAGES, schemaName: getSchemaName(this.schema) })} WHERE id IN ($1:list)`;
     const existingMessagesDb = await this.client.manyOrNone(selectQuery, [messageIds]);
     if (existingMessagesDb.length === 0) {
       return [];
@@ -2212,10 +2233,11 @@ var MemoryPG = class extends storage.MemoryStorage {
       }
     });
     const updatedMessages = await this.client.manyOrNone(selectQuery, [messageIds]);
-    return (updatedMessages || []).map((message) => {
+    return (updatedMessages || []).map((row) => {
+      const message = this.normalizeMessageRow(row);
       if (typeof message.content === "string") {
         try {
-          message.content = JSON.parse(message.content);
+          return { ...message, content: JSON.parse(message.content) };
         } catch {
         }
       }
@@ -2861,11 +2883,13 @@ var StoreOperationsPG = class extends storage.StoreOperations {
    * Set up timestamp triggers for a table to automatically manage createdAt/updatedAt
    */
   async setupTimestampTriggers(tableName) {
-    const fullTableName = getTableName({ indexName: tableName, schemaName: getSchemaName(this.schemaName) });
+    const schemaName = getSchemaName(this.schemaName);
+    const fullTableName = getTableName({ indexName: tableName, schemaName });
+    const functionName = `${schemaName}.trigger_set_timestamps`;
     try {
       const triggerSQL = `
-        -- Create or replace the trigger function
-        CREATE OR REPLACE FUNCTION trigger_set_timestamps()
+        -- Create or replace the trigger function in the schema
+        CREATE OR REPLACE FUNCTION ${functionName}()
         RETURNS TRIGGER AS $$
         BEGIN
             IF TG_OP = 'INSERT' THEN
@@ -2891,7 +2915,7 @@ var StoreOperationsPG = class extends storage.StoreOperations {
         CREATE TRIGGER ${tableName}_timestamps
             BEFORE INSERT OR UPDATE ON ${fullTableName}
             FOR EACH ROW
-            EXECUTE FUNCTION trigger_set_timestamps();
+            EXECUTE FUNCTION ${functionName}();
       `;
       await this.client.none(triggerSQL);
       this.logger?.debug?.(`Set up timestamp triggers for table ${fullTableName}`);