@mastra/pg 1.9.0-alpha.0 → 1.9.1-alpha.0

package/CHANGELOG.md

@@ -1,5 +1,28 @@
 # @mastra/pg
 
+## 1.9.1-alpha.0
+
+### Patch Changes
+
+- Fixed `batchInsert` and `batchUpdate` in `@mastra/pg` to run on a single Postgres transaction connection. ([#15312](https://github.com/mastra-ai/mastra/pull/15312))
+
+  This prevents pooled `BEGIN`/`COMMIT`/`ROLLBACK` calls from landing on different connections and leaving idle transactions open during batch writes.
+
+- Fixed "column does not exist" errors when using experiment review features on databases created before the review pipeline was introduced. Startup now automatically migrates older experiment tables to the latest schema. ([#15304](https://github.com/mastra-ai/mastra/pull/15304))
+
+## 1.9.0
+
+### Minor Changes
+
+- Implemented `updateObservationalMemoryConfig()` in Postgres, LibSQL, and MongoDB storage adapters. This enables per-record config overrides for observational memory thresholds, supporting the new `memory.updateObservationalMemoryConfig()` API in `@mastra/memory`. ([#15115](https://github.com/mastra-ai/mastra/pull/15115))
+
+### Patch Changes
+
+- Fixed observational memory date-range queries to use ISO-string timestamp columns (`createdAtZ`) instead of the raw `createdAt` column, ensuring correct filtering when querying observations by date range. ([#15115](https://github.com/mastra-ai/mastra/pull/15115))
+
+- Updated dependencies [[`f32b9e1`](https://github.com/mastra-ai/mastra/commit/f32b9e115a3c754d1c8cfa3f4256fba87b09cfb7), [`7d6f521`](https://github.com/mastra-ai/mastra/commit/7d6f52164d0cca099f0b07cb2bba334360f1c8ab), [`a50d220`](https://github.com/mastra-ai/mastra/commit/a50d220b01ecbc5644d489a3d446c3bd4ab30245), [`665477b`](https://github.com/mastra-ai/mastra/commit/665477bc104fd52cfef8e7610d7664781a70c220), [`4cc2755`](https://github.com/mastra-ai/mastra/commit/4cc2755a7194cb08720ff2ab4dffb4b4a5103dfd), [`ac7baf6`](https://github.com/mastra-ai/mastra/commit/ac7baf66ef1db15e03975ef4ebb02724f015a391), [`ed425d7`](https://github.com/mastra-ai/mastra/commit/ed425d78e7c66cbda8209fee910856f98c6c6b82), [`1371703`](https://github.com/mastra-ai/mastra/commit/1371703835080450ef3f9aea58059a95d0da2e5a), [`0df8321`](https://github.com/mastra-ai/mastra/commit/0df832196eeb2450ab77ce887e8553abdd44c5a6), [`98f8a8b`](https://github.com/mastra-ai/mastra/commit/98f8a8bdf5761b9982f3ad3acbe7f1cc3efa71f3), [`ba6f7e9`](https://github.com/mastra-ai/mastra/commit/ba6f7e9086d8281393f2acae60fda61de3bff1f9), [`7eb2596`](https://github.com/mastra-ai/mastra/commit/7eb25960d607e07468c9a10c5437abd2deaf1e9a), [`1805ddc`](https://github.com/mastra-ai/mastra/commit/1805ddc9c9b3b14b63749735a13c05a45af43a80), [`fff91cf`](https://github.com/mastra-ai/mastra/commit/fff91cf914de0e731578aacebffdeebef82f0440), [`61109b3`](https://github.com/mastra-ai/mastra/commit/61109b34feb0e38d54bee4b8ca83eb7345b1d557), [`33f1ead`](https://github.com/mastra-ai/mastra/commit/33f1eadfa19c86953f593478e5fa371093b33779)]:
+  - @mastra/core@1.23.0
+
 ## 1.9.0-alpha.0
 
 ### Minor Changes

package/dist/docs/SKILL.md

@@ -3,7 +3,7 @@ name: mastra-pg
 description: Documentation for @mastra/pg. Use when working with @mastra/pg APIs, configuration, or implementation.
 metadata:
   package: "@mastra/pg"
-  version: "1.9.0-alpha.0"
+  version: "1.9.1-alpha.0"
 ---
 
 ## When to use

package/dist/docs/assets/SOURCE_MAP.json

@@ -1,5 +1,5 @@
 {
-  "version": "1.9.0-alpha.0",
+  "version": "1.9.1-alpha.0",
   "package": "@mastra/pg",
   "exports": {},
   "modules": {}

package/dist/docs/references/docs-memory-storage.md

@@ -118,8 +118,6 @@ export const agent = new Agent({
 })
 ```
 
-> **Warning:** Agent-level storage isn't supported when using [Mastra Cloud Store](https://mastra.ai/docs/mastra-cloud/deployment). If you use Mastra Cloud Store, configure storage on the Mastra instance instead. This limitation doesn't apply if you bring your own database.
-
 ## Threads and resources
 
 Mastra organizes conversations using two identifiers:

package/dist/docs/references/docs-memory-working-memory.md

@@ -397,4 +397,4 @@ const response = await agent.generate('What do you know about me?', {
 
 - [Working memory with template](https://github.com/mastra-ai/mastra/tree/main/examples/memory-with-template)
 - [Working memory with schema](https://github.com/mastra-ai/mastra/tree/main/examples/memory-with-schema)
-- [Per-resource working memory](https://github.com/mastra-ai/mastra/tree/main/examples/memory-per-resource-example) - Complete example showing resource-scoped memory persistence
+- [Per-resource working memory](https://github.com/mastra-ai/mastra/tree/main/examples/memory-per-resource-example): Complete example showing resource-scoped memory persistence

package/dist/index.cjs

@@ -2499,36 +2499,39 @@ var PgDB = class extends base.MastraBase {
         return storage.getDefaultValue(type);
     }
   }
-  async insert({ tableName, record }) {
-    try {
-      this.addTimestampZColumns(record);
-      const filteredRecord = await this.filterRecordToKnownColumns(tableName, record);
-      const schemaName = getSchemaName(this.schemaName);
-      const columns = Object.keys(filteredRecord).map((col) => utils.parseSqlIdentifier(col, "column name"));
-      if (columns.length === 0) return;
-      const values = this.prepareValuesForInsert(filteredRecord, tableName);
-      const placeholders = values.map((_, i) => `$${i + 1}`).join(", ");
-      const fullTableName = getTableName({ indexName: tableName, schemaName });
-      const columnList = columns.map((c) => `"${c}"`).join(", ");
-      if (tableName === storage.TABLE_SPANS) {
-        const updateColumns = columns.filter((c) => c !== "traceId" && c !== "spanId");
-        if (updateColumns.length > 0) {
-          const updateClause = updateColumns.map((c) => `"${c}" = EXCLUDED."${c}"`).join(", ");
-          await this.client.none(
-            `INSERT INTO ${fullTableName} (${columnList}) VALUES (${placeholders})
+  async executeInsert(client, { tableName, record }) {
+    this.addTimestampZColumns(record);
+    const filteredRecord = await this.filterRecordToKnownColumns(tableName, record);
+    const schemaName = getSchemaName(this.schemaName);
+    const columns = Object.keys(filteredRecord).map((col) => utils.parseSqlIdentifier(col, "column name"));
+    if (columns.length === 0) return;
+    const values = this.prepareValuesForInsert(filteredRecord, tableName);
+    const placeholders = values.map((_, i) => `$${i + 1}`).join(", ");
+    const fullTableName = getTableName({ indexName: tableName, schemaName });
+    const columnList = columns.map((c) => `"${c}"`).join(", ");
+    if (tableName === storage.TABLE_SPANS) {
+      const updateColumns = columns.filter((c) => c !== "traceId" && c !== "spanId");
+      if (updateColumns.length > 0) {
+        const updateClause = updateColumns.map((c) => `"${c}" = EXCLUDED."${c}"`).join(", ");
+        await client.none(
+          `INSERT INTO ${fullTableName} (${columnList}) VALUES (${placeholders})
              ON CONFLICT ("traceId", "spanId") DO UPDATE SET ${updateClause}`,
-            values
-          );
-        } else {
-          await this.client.none(
-            `INSERT INTO ${fullTableName} (${columnList}) VALUES (${placeholders})
-             ON CONFLICT ("traceId", "spanId") DO NOTHING`,
-            values
-          );
-        }
+          values
+        );
       } else {
-        await this.client.none(`INSERT INTO ${fullTableName} (${columnList}) VALUES (${placeholders})`, values);
+        await client.none(
+          `INSERT INTO ${fullTableName} (${columnList}) VALUES (${placeholders})
+             ON CONFLICT ("traceId", "spanId") DO NOTHING`,
+          values
+        );
       }
+    } else {
+      await client.none(`INSERT INTO ${fullTableName} (${columnList}) VALUES (${placeholders})`, values);
+    }
+  }
+  async insert({ tableName, record }) {
+    try {
+      await this.executeInsert(this.client, { tableName, record });
     } catch (error$1) {
       throw new error.MastraError(
         {
@@ -3003,13 +3006,12 @@ Note: This migration may take some time for large tables.
   }
   async batchInsert({ tableName, records }) {
     try {
-      await this.client.query("BEGIN");
-      for (const record of records) {
-        await this.insert({ tableName, record });
-      }
-      await this.client.query("COMMIT");
+      await this.client.tx(async (tx) => {
+        for (const record of records) {
+          await this.executeInsert(tx, { tableName, record });
+        }
+      });
     } catch (error$1) {
-      await this.client.query("ROLLBACK");
       throw new error.MastraError(
         {
           id: storage.createStorageErrorId("PG", "BATCH_INSERT", "FAILED"),
@@ -3287,30 +3289,7 @@ Note: This migration may take some time for large tables.
     data
   }) {
     try {
-      const filteredData = await this.filterRecordToKnownColumns(tableName, data);
-      if (Object.keys(filteredData).length === 0) return;
-      const setColumns = [];
-      const setValues = [];
-      let paramIndex = 1;
-      Object.entries(filteredData).forEach(([key, value]) => {
-        const parsedKey = utils.parseSqlIdentifier(key, "column name");
-        setColumns.push(`"${parsedKey}" = $${paramIndex++}`);
-        setValues.push(this.prepareValue(value, key, tableName));
-      });
-      const whereConditions = [];
-      const whereValues = [];
-      Object.entries(keys).forEach(([key, value]) => {
-        const parsedKey = utils.parseSqlIdentifier(key, "column name");
-        whereConditions.push(`"${parsedKey}" = $${paramIndex++}`);
-        whereValues.push(this.prepareValue(value, key, tableName));
-      });
-      const tableName_ = getTableName({
-        indexName: tableName,
-        schemaName: getSchemaName(this.schemaName)
-      });
-      const sql = `UPDATE ${tableName_} SET ${setColumns.join(", ")} WHERE ${whereConditions.join(" AND ")}`;
-      const values = [...setValues, ...whereValues];
-      await this.client.none(sql, values);
+      await this.executeUpdate(this.client, { tableName, keys, data });
     } catch (error$1) {
       throw new error.MastraError(
         {
@@ -3330,13 +3309,12 @@ Note: This migration may take some time for large tables.
     updates
   }) {
     try {
-      await this.client.query("BEGIN");
-      for (const { keys, data } of updates) {
-        await this.update({ tableName, keys, data });
-      }
-      await this.client.query("COMMIT");
+      await this.client.tx(async (tx) => {
+        for (const { keys, data } of updates) {
+          await this.executeUpdate(tx, { tableName, keys, data });
+        }
+      });
     } catch (error$1) {
-      await this.client.query("ROLLBACK");
       throw new error.MastraError(
         {
           id: storage.createStorageErrorId("PG", "BATCH_UPDATE", "FAILED"),
@@ -3395,6 +3373,36 @@ Note: This migration may take some time for large tables.
   async deleteData({ tableName }) {
     return this.clearTable({ tableName });
   }
+  async executeUpdate(client, {
+    tableName,
+    keys,
+    data
+  }) {
+    const filteredData = await this.filterRecordToKnownColumns(tableName, data);
+    if (Object.keys(filteredData).length === 0) return;
+    const setColumns = [];
+    const setValues = [];
+    let paramIndex = 1;
+    Object.entries(filteredData).forEach(([key, value]) => {
+      const parsedKey = utils.parseSqlIdentifier(key, "column name");
+      setColumns.push(`"${parsedKey}" = $${paramIndex++}`);
+      setValues.push(this.prepareValue(value, key, tableName));
+    });
+    const whereConditions = [];
+    const whereValues = [];
+    Object.entries(keys).forEach(([key, value]) => {
+      const parsedKey = utils.parseSqlIdentifier(key, "column name");
+      whereConditions.push(`"${parsedKey}" = $${paramIndex++}`);
+      whereValues.push(this.prepareValue(value, key, tableName));
+    });
+    const tableName_ = getTableName({
+      indexName: tableName,
+      schemaName: getSchemaName(this.schemaName)
+    });
+    const sql = `UPDATE ${tableName_} SET ${setColumns.join(", ")} WHERE ${whereConditions.join(" AND ")}`;
+    const values = [...setValues, ...whereValues];
+    await client.none(sql, values);
+  }
 };
 function getSchemaName2(schema) {
   return schema ? `"${utils.parseSqlIdentifier(schema, "schema name")}"` : void 0;
@@ -5308,6 +5316,16 @@ var ExperimentsPG = class _ExperimentsPG extends storage.ExperimentsStorage {
   async init() {
     await this.#db.createTable({ tableName: storage.TABLE_EXPERIMENTS, schema: storage.EXPERIMENTS_SCHEMA });
     await this.#db.createTable({ tableName: storage.TABLE_EXPERIMENT_RESULTS, schema: storage.EXPERIMENT_RESULTS_SCHEMA });
+    await this.#db.alterTable({
+      tableName: storage.TABLE_EXPERIMENTS,
+      schema: storage.EXPERIMENTS_SCHEMA,
+      ifNotExists: ["agentVersion"]
+    });
+    await this.#db.alterTable({
+      tableName: storage.TABLE_EXPERIMENT_RESULTS,
+      schema: storage.EXPERIMENT_RESULTS_SCHEMA,
+      ifNotExists: ["status", "tags"]
+    });
     await this.createDefaultIndexes();
     await this.createCustomIndexes();
   }