Merge branch 'main' into fix/agent_thinking

frontend/src/app/api/queries/useDoclingHealthQuery.ts

@@ -56,8 +56,13 @@ export const useDoclingHealthQuery = (
       queryKey: ["docling-health"],
       queryFn: checkDoclingHealth,
       retry: 1,
-      refetchInterval: 30000, // Check every 30 seconds
+      refetchInterval: (query) => {
-      staleTime: 25000, // Consider data stale after 25 seconds
+        // If healthy, check every 30 seconds; otherwise check every 3 seconds
+        return query.state.data?.status === "healthy" ? 30000 : 3000;
+      },
+      refetchOnWindowFocus: true,
+      refetchOnMount: true,
+      staleTime: 30000, // Consider data stale after 25 seconds
       ...options,
     },
     queryClient,

frontend/src/app/api/queries/useProviderHealthQuery.ts

@@ -92,6 +92,13 @@ export const useProviderHealthQuery = (
       queryKey: ["provider", "health"],
       queryFn: checkProviderHealth,
       retry: false, // Don't retry health checks automatically
+      refetchInterval: (query) => {
+        // If healthy, check every 30 seconds; otherwise check every 3 seconds
+        return query.state.data?.status === "healthy" ? 30000 : 3000;
+      },
+      refetchOnWindowFocus: true,
+      refetchOnMount: true,
+      staleTime: 30000, // Consider data stale after 25 seconds
       enabled: !!settings?.edited && options?.enabled !== false, // Only run after onboarding is complete
       ...options,
     },

src/tui/managers/docling_manager.py

@@ -34,6 +34,7 @@ class DoclingManager:
         # Bind to all interfaces by default (can be overridden with DOCLING_BIND_HOST env var)
         self._host = os.getenv('DOCLING_BIND_HOST', '0.0.0.0')
         self._running = False
+        self._starting = False
         self._external_process = False
 
         # PID file to track docling-serve across sessions (in current working directory)
@@ -126,6 +127,7 @@ class DoclingManager:
         if self._process is not None and self._process.poll() is None:
             self._running = True
             self._external_process = False
+            self._starting = False  # Clear starting flag if service is running
             return True
 
         # Check if we have a PID from file
@@ -133,6 +135,7 @@ class DoclingManager:
         if pid is not None and self._is_process_running(pid):
             self._running = True
             self._external_process = True
+            self._starting = False  # Clear starting flag if service is running
             return True
 
         # No running process found
@@ -142,6 +145,19 @@ class DoclingManager:
     
     def get_status(self) -> Dict[str, Any]:
         """Get current status of docling serve."""
+        # Check for starting state first
+        if self._starting:
+            display_host = "localhost" if self._host == "0.0.0.0" else self._host
+            return {
+                "status": "starting",
+                "port": self._port,
+                "host": self._host,
+                "endpoint": None,
+                "docs_url": None,
+                "ui_url": None,
+                "pid": None
+            }
+
         if self.is_running():
             # Try to get PID from process handle first, then from PID file
             pid = None
@@ -196,6 +212,9 @@ class DoclingManager:
         except Exception as e:
             self._add_log_entry(f"Error checking port availability: {e}")
 
+        # Set starting flag to show "Starting" status in UI
+        self._starting = True
+
         # Clear log buffer when starting
         self._log_buffer = []
         self._add_log_entry("Starting docling serve as external process...")
@@ -261,6 +280,8 @@ class DoclingManager:
 
                     if result == 0:
                         self._add_log_entry(f"Docling-serve is now listening on {self._host}:{self._port}")
+                        # Service is now running, clear starting flag
+                        self._starting = False
                         break
                 except:
                     pass
@@ -294,16 +315,24 @@ class DoclingManager:
                     self._add_log_entry(f"Error reading final output: {e}")
 
                 self._running = False
+                self._starting = False
                 return False, f"Docling serve process exited immediately (code: {return_code})"
 
