diff --git a/src/uipath/runtime/base.py b/src/uipath/runtime/base.py
index 246e974..a9ec469 100644
--- a/src/uipath/runtime/base.py
+++ b/src/uipath/runtime/base.py
@@ -195,8 +195,8 @@ async def stream(
         """Stream runtime execution with context.
 
         Args:
-            runtime: The runtime instance
-            context: The runtime context
+            input: The input data for the runtime
+            options: Streaming-specific execution options
 
         Yields:
             UiPathRuntimeEvent instances during execution and final UiPathRuntimeResult