fixes

Author: maticzav

examples/retrieve.py

@@ -11,19 +11,19 @@
 client = BrowserUse()
 
 # Regular Task
-task = client.tasks.create(
+regular_task = client.tasks.create(
     task="""
     Find top 10 Hacker News articles and return the title and url.
     """
 )
 
-print(f"Task ID: {task.id}")
+print(f"Task ID: {regular_task.id}")
 
 while True:
-    regular = client.tasks.retrieve(task.id)
-    print(regular.status)
-    if regular.status == "finished":
-        print(regular.done_output)
+    regular_status = client.tasks.retrieve(regular_task.id)
+    print(regular_status.status)
+    if regular_status.status == "finished":
+        print(regular_status.done_output)
         break
 
     time.sleep(1)
@@ -39,24 +39,24 @@ class SearchResult(BaseModel):
     posts: List[HackerNewsPost]
 
 
-structured = client.tasks.create(
+structured_task = client.tasks.create(
     task="""
     Find top 10 Hacker News articles and return the title and url.
     """,
     structured_output_json=SearchResult,
 )
 
-print(f"Task ID: {structured.id}")
+print(f"Task ID: {structured_task.id}")
 
 while True:
-    structured = client.tasks.retrieve(task_id=structured.id, structured_output_json=SearchResult)
-    print(structured.status)
+    structured_status = client.tasks.retrieve(task_id=structured_task.id, structured_output_json=SearchResult)
+    print(structured_status.status)
 
-    if structured.status == "finished":
-        if structured.parsed_output is None:
+    if structured_status.status == "finished":
+        if structured_status.parsed_output is None:
             print("No output")
         else:
-            for post in structured.parsed_output.posts:
+            for post in structured_status.parsed_output.posts:
                 print(f" - {post.title} - {post.url}")
 
         break

src/browser_use_sdk/resources/tasks.py

@@ -113,6 +113,9 @@ def run(
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
     ) -> Union[TaskView, TaskViewWithOutput[BaseModel]]:
+        """
+        Run a new task and return the task view.
+        """
         if structured_output_json is not None and isinstance(structured_output_json, type):
             create_task_res = self.create(
                 task=task,
@@ -128,9 +131,9 @@ def run(
                 timeout=timeout,
             )
 
-            for msg in self.stream(create_task_res.id, structured_output_json=structured_output_json):
-                if msg.status == "finished":
-                    return msg
+            for structured_msg in self.stream(create_task_res.id, structured_output_json=structured_output_json):
+                if structured_msg.status == "finished":
+                    return structured_msg
 
             raise ValueError("Task did not finish")
 
@@ -341,7 +344,7 @@ def retrieve(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> TaskView:
+    ) -> Union[TaskView, TaskViewWithOutput[BaseModel]]:
         """
         Get detailed information about a specific AI agent task.
 
@@ -390,14 +393,14 @@ def retrieve(
             )
 
             if res.done_output is None:
-                return TaskViewWithOutput[structured_output_json](
+                return TaskViewWithOutput[BaseModel](
                     **res.model_dump(),
                     parsed_output=None,
                 )
 
             parsed_output = structured_output_json.model_validate_json(res.done_output)
 
-            return TaskViewWithOutput[structured_output_json](
+            return TaskViewWithOutput[BaseModel](
                 **res.model_dump(),
                 parsed_output=parsed_output,
             )
@@ -462,14 +465,14 @@ def stream(
         ):
             if structured_output_json is not None and isinstance(structured_output_json, type):
                 if res.done_output is None:
-                    yield TaskViewWithOutput[structured_output_json](
+                    yield TaskViewWithOutput[BaseModel](
                         **res.model_dump(),
                         parsed_output=None,
                     )
                 else:
                     parsed_output = structured_output_json.model_validate_json(res.done_output)
 
-                    yield TaskViewWithOutput[structured_output_json](
+                    yield TaskViewWithOutput[BaseModel](
                         **res.model_dump(),
                         parsed_output=parsed_output,
                     )