blaxel 0.1.16__py3-none-any.whl → 0.1.17__py3-none-any.whl

blaxel/client/models/preview_spec_request_headers.py

@@ -0,0 +1,48 @@
+from typing import Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="PreviewSpecRequestHeaders")
+
+
+@_attrs_define
+class PreviewSpecRequestHeaders:
+    """Those headers will be set in all requests to your preview. This is especially useful to set the Authorization
+    header.
+
+    """
+
+    additional_properties: dict[str, str] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+        if not src_dict:
+            return None
+        d = src_dict.copy()
+        preview_spec_request_headers = cls()
+
+        preview_spec_request_headers.additional_properties = d
+        return preview_spec_request_headers
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> str:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: str) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

blaxel/{sandbox/client/models/get_process_identifier_logs_response_200.py → client/models/preview_spec_response_headers.py}

@@ -3,12 +3,12 @@ from typing import Any, TypeVar
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
 
-T = TypeVar("T", bound="GetProcessIdentifierLogsResponse200")
+T = TypeVar("T", bound="PreviewSpecResponseHeaders")
 
 
 @_attrs_define
-class GetProcessIdentifierLogsResponse200:
-    """ """
+class PreviewSpecResponseHeaders:
+    """Those headers will be set in all responses of your preview. This is especially useful to set the CORS headers."""
 
     additional_properties: dict[str, str] = _attrs_field(init=False, factory=dict)
 
@@ -23,10 +23,10 @@ class GetProcessIdentifierLogsResponse200:
         if not src_dict:
             return None
         d = src_dict.copy()
-        get_process_identifier_logs_response_200 = cls()
+        preview_spec_response_headers = cls()
 
-        get_process_identifier_logs_response_200.additional_properties = d
-        return get_process_identifier_logs_response_200
+        preview_spec_response_headers.additional_properties = d
+        return preview_spec_response_headers
 
     @property
     def additional_keys(self) -> list[str]:

blaxel/client/models/runtime.py

@@ -29,6 +29,8 @@ class Runtime:
             EnvVar types
         generation (Union[Unset, str]): The generation of the deployment
         image (Union[Unset, str]): The Docker image for the deployment
+        max_concurrent_tasks (Union[Unset, int]): The maximum number of concurrent task for an execution
+        max_retries (Union[Unset, int]): The maximum number of retries for the deployment
         max_scale (Union[Unset, int]): The minimum number of replicas for the deployment. Can be 0 or 1 (in which case
             the deployment is always running in at least one location).
         memory (Union[Unset, int]): The memory for the deployment in MB
@@ -50,6 +52,8 @@ class Runtime:
     envs: Union[Unset, list[Any]] = UNSET
     generation: Union[Unset, str] = UNSET
     image: Union[Unset, str] = UNSET
+    max_concurrent_tasks: Union[Unset, int] = UNSET
+    max_retries: Union[Unset, int] = UNSET
     max_scale: Union[Unset, int] = UNSET
     memory: Union[Unset, int] = UNSET
     metric_port: Union[Unset, int] = UNSET
@@ -93,6 +97,10 @@ class Runtime:
 
         image = self.image
 
+        max_concurrent_tasks = self.max_concurrent_tasks
+
+        max_retries = self.max_retries
+
         max_scale = self.max_scale
 
         memory = self.memory
@@ -148,6 +156,10 @@ class Runtime:
             field_dict["generation"] = generation
         if image is not UNSET:
             field_dict["image"] = image
+        if max_concurrent_tasks is not UNSET:
+            field_dict["maxConcurrentTasks"] = max_concurrent_tasks
+        if max_retries is not UNSET:
+            field_dict["maxRetries"] = max_retries
         if max_scale is not UNSET:
             field_dict["maxScale"] = max_scale
         if memory is not UNSET:
@@ -201,6 +213,10 @@ class Runtime:
 
         image = d.pop("image", UNSET)
 
+        max_concurrent_tasks = d.pop("maxConcurrentTasks", UNSET)
+
+        max_retries = d.pop("maxRetries", UNSET)
+
         max_scale = d.pop("maxScale", UNSET)
 
         memory = d.pop("memory", UNSET)
@@ -240,6 +256,8 @@ class Runtime:
             envs=envs,
             generation=generation,
             image=image,
+            max_concurrent_tasks=max_concurrent_tasks,
+            max_retries=max_retries,
             max_scale=max_scale,
             memory=memory,
             metric_port=metric_port,

blaxel/client/models/serverless_config.py

