@mastra/pg 0.13.0 → 0.13.1

package/dist/index.js

@@ -2017,7 +2017,7 @@ var StoreOperationsPG = class extends StoreOperations {
           const schemaExists = await this.client.oneOrNone(
             `
                 SELECT EXISTS (
-                  SELECT 1 FROM information_schema.schemata 
+                  SELECT 1 FROM information_schema.schemata
                   WHERE schema_name = $1
                 )
               `,
@@ -2130,6 +2130,7 @@ var StoreOperationsPG = class extends StoreOperations {
         await this.setupSchema();
       }
       const finalColumns = [...columns, ...timeZColumns].join(",\n");
+      const constraintPrefix = this.schemaName ? `${this.schemaName}_` : "";
       const sql = `
             CREATE TABLE IF NOT EXISTS ${getTableName({ indexName: tableName, schemaName: getSchemaName(this.schemaName) })} (
               ${finalColumns}
@@ -2137,10 +2138,10 @@ var StoreOperationsPG = class extends StoreOperations {
             ${tableName === TABLE_WORKFLOW_SNAPSHOT ? `
             DO $$ BEGIN
               IF NOT EXISTS (
-                SELECT 1 FROM pg_constraint WHERE conname = 'mastra_workflow_snapshot_workflow_name_run_id_key'
+                SELECT 1 FROM pg_constraint WHERE conname = '${constraintPrefix}mastra_workflow_snapshot_workflow_name_run_id_key'
               ) THEN
                 ALTER TABLE ${getTableName({ indexName: tableName, schemaName: getSchemaName(this.schemaName) })}
-                ADD CONSTRAINT mastra_workflow_snapshot_workflow_name_run_id_key
+                ADD CONSTRAINT ${constraintPrefix}mastra_workflow_snapshot_workflow_name_run_id_key
                 UNIQUE (workflow_name, run_id);
               END IF;
             END $$;
@@ -2307,14 +2308,23 @@ function transformScoreRow(row) {
 var ScoresPG = class extends ScoresStorage {
   client;
   operations;
-  constructor({ client, operations }) {
+  schema;
+  constructor({
+    client,
+    operations,
+    schema
+  }) {
     super();
     this.client = client;
     this.operations = operations;
+    this.schema = schema;
   }
   async getScoreById({ id }) {
     try {
-      const result = await this.client.oneOrNone(`SELECT * FROM ${TABLE_SCORERS} WHERE id = $1`, [id]);
+      const result = await this.client.oneOrNone(
+        `SELECT * FROM ${getTableName({ indexName: TABLE_SCORERS, schemaName: this.schema })} WHERE id = $1`,
+        [id]
+      );
       return transformScoreRow(result);
     } catch (error) {
       throw new MastraError(
@@ -2333,7 +2343,7 @@ var ScoresPG = class extends ScoresStorage {
   }) {
     try {
       const total = await this.client.oneOrNone(
-        `SELECT COUNT(*) FROM ${TABLE_SCORERS} WHERE "scorerId" = $1`,
+        `SELECT COUNT(*) FROM ${getTableName({ indexName: TABLE_SCORERS, schemaName: this.schema })} WHERE "scorerId" = $1`,
         [scorerId]
       );
       if (total?.count === "0" || !total?.count) {
@@ -2348,7 +2358,7 @@ var ScoresPG = class extends ScoresStorage {
         };
       }
       const result = await this.client.manyOrNone(
-        `SELECT * FROM ${TABLE_SCORERS} WHERE "scorerId" = $1 LIMIT $2 OFFSET $3`,
+        `SELECT * FROM ${getTableName({ indexName: TABLE_SCORERS, schemaName: this.schema })} WHERE "scorerId" = $1 LIMIT $2 OFFSET $3`,
         [scorerId, pagination.perPage, pagination.page * pagination.perPage]
       );
       return {
@@ -2402,7 +2412,7 @@ var ScoresPG = class extends ScoresStorage {
   }) {
     try {
       const total = await this.client.oneOrNone(
-        `SELECT COUNT(*) FROM ${TABLE_SCORERS} WHERE "runId" = $1`,
+        `SELECT COUNT(*) FROM ${getTableName({ indexName: TABLE_SCORERS, schemaName: this.schema })} WHERE "runId" = $1`,
         [runId]
       );
       console.log(`total: ${total?.count}`);
@@ -2419,7 +2429,7 @@ var ScoresPG = class extends ScoresStorage {
         };
       }
       const result = await this.client.manyOrNone(
-        `SELECT * FROM ${TABLE_SCORERS} WHERE "runId" = $1 LIMIT $2 OFFSET $3`,
+        `SELECT * FROM ${getTableName({ indexName: TABLE_SCORERS, schemaName: this.schema })} WHERE "runId" = $1 LIMIT $2 OFFSET $3`,
         [runId, pagination.perPage, pagination.page * pagination.perPage]
       );
       return {
@@ -2449,7 +2459,7 @@ var ScoresPG = class extends ScoresStorage {
   }) {
     try {
       const total = await this.client.oneOrNone(
-        `SELECT COUNT(*) FROM ${TABLE_SCORERS} WHERE "entityId" = $1 AND "entityType" = $2`,
+        `SELECT COUNT(*) FROM ${getTableName({ indexName: TABLE_SCORERS, schemaName: this.schema })} WHERE "entityId" = $1 AND "entityType" = $2`,
         [entityId, entityType]
       );
       if (total?.count === "0" || !total?.count) {
@@ -2464,7 +2474,7 @@ var ScoresPG = class extends ScoresStorage {
         };
       }
       const result = await this.client.manyOrNone(
-        `SELECT * FROM ${TABLE_SCORERS} WHERE "entityId" = $1 AND "entityType" = $2 LIMIT $3 OFFSET $4`,
+        `SELECT * FROM ${getTableName({ indexName: TABLE_SCORERS, schemaName: this.schema })} WHERE "entityId" = $1 AND "entityType" = $2 LIMIT $3 OFFSET $4`,
         [entityId, entityType, pagination.perPage, pagination.page * pagination.perPage]
       );
       return {
@@ -2663,7 +2673,7 @@ var WorkflowsPG = class extends WorkflowsStorage {
     try {
       const now = (/* @__PURE__ */ new Date()).toISOString();
       await this.client.none(
-        `INSERT INTO ${TABLE_WORKFLOW_SNAPSHOT} (workflow_name, run_id, snapshot, "createdAt", "updatedAt")
+        `INSERT INTO ${getTableName({ indexName: TABLE_WORKFLOW_SNAPSHOT, schemaName: this.schema })} (workflow_name, run_id, snapshot, "createdAt", "updatedAt")
                  VALUES ($1, $2, $3, $4, $5)
                  ON CONFLICT (workflow_name, run_id) DO UPDATE
                  SET snapshot = $3, "updatedAt" = $5`,
@@ -2721,8 +2731,8 @@ var WorkflowsPG = class extends WorkflowsStorage {
       }
       const whereClause = conditions.length > 0 ? `WHERE ${conditions.join(" AND ")}` : "";
       const query = `
-          SELECT * FROM ${getTableName({ indexName: TABLE_WORKFLOW_SNAPSHOT, schemaName: this.schema })} 
-          ${whereClause} 
+          SELECT * FROM ${getTableName({ indexName: TABLE_WORKFLOW_SNAPSHOT, schemaName: this.schema })}
+          ${whereClause}
         `;
       const queryValues = values;
       const result = await this.client.oneOrNone(query, queryValues);
@@ -2792,8 +2802,8 @@ var WorkflowsPG = class extends WorkflowsStorage {
         total = Number(countResult.count);
       }
       const query = `
-          SELECT * FROM ${getTableName({ indexName: TABLE_WORKFLOW_SNAPSHOT, schemaName: this.schema })} 
-          ${whereClause} 
+          SELECT * FROM ${getTableName({ indexName: TABLE_WORKFLOW_SNAPSHOT, schemaName: this.schema })}
+          ${whereClause}
           ORDER BY "createdAt" DESC
           ${limit !== void 0 && offset !== void 0 ? ` LIMIT $${paramIndex} OFFSET $${paramIndex + 1}` : ""}
         `;
@@ -2859,7 +2869,7 @@ var PostgresStore = class extends MastraStorage {
       );
       this.client = this.db;
       const operations = new StoreOperationsPG({ client: this.client, schemaName: this.schema });
-      const scores = new ScoresPG({ client: this.client, operations });
+      const scores = new ScoresPG({ client: this.client, operations, schema: this.schema });
       const traces = new TracesPG({ client: this.client, operations, schema: this.schema });
       const workflows = new WorkflowsPG({ client: this.client, operations, schema: this.schema });
       const legacyEvals = new LegacyEvalsPG({ client: this.client, schema: this.schema });
@@ -3166,3 +3176,5 @@ Example Complex Query:
 }`;
 
 export { PGVECTOR_PROMPT, PgVector, PostgresStore };
+//# sourceMappingURL=index.js.map
+//# sourceMappingURL=index.js.map