@agent-smith/server 0.0.4 → 0.0.6

package/dist/routes/agents.js

@@ -4,7 +4,8 @@ function getAgentsRoute(r) {
         const agents = db.readFeaturesType("agent");
         ctx.body = agents;
         ctx.status = 200;
-        await next();
+        //console.log("AGENTS", ctx.status);
+        //await next()
     });
 }
 function getAgentRoute(r) {
@@ -13,7 +14,7 @@ function getAgentRoute(r) {
         const taskSpec = fs.openTaskSpec(ctx.params.id, true);
         ctx.body = taskSpec.taskFileSpec;
         ctx.status = 200;
-        await next();
+        //await next()
     });
 }
 export { getAgentRoute, getAgentsRoute, };

package/dist/routes/conf.js

@@ -1,31 +1,16 @@
-import { createConfigFile, db, readConf, updateConfCmd } from '@agent-smith/cli';
+import { createConfigFile, updateConfCmd } from '@agent-smith/cli';
+import { getConfig } from '../utils.js';
 function getConfRoute(r) {
     r.get('/conf', async (ctx, next) => {
-        const fp = db.readFilePaths();
-        let confFilePath = "";
-        //let promptFilePath = "";
-        for (const p of fp) {
-            if (p.name == "conf") {
-                confFilePath = p.path;
-                break;
-            }
-        }
-        if (confFilePath == "") {
+        const { found, conf } = getConfig();
+        if (!found) {
             ctx.body = "can not find config path in db";
             ctx.status = 400;
         }
         else {
-            const { found, data } = readConf(confFilePath);
-            if (!found) {
-                ctx.status = 400;
-                ctx.body = "config file not found at " + confFilePath;
-            }
-            else {
-                ctx.body = data;
-                ctx.status = 200;
-            }
+            ctx.body = conf;
+            ctx.status = 200;
         }
-        await next();
     });
 }
 function createConfRoute(r) {
@@ -44,7 +29,6 @@ function createConfRoute(r) {
             ctx.status = 201;
             await updateConfCmd([cfp]);
         }
-        await next();
     });
 }
 export { createConfRoute, getConfRoute };

package/dist/routes/folders.d.ts

@@ -0,0 +1,3 @@
+import type Router from '@koa/router';
+declare function addFolderRoute(r: Router): void;
+export { addFolderRoute, };

package/dist/routes/folders.js

@@ -0,0 +1,27 @@
+import { updateConfCmd, updateConfigFile, init } from '@agent-smith/cli';
+import { getConfig } from '../utils.js';
+function addFolderRoute(r) {
+    r.post('/folders/add', async (ctx, next) => {
+        const payload = ctx.request.body;
+        await init();
+        const { found, conf, path } = getConfig();
+        if (!found) {
+            throw new Error("no config file found");
+        }
+        ;
+        if (!conf?.features) {
+            conf.features = [];
+        }
+        for (const p of payload) {
+            conf.features.push(p);
+        }
+        ;
+        console.log("Updating config file at", path);
+        console.dir(conf, { depth: 3 });
+        updateConfigFile(conf, path);
+        console.log("Updating db features from config file");
+        await updateConfCmd([path]);
+        ctx.status = 202;
+    });
+}
+export { addFolderRoute, };

package/dist/routes/index.js

@@ -5,5 +5,7 @@ import { getAgentRoute, getAgentsRoute } from "./agents.js";
 import { createConfRoute, getConfRoute } from "./conf.js";
 import { getToolsRoute } from "./tools.js";
 import { getStateRoute } from "./state.js";
-const baseRoutes = new Array(getConfRoute, getTasksRoute, getTaskRoute, getAgentRoute, getAgentsRoute, getModelsCmd, getToolsRoute, getTaskSettingsCmd, getStateRoute, createConfRoute, updateTaskSettingsCmd);
+import { installPluginRoute } from "./plugins.js";
+import { addFolderRoute } from "./folders.js";
+const baseRoutes = new Array(getConfRoute, getTasksRoute, getTaskRoute, getAgentRoute, getAgentsRoute, getModelsCmd, getToolsRoute, getTaskSettingsCmd, getStateRoute, createConfRoute, updateTaskSettingsCmd, installPluginRoute, addFolderRoute);
 export { baseRoutes };

