blaxel 0.2.31__py3-none-any.whl → 0.2.31rc121__py3-none-any.whl

blaxel/core/sandbox/client/api/filesystem/get_filesystem_content_search_path.py

@@ -42,9 +42,7 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(
-    *, client: Client, response: httpx.Response
-) -> Union[ContentSearchResponse, ErrorResponse] | None:
+def _parse_response(*, client: Client, response: httpx.Response) -> Union[ContentSearchResponse, ErrorResponse] | None:
     if response.status_code == 200:
         response_200 = ContentSearchResponse.from_dict(response.json())
 

blaxel/core/sandbox/client/api/filesystem/get_filesystem_find_path.py

@@ -42,9 +42,7 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(
-    *, client: Client, response: httpx.Response
-) -> Union[ErrorResponse, FindResponse] | None:
+def _parse_response(*, client: Client, response: httpx.Response) -> Union[ErrorResponse, FindResponse] | None:
     if response.status_code == 200:
         response_200 = FindResponse.from_dict(response.json())
 
@@ -67,9 +65,7 @@ def _parse_response(
         return None
 
 
-def _build_response(
-    *, client: Client, response: httpx.Response
-) -> Response[Union[ErrorResponse, FindResponse]]:
+def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[ErrorResponse, FindResponse]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

blaxel/core/sandbox/client/api/filesystem/get_filesystem_search_path.py

@@ -39,9 +39,7 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(
-    *, client: Client, response: httpx.Response
-) -> Union[ErrorResponse, FuzzySearchResponse] | None:
+def _parse_response(*, client: Client, response: httpx.Response) -> Union[ErrorResponse, FuzzySearchResponse] | None:
     if response.status_code == 200:
         response_200 = FuzzySearchResponse.from_dict(response.json())
 
@@ -64,9 +62,7 @@ def _parse_response(
         return None
 
 
-def _build_response(
-    *, client: Client, response: httpx.Response
-) -> Response[Union[ErrorResponse, FuzzySearchResponse]]:
+def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[ErrorResponse, FuzzySearchResponse]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

blaxel/core/sandbox/client/api/filesystem/get_watch_filesystem_path.py

@@ -29,9 +29,7 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(
-    *, client: Client, response: httpx.Response
-) -> Union[ErrorResponse, str] | None:
+def _parse_response(*, client: Client, response: httpx.Response) -> Union[ErrorResponse, str] | None:
     if response.status_code == 200:
         response_200 = response.text
         return response_200
@@ -49,9 +47,7 @@ def _parse_response(
         return None
 
 
-def _build_response(
-    *, client: Client, response: httpx.Response
-) -> Response[Union[ErrorResponse, str]]:
+def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[ErrorResponse, str]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

blaxel/core/sandbox/client/api/filesystem/post_filesystem_multipart_upload_id_complete.py

@@ -35,9 +35,7 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(
-    *, client: Client, response: httpx.Response
-) -> Union[ErrorResponse, SuccessResponse] | None:
+def _parse_response(*, client: Client, response: httpx.Response) -> Union[ErrorResponse, SuccessResponse] | None:
     if response.status_code == 200:
         response_200 = SuccessResponse.from_dict(response.json())
 
@@ -60,9 +58,7 @@ def _parse_response(
         return None
 
 
-def _build_response(
-    *, client: Client, response: httpx.Response
-) -> Response[Union[ErrorResponse, SuccessResponse]]:
+def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[ErrorResponse, SuccessResponse]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

blaxel/core/sandbox/client/api/filesystem/put_filesystem_path.py

@@ -35,9 +35,7 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(
-    *, client: Client, response: httpx.Response
-) -> Union[ErrorResponse, SuccessResponse] | None:
+def _parse_response(*, client: Client, response: httpx.Response) -> Union[ErrorResponse, SuccessResponse] | None:
     if response.status_code == 200:
         response_200 = SuccessResponse.from_dict(response.json())
 
@@ -60,9 +58,7 @@ def _parse_response(
         return None
 
 
-def _build_response(
-    *, client: Client, response: httpx.Response
-) -> Response[Union[ErrorResponse, SuccessResponse]]:
+def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[ErrorResponse, SuccessResponse]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

blaxel/core/sandbox/client/api/process/delete_process_identifier.py

@@ -21,9 +21,7 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(
-    *, client: Client, response: httpx.Response
-) -> Union[ErrorResponse, SuccessResponse] | None:
+def _parse_response(*, client: Client, response: httpx.Response) -> Union[ErrorResponse, SuccessResponse] | None:
     if response.status_code == 200:
         response_200 = SuccessResponse.from_dict(response.json())
 
@@ -46,9 +44,7 @@ def _parse_response(
         return None
 
 
-def _build_response(
-    *, client: Client, response: httpx.Response
-) -> Response[Union[ErrorResponse, SuccessResponse]]:
+def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[ErrorResponse, SuccessResponse]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

blaxel/core/sandbox/client/api/process/delete_process_identifier_kill.py

@@ -21,9 +21,7 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(
-    *, client: Client, response: httpx.Response
-) -> Union[ErrorResponse, SuccessResponse] | None:
+def _parse_response(*, client: Client, response: httpx.Response) -> Union[ErrorResponse, SuccessResponse] | None:
     if response.status_code == 200:
         response_200 = SuccessResponse.from_dict(response.json())
 
@@ -46,9 +44,7 @@ def _parse_response(
         return None
 
 
-def _build_response(
-    *, client: Client, response: httpx.Response
-) -> Response[Union[ErrorResponse, SuccessResponse]]:
+def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[ErrorResponse, SuccessResponse]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

blaxel/core/sandbox/client/api/process/get_process.py

@@ -34,9 +34,7 @@ def _parse_response(*, client: Client, response: httpx.Response) -> list["Proces
         return None
 
 
-def _build_response(
-    *, client: Client, response: httpx.Response
-) -> Response[list["ProcessResponse"]]:
+def _build_response(*, client: Client, response: httpx.Response) -> Response[list["ProcessResponse"]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

blaxel/core/sandbox/client/api/process/get_process_identifier.py

@@ -21,9 +21,7 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(
-    *, client: Client, response: httpx.Response
-) -> Union[ErrorResponse, ProcessResponse] | None:
+def _parse_response(*, client: Client, response: httpx.Response) -> Union[ErrorResponse, ProcessResponse] | None:
     if response.status_code == 200:
         response_200 = ProcessResponse.from_dict(response.json())
 
@@ -38,9 +36,7 @@ def _parse_response(
         return None
 
 
-def _build_response(
-    *, client: Client, response: httpx.Response
-) -> Response[Union[ErrorResponse, ProcessResponse]]:
+def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[ErrorResponse, ProcessResponse]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

blaxel/core/sandbox/client/api/process/get_process_identifier_logs.py

@@ -21,9 +21,7 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(
-    *, client: Client, response: httpx.Response
-) -> Union[ErrorResponse, ProcessLogs] | None:
+def _parse_response(*, client: Client, response: httpx.Response) -> Union[ErrorResponse, ProcessLogs] | None:
     if response.status_code == 200:
         response_200 = ProcessLogs.from_dict(response.json())
 
@@ -46,9 +44,7 @@ def _parse_response(
         return None
 
 
-def _build_response(
-    *, client: Client, response: httpx.Response
-) -> Response[Union[ErrorResponse, ProcessLogs]]:
+def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[ErrorResponse, ProcessLogs]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

blaxel/core/sandbox/client/api/process/get_process_identifier_logs_stream.py

@@ -20,9 +20,7 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(
-    *, client: Client, response: httpx.Response
-) -> Union[ErrorResponse, str] | None:
+def _parse_response(*, client: Client, response: httpx.Response) -> Union[ErrorResponse, str] | None:
     if response.status_code == 200:
         response_200 = response.text
         return response_200
@@ -44,9 +42,7 @@ def _parse_response(
         return None
 
 
-def _build_response(
-    *, client: Client, response: httpx.Response
-) -> Response[Union[ErrorResponse, str]]:
+def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[ErrorResponse, str]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

blaxel/core/sandbox/client/api/process/post_process.py

@@ -34,9 +34,7 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(
-    *, client: Client, response: httpx.Response
-) -> Union[ErrorResponse, ProcessResponse] | None:
+def _parse_response(*, client: Client, response: httpx.Response) -> Union[ErrorResponse, ProcessResponse] | None:
     if response.status_code == 200:
         response_200 = ProcessResponse.from_dict(response.json())
 
@@ -59,9 +57,7 @@ def _parse_response(
         return None
 
 
-def _build_response(
-    *, client: Client, response: httpx.Response
-) -> Response[Union[ErrorResponse, ProcessResponse]]:
+def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[ErrorResponse, ProcessResponse]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

blaxel/core/sandbox/client/client.py

@@ -43,9 +43,7 @@ class Client:
     _headers: dict[str, str] = field(factory=dict, kw_only=True, alias="headers")
     _auth: httpx.Auth = field(default=None, alias="auth")
     _timeout: httpx.Timeout | None = field(default=None, kw_only=True, alias="timeout")
-    _verify_ssl: Union[str, bool, ssl.SSLContext] = field(
-        default=True, kw_only=True, alias="verify_ssl"
-    )
+    _verify_ssl: Union[str, bool, ssl.SSLContext] = field(default=True, kw_only=True, alias="verify_ssl")
     _follow_redirects: bool = field(default=False, kw_only=True, alias="follow_redirects")
     _httpx_args: dict[str, Any] = field(factory=dict, kw_only=True, alias="httpx_args")
     _client: httpx.Client | None = field(default=None, init=False)

blaxel/core/sandbox/client/models/filesystem_multipart_upload_parts.py

@@ -14,9 +14,7 @@ T = TypeVar("T", bound="FilesystemMultipartUploadParts")
 class FilesystemMultipartUploadParts:
     """ """
 
-    additional_properties: dict[str, "FilesystemUploadedPart"] = _attrs_field(
-        init=False, factory=dict
-    )
+    additional_properties: dict[str, "FilesystemUploadedPart"] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
         field_dict: dict[str, Any] = {}

blaxel/core/sandbox/default/__init__.py

@@ -1,3 +1,4 @@
+
 from .interpreter import CodeInterpreter
 from .sandbox import (
     SandboxCodegen,

blaxel/core/sandbox/default/action.py

@@ -1,4 +1,6 @@
+
 import httpx
+from contextlib import asynccontextmanager
 
 from ...common.internal import get_forced_url, get_global_unique_hash
 from ...common.settings import settings
@@ -59,9 +61,7 @@ class SandboxAction:
             base_url = self.sandbox_config.force_url or self.url
             self._client = httpx.AsyncClient(
                 base_url=base_url,
-                headers=self.sandbox_config.headers
-                if self.sandbox_config.force_url
-                else {**settings.headers, **self.sandbox_config.headers},
+                headers=self.sandbox_config.headers if self.sandbox_config.force_url else {**settings.headers, **self.sandbox_config.headers},
                 http2=False,
                 limits=httpx.Limits(max_connections=100, max_keepalive_connections=20),
                 timeout=httpx.Timeout(300.0, connect=10.0),

blaxel/core/sandbox/default/codegen.py

@@ -1,3 +1,4 @@
+
 from ...common.settings import settings
 from ..client.api.codegen.get_codegen_reranking_path import (
     asyncio as get_codegen_reranking_path,
@@ -38,7 +39,7 @@ class SandboxCodegen(SandboxAction):
         # Create a Client instance with the sandbox URL and headers
         client = Client(
             base_url=self.url,
-            headers={**settings.headers, **self.sandbox_config.headers},
+            headers={**settings.headers, **self.sandbox_config.headers}
         )
 
         async with client:
@@ -76,7 +77,7 @@ class SandboxCodegen(SandboxAction):
         # Create a Client instance with the sandbox URL and headers
         client = Client(
             base_url=self.url,
-            headers={**settings.headers, **self.sandbox_config.headers},
+            headers={**settings.headers, **self.sandbox_config.headers}
         )
 
         async with client:
@@ -93,3 +94,4 @@ class SandboxCodegen(SandboxAction):
             if isinstance(response, ErrorResponse):
                 raise Exception(f"Reranking failed: {response}")
             return response
+

blaxel/core/sandbox/default/filesystem.py

@@ -47,11 +47,11 @@ class SandboxFileSystem(SandboxAction):
         path = self.format_path(path)
 
         # Calculate content size in bytes
-        content_size = len(content.encode("utf-8"))
+        content_size = len(content.encode('utf-8'))
 
         # Use multipart upload for large files
         if content_size > MULTIPART_THRESHOLD:
-            content_bytes = content.encode("utf-8")
+            content_bytes = content.encode('utf-8')
             return await self._upload_with_multipart(path, content_bytes, "0644")
 
         # Use regular upload for small files
@@ -66,9 +66,7 @@ class SandboxFileSystem(SandboxAction):
         finally:
             await response.aclose()
 
-    async def write_binary(
-        self, path: str, content: Union[bytes, bytearray, str]
-    ) -> SuccessResponse:
+    async def write_binary(self, path: str, content: Union[bytes, bytearray, str]) -> SuccessResponse:
         """Write binary content to a file.
 
         Args:
@@ -98,11 +96,7 @@ class SandboxFileSystem(SandboxAction):
 
         # Prepare multipart form data
         files = {
-            "file": (
-                "binary-file.bin",
-                binary_file,
-                "application/octet-stream",
-            ),
+            "file": ("binary-file.bin", binary_file, "application/octet-stream"),
         }
         data = {"permissions": "0644", "path": path}
 
@@ -115,7 +109,7 @@ class SandboxFileSystem(SandboxAction):
         try:
             content_bytes = await response.aread()
             if not response.is_success:
-                error_text = content_bytes.decode("utf-8", errors="ignore")
+                error_text = content_bytes.decode('utf-8', errors='ignore')
                 raise Exception(f"Failed to write binary: {response.status_code} {error_text}")
             return SuccessResponse.from_dict(json.loads(content_bytes))
         finally:
@@ -252,18 +246,18 @@ class SandboxFileSystem(SandboxAction):
             FindResponse with matching files/directories
         """
         path = self.format_path(path)
-
+        
         params = {}
         if type is not None:
-            params["type"] = type
+            params['type'] = type
         if patterns is not None and len(patterns) > 0:
-            params["patterns"] = ",".join(patterns)
+            params['patterns'] = ','.join(patterns)
         if max_results is not None:
-            params["maxResults"] = max_results
+            params['maxResults'] = max_results
         if exclude_dirs is not None and len(exclude_dirs) > 0:
-            params["excludeDirs"] = ",".join(exclude_dirs)
+            params['excludeDirs'] = ','.join(exclude_dirs)
         if exclude_hidden is not None:
-            params["excludeHidden"] = exclude_hidden
+            params['excludeHidden'] = exclude_hidden
 
         url = f"{self.url}/filesystem-find/{path}"
         headers = {**settings.headers, **self.sandbox_config.headers}
@@ -273,9 +267,8 @@ class SandboxFileSystem(SandboxAction):
         try:
             data = json.loads(await response.aread())
             self.handle_response_error(response)
-
+                
             from ..client.models.find_response import FindResponse
-
             return FindResponse.from_dict(data)
         finally:
             await response.aclose()
@@ -305,18 +298,18 @@ class SandboxFileSystem(SandboxAction):
             ContentSearchResponse with matching lines
         """
         path = self.format_path(path)
-
-        params = {"query": query}
+        
+        params = {'query': query}
         if case_sensitive is not None:
-            params["caseSensitive"] = case_sensitive
+            params['caseSensitive'] = case_sensitive
         if context_lines is not None:
-            params["contextLines"] = context_lines
+            params['contextLines'] = context_lines
         if max_results is not None:
-            params["maxResults"] = max_results
+            params['maxResults'] = max_results
         if file_pattern is not None:
-            params["filePattern"] = file_pattern
+            params['filePattern'] = file_pattern
         if exclude_dirs is not None and len(exclude_dirs) > 0:
-            params["excludeDirs"] = ",".join(exclude_dirs)
+            params['excludeDirs'] = ','.join(exclude_dirs)
 
         url = f"{self.url}/filesystem-content-search/{path}"
         headers = {**settings.headers, **self.sandbox_config.headers}
@@ -326,15 +319,58 @@ class SandboxFileSystem(SandboxAction):
         try:
             data = json.loads(await response.aread())
             self.handle_response_error(response)
-
-            from ..client.models.content_search_response import (
-                ContentSearchResponse,
-            )
-
+                
+            from ..client.models.content_search_response import ContentSearchResponse
             return ContentSearchResponse.from_dict(data)
         finally:
             await response.aclose()
 
+    async def find(
+        self,
+        path: str,
+        type: str | None = None,
+        patterns: List[str] | None = None,
+        max_results: int | None = None,
+        exclude_dirs: List[str] | None = None,
+        exclude_hidden: bool | None = None,
+    ):
+        """Find files and directories.
+
+        Args:
+            path: Path to search in
+            type: Type of search ('file' or 'directory')
+            patterns: File patterns to include (e.g., ['*.py', '*.json'])
+            max_results: Maximum number of results to return
+            exclude_dirs: Directory names to skip
+            exclude_hidden: Exclude hidden files and directories
+
+        Returns:
+            FindResponse with matching files/directories
+        """
+        path = self.format_path(path)
+        
+        params = {}
+        if type is not None:
+            params['type'] = type
+        if patterns is not None and len(patterns) > 0:
+            params['patterns'] = ','.join(patterns)
+        if max_results is not None:
+            params['maxResults'] = max_results
+        if exclude_dirs is not None and len(exclude_dirs) > 0:
+            params['excludeDirs'] = ','.join(exclude_dirs)
+        if exclude_hidden is not None:
+            params['excludeHidden'] = exclude_hidden
+
+        url = f"{self.url}/filesystem-find/{path}"
+        headers = {**settings.headers, **self.sandbox_config.headers}
+
+        async with self.get_client() as client_instance:
+            response = await client_instance.get(url, params=params, headers=headers)
+            self.handle_response_error(response)
+            
+            from ..client.models.find_response import FindResponse
+            return FindResponse.from_dict(response.json())
+
     async def cp(self, source: str, destination: str, max_wait: int = 180000) -> CopyResponse:
         """Copy files or directories using the cp command.
 
@@ -347,7 +383,9 @@ class SandboxFileSystem(SandboxAction):
             raise Exception("Process instance not available. Cannot execute cp command.")
 
         # Execute cp -r command
-        process = await self.process.exec({"command": f"cp -r {source} {destination}"})
+        process = await self.process.exec({
+            "command": f"cp -r {source} {destination}"
+        })
 
         # Wait for process to complete
         process = await self.process.wait(process.pid, max_wait=max_wait, interval=100)
@@ -357,7 +395,11 @@ class SandboxFileSystem(SandboxAction):
             logs = process.logs if hasattr(process, "logs") else "Unknown error"
             raise Exception(f"Could not copy {source} to {destination} cause: {logs}")
 
-        return CopyResponse(message="Files copied", source=source, destination=destination)
+        return CopyResponse(
+            message="Files copied",
+            source=source,
+            destination=destination
+        )
 
     def watch(
         self,
@@ -454,9 +496,7 @@ class SandboxFileSystem(SandboxAction):
         return path
 
     # Multipart upload helper methods
-    async def _initiate_multipart_upload(
-        self, path: str, permissions: str = "0644"
-    ) -> Dict[str, Any]:
+    async def _initiate_multipart_upload(self, path: str, permissions: str = "0644") -> Dict[str, Any]:
         """Initiate a multipart upload session."""
         path = self.format_path(path)
         url = f"{self.url}/filesystem-multipart/initiate/{path}"
@@ -472,7 +512,9 @@ class SandboxFileSystem(SandboxAction):
         finally:
             await response.aclose()
 
-    async def _upload_part(self, upload_id: str, part_number: int, data: bytes) -> Dict[str, Any]:
+    async def _upload_part(
+        self, upload_id: str, part_number: int, data: bytes
+    ) -> Dict[str, Any]:
         """Upload a single part of a multipart upload."""
         url = f"{self.url}/filesystem-multipart/{upload_id}/part"
         headers = {**settings.headers, **self.sandbox_config.headers}
@@ -482,7 +524,9 @@ class SandboxFileSystem(SandboxAction):
         files = {"file": ("part", io.BytesIO(data), "application/octet-stream")}
 
         client = self.get_client()
-        response = await client.put(url, files=files, params=params, headers=headers)
+        response = await client.put(
+            url, files=files, params=params, headers=headers
+        )
         try:
             data = json.loads(await response.aread())
             self.handle_response_error(response)

blaxel/core/sandbox/default/interpreter.py

@@ -17,9 +17,7 @@ class CodeInterpreter(SandboxInstance):
     DEFAULT_PORTS = [
         {"name": "jupyter", "target": 8888, "protocol": "HTTP"},
     ]
-    DEFAULT_LIFECYCLE = {
-        "expirationPolicies": [{"type": "ttl-idle", "value": "30m", "action": "delete"}]
-    }
+    DEFAULT_LIFECYCLE = {"expirationPolicies": [{"type": "ttl-idle", "value": "30m", "action": "delete"}]}
 
     @classmethod
     async def get(cls, sandbox_name: str) -> CodeInterpreter:
@@ -150,15 +148,11 @@ class CodeInterpreter(SandboxInstance):
         elif data_type == "stdout":
             execution.logs.stdout.append(data["text"])
             if on_stdout:
-                return on_stdout(
-                    CodeInterpreter.OutputMessage(data["text"], data.get("timestamp"), False)
-                )
+                return on_stdout(CodeInterpreter.OutputMessage(data["text"], data.get("timestamp"), False))
         elif data_type == "stderr":
             execution.logs.stderr.append(data["text"])
             if on_stderr:
-                return on_stderr(
-                    CodeInterpreter.OutputMessage(data["text"], data.get("timestamp"), True)
-                )
+                return on_stderr(CodeInterpreter.OutputMessage(data["text"], data.get("timestamp"), True))
         elif data_type == "error":
             execution.error = CodeInterpreter.ExecutionError(
                 data.get("name", ""), data.get("value"), data.get("traceback")
@@ -206,10 +200,7 @@ class CodeInterpreter(SandboxInstance):
 
         client = self.process.get_client()
         timeout_cfg = httpx.Timeout(
-            connect=connect_timeout,
-            read=read_timeout,
-            write=write_timeout,
-            pool=pool_timeout,
+            connect=connect_timeout, read=read_timeout, write=write_timeout, pool=pool_timeout
         )
         async with client.stream(
             "POST",
@@ -252,7 +243,7 @@ class CodeInterpreter(SandboxInstance):
                         on_error=on_error,
                     )
                 except json.JSONDecodeError:
-                    # Fallback: treat as stdout text-only message
+                        # Fallback: treat as stdout text-only message
                     execution.logs.stdout.append(decoded)
                     if on_stdout:
                         on_stdout(CodeInterpreter.OutputMessage(decoded, None, False))
@@ -280,10 +271,10 @@ class CodeInterpreter(SandboxInstance):
         try:
             # Always read response body first
             body_bytes = await response.aread()
-
+            
             if response.status_code >= 400:
                 try:
-                    body_text = body_bytes.decode("utf-8", errors="ignore")
+                    body_text = body_bytes.decode('utf-8', errors='ignore')
                 except Exception:
                     body_text = "<unavailable>"
                 method = getattr(response.request, "method", "UNKNOWN")
@@ -296,8 +287,10 @@ class CodeInterpreter(SandboxInstance):
                 )
                 self.logger.debug(details)
                 raise RuntimeError(details)
-
+            
             data = json.loads(body_bytes)
             return CodeInterpreter.Context.from_json(data)
         finally:
             await response.aclose()
+
+