@axiom-lattice/gateway 2.1.77 → 2.1.79

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: {
@@ -2686,13 +2729,17 @@ function registerSandboxProxyRoutes(app2) {
       if (!vmIsolation) {
         return reply.status(500).send({ error: "Assistant sandbox config not found" });
       }
-      const sandboxName = sandboxService.computeSandboxName(
-        assistantId,
-        threadId,
-        vmIsolation
-      );
+      const workspaceId = request.headers["x-workspace-id"];
+      const projectId = request.headers["x-project-id"];
       const sandboxManager = getSandBoxManager2();
-      const sandbox = await sandboxManager.createSandbox(sandboxName);
+      const sandbox = await sandboxManager.getSandboxFromConfig({
+        assistant_id: assistantId,
+        thread_id: threadId,
+        tenantId,
+        workspaceId,
+        projectId,
+        vmIsolation
+      });
       try {
         const data = await request.file();
         if (!data) {
@@ -2730,13 +2777,17 @@ function registerSandboxProxyRoutes(app2) {
       if (!vmIsolation) {
         return reply.status(500).send({ error: "Assistant filesystem vmIsolation not found" });
       }
-      const sandboxName = sandboxService.computeSandboxName(
-        assistantId,
-        threadId,
-        vmIsolation
-      );
+      const workspaceId = request.headers["x-workspace-id"];
+      const projectId = request.headers["x-project-id"];
       const sandboxManager = getSandBoxManager2();
-      const sandbox = await sandboxManager.createSandbox(sandboxName);
+      const sandbox = await sandboxManager.getSandboxFromConfig({
+        assistant_id: assistantId,
+        thread_id: threadId,
+        tenantId,
+        workspaceId,
+        projectId,
+        vmIsolation
+      });
       try {
         const resolvedPath = filePath.startsWith("/") ? filePath : `/${filePath}`;
         const filename = getFilenameFromPath(resolvedPath);
@@ -5787,7 +5838,7 @@ var larkChannelAdapter = {
   },
   async sendReply(replyTarget, message, installation) {
     const { createLarkSender } = await import("./sender-PX32VSHB.mjs");
-    const sender = createLarkSender(installation.config);
+    const sender = await createLarkSender(installation.config);
     await sender.sendTextReply({
       chatId: replyTarget.rawTarget.chatId,
       text: message.text
@@ -5965,12 +6016,13 @@ function createLarkEventHandler(deps) {
       reply.status(404).send({ success: false, message: "Installation not found" });
       return;
     }
-    const body = parseLarkRequestBody(request.body, installation.config.encryptKey);
+    const larkInstallation = installation;
+    const body = parseLarkRequestBody(request.body, larkInstallation.config.encryptKey);
     if (body.type === "url_verification" && body.challenge) {
       reply.status(200).send({ challenge: body.challenge });
       return;
     }
-    const inboundMessage = await larkChannelAdapter.receive(request.body, installation);
+    const inboundMessage = await larkChannelAdapter.receive(request.body, larkInstallation);
     if (!inboundMessage) {
       reply.status(200).send();
       return;
@@ -6524,7 +6576,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 +6617,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 +6656,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 +6682,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 +6708,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 +6729,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 || "",