blaxel 0.2.34__py3-none-any.whl → 0.2.35__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.34"
-__commit__ = "84fc1c14e48dec727c7de1966c51022e09f7c80f"
+__version__ = "0.2.35"
+__commit__ = "4cbf852e4b6e8e7ec6b054ac2a79d2959a4e2fe6"
 __sentry_dsn__ = "https://9711de13cd02b285ca4378c01de8dc30@o4508714045276160.ingest.us.sentry.io/4510461121462272"
 __all__ = ["autoload", "settings", "env"]
 

blaxel/core/client/api/agents/create_agent.py

@@ -1,11 +1,12 @@
 from http import HTTPStatus
-from typing import Any
+from typing import Any, Union
 
 import httpx
 
 from ... import errors
 from ...client import Client
 from ...models.agent import Agent
+from ...models.error import Error
 from ...types import Response
 
 
@@ -32,18 +33,38 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(*, client: Client, response: httpx.Response) -> Agent | None:
+def _parse_response(*, client: Client, response: httpx.Response) -> Union[Agent, Error] | None:
     if response.status_code == 200:
         response_200 = Agent.from_dict(response.json())
 
         return response_200
+    if response.status_code == 400:
+        response_400 = Error.from_dict(response.json())
+
+        return response_400
+    if response.status_code == 401:
+        response_401 = Error.from_dict(response.json())
+
+        return response_401
+    if response.status_code == 403:
+        response_403 = Error.from_dict(response.json())
+
+        return response_403
+    if response.status_code == 409:
+        response_409 = Error.from_dict(response.json())
+
+        return response_409
+    if response.status_code == 500:
+        response_500 = Error.from_dict(response.json())
+
+        return response_500
     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[Agent]:
+def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[Agent, Error]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -56,18 +77,24 @@ def sync_detailed(
     *,
     client: Client,
     body: Agent,
-) -> Response[Agent]:
-    """Create agent by name
+) -> Response[Union[Agent, Error]]:
+    """Create agent
+
+     Creates a new AI agent deployment from your code. The agent will be built and deployed as a
+    serverless auto-scaling endpoint. Use the Blaxel CLI 'bl deploy' for a simpler deployment
+    experience.
 
     Args:
-        body (Agent): Agent
+        body (Agent): Serverless AI agent deployment that runs your custom agent code as an auto-
+            scaling API endpoint. Agents are deployed from your code repository and expose a global
+            inference URL for querying.
 
     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[Agent]
