@wrongstack/webui 0.87.0 → 0.89.3

package/dist/server/index.js

@@ -1,4 +1,5 @@
 // src/server/index.ts
+import { expectDefined as expectDefined2 } from "@wrongstack/core";
 import * as fs4 from "fs/promises";
 import * as path4 from "path";
 
@@ -163,7 +164,7 @@ import {
   collabPauseMiddleware,
   collabInjectMiddleware,
   estimateRequestTokensCalibrated,
-  EventBus,
+  EventBus as EventBus2,
   HybridCompactor as HybridCompactor2,
   ProviderRegistry,
   TOKENS as TOKENS2,
@@ -171,6 +172,7 @@ import {
   atomicWrite as atomicWrite3,
   createDefaultPipelines,
   DEFAULT_CONTEXT_WINDOW_MODE_ID,
+  DEFAULT_SESSION_PRUNE_DAYS,
   DEFAULT_TOOLS_CONFIG,
   listContextWindowModes,
   repairToolUseAdjacency,
@@ -178,7 +180,7 @@ import {
 } from "@wrongstack/core";
 import { ToolExecutor } from "@wrongstack/core/execution";
 import { buildProviderFactoriesFromRegistry, makeProviderFromConfig } from "@wrongstack/providers";
-import { builtinToolsPack, forgetTool, rememberTool } from "@wrongstack/tools";
+import { builtinToolsPack, forgetTool, rememberTool, searchMemoryTool, relatedMemoryTool } from "@wrongstack/tools";
 import { WebSocketServer } from "ws";
 
 // ../runtime/src/container.ts
@@ -223,7 +225,7 @@ function createDefaultContainer(opts) {
       secretScrubber: container.resolve(TOKENS.SecretScrubber)
     })
   );
-  const memoryStore = new DefaultMemoryStore({ paths: wpaths });
+  const memoryStore = new DefaultMemoryStore({ paths: wpaths, events: opts.events });
   container.bind(TOKENS.MemoryStore, () => memoryStore);
   const skillLoader = new DefaultSkillLoader({ paths: wpaths, bundledDir: opts.bundledSkillsDir });
   container.bind(TOKENS.SkillLoader, () => skillLoader);
@@ -1692,12 +1694,7 @@ function createProviderConfigIO(configPath) {
 }
 
 // src/server/provider-keys.ts
-function expectDefined(value) {
-  if (value === null || value === void 0) {
-    throw new Error("Expected value to be defined");
-  }
-  return value;
-}
+import { expectDefined } from "@wrongstack/core";
 function normalizeKeys(cfg) {
   if (Array.isArray(cfg.apiKeys) && cfg.apiKeys.length > 0) {
     return cfg.apiKeys.map((k) => ({ ...k }));
@@ -1829,7 +1826,10 @@ function createProviderHandlers(deps) {
     return loadSavedProviders(globalConfigPath, vault);
   }
   async function saveConfigProviders(providers) {
-    const next = configWriteLock.then(() => saveProviders(globalConfigPath, vault, providers));
+    const next = configWriteLock.then(() => saveProviders(globalConfigPath, vault, providers)).catch((err) => {
+      const msg = err instanceof Error ? err.message : String(err);
+      console.error(`[ProviderHandlers] saveProviders failed: ${msg}`);
+    });
     configWriteLock = next;
     deps.setConfigWriteLock(next);
     await next;
@@ -2003,12 +2003,6 @@ function estimateContextBreakdown(input) {
 }
 
 // src/server/index.ts
-function expectDefined2(value) {
-  if (value === null || value === void 0) {
-    throw new Error("Expected value to be defined");
-  }
-  return value;
-}
 async function startWebUI(opts = {}) {
   const requestedWsPort = opts.wsPort ?? 3457;
   const wsHost = opts.wsHost ?? "127.0.0.1";
@@ -2061,12 +2055,14 @@ async function startWebUI(opts = {}) {
   if (config.features.memory) {
     toolRegistry.register(rememberTool(memoryStore));
     toolRegistry.register(forgetTool(memoryStore));
+    toolRegistry.register(searchMemoryTool(memoryStore));
+    toolRegistry.register(relatedMemoryTool(memoryStore));
   }
   console.log("[WebUI] Tool registry loaded:", toolRegistry.list().length, "tools");
-  const events = new EventBus();
+  const events = new EventBus2();
   events.setLogger(logger);
   const sessionStore = new DefaultSessionStore2({ dir: wpaths.projectSessions });
-  sessionStore.prune(30).then((count) => {
+  sessionStore.prune(DEFAULT_SESSION_PRUNE_DAYS).then((count) => {
     if (count > 0) logger.info(`Pruned ${count} old session${count === 1 ? "" : "s"}.`);
   }).catch(() => void 0);
   const sessionReader = new DefaultSessionReader({ store: sessionStore });
@@ -2360,7 +2356,10 @@ async function startWebUI(opts = {}) {
         if (typeof rawObj === "object" && rawObj !== null) {
           const obj = rawObj;
           if ("__proto__" in obj || "constructor" in obj || "prototype" in obj) {
-            send(ws, { type: "error", payload: { phase: "parse", message: "Invalid message object" } });
+            send(ws, {
+              type: "error",
+              payload: { phase: "parse", message: "Invalid message object" }
+            });
           } else {
             await handleMessage(ws, client, rawObj);
           }
@@ -2961,10 +2960,7 @@ async function startWebUI(opts = {}) {
           break;
         }
         const removed = expectDefined2(context.todos[targetIdx]);
-        const next = [
-          ...context.todos.slice(0, targetIdx),
-          ...context.todos.slice(targetIdx + 1)
-        ];
+        const next = [...context.todos.slice(0, targetIdx), ...context.todos.slice(targetIdx + 1)];
         context.state.replaceTodos(next);
         sendResult(ws, true, `Removed: ${removed.content}`);
         broadcast(clients, { type: "todos.updated", payload: { todos: next } });
@@ -2978,12 +2974,26 @@ async function startWebUI(opts = {}) {
             const plan = await loadPlan(planPath);
             send(ws, {
               type: "plan.updated",
-              payload: { plan: plan ?? { version: 1, sessionId: session.id, updatedAt: (/* @__PURE__ */ new Date()).toISOString(), items: [] } }
+              payload: {
+                plan: plan ?? {
+                  version: 1,
+                  sessionId: session.id,
+                  updatedAt: (/* @__PURE__ */ new Date()).toISOString(),
+                  items: []
+                }
+              }
             });
           } catch {
             send(ws, {
               type: "plan.updated",
-              payload: { plan: { version: 1, sessionId: session.id, updatedAt: (/* @__PURE__ */ new Date()).toISOString(), items: [] } }
+              payload: {
+                plan: {
+                  version: 1,
+                  sessionId: session.id,
+                  updatedAt: (/* @__PURE__ */ new Date()).toISOString(),
+                  items: []
+                }
+              }
             });
           }
         } else {
@@ -3143,9 +3153,14 @@ async function startWebUI(opts = {}) {
       }
       default:
         if (msg.type.startsWith("autophase.")) {
-          await autoPhaseHandler.handleMessage(msg);
+          await autoPhaseHandler.handleMessage(
+            msg
+          );
         } else {
-          send(ws, { type: "error", payload: { phase: "handleMessage", message: `Unknown message type: ${msg.type}` } });
+          send(ws, {
+            type: "error",
+            payload: { phase: "handleMessage", message: `Unknown message type: ${msg.type}` }
+          });
         }
     }
   }