package/dist/routes/models.js

@@ -8,7 +8,6 @@ function getModelsCmd(r) {
         catch (e) {
             ctx.body = "error reading the models";
             ctx.status = 502;
-            await next();
             return;
         }
         const ms = {};
@@ -32,7 +31,6 @@ function getModelsCmd(r) {
         });
         ctx.body = ms;
         ctx.status = 200;
-        await next();
     });
 }
 export { getModelsCmd, };

package/dist/routes/plugins.d.ts

@@ -0,0 +1,3 @@
+import type Router from '@koa/router';
+declare function installPluginRoute(r: Router): void;
+export { installPluginRoute, };

package/dist/routes/plugins.js

@@ -0,0 +1,33 @@
+import { execute, updateConfCmd, updateConfigFile, init } from '@agent-smith/cli';
+import { getConfig } from '../utils.js';
+function installPluginRoute(r) {
+    r.post('/plugins/install', async (ctx, next) => {
+        const payload = ctx.request.body;
+        console.log("P", payload);
+        for (const p of payload) {
+            console.log("Installing", p, "plugin");
+            const res = await execute("npm", ["i", "-g", p]);
+            console.log(p, res);
+        }
+        await init();
+        const { found, conf, path } = getConfig();
+        if (!found) {
+            throw new Error("no config file found");
+        }
+        ;
+        if (!conf?.plugins) {
+            conf.plugins = [];
+        }
+        for (const p of payload) {
+            conf.plugins?.push(p);
+        }
+        ;
+        console.log("Updating config file at", path);
+        console.dir(conf, { depth: 3 });
+        updateConfigFile(conf, path);
+        console.log("Updating db features from config file");
+        await updateConfCmd([path]);
+        ctx.status = 202;
+    });
+}
+export { installPluginRoute, };

package/dist/routes/state.js

@@ -2,9 +2,10 @@ import { getConfigPath, init } from '@agent-smith/cli';
 import fs from "node:fs";
 function getStateRoute(r) {
     r.get('/state', async (ctx, next) => {
-        console.log("STATE ROUTE");
+        //console.log('STATE URL --> ' + ctx.request.url);
+        //console.log("STATE ROUTE");
         const { confDir, dbPath } = getConfigPath("agent-smith", "config.db");
-        console.log("conf paths", confDir, dbPath);
+        //console.log("conf paths", confDir, dbPath);
         if (!fs.existsSync(dbPath)) {
             ctx.body = "no db found at " + dbPath;
             ctx.status = 202;
@@ -13,7 +14,8 @@ function getStateRoute(r) {
             ctx.status = 200;
             await init();
         }
-        await next();
+        //console.log("STATE", ctx.status)
+        //return next()
     });
 }
 export { getStateRoute, };

package/dist/routes/task_settings.js

@@ -4,7 +4,6 @@ function getTaskSettingsCmd(r) {
         const ts = db.getTaskSettings();
         ctx.body = ts;
         ctx.status = 200;
-        await next();
     });
 }
 function updateTaskSettingsCmd(r) {
@@ -17,7 +16,6 @@ function updateTaskSettingsCmd(r) {
         db.getTaskSettings(true);
         ctx.body = ts;
         ctx.status = 200;
-        await next();
     });
 }
 export { getTaskSettingsCmd, updateTaskSettingsCmd, };

package/dist/routes/tasks.js

@@ -4,7 +4,6 @@ function getTasksRoute(r) {
         const tasks = db.readFeaturesType("task");
         ctx.body = tasks;
         ctx.status = 200;
-        await next();
     });
 }
 function getTaskRoute(r) {
@@ -13,7 +12,6 @@ function getTaskRoute(r) {
         const taskSpec = fs.openTaskSpec(ctx.params.id);
         ctx.body = taskSpec.taskFileSpec;
         ctx.status = 200;
-        await next();
     });
 }
 function saveTasksRoute(r) {
@@ -21,7 +19,6 @@ function saveTasksRoute(r) {
         //const payload = ctx.request.body;
         ctx.body = "ok";
         ctx.status = 200;
-        await next();
     });
 }
 export { getTasksRoute, getTaskRoute, };

