@mastra/lance 0.2.0 → 0.2.1-alpha.0

package/dist/index.cjs

@@ -1,267 +1,1104 @@
 'use strict';
 
 var lancedb = require('@lancedb/lancedb');
-var agent = require('@mastra/core/agent');
 var error = require('@mastra/core/error');
 var storage = require('@mastra/core/storage');
+var agent = require('@mastra/core/agent');
 var apacheArrow = require('apache-arrow');
 var vector = require('@mastra/core/vector');
 var filter = require('@mastra/core/vector/filter');
 
 // src/storage/index.ts
-var LanceStorage = class _LanceStorage extends storage.MastraStorage {
-  lanceClient;
-  /**
-   * Creates a new instance of LanceStorage
-   * @param uri The URI to connect to LanceDB
-   * @param options connection options
-   *
-   * Usage:
-   *
-   * Connect to a local database
-   * ```ts
-   * const store = await LanceStorage.create('/path/to/db');
-   * ```
-   *
-   * Connect to a LanceDB cloud database
-   * ```ts
-   * const store = await LanceStorage.create('db://host:port');
-   * ```
-   *
-   * Connect to a cloud database
-   * ```ts
-   * const store = await LanceStorage.create('s3://bucket/db', { storageOptions: { timeout: '60s' } });
-   * ```
-   */
-  static async create(name, uri, options) {
-    const instance = new _LanceStorage(name);
-    try {
-      instance.lanceClient = await lancedb.connect(uri, options);
-      return instance;
-    } catch (e) {
-      throw new error.MastraError(
-        {
-          id: "STORAGE_LANCE_STORAGE_CONNECT_FAILED",
-          domain: error.ErrorDomain.STORAGE,
-          category: error.ErrorCategory.THIRD_PARTY,
-          text: `Failed to connect to LanceDB: ${e.message || e}`,
-          details: { uri, optionsProvided: !!options }
-        },
-        e
-      );
-    }
-  }
-  getPrimaryKeys(tableName) {
-    let primaryId = ["id"];
-    if (tableName === storage.TABLE_WORKFLOW_SNAPSHOT) {
-      primaryId = ["workflow_name", "run_id"];
-    } else if (tableName === storage.TABLE_EVALS) {
-      primaryId = ["agent_name", "metric_name", "run_id"];
-    }
-    return primaryId;
-  }
-  /**
-   * @internal
-   * Private constructor to enforce using the create factory method
-   */
-  constructor(name) {
-    super({ name });
+var StoreLegacyEvalsLance = class extends storage.LegacyEvalsStorage {
+  client;
+  constructor({ client }) {
+    super();
+    this.client = client;
   }
-  async createTable({
-    tableName,
-    schema
-  }) {
+  async getEvalsByAgentName(agentName, type) {
     try {
-      if (!this.lanceClient) {
-        throw new Error("LanceDB client not initialized. Call LanceStorage.create() first.");
-      }
-      if (!tableName) {
-        throw new Error("tableName is required for createTable.");
-      }
-      if (!schema) {
-        throw new Error("schema is required for createTable.");
+      const table = await this.client.openTable(storage.TABLE_EVALS);
+      const query = table.query().where(`agent_name = '${agentName}'`);
+      const records = await query.toArray();
+      let filteredRecords = records;
+      if (type === "live") {
+        filteredRecords = records.filter((record) => record.test_info === null);
+      } else if (type === "test") {
+        filteredRecords = records.filter((record) => record.test_info !== null);
       }
+      return filteredRecords.map((record) => {
+        return {
+          id: record.id,
+          input: record.input,
+          output: record.output,
+          agentName: record.agent_name,
+          metricName: record.metric_name,
+          result: JSON.parse(record.result),
+          instructions: record.instructions,
+          testInfo: record.test_info ? JSON.parse(record.test_info) : null,
+          globalRunId: record.global_run_id,
+          runId: record.run_id,
+          createdAt: new Date(record.created_at).toString()
+        };
+      });
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "STORAGE_LANCE_STORAGE_CREATE_TABLE_INVALID_ARGS",
+          id: "LANCE_STORE_GET_EVALS_BY_AGENT_NAME_FAILED",
           domain: error.ErrorDomain.STORAGE,
-          category: error.ErrorCategory.USER,
-          details: { tableName }
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { agentName }
         },
         error$1
       );
     }
+  }
+  async getEvals(options) {
     try {
-      const arrowSchema = this.translateSchema(schema);
-      await this.lanceClient.createEmptyTable(tableName, arrowSchema);
+      const table = await this.client.openTable(storage.TABLE_EVALS);
+      const conditions = [];
+      if (options.agentName) {
+        conditions.push(`agent_name = '${options.agentName}'`);
+      }
+      if (options.type === "live") {
+        conditions.push("length(test_info) = 0");
+      } else if (options.type === "test") {
+        conditions.push("length(test_info) > 0");
+      }
+      const startDate = options.dateRange?.start || options.fromDate;
+      const endDate = options.dateRange?.end || options.toDate;
+      if (startDate) {
+        conditions.push(`\`created_at\` >= ${startDate.getTime()}`);
+      }
+      if (endDate) {
+        conditions.push(`\`created_at\` <= ${endDate.getTime()}`);
+      }
+      let total = 0;
+      if (conditions.length > 0) {
+        total = await table.countRows(conditions.join(" AND "));
+      } else {
+        total = await table.countRows();
+      }
+      const query = table.query();
+      if (conditions.length > 0) {
+        const whereClause = conditions.join(" AND ");
+        query.where(whereClause);
+      }
+      const records = await query.toArray();
+      const evals = records.sort((a, b) => b.created_at - a.created_at).map((record) => {
+        return {
+          id: record.id,
+          input: record.input,
+          output: record.output,
+          agentName: record.agent_name,
+          metricName: record.metric_name,
+          result: JSON.parse(record.result),
+          instructions: record.instructions,
+          testInfo: record.test_info ? JSON.parse(record.test_info) : null,
+          globalRunId: record.global_run_id,
+          runId: record.run_id,
+          createdAt: new Date(record.created_at).toISOString()
+        };
+      });
+      const page = options.page || 0;
+      const perPage = options.perPage || 10;
+      const pagedEvals = evals.slice(page * perPage, (page + 1) * perPage);
+      return {
+        evals: pagedEvals,
+        total,
+        page,
+        perPage,
+        hasMore: total > (page + 1) * perPage
+      };
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "STORAGE_LANCE_STORAGE_CREATE_TABLE_FAILED",
+          id: "LANCE_STORE_GET_EVALS_FAILED",
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
-          details: { tableName }
+          details: { agentName: options.agentName ?? "" }
         },
         error$1
       );
     }
   }
-  translateSchema(schema) {
-    const fields = Object.entries(schema).map(([name, column]) => {
-      let arrowType;
-      switch (column.type.toLowerCase()) {
-        case "text":
-        case "uuid":
-          arrowType = new apacheArrow.Utf8();
-          break;
-        case "int":
-        case "integer":
-          arrowType = new apacheArrow.Int32();
-          break;
-        case "bigint":
-          arrowType = new apacheArrow.Float64();
-          break;
-        case "float":
-          arrowType = new apacheArrow.Float32();
-          break;
-        case "jsonb":
-        case "json":
-          arrowType = new apacheArrow.Utf8();
-          break;
-        case "binary":
-          arrowType = new apacheArrow.Binary();
-          break;
-        case "timestamp":
-          arrowType = new apacheArrow.Float64();
-          break;
-        default:
-          arrowType = new apacheArrow.Utf8();
+};
+function getPrimaryKeys(tableName) {
+  let primaryId = ["id"];
+  if (tableName === storage.TABLE_WORKFLOW_SNAPSHOT) {
+    primaryId = ["workflow_name", "run_id"];
+  } else if (tableName === storage.TABLE_EVALS) {
+    primaryId = ["agent_name", "metric_name", "run_id"];
+  }
+  return primaryId;
+}
+function validateKeyTypes(keys, tableSchema) {
+  const fieldTypes = new Map(
+    tableSchema.fields.map((field) => [field.name, field.type?.toString().toLowerCase()])
+  );
+  for (const [key, value] of Object.entries(keys)) {
+    const fieldType = fieldTypes.get(key);
+    if (!fieldType) {
+      throw new Error(`Field '${key}' does not exist in table schema`);
+    }
+    if (value !== null) {
+      if ((fieldType.includes("int") || fieldType.includes("bigint")) && typeof value !== "number") {
+        throw new Error(`Expected numeric value for field '${key}', got ${typeof value}`);
+      }
+      if (fieldType.includes("utf8") && typeof value !== "string") {
+        throw new Error(`Expected string value for field '${key}', got ${typeof value}`);
+      }
+      if (fieldType.includes("timestamp") && !(value instanceof Date) && typeof value !== "string") {
+        throw new Error(`Expected Date or string value for field '${key}', got ${typeof value}`);
+      }
+    }
+  }
+}
+function processResultWithTypeConversion(rawResult, tableSchema) {
+  const fieldTypeMap = /* @__PURE__ */ new Map();
+  tableSchema.fields.forEach((field) => {
+    const fieldName = field.name;
+    const fieldTypeStr = field.type.toString().toLowerCase();
+    fieldTypeMap.set(fieldName, fieldTypeStr);
+  });
+  if (Array.isArray(rawResult)) {
+    return rawResult.map((item) => processResultWithTypeConversion(item, tableSchema));
+  }
+  const processedResult = { ...rawResult };
+  for (const key in processedResult) {
+    const fieldTypeStr = fieldTypeMap.get(key);
+    if (!fieldTypeStr) continue;
+    if (typeof processedResult[key] === "string") {
+      if (fieldTypeStr.includes("int32") || fieldTypeStr.includes("float32")) {
+        if (!isNaN(Number(processedResult[key]))) {
+          processedResult[key] = Number(processedResult[key]);
+        }
+      } else if (fieldTypeStr.includes("int64")) {
+        processedResult[key] = Number(processedResult[key]);
+      } else if (fieldTypeStr.includes("utf8") && key !== "id") {
+        try {
+          const parsed = JSON.parse(processedResult[key]);
+          if (typeof parsed === "object") {
+            processedResult[key] = JSON.parse(processedResult[key]);
+          }
+        } catch {
+        }
       }
-      return new apacheArrow.Field(name, arrowType, column.nullable ?? true);
-    });
-    return new apacheArrow.Schema(fields);
+    } else if (typeof processedResult[key] === "bigint") {
+      processedResult[key] = Number(processedResult[key]);
+    } else if (fieldTypeStr.includes("float64") && ["createdAt", "updatedAt"].includes(key)) {
+      processedResult[key] = new Date(processedResult[key]);
+    }
+    console.log(key, "processedResult", processedResult);
   }
-  /**
-   * Drop a table if it exists
-   * @param tableName Name of the table to drop
-   */
-  async dropTable(tableName) {
-    try {
-      if (!this.lanceClient) {
-        throw new Error("LanceDB client not initialized. Call LanceStorage.create() first.");
+  return processedResult;
+}
+async function getTableSchema({
+  tableName,
+  client
+}) {
+  try {
+    if (!client) {
+      throw new Error("LanceDB client not initialized. Call LanceStorage.create() first.");
+    }
+    if (!tableName) {
+      throw new Error("tableName is required for getTableSchema.");
+    }
+  } catch (validationError) {
+    throw new error.MastraError(
+      {
+        id: "STORAGE_LANCE_STORAGE_GET_TABLE_SCHEMA_INVALID_ARGS",
+        domain: error.ErrorDomain.STORAGE,
+        category: error.ErrorCategory.USER,
+        text: validationError.message,
+        details: { tableName }
+      },
+      validationError
+    );
+  }
+  try {
+    const table = await client.openTable(tableName);
+    const rawSchema = await table.schema();
+    const fields = rawSchema.fields;
+    return {
+      fields,
+      metadata: /* @__PURE__ */ new Map(),
+      get names() {
+        return fields.map((field) => field.name);
       }
-      if (!tableName) {
-        throw new Error("tableName is required for dropTable.");
+    };
+  } catch (error$1) {
+    throw new error.MastraError(
+      {
+        id: "STORAGE_LANCE_STORAGE_GET_TABLE_SCHEMA_FAILED",
+        domain: error.ErrorDomain.STORAGE,
+        category: error.ErrorCategory.THIRD_PARTY,
+        details: { tableName }
+      },
+      error$1
+    );
+  }
+}
+
+// src/storage/domains/memory/index.ts
+var StoreMemoryLance = class extends storage.MemoryStorage {
+  client;
+  operations;
+  constructor({ client, operations }) {
+    super();
+    this.client = client;
+    this.operations = operations;
+  }
+  async getThreadById({ threadId }) {
+    try {
+      const thread = await this.operations.load({ tableName: storage.TABLE_THREADS, keys: { id: threadId } });
+      if (!thread) {
+        return null;
       }
-    } catch (validationError) {
+      return {
+        ...thread,
+        createdAt: new Date(thread.createdAt),
+        updatedAt: new Date(thread.updatedAt)
+      };
+    } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "STORAGE_LANCE_STORAGE_DROP_TABLE_INVALID_ARGS",
+          id: "LANCE_STORE_GET_THREAD_BY_ID_FAILED",
           domain: error.ErrorDomain.STORAGE,
-          category: error.ErrorCategory.USER,
-          text: validationError.message,
-          details: { tableName }
+          category: error.ErrorCategory.THIRD_PARTY
         },
-        validationError
+        error$1
       );
     }
+  }
+  async getThreadsByResourceId({ resourceId }) {
     try {
-      await this.lanceClient.dropTable(tableName);
+      const table = await this.client.openTable(storage.TABLE_THREADS);
+      const query = table.query().where(`\`resourceId\` = '${resourceId}'`);
+      const records = await query.toArray();
+      return processResultWithTypeConversion(
+        records,
+        await getTableSchema({ tableName: storage.TABLE_THREADS, client: this.client })
+      );
     } catch (error$1) {
-      if (error$1.toString().includes("was not found") || error$1.message?.includes("Table not found")) {
-        this.logger.debug(`Table '${tableName}' does not exist, skipping drop`);
-        return;
-      }
       throw new error.MastraError(
         {
-          id: "STORAGE_LANCE_STORAGE_DROP_TABLE_FAILED",
+          id: "LANCE_STORE_GET_THREADS_BY_RESOURCE_ID_FAILED",
           domain: error.ErrorDomain.STORAGE,
-          category: error.ErrorCategory.THIRD_PARTY,
-          details: { tableName }
+          category: error.ErrorCategory.THIRD_PARTY
         },
         error$1
       );
     }
   }
   /**
-   * Get table schema
-   * @param tableName Name of the table
-   * @returns Table schema
+   * Saves a thread to the database. This function doesn't overwrite existing threads.
+   * @param thread - The thread to save
+   * @returns The saved thread
    */
-  async getTableSchema(tableName) {
-    try {
-      if (!this.lanceClient) {
-        throw new Error("LanceDB client not initialized. Call LanceStorage.create() first.");
-      }
-      if (!tableName) {
-        throw new Error("tableName is required for getTableSchema.");
-      }
-    } catch (validationError) {
-      throw new error.MastraError(
-        {
-          id: "STORAGE_LANCE_STORAGE_GET_TABLE_SCHEMA_INVALID_ARGS",
-          domain: error.ErrorDomain.STORAGE,
-          category: error.ErrorCategory.USER,
-          text: validationError.message,
-          details: { tableName }
-        },
-        validationError
-      );
-    }
+  async saveThread({ thread }) {
     try {
-      const table = await this.lanceClient.openTable(tableName);
-      const rawSchema = await table.schema();
-      const fields = rawSchema.fields;
-      return {
-        fields,
-        metadata: /* @__PURE__ */ new Map(),
-        get names() {
-          return fields.map((field) => field.name);
-        }
-      };
+      const record = { ...thread, metadata: JSON.stringify(thread.metadata) };
+      const table = await this.client.openTable(storage.TABLE_THREADS);
+      await table.add([record], { mode: "append" });
+      return thread;
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "STORAGE_LANCE_STORAGE_GET_TABLE_SCHEMA_FAILED",
+          id: "LANCE_STORE_SAVE_THREAD_FAILED",
           domain: error.ErrorDomain.STORAGE,
-          category: error.ErrorCategory.THIRD_PARTY,
-          details: { tableName }
+          category: error.ErrorCategory.THIRD_PARTY
         },
         error$1
       );
     }
   }
-  getDefaultValue(type) {
-    switch (type) {
-      case "text":
-        return "''";
-      case "timestamp":
-        return "CURRENT_TIMESTAMP";
-      case "integer":
-      case "bigint":
-        return "0";
-      case "jsonb":
-        return "'{}'";
-      case "uuid":
+  async updateThread({
+    id,
+    title,
+    metadata
+  }) {
+    const maxRetries = 5;
+    for (let attempt = 0; attempt < maxRetries; attempt++) {
+      try {
+        const current = await this.getThreadById({ threadId: id });
+        if (!current) {
+          throw new Error(`Thread with id ${id} not found`);
+        }
+        const mergedMetadata = { ...current.metadata, ...metadata };
+        const record = {
+          id,
+          title,
+          metadata: JSON.stringify(mergedMetadata),
+          updatedAt: (/* @__PURE__ */ new Date()).getTime()
+        };
+        const table = await this.client.openTable(storage.TABLE_THREADS);
+        await table.mergeInsert("id").whenMatchedUpdateAll().whenNotMatchedInsertAll().execute([record]);
+        const updatedThread = await this.getThreadById({ threadId: id });
+        if (!updatedThread) {
+          throw new Error(`Failed to retrieve updated thread ${id}`);
+        }
+        return updatedThread;
+      } catch (error$1) {
+        if (error$1.message?.includes("Commit conflict") && attempt < maxRetries - 1) {
+          const delay = Math.pow(2, attempt) * 10;
+          await new Promise((resolve) => setTimeout(resolve, delay));
+          continue;
+        }
+        throw new error.MastraError(
+          {
+            id: "LANCE_STORE_UPDATE_THREAD_FAILED",
+            domain: error.ErrorDomain.STORAGE,
+            category: error.ErrorCategory.THIRD_PARTY
+          },
+          error$1
+        );
+      }
+    }
+    throw new error.MastraError(
+      {
+        id: "LANCE_STORE_UPDATE_THREAD_FAILED",
+        domain: error.ErrorDomain.STORAGE,
+        category: error.ErrorCategory.THIRD_PARTY
+      },
+      new Error("All retries exhausted")
+    );
+  }
+  async deleteThread({ threadId }) {
+    try {
+      const table = await this.client.openTable(storage.TABLE_THREADS);
+      await table.delete(`id = '${threadId}'`);
+      const messagesTable = await this.client.openTable(storage.TABLE_MESSAGES);
+      await messagesTable.delete(`thread_id = '${threadId}'`);
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "LANCE_STORE_DELETE_THREAD_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY
+        },
+        error$1
+      );
+    }
+  }
+  async getMessages({
+    threadId,
+    resourceId,
+    selectBy,
+    format,
+    threadConfig
+  }) {
+    try {
+      if (threadConfig) {
+        throw new Error("ThreadConfig is not supported by LanceDB storage");
+      }
+      const limit = storage.resolveMessageLimit({ last: selectBy?.last, defaultLimit: Number.MAX_SAFE_INTEGER });
+      const table = await this.client.openTable(storage.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: 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");
+      if (format === "v2") return list.get.all.v2();
+      return list.get.all.v1();
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "LANCE_STORE_GET_MESSAGES_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY
+        },
+        error$1
+      );
+    }
+  }
+  async saveMessages(args) {
+    try {
+      const { messages, format = "v1" } = args;
+      if (messages.length === 0) {
+        return [];
+      }
+      const threadId = messages[0]?.threadId;
+      if (!threadId) {
+        throw new Error("Thread ID is required");
+      }
+      for (const message of messages) {
+        if (!message.id) {
+          throw new Error("Message ID is required");
+        }
+        if (!message.threadId) {
+          throw new Error("Thread ID is required for all messages");
+        }
+        if (message.resourceId === null || message.resourceId === void 0) {
+          throw new Error("Resource ID cannot be null or undefined");
+        }
+        if (!message.content) {
+          throw new Error("Message content is required");
+        }
+      }
+      const transformedMessages = messages.map((message) => {
+        const { threadId: threadId2, type, ...rest } = message;
+        return {
+          ...rest,
+          thread_id: threadId2,
+          type: type ?? "v2",
+          content: JSON.stringify(message.content)
+        };
+      });
+      const table = await this.client.openTable(storage.TABLE_MESSAGES);
+      await table.mergeInsert("id").whenMatchedUpdateAll().whenNotMatchedInsertAll().execute(transformedMessages);
+      const threadsTable = await this.client.openTable(storage.TABLE_THREADS);
+      const currentTime = (/* @__PURE__ */ new Date()).getTime();
+      const updateRecord = { id: threadId, updatedAt: currentTime };
+      await threadsTable.mergeInsert("id").whenMatchedUpdateAll().whenNotMatchedInsertAll().execute([updateRecord]);
+      const list = new agent.MessageList().add(messages, "memory");
+      if (format === `v2`) return list.get.all.v2();
+      return list.get.all.v1();
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "LANCE_STORE_SAVE_MESSAGES_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY
+        },
+        error$1
+      );
+    }
+  }
+  async getThreadsByResourceIdPaginated(args) {
+    try {
+      const { resourceId, page = 0, perPage = 10 } = args;
+      const table = await this.client.openTable(storage.TABLE_THREADS);
+      const total = await table.countRows(`\`resourceId\` = '${resourceId}'`);
+      const query = table.query().where(`\`resourceId\` = '${resourceId}'`);
+      const offset = page * perPage;
+      query.limit(perPage);
+      if (offset > 0) {
+        query.offset(offset);
+      }
+      const records = await query.toArray();
+      records.sort((a, b) => new Date(b.updatedAt).getTime() - new Date(a.updatedAt).getTime());
+      const schema = await getTableSchema({ tableName: storage.TABLE_THREADS, client: this.client });
+      const threads = records.map((record) => processResultWithTypeConversion(record, schema));
+      return {
+        threads,
+        total,
+        page,
+        perPage,
+        hasMore: total > (page + 1) * perPage
+      };
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "LANCE_STORE_GET_THREADS_BY_RESOURCE_ID_PAGINATED_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY
+        },
+        error$1
+      );
+    }
+  }
+  /**
+   * Processes messages to include context messages based on withPreviousMessages and withNextMessages
+   * @param records - The sorted array of records to process
+   * @param include - The array of include specifications with context parameters
+   * @returns The processed array with context messages included
+   */
+  processMessagesWithContext(records, include) {
+    const messagesWithContext = include.filter((item) => item.withPreviousMessages || item.withNextMessages);
+    if (messagesWithContext.length === 0) {
+      return records;
+    }
+    const messageIndexMap = /* @__PURE__ */ new Map();
+    records.forEach((message, index) => {
+      messageIndexMap.set(message.id, index);
+    });
+    const additionalIndices = /* @__PURE__ */ new Set();
+    for (const item of messagesWithContext) {
+      const messageIndex = messageIndexMap.get(item.id);
+      if (messageIndex !== void 0) {
+        if (item.withPreviousMessages) {
+          const startIdx = Math.max(0, messageIndex - item.withPreviousMessages);
+          for (let i = startIdx; i < messageIndex; i++) {
+            additionalIndices.add(i);
+          }
+        }
+        if (item.withNextMessages) {
+          const endIdx = Math.min(records.length - 1, messageIndex + item.withNextMessages);
+          for (let i = messageIndex + 1; i <= endIdx; i++) {
+            additionalIndices.add(i);
+          }
+        }
+      }
+    }
+    if (additionalIndices.size === 0) {
+      return records;
+    }
+    const originalMatchIds = new Set(include.map((item) => item.id));
+    const allIndices = /* @__PURE__ */ new Set();
+    records.forEach((record, index) => {
+      if (originalMatchIds.has(record.id)) {
+        allIndices.add(index);
+      }
+    });
+    additionalIndices.forEach((index) => {
+      allIndices.add(index);
+    });
+    return Array.from(allIndices).sort((a, b) => a - b).map((index) => records[index]);
+  }
+  async getMessagesPaginated(args) {
+    try {
+      const { threadId, resourceId, selectBy, format = "v1" } = args;
+      if (!threadId) {
+        throw new Error("Thread ID is required for getMessagesPaginated");
+      }
+      const page = selectBy?.pagination?.page ?? 0;
+      const perPage = selectBy?.pagination?.perPage ?? 10;
+      const dateRange = selectBy?.pagination?.dateRange;
+      const fromDate = dateRange?.start;
+      const toDate = dateRange?.end;
+      const table = await this.client.openTable(storage.TABLE_MESSAGES);
+      const messages = [];
+      if (selectBy?.include && Array.isArray(selectBy.include)) {
+        const threadIds = [...new Set(selectBy.include.map((item) => item.threadId))];
+        const allThreadMessages = [];
+        for (const threadId2 of threadIds) {
+          const threadQuery = table.query().where(`thread_id = '${threadId2}'`);
+          let threadRecords = await threadQuery.toArray();
+          if (fromDate) threadRecords = threadRecords.filter((m) => m.createdAt >= fromDate.getTime());
+          if (toDate) threadRecords = threadRecords.filter((m) => m.createdAt <= toDate.getTime());
+          allThreadMessages.push(...threadRecords);
+        }
+        allThreadMessages.sort((a, b) => a.createdAt - b.createdAt);
+        const contextMessages = this.processMessagesWithContext(allThreadMessages, selectBy.include);
+        messages.push(...contextMessages);
+      }
+      const conditions = [`thread_id = '${threadId}'`];
+      if (resourceId) {
+        conditions.push(`\`resourceId\` = '${resourceId}'`);
+      }
+      if (fromDate) {
+        conditions.push(`\`createdAt\` >= ${fromDate.getTime()}`);
+      }
+      if (toDate) {
+        conditions.push(`\`createdAt\` <= ${toDate.getTime()}`);
+      }
+      let total = 0;
+      if (conditions.length > 0) {
+        total = await table.countRows(conditions.join(" AND "));
+      } else {
+        total = await table.countRows();
+      }
+      if (total === 0 && messages.length === 0) {
+        return {
+          messages: [],
+          total: 0,
+          page,
+          perPage,
+          hasMore: false
+        };
+      }
+      const excludeIds = messages.map((m) => m.id);
+      let selectedMessages = [];
+      if (selectBy?.last && selectBy.last > 0) {
+        const query = table.query();
+        if (conditions.length > 0) {
+          query.where(conditions.join(" AND "));
+        }
+        let records = await query.toArray();
+        records = records.sort((a, b) => a.createdAt - b.createdAt);
+        if (excludeIds.length > 0) {
+          records = records.filter((m) => !excludeIds.includes(m.id));
+        }
+        selectedMessages = records.slice(-selectBy.last);
+      } else {
+        const query = table.query();
+        if (conditions.length > 0) {
+          query.where(conditions.join(" AND "));
+        }
+        let records = await query.toArray();
+        records = records.sort((a, b) => a.createdAt - b.createdAt);
+        if (excludeIds.length > 0) {
+          records = records.filter((m) => !excludeIds.includes(m.id));
+        }
+        selectedMessages = records.slice(page * perPage, (page + 1) * perPage);
+      }
+      const allMessages = [...messages, ...selectedMessages];
+      const seen = /* @__PURE__ */ new Set();
+      const dedupedMessages = allMessages.filter((m) => {
+        const key = `${m.id}:${m.thread_id}`;
+        if (seen.has(key)) return false;
+        seen.add(key);
+        return true;
+      });
+      const formattedMessages = dedupedMessages.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().add(formattedMessages, "memory");
+      return {
+        messages: format === "v2" ? list.get.all.v2() : list.get.all.v1(),
+        total,
+        // Total should be the count of messages matching the filters
+        page,
+        perPage,
+        hasMore: total > (page + 1) * perPage
+      };
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "LANCE_STORE_GET_MESSAGES_PAGINATED_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY
+        },
+        error$1
+      );
+    }
+  }
+  /**
+   * Parse message data from LanceDB record format to MastraMessageV2 format
+   */
+  parseMessageData(data) {
+    const { thread_id, ...rest } = data;
+    return {
+      ...rest,
+      threadId: thread_id,
+      content: typeof data.content === "string" ? (() => {
+        try {
+          return JSON.parse(data.content);
+        } catch {
+          return data.content;
+        }
+      })() : data.content,
+      createdAt: new Date(data.createdAt),
+      updatedAt: new Date(data.updatedAt)
+    };
+  }
+  async updateMessages(args) {
+    const { messages } = args;
+    this.logger.debug("Updating messages", { count: messages.length });
+    if (!messages.length) {
+      return [];
+    }
+    const updatedMessages = [];
+    const affectedThreadIds = /* @__PURE__ */ new Set();
+    try {
+      for (const updateData of messages) {
+        const { id, ...updates } = updateData;
+        const existingMessage = await this.operations.load({ tableName: storage.TABLE_MESSAGES, keys: { id } });
+        if (!existingMessage) {
+          this.logger.warn("Message not found for update", { id });
+          continue;
+        }
+        const existingMsg = this.parseMessageData(existingMessage);
+        const originalThreadId = existingMsg.threadId;
+        affectedThreadIds.add(originalThreadId);
+        const updatePayload = {};
+        if ("role" in updates && updates.role !== void 0) updatePayload.role = updates.role;
+        if ("type" in updates && updates.type !== void 0) updatePayload.type = updates.type;
+        if ("resourceId" in updates && updates.resourceId !== void 0) updatePayload.resourceId = updates.resourceId;
+        if ("threadId" in updates && updates.threadId !== void 0 && updates.threadId !== null) {
+          updatePayload.thread_id = updates.threadId;
+          affectedThreadIds.add(updates.threadId);
+        }
+        if (updates.content) {
+          const existingContent = existingMsg.content;
+          let newContent = { ...existingContent };
+          if (updates.content.metadata !== void 0) {
+            newContent.metadata = {
+              ...existingContent.metadata || {},
+              ...updates.content.metadata || {}
+            };
+          }
+          if (updates.content.content !== void 0) {
+            newContent.content = updates.content.content;
+          }
+          if ("parts" in updates.content && updates.content.parts !== void 0) {
+            newContent.parts = updates.content.parts;
+          }
+          updatePayload.content = JSON.stringify(newContent);
+        }
+        await this.operations.insert({ tableName: storage.TABLE_MESSAGES, record: { id, ...updatePayload } });
+        const updatedMessage = await this.operations.load({ tableName: storage.TABLE_MESSAGES, keys: { id } });
+        if (updatedMessage) {
+          updatedMessages.push(this.parseMessageData(updatedMessage));
+        }
+      }
+      for (const threadId of affectedThreadIds) {
+        await this.operations.insert({
+          tableName: storage.TABLE_THREADS,
+          record: { id: threadId, updatedAt: Date.now() }
+        });
+      }
+      return updatedMessages;
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "LANCE_STORE_UPDATE_MESSAGES_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { count: messages.length }
+        },
+        error$1
+      );
+    }
+  }
+  async getResourceById({ resourceId }) {
+    try {
+      const resource = await this.operations.load({ tableName: storage.TABLE_RESOURCES, keys: { id: resourceId } });
+      if (!resource) {
+        return null;
+      }
+      let createdAt;
+      let updatedAt;
+      try {
+        if (resource.createdAt instanceof Date) {
+          createdAt = resource.createdAt;
+        } else if (typeof resource.createdAt === "string") {
+          createdAt = new Date(resource.createdAt);
+        } else if (typeof resource.createdAt === "number") {
+          createdAt = new Date(resource.createdAt);
+        } else {
+          createdAt = /* @__PURE__ */ new Date();
+        }
+        if (isNaN(createdAt.getTime())) {
+          createdAt = /* @__PURE__ */ new Date();
+        }
+      } catch {
+        createdAt = /* @__PURE__ */ new Date();
+      }
+      try {
+        if (resource.updatedAt instanceof Date) {
+          updatedAt = resource.updatedAt;
+        } else if (typeof resource.updatedAt === "string") {
+          updatedAt = new Date(resource.updatedAt);
+        } else if (typeof resource.updatedAt === "number") {
+          updatedAt = new Date(resource.updatedAt);
+        } else {
+          updatedAt = /* @__PURE__ */ new Date();
+        }
+        if (isNaN(updatedAt.getTime())) {
+          updatedAt = /* @__PURE__ */ new Date();
+        }
+      } catch {
+        updatedAt = /* @__PURE__ */ new Date();
+      }
+      let workingMemory = resource.workingMemory;
+      if (workingMemory === null || workingMemory === void 0) {
+        workingMemory = void 0;
+      } else if (workingMemory === "") {
+        workingMemory = "";
+      } else if (typeof workingMemory === "object") {
+        workingMemory = JSON.stringify(workingMemory);
+      }
+      let metadata = resource.metadata;
+      if (metadata === "" || metadata === null || metadata === void 0) {
+        metadata = void 0;
+      } else if (typeof metadata === "string") {
+        try {
+          metadata = JSON.parse(metadata);
+        } catch {
+          metadata = metadata;
+        }
+      }
+      return {
+        ...resource,
+        createdAt,
+        updatedAt,
+        workingMemory,
+        metadata
+      };
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "LANCE_STORE_GET_RESOURCE_BY_ID_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY
+        },
+        error$1
+      );
+    }
+  }
+  async saveResource({ resource }) {
+    try {
+      const record = {
+        ...resource,
+        metadata: resource.metadata ? JSON.stringify(resource.metadata) : "",
+        createdAt: resource.createdAt.getTime(),
+        // Store as timestamp (milliseconds)
+        updatedAt: resource.updatedAt.getTime()
+        // Store as timestamp (milliseconds)
+      };
+      const table = await this.client.openTable(storage.TABLE_RESOURCES);
+      await table.add([record], { mode: "append" });
+      return resource;
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "LANCE_STORE_SAVE_RESOURCE_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY
+        },
+        error$1
+      );
+    }
+  }
+  async updateResource({
+    resourceId,
+    workingMemory,
+    metadata
+  }) {
+    const maxRetries = 3;
+    for (let attempt = 0; attempt < maxRetries; attempt++) {
+      try {
+        const existingResource = await this.getResourceById({ resourceId });
+        if (!existingResource) {
+          const newResource = {
+            id: resourceId,
+            workingMemory,
+            metadata: metadata || {},
+            createdAt: /* @__PURE__ */ new Date(),
+            updatedAt: /* @__PURE__ */ new Date()
+          };
+          return this.saveResource({ resource: newResource });
+        }
+        const updatedResource = {
+          ...existingResource,
+          workingMemory: workingMemory !== void 0 ? workingMemory : existingResource.workingMemory,
+          metadata: {
+            ...existingResource.metadata,
+            ...metadata
+          },
+          updatedAt: /* @__PURE__ */ new Date()
+        };
+        const record = {
+          id: resourceId,
+          workingMemory: updatedResource.workingMemory || "",
+          metadata: updatedResource.metadata ? JSON.stringify(updatedResource.metadata) : "",
+          updatedAt: updatedResource.updatedAt.getTime()
+          // Store as timestamp (milliseconds)
+        };
+        const table = await this.client.openTable(storage.TABLE_RESOURCES);
+        await table.mergeInsert("id").whenMatchedUpdateAll().whenNotMatchedInsertAll().execute([record]);
+        return updatedResource;
+      } catch (error$1) {
+        if (error$1.message?.includes("Commit conflict") && attempt < maxRetries - 1) {
+          const delay = Math.pow(2, attempt) * 10;
+          await new Promise((resolve) => setTimeout(resolve, delay));
+          continue;
+        }
+        throw new error.MastraError(
+          {
+            id: "LANCE_STORE_UPDATE_RESOURCE_FAILED",
+            domain: error.ErrorDomain.STORAGE,
+            category: error.ErrorCategory.THIRD_PARTY
+          },
+          error$1
+        );
+      }
+    }
+    throw new Error("Unexpected end of retry loop");
+  }
+};
+var StoreOperationsLance = class extends storage.StoreOperations {
+  client;
+  constructor({ client }) {
+    super();
+    this.client = client;
+  }
+  getDefaultValue(type) {
+    switch (type) {
+      case "text":
+        return "''";
+      case "timestamp":
+        return "CURRENT_TIMESTAMP";
+      case "integer":
+      case "bigint":
+        return "0";
+      case "jsonb":
+        return "'{}'";
+      case "uuid":
         return "''";
       default:
         return super.getDefaultValue(type);
     }
   }
-  /**
-   * Alters table schema to add columns if they don't exist
-   * @param tableName Name of the table
-   * @param schema Schema of the table
-   * @param ifNotExists Array of column names to add if they don't exist
-   */
+  async hasColumn(tableName, columnName) {
+    const table = await this.client.openTable(tableName);
+    const schema = await table.schema();
+    return schema.fields.some((field) => field.name === columnName);
+  }
+  translateSchema(schema) {
+    const fields = Object.entries(schema).map(([name, column]) => {
+      let arrowType;
+      switch (column.type.toLowerCase()) {
+        case "text":
+        case "uuid":
+          arrowType = new apacheArrow.Utf8();
+          break;
+        case "int":
+        case "integer":
+          arrowType = new apacheArrow.Int32();
+          break;
+        case "bigint":
+          arrowType = new apacheArrow.Float64();
+          break;
+        case "float":
+          arrowType = new apacheArrow.Float32();
+          break;
+        case "jsonb":
+        case "json":
+          arrowType = new apacheArrow.Utf8();
+          break;
+        case "binary":
+          arrowType = new apacheArrow.Binary();
+          break;
+        case "timestamp":
+          arrowType = new apacheArrow.Float64();
+          break;
+        default:
+          arrowType = new apacheArrow.Utf8();
+      }
+      return new apacheArrow.Field(name, arrowType, column.nullable ?? true);
+    });
+    return new apacheArrow.Schema(fields);
+  }
+  async createTable({
+    tableName,
+    schema
+  }) {
+    try {
+      if (!this.client) {
+        throw new Error("LanceDB client not initialized. Call LanceStorage.create() first.");
+      }
+      if (!tableName) {
+        throw new Error("tableName is required for createTable.");
+      }
+      if (!schema) {
+        throw new Error("schema is required for createTable.");
+      }
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "STORAGE_LANCE_STORAGE_CREATE_TABLE_INVALID_ARGS",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.USER,
+          details: { tableName }
+        },
+        error$1
+      );
+    }
+    try {
+      const arrowSchema = this.translateSchema(schema);
+      await this.client.createEmptyTable(tableName, arrowSchema);
+    } catch (error$1) {
+      if (error$1.message?.includes("already exists")) {
+        this.logger.debug(`Table '${tableName}' already exists, skipping create`);
+        return;
+      }
+      throw new error.MastraError(
+        {
+          id: "STORAGE_LANCE_STORAGE_CREATE_TABLE_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { tableName }
+        },
+        error$1
+      );
+    }
+  }
+  async dropTable({ tableName }) {
+    try {
+      if (!this.client) {
+        throw new Error("LanceDB client not initialized. Call LanceStorage.create() first.");
+      }
+      if (!tableName) {
+        throw new Error("tableName is required for dropTable.");
+      }
+    } catch (validationError) {
+      throw new error.MastraError(
+        {
+          id: "STORAGE_LANCE_STORAGE_DROP_TABLE_INVALID_ARGS",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.USER,
+          text: validationError.message,
+          details: { tableName }
+        },
+        validationError
+      );
+    }
+    try {
+      await this.client.dropTable(tableName);
+    } catch (error$1) {
+      if (error$1.toString().includes("was not found") || error$1.message?.includes("Table not found")) {
+        this.logger.debug(`Table '${tableName}' does not exist, skipping drop`);
+        return;
+      }
+      throw new error.MastraError(
+        {
+          id: "STORAGE_LANCE_STORAGE_DROP_TABLE_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { tableName }
+        },
+        error$1
+      );
+    }
+  }
   async alterTable({
     tableName,
     schema,
     ifNotExists
   }) {
     try {
-      if (!this.lanceClient) {
+      if (!this.client) {
         throw new Error("LanceDB client not initialized. Call LanceStorage.create() first.");
       }
       if (!tableName) {
@@ -287,7 +1124,7 @@ var LanceStorage = class _LanceStorage extends storage.MastraStorage {
       );
     }
     try {
-      const table = await this.lanceClient.openTable(tableName);
+      const table = await this.client.openTable(tableName);
       const currentSchema = await table.schema();
       const existingFields = new Set(currentSchema.fields.map((f) => f.name));
       const typeMap = {
@@ -323,7 +1160,7 @@ var LanceStorage = class _LanceStorage extends storage.MastraStorage {
   }
   async clearTable({ tableName }) {
     try {
-      if (!this.lanceClient) {
+      if (!this.client) {
         throw new Error("LanceDB client not initialized. Call LanceStorage.create() first.");
       }
       if (!tableName) {
@@ -342,7 +1179,7 @@ var LanceStorage = class _LanceStorage extends storage.MastraStorage {
       );
     }
     try {
-      const table = await this.lanceClient.openTable(tableName);
+      const table = await this.client.openTable(tableName);
       await table.delete("1=1");
     } catch (error$1) {
       throw new error.MastraError(
@@ -356,14 +1193,9 @@ var LanceStorage = class _LanceStorage extends storage.MastraStorage {
       );
     }
   }
-  /**
-   * Insert a single record into a table. This function overwrites the existing record if it exists. Use this function for inserting records into tables with custom schemas.
-   * @param tableName The name of the table to insert into.
-   * @param record The record to insert.
-   */
   async insert({ tableName, record }) {
     try {
-      if (!this.lanceClient) {
+      if (!this.client) {
         throw new Error("LanceDB client not initialized. Call LanceStorage.create() first.");
       }
       if (!tableName) {
@@ -385,8 +1217,8 @@ var LanceStorage = class _LanceStorage extends storage.MastraStorage {
       );
     }
     try {
-      const table = await this.lanceClient.openTable(tableName);
-      const primaryId = this.getPrimaryKeys(tableName);
+      const table = await this.client.openTable(tableName);
+      const primaryId = getPrimaryKeys(tableName);
       const processedRecord = { ...record };
       for (const key in processedRecord) {
         if (processedRecord[key] !== null && typeof processedRecord[key] === "object" && !(processedRecord[key] instanceof Date)) {
@@ -394,6 +1226,7 @@ var LanceStorage = class _LanceStorage extends storage.MastraStorage {
           processedRecord[key] = JSON.stringify(processedRecord[key]);
         }
       }
+      console.log(await table.schema());
       await table.mergeInsert(primaryId).whenMatchedUpdateAll().whenNotMatchedInsertAll().execute([processedRecord]);
     } catch (error$1) {
       throw new error.MastraError(
@@ -407,14 +1240,9 @@ var LanceStorage = class _LanceStorage extends storage.MastraStorage {
       );
     }
   }
-  /**
-   * Insert multiple records into a table. This function overwrites the existing records if they exist. Use this function for inserting records into tables with custom schemas.
-   * @param tableName The name of the table to insert into.
-   * @param records The records to insert.
-   */
   async batchInsert({ tableName, records }) {
     try {
-      if (!this.lanceClient) {
+      if (!this.client) {
         throw new Error("LanceDB client not initialized. Call LanceStorage.create() first.");
       }
       if (!tableName) {
@@ -436,8 +1264,8 @@ var LanceStorage = class _LanceStorage extends storage.MastraStorage {
       );
     }
     try {
-      const table = await this.lanceClient.openTable(tableName);
-      const primaryId = this.getPrimaryKeys(tableName);
+      const table = await this.client.openTable(tableName);
+      const primaryId = getPrimaryKeys(tableName);
       const processedRecords = records.map((record) => {
         const processedRecord = { ...record };
         for (const key in processedRecord) {
@@ -448,6 +1276,7 @@ var LanceStorage = class _LanceStorage extends storage.MastraStorage {
         }
         return processedRecord;
       });
+      console.log(processedRecords);
       await table.mergeInsert(primaryId).whenMatchedUpdateAll().whenNotMatchedInsertAll().execute(processedRecords);
     } catch (error$1) {
       throw new error.MastraError(
@@ -461,16 +1290,9 @@ var LanceStorage = class _LanceStorage extends storage.MastraStorage {
       );
     }
   }
-  /**
-   * Load a record from the database by its key(s)
-   * @param tableName The name of the table to query
-   * @param keys Record of key-value pairs to use for lookup
-   * @throws Error if invalid types are provided for keys
-   * @returns The loaded record with proper type conversions, or null if not found
-   */
   async load({ tableName, keys }) {
     try {
-      if (!this.lanceClient) {
+      if (!this.client) {
         throw new Error("LanceDB client not initialized. Call LanceStorage.create() first.");
       }
       if (!tableName) {
@@ -492,11 +1314,11 @@ var LanceStorage = class _LanceStorage extends storage.MastraStorage {
       );
     }
     try {
-      const table = await this.lanceClient.openTable(tableName);
-      const tableSchema = await this.getTableSchema(tableName);
+      const table = await this.client.openTable(tableName);
+      const tableSchema = await getTableSchema({ tableName, client: this.client });
       const query = table.query();
       if (Object.keys(keys).length > 0) {
-        this.validateKeyTypes(keys, tableSchema);
+        validateKeyTypes(keys, tableSchema);
         const filterConditions = Object.entries(keys).map(([key, value]) => {
           const isCamelCase = /^[a-z][a-zA-Z]*$/.test(key) && /[A-Z]/.test(key);
           const quotedKey = isCamelCase ? `\`${key}\`` : key;
@@ -507,333 +1329,213 @@ var LanceStorage = class _LanceStorage extends storage.MastraStorage {
           } else {
             return `${quotedKey} = ${value}`;
           }
-        }).join(" AND ");
-        this.logger.debug("where clause generated: " + filterConditions);
-        query.where(filterConditions);
-      }
-      const result = await query.limit(1).toArray();
-      if (result.length === 0) {
-        this.logger.debug("No record found");
-        return null;
-      }
-      return this.processResultWithTypeConversion(result[0], tableSchema);
-    } catch (error$1) {
-      if (error$1 instanceof error.MastraError) throw error$1;
-      throw new error.MastraError(
-        {
-          id: "STORAGE_LANCE_STORAGE_LOAD_FAILED",
-          domain: error.ErrorDomain.STORAGE,
-          category: error.ErrorCategory.THIRD_PARTY,
-          details: { tableName, keyCount: Object.keys(keys).length, firstKey: Object.keys(keys)[0] ?? "" }
-        },
-        error$1
-      );
-    }
-  }
-  /**
-   * Validates that key types match the schema definition
-   * @param keys The keys to validate
-   * @param tableSchema The table schema to validate against
-   * @throws Error if a key has an incompatible type
-   */
-  validateKeyTypes(keys, tableSchema) {
-    const fieldTypes = new Map(
-      tableSchema.fields.map((field) => [field.name, field.type?.toString().toLowerCase()])
-    );
-    for (const [key, value] of Object.entries(keys)) {
-      const fieldType = fieldTypes.get(key);
-      if (!fieldType) {
-        throw new Error(`Field '${key}' does not exist in table schema`);
-      }
-      if (value !== null) {
-        if ((fieldType.includes("int") || fieldType.includes("bigint")) && typeof value !== "number") {
-          throw new Error(`Expected numeric value for field '${key}', got ${typeof value}`);
-        }
-        if (fieldType.includes("utf8") && typeof value !== "string") {
-          throw new Error(`Expected string value for field '${key}', got ${typeof value}`);
-        }
-        if (fieldType.includes("timestamp") && !(value instanceof Date) && typeof value !== "string") {
-          throw new Error(`Expected Date or string value for field '${key}', got ${typeof value}`);
-        }
-      }
-    }
-  }
-  /**
-   * Process a database result with appropriate type conversions based on the table schema
-   * @param rawResult The raw result object from the database
-   * @param tableSchema The schema of the table containing type information
-   * @returns Processed result with correct data types
-   */
-  processResultWithTypeConversion(rawResult, tableSchema) {
-    const fieldTypeMap = /* @__PURE__ */ new Map();
-    tableSchema.fields.forEach((field) => {
-      const fieldName = field.name;
-      const fieldTypeStr = field.type.toString().toLowerCase();
-      fieldTypeMap.set(fieldName, fieldTypeStr);
-    });
-    if (Array.isArray(rawResult)) {
-      return rawResult.map((item) => this.processResultWithTypeConversion(item, tableSchema));
-    }
-    const processedResult = { ...rawResult };
-    for (const key in processedResult) {
-      const fieldTypeStr = fieldTypeMap.get(key);
-      if (!fieldTypeStr) continue;
-      if (typeof processedResult[key] === "string") {
-        if (fieldTypeStr.includes("int32") || fieldTypeStr.includes("float32")) {
-          if (!isNaN(Number(processedResult[key]))) {
-            processedResult[key] = Number(processedResult[key]);
-          }
-        } else if (fieldTypeStr.includes("int64")) {
-          processedResult[key] = Number(processedResult[key]);
-        } else if (fieldTypeStr.includes("utf8")) {
-          try {
-            processedResult[key] = JSON.parse(processedResult[key]);
-          } catch (e) {
-            this.logger.debug(`Failed to parse JSON for key ${key}: ${e}`);
-          }
-        }
-      } else if (typeof processedResult[key] === "bigint") {
-        processedResult[key] = Number(processedResult[key]);
-      }
-    }
-    return processedResult;
-  }
-  getThreadById({ threadId }) {
-    try {
-      return this.load({ tableName: storage.TABLE_THREADS, keys: { id: threadId } });
-    } catch (error$1) {
-      throw new error.MastraError(
-        {
-          id: "LANCE_STORE_GET_THREAD_BY_ID_FAILED",
-          domain: error.ErrorDomain.STORAGE,
-          category: error.ErrorCategory.THIRD_PARTY
-        },
-        error$1
-      );
-    }
-  }
-  async getThreadsByResourceId({ resourceId }) {
-    try {
-      const table = await this.lanceClient.openTable(storage.TABLE_THREADS);
-      const query = table.query().where(`\`resourceId\` = '${resourceId}'`);
-      const records = await query.toArray();
-      return this.processResultWithTypeConversion(
-        records,
-        await this.getTableSchema(storage.TABLE_THREADS)
-      );
+        }).join(" AND ");
+        this.logger.debug("where clause generated: " + filterConditions);
+        query.where(filterConditions);
+      }
+      const result = await query.limit(1).toArray();
+      if (result.length === 0) {
+        this.logger.debug("No record found");
+        return null;
+      }
+      return processResultWithTypeConversion(result[0], tableSchema);
     } catch (error$1) {
+      if (error$1 instanceof error.MastraError) throw error$1;
       throw new error.MastraError(
         {
-          id: "LANCE_STORE_GET_THREADS_BY_RESOURCE_ID_FAILED",
+          id: "STORAGE_LANCE_STORAGE_LOAD_FAILED",
           domain: error.ErrorDomain.STORAGE,
-          category: error.ErrorCategory.THIRD_PARTY
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { tableName, keyCount: Object.keys(keys).length, firstKey: Object.keys(keys)[0] ?? "" }
         },
         error$1
       );
     }
   }
-  /**
-   * Saves a thread to the database. This function doesn't overwrite existing threads.
-   * @param thread - The thread to save
-   * @returns The saved thread
-   */
-  async saveThread({ thread }) {
+};
+var StoreScoresLance = class extends storage.ScoresStorage {
+  client;
+  constructor({ client }) {
+    super();
+    this.client = client;
+  }
+  async saveScore(score) {
     try {
-      const record = { ...thread, metadata: JSON.stringify(thread.metadata) };
-      const table = await this.lanceClient.openTable(storage.TABLE_THREADS);
-      await table.add([record], { mode: "append" });
-      return thread;
+      const table = await this.client.openTable(storage.TABLE_SCORERS);
+      const schema = await getTableSchema({ tableName: storage.TABLE_SCORERS, client: this.client });
+      const allowedFields = new Set(schema.fields.map((f) => f.name));
+      const filteredScore = {};
+      Object.keys(score).forEach((key) => {
+        if (allowedFields.has(key)) {
+          filteredScore[key] = score[key];
+        }
+      });
+      for (const key in filteredScore) {
+        if (filteredScore[key] !== null && typeof filteredScore[key] === "object" && !(filteredScore[key] instanceof Date)) {
+          filteredScore[key] = JSON.stringify(filteredScore[key]);
+        }
+      }
+      console.log("Saving score to LanceStorage:", filteredScore);
+      await table.add([filteredScore], { mode: "append" });
+      return { score };
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "LANCE_STORE_SAVE_THREAD_FAILED",
+          id: "LANCE_STORAGE_SAVE_SCORE_FAILED",
+          text: "Failed to save score in LanceStorage",
           domain: error.ErrorDomain.STORAGE,
-          category: error.ErrorCategory.THIRD_PARTY
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { error: error$1?.message }
         },
         error$1
       );
     }
   }
-  async updateThread({
-    id,
-    title,
-    metadata
-  }) {
+  async getScoreById({ id }) {
     try {
-      const record = { id, title, metadata: JSON.stringify(metadata) };
-      const table = await this.lanceClient.openTable(storage.TABLE_THREADS);
-      await table.mergeInsert("id").whenMatchedUpdateAll().whenNotMatchedInsertAll().execute([record]);
-      const query = table.query().where(`id = '${id}'`);
+      const table = await this.client.openTable(storage.TABLE_SCORERS);
+      const query = table.query().where(`id = '${id}'`).limit(1);
       const records = await query.toArray();
-      return this.processResultWithTypeConversion(
-        records[0],
-        await this.getTableSchema(storage.TABLE_THREADS)
-      );
+      if (records.length === 0) return null;
+      const schema = await getTableSchema({ tableName: storage.TABLE_SCORERS, client: this.client });
+      return processResultWithTypeConversion(records[0], schema);
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "LANCE_STORE_UPDATE_THREAD_FAILED",
+          id: "LANCE_STORAGE_GET_SCORE_BY_ID_FAILED",
+          text: "Failed to get score by id in LanceStorage",
           domain: error.ErrorDomain.STORAGE,
-          category: error.ErrorCategory.THIRD_PARTY
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { error: error$1?.message }
         },
         error$1
       );
     }
   }
-  async deleteThread({ threadId }) {
+  async getScoresByScorerId({
+    scorerId,
+    pagination
+  }) {
     try {
-      const table = await this.lanceClient.openTable(storage.TABLE_THREADS);
-      await table.delete(`id = '${threadId}'`);
+      const table = await this.client.openTable(storage.TABLE_SCORERS);
+      const { page = 0, perPage = 10 } = pagination || {};
+      const offset = page * perPage;
+      const query = table.query().where(`\`scorerId\` = '${scorerId}'`).limit(perPage);
+      if (offset > 0) query.offset(offset);
+      const records = await query.toArray();
+      const schema = await getTableSchema({ tableName: storage.TABLE_SCORERS, client: this.client });
+      const scores = processResultWithTypeConversion(records, schema);
+      const allRecords = await table.query().where(`\`scorerId\` = '${scorerId}'`).toArray();
+      const total = allRecords.length;
+      return {
+        pagination: {
+          page,
+          perPage,
+          total,
+          hasMore: offset + scores.length < total
+        },
+        scores
+      };
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "LANCE_STORE_DELETE_THREAD_FAILED",
+          id: "LANCE_STORAGE_GET_SCORES_BY_SCORER_ID_FAILED",
+          text: "Failed to get scores by scorerId in LanceStorage",
           domain: error.ErrorDomain.STORAGE,
-          category: error.ErrorCategory.THIRD_PARTY
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { error: error$1?.message }
         },
         error$1
       );
     }
   }
-  /**
-   * Processes messages to include context messages based on withPreviousMessages and withNextMessages
-   * @param records - The sorted array of records to process
-   * @param include - The array of include specifications with context parameters
-   * @returns The processed array with context messages included
-   */
-  processMessagesWithContext(records, include) {
-    const messagesWithContext = include.filter((item) => item.withPreviousMessages || item.withNextMessages);
-    if (messagesWithContext.length === 0) {
-      return records;
-    }
-    const messageIndexMap = /* @__PURE__ */ new Map();
-    records.forEach((message, index) => {
-      messageIndexMap.set(message.id, index);
-    });
-    const additionalIndices = /* @__PURE__ */ new Set();
-    for (const item of messagesWithContext) {
-      const messageIndex = messageIndexMap.get(item.id);
-      if (messageIndex !== void 0) {
-        if (item.withPreviousMessages) {
-          const startIdx = Math.max(0, messageIndex - item.withPreviousMessages);
-          for (let i = startIdx; i < messageIndex; i++) {
-            additionalIndices.add(i);
-          }
-        }
-        if (item.withNextMessages) {
-          const endIdx = Math.min(records.length - 1, messageIndex + item.withNextMessages);
-          for (let i = messageIndex + 1; i <= endIdx; i++) {
-            additionalIndices.add(i);
-          }
-        }
-      }
-    }
-    if (additionalIndices.size === 0) {
-      return records;
-    }
-    const originalMatchIds = new Set(include.map((item) => item.id));
-    const allIndices = /* @__PURE__ */ new Set();
-    records.forEach((record, index) => {
-      if (originalMatchIds.has(record.id)) {
-        allIndices.add(index);
-      }
-    });
-    additionalIndices.forEach((index) => {
-      allIndices.add(index);
-    });
-    return Array.from(allIndices).sort((a, b) => a - b).map((index) => records[index]);
-  }
-  async getMessages({
-    threadId,
-    resourceId,
-    selectBy,
-    format,
-    threadConfig
+  async getScoresByRunId({
+    runId,
+    pagination
   }) {
     try {
-      if (threadConfig) {
-        throw new Error("ThreadConfig is not supported by LanceDB storage");
-      }
-      const limit = this.resolveMessageLimit({ last: selectBy?.last, defaultLimit: Number.MAX_SAFE_INTEGER });
-      const table = await this.lanceClient.openTable(storage.TABLE_MESSAGES);
-      let query = table.query().where(`\`threadId\` = '${threadId}'`);
-      if (selectBy) {
-        if (selectBy.include && selectBy.include.length > 0) {
-          const includeIds = selectBy.include.map((item) => item.id);
-          const includeClause = includeIds.map((id) => `\`id\` = '${id}'`).join(" OR ");
-          query = query.where(`(\`threadId\` = '${threadId}' OR (${includeClause}))`);
-        }
-      }
-      let records = await query.toArray();
-      records.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) {
-        records = this.processMessagesWithContext(records, selectBy.include);
-      }
-      if (limit !== Number.MAX_SAFE_INTEGER) {
-        records = records.slice(-limit);
-      }
-      const messages = this.processResultWithTypeConversion(records, await this.getTableSchema(storage.TABLE_MESSAGES));
-      const normalized = messages.map((msg) => ({
-        ...msg,
-        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");
-      if (format === "v2") return list.get.all.v2();
-      return list.get.all.v1();
+      const table = await this.client.openTable(storage.TABLE_SCORERS);
+      const { page = 0, perPage = 10 } = pagination || {};
+      const offset = page * perPage;
+      const query = table.query().where(`\`runId\` = '${runId}'`).limit(perPage);
+      if (offset > 0) query.offset(offset);
+      const records = await query.toArray();
+      const schema = await getTableSchema({ tableName: storage.TABLE_SCORERS, client: this.client });
+      const scores = processResultWithTypeConversion(records, schema);
+      const allRecords = await table.query().where(`\`runId\` = '${runId}'`).toArray();
+      const total = allRecords.length;
+      return {
+        pagination: {
+          page,
+          perPage,
+          total,
+          hasMore: offset + scores.length < total
+        },
+        scores
+      };
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "LANCE_STORE_GET_MESSAGES_FAILED",
+          id: "LANCE_STORAGE_GET_SCORES_BY_RUN_ID_FAILED",
+          text: "Failed to get scores by runId in LanceStorage",
           domain: error.ErrorDomain.STORAGE,
-          category: error.ErrorCategory.THIRD_PARTY
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { error: error$1?.message }
         },
         error$1
       );
     }
   }
-  async saveMessages(args) {
+  async getScoresByEntityId({
+    entityId,
+    entityType,
+    pagination
+  }) {
     try {
-      const { messages, format = "v1" } = args;
-      if (messages.length === 0) {
-        return [];
-      }
-      const threadId = messages[0]?.threadId;
-      if (!threadId) {
-        throw new Error("Thread ID is required");
-      }
-      const transformedMessages = messages.map((message) => ({
-        ...message,
-        content: JSON.stringify(message.content)
-      }));
-      const table = await this.lanceClient.openTable(storage.TABLE_MESSAGES);
-      await table.mergeInsert("id").whenMatchedUpdateAll().whenNotMatchedInsertAll().execute(transformedMessages);
-      const list = new agent.MessageList().add(messages, "memory");
-      if (format === `v2`) return list.get.all.v2();
-      return list.get.all.v1();
+      const table = await this.client.openTable(storage.TABLE_SCORERS);
+      const { page = 0, perPage = 10 } = pagination || {};
+      const offset = page * perPage;
+      const query = table.query().where(`\`entityId\` = '${entityId}' AND \`entityType\` = '${entityType}'`).limit(perPage);
+      if (offset > 0) query.offset(offset);
+      const records = await query.toArray();
+      const schema = await getTableSchema({ tableName: storage.TABLE_SCORERS, client: this.client });
+      const scores = processResultWithTypeConversion(records, schema);
+      const allRecords = await table.query().where(`\`entityId\` = '${entityId}' AND \`entityType\` = '${entityType}'`).toArray();
+      const total = allRecords.length;
+      return {
+        pagination: {
+          page,
+          perPage,
+          total,
+          hasMore: offset + scores.length < total
+        },
+        scores
+      };
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "LANCE_STORE_SAVE_MESSAGES_FAILED",
+          id: "LANCE_STORAGE_GET_SCORES_BY_ENTITY_ID_FAILED",
+          text: "Failed to get scores by entityId and entityType in LanceStorage",
           domain: error.ErrorDomain.STORAGE,
-          category: error.ErrorCategory.THIRD_PARTY
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { error: error$1?.message }
         },
         error$1
       );
     }
   }
+};
+var StoreTracesLance = class extends storage.TracesStorage {
+  client;
+  operations;
+  constructor({ client, operations }) {
+    super();
+    this.client = client;
+    this.operations = operations;
+  }
   async saveTrace({ trace }) {
     try {
-      const table = await this.lanceClient.openTable(storage.TABLE_TRACES);
+      const table = await this.client.openTable(storage.TABLE_TRACES);
       const record = {
         ...trace,
         attributes: JSON.stringify(trace.attributes),
@@ -857,10 +1559,10 @@ var LanceStorage = class _LanceStorage extends storage.MastraStorage {
   }
   async getTraceById({ traceId }) {
     try {
-      const table = await this.lanceClient.openTable(storage.TABLE_TRACES);
+      const table = await this.client.openTable(storage.TABLE_TRACES);
       const query = table.query().where(`id = '${traceId}'`);
       const records = await query.toArray();
-      return this.processResultWithTypeConversion(records[0], await this.getTableSchema(storage.TABLE_TRACES));
+      return records[0];
     } catch (error$1) {
       throw new error.MastraError(
         {
@@ -880,7 +1582,7 @@ var LanceStorage = class _LanceStorage extends storage.MastraStorage {
     attributes
   }) {
     try {
-      const table = await this.lanceClient.openTable(storage.TABLE_TRACES);
+      const table = await this.client.openTable(storage.TABLE_TRACES);
       const query = table.query();
       if (name) {
         query.where(`name = '${name}'`);
@@ -898,17 +1600,23 @@ var LanceStorage = class _LanceStorage extends storage.MastraStorage {
       }
       const records = await query.toArray();
       return records.map((record) => {
-        return {
+        const processed = {
           ...record,
-          attributes: JSON.parse(record.attributes),
-          status: JSON.parse(record.status),
-          events: JSON.parse(record.events),
-          links: JSON.parse(record.links),
-          other: JSON.parse(record.other),
+          attributes: record.attributes ? JSON.parse(record.attributes) : {},
+          status: record.status ? JSON.parse(record.status) : {},
+          events: record.events ? JSON.parse(record.events) : [],
+          links: record.links ? JSON.parse(record.links) : [],
+          other: record.other ? JSON.parse(record.other) : {},
           startTime: new Date(record.startTime),
           endTime: new Date(record.endTime),
           createdAt: new Date(record.createdAt)
         };
+        if (processed.parentSpanId === null || processed.parentSpanId === void 0) {
+          processed.parentSpanId = "";
+        } else {
+          processed.parentSpanId = String(processed.parentSpanId);
+        }
+        return processed;
       });
     } catch (error$1) {
       throw new error.MastraError(
@@ -922,131 +1630,177 @@ var LanceStorage = class _LanceStorage extends storage.MastraStorage {
       );
     }
   }
-  async saveEvals({ evals }) {
+  async getTracesPaginated(args) {
     try {
-      const table = await this.lanceClient.openTable(storage.TABLE_EVALS);
-      const transformedEvals = evals.map((evalRecord) => ({
-        input: evalRecord.input,
-        output: evalRecord.output,
-        agent_name: evalRecord.agentName,
-        metric_name: evalRecord.metricName,
-        result: JSON.stringify(evalRecord.result),
-        instructions: evalRecord.instructions,
-        test_info: JSON.stringify(evalRecord.testInfo),
-        global_run_id: evalRecord.globalRunId,
-        run_id: evalRecord.runId,
-        created_at: new Date(evalRecord.createdAt).getTime()
-      }));
-      await table.add(transformedEvals, { mode: "append" });
-      return evals;
+      const table = await this.client.openTable(storage.TABLE_TRACES);
+      const query = table.query();
+      const conditions = [];
+      if (args.name) {
+        conditions.push(`name = '${args.name}'`);
+      }
+      if (args.scope) {
+        conditions.push(`scope = '${args.scope}'`);
+      }
+      if (args.attributes) {
+        const attributesStr = JSON.stringify(args.attributes);
+        conditions.push(`attributes LIKE '%${attributesStr.replace(/"/g, '\\"')}%'`);
+      }
+      if (args.dateRange?.start) {
+        conditions.push(`\`createdAt\` >= ${args.dateRange.start.getTime()}`);
+      }
+      if (args.dateRange?.end) {
+        conditions.push(`\`createdAt\` <= ${args.dateRange.end.getTime()}`);
+      }
+      if (conditions.length > 0) {
+        const whereClause = conditions.join(" AND ");
+        query.where(whereClause);
+      }
+      let total = 0;
+      if (conditions.length > 0) {
+        const countQuery = table.query().where(conditions.join(" AND "));
+        const allRecords = await countQuery.toArray();
+        total = allRecords.length;
+      } else {
+        total = await table.countRows();
+      }
+      const page = args.page || 0;
+      const perPage = args.perPage || 10;
+      const offset = page * perPage;
+      query.limit(perPage);
+      if (offset > 0) {
+        query.offset(offset);
+      }
+      const records = await query.toArray();
+      const traces = records.map((record) => {
+        const processed = {
+          ...record,
+          attributes: record.attributes ? JSON.parse(record.attributes) : {},
+          status: record.status ? JSON.parse(record.status) : {},
+          events: record.events ? JSON.parse(record.events) : [],
+          links: record.links ? JSON.parse(record.links) : [],
+          other: record.other ? JSON.parse(record.other) : {},
+          startTime: new Date(record.startTime),
+          endTime: new Date(record.endTime),
+          createdAt: new Date(record.createdAt)
+        };
+        if (processed.parentSpanId === null || processed.parentSpanId === void 0) {
+          processed.parentSpanId = "";
+        } else {
+          processed.parentSpanId = String(processed.parentSpanId);
+        }
+        return processed;
+      });
+      return {
+        traces,
+        total,
+        page,
+        perPage,
+        hasMore: total > (page + 1) * perPage
+      };
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "LANCE_STORE_SAVE_EVALS_FAILED",
+          id: "LANCE_STORE_GET_TRACES_PAGINATED_FAILED",
           domain: error.ErrorDomain.STORAGE,
-          category: error.ErrorCategory.THIRD_PARTY
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { name: args.name ?? "", scope: args.scope ?? "" }
         },
         error$1
       );
     }
   }
-  async getEvalsByAgentName(agentName, type) {
+  async batchTraceInsert({ records }) {
+    this.logger.debug("Batch inserting traces", { count: records.length });
+    await this.operations.batchInsert({
+      tableName: storage.TABLE_TRACES,
+      records
+    });
+  }
+};
+function parseWorkflowRun(row) {
+  let parsedSnapshot = row.snapshot;
+  if (typeof parsedSnapshot === "string") {
     try {
-      if (type) {
-        this.logger.warn("Type is not implemented yet in LanceDB storage");
-      }
-      const table = await this.lanceClient.openTable(storage.TABLE_EVALS);
-      const query = table.query().where(`agent_name = '${agentName}'`);
+      parsedSnapshot = JSON.parse(row.snapshot);
+    } catch (e) {
+      console.warn(`Failed to parse snapshot for workflow ${row.workflow_name}: ${e}`);
+    }
+  }
+  return {
+    workflowName: row.workflow_name,
+    runId: row.run_id,
+    snapshot: parsedSnapshot,
+    createdAt: storage.ensureDate(row.createdAt),
+    updatedAt: storage.ensureDate(row.updatedAt),
+    resourceId: row.resourceId
+  };
+}
+var StoreWorkflowsLance = class extends storage.WorkflowsStorage {
+  client;
+  constructor({ client }) {
+    super();
+    this.client = client;
+  }
+  async persistWorkflowSnapshot({
+    workflowName,
+    runId,
+    snapshot
+  }) {
+    try {
+      const table = await this.client.openTable(storage.TABLE_WORKFLOW_SNAPSHOT);
+      const query = table.query().where(`workflow_name = '${workflowName}' AND run_id = '${runId}'`);
       const records = await query.toArray();
-      return records.map((record) => {
-        return {
-          id: record.id,
-          input: record.input,
-          output: record.output,
-          agentName: record.agent_name,
-          metricName: record.metric_name,
-          result: JSON.parse(record.result),
-          instructions: record.instructions,
-          testInfo: JSON.parse(record.test_info),
-          globalRunId: record.global_run_id,
-          runId: record.run_id,
-          createdAt: new Date(record.created_at).toString()
-        };
-      });
+      let createdAt;
+      const now = Date.now();
+      if (records.length > 0) {
+        createdAt = records[0].createdAt ?? now;
+      } else {
+        createdAt = now;
+      }
+      const record = {
+        workflow_name: workflowName,
+        run_id: runId,
+        snapshot: JSON.stringify(snapshot),
+        createdAt,
+        updatedAt: now
+      };
+      await table.mergeInsert(["workflow_name", "run_id"]).whenMatchedUpdateAll().whenNotMatchedInsertAll().execute([record]);
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "LANCE_STORE_GET_EVALS_BY_AGENT_NAME_FAILED",
+          id: "LANCE_STORE_PERSIST_WORKFLOW_SNAPSHOT_FAILED",
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
-          details: { agentName }
+          details: { workflowName, runId }
         },
         error$1
       );
     }
   }
-  parseWorkflowRun(row) {
-    let parsedSnapshot = row.snapshot;
-    if (typeof parsedSnapshot === "string") {
-      try {
-        parsedSnapshot = JSON.parse(row.snapshot);
-      } catch (e) {
-        console.warn(`Failed to parse snapshot for workflow ${row.workflow_name}: ${e}`);
-      }
-    }
-    return {
-      workflowName: row.workflow_name,
-      runId: row.run_id,
-      snapshot: parsedSnapshot,
-      createdAt: this.ensureDate(row.createdAt),
-      updatedAt: this.ensureDate(row.updatedAt),
-      resourceId: row.resourceId
-    };
-  }
-  async getWorkflowRuns(args) {
-    try {
-      const table = await this.lanceClient.openTable(storage.TABLE_WORKFLOW_SNAPSHOT);
-      const query = table.query();
-      if (args?.workflowName) {
-        query.where(`workflow_name = '${args.workflowName}'`);
-      }
-      if (args?.fromDate) {
-        query.where(`\`createdAt\` >= ${args.fromDate.getTime()}`);
-      }
-      if (args?.toDate) {
-        query.where(`\`createdAt\` <= ${args.toDate.getTime()}`);
-      }
-      if (args?.limit) {
-        query.limit(args.limit);
-      }
-      if (args?.offset) {
-        query.offset(args.offset);
-      }
+  async loadWorkflowSnapshot({
+    workflowName,
+    runId
+  }) {
+    try {
+      const table = await this.client.openTable(storage.TABLE_WORKFLOW_SNAPSHOT);
+      const query = table.query().where(`workflow_name = '${workflowName}' AND run_id = '${runId}'`);
       const records = await query.toArray();
-      return {
-        runs: records.map((record) => this.parseWorkflowRun(record)),
-        total: records.length
-      };
+      return records.length > 0 ? JSON.parse(records[0].snapshot) : null;
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "LANCE_STORE_GET_WORKFLOW_RUNS_FAILED",
+          id: "LANCE_STORE_LOAD_WORKFLOW_SNAPSHOT_FAILED",
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
-          details: { namespace: args?.namespace ?? "", workflowName: args?.workflowName ?? "" }
+          details: { workflowName, runId }
         },
         error$1
       );
     }
   }
-  /**
-   * Retrieve a single workflow run by its runId.
-   * @param args The ID of the workflow run to retrieve
-   * @returns The workflow run object or null if not found
-   */
   async getWorkflowRunById(args) {
     try {
-      const table = await this.lanceClient.openTable(storage.TABLE_WORKFLOW_SNAPSHOT);
+      const table = await this.client.openTable(storage.TABLE_WORKFLOW_SNAPSHOT);
       let whereClause = `run_id = '${args.runId}'`;
       if (args.workflowName) {
         whereClause += ` AND workflow_name = '${args.workflowName}'`;
@@ -1055,7 +1809,7 @@ var LanceStorage = class _LanceStorage extends storage.MastraStorage {
       const records = await query.toArray();
       if (records.length === 0) return null;
       const record = records[0];
-      return this.parseWorkflowRun(record);
+      return parseWorkflowRun(record);
     } catch (error$1) {
       throw new error.MastraError(
         {
@@ -1068,96 +1822,324 @@ var LanceStorage = class _LanceStorage extends storage.MastraStorage {
       );
     }
   }
-  async persistWorkflowSnapshot({
-    workflowName,
-    runId,
-    snapshot
-  }) {
+  async getWorkflowRuns(args) {
     try {
-      const table = await this.lanceClient.openTable(storage.TABLE_WORKFLOW_SNAPSHOT);
-      const query = table.query().where(`workflow_name = '${workflowName}' AND run_id = '${runId}'`);
-      const records = await query.toArray();
-      let createdAt;
-      const now = Date.now();
-      if (records.length > 0) {
-        createdAt = records[0].createdAt ?? now;
+      const table = await this.client.openTable(storage.TABLE_WORKFLOW_SNAPSHOT);
+      let query = table.query();
+      const conditions = [];
+      if (args?.workflowName) {
+        conditions.push(`workflow_name = '${args.workflowName.replace(/'/g, "''")}'`);
+      }
+      if (args?.resourceId) {
+        conditions.push(`\`resourceId\` = '${args.resourceId}'`);
+      }
+      if (args?.fromDate instanceof Date) {
+        conditions.push(`\`createdAt\` >= ${args.fromDate.getTime()}`);
+      }
+      if (args?.toDate instanceof Date) {
+        conditions.push(`\`createdAt\` <= ${args.toDate.getTime()}`);
+      }
+      let total = 0;
+      if (conditions.length > 0) {
+        query = query.where(conditions.join(" AND "));
+        total = await table.countRows(conditions.join(" AND "));
       } else {
-        createdAt = now;
+        total = await table.countRows();
       }
-      const record = {
-        workflow_name: workflowName,
-        run_id: runId,
-        snapshot: JSON.stringify(snapshot),
-        createdAt,
-        updatedAt: now
+      if (args?.limit) {
+        query.limit(args.limit);
+      }
+      if (args?.offset) {
+        query.offset(args.offset);
+      }
+      const records = await query.toArray();
+      return {
+        runs: records.map((record) => parseWorkflowRun(record)),
+        total: total || records.length
       };
-      await table.mergeInsert(["workflow_name", "run_id"]).whenMatchedUpdateAll().whenNotMatchedInsertAll().execute([record]);
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "LANCE_STORE_PERSIST_WORKFLOW_SNAPSHOT_FAILED",
+          id: "LANCE_STORE_GET_WORKFLOW_RUNS_FAILED",
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
-          details: { workflowName, runId }
+          details: { namespace: args?.namespace ?? "", workflowName: args?.workflowName ?? "" }
         },
         error$1
       );
     }
   }
-  async loadWorkflowSnapshot({
-    workflowName,
-    runId
-  }) {
+};
+
+// src/storage/index.ts
+var LanceStorage = class _LanceStorage extends storage.MastraStorage {
+  stores;
+  lanceClient;
+  /**
+   * Creates a new instance of LanceStorage
+   * @param uri The URI to connect to LanceDB
+   * @param options connection options
+   *
+   * Usage:
+   *
+   * Connect to a local database
+   * ```ts
+   * const store = await LanceStorage.create('/path/to/db');
+   * ```
+   *
+   * Connect to a LanceDB cloud database
+   * ```ts
+   * const store = await LanceStorage.create('db://host:port');
+   * ```
+   *
+   * Connect to a cloud database
+   * ```ts
+   * const store = await LanceStorage.create('s3://bucket/db', { storageOptions: { timeout: '60s' } });
+   * ```
+   */
+  static async create(name, uri, options) {
+    const instance = new _LanceStorage(name);
     try {
-      const table = await this.lanceClient.openTable(storage.TABLE_WORKFLOW_SNAPSHOT);
-      const query = table.query().where(`workflow_name = '${workflowName}' AND run_id = '${runId}'`);
-      const records = await query.toArray();
-      return records.length > 0 ? JSON.parse(records[0].snapshot) : null;
-    } catch (error$1) {
+      instance.lanceClient = await lancedb.connect(uri, options);
+      const operations = new StoreOperationsLance({ client: instance.lanceClient });
+      instance.stores = {
+        operations: new StoreOperationsLance({ client: instance.lanceClient }),
+        workflows: new StoreWorkflowsLance({ client: instance.lanceClient }),
+        traces: new StoreTracesLance({ client: instance.lanceClient, operations }),
+        scores: new StoreScoresLance({ client: instance.lanceClient }),
+        memory: new StoreMemoryLance({ client: instance.lanceClient, operations }),
+        legacyEvals: new StoreLegacyEvalsLance({ client: instance.lanceClient })
+      };
+      return instance;
+    } catch (e) {
       throw new error.MastraError(
         {
-          id: "LANCE_STORE_LOAD_WORKFLOW_SNAPSHOT_FAILED",
+          id: "STORAGE_LANCE_STORAGE_CONNECT_FAILED",
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
-          details: { workflowName, runId }
+          text: `Failed to connect to LanceDB: ${e.message || e}`,
+          details: { uri, optionsProvided: !!options }
         },
-        error$1
+        e
       );
     }
   }
-  async getTracesPaginated(_args) {
-    throw new error.MastraError(
-      {
-        id: "LANCE_STORE_GET_TRACES_PAGINATED_FAILED",
-        domain: error.ErrorDomain.STORAGE,
-        category: error.ErrorCategory.THIRD_PARTY
-      },
-      "Method not implemented."
-    );
+  /**
+   * @internal
+   * Private constructor to enforce using the create factory method
+   */
+  constructor(name) {
+    super({ name });
+    const operations = new StoreOperationsLance({ client: this.lanceClient });
+    this.stores = {
+      operations: new StoreOperationsLance({ client: this.lanceClient }),
+      workflows: new StoreWorkflowsLance({ client: this.lanceClient }),
+      traces: new StoreTracesLance({ client: this.lanceClient, operations }),
+      scores: new StoreScoresLance({ client: this.lanceClient }),
+      legacyEvals: new StoreLegacyEvalsLance({ client: this.lanceClient }),
+      memory: new StoreMemoryLance({ client: this.lanceClient, operations })
+    };
   }
-  async getThreadsByResourceIdPaginated(_args) {
-    throw new error.MastraError(
-      {
-        id: "LANCE_STORE_GET_THREADS_BY_RESOURCE_ID_PAGINATED_FAILED",
-        domain: error.ErrorDomain.STORAGE,
-        category: error.ErrorCategory.THIRD_PARTY
-      },
-      "Method not implemented."
-    );
+  async createTable({
+    tableName,
+    schema
+  }) {
+    return this.stores.operations.createTable({ tableName, schema });
   }
-  async getMessagesPaginated(_args) {
-    throw new error.MastraError(
-      {
-        id: "LANCE_STORE_GET_MESSAGES_PAGINATED_FAILED",
-        domain: error.ErrorDomain.STORAGE,
-        category: error.ErrorCategory.THIRD_PARTY
-      },
-      "Method not implemented."
-    );
+  async dropTable({ tableName }) {
+    return this.stores.operations.dropTable({ tableName });
+  }
+  async alterTable({
+    tableName,
+    schema,
+    ifNotExists
+  }) {
+    return this.stores.operations.alterTable({ tableName, schema, ifNotExists });
+  }
+  async clearTable({ tableName }) {
+    return this.stores.operations.clearTable({ tableName });
+  }
+  async insert({ tableName, record }) {
+    return this.stores.operations.insert({ tableName, record });
+  }
+  async batchInsert({ tableName, records }) {
+    return this.stores.operations.batchInsert({ tableName, records });
+  }
+  async load({ tableName, keys }) {
+    return this.stores.operations.load({ tableName, keys });
+  }
+  async getThreadById({ threadId }) {
+    return this.stores.memory.getThreadById({ threadId });
+  }
+  async getThreadsByResourceId({ resourceId }) {
+    return this.stores.memory.getThreadsByResourceId({ resourceId });
+  }
+  /**
+   * Saves a thread to the database. This function doesn't overwrite existing threads.
+   * @param thread - The thread to save
+   * @returns The saved thread
+   */
+  async saveThread({ thread }) {
+    return this.stores.memory.saveThread({ thread });
+  }
+  async updateThread({
+    id,
+    title,
+    metadata
+  }) {
+    return this.stores.memory.updateThread({ id, title, metadata });
+  }
+  async deleteThread({ threadId }) {
+    return this.stores.memory.deleteThread({ threadId });
+  }
+  get supports() {
+    return {
+      selectByIncludeResourceScope: true,
+      resourceWorkingMemory: true,
+      hasColumn: true,
+      createTable: true
+    };
+  }
+  async getResourceById({ resourceId }) {
+    return this.stores.memory.getResourceById({ resourceId });
+  }
+  async saveResource({ resource }) {
+    return this.stores.memory.saveResource({ resource });
+  }
+  async updateResource({
+    resourceId,
+    workingMemory,
+    metadata
+  }) {
+    return this.stores.memory.updateResource({ resourceId, workingMemory, metadata });
+  }
+  /**
+   * Processes messages to include context messages based on withPreviousMessages and withNextMessages
+   * @param records - The sorted array of records to process
+   * @param include - The array of include specifications with context parameters
+   * @returns The processed array with context messages included
+   */
+  processMessagesWithContext(records, include) {
+    const messagesWithContext = include.filter((item) => item.withPreviousMessages || item.withNextMessages);
+    if (messagesWithContext.length === 0) {
+      return records;
+    }
+    const messageIndexMap = /* @__PURE__ */ new Map();
+    records.forEach((message, index) => {
+      messageIndexMap.set(message.id, index);
+    });
+    const additionalIndices = /* @__PURE__ */ new Set();
+    for (const item of messagesWithContext) {
+      const messageIndex = messageIndexMap.get(item.id);
+      if (messageIndex !== void 0) {
+        if (item.withPreviousMessages) {
+          const startIdx = Math.max(0, messageIndex - item.withPreviousMessages);
+          for (let i = startIdx; i < messageIndex; i++) {
+            additionalIndices.add(i);
+          }
+        }
+        if (item.withNextMessages) {
+          const endIdx = Math.min(records.length - 1, messageIndex + item.withNextMessages);
+          for (let i = messageIndex + 1; i <= endIdx; i++) {
+            additionalIndices.add(i);
+          }
+        }
+      }
+    }
+    if (additionalIndices.size === 0) {
+      return records;
+    }
+    const originalMatchIds = new Set(include.map((item) => item.id));
+    const allIndices = /* @__PURE__ */ new Set();
+    records.forEach((record, index) => {
+      if (originalMatchIds.has(record.id)) {
+        allIndices.add(index);
+      }
+    });
+    additionalIndices.forEach((index) => {
+      allIndices.add(index);
+    });
+    return Array.from(allIndices).sort((a, b) => a - b).map((index) => records[index]);
+  }
+  async getMessages({
+    threadId,
+    resourceId,
+    selectBy,
+    format,
+    threadConfig
+  }) {
+    return this.stores.memory.getMessages({ threadId, resourceId, selectBy, format, threadConfig });
+  }
+  async saveMessages(args) {
+    return this.stores.memory.saveMessages(args);
+  }
+  async getThreadsByResourceIdPaginated(args) {
+    return this.stores.memory.getThreadsByResourceIdPaginated(args);
+  }
+  async getMessagesPaginated(args) {
+    return this.stores.memory.getMessagesPaginated(args);
   }
   async updateMessages(_args) {
-    this.logger.error("updateMessages is not yet implemented in LanceStore");
-    throw new Error("Method not implemented");
+    return this.stores.memory.updateMessages(_args);
+  }
+  async getTraceById(args) {
+    return this.stores.traces.getTraceById(args);
+  }
+  async getTraces(args) {
+    return this.stores.traces.getTraces(args);
+  }
+  async getTracesPaginated(args) {
+    return this.stores.traces.getTracesPaginated(args);
+  }
+  async getEvalsByAgentName(agentName, type) {
+    return this.stores.legacyEvals.getEvalsByAgentName(agentName, type);
+  }
+  async getEvals(options) {
+    return this.stores.legacyEvals.getEvals(options);
+  }
+  async getWorkflowRuns(args) {
+    return this.stores.workflows.getWorkflowRuns(args);
+  }
+  async getWorkflowRunById(args) {
+    return this.stores.workflows.getWorkflowRunById(args);
+  }
+  async persistWorkflowSnapshot({
+    workflowName,
+    runId,
+    snapshot
+  }) {
+    return this.stores.workflows.persistWorkflowSnapshot({ workflowName, runId, snapshot });
+  }
+  async loadWorkflowSnapshot({
+    workflowName,
+    runId
+  }) {
+    return this.stores.workflows.loadWorkflowSnapshot({ workflowName, runId });
+  }
+  async getScoreById({ id: _id }) {
+    return this.stores.scores.getScoreById({ id: _id });
+  }
+  async getScoresByScorerId({
+    scorerId,
+    pagination
+  }) {
+    return this.stores.scores.getScoresByScorerId({ scorerId, pagination });
+  }
+  async saveScore(_score) {
+    return this.stores.scores.saveScore(_score);
+  }
+  async getScoresByRunId({
+    runId,
+    pagination
+  }) {
+    return this.stores.scores.getScoresByRunId({ runId, pagination });
+  }
+  async getScoresByEntityId({
+    entityId,
+    entityType,
+    pagination
+  }) {
+    return this.stores.scores.getScoresByEntityId({ entityId, entityType, pagination });
   }
 };
 var LanceFilterTranslator = class extends filter.BaseFilterTranslator {