@mastra/pg 0.13.3-alpha.0 → 0.14.0-alpha.0

package/.turbo/turbo-build.log

@@ -1,4 +1,4 @@
 
-> @mastra/pg@0.13.3-alpha.0 build /home/runner/work/mastra/mastra/stores/pg
+> @mastra/pg@0.14.0-alpha.0 build /home/runner/work/mastra/mastra/stores/pg
 > tsup --silent --config tsup.config.ts
 

package/CHANGELOG.md

@@ -1,5 +1,29 @@
 # @mastra/pg
 
+## 0.14.0-alpha.0
+
+### Minor Changes
+
+- 79d34ce: improve cloudflare workers compatibility
+
+### Patch Changes
+
+- b32c50d: Filter scores by source
+- Updated dependencies [d5330bf]
+- Updated dependencies [a239d41]
+- Updated dependencies [b32c50d]
+- Updated dependencies [121a3f8]
+- Updated dependencies [ec510e7]
+  - @mastra/core@0.13.2-alpha.2
+
+## 0.13.3
+
+### Patch Changes
+
+- 3e49b7a: Fix PostgreSQL vector metadata filtering for Memory system - changed default minScore to -1 to include all similarity results and added comprehensive metadata filtering tests
+- Updated dependencies [cd0042e]
+  - @mastra/core@0.13.1
+
 ## 0.13.3-alpha.0
 
 ### Patch Changes

package/dist/index.cjs

