@mastra/libsql 1.6.4 → 1.7.0-alpha.0

package/dist/index.cjs

@@ -1320,6 +1320,8 @@ var LibSQLDB = class extends base.MastraBase {
   maxRetries;
   initialBackoffMs;
   executeWriteOperationWithRetry;
+  /** Cache of actual table columns: tableName -> Promise<Set<columnName>> (stores in-flight promise to coalesce concurrent calls) */
+  tableColumnsCache = /* @__PURE__ */ new Map();
   constructor({
     client,
     maxRetries,
@@ -1338,6 +1340,47 @@ var LibSQLDB = class extends base.MastraBase {
       initialBackoffMs: this.initialBackoffMs
     });
   }
+  /**
+   * Gets the set of column names that actually exist in the database table.
+   * Results are cached; the cache is invalidated when alterTable() adds new columns.
+   */
+  async getTableColumns(tableName) {
+    const cached = this.tableColumnsCache.get(tableName);
+    if (cached) return cached;
+    const promise = (async () => {
+      try {
+        const sanitizedTable = utils.parseSqlIdentifier(tableName, "table name");
+        const result = await this.client.execute({
+          sql: `PRAGMA table_info("${sanitizedTable}")`
+        });
+        const columns = new Set((result.rows || []).map((row) => row.name));
+        if (columns.size === 0) {
+          this.tableColumnsCache.delete(tableName);
+        }
+        return columns;
+      } catch (error) {
+        this.tableColumnsCache.delete(tableName);
+        throw error;
+      }
+    })();
+    this.tableColumnsCache.set(tableName, promise);
+    return promise;
+  }
+  /**
+   * Filters a record to only include columns that exist in the actual database table.
+   * Unknown columns are silently dropped to ensure forward compatibility.
+   */
+  async filterRecordToKnownColumns(tableName, record) {
+    const knownColumns = await this.getTableColumns(tableName);
+    if (knownColumns.size === 0) return record;
+    const filtered = {};
+    for (const [key, value] of Object.entries(record)) {
+      if (knownColumns.has(key)) {
+        filtered[key] = value;
+      }
+    }
+    return filtered;
+  }
   /**
    * Checks if a column exists in the specified table.
    *
@@ -1359,10 +1402,12 @@ var LibSQLDB = class extends base.MastraBase {
     tableName,
     record
   }) {
+    const filteredRecord = await this.filterRecordToKnownColumns(tableName, record);
+    if (Object.keys(filteredRecord).length === 0) return;
     await this.client.execute(
       prepareStatement({
         tableName,
-        record
+        record: filteredRecord
       })
     );
   }
@@ -1384,7 +1429,9 @@ var LibSQLDB = class extends base.MastraBase {
     keys,
     data
   }) {
-    await this.client.execute(prepareUpdateStatement({ tableName, updates: data, keys }));
+    const filteredData = await this.filterRecordToKnownColumns(tableName, data);
+    if (Object.keys(filteredData).length === 0) return;
+    await this.client.execute(prepareUpdateStatement({ tableName, updates: filteredData, keys }));
   }
   /**
    * Updates a record in the specified table with automatic retry on lock errors.
@@ -1405,7 +1452,10 @@ var LibSQLDB = class extends base.MastraBase {
     records
   }) {
     if (records.length === 0) return;
-    const batchStatements = records.map((r) => prepareStatement({ tableName, record: r }));
+    const filteredRecords = await Promise.all(records.map((r) => this.filterRecordToKnownColumns(tableName, r)));
+    const nonEmptyRecords = filteredRecords.filter((r) => Object.keys(r).length > 0);
+    if (nonEmptyRecords.length === 0) return;
+    const batchStatements = nonEmptyRecords.map((r) => prepareStatement({ tableName, record: r }));
     await this.client.batch(batchStatements, "write");
   }
   /**
@@ -1443,7 +1493,15 @@ var LibSQLDB = class extends base.MastraBase {
     updates
   }) {
     if (updates.length === 0) return;
-    const batchStatements = updates.map(
+    const filteredUpdates = [];
+    for (const { keys, data } of updates) {
+      const filteredData = await this.filterRecordToKnownColumns(tableName, data);
+      if (Object.keys(filteredData).length > 0) {
+        filteredUpdates.push({ keys, data: filteredData });
+      }
+    }
+    if (filteredUpdates.length === 0) return;
+    const batchStatements = filteredUpdates.map(
       ({ keys, data }) => prepareUpdateStatement({
         tableName,
         updates: data,
@@ -1753,6 +1811,8 @@ var LibSQLDB = class extends base.MastraBase {
         },
         error$1
       );
+    } finally {
+      this.tableColumnsCache.delete(tableName);
     }
   }
   /**
@@ -2032,6 +2092,8 @@ Note: This migration may take some time for large tables.
         },
         error$1
       );
+    } finally {
+      this.tableColumnsCache.delete(tableName);
     }
   }
   /**
@@ -2924,6 +2986,8 @@ var DatasetsLibSQL = class extends storage.DatasetsStorage {
     await this.#db.createTable({ tableName: storage.TABLE_DATASETS, schema: storage.DATASETS_SCHEMA });
     await this.#db.createTable({ tableName: storage.TABLE_DATASET_ITEMS, schema: storage.DATASET_ITEMS_SCHEMA });
     await this.#db.createTable({ tableName: storage.TABLE_DATASET_VERSIONS, schema: storage.DATASET_VERSIONS_SCHEMA });
+    await this.#addColumnIfNotExists(storage.TABLE_DATASETS, "requestContextSchema", "TEXT");
+    await this.#addColumnIfNotExists(storage.TABLE_DATASET_ITEMS, "requestContext", "TEXT");
     await this.#client.execute({
       sql: `CREATE INDEX IF NOT EXISTS idx_dataset_items_dataset_validto ON "${storage.TABLE_DATASET_ITEMS}" ("datasetId", "validTo")`,
       args: []
@@ -2945,6 +3009,12 @@ var DatasetsLibSQL = class extends storage.DatasetsStorage {
       args: []
     });
   }
+  async #addColumnIfNotExists(table, column, sqlType) {
+    const exists = await this.#db.hasColumn(table, column);
+    if (!exists) {
+      await this.#client.execute({ sql: `ALTER TABLE "${table}" ADD COLUMN "${column}" ${sqlType}`, args: [] });
+    }
+  }
   async dangerouslyClearAll() {
     await this.#db.deleteData({ tableName: storage.TABLE_DATASET_VERSIONS });
     await this.#db.deleteData({ tableName: storage.TABLE_DATASET_ITEMS });
@@ -2959,6 +3029,7 @@ var DatasetsLibSQL = class extends storage.DatasetsStorage {
       metadata: row.metadata ? storage.safelyParseJSON(row.metadata) : void 0,
       inputSchema: row.inputSchema ? storage.safelyParseJSON(row.inputSchema) : void 0,
       groundTruthSchema: row.groundTruthSchema ? storage.safelyParseJSON(row.groundTruthSchema) : void 0,
+      requestContextSchema: row.requestContextSchema ? storage.safelyParseJSON(row.requestContextSchema) : void 0,
       version: row.version,
       createdAt: storage.ensureDate(row.createdAt),
       updatedAt: storage.ensureDate(row.updatedAt)
@@ -2971,6 +3042,7 @@ var DatasetsLibSQL = class extends storage.DatasetsStorage {
       datasetVersion: row.datasetVersion,
       input: storage.safelyParseJSON(row.input),
       groundTruth: row.groundTruth ? storage.safelyParseJSON(row.groundTruth) : void 0,
+      requestContext: row.requestContext ? storage.safelyParseJSON(row.requestContext) : void 0,
       metadata: row.metadata ? storage.safelyParseJSON(row.metadata) : void 0,
       createdAt: storage.ensureDate(row.createdAt),
       updatedAt: storage.ensureDate(row.updatedAt)
@@ -2985,6 +3057,7 @@ var DatasetsLibSQL = class extends storage.DatasetsStorage {
       isDeleted: Boolean(row.isDeleted),
       input: storage.safelyParseJSON(row.input),
       groundTruth: row.groundTruth ? storage.safelyParseJSON(row.groundTruth) : void 0,
+      requestContext: row.requestContext ? storage.safelyParseJSON(row.requestContext) : void 0,
       metadata: row.metadata ? storage.safelyParseJSON(row.metadata) : void 0,
       createdAt: storage.ensureDate(row.createdAt),
       updatedAt: storage.ensureDate(row.updatedAt)
@@ -3013,6 +3086,7 @@ var DatasetsLibSQL = class extends storage.DatasetsStorage {
           metadata: input.metadata,
           inputSchema: input.inputSchema ?? null,
           groundTruthSchema: input.groundTruthSchema ?? null,
+          requestContextSchema: input.requestContextSchema ?? null,
           version: 0,
           createdAt: nowIso,
           updatedAt: nowIso
@@ -3025,6 +3099,7 @@ var DatasetsLibSQL = class extends storage.DatasetsStorage {
         metadata: input.metadata,
         inputSchema: input.inputSchema ?? void 0,
         groundTruthSchema: input.groundTruthSchema ?? void 0,
+        requestContextSchema: input.requestContextSchema ?? void 0,
         version: 0,
         createdAt: now,
         updatedAt: now
@@ -3092,6 +3167,10 @@ var DatasetsLibSQL = class extends storage.DatasetsStorage {
         updates.push("groundTruthSchema = ?");
         values.push(args.groundTruthSchema === null ? null : JSON.stringify(args.groundTruthSchema));
       }
+      if (args.requestContextSchema !== void 0) {
+        updates.push("requestContextSchema = ?");
+        values.push(args.requestContextSchema === null ? null : JSON.stringify(args.requestContextSchema));
+      }
       values.push(args.id);
       await this.#client.execute({
         sql: `UPDATE ${storage.TABLE_DATASETS} SET ${updates.join(", ")} WHERE id = ?`,
@@ -3104,6 +3183,7 @@ var DatasetsLibSQL = class extends storage.DatasetsStorage {
         metadata: args.metadata ?? existing.metadata,
         inputSchema: (args.inputSchema !== void 0 ? args.inputSchema : existing.inputSchema) ?? void 0,
         groundTruthSchema: (args.groundTruthSchema !== void 0 ? args.groundTruthSchema : existing.groundTruthSchema) ?? void 0,
+        requestContextSchema: (args.requestContextSchema !== void 0 ? args.requestContextSchema : existing.requestContextSchema) ?? void 0,
         updatedAt: new Date(now)
       };
     } catch (error$1) {
@@ -3209,13 +3289,14 @@ var DatasetsLibSQL = class extends storage.DatasetsStorage {
             args: [args.datasetId]
           },
           {
-            sql: `INSERT INTO ${storage.TABLE_DATASET_ITEMS} (id, datasetId, datasetVersion, validTo, isDeleted, input, groundTruth, metadata, createdAt, updatedAt) VALUES (?, ?, (SELECT version FROM ${storage.TABLE_DATASETS} WHERE id = ?), NULL, 0, jsonb(?), jsonb(?), jsonb(?), ?, ?)`,
+            sql: `INSERT INTO ${storage.TABLE_DATASET_ITEMS} (id, datasetId, datasetVersion, validTo, isDeleted, input, groundTruth, requestContext, metadata, createdAt, updatedAt) VALUES (?, ?, (SELECT version FROM ${storage.TABLE_DATASETS} WHERE id = ?), NULL, 0, jsonb(?), jsonb(?), jsonb(?), jsonb(?), ?, ?)`,
             args: [
               id,
               args.datasetId,
               args.datasetId,
               jsonbArg(args.input),
               jsonbArg(args.groundTruth),
+              jsonbArg(args.requestContext),
               jsonbArg(args.metadata),
               nowIso,
               nowIso
@@ -3235,6 +3316,7 @@ var DatasetsLibSQL = class extends storage.DatasetsStorage {
         datasetVersion: newVersion,
         input: args.input,
         groundTruth: args.groundTruth,
+        requestContext: args.requestContext,
         metadata: args.metadata,
         createdAt: now,
         updatedAt: now
@@ -3275,6 +3357,7 @@ var DatasetsLibSQL = class extends storage.DatasetsStorage {
       const nowIso = now.toISOString();
       const mergedInput = args.input ?? existing.input;
       const mergedGroundTruth = args.groundTruth ?? existing.groundTruth;
+      const mergedRequestContext = args.requestContext ?? existing.requestContext;
       const mergedMetadata = args.metadata ?? existing.metadata;
       const results = await this.#client.batch(
         [
@@ -3287,13 +3370,14 @@ var DatasetsLibSQL = class extends storage.DatasetsStorage {
             args: [args.datasetId, args.id]
           },
           {
-            sql: `INSERT INTO ${storage.TABLE_DATASET_ITEMS} (id, datasetId, datasetVersion, validTo, isDeleted, input, groundTruth, metadata, createdAt, updatedAt) VALUES (?, ?, (SELECT version FROM ${storage.TABLE_DATASETS} WHERE id = ?), NULL, 0, jsonb(?), jsonb(?), jsonb(?), ?, ?)`,
+            sql: `INSERT INTO ${storage.TABLE_DATASET_ITEMS} (id, datasetId, datasetVersion, validTo, isDeleted, input, groundTruth, requestContext, metadata, createdAt, updatedAt) VALUES (?, ?, (SELECT version FROM ${storage.TABLE_DATASETS} WHERE id = ?), NULL, 0, jsonb(?), jsonb(?), jsonb(?), jsonb(?), ?, ?)`,
             args: [
               args.id,
               args.datasetId,
               args.datasetId,
               jsonbArg(mergedInput),
               jsonbArg(mergedGroundTruth),
+              jsonbArg(mergedRequestContext),
               jsonbArg(mergedMetadata),
               existing.createdAt.toISOString(),
               nowIso
@@ -3312,6 +3396,7 @@ var DatasetsLibSQL = class extends storage.DatasetsStorage {
         datasetVersion: newVersion,
         input: mergedInput,
         groundTruth: mergedGroundTruth,
+        requestContext: mergedRequestContext,
         metadata: mergedMetadata,
         updatedAt: now
       };
@@ -3352,13 +3437,14 @@ var DatasetsLibSQL = class extends storage.DatasetsStorage {
             args: [datasetId, id]
           },
           {
-            sql: `INSERT INTO ${storage.TABLE_DATASET_ITEMS} (id, datasetId, datasetVersion, validTo, isDeleted, input, groundTruth, metadata, createdAt, updatedAt) VALUES (?, ?, (SELECT version FROM ${storage.TABLE_DATASETS} WHERE id = ?), NULL, 1, jsonb(?), jsonb(?), jsonb(?), ?, ?)`,
+            sql: `INSERT INTO ${storage.TABLE_DATASET_ITEMS} (id, datasetId, datasetVersion, validTo, isDeleted, input, groundTruth, requestContext, metadata, createdAt, updatedAt) VALUES (?, ?, (SELECT version FROM ${storage.TABLE_DATASETS} WHERE id = ?), NULL, 1, jsonb(?), jsonb(?), jsonb(?), jsonb(?), ?, ?)`,
             args: [
               id,
               datasetId,
               datasetId,
               jsonbArg(existing.input),
               jsonbArg(existing.groundTruth),
+              jsonbArg(existing.requestContext),
               jsonbArg(existing.metadata),
               existing.createdAt.toISOString(),
               nowIso
@@ -3639,13 +3725,14 @@ var DatasetsLibSQL = class extends storage.DatasetsStorage {
         const id = crypto.randomUUID();
         items.push({ id, input: itemInput });
         statements.push({
-          sql: `INSERT INTO ${storage.TABLE_DATASET_ITEMS} (id, datasetId, datasetVersion, validTo, isDeleted, input, groundTruth, metadata, createdAt, updatedAt) VALUES (?, ?, (SELECT version FROM ${storage.TABLE_DATASETS} WHERE id = ?), NULL, 0, jsonb(?), jsonb(?), jsonb(?), ?, ?)`,
+          sql: `INSERT INTO ${storage.TABLE_DATASET_ITEMS} (id, datasetId, datasetVersion, validTo, isDeleted, input, groundTruth, requestContext, metadata, createdAt, updatedAt) VALUES (?, ?, (SELECT version FROM ${storage.TABLE_DATASETS} WHERE id = ?), NULL, 0, jsonb(?), jsonb(?), jsonb(?), jsonb(?), ?, ?)`,
           args: [
             id,
             input.datasetId,
             input.datasetId,
             jsonbArg(itemInput.input),
             jsonbArg(itemInput.groundTruth),
+            jsonbArg(itemInput.requestContext),
             jsonbArg(itemInput.metadata),
             nowIso,
             nowIso
@@ -3664,6 +3751,7 @@ var DatasetsLibSQL = class extends storage.DatasetsStorage {
         datasetVersion: newVersion,
         input: itemInput.input,
         groundTruth: itemInput.groundTruth,
+        requestContext: itemInput.requestContext,
         metadata: itemInput.metadata,
         createdAt: now,
         updatedAt: now
@@ -3713,13 +3801,14 @@ var DatasetsLibSQL = class extends storage.DatasetsStorage {
           args: [input.datasetId, item.id]
         });
         statements.push({
-          sql: `INSERT INTO ${storage.TABLE_DATASET_ITEMS} (id, datasetId, datasetVersion, validTo, isDeleted, input, groundTruth, metadata, createdAt, updatedAt) VALUES (?, ?, (SELECT version FROM ${storage.TABLE_DATASETS} WHERE id = ?), NULL, 1, jsonb(?), jsonb(?), jsonb(?), ?, ?)`,
+          sql: `INSERT INTO ${storage.TABLE_DATASET_ITEMS} (id, datasetId, datasetVersion, validTo, isDeleted, input, groundTruth, requestContext, metadata, createdAt, updatedAt) VALUES (?, ?, (SELECT version FROM ${storage.TABLE_DATASETS} WHERE id = ?), NULL, 1, jsonb(?), jsonb(?), jsonb(?), jsonb(?), ?, ?)`,
           args: [
             item.id,
             input.datasetId,
             input.datasetId,
             jsonbArg(item.input),
             jsonbArg(item.groundTruth),
+            jsonbArg(item.requestContext),
             jsonbArg(item.metadata),
             item.createdAt.toISOString(),
             nowIso
@@ -5214,41 +5303,64 @@ var MemoryLibSQL = class extends storage.MemoryStorage {
     if (row.type && row.type !== `v2`) result.type = row.type;
     return result;
   }
+  _sortMessages(messages, field, direction) {
+    return messages.sort((a, b) => {
+      const isDateField = field === "createdAt" || field === "updatedAt";
+      const aValue = isDateField ? new Date(a[field]).getTime() : a[field];
+      const bValue = isDateField ? new Date(b[field]).getTime() : b[field];
+      if (typeof aValue === "number" && typeof bValue === "number") {
+        return direction === "ASC" ? aValue - bValue : bValue - aValue;
+      }
+      return direction === "ASC" ? String(aValue).localeCompare(String(bValue)) : String(bValue).localeCompare(String(aValue));
+    });
+  }
   async _getIncludedMessages({ include }) {
     if (!include || include.length === 0) return null;
+    const targetIds = include.map((inc) => inc.id).filter(Boolean);
+    if (targetIds.length === 0) return null;
+    const idPlaceholders = targetIds.map(() => "?").join(", ");
+    const targetResult = await this.#client.execute({
+      sql: `SELECT id, thread_id, "createdAt" FROM "${storage.TABLE_MESSAGES}" WHERE id IN (${idPlaceholders})`,
+      args: targetIds
+    });
+    if (!targetResult.rows || targetResult.rows.length === 0) return null;
+    const targetMap = new Map(
+      targetResult.rows.map((r) => [r.id, { threadId: r.thread_id, createdAt: r.createdAt }])
+    );
     const unionQueries = [];
     const params = [];
     for (const inc of include) {
       const { id, withPreviousMessages = 0, withNextMessages = 0 } = inc;
-      unionQueries.push(
-        `
-                SELECT * FROM (
-                  WITH target_thread AS (
-                    SELECT thread_id FROM "${storage.TABLE_MESSAGES}" WHERE id = ?
-                  ),
-                  numbered_messages AS (
-                    SELECT
-                      id, content, role, type, "createdAt", thread_id, "resourceId",
-                      ROW_NUMBER() OVER (ORDER BY "createdAt" ASC) as row_num
-                    FROM "${storage.TABLE_MESSAGES}"
-                    WHERE thread_id = (SELECT thread_id FROM target_thread)
-                  ),
-                  target_positions AS (
-                    SELECT row_num as target_pos
-                    FROM numbered_messages
-                    WHERE id = ?
-                  )
-                  SELECT DISTINCT m.*
-                  FROM numbered_messages m
-                  CROSS JOIN target_positions t
-                  WHERE m.row_num BETWEEN (t.target_pos - ?) AND (t.target_pos + ?)
-                ) 
-                `
-        // Keep ASC for final sorting after fetching context
-      );
-      params.push(id, id, withPreviousMessages, withNextMessages);
-    }
-    const finalQuery = unionQueries.join(" UNION ALL ") + ' ORDER BY "createdAt" ASC';
+      const target = targetMap.get(id);
+      if (!target) continue;
+      unionQueries.push(`SELECT * FROM (
+        SELECT id, content, role, type, "createdAt", thread_id, "resourceId"
+        FROM "${storage.TABLE_MESSAGES}"
+        WHERE thread_id = ?
+          AND "createdAt" <= ?
+        ORDER BY "createdAt" DESC, id DESC
+        LIMIT ?
+      )`);
+      params.push(target.threadId, target.createdAt, withPreviousMessages + 1);
+      if (withNextMessages > 0) {
+        unionQueries.push(`SELECT * FROM (
+          SELECT id, content, role, type, "createdAt", thread_id, "resourceId"
+          FROM "${storage.TABLE_MESSAGES}"
+          WHERE thread_id = ?
+            AND "createdAt" > ?
+          ORDER BY "createdAt" ASC, id ASC
+          LIMIT ?
+        )`);
+        params.push(target.threadId, target.createdAt, withNextMessages);
+      }
+    }
+    if (unionQueries.length === 0) return null;
+    let finalQuery;
+    if (unionQueries.length === 1) {
+      finalQuery = unionQueries[0];
+    } else {
+      finalQuery = `${unionQueries.join(" UNION ALL ")} ORDER BY "createdAt" ASC, id ASC`;
+    }
     const includedResult = await this.#client.execute({ sql: finalQuery, args: params });
     const includedRows = includedResult.rows?.map((row) => this.parseRow(row));
     const seen = /* @__PURE__ */ new Set();
@@ -5343,6 +5455,23 @@ var MemoryLibSQL = class extends storage.MemoryStorage {
         );
       }
       const whereClause = conditions.length > 0 ? `WHERE ${conditions.join(" AND ")}` : "";
