javaos74
diff --git a/‎src/uipath/dev/services/run_service.py‎
Lines changed: 25 additions & 5 deletions b/‎src/uipath/dev/services/run_service.py‎
Lines changed: 25 additions & 5 deletions
@@ -5,7 +5,7 @@
 import json
 import traceback
 from datetime import datetime
-from typing import Any, Callable, cast
+from typing import Any, Callable, Literal, Protocol, cast
 
 from pydantic import BaseModel
 from uipath.core.tracing import UiPathTraceManager
@@ -18,7 +18,11 @@
     UiPathRuntimeStatus,
     UiPathStreamOptions,
 )
-from uipath.runtime.debug import UiPathDebugProtocol, UiPathDebugRuntime
+from uipath.runtime.debug import (
+    UiPathBreakpointResult,
+    UiPathDebugProtocol,
+    UiPathDebugRuntime,
+)
 from uipath.runtime.errors import UiPathErrorContract, UiPathRuntimeError
 from uipath.runtime.events import UiPathRuntimeMessageEvent, UiPathRuntimeStateEvent
 
@@ -30,7 +34,23 @@
 LogCallback = Callable[[LogData], None]
 TraceCallback = Callable[[TraceData], None]
 ChatCallback = Callable[[ChatData], None]
-DebugBridgeFactory = Callable[[ExecutionMode], UiPathDebugProtocol]
+
+
+class DebugBridgeProtocol(UiPathDebugProtocol, Protocol):
+    """Extended debug bridge protocol with control methods and callbacks."""
+
+    on_execution_started: Callable[[], None] | None
+    on_state_update: Callable[[UiPathRuntimeStateEvent], None] | None
+    on_breakpoint_hit: Callable[[UiPathBreakpointResult], None] | None
+    on_execution_completed: Callable[[UiPathRuntimeResult], None] | None
+    on_execution_error: Callable[[str], None] | None
+
+    def resume(self, resume_data: Any) -> None: ...
+    def quit(self) -> None: ...
+    def set_breakpoints(self, breakpoints: list[str] | Literal["*"]) -> None: ...
+
+
+DebugBridgeFactory = Callable[[ExecutionMode], DebugBridgeProtocol]
 
 
 class RunService:
@@ -71,7 +91,7 @@ def __init__(
             batch=False,
         )
 
-        self.debug_bridges: dict[str, UiPathDebugProtocol] = {}
+        self.debug_bridges: dict[str, DebugBridgeProtocol] = {}
 
     def register_run(self, run: ExecutionRun) -> None:
         """Register a new run and emit an initial update."""
@@ -276,7 +296,7 @@ def handle_trace(self, trace_data: TraceData) -> None:
         if self.on_trace is not None:
             self.on_trace(trace_data)
 
-    def get_debug_bridge(self, run_id: str) -> UiPathDebugProtocol | None:
+    def get_debug_bridge(self, run_id: str) -> DebugBridgeProtocol | None:
         """Get the debug bridge for a run."""
         return self.debug_bridges.get(run_id)