@agent-smith/cli 0.0.22 → 0.0.24

package/dist/agent.d.ts

@@ -1,4 +1,4 @@
-import { LmTaskBuilder } from "@agent-smith/lmtask";
+import { LmTaskBuilder } from "../../lmtask/dist/task.js";
 import { marked } from 'marked';
 import { FeatureType } from "./interfaces.js";
 declare let brain: import("@agent-smith/brain").AgentBrain<Record<string, any>>;

package/dist/agent.js

@@ -1,5 +1,5 @@
 import { useAgentBrain } from "@agent-smith/brain";
-import { LmTaskBuilder } from "@agent-smith/lmtask";
+import { LmTaskBuilder } from "../../lmtask/dist/task.js";
 import { marked } from 'marked';
 import { markedTerminal } from "marked-terminal";
 marked.use(markedTerminal());

package/dist/cmd/cmds.d.ts

@@ -4,4 +4,4 @@ declare function initCliCmds(): Promise<void>;
 declare function runCmd(cmdName: string, args?: Array<string>): Promise<void>;
 declare function buildCmds(): Promise<Command>;
 declare function parseCmd(): Promise<void>;
-export { buildCmds, initCliCmds, parseCmd, runCmd, chat };
+export { buildCmds, chat, initCliCmds, parseCmd, runCmd };

package/dist/cmd/cmds.js

@@ -1,11 +1,13 @@
 import { input } from "@inquirer/prompts";
+import { toRaw } from "@vue/reactivity";
 import { Command } from "commander";
 import { brain } from "../agent.js";
+import { query } from "../cli.js";
+import { chatInferenceParams } from "../state/chat.js";
 import { isChatMode, lastCmd, runMode } from "../state/state.js";
 import { cmds, initAliases, initCmds } from "./clicmds/cmds.js";
 import { modes } from "./clicmds/modes.js";
 import { processOutput, setOptions } from "./lib/utils.js";
-import { query } from "../cli.js";
 let cliCmds = {};
 async function chat() {
     const data = { message: '>', default: "" };
@@ -19,7 +21,7 @@ async function chat() {
             await query();
         }
     }
-    await brain.ex.think(prompt);
+    await brain.ex.think(prompt, toRaw(chatInferenceParams));
     console.log();
     await chat();
 }
@@ -75,5 +77,4 @@ async function parseCmd() {
         await chat();
     }
 }
-export { buildCmds, initCliCmds, parseCmd, runCmd, chat };
-;
+export { buildCmds, chat, initCliCmds, parseCmd, runCmd };

package/dist/cmd/lib/execute_job.js

@@ -33,7 +33,7 @@ async function executeJobCmd(name, args = [], options = {}) {
             }
             const taskSpec = taskBuilder.readFromYaml(tres.ymlTask);
             let m = taskSpec.model.name;
-            let t = taskSpec.template.name;
+            let t = taskSpec.model.template;
             if (conf?.model) {
                 m = conf.model;
             }

package/dist/cmd/lib/execute_task.js

@@ -2,6 +2,8 @@ import { brain, initAgent, taskBuilder } from "../../agent.js";
 import { getFeatureSpec } from "../../state/features.js";
 import { isChatMode, isDebug } from "../../state/state.js";
 import { initTaskVars, parseInputOptions, readTask } from "./utils.js";