+      if (perPage === 0 && (!include || include.length === 0)) {
+        return { messages: [], total: 0, page, perPage: perPageForResponse, hasMore: false };
+      }
+      if (perPage === 0 && include && include.length > 0) {
+        const includeMessages = await this._getIncludedMessages({ include });
+        if (!includeMessages || includeMessages.length === 0) {
+          return { messages: [], total: 0, page, perPage: perPageForResponse, hasMore: false };
+        }
+        const list2 = new agent.MessageList().add(includeMessages, "memory");
+        return {
+          messages: this._sortMessages(list2.get.all.db(), field, direction),
+          total: 0,
+          page,
+          perPage: perPageForResponse,
+          hasMore: false
+        };
+      }
       const countResult = await this.#client.execute({
         sql: `SELECT COUNT(*) as count FROM ${storage.TABLE_MESSAGES} ${whereClause}`,
         args: queryParams
@@ -5376,16 +5505,7 @@ var MemoryLibSQL = class extends storage.MemoryStorage {
         }
       }
       const list = new agent.MessageList().add(messages, "memory");
-      let finalMessages = list.get.all.db();
-      finalMessages = finalMessages.sort((a, b) => {
-        const isDateField = field === "createdAt" || field === "updatedAt";
-        const aValue = isDateField ? new Date(a[field]).getTime() : a[field];
-        const bValue = isDateField ? new Date(b[field]).getTime() : b[field];
-        if (typeof aValue === "number" && typeof bValue === "number") {
-          return direction === "ASC" ? aValue - bValue : bValue - aValue;
-        }
-        return direction === "ASC" ? String(aValue).localeCompare(String(bValue)) : String(bValue).localeCompare(String(aValue));
-      });
+      const finalMessages = this._sortMessages(list.get.all.db(), field, direction);
       const threadIdSet = new Set(threadIds);
       const returnedThreadMessageIds = new Set(
         finalMessages.filter((m) => m.threadId && threadIdSet.has(m.threadId)).map((m) => m.id)
@@ -5471,6 +5591,23 @@ var MemoryLibSQL = class extends storage.MemoryStorage {
         );
       }
       const whereClause = `WHERE ${conditions.join(" AND ")}`;
