@agent-smith/cli 0.0.6 → 0.0.8

package/dist/agent.js

@@ -1,6 +1,5 @@
 import { useAgentBrain } from "@agent-smith/brain";
 import { LmTaskBuilder } from "@agent-smith/lmtask";
-;
 import { marked } from 'marked';
 import { markedTerminal } from 'marked-terminal';
 marked.use(markedTerminal());
@@ -9,8 +8,8 @@ const modelsForExpert = {};
 const taskBuilder = new LmTaskBuilder(brain);
 async function initExperts() {
     brain.experts.forEach((ex) => {
-        ex.setOnStartEmit(() => console.log(""));
-        ex.setOnToken((t) => {
+        ex.backend.setOnStartEmit(() => console.log(""));
+        ex.backend.setOnToken((t) => {
             process.stdout.write(t);
         });
     });
@@ -18,9 +17,8 @@ async function initExperts() {
 async function initAgent(mode, isVerbose = false) {
     if (!brain.state.get().isOn) {
         brain.resetExperts();
-        await brain.discoverLocal();
+        await brain.initLocal();
         await initExperts();
-        await brain.expertsForModelsInfo();
     }
     const brainUp = brain.state.get().isOn;
     if (isVerbose) {

package/dist/cmd/clicmds/cmds.d.ts

@@ -1,5 +1,6 @@
 import { Cmd } from "../../interfaces.js";
 declare let cmds: Record<string, Cmd>;
+declare function initAliases(): Record<string, Cmd>;
 declare function initCmds(): Promise<Record<string, Cmd>>;
 declare function pingCmd(args: Array<string> | undefined, options: any): Promise<boolean>;
-export { cmds, initCmds, pingCmd };
+export { cmds, initCmds, pingCmd, initAliases };

package/dist/cmd/clicmds/cmds.js

@@ -1,7 +1,7 @@
 import { formatMode, initFeatures, runMode } from "../../state/state.js";
 import { getFeatureSpec, readFeaturesDirs } from "../../state/features.js";
-import { readFeatures } from "../../db/read.js";
-import { updateFeatures } from "../../db/write.js";
+import { readAliases, readFeatures } from "../../db/read.js";
+import { updateAliases, updateFeatures } from "../../db/write.js";
 import { updateConf } from "../../conf.js";
 import { executeActionCmd } from "../lib/execute_action.js";
 import { initAgent, marked, taskBuilder } from "../../agent.js";
@@ -57,6 +57,35 @@ let cmds = {
         description: "reparse the features dirs and update the list",
     }
 };
+function initAliases() {
+    const aliases = readAliases();
+    const _cmds = {};
+    aliases.forEach((alias) => {
+        switch (alias.type) {
+            case "task":
+                _cmds[alias.name] = {
+                    cmd: (args = [], options) => _executeTaskCmd([alias.name, ...args], options),
+                    description: "task: " + alias.name,
+                    args: "arguments: \n-args: prompt and other arguments if any for the task"
+                };
+                break;
+            case "action":
+                _cmds[alias.name] = {
+                    cmd: (args = [], options = {}, quiet = false) => executeActionCmd([alias.name, ...args], options, quiet),
+                    description: "action: " + alias.name,
+                    args: "arguments: \n-args: other arguments if any for the action"
+                };
+                break;
+            case "job":
+                _cmds[alias.name] = {
+                    cmd: (args = [], options) => _executeJobCmd([alias.name, ...args], options),
+                    description: "job: " + alias.name,
+                    args: "arguments: \n-args: other arguments if any for the job"
+                };
+        }
+    });
+    return _cmds;
+}
 async function initCmds() {
     for (const dirpath of new Set(Object.values(readFeatures().cmd))) {
         const c = await readCmds(`${dirpath}`);
@@ -84,6 +113,7 @@ async function _updateConfCmd(args = [], options) {
     const allPaths = await updateConf(args[0]);
     const feats = readFeaturesDirs(allPaths);
     updateFeatures(feats);
+    updateAliases(feats);
 }
 async function _readJobCmd(args = [], options) {
     if (args.length == 0) {
@@ -140,4 +170,4 @@ async function _readTaskCmd(args = [], options) {
 async function _listTasksCmd(args = [], options) {
     Object.keys(readFeatures().task).forEach((t) => console.log("-", t));
 }
-export { cmds, initCmds, pingCmd };
+export { cmds, initCmds, pingCmd, initAliases };

package/dist/cmd/cmds.js

@@ -2,7 +2,7 @@ import { Command } from "commander";
 import { lastCmd } from "../state/state.js";
 import { modes } from "./clicmds/modes.js";
 import { processOutput, setOptions } from "./lib/utils.js";
-import { cmds, initCmds } from "./clicmds/cmds.js";
+import { cmds, initAliases, initCmds } from "./clicmds/cmds.js";
 let cliCmds = {};
 async function initCliCmds() {
     cliCmds = await initCmds();
@@ -19,7 +19,8 @@ async function runCmd(cmdName, args = []) {
 }
 async function buildCmds() {
     const program = new Command();
-    for (const [name, spec] of Object.entries(cmds)) {
+    const aliases = initAliases();
+    for (const [name, spec] of Object.entries({ ...cmds, ...aliases })) {
         const cmd = program.command(name);
         const _cmd = async (args = [], options = {}) => {
             const _args = await setOptions(options, args);

package/dist/cmd/lib/execute_job.js

@@ -14,8 +14,8 @@ async function executeJobCmd(name, args = []) {
     await job.start();
     let params = args;
     let res = {};
+    brain.backendsForModelsInfo();
     for (const name of Object.keys(job.tasks)) {
-        brain.expertsForModelsInfo();
         try {
             res = await job.runTask(name, params);
             if ("text" in res) {

package/dist/cmd/lib/execute_task.js

@@ -1,4 +1,4 @@
-import { initAgent, taskBuilder } from "../../agent.js";
+import { brain, initAgent, taskBuilder } from "../../agent.js";
 import { getFeatureSpec } from "../../state/features.js";
 import { runMode } from "../../state/state.js";
 import { readTask } from "./utils.js";
@@ -13,6 +13,7 @@ async function executeTaskCmd(args = [], options = {}) {
     if (!res.found) {
         throw new Error(`Task ${name}, ${path} not found`);
     }
+    const taskSpec = taskBuilder.readFromYaml(res.ymlTask);
     const task = taskBuilder.fromYaml(res.ymlTask);
     const pr = args.shift();
     const vars = {};
@@ -22,6 +23,15 @@ async function executeTaskCmd(args = [], options = {}) {
             vars[t[0]] = t[1];
         }
     });
+    const ex = brain.getOrCreateExpertForModel(taskSpec.model.name, taskSpec.template.name);
+    if (!ex) {
+        throw new Error("No expert found for model " + taskSpec.model.name);
+    }
+    ex.checkStatus();
+    ex.backend.setOnToken((t) => {
+        process.stdout.write(t);
+    });
+    vars["expert"] = ex;
     console.log("Ingesting prompt ...");
     const data = await task.run({ prompt: pr, ...vars });
     if (data?.error) {

package/dist/cmd/sys/read_features.d.ts

@@ -1,3 +1,3 @@
-import { Features } from "bin/interfaces";
+import { Features } from "../../interfaces.js";
 declare function readFeaturesDir(dir: string): Features;
 export { readFeaturesDir };

package/dist/db/db.js

@@ -1,7 +1,7 @@
 import DatabaseConstructor from "better-sqlite3";
 import { schemas } from "./schemas.js";
 import { dbPath } from "../conf.js";
-let db;
+let db = new DatabaseConstructor("x", { fileMustExist: false });
 function initDb(isVerbose = false) {
     db = new DatabaseConstructor(dbPath);
     schemas.forEach((s) => {

package/dist/db/read.d.ts

@@ -1,9 +1,13 @@
-import { FeatureSpec, FeatureType } from "../interfaces.js";
+import { AliasType, FeatureSpec, FeatureType } from "../interfaces.js";
 declare function readFeaturePaths(): Array<string>;
 declare function readPlugins(): Array<Record<string, string>>;
 declare function readFeatures(): Record<FeatureType, Record<string, string>>;
+declare function readAliases(): Array<{
+    name: string;
+    type: AliasType;
+}>;
 declare function readFeature(name: string, type: FeatureType): {
     found: boolean;
     feature: FeatureSpec;
 };
-export { readFeatures, readFeaturePaths, readFeature, readPlugins };
+export { readFeatures, readFeaturePaths, readFeature, readPlugins, readAliases };

package/dist/db/read.js

@@ -34,6 +34,15 @@ function readFeatures() {
     feats.cmd = readFeaturesType("cmd");
     return feats;
 }
+function readAliases() {
+    const stmt = db.prepare("SELECT name, type FROM aliases");
+    const data = stmt.all();
+    let f = new Array();
+    data.forEach((row) => {
+        f.push({ name: row.name, type: row.type });
+    });
+    return f;
+}
 function readFeature(name, type) {
     const q = `SELECT id, path, ext FROM ${type} WHERE name='${name}'`;
     const stmt = db.prepare(q);
@@ -50,4 +59,4 @@ function readFeature(name, type) {
     }
     return { found: false, feature: {} };
 }
-export { readFeatures, readFeaturePaths, readFeature, readPlugins };
+export { readFeatures, readFeaturePaths, readFeature, readPlugins, readAliases };

package/dist/db/schemas.js

@@ -36,7 +36,12 @@ const cmds = `CREATE TABLE IF NOT EXISTS cmd (
     path TEXT NOT NULL,
     ext TEXT NOT NULL CHECK ( ext IN ('js') )
 );`;
+const aliases = `CREATE TABLE IF NOT EXISTS aliases (
+    id INTEGER PRIMARY KEY AUTOINCREMENT,
+    name TEXT UNIQUE NOT NULL,
+    type TEXT NOT NULL CHECK ( type IN ('task', 'action', 'job') )
+);`;
 const schemas = [
-    filepaths, featurespaths, tasks, jobs, actions, cmds, plugins
+    filepaths, featurespaths, tasks, jobs, actions, cmds, plugins, aliases
 ];
 export { schemas };

package/dist/db/write.d.ts

@@ -2,5 +2,6 @@ import { Features } from "../interfaces.js";
 declare function insertDefaultFilepaths(): void;
 declare function insertFeaturesPathIfNotExists(path: string): boolean;
 declare function insertPluginIfNotExists(n: string, p: string): boolean;
+declare function updateAliases(feats: Features): void;
 declare function updateFeatures(feats: Features): void;
-export { insertDefaultFilepaths, insertFeaturesPathIfNotExists, insertPluginIfNotExists, updateFeatures, };
+export { insertDefaultFilepaths, insertFeaturesPathIfNotExists, insertPluginIfNotExists, updateFeatures, updateAliases, };

package/dist/db/write.js

@@ -29,6 +29,31 @@ function insertPluginIfNotExists(n, p) {
     stmt.run(n, p);
     return false;
 }
+function _updateAlias(existingAliases, name, type) {
+    if (!existingAliases.includes(name)) {
+        const insertStmt = db.prepare("INSERT INTO aliases (name, type) VALUES (?, ?)");
+        insertStmt.run(name, type);
+    }
+    else {
+        console.log("Can not create command alias", name, ": duplicate name");
+    }
+    existingAliases.push(name);
+    return existingAliases;
+}
+function updateAliases(feats) {
+    const deleteStmt = db.prepare("DELETE FROM aliases");
+    deleteStmt.run();
+    let existingAliases = new Array();
+    feats.task.forEach((feat) => {
+        existingAliases = _updateAlias(existingAliases, feat.name, "task");
+    });
+    feats.action.forEach((feat) => {
+        existingAliases = _updateAlias(existingAliases, feat.name, "action");
+    });
+    feats.job.forEach((feat) => {
+        existingAliases = _updateAlias(existingAliases, feat.name, "job");
+    });
+}
 function upsertAndCleanFeatures(feats, type) {
     const stmt = db.prepare(`SELECT name FROM ${type}`);
     const rows = stmt.all();
@@ -53,4 +78,4 @@ function updateFeatures(feats) {
     upsertAndCleanFeatures(feats.action, "action");
     upsertAndCleanFeatures(feats.cmd, "cmd");
 }
-export { insertDefaultFilepaths, insertFeaturesPathIfNotExists, insertPluginIfNotExists, updateFeatures, };
+export { insertDefaultFilepaths, insertFeaturesPathIfNotExists, insertPluginIfNotExists, updateFeatures, updateAliases, };

package/dist/interfaces.d.ts

@@ -46,4 +46,5 @@ type TaskExtension = "yml";
 type JobExtension = "yml";
 type CmdExtension = "js";
 type FeatureExtension = TaskExtension | JobExtension | CmdExtension | ActionExtension;
-export { Cmd, CmdExecutor, InputMode, OutputMode, RunMode, FormatMode, FeatureType, ActionExtension, TaskExtension, JobExtension, CmdExtension, FeatureSpec, Features, ConfigFile, FeatureExtension, };
+type AliasType = "task" | "action" | "job";
+export { Cmd, CmdExecutor, InputMode, OutputMode, RunMode, FormatMode, FeatureType, ActionExtension, TaskExtension, JobExtension, CmdExtension, FeatureSpec, Features, ConfigFile, FeatureExtension, AliasType, };

package/dist/state/features.js

@@ -9,7 +9,6 @@ function readFeaturesDirs(featuresPaths) {
         cmd: [],
     };
     featuresPaths.forEach((dir) => {
-        console.log("Reading feats in", dir);
         const _f = readFeaturesDir(dir);
         _f.task.forEach((item) => feats.task.push(item));
         _f.job.forEach((item) => feats.job.push(item));

package/dist/state/state.d.ts

@@ -1,11 +1,11 @@
 import { PythonShell } from 'python-shell';
 import { InputMode, RunMode, FormatMode, OutputMode } from "../interfaces.js";
 declare let pyShell: PythonShell;
-declare const inputMode: import("@vue/reactivity").Ref<InputMode>;
-declare const outputMode: import("@vue/reactivity").Ref<OutputMode>;
-declare const runMode: import("@vue/reactivity").Ref<RunMode>;
-declare const formatMode: import("@vue/reactivity").Ref<FormatMode>;
-declare const promptfile: import("@vue/reactivity").Ref<string>;
+declare const inputMode: import("@vue/reactivity").Ref<InputMode, InputMode>;
+declare const outputMode: import("@vue/reactivity").Ref<OutputMode, OutputMode>;
+declare const runMode: import("@vue/reactivity").Ref<RunMode, RunMode>;
+declare const formatMode: import("@vue/reactivity").Ref<FormatMode, FormatMode>;
+declare const promptfile: import("@vue/reactivity").Ref<string, string>;
 declare const lastCmd: {
     name: string;
     args: Array<string>;

package/dist/state/state.js

@@ -2,7 +2,7 @@ import { reactive, ref } from "@vue/reactivity";
 import { createConfDirIfNotExists, confDir } from "../conf.js";
 import { initDb } from "../db/db.js";
 import { readFeaturePaths } from "../db/read.js";
-import { updateFeatures } from "../db/write.js";
+import { updateAliases, updateFeatures } from "../db/write.js";
 import { readFeaturesDirs } from "./features.js";
 import { readPluginsPaths } from "./plugins.js";
 let pyShell;
@@ -28,8 +28,10 @@ async function initFeatures() {
     const p = [...fp, ...pp];
     const feats = readFeaturesDirs(p);
     updateFeatures(feats);
+    updateAliases(feats);
 }
 async function initState() {
     initConf();
+    await initFeatures();
 }
 export { inputMode, outputMode, runMode, formatMode, lastCmd, promptfile, initState, initFeatures, pyShell, };

package/package.json

@@ -2,7 +2,7 @@
   "name": "@agent-smith/cli",
   "description": "Agent Smith: terminal client for language model agents",
   "repository": "https://github.com/synw/agent-smith",
-  "version": "0.0.6",
+  "version": "0.0.8",
   "scripts": {
     "buildrl": "rm -rf dist/* && rollup -c",
     "build": "rm -rf dist/* && tsc",
@@ -10,34 +10,35 @@
     "watch": "tsc -p . -w"
   },
   "dependencies": {
-    "@agent-smith/brain": "^0.0.18",
+    "@agent-smith/brain": "^0.0.24",
     "@agent-smith/jobs": "^0.0.8",
-    "@agent-smith/lmtask": "^0.0.13",
-    "@inquirer/prompts": "^5.3.8",
-    "@inquirer/select": "^2.4.7",
-    "@vue/reactivity": "^3.4.38",
-    "better-sqlite3": "^11.2.1",
+    "@agent-smith/lmtask": "^0.0.17",
+    "@inquirer/prompts": "^6.0.1",
+    "@inquirer/select": "^3.0.1",
+    "@vue/reactivity": "^3.5.6",
+    "better-sqlite3": "^11.3.0",
     "clipboardy": "^4.0.0",
     "commander": "^12.1.0",
     "draftlog": "^1.0.13",
     "marked-terminal": "^7.1.0",
     "modprompt": "^0.7.7",
     "python-shell": "^5.0.0",
-    "yaml": "^2.5.0"
+    "yaml": "^2.5.1"
   },
   "devDependencies": {
     "@agent-smith/tmem-jobs": "^0.0.3",
     "@commander-js/extra-typings": "^12.1.0",
-    "@locallm/types": "^0.0.17",
+    "@locallm/types": "^0.1.5",
     "@rollup/plugin-node-resolve": "^15.2.3",
     "@rollup/plugin-typescript": "^11.1.6",
     "@types/better-sqlite3": "^7.6.11",
     "@types/marked-terminal": "^6.1.1",
-    "@types/node": "^22.5.2",
-    "rollup": "^4.21.2",
+    "@types/node": "^22.5.5",
+    "restmix": "^0.5.0",
+    "rollup": "^4.21.3",
     "ts-node": "^10.9.2",
     "tslib": "2.7.0",
-    "typescript": "^5.5.4"
+    "typescript": "^5.6.2"
   },
   "type": "module",
   "preferGlobal": true,