Use status message to show retry

All-Hands-AI · Jan 30, 2025 · cef27cb · cef27cb
1 parent 825e80e
commit cef27cb
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/frontend/src/i18n/translation.json b/frontend/src/i18n/translation.json
@@ -3816,6 +3816,9 @@
         "es": "Hubo un error al conectar con el entorno de ejecución. Por favor, actualice la página.",
         "tr": "Çalışma zamanına bağlanırken bir hata oluştu. Lütfen sayfayı yenileyin."
     },
+    "STATUS$LLM_RETRY": {
+        "en": "Retrying LLM request"
+    },
     "AGENT_ERROR$BAD_ACTION": {
         "en": "Agent tried to execute a malformed action.",
         "zh-CN": "错误的操作",

diff --git a/openhands/controller/agent_controller.py b/openhands/controller/agent_controller.py
@@ -728,10 +728,11 @@ async def _step(self) -> None:
         self.log(log_level, str(action), extra={'msg_type': 'ACTION'})
 
     def _notify_on_llm_retry(self, retries: int, max: int) -> None:
-        self.event_stream.add_event(
-            ErrorObservation(content=f'Retry {retries} / {max}'),
-            EventSource.ENVIRONMENT,
-        )
+        if self.status_callback is not None:
+            msg_id = 'STATUS$LLM_RETRY'
+            self.status_callback(
+                'info', msg_id, f'Retrying LLM request, {retries} / {max}'
+            )
 
     async def _handle_traffic_control(
         self, limit_type: str, current_value: float, max_value: float