+import { useTemplateForModel } from "@agent-smith/tfm";
+const tfm = useTemplateForModel();
 async function executeTaskCmd(args = [], options = {}) {
     await initAgent();
     if (isDebug.value) {
@@ -36,12 +38,41 @@ async function executeTaskCmd(args = [], options = {}) {
         console.log("Task vars:", vars);
     }
     let m = taskSpec.model.name;
-    let t = taskSpec.template.name;
+    let t = taskSpec.model.template;
+    let c = taskSpec.model.ctx;
     if (conf?.model) {
         m = conf.model;
+        if (conf?.template) {
+            t = conf.template;
+        }
+        else {
+            const gt = tfm.guess(m);
+            if (gt == "none") {
+                throw new Error(`Unable to guess the template for ${conf.model}: please provide a template name: m="modelname/templatename"`);
+            }
+            t = gt;
+        }
+    }
+    else {
+        if (conf?.size) {
+            if (!taskSpec?.models) {
+                throw new Error(`Model ${conf.size} not found in task`);
+            }
+            if (!Object.keys(taskSpec.models).includes(conf.size)) {
+                throw new Error(`Model ${conf.size} not found in task`);
+            }
+            m = taskSpec.models[conf.size].name;
+            t = taskSpec.models[conf.size].template;
+            c = taskSpec.models[conf.size].ctx;
+        }
     }
-    if (conf?.template) {
-        t = conf.template;
+    conf.model = {
+        name: m,
+        template: t,
+        ctx: c,
+    };
+    if (isDebug.value) {
+        console.log("Model:", conf.model);
     }
     const ex = brain.getOrCreateExpertForModel(m, t);
     if (!ex) {
@@ -62,9 +93,6 @@ async function executeTaskCmd(args = [], options = {}) {
     if (isDebug.value) {
         conf.debug = true;
     }
-    if (isDebug.value) {
-        console.log("Vars", vars);
-    }
     const data = await task.run({ prompt: pr, ...vars }, conf);
     if (data?.error) {
         return { ok: false, data: "", conf: conf, error: `Error executing task: ${data.error}` };

package/dist/cmd/lib/utils.js

@@ -99,6 +99,9 @@ function initTaskVars(args) {
                 });
                 conf.inferParams = ip;
             }
+            else if (k == "s") {
+                conf.size = v;
+            }
             else {
                 vars[k] = v;
             }

package/dist/main.d.ts

@@ -1,6 +1,8 @@
 import { execute, run } from "./cmd/sys/execute.js";
 import { executeJobCmd } from "./cmd/lib/execute_job.js";
+import { executeActionCmd } from "./cmd/lib/execute_action.js";
+import { executeTaskCmd } from "./cmd/lib/execute_task.js";
 import { writeToClipboard } from "./cmd/sys/clipboard.js";
 import { pingCmd } from "./cmd/clicmds/cmds.js";
 import { initAgent } from "./agent.js";
-export { execute, run, pingCmd, executeJobCmd, writeToClipboard, initAgent, };
+export { execute, run, pingCmd, executeJobCmd, executeActionCmd, executeTaskCmd, writeToClipboard, initAgent, };

package/dist/main.js

@@ -1,6 +1,8 @@
 import { execute, run } from "./cmd/sys/execute.js";
 import { executeJobCmd } from "./cmd/lib/execute_job.js";
+import { executeActionCmd } from "./cmd/lib/execute_action.js";
+import { executeTaskCmd } from "./cmd/lib/execute_task.js";
 import { writeToClipboard } from "./cmd/sys/clipboard.js";
 import { pingCmd } from "./cmd/clicmds/cmds.js";
 import { initAgent } from "./agent.js";
-export { execute, run, pingCmd, executeJobCmd, writeToClipboard, initAgent, };
+export { execute, run, pingCmd, executeJobCmd, executeActionCmd, executeTaskCmd, writeToClipboard, initAgent, };

package/dist/state/chat.d.ts

@@ -0,0 +1,25 @@
+declare const chatInferenceParams: {
+    stream?: boolean | undefined;
+    model?: {
+        name: string;
+        ctx: number;
+        info?: {
+            size: string;
+            quant: string;
+        } | undefined;
+        extra?: Record<string, any> | undefined;
+    } | undefined;
+    template?: string | undefined;
+    max_tokens?: number | undefined;
+    top_k?: number | undefined;
+    top_p?: number | undefined;
+    min_p?: number | undefined;
+    temperature?: number | undefined;
+    repeat_penalty?: number | undefined;
+    tfs?: number | undefined;
+    stop?: Array<string> | undefined;
+    grammar?: string | undefined;
+    images?: Array<string> | undefined;
+    extra?: Record<string, any> | undefined;
+};
+export { chatInferenceParams, };

package/dist/state/chat.js

@@ -0,0 +1,3 @@
+import { reactive } from "@vue/reactivity";
+const chatInferenceParams = reactive({ temperature: 0.2, min_p: 0.05, max_tokens: 2048 });
+export { chatInferenceParams, };

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.22",
+  "version": "0.0.24",
   "scripts": {
     "buildrl": "rm -rf dist/* && rollup -c",
     "build": "rm -rf dist/* && tsc",
@@ -12,17 +12,18 @@
   "dependencies": {
     "@agent-smith/brain": "^0.0.33",
     "@agent-smith/jobs": "^0.0.11",
-    "@agent-smith/lmtask": "^0.0.24",
-    "@inquirer/prompts": "^7.0.1",
-    "@inquirer/select": "^4.0.1",
-    "@vue/reactivity": "^3.5.12",
+    "@agent-smith/lmtask": "^0.0.25",
+    "@agent-smith/tfm": "^0.1.1",
+    "@inquirer/prompts": "^7.1.0",
+    "@inquirer/select": "^4.0.2",
+    "@vue/reactivity": "^3.5.13",
     "better-sqlite3": "^11.5.0",
     "clipboardy": "^4.0.0",
     "commander": "^12.1.0",
     "marked-terminal": "^7.2.1",
     "modprompt": "^0.9.1",
     "python-shell": "^5.0.0",
-    "yaml": "^2.6.0"
+    "yaml": "^2.6.1"
   },
   "devDependencies": {
     "@agent-smith/tmem-jobs": "^0.0.4",
@@ -30,13 +31,13 @@
     "@locallm/types": "^0.1.5",
     "@rollup/plugin-node-resolve": "^15.3.0",
     "@rollup/plugin-typescript": "^12.1.1",
-    "@types/better-sqlite3": "^7.6.11",
+    "@types/better-sqlite3": "^7.6.12",
     "@types/marked-terminal": "^6.1.1",
-    "@types/node": "^22.8.5",
+    "@types/node": "^22.9.1",
     "restmix": "^0.5.0",
-    "rollup": "^4.24.3",
+    "rollup": "^4.27.3",
     "ts-node": "^10.9.2",
-    "tslib": "2.8.0",
+    "tslib": "2.8.1",
     "typescript": "^5.6.3"
   },
   "type": "module",