add timeout

Author: kcoopermiller

src/scrapybara/client.py

@@ -887,6 +887,7 @@ def bash(
         restart: Optional[bool] = OMIT,
         list_sessions: Optional[bool] = OMIT,
         check_session: Optional[int] = OMIT,
+        timeout: Optional[float] = None,
         request_options: Optional[RequestOptions] = None,
     ) -> Optional[Any]:
         return self._client.instance.bash(
@@ -896,6 +897,7 @@ def bash(
             restart=restart, 
             list_sessions=list_sessions, 
             check_session=check_session, 
+            timeout=timeout,
             request_options=request_options
         )
 
@@ -1411,6 +1413,7 @@ async def bash(
         restart: Optional[bool] = OMIT,
         list_sessions: Optional[bool] = OMIT,
         check_session: Optional[int] = OMIT,
+        timeout: Optional[float] = None,
         request_options: Optional[RequestOptions] = None,
     ) -> Optional[Any]:
         return await self._client.instance.bash(
@@ -1420,6 +1423,7 @@ async def bash(
             restart=restart, 
             list_sessions=list_sessions, 
             check_session=check_session, 
+            timeout=timeout,
             request_options=request_options
         )
 

src/scrapybara/tools/__init__.py

@@ -169,6 +169,7 @@ class BashToolParameters(BaseModel):
     restart: Optional[bool] = Field(False, description="Whether to restart the shell")
     list_sessions: Optional[bool] = Field(None, description="Whether to list all bash sessions")
     check_session: Optional[int] = Field(None, description="Session ID to check status")
+    timeout: Optional[float] = Field(None, description="Timeout for the command")
 
 
 class BashTool(Tool):
@@ -193,6 +194,7 @@ def __call__(self, **kwargs: Any) -> Any:
             session=params.session,
             restart=params.restart,
             list_sessions=params.list_sessions,
-            check_session=params.check_session
+            check_session=params.check_session,
+            timeout=params.timeout,
         )