Add a way to create/store/restore agents in the filesystem at agents/data/characters

Author: maxcoto

.env.example

@@ -616,3 +616,6 @@ INSTAGRAM_POST_INTERVAL_MAX=120   # Default: 120 minutes
 INSTAGRAM_ENABLE_ACTION_PROCESSING=false  # Enable/disable action processing
 INSTAGRAM_ACTION_INTERVAL=5       # Interval between actions in minutes
 INSTAGRAM_MAX_ACTIONS=1          # Maximum number of actions to process at once
+
+# Stores characters set by using the direct API in the data/character folder for further load when the app restarts
+USE_CHARACTER_STORAGE=false

agent/src/index.ts

@@ -249,11 +249,29 @@ function commaSeparatedStringToArray(commaSeparated: string): string[] {
     return commaSeparated?.split(",").map(value => value.trim())
 }
 
+async function readCharactersFromStorage(characterPaths: string[]): Promise<string[]> {
+    try {
+        const uploadDir = path.join(process.cwd(), "data", "characters");
+        const fileNames = await fs.promises.readdir(uploadDir);
+        fileNames.forEach(fileName => {
+            characterPaths.push(uploadDir + '/' + fileName);
+        });
+    } catch (err) {
+        console.error('Error reading directory:', err);
+    }
+
+    return characterPaths;
+};
 
 export async function loadCharacters(
     charactersArg: string
 ): Promise<Character[]> {
     let characterPaths = commaSeparatedStringToArray(charactersArg)
+
+    if(process.env.USE_CHARACTER_STORAGE) {
+        characterPaths = await readCharactersFromStorage(characterPaths);
+    }
+
     const loadedCharacters: Character[] = [];
 
     if (characterPaths?.length > 0) {

packages/client-direct/src/api.ts

@@ -1,6 +1,8 @@
 import express from "express";
 import bodyParser from "body-parser";
 import cors from "cors";
+import path from "path";
+import fs from "fs";
 
 import {
     AgentRuntime,
@@ -79,6 +81,16 @@ export function createApiRouter(
         res.json({ agents: agentsList });
     });
 
+    router.get('/storage', async (req, res) => {
+        try {
+            const uploadDir = path.join(process.cwd(), "data", "characters");
+            const files = await fs.promises.readdir(uploadDir);
+            res.json({ files });
+        } catch (error) {
+            res.status(500).json({ error: error.message });
+        }
+    });
+
     router.get("/agents/:agentId", (req, res) => {
         const { agentId } = validateUUIDParams(req.params, res) ?? {
             agentId: null,
@@ -127,7 +139,7 @@ export function createApiRouter(
         };
         if (!agentId) return;
 
-        const agent: AgentRuntime = agents.get(agentId);
+        let agent: AgentRuntime = agents.get(agentId);
 
         // update character
         if (agent) {
@@ -137,6 +149,9 @@ export function createApiRouter(
             // if it has a different name, the agentId will change
         }
 
+        // stores the json data before it is modified with added data
+        const characterJson = { ...req.body };
+
         // load character from body
         const character = req.body;
         try {
@@ -152,7 +167,7 @@ export function createApiRouter(
 
         // start it up (and register it)
         try {
-            await directClient.startAgent(character);
+            agent = await directClient.startAgent(character);
             elizaLogger.log(`${character.name} started`);
         } catch (e) {
             elizaLogger.error(`Error starting agent: ${e}`);
@@ -162,6 +177,19 @@ export function createApiRouter(
             });
             return;
         }
+
+        if(process.env.USE_CHARACTER_STORAGE === 'true') {
+            let filename = '';
+            try {
+                filename = `${agent.agentId}.json`;
+                const uploadDir = path.join(process.cwd(), "data", "characters");
+                const filepath = path.join(uploadDir, filename);
+                await fs.promises.writeFile(filepath, JSON.stringify({ ...characterJson, id: agent.agentId }, null, 2));
+            } catch (error) {
+                console.log("error:", error.message);
+            }
+        }
+
         res.json({
             id: character.id,
             character: character,