beamlit 0.0.47rc93__py3-none-any.whl → 0.0.48__py3-none-any.whl

beamlit/agents/decorator.py

@@ -11,7 +11,7 @@ from beamlit.authentication import new_client
 from beamlit.common.settings import init
 from beamlit.errors import UnexpectedStatus
 from beamlit.functions import get_functions
-from beamlit.models import Agent, AgentMetadata, AgentSpec
+from beamlit.models import Agent, AgentSpec, EnvironmentMetadata
 
 from .chat import get_chat_model_full
 
@@ -60,7 +60,7 @@ def agent(
             return wrapped
 
         if agent is not None:
-            metadata = AgentMetadata(**agent.get("metadata", {}))
+            metadata = EnvironmentMetadata(**agent.get("metadata", {}))
             spec = AgentSpec(**agent.get("spec", {}))
             agent = Agent(metadata=metadata, spec=spec)
             if agent.spec.model and chat_model is None:
@@ -141,7 +141,7 @@ def agent(
             except AttributeError: # special case for azure-marketplace where it uses the old OpenAI interface (no tools)
                 logger.warning("Using the old OpenAI interface for Azure Marketplace, no tools available")
                 _agent = create_react_agent(chat_model, [], checkpointer=memory)
-              
+
             settings.agent.agent = _agent
         else:
             settings.agent.agent = override_agent

beamlit/api/{model_providers/delete_model_provider.py → accounts/delete_account.py}

@@ -1,37 +1,44 @@
 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_provider import ModelProvider
+from ...models.account import Account
 from ...types import Response
 
 
 def _get_kwargs(
-    model_provider_name: str,
+    account_id: str,
 ) -> dict[str, Any]:
     _kwargs: dict[str, Any] = {
         "method": "delete",
-        "url": f"/model_providers/{model_provider_name}",
+        "url": f"/accounts/{account_id}",
     }
 
     return _kwargs
 
 
-def _parse_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Optional[ModelProvider]:
+def _parse_response(
+    *, client: Union[AuthenticatedClient, Client], response: httpx.Response
+) -> Optional[Union[Account, Any]]:
     if response.status_code == 200:
-        response_200 = ModelProvider.from_dict(response.json())
+        response_200 = Account.from_dict(response.json())
 
         return response_200
+    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:
         return None
 
 
-def _build_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Response[ModelProvider]:
+def _build_response(
+    *, client: Union[AuthenticatedClient, Client], response: httpx.Response
+) -> Response[Union[Account, Any]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -41,27 +48,27 @@ def _build_response(*, client: Union[AuthenticatedClient, Client], response: htt
 
 
 def sync_detailed(
-    model_provider_name: str,
+    account_id: str,
     *,
     client: AuthenticatedClient,
-) -> Response[ModelProvider]:
-    """Delete model provider
+) -> Response[Union[Account, Any]]:
+    """Delete account
 
-     Deletes an integration by ID.
+     Deletes an account by name.
 
     Args:
-        model_provider_name (str):
+        account_id (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[ModelProvider]
+        Response[Union[Account, Any]]
     """
 
     kwargs = _get_kwargs(
-        model_provider_name=model_provider_name,
+        account_id=account_id,
     )
 
     response = client.get_httpx_client().request(
@@ -72,53 +79,53 @@ def sync_detailed(
 
 
 def sync(
-    model_provider_name: str,
+    account_id: str,
     *,
     client: AuthenticatedClient,
-) -> Optional[ModelProvider]:
-    """Delete model provider
+) -> Optional[Union[Account, Any]]:
+    """Delete account
 
-     Deletes an integration by ID.
+     Deletes an account by name.
 
     Args:
-        model_provider_name (str):
+        account_id (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:
-        ModelProvider
+        Union[Account, Any]
     """
 
     return sync_detailed(
-        model_provider_name=model_provider_name,
+        account_id=account_id,
         client=client,
     ).parsed
 
 
 async def asyncio_detailed(
-    model_provider_name: str,
+    account_id: str,
     *,
     client: AuthenticatedClient,
-) -> Response[ModelProvider]:
-    """Delete model provider
+) -> Response[Union[Account, Any]]:
+    """Delete account
 
-     Deletes an integration by ID.
+     Deletes an account by name.
 
     Args:
-        model_provider_name (str):
+        account_id (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[ModelProvider]
+        Response[Union[Account, Any]]
     """
 
     kwargs = _get_kwargs(
-        model_provider_name=model_provider_name,
+        account_id=account_id,
     )
 
     response = await client.get_async_httpx_client().request(**kwargs)
@@ -127,28 +134,28 @@ async def asyncio_detailed(
 
 
 async def asyncio(
-    model_provider_name: str,
+    account_id: str,
     *,
     client: AuthenticatedClient,
-) -> Optional[ModelProvider]:
-    """Delete model provider
+) -> Optional[Union[Account, Any]]:
+    """Delete account
 
-     Deletes an integration by ID.
+     Deletes an account by name.
 
     Args:
-        model_provider_name (str):
+        account_id (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:
-        ModelProvider
+        Union[Account, Any]
     """
 
     return (
         await asyncio_detailed(
-            model_provider_name=model_provider_name,
+            account_id=account_id,
             client=client,
         )
     ).parsed

beamlit/api/{model_providers/get_model_provider.py → accounts/get_account.py}

@@ -5,24 +5,24 @@ import httpx
 
 from ... import errors
 from ...client import AuthenticatedClient, Client
-from ...models.model_provider import ModelProvider
+from ...models.account import Account
 from ...types import Response
 
 
 def _get_kwargs(
-    model_provider_name: str,
+    account_id: str,
 ) -> dict[str, Any]:
     _kwargs: dict[str, Any] = {
         "method": "get",
-        "url": f"/model_providers/{model_provider_name}",
+        "url": f"/accounts/{account_id}",
     }
 
     return _kwargs
 
 
-def _parse_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Optional[ModelProvider]:
+def _parse_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Optional[Account]:
     if response.status_code == 200:
-        response_200 = ModelProvider.from_dict(response.json())
+        response_200 = Account.from_dict(response.json())
 
         return response_200
     if client.raise_on_unexpected_status:
@@ -31,7 +31,7 @@ def _parse_response(*, client: Union[AuthenticatedClient, Client], response: htt
         return None
 
 
-def _build_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Response[ModelProvider]:
+def _build_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Response[Account]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -41,27 +41,27 @@ def _build_response(*, client: Union[AuthenticatedClient, Client], response: htt
 
 
 def sync_detailed(
-    model_provider_name: str,
+    account_id: str,
     *,
     client: AuthenticatedClient,
-) -> Response[ModelProvider]:
-    """Get model provider
+) -> Response[Account]:
+    """Get account by name
 
-     Returns an integration by ID.
+     Returns an account by name.
 
     Args:
-        model_provider_name (str):
+        account_id (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[ModelProvider]
+        Response[Account]
     """
 
     kwargs = _get_kwargs(
-        model_provider_name=model_provider_name,
+        account_id=account_id,
     )
 
     response = client.get_httpx_client().request(
@@ -72,53 +72,53 @@ def sync_detailed(
 
 
 def sync(
-    model_provider_name: str,
+    account_id: str,
     *,
     client: AuthenticatedClient,
-) -> Optional[ModelProvider]:
-    """Get model provider
+) -> Optional[Account]:
+    """Get account by name
 
-     Returns an integration by ID.
+     Returns an account by name.
 
     Args:
-        model_provider_name (str):
+        account_id (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:
-        ModelProvider
+        Account
     """
 
     return sync_detailed(
-        model_provider_name=model_provider_name,
+        account_id=account_id,
         client=client,
     ).parsed
 
 
 async def asyncio_detailed(
-    model_provider_name: str,
+    account_id: str,
     *,
     client: AuthenticatedClient,
-) -> Response[ModelProvider]:
-    """Get model provider
+) -> Response[Account]:
+    """Get account by name
 
-     Returns an integration by ID.
+     Returns an account by name.
 
     Args:
-        model_provider_name (str):
+        account_id (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[ModelProvider]
+        Response[Account]
     """
 
     kwargs = _get_kwargs(
-        model_provider_name=model_provider_name,
+        account_id=account_id,
     )
 
     response = await client.get_async_httpx_client().request(**kwargs)
@@ -127,28 +127,28 @@ async def asyncio_detailed(
 
 
 async def asyncio(
-    model_provider_name: str,
+    account_id: str,
     *,
     client: AuthenticatedClient,
-) -> Optional[ModelProvider]:
-    """Get model provider
+) -> Optional[Account]:
+    """Get account by name
 
-     Returns an integration by ID.
+     Returns an account by name.
 
     Args:
-        model_provider_name (str):
+        account_id (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:
-        ModelProvider
+        Account
     """
 
     return (
         await asyncio_detailed(
-            model_provider_name=model_provider_name,
+            account_id=account_id,
             client=client,
         )
     ).parsed

beamlit/api/{model_providers/list_model_providers.py → accounts/list_accounts.py}

@@ -5,14 +5,14 @@ import httpx
 
 from ... import errors
 from ...client import AuthenticatedClient, Client
-from ...models.model_provider import ModelProvider
+from ...models.account import Account
 from ...types import Response
 
 
 def _get_kwargs() -> dict[str, Any]:
     _kwargs: dict[str, Any] = {
         "method": "get",
-        "url": "/model_providers",
+        "url": "/accounts",
     }
 
     return _kwargs
@@ -20,12 +20,12 @@ def _get_kwargs() -> dict[str, Any]:
 
 def _parse_response(
     *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Optional[list["ModelProvider"]]:
+) -> Optional[list["Account"]]:
     if response.status_code == 200:
         response_200 = []
         _response_200 = response.json()
         for response_200_item_data in _response_200:
-            response_200_item = ModelProvider.from_dict(response_200_item_data)
+            response_200_item = Account.from_dict(response_200_item_data)
 
             response_200.append(response_200_item)
 
@@ -38,7 +38,7 @@ def _parse_response(
 
 def _build_response(
     *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Response[list["ModelProvider"]]:
+) -> Response[list["Account"]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -50,17 +50,17 @@ def _build_response(
 def sync_detailed(
     *,
     client: AuthenticatedClient,
-) -> Response[list["ModelProvider"]]:
-    """List model providers
+) -> Response[list["Account"]]:
+    """List accounts
 
-     Returns a list of all integrations in the workspace.
+     Returns a list of all accounts.
 
     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['ModelProvider']]
+        Response[list['Account']]
     """
 
     kwargs = _get_kwargs()
@@ -75,17 +75,17 @@ def sync_detailed(
 def sync(
     *,
     client: AuthenticatedClient,
-) -> Optional[list["ModelProvider"]]:
-    """List model providers
+) -> Optional[list["Account"]]:
+    """List accounts
 
-     Returns a list of all integrations in the workspace.
+     Returns a list of all accounts.
 
     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['ModelProvider']
+        list['Account']
     """
 
     return sync_detailed(
@@ -96,17 +96,17 @@ def sync(
 async def asyncio_detailed(
     *,
     client: AuthenticatedClient,
-) -> Response[list["ModelProvider"]]:
-    """List model providers
+) -> Response[list["Account"]]:
+    """List accounts
 
-     Returns a list of all integrations in the workspace.
+     Returns a list of all accounts.
 
     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['ModelProvider']]
+        Response[list['Account']]
     """
 
     kwargs = _get_kwargs()
@@ -119,17 +119,17 @@ async def asyncio_detailed(
 async def asyncio(
     *,
     client: AuthenticatedClient,
-) -> Optional[list["ModelProvider"]]:
-    """List model providers
+) -> Optional[list["Account"]]:
+    """List accounts
 
-     Returns a list of all integrations in the workspace.
+     Returns a list of all accounts.
 
     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['ModelProvider']
+        list['Account']
     """
 
     return (