@mastra/clickhouse 1.0.0-beta.1 → 1.0.0-beta.3

package/dist/index.js

@@ -1,6 +1,6 @@
 import { createClient } from '@clickhouse/client';
 import { MastraError, ErrorCategory, ErrorDomain } from '@mastra/core/error';
-import { TABLE_SPANS, TABLE_RESOURCES, TABLE_SCORERS, TABLE_THREADS, TABLE_TRACES, TABLE_WORKFLOW_SNAPSHOT, TABLE_MESSAGES, MastraStorage, StoreOperations, TABLE_SCHEMAS, WorkflowsStorage, normalizePerPage, ScoresStorage, safelyParseJSON, calculatePagination, MemoryStorage } from '@mastra/core/storage';
+import { TABLE_SPANS, TABLE_RESOURCES, TABLE_SCORERS, TABLE_THREADS, TABLE_TRACES, TABLE_WORKFLOW_SNAPSHOT, TABLE_MESSAGES, MastraStorage, createStorageErrorId, StoreOperations, TABLE_SCHEMAS, WorkflowsStorage, normalizePerPage, ScoresStorage, transformScoreRow, SCORERS_SCHEMA, calculatePagination, MemoryStorage, safelyParseJSON } from '@mastra/core/storage';
 import { MessageList } from '@mastra/core/agent';
 import { saveScorePayloadSchema } from '@mastra/core/evals';
 
