@mastra/pg 0.2.7-alpha.2 → 0.2.7-alpha.3

package/.turbo/turbo-build.log

@@ -1,23 +1,23 @@
 
-> @mastra/pg@0.2.7-alpha.2 build /home/runner/work/mastra/mastra/stores/pg
+> @mastra/pg@0.2.7-alpha.3 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 10747ms
+TSC ⚡️ Build success in 10020ms
 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 11651ms
+DTS ⚡️ Build success in 11079ms
 CLI Cleaning output folder
 ESM Build start
 CJS Build start
-CJS dist/index.cjs 37.38 KB
-CJS ⚡️ Build success in 1188ms
-ESM dist/index.js 36.97 KB
-ESM ⚡️ Build success in 1188ms
+ESM dist/index.js 40.32 KB
+ESM ⚡️ Build success in 1253ms
+CJS dist/index.cjs 40.67 KB
+CJS ⚡️ Build success in 1258ms

package/CHANGELOG.md

@@ -1,5 +1,24 @@
 # @mastra/pg
 
+## 0.2.7-alpha.3
+
+### Patch Changes
+
+- 88fa727: Added getWorkflowRuns for libsql, pg, clickhouse and upstash as well as added route getWorkflowRunsHandler
+- 1849b61: adds optional ssl property to PostgresConfig
+- 4d67826: Fix eval writes, remove id column
+- 58a4146: Support missing methods in pg and upstash
+- Updated dependencies [5ae0180]
+- Updated dependencies [9bfa12b]
+- Updated dependencies [515ebfb]
+- Updated dependencies [88fa727]
+- Updated dependencies [f37f535]
+- Updated dependencies [4d67826]
+- Updated dependencies [6330967]
+- Updated dependencies [8393832]
+- Updated dependencies [6330967]
+  - @mastra/core@0.8.0-alpha.3
+
 ## 0.2.7-alpha.2
 
 ### Patch Changes

package/dist/_tsup-dts-rollup.d.cts

@@ -6,6 +6,7 @@ import type { CreateIndexParams } from '@mastra/core/vector';
 import type { ElementOperator } from '@mastra/core/vector/filter';
 import type { EvalRow } from '@mastra/core/storage';
 import type { IndexStats } from '@mastra/core/vector';
+import type { ISSLConfig } from 'pg-promise/typescript/pg-subset';
 import type { LogicalOperator } from '@mastra/core/vector/filter';
 import { MastraStorage } from '@mastra/core/storage';
 import { MastraVector } from '@mastra/core/vector';
@@ -232,6 +233,7 @@ declare type PostgresConfig = {
     database: string;
     user: string;
     password: string;
+    ssl?: boolean | ISSLConfig;
 } | {
     connectionString: string;
 };
