@agent-smith/agent 0.0.2 → 0.0.4

package/dist/agent.d.ts

@@ -6,7 +6,7 @@ declare class Agent {
     tools: Record<string, ToolSpec>;
     history: Array<HistoryTurn>;
     constructor(lm: Lm);
-    run(prompt: string, params: InferenceParams, options?: InferenceOptions, template?: PromptTemplate): Promise<InferenceResult>;
+    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: InferenceResult;

package/dist/agent.js

@@ -1,3 +1,4 @@
+import { PromptTemplate } from 'modprompt';
 class Agent {
     lm;
     tools = {};
@@ -6,23 +7,41 @@ class Agent {
         this.lm = lm;
     }
     async run(prompt, params, options = {}, template) {
-        this.tools = {};
-        if (options?.tools) {
-            options.tools.forEach(t => {
-                this.tools[t.name] = t;
-                if (template) {
-                    template = template.addTool(t);
-                }
-            });
+        let tpl;
+        if (options?.debug) {
+            console.log("Agent inference params:", params);
+            console.log("Agent options:", options);
+            console.log("Agent template:", template);
+            console.log("Prompt:", prompt);
         }
         if (this.lm.providerType == "openai") {
             return await this.runAgentNoTemplate(1, prompt, params, options);
         }
         else {
             if (!template) {
-                throw new Error(`A template is required for provider ${this.lm.provider.name}`);
+                if (params?.template) {
+                    tpl = new PromptTemplate(params.template);
+                }
+                else {
+                    throw new Error(`A template is required for provider ${this.lm.provider.name}`);
+                }
+            }
+            else {
+                if (typeof template == "string") {
+                    tpl = new PromptTemplate(template);
+                }
+                else {
+                    tpl = template;
+                }
             }
-            return (await this.runAgentWithTemplate(1, prompt, params, options, template)).inferenceResult;
+            this.tools = {};
+            if (options?.tools) {
+                options.tools.forEach(t => {
+                    this.tools[t.name] = t;
+                    tpl = tpl.addTool(t);
+                });
+            }
+            return (await this.runAgentWithTemplate(1, prompt, params, options, tpl)).inferenceResult;
         }
     }
     async runAgentNoTemplate(it, prompt, params, options = {}) {
@@ -72,6 +91,12 @@ class Agent {
         return _res;
     }
     async runAgentWithTemplate(it, prompt, params, options = {}, tpl) {
+        if (this.lm.providerType == "ollama") {
+            if (!params?.model) {
+                throw new Error("A model inference parameters is required for provider Ollama");
+            }
+            await this.lm.loadModel(params.model.name, params.model.ctx);
+        }
         let res = await this.lm.infer(tpl.prompt(prompt), params, options);
         const { isToolCall, toolsCall, error } = tpl.processAnswer(res.text);
         if (error) {

package/package.json

@@ -1,8 +1,11 @@
 {
   "name": "@agent-smith/agent",
   "description": "Agent Smith: agent runtime",
-  "repository": "https://github.com/synw/agent-smith",
-  "version": "0.0.2",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/synw/agent-smith.git"
+  },
+  "version": "0.0.4",
   "scripts": {
     "buildrl": "rm -rf dist/* && rollup -c",
     "build": "rm -rf dist/* && tsc"