+      if (perPage === 0 && (!include || include.length === 0)) {
+        return { messages: [], total: 0, page, perPage: perPageForResponse, hasMore: false };
+      }
+      if (perPage === 0 && include && include.length > 0) {
+        const includeMessages = await this._getIncludedMessages({ include });
+        if (!includeMessages || includeMessages.length === 0) {
+          return { messages: [], total: 0, page, perPage: perPageForResponse, hasMore: false };
+        }
+        const list2 = new agent.MessageList().add(includeMessages, "memory");
+        return {
+          messages: this._sortMessages(list2.get.all.db(), field, direction),
+          total: 0,
+          page,
+          perPage: perPageForResponse,
+          hasMore: false
+        };
+      }
       const countResult = await this.#client.execute({
         sql: `SELECT COUNT(*) as count FROM ${storage.TABLE_MESSAGES} ${whereClause}`,
         args: queryParams
@@ -5504,16 +5641,7 @@ var MemoryLibSQL = class extends storage.MemoryStorage {
         }
       }
       const list = new agent.MessageList().add(messages, "memory");
-      let finalMessages = list.get.all.db();
-      finalMessages = finalMessages.sort((a, b) => {
-        const isDateField = field === "createdAt" || field === "updatedAt";
-        const aValue = isDateField ? new Date(a[field]).getTime() : a[field];
-        const bValue = isDateField ? new Date(b[field]).getTime() : b[field];
-        if (typeof aValue === "number" && typeof bValue === "number") {
-          return direction === "ASC" ? aValue - bValue : bValue - aValue;
-        }
-        return direction === "ASC" ? String(aValue).localeCompare(String(bValue)) : String(bValue).localeCompare(String(aValue));
-      });
+      const finalMessages = this._sortMessages(list.get.all.db(), field, direction);
       const hasMore = perPageInput !== false && offset + perPage < total;
       return {
         messages: finalMessages,
@@ -7093,6 +7221,11 @@ var ObservabilityLibSQL = class extends storage.ObservabilityStorage {
   }
   async init() {
     await this.#db.createTable({ tableName: storage.TABLE_SPANS, schema: storage.SPAN_SCHEMA });
+    await this.#db.alterTable({
+      tableName: storage.TABLE_SPANS,
+      schema: storage.SPAN_SCHEMA,
+      ifNotExists: ["requestContext"]
+    });
   }
   async dangerouslyClearAll() {
     await this.#db.deleteData({ tableName: storage.TABLE_SPANS });
@@ -7264,7 +7397,8 @@ var ObservabilityLibSQL = class extends storage.ObservabilityStorage {
   }
   async listTraces(args) {
     const { filters, pagination, orderBy } = storage.listTracesArgsSchema.parse(args);
-    const { page, perPage } = pagination;
+    const page = pagination?.page ?? 0;
+    const perPage = pagination?.perPage ?? 10;
     const tableName = utils.parseSqlIdentifier(storage.TABLE_SPANS, "table name");
     try {
       const conditions = ["parentSpanId IS NULL"];
@@ -7404,8 +7538,8 @@ var ObservabilityLibSQL = class extends storage.ObservabilityStorage {
         }
       }
       const whereClause = conditions.length > 0 ? `WHERE ${conditions.join(" AND ")}` : "";
-      const sortField = orderBy.field;
-      const sortDirection = orderBy.direction;
+      const sortField = orderBy?.field ?? "startedAt";
+      const sortDirection = orderBy?.direction ?? "DESC";
       let orderByClause;
       if (sortField === "endedAt") {
         orderByClause = sortDirection === "DESC" ? `CASE WHEN ${sortField} IS NULL THEN 0 ELSE 1 END, ${sortField} DESC` : `CASE WHEN ${sortField} IS NULL THEN 1 ELSE 0 END, ${sortField} ASC`;