@mastra/pg 1.30.0-alpha.4 → 1.30.0

package/dist/index.js

@@ -702,17 +702,25 @@ var PgVector = class extends 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() {
@@ -1031,7 +1039,7 @@ var PgVector = class extends 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);
@@ -2541,10 +2549,13 @@ function resolvePgConfig(config) {
 		};
 	}
 	let pool;
-	if ("connectionString" in config) pool = new Pool({
+	if ("connectionString" in config) pool = new Pool(buildConnectionStringPoolConfig({
 		connectionString: config.connectionString,
 		ssl: config.ssl
-	});
+	}, {
+		max: 10,
+		idleTimeoutMillis: 1e4
+	}));
 	else pool = new Pool({
 		host: config.host,
 		port: config.port,
@@ -2756,7 +2767,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
@@ -2995,11 +3014,12 @@ var PgDB = class extends 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
 			});
@@ -3025,19 +3045,19 @@ var PgDB = class extends 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;
 	}