@@ -242,17 +244,19 @@ declare class PostgresStore extends MastraStorage {
     private db;
     private pgp;
     constructor(config: PostgresConfig);
-    getEvalsByAgentName(_agentName: string, _type?: 'test' | 'live'): Promise<EvalRow[]>;
+    getEvalsByAgentName(agentName: string, type?: 'test' | 'live'): Promise<EvalRow[]>;
+    private transformEvalRow;
     batchInsert({ tableName, records }: {
         tableName: TABLE_NAMES;
         records: Record<string, any>[];
     }): Promise<void>;
-    getTraces({ name, scope, page, perPage, attributes, }: {
+    getTraces({ name, scope, page, perPage, attributes, filters, }: {
         name?: string;
         scope?: string;
         page: number;
         perPage: number;
         attributes?: Record<string, string>;
+        filters?: Record<string, any>;
     }): Promise<any[]>;
     createTable({ tableName, schema, }: {
         tableName: TABLE_NAMES;
@@ -299,6 +303,22 @@ declare class PostgresStore extends MastraStorage {
         workflowName: string;
         runId: string;
     }): Promise<WorkflowRunState | null>;
+    getWorkflowRuns({ workflowName, fromDate, toDate, limit, offset, }?: {
+        workflowName?: string;
+        fromDate?: Date;
+        toDate?: Date;
+        limit?: number;
+        offset?: number;
+    }): Promise<{
+        runs: Array<{
+            workflowName: string;
+            runId: string;
+            snapshot: WorkflowRunState | string;
+            createdAt: Date;
+            updatedAt: Date;
+        }>;
+        total: number;
+    }>;
     close(): Promise<void>;
 }
 export { PostgresStore }

package/dist/_tsup-dts-rollup.d.ts

@@ -6,6 +6,7 @@ import type { CreateIndexParams } from '@mastra/core/vector';
 import type { ElementOperator } from '@mastra/core/vector/filter';
 import type { EvalRow } from '@mastra/core/storage';
 import type { IndexStats } from '@mastra/core/vector';
+import type { ISSLConfig } from 'pg-promise/typescript/pg-subset';
 import type { LogicalOperator } from '@mastra/core/vector/filter';
 import { MastraStorage } from '@mastra/core/storage';
 import { MastraVector } from '@mastra/core/vector';
@@ -232,6 +233,7 @@ declare type PostgresConfig = {
     database: string;
     user: string;
     password: string;
+    ssl?: boolean | ISSLConfig;
 } | {
     connectionString: string;
 };
@@ -242,17 +244,19 @@ declare class PostgresStore extends MastraStorage {
     private db;
     private pgp;
     constructor(config: PostgresConfig);
-    getEvalsByAgentName(_agentName: string, _type?: 'test' | 'live'): Promise<EvalRow[]>;
+    getEvalsByAgentName(agentName: string, type?: 'test' | 'live'): Promise<EvalRow[]>;
+    private transformEvalRow;
     batchInsert({ tableName, records }: {
         tableName: TABLE_NAMES;
         records: Record<string, any>[];
     }): Promise<void>;
-    getTraces({ name, scope, page, perPage, attributes, }: {
+    getTraces({ name, scope, page, perPage, attributes, filters, }: {
         name?: string;
         scope?: string;
         page: number;
         perPage: number;
         attributes?: Record<string, string>;
+        filters?: Record<string, any>;
     }): Promise<any[]>;
     createTable({ tableName, schema, }: {
         tableName: TABLE_NAMES;
@@ -299,6 +303,22 @@ declare class PostgresStore extends MastraStorage {
         workflowName: string;
         runId: string;
     }): Promise<WorkflowRunState | null>;
+    getWorkflowRuns({ workflowName, fromDate, toDate, limit, offset, }?: {
+        workflowName?: string;
+        fromDate?: Date;
+        toDate?: Date;
+        limit?: number;
+        offset?: number;
+    }): Promise<{
+        runs: Array<{
+            workflowName: string;
+            runId: string;
+            snapshot: WorkflowRunState | string;
+            createdAt: Date;
+            updatedAt: Date;
+        }>;
+        total: number;
+    }>;
     close(): Promise<void>;
 }
 export { PostgresStore }

package/dist/index.cjs

@@ -708,12 +708,46 @@ var PostgresStore = class extends storage.MastraStorage {
         port: config.port,
         database: config.database,
         user: config.user,
-        password: config.password
+        password: config.password,
+        ssl: config.ssl
       }
     );
   }
-  getEvalsByAgentName(_agentName, _type) {
-    throw new Error("Method not implemented.");
+  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)) ?? []);
+    } catch (error) {
+      if (error instanceof Error && error.message.includes("relation") && error.message.includes("does not exist")) {
+        return Promise.resolve([]);
+      }
+      console.error("Failed to get evals for the specified agent: " + error?.message);
+      throw error;
+    }
+  }
+  transformEvalRow(row) {
+    let testInfoValue = null;
+    if (row.test_info) {
+      try {
+        testInfoValue = typeof row.test_info === "string" ? JSON.parse(row.test_info) : row.test_info;
+      } catch (e) {
+        console.warn("Failed to parse test_info:", e);
+      }
+    }
+    return {
+      agentName: row.agent_name,
+      input: row.input,
+      output: row.output,
+      result: row.result,
+      metricName: row.metric_name,
+      instructions: row.instructions,
+      testInfo: testInfoValue,
+      globalRunId: row.global_run_id,
+      runId: row.run_id,
+      createdAt: row.created_at
+    };
   }
   async batchInsert({ tableName, records }) {
     try {
@@ -733,7 +767,8 @@ var PostgresStore = class extends storage.MastraStorage {
     scope,
     page,
     perPage,
-    attributes
+    attributes,
+    filters
   }) {
     let idx = 1;
     const limit = perPage;
@@ -751,6 +786,11 @@ var PostgresStore = class extends storage.MastraStorage {
         conditions.push(`attributes->>'${key}' = $${idx++}`);
       });
     }
