@agent-e/server 1.4.0 → 1.4.4

package/dist/index.d.mts

@@ -3,12 +3,15 @@ import { AgentEConfig, AgentE, EconomyState, EconomicEvent, Diagnosis, AgentEMod
 interface ServerConfig {
     port?: number;
     host?: string;
-    agentEConfig?: Partial<Omit<AgentEConfig, 'adapter'>>;
+    agentE?: Partial<Omit<AgentEConfig, 'adapter'>>;
+    validateState?: boolean;
+    corsOrigin?: string;
 }
-interface QueuedAdjustment {
-    key: string;
+interface EnrichedAdjustment {
+    parameter: string;
     value: number;
     currency?: string;
+    reasoning: string;
 }
 declare class AgentEServer {
     private readonly agentE;
@@ -16,10 +19,12 @@ declare class AgentEServer {
     private lastState;
     private adjustmentQueue;
     private alerts;
-    private readonly port;
+    readonly port: number;
     private readonly host;
     private readonly startedAt;
     private cleanupWs;
+    readonly validateState: boolean;
+    readonly corsOrigin: string;
     constructor(config?: ServerConfig);
     start(): Promise<void>;
     stop(): Promise<void>;
@@ -35,13 +40,14 @@ declare class AgentEServer {
      * 2. Set state
      * 3. Ingest events
      * 4. Run agentE.tick(state)
-     * 5. Drain adjustment queue
+     * 5. Drain adjustment queue, enrich with reasoning from decisions
      * 6. Return response
      */
     processTick(state: EconomyState, events?: EconomicEvent[]): Promise<{
-        adjustments: QueuedAdjustment[];
+        adjustments: EnrichedAdjustment[];
         alerts: Diagnosis[];
         health: number;
+        tick: number;
         decisions: ReturnType<AgentE['getDecisions']>;
     }>;
     /**
@@ -66,10 +72,10 @@ declare class AgentEServer {
  * @example
  * ```ts
  * import { startServer } from '@agent-e/server';
- * const server = await startServer({ port: 3000 });
+ * const server = await startServer({ port: 3100 });
  * // POST /tick, GET /health, etc.
  * ```
  */
 declare function startServer(config?: ServerConfig): Promise<AgentEServer>;
 
-export { AgentEServer, type ServerConfig, startServer };
+export { AgentEServer, type EnrichedAdjustment, type ServerConfig, startServer };

package/dist/index.d.ts

@@ -3,12 +3,15 @@ import { AgentEConfig, AgentE, EconomyState, EconomicEvent, Diagnosis, AgentEMod
 interface ServerConfig {
     port?: number;
     host?: string;
-    agentEConfig?: Partial<Omit<AgentEConfig, 'adapter'>>;
+    agentE?: Partial<Omit<AgentEConfig, 'adapter'>>;
+    validateState?: boolean;
+    corsOrigin?: string;
 }
-interface QueuedAdjustment {
-    key: string;
+interface EnrichedAdjustment {
+    parameter: string;
     value: number;
     currency?: string;
+    reasoning: string;
 }
 declare class AgentEServer {
     private readonly agentE;
@@ -16,10 +19,12 @@ declare class AgentEServer {
     private lastState;
     private adjustmentQueue;
     private alerts;
-    private readonly port;
+    readonly port: number;
     private readonly host;
     private readonly startedAt;
     private cleanupWs;
+    readonly validateState: boolean;
+    readonly corsOrigin: string;
     constructor(config?: ServerConfig);
     start(): Promise<void>;
     stop(): Promise<void>;
@@ -35,13 +40,14 @@ declare class AgentEServer {
      * 2. Set state
      * 3. Ingest events
      * 4. Run agentE.tick(state)
-     * 5. Drain adjustment queue
+     * 5. Drain adjustment queue, enrich with reasoning from decisions
      * 6. Return response
      */
     processTick(state: EconomyState, events?: EconomicEvent[]): Promise<{
-        adjustments: QueuedAdjustment[];
+        adjustments: EnrichedAdjustment[];
         alerts: Diagnosis[];
         health: number;
+        tick: number;
         decisions: ReturnType<AgentE['getDecisions']>;
     }>;
     /**
@@ -66,10 +72,10 @@ declare class AgentEServer {
  * @example
  * ```ts
  * import { startServer } from '@agent-e/server';
- * const server = await startServer({ port: 3000 });
+ * const server = await startServer({ port: 3100 });
  * // POST /tick, GET /health, etc.
  * ```
  */
 declare function startServer(config?: ServerConfig): Promise<AgentEServer>;
 
-export { AgentEServer, type ServerConfig, startServer };
+export { AgentEServer, type EnrichedAdjustment, type ServerConfig, startServer };

package/dist/index.js

@@ -31,31 +31,41 @@ var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__ge
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 
 // src/routes.ts
-function setCorsHeaders(res) {
-  res.setHeader("Access-Control-Allow-Origin", "*");
+function setCorsHeaders(res, origin) {
+  res.setHeader("Access-Control-Allow-Origin", origin);
   res.setHeader("Access-Control-Allow-Methods", "GET, POST, OPTIONS");
   res.setHeader("Access-Control-Allow-Headers", "Content-Type");
 }
-function json(res, status, data) {
-  setCorsHeaders(res);
+function json(res, status, data, origin) {
+  setCorsHeaders(res, origin);
   res.writeHead(status, { "Content-Type": "application/json" });
   res.end(JSON.stringify(data));
 }
 function readBody(req) {
   return new Promise((resolve, reject) => {
     const chunks = [];
-    req.on("data", (chunk) => chunks.push(chunk));
+    let totalBytes = 0;
+    req.on("data", (chunk) => {
+      totalBytes += chunk.length;
+      if (totalBytes > MAX_BODY_BYTES) {
+        req.destroy();
+        reject(new Error("Request body too large"));
+        return;
+      }
+      chunks.push(chunk);
+    });
     req.on("end", () => resolve(Buffer.concat(chunks).toString("utf-8")));
     req.on("error", reject);
   });
 }
 function createRouteHandler(server) {
+  const cors = server.corsOrigin;
   return async (req, res) => {
     const url = new URL(req.url ?? "/", `http://${req.headers.host ?? "localhost"}`);
     const path = url.pathname;
     const method = req.method?.toUpperCase() ?? "GET";
     if (method === "OPTIONS") {
-      setCorsHeaders(res);
+      setCorsHeaders(res, cors);
       res.writeHead(204);
       res.end();
       return;
@@ -67,47 +77,44 @@ function createRouteHandler(server) {
         try {
           parsed = JSON.parse(body);
         } catch {
-          json(res, 400, { error: "Invalid JSON" });
+          json(res, 400, { error: "Invalid JSON" }, cors);
           return;
         }
         if (!parsed || typeof parsed !== "object") {
-          json(res, 400, { error: "Body must be a JSON object" });
+          json(res, 400, { error: "Body must be a JSON object" }, cors);
           return;
         }
         const payload = parsed;
         const state = payload["state"] ?? parsed;
         const events = payload["events"];
-        const validation = (0, import_core.validateEconomyState)(state);
-        if (!validation.valid) {
-          json(res, 400, {
-            error: "Invalid state",
-            validation
-          });
-          return;
+        if (server.validateState) {
+          const validation = (0, import_core.validateEconomyState)(state);
+          if (!validation.valid) {
+            json(res, 400, {
+              error: "invalid_state",
+              validationErrors: validation.errors
+            }, cors);
+            return;
+          }
         }
         const result = await server.processTick(
           state,
           Array.isArray(events) ? events : void 0
         );
+        const warnings = server.validateState ? (0, import_core.validateEconomyState)(state).warnings : [];
         json(res, 200, {
           adjustments: result.adjustments,
           alerts: result.alerts.map((a) => ({
-            principle: a.principle.id,
-            name: a.principle.name,
+            principleId: a.principle.id,
+            principleName: a.principle.name,
             severity: a.violation.severity,
             evidence: a.violation.evidence,
-            suggestedAction: a.violation.suggestedAction
+            reasoning: a.violation.suggestedAction.reasoning
           })),
           health: result.health,
-          decisions: result.decisions.map((d) => ({
-            id: d.id,
-            tick: d.tick,
-            principle: d.diagnosis.principle.id,
-            parameter: d.plan.parameter,
-            result: d.result,
-            reasoning: d.reasoning
-          }))
-        });
+          tick: result.tick,
+          ...warnings.length > 0 ? { validationWarnings: warnings } : {}
+        }, cors);
         return;
       }
       if (path === "/health" && method === "GET") {
@@ -118,11 +125,11 @@ function createRouteHandler(server) {
           mode: agentE.getMode(),
           activePlans: agentE.getActivePlans().length,
           uptime: server.getUptime()
-        });
+        }, cors);
         return;
       }
       if (path === "/decisions" && method === "GET") {
-        const limit = parseInt(url.searchParams.get("limit") ?? "50", 10);
+        const limit = parseInt(url.searchParams.get("limit") ?? "100", 10);
         const since = url.searchParams.get("since");
         const agentE = server.getAgentE();
         let decisions;
@@ -131,20 +138,7 @@ function createRouteHandler(server) {
         } else {
           decisions = agentE.log.latest(limit);
         }
-        json(res, 200, {
-          decisions: decisions.map((d) => ({
-            id: d.id,
-            tick: d.tick,
-            timestamp: d.timestamp,
-            principle: d.diagnosis.principle.id,
-            principeName: d.diagnosis.principle.name,
-            parameter: d.plan.parameter,
-            currentValue: d.plan.currentValue,
-            targetValue: d.plan.targetValue,
-            result: d.result,
-            reasoning: d.reasoning
-          }))
-        });
+        json(res, 200, { decisions }, cors);
         return;
       }
       if (path === "/config" && method === "POST") {
@@ -153,27 +147,32 @@ function createRouteHandler(server) {
         try {
           parsed = JSON.parse(body);
         } catch {
-          json(res, 400, { error: "Invalid JSON" });
+          json(res, 400, { error: "Invalid JSON" }, cors);
           return;
         }
         const config = parsed;
-        if (config["action"] === "lock" && typeof config["param"] === "string") {
-          server.lock(config["param"]);
-          json(res, 200, { ok: true, action: "lock", param: config["param"] });
-        } else if (config["action"] === "unlock" && typeof config["param"] === "string") {
-          server.unlock(config["param"]);
-          json(res, 200, { ok: true, action: "unlock", param: config["param"] });
-        } else if (config["action"] === "constrain" && typeof config["param"] === "string" && typeof config["min"] === "number" && typeof config["max"] === "number") {
-          server.constrain(config["param"], { min: config["min"], max: config["max"] });
-          json(res, 200, { ok: true, action: "constrain", param: config["param"] });
-        } else if (config["action"] === "mode" && (config["mode"] === "autonomous" || config["mode"] === "advisor")) {
+        if (Array.isArray(config["lock"])) {
+          for (const param of config["lock"]) {
+            if (typeof param === "string") server.lock(param);
+          }
+        }
+        if (Array.isArray(config["unlock"])) {
+          for (const param of config["unlock"]) {
+            if (typeof param === "string") server.unlock(param);
+          }
+        }
+        if (Array.isArray(config["constrain"])) {
+          for (const c of config["constrain"]) {
+            if (c && typeof c === "object" && typeof c["param"] === "string" && typeof c["min"] === "number" && typeof c["max"] === "number") {
+              const constraint = c;
+              server.constrain(constraint.param, { min: constraint.min, max: constraint.max });
+            }
+          }
+        }
+        if (config["mode"] === "autonomous" || config["mode"] === "advisor") {
           server.setMode(config["mode"]);
-          json(res, 200, { ok: true, action: "mode", mode: config["mode"] });
-        } else {
-          json(res, 400, {
-            error: "Invalid config action. Use: lock, unlock, constrain, or mode"
-          });
         }
+        json(res, 200, { ok: true }, cors);
         return;
       }
       if (path === "/principles" && method === "GET") {
@@ -186,7 +185,7 @@ function createRouteHandler(server) {
             category: p.category,
             description: p.description
           }))
-        });
+        }, cors);
         return;
       }
       if (path === "/diagnose" && method === "POST") {
@@ -195,40 +194,44 @@ function createRouteHandler(server) {
         try {
           parsed = JSON.parse(body);
         } catch {
-          json(res, 400, { error: "Invalid JSON" });
+          json(res, 400, { error: "Invalid JSON" }, cors);
           return;
         }
         const payload = parsed;
         const state = payload["state"] ?? parsed;
-        const validation = (0, import_core.validateEconomyState)(state);
-        if (!validation.valid) {
-          json(res, 400, { error: "Invalid state", validation });
-          return;
+        if (server.validateState) {
+          const validation = (0, import_core.validateEconomyState)(state);
+          if (!validation.valid) {
+            json(res, 400, { error: "invalid_state", validationErrors: validation.errors }, cors);
+            return;
+          }
         }
         const result = server.diagnoseOnly(state);
         json(res, 200, {
           health: result.health,
           diagnoses: result.diagnoses.map((d) => ({
-            principle: d.principle.id,
-            name: d.principle.name,
+            principleId: d.principle.id,
+            principleName: d.principle.name,
             severity: d.violation.severity,
             evidence: d.violation.evidence,
             suggestedAction: d.violation.suggestedAction
           }))
-        });
+        }, cors);
         return;
       }
-      json(res, 404, { error: "Not found" });
+      json(res, 404, { error: "Not found" }, cors);
     } catch (err) {
-      json(res, 500, { error: "Internal server error" });
+      console.error("[AgentE Server] Unhandled route error:", err);
+      json(res, 500, { error: "Internal server error" }, cors);
     }
   };
 }
-var import_core;
+var import_core, MAX_BODY_BYTES;
 var init_routes = __esm({
   "src/routes.ts"() {
     "use strict";
     import_core = require("@agent-e/core");
+    MAX_BODY_BYTES = 1048576;
   }
 });
 
@@ -240,40 +243,53 @@ function send(ws, data) {
 }
 function createWebSocketHandler(httpServer, server) {
   const wss = new import_ws.WebSocketServer({ server: httpServer });
+  const aliveMap = /* @__PURE__ */ new WeakMap();
   const heartbeatInterval = setInterval(() => {
     for (const ws of wss.clients) {
       if (ws.readyState === import_ws.WebSocket.OPEN) {
+        if (aliveMap.get(ws) === false) {
+          ws.terminate();
+          continue;
+        }
+        aliveMap.set(ws, false);
         ws.ping();
       }
     }
   }, 3e4);
   wss.on("connection", (ws) => {
+    console.log("[AgentE Server] Client connected");
+    aliveMap.set(ws, true);
+    ws.on("pong", () => {
+      aliveMap.set(ws, true);
+    });
+    ws.on("close", () => {
+      console.log("[AgentE Server] Client disconnected");
+    });
     ws.on("message", async (raw) => {
       let msg;
       try {
         msg = JSON.parse(raw.toString());
       } catch {
-        send(ws, { type: "error", error: "Invalid JSON" });
+        send(ws, { type: "error", message: "Malformed JSON" });
         return;
       }
       if (!msg.type || typeof msg.type !== "string") {
-        send(ws, { type: "error", error: 'Missing "type" field' });
+        send(ws, { type: "error", message: 'Missing "type" field' });
         return;
       }
       switch (msg.type) {
         case "tick": {
           const state = msg["state"];
           const events = msg["events"];
-          const validation = (0, import_core2.validateEconomyState)(state);
-          if (!validation.valid) {
-            send(ws, { type: "validation_error", validation });
-            for (const w of validation.warnings) {
-              send(ws, { type: "validation_warning", warning: w });
+          if (server.validateState) {
+            const validation = (0, import_core2.validateEconomyState)(state);
+            if (!validation.valid) {
+              send(ws, { type: "validation_error", validationErrors: validation.errors });
+              return;
+            }
+            if (validation.warnings.length > 0) {
+              send(ws, { type: "validation_warning", validationWarnings: validation.warnings });
             }
-            return;
-          }
-          for (const w of validation.warnings) {
-            send(ws, { type: "validation_warning", warning: w });
           }
           try {
             const result = await server.processTick(
@@ -284,22 +300,16 @@ function createWebSocketHandler(httpServer, server) {
               type: "tick_result",
               adjustments: result.adjustments,
               alerts: result.alerts.map((a) => ({
-                principle: a.principle.id,
-                name: a.principle.name,
+                principleId: a.principle.id,
+                principleName: a.principle.name,
                 severity: a.violation.severity,
-                suggestedAction: a.violation.suggestedAction
+                reasoning: a.violation.suggestedAction.reasoning
               })),
               health: result.health,
-              decisions: result.decisions.map((d) => ({
-                id: d.id,
-                tick: d.tick,
-                principle: d.diagnosis.principle.id,
-                parameter: d.plan.parameter,
-                result: d.result
-              }))
+              tick: result.tick
             });
           } catch (err) {
-            send(ws, { type: "error", error: "Tick processing failed" });
+            send(ws, { type: "error", message: "Tick processing failed" });
           }
           break;
         }
@@ -309,7 +319,7 @@ function createWebSocketHandler(httpServer, server) {
             server.getAgentE().ingest(event);
             send(ws, { type: "event_ack" });
           } else {
-            send(ws, { type: "error", error: 'Missing "event" field' });
+            send(ws, { type: "error", message: 'Missing "event" field' });
           }
           break;
         }
@@ -327,18 +337,20 @@ function createWebSocketHandler(httpServer, server) {
         }
         case "diagnose": {
           const state = msg["state"];
-          const validation = (0, import_core2.validateEconomyState)(state);
-          if (!validation.valid) {
-            send(ws, { type: "validation_error", validation });
-            return;
+          if (server.validateState) {
+            const validation = (0, import_core2.validateEconomyState)(state);
+            if (!validation.valid) {
+              send(ws, { type: "validation_error", validationErrors: validation.errors });
+              return;
+            }
           }
           const result = server.diagnoseOnly(state);
           send(ws, {
             type: "diagnose_result",
             health: result.health,
             diagnoses: result.diagnoses.map((d) => ({
-              principle: d.principle.id,
-              name: d.principle.name,
+              principleId: d.principle.id,
+              principleName: d.principle.name,
               severity: d.violation.severity,
               suggestedAction: d.violation.suggestedAction
             }))
@@ -346,7 +358,7 @@ function createWebSocketHandler(httpServer, server) {
           break;
         }
         default:
-          send(ws, { type: "error", error: `Unknown message type: "${msg.type}"` });
+          send(ws, { type: "error", message: `Unknown message type: "${msg.type}"` });
       }
     });
   });
@@ -384,8 +396,10 @@ var init_AgentEServer = __esm({
         this.alerts = [];
         this.startedAt = Date.now();
         this.cleanupWs = null;
-        this.port = config.port ?? 3e3;
+        this.port = config.port ?? 3100;
         this.host = config.host ?? "0.0.0.0";
+        this.validateState = config.validateState ?? true;
+        this.corsOrigin = config.corsOrigin ?? "*";
         const adapter = {
           getState: () => {
             if (!this.lastState) {
@@ -407,16 +421,17 @@ var init_AgentEServer = __esm({
             this.adjustmentQueue.push({ key, value });
           }
         };
+        const agentECfg = config.agentE ?? {};
         const agentEConfig = {
           adapter,
-          mode: config.agentEConfig?.mode ?? "autonomous",
-          gracePeriod: config.agentEConfig?.gracePeriod ?? 0,
-          checkInterval: config.agentEConfig?.checkInterval ?? 1,
-          ...config.agentEConfig?.dominantRoles ? { dominantRoles: config.agentEConfig.dominantRoles } : {},
-          ...config.agentEConfig?.idealDistribution ? { idealDistribution: config.agentEConfig.idealDistribution } : {},
-          ...config.agentEConfig?.maxAdjustmentPercent !== void 0 ? { maxAdjustmentPercent: config.agentEConfig.maxAdjustmentPercent } : {},
-          ...config.agentEConfig?.cooldownTicks !== void 0 ? { cooldownTicks: config.agentEConfig.cooldownTicks } : {},
-          ...config.agentEConfig?.thresholds ? { thresholds: config.agentEConfig.thresholds } : {}
+          mode: agentECfg.mode ?? "autonomous",
+          gracePeriod: agentECfg.gracePeriod ?? 0,
+          checkInterval: agentECfg.checkInterval ?? 1,
+          ...agentECfg.dominantRoles ? { dominantRoles: agentECfg.dominantRoles } : {},
+          ...agentECfg.idealDistribution ? { idealDistribution: agentECfg.idealDistribution } : {},
+          ...agentECfg.maxAdjustmentPercent !== void 0 ? { maxAdjustmentPercent: agentECfg.maxAdjustmentPercent } : {},
+          ...agentECfg.cooldownTicks !== void 0 ? { cooldownTicks: agentECfg.cooldownTicks } : {},
+          ...agentECfg.thresholds ? { thresholds: agentECfg.thresholds } : {}
         };
         this.agentE = new import_core3.AgentE(agentEConfig);
         this.agentE.on("alert", (diagnosis) => {
@@ -430,6 +445,8 @@ var init_AgentEServer = __esm({
         this.cleanupWs = createWebSocketHandler(this.server, this);
         return new Promise((resolve) => {
           this.server.listen(this.port, this.host, () => {
+            const addr = this.getAddress();
+            console.log(`[AgentE Server] Listening on http://${addr.host}:${addr.port}`);
             resolve();
           });
         });
@@ -463,7 +480,7 @@ var init_AgentEServer = __esm({
        * 2. Set state
        * 3. Ingest events
        * 4. Run agentE.tick(state)
-       * 5. Drain adjustment queue
+       * 5. Drain adjustment queue, enrich with reasoning from decisions
        * 6. Return response
        */
       async processTick(state, events) {
@@ -476,13 +493,25 @@ var init_AgentEServer = __esm({
           }
         }
         await this.agentE.tick(state);
-        const adjustments = [...this.adjustmentQueue];
+        const rawAdj = [...this.adjustmentQueue];
         this.adjustmentQueue = [];
         const decisions = this.agentE.getDecisions({ since: state.tick, until: state.tick });
+        const adjustments = rawAdj.map((adj) => {
+          const decision = decisions.find(
+            (d) => d.plan.parameter === adj.key && d.result === "applied"
+          );
+          return {
+            parameter: adj.key,
+            value: adj.value,
+            ...adj.currency ? { currency: adj.currency } : {},
+            reasoning: decision?.diagnosis.violation.suggestedAction.reasoning ?? ""
+          };
+        });
         return {
           adjustments,
           alerts: [...this.alerts],
           health: this.agentE.getHealth(),
+          tick: state.tick,
           decisions
         };
       }
@@ -492,9 +521,8 @@ var init_AgentEServer = __esm({
       diagnoseOnly(state) {
         const prevState = this.lastState;
         this.lastState = state;
-        const diagnoser = this.agentE;
-        const diagnoses = diagnoser.diagnoseNow();
-        const health = diagnoser.getHealth();
+        const diagnoses = this.agentE.diagnoseNow();
+        const health = this.agentE.getHealth();
         this.lastState = prevState;
         return { diagnoses, health };
       }