@mastra/upstash 0.2.2-alpha.1 → 0.2.2-alpha.3

package/.turbo/turbo-build.log

@@ -1,23 +1,23 @@
 
-> @mastra/upstash@0.2.2-alpha.1 build /home/runner/work/mastra/mastra/stores/upstash
+> @mastra/upstash@0.2.2-alpha.3 build /home/runner/work/mastra/mastra/stores/upstash
 > 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 10071ms
+TSC ⚡️ Build success in 9547ms
 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/upstash/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/upstash/dist/_tsup-dts-rollup.d.cts
-DTS ⚡️ Build success in 10569ms
+DTS ⚡️ Build success in 10120ms
 CLI Cleaning output folder
 ESM Build start
 CJS Build start
-CJS dist/index.cjs 17.20 KB
-CJS ⚡️ Build success in 1018ms
-ESM dist/index.js 17.07 KB
-ESM ⚡️ Build success in 1022ms
+CJS dist/index.cjs 21.72 KB
+CJS ⚡️ Build success in 916ms
+ESM dist/index.js 21.62 KB
+ESM ⚡️ Build success in 917ms

package/CHANGELOG.md

@@ -1,5 +1,32 @@
 # @mastra/upstash
 
+## 0.2.2-alpha.3
+
+### Patch Changes
+
+- 88fa727: Added getWorkflowRuns for libsql, pg, clickhouse and upstash as well as added route getWorkflowRunsHandler
+- 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.2-alpha.2
+
+### Patch Changes
+
+- Updated dependencies [56c31b7]
+- Updated dependencies [dbbbf80]
+- Updated dependencies [99d43b9]
+  - @mastra/core@0.8.0-alpha.2
+
 ## 0.2.2-alpha.1
 
 ### Patch Changes

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

@@ -42,13 +42,15 @@ declare class UpstashStore extends MastraStorage {
         tableName: TABLE_NAMES;
         records: Record<string, any>[];
     }): Promise<void>;
-    getEvalsByAgentName(_agentName: string, _type?: 'test' | 'live'): Promise<EvalRow[]>;
+    getEvalsByAgentName(agentName: string, type?: 'test' | 'live'): Promise<EvalRow[]>;
+    private transformEvalRecord;
     getTraces(_input: {
         name?: string;
         scope?: string;
         page: number;
         perPage: number;
         attributes?: Record<string, string>;
+        filters?: Record<string, any>;
     }): Promise<any[]>;
     private redis;
     constructor(config: UpstashConfig);
@@ -104,6 +106,23 @@ declare class UpstashStore extends MastraStorage {
         workflowName: string;
         runId: string;
     }): Promise<WorkflowRunState | null>;
+    getWorkflowRuns({ namespace, workflowName, fromDate, toDate, limit, offset, }?: {
+        namespace: string;
+        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 { UpstashStore }

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

@@ -42,13 +42,15 @@ declare class UpstashStore extends MastraStorage {
         tableName: TABLE_NAMES;
         records: Record<string, any>[];
     }): Promise<void>;
-    getEvalsByAgentName(_agentName: string, _type?: 'test' | 'live'): Promise<EvalRow[]>;
+    getEvalsByAgentName(agentName: string, type?: 'test' | 'live'): Promise<EvalRow[]>;
+    private transformEvalRecord;
     getTraces(_input: {
         name?: string;
         scope?: string;
         page: number;
         perPage: number;
         attributes?: Record<string, string>;
+        filters?: Record<string, any>;
     }): Promise<any[]>;
     private redis;
     constructor(config: UpstashConfig);
@@ -104,6 +106,23 @@ declare class UpstashStore extends MastraStorage {
         workflowName: string;
         runId: string;
     }): Promise<WorkflowRunState | null>;
