@agent-smith/agent 0.3.0 → 0.3.2

package/dist/agent.js

@@ -148,8 +148,32 @@ class Agent {
             throw new Error(`error processing model answer:\n, ${error}`);
         }
         if (assistant) {
-            if (options?.onAssistant) {
-                options.onAssistant(assistant);
+            if (tpl.tags?.think?.start) {
+                let t = res.text;
+                if (toolsCall.length > 0) {
+                    if (tpl?.tags?.toolCall?.start) {
+                        t = res.text.split(tpl.tags.toolCall.start)[0].trim();
+                    }
+                    else {
+                        console.warn("Model called tools but not tool call tags found in template");
+                    }
+                }
+                const { think, finalAnswer } = splitThinking(t, tpl.tags.think.start, tpl.tags.think.end);
+                if (think.length > 0) {
+                    if (options?.onThink) {
+                        options.onThink(think);
+                    }
+                }
+                if (finalAnswer.length > 0) {
+                    if (options?.onAssistant) {
+                        options.onAssistant(finalAnswer);
+                    }
+                }
+            }
+            else {
+                if (options?.onAssistant) {
+                    options.onAssistant(assistant);
+                }
             }
         }
         const toolResults = new Array();
@@ -210,13 +234,23 @@ class Agent {
             if (options?.onToolsTurnEnd) {
                 options.onToolsTurnEnd(toolResults);
             }
+            let thinking = "";
+            let final = "";
+            if (tpl?.tags?.think) {
+                const { think, finalAnswer } = splitThinking(res.text, tpl.tags.think.start, tpl.tags.think.end);
+                thinking = think;
+                final = finalAnswer;
+            }
             if (it == 1) {
                 const t = {
                     user: prompt,
                     tools: toolResults,
                 };
                 if (assistant) {
-                    t.assistant = assistant;
+                    t.assistant = final;
+                }
+                if (thinking.length > 0) {
+                    t.think = thinking;
                 }
                 this.history.push(t);
             }
@@ -225,7 +259,10 @@ class Agent {
                     tools: toolResults,
                 };
                 if (assistant) {
-                    t.assistant = assistant;
+                    t.assistant = final;
+                }
+                if (thinking.length > 0) {
+                    t.think = thinking;
                 }
                 this.history.push(t);
             }

package/dist/utils.js

@@ -26,7 +26,7 @@ function splitThinking(text, startTag, endTag) {
     let answer = "";
     const st = text.split(endTag);
     if (st.length > 1) {
-        think = extractBetweenTags(text, startTag, endTag);
+        think = extractBetweenTags(text, startTag, endTag).trim();
         answer = st[1].trim();
     }
     else {

package/package.json

@@ -5,22 +5,22 @@
     "type": "git",
     "url": "git+https://github.com/synw/agent-smith.git"
   },
-  "version": "0.3.0",
+  "version": "0.3.2",
   "scripts": {
     "buildrl": "rm -rf dist/* && rollup -c",
     "build": "rm -rf dist/* && tsc"
   },
   "dependencies": {
     "@locallm/api": "^0.7.3",
-    "modprompt": "^0.14.0",
+    "modprompt": "^0.14.1",
     "restmix": "^0.6.1"
   },
   "devDependencies": {
-    "@locallm/types": "^0.7.0",
+    "@locallm/types": "^0.7.1",
     "@rollup/plugin-node-resolve": "^16.0.3",
     "@rollup/plugin-typescript": "^12.3.0",
     "@types/node": "^25.3.0",
-    "openai": "^6.22.0",
+    "openai": "^6.24.0",
     "rollup": "^4.59.0",
     "ts-node": "^10.9.2",
     "tslib": "2.8.1",