@mastra/upstash 0.10.3 → 0.11.0-alpha.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/upstash",
-  "version": "0.10.3",
+  "version": "0.11.0-alpha.0",
   "description": "Upstash provider for Mastra - includes both vector and db storage capabilities",
   "type": "module",
   "main": "dist/index.js",
@@ -25,15 +25,15 @@
   },
   "devDependencies": {
     "@microsoft/api-extractor": "^7.52.8",
-    "@types/node": "^20.17.57",
+    "@types/node": "^20.19.0",
     "dotenv": "^16.5.0",
     "eslint": "^9.28.0",
     "tsup": "^8.5.0",
-    "typescript": "^5.8.2",
-    "vitest": "^3.2.2",
-    "@internal/storage-test-utils": "0.0.7",
-    "@internal/lint": "0.0.11",
-    "@mastra/core": "0.10.4"
+    "typescript": "^5.8.3",
+    "vitest": "^3.2.3",
+    "@internal/storage-test-utils": "0.0.8",
+    "@internal/lint": "0.0.12",
+    "@mastra/core": "0.10.6-alpha.0"
   },
   "peerDependencies": {
     "@mastra/core": ">=0.10.4-0 <0.11.0"

package/src/storage/index.ts

@@ -628,10 +628,15 @@ export class UpstashStore extends MastraStorage {
       _index: index,
     }));
 
+    // Get current thread data once (all messages belong to same thread)
+    const threadKey = this.getKey(TABLE_THREADS, { id: threadId });
+    const existingThread = await this.redis.get<StorageThreadType>(threadKey);
+
     const batchSize = 1000;
     for (let i = 0; i < messagesWithIndex.length; i += batchSize) {
       const batch = messagesWithIndex.slice(i, i + batchSize);
       const pipeline = this.redis.pipeline();
+
       for (const message of batch) {
         const key = this.getMessageKey(message.threadId!, message.id);
         const createdAtScore = new Date(message.createdAt).getTime();
@@ -647,6 +652,15 @@ export class UpstashStore extends MastraStorage {
         });
       }
 
+      // Update the thread's updatedAt field (only in the first batch)
+      if (i === 0 && existingThread) {
+        const updatedThread = {
+          ...existingThread,
+          updatedAt: new Date(),
+        };
+        pipeline.set(threadKey, this.processRecord(TABLE_THREADS, updatedThread).processedRecord);
+      }
+
       await pipeline.exec();
     }
 

package/src/storage/upstash.test.ts

@@ -187,6 +187,29 @@ describe('UpstashStore', () => {
         updated: 'value',
       });
     });
+
+    it('should update thread updatedAt when a message is saved to it', async () => {
+      const thread = createSampleThread();
+      await store.saveThread({ thread });
+
+      // Get the initial thread to capture the original updatedAt
+      const initialThread = await store.getThreadById({ threadId: thread.id });
+      expect(initialThread).toBeDefined();
+      const originalUpdatedAt = initialThread!.updatedAt;
+
+      // Wait a small amount to ensure different timestamp
+      await new Promise(resolve => setTimeout(resolve, 10));
+
+      // Create and save a message to the thread
+      const message = createSampleMessageV2({ threadId: thread.id });
+      await store.saveMessages({ messages: [message], format: 'v2' });
+
+      // Retrieve the thread again and check that updatedAt was updated
+      const updatedThread = await store.getThreadById({ threadId: thread.id });
+      expect(updatedThread).toBeDefined();
+      expect(updatedThread!.updatedAt.getTime()).toBeGreaterThan(originalUpdatedAt.getTime());
+    });
+
     it('should fetch >100000 threads by resource ID', async () => {
       const resourceId = `resource-${randomUUID()}`;
       const total = 100_000;
@@ -669,13 +692,14 @@ describe('UpstashStore', () => {
         activePaths: [],
         suspendedPaths: {},
         timestamp: Date.now(),
+        status: 'success',
       };
 
       await store.persistWorkflowSnapshot({
         namespace: testNamespace,
         workflowName: testWorkflow,
         runId: testRunId,
-        snapshot: mockSnapshot,
+        snapshot: mockSnapshot as WorkflowRunState,
       });
 
       const loadedSnapshot = await store.loadWorkflowSnapshot({