@mastra/pg 0.13.0 → 0.13.1-alpha.0

package/.turbo/turbo-build.log

@@ -1,23 +1,4 @@
 
-> @mastra/pg@0.13.0-alpha.1 build /home/runner/work/mastra/mastra/stores/pg
-> tsup src/index.ts --format esm,cjs --experimental-dts --clean --treeshake=smallest --splitting
+> @mastra/pg@0.13.1-alpha.0 build /home/runner/work/mastra/mastra/stores/pg
+> tsup --silent --config tsup.config.ts
 
-CLI Building entry: src/index.ts
-CLI Using tsconfig: tsconfig.json
-CLI tsup v8.5.0
-TSC Build start
-TSC ⚡️ Build success in 17702ms
-DTS Build start
-CLI Target: es2022
-Analysis will use the bundled TypeScript version 5.8.3
-Writing package typings: /home/runner/work/mastra/mastra/stores/pg/dist/_tsup-dts-rollup.d.ts
-Analysis will use the bundled TypeScript version 5.8.3
-Writing package typings: /home/runner/work/mastra/mastra/stores/pg/dist/_tsup-dts-rollup.d.cts
-DTS ⚡️ Build success in 14252ms
-CLI Cleaning output folder
-ESM Build start
-CJS Build start
-CJS dist/index.cjs 106.34 KB
-CJS ⚡️ Build success in 2113ms
-ESM dist/index.js 104.78 KB
-ESM ⚡️ Build success in 2113ms

package/CHANGELOG.md

@@ -1,5 +1,16 @@
 # @mastra/pg
 
+## 0.13.1-alpha.0
+
+### Patch Changes
+
+- bf6c955: fix: Fix schema support to PostgreSQL scores and workflows storage
+- Updated dependencies [33dcb07]
+- Updated dependencies [d30b1a0]
+- Updated dependencies [bff87f7]
+- Updated dependencies [b4a8df0]
+  - @mastra/core@0.12.1-alpha.0
+
 ## 0.13.0
 
 ### Minor Changes

package/dist/index.cjs

@@ -2025,7 +2025,7 @@ var StoreOperationsPG = class extends storage.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
                 )
               `,
@@ -2138,6 +2138,7 @@ var StoreOperationsPG = class extends storage.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}
@@ -2145,10 +2146,10 @@ var StoreOperationsPG = class extends storage.StoreOperations {
             ${tableName === storage.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 $$;
@@ -2315,14 +2316,23 @@ function transformScoreRow(row) {
 var ScoresPG = class extends storage.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 ${storage.TABLE_SCORERS} WHERE id = $1`, [id]);
