@axiom-lattice/gateway 3.0.7 → 3.0.9

package/dist/index.mjs

@@ -692,6 +692,125 @@ var getAgentGraph = async (request, reply) => {
     });
   }
 };
+var STATS_CACHE_TTL_MS = 5 * 60 * 1e3;
+var statsCache = /* @__PURE__ */ new Map();
+async function getAssistantStats(request, reply) {
+  const tenantId = getTenantId(request);
+  const cached = statsCache.get(tenantId);
+  if (cached && cached.expiresAt > Date.now()) {
+    return {
+      success: true,
+      message: "Ok (cached)",
+      data: cached.data,
+      cache: {
+        expiresAt: new Date(cached.expiresAt).toISOString(),
+        remainingMs: cached.expiresAt - Date.now(),
+        cached: true
+      }
+    };
+  }
+  try {
+    const agentConfigs = await agentLatticeManager.getAllAgentConfigsByTenant(tenantId);
+    const codeConfiguredAssistants = agentConfigs.map(convertAgentConfigToAssistant);
+    const storeLattice = getStoreLattice("default", "assistant");
+    const assistantStore = storeLattice.store;
+    const storedAssistants = await assistantStore.getAllAssistants(tenantId);
+    const assistantMap = /* @__PURE__ */ new Map();
+    codeConfiguredAssistants.forEach((a) => assistantMap.set(a.id, a));
+    storedAssistants.forEach((a) => assistantMap.set(a.id, a));
+    const userId = getUserId(request);
+    const assistants = Array.from(assistantMap.values()).filter((a) => {
+      if (!a.ownerUserId) return true;
+      return a.ownerUserId === userId;
+    });
+    const threadStore = getStoreLattice("default", "thread").store;
+    const evalStore = getStoreLattice("default", "eval").store;
+    const now = Date.now();
+    const ms7d = 7 * 864e5;
+    const ms30d = 30 * 864e5;
+    const records = await Promise.all(
+      assistants.map(async (assistant) => {
+        let threadCount = 0;
+        let last7d = 0;
+        let last30d = 0;
+        let lastActive = null;
+        const daily7d = new Array(7).fill(0);
+        const daily30d = new Array(30).fill(0);
+        try {
+          const threads = await threadStore.getThreadsByAssistantId(tenantId, assistant.id);
+          threadCount = threads.length;
+          for (const t of threads) {
+            const created = new Date(t.createdAt).getTime();
+            if (created >= now - ms7d) last7d++;
+            if (created >= now - ms30d) last30d++;
+            const idx7 = Math.floor((now - created) / 864e5);
+            if (idx7 >= 0 && idx7 < 7) daily7d[6 - idx7]++;
+            const idx30 = Math.floor((now - created) / 864e5);
+            if (idx30 >= 0 && idx30 < 30) daily30d[29 - idx30]++;
+            const updated = new Date(t.updatedAt).getTime();
+            if (lastActive === null || updated > lastActive.getTime()) {
+              lastActive = new Date(t.updatedAt);
+            }
+          }
+        } catch {
+        }
+        let evalScore = null;
+        let evalPassRate = null;
+        let evalRuns = 0;
+        let evalTrend = [];
+        try {
+          const projects = await evalStore.getProjectsByTenant(tenantId);
+          const project = projects.find((p) => p.name === `eval-${assistant.id}`);
+          if (project) {
+            const runs = await evalStore.getRunsByTenant(tenantId, { projectId: project.id });
+            evalRuns = runs.length;
+            const sorted = [...runs].sort(
+              (a, b) => new Date(a.createdAt).getTime() - new Date(b.createdAt).getTime()
+            );
+            evalTrend = sorted.map((r) => typeof r.avgScore === "number" ? r.avgScore : null).filter((v) => v !== null);
+            const latest = sorted[sorted.length - 1];
+            if (latest) {
+              evalScore = typeof latest.avgScore === "number" ? latest.avgScore : null;
+              if (latest.totalCases > 0) {
+                evalPassRate = Math.round(latest.passedCases / latest.totalCases * 100);
+              }
+            }
+          }
+        } catch {
+        }
+        return {
+          id: assistant.id,
+          threadCount,
+          last7d,
+          last30d,
+          lastActive: lastActive ? lastActive.toISOString() : null,
+          daily7d,
+          daily30d,
+          evalScore,
+          evalPassRate,
+          evalRuns,
+          evalTrend: evalTrend.slice(-8)
+        };
+      })
+    );
+    const payload = { records };
+    const expiresAt = Date.now() + STATS_CACHE_TTL_MS;
+    statsCache.set(tenantId, { data: payload, expiresAt });
+    return {
+      success: true,
+      message: "Ok",
+      data: payload,
+      cache: {
+        expiresAt: new Date(expiresAt).toISOString(),
+        remainingMs: STATS_CACHE_TTL_MS,
+        cached: false
+      }
+    };
+  } catch (err) {
+    const message = err instanceof Error ? err.message : "Failed to get assistant stats";
+    return reply.status(500).send({ success: false, message });
+  }
+}
 
 // src/controllers/run.ts
 import { v4 } from "uuid";
