beamlit 0.0.24rc19__py3-none-any.whl → 0.0.24rc21__py3-none-any.whl

beamlit/api/privateclusters/get_private_cluster_health.py

@@ -0,0 +1,97 @@
+from http import HTTPStatus
+from typing import Any, Optional, Union
+
+import httpx
+
+from ... import errors
+from ...client import AuthenticatedClient, Client
+from ...types import Response
+
+
+def _get_kwargs(
+    private_cluster_name: str,
+) -> dict[str, Any]:
+    _kwargs: dict[str, Any] = {
+        "method": "get",
+        "url": f"/privateclusters/{private_cluster_name}/health",
+    }
+
+    return _kwargs
+
+
+def _parse_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Optional[Any]:
+    if response.status_code == 200:
+        return None
+    if response.status_code == 401:
+        return None
+    if response.status_code == 403:
+        return None
+    if client.raise_on_unexpected_status:
+        raise errors.UnexpectedStatus(response.status_code, response.content)
+    else:
+        return None
+
+
+def _build_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Response[Any]:
+    return Response(
+        status_code=HTTPStatus(response.status_code),
+        content=response.content,
+        headers=response.headers,
+        parsed=_parse_response(client=client, response=response),
+    )
+
+
+def sync_detailed(
+    private_cluster_name: str,
+    *,
+    client: Union[AuthenticatedClient, Client],
+) -> Response[Any]:
+    """Get private cluster health
+
+    Args:
+        private_cluster_name (str):
+
+    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[Any]
+    """
+
+    kwargs = _get_kwargs(
+        private_cluster_name=private_cluster_name,
+    )
+
+    response = client.get_httpx_client().request(
+        **kwargs,
+    )
+
+    return _build_response(client=client, response=response)
+
+
+async def asyncio_detailed(
+    private_cluster_name: str,
+    *,
+    client: Union[AuthenticatedClient, Client],
+) -> Response[Any]:
+    """Get private cluster health
+
+    Args:
+        private_cluster_name (str):
+
+    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[Any]
+    """
+
+    kwargs = _get_kwargs(
+        private_cluster_name=private_cluster_name,
+    )
+
+    response = await client.get_async_httpx_client().request(**kwargs)
+
+    return _build_response(client=client, response=response)

beamlit/api/{agents/list_agent_deployments.py → privateclusters/list_private_clusters.py}

@@ -1,20 +1,18 @@
 from http import HTTPStatus
-from typing import Any, Optional, Union
+from typing import Any, Optional, Union, cast
 
 import httpx
 
 from ... import errors
 from ...client import AuthenticatedClient, Client
-from ...models.agent_deployment import AgentDeployment
+from ...models.private_cluster import PrivateCluster
 from ...types import Response
 
 
-def _get_kwargs(
-    agent_name: str,
-) -> dict[str, Any]:
+def _get_kwargs() -> dict[str, Any]:
     _kwargs: dict[str, Any] = {
         "method": "get",
-        "url": f"/agents/{agent_name}/deployments",
+        "url": "/privateclusters",
     }
 
     return _kwargs
@@ -22,16 +20,25 @@ def _get_kwargs(
 
 def _parse_response(
     *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Optional[list["AgentDeployment"]]:
+) -> Optional[Union[Any, list["PrivateCluster"]]]:
     if response.status_code == 200:
         response_200 = []
         _response_200 = response.json()
         for response_200_item_data in _response_200:
-            response_200_item = AgentDeployment.from_dict(response_200_item_data)
+            response_200_item = PrivateCluster.from_dict(response_200_item_data)
 
             response_200.append(response_200_item)
 
         return response_200
+    if response.status_code == 401:
+        response_401 = cast(Any, None)
+        return response_401
+    if response.status_code == 403:
+        response_403 = cast(Any, None)
+        return response_403
+    if response.status_code == 404:
+        response_404 = cast(Any, None)
+        return response_404
     if client.raise_on_unexpected_status:
         raise errors.UnexpectedStatus(response.status_code, response.content)
     else:
@@ -40,7 +47,7 @@ def _parse_response(
 
 def _build_response(
     *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Response[list["AgentDeployment"]]:
+) -> Response[Union[Any, list["PrivateCluster"]]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -50,26 +57,20 @@ def _build_response(
 
 
 def sync_detailed(
-    agent_name: str,
     *,
     client: AuthenticatedClient,
-) -> Response[list["AgentDeployment"]]:
-    """List all agent deployments
-
-    Args:
-        agent_name (str):
+) -> Response[Union[Any, list["PrivateCluster"]]]:
+    """List all private clusters
 
     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['AgentDeployment']]
+        Response[Union[Any, list['PrivateCluster']]]
     """
 
-    kwargs = _get_kwargs(
-        agent_name=agent_name,
-    )
+    kwargs = _get_kwargs()
 
     response = client.get_httpx_client().request(
         **kwargs,
@@ -79,50 +80,39 @@ def sync_detailed(
 
 
 def sync(
-    agent_name: str,
     *,
     client: AuthenticatedClient,
-) -> Optional[list["AgentDeployment"]]:
-    """List all agent deployments
-
-    Args:
-        agent_name (str):
+) -> Optional[Union[Any, list["PrivateCluster"]]]:
+    """List all private clusters
 
     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['AgentDeployment']
+        Union[Any, list['PrivateCluster']]
     """
 
     return sync_detailed(
-        agent_name=agent_name,
         client=client,
     ).parsed
 
 
 async def asyncio_detailed(
-    agent_name: str,
     *,
     client: AuthenticatedClient,
-) -> Response[list["AgentDeployment"]]:
-    """List all agent deployments
-
-    Args:
-        agent_name (str):
+) -> Response[Union[Any, list["PrivateCluster"]]]:
+    """List all private clusters
 
     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['AgentDeployment']]
+        Response[Union[Any, list['PrivateCluster']]]
     """
 
-    kwargs = _get_kwargs(
-        agent_name=agent_name,
-    )
+    kwargs = _get_kwargs()
 
     response = await client.get_async_httpx_client().request(**kwargs)
 
@@ -130,26 +120,21 @@ async def asyncio_detailed(
 
 
 async def asyncio(
-    agent_name: str,
     *,
     client: AuthenticatedClient,
-) -> Optional[list["AgentDeployment"]]:
-    """List all agent deployments
-
-    Args:
-        agent_name (str):
+) -> Optional[Union[Any, list["PrivateCluster"]]]:
+    """List all private clusters
 
     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['AgentDeployment']
+        Union[Any, list['PrivateCluster']]
     """
 
     return (
         await asyncio_detailed(
-            agent_name=agent_name,
             client=client,
         )
     ).parsed

