@axiom-lattice/gateway 2.1.111 → 2.1.112

package/.turbo/turbo-build.log

@@ -1,5 +1,5 @@
 
-> @axiom-lattice/gateway@2.1.111 build /home/runner/work/agentic/agentic/packages/gateway
+> @axiom-lattice/gateway@2.1.112 build /home/runner/work/agentic/agentic/packages/gateway
 > tsup src/index.ts --format cjs,esm --dts --clean --sourcemap
 
 CLI Building entry: src/index.ts
@@ -11,36 +11,36 @@
 ESM Build start
 [warn] ▲ [WARNING] "import.meta" is not available with the "cjs" output format and will be empty [empty-import-meta]
 
-    src/index.ts:199:33:
-      199 │ const __filename = fileURLToPath(import.meta.url);
+    src/index.ts:200:33:
+      200 │ const __filename = fileURLToPath(import.meta.url);
           ╵                                  ~~~~~~~~~~~
 
   You need to set the output format to "esm" for "import.meta" to work correctly.
 
 
+CJS dist/index.js     357.58 KB
+CJS dist/index.js.map 729.13 KB
+CJS ⚡️ Build success in 403ms
+ESM dist/index.mjs                             286.89 KB
 ESM dist/mcp-configs-SHRMQHIL.mjs              685.00 B
 ESM dist/chunk-K4XELOSO.mjs                    12.52 KB
 ESM dist/sender-PX32VSHB.mjs                   873.00 B
 ESM dist/WechatChannelAdapter-WSDKR4OA.mjs     8.27 KB
-ESM dist/a2a-ERG5RMUW.mjs                      15.95 KB
 ESM dist/chunk-6CUQGDJI.mjs                    6.42 KB
-ESM dist/index.mjs                             282.73 KB
+ESM dist/a2a-ERG5RMUW.mjs                      15.95 KB
 ESM dist/resources-VA7LSDKN.mjs                15.81 KB
 ESM dist/chunk-LHQY46YB.mjs                    1.46 KB
 ESM dist/chunk-K4XELOSO.mjs.map                25.53 KB
 ESM dist/sender-PX32VSHB.mjs.map               2.07 KB
 ESM dist/WechatChannelAdapter-WSDKR4OA.mjs.map 16.28 KB
-ESM dist/a2a-ERG5RMUW.mjs.map                  32.14 KB
 ESM dist/chunk-6CUQGDJI.mjs.map                14.04 KB
+ESM dist/a2a-ERG5RMUW.mjs.map                  32.14 KB
 ESM dist/resources-VA7LSDKN.mjs.map            29.34 KB
 ESM dist/chunk-LHQY46YB.mjs.map                2.39 KB
-ESM dist/index.mjs.map                         598.22 KB
+ESM dist/index.mjs.map                         606.92 KB
 ESM dist/mcp-configs-SHRMQHIL.mjs.map          71.00 B
-ESM ⚡️ Build success in 559ms
-CJS dist/index.js     353.42 KB
-CJS dist/index.js.map 720.44 KB
-CJS ⚡️ Build success in 559ms
+ESM ⚡️ Build success in 479ms
 DTS Build start
-DTS ⚡️ Build success in 20544ms
+DTS ⚡️ Build success in 21490ms
 DTS dist/index.d.ts  7.57 KB
 DTS dist/index.d.mts 7.57 KB

package/CHANGELOG.md

@@ -1,5 +1,16 @@
 # @axiom-lattice/gateway
 
+## 2.1.112
+
+### Patch Changes
+
+- 2d1c9f4: enhance core
+- Updated dependencies [2d1c9f4]
+  - @axiom-lattice/core@2.1.99
+  - @axiom-lattice/pg-stores@1.0.90
+  - @axiom-lattice/protocols@2.1.51
+  - @axiom-lattice/queue-redis@1.0.50
+
 ## 2.1.111
 
 ### Patch Changes

package/dist/index.js

