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

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.4"
 ---
 
 ## 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.4",
   "package": "@mastra/pg",
   "exports": {},
   "modules": {}

package/dist/docs/references/reference-tools-vector-query-tool.md

@@ -79,7 +79,7 @@ const queryTool = createVectorQueryTool({
 
 The tool returns an object with:
 
-**relevantContext** (`string`): Combined text from the most relevant document chunks
+**relevantContext** (`any[]`): Array of metadata objects for the most relevant chunks, in rank order (one entry per result). The chunk text is available at relevantContext\[i].text when it was stored in metadata during ingestion.
 
 **sources** (`QueryResult[]`): Array of full retrieval result objects. Each object contains all information needed to reference the original document, chunk, and similarity score.
 
@@ -95,6 +95,8 @@ The tool returns an object with:
 }
 ```
 
+`document` is only populated by vector stores whose `query()` returns document content, such as Chroma, Elasticsearch, LanceDB, and MongoDB. For other stores, such as PgVector, it's an empty string. Read the chunk text from `sources[i].metadata.text` (or whichever metadata key you stored it under).
+
 ## Default tool description
 
 The default description focuses on:
@@ -489,7 +491,7 @@ The tool is created with:
 
 - **ID**: `VectorQuery {vectorStoreName} {indexName} Tool`
 - **Input Schema**: Requires queryText and filter objects
-- **Output Schema**: Returns relevantContext string
+- **Output Schema**: Returns `relevantContext` (array of chunk metadata) and `sources` (array of `QueryResult`)
 
 ## Related
 

package/dist/index.cjs

@@ -21,6 +21,7 @@ var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__ge
 	enumerable: true
 }) : target, mod));
 //#endregion
+let crypto$1 = require("crypto");
 let _mastra_core_error = require("@mastra/core/error");
 let _mastra_core_storage = require("@mastra/core/storage");
 _mastra_core_storage = __toESM(_mastra_core_storage, 1);
@@ -34,7 +35,6 @@ xxhash_wasm = __toESM(xxhash_wasm, 1);
 let pg_connection_string = require("pg-connection-string");
 let _mastra_core_vector_filter = require("@mastra/core/vector/filter");
 let _mastra_core_base = require("@mastra/core/base");
-let crypto$1 = require("crypto");
 let _mastra_core_agent = require("@mastra/core/agent");
 let _mastra_core_features = require("@mastra/core/features");
 let _mastra_core_evals = require("@mastra/core/evals");
@@ -828,6 +828,8 @@ var PgVector = class extends _mastra_core_vector.MastraVector {
          AND attnum > 0
          AND NOT attisdropped`, [tableName])).rowCount === 0) return;
 		await client.query(`ALTER TABLE ${tableName} ADD COLUMN IF NOT EXISTS namespace VARCHAR(255) NOT NULL DEFAULT '${DEFAULT_NAMESPACE}'`);
+		const namespaceIndexName = this.getNamespaceIndexName(parsedIndexName);
+		await client.query(`CREATE UNIQUE INDEX IF NOT EXISTS "${namespaceIndexName}" ON ${tableName} (namespace, vector_id)`);
 		const legacyConstraints = await client.query(`SELECT c.conname
        FROM pg_constraint c
        WHERE c.conrelid = to_regclass($1)
@@ -837,8 +839,21 @@ var PgVector = class extends _mastra_core_vector.MastraVector {
 			const parsedConstraintName = (0, _mastra_core_utils.parseSqlIdentifier)(conname, "constraint name");
 			await client.query(`ALTER TABLE ${tableName} DROP CONSTRAINT "${parsedConstraintName}"`);
 		}
-		const namespaceIndexName = (0, _mastra_core_utils.parseSqlIdentifier)(`${parsedIndexName}_namespace_vector_id_idx`, "index name");
-		await client.query(`CREATE UNIQUE INDEX IF NOT EXISTS "${namespaceIndexName}" ON ${tableName} (namespace, vector_id)`);
+	}
+	/**
+	* Name of the unique (namespace, vector_id) index for a table.
+	*
+	* The full `<index>_namespace_vector_id_idx` name is kept whenever it fits so tables that
+	* were already migrated keep matching `IF NOT EXISTS`. Longer index names would exceed
+	* Postgres' 63-char identifier limit, so those fall back to a truncated prefix plus a hash
+	* of the index name to distinguish tables that
+	* share a long prefix.
+	*/
+	getNamespaceIndexName(parsedIndexName) {
+		const fullName = `${parsedIndexName}_namespace_vector_id_idx`;
+		if (fullName.length <= 63) return fullName;
+		const suffix = `_ns_${(0, crypto$1.createHash)("sha256").update(parsedIndexName).digest("hex").slice(0, 32)}_idx`;
+		return `${parsedIndexName.slice(0, 63 - suffix.length)}${suffix}`;
 	}
 	transformFilter(filter) {
 		return new PGFilterTranslator().translate(filter);
@@ -936,12 +951,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 +1031,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 +1053,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 +1340,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 +1544,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 +1605,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 +1614,7 @@ var PgVector = class extends _mastra_core_vector.MastraVector {
 			this.logger?.trackException(mastraError);
 			throw mastraError;
 		} finally {
-			client.release();
+			if (!existingClient) client.release();
 		}
 	}
 	/**
@@ -1705,10 +1719,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 = [];