@axiom-lattice/gateway 2.1.77 → 2.1.78

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@axiom-lattice/gateway",
-  "version": "2.1.77",
+  "version": "2.1.78",
   "main": "dist/index.js",
   "module": "dist/index.mjs",
   "types": "dist/index.d.ts",
@@ -40,11 +40,11 @@
     "redis": "^5.0.1",
     "uuid": "^9.0.1",
     "zod": "3.25.76",
-    "@axiom-lattice/agent-eval": "2.1.61",
-    "@axiom-lattice/core": "2.1.67",
-    "@axiom-lattice/pg-stores": "1.0.57",
-    "@axiom-lattice/protocols": "2.1.34",
-    "@axiom-lattice/queue-redis": "1.0.33"
+    "@axiom-lattice/agent-eval": "2.1.62",
+    "@axiom-lattice/core": "2.1.68",
+    "@axiom-lattice/pg-stores": "1.0.58",
+    "@axiom-lattice/protocols": "2.1.35",
+    "@axiom-lattice/queue-redis": "1.0.34"
   },
   "devDependencies": {
     "@types/jest": "^29.5.14",

package/src/controllers/workflow-tracking.ts

@@ -1,5 +1,5 @@
 import { FastifyRequest, FastifyReply } from "fastify";
-import { getStoreLattice, agentInstanceManager } from "@axiom-lattice/core";
+import { getStoreLattice, agentInstanceManager, ThreadStatus } from "@axiom-lattice/core";
 import type { WorkflowTrackingStore, WorkflowRun, RunStep } from "@axiom-lattice/protocols";
 import { MessageChunkTypes } from "@axiom-lattice/protocols";
 
@@ -331,6 +331,51 @@ export async function getWorkflowRun(
   }
 }
 
+export async function deleteWorkflowRun(
+  request: FastifyRequest<{ Params: { runId: string } }>,
+  reply: FastifyReply
+): Promise<ApiResponse> {
+  const { runId } = request.params;
+  const tenantId = getTenantId(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 is still running, abort the agent first
+    if (run.status === "running") {
+      try {
+        const workspace_id = request.headers["x-workspace-id"] as string;
+        const project_id = request.headers["x-project-id"] as string;
+        const agent = agentInstanceManager.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 as Error).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" });
+  }
+}
+
 export async function getRunSteps(
   request: FastifyRequest<{ Params: { runId: string }; Querystring: { step_type?: string; status?: string } }>,
   reply: FastifyReply
@@ -435,7 +480,17 @@ export async function replyInboxTask(
       project_id,
     });
 
-    // 3. 直接在后台触发 resume 执行（不等待，不阻塞 HTTP 响应）
+    // 3. 检查 graph 是否处于中断状态
+    const runStatus = await agent.getRunStatus();
+    if (runStatus !== ThreadStatus.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 },
+      });
+    }
+
+    // 4. 直接在后台触发 resume 执行（不等待，不阻塞 HTTP 响应）
     agent.addMessage({
       input: { message: "Clarification answers submitted" },
       command: {
@@ -462,7 +517,7 @@ export async function replyInboxTask(
       request.log.error(error, "Background resume failed");
     });
 
-    // 4. 立即返回成功（不等待 agent 执行完成）
+    // 5. 立即返回成功（不等待 agent 执行完成）
     return reply.status(200).send({
       success: true,
       message: "Resume request submitted successfully",

package/src/router/MessageRouter.ts

@@ -51,11 +51,21 @@ export class MessageRouter {
 
     try {
       await this.runMiddlewares(ctx, async () => {
+        // Resolve tenantId from installation if not provided
+        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) {
@@ -74,11 +84,11 @@ export class MessageRouter {
               id: "fallback",
               channel: message.channel,
               channelInstallationId: message.channelInstallationId,
-              tenantId: message.tenantId,
+              tenantId,
               senderId: message.sender.id,
               agentId: installation.fallbackAgentId,
               threadId: undefined,
-              threadMode: "fixed",
+              threadMode: "fixed" as const,
               enabled: true,
               createdAt: new Date(),
               updatedAt: new Date(),
@@ -103,7 +113,7 @@ export class MessageRouter {
           const threadStore = getStoreLattice("default", "thread").store;
           const newThreadId = randomUUID();
           const newThread = await threadStore.createThread(
-            message.tenantId,
+            tenantId,
             ctx.binding.agentId,
             newThreadId,
             {
@@ -125,7 +135,7 @@ export class MessageRouter {
         }
 
         const agent = agentInstanceManager.getAgent({
-          tenant_id: message.tenantId,
+          tenant_id: tenantId,
           assistant_id: ctx.binding.agentId,
           thread_id: threadId,
           workspace_id: ctx.binding.workspaceId || "",

package/src/routes/index.ts

@@ -378,7 +378,7 @@ export const registerLatticeRoutes = (app: FastifyInstance, channelDeps?: { rout
         const inboundMessage = {
           channel: msg.channel as string,
           channelInstallationId: (msg.channelInstallationId || "") as string,
-          tenantId: (msg.tenantId || "default") as string,
+          tenantId: msg.tenantId as string | undefined,
           sender: {
             id: (msg.sender as Record<string, unknown>).id as string,
             displayName: (msg.sender as Record<string, unknown>).displayName as string | undefined,
@@ -437,6 +437,10 @@ export const registerLatticeRoutes = (app: FastifyInstance, channelDeps?: { rout
     Params: { runId: string };
   }>("/api/workflows/runs/:runId", workflowTrackingController.getWorkflowRun);
 
+  app.delete<{
+    Params: { runId: string };
+  }>("/api/workflows/runs/:runId", workflowTrackingController.deleteWorkflowRun);
+
   app.get<{
     Params: { runId: string };
     Querystring: { step_type?: string; status?: string };