@@ -4332,6 +4332,8 @@ async function getAllWorkflowRuns(request, reply) {
     });
   }
 }
+var INBOX_DEFAULT_PAGE_SIZE = 20;
+var INBOX_MAX_PAGE_SIZE = 100;
 async function getInboxItems(request, reply) {
   const tenantId = getTenantId7(request);
   try {
@@ -4349,10 +4351,30 @@ async function getInboxItems(request, reply) {
       }
     } catch {
     }
-    const runs = await store2.getWorkflowRunsByTenantId(tenantId);
-    const pendingRuns = runs.filter((r) => r.status === "interrupted");
+    const { page: pageParam, pageSize: pageSizeParam } = request.query;
+    const paged = pageParam !== void 0 || pageSizeParam !== void 0;
+    const page = Math.max(1, parseInt(pageParam ?? "", 10) || 1);
+    const pageSize = Math.min(INBOX_MAX_PAGE_SIZE, Math.max(1, parseInt(pageSizeParam ?? "", 10) || INBOX_DEFAULT_PAGE_SIZE));
+    let pendingRuns;
+    let total;
+    if (paged) {
+      const result = await store2.queryWorkflowRuns(tenantId, {
+        status: "interrupted",
+        limit: pageSize,
+        offset: (page - 1) * pageSize
+      });
+      pendingRuns = result.records;
+      total = result.total;
+    } else {
+      const result = await store2.queryWorkflowRuns(tenantId, { status: "interrupted" });
+      pendingRuns = result.records;
+    }
     if (pendingRuns.length === 0) {
-      return { success: true, message: "No pending workflows", data: { records: [] } };
+      return {
+        success: true,
+        message: "No pending workflows",
+        data: paged ? { records: [], total, page, pageSize } : { records: [] }
+      };
     }
     const checkPromises = pendingRuns.map(async (r) => {
       try {
@@ -4409,13 +4431,10 @@ async function getInboxItems(request, reply) {
     });
     const results = await Promise.all(checkPromises);
     const inboxItems = results.flat();
-    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 }
+      data: paged ? { records: inboxItems, total, page, pageSize } : { records: inboxItems }
     };
   } catch (error) {
     request.log.error(error, "Failed to get inbox items");
@@ -4648,6 +4667,54 @@ async function replyInboxTask(request, reply) {
     });
   }
 }
+async function abortWorkflowRun(request, reply) {
+  try {
+    const { runId } = request.params;
+    const tenantId = getTenantId7(request);
+    const store2 = getTrackingStore();
+    if (!store2) {
+      return reply.status(404).send({
+        success: false,
+        message: "No workflow tracking store configured"
+      });
+    }
+    const run = await store2.getWorkflowRun(runId);
+    if (!run) {
+      return reply.status(404).send({
+        success: false,
+        message: "Workflow run not found"
+      });
+    }
+    await store2.updateWorkflowRun(runId, {
+      status: "cancelled",
+      errorMessage: "Aborted by user",
+      completedAt: /* @__PURE__ */ new Date()
+    }).catch(() => {
+    });
+    const workspace_id = request.headers["x-workspace-id"];
+    const project_id = request.headers["x-project-id"];
+    const agent = import_core15.agentInstanceManager.getAgent({
+      assistant_id: run.assistantId,
+      thread_id: run.threadId,
+      tenant_id: tenantId,
+      workspace_id,
+      project_id
+    });
+    await agent.abort();
+    (0, import_core15.abortWorkflowRun)(runId);
+    return reply.status(200).send({
+      success: true,
+      message: "Workflow aborted",
+      data: { runId, assistantId: run.assistantId, threadId: run.threadId }
+    });
+  } catch (error) {
+    request.log.error(error, "Failed to abort workflow run");
+    return reply.status(500).send({
+      success: false,
+      message: `Abort failed: ${error.message}`
+    });
+  }
+}
 
 // src/controllers/personal-assistant.ts
 var import_core16 = require("@axiom-lattice/core");
