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

beamlit/api/agents/get_agent_environment_logs.py

@@ -1,5 +1,5 @@
 from http import HTTPStatus
-from typing import Any, List, Optional, Union
+from typing import Any, Optional, Union
 
 import httpx
 
@@ -22,7 +22,7 @@ def _get_kwargs(
 
 def _parse_response(
     *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Optional[List["ResourceLog"]]:
+) -> Optional[list["ResourceLog"]]:
     if response.status_code == 200:
         response_200 = []
         _response_200 = response.json()
@@ -40,7 +40,7 @@ def _parse_response(
 
 def _build_response(
     *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Response[List["ResourceLog"]]:
+) -> Response[list["ResourceLog"]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -53,7 +53,7 @@ def sync_detailed(
     agent_name: str,
     *,
     client: AuthenticatedClient,
-) -> Response[List["ResourceLog"]]:
+) -> Response[list["ResourceLog"]]:
     """
     Args:
         agent_name (str):
@@ -63,7 +63,7 @@ def sync_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[List['ResourceLog']]
+        Response[list['ResourceLog']]
     """
 
     kwargs = _get_kwargs(
@@ -81,7 +81,7 @@ def sync(
     agent_name: str,
     *,
     client: AuthenticatedClient,
-) -> Optional[List["ResourceLog"]]:
+) -> Optional[list["ResourceLog"]]:
     """
     Args:
         agent_name (str):
@@ -91,7 +91,7 @@ def sync(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        List['ResourceLog']
+        list['ResourceLog']
     """
 
     return sync_detailed(
@@ -104,7 +104,7 @@ async def asyncio_detailed(
     agent_name: str,
     *,
     client: AuthenticatedClient,
-) -> Response[List["ResourceLog"]]:
+) -> Response[list["ResourceLog"]]:
     """
     Args:
         agent_name (str):
@@ -114,7 +114,7 @@ async def asyncio_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[List['ResourceLog']]
+        Response[list['ResourceLog']]
     """
 
     kwargs = _get_kwargs(
@@ -130,7 +130,7 @@ async def asyncio(
     agent_name: str,
     *,
     client: AuthenticatedClient,
-) -> Optional[List["ResourceLog"]]:
+) -> Optional[list["ResourceLog"]]:
     """
     Args:
         agent_name (str):
@@ -140,7 +140,7 @@ async def asyncio(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        List['ResourceLog']
+        list['ResourceLog']
     """
 
     return (

beamlit/api/agents/get_agent_history.py

@@ -21,9 +21,7 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(
-    *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Optional[AgentHistory]:
+def _parse_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Optional[AgentHistory]:
     if response.status_code == 200:
         response_200 = AgentHistory.from_dict(response.json())
 
@@ -34,9 +32,7 @@ def _parse_response(
         return None
 
 
-def _build_response(
-    *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Response[AgentHistory]:
+def _build_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Response[AgentHistory]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

beamlit/api/agents/get_agent_metrics.py

@@ -6,15 +6,24 @@ import httpx
 from ... import errors
 from ...client import AuthenticatedClient, Client
 from ...models.resource_metrics import ResourceMetrics
-from ...types import Response
+from ...types import UNSET, Response, Unset
 
 
 def _get_kwargs(
     agent_name: str,
+    *,
+    environment: Union[Unset, str] = UNSET,
 ) -> dict[str, Any]:
+    params: dict[str, Any] = {}
+
+    params["environment"] = environment
+
+    params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
+
     _kwargs: dict[str, Any] = {
         "method": "get",
         "url": f"/agents/{agent_name}/metrics",
+        "params": params,
     }
 
     return _kwargs
@@ -48,11 +57,13 @@ def sync_detailed(
     agent_name: str,
     *,
     client: AuthenticatedClient,
+    environment: Union[Unset, str] = UNSET,
 ) -> Response[ResourceMetrics]:
     """Get agent metrics
 
     Args:
         agent_name (str):
+        environment (Union[Unset, str]):
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -64,6 +75,7 @@ def sync_detailed(
 
     kwargs = _get_kwargs(
         agent_name=agent_name,
+        environment=environment,
     )
 
     response = client.get_httpx_client().request(
@@ -77,11 +89,13 @@ def sync(
     agent_name: str,
     *,
     client: AuthenticatedClient,
+    environment: Union[Unset, str] = UNSET,
 ) -> Optional[ResourceMetrics]:
     """Get agent metrics
 
     Args:
         agent_name (str):
+        environment (Union[Unset, str]):
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -94,6 +108,7 @@ def sync(
     return sync_detailed(
         agent_name=agent_name,
         client=client,
+        environment=environment,
     ).parsed
 
 
@@ -101,11 +116,13 @@ async def asyncio_detailed(
     agent_name: str,
     *,
     client: AuthenticatedClient,
+    environment: Union[Unset, str] = UNSET,
 ) -> Response[ResourceMetrics]:
     """Get agent metrics
 
     Args:
         agent_name (str):
+        environment (Union[Unset, str]):
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -117,6 +134,7 @@ async def asyncio_detailed(
 
     kwargs = _get_kwargs(
         agent_name=agent_name,
+        environment=environment,
     )
 
     response = await client.get_async_httpx_client().request(**kwargs)
@@ -128,11 +146,13 @@ async def asyncio(
     agent_name: str,
     *,
     client: AuthenticatedClient,
+    environment: Union[Unset, str] = UNSET,
 ) -> Optional[ResourceMetrics]:
     """Get agent metrics
 
     Args:
         agent_name (str):
+        environment (Union[Unset, str]):
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -146,5 +166,6 @@ async def asyncio(
         await asyncio_detailed(
             agent_name=agent_name,
             client=client,
+            environment=environment,
         )
     ).parsed

beamlit/api/agents/list_agent_history.py

@@ -1,5 +1,5 @@
 from http import HTTPStatus
-from typing import Any, List, Optional, Union
+from typing import Any, Optional, Union
 
 import httpx
 
@@ -22,7 +22,7 @@ def _get_kwargs(
 
 def _parse_response(
     *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Optional[List["AgentHistory"]]:
+) -> Optional[list["AgentHistory"]]:
     if response.status_code == 200:
         response_200 = []
         _response_200 = response.json()
@@ -40,7 +40,7 @@ def _parse_response(
 
 def _build_response(
     *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Response[List["AgentHistory"]]:
+) -> Response[list["AgentHistory"]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -53,7 +53,7 @@ def sync_detailed(
     agent_name: str,
     *,
     client: AuthenticatedClient,
-) -> Response[List["AgentHistory"]]:
+) -> Response[list["AgentHistory"]]:
     """
     Args:
         agent_name (str):
@@ -63,7 +63,7 @@ def sync_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[List['AgentHistory']]
+        Response[list['AgentHistory']]
     """
 
     kwargs = _get_kwargs(
@@ -81,7 +81,7 @@ def sync(
     agent_name: str,
     *,
     client: AuthenticatedClient,
-) -> Optional[List["AgentHistory"]]:
+) -> Optional[list["AgentHistory"]]:
     """
     Args:
         agent_name (str):
@@ -91,7 +91,7 @@ def sync(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        List['AgentHistory']
+        list['AgentHistory']
     """
 
     return sync_detailed(
@@ -104,7 +104,7 @@ async def asyncio_detailed(
     agent_name: str,
     *,
     client: AuthenticatedClient,
-) -> Response[List["AgentHistory"]]:
+) -> Response[list["AgentHistory"]]:
     """
     Args:
         agent_name (str):
@@ -114,7 +114,7 @@ async def asyncio_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[List['AgentHistory']]
+        Response[list['AgentHistory']]
     """
 
     kwargs = _get_kwargs(
@@ -130,7 +130,7 @@ async def asyncio(
     agent_name: str,
     *,
     client: AuthenticatedClient,
-) -> Optional[List["AgentHistory"]]:
+) -> Optional[list["AgentHistory"]]:
     """
     Args:
         agent_name (str):
@@ -140,7 +140,7 @@ async def asyncio(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        List['AgentHistory']
+        list['AgentHistory']
     """
 
     return (

beamlit/api/agents/list_agents.py

@@ -6,13 +6,23 @@ import httpx
 from ... import errors
 from ...client import AuthenticatedClient, Client
 from ...models.agent import Agent
-from ...types import Response
+from ...types import UNSET, Response, Unset
 
 
-def _get_kwargs() -> dict[str, Any]:
+def _get_kwargs(
+    *,
+    environment: Union[Unset, str] = UNSET,
+) -> dict[str, Any]:
+    params: dict[str, Any] = {}
+
+    params["environment"] = environment
+
+    params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
+
     _kwargs: dict[str, Any] = {
         "method": "get",
         "url": "/agents",
+        "params": params,
     }
 
     return _kwargs
@@ -46,9 +56,13 @@ def _build_response(*, client: Union[AuthenticatedClient, Client], response: htt
 def sync_detailed(
     *,
     client: AuthenticatedClient,
+    environment: Union[Unset, str] = UNSET,
 ) -> Response[list["Agent"]]:
     """List all agents
 
+    Args:
+        environment (Union[Unset, 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.
@@ -57,7 +71,9 @@ def sync_detailed(
         Response[list['Agent']]
     """
 
-    kwargs = _get_kwargs()
+    kwargs = _get_kwargs(
+        environment=environment,
+    )
 
     response = client.get_httpx_client().request(
         **kwargs,
@@ -69,9 +85,13 @@ def sync_detailed(
 def sync(
     *,
     client: AuthenticatedClient,
+    environment: Union[Unset, str] = UNSET,
 ) -> Optional[list["Agent"]]:
     """List all agents
 
+    Args:
+        environment (Union[Unset, 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.
@@ -82,15 +102,20 @@ def sync(
 
     return sync_detailed(
         client=client,
+        environment=environment,
     ).parsed
 
 
 async def asyncio_detailed(
     *,
     client: AuthenticatedClient,
+    environment: Union[Unset, str] = UNSET,
 ) -> Response[list["Agent"]]:
     """List all agents
 
+    Args:
+        environment (Union[Unset, 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.
@@ -99,7 +124,9 @@ async def asyncio_detailed(
         Response[list['Agent']]
     """
 
-    kwargs = _get_kwargs()
+    kwargs = _get_kwargs(
+        environment=environment,
+    )
 
     response = await client.get_async_httpx_client().request(**kwargs)
 
@@ -109,9 +136,13 @@ async def asyncio_detailed(
 async def asyncio(
     *,
     client: AuthenticatedClient,
+    environment: Union[Unset, str] = UNSET,
 ) -> Optional[list["Agent"]]:
     """List all agents
 
+    Args:
+        environment (Union[Unset, 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.
@@ -123,5 +154,6 @@ async def asyncio(
     return (
         await asyncio_detailed(
             client=client,
+            environment=environment,
         )
     ).parsed

beamlit/api/agents/put_agent_history.py

@@ -31,9 +31,7 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(
-    *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Optional[AgentHistory]:
+def _parse_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Optional[AgentHistory]:
     if response.status_code == 200:
         response_200 = AgentHistory.from_dict(response.json())
 
@@ -44,9 +42,7 @@ def _parse_response(
         return None
 
 
-def _build_response(
-    *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Response[AgentHistory]:
+def _build_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Response[AgentHistory]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

beamlit/api/agents/update_agent.py

@@ -6,14 +6,13 @@ import httpx
 from ... import errors
 from ...client import AuthenticatedClient, Client
 from ...models.agent import Agent
-from ...models.agent_with_deployments import AgentWithDeployments
 from ...types import Response
 
 
 def _get_kwargs(
     agent_name: str,
     *,
-    body: AgentWithDeployments,
+    body: Agent,
 ) -> dict[str, Any]:
     headers: dict[str, Any] = {}
 
@@ -55,14 +54,13 @@ def sync_detailed(
     agent_name: str,
     *,
     client: AuthenticatedClient,
-    body: AgentWithDeployments,
+    body: Agent,
 ) -> Response[Agent]:
     """Update agent by name
 
     Args:
         agent_name (str):
-        body (AgentWithDeployments): Logical object representing an agent but with deployment
-            definition inside
+        body (Agent): Agent
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -88,14 +86,13 @@ def sync(
     agent_name: str,
     *,
     client: AuthenticatedClient,
-    body: AgentWithDeployments,
+    body: Agent,
 ) -> Optional[Agent]:
     """Update agent by name
 
     Args:
         agent_name (str):
-        body (AgentWithDeployments): Logical object representing an agent but with deployment
-            definition inside
+        body (Agent): Agent
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -116,14 +113,13 @@ async def asyncio_detailed(
     agent_name: str,
     *,
     client: AuthenticatedClient,
-    body: AgentWithDeployments,
+    body: Agent,
 ) -> Response[Agent]:
     """Update agent by name
 
     Args:
         agent_name (str):
-        body (AgentWithDeployments): Logical object representing an agent but with deployment
-            definition inside
+        body (Agent): Agent
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -147,14 +143,13 @@ async def asyncio(
     agent_name: str,
     *,
     client: AuthenticatedClient,
-    body: AgentWithDeployments,
+    body: Agent,
 ) -> Optional[Agent]:
     """Update agent by name
 
     Args:
         agent_name (str):
-        body (AgentWithDeployments): Logical object representing an agent but with deployment
-            definition inside
+        body (Agent): Agent
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.

beamlit/api/functions/create_function.py

@@ -6,13 +6,12 @@ import httpx
 from ... import errors
 from ...client import AuthenticatedClient, Client
 from ...models.function import Function
-from ...models.function_with_deployments import FunctionWithDeployments
 from ...types import Response
 
 
 def _get_kwargs(
     *,
-    body: FunctionWithDeployments,
+    body: Function,
 ) -> dict[str, Any]:
     headers: dict[str, Any] = {}
 
@@ -53,13 +52,12 @@ def _build_response(*, client: Union[AuthenticatedClient, Client], response: htt
 def sync_detailed(
     *,
     client: AuthenticatedClient,
-    body: FunctionWithDeployments,
+    body: Function,
 ) -> Response[Function]:
     """Create function
 
     Args:
-        body (FunctionWithDeployments): Logical object representing a function but with deployment
-            definition inside
+        body (Function): Function
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -83,13 +81,12 @@ def sync_detailed(
 def sync(
     *,
     client: AuthenticatedClient,
-    body: FunctionWithDeployments,
+    body: Function,
 ) -> Optional[Function]:
     """Create function
 
     Args:
-        body (FunctionWithDeployments): Logical object representing a function but with deployment
-            definition inside
+        body (Function): Function
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -108,13 +105,12 @@ def sync(
 async def asyncio_detailed(
     *,
     client: AuthenticatedClient,
-    body: FunctionWithDeployments,
+    body: Function,
 ) -> Response[Function]:
     """Create function
 
     Args:
-        body (FunctionWithDeployments): Logical object representing a function but with deployment
-            definition inside
+        body (Function): Function
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -136,13 +132,12 @@ async def asyncio_detailed(
 async def asyncio(
     *,
     client: AuthenticatedClient,
-    body: FunctionWithDeployments,
+    body: Function,
 ) -> Optional[Function]:
     """Create function
 
     Args:
-        body (FunctionWithDeployments): Logical object representing a function but with deployment
-            definition inside
+        body (Function): Function
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.