@mastra/clickhouse 0.13.0 → 0.13.1-alpha.0

package/dist/index.js

@@ -1611,7 +1611,7 @@ var StoreOperationsClickhouse = class extends StoreOperations {
       const hasUpdatedAt = TABLE_SCHEMAS[tableName]?.updatedAt;
       const selectClause = `SELECT *, toDateTime64(createdAt, 3) as createdAt${hasUpdatedAt ? ", toDateTime64(updatedAt, 3) as updatedAt" : ""}`;
       const result = await this.client.query({
-        query: `${selectClause} FROM ${tableName} ${engine.startsWith("ReplacingMergeTree") ? "FINAL" : ""} WHERE ${conditions}`,
+        query: `${selectClause} FROM ${tableName} ${engine.startsWith("ReplacingMergeTree") ? "FINAL" : ""} WHERE ${conditions} ORDER BY createdAt DESC LIMIT 1`,
         query_params: values,
         clickhouse_settings: {
           // Allows to insert serialized JS Dates (such as '2023-12-06T10:54:48.000Z')
@@ -1810,12 +1810,30 @@ var ScoresStorageClickhouse = class extends ScoresStorage {
   }
   async getScoresByScorerId({
     scorerId,
+    entityId,
+    entityType,
+    source,
     pagination
   }) {
+    let whereClause = `scorerId = {var_scorerId:String}`;
+    if (entityId) {
+      whereClause += ` AND entityId = {var_entityId:String}`;
+    }
+    if (entityType) {
+      whereClause += ` AND entityType = {var_entityType:String}`;
+    }
+    if (source) {
+      whereClause += ` AND source = {var_source:String}`;
+    }
     try {
       const countResult = await this.client.query({
-        query: `SELECT COUNT(*) as count FROM ${TABLE_SCORERS} WHERE scorerId = {var_scorerId:String}`,
-        query_params: { var_scorerId: scorerId },
+        query: `SELECT COUNT(*) as count FROM ${TABLE_SCORERS} WHERE ${whereClause}`,
+        query_params: {
+          var_scorerId: scorerId,
+          var_entityId: entityId,
+          var_entityType: entityType,
+          var_source: source
+        },
         format: "JSONEachRow"
       });
       const countRows = await countResult.json();
@@ -1837,11 +1855,14 @@ var ScoresStorageClickhouse = class extends ScoresStorage {
       }
       const offset = pagination.page * pagination.perPage;
       const result = await this.client.query({
-        query: `SELECT * FROM ${TABLE_SCORERS} WHERE scorerId = {var_scorerId:String} ORDER BY createdAt DESC LIMIT {var_limit:Int64} OFFSET {var_offset:Int64}`,
+        query: `SELECT * FROM ${TABLE_SCORERS} WHERE ${whereClause} ORDER BY createdAt DESC LIMIT {var_limit:Int64} OFFSET {var_offset:Int64}`,
         query_params: {
           var_scorerId: scorerId,
           var_limit: pagination.perPage,
-          var_offset: offset
+          var_offset: offset,
+          var_entityId: entityId,
+          var_entityType: entityType,
+          var_source: source
         },
         format: "JSONEachRow",
         clickhouse_settings: {
@@ -2393,6 +2414,7 @@ var WorkflowsStorageClickhouse = class extends WorkflowsStorage {
                 resourceId
               FROM ${TABLE_WORKFLOW_SNAPSHOT} ${TABLE_ENGINES[TABLE_WORKFLOW_SNAPSHOT].startsWith("ReplacingMergeTree") ? "FINAL" : ""}
               ${whereClause}
+              ORDER BY createdAt DESC LIMIT 1
             `,
         query_params: values,
         format: "JSONEachRow"
@@ -2639,9 +2661,12 @@ var ClickhouseStore = class extends MastraStorage {
   }
   async getScoresByScorerId({
     scorerId,
-    pagination
+    pagination,
+    entityId,
+    entityType,
+    source
   }) {
-    return this.stores.scores.getScoresByScorerId({ scorerId, pagination });
+    return this.stores.scores.getScoresByScorerId({ scorerId, pagination, entityId, entityType, source });
   }
   async close() {
     await this.db.close();