+    if (filters) {
+      Object.entries(filters).forEach(([key, value]) => {
+        conditions.push(`${key} = $${idx++}`);
+      });
+    }
     const whereClause = conditions.length > 0 ? `WHERE ${conditions.join(" AND ")}` : "";
     if (name) {
       args.push(name);
@@ -763,15 +803,17 @@ var PostgresStore = class extends storage.MastraStorage {
         args.push(value);
       }
     }
+    if (filters) {
+      for (const [key, value] of Object.entries(filters)) {
+        args.push(value);
+      }
+    }
     console.log(
       "QUERY",
-      `SELECT * FROM ${storage.MastraStorage.TABLE_TRACES} ${whereClause} ORDER BY "createdAt" DESC LIMIT ${limit} OFFSET ${offset}`,
-      args
-    );
-    const result = await this.db.manyOrNone(
-      `SELECT * FROM ${storage.MastraStorage.TABLE_TRACES} ${whereClause} ORDER BY "createdAt" DESC LIMIT ${limit} OFFSET ${offset}`,
+      `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 [];
     }
@@ -807,7 +849,7 @@ var PostgresStore = class extends storage.MastraStorage {
         CREATE TABLE IF NOT EXISTS ${tableName} (
           ${columns}
         );
-        ${tableName === storage.MastraStorage.TABLE_WORKFLOW_SNAPSHOT ? `
+        ${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'
@@ -856,7 +898,7 @@ var PostgresStore = class extends storage.MastraStorage {
       if (!result) {
         return null;
       }
-      if (tableName === storage.MastraStorage.TABLE_WORKFLOW_SNAPSHOT) {
+      if (tableName === storage.TABLE_WORKFLOW_SNAPSHOT) {
         const snapshot = result;
         if (typeof snapshot.snapshot === "string") {
           snapshot.snapshot = JSON.parse(snapshot.snapshot);
@@ -879,7 +921,7 @@ var PostgresStore = class extends storage.MastraStorage {
           metadata,
           "createdAt",
           "updatedAt"
-        FROM "${storage.MastraStorage.TABLE_THREADS}"
+        FROM "${storage.TABLE_THREADS}"
         WHERE id = $1`,
         [threadId]
       );
@@ -907,7 +949,7 @@ var PostgresStore = class extends storage.MastraStorage {
           metadata,
           "createdAt",
           "updatedAt"
-        FROM "${storage.MastraStorage.TABLE_THREADS}"
+        FROM "${storage.TABLE_THREADS}"
         WHERE "resourceId" = $1`,
         [resourceId]
       );
@@ -925,7 +967,7 @@ var PostgresStore = class extends storage.MastraStorage {
   async saveThread({ thread }) {
     try {
       await this.db.none(
-        `INSERT INTO "${storage.MastraStorage.TABLE_THREADS}" (
+        `INSERT INTO "${storage.TABLE_THREADS}" (
           id,
           "resourceId",
           title,
@@ -969,7 +1011,7 @@ var PostgresStore = class extends storage.MastraStorage {
         ...metadata
       };
       const thread = await this.db.one(
-        `UPDATE "${storage.MastraStorage.TABLE_THREADS}"
+        `UPDATE "${storage.TABLE_THREADS}"
         SET title = $1,
             metadata = $2,
             "updatedAt" = $3
@@ -991,8 +1033,8 @@ var PostgresStore = class extends storage.MastraStorage {
   async deleteThread({ threadId }) {
     try {
       await this.db.tx(async (t) => {
-        await t.none(`DELETE FROM "${storage.MastraStorage.TABLE_MESSAGES}" WHERE thread_id = $1`, [threadId]);
-        await t.none(`DELETE FROM "${storage.MastraStorage.TABLE_THREADS}" WHERE id = $1`, [threadId]);
+        await t.none(`DELETE FROM "${storage.TABLE_MESSAGES}" WHERE thread_id = $1`, [threadId]);
+        await t.none(`DELETE FROM "${storage.TABLE_THREADS}" WHERE id = $1`, [threadId]);
       });
     } catch (error) {
       console.error("Error deleting thread:", error);
@@ -1011,7 +1053,7 @@ var PostgresStore = class extends storage.MastraStorage {
             SELECT 
               *,
               ROW_NUMBER() OVER (ORDER BY "createdAt" DESC) as row_num
-            FROM "${storage.MastraStorage.TABLE_MESSAGES}"
+            FROM "${storage.TABLE_MESSAGES}"
             WHERE thread_id = $1
           )
           SELECT
@@ -1054,7 +1096,7 @@ var PostgresStore = class extends storage.MastraStorage {
             type,
             "createdAt", 
             thread_id AS "threadId"
-        FROM "${storage.MastraStorage.TABLE_MESSAGES}"
+        FROM "${storage.TABLE_MESSAGES}"
         WHERE thread_id = $1
         AND id != ALL($2)
         ORDER BY "createdAt" DESC
@@ -1092,7 +1134,7 @@ var PostgresStore = class extends storage.MastraStorage {
       await this.db.tx(async (t) => {
         for (const message of messages) {
           await t.none(
-            `INSERT INTO "${storage.MastraStorage.TABLE_MESSAGES}" (id, thread_id, content, "createdAt", role, type) 
+            `INSERT INTO "${storage.TABLE_MESSAGES}" (id, thread_id, content, "createdAt", role, type) 
              VALUES ($1, $2, $3, $4, $5, $6)`,
             [
               message.id,
@@ -1119,7 +1161,7 @@ var PostgresStore = class extends storage.MastraStorage {
     try {
       const now = (/* @__PURE__ */ new Date()).toISOString();
       await this.db.none(
-        `INSERT INTO "${storage.MastraStorage.TABLE_WORKFLOW_SNAPSHOT}" (
+        `INSERT INTO "${storage.TABLE_WORKFLOW_SNAPSHOT}" (
           workflow_name,
           run_id,
           snapshot,
@@ -1142,7 +1184,7 @@ var PostgresStore = class extends storage.MastraStorage {
   }) {
     try {
       const result = await this.load({
-        tableName: storage.MastraStorage.TABLE_WORKFLOW_SNAPSHOT,
+        tableName: storage.TABLE_WORKFLOW_SNAPSHOT,
         keys: {
           workflow_name: workflowName,
           run_id: runId
@@ -1157,6 +1199,67 @@ var PostgresStore = class extends storage.MastraStorage {
       throw error;
     }
   }
+  async getWorkflowRuns({
+    workflowName,
+    fromDate,
+    toDate,
+    limit,
+    offset
+  } = {}) {
+    const conditions = [];
+    const values = [];
+    let paramIndex = 1;
+    if (workflowName) {
+      conditions.push(`workflow_name = $${paramIndex}`);
+      values.push(workflowName);
+      paramIndex++;
+    }
+    if (fromDate) {
+      conditions.push(`"createdAt" >= $${paramIndex}`);
+      values.push(fromDate);
+      paramIndex++;
+    }
+    if (toDate) {
+      conditions.push(`"createdAt" <= $${paramIndex}`);
+      values.push(toDate);
+      paramIndex++;
+    }
+    const whereClause = conditions.length > 0 ? `WHERE ${conditions.join(" AND ")}` : "";
+    let total = 0;
+    if (limit !== void 0 && offset !== void 0) {
+      const countResult = await this.db.one(
+        `SELECT COUNT(*) as count FROM ${storage.TABLE_WORKFLOW_SNAPSHOT} ${whereClause}`,
+        values
+      );
+      total = Number(countResult.count);
+    }
+    const query = `
+      SELECT * FROM ${storage.TABLE_WORKFLOW_SNAPSHOT} 
+      ${whereClause} 
+      ORDER BY "createdAt" DESC
+      ${limit !== void 0 && offset !== void 0 ? ` LIMIT $${paramIndex} OFFSET $${paramIndex + 1}` : ""}
+    `;
+    const queryValues = limit !== void 0 && offset !== void 0 ? [...values, limit, offset] : values;
+    const result = await this.db.manyOrNone(query, queryValues);
+    const runs = (result || []).map((row) => {
+      let parsedSnapshot = row.snapshot;
+      if (typeof parsedSnapshot === "string") {
+        try {
+          parsedSnapshot = JSON.parse(row.snapshot);
+        } catch (e) {
+          console.warn(`Failed to parse snapshot for workflow ${row.workflow_name}: ${e}`);
+        }
+      }
+      return {
+        workflowName: row.workflow_name,
+        runId: row.run_id,
+        snapshot: parsedSnapshot,
+        createdAt: row.createdAt,
+        updatedAt: row.updatedAt
+      };
+    });
+    return { runs, total: total || runs.length };
+  }
   async close() {
     this.pgp.end();
   }