@mastra/pg 0.2.7-alpha.5 → 0.2.7-alpha.6

package/.turbo/turbo-build.log

@@ -1,23 +1,23 @@
 
-> @mastra/pg@0.2.7-alpha.5 build /home/runner/work/mastra/mastra/stores/pg
+> @mastra/pg@0.2.7-alpha.6 build /home/runner/work/mastra/mastra/stores/pg
 > tsup src/index.ts --format esm,cjs --experimental-dts --clean --treeshake=smallest --splitting
 
 CLI Building entry: src/index.ts
 CLI Using tsconfig: tsconfig.json
 CLI tsup v8.4.0
 TSC Build start
-TSC ⚡️ Build success in 10178ms
+TSC ⚡️ Build success in 10263ms
 DTS Build start
 CLI Target: es2022
 Analysis will use the bundled TypeScript version 5.8.2
 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.2
 Writing package typings: /home/runner/work/mastra/mastra/stores/pg/dist/_tsup-dts-rollup.d.cts
-DTS ⚡️ Build success in 10075ms
+DTS ⚡️ Build success in 11464ms
 CLI Cleaning output folder
 ESM Build start
 CJS Build start
-ESM dist/index.js 40.60 KB
-ESM ⚡️ Build success in 1332ms
-CJS dist/index.cjs 41.03 KB
-CJS ⚡️ Build success in 1332ms
+CJS dist/index.cjs 40.86 KB
+CJS ⚡️ Build success in 1298ms
+ESM dist/index.js 40.44 KB
+ESM ⚡️ Build success in 1321ms

package/CHANGELOG.md

@@ -1,5 +1,13 @@
 # @mastra/pg
 
+## 0.2.7-alpha.6
+
+### Patch Changes
+
+- a3f0e90: Update storage initialization to ensure tables are present
+- Updated dependencies [a3f0e90]
+  - @mastra/core@0.8.0-alpha.6
+
 ## 0.2.7-alpha.5
 
 ### Patch Changes

package/dist/index.cjs

@@ -722,15 +722,16 @@ var PostgresStore = class extends storage.MastraStorage {
       }
     );
   }
