@mastra/lance 0.2.6 → 0.2.7-alpha.0

package/.turbo/turbo-build.log

@@ -1,4 +1,4 @@
 
-> @mastra/lance@0.2.6-alpha.0 build /home/runner/work/mastra/mastra/stores/lance
+> @mastra/lance@0.2.7-alpha.0 build /home/runner/work/mastra/mastra/stores/lance
 > tsup --silent --config tsup.config.ts
 

package/CHANGELOG.md

@@ -1,5 +1,14 @@
 # @mastra/lance
 
+## 0.2.7-alpha.0
+
+### Patch Changes
+
+- [#6700](https://github.com/mastra-ai/mastra/pull/6700) [`a5a23d9`](https://github.com/mastra-ai/mastra/commit/a5a23d981920d458dc6078919992a5338931ef02) Thanks [@gpanakkal](https://github.com/gpanakkal)! - Add `getMessagesById` method to `MastraStorage` adapters
+
+- Updated dependencies [[`6e7e120`](https://github.com/mastra-ai/mastra/commit/6e7e1207d6e8d8b838f9024f90bd10df1181ba27), [`a5a23d9`](https://github.com/mastra-ai/mastra/commit/a5a23d981920d458dc6078919992a5338931ef02)]:
+  - @mastra/core@0.14.1-alpha.0
+
 ## 0.2.6
 
 ### Patch Changes

package/dist/index.cjs

@@ -381,6 +381,20 @@ var StoreMemoryLance = class extends storage.MemoryStorage {
       );
     }
   }
+  normalizeMessage(message) {
+    const { thread_id, ...rest } = message;
+    return {
+      ...rest,
+      threadId: thread_id,
+      content: typeof message.content === "string" ? (() => {
+        try {
+          return JSON.parse(message.content);
+        } catch {
+          return message.content;
+        }
+      })() : message.content
+    };
+  }
   async getMessages({
     threadId,
     resourceId,
@@ -421,21 +435,7 @@ var StoreMemoryLance = class extends storage.MemoryStorage {
         allRecords,
         await getTableSchema({ tableName: storage.TABLE_MESSAGES, client: this.client })
       );
-      const normalized = messages.map((msg) => {
-        const { thread_id, ...rest } = msg;
-        return {
-          ...rest,
-          threadId: thread_id,
-          content: typeof msg.content === "string" ? (() => {
-            try {
-              return JSON.parse(msg.content);
-            } catch {
-              return msg.content;
-            }
-          })() : msg.content
-        };
-      });
-      const list = new agent.MessageList({ threadId, resourceId }).add(normalized, "memory");
+      const list = new agent.MessageList({ threadId, resourceId }).add(messages.map(this.normalizeMessage), "memory");
       if (format === "v2") return list.get.all.v2();
       return list.get.all.v1();
     } catch (error$1) {
@@ -449,6 +449,36 @@ var StoreMemoryLance = class extends storage.MemoryStorage {
       );
     }
   }
+  async getMessagesById({
+    messageIds,
+    format
+  }) {
+    if (messageIds.length === 0) return [];
+    try {
+      const table = await this.client.openTable(storage.TABLE_MESSAGES);
+      const quotedIds = messageIds.map((id) => `'${id}'`).join(", ");
+      const allRecords = await table.query().where(`id IN (${quotedIds})`).toArray();
+      const messages = processResultWithTypeConversion(
+        allRecords,
+        await getTableSchema({ tableName: storage.TABLE_MESSAGES, client: this.client })
+      );
+      const list = new agent.MessageList().add(messages.map(this.normalizeMessage), "memory");
+      if (format === `v1`) return list.get.all.v1();
+      return list.get.all.v2();
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "LANCE_STORE_GET_MESSAGES_BY_ID_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: {
+            messageIds: JSON.stringify(messageIds)
+          }
+        },
+        error$1
+      );
+    }
+  }
   async saveMessages(args) {
     try {
       const { messages, format = "v1" } = args;
@@ -2089,6 +2119,12 @@ var LanceStorage = class _LanceStorage extends storage.MastraStorage {
   }) {
     return this.stores.memory.getMessages({ threadId, resourceId, selectBy, format, threadConfig });
   }
+  async getMessagesById({
+    messageIds,
+    format
+  }) {
+    return this.stores.memory.getMessagesById({ messageIds, format });
+  }
   async saveMessages(args) {
     return this.stores.memory.saveMessages(args);
   }