package/dist/routes/tools.js

@@ -24,7 +24,6 @@ function getToolsRoute(r) {
             ctx.body = tools;
             ctx.status = 200;
         }
-        await next();
     });
 }
 export { getToolsRoute, };

package/dist/server/server.js

@@ -21,17 +21,18 @@ if (argv.length > 2) {
 }*/
 const logger = async (ctx, next) => {
     const start = Date.now();
+    //console.log('LOGGER BEFORE URL --> ' + ctx.request.url);
     await next();
     const duration = Date.now() - start;
     console.log(`${ctx.method} ${ctx.url} - ${ctx.status} - ${duration}ms`);
+    //console.log('LOGGER AFTER URL --> ' + ctx.request.url);
 };
 const app = websockify(new Koa());
 app.use(bodyParser());
-app.use(logger);
 app.use(cors({
     credentials: true
 }));
-app.ws.use(function (ctx, next) {
+app.ws.use((ctx, next) => {
     return next();
 });
 function createAwaiter() {
@@ -48,9 +49,6 @@ function createAwaiter() {
     };
 }
 function runserver(routes, staticDir) {
-    if (staticDir) {
-        app.use(serve(staticDir));
-    }
     const router = useRouter(routes);
     const sendTokensInterval = 100;
     const confirmToolCalls = {};
@@ -326,17 +324,22 @@ function runserver(routes, staticDir) {
             abort = new AbortController();
         });
     }));
+    if (staticDir) {
+        app.use(serve(staticDir));
+    }
+    app.use(logger);
     app.use(router.routes()).use(router.allowedMethods());
     // 404 middleware - runs after router
-    /*app.use((ctx) => {
-      if (!ctx.matched || ctx.matched.length === 0) {
-        ctx.status = 404;
-        ctx.body = {
-          error: 'Not Found',
-          path: ctx.path
-        };
-      }
-    });*/
+    app.use((ctx) => {
+        if (!ctx.matched || ctx.matched.length === 0) {
+            ctx.status = 404;
+            //console.log("404 ROUTE", ctx);
+            ctx.body = {
+                error: 'Not Found',
+                path: ctx.path
+            };
+        }
+    });
     app.listen(5184, () => {
         console.log('Please open url localhost:5184 in a browser');
     });

package/dist/utils.d.ts

@@ -0,0 +1,7 @@
+import type { ConfigFile } from "@agent-smith/types";
+declare function getConfig(): {
+    found: boolean;
+    conf: ConfigFile;
+    path: string;
+};
+export { getConfig, };

package/dist/utils.js

@@ -0,0 +1,18 @@
+import { db, readConf } from "@agent-smith/cli";
+function getConfig() {
+    const fp = db.readFilePaths();
+    let confFilePath = "";
+    //let promptFilePath = "";
+    for (const p of fp) {
+        if (p.name == "conf") {
+            confFilePath = p.path;
+            break;
+        }
+    }
+    if (confFilePath == "") {
+        return { found: false, conf: {}, path: "" };
+    }
+    const c = readConf(confFilePath);
+    return { found: c.found, conf: c.data, path: confFilePath };
+}
+export { getConfig, };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@agent-smith/server",
-  "version": "0.0.4",
+  "version": "0.0.6",
   "description": "Agent Smith Nodejs server",
   "scripts": {
     "test": "echo \"Error: no test specified\" && exit 1",
@@ -10,7 +10,7 @@
   },
   "type": "module",
   "dependencies": {
-    "@agent-smith/cli": "^0.0.113",
+    "@agent-smith/cli": "^0.0.114",
     "@koa/cors": "^5.0.0",
     "@koa/router": "^15.3.0",
     "ansi-colors": "^4.1.3",