beamlit/api/{models/list_model_deployments.py → privateclusters/update_private_cluster.py}

@@ -1,20 +1,20 @@
 from http import HTTPStatus
-from typing import Any, Optional, Union
+from typing import Any, Optional, Union, cast
 
 import httpx
 
 from ... import errors
 from ...client import AuthenticatedClient, Client
-from ...models.model_deployment import ModelDeployment
+from ...models.private_cluster import PrivateCluster
 from ...types import Response
 
 
 def _get_kwargs(
-    model_name: str,
+    private_cluster_name: str,
 ) -> dict[str, Any]:
     _kwargs: dict[str, Any] = {
-        "method": "get",
-        "url": f"/models/{model_name}/deployments",
+        "method": "put",
+        "url": f"/privateclusters/{private_cluster_name}",
     }
 
     return _kwargs
@@ -22,16 +22,17 @@ def _get_kwargs(
 
 def _parse_response(
     *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Optional[list["ModelDeployment"]]:
+) -> Optional[Union[Any, PrivateCluster]]:
     if response.status_code == 200:
-        response_200 = []
-        _response_200 = response.json()
-        for response_200_item_data in _response_200:
-            response_200_item = ModelDeployment.from_dict(response_200_item_data)
-
-            response_200.append(response_200_item)
+        response_200 = PrivateCluster.from_dict(response.json())
 
         return response_200
+    if response.status_code == 401:
+        response_401 = cast(Any, None)
+        return response_401
+    if response.status_code == 403:
+        response_403 = cast(Any, None)
+        return response_403
     if client.raise_on_unexpected_status:
         raise errors.UnexpectedStatus(response.status_code, response.content)
     else:
@@ -40,7 +41,7 @@ def _parse_response(
 
 def _build_response(
     *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Response[list["ModelDeployment"]]:
+) -> Response[Union[Any, PrivateCluster]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -50,27 +51,25 @@ def _build_response(
 
 
 def sync_detailed(
-    model_name: str,
+    private_cluster_name: str,
     *,
     client: AuthenticatedClient,
-) -> Response[list["ModelDeployment"]]:
-    """List model deployments
-
-     Returns a list of all deployments for a model.
+) -> Response[Union[Any, PrivateCluster]]:
+    """Update private cluster
 
     Args:
-        model_name (str):
+        private_cluster_name (str):
 
     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['ModelDeployment']]
+        Response[Union[Any, PrivateCluster]]
     """
 
     kwargs = _get_kwargs(
-        model_name=model_name,
+        private_cluster_name=private_cluster_name,
     )
 
     response = client.get_httpx_client().request(
@@ -81,53 +80,49 @@ def sync_detailed(
 
 
 def sync(
-    model_name: str,
+    private_cluster_name: str,
     *,
     client: AuthenticatedClient,
-) -> Optional[list["ModelDeployment"]]:
-    """List model deployments
-
-     Returns a list of all deployments for a model.
+) -> Optional[Union[Any, PrivateCluster]]:
+    """Update private cluster
 
     Args:
-        model_name (str):
+        private_cluster_name (str):
 
     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['ModelDeployment']
+        Union[Any, PrivateCluster]
     """
 
     return sync_detailed(
-        model_name=model_name,
+        private_cluster_name=private_cluster_name,
         client=client,
     ).parsed
 
 
 async def asyncio_detailed(
-    model_name: str,
+    private_cluster_name: str,
     *,
     client: AuthenticatedClient,
-) -> Response[list["ModelDeployment"]]:
-    """List model deployments
-
-     Returns a list of all deployments for a model.
+) -> Response[Union[Any, PrivateCluster]]:
+    """Update private cluster
 
     Args:
-        model_name (str):
+        private_cluster_name (str):
 
     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['ModelDeployment']]
+        Response[Union[Any, PrivateCluster]]
     """
 
     kwargs = _get_kwargs(
-        model_name=model_name,
+        private_cluster_name=private_cluster_name,
     )
 
     response = await client.get_async_httpx_client().request(**kwargs)
@@ -136,28 +131,26 @@ async def asyncio_detailed(
 
 
 async def asyncio(
-    model_name: str,
+    private_cluster_name: str,
     *,
     client: AuthenticatedClient,
-) -> Optional[list["ModelDeployment"]]:
-    """List model deployments
-
-     Returns a list of all deployments for a model.
+) -> Optional[Union[Any, PrivateCluster]]:
+    """Update private cluster
 
     Args:
-        model_name (str):
+        private_cluster_name (str):
 
     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['ModelDeployment']
+        Union[Any, PrivateCluster]
     """
 
     return (
         await asyncio_detailed(
-            model_name=model_name,
+            private_cluster_name=private_cluster_name,
             client=client,
         )
     ).parsed

