@mastra/clickhouse 0.13.2 → 0.13.3-alpha.0

package/dist/index.js

@@ -1,6 +1,6 @@
 import { createClient } from '@clickhouse/client';
 import { MastraError, ErrorCategory, ErrorDomain } from '@mastra/core/error';
-import { TABLE_RESOURCES, TABLE_SCORERS, TABLE_EVALS, TABLE_THREADS, TABLE_TRACES, TABLE_WORKFLOW_SNAPSHOT, TABLE_MESSAGES, MastraStorage, StoreOperations, TABLE_SCHEMAS, WorkflowsStorage, ScoresStorage, safelyParseJSON, LegacyEvalsStorage, TracesStorage, MemoryStorage, resolveMessageLimit } from '@mastra/core/storage';
+import { TABLE_AI_SPANS, TABLE_RESOURCES, TABLE_SCORERS, TABLE_EVALS, TABLE_THREADS, TABLE_TRACES, TABLE_WORKFLOW_SNAPSHOT, TABLE_MESSAGES, MastraStorage, StoreOperations, TABLE_SCHEMAS, WorkflowsStorage, ScoresStorage, safelyParseJSON, LegacyEvalsStorage, TracesStorage, MemoryStorage, resolveMessageLimit } from '@mastra/core/storage';
 import { MessageList } from '@mastra/core/agent';
 
 // src/storage/index.ts
@@ -11,7 +11,9 @@ var TABLE_ENGINES = {
   [TABLE_THREADS]: `ReplacingMergeTree()`,
   [TABLE_EVALS]: `MergeTree()`,
   [TABLE_SCORERS]: `MergeTree()`,
-  [TABLE_RESOURCES]: `ReplacingMergeTree()`
+  [TABLE_RESOURCES]: `ReplacingMergeTree()`,
+  // TODO: verify this is the correct engine for ai spans when implementing clickhouse storage
+  [TABLE_AI_SPANS]: `ReplacingMergeTree()`
 };
 var COLUMN_TYPES = {
   text: "String",
@@ -20,7 +22,8 @@ var COLUMN_TYPES = {
   jsonb: "String",
   integer: "Int64",
   float: "Float64",
-  bigint: "Int64"
+  bigint: "Int64",
+  boolean: "Bool"
 };
 function transformRow(row) {
   if (!row) {
@@ -2271,6 +2274,22 @@ var WorkflowsStorageClickhouse = class extends WorkflowsStorage {
     this.operations = operations;
     this.client = client;
   }
+  updateWorkflowResults({
+    // workflowName,
+    // runId,
+    // stepId,
+    // result,
+    // runtimeContext,
+  }) {
+    throw new Error("Method not implemented.");
+  }
+  updateWorkflowState({
+    // workflowName,
+    // runId,
+    // opts,
+  }) {
+    throw new Error("Method not implemented.");
+  }
   async persistWorkflowSnapshot({
     workflowName,
     runId,
@@ -2606,6 +2625,22 @@ var ClickhouseStore = class extends MastraStorage {
   async load({ tableName, keys }) {
     return this.stores.operations.load({ tableName, keys });
   }
+  async updateWorkflowResults({
+    workflowName,
+    runId,
+    stepId,
+    result,
+    runtimeContext
+  }) {
+    return this.stores.workflows.updateWorkflowResults({ workflowName, runId, stepId, result, runtimeContext });
+  }
+  async updateWorkflowState({
+    workflowName,
+    runId,
+    opts
+  }) {
+    return this.stores.workflows.updateWorkflowState({ workflowName, runId, opts });
+  }
   async persistWorkflowSnapshot({
     workflowName,
     runId,