@@ -18,6 +18,7 @@ class ServerlessConfig:
 
     Attributes:
         configuration (Union[Unset, ServerlessConfigConfiguration]): The configuration for the deployment
+        max_retries (Union[Unset, int]): The maximum number of retries for the deployment
         max_scale (Union[Unset, int]): The minimum number of replicas for the deployment. Can be 0 or 1 (in which case
             the deployment is always running in at least one location).
         min_scale (Union[Unset, int]): The maximum number of replicas for the deployment.
@@ -25,6 +26,7 @@ class ServerlessConfig:
     """
 
     configuration: Union[Unset, "ServerlessConfigConfiguration"] = UNSET
+    max_retries: Union[Unset, int] = UNSET
     max_scale: Union[Unset, int] = UNSET
     min_scale: Union[Unset, int] = UNSET
     timeout: Union[Unset, int] = UNSET
@@ -41,6 +43,8 @@ class ServerlessConfig:
         elif self.configuration and isinstance(self.configuration, dict):
             configuration = self.configuration
 
+        max_retries = self.max_retries
+
         max_scale = self.max_scale
 
         min_scale = self.min_scale
@@ -52,6 +56,8 @@ class ServerlessConfig:
         field_dict.update({})
         if configuration is not UNSET:
             field_dict["configuration"] = configuration
+        if max_retries is not UNSET:
+            field_dict["maxRetries"] = max_retries
         if max_scale is not UNSET:
             field_dict["maxScale"] = max_scale
         if min_scale is not UNSET:
@@ -75,6 +81,8 @@ class ServerlessConfig:
         else:
             configuration = ServerlessConfigConfiguration.from_dict(_configuration)
 
+        max_retries = d.pop("maxRetries", UNSET)
+
         max_scale = d.pop("maxScale", UNSET)
 
         min_scale = d.pop("minScale", UNSET)
@@ -83,6 +91,7 @@ class ServerlessConfig:
 
         serverless_config = cls(
             configuration=configuration,
+            max_retries=max_retries,
             max_scale=max_scale,
             min_scale=min_scale,
             timeout=timeout,

blaxel/common/internal.py

@@ -1,7 +1,5 @@
-import base64
 import hashlib
 import os
-import re
 from logging import getLogger
 from typing import Optional
 

blaxel/common/logger.py

@@ -5,8 +5,10 @@ to set up logging configurations for Blaxel applications.
 import json
 import logging
 import os
+
 from opentelemetry import trace
 
+
 class JsonFormatter(logging.Formatter):
     """
     A logger compatible with standard json logging.

blaxel/instrumentation/manager.py

@@ -5,9 +5,9 @@ It includes classes and functions for configuring tracers, meters, loggers, and
 
 import importlib
 import logging
+import os
 import signal
 import time
-import os
 from typing import Any, Dict, List, Optional, Type
 
 from opentelemetry import metrics, trace

blaxel/jobs/__init__.py

@@ -1,12 +1,12 @@
 import argparse
+import asyncio
 import os
 import sys
-import asyncio
-import requests
+from logging import getLogger
+from typing import Any, Awaitable, Callable, Dict
 
+import requests
 
-from typing import Any, Dict, Callable, Awaitable
-from logging import getLogger
 from ..client import client
 from ..common.env import env
 from ..common.internal import get_global_unique_hash
@@ -14,7 +14,6 @@ from ..common.settings import settings
 from ..instrumentation.span import SpanManager
 
 
-
 class BlJobWrapper:
     def get_arguments(self) -> Dict[str, Any]:
         if not os.getenv('BL_EXECUTION_DATA_URL'):
