copied request_timeout conventions to get_timeout method

e2b-dev · Dec 26, 2024 · 3edaa2b · 3edaa2b
1 parent 81c1f81
commit 3edaa2b
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/packages/python-sdk/e2b/sandbox_async/main.py b/packages/python-sdk/e2b/sandbox_async/main.py
@@ -366,11 +366,17 @@ async def set_timeout(  # type: ignore
             **self.connection_config.__dict__,
         )
 
-    async def get_timeout(self) -> datetime:  # type: ignore
+    async def get_timeout(  # type: ignore
+        self,
+        request_timeout: Optional[float] = None,
+    ) -> datetime:
         config_dict = self.connection_config.__dict__
         config_dict.pop("access_token", None)
         config_dict.pop("api_url", None)
 
+        if request_timeout:
+            config_dict["request_timeout"] = request_timeout
+
         sbx = await SandboxApi.get(
             sandbox_id=self.sandbox_id,
             **self.connection_config.__dict__,

diff --git a/packages/python-sdk/e2b/sandbox_sync/main.py b/packages/python-sdk/e2b/sandbox_sync/main.py
@@ -359,11 +359,15 @@ def set_timeout(  # type: ignore
 
     def get_timeout(  # type: ignore
         self,
+        request_timeout: Optional[float] = None,
     ) -> datetime:
         config_dict = self.connection_config.__dict__
         config_dict.pop("access_token", None)
         config_dict.pop("api_url", None)
 
+        if request_timeout:
+            config_dict["request_timeout"] = request_timeout
+
         sbx = SandboxApi.get(
             sandbox_id=self.sandbox_id,
             **self.connection_config.__dict__,