@mastra/lance 0.0.0-vnext-20251104230439 → 0.0.0-vnext-20251119160359

package/dist/index.js

@@ -1,6 +1,6 @@
 import { connect, Index } from '@lancedb/lancedb';
 import { MastraError, ErrorCategory, ErrorDomain } from '@mastra/core/error';
-import { MastraStorage, StoreOperations, MemoryStorage, TABLE_THREADS, TABLE_MESSAGES, resolveMessageLimit, normalizePerPage, calculatePagination, TABLE_RESOURCES, ScoresStorage, TABLE_SCORERS, WorkflowsStorage, TABLE_WORKFLOW_SNAPSHOT, ensureDate } from '@mastra/core/storage';
+import { MastraStorage, StoreOperations, MemoryStorage, TABLE_THREADS, TABLE_MESSAGES, normalizePerPage, calculatePagination, TABLE_RESOURCES, ScoresStorage, TABLE_SCORERS, WorkflowsStorage, TABLE_WORKFLOW_SNAPSHOT, ensureDate } from '@mastra/core/storage';
 import { MessageList } from '@mastra/core/agent';
 import { Utf8, Float64, Binary, Float32, Int32, Field, Schema } from 'apache-arrow';
 import { saveScorePayloadSchema } from '@mastra/core/evals';
@@ -261,66 +261,6 @@ var StoreMemoryLance = class extends MemoryStorage {
       })() : message.content
     };
   }
