blaxel 0.2.31rc120__py3-none-any.whl → 0.2.32__py3-none-any.whl

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

@@ -48,17 +48,17 @@ class FilesystemUploadedPart:
         return field_dict
 
     @classmethod
-    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T | None:
         if not src_dict:
             return None
         d = src_dict.copy()
         etag = d.pop("etag", UNSET)
 
-        part_number = d.pop("partNumber", UNSET)
+        part_number = d.pop("partNumber", d.pop("part_number", UNSET))
 
         size = d.pop("size", UNSET)
 
-        uploaded_at = d.pop("uploadedAt", UNSET)
+        uploaded_at = d.pop("uploadedAt", d.pop("uploaded_at", UNSET))
 
         filesystem_uploaded_part = cls(
             etag=etag,

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

@@ -36,13 +36,13 @@ class FindMatch:
         return field_dict
 
     @classmethod
-    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T | None:
         if not src_dict:
             return None
         d = src_dict.copy()
         path = d.pop("path", UNSET)
 
-        type_ = d.pop("type", UNSET)
+        type_ = d.pop("type", d.pop("type_", UNSET))
 
         find_match = cls(
             path=path,

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

@@ -48,7 +48,7 @@ class FindResponse:
         return field_dict
 
     @classmethod
-    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T | None:
         from ..models.find_match import FindMatch
 
         if not src_dict:

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

@@ -42,7 +42,7 @@ class FuzzySearchMatch:
         return field_dict
 
     @classmethod
-    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T | None:
         if not src_dict:
             return None
         d = src_dict.copy()
@@ -50,7 +50,7 @@ class FuzzySearchMatch:
 
         score = d.pop("score", UNSET)
 
-        type_ = d.pop("type", UNSET)
+        type_ = d.pop("type", d.pop("type_", UNSET))
 
         fuzzy_search_match = cls(
             path=path,

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

@@ -48,7 +48,7 @@ class FuzzySearchResponse:
         return field_dict
 
     @classmethod
-    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T | None:
         from ..models.fuzzy_search_match import FuzzySearchMatch
 
         if not src_dict:

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

@@ -19,7 +19,7 @@ class GetNetworkProcessPidPortsResponse200:
         return field_dict
 
     @classmethod
-    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T | None:
         if not src_dict:
             return None
         d = src_dict.copy()

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

@@ -42,7 +42,7 @@ class MultipartCompleteRequest:
         return field_dict
 
     @classmethod
-    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T | None:
         from ..models.multipart_part_info import MultipartPartInfo
 
         if not src_dict:

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

@@ -30,7 +30,7 @@ class MultipartInitiateRequest:
         return field_dict
 
     @classmethod
-    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T | None:
         if not src_dict:
             return None
         d = src_dict.copy()

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

@@ -36,13 +36,13 @@ class MultipartInitiateResponse:
         return field_dict
 
     @classmethod
-    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T | None:
         if not src_dict:
             return None
         d = src_dict.copy()
         path = d.pop("path", UNSET)
 
-        upload_id = d.pop("uploadId", UNSET)
+        upload_id = d.pop("uploadId", d.pop("upload_id", UNSET))
 
         multipart_initiate_response = cls(
             path=path,

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

@@ -48,7 +48,7 @@ class MultipartListPartsResponse:
         return field_dict
 
     @classmethod
-    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T | None:
         from ..models.filesystem_uploaded_part import FilesystemUploadedPart
 
         if not src_dict:
@@ -61,7 +61,7 @@ class MultipartListPartsResponse:
 
             parts.append(parts_item)
 
-        upload_id = d.pop("uploadId", UNSET)
+        upload_id = d.pop("uploadId", d.pop("upload_id", UNSET))
 
         multipart_list_parts_response = cls(
             parts=parts,

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

@@ -42,7 +42,7 @@ class MultipartListUploadsResponse:
         return field_dict
 
     @classmethod
-    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T | None:
         from ..models.filesystem_multipart_upload import FilesystemMultipartUpload
 
         if not src_dict:

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

@@ -36,13 +36,13 @@ class MultipartPartInfo:
         return field_dict
 
     @classmethod
-    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T | None:
         if not src_dict:
             return None
         d = src_dict.copy()
         etag = d.pop("etag", UNSET)
 
-        part_number = d.pop("partNumber", UNSET)
+        part_number = d.pop("partNumber", d.pop("part_number", UNSET))
 
         multipart_part_info = cls(
             etag=etag,

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

@@ -42,13 +42,13 @@ class MultipartUploadPartResponse:
         return field_dict
 
     @classmethod
-    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T | None:
         if not src_dict:
             return None
         d = src_dict.copy()
         etag = d.pop("etag", UNSET)
 
-        part_number = d.pop("partNumber", UNSET)
+        part_number = d.pop("partNumber", d.pop("part_number", UNSET))
 
         size = d.pop("size", UNSET)
 

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

@@ -30,7 +30,7 @@ class PortMonitorRequest:
         return field_dict
 
     @classmethod
-    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T | None:
         if not src_dict:
             return None
         d = src_dict.copy()

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

@@ -19,7 +19,7 @@ class PostNetworkProcessPidMonitorResponse200:
         return field_dict
 
     @classmethod
-    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T | None:
         if not src_dict:
             return None
         d = src_dict.copy()

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

@@ -40,7 +40,7 @@ class ProcessLogs:
         return field_dict
 
     @classmethod
-    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T | None:
         if not src_dict:
             return None
         d = src_dict.copy()

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

@@ -90,7 +90,7 @@ class ProcessRequest:
         return field_dict
 
     @classmethod
-    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T | None:
         from ..models.process_request_env import ProcessRequestEnv
 
         if not src_dict:
@@ -105,19 +105,19 @@ class ProcessRequest:
         else:
             env = ProcessRequestEnv.from_dict(_env)
 
-        max_restarts = d.pop("maxRestarts", UNSET)
+        max_restarts = d.pop("maxRestarts", d.pop("max_restarts", UNSET))
 
         name = d.pop("name", UNSET)
 
-        restart_on_failure = d.pop("restartOnFailure", UNSET)
+        restart_on_failure = d.pop("restartOnFailure", d.pop("restart_on_failure", UNSET))
 
         timeout = d.pop("timeout", UNSET)
 
-        wait_for_completion = d.pop("waitForCompletion", UNSET)
+        wait_for_completion = d.pop("waitForCompletion", d.pop("wait_for_completion", UNSET))
 
-        wait_for_ports = cast(list[int], d.pop("waitForPorts", UNSET))
+        wait_for_ports = cast(list[int], d.pop("waitForPorts", d.pop("wait_for_ports", UNSET)))
 
-        working_dir = d.pop("workingDir", UNSET)
+        working_dir = d.pop("workingDir", d.pop("working_dir", UNSET))
 
         process_request = cls(
             command=command,

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

@@ -23,7 +23,7 @@ class ProcessRequestEnv:
         return field_dict
 
     @classmethod
-    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T | None:
         if not src_dict:
             return None
         d = src_dict.copy()

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

@@ -91,15 +91,15 @@ class ProcessResponse:
         return field_dict
 
     @classmethod
-    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T | None:
         if not src_dict:
             return None
         d = src_dict.copy()
         command = d.pop("command")
 
-        completed_at = d.pop("completedAt")
+        completed_at = d.pop("completedAt") if "completedAt" in d else d.pop("completed_at")
 
-        exit_code = d.pop("exitCode")
+        exit_code = d.pop("exitCode") if "exitCode" in d else d.pop("exit_code")
 
         logs = d.pop("logs")
 
@@ -107,17 +107,17 @@ class ProcessResponse:
 
         pid = d.pop("pid")
 
-        started_at = d.pop("startedAt")
+        started_at = d.pop("startedAt") if "startedAt" in d else d.pop("started_at")
 
         status = ProcessResponseStatus(d.pop("status"))
 
-        working_dir = d.pop("workingDir")
+        working_dir = d.pop("workingDir") if "workingDir" in d else d.pop("working_dir")
 
-        max_restarts = d.pop("maxRestarts", UNSET)
+        max_restarts = d.pop("maxRestarts", d.pop("max_restarts", UNSET))
 
-        restart_count = d.pop("restartCount", UNSET)
+        restart_count = d.pop("restartCount", d.pop("restart_count", UNSET))
 
-        restart_on_failure = d.pop("restartOnFailure", UNSET)
+        restart_on_failure = d.pop("restartOnFailure", d.pop("restart_on_failure", UNSET))
 
         process_response = cls(
             command=command,

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

@@ -48,7 +48,7 @@ class PutFilesystemMultipartUploadIdPartBody:
         return field_dict
 
     @classmethod
-    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T | None:
         if not src_dict:
             return None
         d = src_dict.copy()

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

@@ -42,7 +42,7 @@ class RankedFile:
         return field_dict
 
     @classmethod
-    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T | None:
         if not src_dict:
             return None
         d = src_dict.copy()

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

@@ -54,7 +54,7 @@ class RerankingResponse:
         return field_dict
 
     @classmethod
-    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T | None:
         from ..models.ranked_file import RankedFile
 
         if not src_dict:

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

@@ -35,7 +35,7 @@ class Subdirectory:
         return field_dict
 
     @classmethod
-    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T | None:
         if not src_dict:
             return None
         d = src_dict.copy()

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

@@ -38,7 +38,7 @@ class SuccessResponse:
         return field_dict
 
     @classmethod
-    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T | None:
         if not src_dict:
             return None
         d = src_dict.copy()

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

@@ -39,7 +39,7 @@ class TreeRequest:
         return field_dict
 
     @classmethod
-    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T | None:
         from ..models.tree_request_files import TreeRequestFiles
 
         if not src_dict:

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

@@ -23,7 +23,7 @@ class TreeRequestFiles:
         return field_dict
 
     @classmethod
-    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T | None:
         if not src_dict:
             return None
         d = src_dict.copy()

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

@@ -43,7 +43,7 @@ class WelcomeResponse:
         return field_dict
 
     @classmethod
-    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T | None:
         if not src_dict:
             return None
         d = src_dict.copy()

blaxel/core/sandbox/default/__init__.py

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

blaxel/core/sandbox/default/action.py

@@ -1,6 +1,4 @@
-
 import httpx
-from contextlib import asynccontextmanager
 
 from ...common.internal import get_forced_url, get_global_unique_hash
 from ...common.settings import settings
@@ -61,7 +59,9 @@ 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,4 +1,3 @@
-
 from ...common.settings import settings
 from ..client.api.codegen.get_codegen_reranking_path import (
     asyncio as get_codegen_reranking_path,
@@ -39,7 +38,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:
@@ -77,7 +76,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:
@@ -94,4 +93,3 @@ class SandboxCodegen(SandboxAction):
             if isinstance(response, ErrorResponse):
                 raise Exception(f"Reranking failed: {response}")
             return response
-