@agent-smith/server 0.0.2 → 0.0.3

package/dist/index.d.ts

@@ -0,0 +1,2 @@
+#!/usr/bin/env node
+export {};

package/dist/index.js

@@ -0,0 +1,19 @@
+#!/usr/bin/env node
+import { fileURLToPath } from 'url';
+import path from 'path';
+import { runserver } from './server/server.js';
+import { baseRoutes } from "./routes/index.js";
+const __filename = fileURLToPath(import.meta.url);
+const dirpath = path.resolve(path.dirname(__filename), "../");
+async function main() {
+    //console.log(dirpath, process.env.NODE_ENV);
+    let staticPath = undefined;
+    if (process.env.NODE_ENV != "development") {
+        staticPath = dirpath;
+    }
+    //console.log("B", baseRoutes);
+    runserver(baseRoutes, staticPath);
+}
+(async () => {
+    await main();
+})();

package/dist/routes/agents.d.ts

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

package/dist/routes/agents.js

@@ -0,0 +1,19 @@
+import { db, fs } from '@agent-smith/cli';
+function getAgentsRoute(r) {
+    r.get('/agents', async (ctx, next) => {
+        const agents = db.readFeaturesType("agent");
+        ctx.body = agents;
+        ctx.status = 200;
+        await next();
+    });
+}
+function getAgentRoute(r) {
+    r.get('/agent/:id', async (ctx, next) => {
+        //console.log(ctx.params.id)
+        const taskSpec = fs.openTaskSpec(ctx.params.id, true);
+        ctx.body = taskSpec.taskFileSpec;
+        ctx.status = 200;
+        await next();
+    });
+}
+export { getAgentRoute, getAgentsRoute, };

package/dist/routes/conf.d.ts

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

package/dist/routes/conf.js

@@ -0,0 +1,50 @@
+import { createConfigFile, db, readConf, updateConfCmd } from '@agent-smith/cli';
+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 == "") {
+            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;
+            }
+        }
+        await next();
+    });
+}
+function createConfRoute(r) {
+    r.get('/conf/create', async (ctx, next) => {
+        let cfp = null;
+        try {
+            cfp = createConfigFile();
+        }
+        catch (e) {
+            console.error("500", e);
+            ctx.body = e;
+            ctx.status = 500;
+        }
+        if (cfp) {
+            ctx.body = cfp;
+            ctx.status = 201;
+            await updateConfCmd([cfp]);
+        }
+        await next();
+    });
+}
+export { createConfRoute, getConfRoute };

package/dist/routes/index.js

@@ -1,4 +1,9 @@
 import { getModelsCmd } from "./models.js";
 import { getTaskSettingsCmd, updateTaskSettingsCmd } from "./task_settings.js";
-const baseRoutes = new Array(getModelsCmd, getTaskSettingsCmd, updateTaskSettingsCmd);
+import { getTaskRoute, getTasksRoute } from "./tasks.js";
+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);
 export { baseRoutes };

package/dist/routes/state.d.ts

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

package/dist/routes/state.js

@@ -0,0 +1,19 @@
+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");
+        const { confDir, dbPath } = getConfigPath("agent-smith", "config.db");
+        console.log("conf paths", confDir, dbPath);
+        if (!fs.existsSync(dbPath)) {
+            ctx.body = "no db found at " + dbPath;
+            ctx.status = 202;
+        }
+        else {
+            ctx.status = 200;
+            await init();
+        }
+        await next();
+    });
+}
+export { getStateRoute, };

package/dist/routes/tasks.d.ts

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

package/dist/routes/tasks.js

@@ -0,0 +1,27 @@
+import { db, fs } from '@agent-smith/cli';
+function getTasksRoute(r) {
+    r.get('/tasks', async (ctx, next) => {
+        const tasks = db.readFeaturesType("task");
+        ctx.body = tasks;
+        ctx.status = 200;
+        await next();
+    });
+}
+function getTaskRoute(r) {
+    r.get('/task/:id', async (ctx, next) => {
+        //console.log(ctx.params.id)
+        const taskSpec = fs.openTaskSpec(ctx.params.id);
+        ctx.body = taskSpec.taskFileSpec;
+        ctx.status = 200;
+        await next();
+    });
+}
+function saveTasksRoute(r) {
+    r.post('/save', async (ctx, next) => {
+        //const payload = ctx.request.body;
+        ctx.body = "ok";
+        ctx.status = 200;
+        await next();
+    });
+}
+export { getTasksRoute, getTaskRoute, };

package/dist/routes/tools.d.ts

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

package/dist/routes/tools.js

@@ -0,0 +1,30 @@
+import { db } from '@agent-smith/cli';
+function getToolsRoute(r) {
+    r.post('/tools', async (ctx, next) => {
+        const payload = ctx.request.body;
+        const tools = new Array();
+        let nf = false;
+        for (const tn of payload) {
+            const { found, tool, type } = db.readTool(tn);
+            if (!found) {
+                ctx.status = 400;
+                ctx.body = `Tool ${tn} not found`;
+                nf = true;
+                break;
+            }
+            ;
+            const ts = {
+                name: tool.name,
+                description: tool.description,
+                arguments: tool.arguments,
+            };
+            tools.push({ def: ts, type: type });
+        }
+        if (!nf) {
+            ctx.body = tools;
+            ctx.status = 200;
+        }
+        await next();
+    });
+}
+export { getToolsRoute, };

package/dist/server/router.js

