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

package/dist/index.js

@@ -1214,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);
     }
@@ -1238,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 ? `
@@ -1422,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 };