@mastra/lance 1.0.0-beta.2 → 1.0.0-beta.3

package/dist/index.js

@@ -292,15 +292,16 @@ var StoreMemoryLance = class extends MemoryStorage {
   }
   async listMessages(args) {
     const { threadId, resourceId, include, filter, perPage: perPageInput, page = 0, orderBy } = args;
-    if (!threadId.trim()) {
+    const threadIds = Array.isArray(threadId) ? threadId : [threadId];
+    if (threadIds.length === 0 || threadIds.some((id) => !id.trim())) {
       throw new MastraError(
         {
           id: "STORAGE_LANCE_LIST_MESSAGES_INVALID_THREAD_ID",
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
-          details: { threadId }
+          details: { threadId: Array.isArray(threadId) ? threadId.join(",") : threadId }
         },
-        new Error("threadId must be a non-empty string")
+        new Error("threadId must be a non-empty string or array of non-empty strings")
       );
     }
     const perPage = normalizePerPage(perPageInput, 40);
@@ -319,7 +320,8 @@ var StoreMemoryLance = class extends MemoryStorage {
       }
       const { field, direction } = this.parseOrderBy(orderBy, "ASC");
       const table = await this.client.openTable(TABLE_MESSAGES);
-      const conditions = [`thread_id = '${this.escapeSql(threadId)}'`];
+      const threadCondition = threadIds.length === 1 ? `thread_id = '${this.escapeSql(threadIds[0])}'` : `thread_id IN (${threadIds.map((t) => `'${this.escapeSql(t)}'`).join(", ")})`;
+      const conditions = [threadCondition];
       if (resourceId) {
         conditions.push(`\`resourceId\` = '${this.escapeSql(resourceId)}'`);
       }
@@ -359,9 +361,9 @@ var StoreMemoryLance = class extends MemoryStorage {
       }
       const messageIds = new Set(messages.map((m) => m.id));
       if (include && include.length > 0) {
-        const threadIds = [...new Set(include.map((item) => item.threadId || threadId))];
+        const threadIds2 = [...new Set(include.map((item) => item.threadId || threadId))];
         const allThreadMessages = [];
-        for (const tid of threadIds) {
+        for (const tid of threadIds2) {
           const threadQuery = table.query().where(`thread_id = '${tid}'`);
           let threadRecords = await threadQuery.toArray();
           allThreadMessages.push(...threadRecords);
@@ -407,7 +409,7 @@ var StoreMemoryLance = class extends MemoryStorage {
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
-            threadId,
+            threadId: Array.isArray(threadId) ? threadId.join(",") : threadId,
             resourceId: resourceId ?? ""
           }
         },
@@ -1266,6 +1268,8 @@ var StoreScoresLance = class extends ScoresStorage {
           filteredScore[key] = JSON.stringify(filteredScore[key]);
         }
       }
+      filteredScore.createdAt = /* @__PURE__ */ new Date();
+      filteredScore.updatedAt = /* @__PURE__ */ new Date();
       filteredScore.id = id;
       await table.add([filteredScore], { mode: "append" });
       return { score };
@@ -1288,8 +1292,7 @@ var StoreScoresLance = class extends ScoresStorage {
       const query = table.query().where(`id = '${id}'`).limit(1);
       const records = await query.toArray();
       if (records.length === 0) return null;
-      const schema = await getTableSchema({ tableName: TABLE_SCORERS, client: this.client });
-      return processResultWithTypeConversion(records[0], schema);
+      return await this.transformScoreRow(records[0]);
     } catch (error) {
       throw new MastraError(
         {
@@ -1303,6 +1306,22 @@ var StoreScoresLance = class extends ScoresStorage {
       );
     }
   }
+  /**
+   * LanceDB-specific score row transformation.
+   *
+   * Note: This implementation does NOT use coreTransformScoreRow because:
+   * 1. LanceDB stores schema information in the table itself (requires async fetch)
+   * 2. Uses processResultWithTypeConversion utility for LanceDB-specific type handling
+   */
+  async transformScoreRow(row) {
+    const schema = await getTableSchema({ tableName: TABLE_SCORERS, client: this.client });
+    const transformed = processResultWithTypeConversion(row, schema);
+    return {
+      ...transformed,
+      createdAt: row.createdAt,
+      updatedAt: row.updatedAt
+    };
+  }
   async listScoresByScorerId({
     scorerId,
     pagination,
@@ -1343,8 +1362,7 @@ var StoreScoresLance = class extends ScoresStorage {
         if (start > 0) query = query.offset(start);
       }
       const records = await query.toArray();
-      const schema = await getTableSchema({ tableName: TABLE_SCORERS, client: this.client });
-      const scores = processResultWithTypeConversion(records, schema);
+      const scores = await Promise.all(records.map(async (record) => await this.transformScoreRow(record)));
       return {
         pagination: {
           page,
@@ -1385,8 +1403,7 @@ var StoreScoresLance = class extends ScoresStorage {
         if (start > 0) query = query.offset(start);
       }
       const records = await query.toArray();
-      const schema = await getTableSchema({ tableName: TABLE_SCORERS, client: this.client });
-      const scores = processResultWithTypeConversion(records, schema);
+      const scores = await Promise.all(records.map(async (record) => await this.transformScoreRow(record)));
       return {
         pagination: {
           page,
@@ -1428,8 +1445,7 @@ var StoreScoresLance = class extends ScoresStorage {
         if (start > 0) query = query.offset(start);
       }
       const records = await query.toArray();
-      const schema = await getTableSchema({ tableName: TABLE_SCORERS, client: this.client });
-      const scores = processResultWithTypeConversion(records, schema);
+      const scores = await Promise.all(records.map(async (record) => await this.transformScoreRow(record)));
       return {
         pagination: {
           page,
@@ -1471,8 +1487,7 @@ var StoreScoresLance = class extends ScoresStorage {
         if (start > 0) query = query.offset(start);
       }
       const records = await query.toArray();
-      const schema = await getTableSchema({ tableName: TABLE_SCORERS, client: this.client });
-      const scores = processResultWithTypeConversion(records, schema);
+      const scores = await Promise.all(records.map(async (record) => await this.transformScoreRow(record)));
       return {
         pagination: {
           page,