log streaming

agent/core/agent_loop.py

@@ -203,7 +203,7 @@ class Handlers:
                         )
 
                         output, success = await session.tool_router.call_tool(
-                            tool_name, tool_args
+                            tool_name, tool_args, session=session
                         )
 
                     # Add tool result to history
@@ -376,7 +376,9 @@ class Handlers:
                 )
             )
 
-            output, success = await session.tool_router.call_tool(tool_name, tool_args)
+            output, success = await session.tool_router.call_tool(
+                tool_name, tool_args, session=session
+            )
 
             return (tc, tool_name, output, success)
 

agent/core/tools.py

@@ -219,7 +219,7 @@ class ToolRouter:
 
     @observe(name="call_tool")
     async def call_tool(
-        self, tool_name: str, arguments: dict[str, Any]
+        self, tool_name: str, arguments: dict[str, Any], session: Any = None
     ) -> tuple[str, bool]:
         """
         Call a tool and return (output_string, success_bool).
@@ -230,6 +230,12 @@ class ToolRouter:
         # Check if this is a built-in tool with a handler
         tool = self.tools.get(tool_name)
         if tool and tool.handler:
+            import inspect
+
+            # Check if handler accepts session argument
+            sig = inspect.signature(tool.handler)
+            if "session" in sig.parameters:
+                return await tool.handler(arguments, session=session)
             return await tool.handler(arguments)
 
         # Otherwise, use MCP client

agent/tools/jobs_tool.py

@@ -9,12 +9,13 @@ import base64
 import http.client
 import os
 import re
-from typing import Any, Dict, Literal, Optional
+from typing import Any, Dict, Literal, Optional, Callable, Awaitable
 
 import httpx
 from huggingface_hub import HfApi
 from huggingface_hub.utils import HfHubHTTPError
 
+from agent.core.session import Event
 from agent.tools.types import ToolResult
 from agent.tools.utilities import (
     format_job_details,
@@ -269,9 +270,15 @@ def _scheduled_job_info_to_dict(scheduled_job_info) -> Dict[str, Any]:
 class HfJobsTool:
     """Tool for managing Hugging Face compute jobs using huggingface-hub library"""
 
-    def __init__(self, hf_token: Optional[str] = None, namespace: Optional[str] = None):
+    def __init__(
+        self,
+        hf_token: Optional[str] = None,
+        namespace: Optional[str] = None,
+        log_callback: Optional[Callable[[str], Awaitable[None]]] = None,
+    ):
         self.api = HfApi(token=hf_token)
         self.namespace = namespace
+        self.log_callback = log_callback
 
     async def execute(self, params: Dict[str, Any]) -> ToolResult:
         """Execute the specified operation"""
@@ -366,6 +373,8 @@ class HfJobsTool:
                 # Stream logs in real-time
                 for log_line in logs_gen:
                     print("\t" + log_line)
+                    if self.log_callback:
+                        await self.log_callback(log_line)
                     all_logs.append(log_line)
 
                 # If we get here, streaming completed normally
@@ -961,10 +970,22 @@ HF_JOBS_TOOL_SPEC = {
 }
 
 
-async def hf_jobs_handler(arguments: Dict[str, Any]) -> tuple[str, bool]:
+async def hf_jobs_handler(
+    arguments: Dict[str, Any], session: Any = None
+) -> tuple[str, bool]:
     """Handler for agent tool router"""
     try:
-        tool = HfJobsTool(namespace=os.environ.get("HF_NAMESPACE", ""))
+
+        async def log_callback(log: str):
+            if session:
+                await session.send_event(
+                    Event(event_type="tool_log", data={"tool": "hf_jobs", "log": log})
+                )
+
+        tool = HfJobsTool(
+            namespace=os.environ.get("HF_NAMESPACE", ""),
+            log_callback=log_callback if session else None,
+        )
         result = await tool.execute(arguments)
         return result["formatted"], not result.get("isError", False)
     except Exception as e:

frontend/src/hooks/useAgentWebSocket.ts

@@ -115,6 +115,39 @@ export function useAgentWebSocket({
           break;
         }
 
+        case 'tool_log': {
+          const toolName = (event.data?.tool as string) || 'unknown';
+          const log = (event.data?.log as string) || '';
+
+          if (toolName === 'hf_jobs') {
+            const currentPanel = useAgentStore.getState().panelContent;
+            
+            // If we are already showing logs, append
+            // If we are showing "Compute Job Script", overwrite/switch to logs
+            // Otherwise, initialize
+            
+            let newContent = log;
+            if (currentPanel?.title === 'Job Logs') {
+              newContent = currentPanel.content + '\n' + log;
+            } else if (currentPanel?.title === 'Compute Job Script') {
+               // We were showing the script, now logs start.
+               // Maybe we want to clear and start showing logs.
+               newContent = '--- Starting execution ---\n' + log;
+            }
+
+            setPanelContent({
+              title: 'Job Logs',
+              content: newContent,
+              language: 'text'
+            });
+            
+            if (!useLayoutStore.getState().isRightPanelOpen) {
+                 setRightPanelOpen(true);
+            }
+          }
+          break;
+        }
+
         case 'approval_required': {
           const tools = event.data?.tools as Array<{
             tool: string;

frontend/src/types/events.ts

@@ -8,6 +8,7 @@ export type EventType =
   | 'assistant_message'
   | 'tool_call'
   | 'tool_output'
+  | 'tool_log'
   | 'approval_required'
   | 'turn_complete'
   | 'compacted'
@@ -44,6 +45,11 @@ export interface ToolOutputEventData {
   success: boolean;
 }
 
+export interface ToolLogEventData {
+  tool: string;
+  log: string;
+}
+
 export interface ApprovalRequiredEventData {
   tools: ApprovalToolItem[];
   count: number;