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

blaxel/core/common/settings.py

@@ -69,23 +69,25 @@ class Settings:
             return "https://run.blaxel.ai"
         return "https://run.blaxel.dev"
 
-
     @property
     def sentry_dsn(self) -> str:
         """Get the Sentry DSN (injected at build time)."""
         import blaxel
+
         return blaxel.__sentry_dsn__
 
     @property
     def version(self) -> str:
         """Get the package version (injected at build time)."""
         import blaxel
+
         return blaxel.__version__ or "unknown"
 
     @property
     def commit(self) -> str:
         """Get the commit hash (injected at build time)."""
         import blaxel
+
         return blaxel.__commit__ or "unknown"
 
     @property
@@ -96,7 +98,6 @@ class Settings:
         headers["User-Agent"] = f"blaxel/sdk/python/{self.version} ({os_arch}) blaxel/{self.commit}"
         return headers
 
-
     @property
     def name(self) -> str:
         """Get the name."""
@@ -145,13 +146,13 @@ class Settings:
         Get the tracking setting.
 
         Priority:
-        1. Environment variable BL_TRACKING (true/false)
+        1. Environment variable DO_NOT_TRACK (true/1 to disable, false/0 to enable)
         2. config.yaml tracking field
         3. Default: true
         """
-        env_value = os.environ.get("BL_TRACKING")
+        env_value = os.environ.get("DO_NOT_TRACK")
         if env_value is not None:
-            return env_value.lower() == "true"
+            return env_value.lower() not in ("true", "1")
 
         try:
             home_dir = Path.home()
@@ -165,4 +166,5 @@ class Settings:
 
         return True
 
+
 settings = Settings()

blaxel/core/common/webhook.py

@@ -184,4 +184,3 @@ __all__ = [
     "AsyncSidecarCallback",
     "RequestLike",
 ]
-

blaxel/core/jobs/__init__.py

@@ -14,7 +14,9 @@ from ..client.api.jobs import (
     get_job_execution,
     list_job_executions,
 )
-from ..client.models.create_job_execution_request import CreateJobExecutionRequest
+from ..client.models.create_job_execution_request import (
+    CreateJobExecutionRequest,
+)
 from ..client.models.job_execution import JobExecution
 from ..common.internal import get_forced_url, get_global_unique_hash
 from ..common.settings import settings
@@ -105,7 +107,11 @@ class BlJob:
         body = {"tasks": input_data}
 
         # Merge settings headers with provided headers
-        merged_headers = {**settings.headers, "Content-Type": "application/json", **headers}
+        merged_headers = {
+            **settings.headers,
+            "Content-Type": "application/json",
+            **headers,
+        }
 
         return client.get_httpx_client().post(
             url + "/executions",
@@ -119,7 +125,11 @@ class BlJob:
         body = {"tasks": input_data}
 
         # Merge settings headers with provided headers
-        merged_headers = {**settings.headers, "Content-Type": "application/json", **headers}
+        merged_headers = {
+            **settings.headers,
+            "Content-Type": "application/json",
+            **headers,
+        }
 
         return await client.get_async_httpx_client().post(
             url + "/executions",

blaxel/core/mcp/client.py

@@ -6,7 +6,10 @@ from urllib.parse import urljoin, urlparse
 import anyio
 import mcp.types as types
 from anyio.abc import TaskStatus
-from anyio.streams.memory import MemoryObjectReceiveStream, MemoryObjectSendStream
+from anyio.streams.memory import (
+    MemoryObjectReceiveStream,
+    MemoryObjectSendStream,
+)
 from mcp.shared.message import SessionMessage
 from websockets.asyncio.client import ClientConnection
 from websockets.asyncio.client import connect as ws_connect
@@ -47,7 +50,10 @@ async def websocket_client(
                 logger.debug(f"Connecting to WebSocket endpoint: {remove_request_params(ws_url)}")
 
                 # Use different parameters based on websockets version
-                connection_kwargs = {"additional_headers": headers, "open_timeout": timeout}
+                connection_kwargs = {
+                    "additional_headers": headers,
+                    "open_timeout": timeout,
+                }
 
                 async with ws_connect(ws_url, **connection_kwargs) as websocket:
                     logger.debug("WebSocket connection established")

blaxel/core/mcp/server.py

@@ -6,7 +6,10 @@ from typing import Literal
 
 import anyio
 import mcp.types as types
-from anyio.streams.memory import MemoryObjectReceiveStream, MemoryObjectSendStream
+from anyio.streams.memory import (
+    MemoryObjectReceiveStream,
+    MemoryObjectSendStream,
+)
 from mcp.server.fastmcp import FastMCP as FastMCPBase
 from mcp.shared.message import SessionMessage
 from websockets.asyncio.server import ServerConnection, serve
@@ -134,7 +137,10 @@ class FastMCP(FastMCPBase):
             anyio.run(self.run_sse_async)
 
     async def run_ws_async(self) -> None:
-        async with BlaxelMcpServerTransport().websocket_server() as (read_stream, write_stream):
+        async with BlaxelMcpServerTransport().websocket_server() as (
+            read_stream,
+            write_stream,
+        ):
             await self._mcp_server.run(
                 read_stream,
                 write_stream,

blaxel/core/models/__init__.py

@@ -1,4 +1,3 @@
-
 from ..cache import find_from_cache
 from ..client import client
 from ..client.api.models import get_model
@@ -23,21 +22,21 @@ class BLModel:
         model_data = await self._get_model_metadata()
         if not model_data:
             raise Exception(f"Model {self.model_name} not found")
-        runtime = (model_data.spec and model_data.spec.runtime)
+        runtime = model_data.spec and model_data.spec.runtime
         if not runtime:
             raise Exception(f"Model {self.model_name} has no runtime")
 
-        type = runtime.type_ or 'openai'
+        type = runtime.type_ or "openai"
         model = runtime.model
         self.models[self.model_name] = {
             "url": url,
             "type": type,
-            "model": model
+            "model": model,
         }
         return url, type, model
 
     async def _get_model_metadata(self) -> Model | None:
-        cache_data = await find_from_cache('Model', self.model_name)
+        cache_data = await find_from_cache("Model", self.model_name)
         if cache_data:
             return Model.from_dict(cache_data)
 
@@ -46,7 +45,9 @@ class BLModel:
         except Exception:
             return None
 
+
 def bl_model(model_name, **kwargs):
     return BLModel(model_name, **kwargs)
 
+
 __all__ = ["bl_model"]

blaxel/core/sandbox/__init__.py

@@ -59,4 +59,4 @@ __all__ = [
     "SyncSandboxProcess",
     "SyncCodeInterpreter",
     "CodeInterpreter",
-]
+]

blaxel/core/sandbox/client/api/codegen/get_codegen_reranking_path.py

@@ -39,7 +39,9 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(*, client: Client, response: httpx.Response) -> Union[ErrorResponse, RerankingResponse] | None:
+def _parse_response(
+    *, client: Client, response: httpx.Response
+) -> Union[ErrorResponse, RerankingResponse] | None:
     if response.status_code == 200:
         response_200 = RerankingResponse.from_dict(response.json())
 
@@ -62,7 +64,9 @@ def _parse_response(*, client: Client, response: httpx.Response) -> Union[ErrorR
         return None
 
 
-def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[ErrorResponse, RerankingResponse]]:
+def _build_response(
+    *, client: Client, response: httpx.Response
+) -> Response[Union[ErrorResponse, RerankingResponse]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -74,7 +78,7 @@ def _build_response(*, client: Client, response: httpx.Response) -> Response[Uni
 def sync_detailed(
     path: str,
     *,
-    client: Union[Client],
+    client: Client,
     query: str,
     score_threshold: Union[Unset, float] = UNSET,
     token_limit: Union[Unset, int] = UNSET,
@@ -129,7 +133,7 @@ def sync_detailed(
 def sync(
     path: str,
     *,
-    client: Union[Client],
+    client: Client,
     query: str,
     score_threshold: Union[Unset, float] = UNSET,
     token_limit: Union[Unset, int] = UNSET,
@@ -179,7 +183,7 @@ def sync(
 async def asyncio_detailed(
     path: str,
     *,
-    client: Union[Client],
+    client: Client,
     query: str,
     score_threshold: Union[Unset, float] = UNSET,
     token_limit: Union[Unset, int] = UNSET,
@@ -232,7 +236,7 @@ async def asyncio_detailed(
 async def asyncio(
     path: str,
     *,
-    client: Union[Client],
+    client: Client,
     query: str,
     score_threshold: Union[Unset, float] = UNSET,
     token_limit: Union[Unset, int] = UNSET,

blaxel/core/sandbox/client/api/fastapply/put_codegen_fastapply_path.py

@@ -35,7 +35,9 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(*, client: Client, response: httpx.Response) -> Union[ApplyEditResponse, ErrorResponse] | None:
+def _parse_response(
+    *, client: Client, response: httpx.Response
+) -> Union[ApplyEditResponse, ErrorResponse] | None:
     if response.status_code == 200:
         response_200 = ApplyEditResponse.from_dict(response.json())
 
@@ -58,7 +60,9 @@ def _parse_response(*, client: Client, response: httpx.Response) -> Union[ApplyE
         return None
 
 
-def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[ApplyEditResponse, ErrorResponse]]:
+def _build_response(
+    *, client: Client, response: httpx.Response
+) -> Response[Union[ApplyEditResponse, ErrorResponse]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -70,7 +74,7 @@ def _build_response(*, client: Client, response: httpx.Response) -> Response[Uni
 def sync_detailed(
     path: str,
     *,
-    client: Union[Client],
+    client: Client,
     body: ApplyEditRequest,
 ) -> Response[Union[ApplyEditResponse, ErrorResponse]]:
     r"""Apply code edit