-  async getMessages({
-    threadId,
-    resourceId,
-    selectBy,
-    threadConfig
-  }) {
-    try {
-      if (!threadId.trim()) throw new Error("threadId must be a non-empty string");
-      if (threadConfig) {
-        throw new Error("ThreadConfig is not supported by LanceDB storage");
-      }
-      const limit = resolveMessageLimit({ last: selectBy?.last, defaultLimit: Number.MAX_SAFE_INTEGER });
-      const table = await this.client.openTable(TABLE_MESSAGES);
-      let allRecords = [];
-      if (selectBy?.include && selectBy.include.length > 0) {
-        const threadIds = [...new Set(selectBy.include.map((item) => item.threadId))];
-        for (const threadId2 of threadIds) {
-          const threadQuery = table.query().where(`thread_id = '${threadId2}'`);
-          let threadRecords = await threadQuery.toArray();
-          allRecords.push(...threadRecords);
-        }
-      } else {
-        let query = table.query().where(`\`thread_id\` = '${threadId}'`);
-        allRecords = await query.toArray();
-      }
-      allRecords.sort((a, b) => {
-        const dateA = new Date(a.createdAt).getTime();
-        const dateB = new Date(b.createdAt).getTime();
-        return dateA - dateB;
-      });
-      if (selectBy?.include && selectBy.include.length > 0) {
-        allRecords = this.processMessagesWithContext(allRecords, selectBy.include);
-      }
-      if (limit !== Number.MAX_SAFE_INTEGER) {
-        allRecords = allRecords.slice(-limit);
-      }
-      const messages = processResultWithTypeConversion(
-        allRecords,
-        await getTableSchema({ tableName: TABLE_MESSAGES, client: this.client })
-      );
-      const list = new MessageList({ threadId, resourceId }).add(
-        messages.map(this.normalizeMessage),
-        "memory"
-      );
-      return { messages: list.get.all.db() };
-    } catch (error) {
-      throw new MastraError(
-        {
-          id: "LANCE_STORE_GET_MESSAGES_FAILED",
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          details: {
-            threadId,
-            resourceId: resourceId ?? ""
-          }
-        },
-        error
-      );
-    }
-  }
   async listMessagesById({ messageIds }) {
     if (messageIds.length === 0) return { messages: [] };
     try {
@@ -377,7 +317,7 @@ var StoreMemoryLance = class extends MemoryStorage {
           new Error("page must be >= 0")
         );
       }
-      const { field, direction } = this.parseOrderBy(orderBy);
+      const { field, direction } = this.parseOrderBy(orderBy, "ASC");
       const table = await this.client.openTable(TABLE_MESSAGES);
       const conditions = [`thread_id = '${this.escapeSql(threadId)}'`];
       if (resourceId) {
@@ -1613,11 +1553,13 @@ var StoreWorkflowsLance = class extends WorkflowsStorage {
       } else {
         createdAt = now;
       }
+      const { status, value, ...rest } = snapshot;
       const record = {
         workflow_name: workflowName,
         run_id: runId,
         resourceId,
-        snapshot: JSON.stringify(snapshot),
+        snapshot: JSON.stringify({ status, value, ...rest }),
+        // this is to ensure status is always just before value, for when querying the db by status
         createdAt,
         updatedAt: now
       };
@@ -1687,6 +1629,10 @@ var StoreWorkflowsLance = class extends WorkflowsStorage {
       if (args?.workflowName) {
         conditions.push(`workflow_name = '${args.workflowName.replace(/'/g, "''")}'`);
       }
+      if (args?.status) {
+        const escapedStatus = args.status.replace(/\\/g, "\\\\").replace(/'/g, "''").replace(/%/g, "\\%").replace(/_/g, "\\_");
+        conditions.push(`\`snapshot\` LIKE '%"status":"${escapedStatus}","value"%'`);
+      }
       if (args?.resourceId) {
         conditions.push(`\`resourceId\` = '${args.resourceId}'`);
       }
@@ -1745,6 +1691,8 @@ var LanceStorage = class _LanceStorage extends MastraStorage {
   lanceClient;
   /**
    * Creates a new instance of LanceStorage
+   * @param id The unique identifier for this storage instance
+   * @param name The name for this storage instance
    * @param uri The URI to connect to LanceDB
    * @param options connection options
    *
@@ -1752,21 +1700,21 @@ var LanceStorage = class _LanceStorage extends MastraStorage {
    *
    * Connect to a local database
    * ```ts
-   * const store = await LanceStorage.create('/path/to/db');
+   * const store = await LanceStorage.create('my-storage-id', 'MyStorage', '/path/to/db');
    * ```
    *
    * Connect to a LanceDB cloud database
    * ```ts
-   * const store = await LanceStorage.create('db://host:port');
+   * const store = await LanceStorage.create('my-storage-id', 'MyStorage', 'db://host:port');
    * ```
    *
    * Connect to a cloud database
    * ```ts
-   * const store = await LanceStorage.create('s3://bucket/db', { storageOptions: { timeout: '60s' } });
+   * const store = await LanceStorage.create('my-storage-id', 'MyStorage', 's3://bucket/db', { storageOptions: { timeout: '60s' } });
    * ```
    */
-  static async create(name, uri, options) {
-    const instance = new _LanceStorage(name);
+  static async create(id, name, uri, options) {
+    const instance = new _LanceStorage(id, name);
     try {
       instance.lanceClient = await connect(uri, options);
       const operations = new StoreOperationsLance({ client: instance.lanceClient });
@@ -1794,8 +1742,8 @@ var LanceStorage = class _LanceStorage extends MastraStorage {
    * @internal
    * Private constructor to enforce using the create factory method
    */
-  constructor(name) {
-    super({ name });
+  constructor(id, name) {
+    super({ id, name });
     const operations = new StoreOperationsLance({ client: this.lanceClient });
     this.stores = {
       operations: new StoreOperationsLance({ client: this.lanceClient }),
@@ -1924,14 +1872,6 @@ var LanceStorage = class _LanceStorage extends MastraStorage {
     });
     return Array.from(allIndices).sort((a, b) => a - b).map((index) => records[index]);
   }
-  async getMessages({
-    threadId,
-    resourceId,
-    selectBy,
-    threadConfig
-  }) {
-    return this.stores.memory.getMessages({ threadId, resourceId, selectBy, threadConfig });
-  }
   async listMessagesById({ messageIds }) {
     return this.stores.memory.listMessagesById({ messageIds });
   }
@@ -2359,7 +2299,7 @@ var LanceVectorStore = class _LanceVectorStore extends MastraVector {
    * ```
    */
   static async create(uri, options) {
-    const instance = new _LanceVectorStore();
+    const instance = new _LanceVectorStore(options?.id || crypto.randomUUID());
     try {
       instance.lanceClient = await connect(uri, options);
       return instance;
@@ -2379,8 +2319,8 @@ var LanceVectorStore = class _LanceVectorStore extends MastraVector {
    * @internal
    * Private constructor to enforce using the create factory method
    */
-  constructor() {
-    super();
+  constructor(id) {
+    super({ id });
   }
   close() {
     if (this.lanceClient) {