@agent-smith/cli 0.0.24 → 0.0.26

package/dist/cmd/lib/execute_job.js

@@ -18,11 +18,6 @@ async function executeJobCmd(name, args = [], options = {}) {
     let i = 0;
     for (const [name, task] of Object.entries(job.tasks)) {
         if (task.type == "task") {
-            let conf = {};
-            let vars = {};
-            const tv = initTaskVars(args);
-            conf = tv.conf;
-            vars = i == 0 ? tv.vars : params;
             const { found, path } = getFeatureSpec(name, "task");
             if (!found) {
                 return { ok: false, data: {}, error: `Task ${name} not found` };
@@ -34,6 +29,13 @@ async function executeJobCmd(name, args = [], options = {}) {
             const taskSpec = taskBuilder.readFromYaml(tres.ymlTask);
             let m = taskSpec.model.name;
             let t = taskSpec.model.template;
+            let conf = {};
+            let vars = {};
+            const tv = initTaskVars(args, taskSpec?.inferParams ? taskSpec.inferParams : {});
+            console.log("TIP", taskSpec.inferParams);
+            console.log("IP", tv.conf.inferParams);
+            conf = tv.conf;
+            vars = i == 0 ? tv.vars : params;
             if (conf?.model) {
                 m = conf.model;
             }

package/dist/cmd/lib/execute_task.js

@@ -32,7 +32,7 @@ async function executeTaskCmd(args = [], options = {}) {
     }
     const taskSpec = taskBuilder.readFromYaml(res.ymlTask);
     const task = taskBuilder.fromYaml(res.ymlTask);
-    const { conf, vars } = initTaskVars(args);
+    const { conf, vars } = initTaskVars(args, taskSpec?.inferParams ? taskSpec.inferParams : {});
     if (isDebug.value) {
         console.log("Task conf:", conf);
         console.log("Task vars:", vars);
@@ -48,7 +48,7 @@ async function executeTaskCmd(args = [], options = {}) {
         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"`);
+                throw new Error(`Unable to guess the template for ${conf.model}: please provide a template name"`);
             }
             t = gt;
         }

package/dist/cmd/lib/utils.d.ts

@@ -6,7 +6,7 @@ declare function readTask(taskpath: string): {
     ymlTask: string;
 };
 declare function readTasksDir(dir: string): Array<string>;
-declare function initTaskVars(args: Array<any>): {
+declare function initTaskVars(args: Array<any>, inferParams: Record<string, any>): {
     conf: Record<string, any>;
     vars: Record<string, any>;
 };

package/dist/cmd/lib/utils.js

@@ -73,8 +73,8 @@ function readTasksDir(dir) {
     });
     return tasks;
 }
-function initTaskVars(args) {
-    const conf = {};
+function initTaskVars(args, inferParams) {
+    const conf = { inferParams: inferParams };
     const vars = {};
     args.forEach((a) => {
         if (a.includes("=")) {
@@ -92,12 +92,10 @@ function initTaskVars(args) {
                 }
             }
             else if (k == "ip") {
-                const ip = {};
                 v.split(",").forEach((p) => {
                     const s = p.split(":");
-                    ip[s[0]] = parseFloat(s[1]);
+                    conf["inferParams"][s[0]] = parseFloat(s[1]);
                 });
-                conf.inferParams = ip;
             }
             else if (k == "s") {
                 conf.size = 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.24",
+  "version": "0.0.26",
   "scripts": {
     "buildrl": "rm -rf dist/* && rollup -c",
     "build": "rm -rf dist/* && tsc",