@agent-smith/server 0.0.8 → 0.0.9

package/dist/routes/index.js

@@ -1,6 +1,5 @@
-import { getModelsCmd } from "./models.js";
-import { getTaskSettingsCmd, updateTaskSettingsCmd } from "./task_settings.js";
-import { getTaskRoute, getTasksRoute } from "./tasks.js";
+import { getModelsRoute, getModelsPresetsRoute, upsertModelPresetRoute, delModelPresetRoute } from "./models.js";
+import { getAgentSettingsCmd, updateAgentSettingsCmd } from "./agent_settings.js";
 import { getAgentRoute, getAgentsRoute } from "./agents.js";
 import { createConfRoute, getConfRoute } from "./conf.js";
 import { getToolsRoute } from "./tools.js";
@@ -10,5 +9,8 @@ import { addFolderRoute } from "./folders.js";
 import { getWorkflowRoute, getWorkflowsRoute } from "./workflows.js";
 import { getBackendsRoute, setBackendRoute } from "./backends.js";
 import { getOrCreateAppConfigFileRoute, updateAppConfigFileRoute } from "./apps.js";
-const baseRoutes = new Array(getConfRoute, getTasksRoute, getTaskRoute, getAgentRoute, getAgentsRoute, getModelsCmd, getToolsRoute, getTaskSettingsCmd, getStateRoute, createConfRoute, updateTaskSettingsCmd, installPluginRoute, addFolderRoute, getWorkflowRoute, getWorkflowsRoute, getBackendsRoute, setBackendRoute, getOrCreateAppConfigFileRoute, updateAppConfigFileRoute);
+import { getWorkspaceRoute, updateDefaultWorkspaceRoute, upsertWorkspaceRoute } from "./workspace.js";
+import { getSettingsRoute } from "./settings.js";
+import { applyTemplateRoute } from "./templates.js";
+const baseRoutes = new Array(getConfRoute, getAgentRoute, getAgentsRoute, getModelsRoute, getModelsPresetsRoute, getToolsRoute, getAgentSettingsCmd, getStateRoute, createConfRoute, updateAgentSettingsCmd, installPluginRoute, addFolderRoute, getWorkflowRoute, getWorkflowsRoute, getBackendsRoute, setBackendRoute, getOrCreateAppConfigFileRoute, updateAppConfigFileRoute, getWorkspaceRoute, upsertWorkspaceRoute, updateDefaultWorkspaceRoute, getSettingsRoute, upsertModelPresetRoute, delModelPresetRoute, applyTemplateRoute);
 export { baseRoutes };

package/dist/routes/models.d.ts

@@ -1,3 +1,6 @@
 import type Router from '@koa/router';
-declare function getModelsCmd(r: Router): void;
-export { getModelsCmd, };
+declare function getModelsRoute(r: Router): void;
+declare function getModelsPresetsRoute(r: Router): void;
+declare function upsertModelPresetRoute(r: Router): void;
+declare function delModelPresetRoute(r: Router): void;
+export { getModelsRoute, getModelsPresetsRoute, upsertModelPresetRoute, delModelPresetRoute, };

package/dist/routes/models.js

