osai-agent 4.2.5 → 4.2.8

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "osai-agent",
-  "version": "4.2.5",
+  "version": "4.2.8",
   "type": "module",
   "description": "OS AI Agent - YOUR AI AGENT",
   "main": "src/index.js",

package/src/agent/loop/local.js

@@ -163,7 +163,7 @@ const systemPrompt = buildSystemPrompt(this._userOS, this.mode, {
 
         if (this._shouldCancel()) return null;
 
-        if (retrying) { this.onUpdateLastText(''); retrying = false; }
+        if (retrying) { this.onRetryStatus(''); retrying = false; }
 
         if (!fullResponse) return null;
 
@@ -186,7 +186,7 @@ const systemPrompt = buildSystemPrompt(this._userOS, this.mode, {
         }
         retrying = true;
         const backoffMs = DEFAULTS.API_RETRY_DELAY * (attempt + 1);
-        this.onMarkdown(`⏳ Retrying in ${Math.ceil(backoffMs / 1000)}s...`);
+        this.onRetryStatus(`Retrying in ${Math.ceil(backoffMs / 1000)}s...`);
         const slept = await this._cancellableSleep(backoffMs);
         if (!slept || this._shouldCancel()) return null;
       }

package/src/agent/react-loop.js

@@ -39,7 +39,7 @@ export class AgentLoop {
     onMarkdown, onThought, onToolStart, onToolResult, onObservation,
     onError, onComplete, onTodos, onBlocked, onThinkingStart, onThinkingEnd,
     onBadge, onConfirmPrompt, onAskUserPrompt, onPlanPrompt,
-    onStats, onConnectionChange, onUpdateLastText, readline, noConfirm = false,
+    onStats, onConnectionChange, onUpdateLastText, onRetryStatus, readline, noConfirm = false,
     isSubagent = false, maxIterations = null, local = false,
   }) {
     this.device = device;
@@ -115,6 +115,7 @@ export class AgentLoop {
      this.onPlanPrompt = onPlanPrompt || (() => {});
      this.onStats = onStats || (() => {});
       this.onConnectionChange = onConnectionChange || (() => {});
+     this.onRetryStatus = onRetryStatus || (() => {});
      this.onUpdateLastText = onUpdateLastText || (() => {});
      this.onStats(0);
    }
@@ -299,7 +300,7 @@ export class AgentLoop {
       this.iteration++;
       logger.debug(`Iteration ${this.iteration}/${this._maxIterations}`);
 
-      if (iterationRetrying) { this.onUpdateLastText(''); iterationRetrying = false; }
+      if (iterationRetrying) { this.onRetryStatus(''); iterationRetrying = false; }
 
       try {
         const shouldContinue = await this._executeIteration();
@@ -332,7 +333,7 @@ export class AgentLoop {
         if (this.iteration <= 1 && !this._shouldCancel()) {
           logger.info('Retrying iteration after error...');
           iterationRetrying = true;
-          this.onMarkdown(`\n⏳ Retrying after error: ${error.message}\n`);
+          this.onRetryStatus(`Retrying after error: ${error.message}`);
           this.iteration--;
           const slept = await this._cancellableSleep(DEFAULTS.API_RETRY_DELAY);
           if (!slept || this._shouldCancel()) break;
@@ -618,7 +619,7 @@ export class AgentLoop {
 
           const backoffMs = DEFAULTS.API_RETRY_DELAY * (attempt + 1);
           retrying = true;
-          this.onMarkdown(`⏳ Request failed (${response.status}) — retrying in ${Math.ceil(backoffMs / 1000)}s...`);
+          this.onRetryStatus(`Request failed (${response.status}) — retrying in ${Math.ceil(backoffMs / 1000)}s...`);
           const slept = await this._cancellableSleep(backoffMs);
           if (!slept || this._shouldCancel()) return null;
           continue;
@@ -629,7 +630,7 @@ export class AgentLoop {
         if (contentType.includes('application/json')) {
           const json = await response.json();
           if (json.content || json.message) {
-            if (retrying) { this.onUpdateLastText(''); retrying = false; }
+            if (retrying) { this.onRetryStatus(''); retrying = false; }
             const textContent = json.content || json.message || '';
             const cleanText = textContent
               .replace(/^\s*\{(?:\\")?tool(?:\\")?\s*:\s*.*$/gim, '')
@@ -647,7 +648,7 @@ export class AgentLoop {
             }
             retrying = true;
             const backoffMs = DEFAULTS.API_RETRY_DELAY * (attempt + 1);
-            this.onMarkdown(`⏳ Worker error — retrying in ${Math.ceil(backoffMs / 1000)}s...`);
+            this.onRetryStatus(`Worker error — retrying in ${Math.ceil(backoffMs / 1000)}s...`);
             const slept = await this._cancellableSleep(backoffMs);
             if (!slept || this._shouldCancel()) return null;
             continue;
@@ -655,7 +656,7 @@ export class AgentLoop {
           return null;
         }
 
-        if (retrying) { this.onUpdateLastText(''); retrying = false; }
+        if (retrying) { this.onRetryStatus(''); retrying = false; }
         return response;
       } catch (error) {
         if (error.name === 'AbortError' || this._shouldCancel()) {
@@ -668,7 +669,7 @@ export class AgentLoop {
         }
         retrying = true;
         const backoffMs = DEFAULTS.API_RETRY_DELAY * (attempt + 1);
-        this.onMarkdown(`⏳ Retrying in ${Math.ceil(backoffMs / 1000)}s...`);
+        this.onRetryStatus(`Retrying in ${Math.ceil(backoffMs / 1000)}s...`);
         const slept = await this._cancellableSleep(backoffMs);
         if (!slept || this._shouldCancel()) return null;
       }

package/src/agent/subagent.js

@@ -103,6 +103,9 @@ export async function runSubagent(parent, { prompt, description = '' }) {
        onUpdateLastText: (text) => {
          if (STREAM_SUBAGENT_TO_PARENT) parent.onUpdateLastText(text);
        },
+       onRetryStatus: (text) => {
+         if (STREAM_SUBAGENT_TO_PARENT) parent.onRetryStatus(text);
+       },
        onThought: (text) => {
          if (STREAM_SUBAGENT_TO_PARENT) parent.onThought(text);
        },

package/src/ui/App.js

@@ -719,6 +719,36 @@ export function App({ createAgentLoop, agentConfig, initialSession = null, onExi
           updateLastEvent(ev => ({ ...ev, content: text }));
         }
       },
+      onRetryStatus: (text) => {
+        const arr = eventsRef.current;
+        const findRetryIdx = () => {
+          for (let i = arr.length - 1; i >= 0; i--) {
+            if (arr[i].type === 'retry_status') return i;
+          }
+          return -1;
+        };
+        if (!text) {
+          const idx = findRetryIdx();
+          if (idx !== -1) {
+            const newArr = [...arr];
+            newArr.splice(idx, 1);
+            eventsRef.current = newArr;
+            flushEventsToState(false);
+          }
+          return;
+        }
+        const cleanedText = sanitizeUiText(text);
+        if (!cleanedText.trim()) return;
+        const idx = findRetryIdx();
+        if (idx !== -1) {
+          const newArr = [...arr];
+          newArr[idx] = { ...newArr[idx], content: cleanedText };
+          eventsRef.current = newArr;
+          flushEventsToState(false);
+        } else {
+          addEvent({ type: 'retry_status', content: cleanedText });
+        }
+      },
       onToolStart: (toolCall) => {
         if (toolCall.tool === 'ASK_USER') return;
         const uid = ++toolUiIdCounterRef.current;

package/src/ui/components/MessageHistory.js

@@ -928,6 +928,8 @@ function renderEvent(ev, i, events, expandedOutputIndexes, thoughtStreaming, exp
       const expanded = thoughtId != null && expandedThoughtIds && expandedThoughtIds.has(thoughtId);
       return h(CollapsibleThought, { key: `thought_${thoughtId ?? i}`, id: thoughtId, content: ev.content, expanded, streaming: thoughtStreaming, animate });
     }
+    case 'retry_status':
+      return h(Box, { key: i, paddingLeft: 2 }, h(Text, { color: 'red' }, ev.content));
     case 'text':
       return h(TextContent, { key: i, content: ev.content, events, animate });
     case 'tool_start':