@mastra/clickhouse 1.0.0-beta.2 → 1.0.0-beta.4

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,22 +1524,36 @@ 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 }
+          details: {
+            scorer: score.scorer?.id ?? "unknown",
+            entityId: score.entityId ?? "unknown",
+            entityType: score.entityType ?? "unknown",
+            traceId: score.traceId ?? "",
+            spanId: score.spanId ?? ""
+          }
         },
         error
       );
     }
+    const now = /* @__PURE__ */ new Date();
+    const id = crypto.randomUUID();
+    const createdAt = now;
+    const updatedAt = now;
     try {
       const record = {};
       for (const key of Object.keys(SCORERS_SCHEMA)) {
-        const value = parsedScore[key];
+        if (key === "id") {
+          record[key] = id;
+          continue;
+        }
         if (key === "createdAt" || key === "updatedAt") {
-          record[key] = (/* @__PURE__ */ new Date()).toISOString();
+          record[key] = now.toISOString();
           continue;
         }
+        const value = parsedScore[key];
         record[key] = value === void 0 || value === null ? "_null_" : value;
       }
       await this.client.insert({
@@ -1552,14 +1566,14 @@ var ScoresStorageClickhouse = class extends ScoresStorage {
           output_format_json_quote_64bit_integers: 0
         }
       });
-      return { score };
+      return { score: { ...parsedScore, id, createdAt, updatedAt } };
     } 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 }
+          details: { scoreId: id }
         },
         error
       );
@@ -1626,7 +1640,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 +1731,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 +1803,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 +1878,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 +1903,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 +1961,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 +1989,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 +2096,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 +2145,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 +2162,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 +2208,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 +2225,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 }
@@ -2312,8 +2336,8 @@ var ClickhouseStore = class extends MastraStorage {
   async getScoreById({ id }) {
     return this.stores.scores.getScoreById({ id });
   }
-  async saveScore(_score) {
-    return this.stores.scores.saveScore(_score);
+  async saveScore(score) {
+    return this.stores.scores.saveScore(score);
   }
   async listScoresByRunId({
     runId,