@mastra/pg 1.30.0-alpha.4 → 1.30.0-alpha.6

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.30.0-alpha.4"
+  version: "1.30.0-alpha.6"
 ---
 
 ## When to use

package/dist/docs/assets/SOURCE_MAP.json

@@ -1,5 +1,5 @@
 {
-  "version": "1.30.0-alpha.4",
+  "version": "1.30.0-alpha.6",
   "package": "@mastra/pg",
   "exports": {},
   "modules": {}

package/dist/index.cjs

@@ -726,17 +726,25 @@ var PgVector = class extends _mastra_core_vector.MastraVector {
 			await client.query(`SET search_path TO ${[...schemas].map((s) => `"${s}"`).join(", ")}`);
 		}
 	}
-	/**
-	* Checks if the installed pgvector version supports halfvec type.
-	* halfvec was introduced in pgvector 0.7.0.
-	*/
-	supportsHalfvec() {
+	/** Checks if the installed pgvector version is at least the given major.minor version. */
+	isVectorExtensionAtLeast(minMajor, minMinor) {
 		if (!this.vectorExtensionVersion) return false;
 		const parts = this.vectorExtensionVersion.split(".");
 		const major = parseInt(parts[0] ?? "", 10);
 		const minor = parseInt(parts[1] ?? "", 10);
 		if (isNaN(major) || isNaN(minor)) return false;
-		return major > 0 || major === 0 && minor >= 7;
+		return major > minMajor || major === minMajor && minor >= minMinor;
+	}
+	/**
+	* Checks if the installed pgvector version supports halfvec type.
+	* halfvec was introduced in pgvector 0.7.0.
+	*/
+	supportsHalfvec() {
+		return this.isVectorExtensionAtLeast(0, 7);
+	}
+	/** Checks if pgvector >= 0.8.0 (required for the hnsw.iterative_scan setting). */
+	supportsIterativeScan() {
+		return this.isVectorExtensionAtLeast(0, 8);
 	}
 	/** Checks if pgvector >= 0.7.0 (required for bit type). */
 	supportsBit() {
@@ -1055,7 +1063,7 @@ var PgVector = class extends _mastra_core_vector.MastraVector {
 				const calculatedEf = ef ?? Math.max(topK, (indexInfo?.config?.m ?? 16) * topK);
 				const searchEf = Math.min(1e3, Math.max(1, calculatedEf));
 				await client.query(`SET LOCAL hnsw.ef_search = ${searchEf}`);
-				await client.query(`SET LOCAL hnsw.iterative_scan = strict_order`);
+				if (this.supportsIterativeScan()) await client.query(`SET LOCAL hnsw.iterative_scan = strict_order`);
 			}
 			if (indexInfo.type === "ivfflat" && probes) await client.query(`SET LOCAL ivfflat.probes = ${probes}`);
 			const { tableName } = this.getTableName(indexName);
@@ -2565,10 +2573,13 @@ function resolvePgConfig(config) {
 		};
 	}
 	let pool;
-	if ("connectionString" in config) pool = new pg.Pool({
+	if ("connectionString" in config) pool = new pg.Pool(buildConnectionStringPoolConfig({
 		connectionString: config.connectionString,
 		ssl: config.ssl
-	});
+	}, {
+		max: 10,
+		idleTimeoutMillis: 1e4
+	}));
 	else pool = new pg.Pool({
 		host: config.host,
 		port: config.port,
@@ -2780,7 +2791,15 @@ BEGIN
 END
 $mastra_timestamps_trigger$;`;
 }
-const schemaSetupRegistry = /* @__PURE__ */ new Map();
+const schemaSetupRegistry = /* @__PURE__ */ new WeakMap();
+function getSchemaSetupRegistry(pool) {
+	let registry = schemaSetupRegistry.get(pool);
+	if (!registry) {
+		registry = /* @__PURE__ */ new Map();
+		schemaSetupRegistry.set(pool, registry);
+	}
+	return registry;
+}
 /**
 * Guard prune batch limits: a non-positive limit deletes nothing while callers'
 * drain checks (`affected < limit`) never fire, which turns the prune loop into
@@ -3019,11 +3038,12 @@ var PgDB = class extends _mastra_core_base.MastraBase {
 	}
 	async setupSchema() {
 		if (!this.schemaName) return;
-		let registryEntry = schemaSetupRegistry.get(this.schemaName);
+		const registry = getSchemaSetupRegistry(this.client.$pool);
+		let registryEntry = registry.get(this.schemaName);
 		if (registryEntry?.complete) return;
 		const snapshot = this.schemaSnapshot;
 		if (snapshot && snapshot.tables.size > 0) {
-			schemaSetupRegistry.set(this.schemaName, {
+			registry.set(this.schemaName, {
 				promise: null,
 				complete: true
 			});
@@ -3049,19 +3069,19 @@ var PgDB = class extends _mastra_core_base.MastraBase {
 							throw new Error(`Unable to create schema "${schemaNameCapture}". This requires CREATE privilege on the database. Either create the schema manually or grant CREATE privilege to the user.`);
 						}
 					}
-					const entry = schemaSetupRegistry.get(schemaNameCapture);
+					const entry = registry.get(schemaNameCapture);
 					if (entry) entry.complete = true;
 					this.logger.debug(`Schema "${quotedSchemaName}" is ready for use`);
 				} catch (error) {
-					schemaSetupRegistry.delete(schemaNameCapture);
+					registry.delete(schemaNameCapture);
 					throw error;
 				}
 			})();
-			schemaSetupRegistry.set(this.schemaName, {
+			registry.set(this.schemaName, {
 				promise: setupPromise,
 				complete: false
 			});
-			registryEntry = schemaSetupRegistry.get(this.schemaName);
+			registryEntry = registry.get(this.schemaName);
 		}
 		await registryEntry.promise;
 	}