beamlit/api/privateclusters/update_private_cluster_health.py

@@ -0,0 +1,97 @@
+from http import HTTPStatus
+from typing import Any, Optional, Union
+
+import httpx
+
+from ... import errors
+from ...client import AuthenticatedClient, Client
+from ...types import Response
+
+
+def _get_kwargs(
+    private_cluster_name: str,
+) -> dict[str, Any]:
+    _kwargs: dict[str, Any] = {
+        "method": "post",
+        "url": f"/privateclusters/{private_cluster_name}/health",
+    }
+
+    return _kwargs
+
+
+def _parse_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Optional[Any]:
+    if response.status_code == 200:
+        return None
+    if response.status_code == 401:
+        return None
+    if response.status_code == 403:
+        return None
+    if client.raise_on_unexpected_status:
+        raise errors.UnexpectedStatus(response.status_code, response.content)
+    else:
+        return None
+
+
+def _build_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Response[Any]:
+    return Response(
+        status_code=HTTPStatus(response.status_code),
+        content=response.content,
+        headers=response.headers,
+        parsed=_parse_response(client=client, response=response),
+    )
+
+
+def sync_detailed(
+    private_cluster_name: str,
+    *,
+    client: Union[AuthenticatedClient, Client],
+) -> Response[Any]:
+    """Update private cluster health
+
+    Args:
+        private_cluster_name (str):
+
+    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[Any]
+    """
+
+    kwargs = _get_kwargs(
+        private_cluster_name=private_cluster_name,
+    )
+
+    response = client.get_httpx_client().request(
+        **kwargs,
+    )
+
+    return _build_response(client=client, response=response)
+
+
+async def asyncio_detailed(
+    private_cluster_name: str,
+    *,
+    client: Union[AuthenticatedClient, Client],
+) -> Response[Any]:
+    """Update private cluster health
+
+    Args:
+        private_cluster_name (str):
+
+    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[Any]
+    """
+
+    kwargs = _get_kwargs(
+        private_cluster_name=private_cluster_name,
+    )
+
+    response = await client.get_async_httpx_client().request(**kwargs)
+
+    return _build_response(client=client, response=response)

beamlit/authentication/device_mode.py

@@ -61,9 +61,7 @@ class BearerToken(Auth):
             return Exception("Invalid JWT token format")
 
         try:
-            claims_bytes = base64.urlsafe_b64decode(
-                parts[1] + "=" * (-len(parts[1]) % 4)
-            )
+            claims_bytes = base64.urlsafe_b64decode(parts[1] + "=" * (-len(parts[1]) % 4))
             claims = json.loads(claims_bytes)
         except Exception as e:
             return Exception(f"Failed to decode/parse JWT claims: {str(e)}")
@@ -80,9 +78,7 @@ class BearerToken(Auth):
         if err:
             return err
 
-        request.headers["X-Beamlit-Authorization"] = (
-            f"Bearer {self.credentials.access_token}"
-        )
+        request.headers["X-Beamlit-Authorization"] = f"Bearer {self.credentials.access_token}"
         request.headers["X-Beamlit-Workspace"] = self.workspace_name
         yield request
 
@@ -99,9 +95,7 @@ class BearerToken(Auth):
         }
 
         try:
-            response = post(
-                url, json=refresh_data, headers={"Content-Type": "application/json"}
-            )
+            response = post(url, json=refresh_data, headers={"Content-Type": "application/json"})
             response.raise_for_status()
             finalize_response = DeviceLoginFinalizeResponse(**response.json())