@axiom-lattice/gateway 2.1.76 → 2.1.78

package/dist/index.mjs

@@ -1717,7 +1717,7 @@ async function executeSqlQuery(client, body, reply) {
 }
 
 // src/controllers/workflow-tracking.ts
-import { getStoreLattice as getStoreLattice4, agentInstanceManager as agentInstanceManager4 } from "@axiom-lattice/core";
+import { getStoreLattice as getStoreLattice4, agentInstanceManager as agentInstanceManager4, ThreadStatus as ThreadStatus2 } from "@axiom-lattice/core";
 import { MessageChunkTypes as MessageChunkTypes2 } from "@axiom-lattice/protocols";
 function getTenantId6(request) {
   const userTenantId = request.user?.tenantId;
@@ -1963,6 +1963,41 @@ async function getWorkflowRun(request, reply) {
     return reply.status(500).send({ success: false, message: "Failed to retrieve workflow run" });
   }
 }
+async function deleteWorkflowRun(request, reply) {
+  const { runId } = 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 run = await store.getWorkflowRun(runId);
+    if (!run) {
+      return reply.status(404).send({ success: false, message: "Workflow run not found" });
+    }
+    if (run.status === "running") {
+      try {
+        const workspace_id = request.headers["x-workspace-id"];
+        const project_id = request.headers["x-project-id"];
+        const agent = agentInstanceManager4.getAgent({
+          assistant_id: run.assistantId,
+          thread_id: run.threadId,
+          tenant_id: tenantId,
+          workspace_id,
+          project_id
+        });
+        await agent.abort();
+      } catch (err) {
+        request.log.warn({ runId, error: err.message }, "Failed to abort agent, deleting tracking records anyway");
+      }
+    }
+    await store.deleteWorkflowRun(runId);
+    return { success: true, message: "Workflow run deleted" };
+  } catch (error) {
+    request.log.error(error, "Failed to delete workflow run");
+    return reply.status(500).send({ success: false, message: "Failed to delete workflow run" });
+  }
+}
 async function getRunSteps(request, reply) {
   const { runId } = request.params;
   const { step_type, status: stepStatus } = request.query;
@@ -2035,6 +2070,14 @@ async function replyInboxTask(request, reply) {
       workspace_id,
       project_id
     });
+    const runStatus = await agent.getRunStatus();
+    if (runStatus !== ThreadStatus2.INTERRUPTED) {
+      return reply.status(409).send({
+        success: false,
+        message: `Cannot resume: graph is not interrupted (current status: ${runStatus})`,
+        data: { runId: run.id, assistantId: run.assistantId, threadId: run.threadId, status: runStatus }
+      });
+    }
     agent.addMessage({
       input: { message: "Clarification answers submitted" },
       command: {
@@ -6524,7 +6567,7 @@ var registerLatticeRoutes = (app2, channelDeps) => {
         const inboundMessage = {
           channel: msg.channel,
           channelInstallationId: msg.channelInstallationId || "",
-          tenantId: msg.tenantId || "default",
+          tenantId: msg.tenantId,
           sender: {
             id: msg.sender.id,
             displayName: msg.sender.displayName
@@ -6565,6 +6608,7 @@ var registerLatticeRoutes = (app2, channelDeps) => {
     getWorkflowRuns
   );
   app2.get("/api/workflows/runs/:runId", getWorkflowRun);
+  app2.delete("/api/workflows/runs/:runId", deleteWorkflowRun);
   app2.get("/api/workflows/runs/:runId/steps", getRunSteps);
   app2.get("/api/workflows/runs/:runId/tasks", getRunTasks);
   app2.post("/api/workflows/inbox/reply", replyInboxTask);
@@ -6603,11 +6647,17 @@ var MessageRouter = class {
     };
     try {
       await this.runMiddlewares(ctx, async () => {
+        const tenantId = message.tenantId || (await this.installationStore.getInstallationById(message.channelInstallationId))?.tenantId;
+        if (!tenantId) {
+          throw new Error(
+            "tenantId is required: provide it in the message or ensure the channelInstallation has a tenantId"
+          );
+        }
         let binding = await this.bindingRegistry.resolve({
           channel: message.channel,
           senderId: message.sender.id,
           channelInstallationId: message.channelInstallationId,
-          tenantId: message.tenantId
+          tenantId
         });
         if (!binding) {
           const installation = await this.installationStore.getInstallationById(
@@ -6623,7 +6673,7 @@ var MessageRouter = class {
               id: "fallback",
               channel: message.channel,
               channelInstallationId: message.channelInstallationId,
-              tenantId: message.tenantId,
+              tenantId,
               senderId: message.sender.id,
               agentId: installation.fallbackAgentId,
               threadId: void 0,
@@ -6649,7 +6699,7 @@ var MessageRouter = class {
           const threadStore = getStoreLattice14("default", "thread").store;
           const newThreadId = randomUUID7();
           const newThread = await threadStore.createThread(
-            message.tenantId,
+            tenantId,
             ctx.binding.agentId,
             newThreadId,
             {
@@ -6670,7 +6720,7 @@ var MessageRouter = class {
           }
         }
         const agent = agentInstanceManager6.getAgent({
-          tenant_id: message.tenantId,
+          tenant_id: tenantId,
           assistant_id: ctx.binding.agentId,
           thread_id: threadId,
           workspace_id: ctx.binding.workspaceId || "",