@mastra/pg 1.20.1-alpha.1 → 1.20.1-alpha.3

package/CHANGELOG.md

@@ -1,5 +1,41 @@
 # @mastra/pg
 
+## 1.20.1-alpha.3
+
+### Patch Changes
+
+- Stop a dropped Postgres connection from killing the process while a client is checked out. ([#21765](https://github.com/mastra-ai/mastra/pull/21765))
+
+  `PgFactoryStorage` attached an `error` listener to the pool, but pg only routes pool-level errors for _idle_ clients — it hands ownership of a client to the borrower for the duration of a checkout. A backend restart or network blip that landed on a client mid-transaction therefore reached an emitter with nothing listening, and Node escalated it to an uncaughtException that took the whole server down (`Connection terminated unexpectedly` at `pg/lib/client.js`), even though the idle siblings were logged and discarded cleanly.
+
+  Pools created by `PgFactoryStorage` now attach a listener once per physical connection as it is established, so a client stays covered while borrowed. While the client is idle the pool's own listener already reports the failure, so the extra listener stays quiet and a dropped connection is announced once, as the right thing. The pool still discards the failed connection and reconnects on the next checkout; the failure is now logged instead of fatal. Caller-supplied pools are left untouched, as before.
+
+- Updated dependencies [[`6db7a5d`](https://github.com/mastra-ai/mastra/commit/6db7a5dd3dd2b6f7ef75dcd804fcffef5fa83963), [`0cdc5dc`](https://github.com/mastra-ai/mastra/commit/0cdc5dc69024957815da4f51acc4119eb4f447d7)]:
+  - @mastra/core@1.60.0-alpha.12
+
+## 1.20.1-alpha.2
+
+### Patch Changes
+
+- Added a `durable` option to stored agents so agents created through the Agents API can run with durable execution — no code deployment required. ([#21715](https://github.com/mastra-ai/mastra/pull/21715))
+
+  ```typescript
+  await mastraClient.createStoredAgent({
+    id: 'helper',
+    name: 'Helper',
+    instructions: 'You are a helpful assistant.',
+    model: { provider: 'openai', name: 'gpt-5' },
+    durable: true,
+  });
+  ```
+
+  Pass `true` for defaults, or `{ maxSteps, cleanupTimeoutMs }` to tune the durable loop. Cache and pubsub are inherited from the server's Mastra instance, so configure distributed backends there for durability across replicas. Automatic recovery is still configured in code via `recovery.durableAgents`.
+
+- Improved PgVector upsert performance when writing many vectors at once. Batches are now written with a small number of multi-row inserts instead of one insert per vector, which reduces database round trips and connection pool usage during RAG and memory ingestion. ([#21719](https://github.com/mastra-ai/mastra/pull/21719))
+
+- Updated dependencies [[`6223446`](https://github.com/mastra-ai/mastra/commit/6223446ddce6166e96e0ba5e00d628b615dee8ca), [`583e235`](https://github.com/mastra-ai/mastra/commit/583e23519c13af16c1746f9c49722d011216611b), [`a77f8d4`](https://github.com/mastra-ai/mastra/commit/a77f8d4740d2178a74c41e4bf678b4fcd8fa0bb2), [`40d358e`](https://github.com/mastra-ai/mastra/commit/40d358e29d55543803e64b49241122f598ffabc7), [`e80cd7e`](https://github.com/mastra-ai/mastra/commit/e80cd7e7683e7d732e1cc6784bcac1d2640d2ce3), [`20504b2`](https://github.com/mastra-ai/mastra/commit/20504b2ecebd0e077acda3d457ab57480a98ed3e)]:
+  - @mastra/core@1.60.0-alpha.11
+
 ## 1.20.1-alpha.1
 
 ### Patch 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.20.1-alpha.1"
+  version: "1.20.1-alpha.3"
 ---
 
 ## When to use

package/dist/docs/assets/SOURCE_MAP.json

@@ -1,5 +1,5 @@
 {
-  "version": "1.20.1-alpha.1",
+  "version": "1.20.1-alpha.3",
   "package": "@mastra/pg",
   "exports": {},
   "modules": {}

package/dist/docs/references/reference-memory-memory-class.md

@@ -145,4 +145,5 @@ export const agent = new Agent({
 - [listThreads](https://mastra.ai/reference/memory/listThreads)
 - [deleteMessages](https://mastra.ai/reference/memory/deleteMessages)
 - [cloneThread](https://mastra.ai/reference/memory/cloneThread)
+- [settled](https://mastra.ai/reference/memory/settled)
 - [Clone Utility Methods](https://mastra.ai/reference/memory/clone-utilities)

package/dist/index.cjs

@@ -565,6 +565,7 @@ function buildFilterQuery(filter, minScore, topK) {
 }
 //#endregion
 //#region src/vector/index.ts
+const MAX_UPSERT_ROWS_PER_STATEMENT = Math.floor(65535 / 3);
 var PgVector = class extends _mastra_core_vector.MastraVector {
 	pool;
 	describeIndexCache = /* @__PURE__ */ new Map();
@@ -957,23 +958,41 @@ var PgVector = class extends _mastra_core_vector.MastraVector {
 			const indexInfo = await this.getIndexInfo({ indexName });
 			const qualifiedVectorType = this.getVectorTypeName(indexInfo.vectorType, indexInfo.dimension);
 			const ops = this.getVectorOps(indexInfo.vectorType, indexInfo.metric ?? "cosine");
-			for (let i = 0; i < vectors.length; i++) {
+			if (new Set(vectorIds).size !== vectorIds.length) for (let i = 0; i < vectors.length; i++) {
 				const vectorStr = ops.formatVector(vectors[i], indexInfo.dimension);
 				const query = `
-          INSERT INTO ${tableName} (vector_id, embedding, metadata)
-          VALUES ($1, $2::${qualifiedVectorType}, $3::jsonb)
-          ON CONFLICT (vector_id)
-          DO UPDATE SET
-            embedding = $2::${qualifiedVectorType},
-            metadata = $3::jsonb
-          RETURNING embedding::text
-        `;
+            INSERT INTO ${tableName} (vector_id, embedding, metadata)
+            VALUES ($1, $2::${qualifiedVectorType}, $3::jsonb)
+            ON CONFLICT (vector_id)
+            DO UPDATE SET
+              embedding = $2::${qualifiedVectorType},
+              metadata = $3::jsonb
+          `;
 				await client.query(query, [
 					vectorIds[i],
 					vectorStr,
 					JSON.stringify(metadata?.[i] || {})
 				]);
 			}
+			else for (let start = 0; start < vectors.length; start += MAX_UPSERT_ROWS_PER_STATEMENT) {
+				const end = Math.min(start + MAX_UPSERT_ROWS_PER_STATEMENT, vectors.length);
+				const rows = [];
+				const values = [];
+				for (let i = start; i < end; i++) {
+					const base = values.length;
+					rows.push(`($${base + 1}, $${base + 2}::${qualifiedVectorType}, $${base + 3}::jsonb)`);
+					values.push(vectorIds[i], ops.formatVector(vectors[i], indexInfo.dimension), JSON.stringify(metadata?.[i] || {}));
+				}
+				const query = `
+            INSERT INTO ${tableName} (vector_id, embedding, metadata)
+            VALUES ${rows.join(", ")}
+            ON CONFLICT (vector_id)
+            DO UPDATE SET
+              embedding = EXCLUDED.embedding,
+              metadata = EXCLUDED.metadata
+          `;
+				await client.query(query, values);
+			}
 			await client.query("COMMIT");
 			this.logger?.debug(`Upserted ${vectors.length} vectors to ${indexName}`, {
 				indexName,
@@ -3803,6 +3822,7 @@ var AgentsPG = class AgentsPG extends _mastra_core_storage.AgentsStorage {
 				"workspace",
 				"skills",
 				"skillsFormat",
+				"durable",
 				"browser",
 				"toolProviders"
 			]
@@ -4282,10 +4302,10 @@ var AgentsPG = class AgentsPG extends _mastra_core_storage.AgentsStorage {
           "defaultOptions", workflows, agents, "integrationTools", "toolProviders",
           "inputProcessors", "outputProcessors", memory, scorers,
           "mcpClients", "requestContextSchema", workspace, skills, "skillsFormat",
-          browser,
+          durable, browser,
           "changedFields", "changeMessage",
           "createdAt", "createdAtZ"
-        ) VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14, $15, $16, $17, $18, $19, $20, $21, $22, $23, $24, $25, $26, $27)`, [
+        ) VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14, $15, $16, $17, $18, $19, $20, $21, $22, $23, $24, $25, $26, $27, $28)`, [
 				input.id,
 				input.agentId,
 				input.versionNumber,
@@ -4308,6 +4328,7 @@ var AgentsPG = class AgentsPG extends _mastra_core_storage.AgentsStorage {
 				input.workspace ? JSON.stringify(input.workspace) : null,
 				input.skills ? JSON.stringify(input.skills) : null,
 				input.skillsFormat ?? null,
+				input.durable !== void 0 ? JSON.stringify(input.durable) : null,
 				input.browser ? JSON.stringify(input.browser) : null,
 				input.changedFields ? JSON.stringify(input.changedFields) : null,
 				input.changeMessage ?? null,
@@ -4536,6 +4557,7 @@ var AgentsPG = class AgentsPG extends _mastra_core_storage.AgentsStorage {
 			workspace: parseJsonResilient(row.workspace, "workspace"),
 			skills: parseJsonResilient(row.skills, "skills"),
 			skillsFormat: row.skillsFormat,
+			durable: parseJsonResilient(row.durable, "durable"),
 			browser: parseJsonResilient(row.browser, "browser"),
 			changedFields: parseJsonResilient(row.changedFields, "changedFields"),
 			changeMessage: row.changeMessage,
@@ -20752,6 +20774,38 @@ var PgFactoryStorageOps = class {
 * Supports serializable app-table transactions for cross-replica invariant
 * enforcement and `authDatabase()` exposing the shared pool.
 */
+/**
+* A pool-level `error` listener only covers *idle* clients: pg hands ownership
+* of a client to the borrower for the duration of a checkout and takes its own
+* listener off. So a backend restart or network blip that lands on a client
+* mid-transaction reaches an emitter with no listener, and Node escalates that
+* to an uncaughtException that kills the process — the pool handler logs the
+* idle siblings and the borrowed one still brings everything down.
+*
+* Attach a listener once per physical connection instead. `connect` fires when
+* the pool establishes a client, so the client stays covered for its whole life
+* and there is nothing to remove on release.
+*
+* That listener outlives the checkout, so it also sees the failures the pool is
+* already reporting through its own idle listener. Follow `acquire`/`release`
+* to tell the two apart and stay quiet while the client is idle — otherwise one
+* dropped connection is announced twice, the second time as the wrong thing.
+*
+* The pool still discards the failed connection; this only keeps the failure
+* reportable instead of fatal.
+*/
+function guardCheckedOutClientErrors(pool, warn = console.warn) {
+	const borrowed = /* @__PURE__ */ new WeakSet();
+	pool.on("connect", (client) => {
+		borrowed.add(client);
+		client.on("error", (err) => {
+			if (!borrowed.has(client)) return;
+			warn(`PgFactoryStorage: client error while checked out (the pool discards this connection): ${err instanceof Error ? err.message : String(err)}`);
+		});
+	});
+	pool.on("acquire", (client) => borrowed.add(client));
+	pool.on("release", (_err, client) => borrowed.delete(client));
+}
 var PgFactoryStorage = class extends _mastra_core_storage.FactoryStorage {
 	ops;
 	#pool;
@@ -20772,6 +20826,7 @@ var PgFactoryStorage = class extends _mastra_core_storage.FactoryStorage {
 			this.#pool.on("error", (err) => {
 				console.warn(`PgFactoryStorage: idle pool client error (pool discards the client and reconnects on next checkout): ${err instanceof Error ? err.message : String(err)}`);
 			});
+			guardCheckedOutClientErrors(this.#pool);
 		}
 		this.ops = new PgFactoryStorageOps(this.#pool, this.#schemas);
 	}