@axiom-lattice/gateway 2.1.63 → 2.1.64

package/dist/index.mjs

@@ -1709,6 +1709,296 @@ async function executeSqlQuery(client, body, reply) {
   };
 }
 
+// src/controllers/workflow-tracking.ts
+import { getStoreLattice as getStoreLattice4, agentInstanceManager as agentInstanceManager4 } from "@axiom-lattice/core";
+function getTenantId6(request) {
+  const userTenantId = request.user?.tenantId;
+  if (userTenantId) return userTenantId;
+  return request.headers["x-tenant-id"] || "default";
+}
+function getTrackingStore() {
+  try {
+    const storeLattice = getStoreLattice4("default", "workflowTracking");
+    return storeLattice.store;
+  } catch {
+    return null;
+  }
+}
+async function getDefinitionsFromAssistants(tenantId) {
+  try {
+    const storeLattice = getStoreLattice4("default", "assistant");
+    const assistantStore = storeLattice.store;
+    const assistants = await assistantStore.getAllAssistants(tenantId);
+    const results = [];
+    for (const a of assistants) {
+      const def = a.graphDefinition;
+      if (!def || def.type !== "processing") continue;
+      if (!def.middleware) continue;
+      for (const mw of def.middleware) {
+        if (mw.type === "topology" && mw.enabled && mw.config?.edges?.length > 0) {
+          results.push({
+            assistantId: a.id,
+            assistantName: a.name,
+            topologyEdges: mw.config.edges,
+            totalEdges: mw.config.edges.length
+          });
+        }
+      }
+    }
+    return results;
+  } catch {
+    return [];
+  }
+}
+async function getAllWorkflowDefinitions(request, reply) {
+  const tenantId = getTenantId6(request);
+  try {
+    const configDefs = await getDefinitionsFromAssistants(tenantId);
+    if (configDefs.length === 0) {
+      return {
+        success: true,
+        message: "No workflow definitions found",
+        data: { records: [] }
+      };
+    }
+    const runMap = /* @__PURE__ */ new Map();
+    try {
+      const store = getTrackingStore();
+      if (store) {
+        const runs = await store.getWorkflowRunsByTenantId(tenantId);
+        for (const r of runs) {
+          const entry = runMap.get(r.assistantId) || { runCount: 0 };
+          entry.runCount++;
+          if (!entry.lastRunAt || new Date(r.startedAt) > new Date(entry.lastRunAt)) {
+            entry.lastRunAt = r.startedAt instanceof Date ? r.startedAt.toISOString() : String(r.startedAt);
+          }
+          runMap.set(r.assistantId, entry);
+        }
+      }
+    } catch {
+    }
+    const definitions = configDefs.map((d) => {
+      const runInfo = runMap.get(d.assistantId);
+      return {
+        ...d,
+        lastRunAt: runInfo?.lastRunAt || null,
+        runCount: runInfo?.runCount || 0
+      };
+    });
+    return {
+      success: true,
+      message: "Successfully retrieved workflow definitions",
+      data: { records: definitions }
+    };
+  } catch (error) {
+    request.log.error(error, "Failed to get workflow definitions");
+    return reply.status(500).send({ success: false, message: "Failed to retrieve workflow definitions" });
+  }
+}
+async function getAllWorkflowRuns(request, reply) {
+  const tenantId = getTenantId6(request);
+  const { assistantId, status } = request.query;
+  try {
+    const store = getTrackingStore();
+    if (!store) {
+      return reply.status(404).send({ success: false, message: "No workflow tracking store configured" });
+    }
+    let runs;
+    if (assistantId) {
+      runs = await store.getWorkflowRunsByAssistantId(tenantId, assistantId);
+    } else {
+      runs = await store.getWorkflowRunsByTenantId(tenantId);
+    }
+    if (status) {
+      runs = runs.filter((r) => r.status === status);
+    }
+    return {
+      success: true,
+      message: "Successfully retrieved workflow runs",
+      data: { records: runs, total: runs.length }
+    };
+  } catch (error) {
+    request.log.error(error, "Failed to get workflow runs");
+    return reply.status(500).send({ success: false, message: "Failed to retrieve workflow runs" });
+  }
+}
+async function getInboxItems(request, reply) {
+  const tenantId = getTenantId6(request);
+  try {
+    const store = getTrackingStore();
+    if (!store) {
+      return { success: true, message: "No tracking store configured", data: { records: [] } };
+    }
+    const nameMap = {};
+    try {
+      const asStoreLattice = getStoreLattice4("default", "assistant");
+      const assistantStore = asStoreLattice.store;
+      const assistants = await assistantStore.getAllAssistants(tenantId);
+      for (const a of assistants) {
+        nameMap[a.id] = a.name;
+      }
+    } catch {
+    }
+    const runs = await store.getWorkflowRunsByTenantId(tenantId);
+    const runningRuns = runs.filter((r) => r.status === "running");
+    if (runningRuns.length === 0) {
+      return { success: true, message: "No running workflows", data: { records: [] } };
+    }
+    const inboxItems = [];
+    for (const r of runningRuns) {
+      try {
+        const agent = agentInstanceManager4.getAgent({
+          assistant_id: r.assistantId,
+          thread_id: r.threadId,
+          tenant_id: r.tenantId
+        });
+        const runStatus = await agent.getRunStatus();
+        if (runStatus !== "interrupted") continue;
+        const state = await agent.getCurrentState();
+        const interrupts = state.tasks?.flatMap((t) => t.interrupts || []) || [];
+        for (const i of interrupts) {
+          inboxItems.push({
+            runId: r.id,
+            assistantId: r.assistantId,
+            assistantName: nameMap[r.assistantId] || r.assistantId,
+            threadId: r.threadId,
+            tenantId: r.tenantId,
+            interruptId: i.id,
+            interruptValue: i.value,
+            startedAt: r.startedAt,
+            totalEdges: r.totalEdges,
+            completedEdges: r.completedEdges
+          });
+        }
+      } catch {
+      }
+    }
+    inboxItems.sort(
+      (a, b) => new Date(b.startedAt).getTime() - new Date(a.startedAt).getTime()
+    );
+    return {
+      success: true,
+      message: "Successfully retrieved inbox items",
+      data: { records: inboxItems }
+    };
+  } catch (error) {
+    request.log.error(error, "Failed to get inbox items");
+    return reply.status(500).send({ success: false, message: "Failed to retrieve inbox items" });
+  }
+}
+async function getWorkflowDefinitions(request, reply) {
+  const { assistantId } = request.params;
+  const tenantId = getTenantId6(request);
+  try {
+    const store = getTrackingStore();
+    if (!store) {
+      return reply.status(404).send({ success: false, message: "No workflow tracking store configured" });
+    }
+    const runs = await store.getWorkflowRunsByAssistantId(tenantId, assistantId);
+    const seen = /* @__PURE__ */ new Set();
+    const definitions = runs.filter((r) => {
+      const key = JSON.stringify(r.topologyEdges);
+      if (seen.has(key)) return false;
+      seen.add(key);
+      return true;
+    }).map((r) => ({
+      assistantId: r.assistantId,
+      topologyEdges: r.topologyEdges,
+      totalEdges: r.totalEdges,
+      lastRunAt: r.startedAt
+    }));
+    return {
+      success: true,
+      message: "Successfully retrieved workflow definitions",
+      data: { records: definitions }
+    };
+  } catch (error) {
+    request.log.error(error, "Failed to get workflow definitions");
+    return reply.status(500).send({ success: false, message: "Failed to retrieve workflow definitions" });
+  }
+}
+async function getWorkflowRuns(request, reply) {
+  const { threadId } = request.params;
+  const tenantId = getTenantId6(request);
+  try {
+    const store = getTrackingStore();
+    if (!store) {
+      return reply.status(404).send({ success: false, message: "No workflow tracking store configured" });
+    }
+    const runs = await store.getWorkflowRunsByThreadId(tenantId, threadId);
+    return {
+      success: true,
+      message: "Successfully retrieved workflow runs",
+      data: { records: runs, total: runs.length }
+    };
+  } catch (error) {
+    request.log.error(error, "Failed to get workflow runs");
+    return reply.status(500).send({ success: false, message: "Failed to retrieve workflow runs" });
+  }
+}
+async function getWorkflowRun(request, reply) {
+  const { runId } = request.params;
+  try {
+    const store = getTrackingStore();
+    if (!store) {
+      return reply.status(404).send({ success: false, message: "No workflow tracking store configured" });
+    }
+    const run = await store.getWorkflowRun(runId);
+    if (!run) {
+      return reply.status(404).send({ success: false, message: "Workflow run not found" });
+    }
+    return { success: true, message: "Successfully retrieved workflow run", data: run };
+  } catch (error) {
+    request.log.error(error, "Failed to get workflow run");
+    return reply.status(500).send({ success: false, message: "Failed to retrieve workflow run" });
+  }
+}
+async function getRunSteps(request, reply) {
+  const { runId } = request.params;
+  const { step_type, status: stepStatus } = request.query;
+  try {
+    const store = getTrackingStore();
+    if (!store) {
+      return reply.status(404).send({ success: false, message: "No workflow tracking store configured" });
+    }
+    let steps;
+    if (step_type) {
+      steps = await store.getRunStepsByType(runId, step_type);
+    } else {
+      steps = await store.getRunSteps(runId);
+    }
+    if (stepStatus) {
+      steps = steps.filter((s) => s.status === stepStatus);
+    }
+    return {
+      success: true,
+      message: "Successfully retrieved run steps",
+      data: { records: steps, total: steps.length }
+    };
+  } catch (error) {
+    request.log.error(error, "Failed to get run steps");
+    return reply.status(500).send({ success: false, message: "Failed to retrieve run steps" });
+  }
+}
+async function getRunTasks(request, reply) {
+  const { runId } = request.params;
+  try {
+    const store = getTrackingStore();
+    if (!store) {
+      return reply.status(404).send({ success: false, message: "No workflow tracking store configured" });
+    }
+    const steps = await store.getInterruptedSteps(runId);
+    return {
+      success: true,
+      message: "Successfully retrieved user tasks",
+      data: { records: steps, total: steps.length }
+    };
+  } catch (error) {
+    request.log.error(error, "Failed to get run tasks");
+    return reply.status(500).send({ success: false, message: "Failed to retrieve user tasks" });
+  }
+}
+
 // src/schemas/data-query.ts
 var dataQuerySchema = {
   description: "Execute data query (semantic or SQL)",
@@ -2050,7 +2340,7 @@ var getHealthSchema = {
 };
 
 // src/controllers/thread_status.ts
-import { agentInstanceManager as agentInstanceManager4 } from "@axiom-lattice/core";
+import { agentInstanceManager as agentInstanceManager5 } from "@axiom-lattice/core";
 async function removePendingMessageHandler(request, reply) {
   try {
     const { assistant_id, thread_id, message_id } = request.params;
@@ -2063,7 +2353,7 @@ async function removePendingMessageHandler(request, reply) {
     if (!assistant_id) {
       return reply.code(400).send({ error: "Missing assistant_id parameter" });
     }
-    const agent = agentInstanceManager4.getAgent({
+    const agent = agentInstanceManager5.getAgent({
       assistant_id,
       thread_id,
       tenant_id,
@@ -2393,14 +2683,14 @@ function registerSandboxProxyRoutes(app2) {
 // src/controllers/workspace.ts
 import * as fs from "fs/promises";
 import * as path from "path";
-import { getStoreLattice as getStoreLattice4 } from "@axiom-lattice/core";
+import { getStoreLattice as getStoreLattice5 } from "@axiom-lattice/core";
 import { SandboxFilesystem, FilesystemBackend } from "@axiom-lattice/core";
 import { getSandBoxManager as getSandBoxManager3 } from "@axiom-lattice/core";
 import { v4 as uuidv4 } from "uuid";
 var WorkspaceController = class {
   constructor() {
-    this.workspaceStore = getStoreLattice4("default", "workspace").store;
-    this.projectStore = getStoreLattice4("default", "project").store;
+    this.workspaceStore = getStoreLattice5("default", "workspace").store;
+    this.projectStore = getStoreLattice5("default", "project").store;
   }
   getTenantId(request) {
     const userTenantId = request.user?.tenantId;
@@ -2925,11 +3215,11 @@ function registerWorkspaceRoutes(app2) {
 
 // src/controllers/database-configs.ts
 import {
-  getStoreLattice as getStoreLattice5,
+  getStoreLattice as getStoreLattice6,
   sqlDatabaseManager
 } from "@axiom-lattice/core";
 import { randomUUID as randomUUID3 } from "crypto";
-function getTenantId6(request) {
+function getTenantId7(request) {
   const userTenantId = request.user?.tenantId;
   if (userTenantId) {
     return userTenantId;
@@ -2937,9 +3227,9 @@ function getTenantId6(request) {
   return request.headers["x-tenant-id"] || "default";
 }
 async function getDatabaseConfigList(request, reply) {
-  const tenantId = getTenantId6(request);
+  const tenantId = getTenantId7(request);
   try {
-    const storeLattice = getStoreLattice5("default", "database");
+    const storeLattice = getStoreLattice6("default", "database");
     const store = storeLattice.store;
     const configs = await store.getAllConfigs(tenantId);
     console.log("Backend: getAllConfigs returned:", configs);
@@ -2967,10 +3257,10 @@ async function getDatabaseConfigList(request, reply) {
   }
 }
 async function getDatabaseConfig(request, reply) {
-  const tenantId = getTenantId6(request);
+  const tenantId = getTenantId7(request);
   const { key } = request.params;
   try {
-    const storeLattice = getStoreLattice5("default", "database");
+    const storeLattice = getStoreLattice6("default", "database");
     const store = storeLattice.store;
     const config = await store.getConfigByKey(tenantId, key);
     if (!config) {
@@ -2993,10 +3283,10 @@ async function getDatabaseConfig(request, reply) {
   }
 }
 async function createDatabaseConfig(request, reply) {
-  const tenantId = getTenantId6(request);
+  const tenantId = getTenantId7(request);
   const body = request.body;
   try {
-    const storeLattice = getStoreLattice5("default", "database");
+    const storeLattice = getStoreLattice6("default", "database");
     const store = storeLattice.store;
     const existing = await store.getConfigByKey(tenantId, body.key);
     if (existing) {
@@ -3028,11 +3318,11 @@ async function createDatabaseConfig(request, reply) {
   }
 }
 async function updateDatabaseConfig(request, reply) {
-  const tenantId = getTenantId6(request);
+  const tenantId = getTenantId7(request);
   const { key } = request.params;
   const updates = request.body;
   try {
-    const storeLattice = getStoreLattice5("default", "database");
+    const storeLattice = getStoreLattice6("default", "database");
     const store = storeLattice.store;
     const existing = await store.getConfigByKey(tenantId, key);
     if (!existing) {
@@ -3070,10 +3360,10 @@ async function updateDatabaseConfig(request, reply) {
   }
 }
 async function deleteDatabaseConfig(request, reply) {
-  const tenantId = getTenantId6(request);
+  const tenantId = getTenantId7(request);
   const { keyOrId } = request.params;
   try {
-    const storeLattice = getStoreLattice5("default", "database");
+    const storeLattice = getStoreLattice6("default", "database");
     const store = storeLattice.store;
     console.log("Delete request - keyOrId:", keyOrId);
     let config = await store.getConfigByKey(tenantId, keyOrId);
@@ -3119,10 +3409,10 @@ async function deleteDatabaseConfig(request, reply) {
   }
 }
 async function testDatabaseConnection(request, reply) {
-  const tenantId = getTenantId6(request);
+  const tenantId = getTenantId7(request);
   const { key } = request.params;
   try {
-    const storeLattice = getStoreLattice5("default", "database");
+    const storeLattice = getStoreLattice6("default", "database");
     const store = storeLattice.store;
     const config = await store.getConfigByKey(tenantId, key);
     if (!config) {
@@ -3207,12 +3497,12 @@ function registerDatabaseConfigRoutes(app2) {
 
 // src/controllers/metrics-configs.ts
 import {
-  getStoreLattice as getStoreLattice6,
+  getStoreLattice as getStoreLattice7,
   metricsServerManager as metricsServerManager2,
   SemanticMetricsClient as SemanticMetricsClient2
 } from "@axiom-lattice/core";
 import { randomUUID as randomUUID4 } from "crypto";
-function getTenantId7(request) {
+function getTenantId8(request) {
   const userTenantId = request.user?.tenantId;
   if (userTenantId) {
     return userTenantId;
@@ -3220,9 +3510,9 @@ function getTenantId7(request) {
   return request.headers["x-tenant-id"] || "default";
 }
 async function getMetricsServerConfigList(request, reply) {
-  const tenantId = getTenantId7(request);
+  const tenantId = getTenantId8(request);
   try {
-    const storeLattice = getStoreLattice6("default", "metrics");
+    const storeLattice = getStoreLattice7("default", "metrics");
     const store = storeLattice.store;
     const configs = await store.getAllConfigs(tenantId);
     return {
@@ -3246,10 +3536,10 @@ async function getMetricsServerConfigList(request, reply) {
   }
 }
 async function getMetricsServerConfig(request, reply) {
-  const tenantId = getTenantId7(request);
+  const tenantId = getTenantId8(request);
   const { key } = request.params;
   try {
-    const storeLattice = getStoreLattice6("default", "metrics");
+    const storeLattice = getStoreLattice7("default", "metrics");
     const store = storeLattice.store;
     const config = await store.getConfigByKey(tenantId, key);
     if (!config) {
@@ -3272,10 +3562,10 @@ async function getMetricsServerConfig(request, reply) {
   }
 }
 async function createMetricsServerConfig(request, reply) {
-  const tenantId = getTenantId7(request);
+  const tenantId = getTenantId8(request);
   const body = request.body;
   try {
-    const storeLattice = getStoreLattice6("default", "metrics");
+    const storeLattice = getStoreLattice7("default", "metrics");
     const store = storeLattice.store;
     const existing = await store.getConfigByKey(tenantId, body.key);
     if (existing) {
@@ -3323,11 +3613,11 @@ async function createMetricsServerConfig(request, reply) {
   }
 }
 async function updateMetricsServerConfig(request, reply) {
-  const tenantId = getTenantId7(request);
+  const tenantId = getTenantId8(request);
   const { key } = request.params;
   const updates = request.body;
   try {
-    const storeLattice = getStoreLattice6("default", "metrics");
+    const storeLattice = getStoreLattice7("default", "metrics");
     const store = storeLattice.store;
     const existing = await store.getConfigByKey(tenantId, key);
     if (!existing) {
@@ -3374,10 +3664,10 @@ async function updateMetricsServerConfig(request, reply) {
   }
 }
 async function deleteMetricsServerConfig(request, reply) {
-  const tenantId = getTenantId7(request);
+  const tenantId = getTenantId8(request);
   const { keyOrId } = request.params;
   try {
-    const storeLattice = getStoreLattice6("default", "metrics");
+    const storeLattice = getStoreLattice7("default", "metrics");
     const store = storeLattice.store;
     let config = await store.getConfigByKey(tenantId, keyOrId);
     let configKey = keyOrId;
@@ -3421,10 +3711,10 @@ async function deleteMetricsServerConfig(request, reply) {
   }
 }
 async function testMetricsServerConnection(request, reply) {
-  const tenantId = getTenantId7(request);
+  const tenantId = getTenantId8(request);
   const { key } = request.params;
   try {
-    const storeLattice = getStoreLattice6("default", "metrics");
+    const storeLattice = getStoreLattice7("default", "metrics");
     const store = storeLattice.store;
     const config = await store.getConfigByKey(tenantId, key);
     if (!config) {
@@ -3472,10 +3762,10 @@ async function testMetricsServerConnection(request, reply) {
   }
 }
 async function listAvailableMetrics(request, reply) {
-  const tenantId = getTenantId7(request);
+  const tenantId = getTenantId8(request);
   const { key } = request.params;
   try {
-    const storeLattice = getStoreLattice6("default", "metrics");
+    const storeLattice = getStoreLattice7("default", "metrics");
     const store = storeLattice.store;
     const config = await store.getConfigByKey(tenantId, key);
     if (!config) {
@@ -3510,11 +3800,11 @@ async function listAvailableMetrics(request, reply) {
   }
 }
 async function queryMetricsData(request, reply) {
-  const tenantId = getTenantId7(request);
+  const tenantId = getTenantId8(request);
   const { key } = request.params;
   const { metricName, startTime, endTime, step, labels } = request.body;
   try {
-    const storeLattice = getStoreLattice6("default", "metrics");
+    const storeLattice = getStoreLattice7("default", "metrics");
     const store = storeLattice.store;
     const config = await store.getConfigByKey(tenantId, key);
     if (!config) {
@@ -3558,10 +3848,10 @@ async function queryMetricsData(request, reply) {
   }
 }
 async function getDataSources(request, reply) {
-  const tenantId = getTenantId7(request);
+  const tenantId = getTenantId8(request);
   const { key } = request.params;
   try {
-    const storeLattice = getStoreLattice6("default", "metrics");
+    const storeLattice = getStoreLattice7("default", "metrics");
     const store = storeLattice.store;
     const config = await store.getConfigByKey(tenantId, key);
     if (!config) {
@@ -3599,10 +3889,10 @@ async function getDataSources(request, reply) {
   }
 }
 async function getDatasourceMetrics(request, reply) {
-  const tenantId = getTenantId7(request);
+  const tenantId = getTenantId8(request);
   const { key, datasourceId } = request.params;
   try {
-    const storeLattice = getStoreLattice6("default", "metrics");
+    const storeLattice = getStoreLattice7("default", "metrics");
     const store = storeLattice.store;
     const config = await store.getConfigByKey(tenantId, key);
     if (!config) {
@@ -3636,11 +3926,11 @@ async function getDatasourceMetrics(request, reply) {
   }
 }
 async function querySemanticMetrics(request, reply) {
-  const tenantId = getTenantId7(request);
+  const tenantId = getTenantId8(request);
   const { key } = request.params;
   const body = request.body;
   try {
-    const storeLattice = getStoreLattice6("default", "metrics");
+    const storeLattice = getStoreLattice7("default", "metrics");
     const store = storeLattice.store;
     const config = await store.getConfigByKey(tenantId, key);
     if (!config) {
@@ -3794,12 +4084,12 @@ function registerMetricsServerConfigRoutes(app2) {
 
 // src/controllers/mcp-configs.ts
 import {
-  getStoreLattice as getStoreLattice7,
+  getStoreLattice as getStoreLattice8,
   mcpManager,
   toolLatticeManager as toolLatticeManager2
 } from "@axiom-lattice/core";
 import { randomUUID as randomUUID5 } from "crypto";
-function getTenantId8(request) {
+function getTenantId9(request) {
   const userTenantId = request.user?.tenantId;
   if (userTenantId) {
     return userTenantId;
@@ -3807,9 +4097,9 @@ function getTenantId8(request) {
   return request.headers["x-tenant-id"] || "default";
 }
 async function getMcpServerConfigList(request, reply) {
-  const tenantId = getTenantId8(request);
+  const tenantId = getTenantId9(request);
   try {
-    const storeLattice = getStoreLattice7("default", "mcp");
+    const storeLattice = getStoreLattice8("default", "mcp");
     const store = storeLattice.store;
     const configs = await store.getAllConfigs(tenantId);
     return {
@@ -3833,10 +4123,10 @@ async function getMcpServerConfigList(request, reply) {
   }
 }
 async function getMcpServerConfig(request, reply) {
-  const tenantId = getTenantId8(request);
+  const tenantId = getTenantId9(request);
   const { key } = request.params;
   try {
-    const storeLattice = getStoreLattice7("default", "mcp");
+    const storeLattice = getStoreLattice8("default", "mcp");
     const store = storeLattice.store;
     const config = await store.getConfigByKey(tenantId, key);
     if (!config) {
@@ -3859,10 +4149,10 @@ async function getMcpServerConfig(request, reply) {
   }
 }
 async function createMcpServerConfig(request, reply) {
-  const tenantId = getTenantId8(request);
+  const tenantId = getTenantId9(request);
   const body = request.body;
   try {
-    const storeLattice = getStoreLattice7("default", "mcp");
+    const storeLattice = getStoreLattice8("default", "mcp");
     const store = storeLattice.store;
     const existing = await store.getConfigByKey(tenantId, body.key);
     if (existing) {
@@ -3898,11 +4188,11 @@ async function createMcpServerConfig(request, reply) {
   }
 }
 async function updateMcpServerConfig(request, reply) {
-  const tenantId = getTenantId8(request);
+  const tenantId = getTenantId9(request);
   const { key } = request.params;
   const updates = request.body;
   try {
-    const storeLattice = getStoreLattice7("default", "mcp");
+    const storeLattice = getStoreLattice8("default", "mcp");
     const store = storeLattice.store;
     const existing = await store.getConfigByKey(tenantId, key);
     if (!existing) {
@@ -3948,10 +4238,10 @@ async function updateMcpServerConfig(request, reply) {
   }
 }
 async function deleteMcpServerConfig(request, reply) {
-  const tenantId = getTenantId8(request);
+  const tenantId = getTenantId9(request);
   const { keyOrId } = request.params;
   try {
-    const storeLattice = getStoreLattice7("default", "mcp");
+    const storeLattice = getStoreLattice8("default", "mcp");
     const store = storeLattice.store;
     let config = await store.getConfigByKey(tenantId, keyOrId);
     let configKey = keyOrId;
@@ -3995,10 +4285,10 @@ async function deleteMcpServerConfig(request, reply) {
   }
 }
 async function testMcpServerConnection(request, reply) {
-  const tenantId = getTenantId8(request);
+  const tenantId = getTenantId9(request);
   const { key } = request.params;
   try {
-    const storeLattice = getStoreLattice7("default", "mcp");
+    const storeLattice = getStoreLattice8("default", "mcp");
     const store = storeLattice.store;
     const config = await store.getConfigByKey(tenantId, key);
     if (!config) {
@@ -4048,10 +4338,10 @@ async function testMcpServerConnection(request, reply) {
   }
 }
 async function listMcpServerTools(request, reply) {
-  const tenantId = getTenantId8(request);
+  const tenantId = getTenantId9(request);
   const { key } = request.params;
   try {
-    const storeLattice = getStoreLattice7("default", "mcp");
+    const storeLattice = getStoreLattice8("default", "mcp");
     const store = storeLattice.store;
     const config = await store.getConfigByKey(tenantId, key);
     if (!config) {
@@ -4081,10 +4371,10 @@ async function listMcpServerTools(request, reply) {
   }
 }
 async function connectMcpServer(request, reply) {
-  const tenantId = getTenantId8(request);
+  const tenantId = getTenantId9(request);
   const { key } = request.params;
   try {
-    const storeLattice = getStoreLattice7("default", "mcp");
+    const storeLattice = getStoreLattice8("default", "mcp");
     const store = storeLattice.store;
     const config = await store.getConfigByKey(tenantId, key);
     if (!config) {
@@ -4105,7 +4395,7 @@ async function connectMcpServer(request, reply) {
     };
   } catch (error) {
     console.error("Failed to connect MCP server:", error);
-    const storeLattice = getStoreLattice7("default", "mcp");
+    const storeLattice = getStoreLattice8("default", "mcp");
     const store = storeLattice.store;
     const config = await store.getConfigByKey(tenantId, key);
     if (config) {
@@ -4118,10 +4408,10 @@ async function connectMcpServer(request, reply) {
   }
 }
 async function disconnectMcpServer(request, reply) {
-  const tenantId = getTenantId8(request);
+  const tenantId = getTenantId9(request);
   const { key } = request.params;
   try {
-    const storeLattice = getStoreLattice7("default", "mcp");
+    const storeLattice = getStoreLattice8("default", "mcp");
     const store = storeLattice.store;
     const config = await store.getConfigByKey(tenantId, key);
     if (!config) {
@@ -4226,11 +4516,11 @@ function registerMcpServerConfigRoutes(app2) {
 }
 
 // src/controllers/users.ts
-import { getStoreLattice as getStoreLattice8 } from "@axiom-lattice/core";
+import { getStoreLattice as getStoreLattice9 } from "@axiom-lattice/core";
 import { v4 as uuidv42 } from "uuid";
 var UsersController = class {
   constructor() {
-    this.userStore = getStoreLattice8("default", "user").store;
+    this.userStore = getStoreLattice9("default", "user").store;
   }
   async listUsers(request, reply) {
     const { email } = request.query;
@@ -4308,11 +4598,11 @@ function registerUserRoutes(app2) {
 }
 
 // src/controllers/tenants.ts
-import { getStoreLattice as getStoreLattice9 } from "@axiom-lattice/core";
+import { getStoreLattice as getStoreLattice10 } from "@axiom-lattice/core";
 import { v4 as uuidv43 } from "uuid";
 var TenantsController = class {
   constructor() {
-    this.tenantStore = getStoreLattice9("default", "tenant").store;
+    this.tenantStore = getStoreLattice10("default", "tenant").store;
   }
   // ==================== Tenant CRUD ====================
   async listTenants(request, reply) {
@@ -4376,7 +4666,7 @@ function registerTenantRoutes(app2) {
 }
 
 // src/controllers/auth.ts
-import { getStoreLattice as getStoreLattice10 } from "@axiom-lattice/core";
+import { getStoreLattice as getStoreLattice11 } from "@axiom-lattice/core";
 import { v4 as uuidv44 } from "uuid";
 var defaultAuthConfig = {
   autoApproveUsers: true,
@@ -4385,9 +4675,9 @@ var defaultAuthConfig = {
 };
 var AuthController = class {
   constructor(config = {}) {
-    this.userStore = getStoreLattice10("default", "user").store;
-    this.tenantStore = getStoreLattice10("default", "tenant").store;
-    this.userTenantLinkStore = getStoreLattice10("default", "userTenantLink").store;
+    this.userStore = getStoreLattice11("default", "user").store;
+    this.tenantStore = getStoreLattice11("default", "tenant").store;
+    this.userTenantLinkStore = getStoreLattice11("default", "userTenantLink").store;
     this.config = { ...defaultAuthConfig, ...config };
   }
   async register(request, reply) {
@@ -4759,7 +5049,7 @@ function registerAuthRoutes(app2, config) {
 }
 
 // src/channels/lark/routes.ts
-import { getStoreLattice as getStoreLattice11 } from "@axiom-lattice/core";
+import { getStoreLattice as getStoreLattice12 } from "@axiom-lattice/core";
 import {
   ChannelIdentityMappingStore,
   PostgreSQLChannelInstallationStore
@@ -5010,7 +5300,7 @@ function resolveSubjectType(mappingMode, chatType) {
 }
 
 // src/channels/lark/runner.ts
-import { agentInstanceManager as agentInstanceManager5 } from "@axiom-lattice/core";
+import { agentInstanceManager as agentInstanceManager6 } from "@axiom-lattice/core";
 import { MessageChunkTypes as MessageChunkTypes3 } from "@axiom-lattice/protocols";
 
 // src/channels/lark/aggregator.ts
@@ -5023,7 +5313,7 @@ function aggregateLarkReply(messageId, chunks) {
 
 // src/channels/lark/runner.ts
 async function runAgentAndCollectLarkReply(input) {
-  const agent = agentInstanceManager5.getAgent({
+  const agent = agentInstanceManager6.getAgent({
     tenant_id: input.tenantId,
     assistant_id: input.assistantId,
     thread_id: input.threadId,
@@ -5135,7 +5425,7 @@ function createDefaultLarkDependencies() {
   const installationStore = new PostgreSQLChannelInstallationStore({
     poolConfig: getDatabaseUrl()
   });
-  const threadStore = getStoreLattice11("default", "thread").store;
+  const threadStore = getStoreLattice12("default", "thread").store;
   const mappingStore = new ChannelIdentityMappingStore({
     poolConfig: getDatabaseUrl()
   });
@@ -5216,7 +5506,7 @@ function registerChannelRoutes(app2, dependencies = {}) {
 
 // src/controllers/channel-installations.ts
 import { randomUUID as randomUUID7 } from "crypto";
-function getTenantId9(request) {
+function getTenantId10(request) {
   const userTenantId = request.user?.tenantId;
   if (userTenantId) {
     return userTenantId;
@@ -5234,7 +5524,7 @@ async function getInstallationStore() {
   });
 }
 async function getChannelInstallationList(request, reply) {
-  const tenantId = getTenantId9(request);
+  const tenantId = getTenantId10(request);
   const { channel } = request.query;
   try {
     const store = await getInstallationStore();
@@ -5260,7 +5550,7 @@ async function getChannelInstallationList(request, reply) {
   }
 }
 async function getChannelInstallation(request, reply) {
-  const tenantId = getTenantId9(request);
+  const tenantId = getTenantId10(request);
   const { installationId } = request.params;
   try {
     const store = await getInstallationStore();
@@ -5293,7 +5583,7 @@ async function getChannelInstallation(request, reply) {
   }
 }
 async function createChannelInstallation(request, reply) {
-  const tenantId = getTenantId9(request);
+  const tenantId = getTenantId10(request);
   const body = request.body;
   try {
     if (!body.channel) {
@@ -5356,7 +5646,7 @@ async function createChannelInstallation(request, reply) {
   }
 }
 async function updateChannelInstallation(request, reply) {
-  const tenantId = getTenantId9(request);
+  const tenantId = getTenantId10(request);
   const { installationId } = request.params;
   const body = request.body;
   try {
@@ -5402,7 +5692,7 @@ async function updateChannelInstallation(request, reply) {
   }
 }
 async function deleteChannelInstallation(request, reply) {
-  const tenantId = getTenantId9(request);
+  const tenantId = getTenantId10(request);
   const { installationId } = request.params;
   try {
     const store = await getInstallationStore();
@@ -5588,6 +5878,29 @@ var registerLatticeRoutes = (app2) => {
   });
   registerChannelRoutes(app2);
   registerChannelInstallationRoutes(app2);
+  app2.get(
+    "/api/workflows/definitions",
+    getAllWorkflowDefinitions
+  );
+  app2.get(
+    "/api/workflows/runs",
+    getAllWorkflowRuns
+  );
+  app2.get(
+    "/api/workflows/inbox",
+    getInboxItems
+  );
+  app2.get(
+    "/api/assistants/:assistantId/workflows/definitions",
+    getWorkflowDefinitions
+  );
+  app2.get(
+    "/api/assistants/:assistantId/threads/:threadId/workflows/runs",
+    getWorkflowRuns
+  );
+  app2.get("/api/workflows/runs/:runId", getWorkflowRun);
+  app2.get("/api/workflows/runs/:runId/steps", getRunSteps);
+  app2.get("/api/workflows/runs/:runId/tasks", getRunTasks);
   app2.delete(
     "/api/assistants/:assistant_id/threads/:thread_id/pending-messages/:message_id",
     removePendingMessageHandler
@@ -5657,7 +5970,7 @@ var configureSwagger = async (app2, customSwaggerConfig, customSwaggerUiConfig)
 };
 
 // src/services/agent_task_consumer.ts
-import { eventBus as eventBus2, AGENT_TASK_EVENT, agentInstanceManager as agentInstanceManager6, QueueMode as QueueMode2 } from "@axiom-lattice/core";
+import { eventBus as eventBus2, AGENT_TASK_EVENT, agentInstanceManager as agentInstanceManager7, QueueMode as QueueMode2 } from "@axiom-lattice/core";
 var handleAgentTask = async (taskRequest, retryCount = 0) => {
   const {
     assistant_id,
@@ -5666,13 +5979,16 @@ var handleAgentTask = async (taskRequest, retryCount = 0) => {
     "x-tenant-id": tenant_id,
     command,
     callback_event,
-    runConfig
+    runConfig,
+    main_thread_id,
+    main_tenant_id,
+    main_assistant_id
   } = taskRequest;
   try {
     console.log(
       `\u5F00\u59CB\u5904\u7406\u4EFB\u52A1 [assistant_id: ${assistant_id}, thread_id: ${thread_id}]`
     );
-    const agent = agentInstanceManager6.getAgent({ assistant_id, thread_id, tenant_id, workspace_id: runConfig?.workspaceId, project_id: runConfig?.projectId, custom_run_config: runConfig });
+    const agent = agentInstanceManager7.getAgent({ assistant_id, thread_id, tenant_id, workspace_id: runConfig?.workspaceId, project_id: runConfig?.projectId, custom_run_config: runConfig });
     await agent.addMessage({ input, command, custom_run_config: runConfig }, QueueMode2.STEER);
     if (callback_event) {
       agent.subscribeOnce("message:completed", (evt) => {
@@ -5681,6 +5997,34 @@ var handleAgentTask = async (taskRequest, retryCount = 0) => {
           state: evt.state,
           config: { assistant_id, thread_id, tenant_id }
         });
+        if (main_thread_id && main_tenant_id) {
+          try {
+            const mainAgent = agentInstanceManager7.getAgent({
+              assistant_id: main_assistant_id ?? assistant_id,
+              thread_id: main_thread_id,
+              tenant_id: main_tenant_id
+            });
+            if (mainAgent) {
+              const messages = evt.state?.values?.messages;
+              const lastAIMessage = messages?.filter((m) => m.type === "ai" || m.getType?.() === "ai").pop();
+              const summary = lastAIMessage?.content?.substring(0, 500) ?? "(no output)";
+              mainAgent.addMessage(
+                {
+                  input: {
+                    message: `[Async task completed]
+task_id: ${thread_id}
+${summary}`
+                  }
+                }
+              ).catch((err) => {
+                console.error("Failed to notify main thread:", err);
+              });
+              mainAgent.updateAsyncTaskStatus(thread_id, "completed");
+            }
+          } catch (err) {
+            console.error("Failed to notify main thread:", err);
+          }
+        }
       });
       agent.subscribeOnce("message:interrupted", (evt) => {
         eventBus2.publish(callback_event, {
@@ -5890,8 +6234,9 @@ import {
   loggerLatticeManager,
   sandboxLatticeManager as sandboxLatticeManager2,
   sqlDatabaseManager as sqlDatabaseManager2,
-  getStoreLattice as getStoreLattice12,
-  agentInstanceManager as agentInstanceManager7,
+  getStoreLattice as getStoreLattice13,
+  storeLatticeManager,
+  agentInstanceManager as agentInstanceManager8,
   createSandboxProvider
 } from "@axiom-lattice/core";
 import {
@@ -6039,7 +6384,7 @@ var start = async (config) => {
     app.decorate("loggerLattice", loggerLattice);
     registerLatticeRoutes(app);
     try {
-      const storeLattice = getStoreLattice12("default", "database");
+      const storeLattice = getStoreLattice13("default", "database");
       const store = storeLattice.store;
       sqlDatabaseManager2.setConfigStore(store);
       logger.info("Database config store set for SqlDatabaseManager");
@@ -6050,6 +6395,21 @@ var start = async (config) => {
       sandboxLatticeManager2.registerLattice("default", getConfiguredSandboxProvider());
       logger.info("Registered sandbox manager from env configuration");
     }
+    if (process.env.DATABASE_URL) {
+      try {
+        const { PostgreSQLWorkflowTrackingStore } = await import("@axiom-lattice/pg-stores");
+        const pgStore = new PostgreSQLWorkflowTrackingStore({
+          poolConfig: process.env.DATABASE_URL
+        });
+        if (storeLatticeManager.hasLattice("default", "workflowTracking")) {
+          storeLatticeManager.removeLattice("default", "workflowTracking");
+        }
+        storeLatticeManager.registerLattice("default", "workflowTracking", pgStore);
+        logger.info("Workflow tracking store switched to PostgreSQL");
+      } catch (error) {
+        logger.warn("Failed to switch workflow tracking to PostgreSQL, keeping in-memory: " + (error instanceof Error ? error.message : String(error)));
+      }
+    }
     const target_port = config?.port || Number(process.env.PORT) || 4001;
     await app.listen({ port: target_port, host: "0.0.0.0" });
     logger.info(`Lattice Gateway is running on port: ${target_port}`);
@@ -6068,7 +6428,7 @@ var start = async (config) => {
     }
     try {
       logger.info("Starting agent instance recovery...");
-      const restoreStats = await agentInstanceManager7.restore();
+      const restoreStats = await agentInstanceManager8.restore();
       logger.info(`Agent recovery complete: ${restoreStats.restored} threads restored, ${restoreStats.errors} errors`);
     } catch (error) {
       logger.error("Agent recovery failed", { error });