@@ -114,7 +114,7 @@ var MemoryStorageClickhouse = class extends MemoryStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLICKHOUSE_STORAGE_LIST_MESSAGES_BY_ID_FAILED",
+          id: createStorageErrorId("CLICKHOUSE", "LIST_MESSAGES_BY_ID", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { messageIds: JSON.stringify(messageIds) }
@@ -125,10 +125,12 @@ var MemoryStorageClickhouse = class extends MemoryStorage {
   }
   async listMessages(args) {
     const { threadId, resourceId, include, filter, perPage: perPageInput, page = 0, orderBy } = args;
+    const rawThreadIds = Array.isArray(threadId) ? threadId : [threadId];
+    const threadIds = rawThreadIds.filter((id) => id !== void 0 && id !== null).map((id) => (typeof id === "string" ? id : String(id)).trim()).filter((id) => id.length > 0);
     if (page < 0) {
       throw new MastraError(
         {
-          id: "STORAGE_CLICKHOUSE_LIST_MESSAGES_INVALID_PAGE",
+          id: createStorageErrorId("CLICKHOUSE", "LIST_MESSAGES", "INVALID_PAGE"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER,
           details: { page }
@@ -136,20 +138,21 @@ var MemoryStorageClickhouse = class extends MemoryStorage {
         new Error("page must be >= 0")
       );
     }
-    if (!threadId.trim()) {
+    if (threadIds.length === 0) {
       throw new MastraError(
         {
-          id: "STORAGE_CLICKHOUSE_LIST_MESSAGES_INVALID_THREAD_ID",
+          id: createStorageErrorId("CLICKHOUSE", "LIST_MESSAGES", "INVALID_THREAD_ID"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
-          details: { threadId }
+          details: { threadId: Array.isArray(threadId) ? JSON.stringify(threadId) : String(threadId) }
         },
-        new Error("threadId must be a non-empty string")
+        new Error("threadId must be a non-empty string or array of non-empty strings")
       );
     }
     const perPageForQuery = normalizePerPage(perPageInput, 40);
     const { offset, perPage: perPageForResponse } = calculatePagination(page, perPageInput, perPageForQuery);
     try {
+      const threadCondition = threadIds.length === 1 ? `thread_id = {threadId0:String}` : `thread_id IN (${threadIds.map((_, i) => `{threadId${i}:String}`).join(", ")})`;
       let dataQuery = `
         SELECT 
           id,
@@ -160,9 +163,12 @@ var MemoryStorageClickhouse = class extends MemoryStorage {
           thread_id AS "threadId",
           resourceId
         FROM ${TABLE_MESSAGES}
-        WHERE thread_id = {threadId:String}
+        WHERE ${threadCondition}
       `;
-      const dataParams = { threadId };
+      const dataParams = {};
+      threadIds.forEach((tid, i) => {
+        dataParams[`threadId${i}`] = tid;
+      });
       if (resourceId) {
         dataQuery += ` AND resourceId = {resourceId:String}`;
         dataParams.resourceId = resourceId;
@@ -197,8 +203,11 @@ var MemoryStorageClickhouse = class extends MemoryStorage {
       const rows = await result.json();
       const paginatedMessages = transformRows(rows.data);
       const paginatedCount = paginatedMessages.length;
-      let countQuery = `SELECT count() as total FROM ${TABLE_MESSAGES} WHERE thread_id = {threadId:String}`;
-      const countParams = { threadId };
+      let countQuery = `SELECT count() as total FROM ${TABLE_MESSAGES} WHERE ${threadCondition}`;
+      const countParams = {};
+      threadIds.forEach((tid, i) => {
+        countParams[`threadId${i}`] = tid;
+      });
       if (resourceId) {
         countQuery += ` AND resourceId = {resourceId:String}`;
         countParams.resourceId = resourceId;
@@ -237,12 +246,25 @@ var MemoryStorageClickhouse = class extends MemoryStorage {
       const messageIds = new Set(paginatedMessages.map((m) => m.id));
       let includeMessages = [];
       if (include && include.length > 0) {
+        const includesNeedingThread = include.filter((inc) => !inc.threadId);
+        const threadByMessageId = /* @__PURE__ */ new Map();
+        if (includesNeedingThread.length > 0) {
+          const { messages: includeLookup } = await this.listMessagesById({
+            messageIds: includesNeedingThread.map((inc) => inc.id)
+          });
+          for (const msg of includeLookup) {
+            if (msg.threadId) {
+              threadByMessageId.set(msg.id, msg.threadId);
+            }
+          }
+        }
         const unionQueries = [];
         const params = [];
         let paramIdx = 1;
         for (const inc of include) {
           const { id, withPreviousMessages = 0, withNextMessages = 0 } = inc;
-          const searchId = inc.threadId || threadId;
+          const searchThreadId = inc.threadId ?? threadByMessageId.get(id);
+          if (!searchThreadId) continue;
           unionQueries.push(`
             SELECT * FROM (
               WITH numbered_messages AS (
@@ -264,31 +286,33 @@ var MemoryStorageClickhouse = class extends MemoryStorage {
             ) AS query_${paramIdx}
           `);
           params.push(
-            { [`var_thread_id_${paramIdx}`]: searchId },
+            { [`var_thread_id_${paramIdx}`]: searchThreadId },
             { [`var_include_id_${paramIdx}`]: id },
             { [`var_withPreviousMessages_${paramIdx}`]: withPreviousMessages },
             { [`var_withNextMessages_${paramIdx}`]: withNextMessages }
           );
           paramIdx++;
         }
-        const finalQuery = unionQueries.join(" UNION ALL ") + ' ORDER BY "createdAt" ASC';
-        const mergedParams = params.reduce((acc, paramObj) => ({ ...acc, ...paramObj }), {});
-        const includeResult = await this.client.query({
-          query: finalQuery,
-          query_params: mergedParams,
-          clickhouse_settings: {
-            date_time_input_format: "best_effort",
-            date_time_output_format: "iso",
-            use_client_time_zone: 1,
-            output_format_json_quote_64bit_integers: 0
-          }
-        });
-        const includeRows = await includeResult.json();
-        includeMessages = transformRows(includeRows.data);
-        for (const includeMsg of includeMessages) {
-          if (!messageIds.has(includeMsg.id)) {
-            paginatedMessages.push(includeMsg);
-            messageIds.add(includeMsg.id);
+        if (unionQueries.length > 0) {
+          const finalQuery = unionQueries.join(" UNION ALL ") + ' ORDER BY "createdAt" ASC';
+          const mergedParams = params.reduce((acc, paramObj) => ({ ...acc, ...paramObj }), {});
+          const includeResult = await this.client.query({
+            query: finalQuery,
+            query_params: mergedParams,
+            clickhouse_settings: {
+              date_time_input_format: "best_effort",
+              date_time_output_format: "iso",
+              use_client_time_zone: 1,
+              output_format_json_quote_64bit_integers: 0
+            }
+          });
+          const includeRows = await includeResult.json();
+          includeMessages = transformRows(includeRows.data);
+          for (const includeMsg of includeMessages) {
+            if (!messageIds.has(includeMsg.id)) {
+              paginatedMessages.push(includeMsg);
+              messageIds.add(includeMsg.id);
+            }
           }
         }
       }
@@ -306,7 +330,10 @@ var MemoryStorageClickhouse = class extends MemoryStorage {
         }
         return direction === "ASC" ? String(aValue).localeCompare(String(bValue)) : String(bValue).localeCompare(String(aValue));
       });
-      const returnedThreadMessageIds = new Set(finalMessages.filter((m) => m.threadId === threadId).map((m) => m.id));
+      const threadIdSet = new Set(threadIds);
+      const returnedThreadMessageIds = new Set(
+        finalMessages.filter((m) => m.threadId && threadIdSet.has(m.threadId)).map((m) => m.id)
+      );
       const allThreadMessagesReturned = returnedThreadMessageIds.size >= total;
       const hasMore = perPageForResponse === false ? false : allThreadMessagesReturned ? false : offset + paginatedCount < total;
       return {
@@ -319,11 +346,11 @@ var MemoryStorageClickhouse = class extends MemoryStorage {
     } catch (error) {
       const mastraError = new MastraError(
         {
-          id: "STORAGE_CLICKHOUSE_STORE_LIST_MESSAGES_FAILED",
+          id: createStorageErrorId("CLICKHOUSE", "LIST_MESSAGES", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
-            threadId,
+            threadId: Array.isArray(threadId) ? threadId.join(",") : threadId,
             resourceId: resourceId ?? ""
           }
         },
@@ -482,7 +509,7 @@ var MemoryStorageClickhouse = class extends MemoryStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLICKHOUSE_STORAGE_SAVE_MESSAGES_FAILED",
+          id: createStorageErrorId("CLICKHOUSE", "SAVE_MESSAGES", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY
         },
@@ -527,7 +554,7 @@ var MemoryStorageClickhouse = class extends MemoryStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLICKHOUSE_STORAGE_GET_THREAD_BY_ID_FAILED",
+          id: createStorageErrorId("CLICKHOUSE", "GET_THREAD_BY_ID", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { threadId }
@@ -560,7 +587,7 @@ var MemoryStorageClickhouse = class extends MemoryStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLICKHOUSE_STORAGE_SAVE_THREAD_FAILED",
+          id: createStorageErrorId("CLICKHOUSE", "SAVE_THREAD", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { threadId: thread.id }
@@ -612,7 +639,7 @@ var MemoryStorageClickhouse = class extends MemoryStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLICKHOUSE_STORAGE_UPDATE_THREAD_FAILED",
+          id: createStorageErrorId("CLICKHOUSE", "UPDATE_THREAD", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { threadId: id, title }
@@ -640,7 +667,7 @@ var MemoryStorageClickhouse = class extends MemoryStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLICKHOUSE_STORAGE_DELETE_THREAD_FAILED",
+          id: createStorageErrorId("CLICKHOUSE", "DELETE_THREAD", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { threadId }
@@ -655,7 +682,7 @@ var MemoryStorageClickhouse = class extends MemoryStorage {
     if (page < 0) {
       throw new MastraError(
         {
-          id: "STORAGE_CLICKHOUSE_LIST_THREADS_BY_RESOURCE_ID_INVALID_PAGE",
+          id: createStorageErrorId("CLICKHOUSE", "LIST_THREADS_BY_RESOURCE_ID", "INVALID_PAGE"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER,
           details: { page }
@@ -740,7 +767,7 @@ var MemoryStorageClickhouse = class extends MemoryStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLICKHOUSE_STORAGE_LIST_THREADS_BY_RESOURCE_ID_FAILED",
+          id: createStorageErrorId("CLICKHOUSE", "LIST_THREADS_BY_RESOURCE_ID", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { resourceId, page }
@@ -1022,7 +1049,7 @@ var MemoryStorageClickhouse = class extends MemoryStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLICKHOUSE_STORAGE_UPDATE_MESSAGES_FAILED",
+          id: createStorageErrorId("CLICKHOUSE", "UPDATE_MESSAGES", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { messageIds: messages.map((m) => m.id).join(",") }
@@ -1058,7 +1085,7 @@ var MemoryStorageClickhouse = class extends MemoryStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLICKHOUSE_STORAGE_GET_RESOURCE_BY_ID_FAILED",
+          id: createStorageErrorId("CLICKHOUSE", "GET_RESOURCE_BY_ID", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { resourceId }
@@ -1091,7 +1118,7 @@ var MemoryStorageClickhouse = class extends MemoryStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLICKHOUSE_STORAGE_SAVE_RESOURCE_FAILED",
+          id: createStorageErrorId("CLICKHOUSE", "SAVE_RESOURCE", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { resourceId: resource.id }
@@ -1157,7 +1184,7 @@ var MemoryStorageClickhouse = class extends MemoryStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLICKHOUSE_STORAGE_UPDATE_RESOURCE_FAILED",
+          id: createStorageErrorId("CLICKHOUSE", "UPDATE_RESOURCE", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { resourceId }
@@ -1245,7 +1272,7 @@ var StoreOperationsClickhouse = class extends StoreOperations {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLICKHOUSE_STORAGE_CREATE_TABLE_FAILED",
+          id: createStorageErrorId("CLICKHOUSE", "CREATE_TABLE", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { tableName }
@@ -1284,7 +1311,7 @@ var StoreOperationsClickhouse = class extends StoreOperations {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLICKHOUSE_STORAGE_ALTER_TABLE_FAILED",
+          id: createStorageErrorId("CLICKHOUSE", "ALTER_TABLE", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { tableName }
@@ -1308,7 +1335,7 @@ var StoreOperationsClickhouse = class extends StoreOperations {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLICKHOUSE_STORAGE_CLEAR_TABLE_FAILED",
+          id: createStorageErrorId("CLICKHOUSE", "CLEAR_TABLE", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { tableName }
@@ -1347,7 +1374,7 @@ var StoreOperationsClickhouse = class extends StoreOperations {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLICKHOUSE_STORAGE_INSERT_FAILED",
+          id: createStorageErrorId("CLICKHOUSE", "INSERT", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { tableName }
@@ -1380,7 +1407,7 @@ var StoreOperationsClickhouse = class extends StoreOperations {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLICKHOUSE_STORAGE_BATCH_INSERT_FAILED",
+          id: createStorageErrorId("CLICKHOUSE", "BATCH_INSERT", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { tableName }
@@ -1431,7 +1458,7 @@ var StoreOperationsClickhouse = class extends StoreOperations {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLICKHOUSE_STORAGE_LOAD_FAILED",
+          id: createStorageErrorId("CLICKHOUSE", "LOAD", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { tableName }
@@ -1449,30 +1476,15 @@ var ScoresStorageClickhouse = class extends ScoresStorage {
     this.client = client;
     this.operations = operations;
   }
+  /**
+   * ClickHouse-specific score row transformation.
+   * Converts timestamps to Date objects and filters out '_null_' values.
+   */
   transformScoreRow(row) {
-    const scorer = safelyParseJSON(row.scorer);
-    const preprocessStepResult = safelyParseJSON(row.preprocessStepResult);
-    const analyzeStepResult = safelyParseJSON(row.analyzeStepResult);
-    const metadata = safelyParseJSON(row.metadata);
-    const input = safelyParseJSON(row.input);
-    const output = safelyParseJSON(row.output);
-    const additionalContext = safelyParseJSON(row.additionalContext);
-    const requestContext = safelyParseJSON(row.requestContext);
-    const entity = safelyParseJSON(row.entity);
-    return {
-      ...row,
-      scorer,
-      preprocessStepResult,
-      analyzeStepResult,
-      metadata,
-      input,
-      output,
-      additionalContext,
-      requestContext,
-      entity,
-      createdAt: new Date(row.createdAt),
-      updatedAt: new Date(row.updatedAt)
-    };
+    return transformScoreRow(row, {
+      convertTimestamps: true,
+      nullValuePattern: "_null_"
+    });
   }
   async getScoreById({ id }) {
     try {
@@ -1496,7 +1508,7 @@ var ScoresStorageClickhouse = class extends ScoresStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLICKHOUSE_STORAGE_GET_SCORE_BY_ID_FAILED",
+          id: createStorageErrorId("CLICKHOUSE", "GET_SCORE_BY_ID", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { scoreId: id }
@@ -1512,7 +1524,7 @@ var ScoresStorageClickhouse = class extends ScoresStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLICKHOUSE_STORAGE_SAVE_SCORE_FAILED_INVALID_SCORE_PAYLOAD",
+          id: createStorageErrorId("CLICKHOUSE", "SAVE_SCORE", "VALIDATION_FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER,
           details: { scoreId: score.id }
@@ -1521,9 +1533,15 @@ var ScoresStorageClickhouse = class extends ScoresStorage {
       );
     }
     try {
-      const record = {
-        ...parsedScore
-      };
+      const record = {};
+      for (const key of Object.keys(SCORERS_SCHEMA)) {
+        const value = parsedScore[key];
+        if (key === "createdAt" || key === "updatedAt") {
+          record[key] = (/* @__PURE__ */ new Date()).toISOString();
+          continue;
+        }
+        record[key] = value === void 0 || value === null ? "_null_" : value;
+      }
       await this.client.insert({
         table: TABLE_SCORERS,
         values: [record],
@@ -1538,7 +1556,7 @@ var ScoresStorageClickhouse = class extends ScoresStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLICKHOUSE_STORAGE_SAVE_SCORE_FAILED",
+          id: createStorageErrorId("CLICKHOUSE", "SAVE_SCORE", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { scoreId: score.id }
@@ -1608,7 +1626,7 @@ var ScoresStorageClickhouse = class extends ScoresStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLICKHOUSE_STORAGE_GET_SCORES_BY_RUN_ID_FAILED",
+          id: createStorageErrorId("CLICKHOUSE", "LIST_SCORES_BY_RUN_ID", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { runId }
@@ -1699,7 +1717,7 @@ var ScoresStorageClickhouse = class extends ScoresStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLICKHOUSE_STORAGE_GET_SCORES_BY_SCORER_ID_FAILED",
+          id: createStorageErrorId("CLICKHOUSE", "LIST_SCORES_BY_SCORER_ID", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { scorerId }
@@ -1771,7 +1789,7 @@ var ScoresStorageClickhouse = class extends ScoresStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLICKHOUSE_STORAGE_GET_SCORES_BY_ENTITY_ID_FAILED",
+          id: createStorageErrorId("CLICKHOUSE", "LIST_SCORES_BY_ENTITY_ID", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { entityId, entityType }
@@ -1846,7 +1864,7 @@ var ScoresStorageClickhouse = class extends ScoresStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLICKHOUSE_STORAGE_GET_SCORES_BY_SPAN_FAILED",
+          id: createStorageErrorId("CLICKHOUSE", "LIST_SCORES_BY_SPAN", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { traceId, spanId }
@@ -1871,14 +1889,24 @@ var WorkflowsStorageClickhouse = class extends WorkflowsStorage {
     // result,
     // requestContext,
   }) {
-    throw new Error("Method not implemented.");
+    throw new MastraError({
+      id: createStorageErrorId("CLICKHOUSE", "UPDATE_WORKFLOW_RESULTS", "NOT_IMPLEMENTED"),
+      domain: ErrorDomain.STORAGE,
+      category: ErrorCategory.SYSTEM,
+      text: "Method not implemented."
+    });
   }
   updateWorkflowState({
     // workflowName,
     // runId,
     // opts,
   }) {
-    throw new Error("Method not implemented.");
+    throw new MastraError({
+      id: createStorageErrorId("CLICKHOUSE", "UPDATE_WORKFLOW_STATE", "NOT_IMPLEMENTED"),
+      domain: ErrorDomain.STORAGE,
+      category: ErrorCategory.SYSTEM,
+      text: "Method not implemented."
+    });
   }
   async persistWorkflowSnapshot({
     workflowName,
@@ -1919,7 +1947,7 @@ var WorkflowsStorageClickhouse = class extends WorkflowsStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLICKHOUSE_STORAGE_PERSIST_WORKFLOW_SNAPSHOT_FAILED",
+          id: createStorageErrorId("CLICKHOUSE", "PERSIST_WORKFLOW_SNAPSHOT", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { workflowName, runId }
@@ -1947,7 +1975,7 @@ var WorkflowsStorageClickhouse = class extends WorkflowsStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLICKHOUSE_STORAGE_LOAD_WORKFLOW_SNAPSHOT_FAILED",
+          id: createStorageErrorId("CLICKHOUSE", "LOAD_WORKFLOW_SNAPSHOT", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { workflowName, runId }
@@ -2054,7 +2082,7 @@ var WorkflowsStorageClickhouse = class extends WorkflowsStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLICKHOUSE_STORAGE_LIST_WORKFLOW_RUNS_FAILED",
+          id: createStorageErrorId("CLICKHOUSE", "LIST_WORKFLOW_RUNS", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { workflowName: workflowName ?? "", resourceId: resourceId ?? "" }
@@ -2103,7 +2131,7 @@ var WorkflowsStorageClickhouse = class extends WorkflowsStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLICKHOUSE_STORAGE_GET_WORKFLOW_RUN_BY_ID_FAILED",
+          id: createStorageErrorId("CLICKHOUSE", "GET_WORKFLOW_RUN_BY_ID", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { runId: runId ?? "", workflowName: workflowName ?? "" }
@@ -2120,7 +2148,7 @@ var ClickhouseStore = class extends MastraStorage {
   ttl = {};
   stores;
   constructor(config) {
-    super({ id: config.id, name: "ClickhouseStore" });
+    super({ id: config.id, name: "ClickhouseStore", disableInit: config.disableInit });
     this.db = createClient({
       url: config.url,
       username: config.username,
@@ -2166,7 +2194,7 @@ var ClickhouseStore = class extends MastraStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLICKHOUSE_STORAGE_OPTIMIZE_TABLE_FAILED",
+          id: createStorageErrorId("CLICKHOUSE", "OPTIMIZE_TABLE", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { tableName }
@@ -2183,7 +2211,7 @@ var ClickhouseStore = class extends MastraStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLICKHOUSE_STORAGE_MATERIALIZE_TTL_FAILED",
+          id: createStorageErrorId("CLICKHOUSE", "MATERIALIZE_TTL", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { tableName }