goblin-agents 2.3.1 → 2.4.0

package/lib/llm/aiAgent.js

@@ -146,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
@@ -167,6 +168,7 @@ const defaultValues = {
   role: 'assistant',
   provider: 'open-ai',
   model: '',
+  otherModels: [],
   options: {},
   host: 'http://127.0.0.1:11434',
   headers: {},
@@ -190,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,
@@ -288,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';
@@ -378,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);
 
@@ -869,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.1",
+  "version": "2.4.0",
   "description": "AI Agents framework",
   "author": "Epsitec SA",
   "contributors": [