@mastra/clickhouse 1.0.0-beta.2 → 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, transformScoreRow, SCORERS_SCHEMA, calculatePagination, MemoryStorage, safelyParseJSON } 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) }
@@ -130,7 +130,7 @@ var MemoryStorageClickhouse = class extends MemoryStorage {
     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 }
@@ -141,7 +141,7 @@ var MemoryStorageClickhouse = class extends MemoryStorage {
     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: Array.isArray(threadId) ? JSON.stringify(threadId) : String(threadId) }
@@ -346,7 +346,7 @@ 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: {
@@ -509,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
         },
@@ -554,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 }
@@ -587,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 }
@@ -639,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 }
@@ -667,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 }
@@ -682,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 }
@@ -767,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 }
@@ -1049,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(",") }
@@ -1085,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 }
@@ -1118,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 }
@@ -1184,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 }
@@ -1272,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 }
@@ -1311,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 }
@@ -1335,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 }
@@ -1374,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 }
@@ -1407,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 }
@@ -1458,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 }
@@ -1508,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 }
@@ -1524,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 }
@@ -1556,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 }
@@ -1626,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 }
@@ -1717,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 }
@@ -1789,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 }
@@ -1864,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 }
@@ -1889,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,
@@ -1937,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 }
@@ -1965,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 }
@@ -2072,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 ?? "" }
@@ -2121,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 ?? "" }
@@ -2138,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,
@@ -2184,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 }
@@ -2201,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 }