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

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.23.0-alpha.2"
+  version: "1.23.0-alpha.3"
 ---
 
 ## When to use

package/dist/docs/assets/SOURCE_MAP.json

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

package/dist/index.cjs

@@ -936,12 +936,12 @@ var PgVector = class extends _mastra_core_vector.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);
@@ -1016,6 +1016,7 @@ var PgVector = class extends _mastra_core_vector.MastraVector {
 	async upsert({ indexName, vectors, metadata, ids, deleteFilter, namespace = DEFAULT_NAMESPACE }) {
 		(0, _mastra_core_vector.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);
@@ -1037,7 +1038,6 @@ var PgVector = class extends _mastra_core_vector.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++) {
@@ -1325,7 +1325,7 @@ var PgVector = class extends _mastra_core_vector.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 {
@@ -1529,8 +1529,8 @@ var PgVector = class extends _mastra_core_vector.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(`
@@ -1590,7 +1590,6 @@ var PgVector = class extends _mastra_core_vector.MastraVector {
 				config
 			};
 		} catch (e) {
-			await client.query("ROLLBACK");
 			const mastraError = new _mastra_core_error.MastraError({
 				id: (0, _mastra_core_storage.createVectorErrorId)("PG", "DESCRIBE_INDEX", "FAILED"),
 				domain: _mastra_core_error.ErrorDomain.MASTRA_VECTOR,
@@ -1600,7 +1599,7 @@ var PgVector = class extends _mastra_core_vector.MastraVector {
 			this.logger?.trackException(mastraError);
 			throw mastraError;
 		} finally {
-			client.release();
+			if (!existingClient) client.release();
 		}
 	}
 	/**
@@ -1705,10 +1704,10 @@ var PgVector = class extends _mastra_core_vector.MastraVector {
 				category: _mastra_core_error.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 = [];