fixes

Author: maticzav

examples/async_retrieve.py

@@ -1,14 +1,14 @@
 #!/usr/bin/env -S rye run python
 
 import asyncio
+import os
 from typing import List
 
 from pydantic import BaseModel
 
 from browser_use import AsyncBrowserUse
 
-# gets API Key from environment variable BROWSER_USE_API_KEY
-client = AsyncBrowserUse()
+client = AsyncBrowserUse(api_key=os.getenv("BROWSER_USE_API_KEY"))
 
 
 # Regular Task

examples/async_run.py

@@ -1,14 +1,15 @@
 #!/usr/bin/env -S rye run python
 
 import asyncio
+import os
 from typing import List
 
 from pydantic import BaseModel
 
 from browser_use import AsyncBrowserUse
 
 # gets API Key from environment variable BROWSER_USE_API_KEY
-client = AsyncBrowserUse()
+client = AsyncBrowserUse(api_key=os.getenv("BROWSER_USE_API_KEY"))
 
 
 # Regular Task

examples/async_stream.py

@@ -1,14 +1,14 @@
 #!/usr/bin/env -S rye run python
 
 import asyncio
+import os
 from typing import List
 
 from pydantic import BaseModel
 
 from browser_use import AsyncBrowserUse
 
-# gets API Key from environment variable BROWSER_USE_API_KEY
-client = AsyncBrowserUse()
+client = AsyncBrowserUse(api_key=os.getenv("BROWSER_USE_API_KEY"))
 
 
 # Regular Task

examples/retrieve.py

@@ -1,14 +1,14 @@
 #!/usr/bin/env -S rye run python
 
+import os
 import time
 from typing import List
 
 from pydantic import BaseModel
 
 from browser_use import BrowserUse
 
-# gets API Key from environment variable BROWSER_USE_API_KEY
-client = BrowserUse()
+client = BrowserUse(api_key=os.getenv("BROWSER_USE_API_KEY"))
 
 
 # Regular Task

examples/run.py

@@ -1,13 +1,13 @@
 #!/usr/bin/env -S rye run python
 
+import os
 from typing import List
 
 from pydantic import BaseModel
 
 from browser_use import BrowserUse
 
-# gets API Key from environment variable BROWSER_USE_API_KEY
-client = BrowserUse()
+client = BrowserUse(api_key=os.getenv("BROWSER_USE_API_KEY"))
 
 
 # Regular Task

examples/stream.py

@@ -1,13 +1,13 @@
 #!/usr/bin/env -S rye run python
 
+import os
 from typing import List
 
 from pydantic import BaseModel
 
 from browser_use import BrowserUse
 
-# gets API Key from environment variable BROWSER_USE_API_KEY
-client = BrowserUse()
+client = BrowserUse(api_key=os.getenv("BROWSER_USE_API_KEY"))
 
 
 # Regular Task

src/browser_use/wrapper/parse.py

@@ -80,8 +80,8 @@ class WrappedTaskCreatedResponse(TaskCreatedResponse):
     """TaskCreatedResponse with utility methods for easier interfacing with Browser Use Cloud."""
 
     def __init__(self, id: str, client: TasksClient):
-        super().__init__()
-        self.id = id
+        super().__init__(id=id)
+        self._client = client
 
     def complete(
         self, interval: float = 1, request_options: typing.Optional[RequestOptions] = None
@@ -111,9 +111,9 @@ class WrappedStructuredTaskCreatedResponse(TaskCreatedResponse, Generic[T]):
     """TaskCreatedResponse with structured output."""
 
     def __init__(self, id: str, schema: type[T], client: TasksClient):
-        super().__init__()
+        super().__init__(id=id)
 
-        self.id = id
+        self._client = client
         self._schema = schema
 
     def complete(
@@ -144,8 +144,8 @@ class AsyncWrappedTaskCreatedResponse(TaskCreatedResponse):
     """TaskCreatedResponse with utility methods for easier interfacing with Browser Use Cloud."""
 
     def __init__(self, id: str, client: AsyncTasksClient):
-        super().__init__()
-        self.id = id
+        super().__init__(id=id)
+        self._client = client
 
     async def complete(self, interval: float = 1, request_options: typing.Optional[RequestOptions] = None) -> TaskView:
         """Waits for the task to finish and return the result."""
@@ -173,9 +173,9 @@ class AsyncWrappedStructuredTaskCreatedResponse(TaskCreatedResponse, Generic[T])
     """TaskCreatedResponse with structured output."""
 
     def __init__(self, id: str, schema: type[T], client: AsyncTasksClient):
-        super().__init__()
+        super().__init__(id=id)
 
-        self.id = id
+        self._client = client
         self._schema = schema
 
     async def complete(

src/browser_use/wrapper/tasks/client.py

@@ -105,6 +105,7 @@ def create_task(
                 system_prompt_extension=system_prompt_extension,
                 request_options=request_options,
             )
+
             return WrappedStructuredTaskCreatedResponse[T](id=res.id, schema=schema, client=self)
 
         else:
@@ -125,6 +126,7 @@ def create_task(
                 system_prompt_extension=system_prompt_extension,
                 request_options=request_options,
             )
+
             return WrappedTaskCreatedResponse(id=res.id, client=self)
 
     @typing.overload
@@ -222,7 +224,7 @@ async def create_task(
         if schema is not None:
             structured_output = json.dumps(schema.model_json_schema())
 
-            res = super().create_task(
+            res = await super().create_task(
                 task=task,
                 llm=llm,
                 start_url=start_url,
@@ -242,7 +244,7 @@ async def create_task(
             return AsyncWrappedStructuredTaskCreatedResponse[T](id=res.id, schema=schema, client=self)
 
         else:
-            res = super().create_task(
+            res = await super().create_task(
                 task=task,
                 llm=llm,
                 start_url=start_url,