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

package/dist/index.js

@@ -13,7 +13,8 @@ var TABLE_ENGINES = {
   [TABLE_SCORERS]: `MergeTree()`,
   [TABLE_RESOURCES]: `ReplacingMergeTree()`,
   // TODO: verify this is the correct engine for Spans when implementing clickhouse storage
-  [TABLE_SPANS]: `ReplacingMergeTree()`
+  [TABLE_SPANS]: `ReplacingMergeTree()`,
+  mastra_agents: `ReplacingMergeTree()`
 };
 var COLUMN_TYPES = {
   text: "String",
@@ -1527,19 +1528,33 @@ var ScoresStorageClickhouse = class extends ScoresStorage {
           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 +1567,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: createStorageErrorId("CLICKHOUSE", "SAVE_SCORE", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
-          details: { scoreId: score.id }
+          details: { scoreId: id }
         },
         error
       );
@@ -2140,6 +2155,28 @@ var WorkflowsStorageClickhouse = class extends WorkflowsStorage {
       );
     }
   }
+  async deleteWorkflowRunById({ runId, workflowName }) {
+    try {
+      const values = {
+        var_runId: runId,
+        var_workflow_name: workflowName
+      };
+      await this.client.command({
+        query: `DELETE FROM ${TABLE_WORKFLOW_SNAPSHOT} WHERE run_id = {var_runId:String} AND workflow_name = {var_workflow_name:String}`,
+        query_params: values
+      });
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: createStorageErrorId("CLICKHOUSE", "DELETE_WORKFLOW_RUN_BY_ID", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { runId, workflowName }
+        },
+        error
+      );
+    }
+  }
 };
 
 // src/storage/index.ts
@@ -2284,6 +2321,9 @@ var ClickhouseStore = class extends MastraStorage {
   }) {
     return this.stores.workflows.getWorkflowRunById({ runId, workflowName });
   }
+  async deleteWorkflowRunById({ runId, workflowName }) {
+    return this.stores.workflows.deleteWorkflowRunById({ runId, workflowName });
+  }
   async getThreadById({ threadId }) {
     return this.stores.memory.getThreadById({ threadId });
   }
@@ -2322,8 +2362,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,