@mastra/pg 1.27.1-alpha.0 → 1.27.1

package/dist/index.js

@@ -94,6 +94,34 @@ function buildConnectionStringPoolConfig(config, defaults) {
 	};
 }
 //#endregion
+//#region src/shared/schema-name.ts
+const POSTGRES_IDENTIFIER_MAX_BYTES = 63;
+const UNSAFE_SCHEMA_NAME_CHARS = /["'\\$\u0000-\u001f\u007f]/;
+/**
+* Validates a PostgreSQL schema name.
+*
+* Schema names are always double-quoted in generated SQL, so PostgreSQL accepts any name
+* that fits the identifier limit (for example `my-tenant`). Quotes, backslashes, `$` and
+* control characters are rejected so the name can never break out of a quoted identifier,
+* a string literal, or a dollar-quoted block.
+*/
+function parseSchemaName(name, kind = "schema name") {
+	if (typeof name !== "string" || name.length === 0 || UNSAFE_SCHEMA_NAME_CHARS.test(name) || Buffer.byteLength(name, "utf-8") > POSTGRES_IDENTIFIER_MAX_BYTES) throw new Error(`Invalid ${kind}: ${name}. Must be 1-${POSTGRES_IDENTIFIER_MAX_BYTES} bytes long and must not contain quotes, backslashes, "$", or control characters.`);
+	return name;
+}
+/**
+* Turns a schema name into a string that is safe to use as part of an unquoted identifier,
+* such as the schema prefix of an index or constraint name.
+*
+* Names that are already plain identifiers (letters, digits, underscores) are returned
+* unchanged, so existing index and constraint names keep working. Any other character
+* becomes `_`, e.g. `my-tenant` -> `my_tenant`.
+*/
+function schemaNamePrefix(name) {
+	const sanitized = parseSchemaName(name).replace(/[^A-Za-z0-9_]/g, "_");
+	return /^[0-9]/.test(sanitized) ? `_${sanitized}` : sanitized;
+}
+//#endregion
 //#region src/vector/filter.ts
 /**
 * Translates MongoDB-style filters to PG compatible filters.
@@ -718,7 +746,8 @@ var PgVector = class extends MastraVector {
 		if (vectorType === "bit") return dimension ? `bit(${dimension})` : "bit";
 		if (this.vectorExtensionSchema) {
 			if (this.vectorExtensionSchema === "pg_catalog") return vectorType;
-			return `${parseSqlIdentifier(this.vectorExtensionSchema, "vector extension schema")}.${vectorType}`;
+			const extensionSchema = parseSchemaName(this.vectorExtensionSchema, "vector extension schema");
+			return `${/^[A-Za-z_][A-Za-z0-9_]*$/.test(extensionSchema) ? extensionSchema : `"${extensionSchema}"`}.${vectorType}`;
 		}
 		return vectorType;
 	}
@@ -796,7 +825,7 @@ var PgVector = class extends MastraVector {
 		};
 	}
 	getSchemaName() {
-		return this.schema ? `"${parseSqlIdentifier(this.schema, "schema name")}"` : void 0;
+		return this.schema ? `"${parseSchemaName(this.schema)}"` : void 0;
 	}
 	async ensureNamespaceSchema(indexName, client) {
 		const { tableName, parsedIndexName } = this.getTableName(indexName);
@@ -2551,7 +2580,7 @@ function resolvePgConfig(config) {
 	};
 }
 function getSchemaName$6(schema) {
-	return schema ? `"${parseSqlIdentifier(schema, "schema name")}"` : "\"public\"";
+	return schema ? `"${parseSchemaName(schema)}"` : "\"public\"";
 }
 function getTableName$6({ indexName, schemaName }) {
 	const quotedIndexName = `"${parseSqlIdentifier(indexName, "index name")}"`;
@@ -2590,7 +2619,7 @@ function generateTableSQL({ tableName, schema, schemaName, compositePrimaryKey,
 		...timeZColumns,
 		...tableConstraints
 	].join(",\n");
-	const parsedSchemaName = schemaName ? parseSqlIdentifier(schemaName, "schema name") : "";
+	const parsedSchemaName = schemaName ? schemaNamePrefix(schemaName) : "";
 	const workflowSnapshotConstraint = buildConstraintName({
 		baseName: "mastra_workflow_snapshot_workflow_name_run_id_key",
 		schemaName: parsedSchemaName || void 0
@@ -2600,7 +2629,7 @@ function generateTableSQL({ tableName, schema, schemaName, compositePrimaryKey,
 		schemaName: parsedSchemaName || void 0
 	});
 	const quotedSchemaName = getSchemaName$6(schemaName);
-	const schemaFilter = parsedSchemaName || "public";
+	const schemaFilter = schemaName ? parseSchemaName(schemaName) : "public";
 	return `
             CREATE TABLE IF NOT EXISTS ${getTableName$6({
 		indexName: tableName,
@@ -2726,7 +2755,7 @@ BEGIN
         JOIN pg_catalog.pg_namespace n ON n.oid = c.relnamespace
         WHERE tg.tgname = ${`'${parsedTriggerName}'`}
           AND c.relname = ${`'${parseSqlIdentifier(tableName, "table name")}'`}
-          AND n.nspname = ${schemaName ? `'${parseSqlIdentifier(schemaName, "schema name")}'` : `'public'`}
+          AND n.nspname = ${schemaName ? `'${parseSchemaName(schemaName)}'` : `'public'`}
           AND NOT tg.tgisinternal
           AND tg.tgtype = 23
           AND tg.tgfoid = '${functionName}()'::regprocedure
@@ -2795,7 +2824,7 @@ var PgDB = class extends MastraBase {
 		if (tableName === TABLE_WORKFLOW_SNAPSHOT) {
 			const constraintName = buildConstraintName({
 				baseName: "mastra_workflow_snapshot_workflow_name_run_id_key",
-				schemaName: this.schemaName ? parseSqlIdentifier(this.schemaName, "schema name") : void 0
+				schemaName: this.schemaName ? schemaNamePrefix(this.schemaName) : void 0
 			}).toLowerCase();
 			return snapshot.indexes.has(constraintName) && snapshot.replicaIdentityIndexes.has(constraintName);
 		}
@@ -3400,7 +3429,7 @@ MIGRATION REQUIRED: Duplicate spans detected in ${duplicateInfo.tableName}\n====
 	async spansPrimaryKeyExists() {
 		const constraintName = buildConstraintName({
 			baseName: "mastra_ai_spans_traceid_spanid_pk",
-			schemaName: (this.schemaName ? parseSqlIdentifier(this.schemaName, "schema name") : "") || void 0
+			schemaName: (this.schemaName ? schemaNamePrefix(this.schemaName) : "") || void 0
 		});
 		const schemaFilter = this.schemaName || "public";
 		const snapshot = this.schemaSnapshot;
@@ -3422,7 +3451,7 @@ MIGRATION REQUIRED: Duplicate spans detected in ${duplicateInfo.tableName}\n====
 		});
 		const constraintName = buildConstraintName({
 			baseName: "mastra_ai_spans_traceid_spanid_pk",
-			schemaName: (this.schemaName ? parseSqlIdentifier(this.schemaName, "schema name") : "") || void 0
+			schemaName: (this.schemaName ? schemaNamePrefix(this.schemaName) : "") || void 0
 		});
 		const schemaFilter = this.schemaName || "public";
 		try {
@@ -4015,7 +4044,7 @@ MIGRATION REQUIRED: Duplicate spans detected in ${duplicateInfo.tableName}\n====
 //#endregion
 //#region src/storage/domains/utils.ts
 function getSchemaName$5(schema) {
-	return schema ? `"${parseSqlIdentifier(schema, "schema name")}"` : void 0;
+	return schema ? `"${parseSchemaName(schema)}"` : void 0;
 }
 function getTableName$5({ indexName, schemaName }) {
 	const quotedIndexName = `"${parseSqlIdentifier(indexName, "index name")}"`;
@@ -5069,7 +5098,7 @@ var BackgroundTasksPG = class BackgroundTasksPG extends BackgroundTasksStorage {
 	* so its supporting index is not part of the default index set.
 	*/
 	async ensureRetentionIndexes(policies) {
-		const prefix = this.#schema !== "public" ? `${this.#schema}_` : "";
+		const prefix = this.#schema !== "public" ? `${schemaNamePrefix(this.#schema)}_` : "";
 		for (const [key, entry] of Object.entries(BackgroundTasksPG.retentionTables)) {
 			if (!entry.indexed || !policies[key]) continue;
 			try {
@@ -5124,7 +5153,7 @@ var BackgroundTasksPG = class BackgroundTasksPG extends BackgroundTasksStorage {
 	}
 	static getExportDDL(schemaName) {
 		const statements = [];
-		const parsedSchema = schemaName ? parseSqlIdentifier(schemaName, "schema name") : "";
+		const parsedSchema = schemaName ? schemaNamePrefix(schemaName) : "";
 		const schemaPrefix = parsedSchema && parsedSchema !== "public" ? `${parsedSchema}_` : "";
 		statements.push(generateTableSQL({
 			tableName: TABLE_BACKGROUND_TASKS,
@@ -5136,7 +5165,7 @@ var BackgroundTasksPG = class BackgroundTasksPG extends BackgroundTasksStorage {
 		return statements;
 	}
 	getDefaultIndexDefinitions() {
-		const schemaPrefix = this.#schema !== "public" ? `${this.#schema}_` : "";
+		const schemaPrefix = this.#schema !== "public" ? `${schemaNamePrefix(this.#schema)}_` : "";
 		return BackgroundTasksPG.getDefaultIndexDefs(schemaPrefix);
 	}
 	async createDefaultIndexes() {
@@ -5498,7 +5527,7 @@ var ChannelsPG = class ChannelsPG extends ChannelsStorage {
 	}
 	static getExportDDL(schemaName) {
 		const statements = [];
-		const parsedSchema = schemaName ? parseSqlIdentifier(schemaName, "schema name") : "";
+		const parsedSchema = schemaName ? schemaNamePrefix(schemaName) : "";
 		const schemaPrefix = parsedSchema && parsedSchema !== "public" ? `${parsedSchema}_` : "";
 		for (const tableName of ChannelsPG.MANAGED_TABLES) statements.push(generateTableSQL({
 			tableName,
@@ -5510,7 +5539,7 @@ var ChannelsPG = class ChannelsPG extends ChannelsStorage {
 		return statements;
 	}
 	getDefaultIndexDefinitions() {
-		const schemaPrefix = this.#schema !== "public" ? `${this.#schema}_` : "";
+		const schemaPrefix = this.#schema !== "public" ? `${schemaNamePrefix(this.#schema)}_` : "";
 		return ChannelsPG.getDefaultIndexDefs(schemaPrefix);
 	}
 	async createDefaultIndexes() {
@@ -7022,7 +7051,7 @@ var ExperimentsPG = class ExperimentsPG extends ExperimentsStorage {
 	* so its supporting index is not part of the default index set.
 	*/
 	async ensureRetentionIndexes(policies) {
-		const prefix = this.#schema !== "public" ? `${this.#schema}_` : "";
+		const prefix = this.#schema !== "public" ? `${schemaNamePrefix(this.#schema)}_` : "";
 		for (const [key, entry] of Object.entries(ExperimentsPG.retentionTables)) {
 			if (!entry.indexed || !policies[key]) continue;
 			try {
@@ -8190,7 +8219,7 @@ function postgresSql(sql, schemaName) {
 		return transformed;
 	});
 	if (schemaName) {
-		const quotedSchema = `"${parseSqlIdentifier(schemaName, "schema name")}"`;
+		const quotedSchema = `"${parseSchemaName(schemaName)}"`;
 		normalized = transformSqlCode(normalized, (code) => {
 			let transformed = code;
 			for (const table of [
@@ -8303,7 +8332,7 @@ function parseOutbox(row) {
 function knowledgeIndexes(schemaName) {
 	const table = (name) => {
 		const quotedName = `"${parseSqlIdentifier(name, "table name")}"`;
-		return schemaName ? `"${parseSqlIdentifier(schemaName, "schema name")}".${quotedName}` : quotedName;
+		return schemaName ? `"${parseSchemaName(schemaName)}".${quotedName}` : quotedName;
 	};
 	return [
 		{
@@ -9207,7 +9236,7 @@ var MCPClientsPG = class MCPClientsPG extends MCPClientsStorage {
 	}
 	static getExportDDL(schemaName) {
 		const statements = [];
-		const parsedSchema = schemaName ? parseSqlIdentifier(schemaName, "schema name") : "";
+		const parsedSchema = schemaName ? schemaNamePrefix(schemaName) : "";
 		const schemaPrefix = parsedSchema && parsedSchema !== "public" ? `${parsedSchema}_` : "";
 		for (const tableName of MCPClientsPG.MANAGED_TABLES) statements.push(generateTableSQL({
 			tableName,
@@ -9219,7 +9248,7 @@ var MCPClientsPG = class MCPClientsPG extends MCPClientsStorage {
 		return statements;
 	}
 	getDefaultIndexDefinitions() {
-		const schemaPrefix = this.#schema !== "public" ? `${this.#schema}_` : "";
+		const schemaPrefix = this.#schema !== "public" ? `${schemaNamePrefix(this.#schema)}_` : "";
 		return MCPClientsPG.getDefaultIndexDefs(schemaPrefix);
 	}
 	async createDefaultIndexes() {
@@ -9775,7 +9804,7 @@ var MCPServersPG = class MCPServersPG extends MCPServersStorage {
 	}
 	static getExportDDL(schemaName) {
 		const statements = [];
-		const parsedSchema = schemaName ? parseSqlIdentifier(schemaName, "schema name") : "";
+		const parsedSchema = schemaName ? schemaNamePrefix(schemaName) : "";
 		const schemaPrefix = parsedSchema && parsedSchema !== "public" ? `${parsedSchema}_` : "";
 		for (const tableName of MCPServersPG.MANAGED_TABLES) statements.push(generateTableSQL({
 			tableName,
@@ -9787,7 +9816,7 @@ var MCPServersPG = class MCPServersPG extends MCPServersStorage {
 		return statements;
 	}
 	getDefaultIndexDefinitions() {
-		const schemaPrefix = this.#schema !== "public" ? `${this.#schema}_` : "";
+		const schemaPrefix = this.#schema !== "public" ? `${schemaNamePrefix(this.#schema)}_` : "";
 		return MCPServersPG.getDefaultIndexDefs(schemaPrefix);
 	}
 	async createDefaultIndexes() {
@@ -10497,7 +10526,7 @@ var MemoryPG = class MemoryPG extends MemoryStorage {
 	* so its supporting index is not part of the default index set.
 	*/
 	async ensureRetentionIndexes(policies) {
-		const prefix = this.#schema && this.#schema !== "public" ? `${this.#schema}_` : "";
+		const prefix = this.#schema && this.#schema !== "public" ? `${schemaNamePrefix(this.#schema)}_` : "";
 		for (const [key, entry] of Object.entries(MemoryPG.retentionTables)) {
 			if (!entry.indexed || !policies[key]) continue;
 			try {
@@ -10532,7 +10561,7 @@ var MemoryPG = class MemoryPG extends MemoryStorage {
 	*/
 	static getExportDDL(schemaName) {
 		const statements = [];
-		const parsedSchema = schemaName ? parseSqlIdentifier(schemaName, "schema name") : "";
+		const parsedSchema = schemaName ? schemaNamePrefix(schemaName) : "";
 		const schemaPrefix = parsedSchema && parsedSchema !== "public" ? `${parsedSchema}_` : "";
 		const quotedSchemaName = getSchemaName$6(schemaName);
 		for (const tableName of [
@@ -10567,7 +10596,7 @@ var MemoryPG = class MemoryPG extends MemoryStorage {
 	* Returns default index definitions for this instance's schema.
 	*/
 	getDefaultIndexDefinitions() {
-		const schemaPrefix = this.#schema !== "public" ? `${this.#schema}_` : "";
+		const schemaPrefix = this.#schema !== "public" ? `${schemaNamePrefix(this.#schema)}_` : "";
 		return MemoryPG.getDefaultIndexDefs(schemaPrefix);
 	}
 	/**
@@ -12975,7 +13004,7 @@ var NotificationsPG = class NotificationsPG extends NotificationsStorage {
 	* so its supporting index is not part of the default index set.
 	*/
 	async ensureRetentionIndexes(policies) {
-		const prefix = this.#schema && this.#schema !== "public" ? `${this.#schema}_` : "";
+		const prefix = this.#schema && this.#schema !== "public" ? `${schemaNamePrefix(this.#schema)}_` : "";
 		for (const [key, entry] of Object.entries(NotificationsPG.retentionTables)) {
 			if (!entry.indexed || !policies[key]) continue;
 			try {
@@ -13042,7 +13071,7 @@ var NotificationsPG = class NotificationsPG extends NotificationsStorage {
 	}
 	static getExportDDL(schemaName) {
 		const statements = [];
-		const parsedSchema = schemaName ? parseSqlIdentifier(schemaName, "schema name") : "";
+		const parsedSchema = schemaName ? schemaNamePrefix(schemaName) : "";
 		const schemaPrefix = parsedSchema && parsedSchema !== "public" ? `${parsedSchema}_` : "";
 		statements.push(generateTableSQL({
 			tableName: TABLE_NOTIFICATIONS,
@@ -13054,7 +13083,7 @@ var NotificationsPG = class NotificationsPG extends NotificationsStorage {
 		return statements;
 	}
 	getDefaultIndexDefinitions() {
-		const schemaPrefix = this.#schema !== "public" ? `${this.#schema}_` : "";
+		const schemaPrefix = this.#schema !== "public" ? `${schemaNamePrefix(this.#schema)}_` : "";
 		return NotificationsPG.getDefaultIndexDefs(schemaPrefix);
 	}
 	async createDefaultIndexes() {
@@ -13351,7 +13380,7 @@ var ObservabilityPG = class ObservabilityPG extends ObservabilityStorage {
 	* so its supporting index is not part of the default index set.
 	*/
 	async ensureRetentionIndexes(policies) {
-		const prefix = this.#schema !== "public" ? `${this.#schema}_` : "";
+		const prefix = this.#schema !== "public" ? `${schemaNamePrefix(this.#schema)}_` : "";
 		for (const [key, entry] of Object.entries(ObservabilityPG.retentionTables)) {
 			if (!entry.indexed || !policies[key]) continue;
 			try {
@@ -13432,7 +13461,7 @@ var ObservabilityPG = class ObservabilityPG extends ObservabilityStorage {
 	*/
 	static getExportDDL(schemaName) {
 		const statements = [];
-		const parsedSchema = schemaName ? parseSqlIdentifier(schemaName, "schema name") : "";
+		const parsedSchema = schemaName ? schemaNamePrefix(schemaName) : "";
 		const schemaPrefix = parsedSchema && parsedSchema !== "public" ? `${parsedSchema}_` : "";
 		statements.push(generateTableSQL({
 			tableName: TABLE_SPANS,
@@ -13448,7 +13477,7 @@ var ObservabilityPG = class ObservabilityPG extends ObservabilityStorage {
 	* Returns default index definitions for this instance's schema.
 	*/
 	getDefaultIndexDefinitions() {
-		const schemaPrefix = this.#schema !== "public" ? `${this.#schema}_` : "";
+		const schemaPrefix = this.#schema !== "public" ? `${schemaNamePrefix(this.#schema)}_` : "";
 		return ObservabilityPG.getDefaultIndexDefs(schemaPrefix);
 	}
 	/**
@@ -14659,15 +14688,15 @@ const SIGNAL_TIME_COLUMN = {
 };
 /** Returns a fully-qualified, double-quoted table name. */
 function qualifiedTable(schema, table) {
-	return `"${parseSqlIdentifier(schema, "schema name")}"."${parseSqlIdentifier(table, "table name")}"`;
+	return `"${parseSchemaName(schema)}"."${parseSqlIdentifier(table, "table name")}"`;
 }
 /** Returns a parsed, quoted, schema-prefixed object name (constraint, index, etc.). */
 function qualifiedName(schema, name) {
-	return `"${parseSqlIdentifier(schema, "schema name")}"."${parseSqlIdentifier(name, "object name")}"`;
+	return `"${parseSchemaName(schema)}"."${parseSqlIdentifier(name, "object name")}"`;
 }
 /** Schema CREATE. Safe to run repeatedly before table DDL. */
 function schemaDDL(schema) {
-	return `CREATE SCHEMA IF NOT EXISTS "${parseSqlIdentifier(schema, "schema name")}"`;
+	return `CREATE SCHEMA IF NOT EXISTS "${parseSchemaName(schema)}"`;
 }
 /**
 * Postgres declarative partitioning and Timescale hypertables are mutually
@@ -19207,7 +19236,7 @@ var PromptBlocksPG = class PromptBlocksPG extends PromptBlocksStorage {
 	*/
 	static getExportDDL(schemaName) {
 		const statements = [];
-		const parsedSchema = schemaName ? parseSqlIdentifier(schemaName, "schema name") : "";
+		const parsedSchema = schemaName ? schemaNamePrefix(schemaName) : "";
 		const schemaPrefix = parsedSchema && parsedSchema !== "public" ? `${parsedSchema}_` : "";
 		for (const tableName of PromptBlocksPG.MANAGED_TABLES) statements.push(generateTableSQL({
 			tableName,
@@ -19219,7 +19248,7 @@ var PromptBlocksPG = class PromptBlocksPG extends PromptBlocksStorage {
 		return statements;
 	}
 	getDefaultIndexDefinitions() {
-		const schemaPrefix = this.#schema !== "public" ? `${this.#schema}_` : "";
+		const schemaPrefix = this.#schema !== "public" ? `${schemaNamePrefix(this.#schema)}_` : "";
 		return PromptBlocksPG.getDefaultIndexDefs(schemaPrefix);
 	}
 	async createDefaultIndexes() {
@@ -19854,7 +19883,7 @@ var SchedulesPG = class SchedulesPG extends SchedulesStorage {
 	* so its supporting index is not part of the default index set.
 	*/
 	async ensureRetentionIndexes(policies) {
-		const prefix = this.#schema !== "public" ? `${this.#schema}_` : "";
+		const prefix = this.#schema !== "public" ? `${schemaNamePrefix(this.#schema)}_` : "";
 		for (const [key, entry] of Object.entries(SchedulesPG.retentionTables)) {
 			if (!entry.indexed || !policies[key]) continue;
 			try {
@@ -19902,7 +19931,7 @@ var SchedulesPG = class SchedulesPG extends SchedulesStorage {
 		}];
 	}
 	getDefaultIndexDefinitions() {
-		const schemaPrefix = this.#schema !== "public" ? `${this.#schema}_` : "";
+		const schemaPrefix = this.#schema !== "public" ? `${schemaNamePrefix(this.#schema)}_` : "";
 		return SchedulesPG.getDefaultIndexDefs(schemaPrefix);
 	}
 	async createDefaultIndexes() {
@@ -19923,7 +19952,7 @@ var SchedulesPG = class SchedulesPG extends SchedulesStorage {
 	}
 	static getExportDDL(schemaName) {
 		const statements = [];
-		const parsedSchema = schemaName ? parseSqlIdentifier(schemaName, "schema name") : "";
+		const parsedSchema = schemaName ? schemaNamePrefix(schemaName) : "";
 		const schemaPrefix = parsedSchema && parsedSchema !== "public" ? `${parsedSchema}_` : "";
 		statements.push(generateTableSQL({
 			tableName: TABLE_SCHEDULES,
@@ -19945,7 +19974,7 @@ var SchedulesPG = class SchedulesPG extends SchedulesStorage {
 		await this.#db.clearTable({ tableName: TABLE_SCHEDULES });
 	}
 	#table(tableName) {
-		return getTableName$2(tableName, getSchemaName$2(parseSqlIdentifier(this.#schema, "schema name")));
+		return getTableName$2(tableName, getSchemaName$2(parseSchemaName(this.#schema)));
 	}
 	async createSchedule(schedule) {
 		if (await this.#getSchedule(this.#client, schedule.id)) throw new Error(`Schedule with id "${schedule.id}" already exists`);
@@ -20150,7 +20179,7 @@ var ScorerDefinitionsPG = class ScorerDefinitionsPG extends ScorerDefinitionsSto
 	*/
 	static getExportDDL(schemaName) {
 		const statements = [];
-		const parsedSchema = schemaName ? parseSqlIdentifier(schemaName, "schema name") : "";
+		const parsedSchema = schemaName ? schemaNamePrefix(schemaName) : "";
 		const schemaPrefix = parsedSchema && parsedSchema !== "public" ? `${parsedSchema}_` : "";
 		for (const tableName of ScorerDefinitionsPG.MANAGED_TABLES) statements.push(generateTableSQL({
 			tableName,
@@ -20162,7 +20191,7 @@ var ScorerDefinitionsPG = class ScorerDefinitionsPG extends ScorerDefinitionsSto
 		return statements;
 	}
 	getDefaultIndexDefinitions() {
-		const schemaPrefix = this.#schema !== "public" ? `${this.#schema}_` : "";
+		const schemaPrefix = this.#schema !== "public" ? `${schemaNamePrefix(this.#schema)}_` : "";
 		return ScorerDefinitionsPG.getDefaultIndexDefs(schemaPrefix);
 	}
 	async createDefaultIndexes() {
@@ -20795,7 +20824,7 @@ var ScoresPG = class ScoresPG extends ScoresStorage {
 	* so its supporting index is not part of the default index set.
 	*/
 	async ensureRetentionIndexes(policies) {
-		const prefix = this.#schema && this.#schema !== "public" ? `${this.#schema}_` : "";
+		const prefix = this.#schema && this.#schema !== "public" ? `${schemaNamePrefix(this.#schema)}_` : "";
 		for (const [key, entry] of Object.entries(ScoresPG.retentionTables)) {
 			if (!entry.indexed || !policies[key]) continue;
 			try {
@@ -20830,7 +20859,7 @@ var ScoresPG = class ScoresPG extends ScoresStorage {
 	*/
 	static getExportDDL(schemaName) {
 		const statements = [];
-		const parsedSchema = schemaName ? parseSqlIdentifier(schemaName, "schema name") : "";
+		const parsedSchema = schemaName ? schemaNamePrefix(schemaName) : "";
 		const schemaPrefix = parsedSchema && parsedSchema !== "public" ? `${parsedSchema}_` : "";
 		statements.push(generateTableSQL({
 			tableName: TABLE_SCORERS,
@@ -20845,7 +20874,7 @@ var ScoresPG = class ScoresPG extends ScoresStorage {
 	* Returns default index definitions for this instance's schema.
 	*/
 	getDefaultIndexDefinitions() {
-		const schemaPrefix = this.#schema !== "public" ? `${this.#schema}_` : "";
+		const schemaPrefix = this.#schema !== "public" ? `${schemaNamePrefix(this.#schema)}_` : "";
 		return ScoresPG.getDefaultIndexDefs(schemaPrefix);
 	}
 	/**
@@ -21208,7 +21237,7 @@ var SkillsPG = class SkillsPG extends SkillsStorage {
 	}
 	static getExportDDL(schemaName) {
 		const statements = [];
-		const parsedSchema = schemaName ? parseSqlIdentifier(schemaName, "schema name") : "";
+		const parsedSchema = schemaName ? schemaNamePrefix(schemaName) : "";
 		const schemaPrefix = parsedSchema && parsedSchema !== "public" ? `${parsedSchema}_` : "";
 		for (const tableName of SkillsPG.MANAGED_TABLES) statements.push(generateTableSQL({
 			tableName,
@@ -21220,7 +21249,7 @@ var SkillsPG = class SkillsPG extends SkillsStorage {
 		return statements;
 	}
 	getDefaultIndexDefinitions() {
-		const schemaPrefix = this.#schema !== "public" ? `${this.#schema}_` : "";
+		const schemaPrefix = this.#schema !== "public" ? `${schemaNamePrefix(this.#schema)}_` : "";
 		return SkillsPG.getDefaultIndexDefs(schemaPrefix);
 	}
 	async createDefaultIndexes() {
@@ -21915,7 +21944,7 @@ var ThreadStatePG = class ThreadStatePG extends ThreadStateStorage {
 	* a failure here leaves pruning correct, just slower.
 	*/
 	async #ensureRetentionIndexes(policies) {
-		const prefix = this.#schema && this.#schema !== "public" ? `${this.#schema}_` : "";
+		const prefix = this.#schema && this.#schema !== "public" ? `${schemaNamePrefix(this.#schema)}_` : "";
 		for (const [key, entry] of Object.entries(ThreadStatePG.retentionTables)) {
 			if (!entry.indexed || !policies[key]) continue;
 			try {
@@ -22081,7 +22110,7 @@ var ToolProviderConnectionsPG = class ToolProviderConnectionsPG extends ToolProv
 	}
 	static getExportDDL(schemaName) {
 		const statements = [];
-		const parsedSchema = schemaName ? parseSqlIdentifier(schemaName, "schema name") : "";
+		const parsedSchema = schemaName ? schemaNamePrefix(schemaName) : "";
 		const schemaPrefix = parsedSchema && parsedSchema !== "public" ? `${parsedSchema}_` : "";
 		statements.push(generateTableSQL({
 			tableName: TABLE_TOOL_PROVIDER_CONNECTIONS,
@@ -22098,7 +22127,7 @@ var ToolProviderConnectionsPG = class ToolProviderConnectionsPG extends ToolProv
 		return statements;
 	}
 	getDefaultIndexDefinitions() {
-		const schemaPrefix = this.#schema !== "public" ? `${this.#schema}_` : "";
+		const schemaPrefix = this.#schema !== "public" ? `${schemaNamePrefix(this.#schema)}_` : "";
 		return ToolProviderConnectionsPG.getDefaultIndexDefs(schemaPrefix);
 	}
 	async createDefaultIndexes() {
@@ -22330,7 +22359,7 @@ var WorkflowDefinitionsPG = class WorkflowDefinitionsPG extends WorkflowDefiniti
 	}
 	getDefaultIndexDefinitions() {
 		return [{
-			name: `${this.#schema !== "public" ? `${this.#schema}_` : ""}idx_workflow_definitions_status`,
+			name: `${this.#schema !== "public" ? `${schemaNamePrefix(this.#schema)}_` : ""}idx_workflow_definitions_status`,
 			table: TABLE_WORKFLOW_DEFINITIONS,
 			columns: ["status"]
 		}];
@@ -22607,7 +22636,7 @@ var WorkflowsPG = class WorkflowsPG extends WorkflowsStorage {
 	*/
 	static getExportDDL(schemaName) {
 		const statements = [];
-		const parsedSchema = schemaName ? parseSqlIdentifier(schemaName, "schema name") : "";
+		const parsedSchema = schemaName ? schemaNamePrefix(schemaName) : "";
 		const schemaPrefix = parsedSchema && parsedSchema !== "public" ? `${parsedSchema}_` : "";
 		statements.push(generateTableSQL({
 			tableName: TABLE_WORKFLOW_SNAPSHOT,
@@ -22624,7 +22653,7 @@ var WorkflowsPG = class WorkflowsPG extends WorkflowsStorage {
 	* Returns default index definitions for the workflows domain tables.
 	*/
 	getDefaultIndexDefinitions() {
-		const schemaPrefix = this.#schema !== "public" ? `${this.#schema}_` : "";
+		const schemaPrefix = this.#schema !== "public" ? `${schemaNamePrefix(this.#schema)}_` : "";
 		return WorkflowsPG.getDefaultIndexDefs(schemaPrefix);
 	}
 	/**
@@ -22674,7 +22703,7 @@ var WorkflowsPG = class WorkflowsPG extends WorkflowsStorage {
 	* so its supporting index is not part of the default index set.
 	*/
 	async ensureRetentionIndexes(policies) {
-		const prefix = this.#schema && this.#schema !== "public" ? `${this.#schema}_` : "";
+		const prefix = this.#schema && this.#schema !== "public" ? `${schemaNamePrefix(this.#schema)}_` : "";
 		for (const [key, entry] of Object.entries(WorkflowsPG.retentionTables)) {
 			if (!entry.indexed || !policies[key]) continue;
 			try {
@@ -23052,7 +23081,7 @@ var WorkspacesPG = class WorkspacesPG extends WorkspacesStorage {
 	}
 	static getExportDDL(schemaName) {
 		const statements = [];
-		const parsedSchema = schemaName ? parseSqlIdentifier(schemaName, "schema name") : "";
+		const parsedSchema = schemaName ? schemaNamePrefix(schemaName) : "";
 		const schemaPrefix = parsedSchema && parsedSchema !== "public" ? `${parsedSchema}_` : "";
 		for (const tableName of WorkspacesPG.MANAGED_TABLES) statements.push(generateTableSQL({
 			tableName,
@@ -23064,7 +23093,7 @@ var WorkspacesPG = class WorkspacesPG extends WorkspacesStorage {
 		return statements;
 	}
 	getDefaultIndexDefinitions() {
-		const schemaPrefix = this.#schema !== "public" ? `${this.#schema}_` : "";
+		const schemaPrefix = this.#schema !== "public" ? `${schemaNamePrefix(this.#schema)}_` : "";
 		return WorkspacesPG.getDefaultIndexDefs(schemaPrefix);
 	}
 	async createDefaultIndexes() {
@@ -24192,7 +24221,7 @@ var PostgresStore = class extends MastraCompositeStore {
 				disableInit: config.disableInit,
 				retention: config.retention
 			});
-			this.schema = parseSqlIdentifier(config.schemaName || "public", "schema name");
+			this.schema = parseSchemaName(config.schemaName || "public");
 			if (isPoolConfig(config)) {
 				this.#writePool = config.pool;
 				this.#ownsWritePool = false;