fixes

Author: maticzav

examples/stream.py

@@ -31,7 +31,10 @@ def stream_regular_task() -> None:
             for action in last_step.actions:
                 print(f" - {action}")
 
-    print("Done")
+        if res.status == "finished":
+            print(res.done_output)
+
+    print("Regular: DONE")
 
 
 stream_regular_task()

src/browser_use_sdk/resources/tasks.py

@@ -1307,8 +1307,16 @@ async def _watch(
 
             if res.status == "finished":
                 break
-
-            await asyncio.sleep(interval)
+            if res.status == "paused":
+                break
+            if res.status == "stopped":
+                break
+            if res.status == "started":
+                await asyncio.sleep(interval)
+            else:
+                raise ValueError(
+                    f"Expected one of 'finished', 'paused', 'stopped', or 'started' but received {res.status!r}"
+                )
 
     async def update(
         self,