@agent-smith/agent 0.1.8 → 0.2.1

package/dist/agent.d.ts

@@ -2,10 +2,11 @@ import { Lm } from "@locallm/api";
 import { InferenceParams, ToolSpec, HistoryTurn, InferenceOptions, InferenceResult } from "@locallm/types";
 import { PromptTemplate } from 'modprompt';
 declare class Agent {
+    name: string;
     lm: Lm;
     tools: Record<string, ToolSpec>;
     history: Array<HistoryTurn>;
-    constructor(lm: Lm);
+    constructor(lm: Lm, name?: string);
     run(prompt: string, params: InferenceParams, options?: InferenceOptions, template?: PromptTemplate | string): Promise<InferenceResult>;
     runAgentNoTemplate(it: number, prompt: string, params: InferenceParams, options?: InferenceOptions): Promise<InferenceResult>;
     runAgentWithTemplate(it: number, prompt: string, params: InferenceParams, options: InferenceOptions | undefined, tpl: PromptTemplate): Promise<InferenceResult>;

package/dist/agent.js

@@ -1,17 +1,21 @@
 import { PromptTemplate } from 'modprompt';
 import { splitThinking } from "./utils.js";
 class Agent {
+    name = "unamed";
     lm;
     tools = {};
     history = [];
-    constructor(lm) {
+    constructor(lm, name) {
         this.lm = lm;
+        if (name) {
+            this.name = name;
+        }
     }
     async run(prompt, params, options = {}, template) {
         let tpl;
         if (options?.debug) {
-            console.log("Agent inference params:", params);
-            console.log("Agent options:", options);
+            console.log("Agent", this.name, "inference params:", params);
+            console.log("Agent", this.name, "options:", options);
         }
         if (options?.history) {
             this.history = options.history;
@@ -72,7 +76,7 @@ class Agent {
                 const tool = this.tools[tc.name];
                 let canRun = true;
                 if (tool?.canRun) {
-                    canRun = await tool.canRun(tool);
+                    canRun = await tool.canRun(tc);
                 }
                 if (canRun) {
                     if (options?.onToolCall) {
@@ -94,10 +98,19 @@ class Agent {
                 }
             }
             this.history.push({ tools: toolsResults });
+            if (options?.isToolsRouter) {
+                const fres = {
+                    text: JSON.stringify(toolsResults.map(tr => tr.response)),
+                    stats: res.stats,
+                    serverStats: res.serverStats,
+                    toolCalls: res.toolCalls,
+                };
+                return fres;
+            }
+            const nit = it + 1;
             if (options?.tools) {
                 options.tools = Object.values(this.tools);
             }
-            const nit = it + 1;
             _res = await this.runAgentNoTemplate(nit, "", params, options);
         }
         else {
@@ -138,7 +151,7 @@ class Agent {
                 }
                 let canRun = true;
                 if (tool?.canRun) {
-                    canRun = await tool.canRun(tool);
+                    canRun = await tool.canRun(toolCall);
                 }
                 if (!canRun) {
                     if (options?.debug || options?.verbose) {
@@ -182,6 +195,15 @@ class Agent {
                     tools: toolResults,
                 });
             }
+            if (options?.isToolsRouter && isToolCall) {
+                const fres = {
+                    text: JSON.stringify(toolResults.map(tr => tr.response)),
+                    stats: res.stats,
+                    serverStats: res.serverStats,
+                    toolCalls: res.toolCalls,
+                };
+                return fres;
+            }
             return await this.runAgentWithTemplate(it + 1, prompt, params, options, tpl);
         }
         else {

package/package.json

@@ -5,7 +5,7 @@
     "type": "git",
     "url": "git+https://github.com/synw/agent-smith.git"
   },
-  "version": "0.1.8",
+  "version": "0.2.1",
   "scripts": {
     "buildrl": "rm -rf dist/* && rollup -c",
     "build": "rm -rf dist/* && tsc"
@@ -16,12 +16,12 @@
     "restmix": "^0.6.1"
   },
   "devDependencies": {
-    "@locallm/types": "^0.6.5",
+    "@locallm/types": "^0.6.7",
     "@rollup/plugin-node-resolve": "^16.0.3",
     "@rollup/plugin-typescript": "^12.3.0",
-    "@types/node": "^25.0.3",
-    "openai": "^6.15.0",
-    "rollup": "^4.54.0",
+    "@types/node": "^25.0.8",
+    "openai": "^6.16.0",
+    "rollup": "^4.55.1",
     "ts-node": "^10.9.2",
     "tslib": "2.8.1",
     "typescript": "^5.9.3"