@@ -50,9 +49,6 @@ class BlJobWrapper:
         Handles both async and sync functions.
         Arguments are passed as keyword arguments to the function.
         """
-        attributes = {
-            "span.type": "job.start",
-        }
         try:
             parsed_args = self.get_arguments()
             if asyncio.iscoroutinefunction(func):

blaxel/sandbox/client/api/filesystem/delete_filesystem_path.py

@@ -39,6 +39,10 @@ def _parse_response(*, client: Client, response: httpx.Response) -> Optional[Uni
         response_404 = ErrorResponse.from_dict(response.json())
 
         return response_404
+    if response.status_code == 422:
+        response_422 = ErrorResponse.from_dict(response.json())
+
+        return response_422
     if response.status_code == 500:
         response_500 = ErrorResponse.from_dict(response.json())
 

blaxel/sandbox/client/api/filesystem/get_filesystem_path.py

@@ -49,6 +49,10 @@ def _parse_response(
         response_404 = ErrorResponse.from_dict(response.json())
 
         return response_404
+    if response.status_code == 422:
+        response_422 = ErrorResponse.from_dict(response.json())
+
+        return response_422
     if response.status_code == 500:
         response_500 = ErrorResponse.from_dict(response.json())
 

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

@@ -6,15 +6,24 @@ import httpx
 from ... import errors
 from ...client import Client
 from ...models.error_response import ErrorResponse
-from ...types import Response
+from ...types import UNSET, Response, Unset
 
 
 def _get_kwargs(
     path: str,
+    *,
+    ignore: Union[Unset, str] = UNSET,
 ) -> dict[str, Any]:
+    params: dict[str, Any] = {}
+
+    params["ignore"] = ignore
+
+    params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
+
     _kwargs: dict[str, Any] = {
         "method": "get",
         "url": f"/watch/filesystem/{path}",
+        "params": params,
     }
 
     return _kwargs
@@ -51,6 +60,7 @@ def sync_detailed(
     path: str,
     *,
     client: Union[Client],
+    ignore: Union[Unset, str] = UNSET,
 ) -> Response[Union[ErrorResponse, str]]:
     """Stream file modification events in a directory
 
