fixes

Author: maticzav

examples/utils.py examples/api.pyexamples/utils.py renamed to examples/api.py

@@ -3,8 +3,8 @@
 import asyncio
 from typing import List
 
+from api import API_KEY
 from pydantic import BaseModel
-from utils import API_KEY
 
 from browser_use import AsyncBrowserUse
 

examples/async_retrieve.py

@@ -3,8 +3,8 @@
 import asyncio
 from typing import List
 
+from api import API_KEY
 from pydantic import BaseModel
-from utils import API_KEY
 
 from browser_use import AsyncBrowserUse
 

examples/async_run.py

@@ -3,8 +3,8 @@
 import asyncio
 from typing import List
 
+from api import API_KEY
 from pydantic import BaseModel
-from utils import API_KEY
 
 from browser_use import AsyncBrowserUse
 

examples/async_stream.py

@@ -3,8 +3,8 @@
 import time
 from typing import List
 
+from api import API_KEY
 from pydantic import BaseModel
-from utils import API_KEY
 
 from browser_use import BrowserUse
 

examples/retrieve.py

@@ -2,8 +2,8 @@
 
 from typing import List
 
+from api import API_KEY
 from pydantic import BaseModel
-from utils import API_KEY
 
 from browser_use import BrowserUse
 

examples/run.py

@@ -2,8 +2,8 @@
 
 from typing import List
 
+from api import API_KEY
 from pydantic import BaseModel
-from utils import API_KEY
 
 from browser_use import BrowserUse
 

examples/stream.py

@@ -84,7 +84,7 @@ def create_task(
         system_prompt_extension: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> typing.Union[WrappedStructuredTaskCreatedResponse[T], WrappedTaskCreatedResponse]:
-        if schema is not None:
+        if schema is not None and schema is not OMIT:
             structured_output = json.dumps(schema.model_json_schema())
 
             res = super().create_task(
@@ -145,7 +145,7 @@ def get_task(
     ) -> typing.Union[TaskViewWithOutput[T], TaskView]:
         res = super().get_task(task_id, request_options=request_options)
 
-        if schema is not None:
+        if schema is not None and schema is not OMIT:
             return _parse_task_view_with_output(res, schema)
         else:
             return res
@@ -219,7 +219,7 @@ async def create_task(
         system_prompt_extension: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> typing.Union[AsyncWrappedStructuredTaskCreatedResponse[T], AsyncWrappedTaskCreatedResponse]:
-        if schema is not None:
+        if schema is not None and schema is not OMIT:
             structured_output = json.dumps(schema.model_json_schema())
 
             res = await super().create_task(
@@ -278,7 +278,7 @@ async def get_task(
     ) -> typing.Union[TaskViewWithOutput[T], TaskView]:
         res = await super().get_task(task_id, request_options=request_options)
 
-        if schema is not None:
+        if schema is not None and schema is not OMIT:
             return _parse_task_view_with_output(res, schema)
         else:
             return res