+    getWorkflowRuns({ namespace, workflowName, fromDate, toDate, limit, offset, }?: {
+        namespace: string;
+        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 { UpstashStore }

package/dist/index.cjs

@@ -11,8 +11,82 @@ var UpstashStore = class extends storage.MastraStorage {
   batchInsert(_input) {
     throw new Error("Method not implemented.");
   }
-  getEvalsByAgentName(_agentName, _type) {
-    throw new Error("Method not implemented.");
+  async getEvalsByAgentName(agentName, type) {
+    try {
+      const pattern = `${storage.TABLE_EVALS}:*`;
+      const keys = await this.redis.keys(pattern);
+      const evalRecords = await Promise.all(
+        keys.map(async (key) => {
+          const data = await this.redis.get(key);
+          return data;
+        })
+      );
+      const nonNullRecords = evalRecords.filter(
+        (record) => record !== null && typeof record === "object" && "agent_name" in record && record.agent_name === agentName
+      );
+      let filteredEvals = nonNullRecords;
+      if (type === "test") {
+        filteredEvals = filteredEvals.filter((record) => {
+          if (!record.test_info) return false;
+          try {
+            if (typeof record.test_info === "string") {
+              const parsedTestInfo = JSON.parse(record.test_info);
+              return parsedTestInfo && typeof parsedTestInfo === "object" && "testPath" in parsedTestInfo;
+            }
+            return typeof record.test_info === "object" && "testPath" in record.test_info;
+          } catch (_e) {
+            return false;
+          }
+        });
+      } else if (type === "live") {
+        filteredEvals = filteredEvals.filter((record) => {
+          if (!record.test_info) return true;
+          try {
+            if (typeof record.test_info === "string") {
+              const parsedTestInfo = JSON.parse(record.test_info);
+              return !(parsedTestInfo && typeof parsedTestInfo === "object" && "testPath" in parsedTestInfo);
+            }
+            return !(typeof record.test_info === "object" && "testPath" in record.test_info);
+          } catch (_e) {
+            return true;
+          }
+        });
+      }
+      return filteredEvals.map((record) => this.transformEvalRecord(record));
+    } catch (error) {
+      console.error("Failed to get evals for the specified agent:", error);
+      return [];
+    }
+  }
+  transformEvalRecord(record) {
+    let result = record.result;
+    if (typeof result === "string") {
+      try {
+        result = JSON.parse(result);
+      } catch (_e) {
+        console.warn("Failed to parse result JSON:");
+      }
+    }
+    let testInfo = record.test_info;
+    if (typeof testInfo === "string") {
+      try {
+        testInfo = JSON.parse(testInfo);
+      } catch (_e) {
+        console.warn("Failed to parse test_info JSON:");
+      }
+    }
+    return {
+      agentName: record.agent_name,
+      input: record.input,
+      output: record.output,
+      result,
+      metricName: record.metric_name,
+      instructions: record.instructions,
+      testInfo,
+      globalRunId: record.global_run_id,
+      runId: record.run_id,
+      createdAt: typeof record.created_at === "string" ? record.created_at : record.created_at instanceof Date ? record.created_at.toISOString() : (/* @__PURE__ */ new Date()).toISOString()
+    };
   }
   getTraces(_input) {
     throw new Error("Method not implemented.");
@@ -53,8 +127,16 @@ var UpstashStore = class extends storage.MastraStorage {
   }
   async insert({ tableName, record }) {
     let key;
-    if (tableName === storage.MastraStorage.TABLE_MESSAGES) {
+    if (tableName === storage.TABLE_MESSAGES) {
       key = this.getKey(tableName, { threadId: record.threadId, id: record.id });
+    } else if (tableName === storage.TABLE_WORKFLOW_SNAPSHOT) {
+      key = this.getKey(tableName, {
+        namespace: record.namespace || "workflows",
+        workflow_name: record.workflow_name,
+        run_id: record.run_id
+      });
+    } else if (tableName === storage.TABLE_EVALS) {
+      key = this.getKey(tableName, { id: record.run_id });
     } else {
       key = this.getKey(tableName, { id: record.id });
     }
@@ -72,7 +154,7 @@ var UpstashStore = class extends storage.MastraStorage {
   }
   async getThreadById({ threadId }) {
     const thread = await this.load({
-      tableName: storage.MastraStorage.TABLE_THREADS,
+      tableName: storage.TABLE_THREADS,
       keys: { id: threadId }
     });
     if (!thread) return null;
@@ -84,7 +166,7 @@ var UpstashStore = class extends storage.MastraStorage {
     };
   }
   async getThreadsByResourceId({ resourceId }) {
-    const pattern = `${storage.MastraStorage.TABLE_THREADS}:*`;
+    const pattern = `${storage.TABLE_THREADS}:*`;
     const keys = await this.redis.keys(pattern);
     const threads = await Promise.all(
       keys.map(async (key) => {
@@ -101,7 +183,7 @@ var UpstashStore = class extends storage.MastraStorage {
   }
   async saveThread({ thread }) {
     await this.insert({
-      tableName: storage.MastraStorage.TABLE_THREADS,
+      tableName: storage.TABLE_THREADS,
       record: thread
     });
     return thread;
@@ -127,11 +209,11 @@ var UpstashStore = class extends storage.MastraStorage {
     return updatedThread;
   }
   async deleteThread({ threadId }) {
-    const key = this.getKey(storage.MastraStorage.TABLE_THREADS, { id: threadId });
+    const key = this.getKey(storage.TABLE_THREADS, { id: threadId });
     await this.redis.del(key);
   }
   getMessageKey(threadId, messageId) {
-    return this.getKey(storage.MastraStorage.TABLE_MESSAGES, { threadId, id: messageId });
+    return this.getKey(storage.TABLE_MESSAGES, { threadId, id: messageId });
   }
   getThreadMessagesKey(threadId) {
     return `thread:${threadId}:messages`;
@@ -193,22 +275,71 @@ var UpstashStore = class extends storage.MastraStorage {
   }
   async persistWorkflowSnapshot(params) {
     const { namespace = "workflows", workflowName, runId, snapshot } = params;
-    const key = this.getKey(storage.MastraStorage.TABLE_WORKFLOW_SNAPSHOT, {
-      namespace,
-      workflow_name: workflowName,
-      run_id: runId
+    await this.insert({
+      tableName: storage.TABLE_WORKFLOW_SNAPSHOT,
+      record: {
+        namespace,
+        workflow_name: workflowName,
+        run_id: runId,
+        snapshot,
+        createdAt: /* @__PURE__ */ new Date(),
+        updatedAt: /* @__PURE__ */ new Date()
+      }
     });
-    await this.redis.set(key, snapshot);
   }
   async loadWorkflowSnapshot(params) {
     const { namespace = "workflows", workflowName, runId } = params;
-    const key = this.getKey(storage.MastraStorage.TABLE_WORKFLOW_SNAPSHOT, {
+    const key = this.getKey(storage.TABLE_WORKFLOW_SNAPSHOT, {
       namespace,
       workflow_name: workflowName,
       run_id: runId
     });
     const data = await this.redis.get(key);
-    return data || null;
+    if (!data) return null;
+    return data.snapshot;
+  }
+  async getWorkflowRuns({
+    namespace,
+    workflowName,
+    fromDate,
+    toDate,
+    limit,
+    offset
+  } = { namespace: "workflows" }) {
+    const pattern = workflowName ? this.getKey(storage.TABLE_WORKFLOW_SNAPSHOT, { namespace, workflow_name: workflowName }) + ":*" : this.getKey(storage.TABLE_WORKFLOW_SNAPSHOT, { namespace }) + ":*";
+    const keys = await this.redis.keys(pattern);
+    const workflows = await Promise.all(
+      keys.map(async (key) => {
+        const data = await this.redis.get(key);
+        return data;
+      })
+    );
+    let runs = workflows.filter((w) => w !== null).map((w) => {
+      let parsedSnapshot = w.snapshot;
+      if (typeof parsedSnapshot === "string") {
+        try {
+          parsedSnapshot = JSON.parse(w.snapshot);
+        } catch (_e) {
+          console.warn(`Failed to parse snapshot for workflow ${w.workflow_name}:`);
+        }
+      }
+      return {
+        workflowName: w.workflow_name,
+        runId: w.run_id,
+        snapshot: parsedSnapshot,
+        createdAt: this.ensureDate(w.createdAt),
+        updatedAt: this.ensureDate(w.updatedAt)
+      };
+    }).filter((w) => {
+      if (fromDate && w.createdAt < fromDate) return false;
+      if (toDate && w.createdAt > toDate) return false;
+      return true;
+    }).sort((a, b) => b.createdAt.getTime() - a.createdAt.getTime());
+    const total = runs.length;
+    if (limit !== void 0 && offset !== void 0) {
+      runs = runs.slice(offset, offset + limit);
+    }
+    return { runs, total };
   }
   async close() {
   }

package/dist/index.js

@@ -1,4 +1,4 @@
-import { MastraStorage } from '@mastra/core/storage';
+import { MastraStorage, TABLE_EVALS, TABLE_MESSAGES, TABLE_WORKFLOW_SNAPSHOT, TABLE_THREADS } from '@mastra/core/storage';
 import { Redis } from '@upstash/redis';
 import { MastraVector } from '@mastra/core/vector';
 import { Index } from '@upstash/vector';
@@ -9,8 +9,82 @@ var UpstashStore = class extends MastraStorage {
   batchInsert(_input) {
     throw new Error("Method not implemented.");
   }
-  getEvalsByAgentName(_agentName, _type) {
-    throw new Error("Method not implemented.");
+  async getEvalsByAgentName(agentName, type) {
+    try {
+      const pattern = `${TABLE_EVALS}:*`;
+      const keys = await this.redis.keys(pattern);
+      const evalRecords = await Promise.all(
+        keys.map(async (key) => {
+          const data = await this.redis.get(key);
+          return data;
+        })
+      );
+      const nonNullRecords = evalRecords.filter(
+        (record) => record !== null && typeof record === "object" && "agent_name" in record && record.agent_name === agentName
+      );
+      let filteredEvals = nonNullRecords;
+      if (type === "test") {
+        filteredEvals = filteredEvals.filter((record) => {
+          if (!record.test_info) return false;
+          try {
+            if (typeof record.test_info === "string") {
+              const parsedTestInfo = JSON.parse(record.test_info);
+              return parsedTestInfo && typeof parsedTestInfo === "object" && "testPath" in parsedTestInfo;
+            }
+            return typeof record.test_info === "object" && "testPath" in record.test_info;
+          } catch (_e) {
+            return false;
+          }
+        });
+      } else if (type === "live") {
+        filteredEvals = filteredEvals.filter((record) => {
+          if (!record.test_info) return true;
+          try {
+            if (typeof record.test_info === "string") {
+              const parsedTestInfo = JSON.parse(record.test_info);
+              return !(parsedTestInfo && typeof parsedTestInfo === "object" && "testPath" in parsedTestInfo);
+            }
+            return !(typeof record.test_info === "object" && "testPath" in record.test_info);
+          } catch (_e) {
+            return true;
+          }
+        });
+      }
+      return filteredEvals.map((record) => this.transformEvalRecord(record));
+    } catch (error) {
+      console.error("Failed to get evals for the specified agent:", error);
+      return [];
+    }
+  }
+  transformEvalRecord(record) {
+    let result = record.result;
+    if (typeof result === "string") {
+      try {
+        result = JSON.parse(result);
+      } catch (_e) {
+        console.warn("Failed to parse result JSON:");
+      }
+    }
+    let testInfo = record.test_info;
+    if (typeof testInfo === "string") {
+      try {
+        testInfo = JSON.parse(testInfo);
+      } catch (_e) {
+        console.warn("Failed to parse test_info JSON:");
+      }
+    }
+    return {
+      agentName: record.agent_name,
+      input: record.input,
+      output: record.output,
+      result,
+      metricName: record.metric_name,
+      instructions: record.instructions,
+      testInfo,
+      globalRunId: record.global_run_id,
+      runId: record.run_id,
+      createdAt: typeof record.created_at === "string" ? record.created_at : record.created_at instanceof Date ? record.created_at.toISOString() : (/* @__PURE__ */ new Date()).toISOString()
+    };
   }
   getTraces(_input) {
     throw new Error("Method not implemented.");
@@ -51,8 +125,16 @@ var UpstashStore = class extends MastraStorage {
   }
   async insert({ tableName, record }) {
     let key;
-    if (tableName === MastraStorage.TABLE_MESSAGES) {
+    if (tableName === TABLE_MESSAGES) {
       key = this.getKey(tableName, { threadId: record.threadId, id: record.id });
+    } else if (tableName === TABLE_WORKFLOW_SNAPSHOT) {
+      key = this.getKey(tableName, {
+        namespace: record.namespace || "workflows",
+        workflow_name: record.workflow_name,
+        run_id: record.run_id
+      });
+    } else if (tableName === TABLE_EVALS) {
+      key = this.getKey(tableName, { id: record.run_id });
     } else {
       key = this.getKey(tableName, { id: record.id });
     }
@@ -70,7 +152,7 @@ var UpstashStore = class extends MastraStorage {
   }
   async getThreadById({ threadId }) {
     const thread = await this.load({
-      tableName: MastraStorage.TABLE_THREADS,
+      tableName: TABLE_THREADS,
       keys: { id: threadId }
     });
     if (!thread) return null;
@@ -82,7 +164,7 @@ var UpstashStore = class extends MastraStorage {
     };
   }
   async getThreadsByResourceId({ resourceId }) {
-    const pattern = `${MastraStorage.TABLE_THREADS}:*`;
+    const pattern = `${TABLE_THREADS}:*`;
     const keys = await this.redis.keys(pattern);
     const threads = await Promise.all(
       keys.map(async (key) => {
@@ -99,7 +181,7 @@ var UpstashStore = class extends MastraStorage {
   }
   async saveThread({ thread }) {
     await this.insert({
-      tableName: MastraStorage.TABLE_THREADS,
+      tableName: TABLE_THREADS,
       record: thread
     });
     return thread;
@@ -125,11 +207,11 @@ var UpstashStore = class extends MastraStorage {
     return updatedThread;
   }
   async deleteThread({ threadId }) {
-    const key = this.getKey(MastraStorage.TABLE_THREADS, { id: threadId });
+    const key = this.getKey(TABLE_THREADS, { id: threadId });
     await this.redis.del(key);
   }
   getMessageKey(threadId, messageId) {
-    return this.getKey(MastraStorage.TABLE_MESSAGES, { threadId, id: messageId });
+    return this.getKey(TABLE_MESSAGES, { threadId, id: messageId });
   }
   getThreadMessagesKey(threadId) {
     return `thread:${threadId}:messages`;
@@ -191,22 +273,71 @@ var UpstashStore = class extends MastraStorage {
   }
   async persistWorkflowSnapshot(params) {
     const { namespace = "workflows", workflowName, runId, snapshot } = params;
-    const key = this.getKey(MastraStorage.TABLE_WORKFLOW_SNAPSHOT, {
-      namespace,
-      workflow_name: workflowName,
-      run_id: runId
+    await this.insert({
+      tableName: TABLE_WORKFLOW_SNAPSHOT,
+      record: {
+        namespace,
+        workflow_name: workflowName,
+        run_id: runId,
+        snapshot,
+        createdAt: /* @__PURE__ */ new Date(),
+        updatedAt: /* @__PURE__ */ new Date()
+      }
     });
-    await this.redis.set(key, snapshot);
   }
   async loadWorkflowSnapshot(params) {
     const { namespace = "workflows", workflowName, runId } = params;
-    const key = this.getKey(MastraStorage.TABLE_WORKFLOW_SNAPSHOT, {
+    const key = this.getKey(TABLE_WORKFLOW_SNAPSHOT, {
       namespace,
       workflow_name: workflowName,
       run_id: runId
     });
     const data = await this.redis.get(key);
-    return data || null;
+    if (!data) return null;
+    return data.snapshot;
+  }
+  async getWorkflowRuns({
+    namespace,
+    workflowName,
+    fromDate,
+    toDate,
+    limit,
+    offset
+  } = { namespace: "workflows" }) {
+    const pattern = workflowName ? this.getKey(TABLE_WORKFLOW_SNAPSHOT, { namespace, workflow_name: workflowName }) + ":*" : this.getKey(TABLE_WORKFLOW_SNAPSHOT, { namespace }) + ":*";
+    const keys = await this.redis.keys(pattern);
+    const workflows = await Promise.all(
+      keys.map(async (key) => {
+        const data = await this.redis.get(key);
+        return data;
+      })
+    );
+    let runs = workflows.filter((w) => w !== null).map((w) => {
+      let parsedSnapshot = w.snapshot;
+      if (typeof parsedSnapshot === "string") {
+        try {
+          parsedSnapshot = JSON.parse(w.snapshot);
+        } catch (_e) {
+          console.warn(`Failed to parse snapshot for workflow ${w.workflow_name}:`);
+        }
+      }
+      return {
+        workflowName: w.workflow_name,
+        runId: w.run_id,
+        snapshot: parsedSnapshot,
+        createdAt: this.ensureDate(w.createdAt),
+        updatedAt: this.ensureDate(w.updatedAt)
+      };
+    }).filter((w) => {
+      if (fromDate && w.createdAt < fromDate) return false;
+      if (toDate && w.createdAt > toDate) return false;
+      return true;
+    }).sort((a, b) => b.createdAt.getTime() - a.createdAt.getTime());
+    const total = runs.length;
+    if (limit !== void 0 && offset !== void 0) {
+      runs = runs.slice(offset, offset + limit);
+    }
+    return { runs, total };
   }
   async close() {
   }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/upstash",
-  "version": "0.2.2-alpha.1",
+  "version": "0.2.2-alpha.3",
   "description": "Upstash provider for Mastra - includes both vector and db storage capabilities",
   "type": "module",
   "main": "dist/index.js",
@@ -21,7 +21,7 @@
   "dependencies": {
     "@upstash/redis": "^1.34.5",
     "@upstash/vector": "^1.2.1",
-    "@mastra/core": "^0.8.0-alpha.1"
+    "@mastra/core": "^0.8.0-alpha.3"
   },
   "devDependencies": {
     "@microsoft/api-extractor": "^7.52.1",