fixes

Author: maticzav

src/browser_use/wrapper/parse.py

@@ -2,7 +2,7 @@
 import json
 import typing
 from datetime import datetime
-from typing import Any, AsyncIterator, Generic, Iterator, TypeVar, Union
+from typing import Any, AsyncIterator, Generic, Iterator, Type, TypeVar, Union
 
 from pydantic import BaseModel
 
@@ -110,7 +110,7 @@ def watch(
 class WrappedStructuredTaskCreatedResponse(TaskCreatedResponse, Generic[T]):
     """TaskCreatedResponse with structured output."""
 
-    def __init__(self, id: str, schema: type[T], client: TasksClient):
+    def __init__(self, id: str, schema: Type[T], client: TasksClient):
         super().__init__(id=id)
 
         self._client = client
@@ -172,7 +172,7 @@ async def watch(
 class AsyncWrappedStructuredTaskCreatedResponse(TaskCreatedResponse, Generic[T]):
     """TaskCreatedResponse with structured output."""
 
-    def __init__(self, id: str, schema: type[T], client: AsyncTasksClient):
+    def __init__(self, id: str, schema: Type[T], client: AsyncTasksClient):
         super().__init__(id=id)
 
         self._client = client

src/browser_use/wrapper/tasks/client.py

@@ -31,7 +31,7 @@ def create_task(
         llm: typing.Optional[SupportedLlMs] = OMIT,
         start_url: typing.Optional[str] = OMIT,
         max_steps: typing.Optional[int] = OMIT,
-        schema: type[T],
+        schema: typing.Type[T],
         session_id: typing.Optional[str] = OMIT,
         metadata: typing.Optional[typing.Dict[str, typing.Optional[str]]] = OMIT,
         secrets: typing.Optional[typing.Dict[str, typing.Optional[str]]] = OMIT,
@@ -73,7 +73,7 @@ def create_task(
         start_url: typing.Optional[str] = OMIT,
         max_steps: typing.Optional[int] = OMIT,
         structured_output: typing.Optional[str] = OMIT,
-        schema: typing.Optional[type[BaseModel]] = OMIT,
+        schema: typing.Optional[typing.Type[BaseModel]] = OMIT,
         session_id: typing.Optional[str] = OMIT,
         metadata: typing.Optional[typing.Dict[str, typing.Optional[str]]] = OMIT,
         secrets: typing.Optional[typing.Dict[str, typing.Optional[str]]] = OMIT,
@@ -131,7 +131,7 @@ def create_task(
 
     @typing.overload
     def get_task(
-        self, task_id: str, schema: type[T], *, request_options: typing.Optional[RequestOptions] = None
+        self, task_id: str, schema: typing.Type[T], *, request_options: typing.Optional[RequestOptions] = None
     ) -> TaskViewWithOutput[T]: ...
 
     @typing.overload
@@ -140,7 +140,7 @@ def get_task(self, task_id: str, *, request_options: typing.Optional[RequestOpti
     def get_task(
         self,
         task_id: str,
-        schema: typing.Optional[typing.Union[type[BaseModel], str]] = OMIT,
+        schema: typing.Optional[typing.Union[typing.Type[BaseModel], str]] = OMIT,
         *,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> typing.Union[TaskViewWithOutput[T], TaskView]:
@@ -167,7 +167,7 @@ async def create_task(
         llm: typing.Optional[SupportedLlMs] = OMIT,
         start_url: typing.Optional[str] = OMIT,
         max_steps: typing.Optional[int] = OMIT,
-        schema: type[T],
+        schema: typing.Type[T],
         session_id: typing.Optional[str] = OMIT,
         metadata: typing.Optional[typing.Dict[str, typing.Optional[str]]] = OMIT,
         secrets: typing.Optional[typing.Dict[str, typing.Optional[str]]] = OMIT,
@@ -209,7 +209,7 @@ async def create_task(
         start_url: typing.Optional[str] = OMIT,
         max_steps: typing.Optional[int] = OMIT,
         structured_output: typing.Optional[str] = OMIT,
-        schema: typing.Optional[type[BaseModel]] = OMIT,
+        schema: typing.Optional[typing.Type[BaseModel]] = OMIT,
         session_id: typing.Optional[str] = OMIT,
         metadata: typing.Optional[typing.Dict[str, typing.Optional[str]]] = OMIT,
         secrets: typing.Optional[typing.Dict[str, typing.Optional[str]]] = OMIT,
@@ -265,7 +265,7 @@ async def create_task(
 
     @typing.overload
     async def get_task(
-        self, task_id: str, schema: type[T], *, request_options: typing.Optional[RequestOptions] = None
+        self, task_id: str, schema: typing.Type[T], *, request_options: typing.Optional[RequestOptions] = None
     ) -> TaskViewWithOutput[T]: ...
 
     @typing.overload
@@ -274,7 +274,7 @@ async def get_task(self, task_id: str, *, request_options: typing.Optional[Reque
     async def get_task(
         self,
         task_id: str,
-        schema: typing.Optional[typing.Union[type[BaseModel], str]] = OMIT,
+        schema: typing.Optional[typing.Union[typing.Type[BaseModel], str]] = OMIT,
         *,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> typing.Union[TaskViewWithOutput[T], TaskView]: