@mastra/cloudflare-d1 0.13.8 → 1.0.0-beta.1

package/dist/index.js

@@ -1,9 +1,9 @@
 import { MastraError, ErrorCategory, ErrorDomain } from '@mastra/core/error';
-import { MastraStorage, StoreOperations, ScoresStorage, TABLE_SCORERS, LegacyEvalsStorage, TABLE_EVALS, serializeDate, TracesStorage, TABLE_TRACES, WorkflowsStorage, TABLE_WORKFLOW_SNAPSHOT, ensureDate, MemoryStorage, TABLE_RESOURCES, TABLE_THREADS, TABLE_MESSAGES, resolveMessageLimit, safelyParseJSON } from '@mastra/core/storage';
+import { MastraStorage, StoreOperations, ScoresStorage, TABLE_SCORERS, normalizePerPage, calculatePagination, WorkflowsStorage, TABLE_WORKFLOW_SNAPSHOT, ensureDate, MemoryStorage, TABLE_RESOURCES, TABLE_THREADS, TABLE_MESSAGES, serializeDate, safelyParseJSON } from '@mastra/core/storage';
 import Cloudflare from 'cloudflare';
-import { parseSqlIdentifier } from '@mastra/core/utils';
 import { MessageList } from '@mastra/core/agent';
-import { saveScorePayloadSchema } from '@mastra/core/scores';
+import { parseSqlIdentifier } from '@mastra/core/utils';
+import { saveScorePayloadSchema } from '@mastra/core/evals';
 
 // src/storage/index.ts
 var SqlBuilder = class {
@@ -242,16 +242,6 @@ function isArrayOfRecords(value) {
 }
 function deserializeValue(value, type) {
   if (value === null || value === void 0) return null;
-  if (type === "date" && typeof value === "string") {
-    return new Date(value);
-  }
-  if (type === "jsonb" && typeof value === "string") {
-    try {
-      return JSON.parse(value);
-    } catch {
-      return value;
-    }
-  }
   if (typeof value === "string" && (value.startsWith("{") || value.startsWith("["))) {
     try {
       return JSON.parse(value);
@@ -262,155 +252,7 @@ function deserializeValue(value, type) {
   return value;
 }
 
-// src/storage/domains/legacy-evals/index.ts
-var LegacyEvalsStorageD1 = class extends LegacyEvalsStorage {
-  operations;
-  constructor({ operations }) {
-    super();
-    this.operations = operations;
-  }
-  async getEvals(options) {
-    const { agentName, type, page = 0, perPage = 40, dateRange } = options || {};
-    const fullTableName = this.operations.getTableName(TABLE_EVALS);
-    const conditions = [];
-    const queryParams = [];
-    if (agentName) {
-      conditions.push(`agent_name = ?`);
-      queryParams.push(agentName);
-    }
-    if (type === "test") {
-      conditions.push(`(test_info IS NOT NULL AND json_extract(test_info, '$.testPath') IS NOT NULL)`);
-    } else if (type === "live") {
-      conditions.push(`(test_info IS NULL OR json_extract(test_info, '$.testPath') IS NULL)`);
-    }
-    if (dateRange?.start) {
-      conditions.push(`created_at >= ?`);
-      queryParams.push(serializeDate(dateRange.start));
-    }
-    if (dateRange?.end) {
-      conditions.push(`created_at <= ?`);
-      queryParams.push(serializeDate(dateRange.end));
-    }
-    const countQueryBuilder = createSqlBuilder().count().from(fullTableName);
-    if (conditions.length > 0) {
-      countQueryBuilder.where(conditions.join(" AND "), ...queryParams);
-    }
-    const { sql: countSql, params: countParams } = countQueryBuilder.build();
-    try {
-      const countResult = await this.operations.executeQuery({
-        sql: countSql,
-        params: countParams,
-        first: true
-      });
-      const total = Number(countResult?.count || 0);
-      const currentOffset = page * perPage;
-      if (total === 0) {
-        return {
-          evals: [],
-          total: 0,
-          page,
-          perPage,
-          hasMore: false
-        };
-      }
-      const dataQueryBuilder = createSqlBuilder().select("*").from(fullTableName);
-      if (conditions.length > 0) {
-        dataQueryBuilder.where(conditions.join(" AND "), ...queryParams);
-      }
-      dataQueryBuilder.orderBy("created_at", "DESC").limit(perPage).offset(currentOffset);
-      const { sql: dataSql, params: dataParams } = dataQueryBuilder.build();
-      const rows = await this.operations.executeQuery({
-        sql: dataSql,
-        params: dataParams
-      });
-      const evals = (isArrayOfRecords(rows) ? rows : []).map((row) => {
-        const result = deserializeValue(row.result);
-        const testInfo = row.test_info ? deserializeValue(row.test_info) : void 0;
-        if (!result || typeof result !== "object" || !("score" in result)) {
-          throw new Error(`Invalid MetricResult format: ${JSON.stringify(result)}`);
-        }
-        return {
-          input: row.input,
-          output: row.output,
-          result,
-          agentName: row.agent_name,
-          metricName: row.metric_name,
-          instructions: row.instructions,
-          testInfo,
-          globalRunId: row.global_run_id,
-          runId: row.run_id,
-          createdAt: row.created_at
-        };
-      });
-      const hasMore = currentOffset + evals.length < total;
-      return {
-        evals,
-        total,
-        page,
-        perPage,
-        hasMore
-      };
-    } catch (error) {
-      throw new MastraError(
-        {
-          id: "CLOUDFLARE_D1_STORAGE_GET_EVALS_ERROR",
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          text: `Failed to retrieve evals for agent ${agentName}: ${error instanceof Error ? error.message : String(error)}`,
-          details: { agentName: agentName ?? "", type: type ?? "" }
-        },
-        error
-      );
-    }
-  }
-  /**
-   * @deprecated use getEvals instead
-   */
-  async getEvalsByAgentName(agentName, type) {
-    const fullTableName = this.operations.getTableName(TABLE_EVALS);
-    try {
-      let query = createSqlBuilder().select("*").from(fullTableName).where("agent_name = ?", agentName);
-      if (type === "test") {
-        query = query.andWhere("test_info IS NOT NULL AND json_extract(test_info, '$.testPath') IS NOT NULL");
-      } else if (type === "live") {
-        query = query.andWhere("(test_info IS NULL OR json_extract(test_info, '$.testPath') IS NULL)");
-      }
-      query.orderBy("created_at", "DESC");
-      const { sql, params } = query.build();
-      const results = await this.operations.executeQuery({ sql, params });
-      return isArrayOfRecords(results) ? results.map((row) => {
-        const result = deserializeValue(row.result);
-        const testInfo = row.test_info ? deserializeValue(row.test_info) : void 0;
-        return {
-          input: row.input || "",
-          output: row.output || "",
-          result,
-          agentName: row.agent_name || "",
-          metricName: row.metric_name || "",
-          instructions: row.instructions || "",
-          runId: row.run_id || "",
-          globalRunId: row.global_run_id || "",
-          createdAt: row.created_at || "",
-          testInfo
-        };
-      }) : [];
-    } catch (error) {
-      const mastraError = new MastraError(
-        {
-          id: "CLOUDFLARE_D1_STORAGE_GET_EVALS_ERROR",
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          text: `Failed to retrieve evals for agent ${agentName}: ${error instanceof Error ? error.message : String(error)}`,
-          details: { agentName }
-        },
-        error
-      );
-      this.logger?.error(mastraError.toString());
-      this.logger?.trackException(mastraError);
-      return [];
-    }
-  }
-};
+// src/storage/domains/memory/index.ts
 var MemoryStorageD1 = class extends MemoryStorage {
   operations;
   constructor({ operations }) {
@@ -558,39 +400,22 @@ var MemoryStorageD1 = class extends MemoryStorage {
       return null;
     }
   }
-  /**
-   * @deprecated use getThreadsByResourceIdPaginated instead
-   */
-  async getThreadsByResourceId({ resourceId }) {
-    const fullTableName = this.operations.getTableName(TABLE_THREADS);
-    try {
-      const query = createSqlBuilder().select("*").from(fullTableName).where("resourceId = ?", resourceId);
-      const { sql, params } = query.build();
-      const results = await this.operations.executeQuery({ sql, params });
-      return (isArrayOfRecords(results) ? results : []).map((thread) => ({
-        ...thread,
-        createdAt: ensureDate(thread.createdAt),
-        updatedAt: ensureDate(thread.updatedAt),
-        metadata: typeof thread.metadata === "string" ? JSON.parse(thread.metadata || "{}") : thread.metadata || {}
-      }));
-    } catch (error) {
-      const mastraError = new MastraError(
+  async listThreadsByResourceId(args) {
+    const { resourceId, page = 0, perPage: perPageInput, orderBy } = args;
+    const perPage = normalizePerPage(perPageInput, 100);
+    if (page < 0) {
+      throw new MastraError(
         {
-          id: "CLOUDFLARE_D1_STORAGE_GET_THREADS_BY_RESOURCE_ID_ERROR",
+          id: "STORAGE_CLOUDFLARE_D1_LIST_THREADS_BY_RESOURCE_ID_INVALID_PAGE",
           domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          text: `Error getting threads by resourceId ${resourceId}: ${error instanceof Error ? error.message : String(error)}`,
-          details: { resourceId }
+          category: ErrorCategory.USER,
+          details: { page }
         },
-        error
+        new Error("page must be >= 0")
       );
-      this.logger?.error(mastraError.toString());
-      this.logger?.trackException(mastraError);
-      return [];
     }
-  }
-  async getThreadsByResourceIdPaginated(args) {
-    const { resourceId, page, perPage } = args;
+    const { offset, perPage: perPageForResponse } = calculatePagination(page, perPageInput, perPage);
+    const { field, direction } = this.parseOrderBy(orderBy);
     const fullTableName = this.operations.getTableName(TABLE_THREADS);
     const mapRowToStorageThreadType = (row) => ({
       ...row,
@@ -602,20 +427,21 @@ var MemoryStorageD1 = class extends MemoryStorage {
       const countQuery = createSqlBuilder().count().from(fullTableName).where("resourceId = ?", resourceId);
       const countResult = await this.operations.executeQuery(countQuery.build());
       const total = Number(countResult?.[0]?.count ?? 0);
-      const selectQuery = createSqlBuilder().select("*").from(fullTableName).where("resourceId = ?", resourceId).orderBy("createdAt", "DESC").limit(perPage).offset(page * perPage);
+      const limitValue = perPageInput === false ? total : perPage;
+      const selectQuery = createSqlBuilder().select("*").from(fullTableName).where("resourceId = ?", resourceId).orderBy(field, direction).limit(limitValue).offset(offset);
       const results = await this.operations.executeQuery(selectQuery.build());
       const threads = results.map(mapRowToStorageThreadType);
       return {
         threads,
         total,
         page,
-        perPage,
-        hasMore: page * perPage + threads.length < total
+        perPage: perPageForResponse,
+        hasMore: perPageInput === false ? false : offset + perPage < total
       };
     } catch (error) {
       const mastraError = new MastraError(
         {
-          id: "CLOUDFLARE_D1_STORAGE_GET_THREADS_BY_RESOURCE_ID_PAGINATED_ERROR",
+          id: "CLOUDFLARE_D1_STORAGE_LIST_THREADS_BY_RESOURCE_ID_ERROR",
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           text: `Error getting threads by resourceId ${resourceId}: ${error instanceof Error ? error.message : String(error)}`,
@@ -629,7 +455,7 @@ var MemoryStorageD1 = class extends MemoryStorage {
         threads: [],
         total: 0,
         page,
-        perPage,
+        perPage: perPageForResponse,
         hasMore: false
       };
     }
@@ -739,8 +565,8 @@ var MemoryStorageD1 = class extends MemoryStorage {
     }
   }
   async saveMessages(args) {
-    const { messages, format = "v1" } = args;
-    if (messages.length === 0) return [];
+    const { messages } = args;
+    if (messages.length === 0) return { messages: [] };
     try {
       const now = /* @__PURE__ */ new Date();
       const threadId = messages[0]?.threadId;
@@ -788,8 +614,7 @@ var MemoryStorageD1 = class extends MemoryStorage {
       ]);
       this.logger.debug(`Saved ${messages.length} messages`);
       const list = new MessageList().add(messages, "memory");
-      if (format === `v2`) return list.get.all.v2();
-      return list.get.all.v1();
+      return { messages: list.get.all.db() };
     } catch (error) {
       throw new MastraError(
         {
@@ -802,9 +627,8 @@ var MemoryStorageD1 = class extends MemoryStorage {
       );
     }
   }
-  async _getIncludedMessages(threadId, selectBy) {
+  async _getIncludedMessages(threadId, include) {
     if (!threadId.trim()) throw new Error("threadId must be a non-empty string");
-    const include = selectBy?.include;
     if (!include) return null;
     const unionQueries = [];
     const params = [];
@@ -860,74 +684,8 @@ var MemoryStorageD1 = class extends MemoryStorage {
     });
     return processedMessages;
   }
-  async getMessages({
-    threadId,
-    resourceId,
-    selectBy,
-    format
-  }) {
-    try {
-      if (!threadId.trim()) throw new Error("threadId must be a non-empty string");
-      const fullTableName = this.operations.getTableName(TABLE_MESSAGES);
-      const limit = resolveMessageLimit({
-        last: selectBy?.last,
-        defaultLimit: 40
-      });
-      const include = selectBy?.include || [];
-      const messages = [];
-      if (include.length) {
-        const includeResult = await this._getIncludedMessages(threadId, selectBy);
-        if (Array.isArray(includeResult)) messages.push(...includeResult);
-      }
-      const excludeIds = messages.map((m) => m.id);
-      const query = createSqlBuilder().select(["id", "content", "role", "type", "createdAt", "thread_id AS threadId"]).from(fullTableName).where("thread_id = ?", threadId);
-      if (excludeIds.length > 0) {
-        query.andWhere(`id NOT IN (${excludeIds.map(() => "?").join(",")})`, ...excludeIds);
-      }
-      query.orderBy("createdAt", "DESC").limit(limit);
-      const { sql, params } = query.build();
-      const result = await this.operations.executeQuery({ sql, params });
-      if (Array.isArray(result)) messages.push(...result);
-      messages.sort((a, b) => {
-        const aRecord = a;
-        const bRecord = b;
-        const timeA = new Date(aRecord.createdAt).getTime();
-        const timeB = new Date(bRecord.createdAt).getTime();
-        return timeA - timeB;
-      });
-      const processedMessages = messages.map((message) => {
-        const processedMsg = {};
-        for (const [key, value] of Object.entries(message)) {
-          if (key === `type` && value === `v2`) continue;
-          processedMsg[key] = deserializeValue(value);
-        }
-        return processedMsg;
-      });
-      this.logger.debug(`Retrieved ${messages.length} messages for thread ${threadId}`);
-      const list = new MessageList().add(processedMessages, "memory");
-      if (format === `v2`) return list.get.all.v2();
-      return list.get.all.v1();
-    } catch (error) {
-      const mastraError = new MastraError(
-        {
-          id: "CLOUDFLARE_D1_STORAGE_GET_MESSAGES_ERROR",
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          text: `Failed to retrieve messages for thread ${threadId}: ${error instanceof Error ? error.message : String(error)}`,
-          details: { threadId, resourceId: resourceId ?? "" }
-        },
-        error
-      );
-      this.logger?.error(mastraError.toString());
-      this.logger?.trackException(mastraError);
-      throw mastraError;
-    }
-  }
-  async getMessagesById({
-    messageIds,
-    format
-  }) {
-    if (messageIds.length === 0) return [];
+  async listMessagesById({ messageIds }) {
+    if (messageIds.length === 0) return { messages: [] };
     const fullTableName = this.operations.getTableName(TABLE_MESSAGES);
     const messages = [];
     try {
@@ -946,12 +704,11 @@ var MemoryStorageD1 = class extends MemoryStorage {
       });
       this.logger.debug(`Retrieved ${messages.length} messages`);
       const list = new MessageList().add(processedMessages, "memory");
-      if (format === `v1`) return list.get.all.v1();
-      return list.get.all.v2();
+      return { messages: list.get.all.db() };
     } catch (error) {
       const mastraError = new MastraError(
         {
-          id: "CLOUDFLARE_D1_STORAGE_GET_MESSAGES_BY_ID_ERROR",
+          id: "CLOUDFLARE_D1_STORAGE_LIST_MESSAGES_BY_ID_ERROR",
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           text: `Failed to retrieve messages by ID: ${error instanceof Error ? error.message : String(error)}`,
@@ -964,118 +721,157 @@ var MemoryStorageD1 = class extends MemoryStorage {
       throw mastraError;
     }
   }
-  async getMessagesPaginated({
-    threadId,
-    resourceId,
-    selectBy,
-    format
-  }) {
-    const { dateRange, page = 0, perPage: perPageInput } = selectBy?.pagination || {};
-    const { start: fromDate, end: toDate } = dateRange || {};
-    const perPage = perPageInput !== void 0 ? perPageInput : resolveMessageLimit({ last: selectBy?.last, defaultLimit: 40 });
-    const fullTableName = this.operations.getTableName(TABLE_MESSAGES);
-    const messages = [];
+  async listMessages(args) {
+    const { threadId, resourceId, include, filter, perPage: perPageInput, page = 0, orderBy } = args;
+    if (!threadId.trim()) {
+      throw new MastraError(
+        {
+          id: "STORAGE_CLOUDFLARE_D1_LIST_MESSAGES_INVALID_THREAD_ID",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { threadId }
+        },
+        new Error("threadId must be a non-empty string")
+      );
+    }
+    if (page < 0) {
+      throw new MastraError(
+        {
+          id: "STORAGE_CLOUDFLARE_D1_LIST_MESSAGES_INVALID_PAGE",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.USER,
+          details: { page }
+        },
+        new Error("page must be >= 0")
+      );
+    }
+    const perPage = normalizePerPage(perPageInput, 40);
+    const { offset, perPage: perPageForResponse } = calculatePagination(page, perPageInput, perPage);
     try {
-      if (!threadId.trim()) throw new Error("threadId must be a non-empty string");
-      if (selectBy?.include?.length) {
-        const includeResult = await this._getIncludedMessages(threadId, selectBy);
-        if (Array.isArray(includeResult)) messages.push(...includeResult);
+      const fullTableName = this.operations.getTableName(TABLE_MESSAGES);
+      let query = `
+        SELECT id, content, role, type, createdAt, thread_id AS threadId, resourceId
+        FROM ${fullTableName}
+        WHERE thread_id = ?
+      `;
+      const queryParams = [threadId];
+      if (resourceId) {
+        query += ` AND resourceId = ?`;
+        queryParams.push(resourceId);
       }
-      const countQuery = createSqlBuilder().count().from(fullTableName).where("thread_id = ?", threadId);
-      if (fromDate) {
-        countQuery.andWhere("createdAt >= ?", serializeDate(fromDate));
+      const dateRange = filter?.dateRange;
+      if (dateRange?.start) {
+        const startDate = dateRange.start instanceof Date ? serializeDate(dateRange.start) : serializeDate(new Date(dateRange.start));
+        query += ` AND createdAt >= ?`;
+        queryParams.push(startDate);
       }
-      if (toDate) {
-        countQuery.andWhere("createdAt <= ?", serializeDate(toDate));
+      if (dateRange?.end) {
+        const endDate = dateRange.end instanceof Date ? serializeDate(dateRange.end) : serializeDate(new Date(dateRange.end));
+        query += ` AND createdAt <= ?`;
+        queryParams.push(endDate);
       }
-      const countResult = await this.operations.executeQuery(countQuery.build());
+      const { field, direction } = this.parseOrderBy(orderBy, "ASC");
+      query += ` ORDER BY "${field}" ${direction}`;
+      if (perPage !== Number.MAX_SAFE_INTEGER) {
+        query += ` LIMIT ? OFFSET ?`;
+        queryParams.push(perPage, offset);
+      }
+      const results = await this.operations.executeQuery({ sql: query, params: queryParams });
+      const paginatedMessages = (isArrayOfRecords(results) ? results : []).map((message) => {
+        const processedMsg = {};
+        for (const [key, value] of Object.entries(message)) {
+          if (key === `type` && value === `v2`) continue;
+          processedMsg[key] = deserializeValue(value);
+        }
+        return processedMsg;
+      });
+      const paginatedCount = paginatedMessages.length;
+      let countQuery = `SELECT count() as count FROM ${fullTableName} WHERE thread_id = ?`;
+      const countParams = [threadId];
+      if (resourceId) {
+        countQuery += ` AND resourceId = ?`;
+        countParams.push(resourceId);
+      }
+      if (dateRange?.start) {
+        const startDate = dateRange.start instanceof Date ? serializeDate(dateRange.start) : serializeDate(new Date(dateRange.start));
+        countQuery += ` AND createdAt >= ?`;
+        countParams.push(startDate);
+      }
+      if (dateRange?.end) {
+        const endDate = dateRange.end instanceof Date ? serializeDate(dateRange.end) : serializeDate(new Date(dateRange.end));
+        countQuery += ` AND createdAt <= ?`;
+        countParams.push(endDate);
+      }
+      const countResult = await this.operations.executeQuery({ sql: countQuery, params: countParams });
       const total = Number(countResult[0]?.count ?? 0);
-      if (total === 0 && messages.length === 0) {
+      if (total === 0 && paginatedCount === 0 && (!include || include.length === 0)) {
         return {
           messages: [],
           total: 0,
           page,
-          perPage,
+          perPage: perPageForResponse,
           hasMore: false
         };
       }
-      const excludeIds = messages.map((m) => m.id);
-      const excludeCondition = excludeIds.length > 0 ? `AND id NOT IN (${excludeIds.map(() => "?").join(",")})` : "";
-      let query;
-      let queryParams = [threadId];
-      if (fromDate) {
-        queryParams.push(serializeDate(fromDate));
-      }
-      if (toDate) {
-        queryParams.push(serializeDate(toDate));
-      }
-      if (excludeIds.length > 0) {
-        queryParams.push(...excludeIds);
-      }
-      if (selectBy?.last && selectBy.last > 0) {
-        query = `
-                    SELECT id, content, role, type, createdAt, thread_id AS threadId, resourceId
-                    FROM ${fullTableName}
-                    WHERE thread_id = ?
-                    ${fromDate ? "AND createdAt >= ?" : ""}
-                    ${toDate ? "AND createdAt <= ?" : ""}
-                    ${excludeCondition}
-                    ORDER BY createdAt DESC
-                    LIMIT ?
-                `;
-        queryParams.push(selectBy.last);
-      } else {
-        query = `
-                    SELECT id, content, role, type, createdAt, thread_id AS threadId, resourceId
-                    FROM ${fullTableName}
-                    WHERE thread_id = ?
-                    ${fromDate ? "AND createdAt >= ?" : ""}
-                    ${toDate ? "AND createdAt <= ?" : ""}
-                    ${excludeCondition}
-                    ORDER BY createdAt DESC
-                    LIMIT ? OFFSET ?
-                `;
-        queryParams.push(perPage, page * perPage);
+      const messageIds = new Set(paginatedMessages.map((m) => m.id));
+      let includeMessages = [];
+      if (include && include.length > 0) {
+        const includeResult = await this._getIncludedMessages(threadId, include);
+        if (Array.isArray(includeResult)) {
+          includeMessages = includeResult;
+          for (const includeMsg of includeMessages) {
+            if (!messageIds.has(includeMsg.id)) {
+              paginatedMessages.push(includeMsg);
+              messageIds.add(includeMsg.id);
+            }
+          }
+        }
       }
-      const results = await this.operations.executeQuery({ sql: query, params: queryParams });
-      const processedMessages = results.map((message) => {
-        const processedMsg = {};
-        for (const [key, value] of Object.entries(message)) {
-          if (key === `type` && value === `v2`) continue;
-          processedMsg[key] = deserializeValue(value);
+      const list = new MessageList().add(paginatedMessages, "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 (aValue === bValue) {
+          return a.id.localeCompare(b.id);
         }
-        return processedMsg;
+        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));
       });
-      if (selectBy?.last) {
-        processedMessages.sort((a, b) => new Date(a.createdAt).getTime() - new Date(b.createdAt).getTime());
-      }
-      const list = new MessageList().add(processedMessages, "memory");
-      messages.push(...format === `v2` ? list.get.all.v2() : list.get.all.v1());
+      const returnedThreadMessageIds = new Set(finalMessages.filter((m) => m.threadId === threadId).map((m) => m.id));
+      const allThreadMessagesReturned = returnedThreadMessageIds.size >= total;
+      const hasMore = perPageInput === false ? false : allThreadMessagesReturned ? false : offset + paginatedCount < total;
       return {
-        messages,
+        messages: finalMessages,
         total,
         page,
-        perPage,
-        hasMore: selectBy?.last ? false : page * perPage + messages.length < total
+        perPage: perPageForResponse,
+        hasMore
       };
     } catch (error) {
       const mastraError = new MastraError(
         {
-          id: "CLOUDFLARE_D1_STORAGE_GET_MESSAGES_PAGINATED_ERROR",
+          id: "CLOUDFLARE_D1_STORAGE_LIST_MESSAGES_ERROR",
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
-          text: `Failed to retrieve messages for thread ${threadId}: ${error instanceof Error ? error.message : String(error)}`,
-          details: { threadId, resourceId: resourceId ?? "" }
+          text: `Failed to list messages for thread ${threadId}: ${error instanceof Error ? error.message : String(error)}`,
+          details: {
+            threadId,
+            resourceId: resourceId ?? ""
+          }
         },
         error
       );
-      this.logger?.error(mastraError.toString());
-      this.logger?.trackException(mastraError);
+      this.logger?.error?.(mastraError.toString());
+      this.logger?.trackException?.(mastraError);
       return {
         messages: [],
         total: 0,
         page,
-        perPage,
+        perPage: perPageForResponse,
         hasMore: false
       };
     }
@@ -1576,7 +1372,7 @@ function transformScoreRow(row) {
   deserialized.analyzeStepResult = safelyParseJSON(row.analyzeStepResult);
   deserialized.metadata = safelyParseJSON(row.metadata);
   deserialized.additionalContext = safelyParseJSON(row.additionalContext);
-  deserialized.runtimeContext = safelyParseJSON(row.runtimeContext);
+  deserialized.requestContext = safelyParseJSON(row.requestContext);
   deserialized.entity = safelyParseJSON(row.entity);
   deserialized.createdAt = row.createdAtZ || row.createdAt;
   deserialized.updatedAt = row.updatedAtZ || row.updatedAt;
@@ -1660,7 +1456,7 @@ var ScoresStorageD1 = class extends ScoresStorage {
       );
     }
   }
-  async getScoresByScorerId({
+  async listScoresByScorerId({
     scorerId,
     entityId,
     entityType,
@@ -1668,6 +1464,9 @@ var ScoresStorageD1 = class extends ScoresStorage {
     pagination
   }) {
     try {
+      const { page, perPage: perPageInput } = pagination;
+      const perPage = normalizePerPage(perPageInput, 100);
+      const { offset: start, perPage: perPageForResponse } = calculatePagination(page, perPageInput, perPage);
       const fullTableName = this.operations.getTableName(TABLE_SCORERS);
       const countQuery = createSqlBuilder().count().from(fullTableName).where("scorerId = ?", scorerId);
       if (entityId) {
@@ -1685,13 +1484,15 @@ var ScoresStorageD1 = class extends ScoresStorage {
         return {
           pagination: {
             total: 0,
-            page: pagination.page,
-            perPage: pagination.perPage,
+            page,
+            perPage: perPageForResponse,
             hasMore: false
           },
           scores: []
         };
       }
+      const end = perPageInput === false ? total : start + perPage;
+      const limitValue = perPageInput === false ? total : perPage;
       const selectQuery = createSqlBuilder().select("*").from(fullTableName).where("scorerId = ?", scorerId);
       if (entityId) {
         selectQuery.andWhere("entityId = ?", entityId);
@@ -1702,16 +1503,16 @@ var ScoresStorageD1 = class extends ScoresStorage {
       if (source) {
         selectQuery.andWhere("source = ?", source);
       }
-      selectQuery.limit(pagination.perPage).offset(pagination.page * pagination.perPage);
+      selectQuery.limit(limitValue).offset(start);
       const { sql, params } = selectQuery.build();
       const results = await this.operations.executeQuery({ sql, params });
       const scores = Array.isArray(results) ? results.map(transformScoreRow) : [];
       return {
         pagination: {
           total,
-          page: pagination.page,
-          perPage: pagination.perPage,
-          hasMore: total > (pagination.page + 1) * pagination.perPage
+          page,
+          perPage: perPageForResponse,
+          hasMore: end < total
         },
         scores
       };
@@ -1726,11 +1527,14 @@ var ScoresStorageD1 = class extends ScoresStorage {
       );
     }
   }
-  async getScoresByRunId({
+  async listScoresByRunId({
     runId,
     pagination
   }) {
     try {
+      const { page, perPage: perPageInput } = pagination;
+      const perPage = normalizePerPage(perPageInput, 100);
+      const { offset: start, perPage: perPageForResponse } = calculatePagination(page, perPageInput, perPage);
       const fullTableName = this.operations.getTableName(TABLE_SCORERS);
       const countQuery = createSqlBuilder().count().from(fullTableName).where("runId = ?", runId);
       const countResult = await this.operations.executeQuery(countQuery.build());
@@ -1739,23 +1543,25 @@ var ScoresStorageD1 = class extends ScoresStorage {
         return {
           pagination: {
             total: 0,
-            page: pagination.page,
-            perPage: pagination.perPage,
+            page,
+            perPage: perPageForResponse,
             hasMore: false
           },
           scores: []
         };
       }
-      const selectQuery = createSqlBuilder().select("*").from(fullTableName).where("runId = ?", runId).limit(pagination.perPage).offset(pagination.page * pagination.perPage);
+      const end = perPageInput === false ? total : start + perPage;
+      const limitValue = perPageInput === false ? total : perPage;
+      const selectQuery = createSqlBuilder().select("*").from(fullTableName).where("runId = ?", runId).limit(limitValue).offset(start);
       const { sql, params } = selectQuery.build();
       const results = await this.operations.executeQuery({ sql, params });
       const scores = Array.isArray(results) ? results.map(transformScoreRow) : [];
       return {
         pagination: {
           total,
-          page: pagination.page,
-          perPage: pagination.perPage,
-          hasMore: total > (pagination.page + 1) * pagination.perPage
+          page,
+          perPage: perPageForResponse,
+          hasMore: end < total
         },
         scores
       };
@@ -1770,12 +1576,15 @@ var ScoresStorageD1 = class extends ScoresStorage {
       );
     }
   }
-  async getScoresByEntityId({
+  async listScoresByEntityId({
     entityId,
     entityType,
     pagination
   }) {
     try {
+      const { page, perPage: perPageInput } = pagination;
+      const perPage = normalizePerPage(perPageInput, 100);
+      const { offset: start, perPage: perPageForResponse } = calculatePagination(page, perPageInput, perPage);
       const fullTableName = this.operations.getTableName(TABLE_SCORERS);
       const countQuery = createSqlBuilder().count().from(fullTableName).where("entityId = ?", entityId).andWhere("entityType = ?", entityType);
       const countResult = await this.operations.executeQuery(countQuery.build());
@@ -1784,23 +1593,25 @@ var ScoresStorageD1 = class extends ScoresStorage {
         return {
           pagination: {
             total: 0,
-            page: pagination.page,
-            perPage: pagination.perPage,
+            page,
+            perPage: perPageForResponse,
             hasMore: false
           },
           scores: []
         };
       }
-      const selectQuery = createSqlBuilder().select("*").from(fullTableName).where("entityId = ?", entityId).andWhere("entityType = ?", entityType).limit(pagination.perPage).offset(pagination.page * pagination.perPage);
+      const end = perPageInput === false ? total : start + perPage;
+      const limitValue = perPageInput === false ? total : perPage;
+      const selectQuery = createSqlBuilder().select("*").from(fullTableName).where("entityId = ?", entityId).andWhere("entityType = ?", entityType).limit(limitValue).offset(start);
       const { sql, params } = selectQuery.build();
       const results = await this.operations.executeQuery({ sql, params });
       const scores = Array.isArray(results) ? results.map(transformScoreRow) : [];
       return {
         pagination: {
           total,
-          page: pagination.page,
-          perPage: pagination.perPage,
-          hasMore: total > (pagination.page + 1) * pagination.perPage
+          page,
+          perPage: perPageForResponse,
+          hasMore: end < total
         },
         scores
       };
@@ -1815,12 +1626,15 @@ var ScoresStorageD1 = class extends ScoresStorage {
       );
     }
   }
-  async getScoresBySpan({
+  async listScoresBySpan({
     traceId,
     spanId,
     pagination
   }) {
     try {
+      const { page, perPage: perPageInput } = pagination;
+      const perPage = normalizePerPage(perPageInput, 100);
+      const { offset: start, perPage: perPageForResponse } = calculatePagination(page, perPageInput, perPage);
       const fullTableName = this.operations.getTableName(TABLE_SCORERS);
       const countQuery = createSqlBuilder().count().from(fullTableName).where("traceId = ?", traceId).andWhere("spanId = ?", spanId);
       const countResult = await this.operations.executeQuery(countQuery.build());
@@ -1829,25 +1643,25 @@ var ScoresStorageD1 = class extends ScoresStorage {
         return {
           pagination: {
             total: 0,
-            page: pagination.page,
-            perPage: pagination.perPage,
+            page,
+            perPage: perPageForResponse,
             hasMore: false
           },
           scores: []
         };
       }
-      const limit = pagination.perPage + 1;
-      const selectQuery = createSqlBuilder().select("*").from(fullTableName).where("traceId = ?", traceId).andWhere("spanId = ?", spanId).orderBy("createdAt", "DESC").limit(limit).offset(pagination.page * pagination.perPage);
+      const end = perPageInput === false ? total : start + perPage;
+      const limitValue = perPageInput === false ? total : perPage;
+      const selectQuery = createSqlBuilder().select("*").from(fullTableName).where("traceId = ?", traceId).andWhere("spanId = ?", spanId).orderBy("createdAt", "DESC").limit(limitValue).offset(start);
       const { sql, params } = selectQuery.build();
       const results = await this.operations.executeQuery({ sql, params });
-      const rows = Array.isArray(results) ? results : [];
-      const scores = rows.slice(0, pagination.perPage).map(transformScoreRow);
+      const scores = Array.isArray(results) ? results.map(transformScoreRow) : [];
       return {
         pagination: {
           total,
-          page: pagination.page,
-          perPage: pagination.perPage,
-          hasMore: rows.length > pagination.perPage
+          page,
+          perPage: perPageForResponse,
+          hasMore: end < total
         },
         scores
       };
@@ -1863,128 +1677,6 @@ var ScoresStorageD1 = class extends ScoresStorage {
     }
   }
 };
-function isArrayOfRecords2(value) {
-  return value && Array.isArray(value) && value.length > 0;
-}
-var TracesStorageD1 = class extends TracesStorage {
-  operations;
-  constructor({ operations }) {
-    super();
-    this.operations = operations;
-  }
-  async getTraces(args) {
-    const paginatedArgs = {
-      name: args.name,
-      scope: args.scope,
-      page: args.page,
-      perPage: args.perPage,
-      attributes: args.attributes,
-      filters: args.filters,
-      dateRange: args.fromDate || args.toDate ? {
-        start: args.fromDate,
-        end: args.toDate
-      } : void 0
-    };
-    try {
-      const result = await this.getTracesPaginated(paginatedArgs);
-      return result.traces;
-    } catch (error) {
-      throw new MastraError(
-        {
-          id: "CLOUDFLARE_D1_STORAGE_GET_TRACES_ERROR",
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          text: `Failed to retrieve traces: ${error instanceof Error ? error.message : String(error)}`,
-          details: {
-            name: args.name ?? "",
-            scope: args.scope ?? ""
-          }
-        },
-        error
-      );
-    }
-  }
-  async getTracesPaginated(args) {
-    const { name, scope, page = 0, perPage = 100, attributes, dateRange } = args;
-    const fromDate = dateRange?.start;
-    const toDate = dateRange?.end;
-    const fullTableName = this.operations.getTableName(TABLE_TRACES);
-    try {
-      const dataQuery = createSqlBuilder().select("*").from(fullTableName).where("1=1");
-      const countQuery = createSqlBuilder().count().from(fullTableName).where("1=1");
-      if (name) {
-        dataQuery.andWhere("name LIKE ?", `%${name}%`);
-        countQuery.andWhere("name LIKE ?", `%${name}%`);
-      }
-      if (scope) {
-        dataQuery.andWhere("scope = ?", scope);
-        countQuery.andWhere("scope = ?", scope);
-      }
-      if (attributes && Object.keys(attributes).length > 0) {
-        for (const [key, value] of Object.entries(attributes)) {
-          dataQuery.jsonLike("attributes", key, value);
-          countQuery.jsonLike("attributes", key, value);
-        }
-      }
-      if (fromDate) {
-        const fromDateStr = fromDate instanceof Date ? fromDate.toISOString() : fromDate;
-        dataQuery.andWhere("createdAt >= ?", fromDateStr);
-        countQuery.andWhere("createdAt >= ?", fromDateStr);
-      }
-      if (toDate) {
-        const toDateStr = toDate instanceof Date ? toDate.toISOString() : toDate;
-        dataQuery.andWhere("createdAt <= ?", toDateStr);
-        countQuery.andWhere("createdAt <= ?", toDateStr);
-      }
-      const allDataResult = await this.operations.executeQuery(
-        createSqlBuilder().select("*").from(fullTableName).where("1=1").build()
-      );
-      console.info("allDataResult", allDataResult);
-      const countResult = await this.operations.executeQuery(countQuery.build());
-      const total = Number(countResult?.[0]?.count ?? 0);
-      dataQuery.orderBy("startTime", "DESC").limit(perPage).offset(page * perPage);
-      const results = await this.operations.executeQuery(dataQuery.build());
-      const traces = isArrayOfRecords2(results) ? results.map(
-        (trace) => ({
-          ...trace,
-          attributes: deserializeValue(trace.attributes, "jsonb"),
-          status: deserializeValue(trace.status, "jsonb"),
-          events: deserializeValue(trace.events, "jsonb"),
-          links: deserializeValue(trace.links, "jsonb"),
-          other: deserializeValue(trace.other, "jsonb")
-        })
-      ) : [];
-      return {
-        traces,
-        total,
-        page,
-        perPage,
-        hasMore: page * perPage + traces.length < total
-      };
-    } catch (error) {
-      const mastraError = new MastraError(
-        {
-          id: "CLOUDFLARE_D1_STORAGE_GET_TRACES_PAGINATED_ERROR",
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          text: `Failed to retrieve traces: ${error instanceof Error ? error.message : String(error)}`,
-          details: { name: name ?? "", scope: scope ?? "" }
-        },
-        error
-      );
-      this.logger?.error(mastraError.toString());
-      this.logger?.trackException(mastraError);
-      return { traces: [], total: 0, page, perPage, hasMore: false };
-    }
-  }
-  async batchTraceInsert({ records }) {
-    this.logger.debug("Batch inserting traces", { count: records.length });
-    await this.operations.batchInsert({
-      tableName: TABLE_TRACES,
-      records
-    });
-  }
-};
 var WorkflowsStorageD1 = class extends WorkflowsStorage {
   operations;
   constructor({ operations }) {
@@ -1996,7 +1688,7 @@ var WorkflowsStorageD1 = class extends WorkflowsStorage {
     // runId,
     // stepId,
     // result,
-    // runtimeContext,
+    // requestContext,
   }) {
     throw new Error("Method not implemented.");
   }
@@ -2100,19 +1792,24 @@ var WorkflowsStorageD1 = class extends WorkflowsStorage {
       resourceId: row.resourceId
     };
   }
-  async getWorkflowRuns({
+  async listWorkflowRuns({
     workflowName,
     fromDate,
     toDate,
-    limit,
-    offset,
-    resourceId
+    page,
+    perPage,
+    resourceId,
+    status
   } = {}) {
     const fullTableName = this.operations.getTableName(TABLE_WORKFLOW_SNAPSHOT);
     try {
       const builder = createSqlBuilder().select().from(fullTableName);
       const countBuilder = createSqlBuilder().count().from(fullTableName);
       if (workflowName) builder.whereAnd("workflow_name = ?", workflowName);
+      if (status) {
+        builder.whereAnd("json_extract(snapshot, '$.status') = ?", status);
+        countBuilder.whereAnd("json_extract(snapshot, '$.status') = ?", status);
+      }
       if (resourceId) {
         const hasResourceId = await this.operations.hasColumn(fullTableName, "resourceId");
         if (hasResourceId) {
@@ -2131,11 +1828,14 @@ var WorkflowsStorageD1 = class extends WorkflowsStorage {
         countBuilder.whereAnd("createdAt <= ?", toDate instanceof Date ? toDate.toISOString() : toDate);
       }
       builder.orderBy("createdAt", "DESC");
-      if (typeof limit === "number") builder.limit(limit);
-      if (typeof offset === "number") builder.offset(offset);
+      if (typeof perPage === "number" && typeof page === "number") {
+        const offset = page * perPage;
+        builder.limit(perPage);
+        builder.offset(offset);
+      }
       const { sql, params } = builder.build();
       let total = 0;
-      if (limit !== void 0 && offset !== void 0) {
+      if (perPage !== void 0 && page !== void 0) {
         const { sql: countSql, params: countParams } = countBuilder.build();
         const countResult = await this.operations.executeQuery({
           sql: countSql,
@@ -2150,7 +1850,7 @@ var WorkflowsStorageD1 = class extends WorkflowsStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLOUDFLARE_D1_STORAGE_GET_WORKFLOW_RUNS_ERROR",
+          id: "CLOUDFLARE_D1_STORAGE_LIST_WORKFLOW_RUNS_ERROR",
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           text: `Failed to retrieve workflow runs: ${error instanceof Error ? error.message : String(error)}`,
@@ -2212,7 +1912,7 @@ var D1Store = class extends MastraStorage {
    */
   constructor(config) {
     try {
-      super({ name: "D1" });
+      super({ id: config.id, name: "D1" });
       if (config.tablePrefix && !/^[a-zA-Z0-9_]*$/.test(config.tablePrefix)) {
         throw new Error("Invalid tablePrefix: only letters, numbers, and underscores are allowed.");
       }
@@ -2266,12 +1966,6 @@ var D1Store = class extends MastraStorage {
     const scores = new ScoresStorageD1({
       operations
     });
-    const legacyEvals = new LegacyEvalsStorageD1({
-      operations
-    });
-    const traces = new TracesStorageD1({
-      operations
-    });
     const workflows = new WorkflowsStorageD1({
       operations
     });
@@ -2281,8 +1975,6 @@ var D1Store = class extends MastraStorage {
     this.stores = {
       operations,
       scores,
-      legacyEvals,
-      traces,
       workflows,
       memory
     };
@@ -2294,7 +1986,7 @@ var D1Store = class extends MastraStorage {
       hasColumn: true,
       createTable: true,
       deleteMessages: false,
-      getScoresBySpan: true
+      listScoresBySpan: true
     };
   }
   async createTable({
@@ -2334,15 +2026,6 @@ var D1Store = class extends MastraStorage {
   async getThreadById({ threadId }) {
     return this.stores.memory.getThreadById({ threadId });
   }
-  /**
-   * @deprecated use getThreadsByResourceIdPaginated instead
-   */
-  async getThreadsByResourceId({ resourceId }) {
-    return this.stores.memory.getThreadsByResourceId({ resourceId });
-  }
-  async getThreadsByResourceIdPaginated(args) {
-    return this.stores.memory.getThreadsByResourceIdPaginated(args);
-  }
   async saveThread({ thread }) {
     return this.stores.memory.saveThread({ thread });
   }
@@ -2359,34 +2042,14 @@ var D1Store = class extends MastraStorage {
   async saveMessages(args) {
     return this.stores.memory.saveMessages(args);
   }
-  async getMessages({
-    threadId,
-    selectBy,
-    format
-  }) {
-    return this.stores.memory.getMessages({ threadId, selectBy, format });
-  }
-  async getMessagesById({
-    messageIds,
-    format
-  }) {
-    return this.stores.memory.getMessagesById({ messageIds, format });
-  }
-  async getMessagesPaginated({
-    threadId,
-    selectBy,
-    format
-  }) {
-    return this.stores.memory.getMessagesPaginated({ threadId, selectBy, format });
-  }
   async updateWorkflowResults({
     workflowName,
     runId,
     stepId,
     result,
-    runtimeContext
+    requestContext
   }) {
-    return this.stores.workflows.updateWorkflowResults({ workflowName, runId, stepId, result, runtimeContext });
+    return this.stores.workflows.updateWorkflowResults({ workflowName, runId, stepId, result, requestContext });
   }
   async updateWorkflowState({
     workflowName,
@@ -2406,15 +2069,8 @@ var D1Store = class extends MastraStorage {
   async loadWorkflowSnapshot(params) {
     return this.stores.workflows.loadWorkflowSnapshot(params);
   }
-  async getWorkflowRuns({
-    workflowName,
-    fromDate,
-    toDate,
-    limit,
-    offset,
-    resourceId
-  } = {}) {
-    return this.stores.workflows.getWorkflowRuns({ workflowName, fromDate, toDate, limit, offset, resourceId });
+  async listWorkflowRuns(args = {}) {
+    return this.stores.workflows.listWorkflowRuns(args);
   }
   async getWorkflowRunById({
     runId,
@@ -2430,24 +2086,6 @@ var D1Store = class extends MastraStorage {
   async batchInsert({ tableName, records }) {
     return this.stores.operations.batchInsert({ tableName, records });
   }
-  /**
-   * @deprecated use getTracesPaginated instead
-   */
-  async getTraces(args) {
-    return this.stores.traces.getTraces(args);
-  }
-  async getTracesPaginated(args) {
-    return this.stores.traces.getTracesPaginated(args);
-  }
-  /**
-   * @deprecated use getEvals instead
-   */
-  async getEvalsByAgentName(agentName, type) {
-    return this.stores.legacyEvals.getEvalsByAgentName(agentName, type);
-  }
-  async getEvals(options) {
-    return this.stores.legacyEvals.getEvals(options);
-  }
   async updateMessages(_args) {
     return this.stores.memory.updateMessages(_args);
   }
@@ -2470,38 +2108,38 @@ var D1Store = class extends MastraStorage {
   async saveScore(_score) {
     return this.stores.scores.saveScore(_score);
   }
-  async getScoresByRunId({
+  async listScoresByRunId({
     runId: _runId,
     pagination: _pagination
   }) {
-    return this.stores.scores.getScoresByRunId({ runId: _runId, pagination: _pagination });
+    return this.stores.scores.listScoresByRunId({ runId: _runId, pagination: _pagination });
   }
-  async getScoresByEntityId({
+  async listScoresByEntityId({
     entityId: _entityId,
     entityType: _entityType,
     pagination: _pagination
   }) {
-    return this.stores.scores.getScoresByEntityId({
+    return this.stores.scores.listScoresByEntityId({
       entityId: _entityId,
       entityType: _entityType,
       pagination: _pagination
     });
   }
-  async getScoresByScorerId({
+  async listScoresByScorerId({
     scorerId,
     pagination,
     entityId,
     entityType,
     source
   }) {
-    return this.stores.scores.getScoresByScorerId({ scorerId, pagination, entityId, entityType, source });
+    return this.stores.scores.listScoresByScorerId({ scorerId, pagination, entityId, entityType, source });
   }
-  async getScoresBySpan({
+  async listScoresBySpan({
     traceId,
     spanId,
     pagination
   }) {
-    return this.stores.scores.getScoresBySpan({ traceId, spanId, pagination });
+    return this.stores.scores.listScoresBySpan({ traceId, spanId, pagination });
   }
   /**
    * Close the database connection