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

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

package/dist/docs/assets/SOURCE_MAP.json

@@ -1,5 +1,5 @@
 {
-  "version": "1.23.0-alpha.3",
+  "version": "1.23.0-alpha.5",
   "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);
@@ -23042,6 +23057,11 @@ var PgFactoryStorage = class extends _mastra_core_storage.FactoryStorage {
 			if (!spec.nullable || spec.type === "uuid-pk" || spec.primaryKey) continue;
 			await this.#pool.query(`ALTER TABLE "${schema.name}" ALTER COLUMN "${name}" DROP NOT NULL`);
 		}
+		for (const [name, spec] of Object.entries(schema.columns)) {
+			if (spec.type !== "bigint") continue;
+			const { rows } = await this.#pool.query(`SELECT data_type FROM information_schema.columns WHERE table_schema = current_schema() AND table_name = $1 AND column_name = $2`, [schema.name, name]);
+			if (rows[0]?.data_type === "integer") await this.#pool.query(`ALTER TABLE "${schema.name}" ALTER COLUMN "${name}" TYPE BIGINT`);
+		}
 		for (const index of schema.uniqueIndexes ?? []) {
 			assertIdentifier("index", index.name);
 			index.columns.forEach((column) => assertIdentifier("column", column));