+        Response[Union[Agent, Error]]
     """
 
     kwargs = _get_kwargs(
@@ -85,18 +112,24 @@ def sync(
     *,
     client: Client,
     body: Agent,
-) -> Agent | None:
-    """Create agent by name
+) -> Union[Agent, Error] | None:
+    """Create agent
+
+     Creates a new AI agent deployment from your code. The agent will be built and deployed as a
+    serverless auto-scaling endpoint. Use the Blaxel CLI 'bl deploy' for a simpler deployment
+    experience.
 
     Args:
-        body (Agent): Agent
+        body (Agent): Serverless AI agent deployment that runs your custom agent code as an auto-
+            scaling API endpoint. Agents are deployed from your code repository and expose a global
+            inference URL for querying.
 
     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:
-        Agent
+        Union[Agent, Error]
     """
 
     return sync_detailed(
@@ -109,18 +142,24 @@ async def asyncio_detailed(
     *,
     client: Client,
     body: Agent,
-) -> Response[Agent]:
-    """Create agent by name
+) -> Response[Union[Agent, Error]]:
+    """Create agent
+
+     Creates a new AI agent deployment from your code. The agent will be built and deployed as a
+    serverless auto-scaling endpoint. Use the Blaxel CLI 'bl deploy' for a simpler deployment
+    experience.
 
     Args:
-        body (Agent): Agent
+        body (Agent): Serverless AI agent deployment that runs your custom agent code as an auto-
+            scaling API endpoint. Agents are deployed from your code repository and expose a global
+            inference URL for querying.
 
     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[Agent]
+        Response[Union[Agent, Error]]
     """
 
     kwargs = _get_kwargs(
@@ -136,18 +175,24 @@ async def asyncio(
     *,
     client: Client,
     body: Agent,
-) -> Agent | None:
-    """Create agent by name
+) -> Union[Agent, Error] | None:
+    """Create agent
+
+     Creates a new AI agent deployment from your code. The agent will be built and deployed as a
+    serverless auto-scaling endpoint. Use the Blaxel CLI 'bl deploy' for a simpler deployment
+    experience.
 
     Args:
-        body (Agent): Agent
+        body (Agent): Serverless AI agent deployment that runs your custom agent code as an auto-
+            scaling API endpoint. Agents are deployed from your code repository and expose a global
+            inference URL for querying.
 
     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:
-        Agent
+        Union[Agent, Error]
     """
 
     return (

blaxel/core/client/api/agents/delete_agent.py

@@ -1,11 +1,12 @@
 from http import HTTPStatus
-from typing import Any
+from typing import Any, Union
 
 import httpx
 
 from ... import errors
 from ...client import Client
 from ...models.agent import Agent
+from ...models.error import Error
 from ...types import Response
 
 
@@ -20,18 +21,34 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(*, client: Client, response: httpx.Response) -> Agent | None:
+def _parse_response(*, client: Client, response: httpx.Response) -> Union[Agent, Error] | None:
     if response.status_code == 200:
         response_200 = Agent.from_dict(response.json())
 
         return response_200
+    if response.status_code == 401:
+        response_401 = Error.from_dict(response.json())
+
+        return response_401
+    if response.status_code == 403:
+        response_403 = Error.from_dict(response.json())
+
+        return response_403
+    if response.status_code == 404:
+        response_404 = Error.from_dict(response.json())
+
+        return response_404
+    if response.status_code == 500:
+        response_500 = Error.from_dict(response.json())
+
+        return response_500
     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[Agent]:
+def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[Agent, Error]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -44,8 +61,11 @@ def sync_detailed(
     agent_name: str,
     *,
     client: Client,
-) -> Response[Agent]:
-    """Delete agent by name
+) -> Response[Union[Agent, Error]]:
+    """Delete agent
+
+     Permanently deletes an agent and all its deployment history. The agent's inference endpoint will
+    immediately stop responding. This action cannot be undone.
 
     Args:
         agent_name (str):
@@ -55,7 +75,7 @@ def sync_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[Agent]
+        Response[Union[Agent, Error]]
     """
 
     kwargs = _get_kwargs(
@@ -73,8 +93,11 @@ def sync(
     agent_name: str,
     *,
     client: Client,
-) -> Agent | None:
-    """Delete agent by name
+) -> Union[Agent, Error] | None:
+    """Delete agent
+
+     Permanently deletes an agent and all its deployment history. The agent's inference endpoint will
+    immediately stop responding. This action cannot be undone.
 
     Args:
         agent_name (str):
@@ -84,7 +107,7 @@ def sync(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Agent
+        Union[Agent, Error]
     """
 
     return sync_detailed(
@@ -97,8 +120,11 @@ async def asyncio_detailed(
     agent_name: str,
     *,
     client: Client,
-) -> Response[Agent]:
-    """Delete agent by name
+) -> Response[Union[Agent, Error]]:
+    """Delete agent
+
+     Permanently deletes an agent and all its deployment history. The agent's inference endpoint will
+    immediately stop responding. This action cannot be undone.
 
     Args:
         agent_name (str):
@@ -108,7 +134,7 @@ async def asyncio_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[Agent]
+        Response[Union[Agent, Error]]
     """
 
     kwargs = _get_kwargs(
@@ -124,8 +150,11 @@ async def asyncio(
     agent_name: str,
     *,
     client: Client,
-) -> Agent | None:
-    """Delete agent by name
+) -> Union[Agent, Error] | None:
+    """Delete agent
+
+     Permanently deletes an agent and all its deployment history. The agent's inference endpoint will
+    immediately stop responding. This action cannot be undone.
 
     Args:
         agent_name (str):
@@ -135,7 +164,7 @@ async def asyncio(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Agent
+        Union[Agent, Error]
     """
 
     return (

blaxel/core/client/api/agents/get_agent.py

@@ -6,6 +6,7 @@ import httpx
 from ... import errors
 from ...client import Client
 from ...models.agent import Agent
+from ...models.error import Error
 from ...types import UNSET, Response, Unset
 
 
@@ -29,18 +30,34 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(*, client: Client, response: httpx.Response) -> Agent | None:
+def _parse_response(*, client: Client, response: httpx.Response) -> Union[Agent, Error] | None:
     if response.status_code == 200:
         response_200 = Agent.from_dict(response.json())
 
         return response_200
+    if response.status_code == 401:
+        response_401 = Error.from_dict(response.json())
+
+        return response_401
+    if response.status_code == 403:
+        response_403 = Error.from_dict(response.json())
+
+        return response_403
+    if response.status_code == 404:
+        response_404 = Error.from_dict(response.json())
+
+        return response_404
+    if response.status_code == 500:
+        response_500 = Error.from_dict(response.json())
+
+        return response_500
     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[Agent]:
+def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[Agent, Error]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -54,8 +71,11 @@ def sync_detailed(
     *,
     client: Client,
     show_secrets: Union[Unset, bool] = UNSET,
-) -> Response[Agent]:
-    """Get agent by name
+) -> Response[Union[Agent, Error]]:
+    """Get agent
+
+     Returns detailed information about an agent including its current deployment status, configuration,
+    events history, and inference endpoint URL.
 
     Args:
         agent_name (str):
@@ -66,7 +86,7 @@ def sync_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[Agent]
+        Response[Union[Agent, Error]]
     """
 
     kwargs = _get_kwargs(
@@ -86,8 +106,11 @@ def sync(
     *,
     client: Client,
     show_secrets: Union[Unset, bool] = UNSET,
-) -> Agent | None:
-    """Get agent by name
+) -> Union[Agent, Error] | None:
+    """Get agent
+
+     Returns detailed information about an agent including its current deployment status, configuration,
+    events history, and inference endpoint URL.
 
     Args:
         agent_name (str):
@@ -98,7 +121,7 @@ def sync(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Agent
+        Union[Agent, Error]
     """
 
     return sync_detailed(
@@ -113,8 +136,11 @@ async def asyncio_detailed(
     *,
     client: Client,
     show_secrets: Union[Unset, bool] = UNSET,
-) -> Response[Agent]:
-    """Get agent by name
+) -> Response[Union[Agent, Error]]:
+    """Get agent
+
+     Returns detailed information about an agent including its current deployment status, configuration,
+    events history, and inference endpoint URL.
 
     Args:
         agent_name (str):
@@ -125,7 +151,7 @@ async def asyncio_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[Agent]
+        Response[Union[Agent, Error]]
     """
 
     kwargs = _get_kwargs(
@@ -143,8 +169,11 @@ async def asyncio(
     *,
     client: Client,
     show_secrets: Union[Unset, bool] = UNSET,
-) -> Agent | None:
-    """Get agent by name
+) -> Union[Agent, Error] | None:
+    """Get agent
+
+     Returns detailed information about an agent including its current deployment status, configuration,
+    events history, and inference endpoint URL.
 
     Args:
         agent_name (str):
@@ -155,7 +184,7 @@ async def asyncio(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Agent
+        Union[Agent, Error]
     """
 
     return (

blaxel/core/client/api/agents/list_agents.py

@@ -1,11 +1,12 @@
 from http import HTTPStatus
-from typing import Any
+from typing import Any, Union
 
 import httpx
 
 from ... import errors
 from ...client import Client
 from ...models.agent import Agent
+from ...models.error import Error
 from ...types import Response
 
 
@@ -18,7 +19,9 @@ def _get_kwargs() -> dict[str, Any]:
     return _kwargs
 
 
-def _parse_response(*, client: Client, response: httpx.Response) -> list["Agent"] | None:
+def _parse_response(
+    *, client: Client, response: httpx.Response
+) -> Union[Error, list["Agent"]] | None:
     if response.status_code == 200:
         response_200 = []
         _response_200 = response.json()
@@ -28,13 +31,27 @@ def _parse_response(*, client: Client, response: httpx.Response) -> list["Agent"
             response_200.append(response_200_item)
 
         return response_200
+    if response.status_code == 401:
+        response_401 = Error.from_dict(response.json())
+
+        return response_401
+    if response.status_code == 403:
+        response_403 = Error.from_dict(response.json())
+
+        return response_403
+    if response.status_code == 500:
+        response_500 = Error.from_dict(response.json())
+
+        return response_500
     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[list["Agent"]]:
+def _build_response(
+    *, client: Client, response: httpx.Response
+) -> Response[Union[Error, list["Agent"]]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -46,15 +63,18 @@ def _build_response(*, client: Client, response: httpx.Response) -> Response[lis
 def sync_detailed(
     *,
     client: Client,
-) -> Response[list["Agent"]]:
+) -> Response[Union[Error, list["Agent"]]]:
     """List all agents
 
+     Returns all AI agents deployed in the workspace. Each agent includes its deployment status, runtime
+    configuration, and global inference endpoint URL.
+
     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[list['Agent']]
+        Response[Union[Error, list['Agent']]]
     """
 
     kwargs = _get_kwargs()
@@ -69,15 +89,18 @@ def sync_detailed(
 def sync(
     *,
     client: Client,
-) -> list["Agent"] | None:
+) -> Union[Error, list["Agent"]] | None:
     """List all agents
 
+     Returns all AI agents deployed in the workspace. Each agent includes its deployment status, runtime
+    configuration, and global inference endpoint URL.
+
     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:
-        list['Agent']
+        Union[Error, list['Agent']]
     """
 
     return sync_detailed(
@@ -88,15 +111,18 @@ def sync(
 async def asyncio_detailed(
     *,
     client: Client,
-) -> Response[list["Agent"]]:
+) -> Response[Union[Error, list["Agent"]]]:
     """List all agents
 
+     Returns all AI agents deployed in the workspace. Each agent includes its deployment status, runtime
+    configuration, and global inference endpoint URL.
+
     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[list['Agent']]
+        Response[Union[Error, list['Agent']]]
     """
 
     kwargs = _get_kwargs()
@@ -109,15 +135,18 @@ async def asyncio_detailed(
 async def asyncio(
     *,
     client: Client,
-) -> list["Agent"] | None:
+) -> Union[Error, list["Agent"]] | None:
     """List all agents
 
+     Returns all AI agents deployed in the workspace. Each agent includes its deployment status, runtime
+    configuration, and global inference endpoint URL.
+
     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:
-        list['Agent']
+        Union[Error, list['Agent']]
     """
 
     return (