@hamp10/agentforge 0.2.4 → 0.2.6

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@hamp10/agentforge",
-  "version": "0.2.4",
+  "version": "0.2.6",
   "description": "AgentForge worker — connect your machine to agentforge.ai",
   "type": "module",
   "bin": {

package/src/OllamaAgent.js

@@ -156,21 +156,27 @@ export class OllamaAgent extends EventEmitter {
     return { agentId, workDir };
   }
 
-  async runAgentTask(agentId, task, workDir, sessionId = null, image = null) {
+  async runAgentTask(agentId, task, workDir, sessionId = null, image = null, browserProfile = null, actualWorkDir = null, agentModel = null) {
     const startTime = Date.now();
     const controller = new AbortController();
 
+    // Use per-agent model override if provided (and not the placeholder 'Default').
+    // Strip 'ollama/' prefix — catalog returns IDs like 'ollama/qwen3-vl:8b' but
+    // Ollama's API expects bare names like 'qwen3-vl:8b'.
+    const rawModel = (agentModel && agentModel !== 'Default') ? agentModel : this.model;
+    const effectiveModel = rawModel.startsWith('ollama/') ? rawModel.slice(7) : rawModel;
+
     // Fake proc-like object so worker.js pid checks don't crash
     const fakeProc = { pid: null };
     this.activeAgents.set(agentId, { startTime, task, workDir, abort: () => controller.abort(), proc: fakeProc });
 
-    console.log(`\n🦙 [Ollama/${this.model}] Running agent: ${agentId}`);
+    console.log(`\n🦙 [Ollama/${effectiveModel}] Running agent: ${agentId}`);
     console.log(`   Task: ${task}`);
     console.log(`   Working dir: ${workDir}`);
 
     // Detect model capabilities
-    const isQwen3    = this.model.startsWith('qwen3');
-    const isVision   = /vl|vision|llava|minicpm-v|moondream/i.test(this.model);
+    const isQwen3    = effectiveModel.startsWith('qwen3');
+    const isVision   = /vl|vision|llava|minicpm-v|moondream/i.test(effectiveModel);
 
     try {
       // Load conversation history from disk (session persistence)
@@ -209,12 +215,12 @@ export class OllamaAgent extends EventEmitter {
       for (let turn = 0; turn < MAX_TURNS; turn++) {
         if (controller.signal.aborted) break;
 
-        this.emit('tool_activity', { agentId, event: 'api_call_start', description: `🦙 Calling ${this.model}...` });
+        this.emit('tool_activity', { agentId, event: 'api_call_start', description: `🦙 Calling ${effectiveModel}...` });
 
         let response;
         try {
           const requestBody = {
-            model: this.model,
+            model: effectiveModel,
             messages,
             tools: TOOLS,
             tool_choice: 'auto',
@@ -418,7 +424,7 @@ export class OllamaAgent extends EventEmitter {
             headers: { 'Content-Type': 'application/json' },
             signal: controller.signal,
             body: JSON.stringify({
-              model: this.model,
+              model: effectiveModel,
               messages,
               stream: true,
               ...(isQwen3 ? { options: { think: false } } : {})