@@ -3376,6 +3495,7 @@ async function listTasks(request, reply) {
       priority: query.priority,
       ...query.ownerId && { ownerId: query.ownerId },
       ...query.ownerType && { ownerType: query.ownerType },
+      ...query.parentId && { parentId: query.parentId },
       metadata: query.metadata ? JSON.parse(query.metadata) : void 0,
       projectId: query.projectId,
       limit: query.limit ? parseInt(query.limit) : void 0,
@@ -6214,7 +6334,7 @@ var EvalRunner = class {
   getEventEmitter() {
     return this.eventEmitter;
   }
-  async startRun(tenantId, projectId, suiteIds, caseIds, runConfig) {
+  async startRun(tenantId, projectId, suiteIds, caseIds, runConfig, taskId) {
     const store = this.getEvalStore();
     const project = await store.getProjectById(tenantId, projectId);
     if (!project) throw new Error("Project not found");
@@ -6285,12 +6405,15 @@ var EvalRunner = class {
       }
       resolvedJudgeModelKey = judgeModelKey;
     } else {
-      const firstModel = modelLatticeManager2.getAllLattices()[0];
-      if (!firstModel) {
+      const models = modelLatticeManager2.getAllLattices();
+      const defaultModel = models.find((m) => m.key === "default");
+      resolvedJudgeModelKey = defaultModel?.key ?? models[0]?.key ?? "";
+      if (!resolvedJudgeModelKey) {
         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}"`);
+      if (!defaultModel) {
+        console.warn(`[eval-runner] Project "${project.name}" has no judge modelKey and no "default" model \u2014 falling back to first registered model "${resolvedJudgeModelKey}"`);
+      }
     }
     const runId = uuidv43();
     const concurrency = Math.max(1, Math.floor(project.concurrency) || 3);
@@ -6302,7 +6425,8 @@ var EvalRunner = class {
       concurrency,
       holdout: isHoldout,
       envWorkspaceId: workspaceId || void 0,
-      envProjectId: projectIdFromConfig || void 0
+      envProjectId: projectIdFromConfig || void 0,
+      taskId
     });
     const projectConfig = {
       projectName: project.name,
@@ -6409,7 +6533,7 @@ var EvalRunner = class {
         this.runs.delete(runId);
       }
     })();
-    this.runs.set(runId, { runId, projectId, tenantId, abortController, promise: runPromise });
+    this.runs.set(runId, { runId, projectId, tenantId, abortController, promise: runPromise, taskId });
     runPromise.catch(() => {
     });
     return runId;
@@ -6480,7 +6604,9 @@ async function createProject(request, reply) {
       },
       targetServerConfig: {
         base_url: serverCfg.base_url || "",
-        api_key: serverCfg.api_key || ""
+        api_key: serverCfg.api_key || "",
+        // Project↔agent association pointer (see manage_eval create_project).
+        ...serverCfg.targetAgentId ? { targetAgentId: serverCfg.targetAgentId } : {}
       },
       concurrency: data.concurrency ?? 1,
       reportConfig: data.reportConfig
@@ -6789,7 +6915,7 @@ function registerEvalRoutes(app2) {
         workspaceId: body?.runConfig?.workspaceId ?? bodyWs,
         projectId: body?.runConfig?.projectId ?? bodyPj
       };
-      const runId = await evalRunner.startRun(tenantId, pid, body?.suiteIds, body?.caseIds, runConfig);
+      const runId = await evalRunner.startRun(tenantId, pid, body?.suiteIds, body?.caseIds, runConfig, body?.taskId);
       reply.status(202).send({ success: true, message: "Run started", data: { run_id: runId } });
     } catch (err) {
       const msg = err.message;
@@ -8792,6 +8918,7 @@ var registerLatticeRoutes = (app2, channelDeps) => {
     clearMemory
   );
   app2.get("/api/assistants", getAssistantList);
+  app2.get("/api/assistants/stats", getAssistantStats);
   app2.get("/api/assistants/:id", getAssistant);
   app2.post("/api/assistants", createAssistant);
   app2.put("/api/assistants/:id", updateAssistant);