bash timeout

Author: kcoopermiller

poetry.lock

@@ -3,7 +3,7 @@ name = "scrapybara"
 
 [tool.poetry]
 name = "scrapybara"
-version = "2.6.0-beta.0"
+version = "2.6.0-beta.1"
 description = ""
 readme = "README.md"
 authors = []

pyproject.toml

@@ -562,6 +562,14 @@ client.instance.bash(
 <dl>
 <dd>
 
+**timeout:** `typing.Optional[float]` 
+    
+</dd>
+</dl>
+
+<dl>
+<dd>
+
 **request_options:** `typing.Optional[RequestOptions]` — Request-specific configuration.
 
 </dd>

reference.md

@@ -922,6 +922,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(
@@ -931,6 +932,7 @@ def bash(
             restart=restart, 
             list_sessions=list_sessions, 
             check_session=check_session, 
+            timeout=timeout,
             request_options=request_options
         )
 
@@ -1458,6 +1460,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(
@@ -1467,6 +1470,7 @@ async def bash(
             restart=restart, 
             list_sessions=list_sessions, 
             check_session=check_session, 
+            timeout=timeout,
             request_options=request_options
         )
 

src/scrapybara/client.py

@@ -16,7 +16,7 @@ def get_headers(self) -> typing.Dict[str, str]:
         headers: typing.Dict[str, str] = {
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "scrapybara",
-            "X-Fern-SDK-Version": "2.6.0-beta.0",
+            "X-Fern-SDK-Version": "2.6.0-beta.1",
         }
         headers["x-api-key"] = self.api_key
         return headers

src/scrapybara/core/client_wrapper.py

@@ -219,6 +219,7 @@ def bash(
         restart: typing.Optional[bool] = OMIT,
         list_sessions: typing.Optional[bool] = OMIT,
         check_session: typing.Optional[int] = OMIT,
+        timeout: typing.Optional[float] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> BashResponse:
         """
@@ -236,6 +237,8 @@ def bash(
 
         check_session : typing.Optional[int]
 
+        timeout : typing.Optional[float]
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -264,6 +267,7 @@ def bash(
                 "restart": restart,
                 "list_sessions": list_sessions,
                 "check_session": check_session,
+                "timeout": timeout,
             },
             headers={
                 "content-type": "application/json",
@@ -1131,6 +1135,7 @@ async def bash(
         restart: typing.Optional[bool] = OMIT,
         list_sessions: typing.Optional[bool] = OMIT,
         check_session: typing.Optional[int] = OMIT,
+        timeout: typing.Optional[float] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> BashResponse:
         """
@@ -1148,6 +1153,8 @@ async def bash(
 
         check_session : typing.Optional[int]
 
+        timeout : typing.Optional[float]
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -1184,6 +1191,7 @@ async def main() -> None:
                 "restart": restart,
                 "list_sessions": list_sessions,
                 "check_session": check_session,
+                "timeout": timeout,
             },
             headers={
                 "content-type": "application/json",

src/scrapybara/instance/client.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,
         )