@@ -2,6 +2,7 @@ import Router from '@koa/router';
 const useRouter = (routes = []) => {
     const router = new Router();
     routes.forEach((f) => f(router));
+    //console.log("R", router.routes())
     router.get('/ping', async (ctx, next) => {
         ctx.body = { ok: true };
         ctx.status = 200;

package/dist/server/server.js

@@ -34,7 +34,7 @@ app.use(cors({
 app.ws.use(function (ctx, next) {
     return next();
 });
-function createManualPromise() {
+function createAwaiter() {
     let resolveFn;
     let rejectFn;
     const promise = new Promise((resolve, reject) => {
@@ -109,19 +109,43 @@ function runserver(routes, staticDir) {
                         buf += t;
                         process.stdout.write(t);
                     };
+                    msg.options.onTurnEnd = (ht) => {
+                        const rsm = {
+                            type: "turnend",
+                            msg: JSON.stringify(ht),
+                        };
+                        ctx.websocket.send(JSON.stringify(rsm));
+                    };
+                    msg.options.onAssistant = (txt) => {
+                        const rsm = {
+                            type: "assistant",
+                            msg: txt,
+                        };
+                        ctx.websocket.send(JSON.stringify(rsm));
+                    };
+                    msg.options.onThink = (txt) => {
+                        const rsm = {
+                            type: "think",
+                            msg: txt,
+                        };
+                        ctx.websocket.send(JSON.stringify(rsm));
+                    };
                     try {
                         const it = setInterval(() => {
                             const rsm = {
                                 type: "token",
                                 msg: buf,
                             };
-                            ctx.websocket.send(JSON.stringify(rsm));
+                            if (buf !== "") {
+                                // model is loading if no buffer
+                                ctx.websocket.send(JSON.stringify(rsm));
+                            }
                             buf = "";
                         }, sendTokensInterval);
                         const res = await executeTask(msg.command, msg.payload, msg.options);
-                        setTimeout(() => {
-                            clearInterval(it);
-                        }, sendTokensInterval);
+                        //setTimeout(() => {
+                        clearInterval(it);
+                        //}, sendTokensInterval);
                         //console.dir(res, { depth: 3 });
                         let r;
                         if (res.template.id == "none") {
@@ -171,6 +195,15 @@ function runserver(routes, staticDir) {
                             type: "assistant",
                             msg: txt,
                         };
+                        //console.log("SRV ON ASSISTANT", txt);
+                        ctx.websocket.send(JSON.stringify(rsm));
+                    };
+                    msg.options.onThink = (txt) => {
+                        const rsm = {
+                            type: "think",
+                            msg: txt,
+                        };
+                        //console.log("SRV ON THINK", txt);
                         ctx.websocket.send(JSON.stringify(rsm));
                     };
                     msg.options.onToolCall = (tc) => {
@@ -212,7 +245,7 @@ function runserver(routes, staticDir) {
                             type: "toolcallconfirm",
                             msg: JSON.stringify(tc),
                         };
-                        const { promise, resolve } = createManualPromise();
+                        const { promise, resolve } = createAwaiter();
                         confirmToolCalls[tc.id] = resolve;
                         ctx.websocket.send(JSON.stringify(rsm));
                         const res = await promise;
@@ -238,9 +271,9 @@ function runserver(routes, staticDir) {
                             buf = "";
                         }, sendTokensInterval);
                         const res = await executeTask(msg.command, msg.payload, msg.options);
-                        setTimeout(() => {
-                            clearInterval(it);
-                        }, sendTokensInterval);
+                        //setTimeout(() => {
+                        clearInterval(it);
+                        //}, sendTokensInterval);
                         const ht = JSON.stringify(res.template.history.pop());
                         //console.log("FINAL MSG", ht)
                         const rsm = {
@@ -294,8 +327,18 @@ function runserver(routes, staticDir) {
         });
     }));
     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.listen(5184, () => {
-        console.log('Please open localhost:5184 in a browser');
+        console.log('Please open url localhost:5184 in a browser');
     });
 }
 export { runserver };

package/package.json

@@ -1,29 +1,28 @@
 {
   "name": "@agent-smith/server",
-  "version": "0.0.2",
+  "version": "0.0.3",
   "description": "Agent Smith Nodejs server",
   "scripts": {
     "test": "echo \"Error: no test specified\" && exit 1",
     "build": "rm -rf dist && tsc",
-    "start": "node dist/index.js",
-    "server": "node --loader ts-node/esm src/index.ts",
+    "server": "node dist/index.js",
     "devserver": "node  --loader ts-node/esm --watch src/index.ts"
   },
   "type": "module",
   "dependencies": {
-    "@agent-smith/cli": "^0.0.111",
+    "@agent-smith/cli": "^0.0.112",
     "@koa/cors": "^5.0.0",
     "@koa/router": "^15.3.0",
     "ansi-colors": "^4.1.3",
-    "koa": "^3.1.1",
+    "koa": "^3.1.2",
     "koa-bodyparser": "^4.4.1",
     "koa-route": "^4.0.1",
     "koa-static": "^5.0.0",
     "koa-websocket": "^7.0.0"
   },
   "devDependencies": {
-    "@agent-smith/types": "^0.0.2",
-    "@locallm/types": "^0.7.0",
+    "@agent-smith/types": "^0.0.3",
+    "@locallm/types": "^0.7.1",
     "@types/better-sqlite3": "^7.6.13",
     "@types/koa": "^3.0.1",
     "@types/koa__cors": "^5.0.1",
@@ -31,7 +30,7 @@
     "@types/koa-route": "^3.2.9",
     "@types/koa-static": "^4.0.4",
     "@types/koa-websocket": "^5.0.11",
-    "@types/node": "^25.3.0",
+    "@types/node": "^25.3.1",
     "ts-node": "^10.9.2",
     "tslib": "2.8.1",
     "typescript": "^5.9.3"