@@ -2296,6 +2296,7 @@ var StoreOperationsPG = class extends storage.StoreOperations {
   }
 };
 function transformScoreRow(row) {
+  console.log(`row is`, JSON.stringify(row, null, 2));
   return {
     ...row,
     input: storage.safelyParseJSON(row.input),
@@ -2331,7 +2332,7 @@ var ScoresPG = class extends storage.ScoresStorage {
         `SELECT * FROM ${getTableName({ indexName: storage.TABLE_SCORERS, schemaName: this.schema })} WHERE id = $1`,
         [id]
       );
-      return transformScoreRow(result);
+      return result ? transformScoreRow(result) : null;
     } catch (error$1) {
       throw new error.MastraError(
         {
@@ -2345,12 +2346,31 @@ var ScoresPG = class extends storage.ScoresStorage {
   }
   async getScoresByScorerId({
     scorerId,
-    pagination
+    pagination,
+    entityId,
+    entityType,
+    source
   }) {
     try {
+      const conditions = [`"scorerId" = $1`];
+      const queryParams = [scorerId];
+      let paramIndex = 2;
+      if (entityId) {
+        conditions.push(`"entityId" = $${paramIndex++}`);
+        queryParams.push(entityId);
+      }
+      if (entityType) {
+        conditions.push(`"entityType" = $${paramIndex++}`);
+        queryParams.push(entityType);
+      }
+      if (source) {
+        conditions.push(`"source" = $${paramIndex++}`);
+        queryParams.push(source);
+      }
+      const whereClause = conditions.join(" AND ");
       const total = await this.client.oneOrNone(
-        `SELECT COUNT(*) FROM ${getTableName({ indexName: storage.TABLE_SCORERS, schemaName: this.schema })} WHERE "scorerId" = $1`,
-        [scorerId]
+        `SELECT COUNT(*) FROM ${getTableName({ indexName: storage.TABLE_SCORERS, schemaName: this.schema })} WHERE ${whereClause}`,
+        queryParams
       );
       if (total?.count === "0" || !total?.count) {
         return {
@@ -2364,8 +2384,8 @@ var ScoresPG = class extends storage.ScoresStorage {
         };
       }
       const result = await this.client.manyOrNone(
-        `SELECT * FROM ${getTableName({ indexName: storage.TABLE_SCORERS, schemaName: this.schema })} WHERE "scorerId" = $1 LIMIT $2 OFFSET $3`,
-        [scorerId, pagination.perPage, pagination.page * pagination.perPage]
+        `SELECT * FROM ${getTableName({ indexName: storage.TABLE_SCORERS, schemaName: this.schema })} WHERE ${whereClause} ORDER BY "createdAt" DESC LIMIT $${paramIndex++} OFFSET $${paramIndex++}`,
+        [...queryParams, pagination.perPage, pagination.page * pagination.perPage]
       );
       return {
         pagination: {
@@ -2389,7 +2409,7 @@ var ScoresPG = class extends storage.ScoresStorage {
   }
   async saveScore(score) {
     try {
-      const scoreId = crypto.randomUUID();
+      const id = crypto.randomUUID();
       const {
         scorer,
         preprocessStepResult,
@@ -2402,10 +2422,11 @@ var ScoresPG = class extends storage.ScoresStorage {
         entity,
         ...rest
       } = score;
+      console.log(`saving score with id: ${id}`);
       await this.operations.insert({
         tableName: storage.TABLE_SCORERS,
         record: {
-          id: scoreId,
+          id,
           ...rest,
           input: JSON.stringify(input) || "",
           output: JSON.stringify(output) || "",
@@ -2420,7 +2441,7 @@ var ScoresPG = class extends storage.ScoresStorage {
           updatedAt: (/* @__PURE__ */ new Date()).toISOString()
         }
       });
-      const scoreFromDb = await this.getScoreById({ id: scoreId });
+      const scoreFromDb = await this.getScoreById({ id });
       return { score: scoreFromDb };
     } catch (error$1) {
       throw new error.MastraError(
@@ -2856,10 +2877,11 @@ var WorkflowsPG = class extends storage.WorkflowsStorage {
 
 // src/storage/index.ts
 var PostgresStore = class extends storage.MastraStorage {
-  db;
-  pgp;
-  client;
+  #db;
+  #pgp;
+  #config;
   schema;
+  isConnected = false;
   stores;
   constructor(config) {
     try {
@@ -2880,10 +2902,11 @@ var PostgresStore = class extends storage.MastraStorage {
         }
       }
       super({ name: "PostgresStore" });
-      this.pgp = pgPromise__default.default();
       this.schema = config.schemaName || "public";
-      this.db = this.pgp(
-        `connectionString` in config ? { connectionString: config.connectionString } : {
+      this.#config = {
+        max: config.max,
+        idleTimeoutMillis: config.idleTimeoutMillis,
+        ...`connectionString` in config ? { connectionString: config.connectionString } : {
           host: config.host,
           port: config.port,
           database: config.database,
@@ -2891,14 +2914,33 @@ var PostgresStore = class extends storage.MastraStorage {
           password: config.password,
           ssl: config.ssl
         }
+      };
+      this.stores = {};
+    } catch (e) {
+      throw new error.MastraError(
+        {
+          id: "MASTRA_STORAGE_PG_STORE_INITIALIZATION_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.USER
+        },
+        e
       );
-      this.client = this.db;
-      const operations = new StoreOperationsPG({ client: this.client, schemaName: this.schema });
-      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 });
-      const memory = new MemoryPG({ client: this.client, schema: this.schema, operations });
+    }
+  }
+  async init() {
+    if (this.isConnected) {
+      return;
+    }
+    try {
+      this.isConnected = true;
+      this.#pgp = pgPromise__default.default();
+      this.#db = this.#pgp(this.#config);
+      const operations = new StoreOperationsPG({ client: this.#db, schemaName: this.schema });
+      const scores = new ScoresPG({ client: this.#db, operations, schema: this.schema });
+      const traces = new TracesPG({ client: this.#db, operations, schema: this.schema });
+      const workflows = new WorkflowsPG({ client: this.#db, operations, schema: this.schema });
+      const legacyEvals = new LegacyEvalsPG({ client: this.#db, schema: this.schema });
+      const memory = new MemoryPG({ client: this.#db, schema: this.schema, operations });
       this.stores = {
         operations,
         scores,
@@ -2907,17 +2949,31 @@ var PostgresStore = class extends storage.MastraStorage {
         legacyEvals,
         memory
       };
-    } catch (e) {
+      await super.init();
+    } catch (error$1) {
+      this.isConnected = false;
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_STORE_INITIALIZATION_FAILED",
+          id: "MASTRA_STORAGE_POSTGRES_STORE_INIT_FAILED",
           domain: error.ErrorDomain.STORAGE,
-          category: error.ErrorCategory.USER
+          category: error.ErrorCategory.THIRD_PARTY
         },
-        e
+        error$1
       );
     }
   }
+  get db() {
+    if (!this.#db) {
+      throw new Error(`PostgresStore: Store is not initialized, please call "init()" first.`);
+    }
+    return this.#db;
+  }
+  get pgp() {
+    if (!this.#pgp) {
+      throw new Error(`PostgresStore: Store is not initialized, please call "init()" first.`);
+    }
+    return this.#pgp;
+  }
   get supports() {
     return {
       selectByIncludeResourceScope: true,
@@ -3074,10 +3130,13 @@ var PostgresStore = class extends storage.MastraStorage {
     return this.stores.scores.getScoreById({ id: _id });
   }
   async getScoresByScorerId({
-    scorerId: _scorerId,
-    pagination: _pagination
+    scorerId,
+    pagination,
+    entityId,
+    entityType,
+    source
   }) {
-    return this.stores.scores.getScoresByScorerId({ scorerId: _scorerId, pagination: _pagination });
+    return this.stores.scores.getScoresByScorerId({ scorerId, pagination, entityId, entityType, source });
   }
   async saveScore(_score) {
     return this.stores.scores.saveScore(_score);