@mastra/clickhouse 0.0.0-vector-query-tool-provider-options-20250828222356 → 0.0.0-vector-extension-schema-20250922130418

package/dist/index.js

@@ -255,6 +255,7 @@ var MemoryStorageClickhouse = class extends MemoryStorage {
     format
   }) {
     try {
+      if (!threadId.trim()) throw new Error("threadId must be a non-empty string");
       const messages = [];
       const limit = resolveMessageLimit({ last: selectBy?.last, defaultLimit: 40 });
       const include = selectBy?.include || [];
@@ -847,11 +848,12 @@ var MemoryStorageClickhouse = class extends MemoryStorage {
     }
   }
   async getMessagesPaginated(args) {
+    const { threadId, resourceId, selectBy, format = "v1" } = args;
+    const page = selectBy?.pagination?.page || 0;
+    const perPageInput = selectBy?.pagination?.perPage;
+    const perPage = perPageInput !== void 0 ? perPageInput : resolveMessageLimit({ last: selectBy?.last, defaultLimit: 20 });
     try {
-      const { threadId, selectBy, format = "v1" } = args;
-      const page = selectBy?.pagination?.page || 0;
-      const perPageInput = selectBy?.pagination?.perPage;
-      const perPage = perPageInput !== void 0 ? perPageInput : resolveMessageLimit({ last: selectBy?.last, defaultLimit: 20 });
+      if (!threadId.trim()) throw new Error("threadId must be a non-empty string");
       const offset = page * perPage;
       const dateRange = selectBy?.pagination?.dateRange;
       const fromDate = dateRange?.start;
@@ -1010,14 +1012,21 @@ var MemoryStorageClickhouse = class extends MemoryStorage {
         hasMore: offset + perPage < total
       };
     } catch (error) {
-      throw new MastraError(
+      const mastraError = new MastraError(
         {
           id: "CLICKHOUSE_STORAGE_GET_MESSAGES_PAGINATED_FAILED",
           domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY
+          category: ErrorCategory.THIRD_PARTY,
+          details: {
+            threadId,
+            resourceId: resourceId ?? ""
+          }
         },
         error
       );
+      this.logger?.trackException?.(mastraError);
+      this.logger?.error?.(mastraError.toString());
+      return { messages: [], total: 0, page, perPage: perPageInput || 40, hasMore: false };
     }
   }
   async updateMessages(args) {
@@ -1100,7 +1109,7 @@ var MemoryStorageClickhouse = class extends MemoryStorage {
                 UPDATE ${setClauses.join(", ")}
                 WHERE id = {var_id_${paramIdx}:String}
               `;
-          console.log("Updating message:", id, "with query:", updateQuery, "values:", values);
+          console.info("Updating message:", id, "with query:", updateQuery, "values:", values);
           updatePromises.push(
             this.client.command({
               query: updateQuery,
@@ -1159,7 +1168,7 @@ var MemoryStorageClickhouse = class extends MemoryStorage {
               }
             }
             if (needsRetry) {
-              console.log("Update not applied correctly, retrying with DELETE + INSERT for message:", id);
+              console.info("Update not applied correctly, retrying with DELETE + INSERT for message:", id);
               await this.client.command({
                 query: `DELETE FROM ${TABLE_MESSAGES} WHERE id = {messageId:String}`,
                 query_params: { messageId: id },
@@ -1611,7 +1620,7 @@ var StoreOperationsClickhouse = class extends StoreOperations {
           use_client_time_zone: 1
         }
       });
-      console.log("INSERT RESULT", result);
+      console.info("INSERT RESULT", result);
     } catch (error) {
       throw new MastraError(
         {
@@ -2293,6 +2302,7 @@ var WorkflowsStorageClickhouse = class extends WorkflowsStorage {
   async persistWorkflowSnapshot({
     workflowName,
     runId,
+    resourceId,
     snapshot
   }) {
     try {
@@ -2303,11 +2313,13 @@ var WorkflowsStorageClickhouse = class extends WorkflowsStorage {
       const now = /* @__PURE__ */ new Date();
       const persisting = currentSnapshot ? {
         ...currentSnapshot,
+        resourceId,
         snapshot: JSON.stringify(snapshot),
         updatedAt: now.toISOString()
       } : {
         workflow_name: workflowName,
         run_id: runId,
+        resourceId,
         snapshot: JSON.stringify(snapshot),
         createdAt: now.toISOString(),
         updatedAt: now.toISOString()
@@ -2644,9 +2656,10 @@ var ClickhouseStore = class extends MastraStorage {
   async persistWorkflowSnapshot({
     workflowName,
     runId,
+    resourceId,
     snapshot
   }) {
-    return this.stores.workflows.persistWorkflowSnapshot({ workflowName, runId, snapshot });
+    return this.stores.workflows.persistWorkflowSnapshot({ workflowName, runId, resourceId, snapshot });
   }
   async loadWorkflowSnapshot({
     workflowName,