@@ -6987,6 +7054,10 @@ var testOrDiscoverBody = import_zod.z.object({
   config: import_zod.z.record(import_zod.z.unknown()).optional()
 });
 function getTenant(request) {
+  const userTenantId = request.user?.tenantId;
+  if (userTenantId) {
+    return userTenantId;
+  }
   return request.headers["x-tenant-id"] || "default";
 }
 function maskSecrets(config) {
@@ -7136,7 +7207,6 @@ var import_uuid5 = require("uuid");
 // src/services/eval-runner.ts
 var import_events = require("events");
 var import_core29 = require("@axiom-lattice/core");
-var import_agent_eval = require("@axiom-lattice/agent-eval");
 var import_uuid4 = require("uuid");
 function mapLogs(logs) {
   return logs.map((l) => ({
@@ -7158,6 +7228,11 @@ var EvalRunner = class {
     const store2 = this.getEvalStore();
     const project = await store2.getProjectById(tenantId, projectId);
     if (!project) throw new Error("Project not found");
+    for (const ctx of this.runs.values()) {
+      if (ctx.projectId === projectId) {
+        throw new Error("A run is already in progress for this project");
+      }
+    }
     const existingRuns = await store2.getRunsByTenant(tenantId, { projectId, status: "running" });
     if (existingRuns.length > 0) {
       throw new Error("A run is already in progress for this project");
@@ -7174,7 +7249,7 @@ var EvalRunner = class {
           caseId: c.id,
           input: { message: c.inputMessage, files: c.inputFiles },
           steps: c.steps,
-          output: { type: c.outputType },
+          output: { type: "message_content" },
           eval: {
             content_assertion: c.contentAssertion,
             eval_rubrics: c.rubrics?.map((r) => ({
@@ -7186,35 +7261,32 @@ var EvalRunner = class {
         }))
       });
     }
-    const runId = (0, import_uuid4.v4)();
-    const concurrency = project.concurrency || 3;
-    await store2.createRun(tenantId, projectId, runId, { totalCases, concurrency });
-    const judgeCfg = project.judgeModelConfig;
-    const hasModelKey = Boolean(judgeCfg.modelKey);
-    const hasApiKey = Boolean(judgeCfg.apiKeyEnvName || judgeCfg.apiKey);
-    const hasCredentials = hasApiKey;
-    let judgeModelConfig = {};
-    if (hasModelKey) {
-      judgeModelConfig = { modelKey: judgeCfg.modelKey };
-    } else if (!hasCredentials) {
-      const firstModel = import_core29.modelLatticeManager.getAllLattices()[0];
-      if (firstModel) {
-        judgeModelConfig = { modelKey: firstModel.key };
-      } else {
-        judgeModelConfig = { model: judgeCfg };
+    const judgeCfg = project.judgeModelConfig ?? {};
+    const judgeModelKey = judgeCfg.modelKey || "";
+    let resolvedJudgeModelKey;
+    if (judgeModelKey) {
+      if (!import_core29.modelLatticeManager.hasLattice(judgeModelKey)) {
+        throw new Error(`Judge model "${judgeModelKey}" is not registered`);
       }
+      resolvedJudgeModelKey = judgeModelKey;
     } else {
-      judgeModelConfig = { model: judgeCfg };
+      const firstModel = import_core29.modelLatticeManager.getAllLattices()[0];
+      if (!firstModel) {
+        throw new Error("No model registered \u2014 cannot run eval without a judge model");
+      }
+      resolvedJudgeModelKey = firstModel.key;
+      console.warn(`[eval-runner] Project "${project.name}" has no judge modelKey \u2014 falling back to first registered model "${firstModel.key}"`);
     }
+    const runId = (0, import_uuid4.v4)();
+    const concurrency = Math.max(1, Math.floor(project.concurrency) || 3);
+    await store2.createRun(tenantId, projectId, runId, { totalCases, concurrency });
     const projectConfig = {
       projectName: project.name,
       version: project.version,
       description: project.description,
       suites: evalSuites,
-      judge_agent_config: judgeModelConfig,
+      judge_agent_config: { modelKey: resolvedJudgeModelKey },
       lattice_server_config: {
-        base_url: project.targetServerConfig.base_url || "",
-        api_key: project.targetServerConfig.api_key || "",
         tenant_id: tenantId
       },
       concurrency
@@ -7258,34 +7330,38 @@ var EvalRunner = class {
     };
     const runPromise = (async () => {
       try {
-        const evalProject = new import_agent_eval.LatticeEvalProject(projectConfig, onCaseComplete);
-        const { report } = await evalProject.runAllSuitesBatch(concurrency);
-        const completedCount = stats.passed + stats.failed;
-        await store2.updateRunStatus(tenantId, runId, {
-          status: "completed",
-          completedAt: /* @__PURE__ */ new Date(),
-          passedCases: stats.passed,
-          failedCases: stats.failed,
-          avgScore: completedCount > 0 ? stats.totalScore / completedCount : 0
-        });
-        this.eventEmitter.emit(`run:${runId}`, {
-          type: "completed",
-          runId,
-          data: { passed: stats.passed, failed: stats.failed, avgScore: completedCount > 0 ? stats.totalScore / completedCount : 0 }
-        });
+        const evalProject = new import_core29.LatticeEvalProject(projectConfig, onCaseComplete);
+        const { report } = await evalProject.runAllSuitesBatch(concurrency, abortController.signal);
+        if (!abortController.signal.aborted) {
+          const completedCount = stats.passed + stats.failed;
+          await store2.updateRunStatus(tenantId, runId, {
+            status: "completed",
+            completedAt: /* @__PURE__ */ new Date(),
+            passedCases: stats.passed,
+            failedCases: stats.failed,
+            avgScore: completedCount > 0 ? stats.totalScore / completedCount : 0
+          });
+          this.eventEmitter.emit(`run:${runId}`, {
+            type: "completed",
+            runId,
+            data: { passed: stats.passed, failed: stats.failed, avgScore: completedCount > 0 ? stats.totalScore / completedCount : 0 }
+          });
+        }
         return { report };
       } catch (err) {
         const errorMsg = err.message;
-        await store2.updateRunStatus(tenantId, runId, {
-          status: "failed",
-          error: errorMsg,
-          completedAt: /* @__PURE__ */ new Date()
-        });
-        this.eventEmitter.emit(`run:${runId}`, {
-          type: "error",
-          runId,
-          data: { message: errorMsg }
-        });
+        if (!abortController.signal.aborted) {
+          await store2.updateRunStatus(tenantId, runId, {
+            status: "failed",
+            error: errorMsg,
+            completedAt: /* @__PURE__ */ new Date()
+          });
+          this.eventEmitter.emit(`run:${runId}`, {
+            type: "error",
+            runId,
+            data: { message: errorMsg }
+          });
+        }
         throw err;
       } finally {
         this.runs.delete(runId);
@@ -7330,12 +7406,27 @@ async function createProject(request, reply) {
     const store2 = getEvalStore();
     const id = (0, import_uuid5.v4)();
     const data = request.body;
+    const serverCfg = data.targetServerConfig || {};
+    const judgeCfg = data.judgeModelConfig || {};
+    const modelKey = judgeCfg.modelKey || "";
+    if (modelKey) {
+      const { modelLatticeManager: modelLatticeManager3 } = await import("@axiom-lattice/core");
+      if (!modelLatticeManager3.hasLattice(modelKey)) {
+        return reply.status(400).send({ success: false, message: `Judge model "${modelKey}" is not registered` });
+      }
+    }
     const project = await store2.createProject(tenantId, id, {
       name: data.name,
       description: data.description,
       version: data.version,
-      judgeModelConfig: data.judgeModelConfig ?? {},
-      targetServerConfig: data.targetServerConfig ?? {},
+      judgeModelConfig: {
+        modelKey,
+        displayName: judgeCfg.displayName || ""
+      },
+      targetServerConfig: {
+        base_url: serverCfg.base_url || "",
+        api_key: serverCfg.api_key || ""
+      },
       concurrency: data.concurrency ?? 1,
       reportConfig: data.reportConfig
     });
@@ -7359,7 +7450,7 @@ async function listProjects(request, reply) {
         const { modelLatticeManager: modelLatticeManager3 } = await import("@axiom-lattice/core");
         const models = modelLatticeManager3.getAllLattices();
         const first = models[0];
-        const judgeModel = first ? { modelKey: first.key } : { provider: "openai", model: "gpt-4" };
+        const judgeModel = first ? { modelKey: first.key } : {};
         const host = request.hostname || "localhost";
         const baseUrl = `http://${host}:${process.env.PORT || 4001}`;
         await store2.createProject(tenantId, (0, import_uuid5.v4)(), {
@@ -7412,7 +7503,30 @@ async function updateProject(request, reply) {
     if (!existing) {
       return reply.status(404).send({ success: false, message: "Project not found" });
     }
-    const updated = await store2.updateProject(tenantId, pid, request.body);
+    const body = request.body;
+    const updateData = { ...body };
+    const serverCfg = body.targetServerConfig || {};
+    if (body.targetServerConfig !== void 0) {
+      updateData.targetServerConfig = {
+        base_url: serverCfg.base_url || "",
+        api_key: serverCfg.api_key || ""
+      };
+    }
+    if (body.judgeModelConfig !== void 0) {
+      const judgeCfg = body.judgeModelConfig || {};
+      const modelKey = judgeCfg.modelKey || "";
+      if (modelKey) {
+        const { modelLatticeManager: modelLatticeManager3 } = await import("@axiom-lattice/core");
+        if (!modelLatticeManager3.hasLattice(modelKey)) {
+          return reply.status(400).send({ success: false, message: `Judge model "${modelKey}" is not registered` });
+        }
+      }
+      updateData.judgeModelConfig = {
+        modelKey,
+        displayName: judgeCfg.displayName || ""
+      };
+    }
+    const updated = await store2.updateProject(tenantId, pid, updateData);
     return {
       success: true,
       message: "Successfully updated project",
@@ -9466,6 +9580,7 @@ var registerLatticeRoutes = (app2, channelDeps) => {
   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/runs/:runId/abort", abortWorkflowRun);
   app2.post("/api/workflows/inbox/reply", replyInboxTask);
   app2.delete(
     "/api/assistants/:assistant_id/threads/:thread_id/pending-messages/:message_id",
@@ -10527,6 +10642,7 @@ var start = async (config) => {
         error: err instanceof Error ? err.message : String(err)
       });
     }
+    (0, import_core40.setEvalRunService)(evalRunner);
     try {
       const menuStore = (0, import_core41.getStoreLattice)("default", "menu").store;
       (0, import_core40.setMenuRegistry)(menuStore);
@@ -10587,11 +10703,13 @@ var start = async (config) => {
         agentTaskConsumer.startPollingQueue();
       }
     }
-    import_core41.agentInstanceManager.restore().then((stats) => {
-      logger5.info(`Agent recovery complete: ${stats.restored} threads restored, ${stats.errors} errors`);
-    }).catch((error) => {
-      logger5.error("Agent recovery failed", { error });
-    });
+    if (process.env.AXIOM_RESTORE_PENDING_ON_STARTUP === "true") {
+      import_core41.agentInstanceManager.restore().then((stats) => {
+        logger5.info(`Agent recovery complete: ${stats.restored} threads restored, ${stats.errors} errors`);
+      }).catch((error) => {
+        logger5.error("Agent recovery failed", { error });
+      });
+    }
     restoreMcpConnections().catch((error) => {
       logger5.error("MCP connection restoration failed", { error });
     });