@@ -1,36 +1,73 @@
-import { backend } from '@agent-smith/cli';
-function getModelsCmd(r) {
-    r.get('/models', async (ctx, next) => {
-        let mi;
+import { state } from '@agent-smith/core';
+import { db } from '@agent-smith/core';
+function getModelsRoute(r) {
+    r.get('/models/:backend', async (ctx, next) => {
+        const backend = ctx.params?.backend;
+        if (!Object.keys(state.backends).includes(backend)) {
+            ctx.body = `backend ${backend} not found in config`;
+            ctx.status = 400;
+        }
+        const b = state.backends[backend];
+        //console.log("MB", backend, "/", b);
+        //console.log("SB", state.backends);
+        let mi = new Array();
+        try {
+            mi = await b.modelsInfo() ?? [];
+            //console.log("M", mi);
+        }
+        catch (e) {
+            ctx.body = `error reading the models from backend ${backend}\n${e}`;
+            ctx.status = 502;
+            return;
+        }
+        ctx.body = mi;
+        ctx.status = 200;
+    });
+}
+function getModelsPresetsRoute(r) {
+    r.get('/models/presets/read', async (ctx, next) => {
+        let mp = new Array();
         try {
-            mi = await backend.value?.modelsInfo();
+            //console.log("READ SP");
+            mp = db.readSamplingPresets();
+            //console.log("M", mp);
         }
         catch (e) {
-            ctx.body = "error reading the models";
+            ctx.body = "error reading the models presets";
             ctx.status = 502;
             return;
         }
-        const ms = {};
-        mi?.forEach(m => {
-            let prevArg = "";
-            let ctx = 0;
-            if (backend.value?.providerType == "openai") {
-                ms[m.name] = { status: "unknown", ctx: m?.ctx ?? ctx };
-            }
-            else {
-                //@ts-ignore            
-                m.extra.status.args.forEach(a => {
-                    if (prevArg == "--ctx-size") {
-                        ctx = parseInt(a);
-                    }
-                    //@ts-ignore
-                    ms[m.name] = { status: m.extra.status.value, ctx: ctx };
-                    prevArg = a;
-                });
-            }
-        });
-        ctx.body = ms;
+        ctx.body = mp;
         ctx.status = 200;
     });
 }
-export { getModelsCmd, };
+function upsertModelPresetRoute(r) {
+    r.post('/models/preset/update', async (ctx, next) => {
+        const payload = ctx.request.body;
+        try {
+            db.upsertSamplingPreset(payload);
+            ctx.status = 204;
+        }
+        catch (e) {
+            const err = `error updating model preset:\n ${e}`;
+            console.error(err);
+            ctx.body = err;
+            ctx.status = 500;
+        }
+    });
+}
+function delModelPresetRoute(r) {
+    r.del('/models/preset/delete/:name', async (ctx, next) => {
+        const name = ctx.params?.name;
+        if (!name) {
+            ctx.body = "provide a name to delete model preset";
+            ctx.status = 400;
+        }
+        else {
+            const w = db.deleteSamplingPreset(name);
+            ctx.body = w;
+            ctx.status = 200;
+        }
+    });
+}
+export { getModelsRoute, getModelsPresetsRoute, upsertModelPresetRoute, delModelPresetRoute, };

package/dist/routes/plugins.js

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

package/dist/routes/settings.d.ts

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

package/dist/routes/settings.js

@@ -0,0 +1,20 @@
+import { db } from '@agent-smith/core';
+function getSettingsRoute(r) {
+    r.get('/settings', async (ctx, next) => {
+        let res = {};
+        try {
+            res = db.readSettings();
+            //console.log("M", mi);
+        }
+        catch (e) {
+            const err = `error reading settings ${e}`;
+            console.error(err);
+            ctx.body = err;
+            ctx.status = 500;
+            return;
+        }
+        ctx.body = res;
+        ctx.status = 200;
+    });
+}
+export { getSettingsRoute, };

package/dist/routes/state.js

@@ -1,18 +1,18 @@
-import { getConfigPath, init } from '@agent-smith/cli';
+import { conf, state } from '@agent-smith/core';
 import fs from "node:fs";
 import { getConfig } from '../utils.js';
 function getStateRoute(r) {
     r.get('/state', async (ctx, next) => {
         //console.log('STATE URL --> ' + ctx.request.url);
         //console.log("STATE ROUTE");
-        const { confDir, dbPath } = getConfigPath("agent-smith", "config.db");
+        const { dbPath } = conf.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 {
-            await init();
+            await state.init();
             const { found, conf } = getConfig();
             if (!found) {
                 ctx.body = "can not find config path in db";

package/dist/routes/templates.d.ts

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

package/dist/routes/templates.js

@@ -0,0 +1,18 @@
+import { useAgentExecutor } from '@agent-smith/core';
+function applyTemplateRoute(r) {
+    r.post('/templates/apply', async (ctx, next) => {
+        const payload = ctx.request.body;
+        const ex = await useAgentExecutor("infer", { prompt: "" }, {});
+        try {
+            const res = await ex.agent.lm.applyTemplate(payload.messages, payload.modelName);
+            ctx.status = 200;
+            ctx.body = res.prompt;
+        }
+        catch (e) {
+            console.error(e);
+            ctx.status = 500;
+            ctx.body = e.toString();
+        }
+    });
+}
+export { applyTemplateRoute, };

package/dist/routes/tools.js

@@ -1,11 +1,11 @@
-import { db } from '@agent-smith/cli';
+import { db } from '@agent-smith/core';
 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);
+            const { found, tool } = db.readTool(tn);
             if (!found) {
                 ctx.status = 400;
                 ctx.body = `Tool ${tn} not found`;
@@ -18,7 +18,7 @@ function getToolsRoute(r) {
                 description: tool.description,
                 arguments: tool.arguments,
             };
-            tools.push({ def: ts, type: type });
+            tools.push({ def: ts, type: tool.type });
         }
         if (!nf) {
             ctx.body = tools;

package/dist/routes/workflows.js

@@ -1,4 +1,4 @@
-import { db, fs } from '@agent-smith/cli';
+import { db, fs } from '@agent-smith/core';
 function getWorkflowsRoute(r) {
     r.get('/workflows', async (ctx, next) => {
         const w = db.readFeaturesType("workflow");

package/dist/routes/workspace.d.ts

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

package/dist/routes/workspace.js

@@ -0,0 +1,50 @@
+import { db } from '@agent-smith/core';
+function getWorkspaceRoute(r) {
+    r.get('/workspace', async (ctx, next) => {
+        let ws;
+        try {
+            ws = db.readWorkspaces();
+            //console.log("M", mi);
+        }
+        catch (e) {
+            const err = `error reading the workspaces ${e}`;
+            console.error(err);
+            ctx.body = err;
+            ctx.status = 500;
+            return;
+        }
+        ctx.body = ws;
+        ctx.status = 200;
+    });
+}
+function upsertWorkspaceRoute(r) {
+    r.post('/workspace', async (ctx, next) => {
+        const payload = ctx.request.body;
+        try {
+            db.upsertWorkspace(payload);
+            ctx.status = 204;
+        }
+        catch (e) {
+            const err = `error writing workspace file path into db:\n ${e}`;
+            console.error(err);
+            ctx.body = err;
+            ctx.status = 500;
+        }
+    });
+}
+function updateDefaultWorkspaceRoute(r) {
+    r.post('/workspace/update', async (ctx, next) => {
+        const payload = ctx.request.body;
+        try {
+            db.upsertSetting("workspace", payload.name);
+            ctx.status = 204;
+        }
+        catch (e) {
+            const err = `error updating default workspace:\n ${e}`;
+            console.error(err);
+            ctx.body = err;
+            ctx.status = 500;
+        }
+    });
+}
+export { getWorkspaceRoute, upsertWorkspaceRoute, updateDefaultWorkspaceRoute, };

package/dist/server/router.d.ts

@@ -1,3 +1,6 @@
 import Router from '@koa/router';
-declare const useRouter: (routes?: Array<(r: Router) => void>) => Router<import("koa").DefaultState, import("koa").DefaultContext>;
+declare const useRouter: (apiRoutes?: Array<(r: Router) => void>) => {
+    baseRouter: Router<import("koa").DefaultState, import("koa").DefaultContext>;
+    apiRouter: import("@koa/router").RouterWithMethods<string, import("koa").DefaultState, import("koa").DefaultContext>;
+};
 export { useRouter };

package/dist/server/router.js

@@ -1,13 +1,15 @@
 import Router from '@koa/router';
-const useRouter = (routes = []) => {
+const useRouter = (apiRoutes = []) => {
+    const apiRouter = new Router({ prefix: '/api' });
     const router = new Router();
-    routes.forEach((f) => f(router));
-    //console.log("R", router.routes())
+    apiRoutes.forEach((f) => f(apiRouter));
     router.get('/ping', async (ctx, next) => {
         ctx.body = { ok: true };
         ctx.status = 200;
         await next();
     });
-    return router;
+    //console.log("Routes", router.stack.map(i => i.path));
+    //console.log("Apiroutes", apiRouter.stack.map(i => i.path));
+    return { baseRouter: router, apiRouter };
 };
 export { useRouter };