@agent-smith/cli 0.0.23 → 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/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/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.23",
+  "version": "0.0.24",
   "scripts": {
     "buildrl": "rm -rf dist/* && rollup -c",
     "build": "rm -rf dist/* && tsc",