@mastra/pg 1.8.1 → 1.8.2

package/dist/index.js

@@ -7819,7 +7819,7 @@ var MemoryPG = class _MemoryPG extends MemoryStorage {
     try {
       const tableName = getTableName3({ indexName: TABLE_MESSAGES, schemaName: getSchemaName3(this.#schema) });
       await this.#db.client.tx(async (t) => {
-        const messageInserts = messages.map((message) => {
+        for (const message of messages) {
           if (!message.threadId) {
             throw new Error(
               `Expected to find a threadId for message, but couldn't find one. An unexpected error has occurred.`
@@ -7830,7 +7830,7 @@ var MemoryPG = class _MemoryPG extends MemoryStorage {
               `Expected to find a resourceId for message, but couldn't find one. An unexpected error has occurred.`
             );
           }
-          return t.none(
+          await t.none(
             `INSERT INTO ${tableName} (id, thread_id, content, "createdAt", "createdAtZ", role, type, "resourceId")
              VALUES ($1, $2, $3, $4, $5, $6, $7, $8)
              ON CONFLICT (id) DO UPDATE SET
@@ -7850,19 +7850,17 @@ var MemoryPG = class _MemoryPG extends MemoryStorage {
               message.resourceId
             ]
           );
-        });
+        }
         const threadTableName = getTableName3({ indexName: TABLE_THREADS, schemaName: getSchemaName3(this.#schema) });
         const nowStr = (/* @__PURE__ */ new Date()).toISOString();
-        const threadUpdate = t.none(
+        await t.none(
           `UPDATE ${threadTableName}
-                        SET
-                            "updatedAt" = $1,
-                            "updatedAtZ" = $2
-                        WHERE id = $3
-                    `,
+            SET
+              "updatedAt" = $1,
+              "updatedAtZ" = $2
+            WHERE id = $3`,
           [nowStr, nowStr, threadId]
         );
-        await Promise.all([...messageInserts, threadUpdate]);
       });
       const messagesWithParsedContent = messages.map((message) => {
         if (typeof message.content === "string") {
@@ -7999,10 +7997,10 @@ var MemoryPG = class _MemoryPG extends MemoryStorage {
         const threadIds = messages?.map((msg) => msg.thread_id).filter(Boolean) || [];
         await t.none(`DELETE FROM ${messageTableName} WHERE id IN (${placeholders})`, messageIds);
         if (threadIds.length > 0) {
-          const updatePromises = threadIds.map(
-            (threadId) => t.none(`UPDATE ${threadTableName} SET "updatedAt" = NOW(), "updatedAtZ" = NOW() WHERE id = $1`, [threadId])
+          await t.none(
+            `UPDATE ${threadTableName} SET "updatedAt" = NOW(), "updatedAtZ" = NOW() WHERE id IN (${inPlaceholders(threadIds.length)})`,
+            threadIds
           );
-          await Promise.all(updatePromises);
         }
       });
     } catch (error) {