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

blaxel/core/client/api/models/create_model.py

@@ -1,10 +1,11 @@
 from http import HTTPStatus
-from typing import Any
+from typing import Any, Union
 
 import httpx
 
 from ... import errors
 from ...client import Client
+from ...models.error import Error
 from ...models.model import Model
 from ...types import Response
 
@@ -32,18 +33,38 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(*, client: Client, response: httpx.Response) -> Model | None:
+def _parse_response(*, client: Client, response: httpx.Response) -> Union[Error, Model] | None:
     if response.status_code == 200:
         response_200 = Model.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[Model]:
+def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[Error, Model]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -56,20 +77,22 @@ def sync_detailed(
     *,
     client: Client,
     body: Model,
-) -> Response[Model]:
-    """Create model
+) -> Response[Union[Error, Model]]:
+    """Create model endpoint
 
-     Creates a model.
+     Creates a new model gateway endpoint that proxies requests to an external LLM provider. Requires an
+    integration connection with valid API credentials for the target provider.
 
     Args:
-        body (Model): Logical object representing a model
+        body (Model): Gateway endpoint to external LLM provider APIs (OpenAI, Anthropic, etc.)
+            with unified access control, credentials management, and usage tracking.
 
     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[Model]
+        Response[Union[Error, Model]]
     """
 
     kwargs = _get_kwargs(
@@ -87,20 +110,22 @@ def sync(
     *,
     client: Client,
     body: Model,
-) -> Model | None:
-    """Create model
+) -> Union[Error, Model] | None:
+    """Create model endpoint
 
-     Creates a model.
+     Creates a new model gateway endpoint that proxies requests to an external LLM provider. Requires an
+    integration connection with valid API credentials for the target provider.
 
     Args:
-        body (Model): Logical object representing a model
+        body (Model): Gateway endpoint to external LLM provider APIs (OpenAI, Anthropic, etc.)
+            with unified access control, credentials management, and usage tracking.
 
     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:
-        Model
+        Union[Error, Model]
     """
 
     return sync_detailed(
@@ -113,20 +138,22 @@ async def asyncio_detailed(
     *,
     client: Client,
     body: Model,
-) -> Response[Model]:
-    """Create model
+) -> Response[Union[Error, Model]]:
+    """Create model endpoint
 
-     Creates a model.
+     Creates a new model gateway endpoint that proxies requests to an external LLM provider. Requires an
+    integration connection with valid API credentials for the target provider.
 
     Args:
-        body (Model): Logical object representing a model
+        body (Model): Gateway endpoint to external LLM provider APIs (OpenAI, Anthropic, etc.)
+            with unified access control, credentials management, and usage tracking.
 
     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[Model]
+        Response[Union[Error, Model]]
     """
 
     kwargs = _get_kwargs(
@@ -142,20 +169,22 @@ async def asyncio(
     *,
     client: Client,
     body: Model,
-) -> Model | None:
-    """Create model
+) -> Union[Error, Model] | None:
+    """Create model endpoint
 
-     Creates a model.
+     Creates a new model gateway endpoint that proxies requests to an external LLM provider. Requires an
+    integration connection with valid API credentials for the target provider.
 
     Args:
-        body (Model): Logical object representing a model
+        body (Model): Gateway endpoint to external LLM provider APIs (OpenAI, Anthropic, etc.)
+            with unified access control, credentials management, and usage tracking.
 
     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:
-        Model
+        Union[Error, Model]
     """
 
     return (

blaxel/core/client/api/models/delete_model.py

@@ -1,10 +1,11 @@
 from http import HTTPStatus
-from typing import Any
+from typing import Any, Union
 
 import httpx
 
 from ... import errors
 from ...client import Client
+from ...models.error import Error
 from ...models.model import Model
 from ...types import Response
 
@@ -20,18 +21,34 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(*, client: Client, response: httpx.Response) -> Model | None:
+def _parse_response(*, client: Client, response: httpx.Response) -> Union[Error, Model] | None:
     if response.status_code == 200:
         response_200 = Model.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[Model]:
+def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[Error, Model]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -44,10 +61,11 @@ def sync_detailed(
     model_name: str,
     *,
     client: Client,
-) -> Response[Model]:
-    """Delete model
+) -> Response[Union[Error, Model]]:
+    """Delete model endpoint
 
-     Deletes a model by name.
+     Permanently deletes a model gateway endpoint. Any agents or applications using this endpoint will
+    need to be updated to use a different model.
 
     Args:
         model_name (str):
@@ -57,7 +75,7 @@ def sync_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[Model]
+        Response[Union[Error, Model]]
     """
 
     kwargs = _get_kwargs(
@@ -75,10 +93,11 @@ def sync(
     model_name: str,
     *,
     client: Client,
-) -> Model | None:
-    """Delete model
+) -> Union[Error, Model] | None:
+    """Delete model endpoint
 
-     Deletes a model by name.
+     Permanently deletes a model gateway endpoint. Any agents or applications using this endpoint will
+    need to be updated to use a different model.
 
     Args:
         model_name (str):
@@ -88,7 +107,7 @@ def sync(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Model
+        Union[Error, Model]
     """
 
     return sync_detailed(
@@ -101,10 +120,11 @@ async def asyncio_detailed(
     model_name: str,
     *,
     client: Client,
-) -> Response[Model]:
-    """Delete model
+) -> Response[Union[Error, Model]]:
+    """Delete model endpoint
 
-     Deletes a model by name.
+     Permanently deletes a model gateway endpoint. Any agents or applications using this endpoint will
+    need to be updated to use a different model.
 
     Args:
         model_name (str):
@@ -114,7 +134,7 @@ async def asyncio_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[Model]
+        Response[Union[Error, Model]]
     """
 
     kwargs = _get_kwargs(
@@ -130,10 +150,11 @@ async def asyncio(
     model_name: str,
     *,
     client: Client,
-) -> Model | None:
-    """Delete model
+) -> Union[Error, Model] | None:
+    """Delete model endpoint
 
-     Deletes a model by name.
+     Permanently deletes a model gateway endpoint. Any agents or applications using this endpoint will
+    need to be updated to use a different model.
 
     Args:
         model_name (str):
@@ -143,7 +164,7 @@ async def asyncio(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Model
+        Union[Error, Model]
     """
 
     return (

blaxel/core/client/api/models/get_model.py

@@ -1,10 +1,11 @@
 from http import HTTPStatus
-from typing import Any
+from typing import Any, Union
 
 import httpx
 
 from ... import errors
 from ...client import Client
+from ...models.error import Error
 from ...models.model import Model
 from ...types import Response
 
@@ -20,18 +21,34 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(*, client: Client, response: httpx.Response) -> Model | None:
+def _parse_response(*, client: Client, response: httpx.Response) -> Union[Error, Model] | None:
     if response.status_code == 200:
         response_200 = Model.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[Model]:
+def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[Error, Model]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -44,10 +61,11 @@ def sync_detailed(
     model_name: str,
     *,
     client: Client,
-) -> Response[Model]:
-    """Get model
+) -> Response[Union[Error, Model]]:
+    """Get model endpoint
 
-     Returns a model by name.
+     Returns detailed information about a model gateway endpoint including its provider configuration,
+    integration connection, and usage status.
 
     Args:
         model_name (str):
@@ -57,7 +75,7 @@ def sync_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[Model]
+        Response[Union[Error, Model]]
     """
 
     kwargs = _get_kwargs(
@@ -75,10 +93,11 @@ def sync(
     model_name: str,
     *,
     client: Client,
-) -> Model | None:
-    """Get model
+) -> Union[Error, Model] | None:
+    """Get model endpoint
 
-     Returns a model by name.
+     Returns detailed information about a model gateway endpoint including its provider configuration,
+    integration connection, and usage status.
 
     Args:
         model_name (str):
@@ -88,7 +107,7 @@ def sync(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Model
+        Union[Error, Model]
     """
 
     return sync_detailed(
@@ -101,10 +120,11 @@ async def asyncio_detailed(
     model_name: str,
     *,
     client: Client,
-) -> Response[Model]:
-    """Get model
+) -> Response[Union[Error, Model]]:
+    """Get model endpoint
 
-     Returns a model by name.
+     Returns detailed information about a model gateway endpoint including its provider configuration,
+    integration connection, and usage status.
 
     Args:
         model_name (str):
@@ -114,7 +134,7 @@ async def asyncio_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[Model]
+        Response[Union[Error, Model]]
     """
 
     kwargs = _get_kwargs(
@@ -130,10 +150,11 @@ async def asyncio(
     model_name: str,
     *,
     client: Client,
-) -> Model | None:
-    """Get model
+) -> Union[Error, Model] | None:
+    """Get model endpoint
 
-     Returns a model by name.
+     Returns detailed information about a model gateway endpoint including its provider configuration,
+    integration connection, and usage status.
 
     Args:
         model_name (str):
@@ -143,7 +164,7 @@ async def asyncio(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Model
+        Union[Error, Model]
     """
 
     return (

blaxel/core/client/api/models/list_models.py

@@ -1,10 +1,11 @@
 from http import HTTPStatus
-from typing import Any
+from typing import Any, Union
 
 import httpx
 
 from ... import errors
 from ...client import Client
+from ...models.error import Error
 from ...models.model import Model
 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["Model"] | None:
+def _parse_response(
+    *, client: Client, response: httpx.Response
+) -> Union[Error, list["Model"]] | 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["Model"
             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["Model"]]:
+def _build_response(
+    *, client: Client, response: httpx.Response
+) -> Response[Union[Error, list["Model"]]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -46,17 +63,18 @@ def _build_response(*, client: Client, response: httpx.Response) -> Response[lis
 def sync_detailed(
     *,
     client: Client,
-) -> Response[list["Model"]]:
-    """List models
+) -> Response[Union[Error, list["Model"]]]:
+    """List model endpoints
 
-     Returns a list of all models in the workspace.
+     Returns all model gateway endpoints configured in the workspace. Each model represents a proxy to an
+    external LLM provider (OpenAI, Anthropic, etc.) with unified access control.
 
     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['Model']]
+        Response[Union[Error, list['Model']]]
     """
 
     kwargs = _get_kwargs()
@@ -71,17 +89,18 @@ def sync_detailed(
 def sync(
     *,
     client: Client,
-) -> list["Model"] | None:
-    """List models
+) -> Union[Error, list["Model"]] | None:
+    """List model endpoints
 
-     Returns a list of all models in the workspace.
+     Returns all model gateway endpoints configured in the workspace. Each model represents a proxy to an
+    external LLM provider (OpenAI, Anthropic, etc.) with unified access control.
 
     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['Model']
+        Union[Error, list['Model']]
     """
 
     return sync_detailed(
@@ -92,17 +111,18 @@ def sync(
 async def asyncio_detailed(
     *,
     client: Client,
-) -> Response[list["Model"]]:
-    """List models
+) -> Response[Union[Error, list["Model"]]]:
+    """List model endpoints
 
-     Returns a list of all models in the workspace.
+     Returns all model gateway endpoints configured in the workspace. Each model represents a proxy to an
+    external LLM provider (OpenAI, Anthropic, etc.) with unified access control.
 
     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['Model']]
+        Response[Union[Error, list['Model']]]
     """
 
     kwargs = _get_kwargs()
@@ -115,17 +135,18 @@ async def asyncio_detailed(
 async def asyncio(
     *,
     client: Client,
-) -> list["Model"] | None:
-    """List models
+) -> Union[Error, list["Model"]] | None:
+    """List model endpoints
 
-     Returns a list of all models in the workspace.
+     Returns all model gateway endpoints configured in the workspace. Each model represents a proxy to an
+    external LLM provider (OpenAI, Anthropic, etc.) with unified access control.
 
     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['Model']
+        Union[Error, list['Model']]
     """
 
     return (