@@ -138,7 +142,7 @@ def sync_detailed(
 def sync(
     path: str,
     *,
-    client: Union[Client],
+    client: Client,
     body: ApplyEditRequest,
 ) -> Union[ApplyEditResponse, ErrorResponse] | None:
     r"""Apply code edit
@@ -201,7 +205,7 @@ def sync(
 async def asyncio_detailed(
     path: str,
     *,
-    client: Union[Client],
+    client: Client,
     body: ApplyEditRequest,
 ) -> Response[Union[ApplyEditResponse, ErrorResponse]]:
     r"""Apply code edit
@@ -267,7 +271,7 @@ async def asyncio_detailed(
 async def asyncio(
     path: str,
     *,
-    client: Union[Client],
+    client: Client,
     body: ApplyEditRequest,
 ) -> Union[ApplyEditResponse, ErrorResponse] | None:
     r"""Apply code edit

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

@@ -21,7 +21,9 @@ 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())
 
@@ -44,7 +46,9 @@ def _parse_response(*, client: Client, response: httpx.Response) -> Union[ErrorR
         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,
@@ -56,7 +60,7 @@ def _build_response(*, client: Client, response: httpx.Response) -> Response[Uni
 def sync_detailed(
     upload_id: str,
     *,
-    client: Union[Client],
+    client: Client,
 ) -> Response[Union[ErrorResponse, SuccessResponse]]:
     """Abort multipart upload
 
@@ -87,7 +91,7 @@ def sync_detailed(
 def sync(
     upload_id: str,
     *,
-    client: Union[Client],
+    client: Client,
 ) -> Union[ErrorResponse, SuccessResponse] | None:
     """Abort multipart upload
 
@@ -113,7 +117,7 @@ def sync(
 async def asyncio_detailed(
     upload_id: str,
     *,
-    client: Union[Client],
+    client: Client,
 ) -> Response[Union[ErrorResponse, SuccessResponse]]:
     """Abort multipart upload
 
@@ -142,7 +146,7 @@ async def asyncio_detailed(
 async def asyncio(
     upload_id: str,
     *,
-    client: Union[Client],
+    client: Client,
 ) -> Union[ErrorResponse, SuccessResponse] | None:
     """Abort multipart upload
 

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

@@ -30,7 +30,9 @@ 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())
 
@@ -53,7 +55,9 @@ def _parse_response(*, client: Client, response: httpx.Response) -> Union[ErrorR
         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,
@@ -65,7 +69,7 @@ def _build_response(*, client: Client, response: httpx.Response) -> Response[Uni
 def sync_detailed(
     path: str,
     *,
-    client: Union[Client],
+    client: Client,
     recursive: Union[Unset, bool] = UNSET,
 ) -> Response[Union[ErrorResponse, SuccessResponse]]:
     """Delete file or directory
@@ -99,7 +103,7 @@ def sync_detailed(
 def sync(
     path: str,
     *,
-    client: Union[Client],
+    client: Client,
     recursive: Union[Unset, bool] = UNSET,
 ) -> Union[ErrorResponse, SuccessResponse] | None:
     """Delete file or directory
@@ -128,7 +132,7 @@ def sync(
 async def asyncio_detailed(
     path: str,
     *,
-    client: Union[Client],
+    client: Client,
     recursive: Union[Unset, bool] = UNSET,
 ) -> Response[Union[ErrorResponse, SuccessResponse]]:
     """Delete file or directory
@@ -160,7 +164,7 @@ async def asyncio_detailed(
 async def asyncio(
     path: str,
     *,
-    client: Union[Client],
+    client: Client,
     recursive: Union[Unset, bool] = UNSET,
 ) -> Union[ErrorResponse, SuccessResponse] | None:
     """Delete file or directory

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

@@ -30,7 +30,9 @@ 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())
 
@@ -53,7 +55,9 @@ def _parse_response(*, client: Client, response: httpx.Response) -> Union[ErrorR
         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,
@@ -65,7 +69,7 @@ def _build_response(*, client: Client, response: httpx.Response) -> Response[Uni
 def sync_detailed(
     path: str,
     *,
-    client: Union[Client],
+    client: Client,
     recursive: Union[Unset, bool] = UNSET,
 ) -> Response[Union[ErrorResponse, SuccessResponse]]:
     """Delete directory tree
@@ -99,7 +103,7 @@ def sync_detailed(
 def sync(
     path: str,
     *,
-    client: Union[Client],
+    client: Client,
     recursive: Union[Unset, bool] = UNSET,
 ) -> Union[ErrorResponse, SuccessResponse] | None:
     """Delete directory tree
@@ -128,7 +132,7 @@ def sync(
 async def asyncio_detailed(
     path: str,
     *,
-    client: Union[Client],
+    client: Client,
     recursive: Union[Unset, bool] = UNSET,
 ) -> Response[Union[ErrorResponse, SuccessResponse]]:
     """Delete directory tree
@@ -160,7 +164,7 @@ async def asyncio_detailed(
 async def asyncio(
     path: str,
     *,
-    client: Union[Client],
+    client: Client,
     recursive: Union[Unset, bool] = UNSET,
 ) -> Union[ErrorResponse, SuccessResponse] | None:
     """Delete directory tree

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

@@ -42,7 +42,9 @@ 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())
 
@@ -79,7 +81,7 @@ def _build_response(
 def sync_detailed(
     path: str,
     *,
-    client: Union[Client],
+    client: Client,
     query: str,
     case_sensitive: Union[Unset, bool] = UNSET,
     max_results: Union[Unset, int] = UNSET,
@@ -125,7 +127,7 @@ def sync_detailed(
 def sync(
     path: str,
     *,
-    client: Union[Client],
+    client: Client,
     query: str,
     case_sensitive: Union[Unset, bool] = UNSET,
     max_results: Union[Unset, int] = UNSET,
@@ -166,7 +168,7 @@ def sync(
 async def asyncio_detailed(
     path: str,
     *,
-    client: Union[Client],
+    client: Client,
     query: str,
     case_sensitive: Union[Unset, bool] = UNSET,
     max_results: Union[Unset, int] = UNSET,
@@ -210,7 +212,7 @@ async def asyncio_detailed(
 async def asyncio(
     path: str,
     *,
-    client: Union[Client],
+    client: Client,
     query: str,
     case_sensitive: Union[Unset, bool] = UNSET,
     max_results: Union[Unset, int] = UNSET,

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

@@ -42,7 +42,9 @@ 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())
 
@@ -65,7 +67,9 @@ def _parse_response(*, client: Client, response: httpx.Response) -> Union[ErrorR
         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,
@@ -77,7 +81,7 @@ def _build_response(*, client: Client, response: httpx.Response) -> Response[Uni
 def sync_detailed(
     path: str,
     *,
-    client: Union[Client],
+    client: Client,
     type_: Union[Unset, str] = UNSET,
     patterns: Union[Unset, str] = UNSET,
     max_results: Union[Unset, int] = UNSET,
@@ -123,7 +127,7 @@ def sync_detailed(
 def sync(
     path: str,
     *,
-    client: Union[Client],
+    client: Client,
     type_: Union[Unset, str] = UNSET,
     patterns: Union[Unset, str] = UNSET,
     max_results: Union[Unset, int] = UNSET,
@@ -164,7 +168,7 @@ def sync(
 async def asyncio_detailed(
     path: str,
     *,
-    client: Union[Client],
+    client: Client,
     type_: Union[Unset, str] = UNSET,
     patterns: Union[Unset, str] = UNSET,
     max_results: Union[Unset, int] = UNSET,
@@ -208,7 +212,7 @@ async def asyncio_detailed(
 async def asyncio(
     path: str,
     *,
-    client: Union[Client],
+    client: Client,
     type_: Union[Unset, str] = UNSET,
     patterns: Union[Unset, str] = UNSET,
     max_results: Union[Unset, int] = UNSET,

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

@@ -49,7 +49,7 @@ def _build_response(
 
 def sync_detailed(
     *,
-    client: Union[Client],
+    client: Client,
 ) -> Response[Union[ErrorResponse, MultipartListUploadsResponse]]:
     """List multipart uploads
 
@@ -74,7 +74,7 @@ def sync_detailed(
 
 def sync(
     *,
-    client: Union[Client],
+    client: Client,
 ) -> Union[ErrorResponse, MultipartListUploadsResponse] | None:
     """List multipart uploads
 
@@ -95,7 +95,7 @@ def sync(
 
 async def asyncio_detailed(
     *,
-    client: Union[Client],
+    client: Client,
 ) -> Response[Union[ErrorResponse, MultipartListUploadsResponse]]:
     """List multipart uploads
 
@@ -118,7 +118,7 @@ async def asyncio_detailed(
 
 async def asyncio(
     *,
-    client: Union[Client],
+    client: Client,
 ) -> Union[ErrorResponse, MultipartListUploadsResponse] | None:
     """List multipart uploads
 

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

@@ -60,7 +60,7 @@ def _build_response(
 def sync_detailed(
     upload_id: str,
     *,
-    client: Union[Client],
+    client: Client,
 ) -> Response[Union[ErrorResponse, MultipartListPartsResponse]]:
     """List parts
 
@@ -91,7 +91,7 @@ def sync_detailed(
 def sync(
     upload_id: str,
     *,
-    client: Union[Client],
+    client: Client,
 ) -> Union[ErrorResponse, MultipartListPartsResponse] | None:
     """List parts
 
@@ -117,7 +117,7 @@ def sync(
 async def asyncio_detailed(
     upload_id: str,
     *,
-    client: Union[Client],
+    client: Client,
 ) -> Response[Union[ErrorResponse, MultipartListPartsResponse]]:
     """List parts
 
@@ -146,7 +146,7 @@ async def asyncio_detailed(
 async def asyncio(
     upload_id: str,
     *,
-    client: Union[Client],
+    client: Client,
 ) -> Union[ErrorResponse, MultipartListPartsResponse] | None:
     """List parts