@@ -59,6 +69,7 @@ def sync_detailed(
 
     Args:
         path (str):
+        ignore (Union[Unset, str]):
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -70,6 +81,7 @@ def sync_detailed(
 
     kwargs = _get_kwargs(
         path=path,
+        ignore=ignore,
     )
 
     response = client.get_httpx_client().request(
@@ -83,6 +95,7 @@ def sync(
     path: str,
     *,
     client: Union[Client],
+    ignore: Union[Unset, str] = UNSET,
 ) -> Optional[Union[ErrorResponse, str]]:
     """Stream file modification events in a directory
 
@@ -91,6 +104,7 @@ def sync(
 
     Args:
         path (str):
+        ignore (Union[Unset, str]):
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -103,6 +117,7 @@ def sync(
     return sync_detailed(
         path=path,
         client=client,
+        ignore=ignore,
     ).parsed
 
 
@@ -110,6 +125,7 @@ async def asyncio_detailed(
     path: str,
     *,
     client: Union[Client],
+    ignore: Union[Unset, str] = UNSET,
 ) -> Response[Union[ErrorResponse, str]]:
     """Stream file modification events in a directory
 
@@ -118,6 +134,7 @@ async def asyncio_detailed(
 
     Args:
         path (str):
+        ignore (Union[Unset, str]):
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -129,6 +146,7 @@ async def asyncio_detailed(
 
     kwargs = _get_kwargs(
         path=path,
+        ignore=ignore,
     )
 
     response = await client.get_async_httpx_client().request(**kwargs)
@@ -140,6 +158,7 @@ async def asyncio(
     path: str,
     *,
     client: Union[Client],
+    ignore: Union[Unset, str] = UNSET,
 ) -> Optional[Union[ErrorResponse, str]]:
     """Stream file modification events in a directory
 
@@ -148,6 +167,7 @@ async def asyncio(
 
     Args:
         path (str):
+        ignore (Union[Unset, str]):
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -161,5 +181,6 @@ async def asyncio(
         await asyncio_detailed(
             path=path,
             client=client,
+            ignore=ignore,
         )
     ).parsed

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

@@ -44,6 +44,10 @@ def _parse_response(*, client: Client, response: httpx.Response) -> Optional[Uni
         response_400 = ErrorResponse.from_dict(response.json())
 
         return response_400
+    if response.status_code == 422:
+        response_422 = ErrorResponse.from_dict(response.json())
+
+        return response_422
     if response.status_code == 500:
         response_500 = ErrorResponse.from_dict(response.json())
 
@@ -69,7 +73,7 @@ def sync_detailed(
     client: Union[Client],
     body: FileRequest,
 ) -> Response[Union[ErrorResponse, SuccessResponse]]:
-    """Create or update file or directory
+    """Create or update a file or directory
 
      Create or update a file or directory
 
@@ -103,7 +107,7 @@ def sync(
     client: Union[Client],
     body: FileRequest,
 ) -> Optional[Union[ErrorResponse, SuccessResponse]]:
-    """Create or update file or directory
+    """Create or update a file or directory
 
      Create or update a file or directory
 
@@ -132,7 +136,7 @@ async def asyncio_detailed(
     client: Union[Client],
     body: FileRequest,
 ) -> Response[Union[ErrorResponse, SuccessResponse]]:
-    """Create or update file or directory
+    """Create or update a file or directory
 
      Create or update a file or directory
 
@@ -164,7 +168,7 @@ async def asyncio(
     client: Union[Client],
     body: FileRequest,
 ) -> Optional[Union[ErrorResponse, SuccessResponse]]:
-    """Create or update file or directory
+    """Create or update a file or directory
 
      Create or update a file or directory
 

blaxel/sandbox/client/api/network/delete_network_process_pid_monitor.py

@@ -34,6 +34,10 @@ def _parse_response(
         response_400 = ErrorResponse.from_dict(response.json())
 
         return response_400
+    if response.status_code == 422:
+        response_422 = ErrorResponse.from_dict(response.json())
+
+        return response_422
     if response.status_code == 500:
         response_500 = ErrorResponse.from_dict(response.json())
 

blaxel/sandbox/client/api/network/get_network_process_pid_ports.py

@@ -34,6 +34,10 @@ def _parse_response(
         response_400 = ErrorResponse.from_dict(response.json())
 
         return response_400
+    if response.status_code == 422:
+        response_422 = ErrorResponse.from_dict(response.json())
+
+        return response_422
     if response.status_code == 500:
         response_500 = ErrorResponse.from_dict(response.json())
 

blaxel/sandbox/client/api/network/post_network_process_pid_monitor.py

@@ -48,6 +48,10 @@ def _parse_response(
         response_400 = ErrorResponse.from_dict(response.json())
 
         return response_400
+    if response.status_code == 422:
+        response_422 = ErrorResponse.from_dict(response.json())
+
+        return response_422
     if response.status_code == 500:
         response_500 = ErrorResponse.from_dict(response.json())
 

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

@@ -30,6 +30,10 @@ def _parse_response(*, client: Client, response: httpx.Response) -> Optional[Uni
         response_404 = ErrorResponse.from_dict(response.json())
 
         return response_404
+    if response.status_code == 422:
+        response_422 = ErrorResponse.from_dict(response.json())
+
+        return response_422
     if response.status_code == 500:
         response_500 = ErrorResponse.from_dict(response.json())
 

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

@@ -44,6 +44,10 @@ def _parse_response(*, client: Client, response: httpx.Response) -> Optional[Uni
         response_404 = ErrorResponse.from_dict(response.json())
 
         return response_404
+    if response.status_code == 422:
+        response_422 = ErrorResponse.from_dict(response.json())
+
+        return response_422
     if response.status_code == 500:
         response_500 = ErrorResponse.from_dict(response.json())
 

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

@@ -6,7 +6,7 @@ import httpx
 from ... import errors
 from ...client import Client
 from ...models.error_response import ErrorResponse
-from ...models.get_process_identifier_logs_response_200 import GetProcessIdentifierLogsResponse200
+from ...models.process_logs import ProcessLogs
 from ...types import Response
 
 
@@ -21,17 +21,19 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(
-    *, client: Client, response: httpx.Response
-) -> Optional[Union[ErrorResponse, GetProcessIdentifierLogsResponse200]]:
+def _parse_response(*, client: Client, response: httpx.Response) -> Optional[Union[ErrorResponse, ProcessLogs]]:
     if response.status_code == 200:
-        response_200 = GetProcessIdentifierLogsResponse200.from_dict(response.json())
+        response_200 = ProcessLogs.from_dict(response.json())
 
         return response_200
     if response.status_code == 404:
         response_404 = ErrorResponse.from_dict(response.json())
 
         return response_404
+    if response.status_code == 422:
+        response_422 = ErrorResponse.from_dict(response.json())
+
+        return response_422
     if response.status_code == 500:
         response_500 = ErrorResponse.from_dict(response.json())
 
@@ -42,9 +44,7 @@ def _parse_response(
         return None
 
 
-def _build_response(
-    *, client: Client, response: httpx.Response
-) -> Response[Union[ErrorResponse, GetProcessIdentifierLogsResponse200]]:
+def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[ErrorResponse, ProcessLogs]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -57,7 +57,7 @@ def sync_detailed(
     identifier: str,
     *,
     client: Union[Client],
-) -> Response[Union[ErrorResponse, GetProcessIdentifierLogsResponse200]]:
+) -> Response[Union[ErrorResponse, ProcessLogs]]:
     """Get process logs
 
      Get the stdout and stderr output of a process
@@ -70,7 +70,7 @@ def sync_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[Union[ErrorResponse, GetProcessIdentifierLogsResponse200]]
+        Response[Union[ErrorResponse, ProcessLogs]]
     """
 
     kwargs = _get_kwargs(
@@ -88,7 +88,7 @@ def sync(
     identifier: str,
     *,
     client: Union[Client],
-) -> Optional[Union[ErrorResponse, GetProcessIdentifierLogsResponse200]]:
+) -> Optional[Union[ErrorResponse, ProcessLogs]]:
     """Get process logs
 
      Get the stdout and stderr output of a process
@@ -101,7 +101,7 @@ def sync(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Union[ErrorResponse, GetProcessIdentifierLogsResponse200]
+        Union[ErrorResponse, ProcessLogs]
     """
 
     return sync_detailed(
@@ -114,7 +114,7 @@ async def asyncio_detailed(
     identifier: str,
     *,
     client: Union[Client],
-) -> Response[Union[ErrorResponse, GetProcessIdentifierLogsResponse200]]:
+) -> Response[Union[ErrorResponse, ProcessLogs]]:
     """Get process logs
 
      Get the stdout and stderr output of a process
@@ -127,7 +127,7 @@ async def asyncio_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[Union[ErrorResponse, GetProcessIdentifierLogsResponse200]]
+        Response[Union[ErrorResponse, ProcessLogs]]
     """
 
     kwargs = _get_kwargs(
@@ -143,7 +143,7 @@ async def asyncio(
     identifier: str,
     *,
     client: Union[Client],
-) -> Optional[Union[ErrorResponse, GetProcessIdentifierLogsResponse200]]:
+) -> Optional[Union[ErrorResponse, ProcessLogs]]:
     """Get process logs
 
      Get the stdout and stderr output of a process
@@ -156,7 +156,7 @@ async def asyncio(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Union[ErrorResponse, GetProcessIdentifierLogsResponse200]
+        Union[ErrorResponse, ProcessLogs]
     """
 
     return (

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

@@ -28,6 +28,10 @@ def _parse_response(*, client: Client, response: httpx.Response) -> Optional[Uni
         response_404 = ErrorResponse.from_dict(response.text)
 
         return response_404
+    if response.status_code == 422:
+        response_422 = ErrorResponse.from_dict(response.text)
+
+        return response_422
     if response.status_code == 500:
         response_500 = ErrorResponse.from_dict(response.text)
 

blaxel/sandbox/client/api/process/get_ws_process_identifier_logs_stream.py

@@ -28,6 +28,10 @@ def _parse_response(*, client: Client, response: httpx.Response) -> Optional[Uni
         response_404 = ErrorResponse.from_dict(response.json())
 
         return response_404
+    if response.status_code == 422:
+        response_422 = ErrorResponse.from_dict(response.json())
+
+        return response_422
     if response.status_code == 500:
         response_500 = ErrorResponse.from_dict(response.json())
 
@@ -54,8 +58,7 @@ def sync_detailed(
 ) -> Response[Union[ErrorResponse, str]]:
     """Stream process logs in real time via WebSocket
 
-     Streams the stdout and stderr output of a process in real time as JSON messages. Closes when the
-    process exits or the client disconnects.
+     Streams the stdout and stderr output of a process in real time as JSON messages.
 
     Args:
         identifier (str):
@@ -86,8 +89,7 @@ def sync(
 ) -> Optional[Union[ErrorResponse, str]]:
     """Stream process logs in real time via WebSocket
 
-     Streams the stdout and stderr output of a process in real time as JSON messages. Closes when the
-    process exits or the client disconnects.
+     Streams the stdout and stderr output of a process in real time as JSON messages.
 
     Args:
         identifier (str):
@@ -113,8 +115,7 @@ async def asyncio_detailed(
 ) -> Response[Union[ErrorResponse, str]]:
     """Stream process logs in real time via WebSocket
 
-     Streams the stdout and stderr output of a process in real time as JSON messages. Closes when the
-    process exits or the client disconnects.
+     Streams the stdout and stderr output of a process in real time as JSON messages.
 
     Args:
         identifier (str):
@@ -143,8 +144,7 @@ async def asyncio(
 ) -> Optional[Union[ErrorResponse, str]]:
     """Stream process logs in real time via WebSocket
 
-     Streams the stdout and stderr output of a process in real time as JSON messages. Closes when the
-    process exits or the client disconnects.
+     Streams the stdout and stderr output of a process in real time as JSON messages.
 
     Args:
         identifier (str):