Merge pull request elizaOS#1245 from renlulu/feat/express-payload-config

feat: make express payload limit configurable
AbdelStark · Dec 19, 2024 · 800cd0d · 800cd0d
2 parents da405cd + 31b336f
commit 800cd0d
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 17 deletions.
diff --git a/.env.example b/.env.example
@@ -34,6 +34,9 @@ LARGE_HYPERBOLIC_MODEL=         # Default: meta-llama/Meta-Llama-3.1-405-Instruc
 # Speech Synthesis
 ELEVENLABS_XI_API_KEY=          # API key from elevenlabs
 
+# Direct Client Setting
+EXPRESS_MAX_PAYLOAD= # Default: 100kb
+
 # ElevenLabs Settings
 ELEVENLABS_MODEL_ID=eleven_multilingual_v2
 ELEVENLABS_VOICE_ID=21m00Tcm4TlvDq8ikWAM

diff --git a/packages/client-direct/src/api.ts b/packages/client-direct/src/api.ts
@@ -5,17 +5,27 @@ import cors from "cors";
 import {
     AgentRuntime,
     elizaLogger,
+    getEnvVariable,
     validateCharacterConfig,
 } from "@ai16z/eliza";
 
 import { REST, Routes } from "discord.js";
+import { DirectClient } from ".";
 
-export function createApiRouter(agents: Map<string, AgentRuntime>, directClient) {
+export function createApiRouter(
+    agents: Map<string, AgentRuntime>,
+    directClient: DirectClient
+) {
     const router = express.Router();
 
     router.use(cors());
     router.use(bodyParser.json());
     router.use(bodyParser.urlencoded({ extended: true }));
+    router.use(
+        express.json({
+            limit: getEnvVariable("EXPRESS_MAX_PAYLOAD") || "100kb",
+        })
+    );
 
     router.get("/", (req, res) => {
         res.send("Welcome, this is the REST API!");
@@ -51,41 +61,40 @@ export function createApiRouter(agents: Map<string, AgentRuntime>, directClient)
 
     router.post("/agents/:agentId/set", async (req, res) => {
         const agentId = req.params.agentId;
-        console.log('agentId', agentId)
-        let agent:AgentRuntime = agents.get(agentId);
+        console.log("agentId", agentId);
+        let agent: AgentRuntime = agents.get(agentId);
 
         // update character
         if (agent) {
             // stop agent
-            agent.stop()
-            directClient.unregisterAgent(agent)
+            agent.stop();
+            directClient.unregisterAgent(agent);
             // if it has a different name, the agentId will change
         }
 
         // load character from body
-        const character = req.body
+        const character = req.body;
         try {
-          validateCharacterConfig(character)
-        } catch(e) {
-          elizaLogger.error(`Error parsing character: ${e}`);
-          res.status(400).json({
-            success: false,
-            message: e.message,
-          });
-          return;
+            validateCharacterConfig(character);
+        } catch (e) {
+            elizaLogger.error(`Error parsing character: ${e}`);
+            res.status(400).json({
+                success: false,
+                message: e.message,
+            });
+            return;
         }
 
         // start it up (and register it)
-        agent = await directClient.startAgent(character)
-        elizaLogger.log(`${character.name} started`)
+        agent = await directClient.startAgent(character);
+        elizaLogger.log(`${character.name} started`);
 
         res.json({
             id: character.id,
             character: character,
         });
     });
 
-
     router.get("/agents/:agentId/channels", async (req, res) => {
         const agentId = req.params.agentId;
         const runtime = agents.get(agentId);