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

beamlit/api/models/update_model.py

@@ -6,14 +6,13 @@ import httpx
 from ... import errors
 from ...client import AuthenticatedClient, Client
 from ...models.model import Model
-from ...models.model_with_deployments import ModelWithDeployments
 from ...types import Response
 
 
 def _get_kwargs(
     model_name: str,
     *,
-    body: ModelWithDeployments,
+    body: Model,
 ) -> dict[str, Any]:
     headers: dict[str, Any] = {}
 
@@ -55,7 +54,7 @@ def sync_detailed(
     model_name: str,
     *,
     client: AuthenticatedClient,
-    body: ModelWithDeployments,
+    body: Model,
 ) -> Response[Model]:
     """Create or update model
 
@@ -63,8 +62,8 @@ def sync_detailed(
 
     Args:
         model_name (str):
-        body (ModelWithDeployments): Logical object representing a model but with deployment
-            definition inside
+        body (Model): Logical object representing a model, that can be instantiated in multiple
+            environments as model deployments
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -90,7 +89,7 @@ def sync(
     model_name: str,
     *,
     client: AuthenticatedClient,
-    body: ModelWithDeployments,
+    body: Model,
 ) -> Optional[Model]:
     """Create or update model
 
@@ -98,8 +97,8 @@ def sync(
 
     Args:
         model_name (str):
-        body (ModelWithDeployments): Logical object representing a model but with deployment
-            definition inside
+        body (Model): Logical object representing a model, that can be instantiated in multiple
+            environments as model deployments
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -120,7 +119,7 @@ async def asyncio_detailed(
     model_name: str,
     *,
     client: AuthenticatedClient,
-    body: ModelWithDeployments,
+    body: Model,
 ) -> Response[Model]:
     """Create or update model
 
@@ -128,8 +127,8 @@ async def asyncio_detailed(
 
     Args:
         model_name (str):
-        body (ModelWithDeployments): Logical object representing a model but with deployment
-            definition inside
+        body (Model): Logical object representing a model, that can be instantiated in multiple
+            environments as model deployments
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -153,7 +152,7 @@ async def asyncio(
     model_name: str,
     *,
     client: AuthenticatedClient,
-    body: ModelWithDeployments,
+    body: Model,
 ) -> Optional[Model]:
     """Create or update model
 
@@ -161,8 +160,8 @@ async def asyncio(
 
     Args:
         model_name (str):
-        body (ModelWithDeployments): Logical object representing a model but with deployment
-            definition inside
+        body (Model): Logical object representing a model, that can be instantiated in multiple
+            environments as model deployments
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.

beamlit/api/{agents/get_agent_deployment_metrics.py → privateclusters/create_private_cluster.py}

@@ -1,21 +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.resource_deployment_metrics import ResourceDeploymentMetrics
+from ...models.private_cluster import PrivateCluster
 from ...types import Response
 
 
-def _get_kwargs(
-    agent_name: str,
-    environment_name: str,
-) -> dict[str, Any]:
+def _get_kwargs() -> dict[str, Any]:
     _kwargs: dict[str, Any] = {
-        "method": "get",
-        "url": f"/agents/{agent_name}/deployments/{environment_name}/metrics",
+        "method": "post",
+        "url": "/privateclusters",
     }
 
     return _kwargs
@@ -23,11 +20,17 @@ def _get_kwargs(
 
 def _parse_response(
     *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Optional[ResourceDeploymentMetrics]:
+) -> Optional[Union[Any, PrivateCluster]]:
     if response.status_code == 200:
-        response_200 = ResourceDeploymentMetrics.from_dict(response.json())
+        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:
@@ -36,7 +39,7 @@ def _parse_response(
 
 def _build_response(
     *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Response[ResourceDeploymentMetrics]:
+) -> Response[Union[Any, PrivateCluster]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -46,28 +49,20 @@ def _build_response(
 
 
 def sync_detailed(
-    agent_name: str,
-    environment_name: str,
     *,
     client: AuthenticatedClient,
-) -> Response[ResourceDeploymentMetrics]:
-    """
-    Args:
-        agent_name (str):
-        environment_name (str):
+) -> Response[Union[Any, PrivateCluster]]:
+    """Create private cluster
 
     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[ResourceDeploymentMetrics]
+        Response[Union[Any, PrivateCluster]]
     """
 
-    kwargs = _get_kwargs(
-        agent_name=agent_name,
-        environment_name=environment_name,
-    )
+    kwargs = _get_kwargs()
 
     response = client.get_httpx_client().request(
         **kwargs,
@@ -77,54 +72,39 @@ def sync_detailed(
 
 
 def sync(
-    agent_name: str,
-    environment_name: str,
     *,
     client: AuthenticatedClient,
-) -> Optional[ResourceDeploymentMetrics]:
-    """
-    Args:
-        agent_name (str):
-        environment_name (str):
+) -> Optional[Union[Any, PrivateCluster]]:
+    """Create private cluster
 
     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:
-        ResourceDeploymentMetrics
+        Union[Any, PrivateCluster]
     """
 
     return sync_detailed(
-        agent_name=agent_name,
-        environment_name=environment_name,
         client=client,
     ).parsed
 
 
 async def asyncio_detailed(
-    agent_name: str,
-    environment_name: str,
     *,
     client: AuthenticatedClient,
-) -> Response[ResourceDeploymentMetrics]:
-    """
-    Args:
-        agent_name (str):
-        environment_name (str):
+) -> Response[Union[Any, PrivateCluster]]:
+    """Create private cluster
 
     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[ResourceDeploymentMetrics]
+        Response[Union[Any, PrivateCluster]]
     """
 
-    kwargs = _get_kwargs(
-        agent_name=agent_name,
-        environment_name=environment_name,
-    )
+    kwargs = _get_kwargs()
 
     response = await client.get_async_httpx_client().request(**kwargs)
 
@@ -132,28 +112,21 @@ async def asyncio_detailed(
 
 
 async def asyncio(
-    agent_name: str,
-    environment_name: str,
     *,
     client: AuthenticatedClient,
-) -> Optional[ResourceDeploymentMetrics]:
-    """
-    Args:
-        agent_name (str):
-        environment_name (str):
+) -> Optional[Union[Any, PrivateCluster]]:
+    """Create private cluster
 
     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:
-        ResourceDeploymentMetrics
+        Union[Any, PrivateCluster]
     """
 
     return (
         await asyncio_detailed(
-            agent_name=agent_name,
-            environment_name=environment_name,
             client=client,
         )
     ).parsed

beamlit/api/{functions/list_function_deployments.py → privateclusters/delete_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.function_deployment import FunctionDeployment
+from ...models.private_cluster import PrivateCluster
 from ...types import Response
 
 
 def _get_kwargs(
-    function_name: str,
+    private_cluster_name: str,
 ) -> dict[str, Any]:
     _kwargs: dict[str, Any] = {
-        "method": "get",
-        "url": f"/functions/{function_name}/deployments",
+        "method": "delete",
+        "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["FunctionDeployment"]]:
+) -> 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 = FunctionDeployment.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["FunctionDeployment"]]:
+) -> Response[Union[Any, PrivateCluster]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -50,25 +51,25 @@ def _build_response(
 
 
 def sync_detailed(
-    function_name: str,
+    private_cluster_name: str,
     *,
     client: AuthenticatedClient,
-) -> Response[list["FunctionDeployment"]]:
-    """List all function deployments
+) -> Response[Union[Any, PrivateCluster]]:
+    """Delete private cluster
 
     Args:
-        function_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['FunctionDeployment']]
+        Response[Union[Any, PrivateCluster]]
     """
 
     kwargs = _get_kwargs(
-        function_name=function_name,
+        private_cluster_name=private_cluster_name,
     )
 
     response = client.get_httpx_client().request(
@@ -79,49 +80,49 @@ def sync_detailed(
 
 
 def sync(
-    function_name: str,
+    private_cluster_name: str,
     *,
     client: AuthenticatedClient,
-) -> Optional[list["FunctionDeployment"]]:
-    """List all function deployments
+) -> Optional[Union[Any, PrivateCluster]]:
+    """Delete private cluster
 
     Args:
-        function_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['FunctionDeployment']
+        Union[Any, PrivateCluster]
     """
 
     return sync_detailed(
-        function_name=function_name,
+        private_cluster_name=private_cluster_name,
         client=client,
     ).parsed
 
 
 async def asyncio_detailed(
-    function_name: str,
+    private_cluster_name: str,
     *,
     client: AuthenticatedClient,
-) -> Response[list["FunctionDeployment"]]:
-    """List all function deployments
+) -> Response[Union[Any, PrivateCluster]]:
+    """Delete private cluster
 
     Args:
-        function_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['FunctionDeployment']]
+        Response[Union[Any, PrivateCluster]]
     """
 
     kwargs = _get_kwargs(
-        function_name=function_name,
+        private_cluster_name=private_cluster_name,
     )
 
     response = await client.get_async_httpx_client().request(**kwargs)
@@ -130,26 +131,26 @@ async def asyncio_detailed(
 
 
 async def asyncio(
-    function_name: str,
+    private_cluster_name: str,
     *,
     client: AuthenticatedClient,
-) -> Optional[list["FunctionDeployment"]]:
-    """List all function deployments
+) -> Optional[Union[Any, PrivateCluster]]:
+    """Delete private cluster
 
     Args:
-        function_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['FunctionDeployment']
+        Union[Any, PrivateCluster]
     """
 
     return (
         await asyncio_detailed(
-            function_name=function_name,
+            private_cluster_name=private_cluster_name,
             client=client,
         )
     ).parsed

beamlit/api/{agents/get_agent_deployment.py → privateclusters/get_private_cluster.py}

@@ -1,21 +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.agent_deployment import AgentDeployment
+from ...models.private_cluster import PrivateCluster
 from ...types import Response
 
 
 def _get_kwargs(
-    agent_name: str,
-    environment_name: str,
+    private_cluster_name: str,
 ) -> dict[str, Any]:
     _kwargs: dict[str, Any] = {
         "method": "get",
-        "url": f"/agents/{agent_name}/deployments/{environment_name}",
+        "url": f"/privateclusters/{private_cluster_name}",
     }
 
     return _kwargs
@@ -23,11 +22,20 @@ def _get_kwargs(
 
 def _parse_response(
     *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Optional[AgentDeployment]:
+) -> Optional[Union[Any, PrivateCluster]]:
     if response.status_code == 200:
-        response_200 = AgentDeployment.from_dict(response.json())
+        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 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:
@@ -36,7 +44,7 @@ def _parse_response(
 
 def _build_response(
     *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Response[AgentDeployment]:
+) -> Response[Union[Any, PrivateCluster]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -46,28 +54,25 @@ def _build_response(
 
 
 def sync_detailed(
-    agent_name: str,
-    environment_name: str,
+    private_cluster_name: str,
     *,
     client: AuthenticatedClient,
-) -> Response[AgentDeployment]:
-    """Get agent deployment by environment name
+) -> Response[Union[Any, PrivateCluster]]:
+    """Get private cluster by name
 
     Args:
-        agent_name (str):
-        environment_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[AgentDeployment]
+        Response[Union[Any, PrivateCluster]]
     """
 
     kwargs = _get_kwargs(
-        agent_name=agent_name,
-        environment_name=environment_name,
+        private_cluster_name=private_cluster_name,
     )
 
     response = client.get_httpx_client().request(
@@ -78,55 +83,49 @@ def sync_detailed(
 
 
 def sync(
-    agent_name: str,
-    environment_name: str,
+    private_cluster_name: str,
     *,
     client: AuthenticatedClient,
-) -> Optional[AgentDeployment]:
-    """Get agent deployment by environment name
+) -> Optional[Union[Any, PrivateCluster]]:
+    """Get private cluster by name
 
     Args:
-        agent_name (str):
-        environment_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:
-        AgentDeployment
+        Union[Any, PrivateCluster]
     """
 
     return sync_detailed(
-        agent_name=agent_name,
-        environment_name=environment_name,
+        private_cluster_name=private_cluster_name,
         client=client,
     ).parsed
 
 
 async def asyncio_detailed(
-    agent_name: str,
-    environment_name: str,
+    private_cluster_name: str,
     *,
     client: AuthenticatedClient,
-) -> Response[AgentDeployment]:
-    """Get agent deployment by environment name
+) -> Response[Union[Any, PrivateCluster]]:
+    """Get private cluster by name
 
     Args:
-        agent_name (str):
-        environment_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[AgentDeployment]
+        Response[Union[Any, PrivateCluster]]
     """
 
     kwargs = _get_kwargs(
-        agent_name=agent_name,
-        environment_name=environment_name,
+        private_cluster_name=private_cluster_name,
     )
 
     response = await client.get_async_httpx_client().request(**kwargs)
@@ -135,29 +134,26 @@ async def asyncio_detailed(
 
 
 async def asyncio(
-    agent_name: str,
-    environment_name: str,
+    private_cluster_name: str,
     *,
     client: AuthenticatedClient,
-) -> Optional[AgentDeployment]:
-    """Get agent deployment by environment name
+) -> Optional[Union[Any, PrivateCluster]]:
+    """Get private cluster by name
 
     Args:
-        agent_name (str):
-        environment_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:
-        AgentDeployment
+        Union[Any, PrivateCluster]
     """
 
     return (
         await asyncio_detailed(
-            agent_name=agent_name,
-            environment_name=environment_name,
+            private_cluster_name=private_cluster_name,
             client=client,
         )
     ).parsed