+      const result = await this.client.oneOrNone(
+        `SELECT * FROM ${getTableName({ indexName: storage.TABLE_SCORERS, schemaName: this.schema })} WHERE id = $1`,
+        [id]
+      );
       return transformScoreRow(result);
     } catch (error$1) {
       throw new error.MastraError(
@@ -2341,7 +2351,7 @@ var ScoresPG = class extends storage.ScoresStorage {
   }) {
     try {
       const total = await this.client.oneOrNone(
-        `SELECT COUNT(*) FROM ${storage.TABLE_SCORERS} WHERE "scorerId" = $1`,
+        `SELECT COUNT(*) FROM ${getTableName({ indexName: storage.TABLE_SCORERS, schemaName: this.schema })} WHERE "scorerId" = $1`,
         [scorerId]
       );
       if (total?.count === "0" || !total?.count) {
@@ -2356,7 +2366,7 @@ var ScoresPG = class extends storage.ScoresStorage {
         };
       }
       const result = await this.client.manyOrNone(
-        `SELECT * FROM ${storage.TABLE_SCORERS} WHERE "scorerId" = $1 LIMIT $2 OFFSET $3`,
+        `SELECT * FROM ${getTableName({ indexName: storage.TABLE_SCORERS, schemaName: this.schema })} WHERE "scorerId" = $1 LIMIT $2 OFFSET $3`,
         [scorerId, pagination.perPage, pagination.page * pagination.perPage]
       );
       return {
@@ -2410,7 +2420,7 @@ var ScoresPG = class extends storage.ScoresStorage {
   }) {
     try {
       const total = await this.client.oneOrNone(
-        `SELECT COUNT(*) FROM ${storage.TABLE_SCORERS} WHERE "runId" = $1`,
+        `SELECT COUNT(*) FROM ${getTableName({ indexName: storage.TABLE_SCORERS, schemaName: this.schema })} WHERE "runId" = $1`,
         [runId]
       );
       console.log(`total: ${total?.count}`);
@@ -2427,7 +2437,7 @@ var ScoresPG = class extends storage.ScoresStorage {
         };
       }
       const result = await this.client.manyOrNone(
-        `SELECT * FROM ${storage.TABLE_SCORERS} WHERE "runId" = $1 LIMIT $2 OFFSET $3`,
+        `SELECT * FROM ${getTableName({ indexName: storage.TABLE_SCORERS, schemaName: this.schema })} WHERE "runId" = $1 LIMIT $2 OFFSET $3`,
         [runId, pagination.perPage, pagination.page * pagination.perPage]
       );
       return {
@@ -2457,7 +2467,7 @@ var ScoresPG = class extends storage.ScoresStorage {
   }) {
     try {
       const total = await this.client.oneOrNone(
-        `SELECT COUNT(*) FROM ${storage.TABLE_SCORERS} WHERE "entityId" = $1 AND "entityType" = $2`,
+        `SELECT COUNT(*) FROM ${getTableName({ indexName: storage.TABLE_SCORERS, schemaName: this.schema })} WHERE "entityId" = $1 AND "entityType" = $2`,
         [entityId, entityType]
       );
       if (total?.count === "0" || !total?.count) {
@@ -2472,7 +2482,7 @@ var ScoresPG = class extends storage.ScoresStorage {
         };
       }
       const result = await this.client.manyOrNone(
-        `SELECT * FROM ${storage.TABLE_SCORERS} WHERE "entityId" = $1 AND "entityType" = $2 LIMIT $3 OFFSET $4`,
+        `SELECT * FROM ${getTableName({ indexName: storage.TABLE_SCORERS, schemaName: this.schema })} WHERE "entityId" = $1 AND "entityType" = $2 LIMIT $3 OFFSET $4`,
         [entityId, entityType, pagination.perPage, pagination.page * pagination.perPage]
       );
       return {
@@ -2671,7 +2681,7 @@ var WorkflowsPG = class extends storage.WorkflowsStorage {
     try {
       const now = (/* @__PURE__ */ new Date()).toISOString();
       await this.client.none(
-        `INSERT INTO ${storage.TABLE_WORKFLOW_SNAPSHOT} (workflow_name, run_id, snapshot, "createdAt", "updatedAt")
+        `INSERT INTO ${getTableName({ indexName: storage.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`,
@@ -2729,8 +2739,8 @@ var WorkflowsPG = class extends storage.WorkflowsStorage {
       }
       const whereClause = conditions.length > 0 ? `WHERE ${conditions.join(" AND ")}` : "";
       const query = `
-          SELECT * FROM ${getTableName({ indexName: storage.TABLE_WORKFLOW_SNAPSHOT, schemaName: this.schema })} 
-          ${whereClause} 
+          SELECT * FROM ${getTableName({ indexName: storage.TABLE_WORKFLOW_SNAPSHOT, schemaName: this.schema })}
+          ${whereClause}
         `;
       const queryValues = values;
       const result = await this.client.oneOrNone(query, queryValues);
@@ -2800,8 +2810,8 @@ var WorkflowsPG = class extends storage.WorkflowsStorage {
         total = Number(countResult.count);
       }
       const query = `
-          SELECT * FROM ${getTableName({ indexName: storage.TABLE_WORKFLOW_SNAPSHOT, schemaName: this.schema })} 
-          ${whereClause} 
+          SELECT * FROM ${getTableName({ indexName: storage.TABLE_WORKFLOW_SNAPSHOT, schemaName: this.schema })}
+          ${whereClause}
           ORDER BY "createdAt" DESC
           ${limit !== void 0 && offset !== void 0 ? ` LIMIT $${paramIndex} OFFSET $${paramIndex + 1}` : ""}
         `;
@@ -2867,7 +2877,7 @@ var PostgresStore = class extends storage.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 });
@@ -3176,3 +3186,5 @@ Example Complex Query:
 exports.PGVECTOR_PROMPT = PGVECTOR_PROMPT;
 exports.PgVector = PgVector;
 exports.PostgresStore = PostgresStore;
+//# sourceMappingURL=index.cjs.map
+//# sourceMappingURL=index.cjs.map