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

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",
@@ -1213,14 +1214,18 @@ var StoreOperationsClickhouse = class extends StoreOperations {
   getSqlType(type) {
     switch (type) {
       case "text":
+      case "uuid":
+      case "jsonb":
         return "String";
       case "timestamp":
         return "DateTime64(3)";
       case "integer":
       case "bigint":
         return "Int64";
-      case "jsonb":
-        return "String";
+      case "float":
+        return "Float64";
+      case "boolean":
+        return "Bool";
       default:
         return super.getSqlType(type);
     }
@@ -1237,7 +1242,7 @@ var StoreOperationsClickhouse = class extends StoreOperations {
           constraints.push("DEFAULT '{}'");
         }
         const columnTtl = this.ttl?.[tableName]?.columns?.[name];
-        return `"${name}" ${COLUMN_TYPES[def.type]} ${constraints.join(" ")} ${columnTtl ? `TTL toDateTime(${columnTtl.ttlKey ?? "createdAt"}) + INTERVAL ${columnTtl.interval} ${columnTtl.unit}` : ""}`;
+        return `"${name}" ${this.getSqlType(def.type)} ${constraints.join(" ")} ${columnTtl ? `TTL toDateTime(${columnTtl.ttlKey ?? "createdAt"}) + INTERVAL ${columnTtl.interval} ${columnTtl.unit}` : ""}`;
       }).join(",\n");
       const rowTtl = this.ttl?.[tableName]?.row;
       const sql = tableName === TABLE_WORKFLOW_SNAPSHOT ? `
@@ -1421,7 +1426,7 @@ var StoreOperationsClickhouse = class extends StoreOperations {
       const engine = TABLE_ENGINES[tableName] ?? "MergeTree()";
       const keyEntries = Object.entries(keys);
       const conditions = keyEntries.map(
-        ([key]) => `"${key}" = {var_${key}:${COLUMN_TYPES[TABLE_SCHEMAS[tableName]?.[key]?.type ?? "text"]}}`
+        ([key]) => `"${key}" = {var_${key}:${this.getSqlType(TABLE_SCHEMAS[tableName]?.[key]?.type ?? "text")}}`
       ).join(" AND ");
       const values = keyEntries.reduce((acc, [key, value]) => {
         return { ...acc, [`var_${key}`]: value };
@@ -2154,6 +2159,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
@@ -2298,6 +2325,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 });
   }