+            # If we get here and the process is still running but not listening yet,
+            # clear the starting flag anyway (it's running, just not ready)
+            if self._process.poll() is None:
+                self._starting = False
+
             display_host = "localhost" if self._host == "0.0.0.0" else self._host
             return True, f"Docling serve starting on http://{display_host}:{port}"
 
         except FileNotFoundError:
+            self._starting = False
             return False, "docling-serve not available. Please install: uv add docling-serve"
         except Exception as e:
             self._running = False
             self._process = None
+            self._starting = False
             return False, f"Error starting docling serve: {str(e)}"
 
     def _start_output_capture(self):

src/tui/screens/monitor.py

@@ -206,10 +206,21 @@ class MonitorScreen(Screen):
 
         # Add docling serve to its own table
         docling_status = self.docling_manager.get_status()
-        docling_running = docling_status["status"] == "running"
+        docling_status_value = docling_status["status"]
-        docling_status_text = "running" if docling_running else "stopped"
+        docling_running = docling_status_value == "running"
-        docling_style = "bold green" if docling_running else "bold red"
+        docling_starting = docling_status_value == "starting"
-        docling_port = f"{docling_status['host']}:{docling_status['port']}" if docling_running else "N/A"
+        
+        if docling_running:
+            docling_status_text = "running"
+            docling_style = "bold green"
+        elif docling_starting:
+            docling_status_text = "starting"
+            docling_style = "bold yellow"
+        else:
+            docling_status_text = "stopped"
+            docling_style = "bold red"
+        
+        docling_port = f"{docling_status['host']}:{docling_status['port']}" if (docling_running or docling_starting) else "N/A"
         docling_pid = str(docling_status.get("pid")) if docling_status.get("pid") else "N/A"
 
         if self.docling_table:
@@ -375,15 +386,25 @@ class MonitorScreen(Screen):
         """Start docling serve."""
         self.operation_in_progress = True
         try:
-            success, message = await self.docling_manager.start()
+            # Start the service (this sets _starting = True internally at the start)
+            # Create task and let it begin executing (which sets the flag)
+            start_task = asyncio.create_task(self.docling_manager.start())
+            # Give it a tiny moment to set the _starting flag
+            await asyncio.sleep(0.1)
+            # Refresh immediately to show "Starting" status
+            await self._refresh_services()
+            # Now wait for start to complete
+            success, message = await start_task
             if success:
                 self.notify(message, severity="information")
             else:
                 self.notify(f"Failed to start docling serve: {message}", severity="error")
-            # Refresh the services table to show updated status
+            # Refresh again to show final status (running or stopped)
             await self._refresh_services()
         except Exception as e:
             self.notify(f"Error starting docling serve: {str(e)}", severity="error")
+            # Refresh on error to clear starting status
+            await self._refresh_services()
         finally:
             self.operation_in_progress = False
 
@@ -646,7 +667,11 @@ class MonitorScreen(Screen):
             suffix = f"-{random.randint(10000, 99999)}"
 
             # Add docling serve controls
-            docling_running = self.docling_manager.is_running()
+            docling_status = self.docling_manager.get_status()
+            docling_status_value = docling_status["status"]
+            docling_running = docling_status_value == "running"
+            docling_starting = docling_status_value == "starting"
+            
             if docling_running:
                 docling_controls.mount(
                     Button("Stop", variant="error", id=f"docling-stop-btn{suffix}")
@@ -654,6 +679,11 @@ class MonitorScreen(Screen):
                 docling_controls.mount(
                     Button("Restart", variant="primary", id=f"docling-restart-btn{suffix}")
                 )
+            elif docling_starting:
+                # Show disabled button or no button when starting
+                start_btn = Button("Starting...", variant="warning", id=f"docling-start-btn{suffix}")
+                start_btn.disabled = True
+                docling_controls.mount(start_btn)
             else:
                 docling_controls.mount(
                     Button("Start", variant="success", id=f"docling-start-btn{suffix}")