blaxel 0.2.35__py3-none-any.whl → 0.2.36__py3-none-any.whl

blaxel/__init__.py

@@ -4,8 +4,8 @@ from .core.common.autoload import autoload
 from .core.common.env import env
 from .core.common.settings import settings
 
-__version__ = "0.2.35"
-__commit__ = "4cbf852e4b6e8e7ec6b054ac2a79d2959a4e2fe6"
+__version__ = "0.2.36"
+__commit__ = "70f38e31dd50b26cbb54665420e165b89f3fa4b9"
 __sentry_dsn__ = "https://9711de13cd02b285ca4378c01de8dc30@o4508714045276160.ingest.us.sentry.io/4510461121462272"
 __all__ = ["autoload", "settings", "env"]
 

blaxel/core/client/api/compute/create_sandbox.py

@@ -7,18 +7,26 @@ from ... import errors
 from ...client import Client
 from ...models.sandbox import Sandbox
 from ...models.sandbox_error import SandboxError
-from ...types import Response
+from ...types import UNSET, Response, Unset
 
 
 def _get_kwargs(
     *,
     body: Sandbox,
+    create_if_not_exist: Union[Unset, bool] = False,
 ) -> dict[str, Any]:
     headers: dict[str, Any] = {}
 
+    params: dict[str, Any] = {}
+
+    params["createIfNotExist"] = create_if_not_exist
+
+    params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
+
     _kwargs: dict[str, Any] = {
         "method": "post",
         "url": "/sandboxes",
+        "params": params,
     }
 
     if type(body) is dict:
@@ -81,6 +89,7 @@ def sync_detailed(
     *,
     client: Client,
     body: Sandbox,
+    create_if_not_exist: Union[Unset, bool] = False,
 ) -> Response[Union[Sandbox, SandboxError]]:
     """Create sandbox
 
@@ -88,6 +97,7 @@ def sync_detailed(
     idle and resume instantly, preserving memory state including running processes and filesystem.
 
     Args:
+        create_if_not_exist (Union[Unset, bool]):  Default: False.
         body (Sandbox): Lightweight virtual machine for secure AI code execution. Sandboxes resume
             from standby in under 25ms and automatically scale to zero after inactivity, preserving
             memory state including running processes and filesystem.
@@ -102,6 +112,7 @@ def sync_detailed(
 
     kwargs = _get_kwargs(
         body=body,
+        create_if_not_exist=create_if_not_exist,
     )
 
     response = client.get_httpx_client().request(
@@ -115,6 +126,7 @@ def sync(
     *,
     client: Client,
     body: Sandbox,
+    create_if_not_exist: Union[Unset, bool] = False,
 ) -> Union[Sandbox, SandboxError] | None:
     """Create sandbox
 
