@hamp10/agentforge 0.2.3 → 0.2.5

package/package.json

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

package/src/OllamaAgent.js

@@ -156,21 +156,24 @@ 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')
+    const effectiveModel = (agentModel && agentModel !== 'Default') ? agentModel : this.model;
+
     // 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 +212,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',
@@ -407,6 +410,51 @@ export class OllamaAgent extends EventEmitter {
         finalContent = allOutput;
       }
 
+      // If still no output (model did only tool calls, never wrote text), ask for a summary
+      if (!finalContent && !controller.signal.aborted) {
+        this.emit('agent_output', { agentId, output: '\n' });
+        messages.push({ role: 'user', content: 'Summarize what you just did and show me the result.' });
+
+        try {
+          const summaryRes = await fetch(`${this.baseUrl}/v1/chat/completions`, {
+            method: 'POST',
+            headers: { 'Content-Type': 'application/json' },
+            signal: controller.signal,
+            body: JSON.stringify({
+              model: effectiveModel,
+              messages,
+              stream: true,
+              ...(isQwen3 ? { options: { think: false } } : {})
+            })
+          });
+
+          if (summaryRes.ok) {
+            const reader = summaryRes.body.getReader();
+            const decoder = new TextDecoder();
+            let buf = '';
+            while (true) {
+              const { done, value } = await reader.read();
+              if (done) break;
+              buf += decoder.decode(value, { stream: true });
+              const lines = buf.split('\n');
+              buf = lines.pop();
+              for (const line of lines) {
+                if (!line.startsWith('data: ')) continue;
+                const payload = line.slice(6).trim();
+                if (payload === '[DONE]') continue;
+                let evt;
+                try { evt = JSON.parse(payload); } catch { continue; }
+                const text = evt.choices?.[0]?.delta?.content;
+                if (text) {
+                  finalContent += text;
+                  this.emit('agent_output', { agentId, output: text });
+                }
+              }
+            }
+          }
+        } catch {}
+      }
+
       // Persist history for next task
       if (finalContent && sessionId) {
         this._saveHistory(agentId, workDir, sessionId, [