@mastra/pg 1.23.0-alpha.2 → 1.23.0-alpha.3

package/dist/index.js

@@ -912,12 +912,12 @@ var PgVector = class extends MastraVector {
 				client.release();
 			}
 		}
+		const indexInfo = await this.getIndexMetadata({ indexName });
 		const client = await this.pool.connect();
 		try {
 			await this.ensureSearchPath(client);
 			await client.query("BEGIN");
 			const { sql: filterQuery, values: filterValues } = buildFilterQuery(this.transformFilter(filter), minScore, topK);
-			const indexInfo = await this.getIndexMetadata({ indexName });
 			const metric = indexInfo.metric ?? "cosine";
 			const ops = this.getVectorOps(indexInfo.vectorType, metric);
 			const vectorStr = ops.formatVector(queryVector, indexInfo.dimension);
@@ -992,6 +992,7 @@ var PgVector = class extends MastraVector {
 	async upsert({ indexName, vectors, metadata, ids, deleteFilter, namespace = DEFAULT_NAMESPACE }) {
 		validateUpsertInput("PG", vectors, metadata, ids);
 		const { tableName } = this.getTableName(indexName);
+		const indexInfo = await this.getIndexMetadata({ indexName });
 		const client = await this.pool.connect();
 		try {
 			await this.ensureSearchPath(client);
@@ -1013,7 +1014,6 @@ var PgVector = class extends MastraVector {
 				}
 			}
 			const vectorIds = ids || vectors.map(() => crypto.randomUUID());
-			const indexInfo = await this.getIndexMetadata({ indexName });
 			const qualifiedVectorType = this.getVectorTypeName(indexInfo.vectorType, indexInfo.dimension);
 			const ops = this.getVectorOps(indexInfo.vectorType, indexInfo.metric ?? "cosine");
 			if (new Set(vectorIds).size !== vectorIds.length) for (let i = 0; i < vectors.length; i++) {
@@ -1301,7 +1301,7 @@ var PgVector = class extends MastraVector {
 			let existingIndexInfo = null;
 			let dimension = 0;
 			try {
-				existingIndexInfo = await this.getIndexMetadata({ indexName });
+				existingIndexInfo = await this.describeIndexMetadata({ indexName }, client);
 				dimension = existingIndexInfo.dimension;
 				if (isConfigEmpty && existingIndexInfo.metric === metric) if (existingIndexInfo.type === "flat") this.logger?.debug(`No index exists for ${vectorIndexName}, will create default ivfflat index`);
 				else {
@@ -1505,8 +1505,8 @@ var PgVector = class extends MastraVector {
 	* {@link describeIndex} it issues no `COUNT(*)`, so its cost does not grow with the
 	* number of rows in the table.
 	*/
-	async describeIndexMetadata({ indexName }) {
-		const client = await this.pool.connect();
+	async describeIndexMetadata({ indexName }, existingClient) {
+		const client = existingClient ?? await this.pool.connect();
 		try {
 			const { tableName, parsedIndexName } = this.getTableName(indexName);
 			const tableExists = await client.query(`
@@ -1566,7 +1566,6 @@ var PgVector = class extends MastraVector {
 				config
 			};
 		} catch (e) {
-			await client.query("ROLLBACK");
 			const mastraError = new MastraError({
 				id: createVectorErrorId("PG", "DESCRIBE_INDEX", "FAILED"),
 				domain: ErrorDomain.MASTRA_VECTOR,
@@ -1576,7 +1575,7 @@ var PgVector = class extends MastraVector {
 			this.logger?.trackException(mastraError);
 			throw mastraError;
 		} finally {
-			client.release();
+			if (!existingClient) client.release();
 		}
 	}
 	/**
@@ -1681,10 +1680,10 @@ var PgVector = class extends MastraVector {
 				category: ErrorCategory.USER,
 				details: { indexName }
 			});
+			const indexInfo = await this.getIndexMetadata({ indexName });
 			client = await this.pool.connect();
 			await this.ensureSearchPath(client);
 			const { tableName } = this.getTableName(indexName);
-			const indexInfo = await this.getIndexMetadata({ indexName });
 			const qualifiedVectorType = this.getVectorTypeName(indexInfo.vectorType, indexInfo.dimension);
 			const ops = this.getVectorOps(indexInfo.vectorType, indexInfo.metric ?? "cosine");
 			let updateParts = [];