feat: display HF job URLs immediately when job starts running

- Send job URL immediately after job creation (before waiting for completion)
- Store job URLs in frontend store keyed by tool_call_id
- Display job URL link next to running indicator during execution
- Remove job URL from final output section (only show status there)
- Fix race condition by passing tool_call_id directly to handlers

Co-authored-by: Cursor <cursoragent@cursor.com>

agent/core/agent_loop.py

@@ -612,7 +612,7 @@ class Handlers:
             )
 
             output, success = await session.tool_router.call_tool(
-                tool_name, tool_args, session=session
+                tool_name, tool_args, session=session, tool_call_id=tc.id
             )
 
             return (tc, tool_name, output, success)
@@ -662,8 +662,15 @@ class Handlers:
             rejection_msg = "Job execution cancelled by user"
             user_feedback = approval_decision.get("feedback")
             if user_feedback:
-                rejection_msg += f". User feedback: {user_feedback}"
-
+                # Ensure feedback is a string and sanitize any problematic characters
+                feedback_str = str(user_feedback).strip()
+                # Remove any control characters that might break JSON parsing
+                feedback_str = "".join(char for char in feedback_str if ord(char) >= 32 or char in "\n\t")
+                rejection_msg += f". User feedback: {feedback_str}"
+
+            # Ensure rejection_msg is a clean string
+            rejection_msg = str(rejection_msg).strip()
+            
             tool_msg = Message(
                 role="tool",
                 content=rejection_msg,

agent/core/tools.py

@@ -223,7 +223,7 @@ class ToolRouter:
 
     @observe(name="call_tool")
     async def call_tool(
-        self, tool_name: str, arguments: dict[str, Any], session: Any = None
+        self, tool_name: str, arguments: dict[str, Any], session: Any = None, tool_call_id: str | None = None
     ) -> tuple[str, bool]:
         """
         Call a tool and return (output_string, success_bool).
@@ -239,6 +239,9 @@ class ToolRouter:
             # Check if handler accepts session argument
             sig = inspect.signature(tool.handler)
             if "session" in sig.parameters:
+                # Check if handler also accepts tool_call_id parameter
+                if "tool_call_id" in sig.parameters:
+                    return await tool.handler(arguments, session=session, tool_call_id=tool_call_id)
                 return await tool.handler(arguments, session=session)
             return await tool.handler(arguments)
 

agent/tools/jobs_tool.py

@@ -282,11 +282,15 @@ class HfJobsTool:
         hf_token: Optional[str] = None,
         namespace: Optional[str] = None,
         log_callback: Optional[Callable[[str], Awaitable[None]]] = None,
+        session: Any = None,
+        tool_call_id: Optional[str] = None,
     ):
         self.hf_token = hf_token
         self.api = HfApi(token=hf_token)
         self.namespace = namespace
         self.log_callback = log_callback
+        self.session = session
+        self.tool_call_id = tool_call_id
 
     async def execute(self, params: Dict[str, Any]) -> ToolResult:
         """Execute the specified operation"""
@@ -512,6 +516,20 @@ class HfJobsTool:
                 namespace=self.namespace,
             )
 
+            # Send job URL immediately after job creation (before waiting for completion)
+            if self.session and self.tool_call_id:
+                await self.session.send_event(
+                    Event(
+                        event_type="tool_state_change",
+                        data={
+                            "tool_call_id": self.tool_call_id,
+                            "tool": "hf_jobs",
+                            "state": "running",
+                            "jobUrl": job.url,
+                        },
+                    )
+                )
+
             # Wait for completion and stream logs
             logger.info(f"{job_type} job started: {job.url}")
             logger.info("Streaming logs...")
@@ -1012,7 +1030,7 @@ HF_JOBS_TOOL_SPEC = {
 
 
 async def hf_jobs_handler(
-    arguments: Dict[str, Any], session: Any = None
+    arguments: Dict[str, Any], session: Any = None, tool_call_id: str | None = None
 ) -> tuple[str, bool]:
     """Handler for agent tool router"""
     try:
@@ -1035,6 +1053,8 @@ async def hf_jobs_handler(
             namespace=namespace,
             hf_token=hf_token,
             log_callback=log_callback if session else None,
+            session=session,
+            tool_call_id=tool_call_id,
         )
         result = await tool.execute(arguments)
         return result["formatted"], not result.get("isError", False)

frontend/src/lib/ws-chat-transport.ts

@@ -9,6 +9,7 @@ import type { ChatTransport, UIMessage, UIMessageChunk, ChatRequestOptions } fro
 import { apiFetch, getWebSocketUrl } from '@/utils/api';
 import { logger } from '@/utils/logger';
 import type { AgentEvent } from '@/types/events';
+import { useAgentStore } from '@/store/agentStore';
 
 // ---------------------------------------------------------------------------
 // Side-channel callback interface (non-chat events forwarded to the store)
@@ -499,9 +500,15 @@ export class WebSocketChatTransport implements ChatTransport<UIMessage> {
         if (!this.streamController) break;
         const tcId = (event.data?.tool_call_id as string) || '';
         const state = (event.data?.state as string) || '';
+        const jobUrl = (event.data?.jobUrl as string) || undefined;
 
         if (tcId.startsWith('plan_tool')) break;
 
+        // Store job URL if provided
+        if (jobUrl && tcId) {
+          useAgentStore.getState().setJobUrl(tcId, jobUrl);
+        }
+
         if (state === 'rejected' || state === 'abandoned') {
           this.enqueue({ type: 'tool-output-denied', toolCallId: tcId });
         }

frontend/src/store/agentStore.ts

@@ -59,6 +59,9 @@ interface AgentStore {
   // Edited scripts (tool_call_id -> edited content)
   editedScripts: Record<string, string>;
 
+  // Job URLs (tool_call_id -> job URL) for HF jobs
+  jobUrls: Record<string, string>;
+
   // Actions
   setProcessing: (isProcessing: boolean) => void;
   setConnected: (isConnected: boolean) => void;
@@ -80,6 +83,9 @@ interface AgentStore {
   setEditedScript: (toolCallId: string, content: string) => void;
   getEditedScript: (toolCallId: string) => string | undefined;
   clearEditedScripts: () => void;
+
+  setJobUrl: (toolCallId: string, jobUrl: string) => void;
+  getJobUrl: (toolCallId: string) => string | undefined;
 }
 
 export const useAgentStore = create<AgentStore>()((set, get) => ({
@@ -97,6 +103,7 @@ export const useAgentStore = create<AgentStore>()((set, get) => ({
   plan: [],
 
   editedScripts: {},
+  jobUrls: {},
 
   // ── Global flags ──────────────────────────────────────────────────
 
@@ -180,4 +187,14 @@ export const useAgentStore = create<AgentStore>()((set, get) => ({
   getEditedScript: (toolCallId) => get().editedScripts[toolCallId],
 
   clearEditedScripts: () => set({ editedScripts: {} }),
+
+  // ── Job URLs ────────────────────────────────────────────────────────
+
+  setJobUrl: (toolCallId, jobUrl) => {
+    set((state) => ({
+      jobUrls: { ...state.jobUrls, [toolCallId]: jobUrl },
+    }));
+  },
+
+  getJobUrl: (toolCallId) => get().jobUrls[toolCallId],
 }));