goblin-agents 2.3.0 → 2.4.0

package/lib/llm/aiAgent.js

@@ -134,10 +134,7 @@ function cleanText(text) {
   if (typeof text !== 'string') {
     return text;
   }
-  const {thinks, cleanText} = splitThinkTags(text);
-  for (const think of thinks) {
-    this.log.dbg(`🧐 ${think}`);
-  }
+  const {cleanText} = splitThinkTags(text);
   return cleanText;
 }
 
@@ -149,6 +146,7 @@ class AiAgentShape {
   prompt = string; //default prompt
   provider = enumeration('ollama', 'open-ai');
   model = string;
+  otherModels = array(string);
   host = string;
   headers = record(string, string);
   messages = record(string, array(MessageShape)); //for keeping chat history by context
@@ -170,6 +168,7 @@ const defaultValues = {
   role: 'assistant',
   provider: 'open-ai',
   model: '',
+  otherModels: [],
   options: {},
   host: 'http://127.0.0.1:11434',
   headers: {},
@@ -193,6 +192,7 @@ class AiAgentLogic extends Elf.Archetype {
     prompt: defaultValues.prompt,
     provider: defaultValues.provider,
     model: defaultValues.model,
+    otherModels: defaultValues.otherModels,
     messages: {},
     options: defaultValues.options,
     host: defaultValues.host,
@@ -291,6 +291,12 @@ class AiAgentLogic extends Elf.Archetype {
     state._state.set(path, newValue);
   }
 
+  useOtherModel(model) {
+    this.state.otherModels.deleteByValue(model);
+    this.state.otherModels.push(this.state.model);
+    this.state.model = model;
+  }
+
   publish() {
     const {state} = this;
     state.meta.status = 'published';
@@ -381,6 +387,18 @@ class AiAgent extends Elf {
     await this.persist();
   }
 
+  async useOtherModel(model) {
+    if (!this.state.otherModels.includes(model)) {
+      return false;
+    }
+    this.logic.useOtherModel(model);
+    return true;
+  }
+
+  async getModelList() {
+    return [this.state.model, ...this.state.otherModels];
+  }
+
   async gen(prompt, contextId = 'default') {
     const logger = Logger.get(contextId, this);
 
@@ -612,7 +630,7 @@ class AiAgent extends Elf {
     const res = cleanText(result.message.content);
     if (typeof res === 'string') {
       //keep agent response
-      this.logic.chat(contextId, 'assistant', cleanText);
+      this.logic.chat(contextId, 'assistant', res);
     }
 
     this.log.dbg(`🐈 Chat session ended : ${contextId}`);
@@ -872,8 +890,8 @@ class AiAgent extends Elf {
   }
 
   async getBaseSettings() {
-    const {provider, host, headers, model} = this.state.toJS();
-    return {provider, host, headers, model};
+    const {provider, host, headers, model, otherModels} = this.state.toJS();
+    return {provider, host, headers, model, otherModels};
   }
 
   async trash() {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "goblin-agents",
-  "version": "2.3.0",
+  "version": "2.4.0",
   "description": "AI Agents framework",
   "author": "Epsitec SA",
   "contributors": [