@wrongstack/webui 0.280.1 → 0.281.0

package/dist/server/index.js

@@ -6565,6 +6565,7 @@ function registerShutdown(deps2) {
 import * as path15 from "path";
 import { createRequire } from "module";
 import { WebSocket as WebSocket2 } from "ws";
+import { DefaultTokenCounter } from "@wrongstack/core/infrastructure";
 import {
   AgentStatusTracker,
   AnnotationsStore,
@@ -6577,7 +6578,6 @@ import {
   DefaultSessionStore,
   DefaultSkillLoader,
   DefaultSystemPromptBuilder,
-  DefaultTokenCounter,
   EventBus,
   FleetNotifier,
   GlobalMailbox,
@@ -6593,6 +6593,7 @@ import {
   getSessionRegistry,
   makeMailboxTool,
   makeMailInboxTool,
+  makeFleetStatusTool,
   makeMailSendTool
 } from "@wrongstack/core";
 import { MCPRegistry } from "@wrongstack/mcp";
@@ -6866,6 +6867,7 @@ async function createPreContextServices(input) {
   toolRegistry.register(makeMailboxTool({ projectDir: wpaths.projectDir, events }));
   toolRegistry.register(makeMailSendTool({ projectDir: wpaths.projectDir, events }));
   toolRegistry.register(makeMailInboxTool({ projectDir: wpaths.projectDir, events }));
+  toolRegistry.register(makeFleetStatusTool({ projectDir: wpaths.projectDir, events }));
   applyToolDescriptionModes(toolRegistry, config.tools?.descriptionMode);
   applyToolResultRenderModes(toolRegistry, config.tools?.resultRenderMode);
   configureExecPolicy(config.tools?.exec ?? {});
@@ -6880,7 +6882,25 @@ async function createPreContextServices(input) {
       });
     }
   }
-  const sessionStore = opts.services?.session ?? new DefaultSessionStore({ dir: wpaths.projectSessions });
+  const sessionStore = opts.services?.session ?? new DefaultSessionStore({
+    dir: wpaths.projectSessions,
+    // Cross-process guard for delete(): refuses to remove a session that a
+    // live terminal/TUI/WebUI in this project is using. The store also
+    // checks active.json directly; this widens it to every concurrent
+    // surface via the SessionRegistry.
+    isSessionInUse: async (sessionId) => {
+      try {
+        const registry = getSessionRegistry(wpaths.globalRoot);
+        const live = await registry.listByProject(wpaths.projectSlug);
+        const hit = live.find((e) => e.sessionId === sessionId);
+        if (hit) {
+          return `active in ${hit.projectName} (PID ${hit.pid})`;
+        }
+      } catch {
+      }
+      return null;
+    }
+  });
   if (!opts.services?.session) {
     sessionStore.prune(DEFAULT_SESSION_PRUNE_DAYS).then((count) => {
       if (count > 0) logger.info(`Pruned ${count} old session${count === 1 ? "" : "s"}.`);
@@ -9411,6 +9431,9 @@ async function handleSessionRoute(ws, msg, handlers) {
     case "sessions.list":
       await handlers.listSessions(ws, msg);
       return true;
+    case "session.rename":
+      await handlers.renameSession(ws, msg);
+      return true;
     case "session.delete":
       await handlers.deleteSession(ws, msg);
       return true;
@@ -10987,6 +11010,38 @@ function createSessionHandlers(ctx) {
         sendResult(ws, false, errMessage(err));
       }
     },
+    renameSession: async (ws, msg) => {
+      const payload = msg.payload ?? {};
+      const id = typeof payload.id === "string" ? payload.id : "";
+      const name2 = typeof payload.name === "string" ? payload.name : "";
+      if (!id) {
+        sendResult(ws, false, "Session id is required");
+        return;
+      }
+      try {
+        await ctx.getSessionStore().rename(id, name2);
+        sendResult(ws, true, name2 ? `Renamed session to "${name2}"` : `Cleared session name`);
+        const list = await ctx.getSessionStore().list(50);
+        const currentId = ctx.getSession().id;
+        broadcast(ctx.clients, {
+          type: "sessions.list",
+          payload: {
+            sessions: list.map((s) => ({
+              id: s.id,
+              name: s.name,
+              title: s.title,
+              startedAt: s.startedAt,
+              model: s.model,
+              provider: s.provider,
+              tokenTotal: s.tokenTotal,
+              isCurrent: s.id === currentId
+            }))
+          }
+        });
+      } catch (err) {
+        sendResult(ws, false, errMessage(err));
+      }
+    },
     resumeSession: async (ws, msg) => {
       const { id } = msg.payload;
       try {
@@ -11112,15 +11167,17 @@ function buildRoutes(state, deps2, cb) {
       send(ws, {
         type: "provider.catalog",
         payload: {
-          providers: providers.map((p) => ({
-            id: p.id,
-            name: p.name,
-            family: p.family,
-            apiBase: p.apiBase,
-            envVars: p.envVars,
-            modelCount: p.models.length,
-            hasApiKey: savedIds.has(p.id) || p.envVars.some((v) => !!process.env[v])
-          }))
+          providers: providers.map(
+            (p) => ({
+              id: p.id,
+              name: p.name,
+              family: p.family,
+              apiBase: p.apiBase,
+              envVars: p.envVars,
+              modelCount: p.models.length,
+              hasApiKey: savedIds.has(p.id) || p.envVars.some((v) => !!process.env[v])
+            })
+          )
         }
       });
     },
@@ -11135,16 +11192,12 @@ function buildRoutes(state, deps2, cb) {
       const providerId = msg.payload.providerId;
       const saved = await providerHandlers.loadConfigProviders();
       const cfg = saved[providerId];
-      const provider = await resolveProviderCatalogForModels(
-        deps2.modelsRegistry,
-        providerId,
-        cfg
-      );
+      const provider = await resolveProviderCatalogForModels(deps2.modelsRegistry, providerId, cfg);
       const models = await enrichProviderModelDescriptors(
         deps2.modelsRegistry,
         providerId,
         cfg,
-        resolveProviderModelList(cfg?.models, provider)
+        resolveProviderModelList(cfg?.models, provider, cfg?.type ?? providerId)
       );
       send(ws, {
         type: "provider.models",
@@ -11167,7 +11220,9 @@ function buildRoutes(state, deps2, cb) {
         deps2.configStore.update({ provider: newProvider, model: newModel });
         deps2.context.model = newModel;
         const newCfg = state.getConfig();
-        const providerCfg = newCfg.providers?.[newProvider] ?? { type: newProvider };
+        const providerCfg = newCfg.providers?.[newProvider] ?? {
+          type: newProvider
+        };
         const newProv = deps2.providerRegistry.has(newProvider) ? deps2.providerRegistry.create({ ...providerCfg, type: newProvider }) : makeProviderFromConfig2(newProvider, providerCfg);
         deps2.context.provider = newProv;
         await cb.updateAutoCompactionMaxContext(newProv, newProvider, providerCfg);
@@ -11221,12 +11276,14 @@ function buildRoutes(state, deps2, cb) {
           timeoutMs: 9e4,
           ...reasoning ? { reasoning } : {},
           onError: (reason) => {
-            console.warn(JSON.stringify({
-              level: "warn",
-              event: "model.refine_failed",
-              reason,
-              timestamp: (/* @__PURE__ */ new Date()).toISOString()
-            }));
+            console.warn(
+              JSON.stringify({
+                level: "warn",
+                event: "model.refine_failed",
+                reason,
+                timestamp: (/* @__PURE__ */ new Date()).toISOString()
+              })
+            );
           }
         });
         if (result) {
@@ -11241,12 +11298,14 @@ function buildRoutes(state, deps2, cb) {
           });
         }
       } catch (err) {
-        console.error(JSON.stringify({
-          level: "error",
-          event: "model.refine.error",
-          error: errMessage(err),
-          timestamp: (/* @__PURE__ */ new Date()).toISOString()
-        }));
+        console.error(
+          JSON.stringify({
+            level: "error",
+            event: "model.refine.error",
+            error: errMessage(err),
+            timestamp: (/* @__PURE__ */ new Date()).toISOString()
+          })
+        );
         send(ws, {
           type: "model.refine_result",
           payload: { refined: text, english: text, error: errMessage(err) }
@@ -11328,17 +11387,24 @@ function buildRoutes(state, deps2, cb) {
       const cfg = state.getConfig();
       const features = cfg.features ?? {};
       if (typeof payload["featureMcp"] === "boolean") features["mcp"] = payload["featureMcp"];
-      if (typeof payload["featurePlugins"] === "boolean") features["plugins"] = payload["featurePlugins"];
-      if (typeof payload["featureMemory"] === "boolean") features["memory"] = payload["featureMemory"];
-      if (typeof payload["featureSkills"] === "boolean") features["skills"] = payload["featureSkills"];
-      if (typeof payload["featureModelsRegistry"] === "boolean") features["modelsRegistry"] = payload["featureModelsRegistry"];
+      if (typeof payload["featurePlugins"] === "boolean")
+        features["plugins"] = payload["featurePlugins"];
+      if (typeof payload["featureMemory"] === "boolean")
+        features["memory"] = payload["featureMemory"];
+      if (typeof payload["featureSkills"] === "boolean")
+        features["skills"] = payload["featureSkills"];
+      if (typeof payload["featureModelsRegistry"] === "boolean")
+        features["modelsRegistry"] = payload["featureModelsRegistry"];
       cfg.features = features;
-      if (Array.isArray(payload["fallbackModels"])) cfg.fallbackModels = payload["fallbackModels"];
+      if (Array.isArray(payload["fallbackModels"]))
+        cfg.fallbackModels = payload["fallbackModels"];
       if (payload["fallbackProfiles"] && typeof payload["fallbackProfiles"] === "object" && !Array.isArray(payload["fallbackProfiles"])) {
         cfg.fallbackProfiles = payload["fallbackProfiles"];
       }
-      if (Array.isArray(payload["favoriteModels"])) cfg.favoriteModels = payload["favoriteModels"];
-      if (typeof payload["favoriteModelsOnly"] === "boolean") cfg.favoriteModelsOnly = payload["favoriteModelsOnly"];
+      if (Array.isArray(payload["favoriteModels"]))
+        cfg.favoriteModels = payload["favoriteModels"];
+      if (typeof payload["favoriteModelsOnly"] === "boolean")
+        cfg.favoriteModelsOnly = payload["favoriteModelsOnly"];
       if (payload["modelMatrix"] && typeof payload["modelMatrix"] === "object" && !Array.isArray(payload["modelMatrix"])) {
         cfg.modelMatrix = payload["modelMatrix"];
       }
@@ -11346,7 +11412,10 @@ function buildRoutes(state, deps2, cb) {
       if (typeof payload["contextAutoCompact"] === "boolean") {
         if (payload["contextAutoCompact"] && deps2.autoCompactor) {
           deps2.pipelines.contextWindow.remove("AutoCompaction", { optional: true });
-          deps2.pipelines.contextWindow.use({ name: "AutoCompaction", handler: deps2.autoCompactor.handler() });
+          deps2.pipelines.contextWindow.use({
+            name: "AutoCompaction",
+            handler: deps2.autoCompactor.handler()
+          });
         } else {
           deps2.pipelines.contextWindow.remove("AutoCompaction", { optional: true });
         }
@@ -11381,7 +11450,10 @@ function buildRoutes(state, deps2, cb) {
         sendResult(ws, false, parsed.message);
         return;
       }
-      const result = await handleShellOpen(parsed.value, deps2.logger);
+      const result = await handleShellOpen(
+        parsed.value,
+        deps2.logger
+      );
       sendResult(ws, result.success, result.message);
     }
   };
@@ -11410,10 +11482,10 @@ function buildRoutes(state, deps2, cb) {
         parsed.value
       );
     },
-    clear: (ws) => handleMailboxClear(
-      ws,
-      { projectRoot: state.getProjectRoot(), globalRoot: path21.dirname(deps2.globalConfigPath) }
-    ),
+    clear: (ws) => handleMailboxClear(ws, {
+      projectRoot: state.getProjectRoot(),
+      globalRoot: path21.dirname(deps2.globalConfigPath)
+    }),
     purge: (ws, msg) => {
       const parsed = validateMailboxPurgePayload(msg.payload);
       if (!parsed.ok) {
@@ -11475,7 +11547,10 @@ function buildRoutes(state, deps2, cb) {
           risk: "medium",
           fallback: "ask_human"
         });
-        send(ws, { type: "brain.answer", payload: { sessionId: deps2.context.session?.id, question, decision } });
+        send(ws, {
+          type: "brain.answer",
+          payload: { sessionId: deps2.context.session?.id, question, decision }
+        });
       } catch (err) {
         sendResult(ws, false, `Brain consultation failed: ${errMessage(err)}`);
       }