@mastra/pg 0.17.4 → 0.17.5-alpha.0

package/dist/index.js

@@ -1503,6 +1503,20 @@ var MemoryPG = class extends 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: TABLE_THREADS, schemaName: getSchemaName(this.schema) });
@@ -1770,11 +1784,12 @@ var MemoryPG = class extends 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
@@ -1808,24 +1823,25 @@ var MemoryPG = class extends 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 = resolveMessageLimit({ last: selectBy?.last, defaultLimit: 40 });
     try {
@@ -1849,7 +1865,8 @@ var MemoryPG = class extends 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);
@@ -1888,7 +1905,7 @@ var MemoryPG = class extends 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: TABLE_MESSAGES, schemaName: getSchemaName(this.schema) });
       const query = `
@@ -1897,7 +1914,10 @@ var MemoryPG = class extends MemoryStorage {
         ORDER BY "createdAt" DESC
       `;
       const resultRows = await this.client.manyOrNone(query, messageIds);
-      const list = new MessageList().add(resultRows.map(this.parseRow), "memory");
+      const list = new MessageList().add(
+        resultRows.map((row) => this.parseRow(row)),
+        "memory"
+      );
       if (format === `v1`) return list.get.all.v1();
       return list.get.all.v2();
     } catch (error) {
@@ -1922,7 +1942,7 @@ var MemoryPG = class extends 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 {
@@ -1966,7 +1986,8 @@ var MemoryPG = class extends 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) };
@@ -2111,7 +2132,7 @@ var MemoryPG = class extends 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: 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: TABLE_MESSAGES, schemaName: getSchemaName(this.schema) })} WHERE id IN ($1:list)`;
     const existingMessagesDb = await this.client.manyOrNone(selectQuery, [messageIds]);
     if (existingMessagesDb.length === 0) {
       return [];
@@ -2186,10 +2207,11 @@ var MemoryPG = class extends 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 {
         }
       }
@@ -2835,11 +2857,13 @@ var StoreOperationsPG = class extends 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
@@ -2865,7 +2889,7 @@ var StoreOperationsPG = class extends 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}`);