@mastra/pg 1.22.3-alpha.1 → 1.22.3-alpha.2

package/dist/index.js

@@ -794,19 +794,18 @@ var PgVector = class extends MastraVector {
 	}
 	async ensureNamespaceSchema(indexName, client) {
 		const { tableName, parsedIndexName } = this.getTableName(indexName);
-		const schemaName = this.schema ? parseSqlIdentifier(this.schema, "schema name") : "public";
 		if ((await client.query(`SELECT 1
-       FROM information_schema.columns
-       WHERE table_schema = $1 AND table_name = $2 AND column_name = 'vector_id'`, [schemaName, parsedIndexName])).rowCount === 0) return;
+       FROM pg_attribute
+       WHERE attrelid = to_regclass($1)
+         AND attname = 'vector_id'
+         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 legacyConstraints = await client.query(`SELECT c.conname
        FROM pg_constraint c
-       JOIN pg_class t ON t.oid = c.conrelid
-       JOIN pg_namespace n ON n.oid = t.relnamespace
-       WHERE n.nspname = $1
-         AND t.relname = $2
+       WHERE c.conrelid = to_regclass($1)
          AND c.contype = 'u'
-         AND pg_get_constraintdef(c.oid) = 'UNIQUE (vector_id)'`, [schemaName, parsedIndexName]);
+         AND pg_get_constraintdef(c.oid) = 'UNIQUE (vector_id)'`, [tableName]);
 		for (const { conname } of legacyConstraints.rows) {
 			const parsedConstraintName = parseSqlIdentifier(conname, "constraint name");
 			await client.query(`ALTER TABLE ${tableName} DROP CONSTRAINT "${parsedConstraintName}"`);
@@ -1444,7 +1443,9 @@ var PgVector = class extends MastraVector {
 			return (await client.query(`
         SELECT DISTINCT t.table_name
         FROM information_schema.tables t
-        WHERE t.table_schema = $1
+        WHERE t.table_schema = ANY(
+          CASE WHEN $1::text IS NULL THEN current_schemas(false) ELSE ARRAY[$1::text] END
+        )
         AND EXISTS (
           SELECT 1
           FROM information_schema.columns c
@@ -1469,7 +1470,7 @@ var PgVector = class extends MastraVector {
           AND c.column_name = 'metadata'
           AND c.data_type = 'jsonb'
         );
-      `, [this.schema || "public"])).rows.map((row) => row.table_name);
+      `, [this.schema ?? null])).rows.map((row) => row.table_name);
 		} catch (e) {
 			const mastraError = new MastraError({
 				id: createVectorErrorId("PG", "LIST_INDEXES", "FAILED"),
@@ -1504,15 +1505,18 @@ var PgVector = class extends MastraVector {
 	async describeIndexMetadata({ indexName }) {
 		const client = await this.pool.connect();
 		try {
-			const { tableName } = this.getTableName(indexName);
+			const { tableName, parsedIndexName } = this.getTableName(indexName);
 			const tableExists = await client.query(`
-        SELECT udt_name
-        FROM information_schema.columns
-        WHERE table_schema = $1
-          AND table_name = $2
-          AND udt_name IN ('vector', 'halfvec', 'bit', 'sparsevec')
+        SELECT t.typname AS udt_name
+        FROM pg_attribute a
+        JOIN pg_type t ON t.oid = a.atttypid
+        WHERE a.attrelid = to_regclass($1)
+          AND a.attname = 'embedding'
+          AND a.attnum > 0
+          AND NOT a.attisdropped
+          AND t.typname IN ('vector', 'halfvec', 'bit', 'sparsevec')
         LIMIT 1;
-      `, [this.schema || "public", indexName]);
+      `, [tableName]);
 			if (tableExists.rows.length === 0) throw new Error(`Vector table ${tableName} does not exist`);
 			const udtName = tableExists.rows[0].udt_name;
 			const vectorType = udtName === "halfvec" ? "halfvec" : udtName === "bit" ? "bit" : udtName === "sparsevec" ? "sparsevec" : "vector";
@@ -1531,12 +1535,11 @@ var PgVector = class extends MastraVector {
             JOIN pg_class c ON i.indexrelid = c.oid
             JOIN pg_am am ON c.relam = am.oid
             JOIN pg_opclass opclass ON i.indclass[0] = opclass.oid
-            JOIN pg_namespace n ON c.relnamespace = n.oid
             WHERE c.relname = $1
-            AND n.nspname = $2;
+            AND i.indrelid = to_regclass($2);
             `;
 			const dimResult = await client.query(dimensionQuery, [tableName]);
-			const { index_method, index_def, operator_class } = (await client.query(indexQuery, [`${indexName}_vector_idx`, this.schema || "public"])).rows[0] || {
+			const { index_method, index_def, operator_class } = (await client.query(indexQuery, [`${parsedIndexName}_vector_idx`, tableName])).rows[0] || {
 				index_method: "flat",
 				index_def: "",
 				operator_class: "cosine"
@@ -4474,6 +4477,25 @@ var AgentsPG = class AgentsPG extends AgentsStorage {
 			}, error);
 		}
 	}
+	async getVersions(ids) {
+		if (ids.length === 0) return [];
+		try {
+			const tableName = getTableName$5({
+				indexName: TABLE_AGENT_VERSIONS,
+				schemaName: getSchemaName$5(this.#schema)
+			});
+			const placeholders = ids.map((_, i) => `$${i + 1}`).join(", ");
+			return (await this.#db.client.manyOrNone(`SELECT * FROM ${tableName} WHERE id IN (${placeholders})`, ids)).map((row) => this.parseVersionRow(row));
+		} catch (error) {
+			if (error instanceof MastraError) throw error;
+			throw new MastraError({
+				id: createStorageErrorId("PG", "GET_VERSIONS", "FAILED"),
+				domain: ErrorDomain.STORAGE,
+				category: ErrorCategory.THIRD_PARTY,
+				details: { count: ids.length }
+			}, error);
+		}
+	}
 	async getVersionByNumber(agentId, versionNumber) {
 		try {
 			const tableName = getTableName$5({
@@ -20098,6 +20120,25 @@ var SkillsPG = class SkillsPG extends SkillsStorage {
 			}, error);
 		}
 	}
+	async getVersions(ids) {
+		if (ids.length === 0) return [];
+		try {
+			const tableName = getTableName$5({
+				indexName: TABLE_SKILL_VERSIONS,
+				schemaName: getSchemaName$5(this.#schema)
+			});
+			const placeholders = ids.map((_, i) => `$${i + 1}`).join(", ");
+			return (await this.#db.client.manyOrNone(`SELECT * FROM ${tableName} WHERE id IN (${placeholders})`, ids)).map((row) => this.parseVersionRow(row));
+		} catch (error) {
+			if (error instanceof MastraError) throw error;
+			throw new MastraError({
+				id: createStorageErrorId("PG", "GET_SKILL_VERSIONS", "FAILED"),
+				domain: ErrorDomain.STORAGE,
+				category: ErrorCategory.THIRD_PARTY,
+				details: { count: ids.length }
+			}, error);
+		}
+	}
 	async getVersionByNumber(skillId, versionNumber) {
 		try {
 			const tableName = getTableName$5({