-  getEvalsByAgentName(agentName, type) {
+  async getEvalsByAgentName(agentName, type) {
     try {
       const baseQuery = `SELECT * FROM ${storage.TABLE_EVALS} WHERE agent_name = $1`;
       const typeCondition = type === "test" ? " AND test_info IS NOT NULL AND test_info->>'testPath' IS NOT NULL" : type === "live" ? " AND (test_info IS NULL OR test_info->>'testPath' IS NULL)" : "";
       const query = `${baseQuery}${typeCondition} ORDER BY created_at DESC`;
-      return this.db.manyOrNone(query, [agentName]).then((rows) => rows?.map((row) => this.transformEvalRow(row)) ?? []);
+      const rows = await this.db.manyOrNone(query, [agentName]);
+      return rows?.map((row) => this.transformEvalRow(row)) ?? [];
     } catch (error) {
       if (error instanceof Error && error.message.includes("relation") && error.message.includes("does not exist")) {
-        return Promise.resolve([]);
+        return [];
       }
       console.error("Failed to get evals for the specified agent: " + error?.message);
       throw error;
@@ -817,11 +818,6 @@ var PostgresStore = class extends storage.MastraStorage {
         args.push(value);
       }
     }
-    console.log(
-      "QUERY",
-      `SELECT * FROM ${storage.TABLE_TRACES} ${whereClause} ORDER BY "createdAt" DESC LIMIT ${limit} OFFSET ${offset}`,
-      args
-    );
     const result = await this.db.manyOrNone(`SELECT * FROM ${storage.TABLE_TRACES} ${whereClause} ORDER BY "createdAt" DESC LIMIT ${limit} OFFSET ${offset}`, args);
     if (!result) {
       return [];

package/dist/index.js

@@ -714,15 +714,16 @@ var PostgresStore = class extends MastraStorage {
       }
     );
   }
-  getEvalsByAgentName(agentName, type) {
+  async getEvalsByAgentName(agentName, type) {
     try {
       const baseQuery = `SELECT * FROM ${TABLE_EVALS} WHERE agent_name = $1`;
       const typeCondition = type === "test" ? " AND test_info IS NOT NULL AND test_info->>'testPath' IS NOT NULL" : type === "live" ? " AND (test_info IS NULL OR test_info->>'testPath' IS NULL)" : "";
       const query = `${baseQuery}${typeCondition} ORDER BY created_at DESC`;
-      return this.db.manyOrNone(query, [agentName]).then((rows) => rows?.map((row) => this.transformEvalRow(row)) ?? []);
+      const rows = await this.db.manyOrNone(query, [agentName]);
+      return rows?.map((row) => this.transformEvalRow(row)) ?? [];
     } catch (error) {
       if (error instanceof Error && error.message.includes("relation") && error.message.includes("does not exist")) {
-        return Promise.resolve([]);
+        return [];
       }
       console.error("Failed to get evals for the specified agent: " + error?.message);
       throw error;
@@ -809,11 +810,6 @@ var PostgresStore = class extends MastraStorage {
         args.push(value);
       }
     }
-    console.log(
-      "QUERY",
-      `SELECT * FROM ${TABLE_TRACES} ${whereClause} ORDER BY "createdAt" DESC LIMIT ${limit} OFFSET ${offset}`,
-      args
-    );
     const result = await this.db.manyOrNone(`SELECT * FROM ${TABLE_TRACES} ${whereClause} ORDER BY "createdAt" DESC LIMIT ${limit} OFFSET ${offset}`, args);
     if (!result) {
       return [];

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/pg",
-  "version": "0.2.7-alpha.5",
+  "version": "0.2.7-alpha.6",
   "description": "Postgres provider for Mastra - includes both vector and db storage capabilities",
   "type": "module",
   "main": "dist/index.js",
@@ -23,7 +23,7 @@
     "pg": "^8.13.3",
     "pg-promise": "^11.11.0",
     "xxhash-wasm": "^1.1.0",
-    "@mastra/core": "^0.8.0-alpha.5"
+    "@mastra/core": "^0.8.0-alpha.6"
   },
   "devDependencies": {
     "@microsoft/api-extractor": "^7.52.1",

package/src/storage/index.ts

@@ -47,7 +47,7 @@ export class PostgresStore extends MastraStorage {
     );
   }
 
-  getEvalsByAgentName(agentName: string, type?: 'test' | 'live'): Promise<EvalRow[]> {
+  async getEvalsByAgentName(agentName: string, type?: 'test' | 'live'): Promise<EvalRow[]> {
     try {
       const baseQuery = `SELECT * FROM ${TABLE_EVALS} WHERE agent_name = $1`;
       const typeCondition =
@@ -59,11 +59,12 @@ export class PostgresStore extends MastraStorage {
 
       const query = `${baseQuery}${typeCondition} ORDER BY created_at DESC`;
 
-      return this.db.manyOrNone(query, [agentName]).then(rows => rows?.map(row => this.transformEvalRow(row)) ?? []);
+      const rows = await this.db.manyOrNone(query, [agentName]);
+      return rows?.map(row => this.transformEvalRow(row)) ?? [];
     } catch (error) {
       // Handle case where table doesn't exist yet
       if (error instanceof Error && error.message.includes('relation') && error.message.includes('does not exist')) {
-        return Promise.resolve([]);
+        return [];
       }
       console.error('Failed to get evals for the specified agent: ' + (error as any)?.message);
       throw error;
@@ -170,12 +171,6 @@ export class PostgresStore extends MastraStorage {
       }
     }
 
-    console.log(
-      'QUERY',
-      `SELECT * FROM ${TABLE_TRACES} ${whereClause} ORDER BY "createdAt" DESC LIMIT ${limit} OFFSET ${offset}`,
-      args,
-    );
-
     const result = await this.db.manyOrNone<{
       id: string;
       parentSpanId: string;