@@ -122,6 +134,7 @@ def sync(
     idle and resume instantly, preserving memory state including running processes and filesystem.
 
     Args:
+        create_if_not_exist (Union[Unset, bool]):  Default: False.
         body (Sandbox): Lightweight virtual machine for secure AI code execution. Sandboxes resume
             from standby in under 25ms and automatically scale to zero after inactivity, preserving
             memory state including running processes and filesystem.
@@ -137,6 +150,7 @@ def sync(
     return sync_detailed(
         client=client,
         body=body,
+        create_if_not_exist=create_if_not_exist,
     ).parsed
 
 
@@ -144,6 +158,7 @@ async def asyncio_detailed(
     *,
     client: Client,
     body: Sandbox,
+    create_if_not_exist: Union[Unset, bool] = False,
 ) -> Response[Union[Sandbox, SandboxError]]:
     """Create sandbox
 
@@ -151,6 +166,7 @@ async def asyncio_detailed(
     idle and resume instantly, preserving memory state including running processes and filesystem.
 
     Args:
+        create_if_not_exist (Union[Unset, bool]):  Default: False.
         body (Sandbox): Lightweight virtual machine for secure AI code execution. Sandboxes resume
             from standby in under 25ms and automatically scale to zero after inactivity, preserving
             memory state including running processes and filesystem.
@@ -165,6 +181,7 @@ async def asyncio_detailed(
 
     kwargs = _get_kwargs(
         body=body,
+        create_if_not_exist=create_if_not_exist,
     )
 
     response = await client.get_async_httpx_client().request(**kwargs)
@@ -176,6 +193,7 @@ async def asyncio(
     *,
     client: Client,
     body: Sandbox,
+    create_if_not_exist: Union[Unset, bool] = False,
 ) -> Union[Sandbox, SandboxError] | None:
     """Create sandbox
 
@@ -183,6 +201,7 @@ async def asyncio(
     idle and resume instantly, preserving memory state including running processes and filesystem.
 
     Args:
+        create_if_not_exist (Union[Unset, bool]):  Default: False.
         body (Sandbox): Lightweight virtual machine for secure AI code execution. Sandboxes resume
             from standby in under 25ms and automatically scale to zero after inactivity, preserving
             memory state including running processes and filesystem.
@@ -199,5 +218,6 @@ async def asyncio(
         await asyncio_detailed(
             client=client,
             body=body,
+            create_if_not_exist=create_if_not_exist,
         )
     ).parsed

blaxel/core/client/api/jobs/create_job_execution.py

@@ -5,8 +5,8 @@ import httpx
 
 from ... import errors
 from ...client import Client
+from ...models.create_job_execution_output import CreateJobExecutionOutput
 from ...models.create_job_execution_request import CreateJobExecutionRequest
-from ...models.create_job_execution_response import CreateJobExecutionResponse
 from ...types import Response
 
 
@@ -36,9 +36,9 @@ def _get_kwargs(
 
 def _parse_response(
     *, client: Client, response: httpx.Response
-) -> Union[Any, CreateJobExecutionResponse] | None:
+) -> Union[Any, CreateJobExecutionOutput] | None:
     if response.status_code == 200:
-        response_200 = CreateJobExecutionResponse.from_dict(response.json())
+        response_200 = CreateJobExecutionOutput.from_dict(response.json())
 
         return response_200
     if response.status_code == 400:
@@ -55,7 +55,7 @@ def _parse_response(
 
 def _build_response(
     *, client: Client, response: httpx.Response
-) -> Response[Union[Any, CreateJobExecutionResponse]]:
+) -> Response[Union[Any, CreateJobExecutionOutput]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -69,7 +69,7 @@ def sync_detailed(
     *,
     client: Client,
     body: CreateJobExecutionRequest,
-) -> Response[Union[Any, CreateJobExecutionResponse]]:
+) -> Response[Union[Any, CreateJobExecutionOutput]]:
     """Create job execution
 
      Triggers a new execution of the batch job. Each execution runs multiple tasks in parallel according
@@ -84,7 +84,7 @@ def sync_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[Union[Any, CreateJobExecutionResponse]]
+        Response[Union[Any, CreateJobExecutionOutput]]
     """
 
     kwargs = _get_kwargs(
@@ -104,7 +104,7 @@ def sync(
     *,
     client: Client,
     body: CreateJobExecutionRequest,
-) -> Union[Any, CreateJobExecutionResponse] | None:
+) -> Union[Any, CreateJobExecutionOutput] | None:
     """Create job execution
 
      Triggers a new execution of the batch job. Each execution runs multiple tasks in parallel according
@@ -119,7 +119,7 @@ def sync(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Union[Any, CreateJobExecutionResponse]
+        Union[Any, CreateJobExecutionOutput]
     """
 
     return sync_detailed(
@@ -134,7 +134,7 @@ async def asyncio_detailed(
     *,
     client: Client,
     body: CreateJobExecutionRequest,
-) -> Response[Union[Any, CreateJobExecutionResponse]]:
+) -> Response[Union[Any, CreateJobExecutionOutput]]:
     """Create job execution
 
      Triggers a new execution of the batch job. Each execution runs multiple tasks in parallel according
@@ -149,7 +149,7 @@ async def asyncio_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[Union[Any, CreateJobExecutionResponse]]
+        Response[Union[Any, CreateJobExecutionOutput]]
     """
 
     kwargs = _get_kwargs(
@@ -167,7 +167,7 @@ async def asyncio(
     *,
     client: Client,
     body: CreateJobExecutionRequest,
-) -> Union[Any, CreateJobExecutionResponse] | None:
+) -> Union[Any, CreateJobExecutionOutput] | None:
     """Create job execution
 
      Triggers a new execution of the batch job. Each execution runs multiple tasks in parallel according
@@ -182,7 +182,7 @@ async def asyncio(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Union[Any, CreateJobExecutionResponse]
+        Union[Any, CreateJobExecutionOutput]
     """
 
     return (

blaxel/core/client/api/volumes/update_volume.py

@@ -0,0 +1,187 @@
+from http import HTTPStatus
+from typing import Any
+
+import httpx
+
+from ... import errors
+from ...client import Client
+from ...models.volume import Volume
+from ...types import Response
+
+
+def _get_kwargs(
+    volume_name: str,
+    *,
+    body: Volume,
+) -> dict[str, Any]:
+    headers: dict[str, Any] = {}
+
+    _kwargs: dict[str, Any] = {
+        "method": "put",
+        "url": f"/volumes/{volume_name}",
+    }
+
+    if type(body) is dict:
+        _body = body
+    else:
+        _body = body.to_dict()
+
+    _kwargs["json"] = _body
+    headers["Content-Type"] = "application/json"
+
+    _kwargs["headers"] = headers
+    return _kwargs
+
+
+def _parse_response(*, client: Client, response: httpx.Response) -> Volume | None:
+    if response.status_code == 200:
+        response_200 = Volume.from_dict(response.json())
+
+        return response_200
+    if client.raise_on_unexpected_status:
+        raise errors.UnexpectedStatus(response.status_code, response.content)
+    else:
+        return None
+
+
+def _build_response(*, client: Client, response: httpx.Response) -> Response[Volume]:
+    return Response(
+        status_code=HTTPStatus(response.status_code),
+        content=response.content,
+        headers=response.headers,
+        parsed=_parse_response(client=client, response=response),
+    )
+
+
+def sync_detailed(
+    volume_name: str,
+    *,
+    client: Client,
+    body: Volume,
+) -> Response[Volume]:
+    """Update volume
+
+     Updates a volume.
+
+    Args:
+        volume_name (str):
+        body (Volume): Persistent storage volume that can be attached to sandboxes for durable
+            file storage across sessions. Volumes survive sandbox deletion and can be reattached to
+            new sandboxes.
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Response[Volume]
+    """
+
+    kwargs = _get_kwargs(
+        volume_name=volume_name,
+        body=body,
+    )
+
+    response = client.get_httpx_client().request(
+        **kwargs,
+    )
+
+    return _build_response(client=client, response=response)
+
+
+def sync(
+    volume_name: str,
+    *,
+    client: Client,
+    body: Volume,
+) -> Volume | None:
+    """Update volume
+
+     Updates a volume.
+
+    Args:
+        volume_name (str):
+        body (Volume): Persistent storage volume that can be attached to sandboxes for durable
+            file storage across sessions. Volumes survive sandbox deletion and can be reattached to
+            new sandboxes.
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Volume
+    """
+
+    return sync_detailed(
+        volume_name=volume_name,
+        client=client,
+        body=body,
+    ).parsed
+
+
+async def asyncio_detailed(
+    volume_name: str,
+    *,
+    client: Client,
+    body: Volume,
+) -> Response[Volume]:
+    """Update volume
+
+     Updates a volume.
+
+    Args:
+        volume_name (str):
+        body (Volume): Persistent storage volume that can be attached to sandboxes for durable
+            file storage across sessions. Volumes survive sandbox deletion and can be reattached to
+            new sandboxes.
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Response[Volume]
+    """
+
+    kwargs = _get_kwargs(
+        volume_name=volume_name,
+        body=body,
+    )
+
+    response = await client.get_async_httpx_client().request(**kwargs)
+
+    return _build_response(client=client, response=response)
+
+
+async def asyncio(
+    volume_name: str,
+    *,
+    client: Client,
+    body: Volume,
+) -> Volume | None:
+    """Update volume
+
+     Updates a volume.
+
+    Args:
+        volume_name (str):
+        body (Volume): Persistent storage volume that can be attached to sandboxes for durable
+            file storage across sessions. Volumes survive sandbox deletion and can be reattached to
+            new sandboxes.
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Volume
+    """
+
+    return (
+        await asyncio_detailed(
+            volume_name=volume_name,
+            client=client,
+            body=body,
+        )
+    ).parsed

blaxel/core/client/models/__init__.py

@@ -12,10 +12,11 @@ from .continent import Continent
 from .core_event import CoreEvent
 from .country import Country
 from .create_api_key_for_service_account_body import CreateApiKeyForServiceAccountBody
+from .create_job_execution_output import CreateJobExecutionOutput
+from .create_job_execution_output_tasks_item import CreateJobExecutionOutputTasksItem
 from .create_job_execution_request import CreateJobExecutionRequest
+from .create_job_execution_request_env import CreateJobExecutionRequestEnv
 from .create_job_execution_request_tasks_item import CreateJobExecutionRequestTasksItem
-from .create_job_execution_response import CreateJobExecutionResponse
-from .create_job_execution_response_tasks_item import CreateJobExecutionResponseTasksItem
 from .create_workspace_service_account_body import CreateWorkspaceServiceAccountBody
 from .create_workspace_service_account_response_200 import CreateWorkspaceServiceAccountResponse200
 from .custom_domain import CustomDomain
@@ -43,8 +44,8 @@ from .form_secrets import FormSecrets
 from .function import Function
 from .function_runtime import FunctionRuntime
 from .function_runtime_generation import FunctionRuntimeGeneration
+from .function_runtime_transport import FunctionRuntimeTransport
 from .function_spec import FunctionSpec
-from .function_spec_transport import FunctionSpecTransport
 from .get_workspace_service_accounts_response_200_item import (
     GetWorkspaceServiceAccountsResponse200Item,
 )
@@ -72,6 +73,7 @@ from .job import Job
 from .job_execution import JobExecution
 from .job_execution_metadata import JobExecutionMetadata
 from .job_execution_spec import JobExecutionSpec
+from .job_execution_spec_env_override import JobExecutionSpecEnvOverride
 from .job_execution_stats import JobExecutionStats
 from .job_execution_status import JobExecutionStatus
 from .job_execution_task import JobExecutionTask
@@ -174,10 +176,11 @@ __all__ = (
     "CoreEvent",
     "Country",
     "CreateApiKeyForServiceAccountBody",
+    "CreateJobExecutionOutput",
+    "CreateJobExecutionOutputTasksItem",
     "CreateJobExecutionRequest",
+    "CreateJobExecutionRequestEnv",
     "CreateJobExecutionRequestTasksItem",
-    "CreateJobExecutionResponse",
-    "CreateJobExecutionResponseTasksItem",
     "CreateWorkspaceServiceAccountBody",
     "CreateWorkspaceServiceAccountResponse200",
     "CustomDomain",
@@ -205,8 +208,8 @@ __all__ = (
     "Function",
     "FunctionRuntime",
     "FunctionRuntimeGeneration",
+    "FunctionRuntimeTransport",
     "FunctionSpec",
-    "FunctionSpecTransport",
     "GetWorkspaceServiceAccountsResponse200Item",
     "Image",
     "ImageMetadata",
@@ -232,6 +235,7 @@ __all__ = (
     "JobExecution",
     "JobExecutionMetadata",
     "JobExecutionSpec",
+    "JobExecutionSpecEnvOverride",
     "JobExecutionStats",
     "JobExecutionStatus",
     "JobExecutionTask",

blaxel/core/client/models/{create_job_execution_response.py → create_job_execution_output.py}

@@ -6,16 +6,14 @@ from attrs import field as _attrs_field
 from ..types import UNSET, Unset
 
 if TYPE_CHECKING:
-    from ..models.create_job_execution_response_tasks_item import (
-        CreateJobExecutionResponseTasksItem,
-    )
+    from ..models.create_job_execution_output_tasks_item import CreateJobExecutionOutputTasksItem
 
 
-T = TypeVar("T", bound="CreateJobExecutionResponse")
+T = TypeVar("T", bound="CreateJobExecutionOutput")
 
 
 @_attrs_define
-class CreateJobExecutionResponse:
+class CreateJobExecutionOutput:
     """Response returned when a job execution is successfully created. Contains identifiers and the tasks that will be
     executed.
 
@@ -25,7 +23,7 @@ class CreateJobExecutionResponse:
             id (Union[Unset, str]): Unique identifier for this request, used for idempotency and tracking. Auto-generated if
                 not provided in the request. Example: 550e8400-e29b-41d4-a716-446655440000.
             job_id (Union[Unset, str]): Name of the job that this execution belongs to Example: data-processing-job.
-            tasks (Union[Unset, list['CreateJobExecutionResponseTasksItem']]): Array of task configurations that will be
+            tasks (Union[Unset, list['CreateJobExecutionOutputTasksItem']]): Array of task configurations that will be
                 executed in parallel according to the job's concurrency settings. Each task can have custom parameters.
             workspace_id (Union[Unset, str]): Name of the workspace where the job execution was created Example: my-
                 workspace.
@@ -34,7 +32,7 @@ class CreateJobExecutionResponse:
     execution_id: Union[Unset, str] = UNSET
     id: Union[Unset, str] = UNSET
     job_id: Union[Unset, str] = UNSET
-    tasks: Union[Unset, list["CreateJobExecutionResponseTasksItem"]] = UNSET
+    tasks: Union[Unset, list["CreateJobExecutionOutputTasksItem"]] = UNSET
     workspace_id: Union[Unset, str] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
@@ -75,8 +73,8 @@ class CreateJobExecutionResponse:
 
     @classmethod
     def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T | None:
-        from ..models.create_job_execution_response_tasks_item import (
-            CreateJobExecutionResponseTasksItem,
+        from ..models.create_job_execution_output_tasks_item import (
+            CreateJobExecutionOutputTasksItem,
         )
 
         if not src_dict:
@@ -91,13 +89,13 @@ class CreateJobExecutionResponse:
         tasks = []
         _tasks = d.pop("tasks", UNSET)
         for tasks_item_data in _tasks or []:
-            tasks_item = CreateJobExecutionResponseTasksItem.from_dict(tasks_item_data)
+            tasks_item = CreateJobExecutionOutputTasksItem.from_dict(tasks_item_data)
 
             tasks.append(tasks_item)
 
         workspace_id = d.pop("workspaceId", d.pop("workspace_id", UNSET))
 
-        create_job_execution_response = cls(
+        create_job_execution_output = cls(
             execution_id=execution_id,
             id=id,
             job_id=job_id,
@@ -105,8 +103,8 @@ class CreateJobExecutionResponse:
             workspace_id=workspace_id,
         )
 
-        create_job_execution_response.additional_properties = d
-        return create_job_execution_response
+        create_job_execution_output.additional_properties = d
+        return create_job_execution_output
 
     @property
     def additional_keys(self) -> list[str]:

blaxel/core/client/models/{create_job_execution_response_tasks_item.py → create_job_execution_output_tasks_item.py}

@@ -3,11 +3,11 @@ from typing import Any, TypeVar
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
 
-T = TypeVar("T", bound="CreateJobExecutionResponseTasksItem")
+T = TypeVar("T", bound="CreateJobExecutionOutputTasksItem")
 
 
 @_attrs_define
-class CreateJobExecutionResponseTasksItem:
+class CreateJobExecutionOutputTasksItem:
     """ """
 
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
@@ -23,10 +23,10 @@ class CreateJobExecutionResponseTasksItem:
         if not src_dict:
             return None
         d = src_dict.copy()
-        create_job_execution_response_tasks_item = cls()
+        create_job_execution_output_tasks_item = cls()
 
-        create_job_execution_response_tasks_item.additional_properties = d
-        return create_job_execution_response_tasks_item
+        create_job_execution_output_tasks_item.additional_properties = d
+        return create_job_execution_output_tasks_item
 
     @property
     def additional_keys(self) -> list[str]:

blaxel/core/client/models/create_job_execution_request.py

@@ -6,6 +6,7 @@ from attrs import field as _attrs_field
 from ..types import UNSET, Unset
 
 if TYPE_CHECKING:
+    from ..models.create_job_execution_request_env import CreateJobExecutionRequestEnv
     from ..models.create_job_execution_request_tasks_item import CreateJobExecutionRequestTasksItem
 
 
@@ -17,28 +18,42 @@ class CreateJobExecutionRequest:
     """Request to create a job execution
 
     Attributes:
+        env (Union[Unset, CreateJobExecutionRequestEnv]): Environment variable overrides (optional, will merge with
+            job's environment variables) Example: {"MY_VAR": "custom_value", "BATCH_SIZE": "100"}.
         execution_id (Union[Unset, str]): Execution ID (optional, will be generated if not provided)
         id (Union[Unset, str]): Unique message ID
         job_id (Union[Unset, str]): Job ID Example: data-processing-job.
+        memory (Union[Unset, int]): Memory override in megabytes (optional, must be lower than or equal to job's
+            configured memory) Example: 2048.
         tasks (Union[Unset, list['CreateJobExecutionRequestTasksItem']]): Array of task parameters for parallel
             execution
         workspace_id (Union[Unset, str]): Workspace ID
     """
 
+    env: Union[Unset, "CreateJobExecutionRequestEnv"] = UNSET
     execution_id: Union[Unset, str] = UNSET
     id: Union[Unset, str] = UNSET
     job_id: Union[Unset, str] = UNSET
+    memory: Union[Unset, int] = UNSET
     tasks: Union[Unset, list["CreateJobExecutionRequestTasksItem"]] = UNSET
     workspace_id: Union[Unset, str] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
+        env: Union[Unset, dict[str, Any]] = UNSET
+        if self.env and not isinstance(self.env, Unset) and not isinstance(self.env, dict):
+            env = self.env.to_dict()
+        elif self.env and isinstance(self.env, dict):
+            env = self.env
+
         execution_id = self.execution_id
 
         id = self.id
 
         job_id = self.job_id
 
+        memory = self.memory
+
         tasks: Union[Unset, list[dict[str, Any]]] = UNSET
         if not isinstance(self.tasks, Unset):
             tasks = []
@@ -54,12 +69,16 @@ class CreateJobExecutionRequest:
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
         field_dict.update({})
+        if env is not UNSET:
+            field_dict["env"] = env
         if execution_id is not UNSET:
             field_dict["executionId"] = execution_id
         if id is not UNSET:
             field_dict["id"] = id
         if job_id is not UNSET:
             field_dict["jobId"] = job_id
+        if memory is not UNSET:
+            field_dict["memory"] = memory
         if tasks is not UNSET:
             field_dict["tasks"] = tasks
         if workspace_id is not UNSET:
@@ -69,6 +88,7 @@ class CreateJobExecutionRequest:
 
     @classmethod
     def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T | None:
+        from ..models.create_job_execution_request_env import CreateJobExecutionRequestEnv
         from ..models.create_job_execution_request_tasks_item import (
             CreateJobExecutionRequestTasksItem,
         )
@@ -76,12 +96,21 @@ class CreateJobExecutionRequest:
         if not src_dict:
             return None
         d = src_dict.copy()
+        _env = d.pop("env", UNSET)
+        env: Union[Unset, CreateJobExecutionRequestEnv]
+        if isinstance(_env, Unset):
+            env = UNSET
+        else:
+            env = CreateJobExecutionRequestEnv.from_dict(_env)
+
         execution_id = d.pop("executionId", d.pop("execution_id", UNSET))
 
         id = d.pop("id", UNSET)
 
         job_id = d.pop("jobId", d.pop("job_id", UNSET))
 
+        memory = d.pop("memory", UNSET)
+
         tasks = []
         _tasks = d.pop("tasks", UNSET)
         for tasks_item_data in _tasks or []:
@@ -92,9 +121,11 @@ class CreateJobExecutionRequest:
         workspace_id = d.pop("workspaceId", d.pop("workspace_id", UNSET))
 
         create_job_execution_request = cls(
+            env=env,
             execution_id=execution_id,
             id=id,
             job_id=job_id,
+            memory=memory,
             tasks=tasks,
             workspace_id=workspace_id,
         )